diff runner/AudioDBFeatureWriter.h @ 1:92911f967a16

* some reorganisation
author Chris Cannam
date Thu, 11 Dec 2008 10:26:12 +0000
parents AudioDBFeatureWriter.h@581b1b150a4d
children 18c42fbc1335
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/runner/AudioDBFeatureWriter.h	Thu Dec 11 10:26:12 2008 +0000
@@ -0,0 +1,61 @@
+/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
+
+/*
+    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 _AUDIO_DB_FEATURE_WRITER_H_
+#define _AUDIO_DB_FEATURE_WRITER_H_
+
+#include <string>
+#include <map>
+
+using std::string;
+using std::map;
+
+#include "transform/FeatureWriter.h"
+
+class AudioDBFeatureWriter : public FeatureWriter
+{
+public:
+    AudioDBFeatureWriter();
+    virtual ~AudioDBFeatureWriter();
+
+    virtual ParameterList getSupportedParameters() const;
+    virtual void setParameters(map<string, string> &params);
+
+    virtual void setCatalogueId(const string &);
+    virtual void setBaseDirectory(const string &);
+
+    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:
+    string catalogueId;
+    string baseDir;
+
+    static string catalogueIdParam;
+    static string baseDirParam;
+    
+    struct TrackStream;
+    map<string, TrackStream> dbfiles;
+    
+    bool openDBFile(QString trackid, const string& identifier);
+    bool replaceDBFile(QString trackid, const string& identifier);
+};
+
+#endif