obs-studio/plugins/obs-filters/gain-filter.c

97 lines
2.1 KiB
C

#include <obs-module.h>
#include <media-io/audio-math.h>
#include <math.h>
#define do_log(level, format, ...) \
blog(level, "[gain filter: '%s'] " format, \
obs_source_get_name(gf->context), ##__VA_ARGS__)
#define warn(format, ...) do_log(LOG_WARNING, format, ##__VA_ARGS__)
#define info(format, ...) do_log(LOG_INFO, format, ##__VA_ARGS__)
#define S_GAIN_DB "db"
#define MT_ obs_module_text
#define TEXT_GAIN_DB MT_("Gain.GainDB")
struct gain_data {
obs_source_t *context;
float multiple;
};
static const char *gain_name(void *unused)
{
UNUSED_PARAMETER(unused);
return obs_module_text("Gain");
}
static void gain_destroy(void *data)
{
struct gain_data *gf = data;
bfree(gf);
}
static void gain_update(void *data, obs_data_t *s)
{
struct gain_data *gf = data;
double val = obs_data_get_double(s, S_GAIN_DB);
gf->multiple = db_to_mul((float)val);
}
static void *gain_create(obs_data_t *settings, obs_source_t *filter)
{
struct gain_data *gf = bzalloc(sizeof(*gf));
gf->context = filter;
gain_update(gf, settings);
return gf;
}
static struct obs_audio_data *gain_filter_audio(void *data,
struct obs_audio_data *audio)
{
struct gain_data *gf = data;
float *adata[2] = {(float*)audio->data[0], (float*)audio->data[1]};
const float multiple = gf->multiple;
for (size_t c = 0; c < 2; c++) {
if (audio->data[c]) {
for (size_t i = 0; i < audio->frames; i++) {
adata[c][i] *= multiple;
}
}
}
return audio;
}
static void gain_defaults(obs_data_t *s)
{
obs_data_set_default_double(s, S_GAIN_DB, 0.0f);
}
static obs_properties_t *gain_properties(void *data)
{
obs_properties_t *ppts = obs_properties_create();
obs_properties_add_float_slider(ppts, S_GAIN_DB, TEXT_GAIN_DB,
-30.0, 30.0, 0.1);
UNUSED_PARAMETER(data);
return ppts;
}
struct obs_source_info gain_filter = {
.id = "gain_filter",
.type = OBS_SOURCE_TYPE_FILTER,
.output_flags = OBS_SOURCE_AUDIO,
.get_name = gain_name,
.create = gain_create,
.destroy = gain_destroy,
.update = gain_update,
.filter_audio = gain_filter_audio,
.get_defaults = gain_defaults,
.get_properties = gain_properties,
};