cannam@0: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
cannam@0: 
cannam@0: /*
cannam@0:     QM DSP Library
cannam@0: 
cannam@0:     Centre for Digital Music, Queen Mary, University of London.
cannam@0:     This file copyright 2006 Martin Gasser.
Chris@84: 
Chris@84:     This program is free software; you can redistribute it and/or
Chris@84:     modify it under the terms of the GNU General Public License as
Chris@84:     published by the Free Software Foundation; either version 2 of the
Chris@84:     License, or (at your option) any later version.  See the file
Chris@84:     COPYING included with this distribution for more information.
cannam@0: */
cannam@0: 
cannam@0: #include "TonalEstimator.h"
cannam@0: 
cannam@0: #include <cmath>
cannam@0: #include <iostream>
cannam@0: 
cannam@0: #ifndef PI
cannam@0: #define PI (3.14159265358979232846)
cannam@0: #endif
cannam@0: 
cannam@0: TonalEstimator::TonalEstimator()
cannam@0: {
cannam@0: 	m_Basis.resize(6);
cannam@0: 
cannam@0: 	int i = 0;
cannam@0: 	
cannam@0: 	
cannam@0: 	// circle of fifths
cannam@0: 	m_Basis[i].resize(12);
cannam@0: 	for (int iP = 0; iP < 12; iP++)
cannam@0: 	{
cannam@0: 		m_Basis[i][iP] = std::sin( (7.0 / 6.0) * iP * PI);
cannam@0: 	}
cannam@0: 	
cannam@0: 	i++;
cannam@0: 
cannam@0: 	m_Basis[i].resize(12);
cannam@0: 	for (int iP = 0; iP < 12; iP++)
cannam@0: 	{
cannam@0: 		m_Basis[i][iP] = std::cos( (7.0 / 6.0) * iP * PI);
cannam@0: 	}
cannam@0: 	
cannam@0: 	i++;
cannam@0: 	
cannam@0: 	
cannam@0: 	// circle of major thirds
cannam@0: 	m_Basis[i].resize(12);
cannam@0: 	for (int iP = 0; iP < 12; iP++)
cannam@0: 	{
cannam@0: 		m_Basis[i][iP] = 0.6 * std::sin( (2.0 / 3.0) * iP * PI);
cannam@0: 	}
cannam@0: 	
cannam@0: 	i++;
cannam@0: 
cannam@0: 	m_Basis[i].resize(12);
cannam@0: 	for (int iP = 0; iP < 12; iP++)
cannam@0: 	{
cannam@0: 		m_Basis[i][iP] = 0.6 * std::cos( (2.0 / 3.0) * iP * PI);
cannam@0: 	}
cannam@0: 
cannam@0: 	i++;
cannam@0: 
cannam@0: 
cannam@0: 	// circle of minor thirds
cannam@0: 	m_Basis[i].resize(12);
cannam@0: 	for (int iP = 0; iP < 12; iP++)
cannam@0: 	{
cannam@0: 		m_Basis[i][iP] = 1.1 * std::sin( (3.0 / 2.0) * iP * PI);
cannam@0: 	}
cannam@0: 	
cannam@0: 	i++;
cannam@0: 
cannam@0: 	m_Basis[i].resize(12);
cannam@0: 	for (int iP = 0; iP < 12; iP++)
cannam@0: 	{
cannam@0: 		m_Basis[i][iP] = 1.1 * std::cos( (3.0 / 2.0) * iP * PI);
cannam@0: 	}
cannam@0: 
cannam@0: }
cannam@0: 
cannam@0: TonalEstimator::~TonalEstimator()
cannam@0: {
cannam@0: }
cannam@0: 
cannam@0: TCSVector TonalEstimator::transform2TCS(const ChromaVector& rVector)
cannam@0: {
cannam@0: 	TCSVector vaRetVal;
cannam@0: 	vaRetVal.resize(6, 0.0);
cannam@0: 		
cannam@0: 	for (int i = 0; i < 6; i++)
cannam@0: 	{
cannam@0: 		for (int iP = 0; iP < 12; iP++)
cannam@0: 		{
cannam@0: 			vaRetVal[i] += m_Basis[i][iP] * rVector[iP];
cannam@0: 		}
cannam@0: 	}
cannam@0: 	
cannam@0: 	return vaRetVal;
cannam@0: }