diff options
| author | Robin Gareus <robin@gareus.org> | 2019-09-02 03:12:22 +0200 |
|---|---|---|
| committer | Robin Gareus <robin@gareus.org> | 2019-09-02 03:12:22 +0200 |
| commit | 63994f3b820c8f0754ff59d0d09585405d87ae0e (patch) | |
| tree | 4138d2f4b5d7e7c4ab0f371c08615b5d8fcc7538 /libs/vamp-pyin/Yin.cpp | |
| parent | 1c8b6e1b4296b4fbabc258f9f94635390a319522 (diff) | |
Include vamp-pyin
In preparation for captainMorgan's pitch analysis script.
Diffstat (limited to 'libs/vamp-pyin/Yin.cpp')
| -rw-r--r-- | libs/vamp-pyin/Yin.cpp | 153 |
1 files changed, 153 insertions, 0 deletions
diff --git a/libs/vamp-pyin/Yin.cpp b/libs/vamp-pyin/Yin.cpp new file mode 100644 index 0000000000..024a124c4f --- /dev/null +++ b/libs/vamp-pyin/Yin.cpp @@ -0,0 +1,153 @@ +/* -*- 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;
+}
|