plugins/Notes.h

   1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
   2
   3 /*
   4     Vamp feature extraction plugins using Paul Brossier's Aubio library.
   5
   6     Centre for Digital Music, Queen Mary, University of London.
   7     This file copyright 2006 Chris Cannam.
   8
   9     This program is free software; you can redistribute it and/or
  10     modify it under the terms of the GNU General Public License as
  11     published by the Free Software Foundation; either version 2 of the
  12     License, or (at your option) any later version.  See the file
  13     COPYING included with this distribution for more information.
  14
  15 */
  16
  17 #ifndef _NOTES_PLUGIN_H_
  18 #define _NOTES_PLUGIN_H_
  19
  20 #include <vamp-sdk/Plugin.h>
  21 #include <aubio/aubio.h>
  22
  23 #include <deque>
  24
  25 class Notes : public Vamp::Plugin
  26 {
  27 public:
  28     Notes(float inputSampleRate);
  29     virtual ~Notes();
  30
  31     bool initialise(size_t channels, size_t stepSize, size_t blockSize);
  32     void reset();
  33
  34     InputDomain getInputDomain() const { return TimeDomain; }
  35
  36     std::string getName() const;
  37     std::string getDescription() const;
  38     std::string getMaker() const;
  39     int getPluginVersion() const;
  40     std::string getCopyright() const;
  41
  42     ParameterList getParameterDescriptors() const;
  43     float getParameter(std::string) const;
  44     void setParameter(std::string, float);
  45
  46     size_t getPreferredStepSize() const;
  47     size_t getPreferredBlockSize() const;
  48
  49     OutputList getOutputDescriptors() const;
  50
  51     FeatureSet process(const float *const *inputBuffers,
  52                        Vamp::RealTime timestamp);
  53
  54     FeatureSet getRemainingFeatures();
  55
  56 protected:
  57     fvec_t *m_ibuf;
  58     cvec_t *m_fftgrain;
  59     fvec_t *m_onset;
  60     aubio_pvoc_t *m_pv;
  61     aubio_pickpeak_t *m_peakpick;
  62     aubio_onsetdetection_t *m_onsetdet;
  63     aubio_onsetdetection_type m_onsettype;
  64     aubio_pitchdetection_t *m_pitchdet;
  65     aubio_pitchdetection_type m_pitchtype;
  66     aubio_pitchdetection_mode m_pitchmode;
  67     float m_threshold;
  68     float m_silence;
  69     size_t m_median;
  70     size_t m_stepSize;
  71     size_t m_blockSize;
  72     size_t m_channelCount;
  73     int m_minpitch;
  74     int m_maxpitch;
  75     bool m_wrapRange;
  76     bool m_avoidLeaps;
  77     std::deque<float> m_notebuf;
  78     size_t m_count;
  79     Vamp::RealTime m_delay;
  80     Vamp::RealTime m_currentOnset;
  81     Vamp::RealTime m_lastTimeStamp;
  82     float m_currentLevel;
  83     bool m_haveCurrent;
  84     int m_prevPitch;
  85
  86     void pushNote(FeatureSet &, const Vamp::RealTime &);
  87 };
  88
  89
  90 #endif