mirror of https://git.ffmpeg.org/ffmpeg.git
620 lines
30 KiB
C
620 lines
30 KiB
C
/*
|
|
* Copyright (c) 2013 Seppo Tomperi
|
|
* Copyright (c) 2013 - 2014 Pierre-Edouard Lepere
|
|
*
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
#include "config.h"
|
|
#include "libavutil/cpu.h"
|
|
#include "libavutil/x86/asm.h"
|
|
#include "libavutil/x86/cpu.h"
|
|
#include "libavcodec/get_bits.h" /* required for hevcdsp.h GetBitContext */
|
|
#include "libavcodec/hevcdsp.h"
|
|
#include "libavcodec/x86/hevcdsp.h"
|
|
|
|
#define LFC_FUNC(DIR, DEPTH, OPT) \
|
|
void ff_hevc_ ## DIR ## _loop_filter_chroma_ ## DEPTH ## _ ## OPT(uint8_t *pix, ptrdiff_t stride, int *tc, uint8_t *no_p, uint8_t *no_q);
|
|
|
|
#define LFL_FUNC(DIR, DEPTH, OPT) \
|
|
void ff_hevc_ ## DIR ## _loop_filter_luma_ ## DEPTH ## _ ## OPT(uint8_t *pix, ptrdiff_t stride, int beta, int *tc, uint8_t *no_p, uint8_t *no_q);
|
|
|
|
#define LFC_FUNCS(type, depth, opt) \
|
|
LFC_FUNC(h, depth, opt) \
|
|
LFC_FUNC(v, depth, opt)
|
|
|
|
#define LFL_FUNCS(type, depth, opt) \
|
|
LFL_FUNC(h, depth, opt) \
|
|
LFL_FUNC(v, depth, opt)
|
|
|
|
LFC_FUNCS(uint8_t, 8, sse2)
|
|
LFC_FUNCS(uint8_t, 10, sse2)
|
|
LFC_FUNCS(uint8_t, 12, sse2)
|
|
LFC_FUNCS(uint8_t, 8, avx)
|
|
LFC_FUNCS(uint8_t, 10, avx)
|
|
LFC_FUNCS(uint8_t, 12, avx)
|
|
LFL_FUNCS(uint8_t, 8, sse2)
|
|
LFL_FUNCS(uint8_t, 10, sse2)
|
|
LFL_FUNCS(uint8_t, 12, sse2)
|
|
LFL_FUNCS(uint8_t, 8, ssse3)
|
|
LFL_FUNCS(uint8_t, 10, ssse3)
|
|
LFL_FUNCS(uint8_t, 12, ssse3)
|
|
LFL_FUNCS(uint8_t, 8, avx)
|
|
LFL_FUNCS(uint8_t, 10, avx)
|
|
LFL_FUNCS(uint8_t, 12, avx)
|
|
|
|
#define IDCT_FUNCS(W, opt) \
|
|
void ff_hevc_idct##W##_dc_8_##opt(int16_t *coeffs); \
|
|
void ff_hevc_idct##W##_dc_10_##opt(int16_t *coeffs); \
|
|
void ff_hevc_idct##W##_dc_12_##opt(int16_t *coeffs)
|
|
|
|
IDCT_FUNCS(4x4, mmxext);
|
|
IDCT_FUNCS(8x8, mmxext);
|
|
IDCT_FUNCS(8x8, sse2);
|
|
IDCT_FUNCS(16x16, sse2);
|
|
IDCT_FUNCS(32x32, sse2);
|
|
IDCT_FUNCS(16x16, avx2);
|
|
IDCT_FUNCS(32x32, avx2);
|
|
|
|
#define mc_rep_func(name, bitd, step, W, opt) \
|
|
void ff_hevc_put_hevc_##name##W##_##bitd##_##opt(int16_t *_dst, \
|
|
uint8_t *_src, ptrdiff_t _srcstride, int height, \
|
|
intptr_t mx, intptr_t my, int width) \
|
|
{ \
|
|
int i; \
|
|
uint8_t *src; \
|
|
int16_t *dst; \
|
|
for (i = 0; i < W; i += step) { \
|
|
src = _src + (i * ((bitd + 7) / 8)); \
|
|
dst = _dst + i; \
|
|
ff_hevc_put_hevc_##name##step##_##bitd##_##opt(dst, src, _srcstride, height, mx, my, width); \
|
|
} \
|
|
}
|
|
#define mc_rep_uni_func(name, bitd, step, W, opt) \
|
|
void ff_hevc_put_hevc_uni_##name##W##_##bitd##_##opt(uint8_t *_dst, ptrdiff_t dststride, \
|
|
uint8_t *_src, ptrdiff_t _srcstride, int height, \
|
|
intptr_t mx, intptr_t my, int width) \
|
|
{ \
|
|
int i; \
|
|
uint8_t *src; \
|
|
uint8_t *dst; \
|
|
for (i = 0; i < W; i += step) { \
|
|
src = _src + (i * ((bitd + 7) / 8)); \
|
|
dst = _dst + (i * ((bitd + 7) / 8)); \
|
|
ff_hevc_put_hevc_uni_##name##step##_##bitd##_##opt(dst, dststride, src, _srcstride, \
|
|
height, mx, my, width); \
|
|
} \
|
|
}
|
|
#define mc_rep_bi_func(name, bitd, step, W, opt) \
|
|
void ff_hevc_put_hevc_bi_##name##W##_##bitd##_##opt(uint8_t *_dst, ptrdiff_t dststride, uint8_t *_src, \
|
|
ptrdiff_t _srcstride, int16_t* _src2, \
|
|
int height, intptr_t mx, intptr_t my, int width) \
|
|
{ \
|
|
int i; \
|
|
uint8_t *src; \
|
|
uint8_t *dst; \
|
|
int16_t *src2; \
|
|
for (i = 0; i < W ; i += step) { \
|
|
src = _src + (i * ((bitd + 7) / 8)); \
|
|
dst = _dst + (i * ((bitd + 7) / 8)); \
|
|
src2 = _src2 + i; \
|
|
ff_hevc_put_hevc_bi_##name##step##_##bitd##_##opt(dst, dststride, src, _srcstride, src2, \
|
|
height, mx, my, width); \
|
|
} \
|
|
}
|
|
|
|
#define mc_rep_funcs(name, bitd, step, W, opt) \
|
|
mc_rep_func(name, bitd, step, W, opt); \
|
|
mc_rep_uni_func(name, bitd, step, W, opt); \
|
|
mc_rep_bi_func(name, bitd, step, W, opt)
|
|
|
|
|
|
#if ARCH_X86_64 && HAVE_SSE4_EXTERNAL
|
|
|
|
mc_rep_funcs(pel_pixels, 8, 16, 64, sse4);
|
|
mc_rep_funcs(pel_pixels, 8, 16, 48, sse4);
|
|
mc_rep_funcs(pel_pixels, 8, 16, 32, sse4);
|
|
mc_rep_funcs(pel_pixels, 8, 8, 24, sse4);
|
|
mc_rep_funcs(pel_pixels,10, 8, 64, sse4);
|
|
mc_rep_funcs(pel_pixels,10, 8, 48, sse4);
|
|
mc_rep_funcs(pel_pixels,10, 8, 32, sse4);
|
|
mc_rep_funcs(pel_pixels,10, 8, 24, sse4);
|
|
mc_rep_funcs(pel_pixels,10, 8, 16, sse4);
|
|
mc_rep_funcs(pel_pixels,10, 4, 12, sse4);
|
|
mc_rep_funcs(pel_pixels,12, 8, 64, sse4);
|
|
mc_rep_funcs(pel_pixels,12, 8, 48, sse4);
|
|
mc_rep_funcs(pel_pixels,12, 8, 32, sse4);
|
|
mc_rep_funcs(pel_pixels,12, 8, 24, sse4);
|
|
mc_rep_funcs(pel_pixels,12, 8, 16, sse4);
|
|
mc_rep_funcs(pel_pixels,12, 4, 12, sse4);
|
|
|
|
mc_rep_funcs(epel_h, 8, 16, 64, sse4);
|
|
mc_rep_funcs(epel_h, 8, 16, 48, sse4);
|
|
mc_rep_funcs(epel_h, 8, 16, 32, sse4);
|
|
mc_rep_funcs(epel_h, 8, 8, 24, sse4);
|
|
mc_rep_funcs(epel_h,10, 8, 64, sse4);
|
|
mc_rep_funcs(epel_h,10, 8, 48, sse4);
|
|
mc_rep_funcs(epel_h,10, 8, 32, sse4);
|
|
mc_rep_funcs(epel_h,10, 8, 24, sse4);
|
|
mc_rep_funcs(epel_h,10, 8, 16, sse4);
|
|
mc_rep_funcs(epel_h,10, 4, 12, sse4);
|
|
mc_rep_funcs(epel_h,12, 8, 64, sse4);
|
|
mc_rep_funcs(epel_h,12, 8, 48, sse4);
|
|
mc_rep_funcs(epel_h,12, 8, 32, sse4);
|
|
mc_rep_funcs(epel_h,12, 8, 24, sse4);
|
|
mc_rep_funcs(epel_h,12, 8, 16, sse4);
|
|
mc_rep_funcs(epel_h,12, 4, 12, sse4);
|
|
mc_rep_funcs(epel_v, 8, 16, 64, sse4);
|
|
mc_rep_funcs(epel_v, 8, 16, 48, sse4);
|
|
mc_rep_funcs(epel_v, 8, 16, 32, sse4);
|
|
mc_rep_funcs(epel_v, 8, 8, 24, sse4);
|
|
mc_rep_funcs(epel_v,10, 8, 64, sse4);
|
|
mc_rep_funcs(epel_v,10, 8, 48, sse4);
|
|
mc_rep_funcs(epel_v,10, 8, 32, sse4);
|
|
mc_rep_funcs(epel_v,10, 8, 24, sse4);
|
|
mc_rep_funcs(epel_v,10, 8, 16, sse4);
|
|
mc_rep_funcs(epel_v,10, 4, 12, sse4);
|
|
mc_rep_funcs(epel_v,12, 8, 64, sse4);
|
|
mc_rep_funcs(epel_v,12, 8, 48, sse4);
|
|
mc_rep_funcs(epel_v,12, 8, 32, sse4);
|
|
mc_rep_funcs(epel_v,12, 8, 24, sse4);
|
|
mc_rep_funcs(epel_v,12, 8, 16, sse4);
|
|
mc_rep_funcs(epel_v,12, 4, 12, sse4);
|
|
mc_rep_funcs(epel_hv, 8, 8, 64, sse4);
|
|
mc_rep_funcs(epel_hv, 8, 8, 48, sse4);
|
|
mc_rep_funcs(epel_hv, 8, 8, 32, sse4);
|
|
mc_rep_funcs(epel_hv, 8, 8, 24, sse4);
|
|
mc_rep_funcs(epel_hv, 8, 8, 16, sse4);
|
|
mc_rep_funcs(epel_hv, 8, 4, 12, sse4);
|
|
mc_rep_funcs(epel_hv,10, 8, 64, sse4);
|
|
mc_rep_funcs(epel_hv,10, 8, 48, sse4);
|
|
mc_rep_funcs(epel_hv,10, 8, 32, sse4);
|
|
mc_rep_funcs(epel_hv,10, 8, 24, sse4);
|
|
mc_rep_funcs(epel_hv,10, 8, 16, sse4);
|
|
mc_rep_funcs(epel_hv,10, 4, 12, sse4);
|
|
mc_rep_funcs(epel_hv,12, 8, 64, sse4);
|
|
mc_rep_funcs(epel_hv,12, 8, 48, sse4);
|
|
mc_rep_funcs(epel_hv,12, 8, 32, sse4);
|
|
mc_rep_funcs(epel_hv,12, 8, 24, sse4);
|
|
mc_rep_funcs(epel_hv,12, 8, 16, sse4);
|
|
mc_rep_funcs(epel_hv,12, 4, 12, sse4);
|
|
|
|
mc_rep_funcs(qpel_h, 8, 16, 64, sse4);
|
|
mc_rep_funcs(qpel_h, 8, 16, 48, sse4);
|
|
mc_rep_funcs(qpel_h, 8, 16, 32, sse4);
|
|
mc_rep_funcs(qpel_h, 8, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_h,10, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_h,10, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_h,10, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_h,10, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_h,10, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_h,10, 4, 12, sse4);
|
|
mc_rep_funcs(qpel_h,12, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_h,12, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_h,12, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_h,12, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_h,12, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_h,12, 4, 12, sse4);
|
|
mc_rep_funcs(qpel_v, 8, 16, 64, sse4);
|
|
mc_rep_funcs(qpel_v, 8, 16, 48, sse4);
|
|
mc_rep_funcs(qpel_v, 8, 16, 32, sse4);
|
|
mc_rep_funcs(qpel_v, 8, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_v,10, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_v,10, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_v,10, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_v,10, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_v,10, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_v,10, 4, 12, sse4);
|
|
mc_rep_funcs(qpel_v,12, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_v,12, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_v,12, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_v,12, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_v,12, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_v,12, 4, 12, sse4);
|
|
mc_rep_funcs(qpel_hv, 8, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_hv, 8, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_hv, 8, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_hv, 8, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_hv, 8, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_hv, 8, 4, 12, sse4);
|
|
mc_rep_funcs(qpel_hv,10, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_hv,10, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_hv,10, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_hv,10, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_hv,10, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_hv,10, 4, 12, sse4);
|
|
mc_rep_funcs(qpel_hv,12, 8, 64, sse4);
|
|
mc_rep_funcs(qpel_hv,12, 8, 48, sse4);
|
|
mc_rep_funcs(qpel_hv,12, 8, 32, sse4);
|
|
mc_rep_funcs(qpel_hv,12, 8, 24, sse4);
|
|
mc_rep_funcs(qpel_hv,12, 8, 16, sse4);
|
|
mc_rep_funcs(qpel_hv,12, 4, 12, sse4);
|
|
|
|
#define mc_rep_uni_w(bitd, step, W, opt) \
|
|
void ff_hevc_put_hevc_uni_w##W##_##bitd##_##opt(uint8_t *_dst, ptrdiff_t dststride, int16_t *_src, ptrdiff_t _srcstride,\
|
|
int height, int denom, int _wx, int _ox) \
|
|
{ \
|
|
int i; \
|
|
int16_t *src; \
|
|
uint8_t *dst; \
|
|
for (i = 0; i < W; i += step) { \
|
|
src= _src + i; \
|
|
dst= _dst + (i * ((bitd + 7) / 8)); \
|
|
ff_hevc_put_hevc_uni_w##step##_##bitd##_##opt(dst, dststride, src, _srcstride, \
|
|
height, denom, _wx, _ox); \
|
|
} \
|
|
}
|
|
|
|
mc_rep_uni_w(8, 6, 12, sse4);
|
|
mc_rep_uni_w(8, 8, 16, sse4);
|
|
mc_rep_uni_w(8, 8, 24, sse4);
|
|
mc_rep_uni_w(8, 8, 32, sse4);
|
|
mc_rep_uni_w(8, 8, 48, sse4);
|
|
mc_rep_uni_w(8, 8, 64, sse4);
|
|
|
|
mc_rep_uni_w(10, 6, 12, sse4);
|
|
mc_rep_uni_w(10, 8, 16, sse4);
|
|
mc_rep_uni_w(10, 8, 24, sse4);
|
|
mc_rep_uni_w(10, 8, 32, sse4);
|
|
mc_rep_uni_w(10, 8, 48, sse4);
|
|
mc_rep_uni_w(10, 8, 64, sse4);
|
|
|
|
mc_rep_uni_w(12, 6, 12, sse4);
|
|
mc_rep_uni_w(12, 8, 16, sse4);
|
|
mc_rep_uni_w(12, 8, 24, sse4);
|
|
mc_rep_uni_w(12, 8, 32, sse4);
|
|
mc_rep_uni_w(12, 8, 48, sse4);
|
|
mc_rep_uni_w(12, 8, 64, sse4);
|
|
|
|
#define mc_rep_bi_w(bitd, step, W, opt) \
|
|
void ff_hevc_put_hevc_bi_w##W##_##bitd##_##opt(uint8_t *_dst, ptrdiff_t dststride, int16_t *_src, ptrdiff_t _srcstride, \
|
|
int16_t *_src2, int height, \
|
|
int denom, int _wx0, int _wx1, int _ox0, int _ox1) \
|
|
{ \
|
|
int i; \
|
|
int16_t *src; \
|
|
int16_t *src2; \
|
|
uint8_t *dst; \
|
|
for (i = 0; i < W; i += step) { \
|
|
src = _src + i; \
|
|
src2 = _src2 + i; \
|
|
dst = _dst + (i * ((bitd + 7) / 8)); \
|
|
ff_hevc_put_hevc_bi_w##step##_##bitd##_##opt(dst, dststride, src, _srcstride, src2, \
|
|
height, denom, _wx0, _wx1, _ox0, _ox1); \
|
|
} \
|
|
}
|
|
|
|
mc_rep_bi_w(8, 6, 12, sse4);
|
|
mc_rep_bi_w(8, 8, 16, sse4);
|
|
mc_rep_bi_w(8, 8, 24, sse4);
|
|
mc_rep_bi_w(8, 8, 32, sse4);
|
|
mc_rep_bi_w(8, 8, 48, sse4);
|
|
mc_rep_bi_w(8, 8, 64, sse4);
|
|
|
|
mc_rep_bi_w(10, 6, 12, sse4);
|
|
mc_rep_bi_w(10, 8, 16, sse4);
|
|
mc_rep_bi_w(10, 8, 24, sse4);
|
|
mc_rep_bi_w(10, 8, 32, sse4);
|
|
mc_rep_bi_w(10, 8, 48, sse4);
|
|
mc_rep_bi_w(10, 8, 64, sse4);
|
|
|
|
mc_rep_bi_w(12, 6, 12, sse4);
|
|
mc_rep_bi_w(12, 8, 16, sse4);
|
|
mc_rep_bi_w(12, 8, 24, sse4);
|
|
mc_rep_bi_w(12, 8, 32, sse4);
|
|
mc_rep_bi_w(12, 8, 48, sse4);
|
|
mc_rep_bi_w(12, 8, 64, sse4);
|
|
|
|
#define mc_uni_w_func(name, bitd, W, opt) \
|
|
void ff_hevc_put_hevc_uni_w_##name##W##_##bitd##_##opt(uint8_t *_dst, ptrdiff_t _dststride, \
|
|
uint8_t *_src, ptrdiff_t _srcstride, \
|
|
int height, int denom, \
|
|
int _wx, int _ox, \
|
|
intptr_t mx, intptr_t my, int width) \
|
|
{ \
|
|
LOCAL_ALIGNED_16(int16_t, temp, [71 * MAX_PB_SIZE]); \
|
|
ff_hevc_put_hevc_##name##W##_##bitd##_##opt(temp, _src, _srcstride, height, mx, my, width); \
|
|
ff_hevc_put_hevc_uni_w##W##_##bitd##_##opt(_dst, _dststride, temp, MAX_PB_SIZE, height, denom, _wx, _ox);\
|
|
}
|
|
|
|
#define mc_uni_w_funcs(name, bitd, opt) \
|
|
mc_uni_w_func(name, bitd, 4, opt); \
|
|
mc_uni_w_func(name, bitd, 8, opt); \
|
|
mc_uni_w_func(name, bitd, 12, opt); \
|
|
mc_uni_w_func(name, bitd, 16, opt); \
|
|
mc_uni_w_func(name, bitd, 24, opt); \
|
|
mc_uni_w_func(name, bitd, 32, opt); \
|
|
mc_uni_w_func(name, bitd, 48, opt); \
|
|
mc_uni_w_func(name, bitd, 64, opt)
|
|
|
|
mc_uni_w_funcs(pel_pixels, 8, sse4);
|
|
mc_uni_w_func(pel_pixels, 8, 6, sse4);
|
|
mc_uni_w_funcs(epel_h, 8, sse4);
|
|
mc_uni_w_func(epel_h, 8, 6, sse4);
|
|
mc_uni_w_funcs(epel_v, 8, sse4);
|
|
mc_uni_w_func(epel_v, 8, 6, sse4);
|
|
mc_uni_w_funcs(epel_hv, 8, sse4);
|
|
mc_uni_w_func(epel_hv, 8, 6, sse4);
|
|
mc_uni_w_funcs(qpel_h, 8, sse4);
|
|
mc_uni_w_funcs(qpel_v, 8, sse4);
|
|
mc_uni_w_funcs(qpel_hv, 8, sse4);
|
|
|
|
mc_uni_w_funcs(pel_pixels, 10, sse4);
|
|
mc_uni_w_func(pel_pixels, 10, 6, sse4);
|
|
mc_uni_w_funcs(epel_h, 10, sse4);
|
|
mc_uni_w_func(epel_h, 10, 6, sse4);
|
|
mc_uni_w_funcs(epel_v, 10, sse4);
|
|
mc_uni_w_func(epel_v, 10, 6, sse4);
|
|
mc_uni_w_funcs(epel_hv, 10, sse4);
|
|
mc_uni_w_func(epel_hv, 10, 6, sse4);
|
|
mc_uni_w_funcs(qpel_h, 10, sse4);
|
|
mc_uni_w_funcs(qpel_v, 10, sse4);
|
|
mc_uni_w_funcs(qpel_hv, 10, sse4);
|
|
|
|
mc_uni_w_funcs(pel_pixels, 12, sse4);
|
|
mc_uni_w_func(pel_pixels, 12, 6, sse4);
|
|
mc_uni_w_funcs(epel_h, 12, sse4);
|
|
mc_uni_w_func(epel_h, 12, 6, sse4);
|
|
mc_uni_w_funcs(epel_v, 12, sse4);
|
|
mc_uni_w_func(epel_v, 12, 6, sse4);
|
|
mc_uni_w_funcs(epel_hv, 12, sse4);
|
|
mc_uni_w_func(epel_hv, 12, 6, sse4);
|
|
mc_uni_w_funcs(qpel_h, 12, sse4);
|
|
mc_uni_w_funcs(qpel_v, 12, sse4);
|
|
mc_uni_w_funcs(qpel_hv, 12, sse4);
|
|
|
|
#define mc_bi_w_func(name, bitd, W, opt) \
|
|
void ff_hevc_put_hevc_bi_w_##name##W##_##bitd##_##opt(uint8_t *_dst, ptrdiff_t _dststride, \
|
|
uint8_t *_src, ptrdiff_t _srcstride, \
|
|
int16_t *_src2, \
|
|
int height, int denom, \
|
|
int _wx0, int _wx1, int _ox0, int _ox1, \
|
|
intptr_t mx, intptr_t my, int width) \
|
|
{ \
|
|
LOCAL_ALIGNED_16(int16_t, temp, [71 * MAX_PB_SIZE]); \
|
|
ff_hevc_put_hevc_##name##W##_##bitd##_##opt(temp, _src, _srcstride, height, mx, my, width); \
|
|
ff_hevc_put_hevc_bi_w##W##_##bitd##_##opt(_dst, _dststride, temp, MAX_PB_SIZE, _src2, \
|
|
height, denom, _wx0, _wx1, _ox0, _ox1); \
|
|
}
|
|
|
|
#define mc_bi_w_funcs(name, bitd, opt) \
|
|
mc_bi_w_func(name, bitd, 4, opt); \
|
|
mc_bi_w_func(name, bitd, 8, opt); \
|
|
mc_bi_w_func(name, bitd, 12, opt); \
|
|
mc_bi_w_func(name, bitd, 16, opt); \
|
|
mc_bi_w_func(name, bitd, 24, opt); \
|
|
mc_bi_w_func(name, bitd, 32, opt); \
|
|
mc_bi_w_func(name, bitd, 48, opt); \
|
|
mc_bi_w_func(name, bitd, 64, opt)
|
|
|
|
mc_bi_w_funcs(pel_pixels, 8, sse4);
|
|
mc_bi_w_func(pel_pixels, 8, 6, sse4);
|
|
mc_bi_w_funcs(epel_h, 8, sse4);
|
|
mc_bi_w_func(epel_h, 8, 6, sse4);
|
|
mc_bi_w_funcs(epel_v, 8, sse4);
|
|
mc_bi_w_func(epel_v, 8, 6, sse4);
|
|
mc_bi_w_funcs(epel_hv, 8, sse4);
|
|
mc_bi_w_func(epel_hv, 8, 6, sse4);
|
|
mc_bi_w_funcs(qpel_h, 8, sse4);
|
|
mc_bi_w_funcs(qpel_v, 8, sse4);
|
|
mc_bi_w_funcs(qpel_hv, 8, sse4);
|
|
|
|
mc_bi_w_funcs(pel_pixels, 10, sse4);
|
|
mc_bi_w_func(pel_pixels, 10, 6, sse4);
|
|
mc_bi_w_funcs(epel_h, 10, sse4);
|
|
mc_bi_w_func(epel_h, 10, 6, sse4);
|
|
mc_bi_w_funcs(epel_v, 10, sse4);
|
|
mc_bi_w_func(epel_v, 10, 6, sse4);
|
|
mc_bi_w_funcs(epel_hv, 10, sse4);
|
|
mc_bi_w_func(epel_hv, 10, 6, sse4);
|
|
mc_bi_w_funcs(qpel_h, 10, sse4);
|
|
mc_bi_w_funcs(qpel_v, 10, sse4);
|
|
mc_bi_w_funcs(qpel_hv, 10, sse4);
|
|
|
|
mc_bi_w_funcs(pel_pixels, 12, sse4);
|
|
mc_bi_w_func(pel_pixels, 12, 6, sse4);
|
|
mc_bi_w_funcs(epel_h, 12, sse4);
|
|
mc_bi_w_func(epel_h, 12, 6, sse4);
|
|
mc_bi_w_funcs(epel_v, 12, sse4);
|
|
mc_bi_w_func(epel_v, 12, 6, sse4);
|
|
mc_bi_w_funcs(epel_hv, 12, sse4);
|
|
mc_bi_w_func(epel_hv, 12, 6, sse4);
|
|
mc_bi_w_funcs(qpel_h, 12, sse4);
|
|
mc_bi_w_funcs(qpel_v, 12, sse4);
|
|
mc_bi_w_funcs(qpel_hv, 12, sse4);
|
|
#endif //ARCH_X86_64 && HAVE_SSE4_EXTERNAL
|
|
|
|
|
|
#define EPEL_LINKS(pointer, my, mx, fname, bitd, opt ) \
|
|
PEL_LINK(pointer, 1, my , mx , fname##4 , bitd, opt ); \
|
|
PEL_LINK(pointer, 2, my , mx , fname##6 , bitd, opt ); \
|
|
PEL_LINK(pointer, 3, my , mx , fname##8 , bitd, opt ); \
|
|
PEL_LINK(pointer, 4, my , mx , fname##12, bitd, opt ); \
|
|
PEL_LINK(pointer, 5, my , mx , fname##16, bitd, opt ); \
|
|
PEL_LINK(pointer, 6, my , mx , fname##24, bitd, opt ); \
|
|
PEL_LINK(pointer, 7, my , mx , fname##32, bitd, opt ); \
|
|
PEL_LINK(pointer, 8, my , mx , fname##48, bitd, opt ); \
|
|
PEL_LINK(pointer, 9, my , mx , fname##64, bitd, opt )
|
|
#define QPEL_LINKS(pointer, my, mx, fname, bitd, opt) \
|
|
PEL_LINK(pointer, 1, my , mx , fname##4 , bitd, opt ); \
|
|
PEL_LINK(pointer, 3, my , mx , fname##8 , bitd, opt ); \
|
|
PEL_LINK(pointer, 4, my , mx , fname##12, bitd, opt ); \
|
|
PEL_LINK(pointer, 5, my , mx , fname##16, bitd, opt ); \
|
|
PEL_LINK(pointer, 6, my , mx , fname##24, bitd, opt ); \
|
|
PEL_LINK(pointer, 7, my , mx , fname##32, bitd, opt ); \
|
|
PEL_LINK(pointer, 8, my , mx , fname##48, bitd, opt ); \
|
|
PEL_LINK(pointer, 9, my , mx , fname##64, bitd, opt )
|
|
|
|
|
|
void ff_hevc_dsp_init_x86(HEVCDSPContext *c, const int bit_depth)
|
|
{
|
|
int cpu_flags = av_get_cpu_flags();
|
|
|
|
if (bit_depth == 8) {
|
|
if (EXTERNAL_MMXEXT(cpu_flags)) {
|
|
c->idct_dc[0] = ff_hevc_idct4x4_dc_8_mmxext;
|
|
c->idct_dc[1] = ff_hevc_idct8x8_dc_8_mmxext;
|
|
c->transform_add[0] = ff_hevc_transform_add4_8_mmxext;
|
|
}
|
|
if (EXTERNAL_SSE2(cpu_flags)) {
|
|
c->hevc_v_loop_filter_chroma = ff_hevc_v_loop_filter_chroma_8_sse2;
|
|
c->hevc_h_loop_filter_chroma = ff_hevc_h_loop_filter_chroma_8_sse2;
|
|
if (ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_8_sse2;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_8_sse2;
|
|
}
|
|
c->idct_dc[1] = ff_hevc_idct8x8_dc_8_sse2;
|
|
c->idct_dc[2] = ff_hevc_idct16x16_dc_8_sse2;
|
|
c->idct_dc[3] = ff_hevc_idct32x32_dc_8_sse2;
|
|
|
|
c->transform_add[1] = ff_hevc_transform_add8_8_sse2;
|
|
c->transform_add[2] = ff_hevc_transform_add16_8_sse2;
|
|
c->transform_add[3] = ff_hevc_transform_add32_8_sse2;
|
|
}
|
|
if (EXTERNAL_SSSE3(cpu_flags) && ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_8_ssse3;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_8_ssse3;
|
|
}
|
|
if (EXTERNAL_SSE4(cpu_flags) && ARCH_X86_64) {
|
|
|
|
EPEL_LINKS(c->put_hevc_epel, 0, 0, pel_pixels, 8, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 0, 1, epel_h, 8, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 1, 0, epel_v, 8, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 1, 1, epel_hv, 8, sse4);
|
|
|
|
QPEL_LINKS(c->put_hevc_qpel, 0, 0, pel_pixels, 8, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 0, 1, qpel_h, 8, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 1, 0, qpel_v, 8, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 1, 1, qpel_hv, 8, sse4);
|
|
}
|
|
if (EXTERNAL_AVX(cpu_flags)) {
|
|
c->hevc_v_loop_filter_chroma = ff_hevc_v_loop_filter_chroma_8_avx;
|
|
c->hevc_h_loop_filter_chroma = ff_hevc_h_loop_filter_chroma_8_avx;
|
|
if (ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_8_avx;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_8_avx;
|
|
}
|
|
c->transform_add[1] = ff_hevc_transform_add8_8_avx;
|
|
c->transform_add[2] = ff_hevc_transform_add16_8_avx;
|
|
c->transform_add[3] = ff_hevc_transform_add32_8_avx;
|
|
}
|
|
if (EXTERNAL_AVX2(cpu_flags)) {
|
|
c->idct_dc[2] = ff_hevc_idct16x16_dc_8_avx2;
|
|
c->idct_dc[3] = ff_hevc_idct32x32_dc_8_avx2;
|
|
}
|
|
} else if (bit_depth == 10) {
|
|
if (EXTERNAL_MMXEXT(cpu_flags)) {
|
|
c->transform_add[0] = ff_hevc_transform_add4_10_mmxext;
|
|
c->idct_dc[0] = ff_hevc_idct4x4_dc_10_mmxext;
|
|
c->idct_dc[1] = ff_hevc_idct8x8_dc_10_mmxext;
|
|
}
|
|
if (EXTERNAL_SSE2(cpu_flags)) {
|
|
c->hevc_v_loop_filter_chroma = ff_hevc_v_loop_filter_chroma_10_sse2;
|
|
c->hevc_h_loop_filter_chroma = ff_hevc_h_loop_filter_chroma_10_sse2;
|
|
if (ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_10_sse2;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_10_sse2;
|
|
}
|
|
|
|
c->idct_dc[1] = ff_hevc_idct8x8_dc_10_sse2;
|
|
c->idct_dc[2] = ff_hevc_idct16x16_dc_10_sse2;
|
|
c->idct_dc[3] = ff_hevc_idct32x32_dc_10_sse2;
|
|
|
|
c->transform_add[1] = ff_hevc_transform_add8_10_sse2;
|
|
c->transform_add[2] = ff_hevc_transform_add16_10_sse2;
|
|
c->transform_add[3] = ff_hevc_transform_add32_10_sse2;
|
|
}
|
|
if (EXTERNAL_SSSE3(cpu_flags) && ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_10_ssse3;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_10_ssse3;
|
|
}
|
|
if (EXTERNAL_SSE4(cpu_flags) && ARCH_X86_64) {
|
|
EPEL_LINKS(c->put_hevc_epel, 0, 0, pel_pixels, 10, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 0, 1, epel_h, 10, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 1, 0, epel_v, 10, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 1, 1, epel_hv, 10, sse4);
|
|
|
|
QPEL_LINKS(c->put_hevc_qpel, 0, 0, pel_pixels, 10, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 0, 1, qpel_h, 10, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 1, 0, qpel_v, 10, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 1, 1, qpel_hv, 10, sse4);
|
|
}
|
|
if (EXTERNAL_AVX(cpu_flags)) {
|
|
c->hevc_v_loop_filter_chroma = ff_hevc_v_loop_filter_chroma_10_avx;
|
|
c->hevc_h_loop_filter_chroma = ff_hevc_h_loop_filter_chroma_10_avx;
|
|
if (ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_10_avx;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_10_avx;
|
|
}
|
|
}
|
|
if (EXTERNAL_AVX2(cpu_flags)) {
|
|
|
|
c->idct_dc[2] = ff_hevc_idct16x16_dc_10_avx2;
|
|
c->idct_dc[3] = ff_hevc_idct32x32_dc_10_avx2;
|
|
|
|
c->transform_add[2] = ff_hevc_transform_add16_10_avx2;
|
|
c->transform_add[3] = ff_hevc_transform_add32_10_avx2;
|
|
|
|
}
|
|
} else if (bit_depth == 12) {
|
|
if (EXTERNAL_MMXEXT(cpu_flags)) {
|
|
c->idct_dc[0] = ff_hevc_idct4x4_dc_12_mmxext;
|
|
c->idct_dc[1] = ff_hevc_idct8x8_dc_12_mmxext;
|
|
}
|
|
if (EXTERNAL_SSE2(cpu_flags)) {
|
|
c->hevc_v_loop_filter_chroma = ff_hevc_v_loop_filter_chroma_12_sse2;
|
|
c->hevc_h_loop_filter_chroma = ff_hevc_h_loop_filter_chroma_12_sse2;
|
|
if (ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_12_sse2;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_12_sse2;
|
|
}
|
|
|
|
c->idct_dc[1] = ff_hevc_idct8x8_dc_12_sse2;
|
|
c->idct_dc[2] = ff_hevc_idct16x16_dc_12_sse2;
|
|
c->idct_dc[3] = ff_hevc_idct32x32_dc_12_sse2;
|
|
}
|
|
if (EXTERNAL_SSSE3(cpu_flags) && ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_12_ssse3;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_12_ssse3;
|
|
}
|
|
if (EXTERNAL_SSE4(cpu_flags) && ARCH_X86_64) {
|
|
EPEL_LINKS(c->put_hevc_epel, 0, 0, pel_pixels, 12, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 0, 1, epel_h, 12, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 1, 0, epel_v, 12, sse4);
|
|
EPEL_LINKS(c->put_hevc_epel, 1, 1, epel_hv, 12, sse4);
|
|
|
|
QPEL_LINKS(c->put_hevc_qpel, 0, 0, pel_pixels, 12, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 0, 1, qpel_h, 12, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 1, 0, qpel_v, 12, sse4);
|
|
QPEL_LINKS(c->put_hevc_qpel, 1, 1, qpel_hv, 12, sse4);
|
|
}
|
|
if (EXTERNAL_AVX(cpu_flags)) {
|
|
c->hevc_v_loop_filter_chroma = ff_hevc_v_loop_filter_chroma_12_avx;
|
|
c->hevc_h_loop_filter_chroma = ff_hevc_h_loop_filter_chroma_12_avx;
|
|
if (ARCH_X86_64) {
|
|
c->hevc_v_loop_filter_luma = ff_hevc_v_loop_filter_luma_12_avx;
|
|
c->hevc_h_loop_filter_luma = ff_hevc_h_loop_filter_luma_12_avx;
|
|
}
|
|
}
|
|
if (EXTERNAL_AVX2(cpu_flags)) {
|
|
c->idct_dc[2] = ff_hevc_idct16x16_dc_12_avx2;
|
|
c->idct_dc[3] = ff_hevc_idct32x32_dc_12_avx2;
|
|
}
|
|
}
|
|
}
|