Chris@185
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
Chris@185
|
2
|
Chris@185
|
3 /*
|
Chris@185
|
4 QM DSP Library
|
Chris@185
|
5
|
Chris@185
|
6 Centre for Digital Music, Queen Mary, University of London.
|
Chris@185
|
7 This file copyright 2005-2006 Christian Landone.and Matthew Davies.
|
Chris@84
|
8
|
Chris@84
|
9 This program is free software; you can redistribute it and/or
|
Chris@84
|
10 modify it under the terms of the GNU General Public License as
|
Chris@84
|
11 published by the Free Software Foundation; either version 2 of the
|
Chris@84
|
12 License, or (at your option) any later version. See the file
|
Chris@185
|
13 COPYING included with this distribution for more information.
|
Chris@185
|
14 */
|
Chris@185
|
15
|
Chris@185
|
16 #include "TempoTrack.h"
|
Chris@185
|
17
|
Chris@185
|
18 #include "maths/MathAliases.h"
|
Chris@185
|
19 #include "maths/MathUtilities.h"
|
Chris@185
|
20
|
Chris@185
|
21 #include <iostream>
|
Chris@185
|
22
|
Chris@185
|
23 #include <cassert>
|
Chris@185
|
24
|
Chris@185
|
25 //#define DEBUG_TEMPO_TRACK 1
|
Chris@185
|
26
|
Chris@185
|
27
|
Chris@185
|
28 #define RAY43VAL
|
Chris@185
|
29
|
Chris@185
|
30 //////////////////////////////////////////////////////////////////////
|
Chris@185
|
31 // Construction/Destruction
|
Chris@185
|
32 //////////////////////////////////////////////////////////////////////
|
Chris@185
|
33
|
Chris@185
|
34 TempoTrack::TempoTrack( TTParams Params )
|
Chris@185
|
35 {
|
Chris@185
|
36 m_tempoScratch = NULL;
|
Chris@185
|
37 m_rawDFFrame = NULL;
|
Chris@185
|
38 m_smoothDFFrame = NULL;
|
Chris@185
|
39 m_frameACF = NULL;
|
Chris@185
|
40 m_smoothRCF = NULL;
|
Chris@185
|
41
|
Chris@185
|
42 m_dataLength = 0;
|
Chris@185
|
43 m_winLength = 0;
|
Chris@185
|
44 m_lagLength = 0;
|
Chris@185
|
45
|
Chris@185
|
46 m_rayparam = 0;
|
Chris@185
|
47 m_sigma = 0;
|
Chris@185
|
48 m_DFWVNnorm = 0;
|
Chris@185
|
49
|
Chris@185
|
50 initialise( Params );
|
Chris@185
|
51 }
|
Chris@185
|
52
|
Chris@185
|
53 TempoTrack::~TempoTrack()
|
Chris@185
|
54 {
|
Chris@185
|
55 deInitialise();
|
Chris@185
|
56 }
|
Chris@185
|
57
|
Chris@185
|
58 void TempoTrack::initialise( TTParams Params )
|
Chris@185
|
59 {
|
Chris@185
|
60 m_winLength = Params.winLength;
|
Chris@185
|
61 m_lagLength = Params.lagLength;
|
Chris@185
|
62
|
Chris@185
|
63 m_rayparam = 43.0;
|
Chris@185
|
64 m_sigma = sqrt(3.9017);
|
Chris@185
|
65 m_DFWVNnorm = exp( ( log( 2.0 ) / m_rayparam ) * ( m_winLength + 2 ) );
|
Chris@185
|
66
|
Chris@185
|
67 m_rawDFFrame = new double[ m_winLength ];
|
Chris@185
|
68 m_smoothDFFrame = new double[ m_winLength ];
|
Chris@185
|
69 m_frameACF = new double[ m_winLength ];
|
Chris@185
|
70 m_tempoScratch = new double[ m_lagLength ];
|
Chris@185
|
71 m_smoothRCF = new double[ m_lagLength ];
|
Chris@185
|
72
|
Chris@185
|
73
|
Chris@185
|
74 unsigned int winPre = Params.WinT.pre;
|
Chris@185
|
75 unsigned int winPost = Params.WinT.post;
|
Chris@185
|
76
|
Chris@185
|
77 m_DFFramer.configure( m_winLength, m_lagLength );
|
Chris@185
|
78
|
Chris@185
|
79 m_DFPParams.length = m_winLength;
|
Chris@185
|
80 m_DFPParams.AlphaNormParam = Params.alpha;
|
Chris@185
|
81 m_DFPParams.LPOrd = Params.LPOrd;
|
Chris@185
|
82 m_DFPParams.LPACoeffs = Params.LPACoeffs;
|
Chris@185
|
83 m_DFPParams.LPBCoeffs = Params.LPBCoeffs;
|
Chris@185
|
84 m_DFPParams.winPre = Params.WinT.pre;
|
Chris@185
|
85 m_DFPParams.winPost = Params.WinT.post;
|
Chris@185
|
86 m_DFPParams.isMedianPositive = true;
|
Chris@185
|
87
|
Chris@185
|
88 m_DFConditioning = new DFProcess( m_DFPParams );
|
Chris@185
|
89
|
Chris@185
|
90
|
Chris@185
|
91 // these are parameters for smoothing m_tempoScratch
|
Chris@185
|
92 m_RCFPParams.length = m_lagLength;
|
Chris@185
|
93 m_RCFPParams.AlphaNormParam = Params.alpha;
|
Chris@185
|
94 m_RCFPParams.LPOrd = Params.LPOrd;
|
Chris@185
|
95 m_RCFPParams.LPACoeffs = Params.LPACoeffs;
|
Chris@185
|
96 m_RCFPParams.LPBCoeffs = Params.LPBCoeffs;
|
Chris@185
|
97 m_RCFPParams.winPre = Params.WinT.pre;
|
Chris@185
|
98 m_RCFPParams.winPost = Params.WinT.post;
|
Chris@185
|
99 m_RCFPParams.isMedianPositive = true;
|
Chris@185
|
100
|
Chris@185
|
101 m_RCFConditioning = new DFProcess( m_RCFPParams );
|
Chris@185
|
102
|
Chris@185
|
103 }
|
Chris@185
|
104
|
Chris@185
|
105 void TempoTrack::deInitialise()
|
Chris@185
|
106 {
|
Chris@185
|
107 delete [] m_rawDFFrame;
|
Chris@185
|
108
|
Chris@185
|
109 delete [] m_smoothDFFrame;
|
Chris@185
|
110
|
Chris@185
|
111 delete [] m_smoothRCF;
|
Chris@185
|
112
|
Chris@185
|
113 delete [] m_frameACF;
|
Chris@185
|
114
|
Chris@185
|
115 delete [] m_tempoScratch;
|
Chris@185
|
116
|
Chris@185
|
117 delete m_DFConditioning;
|
Chris@185
|
118
|
Chris@185
|
119 delete m_RCFConditioning;
|
Chris@185
|
120
|
Chris@185
|
121 }
|
Chris@185
|
122
|
Chris@185
|
123 void TempoTrack::createCombFilter(double* Filter, unsigned int winLength, unsigned int TSig, double beatLag)
|
Chris@185
|
124 {
|
Chris@185
|
125 unsigned int i;
|
Chris@185
|
126
|
Chris@185
|
127 if( beatLag == 0 )
|
Chris@185
|
128 {
|
Chris@185
|
129 for( i = 0; i < winLength; i++ )
|
Chris@185
|
130 {
|
Chris@185
|
131 Filter[ i ] = ( ( i + 1 ) / pow( m_rayparam, 2.0) ) * exp( ( -pow(( i + 1 ),2.0 ) / ( 2.0 * pow( m_rayparam, 2.0))));
|
Chris@185
|
132 }
|
Chris@185
|
133 }
|
Chris@185
|
134 else
|
Chris@185
|
135 {
|
Chris@185
|
136 m_sigma = beatLag/4;
|
Chris@185
|
137 for( i = 0; i < winLength; i++ )
|
Chris@185
|
138 {
|
Chris@185
|
139 double dlag = (double)(i+1) - beatLag;
|
Chris@185
|
140 Filter[ i ] = exp(-0.5 * pow(( dlag / m_sigma), 2.0) ) / (sqrt( 2 * PI) * m_sigma);
|
Chris@185
|
141 }
|
Chris@185
|
142 }
|
Chris@185
|
143 }
|
Chris@185
|
144
|
Chris@185
|
145 double TempoTrack::tempoMM(double* ACF, double* weight, int tsig)
|
Chris@185
|
146 {
|
Chris@185
|
147
|
Chris@185
|
148 double period = 0;
|
Chris@185
|
149 double maxValRCF = 0.0;
|
Chris@185
|
150 unsigned int maxIndexRCF = 0;
|
Chris@185
|
151
|
Chris@185
|
152 double* pdPeaks;
|
Chris@185
|
153
|
Chris@185
|
154 unsigned int maxIndexTemp;
|
Chris@185
|
155 double maxValTemp;
|
Chris@185
|
156 unsigned int count;
|
Chris@185
|
157
|
Chris@185
|
158 unsigned int numelem,i,j;
|
Chris@185
|
159 int a, b;
|
Chris@185
|
160
|
Chris@185
|
161 for( i = 0; i < m_lagLength; i++ )
|
Chris@185
|
162 m_tempoScratch[ i ] = 0.0;
|
Chris@185
|
163
|
Chris@185
|
164 if( tsig == 0 )
|
Chris@185
|
165 {
|
Chris@185
|
166 //if time sig is unknown, use metrically unbiased version of Filterbank
|
Chris@185
|
167 numelem = 4;
|
Chris@185
|
168 }
|
Chris@185
|
169 else
|
Chris@185
|
170 {
|
Chris@185
|
171 numelem = tsig;
|
Chris@185
|
172 }
|
Chris@185
|
173
|
Chris@185
|
174 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
175 std::cerr << "tempoMM: m_winLength = " << m_winLength << ", m_lagLength = " << m_lagLength << ", numelem = " << numelem << std::endl;
|
Chris@185
|
176 #endif
|
Chris@185
|
177
|
Chris@185
|
178 for(i=1;i<m_lagLength-1;i++)
|
Chris@185
|
179 {
|
Chris@185
|
180 //first and last output values are left intentionally as zero
|
Chris@185
|
181 for (a=1;a<=numelem;a++)
|
Chris@185
|
182 {
|
Chris@185
|
183 for(b=(1-a);b<a;b++)
|
Chris@185
|
184 {
|
Chris@185
|
185 if( tsig == 0 )
|
Chris@185
|
186 {
|
Chris@185
|
187 m_tempoScratch[i] += ACF[a*(i+1)+b-1] * (1.0 / (2.0 * (double)a-1)) * weight[i];
|
Chris@185
|
188 }
|
Chris@185
|
189 else
|
Chris@185
|
190 {
|
Chris@185
|
191 m_tempoScratch[i] += ACF[a*(i+1)+b-1] * 1 * weight[i];
|
Chris@185
|
192 }
|
Chris@185
|
193 }
|
Chris@185
|
194 }
|
Chris@185
|
195 }
|
Chris@185
|
196
|
Chris@185
|
197
|
Chris@185
|
198 //////////////////////////////////////////////////
|
Chris@185
|
199 // MODIFIED BEAT PERIOD EXTRACTION //////////////
|
Chris@185
|
200 /////////////////////////////////////////////////
|
Chris@185
|
201
|
Chris@185
|
202 // find smoothed version of RCF ( as applied to Detection Function)
|
Chris@185
|
203 m_RCFConditioning->process( m_tempoScratch, m_smoothRCF);
|
Chris@185
|
204
|
Chris@185
|
205 if (tsig != 0) // i.e. in context dependent state
|
Chris@185
|
206 {
|
Chris@185
|
207 // NOW FIND MAX INDEX OF ACFOUT
|
Chris@185
|
208 for( i = 0; i < m_lagLength; i++)
|
Chris@185
|
209 {
|
Chris@185
|
210 if( m_tempoScratch[ i ] > maxValRCF)
|
Chris@185
|
211 {
|
Chris@185
|
212 maxValRCF = m_tempoScratch[ i ];
|
Chris@185
|
213 maxIndexRCF = i;
|
Chris@185
|
214 }
|
Chris@185
|
215 }
|
Chris@185
|
216 }
|
Chris@185
|
217 else // using rayleigh weighting
|
Chris@185
|
218 {
|
Chris@185
|
219 vector <vector<double> > rcfMat;
|
Chris@185
|
220
|
Chris@185
|
221 double sumRcf = 0.;
|
Chris@185
|
222
|
Chris@185
|
223 double maxVal = 0.;
|
Chris@185
|
224 // now find the two values which minimise rcfMat
|
Chris@185
|
225 double minVal = 0.;
|
Chris@185
|
226 int p_i = 1; // periodicity for row i;
|
Chris@185
|
227 int p_j = 1; //periodicity for column j;
|
Chris@185
|
228
|
Chris@185
|
229
|
Chris@185
|
230 for ( i=0; i<m_lagLength; i++)
|
Chris@185
|
231 {
|
Chris@185
|
232 m_tempoScratch[i] =m_smoothRCF[i];
|
Chris@185
|
233 }
|
Chris@185
|
234
|
Chris@185
|
235 // normalise m_tempoScratch so that it sums to zero.
|
Chris@185
|
236 for ( i=0; i<m_lagLength; i++)
|
Chris@185
|
237 {
|
Chris@185
|
238 sumRcf += m_tempoScratch[i];
|
Chris@185
|
239 }
|
Chris@185
|
240
|
Chris@185
|
241 for( i=0; i<m_lagLength; i++)
|
Chris@185
|
242 {
|
Chris@185
|
243 m_tempoScratch[i] /= sumRcf;
|
Chris@185
|
244 }
|
Chris@185
|
245
|
Chris@185
|
246 // create a matrix to store m_tempoScratchValues modified by log2 ratio
|
Chris@185
|
247 for ( i=0; i<m_lagLength; i++)
|
Chris@185
|
248 {
|
Chris@185
|
249 rcfMat.push_back ( vector<double>() ); // adds a new row...
|
Chris@185
|
250 }
|
Chris@185
|
251
|
Chris@185
|
252 for (i=0; i<m_lagLength; i++)
|
Chris@185
|
253 {
|
Chris@185
|
254 for (j=0; j<m_lagLength; j++)
|
Chris@185
|
255 {
|
Chris@185
|
256 rcfMat[i].push_back (0.);
|
Chris@185
|
257 }
|
Chris@185
|
258 }
|
Chris@185
|
259
|
Chris@185
|
260 // the 'i' and 'j' indices deliberately start from '1' and not '0'
|
Chris@185
|
261 for ( i=1; i<m_lagLength; i++)
|
Chris@185
|
262 {
|
Chris@185
|
263 for (j=1; j<m_lagLength; j++)
|
Chris@185
|
264 {
|
Chris@185
|
265 double log2PeriodRatio = log( static_cast<double>(i)/static_cast<double>(j) ) / log(2.0);
|
Chris@185
|
266 rcfMat[i][j] = ( abs(1.0-abs(log2PeriodRatio)) );
|
Chris@185
|
267 rcfMat[i][j] += ( 0.01*( 1./(m_tempoScratch[i]+m_tempoScratch[j]) ) );
|
Chris@185
|
268 }
|
Chris@185
|
269 }
|
Chris@185
|
270
|
Chris@185
|
271 // set diagonal equal to maximum value in rcfMat
|
Chris@185
|
272 // we don't want to pick one strong middle peak - we need a combination of two peaks.
|
Chris@185
|
273
|
Chris@185
|
274 for ( i=1; i<m_lagLength; i++)
|
Chris@185
|
275 {
|
Chris@185
|
276 for (j=1; j<m_lagLength; j++)
|
Chris@185
|
277 {
|
Chris@185
|
278 if (rcfMat[i][j] > maxVal)
|
Chris@185
|
279 {
|
Chris@185
|
280 maxVal = rcfMat[i][j];
|
Chris@185
|
281 }
|
Chris@185
|
282 }
|
Chris@185
|
283 }
|
Chris@185
|
284
|
Chris@185
|
285 for ( i=1; i<m_lagLength; i++)
|
Chris@185
|
286 {
|
Chris@185
|
287 rcfMat[i][i] = maxVal;
|
Chris@185
|
288 }
|
Chris@185
|
289
|
Chris@185
|
290 // now find the row and column number which minimise rcfMat
|
Chris@185
|
291 minVal = maxVal;
|
Chris@185
|
292
|
Chris@185
|
293 for ( i=1; i<m_lagLength; i++)
|
Chris@185
|
294 {
|
Chris@185
|
295 for ( j=1; j<m_lagLength; j++)
|
Chris@185
|
296 {
|
Chris@185
|
297 if (rcfMat[i][j] < minVal)
|
Chris@185
|
298 {
|
Chris@185
|
299 minVal = rcfMat[i][j];
|
Chris@185
|
300 p_i = i;
|
Chris@185
|
301 p_j = j;
|
Chris@185
|
302 }
|
Chris@185
|
303 }
|
Chris@185
|
304 }
|
Chris@185
|
305
|
Chris@185
|
306
|
Chris@185
|
307 // initially choose p_j (arbitrary) - saves on an else statement
|
Chris@185
|
308 int beatPeriod = p_j;
|
Chris@185
|
309 if (m_tempoScratch[p_i] > m_tempoScratch[p_j])
|
Chris@185
|
310 {
|
Chris@185
|
311 beatPeriod = p_i;
|
Chris@185
|
312 }
|
Chris@185
|
313
|
Chris@185
|
314 // now write the output
|
Chris@185
|
315 maxIndexRCF = static_cast<int>(beatPeriod);
|
Chris@185
|
316 }
|
Chris@185
|
317
|
Chris@185
|
318
|
Chris@185
|
319 double locked = 5168.f / maxIndexRCF;
|
Chris@185
|
320 if (locked >= 30 && locked <= 180) {
|
Chris@185
|
321 m_lockedTempo = locked;
|
Chris@185
|
322 }
|
Chris@185
|
323
|
Chris@185
|
324 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
325 std::cerr << "tempoMM: locked tempo = " << m_lockedTempo << std::endl;
|
Chris@185
|
326 #endif
|
Chris@185
|
327
|
Chris@185
|
328 if( tsig == 0 )
|
Chris@185
|
329 tsig = 4;
|
Chris@185
|
330
|
Chris@185
|
331
|
Chris@185
|
332 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
333 std::cerr << "tempoMM: maxIndexRCF = " << maxIndexRCF << std::endl;
|
Chris@185
|
334 #endif
|
Chris@185
|
335
|
Chris@185
|
336 if( tsig == 4 )
|
Chris@185
|
337 {
|
Chris@185
|
338 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
339 std::cerr << "tsig == 4" << std::endl;
|
Chris@185
|
340 #endif
|
Chris@185
|
341
|
Chris@185
|
342 pdPeaks = new double[ 4 ];
|
Chris@185
|
343 for( i = 0; i < 4; i++ ){ pdPeaks[ i ] = 0.0;}
|
Chris@185
|
344
|
Chris@185
|
345 pdPeaks[ 0 ] = ( double )maxIndexRCF + 1;
|
Chris@185
|
346
|
Chris@185
|
347 maxIndexTemp = 0;
|
Chris@185
|
348 maxValTemp = 0.0;
|
Chris@185
|
349 count = 0;
|
Chris@185
|
350
|
Chris@185
|
351 for( i = (2 * maxIndexRCF + 1) - 1; i < (2 * maxIndexRCF + 1) + 2; i++ )
|
Chris@185
|
352 {
|
Chris@185
|
353 if( ACF[ i ] > maxValTemp )
|
Chris@185
|
354 {
|
Chris@185
|
355 maxValTemp = ACF[ i ];
|
Chris@185
|
356 maxIndexTemp = count;
|
Chris@185
|
357 }
|
Chris@185
|
358 count++;
|
Chris@185
|
359 }
|
Chris@185
|
360 pdPeaks[ 1 ] = (double)( maxIndexTemp + 1 + ( (2 * maxIndexRCF + 1 ) - 2 ) + 1 )/2;
|
Chris@185
|
361
|
Chris@185
|
362 maxIndexTemp = 0;
|
Chris@185
|
363 maxValTemp = 0.0;
|
Chris@185
|
364 count = 0;
|
Chris@185
|
365
|
Chris@185
|
366 for( i = (3 * maxIndexRCF + 2 ) - 2; i < (3 * maxIndexRCF + 2 ) + 3; i++ )
|
Chris@185
|
367 {
|
Chris@185
|
368 if( ACF[ i ] > maxValTemp )
|
Chris@185
|
369 {
|
Chris@185
|
370 maxValTemp = ACF[ i ];
|
Chris@185
|
371 maxIndexTemp = count;
|
Chris@185
|
372 }
|
Chris@185
|
373 count++;
|
Chris@185
|
374 }
|
Chris@185
|
375 pdPeaks[ 2 ] = (double)( maxIndexTemp + 1 + ( (3 * maxIndexRCF + 2) - 4 ) + 1 )/3;
|
Chris@185
|
376
|
Chris@185
|
377 maxIndexTemp = 0;
|
Chris@185
|
378 maxValTemp = 0.0;
|
Chris@185
|
379 count = 0;
|
Chris@185
|
380
|
Chris@185
|
381 for( i = ( 4 * maxIndexRCF + 3) - 3; i < ( 4 * maxIndexRCF + 3) + 4; i++ )
|
Chris@185
|
382 {
|
Chris@185
|
383 if( ACF[ i ] > maxValTemp )
|
Chris@185
|
384 {
|
Chris@185
|
385 maxValTemp = ACF[ i ];
|
Chris@185
|
386 maxIndexTemp = count;
|
Chris@185
|
387 }
|
Chris@185
|
388 count++;
|
Chris@185
|
389 }
|
Chris@185
|
390 pdPeaks[ 3 ] = (double)( maxIndexTemp + 1 + ( (4 * maxIndexRCF + 3) - 9 ) + 1 )/4 ;
|
Chris@185
|
391
|
Chris@185
|
392
|
Chris@185
|
393 period = MathUtilities::mean( pdPeaks, 4 );
|
Chris@185
|
394 }
|
Chris@185
|
395 else
|
Chris@185
|
396 {
|
Chris@185
|
397 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
398 std::cerr << "tsig != 4" << std::endl;
|
Chris@185
|
399 #endif
|
Chris@185
|
400
|
Chris@185
|
401 pdPeaks = new double[ 3 ];
|
Chris@185
|
402 for( i = 0; i < 3; i++ ){ pdPeaks[ i ] = 0.0;}
|
Chris@185
|
403
|
Chris@185
|
404 pdPeaks[ 0 ] = ( double )maxIndexRCF + 1;
|
Chris@185
|
405
|
Chris@185
|
406 maxIndexTemp = 0;
|
Chris@185
|
407 maxValTemp = 0.0;
|
Chris@185
|
408 count = 0;
|
Chris@185
|
409
|
Chris@185
|
410 for( i = (2 * maxIndexRCF + 1) - 1; i < (2 * maxIndexRCF + 1) + 2; i++ )
|
Chris@185
|
411 {
|
Chris@185
|
412 if( ACF[ i ] > maxValTemp )
|
Chris@185
|
413 {
|
Chris@185
|
414 maxValTemp = ACF[ i ];
|
Chris@185
|
415 maxIndexTemp = count;
|
Chris@185
|
416 }
|
Chris@185
|
417 count++;
|
Chris@185
|
418 }
|
Chris@185
|
419 pdPeaks[ 1 ] = (double)( maxIndexTemp + 1 + ( (2 * maxIndexRCF + 1 ) - 2 ) + 1 )/2;
|
Chris@185
|
420
|
Chris@185
|
421 maxIndexTemp = 0;
|
Chris@185
|
422 maxValTemp = 0.0;
|
Chris@185
|
423 count = 0;
|
Chris@185
|
424
|
Chris@185
|
425 for( i = (3 * maxIndexRCF + 2 ) - 2; i < (3 * maxIndexRCF + 2 ) + 3; i++ )
|
Chris@185
|
426 {
|
Chris@185
|
427 if( ACF[ i ] > maxValTemp )
|
Chris@185
|
428 {
|
Chris@185
|
429 maxValTemp = ACF[ i ];
|
Chris@185
|
430 maxIndexTemp = count;
|
Chris@185
|
431 }
|
Chris@185
|
432 count++;
|
Chris@185
|
433 }
|
Chris@185
|
434 pdPeaks[ 2 ] = (double)( maxIndexTemp + 1 + ( (3 * maxIndexRCF + 2) - 4 ) + 1 )/3;
|
Chris@185
|
435
|
Chris@185
|
436
|
Chris@185
|
437 period = MathUtilities::mean( pdPeaks, 3 );
|
Chris@185
|
438 }
|
Chris@185
|
439
|
Chris@185
|
440 delete [] pdPeaks;
|
Chris@185
|
441
|
Chris@185
|
442 return period;
|
Chris@185
|
443 }
|
Chris@185
|
444
|
Chris@185
|
445 void TempoTrack::stepDetect( double* periodP, double* periodG, int currentIdx, int* flag )
|
Chris@185
|
446 {
|
Chris@185
|
447 double stepthresh = 1 * 3.9017;
|
Chris@185
|
448
|
Chris@185
|
449 if( *flag )
|
Chris@185
|
450 {
|
Chris@185
|
451 if(abs(periodG[ currentIdx ] - periodP[ currentIdx ]) > stepthresh)
|
Chris@185
|
452 {
|
Chris@185
|
453 // do nuffin'
|
Chris@185
|
454 }
|
Chris@185
|
455 }
|
Chris@185
|
456 else
|
Chris@185
|
457 {
|
Chris@185
|
458 if(fabs(periodG[ currentIdx ]-periodP[ currentIdx ]) > stepthresh)
|
Chris@185
|
459 {
|
Chris@185
|
460 *flag = 3;
|
Chris@185
|
461 }
|
Chris@185
|
462 }
|
Chris@185
|
463 }
|
Chris@185
|
464
|
Chris@185
|
465 void TempoTrack::constDetect( double* periodP, int currentIdx, int* flag )
|
Chris@185
|
466 {
|
Chris@185
|
467 double constthresh = 2 * 3.9017;
|
Chris@185
|
468
|
Chris@185
|
469 if( fabs( 2 * periodP[ currentIdx ] - periodP[ currentIdx - 1] - periodP[ currentIdx - 2] ) < constthresh)
|
Chris@185
|
470 {
|
Chris@185
|
471 *flag = 1;
|
Chris@185
|
472 }
|
Chris@185
|
473 else
|
Chris@185
|
474 {
|
Chris@185
|
475 *flag = 0;
|
Chris@185
|
476 }
|
Chris@185
|
477 }
|
Chris@185
|
478
|
Chris@185
|
479 int TempoTrack::findMeter(double *ACF, unsigned int len, double period)
|
Chris@185
|
480 {
|
Chris@185
|
481 int i;
|
Chris@185
|
482 int p = (int)MathUtilities::round( period );
|
Chris@185
|
483 int tsig;
|
Chris@185
|
484
|
Chris@185
|
485 double Energy_3 = 0.0;
|
Chris@185
|
486 double Energy_4 = 0.0;
|
Chris@185
|
487
|
Chris@185
|
488 double temp3A = 0.0;
|
Chris@185
|
489 double temp3B = 0.0;
|
Chris@185
|
490 double temp4A = 0.0;
|
Chris@185
|
491 double temp4B = 0.0;
|
Chris@185
|
492
|
Chris@185
|
493 double* dbf = new double[ len ]; int t = 0;
|
Chris@185
|
494 for( unsigned int u = 0; u < len; u++ ){ dbf[ u ] = 0.0; }
|
Chris@185
|
495
|
Chris@185
|
496 if( (double)len < 6 * p + 2 )
|
Chris@185
|
497 {
|
Chris@185
|
498 for( i = ( 3 * p - 2 ); i < ( 3 * p + 2 ) + 1; i++ )
|
Chris@185
|
499 {
|
Chris@185
|
500 temp3A += ACF[ i ];
|
Chris@185
|
501 dbf[ t++ ] = ACF[ i ];
|
Chris@185
|
502 }
|
Chris@185
|
503
|
Chris@185
|
504 for( i = ( 4 * p - 2 ); i < ( 4 * p + 2 ) + 1; i++ )
|
Chris@185
|
505 {
|
Chris@185
|
506 temp4A += ACF[ i ];
|
Chris@185
|
507 }
|
Chris@185
|
508
|
Chris@185
|
509 Energy_3 = temp3A;
|
Chris@185
|
510 Energy_4 = temp4A;
|
Chris@185
|
511 }
|
Chris@185
|
512 else
|
Chris@185
|
513 {
|
Chris@185
|
514 for( i = ( 3 * p - 2 ); i < ( 3 * p + 2 ) + 1; i++ )
|
Chris@185
|
515 {
|
Chris@185
|
516 temp3A += ACF[ i ];
|
Chris@185
|
517 }
|
Chris@185
|
518
|
Chris@185
|
519 for( i = ( 4 * p - 2 ); i < ( 4 * p + 2 ) + 1; i++ )
|
Chris@185
|
520 {
|
Chris@185
|
521 temp4A += ACF[ i ];
|
Chris@185
|
522 }
|
Chris@185
|
523
|
Chris@185
|
524 for( i = ( 6 * p - 2 ); i < ( 6 * p + 2 ) + 1; i++ )
|
Chris@185
|
525 {
|
Chris@185
|
526 temp3B += ACF[ i ];
|
Chris@185
|
527 }
|
Chris@185
|
528
|
Chris@185
|
529 for( i = ( 2 * p - 2 ); i < ( 2 * p + 2 ) + 1; i++ )
|
Chris@185
|
530 {
|
Chris@185
|
531 temp4B += ACF[ i ];
|
Chris@185
|
532 }
|
Chris@185
|
533
|
Chris@185
|
534 Energy_3 = temp3A + temp3B;
|
Chris@185
|
535 Energy_4 = temp4A + temp4B;
|
Chris@185
|
536 }
|
Chris@185
|
537
|
Chris@185
|
538 if (Energy_3 > Energy_4)
|
Chris@185
|
539 {
|
Chris@185
|
540 tsig = 3;
|
Chris@185
|
541 }
|
Chris@185
|
542 else
|
Chris@185
|
543 {
|
Chris@185
|
544 tsig = 4;
|
Chris@185
|
545 }
|
Chris@185
|
546
|
Chris@185
|
547
|
Chris@185
|
548 return tsig;
|
Chris@185
|
549 }
|
Chris@185
|
550
|
Chris@185
|
551 void TempoTrack::createPhaseExtractor(double *Filter, unsigned int winLength, double period, unsigned int fsp, unsigned int lastBeat)
|
Chris@185
|
552 {
|
Chris@185
|
553 int p = (int)MathUtilities::round( period );
|
Chris@185
|
554 int predictedOffset = 0;
|
Chris@185
|
555
|
Chris@185
|
556 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
557 std::cerr << "TempoTrack::createPhaseExtractor: period = " << period << ", p = " << p << std::endl;
|
Chris@185
|
558 #endif
|
Chris@185
|
559
|
Chris@185
|
560 if (p > 10000) {
|
Chris@185
|
561 std::cerr << "TempoTrack::createPhaseExtractor: WARNING! Highly implausible period value " << p << "!" << std::endl;
|
Chris@185
|
562 period = 5168 / 120;
|
Chris@185
|
563 }
|
Chris@185
|
564
|
Chris@185
|
565 double* phaseScratch = new double[ p*2 + 2 ];
|
Chris@185
|
566 for (int i = 0; i < p*2 + 2; ++i) phaseScratch[i] = 0.0;
|
Chris@185
|
567
|
Chris@185
|
568
|
Chris@185
|
569 if( lastBeat != 0 )
|
Chris@185
|
570 {
|
Chris@185
|
571 lastBeat = (int)MathUtilities::round((double)lastBeat );///(double)winLength);
|
Chris@185
|
572
|
Chris@185
|
573 predictedOffset = lastBeat + p - fsp;
|
Chris@185
|
574
|
Chris@185
|
575 if (predictedOffset < 0)
|
Chris@185
|
576 {
|
Chris@185
|
577 lastBeat = 0;
|
Chris@185
|
578 }
|
Chris@185
|
579 }
|
Chris@185
|
580
|
Chris@185
|
581 if( lastBeat != 0 )
|
Chris@185
|
582 {
|
Chris@185
|
583 int mu = p;
|
Chris@185
|
584 double sigma = (double)p/8;
|
Chris@185
|
585 double PhaseMin = 0.0;
|
Chris@185
|
586 double PhaseMax = 0.0;
|
Chris@185
|
587 unsigned int scratchLength = p*2;
|
Chris@185
|
588 double temp = 0.0;
|
Chris@185
|
589
|
Chris@185
|
590 for( int i = 0; i < scratchLength; i++ )
|
Chris@185
|
591 {
|
Chris@185
|
592 phaseScratch[ i ] = exp( -0.5 * pow( ( i - mu ) / sigma, 2 ) ) / ( sqrt( 2*PI ) *sigma );
|
Chris@185
|
593 }
|
Chris@185
|
594
|
Chris@185
|
595 MathUtilities::getFrameMinMax( phaseScratch, scratchLength, &PhaseMin, &PhaseMax );
|
Chris@185
|
596
|
Chris@185
|
597 for(int i = 0; i < scratchLength; i ++)
|
Chris@185
|
598 {
|
Chris@185
|
599 temp = phaseScratch[ i ];
|
Chris@185
|
600 phaseScratch[ i ] = (temp - PhaseMin)/PhaseMax;
|
Chris@185
|
601 }
|
Chris@185
|
602
|
Chris@185
|
603 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
604 std::cerr << "predictedOffset = " << predictedOffset << std::endl;
|
Chris@185
|
605 #endif
|
Chris@185
|
606
|
Chris@185
|
607 unsigned int index = 0;
|
Chris@185
|
608 for (int i = p - ( predictedOffset - 1); i < p + ( p - predictedOffset) + 1; i++)
|
Chris@185
|
609 {
|
Chris@185
|
610 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
611 std::cerr << "assigning to filter index " << index << " (size = " << p*2 << ")" << " value " << phaseScratch[i] << " from scratch index " << i << std::endl;
|
Chris@185
|
612 #endif
|
Chris@185
|
613 Filter[ index++ ] = phaseScratch[ i ];
|
Chris@185
|
614 }
|
Chris@185
|
615 }
|
Chris@185
|
616 else
|
Chris@185
|
617 {
|
Chris@185
|
618 for( int i = 0; i < p; i ++)
|
Chris@185
|
619 {
|
Chris@185
|
620 Filter[ i ] = 1;
|
Chris@185
|
621 }
|
Chris@185
|
622 }
|
Chris@185
|
623
|
Chris@185
|
624 delete [] phaseScratch;
|
Chris@185
|
625 }
|
Chris@185
|
626
|
Chris@185
|
627 int TempoTrack::phaseMM(double *DF, double *weighting, unsigned int winLength, double period)
|
Chris@185
|
628 {
|
Chris@185
|
629 int alignment = 0;
|
Chris@185
|
630 int p = (int)MathUtilities::round( period );
|
Chris@185
|
631
|
Chris@185
|
632 double temp = 0.0;
|
Chris@185
|
633
|
Chris@185
|
634 double* y = new double[ winLength ];
|
Chris@185
|
635 double* align = new double[ p ];
|
Chris@185
|
636
|
Chris@185
|
637 for( int i = 0; i < winLength; i++ )
|
Chris@185
|
638 {
|
Chris@185
|
639 y[ i ] = (double)( -i + winLength )/(double)winLength;
|
Chris@185
|
640 y[ i ] = pow(y [i ],2.0); // raise to power 2.
|
Chris@185
|
641 }
|
Chris@185
|
642
|
Chris@185
|
643 for( int o = 0; o < p; o++ )
|
Chris@185
|
644 {
|
Chris@185
|
645 temp = 0.0;
|
Chris@185
|
646 for(int i = 1 + (o - 1); i< winLength; i += (p + 1))
|
Chris@185
|
647 {
|
Chris@185
|
648 temp = temp + DF[ i ] * y[ i ];
|
Chris@185
|
649 }
|
Chris@185
|
650 align[ o ] = temp * weighting[ o ];
|
Chris@185
|
651 }
|
Chris@185
|
652
|
Chris@185
|
653
|
Chris@185
|
654 double valTemp = 0.0;
|
Chris@185
|
655 for(int i = 0; i < p; i++)
|
Chris@185
|
656 {
|
Chris@185
|
657 if( align[ i ] > valTemp )
|
Chris@185
|
658 {
|
Chris@185
|
659 valTemp = align[ i ];
|
Chris@185
|
660 alignment = i;
|
Chris@185
|
661 }
|
Chris@185
|
662 }
|
Chris@185
|
663
|
Chris@185
|
664 delete [] y;
|
Chris@185
|
665 delete [] align;
|
Chris@185
|
666
|
Chris@185
|
667 return alignment;
|
Chris@185
|
668 }
|
Chris@185
|
669
|
Chris@185
|
670 int TempoTrack::beatPredict(unsigned int FSP0, double alignment, double period, unsigned int step )
|
Chris@185
|
671 {
|
Chris@185
|
672 int beat = 0;
|
Chris@185
|
673
|
Chris@185
|
674 int p = (int)MathUtilities::round( period );
|
Chris@185
|
675 int align = (int)MathUtilities::round( alignment );
|
Chris@185
|
676 int FSP = (int)MathUtilities::round( FSP0 );
|
Chris@185
|
677
|
Chris@185
|
678 int FEP = FSP + ( step );
|
Chris@185
|
679
|
Chris@185
|
680 beat = FSP + align;
|
Chris@185
|
681
|
Chris@185
|
682 m_beats.push_back( beat );
|
Chris@185
|
683
|
Chris@185
|
684 while( beat + p < FEP )
|
Chris@185
|
685 {
|
Chris@185
|
686 beat += p;
|
Chris@185
|
687
|
Chris@185
|
688 m_beats.push_back( beat );
|
Chris@185
|
689 }
|
Chris@185
|
690
|
Chris@185
|
691 return beat;
|
Chris@185
|
692 }
|
Chris@185
|
693
|
Chris@185
|
694
|
Chris@185
|
695
|
Chris@185
|
696 vector<int> TempoTrack::process( vector <double> DF,
|
Chris@185
|
697 vector <double> *tempoReturn )
|
Chris@185
|
698 {
|
Chris@185
|
699 m_dataLength = DF.size();
|
Chris@185
|
700
|
Chris@185
|
701 m_lockedTempo = 0.0;
|
Chris@185
|
702
|
Chris@185
|
703 double period = 0.0;
|
Chris@185
|
704 int stepFlag = 0;
|
Chris@185
|
705 int constFlag = 0;
|
Chris@185
|
706 int FSP = 0;
|
Chris@185
|
707 int tsig = 0;
|
Chris@185
|
708 int lastBeat = 0;
|
Chris@185
|
709
|
Chris@185
|
710 vector <double> causalDF;
|
Chris@185
|
711
|
Chris@185
|
712 causalDF = DF;
|
Chris@185
|
713
|
Chris@185
|
714 //Prepare Causal Extension DFData
|
Chris@185
|
715 unsigned int DFCLength = m_dataLength + m_winLength;
|
Chris@185
|
716
|
Chris@185
|
717 for( unsigned int j = 0; j < m_winLength; j++ )
|
Chris@185
|
718 {
|
Chris@185
|
719 causalDF.push_back( 0 );
|
Chris@185
|
720 }
|
Chris@185
|
721
|
Chris@185
|
722
|
Chris@185
|
723 double* RW = new double[ m_lagLength ];
|
Chris@185
|
724 for( unsigned int clear = 0; clear < m_lagLength; clear++){ RW[ clear ] = 0.0;}
|
Chris@185
|
725
|
Chris@185
|
726 double* GW = new double[ m_lagLength ];
|
Chris@185
|
727 for(unsigned int clear = 0; clear < m_lagLength; clear++){ GW[ clear ] = 0.0;}
|
Chris@185
|
728
|
Chris@185
|
729 double* PW = new double[ m_lagLength ];
|
Chris@185
|
730 for(unsigned clear = 0; clear < m_lagLength; clear++){ PW[ clear ] = 0.0;}
|
Chris@185
|
731
|
Chris@185
|
732 m_DFFramer.setSource( &causalDF[0], m_dataLength );
|
Chris@185
|
733
|
Chris@185
|
734 unsigned int TTFrames = m_DFFramer.getMaxNoFrames();
|
Chris@185
|
735
|
Chris@185
|
736 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
737 std::cerr << "TTFrames = " << TTFrames << std::endl;
|
Chris@185
|
738 #endif
|
Chris@185
|
739
|
Chris@185
|
740 double* periodP = new double[ TTFrames ];
|
Chris@185
|
741 for(unsigned clear = 0; clear < TTFrames; clear++){ periodP[ clear ] = 0.0;}
|
Chris@185
|
742
|
Chris@185
|
743 double* periodG = new double[ TTFrames ];
|
Chris@185
|
744 for(unsigned clear = 0; clear < TTFrames; clear++){ periodG[ clear ] = 0.0;}
|
Chris@185
|
745
|
Chris@185
|
746 double* alignment = new double[ TTFrames ];
|
Chris@185
|
747 for(unsigned clear = 0; clear < TTFrames; clear++){ alignment[ clear ] = 0.0;}
|
Chris@185
|
748
|
Chris@185
|
749 m_beats.clear();
|
Chris@185
|
750
|
Chris@185
|
751 createCombFilter( RW, m_lagLength, 0, 0 );
|
Chris@185
|
752
|
Chris@185
|
753 int TTLoopIndex = 0;
|
Chris@185
|
754
|
Chris@185
|
755 for( unsigned int i = 0; i < TTFrames; i++ )
|
Chris@185
|
756 {
|
Chris@185
|
757 m_DFFramer.getFrame( m_rawDFFrame );
|
Chris@185
|
758
|
Chris@185
|
759 m_DFConditioning->process( m_rawDFFrame, m_smoothDFFrame );
|
Chris@185
|
760
|
Chris@185
|
761 m_correlator.doAutoUnBiased( m_smoothDFFrame, m_frameACF, m_winLength );
|
Chris@185
|
762
|
Chris@185
|
763 periodP[ TTLoopIndex ] = tempoMM( m_frameACF, RW, 0 );
|
Chris@185
|
764
|
Chris@185
|
765 if( GW[ 0 ] != 0 )
|
Chris@185
|
766 {
|
Chris@185
|
767 periodG[ TTLoopIndex ] = tempoMM( m_frameACF, GW, tsig );
|
Chris@185
|
768 }
|
Chris@185
|
769 else
|
Chris@185
|
770 {
|
Chris@185
|
771 periodG[ TTLoopIndex ] = 0.0;
|
Chris@185
|
772 }
|
Chris@185
|
773
|
Chris@185
|
774 stepDetect( periodP, periodG, TTLoopIndex, &stepFlag );
|
Chris@185
|
775
|
Chris@185
|
776 if( stepFlag == 1)
|
Chris@185
|
777 {
|
Chris@185
|
778 constDetect( periodP, TTLoopIndex, &constFlag );
|
Chris@185
|
779 stepFlag = 0;
|
Chris@185
|
780 }
|
Chris@185
|
781 else
|
Chris@185
|
782 {
|
Chris@185
|
783 stepFlag -= 1;
|
Chris@185
|
784 }
|
Chris@185
|
785
|
Chris@185
|
786 if( stepFlag < 0 )
|
Chris@185
|
787 {
|
Chris@185
|
788 stepFlag = 0;
|
Chris@185
|
789 }
|
Chris@185
|
790
|
Chris@185
|
791 if( constFlag != 0)
|
Chris@185
|
792 {
|
Chris@185
|
793 tsig = findMeter( m_frameACF, m_winLength, periodP[ TTLoopIndex ] );
|
Chris@185
|
794
|
Chris@185
|
795 createCombFilter( GW, m_lagLength, tsig, periodP[ TTLoopIndex ] );
|
Chris@185
|
796
|
Chris@185
|
797 periodG[ TTLoopIndex ] = tempoMM( m_frameACF, GW, tsig );
|
Chris@185
|
798
|
Chris@185
|
799 period = periodG[ TTLoopIndex ];
|
Chris@185
|
800
|
Chris@185
|
801 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
802 std::cerr << "TempoTrack::process: constFlag == " << constFlag << ", TTLoopIndex = " << TTLoopIndex << ", period from periodG = " << period << std::endl;
|
Chris@185
|
803 #endif
|
Chris@185
|
804
|
Chris@185
|
805 createPhaseExtractor( PW, m_winLength, period, FSP, 0 );
|
Chris@185
|
806
|
Chris@185
|
807 constFlag = 0;
|
Chris@185
|
808
|
Chris@185
|
809 }
|
Chris@185
|
810 else
|
Chris@185
|
811 {
|
Chris@185
|
812 if( GW[ 0 ] != 0 )
|
Chris@185
|
813 {
|
Chris@185
|
814 period = periodG[ TTLoopIndex ];
|
Chris@185
|
815
|
Chris@185
|
816 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
817 std::cerr << "TempoTrack::process: GW[0] == " << GW[0] << ", TTLoopIndex = " << TTLoopIndex << ", period from periodG = " << period << std::endl;
|
Chris@185
|
818 #endif
|
Chris@185
|
819
|
Chris@185
|
820 if (period > 10000) {
|
Chris@185
|
821 std::cerr << "TempoTrack::process: WARNING! Highly implausible period value " << period << "!" << std::endl;
|
Chris@185
|
822 std::cerr << "periodG contains (of " << TTFrames << " frames): " << std::endl;
|
Chris@185
|
823 for (int i = 0; i < TTLoopIndex + 3 && i < TTFrames; ++i) {
|
Chris@185
|
824 std::cerr << i << " -> " << periodG[i] << std::endl;
|
Chris@185
|
825 }
|
Chris@185
|
826 std::cerr << "periodP contains (of " << TTFrames << " frames): " << std::endl;
|
Chris@185
|
827 for (int i = 0; i < TTLoopIndex + 3 && i < TTFrames; ++i) {
|
Chris@185
|
828 std::cerr << i << " -> " << periodP[i] << std::endl;
|
Chris@185
|
829 }
|
Chris@185
|
830 period = 5168 / 120;
|
Chris@185
|
831 }
|
Chris@185
|
832
|
Chris@185
|
833 createPhaseExtractor( PW, m_winLength, period, FSP, lastBeat );
|
Chris@185
|
834
|
Chris@185
|
835 }
|
Chris@185
|
836 else
|
Chris@185
|
837 {
|
Chris@185
|
838 period = periodP[ TTLoopIndex ];
|
Chris@185
|
839
|
Chris@185
|
840 #ifdef DEBUG_TEMPO_TRACK
|
Chris@185
|
841 std::cerr << "TempoTrack::process: GW[0] == " << GW[0] << ", TTLoopIndex = " << TTLoopIndex << ", period from periodP = " << period << std::endl;
|
Chris@185
|
842 #endif
|
Chris@185
|
843
|
Chris@185
|
844 createPhaseExtractor( PW, m_winLength, period, FSP, 0 );
|
Chris@185
|
845 }
|
Chris@185
|
846 }
|
Chris@185
|
847
|
Chris@185
|
848 alignment[ TTLoopIndex ] = phaseMM( m_rawDFFrame, PW, m_winLength, period );
|
Chris@185
|
849
|
Chris@185
|
850 lastBeat = beatPredict(FSP, alignment[ TTLoopIndex ], period, m_lagLength );
|
Chris@185
|
851
|
Chris@185
|
852 FSP += (m_lagLength);
|
Chris@185
|
853
|
Chris@185
|
854 if (tempoReturn) tempoReturn->push_back(m_lockedTempo);
|
Chris@185
|
855
|
Chris@185
|
856 TTLoopIndex++;
|
Chris@185
|
857 }
|
Chris@185
|
858
|
Chris@185
|
859
|
Chris@185
|
860 delete [] periodP;
|
Chris@185
|
861 delete [] periodG;
|
Chris@185
|
862 delete [] alignment;
|
Chris@185
|
863
|
Chris@185
|
864 delete [] RW;
|
Chris@185
|
865 delete [] GW;
|
Chris@185
|
866 delete [] PW;
|
Chris@185
|
867
|
Chris@185
|
868 return m_beats;
|
Chris@185
|
869 }
|
Chris@185
|
870
|