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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
/**
* @file speex_pp.c Speex Pre-processor
*
* Copyright (C) 2010 Creytiv.com
*/
#include <string.h>
#include <stdlib.h>
#include <speex/speex.h>
#include <speex/speex_preprocess.h>
#include <re.h>
#include <baresip.h>
/**
* @defgroup speex_pp speex_pp
*
* Audio pre-processor from libspeexdsp
*/
struct preproc {
struct aufilt_enc_st af; /* base class */
SpeexPreprocessState *state;
};
/** Speex configuration */
static struct {
int denoise_enabled;
int agc_enabled;
int vad_enabled;
int dereverb_enabled;
spx_int32_t agc_level;
} pp_conf = {
1,
1,
1,
1,
8000
};
static void speexpp_destructor(void *arg)
{
struct preproc *st = arg;
if (st->state)
speex_preprocess_state_destroy(st->state);
list_unlink(&st->af.le);
}
static int encode_update(struct aufilt_enc_st **stp, void **ctx,
const struct aufilt *af, struct aufilt_prm *prm)
{
struct preproc *st;
unsigned sampc;
(void)ctx;
if (!stp || !af || !prm || prm->ch != 1)
return EINVAL;
st = mem_zalloc(sizeof(*st), speexpp_destructor);
if (!st)
return ENOMEM;
sampc = prm->srate * prm->ch * prm->ptime / 1000;
st->state = speex_preprocess_state_init(sampc, prm->srate);
if (!st->state)
goto error;
speex_preprocess_ctl(st->state, SPEEX_PREPROCESS_SET_DENOISE,
&pp_conf.denoise_enabled);
speex_preprocess_ctl(st->state, SPEEX_PREPROCESS_SET_AGC,
&pp_conf.agc_enabled);
#ifdef SPEEX_PREPROCESS_SET_AGC_TARGET
if (pp_conf.agc_enabled) {
speex_preprocess_ctl(st->state,
SPEEX_PREPROCESS_SET_AGC_TARGET,
&pp_conf.agc_level);
}
#endif
speex_preprocess_ctl(st->state, SPEEX_PREPROCESS_SET_VAD,
&pp_conf.vad_enabled);
speex_preprocess_ctl(st->state, SPEEX_PREPROCESS_SET_DEREVERB,
&pp_conf.dereverb_enabled);
info("speex_pp: Speex preprocessor loaded: srate = %uHz\n",
prm->srate);
*stp = (struct aufilt_enc_st *)st;
return 0;
error:
mem_deref(st);
return ENOMEM;
}
static int encode(struct aufilt_enc_st *st, int16_t *sampv, size_t *sampc)
{
struct preproc *pp = (struct preproc *)st;
int is_speech = 1;
if (!*sampc)
return 0;
/* NOTE: Using this macro to check libspeex version */
#ifdef SPEEX_PREPROCESS_SET_NOISE_SUPPRESS
/* New API */
is_speech = speex_preprocess_run(pp->state, sampv);
#else
/* Old API - not tested! */
is_speech = speex_preprocess(pp->state, sampv, NULL);
#endif
/* XXX: Handle is_speech and VAD */
(void)is_speech;
return 0;
}
static void config_parse(struct conf *conf)
{
uint32_t v;
if (0 == conf_get_u32(conf, "speex_agc_level", &v))
pp_conf.agc_level = v;
}
static struct aufilt preproc = {
LE_INIT, "speex_pp", encode_update, encode, NULL, NULL
};
static int module_init(void)
{
config_parse(conf_cur());
aufilt_register(baresip_aufiltl(), &preproc);
return 0;
}
static int module_close(void)
{
aufilt_unregister(&preproc);
return 0;
}
EXPORT_SYM const struct mod_export DECL_EXPORTS(speex_pp) = {
"speex_pp",
"filter",
module_init,
module_close
};
|
