searchtab.cpp

Go to the documentation of this file.

/*
  Phrasehunter - index and query text corpora
  Copyright (C) 2006  Torsten Marek (shlomme@gmx.de) &
  Armin Schmidt (armin.sch@gmail.com)
  
  This program is free software; you can redistribute it and/or
  modify it under the terms of the GNU General Public License
  as published by the Free Software Foundation; either version 2
  of the License, or (at your option) any later version.
  
  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.
  
  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
*/

#include <QtGui/QMessageBox>
#include <QtGui/QSortFilterProxyModel>

#include "phrasehunter/corpus.h"
#include "phrasehunter/searchengine.h"
#include "phrasehunter/statistics.h"
#include "phrasehunter/tokencontext.h"
#include "phrasehunter/token.h"
#include "support/unicodehelpers.h"

#include "searchtab.h"

SearchTab::SearchTab(QWidget* parent): 
    QWidget(parent),
    m_searchEngine(NULL),
    m_stats(NULL),
    m_model(NULL)
{
    setupUi(this);
    
    connect(pbSearch, SIGNAL(clicked(bool)), this, SLOT(search()));
    connect(leQuery, SIGNAL(returnPressed()), this, SLOT(search()));
    
    connect(resultsTableView, SIGNAL(doubleClicked(const QModelIndex&)), 
            this, SLOT(emitDocumentRequested(const QModelIndex&)));
}

void SearchTab::emitDocumentRequested(const QModelIndex& index) 
{
    PhraseHunter::TokenContextPtr context = m_model->getContext(index);
    emit documentRequested(context->docID(), context->position(), context->tokenLength());
}

void SearchTab::setContextWidth(int newContextWidth)
{
    m_contextWidth = newContextWidth;
    if(m_model != NULL) {
        m_model->setContextWidth(newContextWidth);
    }
}

void SearchTab::setQueryType(const QString& type) 
{
    queryType = type;
}

void SearchTab::enableSearch(PhraseHunter::CorpusManager* corpus) 
{
    m_searchEngine = corpus->searchEngine();
    
    m_stats = corpus->statisticsEngine();
    
    numberOfDocs = m_stats->getNumberOfDocuments();
    numberOfTokens = m_stats->getSizeOfSampleSpace();
    numberOfTypes = m_stats->getNumberOfTypes();
    
    
    resultsTableView->setModel(NULL);
    delete m_model;
    m_model = new KwicTableModel(this, corpus->contextReader());
    m_model->setContextWidth(m_contextWidth);
    resultsTableView->setModel(m_model);

    /////////////////////////////////////////////////////////////////////////////
    //This shouldn't be neccessary if sorting had been implemented for QTableView
    QSortFilterProxyModel *md = new QSortFilterProxyModel(this);
    md->setSourceModel(m_model);
    resultsTableView->setModel(md);
    /////////////////////////////////////////////////////////////////////////////

    leQuery->setText("");
    leQuery->setEnabled(true);
    pbSearch->setEnabled(true);
}

void SearchTab::search() 
{
    try {
        hideWidgets();
        
        if(m_searchEngine == NULL) {
            QMessageBox::critical(this, QString("Error"), 
                                  QString("No or invalid corpus loaded"), 
                                  QMessageBox::Ok, QMessageBox::NoButton, QMessageBox::NoButton);
            return;
        }

        unsigned int docFreq = 0;
        unsigned int freq = 0;
        int rank = 0;
      
        QString query = leQuery->text().toLower();

        if(queryType == "Regex") {
            std::vector<PhraseHunter::TokenPtr> resultTokens;
          
            if (query.contains(' ')) {
                resultTokens = m_searchEngine->searchPhrasalRegex(schma::UnicodePtr(new UnicodeString(query.toUtf8().data())));
            } else {
                resultTokens = m_searchEngine->searchRegexToken(schma::UnicodePtr(new UnicodeString(query.toUtf8().data())));
            }
            
            for(PhraseHunter::TokenVector::const_iterator i = resultTokens.begin();
                i != resultTokens.end(); ++i) {
                freq += (*i)->corpusFrequency();
            }
            docFreq = PhraseHunter::StatisticsEngine::sizeOfDocSet(resultTokens);
            
            // TODO: sort by rank
            m_model->setData(resultTokens);
            
        } else if(queryType == "Phrase") {

            PhraseHunter::TokenPtr queryToken;
            
            if (query.contains(' ')) {
                queryToken = m_searchEngine->searchPhrase(schma::UnicodePtr(new UnicodeString(query.toUtf8().data())));
            } else {
                queryToken = m_searchEngine->searchToken(schma::UnicodePtr(new UnicodeString(query.toUtf8().data())));
            }
            m_model->setData(queryToken);
            
            docFreq = queryToken->documentFrequency();
            rank = m_stats->rank(queryToken);
            freq = queryToken->corpusFrequency();
        }
        
        setStatistics(docFreq, freq, rank);
        resultsTableView->resizeColumnsToContents();
      
    } catch (PhraseHunter::Exceptions::Exception& _e) {
        QMessageBox::critical(this, QString("Error"),
                              QString("Error while processing search"),
                              QMessageBox::Ok,QMessageBox::NoButton,QMessageBox::NoButton);
    }
}

void SearchTab::hideWidgets()
{
    if(queryType == "Regex") {
        lblRank->setHidden(true);
        label_5->setHidden(true);
        //show 'Sort by rank' checkbox
    } else if (leQuery->text().trimmed().contains(' ')) {
        lblRank->setHidden(true);
        label_5->setHidden(true);
    } else {
        lblRank->setHidden(false);
        label_5->setHidden(false);
    }
}

void SearchTab::setStatistics(int docFreq, int freq, int rank)
{
    lblDocFrequency->setText(QString("%1 of %2").arg(docFreq).arg(numberOfDocs));
    
    if (leQuery->text().contains(' '))
        lblCorpusFrequency->setText(QString("%1").arg(freq));
    else
        lblCorpusFrequency->setText(QString("%1 of %2").arg(freq).arg(numberOfTokens));
    
    lblRank->setText(QString("%1 of %2").arg(rank).arg(numberOfTypes));
}

---