annotate bayesianArraySrc/DynamicVector.cpp @ 0:c4f9e49226eb

Initialising repository. Live osc input registered. Files analysed offline.
author Andrew N Robertson <andrew.robertson@eecs.qmul.ac.uk>
date Tue, 31 Jan 2012 13:54:17 +0000
parents
children 179c09199b3c
rev   line source
andrew@0 1 /*
andrew@0 2 * DynamicVector.cpp
andrew@0 3 * midiCannamReader
andrew@0 4 *
andrew@0 5 * Created by Andrew on 18/07/2011.
andrew@0 6 * Copyright 2011 QMUL. All rights reserved.
andrew@0 7 *
andrew@0 8 */
andrew@0 9
andrew@0 10
andrew@0 11 #include "DynamicVector.h"
andrew@0 12
andrew@0 13 DynamicVector::DynamicVector(){
andrew@0 14 length = 0;
andrew@0 15 arraySize = 0;
andrew@0 16 maximumValue = 0;
andrew@0 17 MAPestimate = 0;
andrew@0 18 offset = 0;
andrew@0 19 scalar = 1;
andrew@0 20 integratedEstimate = length/2;
andrew@0 21
andrew@0 22 gaussianLookupMean = (double) GAUSSIAN_LOOKUP_LENGTH/2;
andrew@0 23 gaussianLookupStdDev = (double)(GAUSSIAN_LOOKUP_LENGTH/16);
andrew@0 24 double factor = 1.0;//(1.0 / (gaussianLookupStdDev*sqrt(2*PI)) );//1.0;//-1.0/(2*PI*sqrt(gaussianLookupStdDev));
andrew@0 25 for (int i = 0;i < GAUSSIAN_LOOKUP_LENGTH;i++){
andrew@0 26 gaussianLookupTable[i] = factor*exp(-1.0*(i-gaussianLookupMean)*(i-gaussianLookupMean)/(2.0*gaussianLookupStdDev*gaussianLookupStdDev));
andrew@0 27 }
andrew@0 28
andrew@0 29 }
andrew@0 30
andrew@0 31 void DynamicVector::copyFromDynamicVector(const DynamicVector& dynamicVec){
andrew@0 32 if (dynamicVec.length == length){
andrew@0 33 for (int i = 0;i < length;i++)
andrew@0 34 array[i] = dynamicVec.array[i];
andrew@0 35 }
andrew@0 36 else{
andrew@0 37 printf("CANNOT COPY VECTORS OF NON SAME LENGTH!!\n");
andrew@0 38 }
andrew@0 39 }
andrew@0 40
andrew@0 41 void DynamicVector::createVector(int len){
andrew@0 42 array.clear();
andrew@0 43 for (int i = 0; i < len;i++){
andrew@0 44 array.push_back(0);
andrew@0 45 }
andrew@0 46 length = len;
andrew@0 47 arraySize = array.size();
andrew@0 48 integratedEstimate = length/2;
andrew@0 49 }
andrew@0 50
andrew@0 51
andrew@0 52 double DynamicVector::getMaximum(){
andrew@0 53 int i;
andrew@0 54 double max = 0;
andrew@0 55 for (i=0;i < length;i++){
andrew@0 56 if (array[i] > max){
andrew@0 57 max = array[i];
andrew@0 58 MAPestimate = i;
andrew@0 59 }
andrew@0 60 }
andrew@0 61 maximumValue = max;
andrew@0 62 return max;
andrew@0 63 }
andrew@0 64
andrew@0 65 double DynamicVector::getIntegratedEstimate(){
andrew@0 66 //returns the index of the integrated average - where the probability distribution is centred
andrew@0 67 integratedEstimate = 0;
andrew@0 68 double integratedTotal = 0;
andrew@0 69 for (int i = 0;i < length;i++){
andrew@0 70 integratedEstimate += array[i]*i;
andrew@0 71 integratedTotal += array[i];
andrew@0 72 }
andrew@0 73 if (integratedTotal > 0){
andrew@0 74 integratedEstimate /= integratedTotal;
andrew@0 75 }
andrew@0 76 return integratedEstimate;
andrew@0 77 }
andrew@0 78
andrew@0 79 void DynamicVector::updateIntegratedEstimate(){
andrew@0 80 //returns the index of the integrated average - where the probability distribution is centred
andrew@0 81 integratedEstimate = 0;
andrew@0 82 double integratedTotal = 0;
andrew@0 83 for (int i = 0;i < length;i++){
andrew@0 84 integratedEstimate += array[i]*i;
andrew@0 85 integratedTotal += array[i];
andrew@0 86 }
andrew@0 87 if (integratedTotal > 0){
andrew@0 88 integratedEstimate /= integratedTotal;
andrew@0 89 }
andrew@0 90
andrew@0 91 }
andrew@0 92
andrew@0 93 void DynamicVector::updateLimitedIntegratedEstimate(){
andrew@0 94 //returns the index of the integrated average - where the probability distribution is centred
andrew@0 95 //but limited round the MAP estimate
andrew@0 96 double tmp = getMaximum();
andrew@0 97 int limit = min(MAPestimate, length - MAPestimate);
andrew@0 98 int start = max(0, MAPestimate - limit);
andrew@0 99 int end = min(MAPestimate + limit, length-1);
andrew@0 100
andrew@0 101 integratedEstimate = 0;
andrew@0 102 double integratedTotal = 0;
andrew@0 103 for (int i = start;i <= end;i++){
andrew@0 104 integratedEstimate += array[i]*i;
andrew@0 105 integratedTotal += array[i];
andrew@0 106 }
andrew@0 107 if (integratedTotal > 0){
andrew@0 108 integratedEstimate /= integratedTotal;
andrew@0 109 }
andrew@0 110
andrew@0 111 }
andrew@0 112
andrew@0 113
andrew@0 114 void DynamicVector::zero(){
andrew@0 115 for (int i = 0;i < array.size();i++)
andrew@0 116 array[i] = 0;
andrew@0 117 }
andrew@0 118
andrew@0 119 void DynamicVector::renormalise(){
andrew@0 120 double tmpMax = getMaximum();
andrew@0 121 if (tmpMax > 0){
andrew@0 122 // printf("renormalise : max is %f and size is %i\n", tmpMax, arraySize);
andrew@0 123 for (int i = 0;i < array.size();i++)
andrew@0 124 array[i] /= tmpMax;
andrew@0 125
andrew@0 126 }
andrew@0 127 //printArray();
andrew@0 128 }
andrew@0 129
andrew@0 130 void DynamicVector::doProduct(DynamicVector& arrayOne, DynamicVector& arrayTwo){
andrew@0 131
andrew@0 132 for (int i = 0;i < arrayOne.length;i++)
andrew@0 133 array[i] = arrayOne.array[i] * arrayTwo.array[i];
andrew@0 134 }
andrew@0 135
andrew@0 136
andrew@0 137 void DynamicVector::printArray(){
andrew@0 138 for (int i = 0;i < arraySize;i++){
andrew@0 139 printf("[%i] = %f\n", i, array[i]);
andrew@0 140 }
andrew@0 141 }
andrew@0 142
andrew@0 143 void DynamicVector::translateDistribution(int translationIndex){
andrew@0 144 int tmpIndex;
andrew@0 145 DoubleVector tmpArray;
andrew@0 146 int i;
andrew@0 147
andrew@0 148 for (i=0;i < arraySize;i++){
andrew@0 149 tmpArray.push_back(array[i]);
andrew@0 150 }
andrew@0 151 //translate values
andrew@0 152 for (i=0;i < arraySize;i++){
andrew@0 153 tmpIndex = (i + translationIndex + arraySize)%arraySize;
andrew@0 154 array[tmpIndex] = tmpArray[i];
andrew@0 155 }
andrew@0 156 tmpArray.clear();
andrew@0 157 //now delete tmp array
andrew@0 158 }
andrew@0 159
andrew@0 160 void DynamicVector::addGaussianShape(const double& mean, const double& StdDev, double factor){
andrew@0 161
andrew@0 162 int i;
andrew@0 163 double std_dev_factor = (2*StdDev*StdDev);
andrew@0 164 factor *= (1/(StdDev*sqrt(2*PI)));
andrew@0 165 int maxVal = min((int) array.size(), (int)(mean + 4.8*StdDev));
andrew@0 166 int minVal = max(0, (int)(mean - 4.8*StdDev));
andrew@0 167
andrew@0 168 for (i=minVal;i < maxVal;i++){
andrew@0 169 array[i] += factor*exp(-1*(i-mean)*(i-mean)/(std_dev_factor));
andrew@0 170 }
andrew@0 171
andrew@0 172 // addGaussianShapeByLookupTable(mean, StdDev, factor);
andrew@0 173 }
andrew@0 174
andrew@0 175 void DynamicVector::addGaussianShapeByLookupTable(double& mean, double& StdDev, double factor){
andrew@0 176 int i;
andrew@0 177 int lookupIndex ;
andrew@0 178 factor *= (1/(StdDev*sqrt(2*PI)));
andrew@0 179 for (i=0;i<array.size()-1;i++){
andrew@0 180 lookupIndex = round(getLookupIndex(i, mean, StdDev));
andrew@0 181 array[i] += factor*gaussianLookupTable[lookupIndex];
andrew@0 182 }
andrew@0 183 //printf("ADDED GAUSSIAN SHAPE %i\n", (int)array.size());
andrew@0 184 }
andrew@0 185
andrew@0 186 double DynamicVector::getLookupIndex(const int& i, const double& mean, const double& StdDev){
andrew@0 187
andrew@0 188 double Z = ((double)i - mean)/StdDev;
andrew@0 189 double lookupIndex = Z*gaussianLookupStdDev + gaussianLookupMean;
andrew@0 190
andrew@0 191 if (lookupIndex < 0)
andrew@0 192 lookupIndex = 0;
andrew@0 193
andrew@0 194 if (lookupIndex >= GAUSSIAN_LOOKUP_LENGTH)
andrew@0 195 lookupIndex = GAUSSIAN_LOOKUP_LENGTH-1;
andrew@0 196
andrew@0 197 // (i - mean)*(i-mean)*(GAUSSIAN_LOOKUP_LENGTH*GAUSSIAN_LOOKUP_LENGTH/16.0)/(StdDev*StdDev);
andrew@0 198 return lookupIndex;
andrew@0 199 }
andrew@0 200
andrew@0 201 void DynamicVector::addTriangularShape(double mean, double width, double factor){
andrew@0 202 int i;
andrew@0 203
andrew@0 204 for (i= max(0., (double)(mean - width));i < min((mean+width), (double)array.size());i++){
andrew@0 205 array[i] += factor * abs(i - mean) / mean;
andrew@0 206 }
andrew@0 207
andrew@0 208 }
andrew@0 209
andrew@0 210 void DynamicVector::addConstant(const double& value){
andrew@0 211 for (int i=0;i<array.size();i++){
andrew@0 212 array[i] += value;
andrew@0 213 }
andrew@0 214 }
andrew@0 215
andrew@0 216
andrew@0 217 void DynamicVector::addToIndex(const int& index, const double& constant){
andrew@0 218 array[index] += constant;
andrew@0 219 }
andrew@0 220
andrew@0 221
andrew@0 222 double DynamicVector::getIndexInRealTerms(const int& index){
andrew@0 223 if (index < arraySize)
andrew@0 224 return (offset + scalar*index);
andrew@0 225 else
andrew@0 226 return 0;
andrew@0 227 }
andrew@0 228
andrew@0 229 double DynamicVector::getRealTermsAsIndex(double value){
andrew@0 230 value -= offset;
andrew@0 231 value /= scalar;
andrew@0 232
andrew@0 233 return value;
andrew@0 234
andrew@0 235 }
andrew@0 236
andrew@0 237 double DynamicVector::getValueAtMillis(const double& millis){
andrew@0 238
andrew@0 239 int index = round(getRealTermsAsIndex(millis));
andrew@0 240 if (index >= 0 && index < length)
andrew@0 241 return array[index];
andrew@0 242 else
andrew@0 243 return 0;
andrew@0 244 }
andrew@0 245
andrew@0 246 void DynamicVector::drawVector(const int& minIndex, const int& maxIndex){
andrew@0 247
andrew@0 248
andrew@0 249 double stepSize = ofGetWidth() / (double)(maxIndex - minIndex);
andrew@0 250 double screenHeight = (double) ofGetHeight();
andrew@0 251 double maxVal = getMaximum();
andrew@0 252
andrew@0 253 int startInt = max(1,minIndex+1);
andrew@0 254 int endInt = min(maxIndex, (int)array.size());
andrew@0 255 double heightConstant = screenHeight / maxVal;
andrew@0 256 int lastHeightPixel = heightConstant * (maxVal - array[startInt-1]);
andrew@0 257 int newHeightPixel;
andrew@0 258 for (int i = startInt;i < endInt;i++){
andrew@0 259 newHeightPixel = (int) heightConstant * (maxVal - array[i]);
andrew@0 260 ofLine (stepSize*(i-1), lastHeightPixel, stepSize*i, newHeightPixel);
andrew@0 261 lastHeightPixel = newHeightPixel;
andrew@0 262 }
andrew@0 263
andrew@0 264 }
andrew@0 265
andrew@0 266
andrew@0 267 void DynamicVector::drawVector(const int& minIndex, const int& maxIndex, ofxWindowRegion window){
andrew@0 268
andrew@0 269
andrew@0 270 double stepSize = window.width / (double)(maxIndex - minIndex);
andrew@0 271 double screenHeight = (double) window.height;
andrew@0 272 double maxVal = getMaximum();
andrew@0 273
andrew@0 274 int startInt = max(1,minIndex+1);
andrew@0 275 int endInt = min(maxIndex, (int)array.size());
andrew@0 276 double heightConstant = screenHeight / maxVal;
andrew@0 277 int lastHeightPixel = heightConstant * (maxVal - array[startInt-1]);
andrew@0 278 int newHeightPixel;
andrew@0 279 for (int i = startInt;i < endInt;i++){
andrew@0 280 newHeightPixel = (int) heightConstant * (maxVal - array[i]);
andrew@0 281 ofLine (window.x+stepSize*(i-1), window.y+lastHeightPixel, window.x+stepSize*i, window.y+newHeightPixel);
andrew@0 282 lastHeightPixel = newHeightPixel;
andrew@0 283 }
andrew@0 284
andrew@0 285 }
andrew@0 286
andrew@0 287
andrew@0 288 void DynamicVector::drawConstrainedVector(const int& minIndex, const int& maxIndex, const int& minScreenIndex, const int& maxScreenIndex){
andrew@0 289 //constrain the height and width
andrew@0 290
andrew@0 291 double stepSize = (maxScreenIndex - minScreenIndex) / (double)(maxIndex - minIndex);//step size in pixels per array bin
andrew@0 292 double screenHeight = ofGetHeight();
andrew@0 293 double maxVal = getMaximum();
andrew@0 294
andrew@0 295 //OPTIMIZE!! XXX could just add stepsize each time
andrew@0 296 //not add minindex each time
andrew@0 297 int i = max(1,minIndex+1);
andrew@0 298 // ofDrawBitmapString("i = "+ofToString(i)+" :: screen min: "+ofToString(minScreenIndex + stepSize*(i-minIndex-1)), 20, 640);
andrew@0 299
andrew@0 300 while ((minScreenIndex + stepSize*(i-minIndex)) < 0)
andrew@0 301 i++;//only draw what is on the screen
andrew@0 302
andrew@0 303 for ( ; i < min(maxIndex+1, (int)array.size());i++){
andrew@0 304 ofLine (minScreenIndex + (stepSize*(i-minIndex-1)), screenHeight * (1 - array[i-1] / maxVal),
andrew@0 305 minScreenIndex + (stepSize*(i-minIndex)), screenHeight * (1 - array[i] / maxVal) );
andrew@0 306
andrew@0 307 }
andrew@0 308
andrew@0 309 ofLine(minScreenIndex, screenHeight, minScreenIndex, screenHeight/2);
andrew@0 310 ofLine(maxScreenIndex, screenHeight, maxScreenIndex, screenHeight/2);
andrew@0 311
andrew@0 312 // ofDrawBitmapString(ofToString(stepSize, 2)+" "+ofToString(maxScreenIndex - minScreenIndex, 0), 20, 600);
andrew@0 313
andrew@0 314 }