1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
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;
}
|
