|
| 1 | +/* |
| 2 | + * ATRAC3+ compatible decoder |
| 3 | + * |
| 4 | + * Copyright (c) 2010-2013 Maxim Poliakovski |
| 5 | + * |
| 6 | + * This file is part of FFmpeg. |
| 7 | + * |
| 8 | + * FFmpeg is free software; you can redistribute it and/or |
| 9 | + * modify it under the terms of the GNU Lesser General Public |
| 10 | + * License as published by the Free Software Foundation; either |
| 11 | + * version 2.1 of the License, or (at your option) any later version. |
| 12 | + * |
| 13 | + * FFmpeg is distributed in the hope that it will be useful, |
| 14 | + * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 16 | + * Lesser General Public License for more details. |
| 17 | + * |
| 18 | + * You should have received a copy of the GNU Lesser General Public |
| 19 | + * License along with FFmpeg; if not, write to the Free Software |
| 20 | + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 21 | + */ |
| 22 | + |
| 23 | +/** |
| 24 | + * @file |
| 25 | + * DSP functions for ATRAC3+ decoder. |
| 26 | + */ |
| 27 | + |
| 28 | +/** |
| 29 | + * This file was borrowed from FFmpeg to test analysis PQF implementation |
| 30 | + */ |
| 31 | + |
| 32 | +#include <stdio.h> |
| 33 | +#include <math.h> |
| 34 | +#include <string.h> |
| 35 | + |
| 36 | +#include "atrac3plusdsp.h" |
| 37 | + |
| 38 | +/* lookup table for fast modulo 23 op required for cyclic buffers of the IPQF */ |
| 39 | +static const int mod23_lut[26] = { |
| 40 | + 23, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, |
| 41 | + 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 0 |
| 42 | +}; |
| 43 | + |
| 44 | +/* First half of the 384-tap IPQF filtering coefficients. */ |
| 45 | +static const float ipqf_coeffs1[ATRAC3P_PQF_FIR_LEN][16] = { |
| 46 | + { -5.8336207e-7, -8.0604229e-7, -4.2005411e-7, -4.4400572e-8, |
| 47 | + 3.226247e-8, 3.530856e-8, 1.2660377e-8, 0.000010516783, |
| 48 | + -0.000011838618, 6.005389e-7, 0.0000014333754, 0.0000023108685, |
| 49 | + 0.0000032569742, 0.0000046192422, 0.0000063894258, 0.0000070302972 }, |
| 50 | + { -0.0000091622824, -0.000010502935, -0.0000079212787, -0.0000041712024, |
| 51 | + -0.0000026336629, -0.0000015432918, -5.7168614e-7, 0.0000018111954, |
| 52 | + 0.000023530851, 0.00002780562, 0.000032302323, 0.000036968919, |
| 53 | + 0.000041575615, 0.000045337845, 0.000046043948, 0.000048585582 }, |
| 54 | + { -0.000064464548, -0.000068306952, -0.000073081472, -0.00007612785, |
| 55 | + -0.000074850752, -0.000070208509, -0.000062285151, -0.000058270442, |
| 56 | + -0.000056296329, -0.000049888811, -0.000035615325, -0.000018532943, |
| 57 | + 0.0000016657353, 0.00002610587, 0.000053397067, 0.00008079566 }, |
| 58 | + { -0.00054488552, -0.00052537228, -0.00049731287, -0.00045778, |
| 59 | + -0.00040612387, -0.00034301577, -0.00026866337, -0.00018248901, |
| 60 | + -0.000084307925, 0.000025081157, 0.00014135583, 0.00026649953, |
| 61 | + 0.00039945057, 0.00053928449, 0.00068422867, 0.00083093712 }, |
| 62 | + { -0.0014771431, -0.001283227, -0.0010566821, -0.00079780724, |
| 63 | + -0.00050782406, -0.00018855913, 0.00015771533, 0.00052769453, |
| 64 | + 0.00091862219, 0.001326357, 0.0017469483, 0.0021754825, |
| 65 | + 0.0026067684, 0.0030352892, 0.0034549395, 0.0038591374 }, |
| 66 | + { -0.0022995141, -0.001443546, -0.00049266568, 0.00055068987, |
| 67 | + 0.001682895, 0.0028992873, 0.0041943151, 0.0055614738, |
| 68 | + 0.0069935122, 0.0084823566, 0.010018963, 0.011593862, |
| 69 | + 0.013196872, 0.014817309, 0.016444042, 0.018065533 }, |
| 70 | + { -0.034426283, -0.034281436, -0.033992987, -0.033563249, |
| 71 | + -0.032995768, -0.032295227, -0.031467363, -0.030518902, |
| 72 | + -0.02945766, -0.028291954, -0.027031265, -0.025685543, |
| 73 | + -0.024265358, -0.022781773, -0.021246184, -0.019670162 }, |
| 74 | + { -0.0030586775, -0.0037203205, -0.0042847847, -0.0047529764, |
| 75 | + -0.0051268316, -0.0054091476, -0.0056034233, -0.005714261, |
| 76 | + -0.0057445862, -0.0057025906, -0.0055920109, -0.0054194843, |
| 77 | + -0.0051914565, -0.0049146507, -0.0045959447, -0.0042418269 }, |
| 78 | + { -0.0016376863, -0.0017651899, -0.0018608454, -0.0019252141, |
| 79 | + -0.0019593791, -0.0019653172, -0.0019450618, -0.0018990048, |
| 80 | + -0.00183808, -0.0017501717, -0.0016481078, -0.0015320742, |
| 81 | + -0.0014046903, -0.0012685474, -0.001125814, -0.00097943726 }, |
| 82 | + { -0.00055432378, -0.00055472925, -0.00054783461, -0.00053276919, |
| 83 | + -0.00051135791, -0.00048466062, -0.00045358928, -0.00042499689, |
| 84 | + -0.00036942671, -0.0003392619, -0.00030001783, -0.00025986304, |
| 85 | + -0.0002197204, -0.00018116167, -0.00014691355, -0.00011279432 }, |
| 86 | + { -0.000064147389, -0.00006174868, -0.000054267788, -0.000047133824, |
| 87 | + -0.000042927582, -0.000039477309, -0.000036340745, -0.000029687517, |
| 88 | + -0.000049787737, -0.000041577889, -0.000033864744, -0.000026534748, |
| 89 | + -0.000019841305, -0.000014789486, -0.000013131184, -0.0000099198869 }, |
| 90 | + { -0.0000062990207, -0.0000072701259, -0.000011984052, -0.000017348082, |
| 91 | + -0.000019907106, -0.000021348773, -0.000021961965, -0.000012203576, |
| 92 | + -0.000010840992, 4.6299544e-7, 5.2588763e-7, 2.7792686e-7, |
| 93 | + -2.3649704e-7, -0.0000010897784, -9.171448e-7, -5.22682e-7 } |
| 94 | +}; |
| 95 | + |
| 96 | +/* Second half of the 384-tap IPQF filtering coefficients. */ |
| 97 | +static const float ipqf_coeffs2[ATRAC3P_PQF_FIR_LEN][16] = { |
| 98 | + { 5.22682e-7, 9.171448e-7, 0.0000010897784, 2.3649704e-7, |
| 99 | + -2.7792686e-7, -5.2588763e-7, -4.6299544e-7, 0.000010840992, |
| 100 | + -0.000012203576, -0.000021961965, -0.000021348773, -0.000019907106, |
| 101 | + -0.000017348082, -0.000011984052, -0.0000072701259, -0.0000062990207 }, |
| 102 | + { 0.0000099198869, 0.000013131184, 0.000014789486, 0.000019841305, |
| 103 | + 0.000026534748, 0.000033864744, 0.000041577889, 0.000049787737, |
| 104 | + -0.000029687517, -0.000036340745, -0.000039477309, -0.000042927582, |
| 105 | + -0.000047133824, -0.000054267788, -0.00006174868, -0.000064147389 }, |
| 106 | + { 0.00011279432, 0.00014691355, 0.00018116167, 0.0002197204, |
| 107 | + 0.00025986304, 0.00030001783, 0.0003392619, 0.00036942671, |
| 108 | + -0.00042499689, -0.00045358928, -0.00048466062, -0.00051135791, |
| 109 | + -0.00053276919, -0.00054783461, -0.00055472925, -0.00055432378 }, |
| 110 | + { 0.00097943726, 0.001125814, 0.0012685474, 0.0014046903, |
| 111 | + 0.0015320742, 0.0016481078, 0.0017501717, 0.00183808, |
| 112 | + -0.0018990048, -0.0019450618, -0.0019653172, -0.0019593791, |
| 113 | + -0.0019252141, -0.0018608454, -0.0017651899, -0.0016376863 }, |
| 114 | + { 0.0042418269, 0.0045959447, 0.0049146507, 0.0051914565, |
| 115 | + 0.0054194843, 0.0055920109, 0.0057025906, 0.0057445862, |
| 116 | + -0.005714261, -0.0056034233, -0.0054091476, -0.0051268316, |
| 117 | + -0.0047529764, -0.0042847847, -0.0037203205, -0.0030586775 }, |
| 118 | + { 0.019670162, 0.021246184, 0.022781773, 0.024265358, |
| 119 | + 0.025685543, 0.027031265, 0.028291954, 0.02945766, |
| 120 | + -0.030518902, -0.031467363, -0.032295227, -0.032995768, |
| 121 | + -0.033563249, -0.033992987, -0.034281436, -0.034426283 }, |
| 122 | + { -0.018065533, -0.016444042, -0.014817309, -0.013196872, |
| 123 | + -0.011593862, -0.010018963, -0.0084823566, -0.0069935122, |
| 124 | + 0.0055614738, 0.0041943151, 0.0028992873, 0.001682895, |
| 125 | + 0.00055068987, -0.00049266568, -0.001443546, -0.0022995141 }, |
| 126 | + { -0.0038591374, -0.0034549395, -0.0030352892, -0.0026067684, |
| 127 | + -0.0021754825, -0.0017469483, -0.001326357, -0.00091862219, |
| 128 | + 0.00052769453, 0.00015771533, -0.00018855913, -0.00050782406, |
| 129 | + -0.00079780724, -0.0010566821, -0.001283227, -0.0014771431 }, |
| 130 | + { -0.00083093712, -0.00068422867, -0.00053928449, -0.00039945057, |
| 131 | + -0.00026649953, -0.00014135583, -0.000025081157, 0.000084307925, |
| 132 | + -0.00018248901, -0.00026866337, -0.00034301577, -0.00040612387, |
| 133 | + -0.00045778, -0.00049731287, -0.00052537228, -0.00054488552 }, |
| 134 | + { -0.00008079566, -0.000053397067, -0.00002610587, -0.0000016657353, |
| 135 | + 0.000018532943, 0.000035615325, 0.000049888811, 0.000056296329, |
| 136 | + -0.000058270442, -0.000062285151, -0.000070208509, -0.000074850752, |
| 137 | + -0.00007612785, -0.000073081472, -0.000068306952, -0.000064464548 }, |
| 138 | + { -0.000048585582, -0.000046043948, -0.000045337845, -0.000041575615, |
| 139 | + -0.000036968919, -0.000032302323, -0.00002780562, -0.000023530851, |
| 140 | + 0.0000018111954, -5.7168614e-7, -0.0000015432918, -0.0000026336629, |
| 141 | + -0.0000041712024, -0.0000079212787, -0.000010502935, -0.0000091622824 }, |
| 142 | + { -0.0000070302972, -0.0000063894258, -0.0000046192422, -0.0000032569742, |
| 143 | + -0.0000023108685, -0.0000014333754, -6.005389e-7, 0.000011838618, |
| 144 | + 0.000010516783, 1.2660377e-8, 3.530856e-8, 3.226247e-8, |
| 145 | + -4.4400572e-8, -4.2005411e-7, -8.0604229e-7, -5.8336207e-7 } |
| 146 | +}; |
| 147 | + |
| 148 | +// Just for test |
| 149 | +static void dct4(float* out, const float* x, int N, float scale) { |
| 150 | + for (int k = 0; k < N; k++) { |
| 151 | + double sum = 0; |
| 152 | + for (int n = 0; n < N; n++) { |
| 153 | + sum += x[n] * cos((M_PI/N) * ((double)n + 0.5) * ((double)k + 0.5)); |
| 154 | + } |
| 155 | + out[N - 1 - k] = sum * scale; |
| 156 | + } |
| 157 | +} |
| 158 | + |
| 159 | +void ff_atrac3p_ipqf(Atrac3pIPQFChannelCtx *hist, const float *in, float *out) |
| 160 | +{ |
| 161 | + int i, s, sb, t, pos_now, pos_next; |
| 162 | + float idct_in[ATRAC3P_SUBBANDS]; |
| 163 | + float idct_out[ATRAC3P_SUBBANDS]; |
| 164 | + |
| 165 | + memset(out, 0, ATRAC3P_FRAME_SAMPLES * sizeof(*out)); |
| 166 | + |
| 167 | + for (s = 0; s < ATRAC3P_SUBBAND_SAMPLES; s++) { |
| 168 | + /* pick up one sample from each subband */ |
| 169 | + for (sb = 0; sb < ATRAC3P_SUBBANDS; sb++) |
| 170 | + idct_in[sb] = in[sb * ATRAC3P_SUBBAND_SAMPLES + s]; |
| 171 | + |
| 172 | + dct4(idct_out, idct_in, ATRAC3P_SUBBANDS, 1.0/1024); |
| 173 | + |
| 174 | + /* append the result to the history */ |
| 175 | + for (i = 0; i < 8; i++) { |
| 176 | + hist->buf1[hist->pos][i] = idct_out[i + 8]; |
| 177 | + hist->buf2[hist->pos][i] = idct_out[7 - i]; |
| 178 | + } |
| 179 | + |
| 180 | + pos_now = hist->pos; |
| 181 | + pos_next = mod23_lut[pos_now + 2]; // pos_next = (pos_now + 1) % 23; |
| 182 | + |
| 183 | + for (t = 0; t < ATRAC3P_PQF_FIR_LEN; t++) { |
| 184 | + for (i = 0; i < 8; i++) { |
| 185 | + out[s * 16 + i + 0] += hist->buf1[pos_now][i] * ipqf_coeffs1[t][i] + |
| 186 | + hist->buf2[pos_next][i] * ipqf_coeffs2[t][i]; |
| 187 | + out[s * 16 + i + 8] += hist->buf1[pos_now][7 - i] * ipqf_coeffs1[t][i + 8] + |
| 188 | + hist->buf2[pos_next][7 - i] * ipqf_coeffs2[t][i + 8]; |
| 189 | + } |
| 190 | + pos_now = mod23_lut[pos_next + 2]; // pos_now = (pos_now + 2) % 23; |
| 191 | + pos_next = mod23_lut[pos_now + 2]; // pos_next = (pos_next + 2) % 23; |
| 192 | + } |
| 193 | + |
| 194 | + hist->pos = mod23_lut[hist->pos]; // hist->pos = (hist->pos - 1) % 23; |
| 195 | + } |
| 196 | +} |
0 commit comments