diff options
Diffstat (limited to 'libs/vamp-pyin/Yin.h')
-rw-r--r-- | libs/vamp-pyin/Yin.h | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/libs/vamp-pyin/Yin.h b/libs/vamp-pyin/Yin.h new file mode 100644 index 0000000000..0fb151a6a5 --- /dev/null +++ b/libs/vamp-pyin/Yin.h @@ -0,0 +1,71 @@ +/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ + +/* + pYIN - A fundamental frequency estimator for monophonic audio + Centre for Digital Music, Queen Mary, University of London. + + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License as + published by the Free Software Foundation; either version 2 of the + License, or (at your option) any later version. See the file + COPYING included with this distribution for more information. +*/ + +#ifndef _YIN_H_ +#define _YIN_H_ + +#include "vamp-sdk/FFT.h" +#include "MeanFilter.h" + +#include <cmath> + +#include <iostream> +#include <vector> +#include <exception> + +using std::vector; +using std::pair; + + + +class Yin +{ +public: + Yin(size_t frameSize, size_t inputSampleRate, double thresh = 0.2, bool fast = true); + virtual ~Yin(); + + struct YinOutput { + double f0; + double periodicity; + double rms; + vector<double> salience; + vector<pair<double, double> > freqProb; + YinOutput() : f0(0), periodicity(0), rms(0), + salience(vector<double>(0)), freqProb(vector<pair<double, double> >(0)) { } + YinOutput(double _f, double _p, double _r) : + f0(_f), periodicity(_p), rms(_r), + salience(vector<double>(0)), freqProb(vector<pair<double, double> >(0)) { } + YinOutput(double _f, double _p, double _r, vector<double> _salience) : + f0(_f), periodicity(_p), rms(_r), salience(_salience), + freqProb(vector<pair<double, double> >(0)) { } + }; + + int setThreshold(double parameter); + int setThresholdDistr(float parameter); + int setFrameSize(size_t frameSize); + int setFast(bool fast); + // int setRemoveUnvoiced(bool frameSize); + YinOutput process(const double *in) const; + YinOutput processProbabilisticYin(const double *in) const; + +private: + mutable size_t m_frameSize; + mutable size_t m_inputSampleRate; + mutable double m_thresh; + mutable size_t m_threshDistr; + mutable size_t m_yinBufferSize; + mutable bool m_fast; + // mutable bool m_removeUnvoiced; +}; + +#endif |