comparison dsp/keydetection/GetKeyMode.cpp @ 15:10c3f9df4a07

Change GetKeyMode to number bins with 1 = C, and shift chroma so that notes are centred correctly.

 #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,
+  0.0126, 0.0121, 0.0086, 0.0364, 0.0623, 0.0279, 0.0275, 0.0414, 0.0186, 
   0.0173, 0.0248, 0.0145, 0.0364, 0.0631, 0.0262, 0.0129, 0.0150, 0.0098,
   0.0312, 0.0521, 0.0235, 0.0129, 0.0142, 0.0095, 0.0289, 0.0478, 0.0239};
 
 static double MinProfile[36] =
 { 0.0375, 0.0682, 0.0299, 0.0119, 0.0138, 0.0093, 0.0296, 0.0543, 0.0257,

 	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);
+	// Set C (= MIDI #12) as our base :
+	// This implies that key = 1 => Cmaj, key = 12 => Bmaj, key = 13 => Cmin, etc.
+	m_ChromaConfig.min = Pitch::getFrequencyForPitch
+		(12, 0, tuningFrequency);
+	m_ChromaConfig.max = Pitch::getFrequencyForPitch
+		(96, 0, tuningFrequency);
 
 	// The chromagram minimum pitch is 1/6 of a tone above A, two
 	// octaves below middle C (for a 36-bin chromagram).  The
 	// maximum pitch is four octaves higher.
-
-	m_ChromaConfig.min = Pitch::getFrequencyForPitch
+/*	m_ChromaConfig.min = Pitch::getFrequencyForPitch
 		(45, 1.f / 3.f, tuningFrequency);
 
 	m_ChromaConfig.max = m_ChromaConfig.min * 2;
 	m_ChromaConfig.max = m_ChromaConfig.max * 2;
 	m_ChromaConfig.max = m_ChromaConfig.max * 2;
 	m_ChromaConfig.max = m_ChromaConfig.max * 2;
-
+*/
 	std::cerr << "Chromagram range: " << m_ChromaConfig.min << " -> " << m_ChromaConfig.max << std::endl;
 
 	m_ChromaConfig.BPO = 36;
 	m_ChromaConfig.CQThresh = 0.0054;
 

 	//////////////////////////////////////////////
 	m_Decimator->process( PCMData, m_DecimatedBuffer);
 
 	m_ChrPointer = m_Chroma->process( m_DecimatedBuffer );		
 
+	
+	// Move bins such that the centre of the base note is in the middle of its three bins :
+	// Added 21.11.07 by Chris Sutton based on debugging with Katy Noland + comparison with Matlab equivalent.
+	MathUtilities::circShift( m_ChrPointer, m_BPO, 1);
+
 	std::cout << "raw chroma: ";
 	for (int ii = 0; ii < m_BPO; ++ii) {
 		std::cout << m_ChrPointer[ii] << " ";
 	}
 	std::cout << std::endl;

 		m_MinCorr[k] = krumCorr( m_MeanHPCP, MinProfile, m_BPO );
 
 		MathUtilities::circShift( MajProfile, m_BPO, 1 );
 		MathUtilities::circShift( MinProfile, m_BPO, 1 );
 	}
-
+	
 	for( k = 0; k < m_BPO; k++ )
 	{
 		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 );
-
-//	std::cout << "key pre-sorting: " << key << std::endl;
+	// '1 +' because we number keys 1-24, not 0-23.
+	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;