diff options
Diffstat (limited to 'newpicsdialog.cpp')
| -rw-r--r-- | newpicsdialog.cpp | 248 |
1 files changed, 248 insertions, 0 deletions
diff --git a/newpicsdialog.cpp b/newpicsdialog.cpp new file mode 100644 index 0000000..673865b --- /dev/null +++ b/newpicsdialog.cpp @@ -0,0 +1,248 @@ +/* + 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. +*/ + +#include <QFileInfo> +#include <QSqlQuery> +#include <QLabel> +#include <QTreeView> +#include <QSortFilterProxyModel> +#include <QPushButton> +#include <QSettings> +#include <QDir> +#include <QFileDialog> +#include <QPersistentModelIndex> +#include <QHBoxLayout> +#include <QVBoxLayout> + +#include "newpicsdialog.h" +#include "smtreeitem.h" +#include "helper.h" + +NewPicsDialog::NewPicsDialog(QWidget *parent) : QWidget(parent) { + //files widget + mFilesWidget = new QWidget; + mFilesV = new QTreeView; + mFilesProxy = new QSortFilterProxyModel(this); + mFilesModel = new NewPicFilesModel(QStringList() << tr("Filename") << tr("Val.") << tr("Md5") << tr("Mime type") << tr("Full path"), this); + mFilesProxy->setSourceModel(mFilesModel); + mFilesV->setModel(mFilesProxy); + mFilesV->setSortingEnabled(true); + mFilesV->setAlternatingRowColors(true); + mFilesV->setColumnHidden(1, true); + mAddFiles = new QPushButton(tr("Select...")); + connect(mAddFiles, SIGNAL(clicked()), this, SLOT(addFiles())); + mRemoveFiles = new QPushButton(tr("Remove")); + connect(mRemoveFiles, SIGNAL(clicked()), this, SLOT(removeFiles())); + QHBoxLayout *fileButtonLayout = new QHBoxLayout; + fileButtonLayout->addStretch(); + fileButtonLayout->addWidget(mRemoveFiles); + fileButtonLayout->addWidget(mAddFiles); + QVBoxLayout *filesLayout = new QVBoxLayout; + filesLayout->addWidget(mFilesV); + filesLayout->addLayout(fileButtonLayout); + mFilesWidget->setLayout(filesLayout); + + //put it together + setMinimumWidth(600); + setMinimumHeight(400); + mTab = new QTabWidget; + mTab->addTab(mFilesWidget, tr("Files")); + QHBoxLayout *mainLayout = new QHBoxLayout; + mainLayout->addWidget(mTab); + setLayout(mainLayout); +} + +void NewPicsDialog::addFiles(){ + QSettings s; + QString startDir = s.value("paths/selecteddir", QDir::homePath()).toString(); + QStringList files = QFileDialog::getOpenFileNames(this, tr("Select pictures"), startDir); + if(files.isEmpty()){ + return; + } + mFilesV->setSortingEnabled(false); + foreach(QString f, files){ + mFilesModel->addFile(f); + } + mFilesV->setSortingEnabled(true); + for(int i = 0; i < NewPicFilesModel::NumFields; ++i){ + mFilesV->resizeColumnToContents(i); + } +} + +void NewPicsDialog::removeFiles(){ + QModelIndexList selected = mFilesV->selectionModel()->selectedRows(); + if(selected.isEmpty()){ + return; + } + QModelIndexList real; + foreach(QModelIndex i, selected){ + real << mFilesProxy->mapToSource(i); + } + QList<QPersistentModelIndex> perm; + foreach(QModelIndex i, real){ + perm << i; + } + foreach(QPersistentModelIndex pi, perm){ + mFilesModel->removeRows(pi.row(), 1, pi.parent()); + } +} + +NewPicFilesModel::NewPicFilesModel(const QStringList &header, QObject *parent) : SmTreeModel(header, parent){ + mDb = QSqlDatabase::database("mDb"); + mMd5Query = new QSqlQuery(mDb); + mMd5Query->prepare("SELECT COUNT(*) FROM pics WHERE cmd5sum = :md5"); +} + +NewPicFilesModel::~NewPicFilesModel(){ + delete mMd5Query; + mDb = QSqlDatabase(); +} + +Qt::ItemFlags NewPicFilesModel::flags(const QModelIndex &index) const{ + Q_UNUSED(index); + return (Qt::ItemIsSelectable | Qt::ItemIsEnabled); +} + +QVariant NewPicFilesModel::data(const QModelIndex &index, int role) const{ + if(!index.isValid()){ + return QVariant(); + } + SmTreeItem *item = itemAt(index); + if(role == FileNameRole){ + return item->data(FileName); + } + if(role == ValidRole){ + return item->data(Valid); + } + if(role == Md5SumRole){ + return item->data(Md5Sum); + } + if(role == MimeTypeRole){ + return item->data(MimeType); + } + if(role == FullPathRole){ + return item->data(FullPath); + } + if(role == Qt::DecorationRole && index.column() == 0){ + bool valid = item->data(Valid).toBool(); + if(valid){ + return QIcon(":/gaping_ass.png"); + }else{ + return QIcon(":/chastity_belt.png"); + } + } + if(role == Qt::FontRole && index.column() == Md5Sum){ + return QFont("courier"); + } + return SmTreeModel::data(index, role); +} + +bool NewPicFilesModel::setData(const QModelIndex &index, const QVariant &value, int role){ + if(role == Qt::EditRole || !index.isValid()){ + return false; + } + SmTreeItem *item = itemAt(index); + if(role == FileNameRole){ + item->setData(FileName, value); + } + if(role == ValidRole){ + item->setData(Valid, value); + } + if(role == Md5SumRole){ + item->setData(Md5Sum, value); + } + if(role == MimeTypeRole){ + item->setData(MimeType, value); + } + if(role == FullPathRole){ + item->setData(FullPath, value); + } + return true; +} + +void NewPicFilesModel::setFiles(const QStringList &paths) { + SmTreeItem *rootItem = new SmTreeItem(NumFields); + foreach(QString file, paths){ + QFileInfo fi(file); + QList<QVariant> fileData; + if(!fi.exists()){ + continue; + } + QString md5 = Helper::md5Sum(fi.absoluteFilePath()); + QString mt = Helper::mimeType(fi.absoluteFilePath()).toLower(); + bool valid = true; + if(!mt.startsWith("image")){ + valid = false; + } + if(haveMd5(md5)){ + valid = false; + } + fileData << fi.fileName() << valid << md5 << mt << fi.absoluteFilePath(); + SmTreeItem *newFile = new SmTreeItem(fileData, rootItem); + rootItem->appendChild(newFile); + } + setRoot(rootItem); +} + +void NewPicFilesModel::addFile(const QString &path){ + QFileInfo fi(path); + if(!fi.exists()){ + return; + } + QString md5 = Helper::md5Sum(fi.absoluteFilePath()); + QString mt = Helper::mimeType(fi.absoluteFilePath()); + bool valid = true; + if(!mt.startsWith("image")){ + valid = false; + } + if(haveMd5(md5)){ + valid = false; + } + insertRows(0, 1, QModelIndex()); + QModelIndex inserted = index(0, 1, QModelIndex()); + setData(inserted, fi.fileName(), FileNameRole); + setData(inserted, valid, ValidRole); + setData(inserted, md5, Md5SumRole); + setData(inserted, mt, MimeTypeRole); + setData(inserted, fi.absoluteFilePath(), FullPathRole); +} + +void NewPicFilesModel::removeFile(const QModelIndex &idx){ + if(!idx.isValid()){ + return; + } + removeRows(idx.row(), 1, idx.parent()); +} + +QList<FileData> NewPicFilesModel::validFiles() const { + SmTreeItem *rootItem = root(); + QList<FileData> retval; + for(int i = 0; i < rootItem->childCount(); ++i){ + SmTreeItem *child = rootItem->child(i); + if(!child->data(Valid).toBool()){ + continue; + } + FileData data; + data.fileName = child->data(FileName).toString(); + data.valid = child->data(Valid).toBool(); + data.md5sum = child->data(Md5Sum).toString(); + data.mimeType = child->data(MimeType).toString(); + data.fullPath = child->data(FullPath).toString(); + retval << data; + } + return retval; +} + +bool NewPicFilesModel::haveMd5(const QString &md5) const { + mMd5Query->bindValue(":md5", md5); + mMd5Query->exec(); + int count = 0; + while(mMd5Query->next()){ + count = mMd5Query->value(0).toInt(); + } + return ((count > 0) ? true : false); +} |