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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
#include "audio_effect_eq.h"
#include "servers/audio_server.h"
void AudioEffectEQInstance::process(const AudioFrame *p_src_frames,AudioFrame *p_dst_frames,int p_frame_count) {
int band_count = bands[0].size();
EQ::BandProcess *proc_l = bands[0].ptr();
EQ::BandProcess *proc_r = bands[1].ptr();
float *bgain = gains.ptr();
for(int i=0;i<band_count;i++) {
bgain[i]=Math::db2linear(base->gain[i]);
}
for(int i=0;i<p_frame_count;i++) {
AudioFrame src = p_src_frames[i];
AudioFrame dst = AudioFrame(0,0);
for(int j=0;j<band_count;j++) {
float l = src.l;
float r = src.r;
proc_l[j].process_one(l);
proc_r[j].process_one(r);
dst.l+=l * bgain[j];
dst.r+=r * bgain[j];
}
p_dst_frames[i]=dst;
}
}
Ref<AudioEffectInstance> AudioEffectEQ::instance() {
Ref<AudioEffectEQInstance> ins;
ins.instance();
ins->base=Ref<AudioEffectEQ>(this);
ins->gains.resize(eq.get_band_count());
for(int i=0;i<2;i++) {
ins->bands[i].resize(eq.get_band_count());
for(int j=0;j<ins->bands[i].size();j++) {
ins->bands[i][j]=eq.get_band_processor(j);
}
}
return ins;
}
void AudioEffectEQ::set_band_gain_db(int p_band,float p_volume) {
ERR_FAIL_INDEX(p_band,gain.size());
gain[p_band]=p_volume;
}
float AudioEffectEQ::get_band_gain_db(int p_band) const {
ERR_FAIL_INDEX_V(p_band,gain.size(),0);
return gain[p_band];
}
int AudioEffectEQ::get_band_count() const {
return gain.size();
}
bool AudioEffectEQ::_set(const StringName& p_name, const Variant& p_value) {
const Map<StringName,int>::Element *E=prop_band_map.find(p_name);
if (E) {
set_band_gain_db(E->get(),p_value);
return true;
}
return false;
}
bool AudioEffectEQ::_get(const StringName& p_name,Variant &r_ret) const{
const Map<StringName,int>::Element *E=prop_band_map.find(p_name);
if (E) {
r_ret=get_band_gain_db(E->get());
return true;
}
return false;
}
void AudioEffectEQ::_get_property_list( List<PropertyInfo> *p_list) const{
for(int i=0;i<band_names.size();i++) {
p_list->push_back(PropertyInfo(Variant::REAL,band_names[i],PROPERTY_HINT_RANGE,"-60,24,0.1"));
}
}
void AudioEffectEQ::_bind_methods() {
ClassDB::bind_method(D_METHOD("set_band_gain_db","band_idx","volume_db"),&AudioEffectEQ::set_band_gain_db);
ClassDB::bind_method(D_METHOD("get_band_gain_db","band_idx"),&AudioEffectEQ::get_band_gain_db);
ClassDB::bind_method(D_METHOD("get_band_count"),&AudioEffectEQ::get_band_count);
}
AudioEffectEQ::AudioEffectEQ(EQ::Preset p_preset)
{
eq.set_mix_rate(AudioServer::get_singleton()->get_mix_rate());
eq.set_preset_band_mode(p_preset);
gain.resize(eq.get_band_count());
for(int i=0;i<gain.size();i++) {
gain[i]=0.0;
String name = "band_db/"+itos(eq.get_band_frequency(i))+"_hz";
prop_band_map[name]=i;
band_names.push_back(name);
}
}
|