1 files changed, 7 insertions, 1 deletions
diff --git a/archivemodel.h b/archivemodel.h
index 62c6508..7e77642 100644
--- a/archivemodel.h
+++ b/archivemodel.h
@@ -76,10 +76,16 @@ class ArchiveFilesModel : public SmTreeModel {
         explicit ArchiveFilesModel(const QStringList &headers, QObject *parent = 0);
         virtual QVariant data(const QModelIndex &index, int role) const;
         virtual bool setData(const QModelIndex &idx, const QVariant &value, int role);
+        virtual Qt::ItemFlags flags(const QModelIndex &index) const;
+        int nextDvd() const;
+        bool isMovie(const QModelIndex &idx) const;
         void populate(const QSet<int> &seriesPartIds);
 
+    public slots:
+        void refresh();
+
     private:
-        QList<int> mSeriesPartIds;
+        QSet<int> mSeriesPartIds;
         QHash<int, QString> mRoleDbColumnMap;
         QSqlDatabase mDb;
 };