lbajardsilogic@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
lbajardsilogic@0: 
lbajardsilogic@0: /*
lbajardsilogic@0:     Sonic Visualiser
lbajardsilogic@0:     An audio file viewer and annotation editor.
lbajardsilogic@0:     Centre for Digital Music, Queen Mary, University of London.
lbajardsilogic@0:     This file copyright 2006 Chris Cannam and QMUL.
lbajardsilogic@0:     
lbajardsilogic@0:     This program is free software; you can redistribute it and/or
lbajardsilogic@0:     modify it under the terms of the GNU General Public License as
lbajardsilogic@0:     published by the Free Software Foundation; either version 2 of the
lbajardsilogic@0:     License, or (at your option) any later version.  See the file
lbajardsilogic@0:     COPYING included with this distribution for more information.
lbajardsilogic@0: */
lbajardsilogic@0: 
lbajardsilogic@0: #ifndef _FEATURE_EXTRACTION_PLUGIN_TRANSFORM_H_
lbajardsilogic@0: #define _FEATURE_EXTRACTION_PLUGIN_TRANSFORM_H_
lbajardsilogic@0: 
lbajardsilogic@0: #include "PluginTransform.h"
lbajardsilogic@0: 
lbajardsilogic@0: class DenseTimeValueModel;
lbajardsilogic@0: 
lbajardsilogic@0: class FeatureExtractionPluginTransform : public PluginTransform
lbajardsilogic@0: {
lbajardsilogic@0:     Q_OBJECT
lbajardsilogic@0: 
lbajardsilogic@0: public:
lbajardsilogic@0:     FeatureExtractionPluginTransform(Model *inputModel,
lbajardsilogic@0: 				     QString plugin,
lbajardsilogic@0:                                      const ExecutionContext &context,
lbajardsilogic@0:                                      QString configurationXml = "",
lbajardsilogic@0: 				     QString outputName = "");
lbajardsilogic@0:     virtual ~FeatureExtractionPluginTransform();
lbajardsilogic@0: 
lbajardsilogic@0: protected:
lbajardsilogic@0:     virtual void run();
lbajardsilogic@0: 
lbajardsilogic@0:     Vamp::Plugin *m_plugin;
lbajardsilogic@0:     Vamp::Plugin::OutputDescriptor *m_descriptor;
lbajardsilogic@0:     int m_outputFeatureNo;
lbajardsilogic@0: 
lbajardsilogic@0:     void addFeature(size_t blockFrame,
lbajardsilogic@0: 		    const Vamp::Plugin::Feature &feature);
lbajardsilogic@0: 
lbajardsilogic@0:     void setCompletion(int);
lbajardsilogic@0: 
lbajardsilogic@0:     void getFrames(int channel, int channelCount,
lbajardsilogic@0:                    long startFrame, long size, float *buffer);
lbajardsilogic@0: 
lbajardsilogic@0:     // just casts
lbajardsilogic@0:     DenseTimeValueModel *getInput();
lbajardsilogic@0:     template <typename ModelClass> ModelClass *getOutput() {
lbajardsilogic@0: 	ModelClass *mc = dynamic_cast<ModelClass *>(m_output);
lbajardsilogic@0: 	if (!mc) {
lbajardsilogic@0: 	    std::cerr << "FeatureExtractionPluginTransform::getOutput: Output model not conformable" << std::endl;
lbajardsilogic@0: 	}
lbajardsilogic@0: 	return mc;
lbajardsilogic@0:     }
lbajardsilogic@0: };
lbajardsilogic@0: 
lbajardsilogic@0: #endif
lbajardsilogic@0: