Chris@320: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@320: 
Chris@320: /*
Chris@320:     Sonic Visualiser
Chris@320:     An audio file viewer and annotation editor.
Chris@320:     Centre for Digital Music, Queen Mary, University of London.
Chris@320:     This file copyright 2006 Chris Cannam and QMUL.
Chris@320:     
Chris@320:     This program is free software; you can redistribute it and/or
Chris@320:     modify it under the terms of the GNU General Public License as
Chris@320:     published by the Free Software Foundation; either version 2 of the
Chris@320:     License, or (at your option) any later version.  See the file
Chris@320:     COPYING included with this distribution for more information.
Chris@320: */
Chris@320: 
Chris@1581: #ifndef SV_FEATURE_EXTRACTION_MODEL_TRANSFORMER_H
Chris@1581: #define SV_FEATURE_EXTRACTION_MODEL_TRANSFORMER_H
Chris@320: 
Chris@350: #include "ModelTransformer.h"
Chris@350: 
Chris@361: #include <QString>
Chris@1211: #include <QMutex>
Chris@1211: #include <QWaitCondition>
Chris@361: 
Chris@475: #include <vamp-hostsdk/Plugin.h>
Chris@350: 
Chris@350: #include <iostream>
Chris@876: #include <map>
Chris@320: 
Chris@320: class DenseTimeValueModel;
Chris@876: class SparseTimeValueModel;
Chris@320: 
Chris@1236: class FeatureExtractionModelTransformer : public ModelTransformer // + is a Thread
Chris@320: {
Chris@320:     Q_OBJECT
Chris@320: 
Chris@320: public:
Chris@350:     FeatureExtractionModelTransformer(Input input,
Chris@859:                                       const Transform &transform);
Chris@848: 
Chris@1739:     /**
Chris@1739:      * Obtain outputs for a set of transforms that all use the same
Chris@1739:      * plugin and input (but with different outputs). i.e. run the
Chris@1739:      * plugin once only and collect more than one output from it.
Chris@1739:      */
Chris@848:     FeatureExtractionModelTransformer(Input input,
Chris@859:                                       const Transforms &relatedTransforms);
gyorgyf@786: 
Chris@331:     virtual ~FeatureExtractionModelTransformer();
Chris@320: 
Chris@877:     // ModelTransformer method, retrieve the additional models
Chris@1580:     Models getAdditionalOutputModels() override;
Chris@1580:     bool willHaveAdditionalOutputModels() override;
Chris@877: 
Chris@320: protected:
Chris@849:     bool initialise();
Chris@1237:     void deinitialise();
Chris@849: 
Chris@1580:     void run() override;
Chris@320: 
Chris@1830:     std::shared_ptr<Vamp::Plugin> m_plugin;
Chris@1739: 
Chris@1739:     // descriptors per transform
Chris@1739:     std::vector<Vamp::Plugin::OutputDescriptor> m_descriptors;
Chris@1739: 
Chris@1739:     // to assign times to FixedSampleRate features
Chris@1739:     std::vector<int> m_fixedRateFeatureNos;
Chris@1739: 
Chris@1739:     // list of plugin output indexes required for this group of transforms
Chris@1739:     std::vector<int> m_outputNos;
Chris@320: 
Chris@876:     void createOutputModels(int n);
Chris@876: 
Chris@1739:     // map from transformNo -> necessity
Chris@1739:     std::map<int, bool> m_needAdditionalModels;
Chris@1739: 
Chris@1739:     // map from transformNo -> binNo -> SparseTimeValueModel id
Chris@1739:     typedef std::map<int, std::map<int, ModelId> > AdditionalModelMap;
Chris@1739:     
Chris@876:     AdditionalModelMap m_additionalModels;
Chris@1739:     
Chris@1739:     ModelId getAdditionalModel(int transformNo, int binNo);
Chris@876: 
Chris@850:     void addFeature(int n,
Chris@1039:                     sv_frame_t blockFrame,
Chris@1429:                     const Vamp::Plugin::Feature &feature);
Chris@320: 
Chris@850:     void setCompletion(int, int);
Chris@320: 
Chris@1039:     void getFrames(int channelCount, sv_frame_t startFrame, sv_frame_t size,
Chris@363:                    float **buffer);
Chris@320: 
Chris@1211:     bool m_haveOutputs;
Chris@1211:     QMutex m_outputMutex;
Chris@1211:     QWaitCondition m_outputsCondition;
Chris@1580:     void awaitOutputModels() override;
Chris@1211:     
Chris@1740:     template <typename T> bool isOutputType(int n) {
Chris@1740:         if (!ModelById::getAs<T>(m_outputs[n])) {
Chris@1740:             return false;
Chris@1740:         } else {
Chris@1740:             return true;
Chris@1740:         }
Chris@441:     }
Chris@441: 
Chris@1740:     template <typename T> bool setOutputCompletion(int n, int completion) {
Chris@1740:         auto model = ModelById::getAs<T>(m_outputs[n]);
Chris@1740:         if (!model) {
Chris@1740:             return false;
Chris@849:         } else {
Chris@1740:             model->setCompletion(completion, true);
Chris@1740:             return true;
Chris@849:         }
Chris@320:     }
Chris@320: };
Chris@320: 
Chris@320: #endif
Chris@320: