comparison dsp/keydetection/GetKeyMode.cpp @ 234:60bbe2d98066

* Various changes in progress for debugging key mode plugin

 //
 //////////////////////////////////////////////////////////////////////
 
 #include "GetKeyMode.h"
 #include "dsp/maths/MathUtilities.h"
+#include "base/Pitch.h"
+
+#include <iostream>
 
 // Chords profile
 static double MajProfile[36] = 
 { 0.0384, 0.0629, 0.0258, 0.0121, 0.0146, 0.0106, 0.0364, 0.0610, 0.0267,
   0.0126, 0.0121, 0.0086, 0.0364, 0.0623, 0.0279, 0.0275, 0.0414, 0.0186,

 
 //////////////////////////////////////////////////////////////////////
 // Construction/Destruction
 //////////////////////////////////////////////////////////////////////
 
-GetKeyMode::GetKeyMode( double hpcpAverage, double medianAverage )
+GetKeyMode::GetKeyMode( int sampleRate, float tuningFrequency,
+			double hpcpAverage, double medianAverage )
 :
 m_hpcpAverage( hpcpAverage ),
 m_medianAverage( medianAverage ),
 m_ChrPointer(0),
 m_DecimatedBuffer(0),

 m_SortedBuffer(0)
 {
 	m_DecimationFactor = 8;
 
 	//Chromagram configuration parameters
-	m_CromaConfig.isNormalised = 1;
-	m_CromaConfig.FS = lrint(44100.0/(double)m_DecimationFactor);
-	m_CromaConfig.min = 111.0641;
-	m_CromaConfig.max = 1.7770e+003;
-	m_CromaConfig.BPO = 36;
-	m_CromaConfig.CQThresh = 0.0054;
+	m_ChromaConfig.isNormalised = 1;
+	m_ChromaConfig.FS = lrint(sampleRate/(double)m_DecimationFactor);
+
+//	m_ChromaConfig.min = 111.0641;
+//	m_ChromaConfig.max = 1.7770e+003;
+
+	m_ChromaConfig.min = Pitch::getFrequencyForPitch
+		(12, 0, tuningFrequency);
+	m_ChromaConfig.max = Pitch::getFrequencyForPitch
+		(96, 0, tuningFrequency);
+
+	m_ChromaConfig.BPO = 36;
+	m_ChromaConfig.CQThresh = 0.0054;
 
 	//Chromagram inst.
-	m_Chroma = new Chromagram( m_CromaConfig );
+	m_Chroma = new Chromagram( m_ChromaConfig );
 
 	//Get calculated parameters from chroma object
 	m_ChromaFrameSize = m_Chroma->getFrameSize();
 	//override hopsize for this application
 	m_ChromaHopSize = m_ChromaFrameSize;//m_Chroma->GetHopSize();
-	m_BPO = m_CromaConfig.BPO;
+	m_BPO = m_ChromaConfig.BPO;
 
 	//Chromagram average and estimated key median filter lengths
-	m_ChromaBuffersize = (int)ceil( m_hpcpAverage * m_CromaConfig.FS/m_ChromaFrameSize );
-	m_MedianWinsize = (int)ceil( m_medianAverage * m_CromaConfig.FS/m_ChromaFrameSize );
+	m_ChromaBuffersize = (int)ceil( m_hpcpAverage * m_ChromaConfig.FS/m_ChromaFrameSize );
+	m_MedianWinsize = (int)ceil( m_medianAverage * m_ChromaConfig.FS/m_ChromaFrameSize );
 
 	//Reset counters
 	m_bufferindex = 0;
 	m_ChromaBufferFilling = 0;
 	m_MedianBufferFilling = 0;

 	//////////////////////////////////////////////
 	m_Decimator->process( PCMData, m_DecimatedBuffer);
 
 	m_ChrPointer = m_Chroma->process( m_DecimatedBuffer );		
 
+	std::cout << "raw chroma: ";
+	for (int ii = 0; ii < m_BPO; ++ii) {
+		std::cout << m_ChrPointer[ii] << " ";
+	}
+	std::cout << std::endl;
+
 	// populate hpcp values;
 	int cbidx;
 	for( j = 0; j < m_BPO; j++ )
 	{
 		cbidx = (m_bufferindex * m_BPO) + j;

 	{
 		m_Keys[k] = m_MajCorr[k];
 		m_Keys[k+m_BPO] = m_MinCorr[k];
 	}
 
+	std::cout << "raw keys: ";
+	for (int ii = 0; ii < 2*m_BPO; ++ii) {
+		std::cout << m_Keys[ii] << " ";
+	}
+	std::cout << std::endl;
 
 	double dummy;
-	key = 1 + (int)ceil( (double)MathUtilities::getMax( m_Keys, 2* m_BPO, &dummy )/3 );
+	key = /*1 +*/ (int)ceil( (double)MathUtilities::getMax( m_Keys, 2* m_BPO, &dummy )/3 );
+
+	std::cout << "key pre-sorting: " << key << std::endl;
 
 
 	//Median filtering
 
 	// track Median buffer initial filling

 
 
 	//quicksort 
 	qsort(m_SortedBuffer, m_MedianBufferFilling, sizeof(unsigned int), MathUtilities::compareInt);
 
+	std::cout << "sorted: ";
+	for (int ii = 0; ii < m_MedianBufferFilling; ++ii) {
+		std::cout << m_SortedBuffer[ii] << " ";
+	}
+	std::cout << std::endl;
+
 	int sortlength = m_MedianBufferFilling;
 	int midpoint = (int)ceil((double)sortlength/2);
 
 	if( midpoint <= 0 )
 		midpoint = 1;