Merge commit 'a1a143adb0fd11c474221431417cff25db7d920f'
[ffmpeg.git] / libavcodec / arm / mlpdsp_init_arm.c
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
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
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 }