annotate constant-q-cpp/src/dsp/KaiserWindow.h @ 372:af71cbdab621 tip

Update bqvec code
author Chris Cannam
date Tue, 19 Nov 2019 10:13:32 +0000
parents 5d0a2ebb4d17
children
rev   line source
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