/
DspBandpassFilter.cpp
107 lines (97 loc) · 3.09 KB
/
DspBandpassFilter.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
* Copyright 2009,2010 Reality Jockey, Ltd.
* info@rjdj.me
* http://rjdj.me/
*
* This file is part of ZenGarden.
*
* ZenGarden is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* ZenGarden is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with ZenGarden. If not, see <http://www.gnu.org/licenses/>.
*
*/
#include "DspBandpassFilter.h"
#include "PdGraph.h"
MessageObject *DspBandpassFilter::newObject(PdMessage *initMessage, PdGraph *graph) {
return new DspBandpassFilter(initMessage, graph);
}
DspBandpassFilter::DspBandpassFilter(PdMessage *initMessage, PdGraph *graph) : DspObject(3, 1, 0, 1, graph) {
sampleRate = graph->getSampleRate();
tap_0 = 0.0f;
tap_1 = 0.0f;
centerFrequency = initMessage->isFloat(0) ? initMessage->getFloat(0) : sampleRate/2.0f;
q = initMessage->isFloat(1) ? initMessage->getFloat(1) : 1.0f;
calculateFilterCoefficients(centerFrequency, q);
}
DspBandpassFilter::~DspBandpassFilter() {
// nothing to do
}
const char *DspBandpassFilter::getObjectLabel() {
return "bp~";
}
void DspBandpassFilter::calculateFilterCoefficients(float f, float q) {
float r, oneminusr, omega;
if (f < 0.001f) f = 10.0f;
if (q < 0.0f) q = 0.0f;
this->centerFrequency = f;
this->q = q;
omega = f * (2.0f * M_PI) / sampleRate;
if (q < 0.001) oneminusr = 1.0f;
else oneminusr = omega/q;
if (oneminusr > 1.0f) oneminusr = 1.0f;
r = 1.0f - oneminusr;
coef1 = 2.0f * sigbp_qcos(omega) * r;
coef2 = - r * r;
gain = 2.0f * oneminusr * (oneminusr + r * omega);
}
float DspBandpassFilter::sigbp_qcos(float f) {
if (f >= -(0.5f * M_PI) && f <= (0.5f * M_PI)) {
float g = f*f;
return (((g*g*g * (-1.0f/720.0f) + g*g*(1.0f/24.0f)) - g*0.5) + 1);
} else {
return 0.0f;
}
}
void DspBandpassFilter::processMessage(int inletIndex, PdMessage *message) {
switch (inletIndex) {
case 0: {
if (message->isSymbol(0, "clear")) {
tap_0 = 0.0f; // TODO(mhroth): how to handle filter resets?
tap_1 = 0.0f;
}
break;
}
case 1: {
if (message->isFloat(0)) {
calculateFilterCoefficients(message->getFloat(0), q);
}
break;
}
case 2: {
if (message->isFloat(0)) {
calculateFilterCoefficients(centerFrequency, message->getFloat(0));
}
break;
}
default: {
break;
}
}
}
void DspBandpassFilter::processDspWithIndex(int fromIndex, int toIndex) {
for (int i = fromIndex; i < toIndex; i++) {
dspBufferAtOutlet0[i] = dspBufferAtInlet0[i] + (coef1 * tap_0) + (coef2 * tap_1);
tap_1 = tap_0;
tap_0 = dspBufferAtOutlet0[i];
dspBufferAtOutlet0[i] *= gain;
}
}