virtualx-engine/servers/audio/effects/audio_effect_chorus.cpp

365 lines
12 KiB
C++

#include "audio_effect_chorus.h"
#include "servers/audio_server.h"
#include "math_funcs.h"
void AudioEffectChorusInstance::process(const AudioFrame *p_src_frames,AudioFrame *p_dst_frames,int p_frame_count) {
int todo = p_frame_count;
while(todo) {
int to_mix = MIN(todo,256); //can't mix too much
_process_chunk(p_src_frames,p_dst_frames,to_mix);
p_src_frames+=to_mix;
p_dst_frames+=to_mix;
todo-=to_mix;
}
}
void AudioEffectChorusInstance::_process_chunk(const AudioFrame *p_src_frames,AudioFrame *p_dst_frames,int p_frame_count) {
//fill ringbuffer
for(int i=0;i<p_frame_count;i++) {
audio_buffer[(buffer_pos+i)&buffer_mask]=p_src_frames[i];
p_dst_frames[i]=p_src_frames[i]*base->dry;
}
float mix_rate = AudioServer::get_singleton()->get_mix_rate();
/* process voices */
for (int vc=0;vc<base->voice_count;vc++) {
AudioEffectChorus::Voice &v=base->voice[vc];
double time_to_mix=(float)p_frame_count/mix_rate;
double cycles_to_mix=time_to_mix*v.rate;
unsigned int local_rb_pos=buffer_pos;
AudioFrame *dst_buff=p_dst_frames;
AudioFrame *rb_buff=audio_buffer.ptr();
double delay_msec=v.delay;
unsigned int delay_frames=Math::fast_ftoi((delay_msec/1000.0)*mix_rate);
float max_depth_frames=(v.depth/1000.0)*mix_rate;
uint64_t local_cycles=cycles[vc];
uint64_t increment=llrint(cycles_to_mix/(double)p_frame_count*(double)(1<<AudioEffectChorus::CYCLES_FRAC));
//check the LFO doesnt read ahead of the write pos
if ((((int)max_depth_frames)+10)>delay_frames) { //10 as some threshold to avoid precision stuff
delay_frames+=(int)max_depth_frames-delay_frames;
delay_frames+=10; //threshold to avoid precision stuff
}
//low pass filter
if (v.cutoff==0)
continue;
float auxlp=expf(-2.0*Math_PI*v.cutoff/mix_rate);
float c1=1.0-auxlp;
float c2=auxlp;
AudioFrame h=filter_h[vc];
if (v.cutoff>=AudioEffectChorus::MS_CUTOFF_MAX) {
c1=1.0; c2=0.0;
}
//vol modifier
AudioFrame vol_modifier=AudioFrame(base->wet,base->wet) * Math::db2linear(v.level);
vol_modifier.l*=CLAMP( 1.0 - v.pan, 0, 1);
vol_modifier.r*=CLAMP( 1.0 + v.pan, 0, 1);
for (int i=0;i<p_frame_count;i++) {
/** COMPUTE WAVEFORM **/
float phase=(float)(local_cycles&AudioEffectChorus::CYCLES_MASK)/(float)(1<<AudioEffectChorus::CYCLES_FRAC);
float wave_delay=sinf(phase*2.0*Math_PI)*max_depth_frames;
int wave_delay_frames=lrint(floor(wave_delay));
float wave_delay_frac=wave_delay-(float)wave_delay_frames;
/** COMPUTE RINGBUFFER POS**/
unsigned int rb_source=local_rb_pos;
rb_source-=delay_frames;
rb_source-=wave_delay_frames;
/** READ FROM RINGBUFFER, LINEARLY INTERPOLATE */
AudioFrame val=rb_buff[rb_source&buffer_mask];
AudioFrame val_next=rb_buff[(rb_source-1)&buffer_mask];
val+=(val_next-val)*wave_delay_frac;
val=val*c1+h*c2;
h=val;
/** MIX VALUE TO OUTPUT **/
dst_buff[i]+=val*vol_modifier;
local_cycles+=increment;
local_rb_pos++;
}
filter_h[vc]=h;
cycles[vc]+=Math::fast_ftoi(cycles_to_mix*(double)(1<<AudioEffectChorus::CYCLES_FRAC));
}
buffer_pos+=p_frame_count;
}
Ref<AudioEffectInstance> AudioEffectChorus::instance() {
Ref<AudioEffectChorusInstance> ins;
ins.instance();
ins->base=Ref<AudioEffectChorus>(this);
for(int i=0;i<4;i++) {
ins->filter_h[i]=AudioFrame(0,0);
ins->cycles[i]=0;
}
float ring_buffer_max_size=AudioEffectChorus::MAX_DELAY_MS+AudioEffectChorus::MAX_DEPTH_MS+AudioEffectChorus::MAX_WIDTH_MS;
ring_buffer_max_size*=2; //just to avoid complications
ring_buffer_max_size/=1000.0;//convert to seconds
ring_buffer_max_size*=AudioServer::get_singleton()->get_mix_rate();
int ringbuff_size=ring_buffer_max_size;
int bits=0;
while(ringbuff_size>0) {
bits++;
ringbuff_size/=2;
}
ringbuff_size=1<<bits;
ins->buffer_mask=ringbuff_size-1;
ins->buffer_pos=0;
ins->audio_buffer.resize(ringbuff_size);
for(int i=0;i<ringbuff_size;i++) {
ins->audio_buffer[i]=AudioFrame(0,0);
}
return ins;
}
void AudioEffectChorus::set_voice_count(int p_voices) {
ERR_FAIL_COND(p_voices<1 || p_voices>=MAX_VOICES);
voice_count=p_voices;
_change_notify();
}
int AudioEffectChorus::get_voice_count() const{
return voice_count;
}
void AudioEffectChorus::set_voice_delay_ms(int p_voice,float p_delay_ms){
ERR_FAIL_INDEX(p_voice,MAX_VOICES);
voice[p_voice].delay=p_delay_ms;
}
float AudioEffectChorus::get_voice_delay_ms(int p_voice) const{
ERR_FAIL_INDEX_V(p_voice,MAX_VOICES,0);
return voice[p_voice].delay;
}
void AudioEffectChorus::set_voice_rate_hz(int p_voice,float p_rate_hz){
ERR_FAIL_INDEX(p_voice,MAX_VOICES);
voice[p_voice].rate=p_rate_hz;
}
float AudioEffectChorus::get_voice_rate_hz(int p_voice) const{
ERR_FAIL_INDEX_V(p_voice,MAX_VOICES,0);
return voice[p_voice].rate;
}
void AudioEffectChorus::set_voice_depth_ms(int p_voice,float p_depth_ms){
ERR_FAIL_INDEX(p_voice,MAX_VOICES);
voice[p_voice].depth=p_depth_ms;
}
float AudioEffectChorus::get_voice_depth_ms(int p_voice) const{
ERR_FAIL_INDEX_V(p_voice,MAX_VOICES,0);
return voice[p_voice].depth;
}
void AudioEffectChorus::set_voice_level_db(int p_voice,float p_level_db){
ERR_FAIL_INDEX(p_voice,MAX_VOICES);
voice[p_voice].level=p_level_db;
}
float AudioEffectChorus::get_voice_level_db(int p_voice) const{
ERR_FAIL_INDEX_V(p_voice,MAX_VOICES,0);
return voice[p_voice].level;
}
void AudioEffectChorus::set_voice_cutoff_hz(int p_voice,float p_cutoff_hz){
ERR_FAIL_INDEX(p_voice,MAX_VOICES);
voice[p_voice].cutoff=p_cutoff_hz;
}
float AudioEffectChorus::get_voice_cutoff_hz(int p_voice) const{
ERR_FAIL_INDEX_V(p_voice,MAX_VOICES,0);
return voice[p_voice].cutoff;
}
void AudioEffectChorus::set_voice_pan(int p_voice,float p_pan){
ERR_FAIL_INDEX(p_voice,MAX_VOICES);
voice[p_voice].pan=p_pan;
}
float AudioEffectChorus::get_voice_pan(int p_voice) const{
ERR_FAIL_INDEX_V(p_voice,MAX_VOICES,0);
return voice[p_voice].pan;
}
void AudioEffectChorus::set_wet(float amount){
wet=amount;
}
float AudioEffectChorus::get_wet() const{
return wet;
}
void AudioEffectChorus::set_dry(float amount){
dry=amount;
}
float AudioEffectChorus::get_dry() const{
return dry;
}
void AudioEffectChorus::_validate_property(PropertyInfo& property) const {
if (property.name.begins_with("voice/")) {
int voice_idx = property.name.get_slice("/",1).to_int();
if (voice_idx>voice_count) {
property.usage=0;
}
}
}
void AudioEffectChorus::_bind_methods() {
ClassDB::bind_method(_MD("set_voice_count","voices"),&AudioEffectChorus::set_voice_count);
ClassDB::bind_method(_MD("get_voice_count"),&AudioEffectChorus::get_voice_count);
ClassDB::bind_method(_MD("set_voice_delay_ms","voice_idx","delay_ms"),&AudioEffectChorus::set_voice_delay_ms);
ClassDB::bind_method(_MD("get_voice_delay_ms","voice_idx"),&AudioEffectChorus::get_voice_delay_ms);
ClassDB::bind_method(_MD("set_voice_rate_hz","voice_idx","rate_hz"),&AudioEffectChorus::set_voice_rate_hz);
ClassDB::bind_method(_MD("get_voice_rate_hz","voice_idx"),&AudioEffectChorus::get_voice_rate_hz);
ClassDB::bind_method(_MD("set_voice_depth_ms","voice_idx","depth_ms"),&AudioEffectChorus::set_voice_depth_ms);
ClassDB::bind_method(_MD("get_voice_depth_ms","voice_idx"),&AudioEffectChorus::get_voice_depth_ms);
ClassDB::bind_method(_MD("set_voice_level_db","voice_idx","level_db"),&AudioEffectChorus::set_voice_level_db);
ClassDB::bind_method(_MD("get_voice_level_db","voice_idx"),&AudioEffectChorus::get_voice_level_db);
ClassDB::bind_method(_MD("set_voice_cutoff_hz","voice_idx","cutoff_hz"),&AudioEffectChorus::set_voice_cutoff_hz);
ClassDB::bind_method(_MD("get_voice_cutoff_hz","voice_idx"),&AudioEffectChorus::get_voice_cutoff_hz);
ClassDB::bind_method(_MD("set_voice_pan","voice_idx","pan"),&AudioEffectChorus::set_voice_pan);
ClassDB::bind_method(_MD("get_voice_pan","voice_idx"),&AudioEffectChorus::get_voice_pan);
ClassDB::bind_method(_MD("set_wet","amount"),&AudioEffectChorus::set_wet);
ClassDB::bind_method(_MD("get_wet"),&AudioEffectChorus::get_wet);
ClassDB::bind_method(_MD("set_dry","amount"),&AudioEffectChorus::set_dry);
ClassDB::bind_method(_MD("get_dry"),&AudioEffectChorus::get_dry);
ADD_PROPERTY(PropertyInfo(Variant::INT,"voice_count",PROPERTY_HINT_RANGE,"1,4,1"),_SCS("set_voice_count"),_SCS("get_voice_count"));
ADD_PROPERTY(PropertyInfo(Variant::REAL,"dry",PROPERTY_HINT_RANGE,"0,1,0.01"),_SCS("set_dry"),_SCS("get_dry"));
ADD_PROPERTY(PropertyInfo(Variant::REAL,"wet",PROPERTY_HINT_RANGE,"0,1,0.01"),_SCS("set_wet"),_SCS("get_wet"));
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/1/delay_ms",PROPERTY_HINT_RANGE,"0,50,0.01"),_SCS("set_voice_delay_ms"),_SCS("get_voice_delay_ms"),0);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/1/rate_hz",PROPERTY_HINT_RANGE,"0.1,20,0.1"),_SCS("set_voice_rate_hz"),_SCS("get_voice_rate_hz"),0);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/1/depth_ms",PROPERTY_HINT_RANGE,"0,20,0.01"),_SCS("set_voice_depth_ms"),_SCS("get_voice_depth_ms"),0);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/1/level_db",PROPERTY_HINT_RANGE,"-60,24,0.1"),_SCS("set_voice_level_db"),_SCS("get_voice_level_db"),0);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/1/cutoff_hz",PROPERTY_HINT_RANGE,"1,16000,1"),_SCS("set_voice_cutoff_hz"),_SCS("get_voice_cutoff_hz"),0);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/1/pan",PROPERTY_HINT_RANGE,"-1,1,0.01"),_SCS("set_voice_pan"),_SCS("get_voice_pan"),0);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/2/delay_ms",PROPERTY_HINT_RANGE,"0,50,0.01"),_SCS("set_voice_delay_ms"),_SCS("get_voice_delay_ms"),1);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/2/rate_hz",PROPERTY_HINT_RANGE,"0.1,20,0.1"),_SCS("set_voice_rate_hz"),_SCS("get_voice_rate_hz"),1);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/2/depth_ms",PROPERTY_HINT_RANGE,"0,20,0.01"),_SCS("set_voice_depth_ms"),_SCS("get_voice_depth_ms"),1);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/2/level_db",PROPERTY_HINT_RANGE,"-60,24,0.1"),_SCS("set_voice_level_db"),_SCS("get_voice_level_db"),1);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/2/cutoff_hz",PROPERTY_HINT_RANGE,"1,16000,1"),_SCS("set_voice_cutoff_hz"),_SCS("get_voice_cutoff_hz"),1);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/2/pan",PROPERTY_HINT_RANGE,"-1,1,0.01"),_SCS("set_voice_pan"),_SCS("get_voice_pan"),1);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/3/delay_ms",PROPERTY_HINT_RANGE,"0,50,0.01"),_SCS("set_voice_delay_ms"),_SCS("get_voice_delay_ms"),2);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/3/rate_hz",PROPERTY_HINT_RANGE,"0.1,20,0.1"),_SCS("set_voice_rate_hz"),_SCS("get_voice_rate_hz"),2);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/3/depth_ms",PROPERTY_HINT_RANGE,"0,20,0.01"),_SCS("set_voice_depth_ms"),_SCS("get_voice_depth_ms"),2);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/3/level_db",PROPERTY_HINT_RANGE,"-60,24,0.1"),_SCS("set_voice_level_db"),_SCS("get_voice_level_db"),2);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/3/cutoff_hz",PROPERTY_HINT_RANGE,"1,16000,1"),_SCS("set_voice_cutoff_hz"),_SCS("get_voice_cutoff_hz"),2);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/3/pan",PROPERTY_HINT_RANGE,"-1,1,0.01"),_SCS("set_voice_pan"),_SCS("get_voice_pan"),2);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/4/delay_ms",PROPERTY_HINT_RANGE,"0,50,0.01"),_SCS("set_voice_delay_ms"),_SCS("get_voice_delay_ms"),3);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/4/rate_hz",PROPERTY_HINT_RANGE,"0.1,20,0.1"),_SCS("set_voice_rate_hz"),_SCS("get_voice_rate_hz"),3);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/4/depth_ms",PROPERTY_HINT_RANGE,"0,20,0.01"),_SCS("set_voice_depth_ms"),_SCS("get_voice_depth_ms"),3);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/4/level_db",PROPERTY_HINT_RANGE,"-60,24,0.1"),_SCS("set_voice_level_db"),_SCS("get_voice_level_db"),3);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/4/cutoff_hz",PROPERTY_HINT_RANGE,"1,16000,1"),_SCS("set_voice_cutoff_hz"),_SCS("get_voice_cutoff_hz"),3);
ADD_PROPERTYI(PropertyInfo(Variant::REAL,"voice/4/pan",PROPERTY_HINT_RANGE,"-1,1,0.01"),_SCS("set_voice_pan"),_SCS("get_voice_pan"),3);
}
AudioEffectChorus::AudioEffectChorus()
{
voice_count=2;
voice[0].delay=15;
voice[1].delay=20;
voice[0].rate=0.8;
voice[1].rate=1.2;
voice[0].depth=2;
voice[1].depth=3;
voice[0].cutoff=8000;
voice[1].cutoff=8000;
voice[0].pan=-0.5;
voice[1].pan=0.5;
wet=0.5;
dry=1.0;
}