diff --git a/dnn/compile.sh b/dnn/compile.sh index 446e7e32..fba4f8e3 100755 --- a/dnn/compile.sh +++ b/dnn/compile.sh @@ -1,4 +1,4 @@ #!/bin/sh -gcc -DTRAINING=1 -Wall -W -O3 -g -I../include dump_data.c kiss_fft.c pitch.c celt_lpc.c -o dump_data -lm -gcc -o test_lpcnet -mavx2 -mfma -g -O3 -Wall -W -Wextra lpcnet.c nnet.c nnet_data.c dump_data.c kiss_fft.c pitch.c celt_lpc.c -lm +gcc -DTRAINING=1 -Wall -W -O3 -g -I../include dump_data.c freq.c kiss_fft.c pitch.c celt_lpc.c -o dump_data -lm +gcc -o test_lpcnet -mavx2 -mfma -g -O3 -Wall -W -Wextra lpcnet.c nnet.c nnet_data.c freq.c kiss_fft.c pitch.c celt_lpc.c -lm diff --git a/dnn/dump_data.c b/dnn/dump_data.c index 8c39c7cf..7bdacc61 100644 --- a/dnn/dump_data.c +++ b/dnn/dump_data.c @@ -40,27 +40,13 @@ #include "celt_lpc.h" #include -#define PREEMPHASIS (0.85f) - -#define FRAME_SIZE_SHIFT 2 -#define FRAME_SIZE (40< +#include +#include +#include "kiss_fft.h" +#include "common.h" +#include +#include "freq.h" +#include "pitch.h" +#include "arch.h" +#include "celt_lpc.h" +#include + +#define SQUARE(x) ((x)*(x)) + +static const opus_int16 eband5ms[] = { +/*0 200 400 600 800 1k 1.2 1.4 1.6 2k 2.4 2.8 3.2 4k 4.8 5.6 6.8 8k*/ + 0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 12, 14, 16, 20, 24, 28, 34, 40 +}; + + +typedef struct { + int init; + kiss_fft_state *kfft; + float half_window[FRAME_SIZE]; + float dct_table[NB_BANDS*NB_BANDS]; +} CommonState; + + + +void compute_band_energy(float *bandE, const kiss_fft_cpx *X) { + int i; + float sum[NB_BANDS] = {0}; + for (i=0;i