diff options
author | Alexandra Hajkova <alexandra@khirnov.net> | 2016-12-11 14:10:19 +0300 |
---|---|---|
committer | Diego Biurrun <diego@biurrun.de> | 2016-12-12 11:25:16 +0300 |
commit | b0e6b3f4777910d61083976aa9fc78a1e0731aae (patch) | |
tree | 5ee56e64ea7481c8c05f6dea539d82e0287e8006 /libavcodec/ppc/hevcdsp.c | |
parent | fc368497f2fff54ddf5316224c573c9d1939fb25 (diff) |
hevc: ppc: Add HEVC 4x4 IDCT for PowerPC
Signed-off-by: Diego Biurrun <diego@biurrun.de>
Diffstat (limited to 'libavcodec/ppc/hevcdsp.c')
-rw-r--r-- | libavcodec/ppc/hevcdsp.c | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/libavcodec/ppc/hevcdsp.c b/libavcodec/ppc/hevcdsp.c new file mode 100644 index 0000000000..9200e27e65 --- /dev/null +++ b/libavcodec/ppc/hevcdsp.c @@ -0,0 +1,108 @@ +/* SIMD-optimized IDCT functions for HEVC decoding + * Copyright (c) Alexandra Hajkova + * + * This file is part of Libav. + * + * Libav 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. + * + * Libav 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 Libav; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "config.h" +#if HAVE_ALTIVEC_H +#include <altivec.h> +#endif + +#include "libavutil/cpu.h" +#include "libavutil/ppc/cpu.h" +#include "libavutil/ppc/types_altivec.h" +#include "libavutil/ppc/util_altivec.h" + +#include "libavcodec/hevcdsp.h" + +#if HAVE_ALTIVEC +static const vector int16_t trans4[4] = { + { 64, 64, 64, 64, 64, 64, 64, 64 }, + { 83, 36, 83, 36, 83, 36, 83, 36 }, + { 64, -64, 64, -64, 64, -64, 64, -64 }, + { 36, -83, 36, -83, 36, -83, 36, -83 }, +}; + +static const vec_u8 mask[2] = { + { 0x00, 0x01, 0x08, 0x09, 0x10, 0x11, 0x18, 0x19, 0x02, 0x03, 0x0A, 0x0B, 0x12, 0x13, 0x1A, 0x1B }, + { 0x04, 0x05, 0x0C, 0x0D, 0x14, 0x15, 0x1C, 0x1D, 0x06, 0x07, 0x0E, 0x0F, 0x16, 0x17, 0x1E, 0x1F }, +}; + +static void transform4x4(vector int16_t src_01, vector int16_t src_23, + vector int32_t res[4], const int shift, int16_t *coeffs) +{ + vector int16_t src_02, src_13; + vector int32_t zero = vec_splat_s32(0); + vector int32_t e0, o0, e1, o1; + vector int32_t add; + + src_13 = vec_mergel(src_01, src_23); + src_02 = vec_mergeh(src_01, src_23); + + e0 = vec_msums(src_02, trans4[0], zero); + o0 = vec_msums(src_13, trans4[1], zero); + e1 = vec_msums(src_02, trans4[2], zero); + o1 = vec_msums(src_13, trans4[3], zero); + + add = vec_sl(vec_splat_s32(1), vec_splat_u32(shift - 1)); + e0 = vec_add(e0, add); + e1 = vec_add(e1, add); + + res[0] = vec_add(e0, o0); + res[1] = vec_add(e1, o1); + res[2] = vec_sub(e1, o1); + res[3] = vec_sub(e0, o0); +} + +static void scale(vector int32_t res[4], vector int16_t res_packed[2], int shift) +{ + int i; + vector unsigned int v_shift = vec_splat_u32(shift); + + for (i = 0; i < 4; i++) + res[i] = vec_sra(res[i], v_shift); + + // clip16 + res_packed[0] = vec_packs(res[0], res[1]); + res_packed[1] = vec_packs(res[2], res[3]); +} + +#define FUNCDECL(a, depth) a ## _ ## depth ## _altivec +#define FUNC(a, b) FUNCDECL(a, b) + +#define BIT_DEPTH 8 +#include "hevcdsp_template.c" +#undef BIT_DEPTH + +#define BIT_DEPTH 10 +#include "hevcdsp_template.c" +#undef BIT_DEPTH +#endif /* HAVE_ALTIVEC */ + +av_cold void ff_hevc_dsp_init_ppc(HEVCDSPContext *c, const int bit_depth) +{ +#if HAVE_ALTIVEC + if (!PPC_ALTIVEC(av_get_cpu_flags())) + return; + + if (bit_depth == 8) + c->idct[0] = ff_hevc_idct_4x4_8_altivec; + if (bit_depth == 10) + c->idct[0] = ff_hevc_idct_4x4_10_altivec; +#endif /* HAVE_ALTIVEC */ +} |