/*
 *  Copyright (c) 2022 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include <assert.h>
#include "./vpx_dsp_rtcd.h"
#include "vpx_dsp/loongarch/vpx_convolve_lsx.h"

static const uint8_t mc_filt_mask_arr[16 * 3] = {
  /* 8 width cases */
  0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 6, 6, 7, 7, 8,
  /* 4 width cases */
  0, 1, 1, 2, 2, 3, 3, 4, 16, 17, 17, 18, 18, 19, 19, 20,
  /* 4 width cases */
  8, 9, 9, 10, 10, 11, 11, 12, 24, 25, 25, 26, 26, 27, 27, 28
};

static void common_hz_8t_4x4_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 const int8_t *filter) {
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out, out0, out1;

  mask0 = __lsx_vld(mc_filt_mask_arr, 16);
  src -= 3;
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);

  LSX_LD_4(src, src_stride, src0, src1, src2, src3);
  DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
            src1, src2, src3);
  HORIZ_8TAP_4WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2, mask3,
                             filter0, filter1, filter2, filter3, out0, out1);
  out = __lsx_vssrarni_b_h(out1, out0, 7);
  out = __lsx_vxori_b(out, 128);
  __lsx_vstelm_w(out, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_w(out, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_w(out, dst, 0, 2);
  dst += dst_stride;
  __lsx_vstelm_w(out, dst, 0, 3);
}

static void common_hz_8t_4x8_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 const int8_t *filter) {
  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride + src_stride2;
  int32_t src_stride4 = src_stride2 << 1;
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out0, out1, out2, out3;
  uint8_t *_src = (uint8_t *)src - 3;

  mask0 = __lsx_vld(mc_filt_mask_arr, 16);
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);

  src0 = __lsx_vld(_src, 0);
  DUP2_ARG2(__lsx_vldx, _src, src_stride, _src, src_stride2, src1, src2);
  src3 = __lsx_vldx(_src, src_stride3);
  _src += src_stride4;
  DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
            src1, src2, src3);
  HORIZ_8TAP_4WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2, mask3,
                             filter0, filter1, filter2, filter3, out0, out1);
  src0 = __lsx_vld(_src, 0);
  DUP2_ARG2(__lsx_vldx, _src, src_stride, _src, src_stride2, src1, src2);
  src3 = __lsx_vldx(_src, src_stride3);
  DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
            src1, src2, src3);
  HORIZ_8TAP_4WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2, mask3,
                             filter0, filter1, filter2, filter3, out2, out3);
  DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
  DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);

  __lsx_vstelm_w(out0, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_w(out0, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_w(out0, dst, 0, 2);
  dst += dst_stride;
  __lsx_vstelm_w(out0, dst, 0, 3);
  dst += dst_stride;
  __lsx_vstelm_w(out1, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_w(out1, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_w(out1, dst, 0, 2);
  dst += dst_stride;
  __lsx_vstelm_w(out1, dst, 0, 3);
}

static void common_hz_8t_4w_lsx(const uint8_t *src, int32_t src_stride,
                                uint8_t *dst, int32_t dst_stride,
                                const int8_t *filter, int32_t height) {
  if (height == 4) {
    common_hz_8t_4x4_lsx(src, src_stride, dst, dst_stride, filter);
  } else if (height == 8) {
    common_hz_8t_4x8_lsx(src, src_stride, dst, dst_stride, filter);
  }
}

static void common_hz_8t_8x4_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 const int8_t *filter) {
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out0, out1, out2, out3;

  mask0 = __lsx_vld(mc_filt_mask_arr, 0);
  src -= 3;
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);

  LSX_LD_4(src, src_stride, src0, src1, src2, src3);
  DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
            src1, src2, src3);
  HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2, mask3,
                             filter0, filter1, filter2, filter3, out0, out1,
                             out2, out3);
  DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
  DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
  __lsx_vstelm_d(out0, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_d(out0, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_d(out1, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_d(out1, dst, 0, 1);
}

static void common_hz_8t_8x8mult_lsx(const uint8_t *src, int32_t src_stride,
                                     uint8_t *dst, int32_t dst_stride,
                                     const int8_t *filter, int32_t height) {
  uint32_t loop_cnt = height >> 2;
  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride + src_stride2;
  int32_t src_stride4 = src_stride2 << 1;
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out0, out1, out2, out3;
  uint8_t *_src = (uint8_t *)src - 3;

  mask0 = __lsx_vld(mc_filt_mask_arr, 0);
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);

  for (; loop_cnt--;) {
    src0 = __lsx_vld(_src, 0);
    DUP2_ARG2(__lsx_vldx, _src, src_stride, _src, src_stride2, src1, src2);
    src3 = __lsx_vldx(_src, src_stride3);
    _src += src_stride4;
    DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
              src1, src2, src3);
    HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2,
                               mask3, filter0, filter1, filter2, filter3, out0,
                               out1, out2, out3);
    DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
    DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
    __lsx_vstelm_d(out0, dst, 0, 0);
    dst += dst_stride;
    __lsx_vstelm_d(out0, dst, 0, 1);
    dst += dst_stride;
    __lsx_vstelm_d(out1, dst, 0, 0);
    dst += dst_stride;
    __lsx_vstelm_d(out1, dst, 0, 1);
    dst += dst_stride;
  }
}

static void common_hz_8t_8w_lsx(const uint8_t *src, int32_t src_stride,
                                uint8_t *dst, int32_t dst_stride,
                                const int8_t *filter, int32_t height) {
  if (height == 4) {
    common_hz_8t_8x4_lsx(src, src_stride, dst, dst_stride, filter);
  } else {
    common_hz_8t_8x8mult_lsx(src, src_stride, dst, dst_stride, filter, height);
  }
}

static void common_hz_8t_16w_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 const int8_t *filter, int32_t height) {
  uint32_t loop_cnt = height >> 1;
  int32_t stride = src_stride << 1;
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out0, out1, out2, out3;

  mask0 = __lsx_vld(mc_filt_mask_arr, 0);
  src -= 3;
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);

  for (; loop_cnt--;) {
    const uint8_t *_src = src + src_stride;
    DUP2_ARG2(__lsx_vld, src, 0, _src, 0, src0, src2);
    DUP2_ARG2(__lsx_vld, src, 8, _src, 8, src1, src3);
    DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
              src1, src2, src3);
    HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2,
                               mask3, filter0, filter1, filter2, filter3, out0,
                               out1, out2, out3);
    DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
    DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
    __lsx_vst(out0, dst, 0);
    dst += dst_stride;
    __lsx_vst(out1, dst, 0);
    dst += dst_stride;
    src += stride;
  }
}

static void common_hz_8t_32w_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 const int8_t *filter, int32_t height) {
  uint32_t loop_cnt = height >> 1;
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out0, out1, out2, out3;
  __m128i shuff = { 0x0F0E0D0C0B0A0908, 0x1716151413121110 };

  mask0 = __lsx_vld(mc_filt_mask_arr, 0);
  src -= 3;
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);

  for (; loop_cnt--;) {
    DUP2_ARG2(__lsx_vld, src, 0, src, 16, src0, src2);
    src3 = __lsx_vld(src, 24);
    src1 = __lsx_vshuf_b(src2, src0, shuff);
    src += src_stride;
    DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
              src1, src2, src3);
    HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2,
                               mask3, filter0, filter1, filter2, filter3, out0,
                               out1, out2, out3);
    DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
    DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
    __lsx_vst(out0, dst, 0);
    __lsx_vst(out1, dst, 16);

    DUP2_ARG2(__lsx_vld, src, 0, src, 16, src0, src2);
    src3 = __lsx_vld(src, 24);
    src1 = __lsx_vshuf_b(src2, src0, shuff);
    src += src_stride;

    dst += dst_stride;
    DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
              src1, src2, src3);
    HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2,
                               mask3, filter0, filter1, filter2, filter3, out0,
                               out1, out2, out3);
    DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
    DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
    __lsx_vst(out0, dst, 0);
    __lsx_vst(out1, dst, 16);
    dst += dst_stride;
  }
}

static void common_hz_8t_64w_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 const int8_t *filter, int32_t height) {
  int32_t loop_cnt = height;
  __m128i src0, src1, src2, src3;
  __m128i filter0, filter1, filter2, filter3;
  __m128i mask0, mask1, mask2, mask3;
  __m128i out0, out1, out2, out3;
  __m128i shuff = { 0x0F0E0D0C0B0A0908, 0x1716151413121110 };

  mask0 = __lsx_vld(mc_filt_mask_arr, 0);
  src -= 3;
  DUP2_ARG2(__lsx_vaddi_bu, mask0, 2, mask0, 4, mask1, mask2);
  mask3 = __lsx_vaddi_bu(mask0, 6);
  DUP4_ARG2(__lsx_vldrepl_h, filter, 0, filter, 2, filter, 4, filter, 6,
            filter0, filter1, filter2, filter3);

  for (; loop_cnt--;) {
    DUP2_ARG2(__lsx_vld, src, 0, src, 16, src0, src2);
    src3 = __lsx_vld(src, 24);
    src1 = __lsx_vshuf_b(src2, src0, shuff);
    DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
              src1, src2, src3);
    HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2,
                               mask3, filter0, filter1, filter2, filter3, out0,
                               out1, out2, out3);
    DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
    DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
    __lsx_vst(out0, dst, 0);
    __lsx_vst(out1, dst, 16);

    DUP2_ARG2(__lsx_vld, src, 32, src, 48, src0, src2);
    src3 = __lsx_vld(src, 56);
    src1 = __lsx_vshuf_b(src2, src0, shuff);
    DUP4_ARG2(__lsx_vxori_b, src0, 128, src1, 128, src2, 128, src3, 128, src0,
              src1, src2, src3);
    HORIZ_8TAP_8WID_4VECS_FILT(src0, src1, src2, src3, mask0, mask1, mask2,
                               mask3, filter0, filter1, filter2, filter3, out0,
                               out1, out2, out3);
    DUP2_ARG3(__lsx_vssrarni_b_h, out1, out0, 7, out3, out2, 7, out0, out1);
    DUP2_ARG2(__lsx_vxori_b, out0, 128, out1, 128, out0, out1);
    __lsx_vst(out0, dst, 32);
    __lsx_vst(out1, dst, 48);
    src += src_stride;
    dst += dst_stride;
  }
}

static void common_hz_2t_4x4_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 int8_t *filter) {
  __m128i src0, src1, src2, src3, mask;
  __m128i filt0, vec0, vec1, res0, res1;
  __m128i vec2, vec3;
  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride + src_stride2;

  int32_t dst_stride2 = dst_stride << 1;
  int32_t dst_stride3 = dst_stride + dst_stride2;

  mask = __lsx_vld(mc_filt_mask_arr, 16);
  /* rearranging filter */
  filt0 = __lsx_vldrepl_h(filter, 0);

  src0 = __lsx_vld(src, 0);
  DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2);
  src3 = __lsx_vldx(src, src_stride3);
  DUP2_ARG3(__lsx_vshuf_b, src1, src0, mask, src3, src2, mask, vec0, vec1);
  DUP2_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, vec3);
  DUP2_ARG2(__lsx_vsrari_h, vec2, FILTER_BITS, vec3, FILTER_BITS, vec2, vec3);
  DUP2_ARG2(__lsx_vpickev_b, vec2, vec2, vec3, vec3, res0, res1);

  __lsx_vstelm_w(res0, dst, 0, 0);
  __lsx_vstelm_w(res0, dst + dst_stride, 0, 1);
  __lsx_vstelm_w(res1, dst + dst_stride2, 0, 0);
  __lsx_vstelm_w(res1, dst + dst_stride3, 0, 1);
}

static void common_hz_2t_4x8_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 int8_t *filter) {
  __m128i vec0, vec1, vec2, vec3, filt0;
  __m128i src0, src1, src2, src3, src4, src5, src6, src7, mask;
  __m128i res0, res1, res2, res3;
  __m128i vec4, vec5, vec6, vec7;
  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride + src_stride2;
  int32_t src_stride4 = src_stride2 << 1;

  int32_t dst_stride2 = dst_stride << 1;
  int32_t dst_stride3 = dst_stride + dst_stride2;

  uint8_t *src_tmp1 = src + src_stride4;

  mask = __lsx_vld(mc_filt_mask_arr, 16);

  /* rearranging filter */
  filt0 = __lsx_vldrepl_h(filter, 0);

  src0 = __lsx_vld(src, 0);
  DUP4_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src, src_stride3,
            src, src_stride4, src1, src2, src3, src4);
  DUP2_ARG2(__lsx_vldx, src_tmp1, src_stride, src_tmp1, src_stride2, src5,
            src6);
  src7 = __lsx_vldx(src_tmp1, src_stride3);

  DUP4_ARG3(__lsx_vshuf_b, src1, src0, mask, src3, src2, mask, src5, src4, mask,
            src7, src6, mask, vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, filt0,
            vec4, vec5, vec6, vec7);
  DUP4_ARG2(__lsx_vsrari_h, vec4, FILTER_BITS, vec5, FILTER_BITS, vec6,
            FILTER_BITS, vec7, FILTER_BITS, vec4, vec5, vec6, vec7);
  DUP4_ARG2(__lsx_vpickev_b, vec4, vec4, vec5, vec5, vec6, vec6, vec7, vec7,
            res0, res1, res2, res3);
  __lsx_vstelm_w(res0, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_w(res0, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_w(res1, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_w(res1, dst, 0, 1);
  dst += dst_stride;

  __lsx_vstelm_w(res2, dst, 0, 0);
  __lsx_vstelm_w(res2, dst + dst_stride, 0, 1);
  __lsx_vstelm_w(res3, dst + dst_stride2, 0, 0);
  __lsx_vstelm_w(res3, dst + dst_stride3, 0, 1);
}

static void common_hz_2t_4w_lsx(const uint8_t *src, int32_t src_stride,
                                uint8_t *dst, int32_t dst_stride,
                                int8_t *filter, int32_t height) {
  if (height == 4) {
    common_hz_2t_4x4_lsx(src, src_stride, dst, dst_stride, filter);
  } else if (height == 8) {
    common_hz_2t_4x8_lsx(src, src_stride, dst, dst_stride, filter);
  }
}

static void common_hz_2t_8x4_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 int8_t *filter) {
  __m128i filt0, mask;
  __m128i src0, src1, src2, src3;
  __m128i vec0, vec1, vec2, vec3;

  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride2 + src_stride;

  int32_t dst_stride2 = dst_stride << 1;
  int32_t dst_stride3 = dst_stride2 + dst_stride;

  mask = __lsx_vld(mc_filt_mask_arr, 0);

  /* rearranging filter */
  filt0 = __lsx_vldrepl_h(filter, 0);

  src0 = __lsx_vld(src, 0);
  DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2);
  src3 = __lsx_vldx(src, src_stride3);

  DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2, mask,
            src3, src3, mask, vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, filt0,
            vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vsrari_h, vec0, FILTER_BITS, vec1, FILTER_BITS, vec2,
            FILTER_BITS, vec3, FILTER_BITS, vec0, vec1, vec2, vec3);
  DUP2_ARG2(__lsx_vpickev_b, vec1, vec0, vec3, vec2, src0, src1);

  __lsx_vstelm_d(src0, dst, 0, 0);
  __lsx_vstelm_d(src0, dst + dst_stride, 0, 1);
  __lsx_vstelm_d(src1, dst + dst_stride2, 0, 0);
  __lsx_vstelm_d(src1, dst + dst_stride3, 0, 1);
}

static void common_hz_2t_8x8mult_lsx(const uint8_t *src, int32_t src_stride,
                                     uint8_t *dst, int32_t dst_stride,
                                     int8_t *filter, int32_t height) {
  __m128i filt0, mask;
  __m128i src0, src1, src2, src3, out0, out1;
  __m128i vec0, vec1, vec2, vec3;

  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride2 + src_stride;
  int32_t src_stride4 = src_stride2 << 1;

  int32_t dst_stride2 = dst_stride << 1;
  int32_t dst_stride3 = dst_stride2 + dst_stride;
  int32_t dst_stride4 = dst_stride2 << 1;

  mask = __lsx_vld(mc_filt_mask_arr, 0);

  /* rearranging filter */
  filt0 = __lsx_vldrepl_h(filter, 0);

  src0 = __lsx_vld(src, 0);
  DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2);
  src3 = __lsx_vldx(src, src_stride3);
  src += src_stride4;

  DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2, mask,
            src3, src3, mask, vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, filt0,
            vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vsrari_h, vec0, FILTER_BITS, vec1, FILTER_BITS, vec2,
            FILTER_BITS, vec3, FILTER_BITS, vec0, vec1, vec2, vec3);

  src0 = __lsx_vld(src, 0);
  DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2);
  src3 = __lsx_vldx(src, src_stride3);
  src += src_stride4;

  DUP2_ARG2(__lsx_vpickev_b, vec1, vec0, vec3, vec2, out0, out1);
  __lsx_vstelm_d(out0, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_d(out0, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_d(out1, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_d(out1, dst, 0, 1);
  dst += dst_stride;

  DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2, mask,
            src3, src3, mask, vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, filt0,
            vec0, vec1, vec2, vec3);
  DUP4_ARG2(__lsx_vsrari_h, vec0, FILTER_BITS, vec1, FILTER_BITS, vec2,
            FILTER_BITS, vec3, FILTER_BITS, vec0, vec1, vec2, vec3);
  DUP2_ARG2(__lsx_vpickev_b, vec1, vec0, vec3, vec2, out0, out1);

  __lsx_vstelm_d(out0, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_d(out0, dst, 0, 1);
  dst += dst_stride;
  __lsx_vstelm_d(out1, dst, 0, 0);
  dst += dst_stride;
  __lsx_vstelm_d(out1, dst, 0, 1);
  dst += dst_stride;

  if (height == 16) {
    uint8_t *dst_tmp1 = dst + dst_stride4;

    src0 = __lsx_vld(src, 0);
    DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2);
    src3 = __lsx_vldx(src, src_stride3);
    src += src_stride4;

    DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2,
              mask, src3, src3, mask, vec0, vec1, vec2, vec3);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
              filt0, vec0, vec1, vec2, vec3);
    DUP4_ARG2(__lsx_vsrari_h, vec0, FILTER_BITS, vec1, FILTER_BITS, vec2,
              FILTER_BITS, vec3, FILTER_BITS, vec0, vec1, vec2, vec3);

    src0 = __lsx_vld(src, 0);
    DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src1, src2);
    src3 = __lsx_vldx(src, src_stride3);
    src += src_stride4;

    DUP2_ARG2(__lsx_vpickev_b, vec1, vec0, vec3, vec2, out0, out1);
    __lsx_vstelm_d(out0, dst, 0, 0);
    __lsx_vstelm_d(out0, dst + dst_stride, 0, 1);
    __lsx_vstelm_d(out1, dst + dst_stride2, 0, 0);
    __lsx_vstelm_d(out1, dst + dst_stride3, 0, 1);

    DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2,
              mask, src3, src3, mask, vec0, vec1, vec2, vec3);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
              filt0, vec0, vec1, vec2, vec3);
    DUP4_ARG2(__lsx_vsrari_h, vec0, FILTER_BITS, vec1, FILTER_BITS, vec2,
              FILTER_BITS, vec3, FILTER_BITS, vec0, vec1, vec2, vec3);
    DUP2_ARG2(__lsx_vpickev_b, vec1, vec0, vec3, vec2, out0, out1);

    __lsx_vstelm_d(out0, dst_tmp1, 0, 0);
    __lsx_vstelm_d(out0, dst_tmp1 + dst_stride, 0, 1);
    __lsx_vstelm_d(out1, dst_tmp1 + dst_stride2, 0, 0);
    __lsx_vstelm_d(out1, dst_tmp1 + dst_stride3, 0, 1);
  }
}

static void common_hz_2t_8w_lsx(const uint8_t *src, int32_t src_stride,
                                uint8_t *dst, int32_t dst_stride,
                                int8_t *filter, int32_t height) {
  if (height == 4) {
    common_hz_2t_8x4_lsx(src, src_stride, dst, dst_stride, filter);
  } else {
    common_hz_2t_8x8mult_lsx(src, src_stride, dst, dst_stride, filter, height);
  }
}

static void common_hz_2t_16w_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 int8_t *filter, int32_t height) {
  uint32_t loop_cnt = (height >> 2) - 1;
  __m128i src0, src1, src2, src3, src4, src5, src6, src7, mask;
  __m128i filt0, vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
  __m128i out0, out1, out2, out3, out4, out5, out6, out7, tmp;

  int32_t src_stride2 = src_stride << 1;
  int32_t src_stride3 = src_stride2 + src_stride;
  int32_t src_stride4 = src_stride2 << 1;

  uint8_t *src_tmp1 = src + 8;
  mask = __lsx_vld(mc_filt_mask_arr, 0);
  filt0 = __lsx_vldrepl_h(filter, 0);

  src0 = __lsx_vld(src, 0);
  DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src2, src4);
  src6 = __lsx_vldx(src, src_stride3);
  src1 = __lsx_vld(src_tmp1, 0);
  DUP2_ARG2(__lsx_vldx, src_tmp1, src_stride, src_tmp1, src_stride2, src3,
            src5);
  src7 = __lsx_vldx(src_tmp1, src_stride3);
  src += src_stride4;

  DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2, mask,
            src3, src3, mask, vec0, vec1, vec2, vec3);
  DUP4_ARG3(__lsx_vshuf_b, src4, src4, mask, src5, src5, mask, src6, src6, mask,
            src7, src7, mask, vec4, vec5, vec6, vec7);
  DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3, filt0,
            out0, out1, out2, out3);
  DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7, filt0,
            out4, out5, out6, out7);
  DUP4_ARG2(__lsx_vsrari_h, out0, FILTER_BITS, out1, FILTER_BITS, out2,
            FILTER_BITS, out3, FILTER_BITS, out0, out1, out2, out3);
  DUP4_ARG2(__lsx_vsrari_h, out4, FILTER_BITS, out5, FILTER_BITS, out6,
            FILTER_BITS, out7, FILTER_BITS, out4, out5, out6, out7);

  tmp = __lsx_vpickev_b(out1, out0);
  __lsx_vst(tmp, dst, 0);
  dst += dst_stride;
  tmp = __lsx_vpickev_b(out3, out2);
  __lsx_vst(tmp, dst, 0);
  dst += dst_stride;
  tmp = __lsx_vpickev_b(out5, out4);
  __lsx_vst(tmp, dst, 0);
  dst += dst_stride;
  tmp = __lsx_vpickev_b(out7, out6);
  __lsx_vst(tmp, dst, 0);
  dst += dst_stride;

  for (; loop_cnt--;) {
    src_tmp1 += src_stride4;

    src0 = __lsx_vld(src, 0);
    DUP2_ARG2(__lsx_vldx, src, src_stride, src, src_stride2, src2, src4);
    src6 = __lsx_vldx(src, src_stride3);

    src1 = __lsx_vld(src_tmp1, 0);
    DUP2_ARG2(__lsx_vldx, src_tmp1, src_stride, src_tmp1, src_stride2, src3,
              src5);
    src7 = __lsx_vldx(src_tmp1, src_stride3);
    src += src_stride4;

    DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2,
              mask, src3, src3, mask, vec0, vec1, vec2, vec3);
    DUP4_ARG3(__lsx_vshuf_b, src4, src4, mask, src5, src5, mask, src6, src6,
              mask, src7, src7, mask, vec4, vec5, vec6, vec7);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
              filt0, out0, out1, out2, out3);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7,
              filt0, out4, out5, out6, out7);
    DUP4_ARG2(__lsx_vsrari_h, out0, FILTER_BITS, out1, FILTER_BITS, out2,
              FILTER_BITS, out3, FILTER_BITS, out0, out1, out2, out3);
    DUP4_ARG2(__lsx_vsrari_h, out4, FILTER_BITS, out5, FILTER_BITS, out6,
              FILTER_BITS, out7, FILTER_BITS, out4, out5, out6, out7);

    tmp = __lsx_vpickev_b(out1, out0);
    __lsx_vst(tmp, dst, 0);
    dst += dst_stride;
    tmp = __lsx_vpickev_b(out3, out2);
    __lsx_vst(tmp, dst, 0);
    dst += dst_stride;
    tmp = __lsx_vpickev_b(out5, out4);
    __lsx_vst(tmp, dst, 0);
    dst += dst_stride;
    tmp = __lsx_vpickev_b(out7, out6);
    __lsx_vst(tmp, dst, 0);
    dst += dst_stride;
  }
}

static void common_hz_2t_32w_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 int8_t *filter, int32_t height) {
  uint32_t loop_cnt = (height >> 1);
  __m128i src0, src1, src2, src3, src4, src5, src6, src7, mask;
  __m128i filt0, vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
  __m128i out0, out1, out2, out3, out4, out5, out6, out7, tmp;
  __m128i shuff = { 0x0F0E0D0C0B0A0908, 0x1716151413121110 };

  mask = __lsx_vld(mc_filt_mask_arr, 0);
  /* rearranging filter */
  filt0 = __lsx_vldrepl_h(filter, 0);

  for (; loop_cnt--;) {
    DUP2_ARG2(__lsx_vld, src, 0, src, 16, src0, src2);
    src3 = __lsx_vld(src, 24);
    src1 = __lsx_vshuf_b(src2, src0, shuff);
    src += src_stride;
    DUP2_ARG2(__lsx_vld, src, 0, src, 16, src4, src6);
    src7 = __lsx_vld(src, 24);
    src5 = __lsx_vshuf_b(src6, src4, shuff);
    src += src_stride;

    DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2,
              mask, src3, src3, mask, vec0, vec1, vec2, vec3);
    DUP4_ARG3(__lsx_vshuf_b, src4, src4, mask, src5, src5, mask, src6, src6,
              mask, src7, src7, mask, vec4, vec5, vec6, vec7);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
              filt0, out0, out1, out2, out3);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7,
              filt0, out4, out5, out6, out7);
    DUP4_ARG2(__lsx_vsrari_h, out0, FILTER_BITS, out1, FILTER_BITS, out2,
              FILTER_BITS, out3, FILTER_BITS, out0, out1, out2, out3);
    DUP4_ARG2(__lsx_vsrari_h, out4, FILTER_BITS, out5, FILTER_BITS, out6,
              FILTER_BITS, out7, FILTER_BITS, out4, out5, out6, out7);

    tmp = __lsx_vpickev_b(out1, out0);
    __lsx_vst(tmp, dst, 0);
    tmp = __lsx_vpickev_b(out3, out2);
    __lsx_vst(tmp, dst, 16);
    dst += dst_stride;

    tmp = __lsx_vpickev_b(out5, out4);
    __lsx_vst(tmp, dst, 0);
    tmp = __lsx_vpickev_b(out7, out6);
    __lsx_vst(tmp, dst, 16);
    dst += dst_stride;
  }
}

static void common_hz_2t_64w_lsx(const uint8_t *src, int32_t src_stride,
                                 uint8_t *dst, int32_t dst_stride,
                                 int8_t *filter, int32_t height) {
  uint32_t loop_cnt = height;
  __m128i src0, src1, src2, src3, src4, src5, src6, src7, mask;
  __m128i filt0, vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
  __m128i out0, out1, out2, out3, out4, out5, out6, out7, tmp;
  __m128i shuff = { 0x0F0E0D0C0B0A0908, 0x1716151413121110 };

  mask = __lsx_vld(mc_filt_mask_arr, 0);

  /* rearranging filter */
  filt0 = __lsx_vldrepl_h(filter, 0);

  for (; loop_cnt--;) {
    DUP4_ARG2(__lsx_vld, src, 0, src, 16, src, 32, src, 48, src0, src2, src4,
              src6);
    src7 = __lsx_vld(src, 56);
    DUP2_ARG3(__lsx_vshuf_b, src2, src0, shuff, src4, src2, shuff, src1, src3);
    src5 = __lsx_vshuf_b(src6, src4, shuff);
    src += src_stride;

    DUP4_ARG3(__lsx_vshuf_b, src0, src0, mask, src1, src1, mask, src2, src2,
              mask, src3, src3, mask, vec0, vec1, vec2, vec3);
    DUP4_ARG3(__lsx_vshuf_b, src4, src4, mask, src5, src5, mask, src6, src6,
              mask, src7, src7, mask, vec4, vec5, vec6, vec7);

    DUP4_ARG2(__lsx_vdp2_h_bu, vec0, filt0, vec1, filt0, vec2, filt0, vec3,
              filt0, out0, out1, out2, out3);
    DUP4_ARG2(__lsx_vdp2_h_bu, vec4, filt0, vec5, filt0, vec6, filt0, vec7,
              filt0, out4, out5, out6, out7);
    DUP4_ARG2(__lsx_vsrari_h, out0, FILTER_BITS, out1, FILTER_BITS, out2,
              FILTER_BITS, out3, FILTER_BITS, out0, out1, out2, out3);
    DUP4_ARG2(__lsx_vsrari_h, out4, FILTER_BITS, out5, FILTER_BITS, out6,
              FILTER_BITS, out7, FILTER_BITS, out4, out5, out6, out7);

    tmp = __lsx_vpickev_b(out1, out0);
    __lsx_vst(tmp, dst, 0);
    tmp = __lsx_vpickev_b(out3, out2);
    __lsx_vst(tmp, dst, 16);
    tmp = __lsx_vpickev_b(out5, out4);
    __lsx_vst(tmp, dst, 32);
    tmp = __lsx_vpickev_b(out7, out6);
    __lsx_vst(tmp, dst, 48);
    dst += dst_stride;
  }
}

void vpx_convolve8_horiz_lsx(const uint8_t *src, ptrdiff_t src_stride,
                             uint8_t *dst, ptrdiff_t dst_stride,
                             const InterpKernel *filter, int x0_q4,
                             int x_step_q4, int y0_q4, int y_step_q4, int w,
                             int h) {
  const int16_t *const filter_x = filter[x0_q4];
  int8_t cnt, filt_hor[8];

  assert(x_step_q4 == 16);
  assert(((const int32_t *)filter_x)[1] != 0x800000);

  for (cnt = 0; cnt < 8; ++cnt) {
    filt_hor[cnt] = filter_x[cnt];
  }
  if (vpx_get_filter_taps(filter_x) == 2) {
    switch (w) {
      case 4:
        common_hz_2t_4w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                            &filt_hor[3], h);
        break;
      case 8:
        common_hz_2t_8w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                            &filt_hor[3], h);
        break;
      case 16:
        common_hz_2t_16w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                             &filt_hor[3], h);
        break;
      case 32:
        common_hz_2t_32w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                             &filt_hor[3], h);
        break;
      case 64:
        common_hz_2t_64w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                             &filt_hor[3], h);
        break;
      default:
        vpx_convolve8_horiz_c(src, src_stride, dst, dst_stride, filter, x0_q4,
                              x_step_q4, y0_q4, y_step_q4, w, h);
        break;
    }
  } else {
    switch (w) {
      case 4:
        common_hz_8t_4w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                            filt_hor, h);
        break;
      case 8:
        common_hz_8t_8w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                            filt_hor, h);
        break;

      case 16:
        common_hz_8t_16w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                             filt_hor, h);
        break;

      case 32:
        common_hz_8t_32w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                             filt_hor, h);
        break;

      case 64:
        common_hz_8t_64w_lsx(src, (int32_t)src_stride, dst, (int32_t)dst_stride,
                             filt_hor, h);
        break;
      default:
        vpx_convolve8_horiz_c(src, src_stride, dst, dst_stride, filter, x0_q4,
                              x_step_q4, y0_q4, y_step_q4, w, h);
        break;
    }
  }
}