annotate dsp/tonal/TonalEstimator.h @ 515:08bcc06c38ec tip master

Remove fast-math
author Chris Cannam <cannam@all-day-breakfast.com>
date Tue, 28 Jan 2020 15:27:37 +0000
parents 701233f8ed41
children
rev   line source
c@225 1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
c@225 2
c@225 3 /*
c@225 4 QM DSP Library
c@225 5
c@225 6 Centre for Digital Music, Queen Mary, University of London.
c@225 7 This file copyright 2006 Martin Gasser.
c@309 8
c@309 9 This program is free software; you can redistribute it and/or
c@309 10 modify it under the terms of the GNU General Public License as
c@309 11 published by the Free Software Foundation; either version 2 of the
c@309 12 License, or (at your option) any later version. See the file
c@309 13 COPYING included with this distribution for more information.
c@225 14 */
c@225 15
cannam@489 16 #ifndef QM_DSP_TONALESTIMATOR_H
cannam@489 17 #define QM_DSP_TONALESTIMATOR_H
c@225 18
c@225 19 #include <valarray>
c@225 20 #include <numeric>
c@225 21 #include <algorithm>
c@225 22 #include <iostream>
c@225 23
c@225 24 class ChromaVector : public std::valarray<double>
c@225 25 {
c@225 26 public:
cannam@482 27 ChromaVector(size_t uSize = 12) : std::valarray<double>() {
cannam@482 28 resize(uSize, 0.0f);
cannam@482 29 }
cannam@482 30
cannam@482 31 virtual ~ChromaVector() {};
cannam@482 32
cannam@482 33 void printDebug() {
cannam@482 34 for (int i = 0; i < int(size()); i++) {
cannam@482 35 std::cout << (*this)[i] << ";";
cannam@482 36 }
cannam@482 37 std::cout << std::endl;
cannam@482 38 }
cannam@482 39
cannam@482 40 void normalizeL1() {
cannam@482 41 // normalize the chroma vector (L1 norm)
cannam@482 42 double dSum = 0.0;
cannam@482 43
cannam@482 44 for (size_t i = 0; i < 12; (dSum += std::abs((*this)[i++]))) ;
cannam@482 45 for (size_t i = 0; i < 12; dSum > 0.0000001?((*this)[i] /= dSum):(*this)[i]=0.0, i++) ;
cannam@482 46 }
c@225 47
cannam@482 48 void clear() {
c@299 49 for (size_t i = 0; i < 12; ++i) (*this)[i] = 0.0;
c@299 50 }
c@225 51 };
c@225 52
c@225 53 class TCSVector : public std::valarray<double>
c@225 54 {
c@225 55 public:
cannam@482 56 TCSVector() : std::valarray<double>() {
cannam@482 57 resize(6, 0.0f);
cannam@482 58 }
cannam@482 59
cannam@482 60 virtual ~TCSVector() {};
c@225 61
cannam@482 62 void printDebug() {
cannam@482 63 for (int i = 0; i < int(size()); i++) {
cannam@482 64 std::cout << (*this)[i] << ";";
cannam@482 65 }
cannam@482 66 std::cout << std::endl;
cannam@482 67 }
cannam@482 68
cannam@482 69 double magnitude() const {
cannam@482 70 double dMag = 0.0;
cannam@482 71
cannam@482 72 for (size_t i = 0; i < 6; i++) {
cannam@482 73 dMag += std::pow((*this)[i], 2.0);
cannam@482 74 }
cannam@482 75
cannam@482 76 return std::sqrt(dMag);
cannam@482 77 }
c@225 78 };
c@225 79
c@225 80 class TonalEstimator
c@225 81 {
c@225 82 public:
cannam@482 83 TonalEstimator();
cannam@482 84 virtual ~TonalEstimator();
cannam@482 85 TCSVector transform2TCS(const ChromaVector& rVector);
cannam@482 86
c@225 87 protected:
cannam@482 88 std::valarray< std::valarray<double> > m_Basis;
c@225 89 };
c@225 90
c@225 91 #endif // _TONALESTIMATOR_