SearchModel.h

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

*/

#ifndef SEARCHMODEL_H
#define SEARCHMODEL_H

#include "Gnutella/Packets/QueryHits.h"

using Gnutella::Packets::QueryHits;

namespace UIs {
namespace Searching {

struct SearchResult {
    SearchResult(const QString &fn, const QString &ext, const quint32 &sz, quint32 sp, QString ht, QString clt)
         : fileName(fn), extension(ext), speed(QString()), size (0),
         rating("Good"), status("Available"), host(ht),
         client(clt), duration("0:53:24"),
         bitrate("8 bps"), /*parent(NULL), */hosts(), showEmphasized (false)
    {
        float fz = sz;      // B
        int i = 0;

        while (fz > 1024) {
            fz /= 1024;
            i++;
        }

        switch (i) {
        case 0:
            size = QString().sprintf ("%.2f B", fz);
            break;
        case 1:
            size = QString().sprintf ("%.2f KB", fz);
            break;
        case 2:
            size = QString().sprintf ("%.2f MB", fz);
            break;
        default:
            size = QString().sprintf ("%.2f GB", fz);
            break;
        }

        if (sp) {
            speed = QString().sprintf ("%d kbps", sp);
        }
    };

    QString         fileName;
    QString         extension;
    QString         speed;
    QString         size;
    QString         rating;
    QString         status;
    QString         host;
    QString         client;
    QString         duration;
    QString         bitrate;
    //mutable SearchResult *parent; // Can we get rid of that?
    QList<SearchResult>  hosts;
    bool            showEmphasized;
};

class SearchModel : public QAbstractTableModel
{
    Q_OBJECT
    REFERENCE_OBJECT (SearchModel)

public:
    SearchModel(QObject *parent);

    int rowCount(const QModelIndex &parent = QModelIndex()) const;
    int columnCount(const QModelIndex &parent = QModelIndex()) const;
    QVariant data(const QModelIndex &index, int role) const;
    QVariant headerData(int section, Qt::Orientation orientation,
                        int role = Qt::DisplayRole) const;
    bool hasChildren (const QModelIndex &index) const;
    QModelIndex index(int row, int column, const QModelIndex &parent) const;
    //QModelIndex parent ( const QModelIndex & index ) const;
    void tabActivated();
    void tabDisactivated();

private:
    int findIndex (SearchResult *result);

private:
    class IconProvider
    {
    public:
        IconProvider();
        ~IconProvider() {}

        QIcon icon (QString fileExtension) const;

    private:
        enum FileTypes {video, audio, compressed, photo, text, unknown};

        QList<QIcon>    iconList;
        QList<QString>  videoFileExtensions;
        QList<QString>  audioFileExtensions;
        QList<QString>  compressedFileExtensions;
        QList<QString>  photoFileExtensions;
        QList<QString>  textFileExtensions;
    };

private:
//  enum SearchModelField {fileName, extension, size, rating, status, hosts, speed, client, duration, bitrate};
    enum SearchModelField {fileName, extension, size, hosts, speed, client};

    QList<SearchResult> resultList;
    IconProvider iconProvider;
    int cCount;
    bool tabActive;

public:
    void addItem(SearchResult *result);
    void addHits(const QueryHits &quieryHits);

signals:
    void itemAdded();

};

} // namespace Searching
} // namespace UIs

#endif  // SEARCHMODEL_H

Copyright © 2006 Peter Dimov.
Powered by Doxygen.