kget/core/transfertreemodel.h

   1 /* This file is part of the KDE project
   2
   3    Copyright (C) 2006 Dario Massarin <nekkar@libero.it>
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public
   7    License as published by the Free Software Foundation; either
   8    version 2 of the License, or (at your option) any later version.
   9 */
  10
  11 #ifndef _TRANSFERTREEMODEL_H
  12 #define _TRANSFERTREEMODEL_H
  13
  14 #include <QAbstractItemModel>
  15 #include <QList>
  16
  17 #include <kget_export.h>
  18
  19 class KUrl;
  20
  21 class TransferGroupHandler;
  22 class TransferGroup;
  23 class TransferHandler;
  24 class Transfer;
  25 class Scheduler;
  26
  27 class TransferTreeModel : public QAbstractItemModel
  28 {
  29     Q_OBJECT
  30
  31     friend class TransferGroupHandler;
  32     friend class TransferGroup;
  33     friend class TransferHandler;
  34     friend class Transfer;
  35
  36     public:
  37         TransferTreeModel(Scheduler * scheduler);
  38         ~TransferTreeModel();
  39
  40         void addGroup(TransferGroup * group);
  41         void delGroup(TransferGroup * group);
  42
  43         void addTransfer(Transfer * transfer, TransferGroup * group);
  44         void delTransfer(Transfer * transfer);
  45
  46         void moveTransfer(Transfer * transfer, TransferGroup * destGroup, Transfer * after=0);
  47
  48         const QList<TransferGroup *> & transferGroups();
  49
  50         TransferGroup * findGroup(const QString & groupName);
  51         Transfer * findTransfer(const KUrl &src);
  52         Transfer *findTransferByDestination(const KUrl &dest);
  53
  54         bool KGET_EXPORT isTransferGroup(const QModelIndex & index) const;
  55
  56         void postDataChangedEvent(TransferHandler * transfer);
  57         void postDataChangedEvent(TransferGroupHandler * group);
  58
  59         QModelIndex createIndex(int row, int column, void * ptr = 0) const;
  60
  61         //QAbstractItemModel functions
  62         int rowCount(const QModelIndex & parent) const;
  63         int columnCount(const QModelIndex & parent) const;
  64         Qt::ItemFlags flags (const QModelIndex & index) const;
  65         QVariant headerData(int section, Qt::Orientation orientation,
  66                             int role = Qt::DisplayRole) const;
  67         QVariant data (const QModelIndex & index, int role) const;
  68         QModelIndex index(int row, int column, const QModelIndex & parent) const;
  69         QModelIndex parent(const QModelIndex & index) const;
  70
  71         //Drag & drop functions
  72         Qt::DropActions supportedDropActions() const;
  73         QStringList mimeTypes() const;
  74         QMimeData * mimeData(const QModelIndexList &indexes) const;
  75         bool dropMimeData(const QMimeData *mdata,
  76                           Qt::DropAction action, int row, int column,
  77                           const QModelIndex &parent);
  78
  79
  80     private:
  81         void timerEvent(QTimerEvent *event);
  82
  83         QList<TransferGroup *> m_transferGroups;
  84         Scheduler * m_scheduler;
  85
  86         // Timer related variables
  87         QList<TransferHandler *> m_changedTransfers;
  88         QList<TransferGroupHandler *> m_changedGroups;
  89
  90         int m_timerId;
  91 };
  92
  93 #endif