annotate bayesianArraySrc/BayesianArrayStructure.cpp @ 6:746a5af43c02

windowed bayesian distributions - drawn within constrained portion of the screen
author Andrew N Robertson <andrew.robertson@eecs.qmul.ac.uk>
date Thu, 02 Feb 2012 17:52:08 +0000
parents 5ef00d1dfe68
children 33dedfe32893
rev   line source
andrew@0 1 /*
andrew@0 2 * BayesianArrayStructure.cpp
andrew@0 3 * midiCannamReader
andrew@0 4 *
andrew@0 5 * Created by Andrew on 17/07/2011.
andrew@0 6 * Copyright 2011 QMUL. All rights reserved.
andrew@0 7 *
andrew@0 8 */
andrew@0 9
andrew@0 10 //look at reset speed to one - what does this do? - get rid of?
andrew@0 11
andrew@0 12
andrew@0 13 #include "BayesianArrayStructure.h"
andrew@0 14
andrew@0 15 BayesianArrayStructure::BayesianArrayStructure(){
andrew@0 16 printf("Bayesian structure: DeFault constructor called");
andrew@0 17
andrew@4 18 usingIntegratedTempoEstimate = false;// use max index
andrew@0 19
andrew@0 20 relativeSpeedLikelihoodStdDev = 5.0;
andrew@0 21
andrew@0 22 prior.createVector(1);
andrew@0 23 likelihood.createVector(1);
andrew@0 24 posterior.createVector(1);
andrew@0 25
andrew@0 26 speedPriorValue = 1.0;//default value for the speed prior
andrew@0 27 speedEstimate = speedPriorValue;
andrew@0 28
andrew@0 29 lastEventTime = 0;//ofGetElapsedTimeMillis();
andrew@0 30
andrew@0 31 tmpBestEstimate = 0;
andrew@5 32 crossUpdateTimeThreshold = 60;
andrew@3 33 priorWidth = 20;
andrew@0 34
andrew@0 35 }
andrew@0 36
andrew@0 37 BayesianArrayStructure::BayesianArrayStructure(int length){
andrew@0 38 printf("BAYESIAN STURCTURE CREATED LENGTH: %i\n", length);
andrew@0 39 //this constructor isnt called it seems
andrew@0 40 prior.createVector(length);
andrew@0 41 likelihood.createVector(length);
andrew@0 42 posterior.createVector(length);
andrew@0 43
andrew@0 44 lastEventTime = 0;
andrew@0 45
andrew@0 46
andrew@0 47 }
andrew@0 48
andrew@0 49
andrew@0 50
andrew@0 51 void BayesianArrayStructure::resetSize(int length){
andrew@0 52 printf("BAYESIAN STRUCTURE size is : %i\n", length);
andrew@0 53
andrew@0 54 prior.createVector(length);
andrew@0 55 likelihood.createVector(length);
andrew@0 56 posterior.createVector(length);
andrew@0 57
andrew@0 58 acceleration.createVector(length);
andrew@0 59
andrew@0 60 }
andrew@0 61
andrew@0 62 void BayesianArrayStructure::resetSpeedSize(int length){
andrew@0 63 printf("BAYESIAN reset SPEED size is : %i\n", length);
andrew@0 64
andrew@0 65 relativeSpeedPrior.createVector(length);
andrew@0 66 relativeSpeedLikelihood.createVector(length);
andrew@0 67 relativeSpeedPosterior.createVector(length);
andrew@0 68 tmpPosteriorForStorage.createVector(length);
andrew@0 69
andrew@0 70
andrew@0 71
andrew@0 72 }
andrew@0 73
andrew@0 74 void BayesianArrayStructure::setRelativeSpeedScalar(double f){
andrew@0 75 relativeSpeedPrior.scalar = f;
andrew@0 76 relativeSpeedPosterior.scalar = f;
andrew@0 77 relativeSpeedLikelihood.scalar = f;
andrew@0 78 }
andrew@0 79
andrew@0 80 void BayesianArrayStructure::resetSpeedToOne(){
andrew@0 81 relativeSpeedPrior.zero();
andrew@0 82 relativeSpeedPosterior.zero();
andrew@0 83 relativeSpeedLikelihood.zero();
andrew@0 84
andrew@0 85
andrew@0 86 relativeSpeedPosterior.addGaussianShape(100, 20, 0.8);
andrew@0 87 relativeSpeedPosterior.renormalise();
andrew@0 88 relativeSpeedPosterior.getMaximum();
andrew@0 89
andrew@0 90 setSpeedPrior(speedPriorValue);
andrew@0 91 speedEstimate = speedPriorValue;
andrew@0 92
andrew@0 93 prior.zero();
andrew@0 94 posterior.zero();
andrew@0 95
andrew@0 96 posterior.addToIndex(0, 1);
andrew@0 97 posterior.renormalise();
andrew@0 98
andrew@0 99 }
andrew@0 100
andrew@0 101 void BayesianArrayStructure::setSpeedPrior(double f){
andrew@0 102 speedPriorValue = f;
andrew@0 103 int index = relativeSpeedPosterior.getRealTermsAsIndex(speedPriorValue);
andrew@0 104 relativeSpeedPosterior.zero();
andrew@0 105 relativeSpeedPosterior.addGaussianShape(index, priorWidth, 0.8);
andrew@4 106 printf("speed adding to index for 1 = %f\n", relativeSpeedPosterior.getRealTermsAsIndex(1));
andrew@4 107 relativeSpeedPosterior.addToIndex(relativeSpeedPosterior.getRealTermsAsIndex(1), 1);
andrew@4 108
andrew@0 109 relativeSpeedPosterior.renormalise();
andrew@0 110 relativeSpeedPosterior.getMaximum();
andrew@0 111 relativeSpeedPrior.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 112 printf("BAYES STRUCTU ' SPEED PRIOR %f . index %i\n", speedPriorValue, index);
andrew@0 113
andrew@0 114 }
andrew@0 115
andrew@0 116
andrew@0 117 void BayesianArrayStructure::setPositionDistributionScalar(double f){
andrew@0 118 if (f > 0){
andrew@0 119 prior.scalar = f;
andrew@0 120 posterior.scalar = f;
andrew@0 121 likelihood.scalar = f;
andrew@0 122 }
andrew@0 123 }
andrew@0 124
andrew@0 125 void BayesianArrayStructure::simpleExample(){
andrew@0 126 relativeSpeedPosterior.getMaximum();
andrew@0 127 relativeSpeedPrior.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 128 }
andrew@0 129
andrew@0 130 void BayesianArrayStructure::copyPriorToPosterior(){
andrew@0 131
andrew@0 132 for (int i = 0;i < prior.arraySize;i++){
andrew@0 133 posterior.array[i] = prior.array[i];
andrew@0 134 }
andrew@0 135 }
andrew@0 136
andrew@0 137 void BayesianArrayStructure::setStartPlaying(){
andrew@0 138
andrew@0 139 lastEventTime = 0;
andrew@0 140 bestEstimate = 0;
andrew@0 141 lastBestEstimateUpdateTime = 0;
andrew@3 142
andrew@0 143 if (*realTimeMode)
andrew@0 144 lastBestEstimateUpdateTime = ofGetElapsedTimeMillis();
andrew@0 145 //cannot just be zero - offline bug
andrew@0 146 //printf("start playing - best estimate %f\n", lastBestEstimateUpdateTime);
andrew@0 147
andrew@0 148 resetArrays();
andrew@0 149 }
andrew@0 150
andrew@0 151 void BayesianArrayStructure::resetArrays(){
andrew@0 152 //called when we start playing
andrew@0 153
andrew@0 154 prior.zero();
andrew@0 155 likelihood.zero();
andrew@0 156 posterior.zero();
andrew@0 157
andrew@0 158 updateCounter = 0;
andrew@0 159
andrew@0 160 posterior.offset = -1000;
andrew@0 161 setNewDistributionOffsets(0);
andrew@0 162
andrew@0 163 int zeroIndex = posterior.getRealTermsAsIndex(0);
andrew@3 164 printf("ZERO INDEX %i\n", zeroIndex);
andrew@0 165
andrew@3 166 posterior.addGaussianShapeFromRealTime(0, 500, 1);//one way to add at x msec
andrew@3 167 posterior.addGaussianShape(posterior.getRealTermsAsIndex(10), 50, 1);//alternative way
andrew@3 168
andrew@3 169 //posterior.addToIndex(0, 1);
andrew@0 170 likelihood.addConstant(1);
andrew@0 171
andrew@0 172 updateCounter = 0;
andrew@0 173
andrew@0 174
andrew@0 175 printf("bayes reset arrays - best estimate %f\n", lastBestEstimateUpdateTime);
andrew@0 176
andrew@0 177 setSpeedPrior(speedPriorValue);
andrew@3 178 relativeSpeedPosterior.copyFromDynamicVector(relativeSpeedPrior);
andrew@3 179
andrew@0 180 }
andrew@0 181
andrew@0 182
andrew@0 183 void BayesianArrayStructure::zeroArrays(){
andrew@0 184 prior.zero();
andrew@0 185 likelihood.zero();
andrew@0 186 posterior.zero();
andrew@0 187
andrew@0 188 relativeSpeedPrior.zero();
andrew@0 189 relativeSpeedPosterior.zero();
andrew@0 190 relativeSpeedLikelihood.zero();
andrew@0 191
andrew@0 192 }
andrew@0 193
andrew@4 194 /*
andrew@0 195 void BayesianArrayStructure::updateTmpBestEstimate(const double& timeDifference){
andrew@0 196 //input is the time since the start of playing
andrew@0 197 // double timeDiff = ofGetElapsedTimeMillis() - lastEventTime;//lastBestEstimateUpdateTime;
andrew@0 198 double timeDiff = timeDifference;
andrew@0 199 if (*realTimeMode)
andrew@0 200 timeDiff = ofGetElapsedTimeMillis() - lastBestEstimateUpdateTime;
andrew@0 201
andrew@0 202 double tmp = relativeSpeedPosterior.getIntegratedEstimate();
andrew@0 203 tmpBestEstimate = posterior.getIndexInRealTerms(posterior.MAPestimate) + timeDiff*relativeSpeedPosterior.getIndexInRealTerms(relativeSpeedPosterior.integratedEstimate);
andrew@0 204 //
andrew@0 205 //printf("tmp best %f and best %f time diff %f posterior MAP %f at speed %f\n", 0Estimate, bestEstimate, timeDifference, posterior.getIndexInRealTerms(posterior.MAPestimate), relativeSpeedPosterior.getIndexInRealTerms(relativeSpeedPosterior.integratedEstimate));
andrew@0 206 //lastBestEstimateUpdateTime = ofGetElapsedTimeMillis();
andrew@0 207 }
andrew@4 208 */
andrew@4 209
andrew@0 210 void BayesianArrayStructure::updateBestEstimate(const double& timeDifference){
andrew@0 211 // double timeDiff = ofGetElapsedTimeMillis() - lastEventTime;//
andrew@3 212 double tmp = bestEstimate;
andrew@4 213 printf("best est routine: posterior offset %f\n", posterior.offset);
andrew@0 214
andrew@0 215 double timeDiff = timeDifference;
andrew@0 216
andrew@0 217 //Using timedifferencfe here will make it go wrong. Is time since beginning of playing
andrew@0 218
andrew@0 219 if (*realTimeMode)
andrew@0 220 timeDiff = ofGetElapsedTimeMillis() - lastBestEstimateUpdateTime;
andrew@0 221
andrew@0 222 //lastbest is time we started playing
andrew@4 223 /*
andrew@0 224 if (usingIntegratedTempoEstimate)
andrew@0 225 speedEstimateIndex = relativeSpeedPosterior.getIntegratedEstimate();
andrew@0 226 else
andrew@4 227 speedEstimateIndex = relativeSpeedPosterior.getMAPestimate();
andrew@4 228 */
andrew@4 229 speedEstimateIndex = getSpeedEstimateIndex();
andrew@0 230
andrew@0 231 speedEstimate = relativeSpeedPosterior.getIndexInRealTerms(speedEstimateIndex);
andrew@0 232 bestEstimate = posterior.getIndexInRealTerms(posterior.MAPestimate) + timeDiff*speedEstimate;
andrew@0 233
andrew@4 234 printf("best estimate update from %f to %f; time diff %f MAP %i = %f ms speed index %f est %f SpeedxTime %f\n", tmp, bestEstimate, timeDiff,
andrew@4 235 posterior.MAPestimate, posterior.getIndexInRealTerms(posterior.MAPestimate), speedEstimateIndex, speedEstimate, timeDiff*speedEstimate);
andrew@0 236 }
andrew@0 237
andrew@0 238 void BayesianArrayStructure::calculatePosterior(){
andrew@0 239 //posterior.doProduct(prior, likelihood);
andrew@4 240 assert(posterior.offset == prior.offset);
andrew@0 241
andrew@0 242 int i;
andrew@0 243 for (i = 0;i < posterior.length;i++){
andrew@0 244 posterior.array[i] = likelihood.array[i] * prior.array[i];
andrew@0 245 }
andrew@0 246
andrew@0 247 posterior.renormalise();
andrew@0 248
andrew@6 249 //tmp print stuff
andrew@6 250 printf("After CALC");
andrew@6 251 printPostOffset();
andrew@6 252
andrew@0 253 }
andrew@0 254
andrew@0 255
andrew@4 256 double BayesianArrayStructure::getSpeedEstimateIndex(){
andrew@4 257 if (usingIntegratedTempoEstimate)
andrew@4 258 return relativeSpeedPosterior.getIntegratedEstimate();
andrew@4 259 else
andrew@4 260 return relativeSpeedPosterior.getMAPestimate();
andrew@4 261 }
andrew@0 262
andrew@0 263
andrew@0 264 void BayesianArrayStructure::setNewDistributionOffsets(const double& newOffset){
andrew@4 265
andrew@4 266 printf("prior offset was %f now %f\n", prior.offset, newOffset);
andrew@4 267
andrew@0 268 prior.offset = newOffset;
andrew@0 269 likelihood.offset = newOffset;
andrew@4 270
andrew@0 271 //posterior.offset = newOffset;
andrew@0 272 }
andrew@0 273
andrew@0 274
andrew@6 275 void BayesianArrayStructure::updateBayesianDistributions(const double& newEventTime){
andrew@6 276
andrew@6 277 //NEED TO CHECK HERE THAT THEY HAVE THE SAME OFFSETS
andrew@6 278 prior.copyFromDynamicVector(posterior);//try the otehr way
andrew@6 279
andrew@6 280 //bayesianStruct.copyPriorToPosterior();
andrew@6 281 //need to get new MAP position and set the offset of the arrays
andrew@6 282 //currently bestEstimate is the approx for the new MAP position
andrew@6 283 // int tmpMap = bayesianStruct.posterior.getMAPestimate();
andrew@6 284
andrew@6 285 double timeDifference = newEventTime - lastEventTime;
andrew@6 286 // printf("updating distributions at time %f diff %f offset %f tmpmap est %i\n", newEventTime, timeDifference, bayesianStruct.posterior.offset, tmpMap);
andrew@6 287
andrew@6 288 //addnoise to the tempo distribution
andrew@6 289 //bayesianStruct.decaySpeedDistribution(timeDifference);
andrew@6 290
andrew@6 291 if (timeDifference > 50){
andrew@6 292 addGaussianNoiseToSpeedPosterior(timeDifference * 10.0 / 100.);
andrew@6 293 }
andrew@6 294
andrew@6 295 printPostOffset();
andrew@6 296
andrew@6 297 updateBestEstimate(timeDifference);
andrew@6 298 lastBestEstimateUpdateTime = newEventTime;//getTimeNow(timePlayed);
andrew@6 299
andrew@6 300
andrew@6 301 setNewDistributionOffsets(max(0., bestEstimate - (prior.scalar*prior.arraySize/2)));
andrew@6 302 crossUpdateArrays(posterior, relativeSpeedPosterior, timeDifference);
andrew@6 303
andrew@6 304 //i.e. using the same offset as prior
andrew@6 305 posterior.offset = prior.offset;//
andrew@6 306
andrew@6 307 // float tmpPrior = max(0., bestEstimate - (prior.scalar*prior.arraySize/2));// prior.offset = max(0., bestEstimate - (prior.scalar*prior.arraySize/2));
andrew@6 308 // printf("Using prior offset of %f not %f\n", tmpPrior, prior.offset);
andrew@6 309
andrew@6 310 lastEventTime = newEventTime;//lastEventTime = ofGetElapsedTimeMillis();
andrew@6 311
andrew@6 312 }
andrew@6 313
andrew@6 314
andrew@0 315 void BayesianArrayStructure::crossUpdateArrays(DynamicVector& position, DynamicVector& speed, double timeDifference){
andrew@0 316 //set the cutoff for offset of position first! XXX
andrew@0 317
andrew@0 318 // printf("time difference %f, ", timeDifference);
andrew@0 319
andrew@0 320 double timeDifferenceInPositionVectorUnits = timeDifference / prior.scalar;
andrew@0 321
andrew@4 322 printf("CROSS UPDATE time diff %f ms is %f units; ", timeDifference, timeDifferenceInPositionVectorUnits);
andrew@0 323 prior.zero();//kill prior
andrew@4 324
andrew@4 325 // calculateNewPriorOffset(timeDifference);//dioesnt do anything
andrew@4 326
andrew@5 327 // printf("new prior offset %f and post offset %f\n", prior.offset, posterior.offset);
andrew@0 328
andrew@0 329 if (timeDifferenceInPositionVectorUnits > crossUpdateTimeThreshold)
andrew@0 330 complexCrossUpdate(timeDifferenceInPositionVectorUnits);
andrew@0 331 else
andrew@0 332 translateByMaximumSpeed(timeDifferenceInPositionVectorUnits);
andrew@0 333
andrew@0 334
andrew@0 335 updateCounter++;
andrew@0 336 prior.renormalise();
andrew@0 337
andrew@0 338 }
andrew@0 339
andrew@0 340 void BayesianArrayStructure::complexCrossUpdate(const double& timeDifferenceInPositionVectorUnits){
andrew@4 341
andrew@5 342
andrew@5 343 printf("before cross c : posterior map is %i = %f ms time diff pos vec %f\n", posterior.getMAPestimate(), posterior.getIndexInRealTerms(prior.getMAPestimate()), timeDifferenceInPositionVectorUnits);
andrew@5 344
andrew@0 345 int distanceMoved, newPriorIndex;
andrew@0 346
andrew@0 347 double speedValue = relativeSpeedPosterior.offset;
andrew@0 348
andrew@0 349 for (int i = 0;i < relativeSpeedPosterior.arraySize;i++){
andrew@0 350
andrew@0 351 // double speedValue = relativeSpeedPosterior.getIndexInRealTerms(i);//so for scalar 0.01, 50 -> speed value of 0.5
andrew@0 352
andrew@0 353 //so we have moved
andrew@0 354 distanceMoved = round(timeDifferenceInPositionVectorUnits * speedValue);//round the value
andrew@5 355 // printf("Speed value %f time %f gives distance %i\n", speedValue, timeDifferenceInPositionVectorUnits, distanceMoved);
andrew@5 356
andrew@0 357 if (relativeSpeedPosterior.array[i] != 0){
andrew@4 358
andrew@0 359 double speedContribution = relativeSpeedPosterior.array[i];
andrew@0 360
andrew@4 361 // printf("speed [%i](val[%f]) gives %f moved %i in %f units \n", i, relativeSpeedPosterior.array[i], speedValue, distanceMoved, timeDifferenceInPositionVectorUnits);
andrew@0 362
andrew@4 363 //1/2/12 deleted line
andrew@4 364 newPriorIndex = posterior.offset - prior.offset + distanceMoved;//i.e. where post[0] goes to in terms of prior at this speed
andrew@5 365 int postIndex = 0;//index of posterior that will contribute
andrew@5 366
andrew@4 367 while (postIndex < posterior.arraySize && newPriorIndex < prior.arraySize){
andrew@4 368
andrew@4 369 //did use a for loop
andrew@4 370 // for (postIndex = 0;postIndex < posterior.arraySize;postIndex++){
andrew@0 371 //old posterior contributing to new prior
andrew@4 372
andrew@4 373 //would use this method
andrew@4 374 //newPriorIndex = postIndex + posterior.offset - prior.offset + distanceMoved;
andrew@0 375
andrew@4 376 if (newPriorIndex >= 0){
andrew@0 377 prior.addToIndex(newPriorIndex, posterior.array[postIndex]*speedContribution);
andrew@4 378 // printf("speed index %i new prior index %i post val %f speed contrib %f dist %i\n", i, newPriorIndex, posterior.array[postIndex], speedContribution, distanceMoved);
andrew@0 379 }
andrew@4 380 //but we actually do this for simplicity
andrew@4 381 newPriorIndex++;
andrew@4 382 postIndex++;
andrew@4 383 }//end for. now while
andrew@0 384
andrew@0 385
andrew@0 386 }//if not zero
andrew@5 387
andrew@5 388 speedValue += relativeSpeedPosterior.scalar;//optimised line
andrew@0 389 //as we wanted:
andrew@0 390 // double speedValue = relativeSpeedPosterior.getIndexInRealTerms(i);//so for scalar 0.01, 50 -> speed value of 0.5
andrew@5 391
andrew@0 392 }//end speed
andrew@5 393
andrew@5 394
andrew@5 395 printf("after cross c : prior map is %i = %f ms\n", prior.getMAPestimate(), prior.getIndexInRealTerms(prior.getMAPestimate()));
andrew@0 396 }
andrew@0 397
andrew@0 398
andrew@0 399
andrew@0 400 void BayesianArrayStructure::translateByMaximumSpeed(const double& timeDifferenceInPositionVectorUnits){
andrew@0 401
andrew@0 402 int distanceMoved, newPriorIndex;
andrew@0 403
andrew@0 404 double speedValue = relativeSpeedPosterior.getIndexInRealTerms(relativeSpeedPosterior.integratedEstimate);
andrew@0 405 //so for scalar 0.01, 50 -> speed value of 0.5
andrew@0 406 double speedContribution = relativeSpeedPosterior.array[relativeSpeedPosterior.integratedEstimate];
andrew@0 407 //so we have moved
andrew@0 408 distanceMoved = round(timeDifferenceInPositionVectorUnits * speedValue);//round the value
andrew@0 409 // printf("speed [%i] gives %f moved %i in %f units \n", i, speedValue, distanceMoved, timeDifferenceInPositionVectorUnits);
andrew@0 410
andrew@0 411 for (int postIndex = 0;postIndex < posterior.arraySize;postIndex++){
andrew@0 412 //old posterior contributing to new prior
andrew@0 413 newPriorIndex = postIndex + posterior.offset - prior.offset + distanceMoved;
andrew@0 414 if (newPriorIndex >= 0 && newPriorIndex < prior.arraySize){
andrew@0 415 prior.addToIndex(newPriorIndex, posterior.array[postIndex]*speedContribution);
andrew@0 416 }
andrew@0 417
andrew@0 418 }
andrew@0 419
andrew@0 420 }
andrew@0 421
andrew@0 422 void BayesianArrayStructure::addGaussianNoiseToSpeedPosterior(const double& std_dev){
andrew@0 423 tmpPosteriorForStorage.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 424
andrew@0 425 for (int i = 0;i < relativeSpeedPosterior.length;i++){
andrew@0 426 tmpPosteriorForStorage.addGaussianShape(i, std_dev, relativeSpeedPosterior.array[i]);
andrew@0 427 }
andrew@0 428
andrew@0 429 tmpPosteriorForStorage.renormalise();
andrew@0 430
andrew@0 431 relativeSpeedPosterior.copyFromDynamicVector(tmpPosteriorForStorage);
andrew@0 432 }
andrew@0 433
andrew@0 434
andrew@0 435 void BayesianArrayStructure::addTriangularNoiseToSpeedPosterior(const double& std_dev){
andrew@0 436 tmpPosteriorForStorage.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 437
andrew@0 438 for (int i = 0;i < relativeSpeedPosterior.length;i++){
andrew@0 439 //adding a linear amount depending on distance
andrew@0 440 tmpPosteriorForStorage.addTriangularShape(i, std_dev*2.0, relativeSpeedPosterior.array[i]);
andrew@0 441 }
andrew@0 442
andrew@0 443 tmpPosteriorForStorage.renormalise();
andrew@0 444
andrew@0 445 relativeSpeedPosterior.copyFromDynamicVector(tmpPosteriorForStorage);
andrew@0 446 }
andrew@0 447
andrew@0 448 void BayesianArrayStructure::calculateNewPriorOffset(const double& timeDifference){
andrew@0 449
andrew@4 450 // double maxSpeed = relativeSpeedPosterior.getIndexInRealTerms(relativeSpeedPosterior.integratedEstimate);
andrew@4 451
andrew@4 452 double maxSpeed = relativeSpeedPosterior.getIndexInRealTerms(getSpeedEstimateIndex());//either integrated or MAP
andrew@0 453 // printf("Maxspeed is %f\n", maxSpeed);
andrew@0 454
andrew@0 455 double priorMax = posterior.getMaximum();
andrew@0 456 double distanceTravelled = maxSpeed * (timeDifference / prior.scalar);
andrew@0 457 double newMaxLocation = posterior.MAPestimate + distanceTravelled;
andrew@0 458 // printf("MAP: %i, tim df %f, distance %f, new location %f\n", posterior.MAPestimate, timeDifference, distanceTravelled, newMaxLocation);
andrew@0 459
andrew@0 460 }
andrew@0 461
andrew@0 462
andrew@0 463 void BayesianArrayStructure::decaySpeedDistribution(double timeDifference){
andrew@0 464
andrew@0 465 // commented for the moment
andrew@0 466 double relativeAmount = max(1.0, timeDifference/1000.);
andrew@0 467 // printf("decay %f around %i \n", timeDifference, relativeSpeedPosterior.MAPestimate);
andrew@0 468 relativeAmount *= speedDecayAmount;
andrew@0 469 relativeSpeedPosterior.renormalise();
andrew@0 470 relativeSpeedPosterior.addGaussianShape(relativeSpeedPosterior.MAPestimate, speedDecayWidth, relativeAmount);
andrew@0 471
andrew@0 472 relativeSpeedPosterior.renormalise();
andrew@0 473 double newMax = relativeSpeedPosterior.getMaximum();
andrew@0 474
andrew@0 475 //old code
andrew@0 476 // relativeSpeedPosterior.addGaussianShape(relativeSpeedPosterior.MAPestimate, speedDecayWidth, 10);
andrew@0 477 //relativeSpeedPosterior.addConstant(1);
andrew@0 478
andrew@0 479 /*
andrew@0 480 relativeSpeedPrior.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 481 relativeSpeedLikelihood.zero();
andrew@0 482 relativeSpeedLikelihood.addConstant(0.2);
andrew@0 483 relativeSpeedLikelihood.addGaussianShape(relativeSpeedPosterior.maximumValue, speedDecayWidth, relativeAmount);
andrew@0 484 relativeSpeedPosterior.doProduct(relativeSpeedPrior, relativeSpeedLikelihood);
andrew@0 485 relativeSpeedPosterior.renormalise();
andrew@0 486 */
andrew@0 487
andrew@0 488
andrew@0 489
andrew@0 490 }
andrew@0 491
andrew@0 492 void BayesianArrayStructure::setLikelihoodToConstant(){
andrew@0 493 //set new likelihood
andrew@0 494 relativeSpeedLikelihood.zero();
andrew@0 495 relativeSpeedLikelihood.addConstant(speedLikelihoodNoise);
andrew@0 496 }
andrew@0 497
andrew@0 498
andrew@0 499 void BayesianArrayStructure::updateTempoLikelihood(const double& speedRatio, const double& matchFactor){
andrew@0 500
andrew@0 501 //speedratio is speed of played relative to the recording
andrew@0 502
andrew@0 503 double index = relativeSpeedLikelihood.getRealTermsAsIndex(speedRatio);
andrew@0 504 // printf("index of likelihood would be %f for ratio %f\n", index, speedRatio);
andrew@0 505 if (index >= 0 && index < relativeSpeedPrior.length){
andrew@0 506 relativeSpeedLikelihood.addGaussianShape(index , relativeSpeedLikelihoodStdDev, matchFactor);
andrew@0 507 }
andrew@0 508 }
andrew@0 509
andrew@0 510
andrew@0 511
andrew@0 512 void BayesianArrayStructure::updateTempoDistribution(){
andrew@0 513
andrew@0 514 //copy posterior to prior
andrew@0 515 relativeSpeedPrior.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 516
andrew@0 517 //update
andrew@0 518 relativeSpeedPosterior.doProduct(relativeSpeedPrior, relativeSpeedLikelihood);
andrew@0 519
andrew@0 520 //normalise
andrew@0 521 relativeSpeedPosterior.renormalise();
andrew@0 522
andrew@0 523 relativeSpeedPosterior.getMaximum();
andrew@0 524
andrew@0 525 //relativeSpeedPosterior.updateIntegratedEstimate(); - could be swayed when off to side
andrew@0 526 //so now limited to where there is room sround the MAP estimate
andrew@0 527 relativeSpeedPosterior.updateLimitedIntegratedEstimate();
andrew@0 528
andrew@0 529 speedEstimate = relativeSpeedPosterior.getIndexInRealTerms(relativeSpeedPosterior.integratedEstimate);
andrew@0 530 }
andrew@0 531
andrew@0 532
andrew@0 533 void BayesianArrayStructure::calculateTempoUpdate(){
andrew@0 534 //copy posterior to prior
andrew@0 535 relativeSpeedPrior.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 536
andrew@0 537 //update
andrew@0 538 relativeSpeedPosterior.doProduct(relativeSpeedPrior, relativeSpeedLikelihood);
andrew@0 539
andrew@0 540 //normalise
andrew@0 541 relativeSpeedPosterior.renormalise();
andrew@0 542
andrew@0 543 relativeSpeedPosterior.getMaximum();
andrew@0 544
andrew@0 545 }
andrew@0 546
andrew@0 547
andrew@0 548 void BayesianArrayStructure::drawArrays(){
andrew@0 549
andrew@0 550 //bayesArray.drawFloatArray(&bayesArray.prior[0], 0, 200);
andrew@0 551 //bayesArray.drawFloatArray(&bayesArray.prior[0], 0, 200);
andrew@0 552
andrew@0 553 int displaySize = prior.arraySize;
andrew@0 554 ofSetColor(0,0,255);
andrew@0 555 prior.drawVector(0, displaySize);
andrew@0 556 ofSetColor(0,255,0);
andrew@0 557 likelihood.drawVector(0, displaySize);
andrew@0 558 ofSetColor(255,0,255);
andrew@0 559 posterior.drawVector(0, displaySize);
andrew@0 560
andrew@0 561
andrew@0 562
andrew@0 563 }
andrew@0 564
andrew@0 565
andrew@0 566 void BayesianArrayStructure::drawTempoArrays(){
andrew@0 567 ofSetColor(0,0,255);
andrew@0 568 // relativeSpeedPrior.drawVector(0, relativeSpeedPrior.arraySize);
andrew@0 569
andrew@0 570 ofSetColor(0,150,255);
andrew@0 571 relativeSpeedLikelihood.drawVector(0, relativeSpeedLikelihood.arraySize);
andrew@0 572
andrew@0 573 // relativeSpeedLikelihood.drawConstrainedVector(0, 199, 0, 1000);// relativeSpeedLikelihood.arraySize);
andrew@0 574 ofSetColor(255,0,0);
andrew@0 575 relativeSpeedPosterior.drawVector(0, relativeSpeedPosterior.arraySize);
andrew@0 576
andrew@0 577 // ofSetColor(0,0,255);
andrew@0 578 // tmpPosteriorForStorage.drawVector(0, tmpPosteriorForStorage.arraySize);
andrew@0 579
andrew@0 580 ofSetColor(255,255, 255);
andrew@0 581 ofLine(screenWidth/2, 0, screenWidth/2, ofGetHeight());//middle of screen
andrew@0 582
andrew@0 583 ofSetColor(25, 0, 250);
andrew@0 584 double fractionOfScreen = ((double)relativeSpeedPosterior.integratedEstimate / relativeSpeedPosterior.length);
andrew@0 585 ofLine(screenWidth * fractionOfScreen, 0, screenWidth * fractionOfScreen, ofGetHeight());
andrew@0 586
andrew@0 587 ofSetColor(255, 0, 20);
andrew@0 588 fractionOfScreen = ((double)relativeSpeedPosterior.MAPestimate / relativeSpeedPosterior.length);
andrew@0 589 ofLine(screenWidth * fractionOfScreen, 0, screenWidth * fractionOfScreen, ofGetHeight());
andrew@0 590
andrew@0 591
andrew@0 592 }
andrew@0 593
andrew@0 594
andrew@0 595 void BayesianArrayStructure::drawArraysRelativeToTimeframe(const double& startTimeMillis, const double& endTimeMillis){
andrew@0 596
andrew@0 597 screenWidth = ofGetWidth();
andrew@0 598
andrew@0 599 int startArrayIndex = 0;
andrew@0 600
andrew@0 601 if (prior.getIndexInRealTerms(prior.arraySize-1) > startTimeMillis){
andrew@0 602 //i.e. the array is on the page
andrew@0 603
andrew@0 604 while (prior.getIndexInRealTerms(startArrayIndex) < startTimeMillis){
andrew@0 605 startArrayIndex++;
andrew@0 606 }
andrew@0 607 int endArrayIndex = prior.arraySize-1;
andrew@0 608 //could find constraints here
andrew@0 609 if (prior.getIndexInRealTerms(prior.arraySize-1) > endTimeMillis)
andrew@0 610 endArrayIndex = (floor)((endTimeMillis - prior.offset)/prior.scalar);
andrew@0 611
andrew@0 612 //so we need to figure where start and end array are on screen
andrew@0 613 int startScreenPosition, endScreenPosition;
andrew@0 614 double screenWidthMillis = endTimeMillis - startTimeMillis;
andrew@0 615
andrew@0 616 startScreenPosition = (prior.getIndexInRealTerms(startArrayIndex) - startTimeMillis)*screenWidth/screenWidthMillis;
andrew@0 617 endScreenPosition = (double)(prior.getIndexInRealTerms(endArrayIndex) - startTimeMillis)*screenWidth/screenWidthMillis;
andrew@0 618
andrew@0 619 ofSetColor(0,0,100);
andrew@0 620 string relativeString = " offset "+ofToString(prior.offset, 1);//starttimes("+ofToString(startTimeMillis)+", "+ofToString(endTimeMillis);
andrew@0 621 relativeString += ": index "+ofToString(startArrayIndex)+" , "+ofToString(endArrayIndex)+" [";
andrew@0 622 // relativeString += ofToString(prior.getIndexInRealTerms(endArrayIndex), 3)+"] (sc-width:"+ofToString(screenWidthMillis, 1)+") ";
andrew@0 623 relativeString += " mapped to screen "+ofToString(startScreenPosition)+" , "+ofToString(endScreenPosition);
andrew@0 624 // ofDrawBitmapString(relativeString, 100, 180);
andrew@0 625
andrew@0 626
andrew@0 627
andrew@0 628 ofSetColor(100,100,100);//255, 255, 0);
andrew@0 629 likelihood.drawConstrainedVector(startArrayIndex, endArrayIndex, startScreenPosition, endScreenPosition);
andrew@0 630
andrew@0 631 // ofSetColor(0,0,200);
andrew@0 632 ofSetColor(170,170,170);//00,200);
andrew@0 633 prior.drawConstrainedVector(startArrayIndex, endArrayIndex, startScreenPosition, endScreenPosition);
andrew@0 634
andrew@0 635 ofSetColor(0,0,150);
andrew@0 636 // ofSetColor(200, 0, 0);
andrew@0 637 posterior.drawConstrainedVector(startArrayIndex, endArrayIndex, startScreenPosition, endScreenPosition);
andrew@0 638
andrew@0 639
andrew@0 640 // ofSetColor(0, 200, 255);
andrew@0 641 // acceleration.drawConstrainedVector(startArrayIndex, endArrayIndex, startScreenPosition, endScreenPosition);
andrew@0 642
andrew@0 643
andrew@0 644 }
andrew@0 645
andrew@0 646 }
andrew@0 647
andrew@0 648
andrew@6 649 void BayesianArrayStructure::printPostOffset(){
andrew@6 650 double tmp = posterior.getMAPestimate();
andrew@6 651 printf(" MAP index %i post offset %f == %f ms\n", posterior.MAPestimate, posterior.offset, posterior.getIndexInRealTerms(posterior.MAPestimate));
andrew@6 652 }
andrew@6 653
andrew@0 654 /*
andrew@0 655
andrew@0 656 void BayesianArrayStructure::updateTempoDistribution(const double& speedRatio, const double& matchFactor){
andrew@0 657 //speedratio is speed of played relative to the recording
andrew@0 658
andrew@0 659 double index = relativeSpeedLikelihood.getRealTermsAsIndex(speedRatio);
andrew@0 660 // printf("\nindex of likelihood would be %f\n", index);
andrew@0 661 if (index >= 0 && index < relativeSpeedPrior.length){
andrew@0 662 //then we can do update
andrew@0 663
andrew@0 664 //set new likelihood
andrew@0 665 relativeSpeedLikelihood.zero();
andrew@0 666 relativeSpeedLikelihood.addConstant(speedLikelihoodNoise);
andrew@0 667
andrew@0 668 relativeSpeedLikelihood.addGaussianShape(index , 5, 0.5*matchFactor);
andrew@0 669
andrew@0 670
andrew@0 671 //copy posterior to prior
andrew@0 672 relativeSpeedPrior.copyFromDynamicVector(relativeSpeedPosterior);
andrew@0 673
andrew@0 674 //update
andrew@0 675 relativeSpeedPosterior.doProduct(relativeSpeedPrior, relativeSpeedLikelihood);
andrew@0 676
andrew@0 677 //normalise
andrew@0 678 relativeSpeedPosterior.renormalise();
andrew@0 679
andrew@0 680 relativeSpeedPosterior.getMaximum();
andrew@0 681 }//end if within range
andrew@0 682
andrew@0 683
andrew@0 684 }
andrew@0 685
andrew@0 686 */