annotate dsp/tempotracking/TempoTrack.cpp @ 185:2ae4ceb76ac3

Fix some uninitialised values
author Chris Cannam
date Mon, 07 Sep 2015 14:00:30 +0100
parents e5907ae6de17
children e4a57215ddee
rev   line source
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