cannam@484
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
cannam@475
|
2 /*
|
cannam@475
|
3 Copyright (c) 2005 Centre for Digital Music ( C4DM )
|
cannam@475
|
4 Queen Mary Univesrity of London
|
cannam@475
|
5
|
cannam@475
|
6 This program is free software; you can redistribute it and/or
|
cannam@475
|
7 modify it under the terms of the GNU General Public License as
|
cannam@475
|
8 published by the Free Software Foundation; either version 2 of the
|
cannam@475
|
9 License, or (at your option) any later version. See the file
|
cannam@475
|
10 COPYING included with this distribution for more information.
|
cannam@475
|
11 */
|
cannam@475
|
12
|
cannam@489
|
13 #ifndef QM_DSP_GETKEYMODE_H
|
cannam@489
|
14 #define QM_DSP_GETKEYMODE_H
|
cannam@475
|
15
|
cannam@475
|
16
|
cannam@475
|
17 #include "dsp/rateconversion/Decimator.h"
|
cannam@475
|
18 #include "dsp/chromagram/Chromagram.h"
|
cannam@475
|
19
|
cannam@475
|
20
|
cannam@475
|
21 class GetKeyMode
|
cannam@475
|
22 {
|
cannam@475
|
23 public:
|
cannam@478
|
24 GetKeyMode( int sampleRate, float tuningFrequency,
|
cannam@478
|
25 double hpcpAverage, double medianAverage );
|
cannam@475
|
26
|
cannam@478
|
27 virtual ~GetKeyMode();
|
cannam@475
|
28
|
cannam@478
|
29 int process( double* PCMData );
|
cannam@475
|
30
|
cannam@478
|
31 double krumCorr( const double *pDataNorm, const double *pProfileNorm,
|
cannam@499
|
32 int shiftProfile, int length );
|
cannam@475
|
33
|
cannam@499
|
34 int getBlockSize() { return m_chromaFrameSize * m_decimationFactor; }
|
cannam@499
|
35 int getHopSize() { return m_chromaHopSize * m_decimationFactor; }
|
cannam@475
|
36
|
cannam@499
|
37 double* getChroma() { return m_chrPointer; }
|
cannam@499
|
38 int getChromaSize();
|
cannam@475
|
39
|
cannam@499
|
40 double* getMeanHPCP() { return m_meanHPCP; }
|
cannam@475
|
41
|
cannam@478
|
42 double* getKeyStrengths();
|
cannam@475
|
43
|
cannam@478
|
44 bool isModeMinor( int key );
|
cannam@475
|
45
|
cannam@475
|
46 protected:
|
cannam@475
|
47
|
cannam@478
|
48 double m_hpcpAverage;
|
cannam@478
|
49 double m_medianAverage;
|
cannam@499
|
50 int m_decimationFactor;
|
cannam@475
|
51
|
cannam@499
|
52 // Decimator (fixed)
|
cannam@499
|
53 Decimator* m_decimator;
|
cannam@475
|
54
|
cannam@499
|
55 // Chroma configuration
|
cannam@499
|
56 ChromaConfig m_chromaConfig;
|
cannam@475
|
57
|
cannam@499
|
58 // Chromagram object
|
cannam@499
|
59 Chromagram* m_chroma;
|
cannam@475
|
60
|
cannam@499
|
61 // Chromagram output pointer
|
cannam@499
|
62 double* m_chrPointer;
|
cannam@475
|
63
|
cannam@499
|
64 // Framesize
|
cannam@499
|
65 int m_chromaFrameSize;
|
cannam@475
|
66
|
cannam@499
|
67 // Hop
|
cannam@499
|
68 int m_chromaHopSize;
|
cannam@475
|
69
|
cannam@499
|
70 int m_chromaBufferSize;
|
cannam@499
|
71 int m_medianWinSize;
|
cannam@478
|
72
|
cannam@499
|
73 int m_bufferIndex;
|
cannam@499
|
74 int m_chromaBufferFilling;
|
cannam@499
|
75 int m_medianBufferFilling;
|
cannam@475
|
76
|
cannam@499
|
77 double* m_decimatedBuffer;
|
cannam@499
|
78 double* m_chromaBuffer;
|
cannam@499
|
79 double* m_meanHPCP;
|
cannam@475
|
80
|
cannam@499
|
81 double* m_majProfileNorm;
|
cannam@499
|
82 double* m_minProfileNorm;
|
cannam@499
|
83 double* m_majCorr;
|
cannam@499
|
84 double* m_minCorr;
|
cannam@499
|
85 int* m_medianFilterBuffer;
|
cannam@499
|
86 int* m_sortedBuffer;
|
cannam@475
|
87
|
cannam@478
|
88 double *m_keyStrengths;
|
cannam@475
|
89 };
|
cannam@475
|
90
|
cannam@499
|
91 #endif // !defined QM_DSP_GETKEYMODE_H
|