view src/AudioEventMatcher.h @ 45:d23685b9e766

Fixed the alignment error caluculations and added histogram plotting

/*
 *  AudioEventMatcher.h
 *  MultipleAudioMathcher
 *
 *  Created by Andrew on 31/01/2012.
 *  Copyright 2012 QMUL. All rights reserved.
 *
 */



#ifndef AUDIO_EVENT_MATCHER_H
#define AUDIO_EVENT_MATCHER_H



#include "ofMain.h"
#include "ChromaOnset.h"
#include "LiveAudioInput.h"
#include "ofxWindowRegion.h"
#include "BayesianArrayStructure.h"
#include "RecordedMultitrackAudio.h"
#include "DynamicVector.h"
#include "AccompanimentSynchroniser.h"
#include "TempoFollower.h"

static const int numberOfChannels = 4;

class AudioEventMatcher{
	
	public:
	AudioEventMatcher();
	
	void setArraySizes();
	
	void updatePosition();
	void updateBestAlignmentPosition();
	
	void draw();
	void drawBayesianDistributions();
	void drawPositionWindow();
	void drawTrackLikelihoods();
	void drawInfo();
	void drawAlignmentTimes();
	
	void setWindowDimensions();
	int getScreenWidthIndexOfEventTime(const double& time);
	int getTimeNow();
	
	void newPitchEvent(const int& channel, const double& pitchIn, const double& timeIn);
	void newKickEvent(const double& timeIn);
	void newKickEvent(const int& channel, const double& timeIn);
	void newSnareEvent(const double& timeIn);
	void newSnareEvent(const int& channel, const double& timeIn);
	
	void newChromaEvent(const int& channel, float* chromaIn, const double& timeIn);

	
	void matchNewPitchEvent(const int& channel, const double& pitchIn, const double& timeIn);
	void matchNewOnsetEvent(const int& channel, const double& timeIn);
	void matchNewChromaEvent(const int& channel, float* chroma, const double& timeIn);

	double getChromaDotProductDistance(float* chromaOne, float* chromaTwo);
	double getChromaEuclideanDistance(float* chromaOne, float* chromaTwo);
	bool useChromaDotProduct;
	
	BayesianArrayStructure bayesianStruct;//hold the probability distriubtions
	
	LiveAudioInput liveInput;//hold the new events that come in
	RecordedMultitrackAudio recordedTracks;
	
	void loadAudioFiles();
	void rescue();//flat prior
	void windowResized(const int& w, const int& h);
	
	ofxWindowRegion bayesTempoWindow;
	ofxWindowRegion bayesPositionWindow;
	ofxWindowRegion bayesLikelihoodWindow;

	bool checkMatch(const double& recordedPitch, const double& livePitch);
	double getPitchDistance(const double& pitchOne, const double& pitchTwo, const double& scale);
	
	void startPlaying();
	void stopPlaying();
	bool startedPlaying;
	
	bool usingRealTime;
	double recentPitch, recentPitchEventTime;
	
	DynamicVector likelihoodVisualisation[numberOfChannels];
	DynamicVector recentPriors[numberOfChannels];
	//DynamicVector recentPrior;
	DynamicVector projectedPrior;
	
	double currentAlignmentPosition;
	double lastAlignmentTime;
	
	double recentEventTime[numberOfChannels];
	int startTime;
	int currentAlignmentTime;
	
	double screenStartTimeMillis, screenEndTimeMillis, screenWidthMillis;
	bool followingLiveInput;
	void setScreenDisplayTimes();
	
	AccompanimentSynchroniser synchroniser;
	
	//params
	double onsetLikelihoodWidth;
	double onsetLikelihoodToNoise;
	double pitchLikelihoodToNoise;//more noise
	double chromaLikelihoodToNoise;
	
	double pitchOfNearestMatch;//for viz purposes
	double distanceOfNearestMatch;
	
	double chromaLikelihoodWidth;
	
	TempoFollower temporal;
	TempoFollower recordedTempoData;
	void calculateRecordedTempoData();
	void setTempoPrior(double tempo);
	
	void setNextOnsetTime(const int&channel, int& time, int* indexForOnsets);
	int recordedTempoIndex;
	double recordedTempo;
	void updateRecordedTempo();
	void drawRecordedTempo();
	double currentSpeedRatio;
	void drawPlayingTempo();
	void setSpeedRatioDistribution(const double& speedRatio);
	
	double euclideanMaximumDistance;
	bool printingData;
};
#endif