diff options
author | Robin Gareus <robin@gareus.org> | 2019-09-02 03:12:22 +0200 |
---|---|---|
committer | Robin Gareus <robin@gareus.org> | 2019-09-02 03:12:22 +0200 |
commit | 63994f3b820c8f0754ff59d0d09585405d87ae0e (patch) | |
tree | 4138d2f4b5d7e7c4ab0f371c08615b5d8fcc7538 /libs/vamp-pyin/Yin.h | |
parent | 1c8b6e1b4296b4fbabc258f9f94635390a319522 (diff) |
Include vamp-pyin
In preparation for captainMorgan's pitch analysis script.
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 |