Chris@320
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
Chris@320
|
2
|
Chris@320
|
3 /*
|
Chris@320
|
4 Sonic Visualiser
|
Chris@320
|
5 An audio file viewer and annotation editor.
|
Chris@320
|
6 Centre for Digital Music, Queen Mary, University of London.
|
Chris@320
|
7 This file copyright 2006 Chris Cannam and QMUL.
|
Chris@320
|
8
|
Chris@320
|
9 This program is free software; you can redistribute it and/or
|
Chris@320
|
10 modify it under the terms of the GNU General Public License as
|
Chris@320
|
11 published by the Free Software Foundation; either version 2 of the
|
Chris@320
|
12 License, or (at your option) any later version. See the file
|
Chris@320
|
13 COPYING included with this distribution for more information.
|
Chris@320
|
14 */
|
Chris@320
|
15
|
Chris@328
|
16 #ifndef _FEATURE_EXTRACTION_PLUGIN_TRANSFORMER_H_
|
Chris@328
|
17 #define _FEATURE_EXTRACTION_PLUGIN_TRANSFORMER_H_
|
Chris@320
|
18
|
Chris@350
|
19 #include "ModelTransformer.h"
|
Chris@350
|
20
|
Chris@361
|
21 #include <QString>
|
Chris@361
|
22
|
Chris@475
|
23 #include <vamp-hostsdk/Plugin.h>
|
Chris@350
|
24
|
Chris@350
|
25 #include <iostream>
|
Chris@320
|
26
|
Chris@320
|
27 class DenseTimeValueModel;
|
Chris@320
|
28
|
Chris@350
|
29 class FeatureExtractionModelTransformer : public ModelTransformer
|
Chris@320
|
30 {
|
Chris@320
|
31 Q_OBJECT
|
Chris@320
|
32
|
Chris@320
|
33 public:
|
Chris@848
|
34 enum PreferredOutputModel {
|
Chris@848
|
35 NoteOutputModel,
|
Chris@848
|
36 FlexiNoteOutputModel,
|
Chris@848
|
37 UndefinedOutputModel = 255
|
Chris@848
|
38 };
|
gyorgyf@786
|
39
|
Chris@350
|
40 FeatureExtractionModelTransformer(Input input,
|
gyorgyf@786
|
41 const Transform &transform,
|
Chris@848
|
42 const PreferredOutputModel outputmodel);
|
Chris@848
|
43
|
Chris@848
|
44 // Obtain outputs for a set of transforms that all use the same
|
Chris@848
|
45 // plugin and input (but with different outputs). i.e. run the
|
Chris@848
|
46 // plugin once only and collect more than one output from it.
|
Chris@848
|
47 FeatureExtractionModelTransformer(Input input,
|
Chris@848
|
48 const Transforms &relatedTransforms,
|
Chris@848
|
49 const PreferredOutputModel outputmodel);
|
gyorgyf@786
|
50
|
Chris@331
|
51 virtual ~FeatureExtractionModelTransformer();
|
Chris@320
|
52
|
Chris@320
|
53 protected:
|
Chris@320
|
54 virtual void run();
|
Chris@320
|
55
|
Chris@320
|
56 Vamp::Plugin *m_plugin;
|
Chris@320
|
57 Vamp::Plugin::OutputDescriptor *m_descriptor;
|
Chris@779
|
58 int m_fixedRateFeatureNo; // to assign times to FixedSampleRate features
|
Chris@778
|
59 int m_outputNo;
|
Chris@822
|
60 PreferredOutputModel m_preferredOutputModel;
|
Chris@320
|
61
|
Chris@558
|
62 void createOutputModel();
|
Chris@558
|
63
|
Chris@320
|
64 void addFeature(size_t blockFrame,
|
Chris@320
|
65 const Vamp::Plugin::Feature &feature);
|
Chris@320
|
66
|
Chris@320
|
67 void setCompletion(int);
|
Chris@320
|
68
|
Chris@363
|
69 void getFrames(int channelCount, long startFrame, long size,
|
Chris@363
|
70 float **buffer);
|
Chris@320
|
71
|
Chris@320
|
72 // just casts
|
Chris@441
|
73
|
Chris@350
|
74 DenseTimeValueModel *getConformingInput();
|
Chris@441
|
75
|
Chris@441
|
76 template <typename ModelClass> bool isOutput() {
|
Chris@441
|
77 return dynamic_cast<ModelClass *>(m_output) != 0;
|
Chris@441
|
78 }
|
Chris@441
|
79
|
Chris@350
|
80 template <typename ModelClass> ModelClass *getConformingOutput() {
|
Chris@320
|
81 ModelClass *mc = dynamic_cast<ModelClass *>(m_output);
|
Chris@320
|
82 if (!mc) {
|
Chris@331
|
83 std::cerr << "FeatureExtractionModelTransformer::getOutput: Output model not conformable" << std::endl;
|
Chris@320
|
84 }
|
Chris@320
|
85 return mc;
|
Chris@320
|
86 }
|
Chris@320
|
87 };
|
Chris@320
|
88
|
Chris@320
|
89 #endif
|
Chris@320
|
90
|