Chris@0: /* -*- c-basic-offset: 4 -*-  vi:set ts=8 sts=4 sw=4: */
Chris@0: 
Chris@0: /*
Chris@0:     A waveform viewer and audio annotation editor.
Chris@2:     Chris Cannam, Queen Mary University of London, 2005-2006
Chris@0:     
Chris@0:     This is experimental software.  Not for distribution.
Chris@0: */
Chris@0: 
Chris@0: #ifndef _FEATURE_EXTRACTION_PLUGIN_TRANSFORM_H_
Chris@0: #define _FEATURE_EXTRACTION_PLUGIN_TRANSFORM_H_
Chris@0: 
Chris@0: #include "Transform.h"
Chris@0: #include "FeatureExtractionPlugin.h"
Chris@0: 
Chris@0: class DenseTimeValueModel;
Chris@0: 
Chris@0: class FeatureExtractionPluginTransform : public Transform
Chris@0: {
Chris@0: public:
Chris@0:     FeatureExtractionPluginTransform(Model *inputModel,
Chris@0: 				     QString plugin,
Chris@0: 				     QString outputName = "");
Chris@0:     virtual ~FeatureExtractionPluginTransform();
Chris@0: 
Chris@0: protected:
Chris@0:     virtual void run();
Chris@0: 
Chris@0:     FeatureExtractionPlugin *m_plugin;
Chris@0:     FeatureExtractionPlugin::OutputDescriptor *m_descriptor;
Chris@0:     int m_outputFeatureNo;
Chris@0: 
Chris@0:     void addFeature(size_t blockFrame,
Chris@0: 		    const FeatureExtractionPlugin::Feature &feature);
Chris@0: 
Chris@0:     void setCompletion(int);
Chris@0: 
Chris@0:     // just casts
Chris@0:     DenseTimeValueModel *getInput();
Chris@0:     template <typename ModelClass> ModelClass *getOutput() {
Chris@0: 	ModelClass *mc = dynamic_cast<ModelClass *>(m_output);
Chris@0: 	if (!mc) {
Chris@0: 	    std::cerr << "FeatureExtractionPluginTransform::getOutput: Output model not conformable" << std::endl;
Chris@0: 	}
Chris@0: 	return mc;
Chris@0:     }
Chris@0: };
Chris@0: 
Chris@0: #endif
Chris@0: