Chris@31: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@31: 
Chris@31: /*
Chris@31:   Silvet
Chris@31: 
Chris@31:   A Vamp plugin for note transcription.
Chris@31:   Centre for Digital Music, Queen Mary University of London.
Chris@31:     
Chris@31:   This program is free software; you can redistribute it and/or
Chris@31:   modify it under the terms of the GNU General Public License as
Chris@31:   published by the Free Software Foundation; either version 2 of the
Chris@31:   License, or (at your option) any later version.  See the file
Chris@31:   COPYING included with this distribution for more information.
Chris@31: */
Chris@31: 
Chris@31: #ifndef SILVET_H
Chris@31: #define SILVET_H
Chris@31: 
Chris@31: #include <vamp-sdk/Plugin.h>
Chris@31: 
Chris@32: #include <vector>
Chris@32: #include <string>
Chris@41: #include <set>
Chris@32: 
Chris@32: #include "maths/MedianFilter.h"
Chris@32: 
Chris@31: using std::string;
Chris@32: using std::vector;
Chris@41: using std::set;
Chris@31: 
Chris@31: class Resampler;
Chris@32: class CQInterpolated;
Chris@31: 
Chris@31: class Silvet : public Vamp::Plugin
Chris@31: {
Chris@31: public:
Chris@31:     Silvet(float inputSampleRate);
Chris@31:     virtual ~Silvet();
Chris@31: 
Chris@31:     string getIdentifier() const;
Chris@31:     string getName() const;
Chris@31:     string getDescription() const;
Chris@31:     string getMaker() const;
Chris@31:     int getPluginVersion() const;
Chris@31:     string getCopyright() const;
Chris@31: 
Chris@31:     InputDomain getInputDomain() const;
Chris@31:     size_t getPreferredBlockSize() const;
Chris@31:     size_t getPreferredStepSize() const;
Chris@31:     size_t getMinChannelCount() const;
Chris@31:     size_t getMaxChannelCount() const;
Chris@31: 
Chris@31:     ParameterList getParameterDescriptors() const;
Chris@31:     float getParameter(string identifier) const;
Chris@31:     void setParameter(string identifier, float value);
Chris@31: 
Chris@31:     ProgramList getPrograms() const;
Chris@31:     string getCurrentProgram() const;
Chris@31:     void selectProgram(string name);
Chris@31: 
Chris@31:     OutputList getOutputDescriptors() const;
Chris@31: 
Chris@31:     bool initialise(size_t channels, size_t stepSize, size_t blockSize);
Chris@31:     void reset();
Chris@31: 
Chris@31:     FeatureSet process(const float *const *inputBuffers,
Chris@31:                        Vamp::RealTime timestamp);
Chris@31: 
Chris@31:     FeatureSet getRemainingFeatures();
Chris@31: 
Chris@31: protected:
Chris@31:     Resampler *m_resampler;
Chris@32:     CQInterpolated *m_cq;
Chris@32: 
Chris@32:     typedef vector<vector<double> > Grid;
Chris@32: 
Chris@41:     vector<MedianFilter<double> *> m_postFilter;
Chris@41:     vector<set<int> > m_pianoRoll;
Chris@41: 
Chris@32:     Grid preProcess(const Grid &);
Chris@41:     FeatureList postProcess(const vector<double> &);
Chris@34:     FeatureSet transcribe(const Grid &);
Chris@31: 
Chris@41:     string noteName(int n) const;
Chris@41:     float noteFrequency(int n) const;
Chris@38: 
Chris@31:     int m_blockSize;
Chris@32:     int m_columnCount;
Chris@32:     int m_reducedColumnCount;
Chris@40:     Vamp::RealTime m_startTime;
Chris@32: 
Chris@32:     mutable int m_notesOutputNo;
Chris@32:     mutable int m_cqOutputNo;
Chris@51:     mutable int m_fcqOutputNo;
Chris@38:     mutable int m_pitchOutputNo;
Chris@31: };
Chris@31: 
Chris@31: #endif