matthiasm@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@9: 
matthiasm@0: /*
Chris@9:     pYIN - A fundamental frequency estimator for monophonic audio
Chris@9:     Centre for Digital Music, Queen Mary, University of London.
Chris@9:     
Chris@9:     This program is free software; you can redistribute it and/or
Chris@9:     modify it under the terms of the GNU General Public License as
Chris@9:     published by the Free Software Foundation; either version 2 of the
Chris@9:     License, or (at your option) any later version.  See the file
Chris@9:     COPYING included with this distribution for more information.
matthiasm@0: */
matthiasm@0: 
matthiasm@0: #ifndef _YINUTIL_H_
matthiasm@0: #define _YINUTIL_H_
matthiasm@0: 
matthiasm@0: #include <cmath>
matthiasm@0: 
matthiasm@0: #include <iostream>
matthiasm@0: #include <vector>
matthiasm@0: #include <exception>
matthiasm@0: 
Chris@136: #include "vamp-sdk/FFT.h"
Chris@136: 
matthiasm@0: using std::vector;
matthiasm@0: 
matthiasm@0: class YinUtil
matthiasm@0: {
matthiasm@0: public:
Chris@136:     YinUtil(size_t yinBufferSize);
Chris@136:     ~YinUtil();
Chris@136:     
Chris@136:     double sumSquare(const double *in, const size_t startInd, const size_t endInd);
Chris@136:     void difference(const double *in, double *yinBuffer);
Chris@136:     void fastDifference(const double *in, double *yinBuffer);
Chris@136:     void slowDifference(const double *in, double *yinBuffer);
Chris@136:     void cumulativeDifference(double *yinBuffer);
Chris@136:     int absoluteThreshold(const double *yinBuffer, const double thresh);
Chris@136:     vector<double> yinProb(const double *yinBuffer, const size_t prior,
Chris@136:                            size_t minTau = 0, size_t maxTau = 0);
Chris@136:     double parabolicInterpolation(const double *yinBuffer, const size_t tau);
Chris@136: 
Chris@136: private:
Chris@136:     const size_t m_yinBufferSize;
Chris@136:     Vamp::FFTReal m_fft;
matthiasm@0: };
matthiasm@0: 
matthiasm@0: #endif