phsearchimpl.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 <string>
#include <iostream>

#include <QtGui/QFileDialog>
#include <QtGui/QMessageBox>
#include <QtGui/QTextBrowser>
#include <QtGui/QToolButton>

#include "phrasehunter/corpus.h"
#include "phrasehunter/phexception.h"
#include "phrasehunter/contextreader.h"

#include "phsearchimpl.h"
#include "searchtab.h"
#include "textbrowsertab.h"


PhSearchImpl::PhSearchImpl()
    : m_corpus(NULL)
{
    setupUi(this);
    setupTabIcons();
    setupConnections();
    
    tabWidget->removeTab(0);
    newSearchTab();
    hideStatistics();

    //similarity functionality does not work, yet:
    searchKindComboBox->removeItem(2);
}

void PhSearchImpl::setupConnections() 
{
    connect(action_Open, SIGNAL(triggered()), this, SLOT(fileOpen()));
    connect(action_Exit, SIGNAL(triggered()), this, SLOT(fileExit()));
    connect(actionSearch_Options, SIGNAL(triggered()), this, SLOT(openOptionsWidget()));
    connect(actionNew_Search_Tab, SIGNAL(triggered()), this, SLOT(newSearchTab()));
    connect(searchKindComboBox, SIGNAL(currentIndexChanged(const QString& )), this, SLOT(hideStatistics()));
}

PhSearchImpl::~PhSearchImpl()
{
    deleteCorpus();
}

void PhSearchImpl::deleteCorpus()
{
    delete m_corpus;
    m_corpus = NULL;
}

void PhSearchImpl::fileOpen()
{
    try {
        deleteCorpus();
        QString corpusDir = 
            QFileDialog::getExistingDirectory(this,
                                              QString("Choose a Corpus"), 
                                              QDir::homePath(),
                                              QFileDialog::ShowDirsOnly);
        if(corpusDir.isEmpty())
            return;
        
        m_corpus = new PhraseHunter::CorpusManager(corpusDir.toStdString());
        m_corpus->statisticsEngine()->setNumberContextWords(40);

        QList<SearchTab*> searchLines = tabWidget->findChildren<SearchTab*>();
        for (int i = 0; i < searchLines.size(); ++i) {
            searchLines.at(i)->enableSearch(m_corpus);
        }
    }
    catch(PhraseHunter::Exceptions::NoSuchCorpusError& _e) {
        QMessageBox::critical(this, QString("Corpus Error"), QString("This directory does not contain a valid corpus: ")+QString(_e.what()),QMessageBox::Ok,QMessageBox::NoButton,QMessageBox::NoButton);
        fileOpen();
    }
    catch (PhraseHunter::Exceptions::Exception& _e) {
        QMessageBox::critical(this, QString("Error"), QString(_e.what()),QMessageBox::Ok,QMessageBox::NoButton,QMessageBox::NoButton );
    }
}

void PhSearchImpl::fileExit()
{
    close();
}

void PhSearchImpl::openOptionsWidget()
{
    optionsDockWidget->setVisible(TRUE);
}

void PhSearchImpl::newSearchTab()
{
    SearchTab* newTab = new SearchTab(this);
    tabWidget->addTab(newTab, QApplication::translate("phMainWindow", "Query", 0,
                                                      QApplication::UnicodeUTF8));
    tabWidget->setCurrentWidget(newTab);
    if(m_corpus != NULL) {
        newTab->enableSearch(m_corpus);
    }
    
    newTab->setQueryType(searchKindComboBox->currentText());
    
    connect(searchKindComboBox, SIGNAL(currentIndexChanged(const QString&)),
            newTab, SLOT(setQueryType(const QString&)));
    
    newTab->setContextWidth(contextWidthSpinBox->value());
    
    connect(contextWidthSpinBox, SIGNAL(valueChanged(int)),
            newTab, SLOT(setContextWidth(int)));

    connect(newTab, SIGNAL(documentRequested(PhraseHunter::DocID, PhraseHunter::IdxPos, size_t)),
            this, SLOT(openDocument(PhraseHunter::DocID, PhraseHunter::IdxPos, size_t)));

}

void PhSearchImpl::setupTabIcons() 
{
    QToolButton *closeTabButton = new QToolButton(tabWidget);
    tabWidget->setCornerWidget(closeTabButton, Qt::TopRightCorner);
    closeTabButton->setAutoRaise(true);
    closeTabButton->setIcon(QIcon(":/icons/closetab.png"));
    closeTabButton->setToolTip(tr("Close page"));
    closeTabButton->setEnabled(true);

    QToolButton *newTabButton = new QToolButton(tabWidget);
    tabWidget->setCornerWidget(newTabButton, Qt::TopLeftCorner);
    newTabButton->setAutoRaise(true);
    newTabButton->setIcon(QIcon(":/icons/addtab.png"));
    newTabButton->setToolTip(tr("New page"));
    newTabButton->setEnabled(true);
    
    connect(closeTabButton, SIGNAL(clicked()), this, SLOT(closeTab()));
    connect(newTabButton, SIGNAL(clicked()), this, SLOT(newSearchTab()));
}

    
void PhSearchImpl::hideStatistics()
{
    sortByRankCheckBox->setHidden(true);
//     if(searchKindComboBox->currentText() == "Regex") {
//         sortByRankCheckBox->setVisible(true);
//     } else {
//         sortByRankCheckBox->setHidden(true);
//     }
}

void PhSearchImpl::closeTab()
{
    if(tabWidget->count() > 1)
        tabWidget->removeTab(tabWidget->currentIndex());
}

void PhSearchImpl::helpAbout()
{
    qWarning( "PhSearchImpl::helpAbout() not yet implemented!" );
}

void PhSearchImpl::openDocument(PhraseHunter::DocID docID, PhraseHunter::IdxPos position, size_t tokenLength) 
{
    PhraseHunter::TokenContextPtr output = 
        m_corpus->contextReader()->getContextFromPosition(docID, position, tokenLength, sourceContextSpinBox->value());
    
    TextBrowserTab* textBrowserTab = new TextBrowserTab(this, output);
    
    tabWidget->addTab(textBrowserTab, QString("<S>") + 
                      tabWidget->currentWidget()->findChild<QLineEdit*>("leQuery")->text());
    
    tabWidget->setCurrentWidget(textBrowserTab);
}

int main(int argc, char* argv[]) 
{
    try {
        QApplication a( argc, argv );
        PhSearchImpl searcher;
        searcher.show();        
        return a.exec();
    }
    catch (PhraseHunter::Exceptions::Exception& _e) {
        QMessageBox::critical(NULL, QString("Error"), QString(_e.what()),QMessageBox::Ok,QMessageBox::NoButton,QMessageBox::NoButton );
    }
    catch (...) {
        QMessageBox::critical(NULL, QString("error"),
                              QString("An unexpected error occurred"), QMessageBox::Ok,QMessageBox::NoButton,QMessageBox::NoButton);
    }
}

---