xref: /aosp_15_r20/external/libopus/dnn/freq.h (revision a58d3d2adb790c104798cd88c8a3aff4fa8b82cc)
1*a58d3d2aSXin Li /* Copyright (c) 2017-2018 Mozilla */
2*a58d3d2aSXin Li /*
3*a58d3d2aSXin Li    Redistribution and use in source and binary forms, with or without
4*a58d3d2aSXin Li    modification, are permitted provided that the following conditions
5*a58d3d2aSXin Li    are met:
6*a58d3d2aSXin Li 
7*a58d3d2aSXin Li    - Redistributions of source code must retain the above copyright
8*a58d3d2aSXin Li    notice, this list of conditions and the following disclaimer.
9*a58d3d2aSXin Li 
10*a58d3d2aSXin Li    - Redistributions in binary form must reproduce the above copyright
11*a58d3d2aSXin Li    notice, this list of conditions and the following disclaimer in the
12*a58d3d2aSXin Li    documentation and/or other materials provided with the distribution.
13*a58d3d2aSXin Li 
14*a58d3d2aSXin Li    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
15*a58d3d2aSXin Li    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
16*a58d3d2aSXin Li    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
17*a58d3d2aSXin Li    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
18*a58d3d2aSXin Li    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
19*a58d3d2aSXin Li    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
20*a58d3d2aSXin Li    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
21*a58d3d2aSXin Li    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
22*a58d3d2aSXin Li    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
23*a58d3d2aSXin Li    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
24*a58d3d2aSXin Li    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25*a58d3d2aSXin Li */
26*a58d3d2aSXin Li 
27*a58d3d2aSXin Li #ifndef FREQ_H
28*a58d3d2aSXin Li #define FREQ_H
29*a58d3d2aSXin Li 
30*a58d3d2aSXin Li #include "kiss_fft.h"
31*a58d3d2aSXin Li 
32*a58d3d2aSXin Li #define LPC_ORDER 16
33*a58d3d2aSXin Li 
34*a58d3d2aSXin Li #define PREEMPHASIS (0.85f)
35*a58d3d2aSXin Li 
36*a58d3d2aSXin Li #define FRAME_SIZE_5MS (2)
37*a58d3d2aSXin Li #define OVERLAP_SIZE_5MS (2)
38*a58d3d2aSXin Li #define TRAINING_OFFSET_5MS (1)
39*a58d3d2aSXin Li 
40*a58d3d2aSXin Li #define WINDOW_SIZE_5MS (FRAME_SIZE_5MS + OVERLAP_SIZE_5MS)
41*a58d3d2aSXin Li 
42*a58d3d2aSXin Li #define FRAME_SIZE (80*FRAME_SIZE_5MS)
43*a58d3d2aSXin Li #define OVERLAP_SIZE (80*OVERLAP_SIZE_5MS)
44*a58d3d2aSXin Li #define TRAINING_OFFSET (80*TRAINING_OFFSET_5MS)
45*a58d3d2aSXin Li #define WINDOW_SIZE (FRAME_SIZE + OVERLAP_SIZE)
46*a58d3d2aSXin Li #define FREQ_SIZE (WINDOW_SIZE/2 + 1)
47*a58d3d2aSXin Li 
48*a58d3d2aSXin Li #define NB_BANDS 18
49*a58d3d2aSXin Li #define NB_BANDS_1 (NB_BANDS - 1)
50*a58d3d2aSXin Li 
51*a58d3d2aSXin Li void lpcn_compute_band_energy(float *bandE, const kiss_fft_cpx *X);
52*a58d3d2aSXin Li void burg_cepstral_analysis(float *ceps, const float *x);
53*a58d3d2aSXin Li 
54*a58d3d2aSXin Li void apply_window(float *x);
55*a58d3d2aSXin Li void dct(float *out, const float *in);
56*a58d3d2aSXin Li void forward_transform(kiss_fft_cpx *out, const float *in);
57*a58d3d2aSXin Li float lpc_from_cepstrum(float *lpc, const float *cepstrum);
58*a58d3d2aSXin Li void apply_window(float *x);
59*a58d3d2aSXin Li void lpc_weighting(float *lpc, float gamma);
60*a58d3d2aSXin Li 
61*a58d3d2aSXin Li #endif
62