| @@ -141,24 +141,25 @@ typedef struct AudioFFTDeNoiseContext { | |||||
| } AudioFFTDeNoiseContext; | } AudioFFTDeNoiseContext; | ||||
| #define OFFSET(x) offsetof(AudioFFTDeNoiseContext, x) | #define OFFSET(x) offsetof(AudioFFTDeNoiseContext, x) | ||||
| #define A AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM | |||||
| #define AF AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM | |||||
| #define AFR AV_OPT_FLAG_AUDIO_PARAM|AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_RUNTIME_PARAM | |||||
| static const AVOption afftdn_options[] = { | static const AVOption afftdn_options[] = { | ||||
| { "nr", "set the noise reduction", OFFSET(noise_reduction), AV_OPT_TYPE_FLOAT, {.dbl = 12}, .01, 97, A }, | |||||
| { "nf", "set the noise floor", OFFSET(noise_floor), AV_OPT_TYPE_FLOAT, {.dbl =-50}, -80,-20, A }, | |||||
| { "nt", "set the noise type", OFFSET(noise_type), AV_OPT_TYPE_INT, {.i64 = WHITE_NOISE}, WHITE_NOISE, NB_NOISE-1, A, "type" }, | |||||
| { "w", "white noise", 0, AV_OPT_TYPE_CONST, {.i64 = WHITE_NOISE}, 0, 0, A, "type" }, | |||||
| { "v", "vinyl noise", 0, AV_OPT_TYPE_CONST, {.i64 = VINYL_NOISE}, 0, 0, A, "type" }, | |||||
| { "s", "shellac noise", 0, AV_OPT_TYPE_CONST, {.i64 = SHELLAC_NOISE}, 0, 0, A, "type" }, | |||||
| { "c", "custom noise", 0, AV_OPT_TYPE_CONST, {.i64 = CUSTOM_NOISE}, 0, 0, A, "type" }, | |||||
| { "bn", "set the custom bands noise", OFFSET(band_noise_str), AV_OPT_TYPE_STRING, {.str = 0}, 0, 0, A }, | |||||
| { "rf", "set the residual floor", OFFSET(residual_floor), AV_OPT_TYPE_FLOAT, {.dbl =-38}, -80,-20, A }, | |||||
| { "tn", "track noise", OFFSET(track_noise), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, A }, | |||||
| { "tr", "track residual", OFFSET(track_residual), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, A }, | |||||
| { "om", "set output mode", OFFSET(output_mode), AV_OPT_TYPE_INT, {.i64 = OUT_MODE}, 0, NB_MODES-1, A, "mode" }, | |||||
| { "i", "input", 0, AV_OPT_TYPE_CONST, {.i64 = IN_MODE}, 0, 0, A, "mode" }, | |||||
| { "o", "output", 0, AV_OPT_TYPE_CONST, {.i64 = OUT_MODE}, 0, 0, A, "mode" }, | |||||
| { "n", "noise", 0, AV_OPT_TYPE_CONST, {.i64 = NOISE_MODE}, 0, 0, A, "mode" }, | |||||
| { "nr", "set the noise reduction", OFFSET(noise_reduction), AV_OPT_TYPE_FLOAT, {.dbl = 12}, .01, 97, AFR }, | |||||
| { "nf", "set the noise floor", OFFSET(noise_floor), AV_OPT_TYPE_FLOAT, {.dbl =-50}, -80,-20, AFR }, | |||||
| { "nt", "set the noise type", OFFSET(noise_type), AV_OPT_TYPE_INT, {.i64 = WHITE_NOISE}, WHITE_NOISE, NB_NOISE-1, AF, "type" }, | |||||
| { "w", "white noise", 0, AV_OPT_TYPE_CONST, {.i64 = WHITE_NOISE}, 0, 0, AF, "type" }, | |||||
| { "v", "vinyl noise", 0, AV_OPT_TYPE_CONST, {.i64 = VINYL_NOISE}, 0, 0, AF, "type" }, | |||||
| { "s", "shellac noise", 0, AV_OPT_TYPE_CONST, {.i64 = SHELLAC_NOISE}, 0, 0, AF, "type" }, | |||||
| { "c", "custom noise", 0, AV_OPT_TYPE_CONST, {.i64 = CUSTOM_NOISE}, 0, 0, AF, "type" }, | |||||
| { "bn", "set the custom bands noise", OFFSET(band_noise_str), AV_OPT_TYPE_STRING, {.str = 0}, 0, 0, AF }, | |||||
| { "rf", "set the residual floor", OFFSET(residual_floor), AV_OPT_TYPE_FLOAT, {.dbl =-38}, -80,-20, AFR }, | |||||
| { "tn", "track noise", OFFSET(track_noise), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AFR }, | |||||
| { "tr", "track residual", OFFSET(track_residual), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, AFR }, | |||||
| { "om", "set output mode", OFFSET(output_mode), AV_OPT_TYPE_INT, {.i64 = OUT_MODE}, 0, NB_MODES-1, AFR, "mode" }, | |||||
| { "i", "input", 0, AV_OPT_TYPE_CONST, {.i64 = IN_MODE}, 0, 0, AFR, "mode" }, | |||||
| { "o", "output", 0, AV_OPT_TYPE_CONST, {.i64 = OUT_MODE}, 0, 0, AFR, "mode" }, | |||||
| { "n", "noise", 0, AV_OPT_TYPE_CONST, {.i64 = NOISE_MODE}, 0, 0, AFR, "mode" }, | |||||
| { NULL } | { NULL } | ||||
| }; | }; | ||||
| @@ -1375,6 +1376,7 @@ static int process_command(AVFilterContext *ctx, const char *cmd, const char *ar | |||||
| { | { | ||||
| AudioFFTDeNoiseContext *s = ctx->priv; | AudioFFTDeNoiseContext *s = ctx->priv; | ||||
| int need_reset = 0; | int need_reset = 0; | ||||
| int ret = 0; | |||||
| if (!strcmp(cmd, "sample_noise") || | if (!strcmp(cmd, "sample_noise") || | ||||
| !strcmp(cmd, "sn")) { | !strcmp(cmd, "sn")) { | ||||
| @@ -1386,31 +1388,11 @@ static int process_command(AVFilterContext *ctx, const char *cmd, const char *ar | |||||
| s->sample_noise_start = 0; | s->sample_noise_start = 0; | ||||
| s->sample_noise_end = 1; | s->sample_noise_end = 1; | ||||
| } | } | ||||
| } else if (!strcmp(cmd, "nr") || | |||||
| !strcmp(cmd, "noise_reduction")) { | |||||
| float nr; | |||||
| if (av_sscanf(args, "%f", &nr) == 1) { | |||||
| s->noise_reduction = av_clipf(nr, 0.01, 97); | |||||
| need_reset = 1; | |||||
| } | |||||
| } else if (!strcmp(cmd, "nf") || | |||||
| !strcmp(cmd, "noise_floor")) { | |||||
| float nf; | |||||
| if (av_sscanf(args, "%f", &nf) == 1) { | |||||
| s->noise_floor = av_clipf(nf, -80, -20); | |||||
| need_reset = 1; | |||||
| } | |||||
| } else if (!strcmp(cmd, "output_mode") || | |||||
| !strcmp(cmd, "om")) { | |||||
| if (!strcmp(args, "i")) { | |||||
| s->output_mode = IN_MODE; | |||||
| } else if (!strcmp(args, "o")) { | |||||
| s->output_mode = OUT_MODE; | |||||
| } else if (!strcmp(args, "n")) { | |||||
| s->output_mode = NOISE_MODE; | |||||
| } | |||||
| } else { | |||||
| ret = ff_filter_process_command(ctx, cmd, args, res, res_len, flags); | |||||
| if (ret < 0) | |||||
| return ret; | |||||
| need_reset = 1; | |||||
| } | } | ||||
| if (need_reset) | if (need_reset) | ||||