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 Chris@31: Chris@32: #include Chris@32: #include Chris@41: #include 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@246: class FlattenDynamics; 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@316: enum ProcessingMode { // ordered so draft==0 and hq==1 as in prior releases Chris@316: DraftMode = 0, Chris@316: HighQualityMode = 1, Chris@316: LiveMode = 2, Chris@316: }; Chris@316: Chris@31: protected: Chris@161: const std::vector m_instruments; Chris@298: const std::vector m_liveInstruments; Chris@161: Chris@298: const InstrumentPack &getPack(int instrument) const { Chris@298: if (m_mode == LiveMode) { Chris@298: return m_liveInstruments[instrument]; Chris@298: } else { Chris@298: return m_instruments[instrument]; Chris@298: } Chris@298: } Chris@298: Chris@31: Resampler *m_resampler; Chris@246: FlattenDynamics *m_flattener; Chris@145: CQSpectrogram *m_cq; Chris@32: Chris@297: ProcessingMode m_mode; Chris@297: Chris@166: bool m_fineTuning; Chris@162: int m_instrument; Chris@165: int m_colsPerSec; Chris@110: Chris@32: typedef vector > Grid; Chris@32: Chris@41: vector *> m_postFilter; Chris@55: vector > m_pianoRoll; Chris@168: vector > m_pianoRollShifts; Chris@252: map m_inputGains; Chris@41: Chris@32: Grid preProcess(const Grid &); Chris@166: Chris@311: std::pair, vector > applyEM(const InstrumentPack &pack, Chris@311: const vector &column, Chris@311: bool wantShifts); Chris@311: Chris@294: vector postProcess(const vector &pitches, Chris@294: const vector &bestShifts, Chris@294: bool wantShifts); // -> piano roll column Chris@166: Chris@319: std::pair noteTrack(int shiftCount); // notes, onsets Chris@166: Chris@169: void emitNote(int start, int end, int note, int shiftCount, Chris@169: FeatureList ¬eFeatures); Chris@319: Chris@319: void emitOnset(int start, int note, int shiftCount, Chris@319: FeatureList ¬eFeatures); Chris@309: Chris@309: Vamp::RealTime getColumnTimestamp(int column); Chris@309: Chris@252: Feature makeNoteFeature(int start, int end, int note, int shift, Chris@320: int shiftCount, double strength); Chris@319: Feature makeOnsetFeature(int start, int note, int shift, Chris@320: int shiftCount, double strength); Chris@252: Chris@320: int getVelocityFor(double strength, int column); Chris@320: Chris@252: float getInputGainAt(Vamp::RealTime t); Chris@252: Chris@302: void insertTemplateFeatures(FeatureSet &); Chris@302: Chris@302: void transcribe(const Grid &, FeatureSet &); Chris@31: Chris@320: string getChromaName(int n) const; Chris@320: string getNoteName(int n, int shift, int shiftCount) const; Chris@320: float getNoteFrequency(int n, int shift, int shiftCount) const; Chris@38: Chris@31: int m_blockSize; Chris@32: int m_columnCount; Chris@272: int m_resampledCount; Chris@40: Vamp::RealTime m_startTime; Chris@313: bool m_haveStartTime; Chris@32: Chris@32: mutable int m_notesOutputNo; Chris@319: mutable int m_onsetsOutputNo; Chris@178: mutable int m_fcqOutputNo; Chris@294: mutable int m_pitchOutputNo; Chris@302: mutable int m_templateOutputNo; Chris@309: mutable int m_chromaOutputNo; Chris@31: }; Chris@31: Chris@31: #endif