mirror of https://github.com/drowe67/LPCNet.git
More cleanup
parent
7c7cd2d478
commit
0f81f22044
|
|
@ -52,7 +52,7 @@
|
|||
#define NB_FEATURES (2*NB_BANDS+3+LPC_ORDER)
|
||||
|
||||
|
||||
struct DenoiseState {
|
||||
typedef struct {
|
||||
float analysis_mem[FRAME_SIZE];
|
||||
float cepstral_mem[CEPS_MEM][NB_BANDS];
|
||||
int memid;
|
||||
|
|
@ -63,29 +63,29 @@ struct DenoiseState {
|
|||
int last_period;
|
||||
float mem_hp_x[2];
|
||||
float lastg[NB_BANDS];
|
||||
};
|
||||
} DenoiseState;
|
||||
|
||||
int rnnoise_get_size() {
|
||||
static int rnnoise_get_size() {
|
||||
return sizeof(DenoiseState);
|
||||
}
|
||||
|
||||
int rnnoise_init(DenoiseState *st) {
|
||||
static int rnnoise_init(DenoiseState *st) {
|
||||
memset(st, 0, sizeof(*st));
|
||||
return 0;
|
||||
}
|
||||
|
||||
DenoiseState *rnnoise_create() {
|
||||
static DenoiseState *rnnoise_create() {
|
||||
DenoiseState *st;
|
||||
st = malloc(rnnoise_get_size());
|
||||
rnnoise_init(st);
|
||||
return st;
|
||||
}
|
||||
|
||||
void rnnoise_destroy(DenoiseState *st) {
|
||||
static void rnnoise_destroy(DenoiseState *st) {
|
||||
free(st);
|
||||
}
|
||||
|
||||
short float2short(float x)
|
||||
static short float2short(float x)
|
||||
{
|
||||
int i;
|
||||
i = (int)floor(.5+x);
|
||||
|
|
@ -314,10 +314,10 @@ int main(int argc, char **argv) {
|
|||
old_speech_gain = speech_gain;
|
||||
count++;
|
||||
}
|
||||
//fprintf(stderr, "matrix size: %d x %d\n", count, NB_FEATURES + 2*NB_BANDS + 1);
|
||||
fclose(f1);
|
||||
fclose(ffeat);
|
||||
if (fpcm) fclose(fpcm);
|
||||
rnnoise_destroy(st);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
|
|
|||
28
src/freq.h
28
src/freq.h
|
|
@ -33,34 +33,6 @@
|
|||
|
||||
#define NB_BANDS 18
|
||||
|
||||
|
||||
#ifndef RNNOISE_EXPORT
|
||||
# if defined(WIN32)
|
||||
# if defined(RNNOISE_BUILD) && defined(DLL_EXPORT)
|
||||
# define RNNOISE_EXPORT __declspec(dllexport)
|
||||
# else
|
||||
# define RNNOISE_EXPORT
|
||||
# endif
|
||||
# elif defined(__GNUC__) && defined(RNNOISE_BUILD)
|
||||
# define RNNOISE_EXPORT __attribute__ ((visibility ("default")))
|
||||
# else
|
||||
# define RNNOISE_EXPORT
|
||||
# endif
|
||||
#endif
|
||||
|
||||
|
||||
typedef struct DenoiseState DenoiseState;
|
||||
|
||||
RNNOISE_EXPORT int rnnoise_get_size();
|
||||
|
||||
RNNOISE_EXPORT int rnnoise_init(DenoiseState *st);
|
||||
|
||||
RNNOISE_EXPORT DenoiseState *rnnoise_create();
|
||||
|
||||
RNNOISE_EXPORT void rnnoise_destroy(DenoiseState *st);
|
||||
|
||||
RNNOISE_EXPORT float rnnoise_process_frame(DenoiseState *st, float *out, const float *in);
|
||||
|
||||
void compute_band_energy(float *bandE, const kiss_fft_cpx *X);
|
||||
void compute_band_corr(float *bandE, const kiss_fft_cpx *X, const kiss_fft_cpx *P);
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue