annotate dsp/tempotracking/TempoTrack.cpp @ 501:12b5a9244bb0

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