annotate transform/FileFeatureWriter.h @ 625:608b4dc5ff34

* Improve behaviour when output file write fails (i.e. fail earlier)
author Chris Cannam
date Fri, 21 May 2010 10:21:18 +0000
parents 7065e921f122
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