154 lines
4.2 KiB
C++
154 lines
4.2 KiB
C++
/* -*- 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.
|
|
*/
|
|
|
|
#include "Yin.h"
|
|
|
|
#include "vamp-sdk/FFT.h"
|
|
#include "MeanFilter.h"
|
|
#include "YinUtil.h"
|
|
|
|
#include <vector>
|
|
#include <cstdlib>
|
|
#include <cstdio>
|
|
#include <cmath>
|
|
#include <complex>
|
|
|
|
using std::vector;
|
|
|
|
Yin::Yin(size_t frameSize, size_t inputSampleRate, double thresh, bool fast) :
|
|
m_frameSize(frameSize),
|
|
m_inputSampleRate(inputSampleRate),
|
|
m_thresh(thresh),
|
|
m_threshDistr(2),
|
|
m_yinBufferSize(frameSize/2),
|
|
m_fast(fast)
|
|
{
|
|
if (frameSize & (frameSize-1)) {
|
|
// throw "N must be a power of two";
|
|
}
|
|
}
|
|
|
|
Yin::~Yin()
|
|
{
|
|
}
|
|
|
|
Yin::YinOutput
|
|
Yin::process(const double *in) const {
|
|
|
|
double* yinBuffer = new double[m_yinBufferSize];
|
|
|
|
// calculate aperiodicity function for all periods
|
|
if (m_fast) YinUtil::fastDifference(in, yinBuffer, m_yinBufferSize);
|
|
else YinUtil::slowDifference(in, yinBuffer, m_yinBufferSize);
|
|
|
|
YinUtil::cumulativeDifference(yinBuffer, m_yinBufferSize);
|
|
|
|
int tau = 0;
|
|
tau = YinUtil::absoluteThreshold(yinBuffer, m_yinBufferSize, m_thresh);
|
|
|
|
double interpolatedTau;
|
|
double aperiodicity;
|
|
double f0;
|
|
|
|
if (tau!=0)
|
|
{
|
|
interpolatedTau = YinUtil::parabolicInterpolation(yinBuffer, abs(tau), m_yinBufferSize);
|
|
f0 = m_inputSampleRate * (1.0 / interpolatedTau);
|
|
} else {
|
|
interpolatedTau = 0;
|
|
f0 = 0;
|
|
}
|
|
double rms = std::sqrt(YinUtil::sumSquare(in, 0, m_yinBufferSize)/m_yinBufferSize);
|
|
aperiodicity = yinBuffer[abs(tau)];
|
|
// std::cerr << aperiodicity << std::endl;
|
|
if (tau < 0) f0 = -f0;
|
|
|
|
Yin::YinOutput yo(f0, 1-aperiodicity, rms);
|
|
for (size_t iBuf = 0; iBuf < m_yinBufferSize; ++iBuf)
|
|
{
|
|
yo.salience.push_back(yinBuffer[iBuf] < 1 ? 1-yinBuffer[iBuf] : 0); // why are the values sometimes < 0 if I don't check?
|
|
}
|
|
|
|
delete [] yinBuffer;
|
|
return yo;
|
|
}
|
|
|
|
Yin::YinOutput
|
|
Yin::processProbabilisticYin(const double *in) const {
|
|
|
|
double* yinBuffer = new double[m_yinBufferSize];
|
|
|
|
// calculate aperiodicity function for all periods
|
|
if (m_fast) YinUtil::fastDifference(in, yinBuffer, m_yinBufferSize);
|
|
else YinUtil::slowDifference(in, yinBuffer, m_yinBufferSize);
|
|
|
|
YinUtil::cumulativeDifference(yinBuffer, m_yinBufferSize);
|
|
|
|
vector<double> peakProbability = YinUtil::yinProb(yinBuffer, m_threshDistr, m_yinBufferSize);
|
|
|
|
// calculate overall "probability" from peak probability
|
|
double probSum = 0;
|
|
for (size_t iBin = 0; iBin < m_yinBufferSize; ++iBin)
|
|
{
|
|
probSum += peakProbability[iBin];
|
|
}
|
|
double rms = std::sqrt(YinUtil::sumSquare(in, 0, m_yinBufferSize)/m_yinBufferSize);
|
|
Yin::YinOutput yo(0,0,rms);
|
|
for (size_t iBuf = 0; iBuf < m_yinBufferSize; ++iBuf)
|
|
{
|
|
yo.salience.push_back(peakProbability[iBuf]);
|
|
if (peakProbability[iBuf] > 0)
|
|
{
|
|
double currentF0 =
|
|
m_inputSampleRate * (1.0 /
|
|
YinUtil::parabolicInterpolation(yinBuffer, iBuf, m_yinBufferSize));
|
|
yo.freqProb.push_back(pair<double, double>(currentF0, peakProbability[iBuf]));
|
|
}
|
|
}
|
|
|
|
// std::cerr << yo.freqProb.size() << std::endl;
|
|
|
|
delete [] yinBuffer;
|
|
return yo;
|
|
}
|
|
|
|
|
|
int
|
|
Yin::setThreshold(double parameter)
|
|
{
|
|
m_thresh = static_cast<float>(parameter);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
Yin::setThresholdDistr(float parameter)
|
|
{
|
|
m_threshDistr = static_cast<size_t>(parameter);
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
Yin::setFrameSize(size_t parameter)
|
|
{
|
|
m_frameSize = parameter;
|
|
m_yinBufferSize = m_frameSize/2;
|
|
return 0;
|
|
}
|
|
|
|
int
|
|
Yin::setFast(bool parameter)
|
|
{
|
|
m_fast = parameter;
|
|
return 0;
|
|
}
|