comparison rdf/RDFFeatureWriter.h @ 498:fdf5930b7ccc

* Bring FeatureWriter and RDFFeatureWriter into the fold (from Runner)
  so that we can use them to export features from SV as well

+/* -*- 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.
+
+    Sonic Annotator
+    A utility for batch feature extraction from audio files.
+
+    Mark Levy, Chris Sutton and Chris Cannam, Queen Mary, University of London.
+    Copyright 2007-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.
+*/
+
+#ifndef _RDF_FEATURE_WRITER_H_
+#define _RDF_FEATURE_WRITER_H_
+
+#include <string>
+#include <map>
+#include <set>
+
+#include <QString>
+
+#include "transform/FileFeatureWriter.h"
+
+#include "PluginRDFDescription.h"
+
+using std::string;
+using std::map;
+using std::set;
+using std::pair;
+
+class QTextStream;
+class QFile;
+
+class RDFFeatureWriter : public FileFeatureWriter
+{
+public:
+    RDFFeatureWriter();
+    virtual ~RDFFeatureWriter();
+
+    virtual ParameterList getSupportedParameters() const;
+    virtual void setParameters(map<string, string> &params);
+
+    virtual void write(QString trackid,
+                       const Transform &transform,
+                       const Vamp::Plugin::OutputDescriptor &output,
+                       const Vamp::Plugin::FeatureList &features,
+                       std::string summaryType = "");
+
+    virtual void finish();
+
+private:
+    typedef map<QString, PluginRDFDescription> RDFDescriptionMap; // by plugin id
+    RDFDescriptionMap m_rdfDescriptions;
+
+    void writePrefixes(QTextStream *);
+    void writeSignalDescription(QTextStream *, QString);
+    void writeLocalFeatureTypes(QTextStream *,
+                                const Transform &,
+                                const Vamp::Plugin::OutputDescriptor &,
+                                PluginRDFDescription &);
+
+    void writeSparseRDF(QTextStream *stream,
+                        const Transform &transform,
+                        const Vamp::Plugin::OutputDescriptor &output,
+                        const Vamp::Plugin::FeatureList &features,
+                        PluginRDFDescription &desc,
+                        QString timelineURI);
+
+    void writeDenseRDF(QTextStream *stream,
+                       const Transform &transform,
+                       const Vamp::Plugin::OutputDescriptor &output,
+                       const Vamp::Plugin::FeatureList &features,
+                       PluginRDFDescription &desc,
+                       QString signalURI,
+                       QString timelineURI);
+
+    set<QString> m_startedTrackIds;
+
+    map<QTextStream *, set<Transform> > m_startedStreamTransforms;
+
+    map<QString, QString> m_trackTimelineURIs;
+    map<QString, QString> m_trackSignalURIs;
+
+    map<Transform, QString> m_transformURIs;
+    map<Transform, QString> m_syntheticEventTypeURIs;
+    map<Transform, QString> m_syntheticSignalTypeURIs;
+
+    typedef pair<QString, Transform> StringTransformPair;
+    typedef pair<QTextStream *, QString> StreamBuffer;
+    map<StringTransformPair, StreamBuffer> m_openDenseFeatures; // signal URI + transform -> stream + text
+    QString m_suri;
+
+    bool m_plain;
+
+    uint64_t m_count;
+};
+
+#endif