HEVC : added assembly MC functions

pretty print x86

Reviewed-by: "Ronald S. Bultje" <rsbultje@gmail.com>
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>

1 files changed, 385 insertions, 0 deletions

diff --git a/libavcodec/x86/hevcdsp_init.c b/libavcodec/x86/hevcdsp_init.c
new file mode 100644
index 0000000000..565c4a0a6e
--- /dev/null
+++ b/libavcodec/x86/hevcdsp_init.c
@@ -0,0 +1,385 @@
+/*
+ * 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 mc_rep_func(name, bitd, step, W) \
+void ff_hevc_put_hevc_##name##W##_##bitd##_sse4(int16_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;                                                                                               \
+    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##_sse4(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,                         \
+                                                    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##_sse4(dst, dststride, src, _srcstride,                      \
+                                                          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,           \
+                                                   ptrdiff_t _srcstride, int16_t* _src2, ptrdiff_t _src2stride, \
+                                                   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##_sse4(dst, dststride, src, _srcstride, src2,                 \
+                                                         _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, \
+                                               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##_sse4(dst, dststride, src, _srcstride,                                   \
+                                                     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,  \
+                                              int16_t *_src2, ptrdiff_t _src2stride, 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##_sse4(dst, dststride, src, _srcstride, src2, _src2stride,                 \
+                                                    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,          \
+                                                      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); \
+}
+
+#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,            \
+                                                     uint8_t *_src, ptrdiff_t _srcstride,            \
+                                                     int16_t *_src2, ptrdiff_t _src2stride,          \
+                                                     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 * 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,         \
+                                             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 )
+
+
+void ff_hevcdsp_init_x86(HEVCDSPContext *c, const int bit_depth)
+{
+    int mm_flags = av_get_cpu_flags();
+
+    if (bit_depth == 8) {
+        if (EXTERNAL_MMX(mm_flags)) {
+
+            if (EXTERNAL_MMXEXT(mm_flags)) {
+
+                if (EXTERNAL_SSSE3(mm_flags) && ARCH_X86_64) {
+
+                    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);
+
+                    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);
+
+                }
+            }
+        }
+    } else if (bit_depth == 10) {
+        if (EXTERNAL_MMX(mm_flags)) {
+            if (EXTERNAL_MMXEXT(mm_flags) && ARCH_X86_64) {
+
+                if (EXTERNAL_SSSE3(mm_flags)) {
+
+                    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);
+                }
+
+            }
+        }
+    }
+}