comparison Yin.cpp @ 0:99bac62ee2da

added PYIN sources, should be compileable

+#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) : 
+    m_frameSize(frameSize),
+    m_inputSampleRate(inputSampleRate),
+    m_thresh(thresh),
+    m_threshDistr(2),
+    m_yinBufferSize(frameSize/2)
+{
+    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
+    YinUtil::fastDifference(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
+    YinUtil::fastDifference(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];
+    }
+        
+    Yin::YinOutput yo(0,0,0);
+    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::setRemoveUnvoiced(bool parameter)
+// {
+//     m_removeUnvoiced = parameter;
+//     return 0;
+// }