/*****************************************************************************
 * pixel-c.c: msa pixel metrics
 *****************************************************************************
 * Copyright (C) 2015-2024 x264 project
 *
 * Authors: Mandar Sahastrabuddhe <mandar.sahastrabuddhe@imgtec.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
 *
 * This program is also available under a commercial proprietary license.
 * For more information, contact us at licensing@x264.com.
 *****************************************************************************/

#include "common/common.h"
#include "macros.h"
#include "pixel.h"
#include "predict.h"

#if !HIGH_BIT_DEPTH
#define CALC_MSE_B( src, ref, var )                                    \
{                                                                      \
    v16u8 src_l0_m, src_l1_m;                                          \
    v8i16 res_l0_m, res_l1_m;                                          \
                                                                       \
    ILVRL_B2_UB( src, ref, src_l0_m, src_l1_m );                       \
    HSUB_UB2_SH( src_l0_m, src_l1_m, res_l0_m, res_l1_m );             \
    DPADD_SH2_SW( res_l0_m, res_l1_m, res_l0_m, res_l1_m, var, var );  \
}

#define CALC_MSE_AVG_B( src, ref, var, sub )                           \
{                                                                      \
    v16u8 src_l0_m, src_l1_m;                                          \
    v8i16 res_l0_m, res_l1_m;                                          \
                                                                       \
    ILVRL_B2_UB( src, ref, src_l0_m, src_l1_m );                       \
    HSUB_UB2_SH( src_l0_m, src_l1_m, res_l0_m, res_l1_m );             \
    DPADD_SH2_SW( res_l0_m, res_l1_m, res_l0_m, res_l1_m, var, var );  \
                                                                       \
    sub += res_l0_m + res_l1_m;                                        \
}

#define VARIANCE_WxH( sse, diff, shift )                                \
    ( ( sse ) - ( ( ( uint32_t )( diff ) * ( diff ) ) >> ( shift ) ) )

static uint32_t sad_4width_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_ref, int32_t i_ref_stride,
                                int32_t i_height )
{
    int32_t i_ht_cnt;
    uint32_t u_src0, u_src1, u_src2, u_src3, u_ref0, u_ref1, u_ref2, u_ref3;
    v16u8 src = { 0 };
    v16u8 ref = { 0 };
    v16u8 diff;
    v8u16 sad = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LW4( p_src, i_src_stride, u_src0, u_src1, u_src2, u_src3 );
        p_src += ( 4 * i_src_stride );
        LW4( p_ref, i_ref_stride, u_ref0, u_ref1, u_ref2, u_ref3 );
        p_ref += ( 4 * i_ref_stride );

        INSERT_W4_UB( u_src0, u_src1, u_src2, u_src3, src );
        INSERT_W4_UB( u_ref0, u_ref1, u_ref2, u_ref3, ref );

        diff = __msa_asub_u_b( src, ref );
        sad += __msa_hadd_u_h( diff, diff );
    }

    return ( HADD_UH_U32( sad ) );
}

static uint32_t sad_8width_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_ref, int32_t i_ref_stride,
                                int32_t i_height )
{
    int32_t i_ht_cnt;
    v16u8 src0, src1, src2, src3, ref0, ref1, ref2, ref3;
    v8u16 sad = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LD_UB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += ( 4 * i_src_stride );
        LD_UB4( p_ref, i_ref_stride, ref0, ref1, ref2, ref3 );
        p_ref += ( 4 * i_ref_stride );

        PCKEV_D4_UB( src1, src0, src3, src2, ref1, ref0, ref3, ref2,
                     src0, src1, ref0, ref1 );
        sad += SAD_UB2_UH( src0, src1, ref0, ref1 );
    }

    return ( HADD_UH_U32( sad ) );
}

static uint32_t sad_16width_msa( uint8_t *p_src, int32_t i_src_stride,
                                 uint8_t *p_ref, int32_t i_ref_stride,
                                 int32_t i_height )
{
    int32_t i_ht_cnt;
    v16u8 src0, src1, ref0, ref1;
    v8u16 sad = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LD_UB2( p_src, i_src_stride, src0, src1 );
        p_src += ( 2 * i_src_stride );
        LD_UB2( p_ref, i_ref_stride, ref0, ref1 );
        p_ref += ( 2 * i_ref_stride );
        sad += SAD_UB2_UH( src0, src1, ref0, ref1 );

        LD_UB2( p_src, i_src_stride, src0, src1 );
        p_src += ( 2 * i_src_stride );
        LD_UB2( p_ref, i_ref_stride, ref0, ref1 );
        p_ref += ( 2 * i_ref_stride );
        sad += SAD_UB2_UH( src0, src1, ref0, ref1 );
    }

    return ( HADD_UH_U32( sad ) );
}

static void sad_4width_x3d_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_ref0, uint8_t *p_ref1,
                                uint8_t *p_ref2, int32_t i_ref_stride,
                                int32_t i_height, uint32_t *pu_sad_array )
{
    int32_t i_ht_cnt;
    v16u8 src = { 0 };
    uint32_t src0, src1, src2, src3, load0, load1, load2, load3;
    v16u8 ref0 = { 0 };
    v16u8 ref1 = { 0 };
    v16u8 ref2 = { 0 };
    v16u8 diff;
    v8u16 sad0 = { 0 };
    v8u16 sad1 = { 0 };
    v8u16 sad2 = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LW4( p_src, i_src_stride, src0, src1, src2, src3 );
        INSERT_W4_UB( src0, src1, src2, src3, src );
        p_src += ( 4 * i_src_stride );

        LW4( p_ref0, i_ref_stride, load0, load1, load2, load3 );
        INSERT_W4_UB( load0, load1, load2, load3, ref0 );
        p_ref0 += ( 4 * i_ref_stride );

        LW4( p_ref1, i_ref_stride, load0, load1, load2, load3 );
        INSERT_W4_UB( load0, load1, load2, load3, ref1 );
        p_ref1 += ( 4 * i_ref_stride );

        LW4( p_ref2, i_ref_stride, load0, load1, load2, load3 );
        INSERT_W4_UB( load0, load1, load2, load3, ref2 );
        p_ref2 += ( 4 * i_ref_stride );

        diff = __msa_asub_u_b( src, ref0 );
        sad0 += __msa_hadd_u_h( diff, diff );

        diff = __msa_asub_u_b( src, ref1 );
        sad1 += __msa_hadd_u_h( diff, diff );

        diff = __msa_asub_u_b( src, ref2 );
        sad2 += __msa_hadd_u_h( diff, diff );
    }

    pu_sad_array[0] = HADD_UH_U32( sad0 );
    pu_sad_array[1] = HADD_UH_U32( sad1 );
    pu_sad_array[2] = HADD_UH_U32( sad2 );
}

static void sad_8width_x3d_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_ref0, uint8_t *p_ref1,
                                uint8_t *p_ref2, int32_t i_ref_stride,
                                int32_t i_height, uint32_t *pu_sad_array )
{
    int32_t i_ht_cnt;
    v16u8 src0, src1, src2, src3;
    v16u8 ref0, ref1, ref00, ref11, ref22, ref33;
    v8u16 sad0 = { 0 };
    v8u16 sad1 = { 0 };
    v8u16 sad2 = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LD_UB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += ( 4 * i_src_stride );
        LD_UB4( p_ref0, i_ref_stride, ref00, ref11, ref22, ref33 );
        p_ref0 += ( 4 * i_ref_stride );

        PCKEV_D4_UB( src1, src0, src3, src2, ref11, ref00, ref33, ref22,
                     src0, src1, ref0, ref1 );
        sad0 += SAD_UB2_UH( src0, src1, ref0, ref1 );

        LD_UB4( p_ref1, i_ref_stride, ref00, ref11, ref22, ref33 );
        p_ref1 += ( 4 * i_ref_stride );

        PCKEV_D2_UB( ref11, ref00, ref33, ref22, ref0, ref1 );
        sad1 += SAD_UB2_UH( src0, src1, ref0, ref1 );

        LD_UB4( p_ref2, i_ref_stride, ref00, ref11, ref22, ref33 );
        p_ref2 += ( 4 * i_ref_stride );

        PCKEV_D2_UB( ref11, ref00, ref33, ref22, ref0, ref1 );
        sad2 += SAD_UB2_UH( src0, src1, ref0, ref1 );
    }

    pu_sad_array[0] = HADD_UH_U32( sad0 );
    pu_sad_array[1] = HADD_UH_U32( sad1 );
    pu_sad_array[2] = HADD_UH_U32( sad2 );
}

static void sad_16width_x3d_msa( uint8_t *p_src, int32_t i_src_stride,
                                 uint8_t *p_ref0, uint8_t *p_ref1,
                                 uint8_t *p_ref2, int32_t i_ref_stride,
                                 int32_t i_height, uint32_t *pu_sad_array )
{
    int32_t i_ht_cnt;
    v16u8 src, ref;
    v16u8 diff;
    v8u16 sad0 = { 0 };
    v8u16 sad1 = { 0 };
    v8u16 sad2 = { 0 };

    for( i_ht_cnt = ( i_height >> 1 ); i_ht_cnt--; )
    {
        src = LD_UB( p_src );
        p_src += i_src_stride;

        ref = LD_UB( p_ref0 );
        p_ref0 += i_ref_stride;
        diff = __msa_asub_u_b( src, ref );
        sad0 += __msa_hadd_u_h( diff, diff );

        ref = LD_UB( p_ref1 );
        p_ref1 += i_ref_stride;
        diff = __msa_asub_u_b( src, ref );
        sad1 += __msa_hadd_u_h( diff, diff );

        ref = LD_UB( p_ref2 );
        p_ref2 += i_ref_stride;
        diff = __msa_asub_u_b( src, ref );
        sad2 += __msa_hadd_u_h( diff, diff );

        src = LD_UB( p_src );
        p_src += i_src_stride;

        ref = LD_UB( p_ref0 );
        p_ref0 += i_ref_stride;
        diff = __msa_asub_u_b( src, ref );
        sad0 += __msa_hadd_u_h( diff, diff );

        ref = LD_UB( p_ref1 );
        p_ref1 += i_ref_stride;
        diff = __msa_asub_u_b( src, ref );
        sad1 += __msa_hadd_u_h( diff, diff );

        ref = LD_UB( p_ref2 );
        p_ref2 += i_ref_stride;
        diff = __msa_asub_u_b( src, ref );
        sad2 += __msa_hadd_u_h( diff, diff );
    }

    pu_sad_array[0] = HADD_UH_U32( sad0 );
    pu_sad_array[1] = HADD_UH_U32( sad1 );
    pu_sad_array[2] = HADD_UH_U32( sad2 );
}

static void sad_4width_x4d_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_aref[], int32_t i_ref_stride,
                                int32_t i_height, uint32_t *pu_sad_array )
{
    uint8_t *p_ref0, *p_ref1, *p_ref2, *p_ref3;
    int32_t i_ht_cnt;
    uint32_t src0, src1, src2, src3;
    uint32_t ref0, ref1, ref2, ref3;
    v16u8 src = { 0 };
    v16u8 ref = { 0 };
    v16u8 diff;
    v8u16 sad0 = { 0 };
    v8u16 sad1 = { 0 };
    v8u16 sad2 = { 0 };
    v8u16 sad3 = { 0 };

    p_ref0 = p_aref[0];
    p_ref1 = p_aref[1];
    p_ref2 = p_aref[2];
    p_ref3 = p_aref[3];

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LW4( p_src, i_src_stride, src0, src1, src2, src3 );
        INSERT_W4_UB( src0, src1, src2, src3, src );
        p_src += ( 4 * i_src_stride );

        LW4( p_ref0, i_ref_stride, ref0, ref1, ref2, ref3 );
        INSERT_W4_UB( ref0, ref1, ref2, ref3, ref );
        p_ref0 += ( 4 * i_ref_stride );

        diff = __msa_asub_u_b( src, ref );
        sad0 += __msa_hadd_u_h( diff, diff );

        LW4( p_ref1, i_ref_stride, ref0, ref1, ref2, ref3 );
        INSERT_W4_UB( ref0, ref1, ref2, ref3, ref );
        p_ref1 += ( 4 * i_ref_stride );

        diff = __msa_asub_u_b( src, ref );
        sad1 += __msa_hadd_u_h( diff, diff );

        LW4( p_ref2, i_ref_stride, ref0, ref1, ref2, ref3 );
        INSERT_W4_UB( ref0, ref1, ref2, ref3, ref );
        p_ref2 += ( 4 * i_ref_stride );

        diff = __msa_asub_u_b( src, ref );
        sad2 += __msa_hadd_u_h( diff, diff );

        LW4( p_ref3, i_ref_stride, ref0, ref1, ref2, ref3 );
        INSERT_W4_UB( ref0, ref1, ref2, ref3, ref );
        p_ref3 += ( 4 * i_ref_stride );

        diff = __msa_asub_u_b( src, ref );
        sad3 += __msa_hadd_u_h( diff, diff );
    }

    pu_sad_array[0] = HADD_UH_U32( sad0 );
    pu_sad_array[1] = HADD_UH_U32( sad1 );
    pu_sad_array[2] = HADD_UH_U32( sad2 );
    pu_sad_array[3] = HADD_UH_U32( sad3 );
}

static void sad_8width_x4d_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_aref[], int32_t i_ref_stride,
                                int32_t i_height, uint32_t *pu_sad_array )
{
    int32_t i_ht_cnt;
    uint8_t *p_ref0, *p_ref1, *p_ref2, *p_ref3;
    v16u8 src0, src1, src2, src3;
    v16u8 ref0, ref1, ref2, ref3, ref4, ref5, ref6, ref7;
    v16u8 ref8, ref9, ref10, ref11, ref12, ref13, ref14, ref15;
    v8u16 sad0 = { 0 };
    v8u16 sad1 = { 0 };
    v8u16 sad2 = { 0 };
    v8u16 sad3 = { 0 };

    p_ref0 = p_aref[0];
    p_ref1 = p_aref[1];
    p_ref2 = p_aref[2];
    p_ref3 = p_aref[3];

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LD_UB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += ( 4 * i_src_stride );
        LD_UB4( p_ref0, i_ref_stride, ref0, ref1, ref2, ref3 );
        p_ref0 += ( 4 * i_ref_stride );
        LD_UB4( p_ref1, i_ref_stride, ref4, ref5, ref6, ref7 );
        p_ref1 += ( 4 * i_ref_stride );
        LD_UB4( p_ref2, i_ref_stride, ref8, ref9, ref10, ref11 );
        p_ref2 += ( 4 * i_ref_stride );
        LD_UB4( p_ref3, i_ref_stride, ref12, ref13, ref14, ref15 );
        p_ref3 += ( 4 * i_ref_stride );

        PCKEV_D2_UB( src1, src0, src3, src2, src0, src1 );
        PCKEV_D2_UB( ref1, ref0, ref3, ref2, ref0, ref1 );
        sad0 += SAD_UB2_UH( src0, src1, ref0, ref1 );

        PCKEV_D2_UB( ref5, ref4, ref7, ref6, ref0, ref1 );
        sad1 += SAD_UB2_UH( src0, src1, ref0, ref1 );

        PCKEV_D2_UB( ref9, ref8, ref11, ref10, ref0, ref1 );
        sad2 += SAD_UB2_UH( src0, src1, ref0, ref1 );

        PCKEV_D2_UB( ref13, ref12, ref15, ref14, ref0, ref1 );
        sad3 += SAD_UB2_UH( src0, src1, ref0, ref1 );
    }

    pu_sad_array[0] = HADD_UH_U32( sad0 );
    pu_sad_array[1] = HADD_UH_U32( sad1 );
    pu_sad_array[2] = HADD_UH_U32( sad2 );
    pu_sad_array[3] = HADD_UH_U32( sad3 );
}

static void sad_16width_x4d_msa( uint8_t *p_src, int32_t i_src_stride,
                                 uint8_t *p_aref[], int32_t i_ref_stride,
                                 int32_t i_height, uint32_t *pu_sad_array )
{
    int32_t i_ht_cnt;
    uint8_t *p_ref0, *p_ref1, *p_ref2, *p_ref3;
    v16u8 src, ref0, ref1, ref2, ref3, diff;
    v8u16 sad0 = { 0 };
    v8u16 sad1 = { 0 };
    v8u16 sad2 = { 0 };
    v8u16 sad3 = { 0 };

    p_ref0 = p_aref[0];
    p_ref1 = p_aref[1];
    p_ref2 = p_aref[2];
    p_ref3 = p_aref[3];

    for( i_ht_cnt = ( i_height >> 1 ); i_ht_cnt--; )
    {
        src = LD_UB( p_src );
        p_src += i_src_stride;
        ref0 = LD_UB( p_ref0 );
        p_ref0 += i_ref_stride;
        ref1 = LD_UB( p_ref1 );
        p_ref1 += i_ref_stride;
        ref2 = LD_UB( p_ref2 );
        p_ref2 += i_ref_stride;
        ref3 = LD_UB( p_ref3 );
        p_ref3 += i_ref_stride;

        diff = __msa_asub_u_b( src, ref0 );
        sad0 += __msa_hadd_u_h( diff, diff );
        diff = __msa_asub_u_b( src, ref1 );
        sad1 += __msa_hadd_u_h( diff, diff );
        diff = __msa_asub_u_b( src, ref2 );
        sad2 += __msa_hadd_u_h( diff, diff );
        diff = __msa_asub_u_b( src, ref3 );
        sad3 += __msa_hadd_u_h( diff, diff );

        src = LD_UB( p_src );
        p_src += i_src_stride;
        ref0 = LD_UB( p_ref0 );
        p_ref0 += i_ref_stride;
        ref1 = LD_UB( p_ref1 );
        p_ref1 += i_ref_stride;
        ref2 = LD_UB( p_ref2 );
        p_ref2 += i_ref_stride;
        ref3 = LD_UB( p_ref3 );
        p_ref3 += i_ref_stride;

        diff = __msa_asub_u_b( src, ref0 );
        sad0 += __msa_hadd_u_h( diff, diff );
        diff = __msa_asub_u_b( src, ref1 );
        sad1 += __msa_hadd_u_h( diff, diff );
        diff = __msa_asub_u_b( src, ref2 );
        sad2 += __msa_hadd_u_h( diff, diff );
        diff = __msa_asub_u_b( src, ref3 );
        sad3 += __msa_hadd_u_h( diff, diff );
    }

    pu_sad_array[0] = HADD_UH_U32( sad0 );
    pu_sad_array[1] = HADD_UH_U32( sad1 );
    pu_sad_array[2] = HADD_UH_U32( sad2 );
    pu_sad_array[3] = HADD_UH_U32( sad3 );
}

static uint64_t avc_pixel_var16width_msa( uint8_t *p_pix, int32_t i_stride,
                                          uint8_t i_height )
{
    uint32_t u_sum = 0, u_sqr_out = 0, u_cnt;
    v16i8 pix, zero = { 0 };
    v8u16 add, pix_r, pix_l;
    v4u32 sqr = { 0 };

    for( u_cnt = i_height; u_cnt--; )
    {
        pix = LD_SB( p_pix );
        p_pix += i_stride;
        add = __msa_hadd_u_h( ( v16u8 ) pix, ( v16u8 ) pix );
        u_sum += HADD_UH_U32( add );
        ILVRL_B2_UH( zero, pix, pix_r, pix_l );
        sqr = __msa_dpadd_u_w( sqr, pix_r, pix_r );
        sqr = __msa_dpadd_u_w( sqr, pix_l, pix_l );
    }

    u_sqr_out = HADD_SW_S32( sqr );

    return ( u_sum + ( ( uint64_t ) u_sqr_out << 32 ) );
}

static uint64_t avc_pixel_var8width_msa( uint8_t *p_pix, int32_t i_stride,
                                         uint8_t i_height )
{
    uint32_t u_sum = 0, u_sqr_out = 0, u_cnt;
    v16i8 pix, zero = { 0 };
    v8u16 add, pix_r;
    v4u32 sqr = { 0 };

    for( u_cnt = i_height; u_cnt--; )
    {
        pix = LD_SB( p_pix );
        p_pix += i_stride;
        pix_r = ( v8u16 ) __msa_ilvr_b( zero, pix );
        add = __msa_hadd_u_h( ( v16u8 ) pix_r, ( v16u8 ) pix_r );
        u_sum += HADD_UH_U32( add );
        sqr = __msa_dpadd_u_w( sqr, pix_r, pix_r );
    }

    u_sqr_out = HADD_SW_S32( sqr );

    return ( u_sum + ( ( uint64_t ) u_sqr_out << 32 ) );
}

static uint32_t sse_diff_8width_msa( uint8_t *p_src, int32_t i_src_stride,
                                     uint8_t *p_ref, int32_t i_ref_stride,
                                     int32_t i_height, int32_t *p_diff )
{
    int32_t i_ht_cnt;
    uint32_t u_sse;
    v16u8 src0, src1, src2, src3;
    v16u8 ref0, ref1, ref2, ref3;
    v8i16 avg = { 0 };
    v4i32 vec, var = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LD_UB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += ( 4 * i_src_stride );
        LD_UB4( p_ref, i_ref_stride, ref0, ref1, ref2, ref3 );
        p_ref += ( 4 * i_ref_stride );

        PCKEV_D4_UB( src1, src0, src3, src2, ref1, ref0, ref3, ref2,
                     src0, src1, ref0, ref1 );
        CALC_MSE_AVG_B( src0, ref0, var, avg );
        CALC_MSE_AVG_B( src1, ref1, var, avg );
    }

    vec = __msa_hadd_s_w( avg, avg );
    *p_diff = HADD_SW_S32( vec );
    u_sse = HADD_SW_S32( var );

    return u_sse;
}

static uint32_t sse_4width_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_ref, int32_t i_ref_stride,
                                int32_t i_height )
{
    int32_t i_ht_cnt;
    uint32_t u_sse;
    uint32_t u_src0, u_src1, u_src2, u_src3;
    uint32_t u_ref0, u_ref1, u_ref2, u_ref3;
    v16u8 src = { 0 };
    v16u8 ref = { 0 };
    v4i32 var = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LW4( p_src, i_src_stride, u_src0, u_src1, u_src2, u_src3 );
        p_src += ( 4 * i_src_stride );
        LW4( p_ref, i_ref_stride, u_ref0, u_ref1, u_ref2, u_ref3 );
        p_ref += ( 4 * i_ref_stride );

        INSERT_W4_UB( u_src0, u_src1, u_src2, u_src3, src );
        INSERT_W4_UB( u_ref0, u_ref1, u_ref2, u_ref3, ref );
        CALC_MSE_B( src, ref, var );
    }

    u_sse = HADD_SW_S32( var );

    return u_sse;
}

static uint32_t sse_8width_msa( uint8_t *p_src, int32_t i_src_stride,
                                uint8_t *p_ref, int32_t i_ref_stride,
                                int32_t i_height )
{
    int32_t i_ht_cnt;
    uint32_t u_sse;
    v16u8 src0, src1, src2, src3;
    v16u8 ref0, ref1, ref2, ref3;
    v4i32 var = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        LD_UB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += ( 4 * i_src_stride );
        LD_UB4( p_ref, i_ref_stride, ref0, ref1, ref2, ref3 );
        p_ref += ( 4 * i_ref_stride );

        PCKEV_D4_UB( src1, src0, src3, src2, ref1, ref0, ref3, ref2,
                     src0, src1, ref0, ref1 );
        CALC_MSE_B( src0, ref0, var );
        CALC_MSE_B( src1, ref1, var );
    }

    u_sse = HADD_SW_S32( var );

    return u_sse;
}

static uint32_t sse_16width_msa( uint8_t *p_src, int32_t i_src_stride,
                                 uint8_t *p_ref, int32_t i_ref_stride,
                                 int32_t i_height )
{
    int32_t i_ht_cnt;
    uint32_t u_sse;
    v16u8 src, ref;
    v4i32 var = { 0 };

    for( i_ht_cnt = ( i_height >> 2 ); i_ht_cnt--; )
    {
        src = LD_UB( p_src );
        p_src += i_src_stride;
        ref = LD_UB( p_ref );
        p_ref += i_ref_stride;
        CALC_MSE_B( src, ref, var );

        src = LD_UB( p_src );
        p_src += i_src_stride;
        ref = LD_UB( p_ref );
        p_ref += i_ref_stride;
        CALC_MSE_B( src, ref, var );

        src = LD_UB( p_src );
        p_src += i_src_stride;
        ref = LD_UB( p_ref );
        p_ref += i_ref_stride;
        CALC_MSE_B( src, ref, var );

        src = LD_UB( p_src );
        p_src += i_src_stride;
        ref = LD_UB( p_ref );
        p_ref += i_ref_stride;
        CALC_MSE_B( src, ref, var );
    }

    u_sse = HADD_SW_S32( var );

    return u_sse;
}

static void ssim_4x4x2_core_msa( const uint8_t *p_src, int32_t i_src_stride,
                                 const uint8_t *p_ref, int32_t i_ref_stride,
                                 int32_t pi_sum_array[2][4] )
{
    v16i8 zero = { 0 };
    v16u8 src0, src1, src2, src3, ref0, ref1, ref2, ref3;
    v8u16 temp0, temp1, temp2, temp3;
    v8u16 vec0, vec1, vec2, vec3, vec4, vec5, vec6, vec7;
    v4u32 tmp0;
    v4i32 tmp2, tmp3;

    LD_UB4( p_src, i_src_stride, src0, src1, src2, src3 );
    p_src += ( 4 * i_src_stride );
    LD_UB4( p_ref, i_ref_stride, ref0, ref1, ref2, ref3 );
    p_ref += ( 4 * i_ref_stride );

    ILVR_D2_UB( src1, src0, src3, src2, src0, src2 );
    ILVR_D2_UB( ref1, ref0, ref3, ref2, ref0, ref2 );
    HADD_UB2_UH( src0, src2, temp0, temp1 );

    temp2 = ( v8u16 ) __msa_ilvev_w( ( v4i32 ) temp1, ( v4i32 ) temp0 );
    temp3 = ( v8u16 ) __msa_ilvod_w( ( v4i32 ) temp1, ( v4i32 ) temp0 );

    pi_sum_array[0][0] = ( int32_t ) HADD_UH_U32( temp2 );
    pi_sum_array[1][0] = ( int32_t ) HADD_UH_U32( temp3 );

    HADD_UB2_UH( ref0, ref2, temp0, temp1 );

    temp2 = ( v8u16 ) __msa_ilvev_w( ( v4i32 ) temp1, ( v4i32 ) temp0 );
    temp3 = ( v8u16 ) __msa_ilvod_w( ( v4i32 ) temp1, ( v4i32 ) temp0 );

    pi_sum_array[0][1] = ( int32_t ) HADD_UH_U32( temp2 );
    pi_sum_array[1][1] = ( int32_t ) HADD_UH_U32( temp3 );

    ILVR_B4_UH( zero, src0, zero, src2, zero, ref0, zero, ref2, vec0, vec2,
                vec4, vec6 );
    ILVL_B4_UH( zero, src0, zero, src2, zero, ref0, zero, ref2, vec1, vec3,
                vec5, vec7 );

    tmp0 = __msa_dotp_u_w( vec0, vec0 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec1, vec1 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec2, vec2 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec3, vec3 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec4, vec4 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec5, vec5 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec6, vec6 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec7, vec7 );

    tmp2 = ( v4i32 ) __msa_ilvev_d( ( v2i64 ) tmp0, ( v2i64 ) tmp0 );
    tmp3 = ( v4i32 ) __msa_ilvod_d( ( v2i64 ) tmp0, ( v2i64 ) tmp0 );
    tmp2 = ( v4i32 ) __msa_hadd_u_d( ( v4u32 ) tmp2, ( v4u32 ) tmp2 );
    tmp3 = ( v4i32 ) __msa_hadd_u_d( ( v4u32 ) tmp3, ( v4u32 ) tmp3 );

    pi_sum_array[0][2] = __msa_copy_u_w( tmp2, 0 );
    pi_sum_array[1][2] = __msa_copy_u_w( tmp3, 0 );

    tmp0 = __msa_dotp_u_w( vec4, vec0 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec5, vec1 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec6, vec2 );
    tmp0 = __msa_dpadd_u_w( tmp0, vec7, vec3 );

    tmp2 = ( v4i32 ) __msa_ilvev_d( ( v2i64 ) tmp0, ( v2i64 ) tmp0 );
    tmp3 = ( v4i32 ) __msa_ilvod_d( ( v2i64 ) tmp0, ( v2i64 ) tmp0 );
    tmp2 = ( v4i32 ) __msa_hadd_u_d( ( v4u32 ) tmp2, ( v4u32 ) tmp2 );
    tmp3 = ( v4i32 ) __msa_hadd_u_d( ( v4u32 ) tmp3, ( v4u32 ) tmp3 );

    pi_sum_array[0][3] = __msa_copy_u_w( tmp2, 0 );
    pi_sum_array[1][3] = __msa_copy_u_w( tmp3, 0 );
}

static int32_t pixel_satd_4width_msa( uint8_t *p_src, int32_t i_src_stride,
                                      uint8_t *p_ref, int32_t i_ref_stride,
                                      uint8_t i_height )
{
    int32_t cnt;
    uint32_t u_sum = 0;
    v16i8 src0, src1, src2, src3;
    v16i8 ref0, ref1, ref2, ref3;
    v8i16 zero = { 0 };
    v8i16 diff0, diff1, diff2, diff3;
    v8i16 temp0, temp1, temp2, temp3;

    for( cnt = i_height >> 2; cnt--; )
    {
        LD_SB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += 4 * i_src_stride;
        LD_SB4( p_ref, i_ref_stride, ref0, ref1, ref2, ref3 );
        p_ref += 4 * i_ref_stride;

        ILVR_B4_SH( src0, ref0, src1, ref1, src2, ref2, src3, ref3,
                    diff0, diff1, diff2, diff3 );
        HSUB_UB4_SH( diff0, diff1, diff2, diff3, diff0, diff1, diff2, diff3 );
        TRANSPOSE4x4_SH_SH( diff0, diff1, diff2, diff3,
                            diff0, diff1, diff2, diff3 );
        BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
        BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );
        TRANSPOSE4x4_SH_SH( diff0, diff1, diff2, diff3,
                            diff0, diff1, diff2, diff3 );
        BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
        BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );

        diff0 = __msa_add_a_h( diff0, zero );
        diff1 = __msa_add_a_h( diff1, zero );
        diff2 = __msa_add_a_h( diff2, zero );
        diff3 = __msa_add_a_h( diff3, zero );
        diff0 = ( diff0 + diff1 + diff2 + diff3 );
        diff0 = ( v8i16 ) __msa_hadd_u_w( ( v8u16 ) diff0, ( v8u16 ) diff0 );
        diff0 = ( v8i16 ) __msa_hadd_u_d( ( v4u32 ) diff0, ( v4u32 ) diff0 );
        u_sum += __msa_copy_u_w( ( v4i32 ) diff0, 0 );
    }

    return ( u_sum >> 1 );
}

static int32_t pixel_satd_8width_msa( uint8_t *p_src, int32_t i_src_stride,
                                      uint8_t *p_ref, int32_t i_ref_stride,
                                      uint8_t i_height )
{
    int32_t cnt;
    uint32_t u_sum = 0;
    v16i8 src0, src1, src2, src3;
    v16i8 ref0, ref1, ref2, ref3;
    v8i16 zero = { 0 };
    v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
    v8i16 temp0, temp1, temp2, temp3;

    for( cnt = i_height >> 2; cnt--; )
    {
        LD_SB4( p_src, i_src_stride, src0, src1, src2, src3 );
        p_src += 4 * i_src_stride;
        LD_SB4( p_ref, i_ref_stride, ref0, ref1, ref2, ref3 );
        p_ref += 4 * i_ref_stride;

        ILVR_B4_SH( src0, ref0, src1, ref1, src2, ref2, src3, ref3,
                    diff0, diff1, diff2, diff3 );
        HSUB_UB4_SH( diff0, diff1, diff2, diff3, diff0, diff1, diff2, diff3 );
        TRANSPOSE8X4_SH_SH( diff0, diff1, diff2, diff3,
                            diff0, diff2, diff4, diff6 );

        diff1 = ( v8i16 ) __msa_splati_d( ( v2i64 ) diff0, 1 );
        diff3 = ( v8i16 ) __msa_splati_d( ( v2i64 ) diff2, 1 );
        diff5 = ( v8i16 ) __msa_splati_d( ( v2i64 ) diff4, 1 );
        diff7 = ( v8i16 ) __msa_splati_d( ( v2i64 ) diff6, 1 );

        BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
        BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );
        BUTTERFLY_4( diff4, diff6, diff7, diff5, temp0, temp2, temp3, temp1 );
        BUTTERFLY_4( temp0, temp1, temp3, temp2, diff4, diff5, diff7, diff6 );
        TRANSPOSE4X8_SH_SH( diff0, diff1, diff2, diff3, diff4, diff5, diff6,
                            diff7, diff0, diff1, diff2, diff3, diff4, diff5,
                            diff6, diff7 );
        BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
        BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );

        diff0 = __msa_add_a_h( diff0, zero );
        diff1 = __msa_add_a_h( diff1, zero );
        diff2 = __msa_add_a_h( diff2, zero );
        diff3 = __msa_add_a_h( diff3, zero );
        diff0 = ( diff0 + diff1 + diff2 + diff3 );
        u_sum += HADD_UH_U32( diff0 );
    }

    return ( u_sum >> 1 );
}

static int32_t sa8d_8x8_msa( uint8_t *p_src, int32_t i_src_stride,
                             uint8_t *p_ref, int32_t i_ref_stride )
{
    uint32_t u_sum = 0;
    v16i8 src0, src1, src2, src3, src4, src5, src6, src7;
    v16i8 ref0, ref1, ref2, ref3, ref4, ref5, ref6, ref7;
    v8i16 zero = { 0 };
    v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
    v8i16 sub0, sub1, sub2, sub3, sub4, sub5, sub6, sub7;
    v8i16 temp0, temp1, temp2, temp3;

    LD_SB8( p_src, i_src_stride, src0, src1, src2, src3, src4, src5, src6, src7 );
    LD_SB8( p_ref, i_ref_stride, ref0, ref1, ref2, ref3, ref4, ref5, ref6, ref7 );
    ILVR_B4_SH( src0, ref0, src1, ref1, src2, ref2, src3, ref3, sub0, sub1,
                sub2, sub3 );
    ILVR_B4_SH( src4, ref4, src5, ref5, src6, ref6, src7, ref7, sub4, sub5,
               sub6, sub7 );
    HSUB_UB4_SH( sub0, sub1, sub2, sub3, sub0, sub1, sub2, sub3 );
    HSUB_UB4_SH( sub4, sub5, sub6, sub7, sub4, sub5, sub6, sub7 );
    TRANSPOSE8x8_SH_SH( sub0, sub1, sub2, sub3, sub4, sub5, sub6, sub7,
                        sub0, sub1, sub2, sub3, sub4, sub5, sub6, sub7 );
    BUTTERFLY_4( sub0, sub2, sub3, sub1, diff0, diff1, diff4, diff5 );
    BUTTERFLY_4( sub4, sub6, sub7, sub5, diff2, diff3, diff7, diff6 );
    BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );
    BUTTERFLY_4( diff4, diff6, diff7, diff5, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff4, diff5, diff7, diff6 );
    TRANSPOSE8x8_SH_SH( diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7,
                        diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7 );
    BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );
    BUTTERFLY_4( diff4, diff6, diff7, diff5, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff4, diff5, diff7, diff6 );

    temp0 = diff0 + diff4;
    temp1 = diff1 + diff5;
    temp2 = diff2 + diff6;
    temp3 = diff3 + diff7;

    temp0 = __msa_add_a_h( temp0, zero );
    temp1 = __msa_add_a_h( temp1, zero );
    temp2 = __msa_add_a_h( temp2, zero );
    temp3 = __msa_add_a_h( temp3, zero );

    diff0 = temp0 + __msa_asub_s_h( diff0, diff4 );
    diff1 = temp1 + __msa_asub_s_h( diff1, diff5 );
    diff2 = temp2 + __msa_asub_s_h( diff2, diff6 );
    diff3 = temp3 + __msa_asub_s_h( diff3, diff7 );
    diff0 = ( diff0 + diff1 + diff2 + diff3 );

    u_sum = HADD_UH_U32( diff0 );

    return u_sum;
}

static uint64_t pixel_hadamard_ac_8x8_msa( uint8_t *p_pix, int32_t i_stride )
{
    int16_t tmp0, tmp1, tmp2, tmp3;
    uint32_t u_sum4 = 0, u_sum8 = 0, u_dc;
    v16u8 src0, src1, src2, src3, src4, src5, src6, src7;
    v8i16 zero = { 0 };
    v8i16 diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7;
    v8i16 sub0, sub1, sub2, sub3, sub4, sub5, sub6, sub7;
    v8i16 temp0, temp1, temp2, temp3;

    LD_UB8( p_pix, i_stride, src0, src1, src2, src3, src4, src5, src6, src7 );

    ILVR_B4_SH( zero, src0, zero, src1, zero, src2, zero, src3, diff0, diff1,
                diff2, diff3 );
    ILVR_B4_SH( zero, src4, zero, src5, zero, src6, zero, src7, diff4, diff5,
                diff6, diff7 );
    TRANSPOSE8x8_SH_SH( diff0, diff1, diff2, diff3,
                        diff4, diff5, diff6, diff7,
                        diff0, diff1, diff2, diff3,
                        diff4, diff5, diff6, diff7 );
    BUTTERFLY_4( diff0, diff2, diff3, diff1,
                 temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2,
                 diff0, diff1, diff3, diff2 );
    BUTTERFLY_4( diff4, diff6, diff7, diff5,
                 temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2,
                 diff4, diff5, diff7, diff6 );
    TRANSPOSE8x8_SH_SH( diff0, diff1, diff2, diff3,
                        diff4, diff5, diff6, diff7,
                        diff0, diff1, diff2, diff3,
                        diff4, diff5, diff6, diff7 );
    BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );
    BUTTERFLY_4( diff4, diff6, diff7, diff5, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff4, diff5, diff7, diff6 );

    tmp0 = diff0[0];
    tmp1 = diff0[4];
    tmp2 = diff4[0];
    tmp3 = diff4[4];

    sub0 = __msa_add_a_h( diff0, zero );
    sub1 = __msa_add_a_h( diff1, zero );
    sub2 = __msa_add_a_h( diff2, zero );
    sub3 = __msa_add_a_h( diff3, zero );
    sub4 = __msa_add_a_h( diff4, zero );
    sub5 = __msa_add_a_h( diff5, zero );
    sub6 = __msa_add_a_h( diff6, zero );
    sub7 = __msa_add_a_h( diff7, zero );

    sub0 = ( sub0 + sub1 + sub2 + sub3 );
    sub1 = ( sub4 + sub5 + sub6 + sub7 );
    sub0 += sub1;

    u_sum4 += HADD_UH_U32( sub0 );

    TRANSPOSE8x8_SH_SH( diff0, diff1, diff2, diff3, diff4, diff5, diff6, diff7,
                        sub0, sub1, sub2, sub3, sub4, sub5, sub6, sub7 );

    ILVR_D2_SH( sub2, sub0, sub6, sub4, diff0, diff1 );
    ILVR_D2_SH( sub3, sub1, sub7, sub5, diff4, diff6 );

    diff2 = ( v8i16 ) __msa_ilvl_d( ( v2i64 ) sub2, ( v2i64 ) sub0 );
    diff3 = ( v8i16 ) __msa_ilvl_d( ( v2i64 ) sub6, ( v2i64 ) sub4 );
    diff5 = ( v8i16 ) __msa_ilvl_d( ( v2i64 ) sub3, ( v2i64 ) sub1 );
    diff7 = ( v8i16 ) __msa_ilvl_d( ( v2i64 ) sub7, ( v2i64 ) sub5 );

    BUTTERFLY_4( diff0, diff2, diff3, diff1, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff0, diff1, diff3, diff2 );
    BUTTERFLY_4( diff4, diff6, diff7, diff5, temp0, temp2, temp3, temp1 );
    BUTTERFLY_4( temp0, temp1, temp3, temp2, diff4, diff5, diff7, diff6 );

    sub0 = __msa_add_a_h( diff0, zero );
    sub1 = __msa_add_a_h( diff1, zero );
    sub2 = __msa_add_a_h( diff2, zero );
    sub3 = __msa_add_a_h( diff3, zero );
    sub4 = __msa_add_a_h( diff4, zero );
    sub5 = __msa_add_a_h( diff5, zero );
    sub6 = __msa_add_a_h( diff6, zero );
    sub7 = __msa_add_a_h( diff7, zero );

    sub0 = ( sub0 + sub1 + sub2 + sub3 );
    sub1 = ( sub4 + sub5 + sub6 + sub7 );
    sub0 += sub1;

    u_sum8 += HADD_UH_U32( sub0 );

    u_dc = ( uint16_t ) ( tmp0 + tmp1 + tmp2 + tmp3 );
    u_sum4 = u_sum4 - u_dc;
    u_sum8 = u_sum8 - u_dc;

    return ( ( uint64_t ) u_sum8 << 32 ) + u_sum4;
}

int32_t x264_pixel_sad_16x16_msa( uint8_t *p_src, intptr_t i_src_stride,
                                  uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_16width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 16 );
}

int32_t x264_pixel_sad_16x8_msa( uint8_t *p_src, intptr_t i_src_stride,
                                 uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_16width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 8 );
}

int32_t x264_pixel_sad_8x16_msa( uint8_t *p_src, intptr_t i_src_stride,
                                 uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_8width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 16 );
}

int32_t x264_pixel_sad_8x8_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_8width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 8 );
}

int32_t x264_pixel_sad_8x4_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_8width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 4 );
}

int32_t x264_pixel_sad_4x16_msa( uint8_t *p_src, intptr_t i_src_stride,
                                 uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_4width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 16 );
}

int32_t x264_pixel_sad_4x8_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_4width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 8 );
}

int32_t x264_pixel_sad_4x4_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sad_4width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 4 );
}

void x264_pixel_sad_x4_16x16_msa( uint8_t *p_src, uint8_t *p_ref0,
                                  uint8_t *p_ref1, uint8_t *p_ref2,
                                  uint8_t *p_ref3, intptr_t i_ref_stride,
                                  int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_16width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 16,
                         ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x4_16x8_msa( uint8_t *p_src, uint8_t *p_ref0,
                                 uint8_t *p_ref1, uint8_t *p_ref2,
                                 uint8_t *p_ref3, intptr_t i_ref_stride,
                                 int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_16width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 8,
                         ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x4_8x16_msa( uint8_t *p_src, uint8_t *p_ref0,
                                 uint8_t *p_ref1, uint8_t *p_ref2,
                                 uint8_t *p_ref3, intptr_t i_ref_stride,
                                 int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_8width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 16,
                        ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x4_8x8_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                uint8_t *p_ref3, intptr_t i_ref_stride,
                                int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_8width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 8,
                        ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x4_8x4_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                uint8_t *p_ref3, intptr_t i_ref_stride,
                                int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_8width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 4,
                        ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x4_4x8_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                uint8_t *p_ref3, intptr_t i_ref_stride,
                                int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_4width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 8,
                        ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x4_4x4_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                uint8_t *p_ref3, intptr_t i_ref_stride,
                                int32_t p_sad_array[4] )
{
    uint8_t *p_aref[4] = { p_ref0, p_ref1, p_ref2, p_ref3 };

    sad_4width_x4d_msa( p_src, FENC_STRIDE, p_aref, i_ref_stride, 4,
                        ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_16x16_msa( uint8_t *p_src, uint8_t *p_ref0,
                                  uint8_t *p_ref1, uint8_t *p_ref2,
                                  intptr_t i_ref_stride,
                                  int32_t p_sad_array[3] )
{
    sad_16width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                         i_ref_stride, 16, ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_16x8_msa( uint8_t *p_src, uint8_t *p_ref0,
                                 uint8_t *p_ref1, uint8_t *p_ref2,
                                 intptr_t i_ref_stride,
                                 int32_t p_sad_array[3] )
{
    sad_16width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                         i_ref_stride, 8, ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_8x16_msa( uint8_t *p_src, uint8_t *p_ref0,
                                 uint8_t *p_ref1, uint8_t *p_ref2,
                                 intptr_t i_ref_stride,
                                 int32_t p_sad_array[3] )
{
    sad_8width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                        i_ref_stride, 16, ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_8x8_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                intptr_t i_ref_stride,
                                int32_t p_sad_array[3] )
{
    sad_8width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                        i_ref_stride, 8, ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_8x4_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                intptr_t i_ref_stride,
                                int32_t p_sad_array[3] )
{
    sad_8width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                        i_ref_stride, 4, ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_4x8_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                intptr_t i_ref_stride,
                                int32_t p_sad_array[3] )
{
    sad_4width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                        i_ref_stride, 8, ( uint32_t * ) p_sad_array );
}

void x264_pixel_sad_x3_4x4_msa( uint8_t *p_src, uint8_t *p_ref0,
                                uint8_t *p_ref1, uint8_t *p_ref2,
                                intptr_t i_ref_stride,
                                int32_t p_sad_array[3] )
{
    sad_4width_x3d_msa( p_src, FENC_STRIDE, p_ref0, p_ref1, p_ref2,
                        i_ref_stride, 4, ( uint32_t * ) p_sad_array );
}

int32_t x264_pixel_ssd_16x16_msa( uint8_t *p_src, intptr_t i_src_stride,
                                  uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_16width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 16 );
}

int32_t x264_pixel_ssd_16x8_msa( uint8_t *p_src, intptr_t i_src_stride,
                                 uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_16width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 8 );
}

int32_t x264_pixel_ssd_8x16_msa( uint8_t *p_src, intptr_t i_src_stride,
                                 uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_8width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 16 );
}

int32_t x264_pixel_ssd_8x8_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_8width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 8 );
}

int32_t x264_pixel_ssd_8x4_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_8width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 4 );
}

int32_t x264_pixel_ssd_4x16_msa( uint8_t *p_src, intptr_t i_src_stride,
                                 uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_4width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 16 );
}

int32_t x264_pixel_ssd_4x8_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_4width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 8 );
}

int32_t x264_pixel_ssd_4x4_msa( uint8_t *p_src, intptr_t i_src_stride,
                                uint8_t *p_ref, intptr_t i_ref_stride )
{
    return sse_4width_msa( p_src, i_src_stride, p_ref, i_ref_stride, 4 );
}

void x264_intra_sad_x3_4x4_msa( uint8_t *p_enc, uint8_t *p_dec,
                                int32_t p_sad_array[3] )
{
    x264_intra_predict_vert_4x4_msa( p_dec );
    p_sad_array[0] = x264_pixel_sad_4x4_msa( p_dec, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );

    x264_intra_predict_hor_4x4_msa( p_dec );
    p_sad_array[1] = x264_pixel_sad_4x4_msa( p_dec, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );

    x264_intra_predict_dc_4x4_msa( p_dec );
    p_sad_array[2] = x264_pixel_sad_4x4_msa( p_dec, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );
}

void x264_intra_sad_x3_16x16_msa( uint8_t *p_enc, uint8_t *p_dec,
                                  int32_t p_sad_array[3] )
{
    x264_intra_predict_vert_16x16_msa( p_dec );
    p_sad_array[0] = x264_pixel_sad_16x16_msa( p_dec, FDEC_STRIDE,
                                               p_enc, FENC_STRIDE );

    x264_intra_predict_hor_16x16_msa( p_dec );
    p_sad_array[1] = x264_pixel_sad_16x16_msa( p_dec, FDEC_STRIDE,
                                               p_enc, FENC_STRIDE );

    x264_intra_predict_dc_16x16_msa( p_dec );
    p_sad_array[2] = x264_pixel_sad_16x16_msa( p_dec, FDEC_STRIDE,
                                               p_enc, FENC_STRIDE );
}

void x264_intra_sad_x3_8x8_msa( uint8_t *p_enc, uint8_t p_edge[36],
                                int32_t p_sad_array[3] )
{
    ALIGNED_ARRAY_16( uint8_t, pix, [8 * FDEC_STRIDE] );

    x264_intra_predict_v_8x8_msa( pix, p_edge );
    p_sad_array[0] = x264_pixel_sad_8x8_msa( pix, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );

    x264_intra_predict_h_8x8_msa( pix, p_edge );
    p_sad_array[1] = x264_pixel_sad_8x8_msa( pix, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );

    x264_intra_predict_dc_8x8_msa( pix, p_edge );
    p_sad_array[2] = x264_pixel_sad_8x8_msa( pix, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );
}

void x264_intra_sad_x3_8x8c_msa( uint8_t *p_enc, uint8_t *p_dec,
                                 int32_t p_sad_array[3] )
{
    x264_intra_predict_dc_4blk_8x8_msa( p_dec );
    p_sad_array[0] = x264_pixel_sad_8x8_msa( p_dec, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );

    x264_intra_predict_hor_8x8_msa( p_dec );
    p_sad_array[1] = x264_pixel_sad_8x8_msa( p_dec, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );

    x264_intra_predict_vert_8x8_msa( p_dec );
    p_sad_array[2] = x264_pixel_sad_8x8_msa( p_dec, FDEC_STRIDE,
                                             p_enc, FENC_STRIDE );
}

void x264_ssim_4x4x2_core_msa( const uint8_t *p_pix1, intptr_t i_stride1,
                               const uint8_t *p_pix2, intptr_t i_stride2,
                               int32_t i_sums[2][4] )
{
    ssim_4x4x2_core_msa( p_pix1, i_stride1, p_pix2, i_stride2, i_sums );
}

uint64_t x264_pixel_hadamard_ac_8x8_msa( uint8_t *p_pix, intptr_t i_stride )
{
    uint64_t u_sum;

    u_sum = pixel_hadamard_ac_8x8_msa( p_pix, i_stride );

    return ( ( u_sum >> 34 ) << 32 ) + ( ( uint32_t ) u_sum >> 1 );
}

uint64_t x264_pixel_hadamard_ac_8x16_msa( uint8_t *p_pix, intptr_t i_stride )
{
    uint64_t u_sum;

    u_sum = pixel_hadamard_ac_8x8_msa( p_pix, i_stride );
    u_sum += pixel_hadamard_ac_8x8_msa( p_pix + 8 * i_stride, i_stride );

    return ( ( u_sum >> 34 ) << 32 ) + ( ( uint32_t ) u_sum >> 1 );
}

uint64_t x264_pixel_hadamard_ac_16x8_msa( uint8_t *p_pix, intptr_t i_stride )
{
    uint64_t u_sum;

    u_sum = pixel_hadamard_ac_8x8_msa( p_pix, i_stride );
    u_sum += pixel_hadamard_ac_8x8_msa( p_pix + 8, i_stride );

    return ( ( u_sum >> 34 ) << 32 ) + ( ( uint32_t ) u_sum >> 1 );
}

uint64_t x264_pixel_hadamard_ac_16x16_msa( uint8_t *p_pix, intptr_t i_stride )
{
    uint64_t u_sum;

    u_sum = pixel_hadamard_ac_8x8_msa( p_pix, i_stride );
    u_sum += pixel_hadamard_ac_8x8_msa( p_pix + 8, i_stride );
    u_sum += pixel_hadamard_ac_8x8_msa( p_pix + 8 * i_stride, i_stride );
    u_sum += pixel_hadamard_ac_8x8_msa( p_pix + 8 * i_stride + 8, i_stride );

    return ( ( u_sum >> 34 ) << 32 ) + ( ( uint32_t ) u_sum >> 1 );
}

int32_t x264_pixel_satd_4x4_msa( uint8_t *p_pix1, intptr_t i_stride,
                                 uint8_t *p_pix2, intptr_t i_stride2 )
{
    return pixel_satd_4width_msa( p_pix1, i_stride, p_pix2, i_stride2, 4 );
}

int32_t x264_pixel_satd_4x8_msa( uint8_t *p_pix1, intptr_t i_stride,
                                 uint8_t *p_pix2, intptr_t i_stride2 )
{
    return pixel_satd_4width_msa( p_pix1, i_stride, p_pix2, i_stride2, 8 );
}

int32_t x264_pixel_satd_4x16_msa( uint8_t *p_pix1, intptr_t i_stride,
                                  uint8_t *p_pix2, intptr_t i_stride2 )
{
    return pixel_satd_4width_msa( p_pix1, i_stride, p_pix2, i_stride2, 16 );
}

int32_t x264_pixel_satd_8x4_msa( uint8_t *p_pix1, intptr_t i_stride,
                                 uint8_t *p_pix2, intptr_t i_stride2 )
{
    return pixel_satd_8width_msa( p_pix1, i_stride, p_pix2, i_stride2, 4 );
}

int32_t x264_pixel_satd_8x8_msa( uint8_t *p_pix1, intptr_t i_stride,
                                 uint8_t *p_pix2, intptr_t i_stride2 )
{
    return pixel_satd_8width_msa( p_pix1, i_stride, p_pix2, i_stride2, 8 );
}

int32_t x264_pixel_satd_8x16_msa( uint8_t *p_pix1, intptr_t i_stride,
                                  uint8_t *p_pix2, intptr_t i_stride2 )
{
    return pixel_satd_8width_msa( p_pix1, i_stride, p_pix2, i_stride2, 16 );
}

int32_t x264_pixel_satd_16x8_msa( uint8_t *p_pix1, intptr_t i_stride,
                                  uint8_t *p_pix2, intptr_t i_stride2 )
{
    uint32_t u32Sum = 0;

    u32Sum = pixel_satd_8width_msa( p_pix1, i_stride, p_pix2, i_stride2, 8 );
    u32Sum += pixel_satd_8width_msa( p_pix1 + 8, i_stride,
                                     p_pix2 + 8, i_stride2, 8 );

    return u32Sum;
}

int32_t x264_pixel_satd_16x16_msa( uint8_t *p_pix1, intptr_t i_stride,
                                   uint8_t *p_pix2, intptr_t i_stride2 )
{
    uint32_t u32Sum = 0;

    u32Sum = pixel_satd_8width_msa( p_pix1, i_stride, p_pix2, i_stride2, 16 );
    u32Sum += pixel_satd_8width_msa( p_pix1 + 8, i_stride,
                                     p_pix2 + 8, i_stride2, 16 );

    return u32Sum;
}

int32_t x264_pixel_sa8d_8x8_msa( uint8_t *p_pix1, intptr_t i_stride,
                                 uint8_t *p_pix2, intptr_t i_stride2 )
{
    int32_t i32Sum = sa8d_8x8_msa( p_pix1, i_stride, p_pix2, i_stride2 );

    return ( i32Sum + 2 ) >> 2;
}

int32_t x264_pixel_sa8d_16x16_msa( uint8_t *p_pix1, intptr_t i_stride,
                                   uint8_t *p_pix2, intptr_t i_stride2 )
{
    int32_t i32Sum = sa8d_8x8_msa( p_pix1, i_stride, p_pix2, i_stride2 ) +
                     sa8d_8x8_msa( p_pix1 + 8, i_stride,
                                   p_pix2 + 8, i_stride2 ) +
                     sa8d_8x8_msa( p_pix1 + 8 * i_stride, i_stride,
                                   p_pix2 + 8 * i_stride2, i_stride2 ) +
                     sa8d_8x8_msa( p_pix1 + 8 + 8 * i_stride, i_stride,
                                   p_pix2 + 8 + 8 * i_stride2, i_stride2 );

    return ( i32Sum + 2 ) >> 2;
}

void x264_intra_satd_x3_4x4_msa( uint8_t *p_enc, uint8_t *p_dec,
                                 int32_t p_sad_array[3] )
{
    x264_intra_predict_vert_4x4_msa( p_dec );
    p_sad_array[0] = x264_pixel_satd_4x4_msa( p_dec, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );

    x264_intra_predict_hor_4x4_msa( p_dec );
    p_sad_array[1] = x264_pixel_satd_4x4_msa( p_dec, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );

    x264_intra_predict_dc_4x4_msa( p_dec );
    p_sad_array[2] = x264_pixel_satd_4x4_msa( p_dec, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );
}

void x264_intra_satd_x3_16x16_msa( uint8_t *p_enc, uint8_t *p_dec,
                                   int32_t p_sad_array[3] )
{
    x264_intra_predict_vert_16x16_msa( p_dec );
    p_sad_array[0] = x264_pixel_satd_16x16_msa( p_dec, FDEC_STRIDE,
                                                p_enc, FENC_STRIDE );

    x264_intra_predict_hor_16x16_msa( p_dec );
    p_sad_array[1] = x264_pixel_satd_16x16_msa( p_dec, FDEC_STRIDE,
                                                p_enc, FENC_STRIDE );

    x264_intra_predict_dc_16x16_msa( p_dec );
    p_sad_array[2] = x264_pixel_satd_16x16_msa( p_dec, FDEC_STRIDE,
                                                p_enc, FENC_STRIDE );
}

void x264_intra_sa8d_x3_8x8_msa( uint8_t *p_enc, uint8_t p_edge[36],
                                 int32_t p_sad_array[3] )
{
    ALIGNED_ARRAY_16( uint8_t, pix, [8 * FDEC_STRIDE] );

    x264_intra_predict_v_8x8_msa( pix, p_edge );
    p_sad_array[0] = x264_pixel_sa8d_8x8_msa( pix, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );

    x264_intra_predict_h_8x8_msa( pix, p_edge );
    p_sad_array[1] = x264_pixel_sa8d_8x8_msa( pix, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );

    x264_intra_predict_dc_8x8_msa( pix, p_edge );
    p_sad_array[2] = x264_pixel_sa8d_8x8_msa( pix, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );
}

void x264_intra_satd_x3_8x8c_msa( uint8_t *p_enc, uint8_t *p_dec,
                                  int32_t p_sad_array[3] )
{
    x264_intra_predict_dc_4blk_8x8_msa( p_dec );
    p_sad_array[0] = x264_pixel_satd_8x8_msa( p_dec, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );

    x264_intra_predict_hor_8x8_msa( p_dec );
    p_sad_array[1] = x264_pixel_satd_8x8_msa( p_dec, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );

    x264_intra_predict_vert_8x8_msa( p_dec );
    p_sad_array[2] = x264_pixel_satd_8x8_msa( p_dec, FDEC_STRIDE,
                                              p_enc, FENC_STRIDE );
}

uint64_t x264_pixel_var_16x16_msa( uint8_t *p_pix, intptr_t i_stride )
{
    return avc_pixel_var16width_msa( p_pix, i_stride, 16 );
}

uint64_t x264_pixel_var_8x16_msa( uint8_t *p_pix, intptr_t i_stride )
{
    return avc_pixel_var8width_msa( p_pix, i_stride, 16 );
}

uint64_t x264_pixel_var_8x8_msa( uint8_t *p_pix, intptr_t i_stride )
{
    return avc_pixel_var8width_msa( p_pix, i_stride, 8 );
}

int32_t x264_pixel_var2_8x16_msa( uint8_t *p_pix1, intptr_t i_stride1,
                                  uint8_t *p_pix2, intptr_t i_stride2,
                                  int32_t *p_ssd )
{
    int32_t i_var = 0, i_diff = 0, i_sqr = 0;

    i_sqr = sse_diff_8width_msa( p_pix1, i_stride1, p_pix2, i_stride2, 16,
                                 &i_diff );
    i_var = VARIANCE_WxH( i_sqr, i_diff, 7 );
    *p_ssd = i_sqr;

    return i_var;
}

int32_t x264_pixel_var2_8x8_msa( uint8_t *p_pix1, intptr_t i_stride1,
                                 uint8_t *p_pix2, intptr_t i_stride2,
                                 int32_t *p_ssd )
{
    int32_t i_var = 0, i_diff = 0, i_sqr = 0;

    i_sqr = sse_diff_8width_msa( p_pix1, i_stride1,
                                 p_pix2, i_stride2, 8, &i_diff );
    i_var = VARIANCE_WxH( i_sqr, i_diff, 6 );
    *p_ssd = i_sqr;

    return i_var;
}
#endif
