135 lines
3.4 KiB
C++
135 lines
3.4 KiB
C++
#include "audio_effect_stereo_enhance.h"
|
|
#include "servers/audio_server.h"
|
|
void AudioEffectStereoEnhanceInstance::process(const AudioFrame *p_src_frames,AudioFrame *p_dst_frames,int p_frame_count) {
|
|
|
|
|
|
float intensity=base->pan_pullout;
|
|
bool surround_mode=base->surround>0;
|
|
float surround_amount=base->surround;
|
|
unsigned int delay_frames=(base->time_pullout/1000.0)*AudioServer::get_singleton()->get_mix_rate();
|
|
|
|
for (int i=0;i<p_frame_count;i++) {
|
|
|
|
float l=p_src_frames[i].l;
|
|
float r=p_src_frames[i].r;
|
|
|
|
float center=(l+r)/2.0f;
|
|
|
|
l=( center+(l-center)*intensity );
|
|
r=( center+(r-center)*intensity );
|
|
|
|
if (surround_mode) {
|
|
|
|
float val=(l+r)/2.0;
|
|
|
|
delay_ringbuff[ringbuff_pos&ringbuff_mask]=val;
|
|
|
|
float out=delay_ringbuff[(ringbuff_pos-delay_frames)&ringbuff_mask]*surround_amount;
|
|
|
|
l+=out;
|
|
r+=-out;
|
|
} else {
|
|
|
|
float val=r;
|
|
|
|
delay_ringbuff[ringbuff_pos&ringbuff_mask]=val;
|
|
|
|
//r is delayed
|
|
r=delay_ringbuff[(ringbuff_pos-delay_frames)&ringbuff_mask];;
|
|
|
|
|
|
}
|
|
|
|
p_dst_frames[i].l=l;
|
|
p_dst_frames[i].r=r;
|
|
ringbuff_pos++;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
AudioEffectStereoEnhanceInstance::~AudioEffectStereoEnhanceInstance() {
|
|
|
|
memdelete_arr(delay_ringbuff);
|
|
}
|
|
|
|
Ref<AudioEffectInstance> AudioEffectStereoEnhance::instance() {
|
|
Ref<AudioEffectStereoEnhanceInstance> ins;
|
|
ins.instance();
|
|
|
|
ins->base=Ref<AudioEffectStereoEnhance>(this);
|
|
|
|
|
|
float ring_buffer_max_size=AudioEffectStereoEnhanceInstance::MAX_DELAY_MS+2;
|
|
ring_buffer_max_size/=1000.0;//convert to seconds
|
|
ring_buffer_max_size*=AudioServer::get_singleton()->get_mix_rate();
|
|
|
|
int ringbuff_size=(int)ring_buffer_max_size;
|
|
|
|
int bits=0;
|
|
|
|
while(ringbuff_size>0) {
|
|
bits++;
|
|
ringbuff_size/=2;
|
|
}
|
|
|
|
ringbuff_size=1<<bits;
|
|
ins->ringbuff_mask=ringbuff_size-1;
|
|
ins->ringbuff_pos=0;
|
|
|
|
ins->delay_ringbuff = memnew_arr(float,ringbuff_size );
|
|
|
|
return ins;
|
|
}
|
|
|
|
void AudioEffectStereoEnhance::set_pan_pullout(float p_amount) {
|
|
|
|
pan_pullout=p_amount;
|
|
}
|
|
|
|
float AudioEffectStereoEnhance::get_pan_pullout() const {
|
|
|
|
return pan_pullout;
|
|
}
|
|
|
|
void AudioEffectStereoEnhance::set_time_pullout(float p_amount) {
|
|
|
|
time_pullout=p_amount;
|
|
}
|
|
float AudioEffectStereoEnhance::get_time_pullout() const {
|
|
|
|
return time_pullout;
|
|
}
|
|
|
|
void AudioEffectStereoEnhance::set_surround(float p_amount) {
|
|
|
|
surround=p_amount;
|
|
}
|
|
float AudioEffectStereoEnhance::get_surround() const {
|
|
|
|
return surround;
|
|
}
|
|
|
|
void AudioEffectStereoEnhance::_bind_methods() {
|
|
|
|
ClassDB::bind_method(D_METHOD("set_pan_pullout","amount"),&AudioEffectStereoEnhance::set_pan_pullout);
|
|
ClassDB::bind_method(D_METHOD("get_pan_pullout"),&AudioEffectStereoEnhance::get_pan_pullout);
|
|
|
|
ClassDB::bind_method(D_METHOD("set_time_pullout","amount"),&AudioEffectStereoEnhance::set_time_pullout);
|
|
ClassDB::bind_method(D_METHOD("get_time_pullout"),&AudioEffectStereoEnhance::get_time_pullout);
|
|
|
|
ClassDB::bind_method(D_METHOD("set_surround","amount"),&AudioEffectStereoEnhance::set_surround);
|
|
ClassDB::bind_method(D_METHOD("get_surround"),&AudioEffectStereoEnhance::get_surround);
|
|
|
|
ADD_PROPERTY(PropertyInfo(Variant::REAL,"pan_pullout",PROPERTY_HINT_RANGE,"0,4,0.01"),"set_pan_pullout","get_pan_pullout");
|
|
ADD_PROPERTY(PropertyInfo(Variant::REAL,"time_pullout_ms",PROPERTY_HINT_RANGE,"0,50,0.01"),"set_time_pullout","get_time_pullout");
|
|
ADD_PROPERTY(PropertyInfo(Variant::REAL,"surround",PROPERTY_HINT_RANGE,"0,1,0.01"),"set_surround","get_surround");
|
|
}
|
|
|
|
AudioEffectStereoEnhance::AudioEffectStereoEnhance()
|
|
{
|
|
pan_pullout=1;
|
|
time_pullout=0;
|
|
surround=0;
|
|
}
|