kwicmodels.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 "kwicmodels.h"

#include "phrasehunter/contextreader.h"
#include "phrasehunter/token.h"
#include "phrasehunter/tokencontext.h"

#include "support/unicodehelpers.h"

#include <boost/lexical_cast.hpp>

#include "qt_utils.h"

using namespace PhraseHunter;

void addAllOccurrences(TokenPtr token, 
                       std::vector<TokenContextPtr>& entries) 
{
    const OccurrenceMap& all_occurrences = token->allOccurrences();
    for(OccurrenceMap::const_iterator it = all_occurrences.begin();
        it != all_occurrences.end(); ++it) {            
      
        for(PositionList::const_iterator it_inner = it->second.begin();
            it_inner != it->second.end(); ++it_inner) {
            entries.push_back(TokenContext::emptyContext(it->first, *it_inner, token->length()));
        }
    }
}

KwicTableModel::KwicTableModel(QObject* parent, ContextReader* reader):
    QAbstractTableModel(parent), m_reader(reader), m_width(40)
{
}

QVariant KwicTableModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if(orientation != Qt::Horizontal || role != Qt::DisplayRole) {
        return QVariant();
    }

    switch(section){
    case 0:
        return QVariant("");
    case 1:
        return QVariant("Left Context");
    case 2:
        return QVariant("Token");
    case 3:
        return QVariant("Right Context");
    default:
        return QVariant();
    }
}

    
int KwicTableModel::columnCount(const QModelIndex& parent) const
{
    return 4;
}

int KwicTableModel::rowCount(const QModelIndex& parent) const
{
    return entries.size();
}

QVariant KwicTableModel::getContextString(TokenContextPtr context, const QModelIndex& index) const
{
    if(context->empty()) {
        m_reader->fillContext(context, m_width);
    }
    
    schma::UnicodePtr unistr;

    switch(index.column()) {
    case 0:
        return QVariant(QString::number(index.row()+1));
    case 1:
        unistr = context->leftContext();
        break;
    case 2:
        unistr = context->token();
        break;
    case 3:
        unistr = context->rightContext();
        break;
    }
    return QVariant(icuToQString(unistr));
}

QVariant KwicTableModel::data(const QModelIndex& index, int role) const
{
    return (index.isValid() && role == Qt::DisplayRole) 
        ? getContextString(entries[index.row()], index)
        : QVariant();
}

void KwicTableModel::setData(TokenPtr token) 
{
    entries.clear();
    addAllOccurrences(token, entries);
    reset();
}

void KwicTableModel::setData(const TokenVector& tokens) 
{
    entries.clear();
    for(TokenVector::const_iterator it = tokens.begin();
        it != tokens.end(); ++it) {
        addAllOccurrences(*it, entries);
    }
    reset();
}

---