Start overhaul for (current git, presumably later v0.4.0) aubio API
[vamp-aubio-plugins.git] / 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 #include "Types.h"
26
27 class Notes : public Vamp::Plugin
28 {
29 public:
30     Notes(float inputSampleRate, unsigned int apiVersion);
31     virtual ~Notes();
32
33     bool initialise(size_t channels, size_t stepSize, size_t blockSize);
34     void reset();
35
36     InputDomain getInputDomain() const { return TimeDomain; }
37
38     std::string getIdentifier() const;
39     std::string getName() const;
40     std::string getDescription() const;
41     std::string getMaker() const;
42     int getPluginVersion() const;
43     std::string getCopyright() const;
44
45     ParameterList getParameterDescriptors() const;
46     float getParameter(std::string) const;
47     void setParameter(std::string, float);
48
49     size_t getPreferredStepSize() const;
50     size_t getPreferredBlockSize() const;
51
52     OutputList getOutputDescriptors() const;
53
54     FeatureSet process(const float *const *inputBuffers,
55                        Vamp::RealTime timestamp);
56
57     FeatureSet getRemainingFeatures();
58
59 protected:
60     int m_apiVersion;
61     fvec_t *m_ibuf;
62     cvec_t *m_fftgrain;
63     fvec_t *m_onset;
64     aubio_pvoc_t *m_pv;
65     aubio_peakpicker_t *m_peakpick;
66     aubio_onset_t *m_onsetdet;
67     OnsetType m_onsettype;
68     aubio_pitch_t *m_pitchdet;
69     PitchType m_pitchtype;
70     float m_threshold;
71     float m_silence;
72     size_t m_median;
73     size_t m_stepSize;
74     size_t m_blockSize;
75     size_t m_channelCount;
76     int m_minpitch;
77     int m_maxpitch;
78     bool m_wrapRange;
79     bool m_avoidLeaps;
80     std::deque<float> m_notebuf;
81     size_t m_count;
82     Vamp::RealTime m_delay;
83     Vamp::RealTime m_currentOnset;
84     Vamp::RealTime m_lastTimeStamp;
85     float m_currentLevel;
86     bool m_haveCurrent;
87     int m_prevPitch;
88
89     void pushNote(FeatureSet &, const Vamp::RealTime &);
90 };
91
92
93 #endif