/*
 * Copyright (c) 2016, Alliance for Open Media. All rights reserved
 *
 * This source code is subject to the terms of the BSD 2 Clause License and
 * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
 * was not distributed with this source code in the LICENSE file, you can
 * obtain it at www.aomedia.org/license/software. If the Alliance for Open
 * Media Patent License 1.0 was not distributed with this source code in the
 * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
 */

#ifndef AOM_DSP_MIPS_FWD_TXFM_MSA_H_
#define AOM_DSP_MIPS_FWD_TXFM_MSA_H_

#include "aom_dsp/mips/txfm_macros_msa.h"
#include "aom_dsp/txfm_common.h"

#define LD_HADD(psrc, stride)                                                  \
  ({                                                                           \
    v8i16 in0_m, in1_m, in2_m, in3_m, in4_m, in5_m, in6_m, in7_m;              \
    v4i32 vec_w_m;                                                             \
                                                                               \
    LD_SH4((psrc), stride, in0_m, in1_m, in2_m, in3_m);                        \
    ADD2(in0_m, in1_m, in2_m, in3_m, in0_m, in2_m);                            \
    LD_SH4(((psrc) + 4 * stride), stride, in4_m, in5_m, in6_m, in7_m);         \
    ADD4(in4_m, in5_m, in6_m, in7_m, in0_m, in2_m, in4_m, in6_m, in4_m, in6_m, \
         in0_m, in4_m);                                                        \
    in0_m += in4_m;                                                            \
                                                                               \
    vec_w_m = __msa_hadd_s_w(in0_m, in0_m);                                    \
    HADD_SW_S32(vec_w_m);                                                      \
  })

#define AOM_FDCT4(in0, in1, in2, in3, out0, out1, out2, out3)                  \
  {                                                                            \
    v8i16 cnst0_m, cnst1_m, cnst2_m, cnst3_m;                                  \
    v8i16 vec0_m, vec1_m, vec2_m, vec3_m;                                      \
    v4i32 vec4_m, vec5_m, vec6_m, vec7_m;                                      \
    v8i16 coeff_m = {                                                          \
      cospi_16_64, -cospi_16_64, cospi_8_64, cospi_24_64, -cospi_8_64, 0, 0, 0 \
    };                                                                         \
                                                                               \
    BUTTERFLY_4(in0, in1, in2, in3, vec0_m, vec1_m, vec2_m, vec3_m);           \
    ILVR_H2_SH(vec1_m, vec0_m, vec3_m, vec2_m, vec0_m, vec2_m);                \
    SPLATI_H2_SH(coeff_m, 0, 1, cnst0_m, cnst1_m);                             \
    cnst1_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
    vec5_m = __msa_dotp_s_w(vec0_m, cnst1_m);                                  \
                                                                               \
    SPLATI_H2_SH(coeff_m, 4, 3, cnst2_m, cnst3_m);                             \
    cnst2_m = __msa_ilvev_h(cnst3_m, cnst2_m);                                 \
    vec7_m = __msa_dotp_s_w(vec2_m, cnst2_m);                                  \
                                                                               \
    vec4_m = __msa_dotp_s_w(vec0_m, cnst0_m);                                  \
    cnst2_m = __msa_splati_h(coeff_m, 2);                                      \
    cnst2_m = __msa_ilvev_h(cnst2_m, cnst3_m);                                 \
    vec6_m = __msa_dotp_s_w(vec2_m, cnst2_m);                                  \
                                                                               \
    SRARI_W4_SW(vec4_m, vec5_m, vec6_m, vec7_m, DCT_CONST_BITS);               \
    PCKEV_H4_SH(vec4_m, vec4_m, vec5_m, vec5_m, vec6_m, vec6_m, vec7_m,        \
                vec7_m, out0, out2, out1, out3);                               \
  }

#define SRLI_AVE_S_4V_H(in0, in1, in2, in3, in4, in5, in6, in7)              \
  {                                                                          \
    v8i16 vec0_m, vec1_m, vec2_m, vec3_m, vec4_m, vec5_m, vec6_m, vec7_m;    \
                                                                             \
    SRLI_H4_SH(in0, in1, in2, in3, vec0_m, vec1_m, vec2_m, vec3_m, 15);      \
    SRLI_H4_SH(in4, in5, in6, in7, vec4_m, vec5_m, vec6_m, vec7_m, 15);      \
    AVE_SH4_SH(vec0_m, in0, vec1_m, in1, vec2_m, in2, vec3_m, in3, in0, in1, \
               in2, in3);                                                    \
    AVE_SH4_SH(vec4_m, in4, vec5_m, in5, vec6_m, in6, vec7_m, in7, in4, in5, \
               in6, in7);                                                    \
  }

#define AOM_FDCT8(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1, out2,  \
                  out3, out4, out5, out6, out7)                              \
  {                                                                          \
    v8i16 s0_m, s1_m, s2_m, s3_m, s4_m, s5_m, s6_m;                          \
    v8i16 s7_m, x0_m, x1_m, x2_m, x3_m;                                      \
    v8i16 coeff_m = { cospi_16_64, -cospi_16_64, cospi_8_64,  cospi_24_64,   \
                      cospi_4_64,  cospi_28_64,  cospi_12_64, cospi_20_64 }; \
                                                                             \
    /* FDCT stage1 */                                                        \
    BUTTERFLY_8(in0, in1, in2, in3, in4, in5, in6, in7, s0_m, s1_m, s2_m,    \
                s3_m, s4_m, s5_m, s6_m, s7_m);                               \
    BUTTERFLY_4(s0_m, s1_m, s2_m, s3_m, x0_m, x1_m, x2_m, x3_m);             \
    ILVL_H2_SH(x1_m, x0_m, x3_m, x2_m, s0_m, s2_m);                          \
    ILVR_H2_SH(x1_m, x0_m, x3_m, x2_m, s1_m, s3_m);                          \
    SPLATI_H2_SH(coeff_m, 0, 1, x0_m, x1_m);                                 \
    x1_m = __msa_ilvev_h(x1_m, x0_m);                                        \
    out4 = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x1_m);                          \
                                                                             \
    SPLATI_H2_SH(coeff_m, 2, 3, x2_m, x3_m);                                 \
    x2_m = -x2_m;                                                            \
    x2_m = __msa_ilvev_h(x3_m, x2_m);                                        \
    out6 = DOT_SHIFT_RIGHT_PCK_H(s2_m, s3_m, x2_m);                          \
                                                                             \
    out0 = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x0_m);                          \
    x2_m = __msa_splati_h(coeff_m, 2);                                       \
    x2_m = __msa_ilvev_h(x2_m, x3_m);                                        \
    out2 = DOT_SHIFT_RIGHT_PCK_H(s2_m, s3_m, x2_m);                          \
                                                                             \
    /* stage2 */                                                             \
    ILVRL_H2_SH(s5_m, s6_m, s1_m, s0_m);                                     \
                                                                             \
    s6_m = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x0_m);                          \
    s5_m = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x1_m);                          \
                                                                             \
    /* stage3 */                                                             \
    BUTTERFLY_4(s4_m, s7_m, s6_m, s5_m, x0_m, x3_m, x2_m, x1_m);             \
                                                                             \
    /* stage4 */                                                             \
    ILVL_H2_SH(x3_m, x0_m, x2_m, x1_m, s4_m, s6_m);                          \
    ILVR_H2_SH(x3_m, x0_m, x2_m, x1_m, s5_m, s7_m);                          \
                                                                             \
    SPLATI_H2_SH(coeff_m, 4, 5, x0_m, x1_m);                                 \
    x1_m = __msa_ilvev_h(x0_m, x1_m);                                        \
    out1 = DOT_SHIFT_RIGHT_PCK_H(s4_m, s5_m, x1_m);                          \
                                                                             \
    SPLATI_H2_SH(coeff_m, 6, 7, x2_m, x3_m);                                 \
    x2_m = __msa_ilvev_h(x3_m, x2_m);                                        \
    out5 = DOT_SHIFT_RIGHT_PCK_H(s6_m, s7_m, x2_m);                          \
                                                                             \
    x1_m = __msa_splati_h(coeff_m, 5);                                       \
    x0_m = -x0_m;                                                            \
    x0_m = __msa_ilvev_h(x1_m, x0_m);                                        \
    out7 = DOT_SHIFT_RIGHT_PCK_H(s4_m, s5_m, x0_m);                          \
                                                                             \
    x2_m = __msa_splati_h(coeff_m, 6);                                       \
    x3_m = -x3_m;                                                            \
    x2_m = __msa_ilvev_h(x2_m, x3_m);                                        \
    out3 = DOT_SHIFT_RIGHT_PCK_H(s6_m, s7_m, x2_m);                          \
  }

#define FDCT8x16_EVEN(in0, in1, in2, in3, in4, in5, in6, in7, out0, out1,    \
                      out2, out3, out4, out5, out6, out7)                    \
  {                                                                          \
    v8i16 s0_m, s1_m, s2_m, s3_m, s4_m, s5_m, s6_m, s7_m;                    \
    v8i16 x0_m, x1_m, x2_m, x3_m;                                            \
    v8i16 coeff_m = { cospi_16_64, -cospi_16_64, cospi_8_64,  cospi_24_64,   \
                      cospi_4_64,  cospi_28_64,  cospi_12_64, cospi_20_64 }; \
                                                                             \
    /* FDCT stage1 */                                                        \
    BUTTERFLY_8(in0, in1, in2, in3, in4, in5, in6, in7, s0_m, s1_m, s2_m,    \
                s3_m, s4_m, s5_m, s6_m, s7_m);                               \
    BUTTERFLY_4(s0_m, s1_m, s2_m, s3_m, x0_m, x1_m, x2_m, x3_m);             \
    ILVL_H2_SH(x1_m, x0_m, x3_m, x2_m, s0_m, s2_m);                          \
    ILVR_H2_SH(x1_m, x0_m, x3_m, x2_m, s1_m, s3_m);                          \
    SPLATI_H2_SH(coeff_m, 0, 1, x0_m, x1_m);                                 \
    x1_m = __msa_ilvev_h(x1_m, x0_m);                                        \
    out4 = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x1_m);                          \
                                                                             \
    SPLATI_H2_SH(coeff_m, 2, 3, x2_m, x3_m);                                 \
    x2_m = -x2_m;                                                            \
    x2_m = __msa_ilvev_h(x3_m, x2_m);                                        \
    out6 = DOT_SHIFT_RIGHT_PCK_H(s2_m, s3_m, x2_m);                          \
                                                                             \
    out0 = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x0_m);                          \
    x2_m = __msa_splati_h(coeff_m, 2);                                       \
    x2_m = __msa_ilvev_h(x2_m, x3_m);                                        \
    out2 = DOT_SHIFT_RIGHT_PCK_H(s2_m, s3_m, x2_m);                          \
                                                                             \
    /* stage2 */                                                             \
    ILVRL_H2_SH(s5_m, s6_m, s1_m, s0_m);                                     \
                                                                             \
    s6_m = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x0_m);                          \
    s5_m = DOT_SHIFT_RIGHT_PCK_H(s0_m, s1_m, x1_m);                          \
                                                                             \
    /* stage3 */                                                             \
    BUTTERFLY_4(s4_m, s7_m, s6_m, s5_m, x0_m, x3_m, x2_m, x1_m);             \
                                                                             \
    /* stage4 */                                                             \
    ILVL_H2_SH(x3_m, x0_m, x2_m, x1_m, s4_m, s6_m);                          \
    ILVR_H2_SH(x3_m, x0_m, x2_m, x1_m, s5_m, s7_m);                          \
                                                                             \
    SPLATI_H2_SH(coeff_m, 4, 5, x0_m, x1_m);                                 \
    x1_m = __msa_ilvev_h(x0_m, x1_m);                                        \
    out1 = DOT_SHIFT_RIGHT_PCK_H(s4_m, s5_m, x1_m);                          \
                                                                             \
    SPLATI_H2_SH(coeff_m, 6, 7, x2_m, x3_m);                                 \
    x2_m = __msa_ilvev_h(x3_m, x2_m);                                        \
    out5 = DOT_SHIFT_RIGHT_PCK_H(s6_m, s7_m, x2_m);                          \
                                                                             \
    x1_m = __msa_splati_h(coeff_m, 5);                                       \
    x0_m = -x0_m;                                                            \
    x0_m = __msa_ilvev_h(x1_m, x0_m);                                        \
    out7 = DOT_SHIFT_RIGHT_PCK_H(s4_m, s5_m, x0_m);                          \
                                                                             \
    x2_m = __msa_splati_h(coeff_m, 6);                                       \
    x3_m = -x3_m;                                                            \
    x2_m = __msa_ilvev_h(x2_m, x3_m);                                        \
    out3 = DOT_SHIFT_RIGHT_PCK_H(s6_m, s7_m, x2_m);                          \
  }

#define FDCT8x16_ODD(input0, input1, input2, input3, input4, input5, input6,   \
                     input7, out1, out3, out5, out7, out9, out11, out13,       \
                     out15)                                                    \
  {                                                                            \
    v8i16 stp21_m, stp22_m, stp23_m, stp24_m, stp25_m, stp26_m;                \
    v8i16 stp30_m, stp31_m, stp32_m, stp33_m, stp34_m, stp35_m;                \
    v8i16 stp36_m, stp37_m, vec0_m, vec1_m;                                    \
    v8i16 vec2_m, vec3_m, vec4_m, vec5_m, vec6_m;                              \
    v8i16 cnst0_m, cnst1_m, cnst4_m, cnst5_m;                                  \
    v8i16 coeff_m = { cospi_16_64, -cospi_16_64, cospi_8_64,  cospi_24_64,     \
                      -cospi_8_64, -cospi_24_64, cospi_12_64, cospi_20_64 };   \
    v8i16 coeff1_m = { cospi_2_64,  cospi_30_64, cospi_14_64, cospi_18_64,     \
                       cospi_10_64, cospi_22_64, cospi_6_64,  cospi_26_64 };   \
    v8i16 coeff2_m = {                                                         \
      -cospi_2_64, -cospi_10_64, -cospi_18_64, -cospi_26_64, 0, 0, 0, 0        \
    };                                                                         \
                                                                               \
    /* stp 1 */                                                                \
    ILVL_H2_SH(input2, input5, input3, input4, vec2_m, vec4_m);                \
    ILVR_H2_SH(input2, input5, input3, input4, vec3_m, vec5_m);                \
                                                                               \
    cnst4_m = __msa_splati_h(coeff_m, 0);                                      \
    stp25_m = DOT_SHIFT_RIGHT_PCK_H(vec2_m, vec3_m, cnst4_m);                  \
                                                                               \
    cnst5_m = __msa_splati_h(coeff_m, 1);                                      \
    cnst5_m = __msa_ilvev_h(cnst5_m, cnst4_m);                                 \
    stp22_m = DOT_SHIFT_RIGHT_PCK_H(vec2_m, vec3_m, cnst5_m);                  \
    stp24_m = DOT_SHIFT_RIGHT_PCK_H(vec4_m, vec5_m, cnst4_m);                  \
    stp23_m = DOT_SHIFT_RIGHT_PCK_H(vec4_m, vec5_m, cnst5_m);                  \
                                                                               \
    /* stp2 */                                                                 \
    BUTTERFLY_4(input0, input1, stp22_m, stp23_m, stp30_m, stp31_m, stp32_m,   \
                stp33_m);                                                      \
    BUTTERFLY_4(input7, input6, stp25_m, stp24_m, stp37_m, stp36_m, stp35_m,   \
                stp34_m);                                                      \
                                                                               \
    ILVL_H2_SH(stp36_m, stp31_m, stp35_m, stp32_m, vec2_m, vec4_m);            \
    ILVR_H2_SH(stp36_m, stp31_m, stp35_m, stp32_m, vec3_m, vec5_m);            \
                                                                               \
    SPLATI_H2_SH(coeff_m, 2, 3, cnst0_m, cnst1_m);                             \
    cnst0_m = __msa_ilvev_h(cnst0_m, cnst1_m);                                 \
    stp26_m = DOT_SHIFT_RIGHT_PCK_H(vec2_m, vec3_m, cnst0_m);                  \
                                                                               \
    cnst0_m = __msa_splati_h(coeff_m, 4);                                      \
    cnst1_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
    stp21_m = DOT_SHIFT_RIGHT_PCK_H(vec2_m, vec3_m, cnst1_m);                  \
                                                                               \
    SPLATI_H2_SH(coeff_m, 5, 2, cnst0_m, cnst1_m);                             \
    cnst1_m = __msa_ilvev_h(cnst0_m, cnst1_m);                                 \
    stp25_m = DOT_SHIFT_RIGHT_PCK_H(vec4_m, vec5_m, cnst1_m);                  \
                                                                               \
    cnst0_m = __msa_splati_h(coeff_m, 3);                                      \
    cnst1_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
    stp22_m = DOT_SHIFT_RIGHT_PCK_H(vec4_m, vec5_m, cnst1_m);                  \
                                                                               \
    /* stp4 */                                                                 \
    BUTTERFLY_4(stp30_m, stp37_m, stp26_m, stp21_m, vec6_m, vec2_m, vec4_m,    \
                vec5_m);                                                       \
    BUTTERFLY_4(stp33_m, stp34_m, stp25_m, stp22_m, stp21_m, stp23_m, stp24_m, \
                stp31_m);                                                      \
                                                                               \
    ILVRL_H2_SH(vec2_m, vec6_m, vec1_m, vec0_m);                               \
    SPLATI_H2_SH(coeff1_m, 0, 1, cnst0_m, cnst1_m);                            \
    cnst0_m = __msa_ilvev_h(cnst0_m, cnst1_m);                                 \
                                                                               \
    out1 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst0_m);                     \
                                                                               \
    cnst0_m = __msa_splati_h(coeff2_m, 0);                                     \
    cnst0_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
    out15 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst0_m);                    \
                                                                               \
    ILVRL_H2_SH(vec4_m, vec5_m, vec1_m, vec0_m);                               \
    SPLATI_H2_SH(coeff1_m, 2, 3, cnst0_m, cnst1_m);                            \
    cnst1_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
                                                                               \
    out9 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst1_m);                     \
                                                                               \
    cnst1_m = __msa_splati_h(coeff2_m, 2);                                     \
    cnst0_m = __msa_ilvev_h(cnst0_m, cnst1_m);                                 \
    out7 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst0_m);                     \
                                                                               \
    ILVRL_H2_SH(stp23_m, stp21_m, vec1_m, vec0_m);                             \
    SPLATI_H2_SH(coeff1_m, 4, 5, cnst0_m, cnst1_m);                            \
    cnst0_m = __msa_ilvev_h(cnst0_m, cnst1_m);                                 \
    out5 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst0_m);                     \
                                                                               \
    cnst0_m = __msa_splati_h(coeff2_m, 1);                                     \
    cnst0_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
    out11 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst0_m);                    \
                                                                               \
    ILVRL_H2_SH(stp24_m, stp31_m, vec1_m, vec0_m);                             \
    SPLATI_H2_SH(coeff1_m, 6, 7, cnst0_m, cnst1_m);                            \
    cnst1_m = __msa_ilvev_h(cnst1_m, cnst0_m);                                 \
                                                                               \
    out13 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst1_m);                    \
                                                                               \
    cnst1_m = __msa_splati_h(coeff2_m, 3);                                     \
    cnst0_m = __msa_ilvev_h(cnst0_m, cnst1_m);                                 \
    out3 = DOT_SHIFT_RIGHT_PCK_H(vec0_m, vec1_m, cnst0_m);                     \
  }

#define FDCT_POSTPROC_2V_NEG_H(vec0, vec1) \
  {                                        \
    v8i16 tp0_m, tp1_m;                    \
    v8i16 one_m = __msa_ldi_h(1);          \
                                           \
    tp0_m = __msa_clti_s_h(vec0, 0);       \
    tp1_m = __msa_clti_s_h(vec1, 0);       \
    vec0 += 1;                             \
    vec1 += 1;                             \
    tp0_m = one_m & tp0_m;                 \
    tp1_m = one_m & tp1_m;                 \
    vec0 += tp0_m;                         \
    vec1 += tp1_m;                         \
    vec0 >>= 2;                            \
    vec1 >>= 2;                            \
  }

#define FDCT32_POSTPROC_NEG_W(vec)   \
  {                                  \
    v4i32 temp_m;                    \
    v4i32 one_m = __msa_ldi_w(1);    \
                                     \
    temp_m = __msa_clti_s_w(vec, 0); \
    vec += 1;                        \
    temp_m = one_m & temp_m;         \
    vec += temp_m;                   \
    vec >>= 2;                       \
  }

#define FDCT32_POSTPROC_2V_POS_H(vec0, vec1)        \
  {                                                 \
    v8i16 tp0_m, tp1_m;                             \
    v8i16 one = __msa_ldi_h(1);                     \
                                                    \
    tp0_m = __msa_clei_s_h(vec0, 0);                \
    tp1_m = __msa_clei_s_h(vec1, 0);                \
    tp0_m = (v8i16)__msa_xori_b((v16u8)tp0_m, 255); \
    tp1_m = (v8i16)__msa_xori_b((v16u8)tp1_m, 255); \
    vec0 += 1;                                      \
    vec1 += 1;                                      \
    tp0_m = one & tp0_m;                            \
    tp1_m = one & tp1_m;                            \
    vec0 += tp0_m;                                  \
    vec1 += tp1_m;                                  \
    vec0 >>= 2;                                     \
    vec1 >>= 2;                                     \
  }

#define DOTP_CONST_PAIR_W(reg0_left, reg1_left, reg0_right, reg1_right, \
                          const0, const1, out0, out1, out2, out3)       \
  {                                                                     \
    v4i32 s0_m, s1_m, s2_m, s3_m, s4_m, s5_m, s6_m, s7_m;               \
    v2i64 tp0_m, tp1_m, tp2_m, tp3_m;                                   \
    v4i32 k0_m = __msa_fill_w((int32_t)const0);                         \
                                                                        \
    s0_m = __msa_fill_w((int32_t)const1);                               \
    k0_m = __msa_ilvev_w(s0_m, k0_m);                                   \
                                                                        \
    ILVRL_W2_SW(-reg1_left, reg0_left, s1_m, s0_m);                     \
    ILVRL_W2_SW(reg0_left, reg1_left, s3_m, s2_m);                      \
    ILVRL_W2_SW(-reg1_right, reg0_right, s5_m, s4_m);                   \
    ILVRL_W2_SW(reg0_right, reg1_right, s7_m, s6_m);                    \
                                                                        \
    DOTP_SW2_SD(s0_m, s1_m, k0_m, k0_m, tp0_m, tp1_m);                  \
    DOTP_SW2_SD(s4_m, s5_m, k0_m, k0_m, tp2_m, tp3_m);                  \
    tp0_m = __msa_srari_d(tp0_m, DCT_CONST_BITS);                       \
    tp1_m = __msa_srari_d(tp1_m, DCT_CONST_BITS);                       \
    tp2_m = __msa_srari_d(tp2_m, DCT_CONST_BITS);                       \
    tp3_m = __msa_srari_d(tp3_m, DCT_CONST_BITS);                       \
    out0 = __msa_pckev_w((v4i32)tp0_m, (v4i32)tp1_m);                   \
    out1 = __msa_pckev_w((v4i32)tp2_m, (v4i32)tp3_m);                   \
                                                                        \
    DOTP_SW2_SD(s2_m, s3_m, k0_m, k0_m, tp0_m, tp1_m);                  \
    DOTP_SW2_SD(s6_m, s7_m, k0_m, k0_m, tp2_m, tp3_m);                  \
    tp0_m = __msa_srari_d(tp0_m, DCT_CONST_BITS);                       \
    tp1_m = __msa_srari_d(tp1_m, DCT_CONST_BITS);                       \
    tp2_m = __msa_srari_d(tp2_m, DCT_CONST_BITS);                       \
    tp3_m = __msa_srari_d(tp3_m, DCT_CONST_BITS);                       \
    out2 = __msa_pckev_w((v4i32)tp0_m, (v4i32)tp1_m);                   \
    out3 = __msa_pckev_w((v4i32)tp2_m, (v4i32)tp3_m);                   \
  }

void fdct8x16_1d_column(const int16_t *input, int16_t *tmp_ptr,
                        int32_t src_stride);
void fdct16x8_1d_row(int16_t *input, int16_t *output);
#endif  // AOM_DSP_MIPS_FWD_TXFM_MSA_H_