1/*
2 * Copyright (c) 2007-2008 Ian Caulfield
3 *               2009 Ramiro Polla
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg 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 * FFmpeg 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 FFmpeg; 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#include "libavutil/attributes.h"
24#include "mlpdsp.h"
25#include "mlp.h"
26
27static void mlp_filter_channel(int32_t *state, const int32_t *coeff,
28                               int firorder, int iirorder,
29                               unsigned int filter_shift, int32_t mask,
30                               int blocksize, int32_t *sample_buffer)
31{
32    int32_t *firbuf = state;
33    int32_t *iirbuf = state + MAX_BLOCKSIZE + MAX_FIR_ORDER;
34    const int32_t *fircoeff = coeff;
35    const int32_t *iircoeff = coeff + MAX_FIR_ORDER;
36    int i;
37
38    for (i = 0; i < blocksize; i++) {
39        int32_t residual = *sample_buffer;
40        unsigned int order;
41        int64_t accum = 0;
42        int32_t result;
43
44        for (order = 0; order < firorder; order++)
45            accum += (int64_t) firbuf[order] * fircoeff[order];
46        for (order = 0; order < iirorder; order++)
47            accum += (int64_t) iirbuf[order] * iircoeff[order];
48
49        accum  = accum >> filter_shift;
50        result = (accum + residual) & mask;
51
52        *--firbuf = result;
53        *--iirbuf = result - accum;
54
55        *sample_buffer = result;
56        sample_buffer += MAX_CHANNELS;
57    }
58}
59
60void ff_mlp_rematrix_channel(int32_t *samples,
61                             const int32_t *coeffs,
62                             const uint8_t *bypassed_lsbs,
63                             const int8_t *noise_buffer,
64                             int index,
65                             unsigned int dest_ch,
66                             uint16_t blockpos,
67                             unsigned int maxchan,
68                             int matrix_noise_shift,
69                             int access_unit_size_pow2,
70                             int32_t mask)
71{
72    unsigned int src_ch, i;
73    int index2 = 2 * index + 1;
74    for (i = 0; i < blockpos; i++) {
75        int64_t accum = 0;
76
77        for (src_ch = 0; src_ch <= maxchan; src_ch++)
78            accum += (int64_t) samples[src_ch] * coeffs[src_ch];
79
80        if (matrix_noise_shift) {
81            index &= access_unit_size_pow2 - 1;
82            accum += noise_buffer[index] << (matrix_noise_shift + 7);
83            index += index2;
84        }
85
86        samples[dest_ch] = ((accum >> 14) & mask) + *bypassed_lsbs;
87        bypassed_lsbs += MAX_CHANNELS;
88        samples += MAX_CHANNELS;
89    }
90}
91
92static int32_t (*mlp_select_pack_output(uint8_t *ch_assign,
93                                        int8_t *output_shift,
94                                        uint8_t max_matrix_channel,
95                                        int is32))(int32_t, uint16_t, int32_t (*)[], void *, uint8_t*, int8_t *, uint8_t, int)
96{
97    return ff_mlp_pack_output;
98}
99
100int32_t ff_mlp_pack_output(int32_t lossless_check_data,
101                           uint16_t blockpos,
102                           int32_t (*sample_buffer)[MAX_CHANNELS],
103                           void *data,
104                           uint8_t *ch_assign,
105                           int8_t *output_shift,
106                           uint8_t max_matrix_channel,
107                           int is32)
108{
109    unsigned int i, out_ch = 0;
110    int32_t *data_32 = data;
111    int16_t *data_16 = data;
112
113    for (i = 0; i < blockpos; i++) {
114        for (out_ch = 0; out_ch <= max_matrix_channel; out_ch++) {
115            int mat_ch = ch_assign[out_ch];
116            int32_t sample = sample_buffer[i][mat_ch]
117                          << output_shift[mat_ch];
118            lossless_check_data ^= (sample & 0xffffff) << mat_ch;
119            if (is32)
120                *data_32++ = sample << 8;
121            else
122                *data_16++ = sample >> 8;
123        }
124    }
125    return lossless_check_data;
126}
127
128av_cold void ff_mlpdsp_init(MLPDSPContext *c)
129{
130    c->mlp_filter_channel = mlp_filter_channel;
131    c->mlp_rematrix_channel = ff_mlp_rematrix_channel;
132    c->mlp_select_pack_output = mlp_select_pack_output;
133    c->mlp_pack_output = ff_mlp_pack_output;
134    if (ARCH_ARM)
135        ff_mlpdsp_init_arm(c);
136    if (ARCH_X86)
137        ff_mlpdsp_init_x86(c);
138}
139