| 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 | } |