comparison rdf/RDFTransformFactory.cpp @ 439:beb2948baa77

* Merge revisions 1041 to 1130 from sv-rdf-import branch

+/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
+
+/*
+    Sonic Visualiser
+    An audio file viewer and annotation editor.
+    Centre for Digital Music, Queen Mary, University of London.
+    This file copyright 2008 QMUL.
+   
+    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.  See the file
+    COPYING included with this distribution for more information.
+*/
+
+#include "RDFTransformFactory.h"
+
+#include <map>
+#include <vector>
+
+#include <redland.h>
+#include <rasqal.h>
+
+#include <iostream>
+#include <cmath>
+
+#include "SimpleSPARQLQuery.h"
+#include "PluginRDFIndexer.h"
+#include "base/ProgressReporter.h"
+
+#include "transform/TransformFactory.h"
+
+using std::cerr;
+using std::endl;
+
+typedef const unsigned char *STR; // redland's expected string type
+
+
+class RDFTransformFactoryImpl
+{
+public:
+    RDFTransformFactoryImpl(QString url);
+    virtual ~RDFTransformFactoryImpl();
+    
+    bool isOK();
+    QString getErrorString() const;
+
+    std::vector<Transform> getTransforms(ProgressReporter *);
+
+protected:
+    QString m_urlString;
+    QString m_errorString;
+};
+
+
+QString
+RDFTransformFactory::getKnownExtensions()
+{
+    return "*.rdf *.n3 *.ttl";
+}
+
+RDFTransformFactory::RDFTransformFactory(QString url) :
+    m_d(new RDFTransformFactoryImpl(url)) 
+{
+}
+
+RDFTransformFactory::~RDFTransformFactory()
+{
+    delete m_d;
+}
+
+bool
+RDFTransformFactory::isOK()
+{
+    return m_d->isOK();
+}
+
+QString
+RDFTransformFactory::getErrorString() const
+{
+    return m_d->getErrorString();
+}
+
+std::vector<Transform>
+RDFTransformFactory::getTransforms(ProgressReporter *r)
+{
+    return m_d->getTransforms(r);
+}
+
+RDFTransformFactoryImpl::RDFTransformFactoryImpl(QString url) :
+    m_urlString(url)
+{
+}
+
+RDFTransformFactoryImpl::~RDFTransformFactoryImpl()
+{
+}
+
+bool
+RDFTransformFactoryImpl::isOK()
+{
+    return (m_errorString == "");
+}
+
+QString
+RDFTransformFactoryImpl::getErrorString() const
+{
+    return m_errorString;
+}
+
+std::vector<Transform>
+RDFTransformFactoryImpl::getTransforms(ProgressReporter *reporter)
+{
+    std::vector<Transform> transforms;
+
+    SimpleSPARQLQuery query
+        (QString
+         (
+             " PREFIX vamp: <http://purl.org/ontology/vamp/> "
+
+             " SELECT ?transform ?plugin ?output ?program "
+             "        ?step_size ?block_size ?window_type "
+             "        ?sample_rate ?start ?duration "
+
+             " FROM <%1> "
+
+             " WHERE { "
+             "   ?transform a vamp:Transform ; "
+             "              vamp:plugin ?plugin . "
+             "   OPTIONAL { ?transform vamp:output ?output } . "
+             "   OPTIONAL { ?transform vamp:program ?program } . "
+             "   OPTIONAL { ?transform vamp:step_size ?step_size } . "
+             "   OPTIONAL { ?transform vamp:block_size ?block_size } . "
+             "   OPTIONAL { ?transform vamp:window_type ?window_type } . "
+             "   OPTIONAL { ?transform vamp:sample_rate ?sample_rate } . "
+             "   OPTIONAL { ?transform vamp:start ?start } . "
+             "   OPTIONAL { ?transform vamp:duration ?duration } "
+             " } "
+             )
+         .arg(m_urlString));
+
+    SimpleSPARQLQuery::ResultList results = query.execute();
+
+    if (!query.isOK()) {
+        m_errorString = query.getErrorString();
+        return transforms;
+    }
+
+    if (query.wasCancelled()) {
+        m_errorString = "Query cancelled";
+        return transforms;
+    }
+
+    PluginRDFIndexer *indexer = PluginRDFIndexer::getInstance();
+
+    for (int i = 0; i < results.size(); ++i) {
+
+        SimpleSPARQLQuery::KeyValueMap &result = results[i];
+
+        QString transformUri = result["transform"].value;
+        QString pluginUri = result["plugin"].value;
+
+        QString pluginId = indexer->getIdForPluginURI(pluginUri);
+
+        if (pluginId == "") {
+            cerr << "RDFTransformFactory: WARNING: Unknown plugin <"
+                 << pluginUri.toStdString() << "> for transform <"
+                 << transformUri.toStdString() << ">" << endl;
+            continue;
+        }
+
+        Transform transform;
+        transform.setPluginIdentifier(pluginId);
+        
+        if (result["output"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setOutput(result["output"].value);
+        }
+
+        if (result["program"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setProgram(result["program"].value);
+        }
+        
+        if (result["step_size"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setStepSize(result["step_size"].value.toUInt());
+        }
+        
+        if (result["block_size"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setBlockSize(result["block_size"].value.toUInt());
+        }
+        
+        if (result["window_type"].type == SimpleSPARQLQuery::LiteralValue) {
+            cerr << "NOTE: can't handle window type yet (value is \""
+                 << result["window_type"].value.toStdString() << "\")" << endl;
+        }
+        
+        if (result["sample_rate"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setStepSize(result["sample_rate"].value.toFloat());
+        }
+
+        if (result["start"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setStartTime(RealTime::fromXsdDuration
+                                   (result["start"].value.toStdString()));
+        }
+
+        if (result["duration"].type == SimpleSPARQLQuery::LiteralValue) {
+            transform.setDuration(RealTime::fromXsdDuration
+                                  (result["duration"].value.toStdString()));
+        }
+
+        SimpleSPARQLQuery paramQuery
+            (QString
+             (
+                 " PREFIX vamp: <http://purl.org/ontology/vamp/> "
+
+                 " SELECT ?param_id ?param_value "
+
+                 " FROM <%1> "
+
+                 " WHERE { "
+                 "   <%2> vamp:parameter ?param . "
+                 "   ?param vamp:identifier ?param_id ; "
+                 "          vamp:value ?param_value "
+                 " } "
+                 )
+             .arg(m_urlString)
+             .arg(transformUri));
+        
+        SimpleSPARQLQuery::ResultList paramResults = paramQuery.execute();
+
+        if (!paramQuery.isOK()) {
+            m_errorString = paramQuery.getErrorString();
+            return transforms;
+        }
+
+        if (paramQuery.wasCancelled()) {
+            m_errorString = "Query cancelled";
+            return transforms;
+        }
+
+        for (int j = 0; j < paramResults.size(); ++j) {
+
+            QString paramId = paramResults[j]["param_id"].value;
+            QString paramValue = paramResults[j]["param_value"].value;
+
+            if (paramId == "" || paramValue == "") continue;
+
+            transform.setParameter(paramId, paramValue.toFloat());
+        }
+
+        cerr << "RDFTransformFactory: NOTE: Transform is: " << endl;
+        cerr << transform.toXmlString().toStdString() << endl;
+
+        transforms.push_back(transform);
+    }
+        
+    return transforms;
+}
+