Mercurial > hg > svcore
view transform/CSVFeatureWriter.h @ 1064:c9fdd9921146 tonioni
Implement FillGaps option in SparseModel export
author | Chris Cannam |
---|---|
date | Tue, 31 Mar 2015 11:54:57 +0100 |
parents | d954e03274e8 |
children | 870acd589a38 |
line wrap: on
line source
/* -*- 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 _CSV_FEATURE_WRITER_H_ #define _CSV_FEATURE_WRITER_H_ #include <string> #include <map> #include <set> #include <QString> #include "FileFeatureWriter.h" using std::string; using std::map; class QTextStream; class QFile; class CSVFeatureWriter : public FileFeatureWriter { public: CSVFeatureWriter(); virtual ~CSVFeatureWriter(); virtual string getDescription() const; virtual ParameterList getSupportedParameters() const; virtual void setParameters(map<string, string> ¶ms); virtual void write(QString trackid, const Transform &transform, const Vamp::Plugin::OutputDescriptor &output, const Vamp::Plugin::FeatureList &features, std::string summaryType = ""); virtual void finish(); virtual QString getWriterTag() const { return "csv"; } private: QString m_separator; bool m_sampleTiming; bool m_endTimes; bool m_forceEnd; bool m_omitFilename; QString m_prevPrintedTrackId; typedef pair<QString, Transform> DataId; // track id, transform typedef map<DataId, Vamp::Plugin::Feature> PendingFeatures; typedef map<DataId, std::string> PendingSummaryTypes; PendingFeatures m_pending; PendingSummaryTypes m_pendingSummaryTypes; void writeFeature(DataId, QTextStream &, const Vamp::Plugin::Feature &f, const Vamp::Plugin::Feature *optionalNextFeature, std::string summaryType); }; #endif