rtaudio_speex/speex_func.cpp at master · graylin2025/rtaudio_speex · GitHub

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
#include <speex/speex_echo.h>
#include <speex/speex_preprocess.h>
#include "speex_func.h"

int speex_func_init(int frame_size, int filter_length, int sample_rate);
int speex_func_destroy();
int speex_func_echo_cancel(short* mic, short* play, short* out);

static SpeexPreprocessState* g_preprocess_state = nullptr;
static SpeexEchoState *g_echo_state = nullptr;
static int g_frame_size = 0;
static int g_filter_length = 0;

int speex_func_init(int frame_size, int filter_length, int sample_rate) {
	g_frame_size = frame_size;
	g_filter_length = filter_length;

	g_echo_state = speex_echo_state_init(
		g_frame_size,
		g_filter_length
		);
	g_preprocess_state = speex_preprocess_state_init(
		g_frame_size,
		sample_rate);

	int _db = 60;
	int _denose = 1;
	int _noiseSuppress = -25;
	speex_echo_ctl(g_echo_state, SPEEX_ECHO_SET_SAMPLING_RATE, &sample_rate);
	speex_preprocess_ctl(g_preprocess_state, SPEEX_PREPROCESS_SET_DENOISE, &_denose);
	speex_preprocess_ctl(g_preprocess_state, SPEEX_PREPROCESS_SET_NOISE_SUPPRESS, &_noiseSuppress); //设置噪声的dB
	speex_preprocess_ctl(g_preprocess_state, SPEEX_PREPROCESS_SET_ECHO_STATE, g_echo_state);
	speex_preprocess_ctl(g_preprocess_state, SPEEX_PREPROCESS_SET_ECHO_SUPPRESS, &_db);

	return (g_echo_state ? 1 : 0) && (g_preprocess_state? 1: 0);
}


int speex_func_destroy() {
	if (g_preprocess_state) {
		speex_preprocess_state_destroy(g_preprocess_state);
		g_preprocess_state = nullptr;
	}
	if (g_echo_state) {
		speex_echo_state_destroy(g_echo_state);
		g_echo_state = nullptr;
	}
	return 0;
}

int speex_func_echo_cancel(short* mic, short* play, short* out) {
	if (!g_echo_state || !g_preprocess_state) {
		return 0;
	}
	speex_echo_cancellation(g_echo_state,
		(const spx_int16_t*)mic,
		(const spx_int16_t*)play,
		(spx_int16_t*)out
		);
	speex_preprocess_run(g_preprocess_state, out);
	return 0;
}