Mercurial > hg > pyin
diff Yin.h @ 0:99bac62ee2da
added PYIN sources, should be compileable
author | matthiasm |
---|---|
date | Wed, 27 Nov 2013 11:59:49 +0000 |
parents | |
children | 5945b8905d1f |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/Yin.h Wed Nov 27 11:59:49 2013 +0000 @@ -0,0 +1,80 @@ +/* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */ +/* + This file is Copyright (c) 2012 Chris Cannam + + Permission is hereby granted, free of charge, to any person + obtaining a copy of this software and associated documentation + files (the "Software"), to deal in the Software without + restriction, including without limitation the rights to use, copy, + modify, merge, publish, distribute, sublicense, and/or sell copies + of the Software, and to permit persons to whom the Software is + furnished to do so, subject to the following conditions: + + The above copyright notice and this permission notice shall be + included in all copies or substantial portions of the Software. + + THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND + NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR + ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF + CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION + WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. +*/ + +#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); + 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 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_removeUnvoiced; +}; + +#endif