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: #ifndef _TONALESTIMATOR_
cannam@0: #define _TONALESTIMATOR_
cannam@0: 
cannam@0: 
cannam@0: #include <valarray>
cannam@0: #include <numeric>
cannam@0: #include <algorithm>
cannam@0: #include <iostream>
cannam@0: 
cannam@0: class ChromaVector : public std::valarray<double>
cannam@0: {
cannam@0: public:
cannam@0: 	ChromaVector(size_t uSize = 12) : std::valarray<double>()
cannam@0: 	{ resize(uSize, 0.0f); }
cannam@0: 	
cannam@0: 	virtual ~ChromaVector() {};
cannam@0: 	
cannam@0: 	void printDebug()
cannam@0: 	{
Chris@189: 		for (int i = 0; i < int(size()); i++)
cannam@0: 		{
cannam@0: 			std::cout <<  (*this)[i] << ";";
cannam@0: 		}
cannam@0: 		
cannam@0: 		std::cout << std::endl;
cannam@0: 	}
cannam@0: 	
cannam@0: 	void normalizeL1()
cannam@0: 	{
cannam@0: 		// normalize the chroma vector (L1 norm)
cannam@0: 		double dSum = 0.0;
cannam@0: 	
cannam@60: 		for (size_t i = 0; i < 12; (dSum += std::abs((*this)[i++]))) ;
cannam@60: 		for (size_t i = 0; i < 12; dSum > 0.0000001?((*this)[i] /= dSum):(*this)[i]=0.0, i++) ;
cannam@0: 
cannam@0: 	}
cannam@74: 
cannam@74:     void clear()
cannam@74:     {
cannam@74:         for (size_t i = 0; i < 12; ++i) (*this)[i] = 0.0;
cannam@74:     }
cannam@74:         
cannam@0: 	
cannam@0: };
cannam@0: 
cannam@0: class TCSVector : public std::valarray<double>
cannam@0: {
cannam@0: public:
cannam@0: 	TCSVector() : std::valarray<double>()
cannam@0: 	{ resize(6, 0.0f); }
cannam@0: 	
cannam@0: 	virtual ~TCSVector() {};
cannam@0: 
cannam@0: 	void printDebug()
cannam@0: 	{
Chris@189: 		for (int i = 0; i < int(size()); i++)
cannam@0: 		{
cannam@0: 			std::cout <<  (*this)[i] << ";";
cannam@0: 		}
cannam@0: 		
cannam@0: 		std::cout << std::endl;
cannam@0: 	}
cannam@0: 	
cannam@0: 	double magnitude() const
cannam@0: 	{
cannam@0: 		double dMag = 0.0;
cannam@0: 		
cannam@0: 		for (size_t i = 0; i < 6; i++)
cannam@0: 		{
cannam@0: 			dMag += std::pow((*this)[i], 2.0);
cannam@0: 		}
cannam@0: 		
cannam@0: 		return std::sqrt(dMag);
cannam@0: 	}
cannam@0: 
cannam@0: };
cannam@0: 
cannam@0: 
cannam@0: 
cannam@0: class TonalEstimator
cannam@0: {
cannam@0: public:
cannam@0: 	TonalEstimator();
cannam@0: 	virtual ~TonalEstimator();
cannam@0: 	TCSVector transform2TCS(const ChromaVector& rVector);
cannam@0: protected:
cannam@0: 	std::valarray< std::valarray<double> > m_Basis;
cannam@0: };
cannam@0: 
cannam@0: #endif // _TONALESTIMATOR_