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@152: #include "MedianFilter.h"
Chris@161: #include "Instruments.h"
Chris@32: 
Chris@31: using std::string;
Chris@32: using std::vector;
Chris@41: using std::set;
Chris@55: using std::map;
Chris@31: 
Chris@31: class Resampler;
Chris@145: class CQSpectrogram;
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@161:     const std::vector<InstrumentPack> m_instruments;
Chris@161: 
Chris@31:     Resampler *m_resampler;
Chris@145:     CQSpectrogram *m_cq;
Chris@32: 
Chris@110:     bool m_hqMode;
Chris@166:     bool m_fineTuning;
Chris@162:     int m_instrument;
Chris@165:     int m_colsPerSec;
Chris@110: 
Chris@32:     typedef vector<vector<double> > Grid;
Chris@32: 
Chris@41:     vector<MedianFilter<double> *> m_postFilter;
Chris@55:     vector<map<int, double> > m_pianoRoll;
Chris@168:     vector<map<int, int> > m_pianoRollShifts;
Chris@41: 
Chris@32:     Grid preProcess(const Grid &);
Chris@166: 
Chris@168:     void postProcess(const float *pitches,
Chris@168:                      const float *const *shifts,
Chris@168:                      int shiftCount,
Chris@168:                      double gain); // -> piano roll column
Chris@166: 
Chris@168:     FeatureList noteTrack(int shiftCount);
Chris@166: 
Chris@169:     void emitNote(int start, int end, int note, int shiftCount,
Chris@169:                   FeatureList &noteFeatures);
Chris@169: 
Chris@34:     FeatureSet transcribe(const Grid &);
Chris@31: 
Chris@41:     string noteName(int n) const;
Chris@169:     float noteFrequency(int n, int shift, int shiftCount) const;
Chris@38: 
Chris@31:     int m_blockSize;
Chris@32:     int m_columnCount;
Chris@40:     Vamp::RealTime m_startTime;
Chris@32: 
Chris@32:     mutable int m_notesOutputNo;
Chris@31: };
Chris@31: 
Chris@31: #endif