ec3561dd54c4b24dded4ceb4910b6fe3e206f4ce
[ffmpeg.git] / libavcodec / dcadsp.c
1 /*
2  * Copyright (c) 2004 Gildas Bazin
3  * Copyright (c) 2010 Mans Rullgard <mans@mansr.com>
4  *
5  * This file is part of Libav.
6  *
7  * Libav is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * Libav is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with Libav; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include "config.h"
23
24 #include "libavutil/attributes.h"
25 #include "libavutil/intreadwrite.h"
26
27 #include "dcadsp.h"
28 #include "dcamath.h"
29
30 static void decode_hf_c(int32_t dst[DCA_SUBBANDS][8],
31                         const int32_t vq_num[DCA_SUBBANDS],
32                         const int8_t hf_vq[1024][32], intptr_t vq_offset,
33                         int32_t scale[DCA_SUBBANDS][2],
34                         intptr_t start, intptr_t end)
35 {
36     int i, j;
37
38     for (j = start; j < end; j++) {
39         const int8_t *ptr = &hf_vq[vq_num[j]][vq_offset];
40         for (i = 0; i < 8; i++)
41             dst[j][i] = ptr[i] * scale[j][0] + 8 >> 4;
42     }
43 }
44
45 static inline void dca_lfe_fir(float *out, const float *in, const float *coefs,
46                                int decifactor)
47 {
48     float *out2    = out + 2 * decifactor - 1;
49     int num_coeffs = 256 / decifactor;
50     int j, k;
51
52     /* One decimated sample generates 2*decifactor interpolated ones */
53     for (k = 0; k < decifactor; k++) {
54         float v0 = 0.0;
55         float v1 = 0.0;
56         for (j = 0; j < num_coeffs; j++, coefs++) {
57             v0 += in[-j]                 * *coefs;
58             v1 += in[j + 1 - num_coeffs] * *coefs;
59         }
60         *out++  = v0;
61         *out2-- = v1;
62     }
63 }
64
65 static void dca_qmf_32_subbands(float samples_in[32][8], int sb_act,
66                                 SynthFilterContext *synth, FFTContext *imdct,
67                                 float synth_buf_ptr[512],
68                                 int *synth_buf_offset, float synth_buf2[32],
69                                 const float window[512], float *samples_out,
70                                 float raXin[32], float scale)
71 {
72     int i;
73     int subindex;
74
75     for (i = sb_act; i < 32; i++)
76         raXin[i] = 0.0;
77
78     /* Reconstructed channel sample index */
79     for (subindex = 0; subindex < 8; subindex++) {
80         /* Load in one sample from each subband and clear inactive subbands */
81         for (i = 0; i < sb_act; i++) {
82             unsigned sign = (i - 1) & 2;
83             uint32_t v    = AV_RN32A(&samples_in[i][subindex]) ^ sign << 30;
84             AV_WN32A(&raXin[i], v);
85         }
86
87         synth->synth_filter_float(imdct, synth_buf_ptr, synth_buf_offset,
88                                   synth_buf2, window, samples_out, raXin,
89                                   scale);
90         samples_out += 32;
91     }
92 }
93
94 static void dequantize_c(int32_t *samples, uint32_t step_size, uint32_t scale)
95 {
96     int64_t step = (int64_t)step_size * scale;
97     int shift, i;
98     int32_t step_scale;
99
100     if (step > (1 << 23))
101         shift = av_log2(step >> 23) + 1;
102     else
103         shift = 0;
104     step_scale = (int32_t)(step >> shift);
105
106     for (i = 0; i < 8; i++)
107         samples[i] = dca_clip23(dca_norm((int64_t)samples[i] * step_scale, 22 - shift));
108 }
109
110 static void dca_lfe_fir0_c(float *out, const float *in, const float *coefs)
111 {
112     dca_lfe_fir(out, in, coefs, 32);
113 }
114
115 static void dca_lfe_fir1_c(float *out, const float *in, const float *coefs)
116 {
117     dca_lfe_fir(out, in, coefs, 64);
118 }
119
120 av_cold void ff_dcadsp_init(DCADSPContext *s)
121 {
122     s->lfe_fir[0]      = dca_lfe_fir0_c;
123     s->lfe_fir[1]      = dca_lfe_fir1_c;
124     s->qmf_32_subbands = dca_qmf_32_subbands;
125     s->decode_hf       = decode_hf_c;
126     s->dequantize      = dequantize_c;
127
128     if (ARCH_AARCH64)
129         ff_dcadsp_init_aarch64(s);
130     if (ARCH_ARM)
131         ff_dcadsp_init_arm(s);
132     if (ARCH_X86)
133         ff_dcadsp_init_x86(s);
134 }