summaryrefslogtreecommitdiffstats
path: root/pictureswidget.h
blob: 1b7fa09bfd9dafaee863417cfe923e6d9cc1f1cc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
/*
  This program 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.
*/

#ifndef PICTURESWIDGET_H
#define PICTURESWIDGET_H

#include <QWidget>
#include <QTreeView>
#include <QSqlDatabase>

#include "smtreemodel.h"
#include "pictureviewer2.h"

class PictureView;
class MappingTreeWidget;
class MappingEditWidget;
class MappingEditDialog;
class QSqlQuery;
class PicFilesModel;
class QSortFilterProxyModel;
class QHideEvent;
class QEvent;
class HoverWindow;
class PictureViewer2;

class PicturesWidget : public QWidget {
    Q_OBJECT
    public:
        explicit PicturesWidget(QWidget *parent = 0);
        PictureView *picView() { return mPictureView; }

    public slots:
        void showPicViewer(bool toggled);

    private slots:
        void setMappingColors();
        void editMappings();
        void constructWindowTitle();

    signals:
        void needWindowTitleChange(QString);

    private:
        MappingTreeWidget *mMappingTree;
        PictureView *mPictureView;
        PictureViewer2 *mPicViewer;
        MappingEditDialog *mEditDialog;
        const QString mWindowTitleBase;
};

class PictureView : public QTreeView {
    Q_OBJECT
    public:
        explicit PictureView(QWidget *parent = 0);
        QList<int> fileMappings() { return mFilesMappings; }
        PicFilesModel *filesModel() { return mModel; }

    public slots:
        void mappingChanged(int mapping);
        void deletePics();
        void refresh();

    protected:
        virtual void hideEvent(QHideEvent *);
        virtual bool event(QEvent *e);
        virtual void contextMenuEvent(QContextMenuEvent *e);

    private slots:
        void setFileMappings();

    signals:
        void newFileMappigs();
        void editPicsMappings();

    private:
        HoverWindow *mHoverWin;
        QModelIndex mCurHover;
        QSortFilterProxyModel *mProxy;
        PicFilesModel *mModel;
        int mCursorOffset;
        QList<int> mFilesMappings;
};

class PicFilesModel : public SmTreeModel {
    Q_OBJECT
    public:
        enum Roles { FileNameRole = Qt::UserRole + 1, SizeRole = Qt::UserRole + 2, MimeTypeRole = Qt::UserRole + 3, FullPathRole = Qt::UserRole + 4, IdRole = Qt::UserRole + 5, AddedRole = Qt::UserRole + 6, Md5SumRole = Qt::UserRole + 7, SizeDisplayRole = Qt::UserRole + 8 };
        enum Fields { FileName = 0, Size = 1, MimeType = 2, FullPath = 3, Id = 4, Added = 5, Md5Sum = 6, SizeDisplay = 7 };
        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; }
        QVariant data(const QModelIndex &index, int role) const;
        void removeFiles(const QList<QPersistentModelIndex> &files);
        bool changeMappings(const QList<int> &fileIds, const QList<int> &mappingIds);

    public slots:
        void populate();

    private:
        QSqlDatabase mDb;
        QSqlQuery *mPopulateQ;
        QSqlQuery *mDeleteFileQ;
        QSqlQuery *mCurMappingIdsQ;
        QSqlQuery *mDeleteMappingsQ;
        QSqlQuery *mAddMappingsQ;
        QString mCurMappingIdsQS;
        int mMappingId;
        QList<int> mCurMappingsIds;
};

#endif // PICTURESWIDGET_H