Chromium Code Reviews| Index: source/row_msa.cc |
| diff --git a/source/row_msa.cc b/source/row_msa.cc |
| index 3a55b0b7554a2a0e4506a633e73242c879187f74..d0990b4ff8f799d82157da76e6a145e7e22df81c 100644 |
| --- a/source/row_msa.cc |
| +++ b/source/row_msa.cc |
| @@ -19,6 +19,66 @@ namespace libyuv { |
| extern "C" { |
| #endif |
| +#define I422TORGB(in0, in1, in2, ub, vr, ug, vg, \ |
|
fbarchard1
2016/10/24 17:36:32
Could use a brief comment to describe what this ma
|
| + bb, bg, br, yg, out0, out1, out2) { \ |
| + v8i16 vec0_m; \ |
| + v4i32 reg0_m, reg1_m, reg2_m, reg3_m, reg4_m; \ |
| + v4i32 reg5_m, reg6_m, reg7_m, reg8_m, reg9_m; \ |
| + v4i32 max_val_m = __msa_ldi_w(255); \ |
| + v8i16 zero_m = { 0 }; \ |
| + \ |
| + in1 = (v16u8) __msa_ilvr_b((v16i8) in1, (v16i8) in1); \ |
| + in2 = (v16u8) __msa_ilvr_b((v16i8) in2, (v16i8) in2); \ |
| + vec0_m = (v8i16) __msa_ilvr_b((v16i8) in0, (v16i8) in0); \ |
| + reg0_m = (v4i32) __msa_ilvr_h(zero_m, vec0_m); \ |
| + reg1_m = (v4i32) __msa_ilvl_h(zero_m, vec0_m); \ |
| + reg0_m *= vec_yg; \ |
| + reg1_m *= vec_yg; \ |
| + reg0_m = __msa_srai_w(reg0_m, 16); \ |
| + reg1_m = __msa_srai_w(reg1_m, 16); \ |
| + reg4_m = reg0_m + br; \ |
| + reg5_m = reg1_m + br; \ |
| + reg2_m = reg0_m + bg; \ |
| + reg3_m = reg1_m + bg; \ |
| + reg0_m += bb; \ |
| + reg1_m += bb; \ |
| + vec0_m = (v8i16) __msa_ilvr_b((v16i8) zero_m, (v16i8) in1); \ |
| + reg6_m = (v4i32) __msa_ilvr_h(zero_m, (v8i16) vec0_m); \ |
| + reg7_m = (v4i32) __msa_ilvl_h(zero_m, (v8i16) vec0_m); \ |
| + vec0_m = (v8i16) __msa_ilvr_b((v16i8) zero_m, (v16i8) in2); \ |
| + reg8_m = (v4i32) __msa_ilvr_h(zero_m, (v8i16) vec0_m); \ |
| + reg9_m = (v4i32) __msa_ilvl_h(zero_m, (v8i16) vec0_m); \ |
| + reg0_m -= reg6_m * ub; \ |
| + reg1_m -= reg7_m * ub; \ |
| + reg2_m -= reg6_m * ug; \ |
| + reg3_m -= reg7_m * ug; \ |
| + reg4_m -= reg8_m * vr; \ |
| + reg5_m -= reg9_m * vr; \ |
| + reg2_m -= reg8_m * vg; \ |
| + reg3_m -= reg9_m * vg; \ |
| + reg0_m = __msa_srai_w(reg0_m, 6); \ |
| + reg1_m = __msa_srai_w(reg1_m, 6); \ |
| + reg2_m = __msa_srai_w(reg2_m, 6); \ |
| + reg3_m = __msa_srai_w(reg3_m, 6); \ |
| + reg4_m = __msa_srai_w(reg4_m, 6); \ |
| + reg5_m = __msa_srai_w(reg5_m, 6); \ |
| + reg0_m = __msa_maxi_s_w(reg0_m, 0); \ |
| + reg1_m = __msa_maxi_s_w(reg1_m, 0); \ |
| + reg2_m = __msa_maxi_s_w(reg2_m, 0); \ |
| + reg3_m = __msa_maxi_s_w(reg3_m, 0); \ |
| + reg4_m = __msa_maxi_s_w(reg4_m, 0); \ |
| + reg5_m = __msa_maxi_s_w(reg5_m, 0); \ |
| + reg0_m = __msa_min_s_w(reg0_m, max_val_m); \ |
| + reg1_m = __msa_min_s_w(reg1_m, max_val_m); \ |
| + reg2_m = __msa_min_s_w(reg2_m, max_val_m); \ |
| + reg3_m = __msa_min_s_w(reg3_m, max_val_m); \ |
| + reg4_m = __msa_min_s_w(reg4_m, max_val_m); \ |
| + reg5_m = __msa_min_s_w(reg5_m, max_val_m); \ |
| + out0 = __msa_pckev_h((v8i16) reg1_m, (v8i16) reg0_m); \ |
| + out1 = __msa_pckev_h((v8i16) reg3_m, (v8i16) reg2_m); \ |
| + out2 = __msa_pckev_h((v8i16) reg5_m, (v8i16) reg4_m); \ |
| +} |
| + |
| void MirrorRow_MSA(const uint8* src, uint8* dst, int width) { |
| int x; |
| v16u8 src0, src1, src2, src3; |
| @@ -101,6 +161,90 @@ void I422ToUYVYRow_MSA(const uint8* src_y, |
| } |
| } |
| +void I422ToARGBRow_MSA(const uint8* src_y, const uint8* src_u, |
| + const uint8* src_v, uint8* rgb_buf, |
| + const struct YuvConstants* yuvconstants, int width) { |
| + int x; |
| + int32 data_u, data_v; |
| + int64 data_y; |
| + v16u8 src0, src1, src2, dst0, dst1; |
| + v8i16 vec0, vec1, vec2; |
| + v4i32 vec_ub, vec_vr, vec_ug, vec_vg, vec_bb, vec_bg, vec_br, vec_yg; |
| + v16u8 const_255 = (v16u8) __msa_ldi_b(255); |
| + v4i32 zero = { 0 }; |
| + |
| + vec_ub = __msa_fill_w(yuvconstants->kUVToB[0]); |
| + vec_vr = __msa_fill_w(yuvconstants->kUVToR[1]); |
| + vec_ug = __msa_fill_w(yuvconstants->kUVToG[0]); |
| + vec_vg = __msa_fill_w(yuvconstants->kUVToG[1]); |
| + vec_bb = __msa_fill_w(yuvconstants->kUVBiasB[0]); |
| + vec_bg = __msa_fill_w(yuvconstants->kUVBiasG[0]); |
| + vec_br = __msa_fill_w(yuvconstants->kUVBiasR[0]); |
| + vec_yg = __msa_fill_w(yuvconstants->kYToRgb[0]); |
| + |
| + for (x = 0; x < width; x += 8) { |
| + data_y = LD(src_y); |
|
fbarchard1
2016/10/24 17:36:32
Consider a macro to read I422, as there are many Y
|
| + data_u = LW(src_u); |
| + data_v = LW(src_v); |
| + src0 = (v16u8) __msa_insert_d((v2i64) zero, 0, data_y); |
| + src1 = (v16u8) __msa_insert_w(zero, 0, data_u); |
| + src2 = (v16u8) __msa_insert_w(zero, 0, data_v); |
|
fbarchard1
2016/10/24 17:36:32
No way to load 4 or 8 bytes directly into an MSA r
|
| + I422TORGB(src0, src1, src2, vec_ub, vec_vr, vec_ug, vec_vg, |
| + vec_bb, vec_bg, vec_br, vec_yg, vec0, vec1, vec2); |
| + vec0 = (v8i16) __msa_ilvev_b((v16i8) vec1, (v16i8) vec0); |
| + vec1 = (v8i16) __msa_ilvev_b((v16i8) const_255, (v16i8) vec2); |
| + dst0 = (v16u8) __msa_ilvr_h((v8i16) vec1, (v8i16) vec0); |
| + dst1 = (v16u8) __msa_ilvl_h((v8i16) vec1, (v8i16) vec0); |
| + ST_UB2(dst0, dst1, rgb_buf, 16); |
| + src_y += 8; |
| + src_u += 4; |
| + src_v += 4; |
| + rgb_buf += 32; |
| + } |
| +} |
| + |
| +void I422ToRGBARow_MSA(const uint8* src_y, const uint8* src_u, |
| + const uint8* src_v, uint8* rgb_buf, |
| + const struct YuvConstants* yuvconstants, int width) { |
| + int x; |
| + int64 data_y; |
| + int32 data_u, data_v; |
| + v16u8 src0, src1, src2, dst0, dst1; |
| + v8i16 vec0, vec1, vec2; |
| + v4i32 vec_ub, vec_vr, vec_ug, vec_vg, vec_bb, vec_bg, vec_br, vec_yg; |
| + v16u8 const_255 = (v16u8) __msa_ldi_b(255); |
| + v4i32 zero = { 0 }; |
| + |
| + vec_ub = __msa_fill_w(yuvconstants->kUVToB[0]); |
| + vec_vr = __msa_fill_w(yuvconstants->kUVToR[1]); |
| + vec_ug = __msa_fill_w(yuvconstants->kUVToG[0]); |
| + vec_vg = __msa_fill_w(yuvconstants->kUVToG[1]); |
| + vec_bb = __msa_fill_w(yuvconstants->kUVBiasB[0]); |
| + vec_bg = __msa_fill_w(yuvconstants->kUVBiasG[0]); |
| + vec_br = __msa_fill_w(yuvconstants->kUVBiasR[0]); |
| + vec_yg = __msa_fill_w(yuvconstants->kYToRgb[0]); |
| + |
| + for (x = 0; x < width; x += 8) { |
| + data_y = LD(src_y); |
| + data_u = LW(src_u); |
| + data_v = LW(src_v); |
| + src0 = (v16u8) __msa_insert_d((v2i64) zero, 0, data_y); |
| + src1 = (v16u8) __msa_insert_w(zero, 0, data_u); |
| + src2 = (v16u8) __msa_insert_w(zero, 0, data_v); |
| + I422TORGB(src0, src1, src2, vec_ub, vec_vr, vec_ug, vec_vg, |
| + vec_bb, vec_bg, vec_br, vec_yg, vec0, vec1, vec2); |
| + vec0 = (v8i16) __msa_ilvev_b((v16i8) vec0, (v16i8) const_255); |
| + vec1 = (v8i16) __msa_ilvev_b((v16i8) vec2, (v16i8) vec1); |
| + dst0 = (v16u8) __msa_ilvr_h(vec1, vec0); |
| + dst1 = (v16u8) __msa_ilvl_h(vec1, vec0); |
| + ST_UB2(dst0, dst1, rgb_buf, 16); |
| + src_y += 8; |
| + src_u += 4; |
| + src_v += 4; |
| + rgb_buf += 32; |
| + } |
| +} |
| + |
| void YUY2ToYRow_MSA(const uint8* src_yuy2, uint8* dst_y, int width) { |
| int x; |
| v16u8 src0, src1, src2, src3, dst0, dst1; |