avcodec/hevc: new idct + asm
[ffmpeg.git] / libavcodec / x86 / hevcdsp_init.c
index 565c4a0..58a0891 100644 (file)
 #include "libavcodec/x86/hevcdsp.h"
 
 
 #include "libavcodec/x86/hevcdsp.h"
 
 
-#define mc_rep_func(name, bitd, step, W) \
-void ff_hevc_put_hevc_##name##W##_##bitd##_sse4(int16_t *_dst, ptrdiff_t dststride,                             \
+#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) \
+LFC_FUNC(h, depth, sse2)  \
+LFC_FUNC(v, depth, sse2)
+
+#define LFL_FUNCS(type, depth) \
+LFL_FUNC(h, depth, ssse3)  \
+LFL_FUNC(v, depth, ssse3)
+
+LFC_FUNCS(uint8_t,   8)
+LFC_FUNCS(uint8_t,  10)
+LFL_FUNCS(uint8_t,   8)
+LFL_FUNCS(uint8_t,  10)
+
+#if HAVE_SSE2_EXTERNAL
+void ff_hevc_idct32_dc_add_8_sse2(uint8_t *dst, int16_t *coeffs, ptrdiff_t stride)
+{
+    ff_hevc_idct16_dc_add_8_sse2(dst, coeffs, stride);
+    ff_hevc_idct16_dc_add_8_sse2(dst+16, coeffs, stride);
+    ff_hevc_idct16_dc_add_8_sse2(dst+16*stride, coeffs, stride);
+    ff_hevc_idct16_dc_add_8_sse2(dst+16*stride+16, coeffs, stride);
+}
+
+void ff_hevc_idct16_dc_add_10_sse2(uint8_t *dst, int16_t *coeffs, ptrdiff_t stride)
+{
+    ff_hevc_idct8_dc_add_10_sse2(dst, coeffs, stride);
+    ff_hevc_idct8_dc_add_10_sse2(dst+16, coeffs, stride);
+    ff_hevc_idct8_dc_add_10_sse2(dst+8*stride, coeffs, stride);
+    ff_hevc_idct8_dc_add_10_sse2(dst+8*stride+16, coeffs, stride);
+}
+
+void ff_hevc_idct32_dc_add_10_sse2(uint8_t *dst, int16_t *coeffs, ptrdiff_t stride)
+{
+    ff_hevc_idct16_dc_add_10_sse2(dst, coeffs, stride);
+    ff_hevc_idct16_dc_add_10_sse2(dst+32, coeffs, stride);
+    ff_hevc_idct16_dc_add_10_sse2(dst+16*stride, coeffs, stride);
+    ff_hevc_idct16_dc_add_10_sse2(dst+16*stride+32, coeffs, stride);
+}
+#endif //HAVE_SSE2_EXTERNAL
+#if HAVE_AVX_EXTERNAL
+void ff_hevc_idct16_dc_add_10_avx(uint8_t *dst, int16_t *coeffs, ptrdiff_t stride)
+{
+    ff_hevc_idct8_dc_add_10_avx(dst, coeffs, stride);
+    ff_hevc_idct8_dc_add_10_avx(dst+16, coeffs, stride);
+    ff_hevc_idct8_dc_add_10_avx(dst+8*stride, coeffs, stride);
+    ff_hevc_idct8_dc_add_10_avx(dst+8*stride+16, coeffs, stride);
+}
+
+void ff_hevc_idct32_dc_add_10_avx(uint8_t *dst, int16_t *coeffs, ptrdiff_t stride)
+{
+    ff_hevc_idct16_dc_add_10_avx(dst, coeffs, stride);
+    ff_hevc_idct16_dc_add_10_avx(dst+32, coeffs, stride);
+    ff_hevc_idct16_dc_add_10_avx(dst+16*stride, coeffs, stride);
+    ff_hevc_idct16_dc_add_10_avx(dst+16*stride+32, coeffs, stride);
+}
+#endif //HAVE_AVX_EXTERNAL
+
+#define mc_rep_func(name, bitd, step, W, opt) \
+void ff_hevc_put_hevc_##name##W##_##bitd##_##opt(int16_t *_dst, ptrdiff_t dststride,                            \
                                                 uint8_t *_src, ptrdiff_t _srcstride, int height,                \
                                                 intptr_t mx, intptr_t my, int width)                            \
 {                                                                                                               \
                                                 uint8_t *_src, ptrdiff_t _srcstride, int height,                \
                                                 intptr_t mx, intptr_t my, int width)                            \
 {                                                                                                               \
@@ -40,11 +103,11 @@ void ff_hevc_put_hevc_##name##W##_##bitd##_sse4(int16_t *_dst, ptrdiff_t dststri
     for (i = 0; i < W; i += step) {                                                                             \
         src  = _src + (i * ((bitd + 7) / 8));                                                                   \
         dst = _dst + i;                                                                                         \
     for (i = 0; i < W; i += step) {                                                                             \
         src  = _src + (i * ((bitd + 7) / 8));                                                                   \
         dst = _dst + i;                                                                                         \
-        ff_hevc_put_hevc_##name##step##_##bitd##_sse4(dst, dststride, src, _srcstride, height, mx, my, width);  \
+        ff_hevc_put_hevc_##name##step##_##bitd##_##opt(dst, dststride, src, _srcstride, height, mx, my, width); \
     }                                                                                                           \
 }
     }                                                                                                           \
 }
-#define mc_rep_uni_func(name, bitd, step, W) \
-void ff_hevc_put_hevc_uni_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dststride,                         \
+#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)                        \
 {                                                                                                               \
                                                     uint8_t *_src, ptrdiff_t _srcstride, int height,            \
                                                     intptr_t mx, intptr_t my, int width)                        \
 {                                                                                                               \
@@ -54,12 +117,12 @@ void ff_hevc_put_hevc_uni_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dst
     for (i = 0; i < W; i += step) {                                                                             \
         src = _src + (i * ((bitd + 7) / 8));                                                                    \
         dst = _dst + (i * ((bitd + 7) / 8));                                                                    \
     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##_sse4(dst, dststride, src, _srcstride,                      \
+        ff_hevc_put_hevc_uni_##name##step##_##bitd##_##opt(dst, dststride, src, _srcstride,                     \
                                                           height, mx, my, width);                               \
     }                                                                                                           \
 }
                                                           height, mx, my, width);                               \
     }                                                                                                           \
 }
-#define mc_rep_bi_func(name, bitd, step, W) \
-void ff_hevc_put_hevc_bi_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dststride, uint8_t *_src,           \
+#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, ptrdiff_t _src2stride, \
                                                    int height, intptr_t mx, intptr_t my, int width)             \
 {                                                                                                               \
                                                    ptrdiff_t _srcstride, int16_t* _src2, ptrdiff_t _src2stride, \
                                                    int height, intptr_t mx, intptr_t my, int width)             \
 {                                                                                                               \
@@ -71,97 +134,99 @@ void ff_hevc_put_hevc_bi_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dsts
         src  = _src + (i * ((bitd + 7) / 8));                                                                   \
         dst  = _dst + (i * ((bitd + 7) / 8));                                                                   \
         src2 = _src2 + i;                                                                                       \
         src  = _src + (i * ((bitd + 7) / 8));                                                                   \
         dst  = _dst + (i * ((bitd + 7) / 8));                                                                   \
         src2 = _src2 + i;                                                                                       \
-        ff_hevc_put_hevc_bi_##name##step##_##bitd##_sse4(dst, dststride, src, _srcstride, src2,                 \
+        ff_hevc_put_hevc_bi_##name##step##_##bitd##_##opt(dst, dststride, src, _srcstride, src2,                \
                                                          _src2stride, height, mx, my, width);                   \
     }                                                                                                           \
 }
 
                                                          _src2stride, height, mx, my, width);                   \
     }                                                                                                           \
 }
 
-#define mc_rep_funcs(name, bitd, step, W)        \
-    mc_rep_func(name, bitd, step, W);            \
-    mc_rep_uni_func(name, bitd, step, W);        \
-    mc_rep_bi_func(name, bitd, step, W)
-
-
-mc_rep_funcs(pel_pixels, 8, 16, 64);
-mc_rep_funcs(pel_pixels, 8, 16, 48);
-mc_rep_funcs(pel_pixels, 8, 16, 32);
-mc_rep_funcs(pel_pixels, 8,  8, 24);
-
-mc_rep_funcs(pel_pixels,10,  8, 64);
-mc_rep_funcs(pel_pixels,10,  8, 48);
-mc_rep_funcs(pel_pixels,10,  8, 32);
-mc_rep_funcs(pel_pixels,10,  8, 24);
-mc_rep_funcs(pel_pixels,10,  8, 16);
-mc_rep_funcs(pel_pixels,10,  4, 12);
-
-mc_rep_funcs(epel_h, 8, 16, 64);
-mc_rep_funcs(epel_h, 8, 16, 48);
-mc_rep_funcs(epel_h, 8, 16, 32);
-mc_rep_funcs(epel_h, 8,  8, 24);
-mc_rep_funcs(epel_h,10,  8, 64);
-mc_rep_funcs(epel_h,10,  8, 48);
-mc_rep_funcs(epel_h,10,  8, 32);
-mc_rep_funcs(epel_h,10,  8, 24);
-mc_rep_funcs(epel_h,10,  8, 16);
-mc_rep_funcs(epel_h,10,  4, 12);
-mc_rep_funcs(epel_v, 8, 16, 64);
-mc_rep_funcs(epel_v, 8, 16, 48);
-mc_rep_funcs(epel_v, 8, 16, 32);
-mc_rep_funcs(epel_v, 8,  8, 24);
-mc_rep_funcs(epel_v,10,  8, 64);
-mc_rep_funcs(epel_v,10,  8, 48);
-mc_rep_funcs(epel_v,10,  8, 32);
-mc_rep_funcs(epel_v,10,  8, 24);
-mc_rep_funcs(epel_v,10,  8, 16);
-mc_rep_funcs(epel_v,10,  4, 12);
-mc_rep_funcs(epel_hv, 8,  8, 64);
-mc_rep_funcs(epel_hv, 8,  8, 48);
-mc_rep_funcs(epel_hv, 8,  8, 32);
-mc_rep_funcs(epel_hv, 8,  8, 24);
-mc_rep_funcs(epel_hv, 8,  8, 16);
-mc_rep_funcs(epel_hv, 8,  4, 12);
-mc_rep_funcs(epel_hv,10,  8, 64);
-mc_rep_funcs(epel_hv,10,  8, 48);
-mc_rep_funcs(epel_hv,10,  8, 32);
-mc_rep_funcs(epel_hv,10,  8, 24);
-mc_rep_funcs(epel_hv,10,  8, 16);
-mc_rep_funcs(epel_hv,10,  4, 12);
-
-mc_rep_funcs(qpel_h, 8, 16, 64);
-mc_rep_funcs(qpel_h, 8, 16, 48);
-mc_rep_funcs(qpel_h, 8, 16, 32);
-mc_rep_funcs(qpel_h, 8,  8, 24);
-mc_rep_funcs(qpel_h,10,  8, 64);
-mc_rep_funcs(qpel_h,10,  8, 48);
-mc_rep_funcs(qpel_h,10,  8, 32);
-mc_rep_funcs(qpel_h,10,  8, 24);
-mc_rep_funcs(qpel_h,10,  8, 16);
-mc_rep_funcs(qpel_h,10,  4, 12);
-mc_rep_funcs(qpel_v, 8, 16, 64);
-mc_rep_funcs(qpel_v, 8, 16, 48);
-mc_rep_funcs(qpel_v, 8, 16, 32);
-mc_rep_funcs(qpel_v, 8,  8, 24);
-mc_rep_funcs(qpel_v,10,  8, 64);
-mc_rep_funcs(qpel_v,10,  8, 48);
-mc_rep_funcs(qpel_v,10,  8, 32);
-mc_rep_funcs(qpel_v,10,  8, 24);
-mc_rep_funcs(qpel_v,10,  8, 16);
-mc_rep_funcs(qpel_v,10,  4, 12);
-mc_rep_funcs(qpel_hv, 8,  8, 64);
-mc_rep_funcs(qpel_hv, 8,  8, 48);
-mc_rep_funcs(qpel_hv, 8,  8, 32);
-mc_rep_funcs(qpel_hv, 8,  8, 24);
-mc_rep_funcs(qpel_hv, 8,  8, 16);
-mc_rep_funcs(qpel_hv, 8,  4, 12);
-mc_rep_funcs(qpel_hv,10,  8, 64);
-mc_rep_funcs(qpel_hv,10,  8, 48);
-mc_rep_funcs(qpel_hv,10,  8, 32);
-mc_rep_funcs(qpel_hv,10,  8, 24);
-mc_rep_funcs(qpel_hv,10,  8, 16);
-mc_rep_funcs(qpel_hv,10,  4, 12);
-
-#define mc_rep_uni_w(bitd, step, W) \
-void ff_hevc_put_hevc_uni_w##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dststride, int16_t *_src, ptrdiff_t _srcstride, \
+#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(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_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_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(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_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_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);
+
+#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;                                                                                                              \
                                                int height, int denom,  int _wx, int _ox)                                \
 {                                                                                                                       \
     int i;                                                                                                              \
@@ -170,27 +235,27 @@ void ff_hevc_put_hevc_uni_w##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dststrid
     for (i = 0; i < W; i += step) {                                                                                     \
         src= _src + i;                                                                                                  \
         dst= _dst + (i * ((bitd + 7) / 8));                                                                             \
     for (i = 0; i < W; i += step) {                                                                                     \
         src= _src + i;                                                                                                  \
         dst= _dst + (i * ((bitd + 7) / 8));                                                                             \
-        ff_hevc_put_hevc_uni_w##step##_##bitd##_sse4(dst, dststride, src, _srcstride,                                   \
+        ff_hevc_put_hevc_uni_w##step##_##bitd##_##opt(dst, dststride, src, _srcstride,                                  \
                                                      height, denom, _wx, _ox);                                          \
     }                                                                                                                   \
 }
 
                                                      height, denom, _wx, _ox);                                          \
     }                                                                                                                   \
 }
 
-mc_rep_uni_w(8, 6, 12);
-mc_rep_uni_w(8, 8, 16);
-mc_rep_uni_w(8, 8, 24);
-mc_rep_uni_w(8, 8, 32);
-mc_rep_uni_w(8, 8, 48);
-mc_rep_uni_w(8, 8, 64);
-
-mc_rep_uni_w(10, 6, 12);
-mc_rep_uni_w(10, 8, 16);
-mc_rep_uni_w(10, 8, 24);
-mc_rep_uni_w(10, 8, 32);
-mc_rep_uni_w(10, 8, 48);
-mc_rep_uni_w(10, 8, 64);
-
-#define mc_rep_bi_w(bitd, step, W) \
-void ff_hevc_put_hevc_bi_w##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dststride, int16_t *_src, ptrdiff_t _srcstride,  \
+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);
+
+#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, ptrdiff_t _src2stride, int height,                        \
                                               int denom,  int _wx0,  int _wx1, int _ox0, int _ox1)                      \
 {                                                                                                                       \
                                               int16_t *_src2, ptrdiff_t _src2stride, int height,                        \
                                               int denom,  int _wx0,  int _wx1, int _ox0, int _ox1)                      \
 {                                                                                                                       \
@@ -202,74 +267,74 @@ void ff_hevc_put_hevc_bi_w##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t dststride
         src  = _src  + i;                                                                                               \
         src2 = _src2 + i;                                                                                               \
         dst  = _dst  + (i * ((bitd + 7) / 8));                                                                          \
         src  = _src  + i;                                                                                               \
         src2 = _src2 + i;                                                                                               \
         dst  = _dst  + (i * ((bitd + 7) / 8));                                                                          \
-        ff_hevc_put_hevc_bi_w##step##_##bitd##_sse4(dst, dststride, src, _srcstride, src2, _src2stride,                 \
+        ff_hevc_put_hevc_bi_w##step##_##bitd##_##opt(dst, dststride, src, _srcstride, src2, _src2stride,                \
                                                     height, denom, _wx0, _wx1, _ox0, _ox1);                             \
     }                                                                                                                   \
 }
 
                                                     height, denom, _wx0, _wx1, _ox0, _ox1);                             \
     }                                                                                                                   \
 }
 
-mc_rep_bi_w(8, 6, 12);
-mc_rep_bi_w(8, 8, 16);
-mc_rep_bi_w(8, 8, 24);
-mc_rep_bi_w(8, 8, 32);
-mc_rep_bi_w(8, 8, 48);
-mc_rep_bi_w(8, 8, 64);
-
-mc_rep_bi_w(10, 6, 12);
-mc_rep_bi_w(10, 8, 16);
-mc_rep_bi_w(10, 8, 24);
-mc_rep_bi_w(10, 8, 32);
-mc_rep_bi_w(10, 8, 48);
-mc_rep_bi_w(10, 8, 64);
-
-#define mc_uni_w_func(name, bitd, W) \
-void ff_hevc_put_hevc_uni_w_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t _dststride,          \
+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);
+
+#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 * 64]);                                                     \
                                                       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 * 64]);                                                     \
-    ff_hevc_put_hevc_##name##W##_##bitd##_sse4(temp, 64, _src, _srcstride, height, mx, my, width);  \
-    ff_hevc_put_hevc_uni_w##W##_##bitd##_sse4(_dst, _dststride, temp, 64, height, denom, _wx, _ox); \
+    ff_hevc_put_hevc_##name##W##_##bitd##_##opt(temp, 64, _src, _srcstride, height, mx, my, width); \
+    ff_hevc_put_hevc_uni_w##W##_##bitd##_##opt(_dst, _dststride, temp, 64, height, denom, _wx, _ox);\
 }
 
 }
 
-#define mc_uni_w_funcs(name, bitd)       \
-        mc_uni_w_func(name, bitd, 4);    \
-        mc_uni_w_func(name, bitd, 8);    \
-        mc_uni_w_func(name, bitd, 12);   \
-        mc_uni_w_func(name, bitd, 16);   \
-        mc_uni_w_func(name, bitd, 24);   \
-        mc_uni_w_func(name, bitd, 32);   \
-        mc_uni_w_func(name, bitd, 48);   \
-        mc_uni_w_func(name, bitd, 64)
-
-mc_uni_w_funcs(pel_pixels, 8);
-mc_uni_w_func(pel_pixels, 8, 6);
-mc_uni_w_funcs(epel_h, 8);
-mc_uni_w_func(epel_h, 8, 6);
-mc_uni_w_funcs(epel_v, 8);
-mc_uni_w_func(epel_v, 8, 6);
-mc_uni_w_funcs(epel_hv, 8);
-mc_uni_w_func(epel_hv, 8, 6);
-mc_uni_w_funcs(qpel_h, 8);
-mc_uni_w_funcs(qpel_v, 8);
-mc_uni_w_funcs(qpel_hv, 8);
-
-mc_uni_w_funcs(pel_pixels, 10);
-mc_uni_w_func(pel_pixels, 10, 6);
-mc_uni_w_funcs(epel_h, 10);
-mc_uni_w_func(epel_h, 10, 6);
-mc_uni_w_funcs(epel_v, 10);
-mc_uni_w_func(epel_v, 10, 6);
-mc_uni_w_funcs(epel_hv, 10);
-mc_uni_w_func(epel_hv, 10, 6);
-mc_uni_w_funcs(qpel_h, 10);
-mc_uni_w_funcs(qpel_v, 10);
-mc_uni_w_funcs(qpel_hv, 10);
-
-
-#define mc_bi_w_func(name, bitd, W) \
-void ff_hevc_put_hevc_bi_w_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t _dststride,            \
+#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);
+
+
+#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, ptrdiff_t _src2stride,          \
                                                      int height, int denom,                          \
                                                      uint8_t *_src, ptrdiff_t _srcstride,            \
                                                      int16_t *_src2, ptrdiff_t _src2stride,          \
                                                      int height, int denom,                          \
@@ -277,65 +342,67 @@ void ff_hevc_put_hevc_bi_w_##name##W##_##bitd##_sse4(uint8_t *_dst, ptrdiff_t _d
                                                      intptr_t mx, intptr_t my, int width)            \
 {                                                                                                    \
     LOCAL_ALIGNED_16(int16_t, temp, [71 * 64]);                                                      \
                                                      intptr_t mx, intptr_t my, int width)            \
 {                                                                                                    \
     LOCAL_ALIGNED_16(int16_t, temp, [71 * 64]);                                                      \
-    ff_hevc_put_hevc_##name##W##_##bitd##_sse4(temp, 64, _src, _srcstride, height, mx, my, width);   \
-    ff_hevc_put_hevc_bi_w##W##_##bitd##_sse4(_dst, _dststride, temp, 64, _src2, _src2stride,         \
+    ff_hevc_put_hevc_##name##W##_##bitd##_##opt(temp, 64, _src, _srcstride, height, mx, my, width);  \
+    ff_hevc_put_hevc_bi_w##W##_##bitd##_##opt(_dst, _dststride, temp, 64, _src2, _src2stride,        \
                                              height, denom, _wx0, _wx1, _ox0, _ox1);                 \
 }
 
                                              height, denom, _wx0, _wx1, _ox0, _ox1);                 \
 }
 
-#define mc_bi_w_funcs(name, bitd)       \
-        mc_bi_w_func(name, bitd, 4);    \
-        mc_bi_w_func(name, bitd, 8);    \
-        mc_bi_w_func(name, bitd, 12);   \
-        mc_bi_w_func(name, bitd, 16);   \
-        mc_bi_w_func(name, bitd, 24);   \
-        mc_bi_w_func(name, bitd, 32);   \
-        mc_bi_w_func(name, bitd, 48);   \
-        mc_bi_w_func(name, bitd, 64)
-
-mc_bi_w_funcs(pel_pixels, 8);
-mc_bi_w_func(pel_pixels, 8, 6);
-mc_bi_w_funcs(epel_h, 8);
-mc_bi_w_func(epel_h, 8, 6);
-mc_bi_w_funcs(epel_v, 8);
-mc_bi_w_func(epel_v, 8, 6);
-mc_bi_w_funcs(epel_hv, 8);
-mc_bi_w_func(epel_hv, 8, 6);
-mc_bi_w_funcs(qpel_h, 8);
-mc_bi_w_funcs(qpel_v, 8);
-mc_bi_w_funcs(qpel_hv, 8);
-
-mc_bi_w_funcs(pel_pixels, 10);
-mc_bi_w_func(pel_pixels, 10, 6);
-mc_bi_w_funcs(epel_h, 10);
-mc_bi_w_func(epel_h, 10, 6);
-mc_bi_w_funcs(epel_v, 10);
-mc_bi_w_func(epel_v, 10, 6);
-mc_bi_w_funcs(epel_hv, 10);
-mc_bi_w_func(epel_hv, 10, 6);
-mc_bi_w_funcs(qpel_h, 10);
-mc_bi_w_funcs(qpel_v, 10);
-mc_bi_w_funcs(qpel_hv, 10);
-
-
-#define EPEL_LINKS(pointer, my, mx, fname, bitd)           \
-        PEL_LINK(pointer, 1, my , mx , fname##4 ,  bitd ); \
-        PEL_LINK(pointer, 2, my , mx , fname##6 ,  bitd ); \
-        PEL_LINK(pointer, 3, my , mx , fname##8 ,  bitd ); \
-        PEL_LINK(pointer, 4, my , mx , fname##12,  bitd ); \
-        PEL_LINK(pointer, 5, my , mx , fname##16,  bitd ); \
-        PEL_LINK(pointer, 6, my , mx , fname##24,  bitd ); \
-        PEL_LINK(pointer, 7, my , mx , fname##32,  bitd ); \
-        PEL_LINK(pointer, 8, my , mx , fname##48,  bitd ); \
-        PEL_LINK(pointer, 9, my , mx , fname##64,  bitd )
-#define QPEL_LINKS(pointer, my, mx, fname, bitd)           \
-        PEL_LINK(pointer, 1, my , mx , fname##4 ,  bitd ); \
-        PEL_LINK(pointer, 3, my , mx , fname##8 ,  bitd ); \
-        PEL_LINK(pointer, 4, my , mx , fname##12,  bitd ); \
-        PEL_LINK(pointer, 5, my , mx , fname##16,  bitd ); \
-        PEL_LINK(pointer, 6, my , mx , fname##24,  bitd ); \
-        PEL_LINK(pointer, 7, my , mx , fname##32,  bitd ); \
-        PEL_LINK(pointer, 8, my , mx , fname##48,  bitd ); \
-        PEL_LINK(pointer, 9, my , mx , fname##64,  bitd )
+#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);
+
+#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_hevcdsp_init_x86(HEVCDSPContext *c, const int bit_depth)
 
 
 void ff_hevcdsp_init_x86(HEVCDSPContext *c, const int bit_depth)
@@ -343,43 +410,69 @@ void ff_hevcdsp_init_x86(HEVCDSPContext *c, const int bit_depth)
     int mm_flags = av_get_cpu_flags();
 
     if (bit_depth == 8) {
     int mm_flags = av_get_cpu_flags();
 
     if (bit_depth == 8) {
-        if (EXTERNAL_MMX(mm_flags)) {
+        if (EXTERNAL_MMXEXT(mm_flags)) {
+                c->transform_dc_add[0]    =  ff_hevc_idct4_dc_add_8_mmxext;
+                c->transform_dc_add[1]    =  ff_hevc_idct8_dc_add_8_mmxext;
 
 
-            if (EXTERNAL_MMXEXT(mm_flags)) {
-
-                if (EXTERNAL_SSSE3(mm_flags) && ARCH_X86_64) {
+        }
+        if (EXTERNAL_SSE2(mm_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;
 
 
-                    EPEL_LINKS(c->put_hevc_epel, 0, 0, pel_pixels,  8);
-                    EPEL_LINKS(c->put_hevc_epel, 0, 1, epel_h,      8);
-                    EPEL_LINKS(c->put_hevc_epel, 1, 0, epel_v,      8);
-                    EPEL_LINKS(c->put_hevc_epel, 1, 1, epel_hv,     8);
+                    c->transform_dc_add[2]    =  ff_hevc_idct16_dc_add_8_sse2;
+                    c->transform_dc_add[3]    =  ff_hevc_idct32_dc_add_8_sse2;
+        }
+        if (EXTERNAL_SSSE3(mm_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(mm_flags) && ARCH_X86_64) {
 
 
-                    QPEL_LINKS(c->put_hevc_qpel, 0, 0, pel_pixels, 8);
-                    QPEL_LINKS(c->put_hevc_qpel, 0, 1, qpel_h,     8);
-                    QPEL_LINKS(c->put_hevc_qpel, 1, 0, qpel_v,     8);
-                    QPEL_LINKS(c->put_hevc_qpel, 1, 1, qpel_hv,    8);
+            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);
         }
     } else if (bit_depth == 10) {
         }
     } else if (bit_depth == 10) {
-        if (EXTERNAL_MMX(mm_flags)) {
-            if (EXTERNAL_MMXEXT(mm_flags) && ARCH_X86_64) {
+        if (EXTERNAL_MMXEXT(mm_flags)) {
+                c->transform_dc_add[0]    =  ff_hevc_idct4_dc_add_10_mmxext;
 
 
-                if (EXTERNAL_SSSE3(mm_flags)) {
+        }
+        if (EXTERNAL_SSE2(mm_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;
 
 
-                    EPEL_LINKS(c->put_hevc_epel, 0, 0, pel_pixels, 10);
-                    EPEL_LINKS(c->put_hevc_epel, 0, 1, epel_h,     10);
-                    EPEL_LINKS(c->put_hevc_epel, 1, 0, epel_v,     10);
-                    EPEL_LINKS(c->put_hevc_epel, 1, 1, epel_hv,    10);
 
 
-                    QPEL_LINKS(c->put_hevc_qpel, 0, 0, pel_pixels, 10);
-                    QPEL_LINKS(c->put_hevc_qpel, 0, 1, qpel_h,     10);
-                    QPEL_LINKS(c->put_hevc_qpel, 1, 0, qpel_v,     10);
-                    QPEL_LINKS(c->put_hevc_qpel, 1, 1, qpel_hv,    10);
+                    c->transform_dc_add[1]    =  ff_hevc_idct8_dc_add_10_sse2;
+                    c->transform_dc_add[2]    =  ff_hevc_idct16_dc_add_10_sse2;
+                    c->transform_dc_add[3]    =  ff_hevc_idct32_dc_add_10_sse2;
                 }
                 }
+        if (EXTERNAL_SSSE3(mm_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(mm_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(mm_flags)) {
+            c->transform_dc_add[1]    =  ff_hevc_idct8_dc_add_10_avx;
+            c->transform_dc_add[2]    =  ff_hevc_idct16_dc_add_10_avx;
+            c->transform_dc_add[3]    =  ff_hevc_idct32_dc_add_10_avx;
         }
         }
+
     }
 }
     }
 }