2 * Copyright (c) 2014 RISC OS Open Ltd
3 * Author: Ben Avison <bavison@riscosopen.org>
5 * This file is part of FFmpeg.
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.
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.
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
24 #include "libavutil/arm/cpu.h"
25 #include "libavutil/attributes.h"
26 #include "libavcodec/mlpdsp.h"
28 void ff_mlp_filter_channel_arm(int32_t *state, const int32_t *coeff,
29 int firorder, int iirorder,
30 unsigned int filter_shift, int32_t mask,
31 int blocksize, int32_t *sample_buffer);
32 void ff_mlp_rematrix_channel_arm(int32_t *samples,
33 const int32_t *coeffs,
34 const uint8_t *bypassed_lsbs,
35 const int8_t *noise_buffer,
40 int matrix_noise_shift,
41 int access_unit_size_pow2,
44 #define DECLARE_PACK(order,channels,shift) \
45 int32_t ff_mlp_pack_output_##order##order_##channels##ch_##shift##shift_armv6(int32_t, uint16_t, int32_t (*)[], void *, uint8_t*, int8_t *, uint8_t, int);
46 #define ENUMERATE_PACK(order,channels,shift) \
47 ff_mlp_pack_output_##order##order_##channels##ch_##shift##shift_armv6,
48 #define PACK_CHANNELS(macro,order,channels) \
49 macro(order,channels,0) \
50 macro(order,channels,1) \
51 macro(order,channels,2) \
52 macro(order,channels,3) \
53 macro(order,channels,4) \
54 macro(order,channels,5) \
55 macro(order,channels,mixed)
56 #define PACK_ORDER(macro,order) \
57 PACK_CHANNELS(macro,order,2) \
58 PACK_CHANNELS(macro,order,6) \
59 PACK_CHANNELS(macro,order,8)
60 #define PACK_ALL(macro) \
61 PACK_ORDER(macro,outof) \
63 PACK_ALL(DECLARE_PACK)
65 #define ff_mlp_pack_output_outoforder_2ch_mixedshift_armv6 0
66 #define ff_mlp_pack_output_outoforder_6ch_mixedshift_armv6 0
67 #define ff_mlp_pack_output_outoforder_8ch_mixedshift_armv6 0
69 #define ff_mlp_pack_output_outoforder_2ch_0shift_armv6 0
70 #define ff_mlp_pack_output_outoforder_2ch_1shift_armv6 0
71 #define ff_mlp_pack_output_outoforder_2ch_2shift_armv6 0
72 #define ff_mlp_pack_output_outoforder_2ch_3shift_armv6 0
73 #define ff_mlp_pack_output_outoforder_2ch_4shift_armv6 0
74 #define ff_mlp_pack_output_outoforder_2ch_5shift_armv6 0
75 #define ff_mlp_pack_output_outoforder_6ch_0shift_armv6 0
76 #define ff_mlp_pack_output_outoforder_6ch_1shift_armv6 0
77 #define ff_mlp_pack_output_outoforder_6ch_2shift_armv6 0
78 #define ff_mlp_pack_output_outoforder_6ch_3shift_armv6 0
79 #define ff_mlp_pack_output_outoforder_6ch_4shift_armv6 0
80 #define ff_mlp_pack_output_outoforder_6ch_5shift_armv6 0
81 #define ff_mlp_pack_output_outoforder_8ch_0shift_armv6 0
82 #define ff_mlp_pack_output_outoforder_8ch_1shift_armv6 0
83 #define ff_mlp_pack_output_outoforder_8ch_2shift_armv6 0
84 #define ff_mlp_pack_output_outoforder_8ch_3shift_armv6 0
85 #define ff_mlp_pack_output_outoforder_8ch_4shift_armv6 0
86 #define ff_mlp_pack_output_outoforder_8ch_5shift_armv6 0
89 static int32_t (*mlp_select_pack_output_armv6(uint8_t *ch_assign,
91 uint8_t max_matrix_channel,
92 int is32))(int32_t, uint16_t, int32_t (*)[], void *, uint8_t*, int8_t *, uint8_t, int)
95 int shift = output_shift[0] < 0 || output_shift[0] > 5 ? 6 : output_shift[0];
97 static int32_t (*const routine[2*3*7])(int32_t, uint16_t, int32_t (*)[], void *, uint8_t*, int8_t *, uint8_t, int) = {
98 PACK_ALL(ENUMERATE_PACK)
102 if (!is32) // don't support 16-bit output (it's not used by TrueHD)
103 return ff_mlp_pack_output;
105 switch (max_matrix_channel) {
116 return ff_mlp_pack_output;
119 for (i = 0; i <= max_matrix_channel; i++) {
120 if (shift != 6 && output_shift[i] != shift)
121 shift = 6; // indicate mixed shifts
122 if (ch_assign[i] != i)
127 return ff_mlp_pack_output; // can't currently handle an order array except in ARM mode
129 if (shift == 6 && !inorder)
130 return ff_mlp_pack_output; // can't currently handle both an order array and a shift array
133 return routine[(inorder*3+ch_index)*7+shift];
136 av_cold void ff_mlpdsp_init_arm(MLPDSPContext *c)
138 int cpu_flags = av_get_cpu_flags();
140 if (have_armv5te(cpu_flags)) {
141 c->mlp_filter_channel = ff_mlp_filter_channel_arm;
142 c->mlp_rematrix_channel = ff_mlp_rematrix_channel_arm;
144 if (have_armv6(cpu_flags))
145 c->mlp_select_pack_output = mlp_select_pack_output_armv6;