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/Utils/SimpleFacet>
  26 #include <Nepomuk/Utils/ProxyFacet>
  27 #include <Nepomuk/Utils/DynamicResourceFacet>
  28 #include <Nepomuk/Query/FileQuery>
  29 #include <Nepomuk/Query/ResourceTypeTerm>
  30 #include <Nepomuk/Query/LiteralTerm>
  31 #include <Nepomuk/Query/ComparisonTerm>
  32 #include <Nepomuk/Vocabulary/NFO>
  33 #include <Nepomuk/Vocabulary/NMM>
  34
  35 #include <kfileitem.h>
  36 #include <kio/jobclasses.h>
  37 #include <kio/job.h>
  38
  39 #include <QPushButton>
  40 #include <QShowEvent>
  41 #include <QTreeView>
  42 #include <QVBoxLayout>
  43
  44 FilterPanel::FilterPanel(QWidget* parent) :
  45     Panel(parent),
  46     m_initialized(false),
  47     m_lastSetUrlStatJob(0),
  48     m_removeFolderRestrictionButton(0),
  49     m_facetWidget(0),
  50     m_unfacetedRestQuery()
  51 {
  52 }
  53
  54 FilterPanel::~FilterPanel()
  55 {
  56 }
  57
  58 bool FilterPanel::urlChanged()
  59 {
  60     if (isVisible()) {
  61         setQuery(Nepomuk::Query::Query());
  62
  63         delete m_lastSetUrlStatJob;
  64
  65         m_lastSetUrlStatJob = KIO::stat(url(), KIO::HideProgressInfo);
  66         connect(m_lastSetUrlStatJob, SIGNAL(result(KJob*)),
  67                 this, SLOT(slotSetUrlStatFinished(KJob*)));
  68     }
  69
  70     return true;
  71 }
  72
  73 void FilterPanel::showEvent(QShowEvent* event)
  74 {
  75     if (event->spontaneous()) {
  76         Panel::showEvent(event);
  77         return;
  78     }
  79
  80     if (!m_initialized) {
  81         QVBoxLayout* layout = new QVBoxLayout(this);
  82         Q_ASSERT(m_removeFolderRestrictionButton == 0);
  83         m_removeFolderRestrictionButton = new QPushButton(i18n("Remove folder restriction"), this);
  84         connect(m_removeFolderRestrictionButton, SIGNAL(clicked()), SLOT(slotRemoveFolderRestrictionClicked()));
  85
  86         layout->addWidget(m_removeFolderRestrictionButton);
  87
  88         Q_ASSERT(m_facetWidget == 0);
  89         m_facetWidget = new Nepomuk::Utils::FacetWidget(this);
  90         layout->addWidget(m_facetWidget, 1);
  91
  92         // File Type
  93         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createFileTypeFacet());
  94
  95         // Image Size
  96         Nepomuk::Utils::ProxyFacet* imageSizeProxy = new Nepomuk::Utils::ProxyFacet();
  97         imageSizeProxy->setFacetCondition(Nepomuk::Query::ResourceTypeTerm(Nepomuk::Vocabulary::NFO::Image()));
  98         Nepomuk::Utils::SimpleFacet* imageSizeFacet = new Nepomuk::Utils::SimpleFacet(imageSizeProxy);
  99         imageSizeFacet->setSelectionMode(Nepomuk::Utils::Facet::MatchAny);
 100         imageSizeFacet->addTerm( i18nc("option:check Refers to a filter on image size", "Small"),
 101                                 Nepomuk::Vocabulary::NFO::width() <= Nepomuk::Query::LiteralTerm(300));
 102         imageSizeFacet->addTerm( i18nc("option:check Refers to a filter on image size", "Medium"),
 103                                 (Nepomuk::Vocabulary::NFO::width() > Nepomuk::Query::LiteralTerm(300)) &&
 104                                 (Nepomuk::Vocabulary::NFO::width() <= Nepomuk::Query::LiteralTerm(800)));
 105         imageSizeFacet->addTerm( i18nc("option:check Refers to a filter on image size", "Large"),
 106                                 Nepomuk::Vocabulary::NFO::width() > Nepomuk::Query::LiteralTerm(800));
 107         imageSizeProxy->setSourceFacet(imageSizeFacet);
 108         m_facetWidget->addFacet(imageSizeProxy);
 109
 110         // Artists
 111         Nepomuk::Utils::ProxyFacet* artistProxy = new Nepomuk::Utils::ProxyFacet();
 112         artistProxy->setFacetCondition(Nepomuk::Query::ResourceTypeTerm(Nepomuk::Vocabulary::NFO::Audio()));
 113         Nepomuk::Utils::DynamicResourceFacet* artistFacet = new Nepomuk::Utils::DynamicResourceFacet(artistProxy);
 114         artistFacet->setSelectionMode(Nepomuk::Utils::Facet::MatchAny);
 115         artistFacet->setRelation(Nepomuk::Vocabulary::NMM::performer());
 116         artistProxy->setSourceFacet(artistFacet);
 117         m_facetWidget->addFacet(artistProxy);
 118
 119         // Misc
 120         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createDateFacet());
 121         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createRatingFacet());
 122         m_facetWidget->addFacet(Nepomuk::Utils::Facet::createTagFacet());
 123
 124         Q_ASSERT(m_lastSetUrlStatJob == 0);
 125         m_lastSetUrlStatJob = KIO::stat(url(), KIO::HideProgressInfo);
 126         connect(m_lastSetUrlStatJob, SIGNAL(result(KJob*)),
 127                 this, SLOT(slotSetUrlStatFinished(KJob*)));
 128
 129         connect(m_facetWidget, SIGNAL(queryTermChanged(Nepomuk::Query::Term)),
 130                 this, SLOT(slotQueryTermChanged(Nepomuk::Query::Term)));
 131
 132         m_initialized = true;
 133     }
 134
 135     Panel::showEvent(event);
 136 }
 137
 138 void FilterPanel::slotSetUrlStatFinished(KJob* job)
 139 {
 140     m_lastSetUrlStatJob = 0;
 141
 142     const KIO::UDSEntry uds = static_cast<KIO::StatJob*>(job)->statResult();
 143     const QString nepomukQueryStr = uds.stringValue(KIO::UDSEntry::UDS_NEPOMUK_QUERY);
 144     Nepomuk::Query::FileQuery nepomukQuery;
 145     if (!nepomukQueryStr.isEmpty()) {
 146         nepomukQuery = Nepomuk::Query::Query::fromString(nepomukQueryStr);
 147     } else if (url().isLocalFile()) {
 148         // Fallback query for local file URLs
 149         nepomukQuery.addIncludeFolder(url(), false);
 150     }
 151     setQuery(nepomukQuery);
 152 }
 153
 154 void FilterPanel::slotQueryTermChanged(const Nepomuk::Query::Term& term)
 155 {
 156     Nepomuk::Query::Query query(m_unfacetedRestQuery && term);
 157     emit urlActivated(query.toSearchUrl());
 158 }
 159
 160 void FilterPanel::slotRemoveFolderRestrictionClicked()
 161 {
 162     Nepomuk::Query::FileQuery query(m_unfacetedRestQuery && m_facetWidget->queryTerm());
 163     query.setIncludeFolders(KUrl::List());
 164     query.setExcludeFolders(KUrl::List());
 165     m_facetWidget->setClientQuery(query);
 166     emit urlActivated(query.toSearchUrl());
 167 }
 168
 169 void FilterPanel::setQuery(const Nepomuk::Query::Query& query)
 170 {
 171     if (query.isValid()) {
 172         m_removeFolderRestrictionButton->setVisible(query.isFileQuery() && !query.toFileQuery().includeFolders().isEmpty());
 173         m_unfacetedRestQuery = m_facetWidget->extractFacetsFromQuery(query);
 174         m_facetWidget->setClientQuery(query);
 175         setEnabled(true);
 176     } else {
 177         m_unfacetedRestQuery = Nepomuk::Query::Query();
 178         setEnabled(false);
 179     }
 180 }