/*
 * 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/attributes.h"
#include "libavutil/cpu.h"
#include "libavutil/mem_internal.h"
#include "libavutil/x86/asm.h"
#include "libavutil/x86/cpu.h"
#include "libavcodec/mpeg4videodsp.h"
#include "videodsp.h"

#if HAVE_SSSE3_INLINE

#define SPLATW(reg) "pshuflw  $0, %%" #reg ", %%" #reg "\n\t" \
                    "punpcklqdq   %%" #reg ", %%" #reg "\n\t"

typedef struct {
    DECLARE_ALIGNED_16(uint16_t, u16)[8];
} xmm_u16;

DECLARE_ASM_CONST(16, xmm_u16, pw_0to7) = { { 0, 1, 2, 3, 4, 5, 6, 7 } };

static void gmc_ssse3(uint8_t *dst, const uint8_t *src,
                      ptrdiff_t stride, int h, int ox, int oy,
                      int dxx, int dxy, int dyx, int dyy,
                      int shift, int r, int width, int height)
{
    enum {
        W               = 8,
        EDGE_EMU_STRIDE = 16, //< anything >= W+1 will do
        MAX_H           = 16,
    };
    const int w    = 8;
    const int ix   = ox  >> (16 + shift);
    const int iy   = oy  >> (16 + shift);
    const int ox2  = ox & (1 << (16 + shift)) - 1;
    const int oy2  = oy & (1 << (16 + shift)) - 1;
    const int oxs  = ox2 >> 4;
    const int oys  = oy2 >> 4;
    const int dxx2 = dxx - (1 << (16 + shift));
    const int dyy2 = dyy - (1 << (16 + shift));
    const int dxxs = dxx2 >> 4;
    const int dxys = dxy >> 4;
    const int dyxs = dyx >> 4;
    const int dyys = dyy2 >> 4;

    const int dxw = dxx2 * (w - 1);
    const int dyh = dyy2 * (h - 1);
    const int dxh = dxy * (h - 1);
    const int dyw = dyx * (w - 1);
    int need_emu  =  (unsigned) ix >= width  - w || width < w ||
                     (unsigned) iy >= height - h || height< h
                     ;

    if ( // non-constant fullpel offset (3% of blocks)
        ((ox2 + dxw) | (ox2 + dxh) | (ox2 + dxw + dxh) |
         (oy2 + dyw) | (oy2 + dyh) | (oy2 + dyw + dyh)) >> (16 + shift) ||
        // uses more than 16 bits of subpel mv (only at huge resolution)
        (dxx | dxy | dyx | dyy) & 15 ||
        (!HAVE_SSE2_EXTERNAL && need_emu)) {
        ff_gmc_c(dst, src, stride, h, ox, oy, dxx, dxy, dyx, dyy,
                 shift, r, width, height);
        return;
    }

    src += ix + iy * stride;
    const ptrdiff_t dst_stride = stride;
    ptrdiff_t src_stride = stride;
#if HAVE_SSE2_EXTERNAL
    uint8_t edge_buf[(MAX_H + 1) * EDGE_EMU_STRIDE];
    if (need_emu) {
        ff_emulated_edge_mc_sse2(edge_buf, src, EDGE_EMU_STRIDE, src_stride,
                                 w + 1, h + 1, ix, iy, width, height);
        src        = edge_buf;
        src_stride = EDGE_EMU_STRIDE;
    }
#endif

#if ARCH_X86_32
    xmm_u16 dxy8, dyy8, r8;
    DECLARE_ALIGNED_16(uint64_t, shift2) = 2 * shift;
#endif

    __asm__ volatile (
        "movd             %[dxxs], %%xmm2     \n\t"
        "movd             %[dyxs], %%xmm3     \n\t"
        "movd              %[oxs], %%xmm1     \n\t"
        SPLATW(xmm2)
        "movd              %[oys], %%xmm7     \n\t"
        SPLATW(xmm3)
        "pmullw "MANGLE(pw_0to7)", %%xmm2     \n\t"
        SPLATW(xmm1)
        "movd                %[s], %%xmm6     \n\t"
        "pmullw "MANGLE(pw_0to7)", %%xmm3     \n\t"
        "movq            (%[src]), %%xmm5     \n\t"
        SPLATW(xmm7)
#if ARCH_X86_32
        "movd             %[dxys], %%xmm0     \n\t"
#else
        "movd             %[dxys], %%xmm11    \n\t"
#endif
        "paddw             %%xmm2, %%xmm1     \n\t"
        "movq           1(%[src]), %%xmm2     \n\t"
        SPLATW(xmm6)
#if ARCH_X86_32
        "movd             %[dyys], %%xmm4     \n\t"
#else
        "movd             %[dyys], %%xmm9     \n\t"
#endif
        "paddw             %%xmm3, %%xmm7     \n\t"
        "punpcklbw         %%xmm2, %%xmm5     \n\t"
#if ARCH_X86_32
        SPLATW(xmm0)
        "movd                %[r], %%xmm2     \n\t"
        SPLATW(xmm4)
        "movdqa            %%xmm0, %[dxy8]    \n\t"
        SPLATW(xmm2)
        "movdqa            %%xmm4, %[dyy8]    \n\t"
        "movdqa            %%xmm2, %[r8]      \n\t"
#else
        SPLATW(xmm11)
        "movd                %[r], %%xmm8     \n\t"
        SPLATW(xmm9)
        SPLATW(xmm8)
        "movd           %[shift2], %%xmm12    \n\t"
#endif

        "1:                                   \n\t"
        "add        %[src_stride], %[src]     \n\t"
        "movq            (%[src]), %%xmm3     \n\t"
        "movq           1(%[src]), %%xmm0     \n\t"
        "movdqa            %%xmm1, %%xmm4     \n\t"
        "psrlw                $12, %%xmm4     \n\t" // dx
        "movdqa            %%xmm6, %%xmm2     \n\t"
        "psubw             %%xmm4, %%xmm2     \n\t" // (s-dx)
        "psllw                 $8, %%xmm4     \n\t"
        "por               %%xmm4, %%xmm2     \n\t" // s-dx,dx,s-dx,dx (bytes)
        "pmaddubsw         %%xmm2, %%xmm5     \n\t" // src[0, 0] * (s - dx) + src[1,0] * dx
        "punpcklbw         %%xmm0, %%xmm3     \n\t"
        "movdqa            %%xmm3, %%xmm0     \n\t"
        "pmaddubsw         %%xmm2, %%xmm3     \n\t" // src[0, 1] * (s - dx) + src[1,1] * dx
#if ARCH_X86_32
        "paddw            %[dxy8], %%xmm1     \n\t"
#else
        "paddw            %%xmm11, %%xmm1     \n\t"
#endif
        "movdqa            %%xmm7, %%xmm4     \n\t"
        "movdqa            %%xmm6, %%xmm2     \n\t"
        "psrlw                $12, %%xmm4     \n\t" // dy
        "psubw             %%xmm4, %%xmm2     \n\t" // (s-dy)
        "pmullw            %%xmm5, %%xmm2     \n\t" // (src[0, 0] * (s - dx) + src[1,0] * dx) * (s - dy)
#if ARCH_X86_32
        "paddw            %[dyy8], %%xmm7     \n\t"
#else
        "paddw             %%xmm9, %%xmm7     \n\t"
#endif
        "pmullw            %%xmm3, %%xmm4     \n\t" // (src[0, 1] * (s - dx) + src[1,1] * dx) * dy

#if ARCH_X86_32
        "paddw              %[r8], %%xmm2     \n\t"
#else
        "paddw             %%xmm8, %%xmm2     \n\t"
#endif
        "paddw             %%xmm2, %%xmm4     \n\t"

#if ARCH_X86_32
        "psrlw          %[shift2], %%xmm4     \n\t"
#else
        "psrlw            %%xmm12, %%xmm4     \n\t"
#endif
        "packuswb          %%xmm4, %%xmm4     \n\t"
        "movq              %%xmm4, (%[dst])   \n\t"
        "movdqa            %%xmm0, %%xmm5     \n\t"
        "add        %[dst_stride], %[dst]     \n\t"

        "decl                %[h]             \n\t"
        "jnz                   1b             \n\t"
        : [dst]"+r"(dst), [src]"+r"(src),
#if HAVE_X86_6REGS || HAVE_INLINE_ASM_DIRECT_SYMBOL_REFS
        [h]"+r"(h)
#else
        [h]"+m"(h)
#endif
#if ARCH_X86_32
          , [dxy8]"=m" (dxy8), [dyy8]"=m" (dyy8), [r8]"=m" (r8)
#endif
        : [dst_stride]"r"(dst_stride), [src_stride]"r"(src_stride),
          [s]"g" (1 << shift),
#if ARCH_X86_32
          [shift2]"m" (shift2),
#else
          [shift2]"g" (2*shift),
#endif
          [oxs]"g"(oxs),  [oys]"g"(oys),  [dxxs]"g"(dxxs), [dyxs]"g"(dyxs),
          [dxys]"g"(dxys), [dyys]"g"(dyys), [r]"g"(r) NAMED_CONSTRAINTS_ADD(pw_0to7)
        : XMM_CLOBBERS("xmm0", "xmm1", "xmm2", "xmm3", "xmm4", "xmm5", "xmm6", "xmm7",)
#if ARCH_X86_64
          XMM_CLOBBERS("xmm8", "xmm9", "xmm10", "xmm11", "xmm12",)
#endif
         "memory");
}

#endif /* HAVE_SSSE3_INLINE */

av_cold void ff_mpeg4videodsp_init_x86(Mpeg4VideoDSPContext *c)
{
#if HAVE_SSSE3_INLINE
    int cpu_flags = av_get_cpu_flags();

    if (INLINE_SSSE3(cpu_flags))
        c->gmc = gmc_ssse3;
#endif /* HAVE_SSSE3_INLINE */
}
