1 files changed, 15 insertions, 0 deletions
diff --git a/pictureswidget.h b/pictureswidget.h
index b0bd277..b8a7b40 100644
--- a/pictureswidget.h
+++ b/pictureswidget.h
@@ -29,6 +29,9 @@ class PicturesWidget : public QWidget {
         explicit PicturesWidget(QWidget *parent = 0);
         PictureView *picView() { return mPictureView; }
 
+    private slots:
+        void setMappingColors();
+
     private:
         MappingTreeWidget *mMappingTree;
         PictureView *mPictureView;
@@ -38,6 +41,7 @@ class PictureView : public QTreeView {
     Q_OBJECT
     public:
         explicit PictureView(QWidget *parent = 0);
+        QList<int> fileMappings() { return mFilesMappings; }
 
     public slots:
         void mappingChanged(int mapping);
@@ -48,12 +52,19 @@ class PictureView : public QTreeView {
         virtual bool event(QEvent *e);
         virtual void contextMenuEvent(QContextMenuEvent *e);
 
+    private slots:
+        void setFileMappings();
+
+    signals:
+        void newFileMappigs();
+
     private:
         HoverWindow *mHoverWin;
         QModelIndex mCurHover;
         QSortFilterProxyModel *mProxy;
         PicFilesModel *mModel;
         int mCursorOffset;
+        QList<int> mFilesMappings;
 };
 
 class PicFilesModel : public SmTreeModel {
@@ -64,6 +75,7 @@ class PicFilesModel : public SmTreeModel {
         enum { NumFields = 8 };
         explicit PicFilesModel(const QStringList &headers, QObject *parent = 0);
         void setMapping(int mappingId) { mMappingId = mappingId; }
+        QList<int> mappingIds(const QList<QVariant> &fileIds);
 
         //some data
         Qt::ItemFlags flags(const QModelIndex &) const { return Qt::ItemIsEnabled | Qt::ItemIsSelectable; }
@@ -77,7 +89,10 @@ class PicFilesModel : public SmTreeModel {
         QSqlDatabase mDb;
         QSqlQuery *mPopulateQ;
         QSqlQuery *mDeleteFileQ;
+        QSqlQuery *mCurMappingIdsQ;
+        QString mCurMappingIdsQS;
         int mMappingId;
+        QList<int> mCurMappingsIds;
 };
 
 #endif // PICTURESWIDGET_H