1 /*
2 * Copyright (c) 2014 RISC OS Open Ltd
3 * Author: Ben Avison <bavison@riscosopen.org>
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 <stdint.h>
23
24 #include "libavutil/arm/cpu.h"
25 #include "libavutil/attributes.h"
26 #include "libavcodec/mlpdsp.h"
27
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,
36 int index,
37 unsigned int dest_ch,
38 uint16_t blockpos,
39 unsigned int maxchan,
40 int matrix_noise_shift,
41 int access_unit_size_pow2,
42 int32_t mask);
43
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) \
62 PACK_ORDER(macro,in)
63 PACK_ALL(DECLARE_PACK)
64
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
68 #if CONFIG_THUMB
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
87 #endif
88
mlp_select_pack_output_armv6(uint8_t * ch_assign,int8_t * output_shift,uint8_t max_matrix_channel,int is32)89 static int32_t (*mlp_select_pack_output_armv6(uint8_t *ch_assign,
90 int8_t *output_shift,
91 uint8_t max_matrix_channel,
92 int is32))(int32_t, uint16_t, int32_t (*)[], void *, uint8_t*, int8_t *, uint8_t, int)
93 {
94 int ch_index;
95 int shift = output_shift[0] < 0 || output_shift[0] > 5 ? 6 : output_shift[0];
96 int inorder = 1;
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)
99 };
100 int i;
101
102 if (!is32) // don't support 16-bit output (it's not used by TrueHD)
103 return ff_mlp_pack_output;
104
105 switch (max_matrix_channel) {
106 case 1:
107 ch_index = 0;
108 break;
109 case 5:
110 ch_index = 1;
111 break;
112 case 7:
113 ch_index = 2;
114 break;
115 default:
116 return ff_mlp_pack_output;
117 }
118
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)
123 inorder = 0;
124 }
125 #if CONFIG_THUMB
126 if (!inorder)
127 return ff_mlp_pack_output; // can't currently handle an order array except in ARM mode
128 #else
129 if (shift == 6 && !inorder)
130 return ff_mlp_pack_output; // can't currently handle both an order array and a shift array
131 #endif
132
133 return routine[(inorder*3+ch_index)*7+shift];
134 }
135
ff_mlpdsp_init_arm(MLPDSPContext * c)136 av_cold void ff_mlpdsp_init_arm(MLPDSPContext *c)
137 {
138 int cpu_flags = av_get_cpu_flags();
139
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;
143 }
144 if (have_armv6(cpu_flags))
145 c->mlp_select_pack_output = mlp_select_pack_output_armv6;
146 }
147