/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * Optimized YUV/RGB conversion operations

 *

 * Copyright 2014 Thomas Erbesdobler

 * Copyright 2016-2017 Armin Novak <armin.novak@thincast.com>

 * Copyright 2016-2017 Norbert Federa <norbert.federa@thincast.com>

 * Copyright 2016-2017 Thincast Technologies GmbH

 *

 * 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 <winpr/sysinfo.h>

#include <winpr/crt.h>

#include <freerdp/types.h>

#include <freerdp/primitives.h>

#include "prim_internal.h"

#include "prim_YUV.h"

#if defined(NEON_INTRINSICS_ENABLED)

#include <arm_neon.h>

static primitives_t* generic = NULL;

static INLINE uint8x8_t neon_YUV2R_single(uint16x8_t C, int16x8_t D, int16x8_t E)

{

  /* R = (256 * Y + 403 * (V - 128)) >> 8 */

  const int32x4_t Ch = vreinterpretq_s32_u32(vmovl_u16(vget_high_u16(C)));

  const int32x4_t e403h = vmull_n_s16(vget_high_s16(E), 403);

  const int32x4_t cehm = vaddq_s32(Ch, e403h);

  const int32x4_t ceh = vshrq_n_s32(cehm, 8);

  const int32x4_t Cl = vreinterpretq_s32_u32(vmovl_u16(vget_low_u16(C)));

  const int32x4_t e403l = vmull_n_s16(vget_low_s16(E), 403);

  const int32x4_t celm = vaddq_s32(Cl, e403l);

  const int32x4_t cel = vshrq_n_s32(celm, 8);

  const int16x8_t ce = vcombine_s16(vqmovn_s32(cel), vqmovn_s32(ceh));

  return vqmovun_s16(ce);

}

static INLINE uint8x8x2_t neon_YUV2R(uint16x8x2_t C, int16x8x2_t D, int16x8x2_t E)

{

  uint8x8x2_t res = { { neon_YUV2R_single(C.val[0], D.val[0], E.val[0]),

                      neon_YUV2R_single(C.val[1], D.val[1], E.val[1]) } };

  return res;

}

static INLINE uint8x8_t neon_YUV2G_single(uint16x8_t C, int16x8_t D, int16x8_t E)

{

  /* G = (256L * Y -  48 * (U - 128) - 120 * (V - 128)) >> 8 */

  const int16x8_t d48 = vmulq_n_s16(D, 48);

  const int16x8_t e120 = vmulq_n_s16(E, 120);

  const int32x4_t deh = vaddl_s16(vget_high_s16(d48), vget_high_s16(e120));

  const int32x4_t Ch = vreinterpretq_s32_u32(vmovl_u16(vget_high_u16(C)));

  const int32x4_t cdeh32m = vsubq_s32(Ch, deh);

  const int32x4_t cdeh32 = vshrq_n_s32(cdeh32m, 8);

  const int16x4_t cdeh = vqmovn_s32(cdeh32);

  const int32x4_t del = vaddl_s16(vget_low_s16(d48), vget_low_s16(e120));

  const int32x4_t Cl = vreinterpretq_s32_u32(vmovl_u16(vget_low_u16(C)));

  const int32x4_t cdel32m = vsubq_s32(Cl, del);

  const int32x4_t cdel32 = vshrq_n_s32(cdel32m, 8);

  const int16x4_t cdel = vqmovn_s32(cdel32);

  const int16x8_t cde = vcombine_s16(cdel, cdeh);

  return vqmovun_s16(cde);

}

static INLINE uint8x8x2_t neon_YUV2G(uint16x8x2_t C, int16x8x2_t D, int16x8x2_t E)

{

  uint8x8x2_t res = { { neon_YUV2G_single(C.val[0], D.val[0], E.val[0]),

                      neon_YUV2G_single(C.val[1], D.val[1], E.val[1]) } };

  return res;

}

static INLINE uint8x8_t neon_YUV2B_single(uint16x8_t C, int16x8_t D, int16x8_t E)

{

  /* B = (256L * Y + 475 * (U - 128)) >> 8*/

  const int32x4_t Ch = vreinterpretq_s32_u32(vmovl_u16(vget_high_u16(C)));

  const int32x4_t d475h = vmull_n_s16(vget_high_s16(D), 475);

  const int32x4_t cdhm = vaddq_s32(Ch, d475h);

  const int32x4_t cdh = vshrq_n_s32(cdhm, 8);

  const int32x4_t Cl = vreinterpretq_s32_u32(vmovl_u16(vget_low_u16(C)));

  const int32x4_t d475l = vmull_n_s16(vget_low_s16(D), 475);

  const int32x4_t cdlm = vaddq_s32(Cl, d475l);

  const int32x4_t cdl = vshrq_n_s32(cdlm, 8);

  const int16x8_t cd = vcombine_s16(vqmovn_s32(cdl), vqmovn_s32(cdh));

  return vqmovun_s16(cd);

}

static INLINE uint8x8x2_t neon_YUV2B(uint16x8x2_t C, int16x8x2_t D, int16x8x2_t E)

{

  uint8x8x2_t res = { { neon_YUV2B_single(C.val[0], D.val[0], E.val[0]),

                      neon_YUV2B_single(C.val[1], D.val[1], E.val[1]) } };

  return res;

}

static inline void neon_store_bgrx(BYTE* WINPR_RESTRICT pRGB, uint8x8_t r, uint8x8_t g, uint8x8_t b,

                                   uint8_t rPos, uint8_t gPos, uint8_t bPos, uint8_t aPos)

{

  uint8x8x4_t bgrx = vld4_u8(pRGB);

  bgrx.val[rPos] = r;

  bgrx.val[gPos] = g;

  bgrx.val[bPos] = b;

  vst4_u8(pRGB, bgrx);

}

static INLINE void neon_YuvToRgbPixel(BYTE* pRGB, uint8x8x2_t Y, int16x8x2_t D, int16x8x2_t E,

                                      const uint8_t rPos, const uint8_t gPos, const uint8_t bPos,

                                      const uint8_t aPos)

{

  /* Y * 256 == Y << 8  */

  const uint16x8x2_t C = { { vshlq_n_u16(vmovl_u8(Y.val[0]), 8),

                           vshlq_n_u16(vmovl_u8(Y.val[1]), 8) } };

  const uint8x8x2_t r = neon_YUV2R(C, D, E);

  const uint8x8x2_t g = neon_YUV2G(C, D, E);

  const uint8x8x2_t b = neon_YUV2B(C, D, E);

  neon_store_bgrx(pRGB, r.val[0], g.val[0], b.val[0], rPos, gPos, bPos, aPos);

  neon_store_bgrx(pRGB + sizeof(uint8x8x4_t), r.val[1], g.val[1], b.val[1], rPos, gPos, bPos,

                  aPos);

}

static inline int16x8x2_t loadUV(const BYTE* WINPR_RESTRICT pV, size_t x)

{

  const uint8x8_t Vraw = vld1_u8(&pV[x / 2]);

  const int16x8_t V = vreinterpretq_s16_u16(vmovl_u8(Vraw));

  const int16x8_t c128 = vdupq_n_s16(128);

  const int16x8_t E = vsubq_s16(V, c128);

  return vzipq_s16(E, E);

}

static INLINE void neon_write_pixel(BYTE* pRGB, BYTE Y, BYTE U, BYTE V, const uint8_t rPos,

                                    const uint8_t gPos, const uint8_t bPos, const uint8_t aPos)

{

  const BYTE r = YUV2R(Y, U, V);

  const BYTE g = YUV2G(Y, U, V);

  const BYTE b = YUV2B(Y, U, V);

  pRGB[rPos] = r;

  pRGB[gPos] = g;

  pRGB[bPos] = b;

}

static INLINE void neon_YUV420ToX_DOUBLE_ROW(const BYTE* WINPR_RESTRICT pY[2],

                                             const BYTE* WINPR_RESTRICT pU,

                                             const BYTE* WINPR_RESTRICT pV,

                                             BYTE* WINPR_RESTRICT pRGB[2], size_t width,

                                             const uint8_t rPos, const uint8_t gPos,

                                             const uint8_t bPos, const uint8_t aPos)

{

  UINT32 x = 0;

  for (; x < width - width % 16; x += 16)

  {

    const uint8x16_t Y0raw = vld1q_u8(&pY[0][x]);

    const uint8x8x2_t Y0 = { { vget_low_u8(Y0raw), vget_high_u8(Y0raw) } };

    const int16x8x2_t D = loadUV(pU, x);

    const int16x8x2_t E = loadUV(pV, x);

    neon_YuvToRgbPixel(&pRGB[0][4ULL * x], Y0, D, E, rPos, gPos, bPos, aPos);

    const uint8x16_t Y1raw = vld1q_u8(&pY[1][x]);

    const uint8x8x2_t Y1 = { { vget_low_u8(Y1raw), vget_high_u8(Y1raw) } };

    neon_YuvToRgbPixel(&pRGB[1][4ULL * x], Y1, D, E, rPos, gPos, bPos, aPos);

  }

  for (; x < width - width % 2; x += 2)

  {

    const BYTE U = pU[x / 2];

    const BYTE V = pV[x / 2];

    neon_write_pixel(&pRGB[0][4 * x], pY[0][x], U, V, rPos, gPos, bPos, aPos);

    neon_write_pixel(&pRGB[0][4 * (1ULL + x)], pY[0][1ULL + x], U, V, rPos, gPos, bPos, aPos);

    neon_write_pixel(&pRGB[1][4 * x], pY[1][x], U, V, rPos, gPos, bPos, aPos);

    neon_write_pixel(&pRGB[1][4 * (1ULL + x)], pY[1][1ULL + x], U, V, rPos, gPos, bPos, aPos);

  }

  for (; x < width; x++)

  {

    const BYTE U = pU[x / 2];

    const BYTE V = pV[x / 2];

    neon_write_pixel(&pRGB[0][4 * x], pY[0][x], U, V, rPos, gPos, bPos, aPos);

    neon_write_pixel(&pRGB[1][4 * x], pY[1][x], U, V, rPos, gPos, bPos, aPos);

  }

}

static INLINE void neon_YUV420ToX_SINGLE_ROW(const BYTE* WINPR_RESTRICT pY,

                                             const BYTE* WINPR_RESTRICT pU,

                                             const BYTE* WINPR_RESTRICT pV,

                                             BYTE* WINPR_RESTRICT pRGB, size_t width,

                                             const uint8_t rPos, const uint8_t gPos,

                                             const uint8_t bPos, const uint8_t aPos)

{

  UINT32 x = 0;

  for (; x < width - width % 16; x += 16)

  {

    const uint8x16_t Y0raw = vld1q_u8(&pY[x]);

    const uint8x8x2_t Y0 = { { vget_low_u8(Y0raw), vget_high_u8(Y0raw) } };

    const int16x8x2_t D = loadUV(pU, x);

    const int16x8x2_t E = loadUV(pV, x);

    neon_YuvToRgbPixel(&pRGB[4ULL * x], Y0, D, E, rPos, gPos, bPos, aPos);

  }

  for (; x < width - width % 2; x += 2)

  {

    const BYTE U = pU[x / 2];

    const BYTE V = pV[x / 2];

    neon_write_pixel(&pRGB[4 * x], pY[x], U, V, rPos, gPos, bPos, aPos);

    neon_write_pixel(&pRGB[4 * (1ULL + x)], pY[1ULL + x], U, V, rPos, gPos, bPos, aPos);

  }

  for (; x < width; x++)

  {

    const BYTE U = pU[x / 2];

    const BYTE V = pV[x / 2];

    neon_write_pixel(&pRGB[4 * x], pY[x], U, V, rPos, gPos, bPos, aPos);

  }

}

static INLINE pstatus_t neon_YUV420ToX(const BYTE* WINPR_RESTRICT pSrc[3], const UINT32 srcStep[3],

                                       BYTE* WINPR_RESTRICT pDst, UINT32 dstStep,

                                       const prim_size_t* WINPR_RESTRICT roi, const uint8_t rPos,

                                       const uint8_t gPos, const uint8_t bPos, const uint8_t aPos)

{

  const UINT32 nWidth = roi->width;

  const UINT32 nHeight = roi->height;

  WINPR_ASSERT(nHeight > 0);

  UINT32 y = 0;

  for (; y < (nHeight - 1); y += 2)

  {

    const uint8_t* pY[2] = { pSrc[0] + y * srcStep[0], pSrc[0] + (1ULL + y) * srcStep[0] };

    const uint8_t* pU = pSrc[1] + (y / 2) * srcStep[1];

    const uint8_t* pV = pSrc[2] + (y / 2) * srcStep[2];

    uint8_t* pRGB[2] = { pDst + y * dstStep, pDst + (1ULL + y) * dstStep };

    neon_YUV420ToX_DOUBLE_ROW(pY, pU, pV, pRGB, nWidth, rPos, gPos, bPos, aPos);

  }

  for (; y < nHeight; y++)

  {

    const uint8_t* pY = pSrc[0] + y * srcStep[0];

    const uint8_t* pU = pSrc[1] + (y / 2) * srcStep[1];

    const uint8_t* pV = pSrc[2] + (y / 2) * srcStep[2];

    uint8_t* pRGB = pDst + y * dstStep;

    neon_YUV420ToX_SINGLE_ROW(pY, pU, pV, pRGB, nWidth, rPos, gPos, bPos, aPos);

  }

  return PRIMITIVES_SUCCESS;

}

static pstatus_t neon_YUV420ToRGB_8u_P3AC4R(const BYTE* WINPR_RESTRICT pSrc[3],

                                            const UINT32 srcStep[3], BYTE* WINPR_RESTRICT pDst,

                                            UINT32 dstStep, UINT32 DstFormat,

                                            const prim_size_t* WINPR_RESTRICT roi)

{

  switch (DstFormat)

  {

    case PIXEL_FORMAT_BGRA32:

    case PIXEL_FORMAT_BGRX32:

      return neon_YUV420ToX(pSrc, srcStep, pDst, dstStep, roi, 2, 1, 0, 3);

    case PIXEL_FORMAT_RGBA32:

    case PIXEL_FORMAT_RGBX32:

      return neon_YUV420ToX(pSrc, srcStep, pDst, dstStep, roi, 0, 1, 2, 3);

    case PIXEL_FORMAT_ARGB32:

    case PIXEL_FORMAT_XRGB32:

      return neon_YUV420ToX(pSrc, srcStep, pDst, dstStep, roi, 1, 2, 3, 0);

    case PIXEL_FORMAT_ABGR32:

    case PIXEL_FORMAT_XBGR32:

      return neon_YUV420ToX(pSrc, srcStep, pDst, dstStep, roi, 3, 2, 1, 0);

    default:

      return generic->YUV420ToRGB_8u_P3AC4R(pSrc, srcStep, pDst, dstStep, DstFormat, roi);

  }

}

static inline int16x8_t loadUVreg(uint8x8_t Vraw)

{

  const int16x8_t V = vreinterpretq_s16_u16(vmovl_u8(Vraw));

  const int16x8_t c128 = vdupq_n_s16(128);

  const int16x8_t E = vsubq_s16(V, c128);

  return E;

}

static inline int16x8x2_t loadUV444(uint8x16_t Vld)

{

  const uint8x8x2_t V = { { vget_low_u8(Vld), vget_high_u8(Vld) } };

  const int16x8x2_t res = { {

    loadUVreg(V.val[0]),

    loadUVreg(V.val[1]),

  } };

  return res;

}

static inline void avgUV(BYTE U[2][2])

{

  const BYTE u00 = U[0][0];

  const INT16 umul = (INT16)u00 << 2;

  const INT16 sum = (INT16)U[0][1] + U[1][0] + U[1][1];

  const INT16 wavg = umul - sum;

  const BYTE val = CONDITIONAL_CLIP(wavg, u00);

  U[0][0] = val;

}

static inline void neon_avgUV(uint8x16_t pU[2])

{

  /* put even and odd values into different registers.

   * U 0/0 is in lower half */

  const uint8x16x2_t usplit = vuzpq_u8(pU[0], pU[1]);

  const uint8x16_t ueven = usplit.val[0];

  const uint8x16_t uodd = usplit.val[1];

  const uint8x8_t u00 = vget_low_u8(ueven);

  const uint8x8_t u01 = vget_low_u8(uodd);

  const uint8x8_t u10 = vget_high_u8(ueven);

  const uint8x8_t u11 = vget_high_u8(uodd);

  /* Create sum of U01 + U10 + U11 */

  const uint16x8_t uoddsum = vaddl_u8(u01, u10);

  const uint16x8_t usum = vaddq_u16(uoddsum, vmovl_u8(u11));

  /* U00 * 4 */

  const uint16x8_t umul = vshll_n_u8(u00, 2);

  /* U00 - (U01 + U10 + U11) */

  const int16x8_t wavg = vsubq_s16(vreinterpretq_s16_u16(umul), vreinterpretq_s16_u16(usum));

  const uint8x8_t avg = vqmovun_s16(wavg);

  /* abs(u00 - avg) */

  const uint8x8_t absdiff = vabd_u8(avg, u00);

  /* (diff < 30) ? u00 : avg */

  const uint8x8_t mask = vclt_u8(absdiff, vdup_n_u8(30));

  /* out1 = u00 & mask */

  const uint8x8_t out1 = vand_u8(u00, mask);

  /* invmask = ~mask */

  const uint8x8_t notmask = vmvn_u8(mask);

  /* out2 = avg & invmask */

  const uint8x8_t out2 = vand_u8(avg, notmask);

  /* out = out1 | out2 */

  const uint8x8_t out = vorr_u8(out1, out2);

  const uint8x8x2_t ua = vzip_u8(out, u01);

  const uint8x16_t u = vcombine_u8(ua.val[0], ua.val[1]);

  pU[0] = u;

}

static INLINE pstatus_t neon_YUV444ToX_SINGLE_ROW(const BYTE* WINPR_RESTRICT pY,

                                                  const BYTE* WINPR_RESTRICT pU,

                                                  const BYTE* WINPR_RESTRICT pV,

                                                  BYTE* WINPR_RESTRICT pRGB, size_t width,

                                                  const uint8_t rPos, const uint8_t gPos,

                                                  const uint8_t bPos, const uint8_t aPos)

{

  WINPR_ASSERT(width % 2 == 0);

  size_t x = 0;

  for (; x < width - width % 16; x += 16)

  {

    uint8x16_t U = vld1q_u8(&pU[x]);

    uint8x16_t V = vld1q_u8(&pV[x]);

    const uint8x16_t Y0raw = vld1q_u8(&pY[x]);

    const uint8x8x2_t Y0 = { { vget_low_u8(Y0raw), vget_high_u8(Y0raw) } };

    const int16x8x2_t D0 = loadUV444(U);

    const int16x8x2_t E0 = loadUV444(V);

    neon_YuvToRgbPixel(&pRGB[4ULL * x], Y0, D0, E0, rPos, gPos, bPos, aPos);

  }

  for (; x < width; x += 2)

  {

    BYTE* rgb = &pRGB[x * 4];

    for (size_t j = 0; j < 2; j++)

    {

      const BYTE y = pY[x + j];

      const BYTE u = pU[x + j];

      const BYTE v = pV[x + j];

      neon_write_pixel(&rgb[4 * (j)], y, u, v, rPos, gPos, bPos, aPos);

    }

  }

  return PRIMITIVES_SUCCESS;

}

static INLINE pstatus_t neon_YUV444ToX_DOUBLE_ROW(const BYTE* WINPR_RESTRICT pY[2],

                                                  const BYTE* WINPR_RESTRICT pU[2],

                                                  const BYTE* WINPR_RESTRICT pV[2],

                                                  BYTE* WINPR_RESTRICT pRGB[2], size_t width,

                                                  const uint8_t rPos, const uint8_t gPos,

                                                  const uint8_t bPos, const uint8_t aPos)

{

  WINPR_ASSERT(width % 2 == 0);

  size_t x = 0;

  for (; x < width - width % 16; x += 16)

  {

    uint8x16_t U[2] = { vld1q_u8(&pU[0][x]), vld1q_u8(&pU[1][x]) };

    neon_avgUV(U);

    uint8x16_t V[2] = { vld1q_u8(&pV[0][x]), vld1q_u8(&pV[1][x]) };

    neon_avgUV(V);

    const uint8x16_t Y0raw = vld1q_u8(&pY[0][x]);

    const uint8x8x2_t Y0 = { { vget_low_u8(Y0raw), vget_high_u8(Y0raw) } };

    const int16x8x2_t D0 = loadUV444(U[0]);

    const int16x8x2_t E0 = loadUV444(V[0]);

    neon_YuvToRgbPixel(&pRGB[0][4ULL * x], Y0, D0, E0, rPos, gPos, bPos, aPos);

    const uint8x16_t Y1raw = vld1q_u8(&pY[1][x]);

    const uint8x8x2_t Y1 = { { vget_low_u8(Y1raw), vget_high_u8(Y1raw) } };

    const int16x8x2_t D1 = loadUV444(U[1]);

    const int16x8x2_t E1 = loadUV444(V[1]);

    neon_YuvToRgbPixel(&pRGB[1][4ULL * x], Y1, D1, E1, rPos, gPos, bPos, aPos);

  }

  for (; x < width; x += 2)

  {

    BYTE* rgb[2] = { &pRGB[0][x * 4], &pRGB[1][x * 4] };

    BYTE U[2][2] = { { pU[0][x], pU[0][x + 1] }, { pU[1][x], pU[1][x + 1] } };

    avgUV(U);

    BYTE V[2][2] = { { pV[0][x], pV[0][x + 1] }, { pV[1][x], pV[1][x + 1] } };

    avgUV(V);

    for (size_t i = 0; i < 2; i++)

    {

      for (size_t j = 0; j < 2; j++)

      {

        const BYTE y = pY[i][x + j];

        const BYTE u = U[i][j];

        const BYTE v = V[i][j];

        neon_write_pixel(&rgb[i][4 * (j)], y, u, v, rPos, gPos, bPos, aPos);

      }

    }

  }

  return PRIMITIVES_SUCCESS;

}

static INLINE pstatus_t neon_YUV444ToX(const BYTE* WINPR_RESTRICT pSrc[3], const UINT32 srcStep[3],

                                       BYTE* WINPR_RESTRICT pDst, UINT32 dstStep,

                                       const prim_size_t* WINPR_RESTRICT roi, const uint8_t rPos,

                                       const uint8_t gPos, const uint8_t bPos, const uint8_t aPos)

{

  WINPR_ASSERT(roi);

  const UINT32 nWidth = roi->width;

  const UINT32 nHeight = roi->height;

  size_t y = 0;

  for (; y < nHeight - nHeight % 2; y += 2)

  {

    const uint8_t* WINPR_RESTRICT pY[2] = { pSrc[0] + y * srcStep[0],

                                          pSrc[0] + (y + 1) * srcStep[0] };

    const uint8_t* WINPR_RESTRICT pU[2] = { pSrc[1] + y * srcStep[1],

                                          pSrc[1] + (y + 1) * srcStep[1] };

    const uint8_t* WINPR_RESTRICT pV[2] = { pSrc[2] + y * srcStep[2],

                                          pSrc[2] + (y + 1) * srcStep[2] };

    uint8_t* WINPR_RESTRICT pRGB[2] = { &pDst[y * dstStep], &pDst[(y + 1) * dstStep] };

    const pstatus_t rc =

        neon_YUV444ToX_DOUBLE_ROW(pY, pU, pV, pRGB, nWidth, rPos, gPos, bPos, aPos);

    if (rc != PRIMITIVES_SUCCESS)

      return rc;

  }

  for (; y < nHeight; y++)

  {

    const uint8_t* WINPR_RESTRICT pY = pSrc[0] + y * srcStep[0];

    const uint8_t* WINPR_RESTRICT pU = pSrc[1] + y * srcStep[1];

    const uint8_t* WINPR_RESTRICT pV = pSrc[2] + y * srcStep[2];

    uint8_t* WINPR_RESTRICT pRGB = &pDst[y * dstStep];

    const pstatus_t rc =

        neon_YUV444ToX_SINGLE_ROW(pY, pU, pV, pRGB, nWidth, rPos, gPos, bPos, aPos);

    if (rc != PRIMITIVES_SUCCESS)

      return rc;

  }

  return PRIMITIVES_SUCCESS;

}

static pstatus_t neon_YUV444ToRGB_8u_P3AC4R(const BYTE* WINPR_RESTRICT pSrc[3],

                                            const UINT32 srcStep[3], BYTE* WINPR_RESTRICT pDst,

                                            UINT32 dstStep, UINT32 DstFormat,

                                            const prim_size_t* WINPR_RESTRICT roi)

{

  switch (DstFormat)

  {

    case PIXEL_FORMAT_BGRA32:

    case PIXEL_FORMAT_BGRX32:

      return neon_YUV444ToX(pSrc, srcStep, pDst, dstStep, roi, 2, 1, 0, 3);

    case PIXEL_FORMAT_RGBA32:

    case PIXEL_FORMAT_RGBX32:

      return neon_YUV444ToX(pSrc, srcStep, pDst, dstStep, roi, 0, 1, 2, 3);

    case PIXEL_FORMAT_ARGB32:

    case PIXEL_FORMAT_XRGB32:

      return neon_YUV444ToX(pSrc, srcStep, pDst, dstStep, roi, 1, 2, 3, 0);

    case PIXEL_FORMAT_ABGR32:

    case PIXEL_FORMAT_XBGR32:

      return neon_YUV444ToX(pSrc, srcStep, pDst, dstStep, roi, 3, 2, 1, 0);

    default:

      return generic->YUV444ToRGB_8u_P3AC4R(pSrc, srcStep, pDst, dstStep, DstFormat, roi);

  }

}

static pstatus_t neon_LumaToYUV444(const BYTE* WINPR_RESTRICT pSrcRaw[3], const UINT32 srcStep[3],

                                   BYTE* WINPR_RESTRICT pDstRaw[3], const UINT32 dstStep[3],

                                   const RECTANGLE_16* WINPR_RESTRICT roi)

{

  const UINT32 nWidth = roi->right - roi->left;

  const UINT32 nHeight = roi->bottom - roi->top;

  const UINT32 halfWidth = (nWidth + 1) / 2;

  const UINT32 halfHeight = (nHeight + 1) / 2;

  const UINT32 evenY = 0;

  const BYTE* pSrc[3] = { pSrcRaw[0] + roi->top * srcStep[0] + roi->left,

                        pSrcRaw[1] + roi->top / 2 * srcStep[1] + roi->left / 2,

                        pSrcRaw[2] + roi->top / 2 * srcStep[2] + roi->left / 2 };

  BYTE* pDst[3] = { pDstRaw[0] + roi->top * dstStep[0] + roi->left,

                  pDstRaw[1] + roi->top * dstStep[1] + roi->left,

                  pDstRaw[2] + roi->top * dstStep[2] + roi->left };

  /* Y data is already here... */

  /* B1 */

  for (UINT32 y = 0; y < nHeight; y++)

  {

    const BYTE* Ym = pSrc[0] + srcStep[0] * y;

    BYTE* pY = pDst[0] + dstStep[0] * y;

    memcpy(pY, Ym, nWidth);

  }

  /* The first half of U, V are already here part of this frame. */

  /* B2 and B3 */

  for (UINT32 y = 0; y < halfHeight; y++)

  {

    const UINT32 val2y = (2 * y + evenY);

    const BYTE* Um = pSrc[1] + srcStep[1] * y;

    const BYTE* Vm = pSrc[2] + srcStep[2] * y;

    BYTE* pU = pDst[1] + dstStep[1] * val2y;

    BYTE* pV = pDst[2] + dstStep[2] * val2y;

    BYTE* pU1 = pU + dstStep[1];

    BYTE* pV1 = pV + dstStep[2];

    UINT32 x = 0;

    for (; x + 16 < halfWidth; x += 16)

    {

      {

        const uint8x16_t u = vld1q_u8(Um);

        uint8x16x2_t u2x;

        u2x.val[0] = u;

        u2x.val[1] = u;

        vst2q_u8(pU, u2x);

        vst2q_u8(pU1, u2x);

        Um += 16;

        pU += 32;

        pU1 += 32;

      }

      {

        const uint8x16_t v = vld1q_u8(Vm);

        uint8x16x2_t v2x;

        v2x.val[0] = v;

        v2x.val[1] = v;

        vst2q_u8(pV, v2x);

        vst2q_u8(pV1, v2x);

        Vm += 16;

        pV += 32;

        pV1 += 32;

      }

    }

    for (; x < halfWidth; x++)

    {

      const BYTE u = *Um++;

      const BYTE v = *Vm++;

      *pU++ = u;

      *pU++ = u;

      *pU1++ = u;

      *pU1++ = u;

      *pV++ = v;

      *pV++ = v;

      *pV1++ = v;

      *pV1++ = v;

    }

  }

  return PRIMITIVES_SUCCESS;

}

static pstatus_t neon_ChromaV1ToYUV444(const BYTE* WINPR_RESTRICT pSrcRaw[3],

                                       const UINT32 srcStep[3], BYTE* WINPR_RESTRICT pDstRaw[3],

                                       const UINT32 dstStep[3],

                                       const RECTANGLE_16* WINPR_RESTRICT roi)

{

  const UINT32 mod = 16;

  UINT32 uY = 0;

  UINT32 vY = 0;

  const UINT32 nWidth = roi->right - roi->left;

  const UINT32 nHeight = roi->bottom - roi->top;

  const UINT32 halfWidth = (nWidth) / 2;

  const UINT32 halfHeight = (nHeight) / 2;

  const UINT32 oddY = 1;

  const UINT32 evenY = 0;

  const UINT32 oddX = 1;

  /* The auxiliary frame is aligned to multiples of 16x16.

   * We need the padded height for B4 and B5 conversion. */

  const UINT32 padHeigth = nHeight + 16 - nHeight % 16;

  const UINT32 halfPad = halfWidth % 16;

  const BYTE* pSrc[3] = { pSrcRaw[0] + roi->top * srcStep[0] + roi->left,

                        pSrcRaw[1] + roi->top / 2 * srcStep[1] + roi->left / 2,

                        pSrcRaw[2] + roi->top / 2 * srcStep[2] + roi->left / 2 };

  BYTE* pDst[3] = { pDstRaw[0] + roi->top * dstStep[0] + roi->left,

                  pDstRaw[1] + roi->top * dstStep[1] + roi->left,

                  pDstRaw[2] + roi->top * dstStep[2] + roi->left };

  /* The second half of U and V is a bit more tricky... */

  /* B4 and B5 */

  for (UINT32 y = 0; y < padHeigth; y++)

  {

    const BYTE* Ya = pSrc[0] + srcStep[0] * y;

    BYTE* pX;

    if ((y) % mod < (mod + 1) / 2)

    {

      const UINT32 pos = (2 * uY++ + oddY);

      if (pos >= nHeight)

        continue;

      pX = pDst[1] + dstStep[1] * pos;

    }

    else

    {

      const UINT32 pos = (2 * vY++ + oddY);

      if (pos >= nHeight)

        continue;

      pX = pDst[2] + dstStep[2] * pos;

    }

    memcpy(pX, Ya, nWidth);

  }

  /* B6 and B7 */

  for (UINT32 y = 0; y < halfHeight; y++)

  {

    const UINT32 val2y = (y * 2 + evenY);

    const BYTE* Ua = pSrc[1] + srcStep[1] * y;

    const BYTE* Va = pSrc[2] + srcStep[2] * y;

    BYTE* pU = pDst[1] + dstStep[1] * val2y;

    BYTE* pV = pDst[2] + dstStep[2] * val2y;

    UINT32 x = 0;

    for (; x < halfWidth - halfPad; x += 16)

    {

      {

        uint8x16x2_t u = vld2q_u8(&pU[2 * x]);

        u.val[1] = vld1q_u8(&Ua[x]);

        vst2q_u8(&pU[2 * x], u);

      }

      {

        uint8x16x2_t v = vld2q_u8(&pV[2 * x]);

        v.val[1] = vld1q_u8(&Va[x]);

        vst2q_u8(&pV[2 * x], v);

      }

    }

    for (; x < halfWidth; x++)

    {

      const UINT32 val2x1 = (x * 2 + oddX);

      pU[val2x1] = Ua[x];

      pV[val2x1] = Va[x];

    }

  }

  return PRIMITIVES_SUCCESS;

}

static pstatus_t neon_ChromaV2ToYUV444(const BYTE* WINPR_RESTRICT pSrc[3], const UINT32 srcStep[3],

                                       UINT32 nTotalWidth, UINT32 nTotalHeight,

                                       BYTE* WINPR_RESTRICT pDst[3], const UINT32 dstStep[3],

                                       const RECTANGLE_16* WINPR_RESTRICT roi)

{

  const UINT32 nWidth = roi->right - roi->left;

  const UINT32 nHeight = roi->bottom - roi->top;

  const UINT32 halfWidth = (nWidth + 1) / 2;

  const UINT32 halfPad = halfWidth % 16;

  const UINT32 halfHeight = (nHeight + 1) / 2;

  const UINT32 quaterWidth = (nWidth + 3) / 4;

  const UINT32 quaterPad = quaterWidth % 16;

  /* B4 and B5: odd UV values for width/2, height */

  for (UINT32 y = 0; y < nHeight; y++)

  {

    const UINT32 yTop = y + roi->top;

    const BYTE* pYaU = pSrc[0] + srcStep[0] * yTop + roi->left / 2;

    const BYTE* pYaV = pYaU + nTotalWidth / 2;

    BYTE* pU = pDst[1] + dstStep[1] * yTop + roi->left;

    BYTE* pV = pDst[2] + dstStep[2] * yTop + roi->left;

    UINT32 x = 0;

    for (; x < halfWidth - halfPad; x += 16)

    {

      {

        uint8x16x2_t u = vld2q_u8(&pU[2 * x]);

        u.val[1] = vld1q_u8(&pYaU[x]);

        vst2q_u8(&pU[2 * x], u);

      }

      {

        uint8x16x2_t v = vld2q_u8(&pV[2 * x]);

        v.val[1] = vld1q_u8(&pYaV[x]);

        vst2q_u8(&pV[2 * x], v);

      }

    }

    for (; x < halfWidth; x++)

    {

      const UINT32 odd = 2 * x + 1;

      pU[odd] = pYaU[x];

      pV[odd] = pYaV[x];

    }

  }

  /* B6 - B9 */

  for (UINT32 y = 0; y < halfHeight; y++)

  {

    const BYTE* pUaU = pSrc[1] + srcStep[1] * (y + roi->top / 2) + roi->left / 4;

    const BYTE* pUaV = pUaU + nTotalWidth / 4;

    const BYTE* pVaU = pSrc[2] + srcStep[2] * (y + roi->top / 2) + roi->left / 4;

    const BYTE* pVaV = pVaU + nTotalWidth / 4;

    BYTE* pU = pDst[1] + dstStep[1] * (2 * y + 1 + roi->top) + roi->left;

    BYTE* pV = pDst[2] + dstStep[2] * (2 * y + 1 + roi->top) + roi->left;

    UINT32 x = 0;

    for (; x < quaterWidth - quaterPad; x += 16)

    {

      {

        uint8x16x4_t u = vld4q_u8(&pU[4 * x]);

        u.val[0] = vld1q_u8(&pUaU[x]);

        u.val[2] = vld1q_u8(&pVaU[x]);

        vst4q_u8(&pU[4 * x], u);

      }

      {

        uint8x16x4_t v = vld4q_u8(&pV[4 * x]);

        v.val[0] = vld1q_u8(&pUaV[x]);

        v.val[2] = vld1q_u8(&pVaV[x]);

        vst4q_u8(&pV[4 * x], v);

      }

    }

    for (; x < quaterWidth; x++)

    {

      pU[4 * x + 0] = pUaU[x];

      pV[4 * x + 0] = pUaV[x];

      pU[4 * x + 2] = pVaU[x];

      pV[4 * x + 2] = pVaV[x];

    }

  }

  return PRIMITIVES_SUCCESS;

}

static pstatus_t neon_YUV420CombineToYUV444(avc444_frame_type type,

                                            const BYTE* WINPR_RESTRICT pSrc[3],

                                            const UINT32 srcStep[3], UINT32 nWidth, UINT32 nHeight,

                                            BYTE* WINPR_RESTRICT pDst[3], const UINT32 dstStep[3],

                                            const RECTANGLE_16* WINPR_RESTRICT roi)

{

  if (!pSrc || !pSrc[0] || !pSrc[1] || !pSrc[2])

    return -1;

  if (!pDst || !pDst[0] || !pDst[1] || !pDst[2])

    return -1;

  if (!roi)

    return -1;

  switch (type)

  {

    case AVC444_LUMA:

      return neon_LumaToYUV444(pSrc, srcStep, pDst, dstStep, roi);

    case AVC444_CHROMAv1:

      return neon_ChromaV1ToYUV444(pSrc, srcStep, pDst, dstStep, roi);

    case AVC444_CHROMAv2:

      return neon_ChromaV2ToYUV444(pSrc, srcStep, nWidth, nHeight, pDst, dstStep, roi);

    default:

      return -1;

  }

}

#endif

void primitives_init_YUV_neon_int(primitives_t* WINPR_RESTRICT prims)

{

#if defined(NEON_INTRINSICS_ENABLED)

  generic = primitives_get_generic();

  WLog_VRB(PRIM_TAG, "NEON optimizations");

  prims->YUV420ToRGB_8u_P3AC4R = neon_YUV420ToRGB_8u_P3AC4R;

  prims->YUV444ToRGB_8u_P3AC4R = neon_YUV444ToRGB_8u_P3AC4R;

  prims->YUV420CombineToYUV444 = neon_YUV420CombineToYUV444;

#else

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

  WINPR_UNUSED(prims);

#endif

}