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@352: #ifndef MAX_EM_THREADS Chris@352: #define MAX_EM_THREADS 8 Chris@352: #endif Chris@352: 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@341: enum ProcessingMode { Chris@341: LiveMode = 0, Chris@316: HighQualityMode = 1, 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@325: set m_current; Chris@41: Chris@32: Grid preProcess(const Grid &); Chris@166: Chris@311: std::pair, vector > applyEM(const InstrumentPack &pack, Chris@336: const vector &column); Chris@311: Chris@321: void postProcess(const vector &pitches, Chris@336: const vector &bestShifts); // -> piano roll column Chris@166: Chris@336: struct FeatureChunk { Chris@336: FeatureList notes; Chris@336: FeatureList onsets; Chris@336: FeatureList onOffsets; Chris@336: }; Chris@166: Chris@336: int getShiftCount() const; Chris@336: Chris@336: FeatureChunk noteTrack(); // notes, on/offsets Chris@336: Chris@343: void emitNoteAndOffset(int start, int end, int note, Chris@343: FeatureList ¬eFeatures, Chris@343: FeatureList &onOffsetFeatures); Chris@319: Chris@336: void emitOnset(int start, int note, Chris@336: FeatureList &onOffsetFeatures); Chris@336: Chris@309: Vamp::RealTime getColumnTimestamp(int column); Chris@309: Chris@336: Feature makeNoteFeature(int start, int end, int note, int shift, double strength); Chris@336: Feature makeOnsetFeature(int start, int note, int shift, double strength); Chris@336: Feature makeOffsetFeature(int col, int note, int shift); 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@336: string getNoteName(int n, int shift) const; Chris@336: float getNoteFrequency(int n, int shift) 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@336: mutable int m_onOffsetsOutputNo; 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