Chris@21: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@21: 
Chris@21: /*
Chris@21:   Centre for Digital Music, Queen Mary University of London.
Chris@21:     
Chris@21:   This program is free software; you can redistribute it and/or
Chris@21:   modify it under the terms of the GNU General Public License as
Chris@21:   published by the Free Software Foundation; either version 2 of the
Chris@21:   License, or (at your option) any later version.  See the file
Chris@21:   COPYING included with this distribution for more information.
Chris@21: */
Chris@21: 
Chris@0: #ifndef TUNING_DIFFERENCE_H
Chris@0: #define TUNING_DIFFERENCE_H
Chris@0: 
Chris@0: #include <vamp-sdk/Plugin.h>
Chris@0: 
Chris@15: #include <cq/Chromagram.h>
Chris@13: 
Chris@13: #include <memory>
Chris@13: 
Chris@0: using std::string;
Chris@1: using std::vector;
Chris@0: 
Chris@0: class TuningDifference : public Vamp::Plugin
Chris@0: {
Chris@0: public:
Chris@0:     TuningDifference(float inputSampleRate);
Chris@0:     virtual ~TuningDifference();
Chris@0: 
Chris@0:     string getIdentifier() const;
Chris@0:     string getName() const;
Chris@0:     string getDescription() const;
Chris@0:     string getMaker() const;
Chris@0:     int getPluginVersion() const;
Chris@0:     string getCopyright() const;
Chris@0: 
Chris@0:     InputDomain getInputDomain() const;
Chris@0:     size_t getPreferredBlockSize() const;
Chris@0:     size_t getPreferredStepSize() const;
Chris@0:     size_t getMinChannelCount() const;
Chris@0:     size_t getMaxChannelCount() const;
Chris@0: 
Chris@0:     ParameterList getParameterDescriptors() const;
Chris@0:     float getParameter(string identifier) const;
Chris@0:     void setParameter(string identifier, float value);
Chris@0: 
Chris@0:     ProgramList getPrograms() const;
Chris@0:     string getCurrentProgram() const;
Chris@0:     void selectProgram(string name);
Chris@0: 
Chris@0:     OutputList getOutputDescriptors() const;
Chris@0: 
Chris@0:     bool initialise(size_t channels, size_t stepSize, size_t blockSize);
Chris@0:     void reset();
Chris@0: 
Chris@0:     FeatureSet process(const float *const *inputBuffers,
Chris@0:                        Vamp::RealTime timestamp);
Chris@0: 
Chris@0:     FeatureSet getRemainingFeatures();
Chris@0: 
Chris@0: protected:
Chris@13:     typedef vector<float> Signal;
Chris@13:     typedef vector<double> TFeature;
Chris@13: 
Chris@13:     int m_bpo;
Chris@15:     std::unique_ptr<Chromagram> m_refChroma;
Chris@15:     TFeature m_refTotals;
Chris@13:     TFeature m_refFeature;
Chris@13:     Signal m_other;
Chris@1:     int m_blockSize;
Chris@1:     int m_frameCount;
Chris@13: 
Chris@15:     Chromagram::Parameters paramsForTuningFrequency(double hz) const;
Chris@15:     TFeature computeFeatureFromTotals(const TFeature &totals) const;
Chris@13:     TFeature computeFeatureFromSignal(const Signal &signal, double hz) const;
Chris@13:     double featureDistance(const TFeature &other, int rotation = 0) const;
Chris@13:     int findBestRotation(const TFeature &other) const;
Chris@19:     std::pair<int, double> findFineFrequency(int coarseCents, double coarseScore);
Chris@13: 
Chris@13:     mutable std::map<string, int> m_outputs;
Chris@0: };
Chris@0: 
Chris@0: 
Chris@0: #endif