Chris@366
|
1 /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*- vi:set ts=8 sts=4 sw=4: */
|
Chris@366
|
2 /*
|
Chris@366
|
3 Constant-Q library
|
Chris@366
|
4 Copyright (c) 2013-2014 Queen Mary, University of London
|
Chris@366
|
5
|
Chris@366
|
6 Permission is hereby granted, free of charge, to any person
|
Chris@366
|
7 obtaining a copy of this software and associated documentation
|
Chris@366
|
8 files (the "Software"), to deal in the Software without
|
Chris@366
|
9 restriction, including without limitation the rights to use, copy,
|
Chris@366
|
10 modify, merge, publish, distribute, sublicense, and/or sell copies
|
Chris@366
|
11 of the Software, and to permit persons to whom the Software is
|
Chris@366
|
12 furnished to do so, subject to the following conditions:
|
Chris@366
|
13
|
Chris@366
|
14 The above copyright notice and this permission notice shall be
|
Chris@366
|
15 included in all copies or substantial portions of the Software.
|
Chris@366
|
16
|
Chris@366
|
17 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
Chris@366
|
18 EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
Chris@366
|
19 MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
Chris@366
|
20 NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
|
Chris@366
|
21 CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
|
Chris@366
|
22 CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
|
Chris@366
|
23 WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
Chris@366
|
24
|
Chris@366
|
25 Except as contained in this notice, the names of the Centre for
|
Chris@366
|
26 Digital Music; Queen Mary, University of London; and Chris Cannam
|
Chris@366
|
27 shall not be used in advertising or otherwise to promote the sale,
|
Chris@366
|
28 use or other dealings in this Software without prior written
|
Chris@366
|
29 authorization.
|
Chris@366
|
30 */
|
Chris@366
|
31
|
Chris@366
|
32 #ifndef KAISER_WINDOW_H
|
Chris@366
|
33 #define KAISER_WINDOW_H
|
Chris@366
|
34
|
Chris@366
|
35 #include <vector>
|
Chris@366
|
36 #include <cmath>
|
Chris@366
|
37
|
Chris@366
|
38 #include "pi.h"
|
Chris@366
|
39
|
Chris@366
|
40 /**
|
Chris@366
|
41 * Kaiser window: A windower whose bandwidth and sidelobe height
|
Chris@366
|
42 * (signal-noise ratio) can be specified. These parameters are traded
|
Chris@366
|
43 * off against the window length.
|
Chris@366
|
44 */
|
Chris@366
|
45 class KaiserWindow
|
Chris@366
|
46 {
|
Chris@366
|
47 public:
|
Chris@366
|
48 struct Parameters {
|
Chris@366
|
49 int length;
|
Chris@366
|
50 double beta;
|
Chris@366
|
51 };
|
Chris@366
|
52
|
Chris@366
|
53 /**
|
Chris@366
|
54 * Construct a Kaiser windower with the given length and beta
|
Chris@366
|
55 * parameter.
|
Chris@366
|
56 */
|
Chris@366
|
57 KaiserWindow(Parameters p) : m_length(p.length), m_beta(p.beta) { init(); }
|
Chris@366
|
58
|
Chris@366
|
59 /**
|
Chris@366
|
60 * Construct a Kaiser windower with the given attenuation in dB
|
Chris@366
|
61 * and transition width in samples.
|
Chris@366
|
62 */
|
Chris@366
|
63 static KaiserWindow byTransitionWidth(double attenuation,
|
Chris@366
|
64 double transition) {
|
Chris@366
|
65 return KaiserWindow
|
Chris@366
|
66 (parametersForTransitionWidth(attenuation, transition));
|
Chris@366
|
67 }
|
Chris@366
|
68
|
Chris@366
|
69 /**
|
Chris@366
|
70 * Construct a Kaiser windower with the given attenuation in dB
|
Chris@366
|
71 * and transition bandwidth in Hz for the given samplerate.
|
Chris@366
|
72 */
|
Chris@366
|
73 static KaiserWindow byBandwidth(double attenuation,
|
Chris@366
|
74 double bandwidth,
|
Chris@366
|
75 double samplerate) {
|
Chris@366
|
76 return KaiserWindow
|
Chris@366
|
77 (parametersForBandwidth(attenuation, bandwidth, samplerate));
|
Chris@366
|
78 }
|
Chris@366
|
79
|
Chris@366
|
80 /**
|
Chris@366
|
81 * Obtain the parameters necessary for a Kaiser window of the
|
Chris@366
|
82 * given attenuation in dB and transition width in samples.
|
Chris@366
|
83 */
|
Chris@366
|
84 static Parameters parametersForTransitionWidth(double attenuation,
|
Chris@366
|
85 double transition);
|
Chris@366
|
86
|
Chris@366
|
87 /**
|
Chris@366
|
88 * Obtain the parameters necessary for a Kaiser window of the
|
Chris@366
|
89 * given attenuation in dB and transition bandwidth in Hz for the
|
Chris@366
|
90 * given samplerate.
|
Chris@366
|
91 */
|
Chris@366
|
92 static Parameters parametersForBandwidth(double attenuation,
|
Chris@366
|
93 double bandwidth,
|
Chris@366
|
94 double samplerate) {
|
Chris@366
|
95 return parametersForTransitionWidth
|
Chris@366
|
96 (attenuation, (bandwidth * 2 * M_PI) / samplerate);
|
Chris@366
|
97 }
|
Chris@366
|
98
|
Chris@366
|
99 int getLength() const {
|
Chris@366
|
100 return m_length;
|
Chris@366
|
101 }
|
Chris@366
|
102
|
Chris@366
|
103 const double *getWindow() const {
|
Chris@366
|
104 return m_window.data();
|
Chris@366
|
105 }
|
Chris@366
|
106
|
Chris@366
|
107 void cut(double *src) const {
|
Chris@366
|
108 cut(src, src);
|
Chris@366
|
109 }
|
Chris@366
|
110
|
Chris@366
|
111 void cut(const double *src, double *dst) const {
|
Chris@366
|
112 for (int i = 0; i < m_length; ++i) {
|
Chris@366
|
113 dst[i] = src[i] * m_window[i];
|
Chris@366
|
114 }
|
Chris@366
|
115 }
|
Chris@366
|
116
|
Chris@366
|
117 private:
|
Chris@366
|
118 int m_length;
|
Chris@366
|
119 double m_beta;
|
Chris@366
|
120 std::vector<double> m_window;
|
Chris@366
|
121
|
Chris@366
|
122 void init();
|
Chris@366
|
123 };
|
Chris@366
|
124
|
Chris@366
|
125 #endif
|