SearchModel.cpp

Go to the documentation of this file.

/*

Copyright (C) 2005-2007 by Peter Dimov.

This file is part of Calitko (http://www.calitko.org).

Calitko 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.

Calitko 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 Calitko; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/

#include "Qt.h"
#include "SearchModel.h"
#include "Gnutella/Packets/QueryHits.h"

using UIs::Searching::SearchModel;
using UIs::Searching::SearchResult;
using Gnutella::Packets::QueryHits;

SearchModel::IconProvider::IconProvider()
 :  iconList(), videoFileExtensions(), audioFileExtensions(),
    compressedFileExtensions(), photoFileExtensions(), textFileExtensions()
{
        videoFileExtensions.append ("avi");
        videoFileExtensions.append ("mpeg");
        videoFileExtensions.append ("mpg");
        videoFileExtensions.append ("wmv");
        videoFileExtensions.append ("mov");
        audioFileExtensions.append ("mp3");
        audioFileExtensions.append ("mp4");
        audioFileExtensions.append ("wave");
        audioFileExtensions.append ("midi");
        audioFileExtensions.append ("wma");
        compressedFileExtensions.append ("zip");
        compressedFileExtensions.append ("rar");
        compressedFileExtensions.append ("gz");
        compressedFileExtensions.append ("tar.gz");
        photoFileExtensions.append ("jpeg");
        photoFileExtensions.append ("jpg");
        photoFileExtensions.append ("png");
        photoFileExtensions.append ("bmp");

        textFileExtensions.append ("txt");
        textFileExtensions.append ("rtf");
        textFileExtensions.append ("doc");
        textFileExtensions.append ("pdf");

        iconList.append (QIcon ("Icons/video.png"));
        iconList.append (QIcon ("Icons/audio.png"));
        iconList.append (QIcon ("Icons/compressed.png"));
        iconList.append (QIcon ("Icons/photo.png"));
        iconList.append (QIcon ("Icons/text.png"));
        iconList.append (QIcon ("Icons/unknown.png"));
}

QIcon SearchModel::IconProvider::icon (QString fileExtension) const
{
    FileTypes fileType = unknown;

    if (videoFileExtensions.contains (fileExtension)) {
        fileType = video;
    } else if (audioFileExtensions.contains (fileExtension)) {
        fileType = audio;
    } else if (compressedFileExtensions.contains (fileExtension)) {
        fileType = compressed;
    } else if (photoFileExtensions.contains (fileExtension)) {
        fileType = photo;
    } else if (textFileExtensions.contains (fileExtension)) {
        fileType = text;
    } else {
        fileType = unknown;
    }

    return iconList.at (fileType);

}

SearchModel::SearchModel (QObject *parent)
 :  QAbstractTableModel (parent), resultList(), iconProvider(),
    cCount (6), tabActive (true)
{
/*  SearchResult res("file1", "txt", 1343, 234, "123.34", "LimeWire");
    SearchResult res1("file2", "jpg", 1343, 234, "123.34", "LimeWire");
    SearchResult res2("file3", "zip", 1343, 234, "123.34", "LimeWire");
    SearchResult res3("file4", "avi", 1343, 234, "123.34", "LimeWire");
    SearchResult res4("file5", "mp3", 1343, 234, "123.34", "LimeWire");
    SearchResult res5("file6", "abc", 1343, 234, "123.34", "LimeWire");

    addItem (&res);
    addItem (&res1);
    addItem (&res2);
    addItem (&res3);
    addItem (&res4);
    addItem (&res5);
*/}

void SearchModel::addHits(const QueryHits &)
{
}

void SearchModel::addItem(SearchResult *result)
{
    int count = resultList.count();
    beginInsertRows (QModelIndex(), count, count);

    result->showEmphasized = !tabActive;

    int index = findIndex (result);

    resultList.insert (index, *result);
/*  for (int i=0; i<resultList.at(index).hosts.count(); i++) {
        resultList.at(index).hosts.at(i).parent = (SearchResult*) &resultList.at(index);
    }*/

    endInsertRows();

    emit (itemAdded());

    return;
}

int SearchModel::rowCount (const QModelIndex &index) const
{
    if (!index.isValid()) {
        return resultList.count();
    }

    SearchResult *result = static_cast<SearchResult*> (index.internalPointer());

    return result->hosts.count();
}

int SearchModel::columnCount (const QModelIndex &) const
{
    return cCount;
}

QVariant SearchModel::data (const QModelIndex &index, int role) const
{
    static QIcon icon ("glmdiggp.png");

    if (!index.isValid())
        return QVariant();

    SearchResult* result = static_cast <SearchResult*> (index.internalPointer());
    Q_ASSERT (result);

    if (role == Qt::DisplayRole) {
        SearchModelField field = static_cast <SearchModelField> (index.column());
        switch (field) {
        case fileName:
            return result->fileName;
        case extension:
            return result->extension;
        case size:
            return result->size;
//      case rating:
//          return result->rating;
//      case status:
//          return result->status;
        case hosts:
            return result->host;
        case speed:
            return result->speed;
        case client:
            return result->client;
//      case duration:
//          return result->duration;
//      case bitrate:
//          return result->bitrate;
        default:
            return QVariant();
        }
    }

    if (role == Qt::DecorationRole && index.column() == 0)
        return iconProvider.icon (result->extension.toLower());

    if (role == Qt::TextColorRole)
        if (result->showEmphasized)
        return QColor (0, 0, 128);

    return QVariant();

}

bool SearchModel::hasChildren (const QModelIndex &index) const
{
    static SearchResult *result;

    if(index.isValid()) {
        result = static_cast <SearchResult*> (index.internalPointer());
        return (result->hosts.count() > 0);
    }

    return (resultList.count() > 0);
}
QModelIndex SearchModel::index(int row, int column, const QModelIndex &parent) const
{
    static SearchResult *result;
    //static SearchResult* host;

    if (!parent.isValid()) {
        if (row >= resultList.count())
            return QModelIndex();

        result = const_cast <SearchResult *> (&resultList.at (row));
        return createIndex (row, column, result);
    } else {
        Q_ASSERT (false);
/*      result = static_cast <SearchResult*> (parent.internalPointer());
        Q_ASSERT(result);
        if (row < result->hosts.count()) {
            host = (SearchResult*) &result->hosts.at (row);
            return createIndex (row, column, (void*) host);
        }*/
    }

    return QModelIndex();
}
/*
QModelIndex SearchModel::parent ( const QModelIndex & index ) const
{
    static SearchResult * host;

    if (index.isValid()) {
        host = static_cast<SearchResult*> (index.internalPointer());
//          qDebug() << child->fileName << "parent child.fileName";
        if (host->parent) {
            return createIndex (0, 0, host->parent);
        }
    }

    return QModelIndex();
}*/

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

    if (orientation == Qt::Horizontal) {
        SearchModelField field = static_cast <SearchModelField> (section);
        switch (field) {
        case fileName:
            return QString ("File");
        case extension:
            return QString ("Extension");
        case size:
            return QString ("Size");
//      case rating:
//          return QString ("Rating");
//      case staus:
//          return QString ("Status");
        case hosts:
            return QString ("Host(s)");
        case speed:
            return QString ("Speed");
        case client:
            return QString ("Client");
//      case duration:
//          return QString ("Duration");
//      case bitrate:
//          return QString ("Bitrate");
        default:
            return QString ("Column %1").arg (section);
        }
    }

    return QString ("");

}

void SearchModel::tabActivated()
{
    tabActive = true;
}

void SearchModel::tabDisactivated()
{
    tabActive = false;
    if (resultList.isEmpty()) return;

    QList<SearchResult>::iterator it = resultList.begin();
    do {
        it->showEmphasized = false;
        it++;
    } while (it != resultList.end());

    dataChanged (index (0, 0, QModelIndex()), index (resultList.count()-1, resultList.count()-1, QModelIndex()));
}

int SearchModel::findIndex (SearchResult *result)
{
    int i = 1;
    QString fileName = result->fileName;

    if (resultList.isEmpty ()) return 0;

    SearchResult res = resultList.first();

    while ((fileName > res.fileName) && (i < resultList.count())) {
        res = resultList.at (i++);
    }

    return i-1;
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.