Chris@0
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
Chris@0
|
2
|
Chris@0
|
3 /*
|
Chris@0
|
4 Sonic Visualiser
|
Chris@0
|
5 An audio file viewer and annotation editor.
|
Chris@0
|
6 Centre for Digital Music, Queen Mary, University of London.
|
Chris@0
|
7 This file copyright 2006 Chris Cannam.
|
Chris@0
|
8
|
Chris@0
|
9 This program is free software; you can redistribute it and/or
|
Chris@0
|
10 modify it under the terms of the GNU General Public License as
|
Chris@0
|
11 published by the Free Software Foundation; either version 2 of the
|
Chris@0
|
12 License, or (at your option) any later version. See the file
|
Chris@0
|
13 COPYING included with this distribution for more information.
|
Chris@0
|
14 */
|
Chris@0
|
15
|
Chris@0
|
16 #ifndef _FEATURE_EXTRACTION_PLUGIN_TRANSFORM_H_
|
Chris@0
|
17 #define _FEATURE_EXTRACTION_PLUGIN_TRANSFORM_H_
|
Chris@0
|
18
|
Chris@0
|
19 #include "Transform.h"
|
Chris@0
|
20
|
Chris@26
|
21 #include "base/Window.h"
|
Chris@26
|
22
|
Chris@0
|
23 #include "vamp-sdk/Plugin.h"
|
Chris@0
|
24
|
Chris@0
|
25 class DenseTimeValueModel;
|
Chris@0
|
26
|
Chris@0
|
27 class FeatureExtractionPluginTransform : public Transform
|
Chris@0
|
28 {
|
Chris@0
|
29 public:
|
Chris@0
|
30 FeatureExtractionPluginTransform(Model *inputModel,
|
Chris@0
|
31 QString plugin,
|
Chris@0
|
32 int channel,
|
Chris@0
|
33 QString configurationXml = "",
|
Chris@26
|
34 QString outputName = "",
|
Chris@26
|
35 size_t stepSize = 0,
|
Chris@26
|
36 size_t blockSize = 0,
|
Chris@26
|
37 WindowType windowType = HanningWindow);
|
Chris@0
|
38 virtual ~FeatureExtractionPluginTransform();
|
Chris@0
|
39
|
Chris@0
|
40 protected:
|
Chris@0
|
41 virtual void run();
|
Chris@0
|
42
|
Chris@0
|
43 Vamp::Plugin *m_plugin;
|
Chris@0
|
44 int m_channel;
|
Chris@0
|
45 size_t m_stepSize;
|
Chris@0
|
46 size_t m_blockSize;
|
Chris@26
|
47 WindowType m_windowType;
|
Chris@0
|
48 Vamp::Plugin::OutputDescriptor *m_descriptor;
|
Chris@0
|
49 int m_outputFeatureNo;
|
Chris@0
|
50
|
Chris@0
|
51 void addFeature(size_t blockFrame,
|
Chris@0
|
52 const Vamp::Plugin::Feature &feature);
|
Chris@0
|
53
|
Chris@0
|
54 void setCompletion(int);
|
Chris@0
|
55
|
Chris@0
|
56 void getFrames(int channel, int channelCount,
|
Chris@0
|
57 long startFrame, long size, float *buffer);
|
Chris@0
|
58
|
Chris@0
|
59 // just casts
|
Chris@0
|
60 DenseTimeValueModel *getInput();
|
Chris@0
|
61 template <typename ModelClass> ModelClass *getOutput() {
|
Chris@0
|
62 ModelClass *mc = dynamic_cast<ModelClass *>(m_output);
|
Chris@0
|
63 if (!mc) {
|
Chris@0
|
64 std::cerr << "FeatureExtractionPluginTransform::getOutput: Output model not conformable" << std::endl;
|
Chris@0
|
65 }
|
Chris@0
|
66 return mc;
|
Chris@0
|
67 }
|
Chris@0
|
68 };
|
Chris@0
|
69
|
Chris@0
|
70 #endif
|
Chris@0
|
71
|