diff options
author | Daniil Cherednik <dan.cherednik@gmail.com> | 2024-05-17 21:31:15 +0000 |
---|---|---|
committer | Daniil Cherednik <dan.cherednik@gmail.com> | 2024-06-14 18:58:21 +0200 |
commit | f018c661a7a6093cabc40784ae3b1fefb96c82de (patch) | |
tree | 52e057234a8b494ac583a75ca4e5833813933476 | |
parent | 065121a354e99e59dfd57124d7650d82213db094 (diff) | |
download | atracdenc-f018c661a7a6093cabc40784ae3b1fefb96c82de.tar.gz |
Simple ipqf test and ref files
ref files contains samples just before and after ipqf
-rw-r--r-- | src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.c | 196 | ||||
-rw-r--r-- | src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.h | 48 | ||||
-rw-r--r-- | src/atrac/atrac3plus_pqf/ut/ipqf_ut.cpp | 80 | ||||
-rwxr-xr-x | src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_mr.dat | bin | 0 -> 32768 bytes | |||
-rwxr-xr-x | src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_out.dat | bin | 0 -> 32768 bytes | |||
-rw-r--r-- | src/util_ut.cpp | 1 | ||||
-rw-r--r-- | test/CMakeLists.txt | 25 |
7 files changed, 349 insertions, 1 deletions
diff --git a/src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.c b/src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.c new file mode 100644 index 0000000..940cf05 --- /dev/null +++ b/src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.c @@ -0,0 +1,196 @@ +/* + * ATRAC3+ compatible decoder + * + * Copyright (c) 2010-2013 Maxim Poliakovski + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +/** + * @file + * DSP functions for ATRAC3+ decoder. + */ + +/** + * This file was borrowed from FFmpeg to test analysis PQF implementation + */ + +#include <stdio.h> +#include <math.h> +#include <string.h> + +#include "atrac3plusdsp.h" + +/* lookup table for fast modulo 23 op required for cyclic buffers of the IPQF */ +static const int mod23_lut[26] = { + 23, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, + 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 0 +}; + +/* First half of the 384-tap IPQF filtering coefficients. */ +static const float ipqf_coeffs1[ATRAC3P_PQF_FIR_LEN][16] = { + { -5.8336207e-7, -8.0604229e-7, -4.2005411e-7, -4.4400572e-8, + 3.226247e-8, 3.530856e-8, 1.2660377e-8, 0.000010516783, + -0.000011838618, 6.005389e-7, 0.0000014333754, 0.0000023108685, + 0.0000032569742, 0.0000046192422, 0.0000063894258, 0.0000070302972 }, + { -0.0000091622824, -0.000010502935, -0.0000079212787, -0.0000041712024, + -0.0000026336629, -0.0000015432918, -5.7168614e-7, 0.0000018111954, + 0.000023530851, 0.00002780562, 0.000032302323, 0.000036968919, + 0.000041575615, 0.000045337845, 0.000046043948, 0.000048585582 }, + { -0.000064464548, -0.000068306952, -0.000073081472, -0.00007612785, + -0.000074850752, -0.000070208509, -0.000062285151, -0.000058270442, + -0.000056296329, -0.000049888811, -0.000035615325, -0.000018532943, + 0.0000016657353, 0.00002610587, 0.000053397067, 0.00008079566 }, + { -0.00054488552, -0.00052537228, -0.00049731287, -0.00045778, + -0.00040612387, -0.00034301577, -0.00026866337, -0.00018248901, + -0.000084307925, 0.000025081157, 0.00014135583, 0.00026649953, + 0.00039945057, 0.00053928449, 0.00068422867, 0.00083093712 }, + { -0.0014771431, -0.001283227, -0.0010566821, -0.00079780724, + -0.00050782406, -0.00018855913, 0.00015771533, 0.00052769453, + 0.00091862219, 0.001326357, 0.0017469483, 0.0021754825, + 0.0026067684, 0.0030352892, 0.0034549395, 0.0038591374 }, + { -0.0022995141, -0.001443546, -0.00049266568, 0.00055068987, + 0.001682895, 0.0028992873, 0.0041943151, 0.0055614738, + 0.0069935122, 0.0084823566, 0.010018963, 0.011593862, + 0.013196872, 0.014817309, 0.016444042, 0.018065533 }, + { -0.034426283, -0.034281436, -0.033992987, -0.033563249, + -0.032995768, -0.032295227, -0.031467363, -0.030518902, + -0.02945766, -0.028291954, -0.027031265, -0.025685543, + -0.024265358, -0.022781773, -0.021246184, -0.019670162 }, + { -0.0030586775, -0.0037203205, -0.0042847847, -0.0047529764, + -0.0051268316, -0.0054091476, -0.0056034233, -0.005714261, + -0.0057445862, -0.0057025906, -0.0055920109, -0.0054194843, + -0.0051914565, -0.0049146507, -0.0045959447, -0.0042418269 }, + { -0.0016376863, -0.0017651899, -0.0018608454, -0.0019252141, + -0.0019593791, -0.0019653172, -0.0019450618, -0.0018990048, + -0.00183808, -0.0017501717, -0.0016481078, -0.0015320742, + -0.0014046903, -0.0012685474, -0.001125814, -0.00097943726 }, + { -0.00055432378, -0.00055472925, -0.00054783461, -0.00053276919, + -0.00051135791, -0.00048466062, -0.00045358928, -0.00042499689, + -0.00036942671, -0.0003392619, -0.00030001783, -0.00025986304, + -0.0002197204, -0.00018116167, -0.00014691355, -0.00011279432 }, + { -0.000064147389, -0.00006174868, -0.000054267788, -0.000047133824, + -0.000042927582, -0.000039477309, -0.000036340745, -0.000029687517, + -0.000049787737, -0.000041577889, -0.000033864744, -0.000026534748, + -0.000019841305, -0.000014789486, -0.000013131184, -0.0000099198869 }, + { -0.0000062990207, -0.0000072701259, -0.000011984052, -0.000017348082, + -0.000019907106, -0.000021348773, -0.000021961965, -0.000012203576, + -0.000010840992, 4.6299544e-7, 5.2588763e-7, 2.7792686e-7, + -2.3649704e-7, -0.0000010897784, -9.171448e-7, -5.22682e-7 } +}; + +/* Second half of the 384-tap IPQF filtering coefficients. */ +static const float ipqf_coeffs2[ATRAC3P_PQF_FIR_LEN][16] = { + { 5.22682e-7, 9.171448e-7, 0.0000010897784, 2.3649704e-7, + -2.7792686e-7, -5.2588763e-7, -4.6299544e-7, 0.000010840992, + -0.000012203576, -0.000021961965, -0.000021348773, -0.000019907106, + -0.000017348082, -0.000011984052, -0.0000072701259, -0.0000062990207 }, + { 0.0000099198869, 0.000013131184, 0.000014789486, 0.000019841305, + 0.000026534748, 0.000033864744, 0.000041577889, 0.000049787737, + -0.000029687517, -0.000036340745, -0.000039477309, -0.000042927582, + -0.000047133824, -0.000054267788, -0.00006174868, -0.000064147389 }, + { 0.00011279432, 0.00014691355, 0.00018116167, 0.0002197204, + 0.00025986304, 0.00030001783, 0.0003392619, 0.00036942671, + -0.00042499689, -0.00045358928, -0.00048466062, -0.00051135791, + -0.00053276919, -0.00054783461, -0.00055472925, -0.00055432378 }, + { 0.00097943726, 0.001125814, 0.0012685474, 0.0014046903, + 0.0015320742, 0.0016481078, 0.0017501717, 0.00183808, + -0.0018990048, -0.0019450618, -0.0019653172, -0.0019593791, + -0.0019252141, -0.0018608454, -0.0017651899, -0.0016376863 }, + { 0.0042418269, 0.0045959447, 0.0049146507, 0.0051914565, + 0.0054194843, 0.0055920109, 0.0057025906, 0.0057445862, + -0.005714261, -0.0056034233, -0.0054091476, -0.0051268316, + -0.0047529764, -0.0042847847, -0.0037203205, -0.0030586775 }, + { 0.019670162, 0.021246184, 0.022781773, 0.024265358, + 0.025685543, 0.027031265, 0.028291954, 0.02945766, + -0.030518902, -0.031467363, -0.032295227, -0.032995768, + -0.033563249, -0.033992987, -0.034281436, -0.034426283 }, + { -0.018065533, -0.016444042, -0.014817309, -0.013196872, + -0.011593862, -0.010018963, -0.0084823566, -0.0069935122, + 0.0055614738, 0.0041943151, 0.0028992873, 0.001682895, + 0.00055068987, -0.00049266568, -0.001443546, -0.0022995141 }, + { -0.0038591374, -0.0034549395, -0.0030352892, -0.0026067684, + -0.0021754825, -0.0017469483, -0.001326357, -0.00091862219, + 0.00052769453, 0.00015771533, -0.00018855913, -0.00050782406, + -0.00079780724, -0.0010566821, -0.001283227, -0.0014771431 }, + { -0.00083093712, -0.00068422867, -0.00053928449, -0.00039945057, + -0.00026649953, -0.00014135583, -0.000025081157, 0.000084307925, + -0.00018248901, -0.00026866337, -0.00034301577, -0.00040612387, + -0.00045778, -0.00049731287, -0.00052537228, -0.00054488552 }, + { -0.00008079566, -0.000053397067, -0.00002610587, -0.0000016657353, + 0.000018532943, 0.000035615325, 0.000049888811, 0.000056296329, + -0.000058270442, -0.000062285151, -0.000070208509, -0.000074850752, + -0.00007612785, -0.000073081472, -0.000068306952, -0.000064464548 }, + { -0.000048585582, -0.000046043948, -0.000045337845, -0.000041575615, + -0.000036968919, -0.000032302323, -0.00002780562, -0.000023530851, + 0.0000018111954, -5.7168614e-7, -0.0000015432918, -0.0000026336629, + -0.0000041712024, -0.0000079212787, -0.000010502935, -0.0000091622824 }, + { -0.0000070302972, -0.0000063894258, -0.0000046192422, -0.0000032569742, + -0.0000023108685, -0.0000014333754, -6.005389e-7, 0.000011838618, + 0.000010516783, 1.2660377e-8, 3.530856e-8, 3.226247e-8, + -4.4400572e-8, -4.2005411e-7, -8.0604229e-7, -5.8336207e-7 } +}; + +// Just for test +static void dct4(float* out, const float* x, int N, float scale) { + for (int k = 0; k < N; k++) { + double sum = 0; + for (int n = 0; n < N; n++) { + sum += x[n] * cos((M_PI/N) * ((double)n + 0.5) * ((double)k + 0.5)); + } + out[N - 1 - k] = sum * scale; + } +} + +void ff_atrac3p_ipqf(Atrac3pIPQFChannelCtx *hist, const float *in, float *out) +{ + int i, s, sb, t, pos_now, pos_next; + float idct_in[ATRAC3P_SUBBANDS]; + float idct_out[ATRAC3P_SUBBANDS]; + + memset(out, 0, ATRAC3P_FRAME_SAMPLES * sizeof(*out)); + + for (s = 0; s < ATRAC3P_SUBBAND_SAMPLES; s++) { + /* pick up one sample from each subband */ + for (sb = 0; sb < ATRAC3P_SUBBANDS; sb++) + idct_in[sb] = in[sb * ATRAC3P_SUBBAND_SAMPLES + s]; + + dct4(idct_out, idct_in, ATRAC3P_SUBBANDS, 1.0/1024); + + /* append the result to the history */ + for (i = 0; i < 8; i++) { + hist->buf1[hist->pos][i] = idct_out[i + 8]; + hist->buf2[hist->pos][i] = idct_out[7 - i]; + } + + pos_now = hist->pos; + pos_next = mod23_lut[pos_now + 2]; // pos_next = (pos_now + 1) % 23; + + for (t = 0; t < ATRAC3P_PQF_FIR_LEN; t++) { + for (i = 0; i < 8; i++) { + out[s * 16 + i + 0] += hist->buf1[pos_now][i] * ipqf_coeffs1[t][i] + + hist->buf2[pos_next][i] * ipqf_coeffs2[t][i]; + out[s * 16 + i + 8] += hist->buf1[pos_now][7 - i] * ipqf_coeffs1[t][i + 8] + + hist->buf2[pos_next][7 - i] * ipqf_coeffs2[t][i + 8]; + } + pos_now = mod23_lut[pos_next + 2]; // pos_now = (pos_now + 2) % 23; + pos_next = mod23_lut[pos_now + 2]; // pos_next = (pos_next + 2) % 23; + } + + hist->pos = mod23_lut[hist->pos]; // hist->pos = (hist->pos - 1) % 23; + } +} diff --git a/src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.h b/src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.h new file mode 100644 index 0000000..af45d92 --- /dev/null +++ b/src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.h @@ -0,0 +1,48 @@ +/* + * This file is part of AtracDEnc. + * + * AtracDEnc is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * AtracDEnc is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with AtracDEnc; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#ifndef ATRAC3PLUSDSP_H +#define ATRAC3PLUSDSP_H + +#define ATRAC3P_SUBBANDS 16 ///< number of PQF subbands +#define ATRAC3P_SUBBAND_SAMPLES 128 ///< number of samples per subband +#define ATRAC3P_FRAME_SAMPLES (ATRAC3P_SUBBAND_SAMPLES * ATRAC3P_SUBBANDS) + +#define ATRAC3P_PQF_FIR_LEN 12 + +typedef struct Atrac3pIPQFChannelCtx { + float buf1[ATRAC3P_PQF_FIR_LEN * 2][8]; + float buf2[ATRAC3P_PQF_FIR_LEN * 2][8]; + int pos; +} Atrac3pIPQFChannelCtx; + +#ifdef __cplusplus +extern "C" +{ +#endif /* __cplusplus */ + +void ff_atrac3p_ipqf(Atrac3pIPQFChannelCtx *hist, const float *in, float *out); + +#ifdef __cplusplus +} +#endif /* __cplusplus */ + + + +#endif + diff --git a/src/atrac/atrac3plus_pqf/ut/ipqf_ut.cpp b/src/atrac/atrac3plus_pqf/ut/ipqf_ut.cpp new file mode 100644 index 0000000..ababfcd --- /dev/null +++ b/src/atrac/atrac3plus_pqf/ut/ipqf_ut.cpp @@ -0,0 +1,80 @@ +/* + * This file is part of AtracDEnc. + * + * AtracDEnc is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * AtracDEnc is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with AtracDEnc; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "atrac3plusdsp.h" + +#include <gtest/gtest.h> + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <math.h> + +#define SAMPLES 8192 + +static int read_file(FILE* f, float buf[SAMPLES]) { + size_t ret = fread(&buf[0], sizeof(buf[0]), SAMPLES, f); + + if (ret != SAMPLES) { + return -1; + } + return 0; +} + +TEST(ipqf, CheckOnRefData) { + FILE* mr_f = fopen("test_data/ipqftest_pcm_mr.dat", "r"); + if (!mr_f) { + fprintf(stderr, "unable to open multirate file\n"); + FAIL(); + } + + FILE* ref_f = fopen("test_data/ipqftest_pcm_out.dat", "r"); + if (!ref_f) { + fclose(mr_f); + fprintf(stderr, "unable to open reference file\n"); + FAIL(); + } + + float mr_data[SAMPLES] = {0}; + float ref_data[SAMPLES] = {0}; + + if (read_file(mr_f, mr_data) < 0) { + fprintf(stderr, "unable to read multirate file\n"); + FAIL(); + } + + if (read_file(ref_f, ref_data) < 0) { + fprintf(stderr, "unable to read reference file\n"); + FAIL(); + } + + Atrac3pIPQFChannelCtx ctx; + memset(&ctx, 0, sizeof(ctx)); + + float tmp[SAMPLES] = {0}; + + for (int i = 0; i < SAMPLES; i+= 2048) { + ff_atrac3p_ipqf(&ctx, &mr_data[i], &tmp[i]); + } + + const static float err = 1.0 / (float)(1<<26); + for (int i = 0; i < SAMPLES; i++) { + //fprintf(stdout, "%f %f\n", tmp[i], ref_data[i]); + EXPECT_NEAR(tmp[i], ref_data[i], err); + } +} diff --git a/src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_mr.dat b/src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_mr.dat Binary files differnew file mode 100755 index 0000000..4d0cf30 --- /dev/null +++ b/src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_mr.dat diff --git a/src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_out.dat b/src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_out.dat Binary files differnew file mode 100755 index 0000000..9ff2491 --- /dev/null +++ b/src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_out.dat diff --git a/src/util_ut.cpp b/src/util_ut.cpp index 80217a9..9fb61c4 100644 --- a/src/util_ut.cpp +++ b/src/util_ut.cpp @@ -21,7 +21,6 @@ #include <vector> - TEST(Util, SwapArrayTest) { TFloat arr[8] = { 0, 1, 2, 3, 4, 5, 6, 7 }; SwapArray(arr, 8); diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt index 4b5c295..807b4ab 100644 --- a/test/CMakeLists.txt +++ b/test/CMakeLists.txt @@ -21,5 +21,30 @@ target_link_libraries(atracdenc_ut GTest::gtest_main ) +configure_file( + ../src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_mr.dat + ${CMAKE_CURRENT_BINARY_DIR}/test_data/ipqftest_pcm_mr.dat + COPYONLY +) + +configure_file( + ../src/atrac/atrac3plus_pqf/ut/test_data/ipqftest_pcm_out.dat + ${CMAKE_CURRENT_BINARY_DIR}/test_data/ipqftest_pcm_out.dat + COPYONLY +) + +set(at3plus_pqf_ut + ../src/atrac/atrac3plus_pqf/ut/ipqf_ut.cpp + ../src/atrac/atrac3plus_pqf/ut/atrac3plusdsp.c +) + +add_executable(at3plus_pqf_ut ${at3plus_pqf_ut}) + +target_link_libraries(at3plus_pqf_ut + m + GTest::gtest_main +) + enable_testing() add_test(ut atracdenc_ut) +add_test(at3plus_pqf_ut at3plus_pqf_ut) |