/* FreeRDP: A Remote Desktop Protocol Client

 * Shift operations.

 * vi:ts=4 sw=4:

 *

 * (c) Copyright 2012 Hewlett-Packard Development Company, L.P.

 * Licensed under the Apache License, Version 2.0 (the "License"); you may

 * not use this file except in compliance with the License. You may obtain

 * a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express

 * or implied. See the License for the specific language governing

 * permissions and limitations under the License.

 */

#include <freerdp/config.h>

#include <freerdp/types.h>

#include <freerdp/primitives.h>

#include <winpr/sysinfo.h>

#include "prim_shift.h"

#include "prim_internal.h"

#include "prim_templates.h"

#if defined(SSE_AVX_INTRINSICS_ENABLED)

#include <emmintrin.h>

#include <pmmintrin.h>

static primitives_t* generic = NULL;

/* ------------------------------------------------------------------------- */

SSE3_SCD_ROUTINE(sse2_lShiftC_16s, INT16, generic->lShiftC_16s, _mm_slli_epi16, int16_t,

                 *dptr++ = (INT16)(((UINT16)*sptr++ << val) & 0xFFFF))

/* ------------------------------------------------------------------------- */

SSE3_SCD_ROUTINE(sse2_rShiftC_16s, INT16, generic->rShiftC_16s, _mm_srai_epi16, int16_t,

                 *dptr++ = *sptr++ >> val)

/* ------------------------------------------------------------------------- */

SSE3_SCD_ROUTINE(sse2_lShiftC_16u, UINT16, generic->lShiftC_16u, _mm_slli_epi16, int16_t,

                 *dptr++ = (((UINT16)*sptr++ << val) & 0xFFFF))

/* ------------------------------------------------------------------------- */

SSE3_SCD_ROUTINE(sse2_rShiftC_16u, UINT16, generic->rShiftC_16u, _mm_srli_epi16, int16_t,

                 *dptr++ = *sptr++ >> val)

static pstatus_t sse2_lShiftC_16s_inplace(INT16* WINPR_RESTRICT pSrcDst, UINT32 val, UINT32 ulen)

{

  size_t len = ulen;

  const INT32 shifts = 2;

  if (val == 0)

    return PRIMITIVES_SUCCESS;

  if (val >= 16)

    return -1;

  if (len < 16) /* pointless if too small */

    return generic->lShiftC_16s_inplace(pSrcDst, val, ulen);

  UINT32 offBeatMask = (1 << (shifts - 1)) - 1;

  if ((ULONG_PTR)pSrcDst & offBeatMask)

  {

    /* Incrementing the pointer skips over 16-byte boundary. */

    return generic->lShiftC_16s_inplace(pSrcDst, val, ulen);

  }

  /* Get to the 16-byte boundary now. */

  const UINT32 rem = ((UINT_PTR)pSrcDst & 0x0f) / sizeof(INT16);

  if (rem > 0)

  {

    const UINT32 add = 16 - rem;

    pstatus_t status = generic->lShiftC_16s_inplace(pSrcDst, val, add);

    if (status != PRIMITIVES_SUCCESS)

      return status;

    pSrcDst += add;

    len -= add;

  }

  /* Use 8 128-bit SSE registers. */

  size_t count = len >> (8 - shifts);

  len -= count << (8 - shifts);

  while (count--)

  {

    const __m128i* src = (const __m128i*)pSrcDst;

    __m128i xmm0 = LOAD_SI128(src++);

    __m128i xmm1 = LOAD_SI128(src++);

    __m128i xmm2 = LOAD_SI128(src++);

    __m128i xmm3 = LOAD_SI128(src++);

    __m128i xmm4 = LOAD_SI128(src++);

    __m128i xmm5 = LOAD_SI128(src++);

    __m128i xmm6 = LOAD_SI128(src++);

    __m128i xmm7 = LOAD_SI128(src);

    xmm0 = _mm_slli_epi16(xmm0, (int16_t)val);

    xmm1 = _mm_slli_epi16(xmm1, (int16_t)val);

    xmm2 = _mm_slli_epi16(xmm2, (int16_t)val);

    xmm3 = _mm_slli_epi16(xmm3, (int16_t)val);

    xmm4 = _mm_slli_epi16(xmm4, (int16_t)val);

    xmm5 = _mm_slli_epi16(xmm5, (int16_t)val);

    xmm6 = _mm_slli_epi16(xmm6, (int16_t)val);

    xmm7 = _mm_slli_epi16(xmm7, (int16_t)val);

    __m128i* dst = (__m128i*)pSrcDst;

    STORE_SI128(dst++, xmm0);

    STORE_SI128(dst++, xmm1);

    STORE_SI128(dst++, xmm2);

    STORE_SI128(dst++, xmm3);

    STORE_SI128(dst++, xmm4);

    STORE_SI128(dst++, xmm5);

    STORE_SI128(dst++, xmm6);

    STORE_SI128(dst++, xmm7);

    pSrcDst = (INT16*)dst;

  }

  /* Use a single 128-bit SSE register. */

  count = len >> (5 - shifts);

  len -= count << (5 - shifts);

  while (count--)

  {

    const __m128i* src = (const __m128i*)pSrcDst;

    __m128i xmm0 = LOAD_SI128(src);

    xmm0 = _mm_slli_epi16(xmm0, (int16_t)val);

    __m128i* dst = (__m128i*)pSrcDst;

    STORE_SI128(dst++, xmm0);

    pSrcDst = (INT16*)dst;

  }

  /* Finish off the remainder. */

  if (len > 0)

    return generic->lShiftC_16s_inplace(pSrcDst, val, WINPR_ASSERTING_INT_CAST(uint32_t, len));

  return PRIMITIVES_SUCCESS;

}

#endif

/* Note: the IPP version will have to call ippLShiftC_16s or ippRShiftC_16s

 * depending on the sign of val.  To avoid using the deprecated inplace

 * routines, a wrapper can use the src for the dest.

 */

/* ------------------------------------------------------------------------- */

void primitives_init_shift_sse3_int(primitives_t* WINPR_RESTRICT prims)

{

#if defined(SSE_AVX_INTRINSICS_ENABLED)

  generic = primitives_get_generic();

  WLog_VRB(PRIM_TAG, "SSE2/SSE3 optimizations");

  prims->lShiftC_16s_inplace = sse2_lShiftC_16s_inplace;

  prims->lShiftC_16s = sse2_lShiftC_16s;

  prims->rShiftC_16s = sse2_rShiftC_16s;

  prims->lShiftC_16u = sse2_lShiftC_16u;

  prims->rShiftC_16u = sse2_rShiftC_16u;

#else

  WLog_VRB(PRIM_TAG, "undefined WITH_SIMD or SSE3 intrinsics not available");

  WINPR_UNUSED(prims);

#endif

}