src/GUI/AircraftModel.hxx

   1 // AircraftModel.hxx - part of GUI launcher using Qt5
   2 //
   3 // Written by James Turner, started March 2015.
   4 //
   5 // Copyright (C) 2015 James Turner <zakalawe@mac.com>
   6 //
   7 // This program is free software; you can redistribute it and/or
   8 // modify it under the terms of the GNU General Public License as
   9 // published by the Free Software Foundation; either version 2 of the
  10 // License, or (at your option) any later version.
  11 //
  12 // This program is distributed in the hope that it will be useful, but
  13 // WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 // General Public License for more details.
  16 //
  17 // You should have received a copy of the GNU General Public License
  18 // along with this program; if not, write to the Free Software
  19 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  20
  21 #ifndef FG_GUI_AIRCRAFT_MODEL
  22 #define FG_GUI_AIRCRAFT_MODEL
  23
  24 #include <QAbstractListModel>
  25 #include <QDateTime>
  26 #include <QDir>
  27 #include <QPixmap>
  28 #include <QStringList>
  29 #include <QSharedPointer>
  30 #include <QUrl>
  31
  32 #include <simgear/package/Root.hxx>
  33
  34 const int AircraftPathRole = Qt::UserRole + 1;
  35 const int AircraftAuthorsRole = Qt::UserRole + 2;
  36 const int AircraftVariantRole = Qt::UserRole + 3;
  37 const int AircraftVariantCountRole = Qt::UserRole + 4;
  38 const int AircraftThumbnailCountRole = Qt::UserRole + 5;
  39 const int AircraftPackageIdRole = Qt::UserRole + 6;
  40 const int AircraftPackageStatusRole = Qt::UserRole + 7;
  41 const int AircraftPackageProgressRole = Qt::UserRole + 8;
  42 const int AircraftLongDescriptionRole = Qt::UserRole + 9;
  43 const int AircraftHasRatingsRole = Qt::UserRole + 10;
  44 const int AircraftInstallPercentRole = Qt::UserRole + 11;
  45 const int AircraftPackageSizeRole = Qt::UserRole + 12;
  46 const int AircraftInstallDownloadedSizeRole = Qt::UserRole + 13;
  47 const int AircraftURIRole = Qt::UserRole + 14;
  48 const int AircraftThumbnailSizeRole = Qt::UserRole + 15;
  49 const int AircraftIsHelicopterRole = Qt::UserRole + 16;
  50 const int AircraftIsSeaplaneRole = Qt::UserRole + 17;
  51
  52 const int AircraftRatingRole = Qt::UserRole + 100;
  53 const int AircraftVariantDescriptionRole = Qt::UserRole + 200;
  54 const int AircraftThumbnailRole = Qt::UserRole + 300;
  55
  56 class AircraftScanThread;
  57 class QDataStream;
  58 class PackageDelegate;
  59 struct AircraftItem;
  60 typedef QSharedPointer<AircraftItem> AircraftItemPtr;
  61
  62 struct AircraftItem
  63 {
  64     AircraftItem();
  65
  66     AircraftItem(QDir dir, QString filePath);
  67
  68     // the file-name without -set.xml suffix
  69     QString baseName() const;
  70
  71     void fromDataStream(QDataStream& ds);
  72
  73     void toDataStream(QDataStream& ds) const;
  74
  75     QPixmap thumbnail() const;
  76
  77     bool excluded;
  78     QString path;
  79     QString description;
  80     QString authors;
  81     int ratings[4];
  82     QString variantOf;
  83     QDateTime pathModTime;
  84     QList<AircraftItemPtr> variants;
  85     bool usesHeliports;
  86     bool usesSeaports;
  87 private:
  88     mutable QPixmap m_thumbnail;
  89 };
  90
  91
  92 enum AircraftItemStatus {
  93     PackageNotInstalled = 0,
  94     PackageInstalled,
  95     PackageUpdateAvailable,
  96     PackageQueued,
  97     PackageDownloading
  98 };
  99
 100 class AircraftItemModel : public QAbstractListModel
 101 {
 102     Q_OBJECT
 103 public:
 104     AircraftItemModel(QObject* pr, const simgear::pkg::RootRef& root);
 105
 106     ~AircraftItemModel();
 107
 108     void setPaths(QStringList paths);
 109
 110     void scanDirs();
 111
 112     virtual int rowCount(const QModelIndex& parent) const;
 113
 114     virtual QVariant data(const QModelIndex& index, int role) const;
 115
 116     virtual bool setData(const QModelIndex &index, const QVariant &value, int role);
 117
 118     /**
 119      * given a -set.xml path, return the corresponding model index, if one
 120      * exists.
 121      */
 122
 123     QModelIndex indexOfAircraftURI(QUrl uri) const;
 124
 125     /**
 126      * return if a given aircraft is ready to be run, or not. Aircraft which
 127      * are not installed, or are downloading, are not runnable.
 128      */
 129     bool isIndexRunnable(const QModelIndex& index) const;
 130
 131 signals:
 132     void aircraftInstallFailed(QModelIndex index, QString errorMessage);
 133
 134     void aircraftInstallCompleted(QModelIndex index);
 135
 136     void scanCompleted();
 137
 138 private slots:
 139     void onScanResults();
 140
 141     void onScanFinished();
 142
 143 private:
 144     friend class PackageDelegate;
 145
 146     QVariant dataFromItem(AircraftItemPtr item, quint32 variantIndex, int role) const;
 147
 148     QVariant dataFromPackage(const simgear::pkg::PackageRef& item,
 149                              quint32 variantIndex, int role) const;
 150
 151     QVariant packageThumbnail(simgear::pkg::PackageRef p, int index, bool download = true) const;
 152
 153     void abandonCurrentScan();
 154     void refreshPackages();
 155
 156     void installSucceeded(QModelIndex index);
 157     void installFailed(QModelIndex index, simgear::pkg::Delegate::StatusCode reason);
 158
 159     QStringList m_paths;
 160     AircraftScanThread* m_scanThread;
 161     QVector<AircraftItemPtr> m_items;
 162     PackageDelegate* m_delegate;
 163
 164     QVector<quint32> m_activeVariant;
 165     QVector<quint32> m_packageVariant;
 166
 167     simgear::pkg::RootRef m_packageRoot;
 168     simgear::pkg::PackageList m_packages;
 169
 170     mutable QHash<QString, QPixmap> m_thumbnailPixmapCache;
 171 };
 172
 173 #endif // of FG_GUI_AIRCRAFT_MODEL