1/*
2 * Copyright (c) Alexandra Hajkova
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21static void FUNC(ff_hevc_idct_4x4, BIT_DEPTH)(int16_t *coeffs, int col_limit)
22{
23    const int shift = 7;
24    const int shift2 = 20 - BIT_DEPTH;
25    vec_s16 src_01, src_23;
26    vec_s32 res[4];
27    vec_s16 res_packed[2];
28
29    src_01 = vec_ld(0, coeffs);
30    src_23 = vec_ld(16, coeffs);
31
32    transform4x4(src_01, src_23, res, shift, coeffs);
33    src_01 = vec_packs(res[0], res[1]);
34    src_23 = vec_packs(res[2], res[3]);
35    scale(res, res_packed, shift);
36    // transpose
37    src_01 = vec_perm(res_packed[0], res_packed[1], mask[0]);
38    src_23 = vec_perm(res_packed[0], res_packed[1], mask[1]);
39
40    transform4x4(src_01, src_23, res, shift2, coeffs);
41    scale(res, res_packed, shift2);
42    // transpose
43    src_01 = vec_perm(res_packed[0], res_packed[1], mask[0]);
44    src_23 = vec_perm(res_packed[0], res_packed[1], mask[1]);
45
46    vec_st(src_01, 0, coeffs);
47    vec_st(src_23, 16, coeffs);
48}
49