CoCalc -- dec

GitHub Repository: Tetragramm/opencv
Path: blob/master/3rdparty/libwebp/src/dsp/dec_msa.c
¹⁶³⁴⁸ views
1
// Copyright 2016 Google Inc. All Rights Reserved.
2
//
3
// Use of this source code is governed by a BSD-style license
4
// that can be found in the COPYING file in the root of the source
5
// tree. An additional intellectual property rights grant can be found
6
// in the file PATENTS. All contributing project authors may
7
// be found in the AUTHORS file in the root of the source tree.
8
// -----------------------------------------------------------------------------
9
//
10
// MSA version of dsp functions
11
//
12
// Author(s):  Prashant Patil   ([email protected])
13

14

15
#include "src/dsp/dsp.h"
16

17
#if defined(WEBP_USE_MSA)
18

19
#include "src/dsp/msa_macro.h"
20

21
//------------------------------------------------------------------------------
22
// Transforms
23

24
#define IDCT_1D_W(in0, in1, in2, in3, out0, out1, out2, out3) {  \
25
  v4i32 a1_m, b1_m, c1_m, d1_m;                                  \
26
  v4i32 c_tmp1_m, c_tmp2_m, d_tmp1_m, d_tmp2_m;                  \
27
  const v4i32 cospi8sqrt2minus1 = __msa_fill_w(20091);           \
28
  const v4i32 sinpi8sqrt2 = __msa_fill_w(35468);                 \
29
                                                                 \
30
  a1_m = in0 + in2;                                              \
31
  b1_m = in0 - in2;                                              \
32
  c_tmp1_m = (in1 * sinpi8sqrt2) >> 16;                          \
33
  c_tmp2_m = in3 + ((in3 * cospi8sqrt2minus1) >> 16);            \
34
  c1_m = c_tmp1_m - c_tmp2_m;                                    \
35
  d_tmp1_m = in1 + ((in1 * cospi8sqrt2minus1) >> 16);            \
36
  d_tmp2_m = (in3 * sinpi8sqrt2) >> 16;                          \
37
  d1_m = d_tmp1_m + d_tmp2_m;                                    \
38
  BUTTERFLY_4(a1_m, b1_m, c1_m, d1_m, out0, out1, out2, out3);   \
39
}
40
#define MULT1(a) ((((a) * 20091) >> 16) + (a))
41
#define MULT2(a) (((a) * 35468) >> 16)
42

43
static void TransformOne(const int16_t* in, uint8_t* dst) {
44
  v8i16 input0, input1;
45
  v4i32 in0, in1, in2, in3, hz0, hz1, hz2, hz3, vt0, vt1, vt2, vt3;
46
  v4i32 res0, res1, res2, res3;
47
  const v16i8 zero = { 0 };
48
  v16i8 dest0, dest1, dest2, dest3;
49

50
  LD_SH2(in, 8, input0, input1);
51
  UNPCK_SH_SW(input0, in0, in1);
52
  UNPCK_SH_SW(input1, in2, in3);
53
  IDCT_1D_W(in0, in1, in2, in3, hz0, hz1, hz2, hz3);
54
  TRANSPOSE4x4_SW_SW(hz0, hz1, hz2, hz3, hz0, hz1, hz2, hz3);
55
  IDCT_1D_W(hz0, hz1, hz2, hz3, vt0, vt1, vt2, vt3);
56
  SRARI_W4_SW(vt0, vt1, vt2, vt3, 3);
57
  TRANSPOSE4x4_SW_SW(vt0, vt1, vt2, vt3, vt0, vt1, vt2, vt3);
58
  LD_SB4(dst, BPS, dest0, dest1, dest2, dest3);
59
  ILVR_B4_SW(zero, dest0, zero, dest1, zero, dest2, zero, dest3,
60
             res0, res1, res2, res3);
61
  ILVR_H4_SW(zero, res0, zero, res1, zero, res2, zero, res3,
62
             res0, res1, res2, res3);
63
  ADD4(res0, vt0, res1, vt1, res2, vt2, res3, vt3, res0, res1, res2, res3);
64
  CLIP_SW4_0_255(res0, res1, res2, res3);
65
  PCKEV_B2_SW(res0, res1, res2, res3, vt0, vt1);
66
  res0 = (v4i32)__msa_pckev_b((v16i8)vt0, (v16i8)vt1);
67
  ST4x4_UB(res0, res0, 3, 2, 1, 0, dst, BPS);
68
}
69

70
static void TransformTwo(const int16_t* in, uint8_t* dst, int do_two) {
71
  TransformOne(in, dst);
72
  if (do_two) {
73
    TransformOne(in + 16, dst + 4);
74
  }
75
}
76

77
static void TransformWHT(const int16_t* in, int16_t* out) {
78
  v8i16 input0, input1;
79
  const v8i16 mask0 = { 0, 1, 2, 3, 8, 9, 10, 11 };
80
  const v8i16 mask1 = { 4, 5, 6, 7, 12, 13, 14, 15 };
81
  const v8i16 mask2 = { 0, 4, 8, 12, 1, 5, 9, 13 };
82
  const v8i16 mask3 = { 3, 7, 11, 15, 2, 6, 10, 14 };
83
  v8i16 tmp0, tmp1, tmp2, tmp3;
84
  v8i16 out0, out1;
85

86
  LD_SH2(in, 8, input0, input1);
87
  input1 = SLDI_SH(input1, input1, 8);
88
  tmp0 = input0 + input1;
89
  tmp1 = input0 - input1;
90
  VSHF_H2_SH(tmp0, tmp1, tmp0, tmp1, mask0, mask1, tmp2, tmp3);
91
  out0 = tmp2 + tmp3;
92
  out1 = tmp2 - tmp3;
93
  VSHF_H2_SH(out0, out1, out0, out1, mask2, mask3, input0, input1);
94
  tmp0 = input0 + input1;
95
  tmp1 = input0 - input1;
96
  VSHF_H2_SH(tmp0, tmp1, tmp0, tmp1, mask0, mask1, tmp2, tmp3);
97
  tmp0 = tmp2 + tmp3;
98
  tmp1 = tmp2 - tmp3;
99
  ADDVI_H2_SH(tmp0, 3, tmp1, 3, out0, out1);
100
  SRAI_H2_SH(out0, out1, 3);
101
  out[0] = __msa_copy_s_h(out0, 0);
102
  out[16] = __msa_copy_s_h(out0, 4);
103
  out[32] = __msa_copy_s_h(out1, 0);
104
  out[48] = __msa_copy_s_h(out1, 4);
105
  out[64] = __msa_copy_s_h(out0, 1);
106
  out[80] = __msa_copy_s_h(out0, 5);
107
  out[96] = __msa_copy_s_h(out1, 1);
108
  out[112] = __msa_copy_s_h(out1, 5);
109
  out[128] = __msa_copy_s_h(out0, 2);
110
  out[144] = __msa_copy_s_h(out0, 6);
111
  out[160] = __msa_copy_s_h(out1, 2);
112
  out[176] = __msa_copy_s_h(out1, 6);
113
  out[192] = __msa_copy_s_h(out0, 3);
114
  out[208] = __msa_copy_s_h(out0, 7);
115
  out[224] = __msa_copy_s_h(out1, 3);
116
  out[240] = __msa_copy_s_h(out1, 7);
117
}
118

119
static void TransformDC(const int16_t* in, uint8_t* dst) {
120
  const int DC = (in[0] + 4) >> 3;
121
  const v8i16 tmp0 = __msa_fill_h(DC);
122
  ADDBLK_ST4x4_UB(tmp0, tmp0, tmp0, tmp0, dst, BPS);
123
}
124

125
static void TransformAC3(const int16_t* in, uint8_t* dst) {
126
  const int a = in[0] + 4;
127
  const int c4 = MULT2(in[4]);
128
  const int d4 = MULT1(in[4]);
129
  const int in2 = MULT2(in[1]);
130
  const int in3 = MULT1(in[1]);
131
  v4i32 tmp0 = { 0 };
132
  v4i32 out0 = __msa_fill_w(a + d4);
133
  v4i32 out1 = __msa_fill_w(a + c4);
134
  v4i32 out2 = __msa_fill_w(a - c4);
135
  v4i32 out3 = __msa_fill_w(a - d4);
136
  v4i32 res0, res1, res2, res3;
137
  const v4i32 zero = { 0 };
138
  v16u8 dest0, dest1, dest2, dest3;
139

140
  INSERT_W4_SW(in3, in2, -in2, -in3, tmp0);
141
  ADD4(out0, tmp0, out1, tmp0, out2, tmp0, out3, tmp0,
142
       out0, out1, out2, out3);
143
  SRAI_W4_SW(out0, out1, out2, out3, 3);
144
  LD_UB4(dst, BPS, dest0, dest1, dest2, dest3);
145
  ILVR_B4_SW(zero, dest0, zero, dest1, zero, dest2, zero, dest3,
146
             res0, res1, res2, res3);
147
  ILVR_H4_SW(zero, res0, zero, res1, zero, res2, zero, res3,
148
             res0, res1, res2, res3);
149
  ADD4(res0, out0, res1, out1, res2, out2, res3, out3, res0, res1, res2, res3);
150
  CLIP_SW4_0_255(res0, res1, res2, res3);
151
  PCKEV_B2_SW(res0, res1, res2, res3, out0, out1);
152
  res0 = (v4i32)__msa_pckev_b((v16i8)out0, (v16i8)out1);
153
  ST4x4_UB(res0, res0, 3, 2, 1, 0, dst, BPS);
154
}
155

156
//------------------------------------------------------------------------------
157
// Edge filtering functions
158

159
#define FLIP_SIGN2(in0, in1, out0, out1) {  \
160
  out0 = (v16i8)__msa_xori_b(in0, 0x80);    \
161
  out1 = (v16i8)__msa_xori_b(in1, 0x80);    \
162
}
163

164
#define FLIP_SIGN4(in0, in1, in2, in3, out0, out1, out2, out3) {  \
165
  FLIP_SIGN2(in0, in1, out0, out1);                               \
166
  FLIP_SIGN2(in2, in3, out2, out3);                               \
167
}
168

169
#define FILT_VAL(q0_m, p0_m, mask, filt) do {  \
170
  v16i8 q0_sub_p0;                             \
171
  q0_sub_p0 = __msa_subs_s_b(q0_m, p0_m);      \
172
  filt = __msa_adds_s_b(filt, q0_sub_p0);      \
173
  filt = __msa_adds_s_b(filt, q0_sub_p0);      \
174
  filt = __msa_adds_s_b(filt, q0_sub_p0);      \
175
  filt = filt & mask;                          \
176
} while (0)
177

178
#define FILT2(q_m, p_m, q, p) do {            \
179
  u_r = SRAI_H(temp1, 7);                     \
180
  u_r = __msa_sat_s_h(u_r, 7);                \
181
  u_l = SRAI_H(temp3, 7);                     \
182
  u_l = __msa_sat_s_h(u_l, 7);                \
183
  u = __msa_pckev_b((v16i8)u_l, (v16i8)u_r);  \
184
  q_m = __msa_subs_s_b(q_m, u);               \
185
  p_m = __msa_adds_s_b(p_m, u);               \
186
  q = __msa_xori_b((v16u8)q_m, 0x80);         \
187
  p = __msa_xori_b((v16u8)p_m, 0x80);         \
188
} while (0)
189

190
#define LPF_FILTER4_4W(p1, p0, q0, q1, mask, hev) do {  \
191
  v16i8 p1_m, p0_m, q0_m, q1_m;                         \
192
  v16i8 filt, t1, t2;                                   \
193
  const v16i8 cnst4b = __msa_ldi_b(4);                  \
194
  const v16i8 cnst3b = __msa_ldi_b(3);                  \
195
                                                        \
196
  FLIP_SIGN4(p1, p0, q0, q1, p1_m, p0_m, q0_m, q1_m);   \
197
  filt = __msa_subs_s_b(p1_m, q1_m);                    \
198
  filt = filt & hev;                                    \
199
  FILT_VAL(q0_m, p0_m, mask, filt);                     \
200
  t1 = __msa_adds_s_b(filt, cnst4b);                    \
201
  t1 = SRAI_B(t1, 3);                                   \
202
  t2 = __msa_adds_s_b(filt, cnst3b);                    \
203
  t2 = SRAI_B(t2, 3);                                   \
204
  q0_m = __msa_subs_s_b(q0_m, t1);                      \
205
  q0 = __msa_xori_b((v16u8)q0_m, 0x80);                 \
206
  p0_m = __msa_adds_s_b(p0_m, t2);                      \
207
  p0 = __msa_xori_b((v16u8)p0_m, 0x80);                 \
208
  filt = __msa_srari_b(t1, 1);                          \
209
  hev = __msa_xori_b(hev, 0xff);                        \
210
  filt = filt & hev;                                    \
211
  q1_m = __msa_subs_s_b(q1_m, filt);                    \
212
  q1 = __msa_xori_b((v16u8)q1_m, 0x80);                 \
213
  p1_m = __msa_adds_s_b(p1_m, filt);                    \
214
  p1 = __msa_xori_b((v16u8)p1_m, 0x80);                 \
215
} while (0)
216

217
#define LPF_MBFILTER(p2, p1, p0, q0, q1, q2, mask, hev) do {  \
218
  v16i8 p2_m, p1_m, p0_m, q2_m, q1_m, q0_m;                   \
219
  v16i8 u, filt, t1, t2, filt_sign;                           \
220
  v8i16 filt_r, filt_l, u_r, u_l;                             \
221
  v8i16 temp0, temp1, temp2, temp3;                           \
222
  const v16i8 cnst4b = __msa_ldi_b(4);                        \
223
  const v16i8 cnst3b = __msa_ldi_b(3);                        \
224
  const v8i16 cnst9h = __msa_ldi_h(9);                        \
225
  const v8i16 cnst63h = __msa_ldi_h(63);                      \
226
                                                              \
227
  FLIP_SIGN4(p1, p0, q0, q1, p1_m, p0_m, q0_m, q1_m);         \
228
  filt = __msa_subs_s_b(p1_m, q1_m);                          \
229
  FILT_VAL(q0_m, p0_m, mask, filt);                           \
230
  FLIP_SIGN2(p2, q2, p2_m, q2_m);                             \
231
  t2 = filt & hev;                                            \
232
  /* filt_val &= ~hev */                                      \
233
  hev = __msa_xori_b(hev, 0xff);                              \
234
  filt = filt & hev;                                          \
235
  t1 = __msa_adds_s_b(t2, cnst4b);                            \
236
  t1 = SRAI_B(t1, 3);                                         \
237
  t2 = __msa_adds_s_b(t2, cnst3b);                            \
238
  t2 = SRAI_B(t2, 3);                                         \
239
  q0_m = __msa_subs_s_b(q0_m, t1);                            \
240
  p0_m = __msa_adds_s_b(p0_m, t2);                            \
241
  filt_sign = __msa_clti_s_b(filt, 0);                        \
242
  ILVRL_B2_SH(filt_sign, filt, filt_r, filt_l);               \
243
  /* update q2/p2 */                                          \
244
  temp0 = filt_r * cnst9h;                                    \
245
  temp1 = temp0 + cnst63h;                                    \
246
  temp2 = filt_l * cnst9h;                                    \
247
  temp3 = temp2 + cnst63h;                                    \
248
  FILT2(q2_m, p2_m, q2, p2);                                  \
249
  /* update q1/p1 */                                          \
250
  temp1 = temp1 + temp0;                                      \
251
  temp3 = temp3 + temp2;                                      \
252
  FILT2(q1_m, p1_m, q1, p1);                                  \
253
  /* update q0/p0 */                                          \
254
  temp1 = temp1 + temp0;                                      \
255
  temp3 = temp3 + temp2;                                      \
256
  FILT2(q0_m, p0_m, q0, p0);                                  \
257
} while (0)
258

259
#define LPF_MASK_HEV(p3_in, p2_in, p1_in, p0_in,                 \
260
                     q0_in, q1_in, q2_in, q3_in,                 \
261
                     limit_in, b_limit_in, thresh_in,            \
262
                     hev_out, mask_out) do {                     \
263
  v16u8 p3_asub_p2_m, p2_asub_p1_m, p1_asub_p0_m, q1_asub_q0_m;  \
264
  v16u8 p1_asub_q1_m, p0_asub_q0_m, q3_asub_q2_m, q2_asub_q1_m;  \
265
  v16u8 flat_out;                                                \
266
                                                                 \
267
  /* absolute subtraction of pixel values */                     \
268
  p3_asub_p2_m = __msa_asub_u_b(p3_in, p2_in);                   \
269
  p2_asub_p1_m = __msa_asub_u_b(p2_in, p1_in);                   \
270
  p1_asub_p0_m = __msa_asub_u_b(p1_in, p0_in);                   \
271
  q1_asub_q0_m = __msa_asub_u_b(q1_in, q0_in);                   \
272
  q2_asub_q1_m = __msa_asub_u_b(q2_in, q1_in);                   \
273
  q3_asub_q2_m = __msa_asub_u_b(q3_in, q2_in);                   \
274
  p0_asub_q0_m = __msa_asub_u_b(p0_in, q0_in);                   \
275
  p1_asub_q1_m = __msa_asub_u_b(p1_in, q1_in);                   \
276
  /* calculation of hev */                                       \
277
  flat_out = __msa_max_u_b(p1_asub_p0_m, q1_asub_q0_m);          \
278
  hev_out = (thresh_in < flat_out);                              \
279
  /* calculation of mask */                                      \
280
  p0_asub_q0_m = __msa_adds_u_b(p0_asub_q0_m, p0_asub_q0_m);     \
281
  p1_asub_q1_m = SRAI_B(p1_asub_q1_m, 1);                        \
282
  p0_asub_q0_m = __msa_adds_u_b(p0_asub_q0_m, p1_asub_q1_m);     \
283
  mask_out = (b_limit_in < p0_asub_q0_m);                        \
284
  mask_out = __msa_max_u_b(flat_out, mask_out);                  \
285
  p3_asub_p2_m = __msa_max_u_b(p3_asub_p2_m, p2_asub_p1_m);      \
286
  mask_out = __msa_max_u_b(p3_asub_p2_m, mask_out);              \
287
  q2_asub_q1_m = __msa_max_u_b(q2_asub_q1_m, q3_asub_q2_m);      \
288
  mask_out = __msa_max_u_b(q2_asub_q1_m, mask_out);              \
289
  mask_out = (limit_in < mask_out);                              \
290
  mask_out = __msa_xori_b(mask_out, 0xff);                       \
291
} while (0)
292

293
#define ST6x1_UB(in0, in0_idx, in1, in1_idx, pdst, stride) do { \
294
  const uint16_t tmp0_h = __msa_copy_s_h((v8i16)in1, in1_idx);  \
295
  const uint32_t tmp0_w = __msa_copy_s_w((v4i32)in0, in0_idx);  \
296
  SW(tmp0_w, pdst);                                             \
297
  SH(tmp0_h, pdst + stride);                                    \
298
} while (0)
299

300
#define ST6x4_UB(in0, start_in0_idx, in1, start_in1_idx, pdst, stride) do { \
301
  uint8_t* ptmp1 = (uint8_t*)pdst;                                          \
302
  ST6x1_UB(in0, start_in0_idx, in1, start_in1_idx, ptmp1, 4);               \
303
  ptmp1 += stride;                                                          \
304
  ST6x1_UB(in0, start_in0_idx + 1, in1, start_in1_idx + 1, ptmp1, 4);       \
305
  ptmp1 += stride;                                                          \
306
  ST6x1_UB(in0, start_in0_idx + 2, in1, start_in1_idx + 2, ptmp1, 4);       \
307
  ptmp1 += stride;                                                          \
308
  ST6x1_UB(in0, start_in0_idx + 3, in1, start_in1_idx + 3, ptmp1, 4);       \
309
} while (0)
310

311
#define LPF_SIMPLE_FILT(p1_in, p0_in, q0_in, q1_in, mask) do {       \
312
    v16i8 p1_m, p0_m, q0_m, q1_m, filt, filt1, filt2;                \
313
    const v16i8 cnst4b = __msa_ldi_b(4);                             \
314
    const v16i8 cnst3b =  __msa_ldi_b(3);                            \
315
                                                                     \
316
    FLIP_SIGN4(p1_in, p0_in, q0_in, q1_in, p1_m, p0_m, q0_m, q1_m);  \
317
    filt = __msa_subs_s_b(p1_m, q1_m);                               \
318
    FILT_VAL(q0_m, p0_m, mask, filt);                                \
319
    filt1 = __msa_adds_s_b(filt, cnst4b);                            \
320
    filt1 = SRAI_B(filt1, 3);                                        \
321
    filt2 = __msa_adds_s_b(filt, cnst3b);                            \
322
    filt2 = SRAI_B(filt2, 3);                                        \
323
    q0_m = __msa_subs_s_b(q0_m, filt1);                              \
324
    p0_m = __msa_adds_s_b(p0_m, filt2);                              \
325
    q0_in = __msa_xori_b((v16u8)q0_m, 0x80);                         \
326
    p0_in = __msa_xori_b((v16u8)p0_m, 0x80);                         \
327
} while (0)
328

329
#define LPF_SIMPLE_MASK(p1, p0, q0, q1, b_limit, mask) do {    \
330
    v16u8 p1_a_sub_q1, p0_a_sub_q0;                            \
331
                                                               \
332
    p0_a_sub_q0 = __msa_asub_u_b(p0, q0);                      \
333
    p1_a_sub_q1 = __msa_asub_u_b(p1, q1);                      \
334
    p1_a_sub_q1 = (v16u8)__msa_srli_b((v16i8)p1_a_sub_q1, 1);  \
335
    p0_a_sub_q0 = __msa_adds_u_b(p0_a_sub_q0, p0_a_sub_q0);    \
336
    mask = __msa_adds_u_b(p0_a_sub_q0, p1_a_sub_q1);           \
337
    mask = (mask <= b_limit);                                  \
338
} while (0)
339

340
static void VFilter16(uint8_t* src, int stride,
341
                      int b_limit_in, int limit_in, int thresh_in) {
342
  uint8_t* ptemp = src - 4 * stride;
343
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0;
344
  v16u8 mask, hev;
345
  const v16u8 thresh = (v16u8)__msa_fill_b(thresh_in);
346
  const v16u8 limit = (v16u8)__msa_fill_b(limit_in);
347
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
348

349
  LD_UB8(ptemp, stride, p3, p2, p1, p0, q0, q1, q2, q3);
350
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit, b_limit, thresh,
351
               hev, mask);
352
  LPF_MBFILTER(p2, p1, p0, q0, q1, q2, mask, hev);
353
  ptemp = src - 3 * stride;
354
  ST_UB4(p2, p1, p0, q0, ptemp, stride);
355
  ptemp += (4 * stride);
356
  ST_UB2(q1, q2, ptemp, stride);
357
}
358

359
static void HFilter16(uint8_t* src, int stride,
360
                      int b_limit_in, int limit_in, int thresh_in) {
361
  uint8_t* ptmp  = src - 4;
362
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0;
363
  v16u8 mask, hev;
364
  v16u8 row0, row1, row2, row3, row4, row5, row6, row7, row8;
365
  v16u8 row9, row10, row11, row12, row13, row14, row15;
366
  v8i16 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
367
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
368
  const v16u8 limit = (v16u8)__msa_fill_b(limit_in);
369
  const v16u8 thresh = (v16u8)__msa_fill_b(thresh_in);
370

371
  LD_UB8(ptmp, stride, row0, row1, row2, row3, row4, row5, row6, row7);
372
  ptmp += (8 * stride);
373
  LD_UB8(ptmp, stride, row8, row9, row10, row11, row12, row13, row14, row15);
374
  TRANSPOSE16x8_UB_UB(row0, row1, row2, row3, row4, row5, row6, row7,
375
                      row8, row9, row10, row11, row12, row13, row14, row15,
376
                      p3, p2, p1, p0, q0, q1, q2, q3);
377
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit, b_limit, thresh,
378
               hev, mask);
379
  LPF_MBFILTER(p2, p1, p0, q0, q1, q2, mask, hev);
380
  ILVR_B2_SH(p1, p2, q0, p0, tmp0, tmp1);
381
  ILVRL_H2_SH(tmp1, tmp0, tmp3, tmp4);
382
  ILVL_B2_SH(p1, p2, q0, p0, tmp0, tmp1);
383
  ILVRL_H2_SH(tmp1, tmp0, tmp6, tmp7);
384
  ILVRL_B2_SH(q2, q1, tmp2, tmp5);
385
  ptmp = src - 3;
386
  ST6x1_UB(tmp3, 0, tmp2, 0, ptmp, 4);
387
  ptmp += stride;
388
  ST6x1_UB(tmp3, 1, tmp2, 1, ptmp, 4);
389
  ptmp += stride;
390
  ST6x1_UB(tmp3, 2, tmp2, 2, ptmp, 4);
391
  ptmp += stride;
392
  ST6x1_UB(tmp3, 3, tmp2, 3, ptmp, 4);
393
  ptmp += stride;
394
  ST6x1_UB(tmp4, 0, tmp2, 4, ptmp, 4);
395
  ptmp += stride;
396
  ST6x1_UB(tmp4, 1, tmp2, 5, ptmp, 4);
397
  ptmp += stride;
398
  ST6x1_UB(tmp4, 2, tmp2, 6, ptmp, 4);
399
  ptmp += stride;
400
  ST6x1_UB(tmp4, 3, tmp2, 7, ptmp, 4);
401
  ptmp += stride;
402
  ST6x1_UB(tmp6, 0, tmp5, 0, ptmp, 4);
403
  ptmp += stride;
404
  ST6x1_UB(tmp6, 1, tmp5, 1, ptmp, 4);
405
  ptmp += stride;
406
  ST6x1_UB(tmp6, 2, tmp5, 2, ptmp, 4);
407
  ptmp += stride;
408
  ST6x1_UB(tmp6, 3, tmp5, 3, ptmp, 4);
409
  ptmp += stride;
410
  ST6x1_UB(tmp7, 0, tmp5, 4, ptmp, 4);
411
  ptmp += stride;
412
  ST6x1_UB(tmp7, 1, tmp5, 5, ptmp, 4);
413
  ptmp += stride;
414
  ST6x1_UB(tmp7, 2, tmp5, 6, ptmp, 4);
415
  ptmp += stride;
416
  ST6x1_UB(tmp7, 3, tmp5, 7, ptmp, 4);
417
}
418

419
// on three inner edges
420
static void VFilterHorEdge16i(uint8_t* src, int stride,
421
                              int b_limit, int limit, int thresh) {
422
  v16u8 mask, hev;
423
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0;
424
  const v16u8 thresh0 = (v16u8)__msa_fill_b(thresh);
425
  const v16u8 b_limit0 = (v16u8)__msa_fill_b(b_limit);
426
  const v16u8 limit0 = (v16u8)__msa_fill_b(limit);
427

428
  LD_UB8((src - 4 * stride), stride, p3, p2, p1, p0, q0, q1, q2, q3);
429
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit0, b_limit0, thresh0,
430
               hev, mask);
431
  LPF_FILTER4_4W(p1, p0, q0, q1, mask, hev);
432
  ST_UB4(p1, p0, q0, q1, (src - 2 * stride), stride);
433
}
434

435
static void VFilter16i(uint8_t* src_y, int stride,
436
                       int b_limit, int limit, int thresh) {
437
  VFilterHorEdge16i(src_y +  4 * stride, stride, b_limit, limit, thresh);
438
  VFilterHorEdge16i(src_y +  8 * stride, stride, b_limit, limit, thresh);
439
  VFilterHorEdge16i(src_y + 12 * stride, stride, b_limit, limit, thresh);
440
}
441

442
static void HFilterVertEdge16i(uint8_t* src, int stride,
443
                               int b_limit, int limit, int thresh) {
444
  v16u8 mask, hev;
445
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0;
446
  v16u8 row0, row1, row2, row3, row4, row5, row6, row7;
447
  v16u8 row8, row9, row10, row11, row12, row13, row14, row15;
448
  v8i16 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5;
449
  const v16u8 thresh0 = (v16u8)__msa_fill_b(thresh);
450
  const v16u8 b_limit0 = (v16u8)__msa_fill_b(b_limit);
451
  const v16u8 limit0 = (v16u8)__msa_fill_b(limit);
452

453
  LD_UB8(src - 4, stride, row0, row1, row2, row3, row4, row5, row6, row7);
454
  LD_UB8(src - 4 + (8 * stride), stride,
455
         row8, row9, row10, row11, row12, row13, row14, row15);
456
  TRANSPOSE16x8_UB_UB(row0, row1, row2, row3, row4, row5, row6, row7,
457
                      row8, row9, row10, row11, row12, row13, row14, row15,
458
                      p3, p2, p1, p0, q0, q1, q2, q3);
459
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit0, b_limit0, thresh0,
460
               hev, mask);
461
  LPF_FILTER4_4W(p1, p0, q0, q1, mask, hev);
462
  ILVR_B2_SH(p0, p1, q1, q0, tmp0, tmp1);
463
  ILVRL_H2_SH(tmp1, tmp0, tmp2, tmp3);
464
  ILVL_B2_SH(p0, p1, q1, q0, tmp0, tmp1);
465
  ILVRL_H2_SH(tmp1, tmp0, tmp4, tmp5);
466
  src -= 2;
467
  ST4x8_UB(tmp2, tmp3, src, stride);
468
  src += (8 * stride);
469
  ST4x8_UB(tmp4, tmp5, src, stride);
470
}
471

472
static void HFilter16i(uint8_t* src_y, int stride,
473
                       int b_limit, int limit, int thresh) {
474
  HFilterVertEdge16i(src_y +  4, stride, b_limit, limit, thresh);
475
  HFilterVertEdge16i(src_y +  8, stride, b_limit, limit, thresh);
476
  HFilterVertEdge16i(src_y + 12, stride, b_limit, limit, thresh);
477
}
478

479
// 8-pixels wide variants, for chroma filtering
480
static void VFilter8(uint8_t* src_u, uint8_t* src_v, int stride,
481
                     int b_limit_in, int limit_in, int thresh_in) {
482
  uint8_t* ptmp_src_u = src_u - 4 * stride;
483
  uint8_t* ptmp_src_v = src_v - 4 * stride;
484
  uint64_t p2_d, p1_d, p0_d, q0_d, q1_d, q2_d;
485
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0, mask, hev;
486
  v16u8 p3_u, p2_u, p1_u, p0_u, q3_u, q2_u, q1_u, q0_u;
487
  v16u8 p3_v, p2_v, p1_v, p0_v, q3_v, q2_v, q1_v, q0_v;
488
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
489
  const v16u8 limit = (v16u8)__msa_fill_b(limit_in);
490
  const v16u8 thresh = (v16u8)__msa_fill_b(thresh_in);
491

492
  LD_UB8(ptmp_src_u, stride, p3_u, p2_u, p1_u, p0_u, q0_u, q1_u, q2_u, q3_u);
493
  LD_UB8(ptmp_src_v, stride, p3_v, p2_v, p1_v, p0_v, q0_v, q1_v, q2_v, q3_v);
494
  ILVR_D4_UB(p3_v, p3_u, p2_v, p2_u, p1_v, p1_u, p0_v, p0_u, p3, p2, p1, p0);
495
  ILVR_D4_UB(q0_v, q0_u, q1_v, q1_u, q2_v, q2_u, q3_v, q3_u, q0, q1, q2, q3);
496
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit, b_limit, thresh,
497
               hev, mask);
498
  LPF_MBFILTER(p2, p1, p0, q0, q1, q2, mask, hev);
499
  p2_d = __msa_copy_s_d((v2i64)p2, 0);
500
  p1_d = __msa_copy_s_d((v2i64)p1, 0);
501
  p0_d = __msa_copy_s_d((v2i64)p0, 0);
502
  q0_d = __msa_copy_s_d((v2i64)q0, 0);
503
  q1_d = __msa_copy_s_d((v2i64)q1, 0);
504
  q2_d = __msa_copy_s_d((v2i64)q2, 0);
505
  ptmp_src_u += stride;
506
  SD4(p2_d, p1_d, p0_d, q0_d, ptmp_src_u, stride);
507
  ptmp_src_u += (4 * stride);
508
  SD(q1_d, ptmp_src_u);
509
  ptmp_src_u += stride;
510
  SD(q2_d, ptmp_src_u);
511
  p2_d = __msa_copy_s_d((v2i64)p2, 1);
512
  p1_d = __msa_copy_s_d((v2i64)p1, 1);
513
  p0_d = __msa_copy_s_d((v2i64)p0, 1);
514
  q0_d = __msa_copy_s_d((v2i64)q0, 1);
515
  q1_d = __msa_copy_s_d((v2i64)q1, 1);
516
  q2_d = __msa_copy_s_d((v2i64)q2, 1);
517
  ptmp_src_v += stride;
518
  SD4(p2_d, p1_d, p0_d, q0_d, ptmp_src_v, stride);
519
  ptmp_src_v += (4 * stride);
520
  SD(q1_d, ptmp_src_v);
521
  ptmp_src_v += stride;
522
  SD(q2_d, ptmp_src_v);
523
}
524

525
static void HFilter8(uint8_t* src_u, uint8_t* src_v, int stride,
526
                     int b_limit_in, int limit_in, int thresh_in) {
527
  uint8_t* ptmp_src_u = src_u - 4;
528
  uint8_t* ptmp_src_v = src_v - 4;
529
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0, mask, hev;
530
  v16u8 row0, row1, row2, row3, row4, row5, row6, row7, row8;
531
  v16u8 row9, row10, row11, row12, row13, row14, row15;
532
  v8i16 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5, tmp6, tmp7;
533
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
534
  const v16u8 limit = (v16u8)__msa_fill_b(limit_in);
535
  const v16u8 thresh = (v16u8)__msa_fill_b(thresh_in);
536

537
  LD_UB8(ptmp_src_u, stride, row0, row1, row2, row3, row4, row5, row6, row7);
538
  LD_UB8(ptmp_src_v, stride,
539
         row8, row9, row10, row11, row12, row13, row14, row15);
540
  TRANSPOSE16x8_UB_UB(row0, row1, row2, row3, row4, row5, row6, row7,
541
                      row8, row9, row10, row11, row12, row13, row14, row15,
542
                      p3, p2, p1, p0, q0, q1, q2, q3);
543
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit, b_limit, thresh,
544
               hev, mask);
545
  LPF_MBFILTER(p2, p1, p0, q0, q1, q2, mask, hev);
546
  ILVR_B2_SH(p1, p2, q0, p0, tmp0, tmp1);
547
  ILVRL_H2_SH(tmp1, tmp0, tmp3, tmp4);
548
  ILVL_B2_SH(p1, p2, q0, p0, tmp0, tmp1);
549
  ILVRL_H2_SH(tmp1, tmp0, tmp6, tmp7);
550
  ILVRL_B2_SH(q2, q1, tmp2, tmp5);
551
  ptmp_src_u += 1;
552
  ST6x4_UB(tmp3, 0, tmp2, 0, ptmp_src_u, stride);
553
  ptmp_src_u += 4 * stride;
554
  ST6x4_UB(tmp4, 0, tmp2, 4, ptmp_src_u, stride);
555
  ptmp_src_v += 1;
556
  ST6x4_UB(tmp6, 0, tmp5, 0, ptmp_src_v, stride);
557
  ptmp_src_v += 4 * stride;
558
  ST6x4_UB(tmp7, 0, tmp5, 4, ptmp_src_v, stride);
559
}
560

561
static void VFilter8i(uint8_t* src_u, uint8_t* src_v, int stride,
562
                      int b_limit_in, int limit_in, int thresh_in) {
563
  uint64_t p1_d, p0_d, q0_d, q1_d;
564
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0, mask, hev;
565
  v16u8 p3_u, p2_u, p1_u, p0_u, q3_u, q2_u, q1_u, q0_u;
566
  v16u8 p3_v, p2_v, p1_v, p0_v, q3_v, q2_v, q1_v, q0_v;
567
  const v16u8 thresh = (v16u8)__msa_fill_b(thresh_in);
568
  const v16u8 limit = (v16u8)__msa_fill_b(limit_in);
569
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
570

571
  LD_UB8(src_u, stride, p3_u, p2_u, p1_u, p0_u, q0_u, q1_u, q2_u, q3_u);
572
  src_u += (5 * stride);
573
  LD_UB8(src_v, stride, p3_v, p2_v, p1_v, p0_v, q0_v, q1_v, q2_v, q3_v);
574
  src_v += (5 * stride);
575
  ILVR_D4_UB(p3_v, p3_u, p2_v, p2_u, p1_v, p1_u, p0_v, p0_u, p3, p2, p1, p0);
576
  ILVR_D4_UB(q0_v, q0_u, q1_v, q1_u, q2_v, q2_u, q3_v, q3_u, q0, q1, q2, q3);
577
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit, b_limit, thresh,
578
               hev, mask);
579
  LPF_FILTER4_4W(p1, p0, q0, q1, mask, hev);
580
  p1_d = __msa_copy_s_d((v2i64)p1, 0);
581
  p0_d = __msa_copy_s_d((v2i64)p0, 0);
582
  q0_d = __msa_copy_s_d((v2i64)q0, 0);
583
  q1_d = __msa_copy_s_d((v2i64)q1, 0);
584
  SD4(q1_d, q0_d, p0_d, p1_d, src_u, -stride);
585
  p1_d = __msa_copy_s_d((v2i64)p1, 1);
586
  p0_d = __msa_copy_s_d((v2i64)p0, 1);
587
  q0_d = __msa_copy_s_d((v2i64)q0, 1);
588
  q1_d = __msa_copy_s_d((v2i64)q1, 1);
589
  SD4(q1_d, q0_d, p0_d, p1_d, src_v, -stride);
590
}
591

592
static void HFilter8i(uint8_t* src_u, uint8_t* src_v, int stride,
593
                      int b_limit_in, int limit_in, int thresh_in) {
594
  v16u8 p3, p2, p1, p0, q3, q2, q1, q0, mask, hev;
595
  v16u8 row0, row1, row2, row3, row4, row5, row6, row7, row8;
596
  v16u8 row9, row10, row11, row12, row13, row14, row15;
597
  v4i32 tmp0, tmp1, tmp2, tmp3, tmp4, tmp5;
598
  const v16u8 thresh = (v16u8)__msa_fill_b(thresh_in);
599
  const v16u8 limit = (v16u8)__msa_fill_b(limit_in);
600
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
601

602
  LD_UB8(src_u, stride, row0, row1, row2, row3, row4, row5, row6, row7);
603
  LD_UB8(src_v, stride,
604
         row8, row9, row10, row11, row12, row13, row14, row15);
605
  TRANSPOSE16x8_UB_UB(row0, row1, row2, row3, row4, row5, row6, row7,
606
                      row8, row9, row10, row11, row12, row13, row14, row15,
607
                      p3, p2, p1, p0, q0, q1, q2, q3);
608
  LPF_MASK_HEV(p3, p2, p1, p0, q0, q1, q2, q3, limit, b_limit, thresh,
609
               hev, mask);
610
  LPF_FILTER4_4W(p1, p0, q0, q1, mask, hev);
611
  ILVR_B2_SW(p0, p1, q1, q0, tmp0, tmp1);
612
  ILVRL_H2_SW(tmp1, tmp0, tmp2, tmp3);
613
  ILVL_B2_SW(p0, p1, q1, q0, tmp0, tmp1);
614
  ILVRL_H2_SW(tmp1, tmp0, tmp4, tmp5);
615
  src_u += 2;
616
  ST4x4_UB(tmp2, tmp2, 0, 1, 2, 3, src_u, stride);
617
  src_u += 4 * stride;
618
  ST4x4_UB(tmp3, tmp3, 0, 1, 2, 3, src_u, stride);
619
  src_v += 2;
620
  ST4x4_UB(tmp4, tmp4, 0, 1, 2, 3, src_v, stride);
621
  src_v += 4 * stride;
622
  ST4x4_UB(tmp5, tmp5, 0, 1, 2, 3, src_v, stride);
623
}
624

625
static void SimpleVFilter16(uint8_t* src, int stride, int b_limit_in) {
626
  v16u8 p1, p0, q1, q0, mask;
627
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
628

629
  LD_UB4(src - 2 * stride, stride, p1, p0, q0, q1);
630
  LPF_SIMPLE_MASK(p1, p0, q0, q1, b_limit, mask);
631
  LPF_SIMPLE_FILT(p1, p0, q0, q1, mask);
632
  ST_UB2(p0, q0, src - stride, stride);
633
}
634

635
static void SimpleHFilter16(uint8_t* src, int stride, int b_limit_in) {
636
  v16u8 p1, p0, q1, q0, mask, row0, row1, row2, row3, row4, row5, row6, row7;
637
  v16u8 row8, row9, row10, row11, row12, row13, row14, row15;
638
  v8i16 tmp0, tmp1;
639
  const v16u8 b_limit = (v16u8)__msa_fill_b(b_limit_in);
640
  uint8_t* ptemp_src = src - 2;
641

642
  LD_UB8(ptemp_src, stride, row0, row1, row2, row3, row4, row5, row6, row7);
643
  LD_UB8(ptemp_src + 8 * stride, stride,
644
         row8, row9, row10, row11, row12, row13, row14, row15);
645
  TRANSPOSE16x4_UB_UB(row0, row1, row2, row3, row4, row5, row6, row7,
646
                      row8, row9, row10, row11, row12, row13, row14, row15,
647
                      p1, p0, q0, q1);
648
  LPF_SIMPLE_MASK(p1, p0, q0, q1, b_limit, mask);
649
  LPF_SIMPLE_FILT(p1, p0, q0, q1, mask);
650
  ILVRL_B2_SH(q0, p0, tmp1, tmp0);
651
  ptemp_src += 1;
652
  ST2x4_UB(tmp1, 0, ptemp_src, stride);
653
  ptemp_src += 4 * stride;
654
  ST2x4_UB(tmp1, 4, ptemp_src, stride);
655
  ptemp_src += 4 * stride;
656
  ST2x4_UB(tmp0, 0, ptemp_src, stride);
657
  ptemp_src += 4 * stride;
658
  ST2x4_UB(tmp0, 4, ptemp_src, stride);
659
  ptemp_src += 4 * stride;
660
}
661

662
static void SimpleVFilter16i(uint8_t* src_y, int stride, int b_limit_in) {
663
  SimpleVFilter16(src_y +  4 * stride, stride, b_limit_in);
664
  SimpleVFilter16(src_y +  8 * stride, stride, b_limit_in);
665
  SimpleVFilter16(src_y + 12 * stride, stride, b_limit_in);
666
}
667

668
static void SimpleHFilter16i(uint8_t* src_y, int stride, int b_limit_in) {
669
  SimpleHFilter16(src_y +  4, stride, b_limit_in);
670
  SimpleHFilter16(src_y +  8, stride, b_limit_in);
671
  SimpleHFilter16(src_y + 12, stride, b_limit_in);
672
}
673

674
//------------------------------------------------------------------------------
675
// Intra predictions
676
//------------------------------------------------------------------------------
677

678
// 4x4
679

680
static void DC4(uint8_t* dst) {   // DC
681
  uint32_t dc = 4;
682
  int i;
683
  for (i = 0; i < 4; ++i) dc += dst[i - BPS] + dst[-1 + i * BPS];
684
  dc >>= 3;
685
  dc = dc | (dc << 8) | (dc << 16) | (dc << 24);
686
  SW4(dc, dc, dc, dc, dst, BPS);
687
}
688

689
static void TM4(uint8_t* dst) {
690
  const uint8_t* const ptemp = dst - BPS - 1;
691
  v8i16 T, d, r0, r1, r2, r3;
692
  const v16i8 zero = { 0 };
693
  const v8i16 TL = (v8i16)__msa_fill_h(ptemp[0 * BPS]);
694
  const v8i16 L0 = (v8i16)__msa_fill_h(ptemp[1 * BPS]);
695
  const v8i16 L1 = (v8i16)__msa_fill_h(ptemp[2 * BPS]);
696
  const v8i16 L2 = (v8i16)__msa_fill_h(ptemp[3 * BPS]);
697
  const v8i16 L3 = (v8i16)__msa_fill_h(ptemp[4 * BPS]);
698
  const v16u8 T1 = LD_UB(ptemp + 1);
699

700
  T  = (v8i16)__msa_ilvr_b(zero, (v16i8)T1);
701
  d = T - TL;
702
  ADD4(d, L0, d, L1, d, L2, d, L3, r0, r1, r2, r3);
703
  CLIP_SH4_0_255(r0, r1, r2, r3);
704
  PCKEV_ST4x4_UB(r0, r1, r2, r3, dst, BPS);
705
}
706

707
static void VE4(uint8_t* dst) {    // vertical
708
  const uint8_t* const ptop = dst - BPS - 1;
709
  const uint32_t val0 = LW(ptop + 0);
710
  const uint32_t val1 = LW(ptop + 4);
711
  uint32_t out;
712
  v16u8 A = { 0 }, B, C, AC, B2, R;
713

714
  INSERT_W2_UB(val0, val1, A);
715
  B = SLDI_UB(A, A, 1);
716
  C = SLDI_UB(A, A, 2);
717
  AC = __msa_ave_u_b(A, C);
718
  B2 = __msa_ave_u_b(B, B);
719
  R = __msa_aver_u_b(AC, B2);
720
  out = __msa_copy_s_w((v4i32)R, 0);
721
  SW4(out, out, out, out, dst, BPS);
722
}
723

724
static void RD4(uint8_t* dst) {   // Down-right
725
  const uint8_t* const ptop = dst - 1 - BPS;
726
  uint32_t val0 = LW(ptop + 0);
727
  uint32_t val1 = LW(ptop + 4);
728
  uint32_t val2, val3;
729
  v16u8 A, B, C, AC, B2, R, A1 = { 0 };
730

731
  INSERT_W2_UB(val0, val1, A1);
732
  A = SLDI_UB(A1, A1, 12);
733
  A = (v16u8)__msa_insert_b((v16i8)A, 3, ptop[1 * BPS]);
734
  A = (v16u8)__msa_insert_b((v16i8)A, 2, ptop[2 * BPS]);
735
  A = (v16u8)__msa_insert_b((v16i8)A, 1, ptop[3 * BPS]);
736
  A = (v16u8)__msa_insert_b((v16i8)A, 0, ptop[4 * BPS]);
737
  B = SLDI_UB(A, A, 1);
738
  C = SLDI_UB(A, A, 2);
739
  AC = __msa_ave_u_b(A, C);
740
  B2 = __msa_ave_u_b(B, B);
741
  R = __msa_aver_u_b(AC, B2);
742
  val3 = __msa_copy_s_w((v4i32)R, 0);
743
  R = SLDI_UB(R, R, 1);
744
  val2 = __msa_copy_s_w((v4i32)R, 0);
745
  R = SLDI_UB(R, R, 1);
746
  val1 = __msa_copy_s_w((v4i32)R, 0);
747
  R = SLDI_UB(R, R, 1);
748
  val0 = __msa_copy_s_w((v4i32)R, 0);
749
  SW4(val0, val1, val2, val3, dst, BPS);
750
}
751

752
static void LD4(uint8_t* dst) {   // Down-Left
753
  const uint8_t* const ptop = dst - BPS;
754
  uint32_t val0 = LW(ptop + 0);
755
  uint32_t val1 = LW(ptop + 4);
756
  uint32_t val2, val3;
757
  v16u8 A = { 0 }, B, C, AC, B2, R;
758

759
  INSERT_W2_UB(val0, val1, A);
760
  B = SLDI_UB(A, A, 1);
761
  C = SLDI_UB(A, A, 2);
762
  C = (v16u8)__msa_insert_b((v16i8)C, 6, ptop[7]);
763
  AC = __msa_ave_u_b(A, C);
764
  B2 = __msa_ave_u_b(B, B);
765
  R = __msa_aver_u_b(AC, B2);
766
  val0 = __msa_copy_s_w((v4i32)R, 0);
767
  R = SLDI_UB(R, R, 1);
768
  val1 = __msa_copy_s_w((v4i32)R, 0);
769
  R = SLDI_UB(R, R, 1);
770
  val2 = __msa_copy_s_w((v4i32)R, 0);
771
  R = SLDI_UB(R, R, 1);
772
  val3 = __msa_copy_s_w((v4i32)R, 0);
773
  SW4(val0, val1, val2, val3, dst, BPS);
774
}
775

776
// 16x16
777

778
static void DC16(uint8_t* dst) {   // DC
779
  uint32_t dc = 16;
780
  int i;
781
  const v16u8 rtop = LD_UB(dst - BPS);
782
  const v8u16 dctop = __msa_hadd_u_h(rtop, rtop);
783
  v16u8 out;
784

785
  for (i = 0; i < 16; ++i) {
786
    dc += dst[-1 + i * BPS];
787
  }
788
  dc += HADD_UH_U32(dctop);
789
  out = (v16u8)__msa_fill_b(dc >> 5);
790
  ST_UB8(out, out, out, out, out, out, out, out, dst, BPS);
791
  ST_UB8(out, out, out, out, out, out, out, out, dst + 8 * BPS, BPS);
792
}
793

794
static void TM16(uint8_t* dst) {
795
  int j;
796
  v8i16 d1, d2;
797
  const v16i8 zero = { 0 };
798
  const v8i16 TL = (v8i16)__msa_fill_h(dst[-1 - BPS]);
799
  const v16i8 T = LD_SB(dst - BPS);
800

801
  ILVRL_B2_SH(zero, T, d1, d2);
802
  SUB2(d1, TL, d2, TL, d1, d2);
803
  for (j = 0; j < 16; j += 4) {
804
    v16i8 t0, t1, t2, t3;
805
    v8i16 r0, r1, r2, r3, r4, r5, r6, r7;
806
    const v8i16 L0 = (v8i16)__msa_fill_h(dst[-1 + 0 * BPS]);
807
    const v8i16 L1 = (v8i16)__msa_fill_h(dst[-1 + 1 * BPS]);
808
    const v8i16 L2 = (v8i16)__msa_fill_h(dst[-1 + 2 * BPS]);
809
    const v8i16 L3 = (v8i16)__msa_fill_h(dst[-1 + 3 * BPS]);
810
    ADD4(d1, L0, d1, L1, d1, L2, d1, L3, r0, r1, r2, r3);
811
    ADD4(d2, L0, d2, L1, d2, L2, d2, L3, r4, r5, r6, r7);
812
    CLIP_SH4_0_255(r0, r1, r2, r3);
813
    CLIP_SH4_0_255(r4, r5, r6, r7);
814
    PCKEV_B4_SB(r4, r0, r5, r1, r6, r2, r7, r3, t0, t1, t2, t3);
815
    ST_SB4(t0, t1, t2, t3, dst, BPS);
816
    dst += 4 * BPS;
817
  }
818
}
819

820
static void VE16(uint8_t* dst) {   // vertical
821
  const v16u8 rtop = LD_UB(dst - BPS);
822
  ST_UB8(rtop, rtop, rtop, rtop, rtop, rtop, rtop, rtop, dst, BPS);
823
  ST_UB8(rtop, rtop, rtop, rtop, rtop, rtop, rtop, rtop, dst + 8 * BPS, BPS);
824
}
825

826
static void HE16(uint8_t* dst) {   // horizontal
827
  int j;
828
  for (j = 16; j > 0; j -= 4) {
829
    const v16u8 L0 = (v16u8)__msa_fill_b(dst[-1 + 0 * BPS]);
830
    const v16u8 L1 = (v16u8)__msa_fill_b(dst[-1 + 1 * BPS]);
831
    const v16u8 L2 = (v16u8)__msa_fill_b(dst[-1 + 2 * BPS]);
832
    const v16u8 L3 = (v16u8)__msa_fill_b(dst[-1 + 3 * BPS]);
833
    ST_UB4(L0, L1, L2, L3, dst, BPS);
834
    dst += 4 * BPS;
835
  }
836
}
837

838
static void DC16NoTop(uint8_t* dst) {   // DC with top samples not available
839
  int j;
840
  uint32_t dc = 8;
841
  v16u8 out;
842

843
  for (j = 0; j < 16; ++j) {
844
    dc += dst[-1 + j * BPS];
845
  }
846
  out = (v16u8)__msa_fill_b(dc >> 4);
847
  ST_UB8(out, out, out, out, out, out, out, out, dst, BPS);
848
  ST_UB8(out, out, out, out, out, out, out, out, dst + 8 * BPS, BPS);
849
}
850

851
static void DC16NoLeft(uint8_t* dst) {   // DC with left samples not available
852
  uint32_t dc = 8;
853
  const v16u8 rtop = LD_UB(dst - BPS);
854
  const v8u16 dctop = __msa_hadd_u_h(rtop, rtop);
855
  v16u8 out;
856

857
  dc += HADD_UH_U32(dctop);
858
  out = (v16u8)__msa_fill_b(dc >> 4);
859
  ST_UB8(out, out, out, out, out, out, out, out, dst, BPS);
860
  ST_UB8(out, out, out, out, out, out, out, out, dst + 8 * BPS, BPS);
861
}
862

863
static void DC16NoTopLeft(uint8_t* dst) {   // DC with nothing
864
  const v16u8 out = (v16u8)__msa_fill_b(0x80);
865
  ST_UB8(out, out, out, out, out, out, out, out, dst, BPS);
866
  ST_UB8(out, out, out, out, out, out, out, out, dst + 8 * BPS, BPS);
867
}
868

869
// Chroma
870

871
#define STORE8x8(out, dst) do {                 \
872
  SD4(out, out, out, out, dst + 0 * BPS, BPS);  \
873
  SD4(out, out, out, out, dst + 4 * BPS, BPS);  \
874
} while (0)
875

876
static void DC8uv(uint8_t* dst) {   // DC
877
  uint32_t dc = 8;
878
  int i;
879
  uint64_t out;
880
  const v16u8 rtop = LD_UB(dst - BPS);
881
  const v8u16 temp0 = __msa_hadd_u_h(rtop, rtop);
882
  const v4u32 temp1 = __msa_hadd_u_w(temp0, temp0);
883
  const v2u64 temp2 = __msa_hadd_u_d(temp1, temp1);
884
  v16u8 dctemp;
885

886
  for (i = 0; i < 8; ++i) {
887
    dc += dst[-1 + i * BPS];
888
  }
889
  dc += __msa_copy_s_w((v4i32)temp2, 0);
890
  dctemp = (v16u8)__msa_fill_b(dc >> 4);
891
  out = __msa_copy_s_d((v2i64)dctemp, 0);
892
  STORE8x8(out, dst);
893
}
894

895
static void TM8uv(uint8_t* dst) {
896
  int j;
897
  const v16i8 T1 = LD_SB(dst - BPS);
898
  const v16i8 zero = { 0 };
899
  const v8i16 T  = (v8i16)__msa_ilvr_b(zero, T1);
900
  const v8i16 TL = (v8i16)__msa_fill_h(dst[-1 - BPS]);
901
  const v8i16 d = T - TL;
902

903
  for (j = 0; j < 8; j += 4) {
904
    v16i8 t0, t1;
905
    v8i16 r0 = (v8i16)__msa_fill_h(dst[-1 + 0 * BPS]);
906
    v8i16 r1 = (v8i16)__msa_fill_h(dst[-1 + 1 * BPS]);
907
    v8i16 r2 = (v8i16)__msa_fill_h(dst[-1 + 2 * BPS]);
908
    v8i16 r3 = (v8i16)__msa_fill_h(dst[-1 + 3 * BPS]);
909
    ADD4(d, r0, d, r1, d, r2, d, r3, r0, r1, r2, r3);
910
    CLIP_SH4_0_255(r0, r1, r2, r3);
911
    PCKEV_B2_SB(r1, r0, r3, r2, t0, t1);
912
    ST4x4_UB(t0, t1, 0, 2, 0, 2, dst, BPS);
913
    ST4x4_UB(t0, t1, 1, 3, 1, 3, dst + 4, BPS);
914
    dst += 4 * BPS;
915
  }
916
}
917

918
static void VE8uv(uint8_t* dst) {   // vertical
919
  const v16u8 rtop = LD_UB(dst - BPS);
920
  const uint64_t out = __msa_copy_s_d((v2i64)rtop, 0);
921
  STORE8x8(out, dst);
922
}
923

924
static void HE8uv(uint8_t* dst) {   // horizontal
925
  int j;
926
  for (j = 0; j < 8; j += 4) {
927
    const v16u8 L0 = (v16u8)__msa_fill_b(dst[-1 + 0 * BPS]);
928
    const v16u8 L1 = (v16u8)__msa_fill_b(dst[-1 + 1 * BPS]);
929
    const v16u8 L2 = (v16u8)__msa_fill_b(dst[-1 + 2 * BPS]);
930
    const v16u8 L3 = (v16u8)__msa_fill_b(dst[-1 + 3 * BPS]);
931
    const uint64_t out0 = __msa_copy_s_d((v2i64)L0, 0);
932
    const uint64_t out1 = __msa_copy_s_d((v2i64)L1, 0);
933
    const uint64_t out2 = __msa_copy_s_d((v2i64)L2, 0);
934
    const uint64_t out3 = __msa_copy_s_d((v2i64)L3, 0);
935
    SD4(out0, out1, out2, out3, dst, BPS);
936
    dst += 4 * BPS;
937
  }
938
}
939

940
static void DC8uvNoLeft(uint8_t* dst) {   // DC with no left samples
941
  const uint32_t dc = 4;
942
  const v16u8 rtop = LD_UB(dst - BPS);
943
  const v8u16 temp0 = __msa_hadd_u_h(rtop, rtop);
944
  const v4u32 temp1 = __msa_hadd_u_w(temp0, temp0);
945
  const v2u64 temp2 = __msa_hadd_u_d(temp1, temp1);
946
  const uint32_t sum_m = __msa_copy_s_w((v4i32)temp2, 0);
947
  const v16u8 dcval = (v16u8)__msa_fill_b((dc + sum_m) >> 3);
948
  const uint64_t out = __msa_copy_s_d((v2i64)dcval, 0);
949
  STORE8x8(out, dst);
950
}
951

952
static void DC8uvNoTop(uint8_t* dst) {   // DC with no top samples
953
  uint32_t dc = 4;
954
  int i;
955
  uint64_t out;
956
  v16u8 dctemp;
957

958
  for (i = 0; i < 8; ++i) {
959
    dc += dst[-1 + i * BPS];
960
  }
961
  dctemp = (v16u8)__msa_fill_b(dc >> 3);
962
  out = __msa_copy_s_d((v2i64)dctemp, 0);
963
  STORE8x8(out, dst);
964
}
965

966
static void DC8uvNoTopLeft(uint8_t* dst) {   // DC with nothing
967
  const uint64_t out = 0x8080808080808080ULL;
968
  STORE8x8(out, dst);
969
}
970

971
//------------------------------------------------------------------------------
972
// Entry point
973

974
extern void VP8DspInitMSA(void);
975

976
WEBP_TSAN_IGNORE_FUNCTION void VP8DspInitMSA(void) {
977
  VP8TransformWHT = TransformWHT;
978
  VP8Transform = TransformTwo;
979
  VP8TransformDC = TransformDC;
980
  VP8TransformAC3 = TransformAC3;
981

982
  VP8VFilter16  = VFilter16;
983
  VP8HFilter16  = HFilter16;
984
  VP8VFilter16i = VFilter16i;
985
  VP8HFilter16i = HFilter16i;
986
  VP8VFilter8  = VFilter8;
987
  VP8HFilter8  = HFilter8;
988
  VP8VFilter8i = VFilter8i;
989
  VP8HFilter8i = HFilter8i;
990
  VP8SimpleVFilter16  = SimpleVFilter16;
991
  VP8SimpleHFilter16  = SimpleHFilter16;
992
  VP8SimpleVFilter16i = SimpleVFilter16i;
993
  VP8SimpleHFilter16i = SimpleHFilter16i;
994

995
  VP8PredLuma4[0] = DC4;
996
  VP8PredLuma4[1] = TM4;
997
  VP8PredLuma4[2] = VE4;
998
  VP8PredLuma4[4] = RD4;
999
  VP8PredLuma4[6] = LD4;
1000
  VP8PredLuma16[0] = DC16;
1001
  VP8PredLuma16[1] = TM16;
1002
  VP8PredLuma16[2] = VE16;
1003
  VP8PredLuma16[3] = HE16;
1004
  VP8PredLuma16[4] = DC16NoTop;
1005
  VP8PredLuma16[5] = DC16NoLeft;
1006
  VP8PredLuma16[6] = DC16NoTopLeft;
1007
  VP8PredChroma8[0] = DC8uv;
1008
  VP8PredChroma8[1] = TM8uv;
1009
  VP8PredChroma8[2] = VE8uv;
1010
  VP8PredChroma8[3] = HE8uv;
1011
  VP8PredChroma8[4] = DC8uvNoTop;
1012
  VP8PredChroma8[5] = DC8uvNoLeft;
1013
  VP8PredChroma8[6] = DC8uvNoTopLeft;
1014
}
1015

1016
#else  // !WEBP_USE_MSA
1017

1018
WEBP_DSP_INIT_STUB(VP8DspInitMSA)
1019

1020
#endif  // WEBP_USE_MSA
1021

1022
Product

Resources

Company
