view Yin.h @ 137:109c3a2ad930 vamp-fft-revision

Make use of new Vamp FFT interface. This reduces the runtime of the regression test from 5.7 to 2.2 seconds on this machine, but it does need the right version of the SDK, which is currently only available in the vampipe branch.

/* -*- 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 YinUtil;

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;
    YinUtil *m_yinUtil;
};

#endif