blob: 57f705d25ec56986d47753e2c35f1e3051268060 [file] [log] [blame]
/*
* Copyright (c) 2015 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 "./vp8_rtcd.h"
#include "vp8/common/blockd.h"
#include "vp8/common/mips/msa/vp8_macros_msa.h"
static void intra_predict_vert_8x8_msa(uint8_t *src, uint8_t *dst,
int32_t dst_stride)
{
uint64_t out = LD(src);
SD4(out, out, out, out, dst, dst_stride);
dst += (4 * dst_stride);
SD4(out, out, out, out, dst, dst_stride);
}
static void intra_predict_vert_16x16_msa(uint8_t *src, uint8_t *dst,
int32_t dst_stride)
{
v16u8 out = LD_UB(src);
ST_UB8(out, out, out, out, out, out, out, out, dst, dst_stride);
dst += (8 * dst_stride);
ST_UB8(out, out, out, out, out, out, out, out, dst, dst_stride);
}
static void intra_predict_horiz_8x8_msa(uint8_t *src, int32_t src_stride,
uint8_t *dst, int32_t dst_stride)
{
uint64_t out0, out1, out2, out3, out4, out5, out6, out7;
out0 = src[0 * src_stride] * 0x0101010101010101ull;
out1 = src[1 * src_stride] * 0x0101010101010101ull;
out2 = src[2 * src_stride] * 0x0101010101010101ull;
out3 = src[3 * src_stride] * 0x0101010101010101ull;
out4 = src[4 * src_stride] * 0x0101010101010101ull;
out5 = src[5 * src_stride] * 0x0101010101010101ull;
out6 = src[6 * src_stride] * 0x0101010101010101ull;
out7 = src[7 * src_stride] * 0x0101010101010101ull;
SD4(out0, out1, out2, out3, dst, dst_stride);
dst += (4 * dst_stride);
SD4(out4, out5, out6, out7, dst, dst_stride);
}
static void intra_predict_horiz_16x16_msa(uint8_t *src, int32_t src_stride,
uint8_t *dst, int32_t dst_stride)
{
uint32_t row;
uint8_t inp0, inp1, inp2, inp3;
v16u8 src0, src1, src2, src3;
for (row = 4; row--;)
{
inp0 = src[0];
src += src_stride;
inp1 = src[0];
src += src_stride;
inp2 = src[0];
src += src_stride;
inp3 = src[0];
src += src_stride;
src0 = (v16u8)__msa_fill_b(inp0);
src1 = (v16u8)__msa_fill_b(inp1);
src2 = (v16u8)__msa_fill_b(inp2);
src3 = (v16u8)__msa_fill_b(inp3);
ST_UB4(src0, src1, src2, src3, dst, dst_stride);
dst += (4 * dst_stride);
}
}
static void intra_predict_dc_8x8_msa(uint8_t *src_top, uint8_t *src_left,
int32_t src_stride_left,
uint8_t *dst, int32_t dst_stride,
uint8_t is_above, uint8_t is_left)
{
uint32_t row, addition = 0;
uint64_t out;
v16u8 src_above, store;
v8u16 sum_above;
v4u32 sum_top;
v2u64 sum;
if (is_left && is_above)
{
src_above = LD_UB(src_top);
sum_above = __msa_hadd_u_h(src_above, src_above);
sum_top = __msa_hadd_u_w(sum_above, sum_above);
sum = __msa_hadd_u_d(sum_top, sum_top);
addition = __msa_copy_u_w((v4i32)sum, 0);
for (row = 0; row < 8; ++row)
{
addition += src_left[row * src_stride_left];
}
addition = (addition + 8) >> 4;
store = (v16u8)__msa_fill_b(addition);
}
else if (is_left)
{
for (row = 0; row < 8; ++row)
{
addition += src_left[row * src_stride_left];
}
addition = (addition + 4) >> 3;
store = (v16u8)__msa_fill_b(addition);
}
else if (is_above)
{
src_above = LD_UB(src_top);
sum_above = __msa_hadd_u_h(src_above, src_above);
sum_top = __msa_hadd_u_w(sum_above, sum_above);
sum = __msa_hadd_u_d(sum_top, sum_top);
sum = (v2u64)__msa_srari_d((v2i64)sum, 3);
store = (v16u8)__msa_splati_b((v16i8)sum, 0);
}
else
{
store = (v16u8)__msa_ldi_b(128);
}
out = __msa_copy_u_d((v2i64)store, 0);
SD4(out, out, out, out, dst, dst_stride);
dst += (4 * dst_stride);
SD4(out, out, out, out, dst, dst_stride);
}
static void intra_predict_dc_16x16_msa(uint8_t *src_top, uint8_t *src_left,
int32_t src_stride_left,
uint8_t *dst, int32_t dst_stride,
uint8_t is_above, uint8_t is_left)
{
uint32_t row;
uint32_t addition = 0;
v16u8 src_above, out;
v8u16 sum_above;
v4u32 sum_top;
v2u64 sum;
if (is_left && is_above)
{
src_above = LD_UB(src_top);
sum_above = __msa_hadd_u_h(src_above, src_above);
sum_top = __msa_hadd_u_w(sum_above, sum_above);
sum = __msa_hadd_u_d(sum_top, sum_top);
sum_top = (v4u32)__msa_pckev_w((v4i32)sum, (v4i32)sum);
sum = __msa_hadd_u_d(sum_top, sum_top);
addition = __msa_copy_u_w((v4i32)sum, 0);
for (row = 0; row < 16; ++row)
{
addition += src_left[row * src_stride_left];
}
addition = (addition + 16) >> 5;
out = (v16u8)__msa_fill_b(addition);
}
else if (is_left)
{
for (row = 0; row < 16; ++row)
{
addition += src_left[row * src_stride_left];
}
addition = (addition + 8) >> 4;
out = (v16u8)__msa_fill_b(addition);
}
else if (is_above)
{
src_above = LD_UB(src_top);
sum_above = __msa_hadd_u_h(src_above, src_above);
sum_top = __msa_hadd_u_w(sum_above, sum_above);
sum = __msa_hadd_u_d(sum_top, sum_top);
sum_top = (v4u32)__msa_pckev_w((v4i32)sum, (v4i32)sum);
sum = __msa_hadd_u_d(sum_top, sum_top);
sum = (v2u64)__msa_srari_d((v2i64)sum, 4);
out = (v16u8)__msa_splati_b((v16i8)sum, 0);
}
else
{
out = (v16u8)__msa_ldi_b(128);
}
ST_UB8(out, out, out, out, out, out, out, out, dst, dst_stride);
dst += (8 * dst_stride);
ST_UB8(out, out, out, out, out, out, out, out, dst, dst_stride);
}
void vp8_build_intra_predictors_mby_s_msa(struct macroblockd *x,
unsigned char *yabove_row,
unsigned char *yleft,
int left_stride,
unsigned char *ypred_ptr,
int y_stride)
{
uint32_t row, col;
uint8_t ytop_left = yabove_row[-1];
switch (x->mode_info_context->mbmi.mode)
{
case DC_PRED:
intra_predict_dc_16x16_msa(yabove_row, yleft, left_stride,
ypred_ptr, y_stride,
x->up_available, x->left_available);
break;
case V_PRED:
intra_predict_vert_16x16_msa(yabove_row, ypred_ptr, y_stride);
break;
case H_PRED:
intra_predict_horiz_16x16_msa(yleft, left_stride, ypred_ptr,
y_stride);
break;
case TM_PRED:
for (row = 0; row < 16; ++row)
{
for (col = 0; col < 16; ++col)
{
int pred = yleft[row * left_stride] + yabove_row[col] -
ytop_left;
if (pred < 0)
pred = 0;
if (pred > 255)
pred = 255;
ypred_ptr[col] = pred;
}
ypred_ptr += y_stride;
}
break;
case B_PRED:
case NEARESTMV:
case NEARMV:
case ZEROMV:
case NEWMV:
case SPLITMV:
case MB_MODE_COUNT:
break;
}
}
void vp8_build_intra_predictors_mbuv_s_msa(struct macroblockd *x,
unsigned char *uabove_row,
unsigned char *vabove_row,
unsigned char *uleft,
unsigned char *vleft,
int left_stride,
unsigned char *upred_ptr,
unsigned char *vpred_ptr,
int pred_stride)
{
uint32_t row, col;
uint8_t utop_left = uabove_row[-1];
uint8_t vtop_left = vabove_row[-1];
switch (x->mode_info_context->mbmi.uv_mode)
{
case DC_PRED:
intra_predict_dc_8x8_msa(uabove_row, uleft, left_stride,
upred_ptr, pred_stride,
x->up_available, x->left_available);
intra_predict_dc_8x8_msa(vabove_row, vleft, left_stride,
vpred_ptr, pred_stride,
x->up_available, x->left_available);
break;
case V_PRED:
intra_predict_vert_8x8_msa(uabove_row, upred_ptr, pred_stride);
intra_predict_vert_8x8_msa(vabove_row, vpred_ptr, pred_stride);
break;
case H_PRED:
intra_predict_horiz_8x8_msa(uleft, left_stride, upred_ptr,
pred_stride);
intra_predict_horiz_8x8_msa(vleft, left_stride, vpred_ptr,
pred_stride);
break;
case TM_PRED:
for (row = 0; row < 8; ++row)
{
for (col = 0; col < 8; ++col)
{
int predu = uleft[row * left_stride] + uabove_row[col] -
utop_left;
int predv = vleft[row * left_stride] + vabove_row[col] -
vtop_left;
if (predu < 0)
predu = 0;
if (predu > 255)
predu = 255;
if (predv < 0)
predv = 0;
if (predv > 255)
predv = 255;
upred_ptr[col] = predu;
vpred_ptr[col] = predv;
}
upred_ptr += pred_stride;
vpred_ptr += pred_stride;
}
break;
case B_PRED:
case NEARESTMV:
case NEARMV:
case ZEROMV:
case NEWMV:
case SPLITMV:
case MB_MODE_COUNT:
break;
}
}