Files
cantata/dynamic/dynamicrulesdialog.cpp
2017-04-08 10:35:31 +01:00

403 lines
12 KiB
C++

/*
* Cantata
*
* Copyright (c) 2011-2017 Craig Drummond <craig.p.drummond@gmail.com>
*
* ----
*
* 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.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; see the file COPYING. If not, write to
* the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#include "dynamicrulesdialog.h"
#include "dynamicruledialog.h"
#include "dynamic.h"
#include "support/messagebox.h"
#include "widgets/icons.h"
#include "widgets/basicitemdelegate.h"
#include <QIcon>
#include <QStandardItem>
#include <QStandardItemModel>
#include <QSortFilterProxyModel>
#include <QSpinBox>
class RulesSort : public QSortFilterProxyModel
{
public:
RulesSort(QObject *parent)
: QSortFilterProxyModel(parent) {
setDynamicSortFilter(true);
setFilterCaseSensitivity(Qt::CaseInsensitive);
setSortCaseSensitivity(Qt::CaseInsensitive);
setSortLocaleAware(true);
sort(0);
}
virtual ~RulesSort() {
}
bool lessThan(const QModelIndex &left, const QModelIndex &right) const {
bool lInc=left.data(Qt::UserRole+2).toBool();
bool rInc=right.data(Qt::UserRole+2).toBool();
if (lInc==rInc) {
return left.data(Qt::DisplayRole).toString().localeAwareCompare(right.data(Qt::DisplayRole).toString())<0;
} else {
return lInc ? true : false;
}
}
};
static QString translateStr(const QString &key)
{
if (Dynamic::constArtistKey==key) {
return QObject::tr("Artist");
} else if (Dynamic::constSimilarArtistsKey==key) {
return QObject::tr("SimilarArtists");
} else if (Dynamic::constAlbumArtistKey==key) {
return QObject::tr("AlbumArtist");
} else if (Dynamic::constComposerKey==key) {
return QObject::tr("Composer");
} else if (Dynamic::constCommentKey==key) {
return QObject::tr("Comment");
} else if (Dynamic::constAlbumKey==key) {
return QObject::tr("Album");
} else if (Dynamic::constTitleKey==key) {
return QObject::tr("Title");
} else if (Dynamic::constGenreKey==key) {
return QObject::tr("Genre");
} else if (Dynamic::constDateKey==key) {
return QObject::tr("Date");
} else if (Dynamic::constFileKey==key) {
return QObject::tr("File");
} else {
return key;
}
}
static void update(QStandardItem *i, const Dynamic::Rule &rule)
{
Dynamic::Rule::ConstIterator it(rule.constBegin());
Dynamic::Rule::ConstIterator end(rule.constEnd());
QMap<QString, QVariant> v;
QString str;
QString type=QObject::tr("Include");
bool exact=true;
bool include=true;
for (int count=0; it!=end; ++it, ++count) {
if (Dynamic::constExcludeKey==it.key()) {
if (QLatin1String("true")==it.value()) {
type=QObject::tr("Exclude");
include=false;
}
} else if (Dynamic::constExactKey==it.key()) {
if (QLatin1String("false")==it.value()) {
exact=false;
}
} else {
str+=QString("%1=%2").arg(translateStr(it.key()), it.value());
if (count<rule.count()-1) {
str+=' ';
}
}
v.insert(it.key(), it.value());
}
if (str.isEmpty()) {
str=type;
} else {
str=type+' '+str;
}
if (exact) {
str+=QObject::tr(" (Exact)");
}
i->setText(str);
i->setData(v);
i->setData(include, Qt::UserRole+2);
i->setFlags(Qt::ItemIsSelectable| Qt::ItemIsEnabled);
}
DynamicRulesDialog::DynamicRulesDialog(QWidget *parent)
: Dialog(parent, "DynamicRulesDialog")
, dlg(0)
{
QWidget *mainWidet = new QWidget(this);
setupUi(mainWidet);
setMainWidget(mainWidet);
setButtons(Ok|Cancel);
enableButton(Ok, false);
setCaption(tr("Dynamic Rules"));
setAttribute(Qt::WA_DeleteOnClose);
connect(addBtn, SIGNAL(clicked()), SLOT(add()));
connect(editBtn, SIGNAL(clicked()), SLOT(edit()));
connect(removeBtn, SIGNAL(clicked()), SLOT(remove()));
addBtn->setIcon(Icons::self()->addIcon);
editBtn->setIcon(Icons::self()->editIcon);
removeBtn->setIcon(Icons::self()->minusIcon);
connect(rulesList, SIGNAL(itemsSelected(bool)), SLOT(controlButtons()));
connect(nameText, SIGNAL(textChanged(const QString &)), SLOT(enableOkButton()));
connect(aboutLabel, SIGNAL(leftClickedUrl()), this, SLOT(showAbout()));
connect(Dynamic::self(), SIGNAL(saved(bool)), SLOT(saved(bool)));
messageWidget->setVisible(false);
model=new QStandardItemModel(this);
proxy=new RulesSort(this);
proxy->setSourceModel(model);
rulesList->setModel(proxy);
rulesList->setItemDelegate(new BasicItemDelegate(rulesList));
rulesList->setAlternatingRowColors(false);
minDuration->setSpecialValueText(tr("None"));
maxDuration->setSpecialValueText(tr("None"));
controlButtons();
resize(500, 240);
static bool registered=false;
if (!registered) {
qRegisterMetaType<Dynamic::Rule>("Dynamic::Rule");
registered=true;
}
}
DynamicRulesDialog::~DynamicRulesDialog()
{
}
void DynamicRulesDialog::edit(const QString &name)
{
Dynamic::Entry e=Dynamic::self()->entry(name);
if (model->rowCount()) {
model->removeRows(0, model->rowCount());
}
nameText->setText(name);
foreach (const Dynamic::Rule &r, e.rules) {
QStandardItem *item = new QStandardItem();
::update(item, r);
model->setItem(model->rowCount(), 0, item);
}
origName=name;
ratingFrom->setValue(e.ratingFrom);
ratingTo->setValue(e.ratingTo);
minDuration->setValue(e.minDuration);
maxDuration->setValue(e.maxDuration);
show();
}
void DynamicRulesDialog::enableOkButton()
{
bool enable=!nameText->text().trimmed().isEmpty();
enableButton(Ok, enable);
}
void DynamicRulesDialog::slotButtonClicked(int button)
{
switch (button) {
case Ok: {
if (save()) {
accept();
}
break;
}
case Cancel:
reject();
// Need to call this - if not, when dialog is closed by window X control, it is not deleted!!!!
Dialog::slotButtonClicked(button);
break;
default:
break;
}
}
void DynamicRulesDialog::controlButtons()
{
int numSel=rulesList->selectedIndexes().count();
removeBtn->setEnabled(numSel>0);
editBtn->setEnabled(1==numSel);
}
void DynamicRulesDialog::add()
{
if (!dlg) {
dlg=new DynamicRuleDialog(this);
connect(dlg, SIGNAL(addRule(const Dynamic::Rule&)), SLOT(addRule(const Dynamic::Rule&)));
}
dlg->createNew();
}
void DynamicRulesDialog::addRule(const Dynamic::Rule &rule)
{
QStandardItem *item = new QStandardItem();
::update(item, rule);
int index=indexOf(item);
if (-1!=index) {
delete item;
} else {
model->setItem(model->rowCount(), 0, item);
}
}
void DynamicRulesDialog::edit()
{
QModelIndexList items=rulesList->selectedIndexes();
if (1!=items.count()) {
return;
}
if (!dlg) {
dlg=new DynamicRuleDialog(this);
connect(dlg, SIGNAL(addRule(const Dynamic::Rule&)), SLOT(addRule(const Dynamic::Rule&)));
}
QModelIndex index=proxy->mapToSource(items.at(0));
QStandardItem *item=model->itemFromIndex(index);
Dynamic::Rule rule;
QMap<QString, QVariant> v=item->data().toMap();
QMap<QString, QVariant>::ConstIterator it(v.constBegin());
QMap<QString, QVariant>::ConstIterator end(v.constEnd());
for (; it!=end; ++it) {
rule.insert(it.key(), it.value().toString());
}
if (dlg->edit(rule)) {
::update(item, dlg->rule());
int idx=indexOf(item, true);
if (-1!=idx && idx!=index.row()) {
model->removeRow(index.row());
}
}
}
void DynamicRulesDialog::remove()
{
QModelIndexList items=rulesList->selectedIndexes();
QList<int> rows;
foreach (const QModelIndex &i, items) {
rows.append(proxy->mapToSource(i).row());
}
qSort(rows);
while (rows.count()) {
model->removeRow(rows.takeLast());
}
}
void DynamicRulesDialog::showAbout()
{
MessageBox::information(this,
#ifdef Q_OS_MAC
tr("About dynamic rules")+QLatin1String("<br/><br/>")+
#endif
tr("<p>Cantata will query your library using all of the rules listed. "
"The list of <i>Include</i> rules will be used to build a set of songs that can be used. "
"The list of <i>Exclude</i> rules will be used to build a set of songs that cannot be used. "
"If there are no <i>Include</i> rules, Cantata will assume that all songs (bar those from <i>Exclude</i>) can be used.</p>"
"<p>e.g. to have Cantata look for 'Rock songs by Wibble OR songs by Various Artists', you would need the following: "
"<ul><li>Include AlbumArtist=Wibble Genre=Rock</li><li>Include AlbumArtist=Various Artists</li></ul> "
"To have Cantata look for 'Songs by Wibble but not from album Abc', you would need the following: "
"<ul><li>Include AlbumArtist=Wibble</li><li>Exclude AlbumArtist=Wibble Album=Abc</li></ul>"
"After the set of usable songs has been created, Cantata will randomly select songs to "
"keep the play queue filled with 10 entries. If a range of ratings has been specified, then "
"only songs with a rating within this range will be used. Likewise, if a duration has been set.</p>")
);
}
void DynamicRulesDialog::saved(bool s)
{
if (s) {
accept();
} else {
messageWidget->setError(tr("Failed to save %1").arg(nameText->text().trimmed()));
controls->setEnabled(true);
}
}
bool DynamicRulesDialog::save()
{
if (!controls->isEnabled()) {
return false;
}
QString name=nameText->text().trimmed();
if (name.isEmpty()) {
return false;
}
if (name!=origName && Dynamic::self()->exists(name) &&
MessageBox::No==MessageBox::warningYesNo(this, tr("A set of rules named '%1' already exists!\n\nOverwrite?").arg(name),
tr("Overwrite Rules"), StdGuiItem::overwrite(), StdGuiItem::cancel())) {
return false;
}
Dynamic::Entry entry;
entry.name=name;
int from=ratingFrom->value();
int to=ratingTo->value();
entry.ratingFrom=qMin(from, to);
entry.ratingTo=qMax(from, to);
from=minDuration->value();
to=maxDuration->value();
entry.minDuration=qMin(from, to);
entry.maxDuration=qMax(from, to);
for (int i=0; i<model->rowCount(); ++i) {
QStandardItem *itm=model->item(i);
if (itm) {
QMap<QString, QVariant> v=itm->data().toMap();
QMap<QString, QVariant>::ConstIterator it(v.constBegin());
QMap<QString, QVariant>::ConstIterator end(v.constEnd());
Dynamic::Rule rule;
for (; it!=end; ++it) {
rule.insert(it.key(), it.value().toString());
}
entry.rules.append(rule);
}
}
bool saved=Dynamic::self()->save(entry);
if (Dynamic::self()->isRemote()) {
if (saved) {
messageWidget->setInformation(tr("Saving %1").arg(name));
controls->setEnabled(false);
enableButton(Ok, false);
}
return false;
} else {
if (saved && !origName.isEmpty() && entry.name!=origName) {
Dynamic::self()->del(origName);
}
return saved;
}
}
int DynamicRulesDialog::indexOf(QStandardItem *item, bool diff)
{
QMap<QString, QVariant> v=item->data().toMap();
for (int i=0; i<model->rowCount(); ++i) {
QStandardItem *itm=model->item(i);
if (itm) {
if (itm->data().toMap()==v && (!diff || itm!=item)) {
return i;
}
}
}
return -1;
}