annotate transform/FileFeatureWriter.h @ 876:47aa3aeb687b tonioni

For outputs with unknown bin count or multiple bins with variable sample rate, create additional output models for bins beyond the first
author Chris Cannam
date Wed, 29 Jan 2014 09:31:22 +0000
parents 608b4dc5ff34
children 06579b8ffb7b
rev   line source
Chris@498 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
Chris@498 2
Chris@498 3 /*
Chris@498 4 Sonic Visualiser
Chris@498 5 An audio file viewer and annotation editor.
Chris@498 6
Chris@498 7 Sonic Annotator
Chris@498 8 A utility for batch feature extraction from audio files.
Chris@498 9
Chris@498 10 Mark Levy, Chris Sutton and Chris Cannam, Queen Mary, University of London.
Chris@498 11 Copyright 2007-2008 QMUL.
Chris@498 12
Chris@498 13 This program is free software; you can redistribute it and/or
Chris@498 14 modify it under the terms of the GNU General Public License as
Chris@498 15 published by the Free Software Foundation; either version 2 of the
Chris@498 16 License, or (at your option) any later version. See the file
Chris@498 17 COPYING included with this distribution for more information.
Chris@498 18 */
Chris@498 19
Chris@498 20 #ifndef _FILE_FEATURE_WRITER_H_
Chris@498 21 #define _FILE_FEATURE_WRITER_H_
Chris@498 22
Chris@498 23 #include <string>
Chris@498 24 #include <map>
Chris@498 25 #include <set>
Chris@498 26
Chris@498 27 #include "FeatureWriter.h"
Chris@498 28
Chris@498 29 using std::string;
Chris@498 30 using std::map;
Chris@498 31 using std::set;
Chris@498 32 using std::pair;
Chris@498 33
Chris@498 34 class QTextStream;
Chris@498 35 class QFile;
Chris@498 36
Chris@498 37 class FileFeatureWriter : public FeatureWriter
Chris@498 38 {
Chris@498 39 public:
Chris@498 40 virtual ~FileFeatureWriter();
Chris@498 41
Chris@498 42 virtual ParameterList getSupportedParameters() const;
Chris@498 43 virtual void setParameters(map<string, string> &params);
Chris@498 44
Chris@625 45 virtual void testOutputFile(QString trackId, TransformId transformId);
Chris@515 46 virtual void flush();
Chris@531 47 virtual void finish();
Chris@515 48
Chris@498 49 protected:
Chris@498 50 enum FileWriteSupport {
Chris@498 51 SupportOneFilePerTrackTransform = 1,
Chris@498 52 SupportOneFilePerTrack = 2,
Chris@498 53 SupportOneFileTotal = 4
Chris@498 54 };
Chris@498 55
Chris@498 56 FileFeatureWriter(int support, QString extension);
Chris@498 57 QTextStream *getOutputStream(QString, TransformId);
Chris@498 58
Chris@498 59 typedef pair<QString, TransformId> TrackTransformPair;
Chris@498 60 typedef map<TrackTransformPair, QFile *> FileMap;
Chris@498 61 typedef map<QFile *, QTextStream *> FileStreamMap;
Chris@498 62 FileMap m_files;
Chris@498 63 FileStreamMap m_streams;
Chris@512 64 QTextStream *m_prevstream;
Chris@498 65
Chris@498 66 QString getOutputFilename(QString, TransformId);
Chris@498 67 QFile *getOutputFile(QString, TransformId);
Chris@591 68
Chris@591 69 // subclass can implement this to be called before file is opened for append
Chris@591 70 virtual void reviewFileForAppending(QString filename) { }
Chris@498 71
Chris@498 72 int m_support;
Chris@498 73 QString m_extension;
Chris@498 74 QString m_baseDir;
Chris@498 75 bool m_manyFiles;
Chris@498 76 QString m_singleFileName;
Chris@498 77 bool m_stdout;
Chris@498 78 bool m_append;
Chris@498 79 bool m_force;
Chris@498 80 };
Chris@498 81
Chris@498 82 #endif