src/panels/filter/filterpanel.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2010 by Sebastian Trueg <trueg@kde.org>                  *
   3  *                                                                         *
   4  *   This program is free software; you can redistribute it and/or modify  *
   5  *   it under the terms of the GNU General Public License as published by  *
   6  *   the Free Software Foundation; either version 2 of the License, or     *
   7  *   (at your option) any later version.                                   *
   8  *                                                                         *
   9  *   This program is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU General Public License for more details.                          *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU General Public License     *
  15  *   along with this program; if not, write to the                         *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA            *
  18  ***************************************************************************/
  19
  20 #include "filterpanel.h"
  21
  22 #include <nepomuk/filequery.h>
  23 #include <nepomuk/facetwidget.h>
  24 #include <nepomuk/facet.h>
  25 #include <nepomuk/resourcemanager.h>
  26 #include <Nepomuk/Utils/SimpleFacet>
  27 #include <Nepomuk/Utils/ProxyFacet>
  28 #include <Nepomuk/Utils/DynamicResourceFacet>
  29 #include <Nepomuk/Query/FileQuery>
  30 #include <Nepomuk/Query/ResourceTypeTerm>
  31 #include <Nepomuk/Query/LiteralTerm>
  32 #include <Nepomuk/Query/ComparisonTerm>
  33 #include <Nepomuk/Vocabulary/NFO>
  34 #include <Nepomuk/Vocabulary/NMM>
  35
  36 #include <kfileitem.h>
  37 #include <kio/jobclasses.h>
  38 #include <kio/job.h>
  39
  40 #include <QPushButton>
  41 #include <QShowEvent>
  42 #include <QTreeView>
  43 #include <QVBoxLayout>
  44
  45 FilterPanel::FilterPanel(QWidget* parent) :
  46     Panel(parent),
  47     m_initialized(false),
  48     m_nepomukEnabled(false),
  49     m_lastSetUrlStatJob(0),
  50     m_removeFolderRestrictionButton(0),
  51     m_facetWidget(0),
  52     m_unfacetedRestQuery()
  53 {
  54 }
  55
  56 FilterPanel::~FilterPanel()
  57 {
  58 }
  59
  60 bool FilterPanel::urlChanged()
  61 {
  62     if (isVisible() && m_nepomukEnabled) {
  63         setQuery(Nepomuk::Query::Query());
  64
  65         delete m_lastSetUrlStatJob;
  66
  67         m_lastSetUrlStatJob = KIO::stat(url(), KIO::HideProgressInfo);
  68         connect(m_lastSetUrlStatJob, SIGNAL(result(KJob*)),
  69                 this, SLOT(slotSetUrlStatFinished(KJob*)));
  70     }
  71
  72     return true;
  73 }
  74
  75 void FilterPanel::showEvent(QShowEvent* event)
  76 {
  77     if (event->spontaneous()) {
  78         Panel::showEvent(event);
  79         return;
  80     }
  81
  82     if (!m_initialized) {
  83         QVBoxLayout* layout = new QVBoxLayout(this);
  84         Q_ASSERT(m_removeFolderRestrictionButton == 0);
  85         m_removeFolderRestrictionButton = new QPushButton(i18n("Remove folder restriction"), this);
  86         connect(m_removeFolderRestrictionButton, SIGNAL(clicked()), SLOT(slotRemoveFolderRestrictionClicked()));
  87
  88         layout->addWidget(m_removeFolderRestrictionButton);
  89
  90         Q_ASSERT(m_facetWidget == 0);
  91         m_facetWidget = new Nepomuk::Utils::FacetWidget(this);
  92         layout->addWidget(m_facetWidget, 1);
  93
  94         // File Type
  95         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createFileTypeFacet());
  96
  97         // Image Size
  98         Nepomuk::Utils::ProxyFacet* imageSizeProxy = new Nepomuk::Utils::ProxyFacet();
  99         imageSizeProxy->setFacetCondition(Nepomuk::Query::ResourceTypeTerm(Nepomuk::Vocabulary::NFO::Image()));
 100         Nepomuk::Utils::SimpleFacet* imageSizeFacet = new Nepomuk::Utils::SimpleFacet(imageSizeProxy);
 101         imageSizeFacet->setSelectionMode(Nepomuk::Utils::Facet::MatchAny);
 102         imageSizeFacet->addTerm( i18nc("option:check Refers to a filter on image size", "Small"),
 103                                 Nepomuk::Vocabulary::NFO::width() <= Nepomuk::Query::LiteralTerm(300));
 104         imageSizeFacet->addTerm( i18nc("option:check Refers to a filter on image size", "Medium"),
 105                                 (Nepomuk::Vocabulary::NFO::width() > Nepomuk::Query::LiteralTerm(300)) &&
 106                                 (Nepomuk::Vocabulary::NFO::width() <= Nepomuk::Query::LiteralTerm(800)));
 107         imageSizeFacet->addTerm( i18nc("option:check Refers to a filter on image size", "Large"),
 108                                 Nepomuk::Vocabulary::NFO::width() > Nepomuk::Query::LiteralTerm(800));
 109         imageSizeProxy->setSourceFacet(imageSizeFacet);
 110         m_facetWidget->addFacet(imageSizeProxy);
 111
 112         // Artists
 113         Nepomuk::Utils::ProxyFacet* artistProxy = new Nepomuk::Utils::ProxyFacet();
 114         artistProxy->setFacetCondition(Nepomuk::Query::ResourceTypeTerm(Nepomuk::Vocabulary::NFO::Audio()));
 115         Nepomuk::Utils::DynamicResourceFacet* artistFacet = new Nepomuk::Utils::DynamicResourceFacet(artistProxy);
 116         artistFacet->setSelectionMode(Nepomuk::Utils::Facet::MatchAny);
 117         artistFacet->setRelation(Nepomuk::Vocabulary::NMM::performer());
 118         artistProxy->setSourceFacet(artistFacet);
 119         m_facetWidget->addFacet(artistProxy);
 120
 121         // Misc
 122         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createDateFacet());
 123         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createRatingFacet());
 124         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createTagFacet());
 125
 126         Q_ASSERT(m_lastSetUrlStatJob == 0);
 127         m_lastSetUrlStatJob = KIO::stat(url(), KIO::HideProgressInfo);
 128         connect(m_lastSetUrlStatJob, SIGNAL(result(KJob*)),
 129                 this, SLOT(slotSetUrlStatFinished(KJob*)));
 130
 131         connect(m_facetWidget, SIGNAL(queryTermChanged(Nepomuk::Query::Term)),
 132                 this, SLOT(slotQueryTermChanged(Nepomuk::Query::Term)));
 133
 134         m_nepomukEnabled = (Nepomuk::ResourceManager::instance()->init() == 0);
 135         m_facetWidget->setEnabled(m_nepomukEnabled);
 136
 137         m_initialized = true;
 138     }
 139
 140     Panel::showEvent(event);
 141 }
 142
 143 void FilterPanel::slotSetUrlStatFinished(KJob* job)
 144 {
 145     m_lastSetUrlStatJob = 0;
 146
 147     const KIO::UDSEntry uds = static_cast<KIO::StatJob*>(job)->statResult();
 148     const QString nepomukQueryStr = uds.stringValue(KIO::UDSEntry::UDS_NEPOMUK_QUERY);
 149     Nepomuk::Query::FileQuery nepomukQuery;
 150     if (!nepomukQueryStr.isEmpty()) {
 151         nepomukQuery = Nepomuk::Query::Query::fromString(nepomukQueryStr);
 152     } else if (url().isLocalFile()) {
 153         // Fallback query for local file URLs
 154         nepomukQuery.addIncludeFolder(url(), false);
 155     }
 156     setQuery(nepomukQuery);
 157 }
 158
 159 void FilterPanel::slotQueryTermChanged(const Nepomuk::Query::Term& term)
 160 {
 161     Nepomuk::Query::Query query(m_unfacetedRestQuery && term);
 162     emit urlActivated(query.toSearchUrl());
 163 }
 164
 165 void FilterPanel::slotRemoveFolderRestrictionClicked()
 166 {
 167     Nepomuk::Query::FileQuery query(m_unfacetedRestQuery && m_facetWidget->queryTerm());
 168     query.setIncludeFolders(KUrl::List());
 169     query.setExcludeFolders(KUrl::List());
 170     m_facetWidget->setClientQuery(query);
 171     emit urlActivated(query.toSearchUrl());
 172 }
 173
 174 void FilterPanel::setQuery(const Nepomuk::Query::Query& query)
 175 {
 176     if (query.isValid()) {
 177         const bool block = m_facetWidget->blockSignals(true);
 178
 179         m_removeFolderRestrictionButton->setVisible(query.isFileQuery() && !query.toFileQuery().includeFolders().isEmpty());
 180         m_unfacetedRestQuery = m_facetWidget->extractFacetsFromQuery(query);
 181         m_facetWidget->setClientQuery(query);
 182         setEnabled(true);
 183
 184         m_facetWidget->blockSignals(block);
 185     } else {
 186         m_unfacetedRestQuery = Nepomuk::Query::Query();
 187         setEnabled(false);
 188     }
 189 }