c@116
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
c@116
|
2 /*
|
c@116
|
3 Constant-Q library
|
c@116
|
4 Copyright (c) 2013-2014 Queen Mary, University of London
|
c@116
|
5
|
c@116
|
6 Permission is hereby granted, free of charge, to any person
|
c@116
|
7 obtaining a copy of this software and associated documentation
|
c@116
|
8 files (the "Software"), to deal in the Software without
|
c@116
|
9 restriction, including without limitation the rights to use, copy,
|
c@116
|
10 modify, merge, publish, distribute, sublicense, and/or sell copies
|
c@116
|
11 of the Software, and to permit persons to whom the Software is
|
c@116
|
12 furnished to do so, subject to the following conditions:
|
c@116
|
13
|
c@116
|
14 The above copyright notice and this permission notice shall be
|
c@116
|
15 included in all copies or substantial portions of the Software.
|
c@116
|
16
|
c@116
|
17 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
c@116
|
18 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
c@116
|
19 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
c@116
|
20 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
c@116
|
21 CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
|
c@116
|
22 CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
c@116
|
23 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
c@116
|
24
|
c@116
|
25 Except as contained in this notice, the names of the Centre for
|
c@116
|
26 Digital Music; Queen Mary, University of London; and Chris Cannam
|
c@116
|
27 shall not be used in advertising or otherwise to promote the sale,
|
c@116
|
28 use or other dealings in this Software without prior written
|
c@116
|
29 authorization.
|
c@116
|
30 */
|
c@116
|
31
|
c@116
|
32 #ifndef CQ_KERNEL_H
|
c@116
|
33 #define CQ_KERNEL_H
|
c@116
|
34
|
c@127
|
35 #include "CQParameters.h"
|
c@127
|
36
|
c@116
|
37 #include <vector>
|
c@116
|
38 #include <complex>
|
c@116
|
39
|
c@116
|
40 class FFT;
|
c@116
|
41
|
c@116
|
42 class CQKernel
|
c@116
|
43 {
|
c@116
|
44 public:
|
c@127
|
45 CQKernel(CQParameters params);
|
c@116
|
46 ~CQKernel();
|
c@147
|
47
|
c@147
|
48 bool isValid() const { return m_valid; }
|
c@116
|
49
|
c@116
|
50 struct Properties {
|
c@116
|
51 double sampleRate;
|
c@116
|
52 double maxFrequency;
|
c@116
|
53 double minFrequency;
|
c@116
|
54 int binsPerOctave;
|
c@116
|
55 int fftSize;
|
c@116
|
56 int fftHop;
|
c@116
|
57 int atomsPerFrame;
|
c@116
|
58 int atomSpacing;
|
c@116
|
59 int firstCentre;
|
c@116
|
60 int lastCentre;
|
c@116
|
61 double Q;
|
c@116
|
62 };
|
c@116
|
63
|
c@116
|
64 Properties getProperties() const { return m_p; }
|
c@116
|
65
|
c@116
|
66 std::vector<std::complex<double> > processForward
|
c@116
|
67 (const std::vector<std::complex<double> > &);
|
c@116
|
68
|
c@116
|
69 std::vector<std::complex<double> > processInverse
|
c@116
|
70 (const std::vector<std::complex<double> > &);
|
c@116
|
71
|
c@116
|
72 private:
|
c@127
|
73 const CQParameters m_inparams;
|
c@116
|
74 Properties m_p;
|
c@147
|
75 bool m_valid;
|
c@116
|
76 FFT *m_fft;
|
c@116
|
77
|
c@116
|
78 struct KernelMatrix {
|
c@116
|
79 std::vector<int> origin;
|
c@116
|
80 std::vector<std::vector<std::complex<double> > > data;
|
c@116
|
81 };
|
c@116
|
82 KernelMatrix m_kernel;
|
c@116
|
83
|
c@127
|
84 std::vector<double> makeWindow(int len) const;
|
c@147
|
85 bool generateKernel();
|
c@116
|
86 void finaliseKernel();
|
c@116
|
87 };
|
c@116
|
88
|
c@116
|
89 #endif
|