scale_neon64.cc 30.9 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
/*
 *  Copyright 2014 The LibYuv 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.
 */

11
#include "libyuv/scale.h"
12
#include "libyuv/row.h"
13
#include "libyuv/scale_row.h"
14 15 16 17 18 19

#ifdef __cplusplus
namespace libyuv {
extern "C" {
#endif

20
// This module is for GCC Neon armv8 64 bit.
21
#if !defined(LIBYUV_DISABLE_NEON) && defined(__aarch64__)
22

23 24 25 26 27 28
#ifdef HAS_SCALEROWDOWN2_NEON
// Read 32x1 throw away even pixels, and write 16x1.
void ScaleRowDown2_NEON(const uint8* src_ptr, ptrdiff_t src_stride,
                        uint8* dst, int dst_width) {
  asm volatile (
  "1:                                          \n"
29
    // load even pixels into v0, odd into v1
30
    MEMACCESS(0)
31
    "ld2        {v0.16b,v1.16b}, [%0], #32    \n"
32 33
    "subs       %2, %2, #16                    \n"  // 16 processed per loop
    MEMACCESS(1)
34
    "st1        {v1.16b}, [%1], #16            \n"  // store odd pixels
35
    "b.gt       1b                             \n"
36 37 38 39
  : "+r"(src_ptr),          // %0
    "+r"(dst),              // %1
    "+r"(dst_width)         // %2
  :
40
  : "v0", "v1"              // Clobber List
41 42 43 44 45 46 47 48 49 50
  );
}
#endif //HAS_SCALEROWDOWN2_NEON

#ifdef HAS_SCALEROWDOWN2_NEON
// Read 32x2 average down and write 16x1.
void ScaleRowDown2Box_NEON(const uint8* src_ptr, ptrdiff_t src_stride,
                           uint8* dst, int dst_width) {
  asm volatile (
    // change the stride to row 2 pointer
51
    "add        %1, %1, %0                     \n"
52 53
  "1:                                          \n"
    MEMACCESS(0)
54
    "ld1        {v0.16b,v1.16b}, [%0], #32    \n"  // load row 1 and post inc
55
    MEMACCESS(1)
56
    "ld1        {v2.16b, v3.16b}, [%1], #32    \n"  // load row 2 and post inc
57
    "subs       %3, %3, #16                    \n"  // 16 processed per loop
58 59 60 61 62 63
    "uaddlp     v0.8h, v0.16b                  \n"  // row 1 add adjacent
    "uaddlp     v1.8h, v1.16b                  \n"
    "uadalp     v0.8h, v2.16b                  \n"  // row 2 add adjacent + row1
    "uadalp     v1.8h, v3.16b                  \n"
    "rshrn      v0.8b, v0.8h, #2               \n"  // downshift, round and pack
    "rshrn2     v0.16b, v1.8h, #2              \n"
64
    MEMACCESS(2)
65
    "st1        {v0.16b}, [%2], #16            \n"
66
    "b.gt       1b                             \n"
67 68 69 70 71
  : "+r"(src_ptr),          // %0
    "+r"(src_stride),       // %1
    "+r"(dst),              // %2
    "+r"(dst_width)         // %3
  :
72
  : "v0", "v1", "v2", "v3"     // Clobber List
73 74 75 76 77 78 79 80 81 82
  );
}
#endif //HAS_SCALEROWDOWN2_NEON

#ifdef HAS_SCALEROWDOWN4_NEON
void ScaleRowDown4_NEON(const uint8* src_ptr, ptrdiff_t src_stride,
                        uint8* dst_ptr, int dst_width) {
  asm volatile (
  "1:                                          \n"
    MEMACCESS(0)
83
    "ld4     {v0.8b,v1.8b,v2.8b,v3.8b}, [%0], #32          \n"  // src line 0
84
    "subs       %2, %2, #8                     \n"  // 8 processed per loop
85
    MEMACCESS(1)
86
    "st1     {v2.8b}, [%1], #8                 \n"
87
    "b.gt       1b                             \n"
88 89 90 91
  : "+r"(src_ptr),          // %0
    "+r"(dst_ptr),          // %1
    "+r"(dst_width)         // %2
  :
92
  : "v0", "v1", "v2", "v3", "memory", "cc"
93 94 95 96 97 98 99 100 101 102 103 104 105
  );
}
#endif //HAS_SCALEROWDOWN4_NEON

#ifdef HAS_SCALEROWDOWN4_NEON
void ScaleRowDown4Box_NEON(const uint8* src_ptr, ptrdiff_t src_stride,
                           uint8* dst_ptr, int dst_width) {
  const uint8* src_ptr1 = src_ptr + src_stride;
  const uint8* src_ptr2 = src_ptr + src_stride * 2;
  const uint8* src_ptr3 = src_ptr + src_stride * 3;
asm volatile (
  "1:                                          \n"
    MEMACCESS(0)
106
    "ld1     {v0.16b}, [%0], #16               \n"   // load up 16x4
107
    MEMACCESS(3)
108
    "ld1     {v1.16b}, [%2], #16               \n"
109
    MEMACCESS(4)
110
    "ld1     {v2.16b}, [%3], #16               \n"
111
    MEMACCESS(5)
112 113
    "ld1     {v3.16b}, [%4], #16               \n"
    "subs    %5, %5, #4                        \n"
114 115 116 117 118 119
    "uaddlp  v0.8h, v0.16b                     \n"
    "uadalp  v0.8h, v1.16b                     \n"
    "uadalp  v0.8h, v2.16b                     \n"
    "uadalp  v0.8h, v3.16b                     \n"
    "addp    v0.8h, v0.8h, v0.8h               \n"
    "rshrn   v0.8b, v0.8h, #4                  \n"   // divide by 16 w/rounding
120
    MEMACCESS(1)
121
    "st1    {v0.s}[0], [%1], #4                \n"
122
    "b.gt       1b                             \n"
123 124
  : "+r"(src_ptr),   // %0
    "+r"(dst_ptr),   // %1
125 126 127 128
    "+r"(src_ptr1),  // %2
    "+r"(src_ptr2),  // %3
    "+r"(src_ptr3),  // %4
    "+r"(dst_width)  // %5
129
  :
130
  : "v0", "v1", "v2", "v3", "memory", "cc"
131 132 133 134 135 136 137 138 139 140 141 142
  );
}
#endif //HAS_SCALEROWDOWN4_NEON

#ifdef HAS_SCALEROWDOWN34_NEON
// Down scale from 4 to 3 pixels. Use the neon multilane read/write
// to load up the every 4th pixel into a 4 different registers.
// Point samples 32 pixels to 24 pixels.
void ScaleRowDown34_NEON(const uint8* src_ptr,
                         ptrdiff_t src_stride,
                         uint8* dst_ptr, int dst_width) {
  asm volatile (
143
  "1:                                                  \n"
144
    MEMACCESS(0)
145
    "ld4       {v0.8b,v1.8b,v2.8b,v3.8b}, [%0], #32                \n"  // src line 0
146
    "subs      %2, %2, #24                             \n"
147
    "orr       v2.16b, v3.16b, v3.16b                  \n"  // order v0, v1, v2
148
    MEMACCESS(1)
149
    "st3       {v0.8b,v1.8b,v2.8b}, [%1], #24                \n"
150
    "b.gt      1b                                      \n"
151 152 153 154
  : "+r"(src_ptr),          // %0
    "+r"(dst_ptr),          // %1
    "+r"(dst_width)         // %2
  :
155
  : "v0", "v1", "v2", "v3", "memory", "cc"
156 157 158 159 160 161 162 163 164
  );
}
#endif //HAS_SCALEROWDOWN34_NEON

#ifdef HAS_SCALEROWDOWN34_NEON
void ScaleRowDown34_0_Box_NEON(const uint8* src_ptr,
                               ptrdiff_t src_stride,
                               uint8* dst_ptr, int dst_width) {
  asm volatile (
165 166 167
    "movi      v20.8b, #3                              \n"
    "add       %3, %3, %0                              \n"
  "1:                                                  \n"
168
    MEMACCESS(0)
169
    "ld4       {v0.8b,v1.8b,v2.8b,v3.8b}, [%0], #32                \n"  // src line 0
170
    MEMACCESS(3)
171
    "ld4       {v4.8b,v5.8b,v6.8b,v7.8b}, [%3], #32                \n"  // src line 1
172
    "subs         %2, %2, #24                          \n"
173 174 175 176

    // filter src line 0 with src line 1
    // expand chars to shorts to allow for room
    // when adding lines together
177 178 179 180
    "ushll     v16.8h, v4.8b, #0                       \n"
    "ushll     v17.8h, v5.8b, #0                       \n"
    "ushll     v18.8h, v6.8b, #0                       \n"
    "ushll     v19.8h, v7.8b, #0                       \n"
181 182

    // 3 * line_0 + line_1
183 184 185 186
    "umlal     v16.8h, v0.8b, v20.8b                   \n"
    "umlal     v17.8h, v1.8b, v20.8b                   \n"
    "umlal     v18.8h, v2.8b, v20.8b                   \n"
    "umlal     v19.8h, v3.8b, v20.8b                   \n"
187 188

    // (3 * line_0 + line_1) >> 2
189 190 191 192
    "uqrshrn   v0.8b, v16.8h, #2                       \n"
    "uqrshrn   v1.8b, v17.8h, #2                       \n"
    "uqrshrn   v2.8b, v18.8h, #2                       \n"
    "uqrshrn   v3.8b, v19.8h, #2                       \n"
193 194

    // a0 = (src[0] * 3 + s[1] * 1) >> 2
195 196 197
    "ushll     v16.8h, v1.8b, #0                       \n"
    "umlal     v16.8h, v0.8b, v20.8b                   \n"
    "uqrshrn   v0.8b, v16.8h, #2                       \n"
198 199

    // a1 = (src[1] * 1 + s[2] * 1) >> 1
200
    "urhadd    v1.8b, v1.8b, v2.8b                     \n"
201 202

    // a2 = (src[2] * 1 + s[3] * 3) >> 2
203 204 205
    "ushll     v16.8h, v2.8b, #0                       \n"
    "umlal     v16.8h, v3.8b, v20.8b                   \n"
    "uqrshrn   v2.8b, v16.8h, #2                       \n"
206 207

    MEMACCESS(1)
208
    "st3       {v0.8b,v1.8b,v2.8b}, [%1], #24                \n"
209

210
    "b.gt      1b                                      \n"
211 212 213 214 215
  : "+r"(src_ptr),          // %0
    "+r"(dst_ptr),          // %1
    "+r"(dst_width),        // %2
    "+r"(src_stride)        // %3
  :
216 217
  : "v0", "v1", "v2", "v3", "v4", "v5", "v6", "v7", "v16", "v17", "v18", "v19",
    "v20", "memory", "cc"
218 219 220 221 222 223 224 225 226
  );
}
#endif //ScaleRowDown34_0_Box_NEON

#ifdef HAS_SCALEROWDOWN34_NEON
void ScaleRowDown34_1_Box_NEON(const uint8* src_ptr,
                               ptrdiff_t src_stride,
                               uint8* dst_ptr, int dst_width) {
  asm volatile (
227 228 229
    "movi      v20.8b, #3                              \n"
    "add       %3, %3, %0                              \n"
  "1:                                                  \n"
230
    MEMACCESS(0)
231
    "ld4       {v0.8b,v1.8b,v2.8b,v3.8b}, [%0], #32                \n"  // src line 0
232
    MEMACCESS(3)
233
    "ld4       {v4.8b,v5.8b,v6.8b,v7.8b}, [%3], #32                \n"  // src line 1
234
    "subs         %2, %2, #24                          \n"
235
    // average src line 0 with src line 1
236 237 238 239
    "urhadd    v0.8b, v0.8b, v4.8b                     \n"
    "urhadd    v1.8b, v1.8b, v5.8b                     \n"
    "urhadd    v2.8b, v2.8b, v6.8b                     \n"
    "urhadd    v3.8b, v3.8b, v7.8b                     \n"
240 241

    // a0 = (src[0] * 3 + s[1] * 1) >> 2
242 243 244
    "ushll     v4.8h, v1.8b, #0                        \n"
    "umlal     v4.8h, v0.8b, v20.8b                    \n"
    "uqrshrn   v0.8b, v4.8h, #2                        \n"
245 246

    // a1 = (src[1] * 1 + s[2] * 1) >> 1
247
    "urhadd    v1.8b, v1.8b, v2.8b                     \n"
248 249

    // a2 = (src[2] * 1 + s[3] * 3) >> 2
250 251 252
    "ushll     v4.8h, v2.8b, #0                        \n"
    "umlal     v4.8h, v3.8b, v20.8b                    \n"
    "uqrshrn   v2.8b, v4.8h, #2                        \n"
253 254

    MEMACCESS(1)
255
    "st3       {v0.8b,v1.8b,v2.8b}, [%1], #24                \n"
256
    "b.gt      1b                                      \n"
257 258 259 260 261
  : "+r"(src_ptr),          // %0
    "+r"(dst_ptr),          // %1
    "+r"(dst_width),        // %2
    "+r"(src_stride)        // %3
  :
262
  : "v0", "v1", "v2", "v3", "v4", "v5", "v6", "v7", "v20", "memory", "cc"
263 264 265 266 267 268 269 270
  );
}
#endif //HAS_SCALEROWDOWN34_NEON

#ifdef HAS_SCALEROWDOWN38_NEON
static uvec8 kShuf38 =
  { 0, 3, 6, 8, 11, 14, 16, 19, 22, 24, 27, 30, 0, 0, 0, 0 };
static uvec8 kShuf38_2 =
271
  { 0, 16, 32, 2, 18, 33, 4, 20, 34, 6, 22, 35, 0, 0, 0, 0 };
272 273 274 275 276 277 278 279 280 281 282 283 284
static vec16 kMult38_Div6 =
  { 65536 / 12, 65536 / 12, 65536 / 12, 65536 / 12,
    65536 / 12, 65536 / 12, 65536 / 12, 65536 / 12 };
static vec16 kMult38_Div9 =
  { 65536 / 18, 65536 / 18, 65536 / 18, 65536 / 18,
    65536 / 18, 65536 / 18, 65536 / 18, 65536 / 18 };

// 32 -> 12
void ScaleRowDown38_NEON(const uint8* src_ptr,
                         ptrdiff_t src_stride,
                         uint8* dst_ptr, int dst_width) {
  asm volatile (
    MEMACCESS(3)
285 286
    "ld1       {v3.16b}, [%3]                          \n"
  "1:                                                  \n"
287
    MEMACCESS(0)
288
    "ld1       {v0.16b,v1.16b}, [%0], #32             \n"
289
    "subs      %2, %2, #12                             \n"
290
    "tbl       v2.16b, {v0.16b,v1.16b}, v3.16b        \n"
291
    MEMACCESS(1)
292
    "st1       {v2.8b}, [%1], #8                       \n"
293
    MEMACCESS(1)
294
    "st1       {v2.s}[2], [%1], #4                     \n"
295
    "b.gt      1b                                      \n"
296 297 298 299
  : "+r"(src_ptr),          // %0
    "+r"(dst_ptr),          // %1
    "+r"(dst_width)         // %2
  : "r"(&kShuf38)           // %3
300
  : "v0", "v1", "v2", "v3", "memory", "cc"
301 302 303 304 305 306 307 308 309 310 311
  );
}

#endif //HAS_SCALEROWDOWN38_NEON

#ifdef HAS_SCALEROWDOWN38_NEON
// 32x3 -> 12x1
void OMITFP ScaleRowDown38_3_Box_NEON(const uint8* src_ptr,
                                      ptrdiff_t src_stride,
                                      uint8* dst_ptr, int dst_width) {
  const uint8* src_ptr1 = src_ptr + src_stride * 2;
312
  ptrdiff_t tmp_src_stride = src_stride;
313 314 315

  asm volatile (
    MEMACCESS(5)
316
    "ld1       {v29.8h}, [%5]                          \n"
317
    MEMACCESS(6)
318
    "ld1       {v30.16b}, [%6]                         \n"
319
    MEMACCESS(7)
320
    "ld1       {v31.8h}, [%7]                          \n"
321
    "add       %2, %2, %0                              \n"
322
  "1:                                                  \n"
323

324 325 326 327
    // 00 40 01 41 02 42 03 43
    // 10 50 11 51 12 52 13 53
    // 20 60 21 61 22 62 23 63
    // 30 70 31 71 32 72 33 73
328
    MEMACCESS(0)
329
    "ld4       {v0.8b,v1.8b,v2.8b,v3.8b}, [%0], #32                \n"
330
    MEMACCESS(3)
331
    "ld4       {v4.8b,v5.8b,v6.8b,v7.8b}, [%2], #32                \n"
332
    MEMACCESS(4)
333 334
    "ld4       {v16.8b,v17.8b,v18.8b,v19.8b}, [%3], #32              \n"
    "subs      %4, %4, #12                             \n"
335 336 337

    // Shuffle the input data around to get align the data
    //  so adjacent data can be added. 0,1 - 2,3 - 4,5 - 6,7
338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368
    // 00 10 01 11 02 12 03 13
    // 40 50 41 51 42 52 43 53
    "trn1      v20.8b, v0.8b, v1.8b                    \n"
    "trn2      v21.8b, v0.8b, v1.8b                    \n"
    "trn1      v22.8b, v4.8b, v5.8b                    \n"
    "trn2      v23.8b, v4.8b, v5.8b                    \n"
    "trn1      v24.8b, v16.8b, v17.8b                  \n"
    "trn2      v25.8b, v16.8b, v17.8b                  \n"

    // 20 30 21 31 22 32 23 33
    // 60 70 61 71 62 72 63 73
    "trn1      v0.8b, v2.8b, v3.8b                     \n"
    "trn2      v1.8b, v2.8b, v3.8b                     \n"
    "trn1      v4.8b, v6.8b, v7.8b                     \n"
    "trn2      v5.8b, v6.8b, v7.8b                     \n"
    "trn1      v16.8b, v18.8b, v19.8b                  \n"
    "trn2      v17.8b, v18.8b, v19.8b                  \n"

    // 00+10 01+11 02+12 03+13
    // 40+50 41+51 42+52 43+53
    "uaddlp    v20.4h, v20.8b                          \n"
    "uaddlp    v21.4h, v21.8b                          \n"
    "uaddlp    v22.4h, v22.8b                          \n"
    "uaddlp    v23.4h, v23.8b                          \n"
    "uaddlp    v24.4h, v24.8b                          \n"
    "uaddlp    v25.4h, v25.8b                          \n"

    // 60+70 61+71 62+72 63+73
    "uaddlp    v1.4h, v1.8b                            \n"
    "uaddlp    v5.4h, v5.8b                            \n"
    "uaddlp    v17.4h, v17.8b                          \n"
369 370

    // combine source lines
371 372 373 374 375 376
    "add       v20.4h, v20.4h, v22.4h                  \n"
    "add       v21.4h, v21.4h, v23.4h                  \n"
    "add       v20.4h, v20.4h, v24.4h                  \n"
    "add       v21.4h, v21.4h, v25.4h                  \n"
    "add       v2.4h, v1.4h, v5.4h                     \n"
    "add       v2.4h, v2.4h, v17.4h                    \n"
377 378 379 380

    // dst_ptr[3] = (s[6 + st * 0] + s[7 + st * 0]
    //             + s[6 + st * 1] + s[7 + st * 1]
    //             + s[6 + st * 2] + s[7 + st * 2]) / 6
381 382
    "sqrdmulh  v2.8h, v2.8h, v29.8h                    \n"
    "xtn       v2.8b,  v2.8h                           \n"
383 384 385 386 387 388

    // Shuffle 2,3 reg around so that 2 can be added to the
    //  0,1 reg and 3 can be added to the 4,5 reg. This
    //  requires expanding from u8 to u16 as the 0,1 and 4,5
    //  registers are already expanded. Then do transposes
    //  to get aligned.
389 390 391
    // xx 20 xx 30 xx 21 xx 31 xx 22 xx 32 xx 23 xx 33
    "ushll     v16.8h, v16.8b, #0                      \n"
    "uaddl     v0.8h, v0.8b, v4.8b                     \n"
392 393

    // combine source lines
394
    "add       v0.8h, v0.8h, v16.8h                    \n"
395

396 397 398 399 400 401
    // xx 20 xx 21 xx 22 xx 23
    // xx 30 xx 31 xx 32 xx 33
    "trn1      v1.8h, v0.8h, v0.8h                     \n"
    "trn2      v4.8h, v0.8h, v0.8h                     \n"
    "xtn       v0.4h, v1.4s                            \n"
    "xtn       v4.4h, v4.4s                            \n"
402 403

    // 0+1+2, 3+4+5
404 405
    "add       v20.8h, v20.8h, v0.8h                   \n"
    "add       v21.8h, v21.8h, v4.8h                   \n"
406 407 408 409

    // Need to divide, but can't downshift as the the value
    //  isn't a power of 2. So multiply by 65536 / n
    //  and take the upper 16 bits.
410 411
    "sqrdmulh  v0.8h, v20.8h, v31.8h                   \n"
    "sqrdmulh  v1.8h, v21.8h, v31.8h                   \n"
412 413 414

    // Align for table lookup, vtbl requires registers to
    //  be adjacent
415
    "tbl       v3.16b, {v0.16b, v1.16b, v2.16b}, v30.16b \n"
416 417

    MEMACCESS(1)
418
    "st1       {v3.8b}, [%1], #8                       \n"
419
    MEMACCESS(1)
420
    "st1       {v3.s}[2], [%1], #4                     \n"
421
    "b.gt      1b                                      \n"
422 423
  : "+r"(src_ptr),          // %0
    "+r"(dst_ptr),          // %1
424 425 426
    "+r"(tmp_src_stride),   // %2
    "+r"(src_ptr1),         // %3
    "+r"(dst_width)         // %4
427 428 429
  : "r"(&kMult38_Div6),     // %5
    "r"(&kShuf38_2),        // %6
    "r"(&kMult38_Div9)      // %7
430 431 432
  : "v0", "v1", "v2", "v3", "v4", "v5", "v6", "v7", "v16", "v17",
    "v18", "v19", "v20", "v21", "v22", "v23", "v24", "v25", "v29",
    "v30", "v31", "memory", "cc"
433 434 435 436 437 438 439 440 441
  );
}
#endif //HAS_SCALEROWDOWN38_NEON

#ifdef HAS_SCALEROWDOWN38_NEON
// 32x2 -> 12x1
void ScaleRowDown38_2_Box_NEON(const uint8* src_ptr,
                               ptrdiff_t src_stride,
                               uint8* dst_ptr, int dst_width) {
442 443
  // TODO(fbarchard): use src_stride directly for clang 3.5+.
  ptrdiff_t tmp_src_stride = src_stride;
444 445
  asm volatile (
    MEMACCESS(4)
446
    "ld1       {v30.8h}, [%4]                          \n"
447
    MEMACCESS(5)
448
    "ld1       {v31.16b}, [%5]                         \n"
449
    "add       %2, %2, %0                              \n"
450
  "1:                                                  \n"
451

452 453 454 455
    // 00 40 01 41 02 42 03 43
    // 10 50 11 51 12 52 13 53
    // 20 60 21 61 22 62 23 63
    // 30 70 31 71 32 72 33 73
456
    MEMACCESS(0)
457
    "ld4       {v0.8b,v1.8b,v2.8b,v3.8b}, [%0], #32                \n"
458
    MEMACCESS(3)
459
    "ld4       {v4.8b,v5.8b,v6.8b,v7.8b}, [%3], #32                \n"
460
    "subs      %3, %3, #12                             \n"
461 462 463

    // Shuffle the input data around to get align the data
    //  so adjacent data can be added. 0,1 - 2,3 - 4,5 - 6,7
464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487
    // 00 10 01 11 02 12 03 13
    // 40 50 41 51 42 52 43 53
    "trn1      v16.8b, v0.8b, v1.8b                    \n"
    "trn2      v17.8b, v0.8b, v1.8b                    \n"
    "trn1      v18.8b, v4.8b, v5.8b                    \n"
    "trn2      v19.8b, v4.8b, v5.8b                    \n"

    // 20 30 21 31 22 32 23 33
    // 60 70 61 71 62 72 63 73
    "trn1      v0.8b, v2.8b, v3.8b                     \n"
    "trn2      v1.8b, v2.8b, v3.8b                     \n"
    "trn1      v4.8b, v6.8b, v7.8b                     \n"
    "trn2      v5.8b, v6.8b, v7.8b                     \n"

    // 00+10 01+11 02+12 03+13
    // 40+50 41+51 42+52 43+53
    "uaddlp    v16.4h, v16.8b                          \n"
    "uaddlp    v17.4h, v17.8b                          \n"
    "uaddlp    v18.4h, v18.8b                          \n"
    "uaddlp    v19.4h, v19.8b                          \n"

    // 60+70 61+71 62+72 63+73
    "uaddlp    v1.4h, v1.8b                            \n"
    "uaddlp    v5.4h, v5.8b                            \n"
488 489

    // combine source lines
490 491 492
    "add       v16.4h, v16.4h, v18.4h                  \n"
    "add       v17.4h, v17.4h, v19.4h                  \n"
    "add       v2.4h, v1.4h, v5.4h                     \n"
493 494

    // dst_ptr[3] = (s[6] + s[7] + s[6+st] + s[7+st]) / 4
495
    "uqrshrn   v2.8b, v2.8h, #2                        \n"
496 497 498 499 500 501

    // Shuffle 2,3 reg around so that 2 can be added to the
    //  0,1 reg and 3 can be added to the 4,5 reg. This
    //  requires expanding from u8 to u16 as the 0,1 and 4,5
    //  registers are already expanded. Then do transposes
    //  to get aligned.
502
    // xx 20 xx 30 xx 21 xx 31 xx 22 xx 32 xx 23 xx 33
503 504

    // combine source lines
505
    "uaddl     v0.8h, v0.8b, v4.8b                     \n"
506

507 508 509 510 511 512
    // xx 20 xx 21 xx 22 xx 23
    // xx 30 xx 31 xx 32 xx 33
    "trn1      v1.8h, v0.8h, v0.8h                     \n"
    "trn2      v4.8h, v0.8h, v0.8h                     \n"
    "xtn       v0.4h, v1.4s                            \n"
    "xtn       v4.4h, v4.4s                            \n"
513 514

    // 0+1+2, 3+4+5
515 516
    "add       v16.8h, v16.8h, v0.8h                   \n"
    "add       v17.8h, v17.8h, v4.8h                   \n"
517 518 519 520

    // Need to divide, but can't downshift as the the value
    //  isn't a power of 2. So multiply by 65536 / n
    //  and take the upper 16 bits.
521 522
    "sqrdmulh  v0.8h, v16.8h, v30.8h                   \n"
    "sqrdmulh  v1.8h, v17.8h, v30.8h                   \n"
523 524 525 526

    // Align for table lookup, vtbl requires registers to
    //  be adjacent

527
    "tbl       v3.16b, {v0.16b, v1.16b, v2.16b}, v31.16b \n"
528 529

    MEMACCESS(1)
530
    "st1       {v3.8b}, [%1], #8                       \n"
531
    MEMACCESS(1)
532
    "st1       {v3.s}[2], [%1], #4                     \n"
533
    "b.gt      1b                                      \n"
534 535 536 537 538 539
  : "+r"(src_ptr),         // %0
    "+r"(dst_ptr),         // %1
    "+r"(tmp_src_stride),  // %2
    "+r"(dst_width)        // %3
  : "r"(&kMult38_Div6),    // %4
    "r"(&kShuf38_2)        // %5
540 541
  : "v0", "v1", "v2", "v3", "v4", "v5", "v6", "v7", "v16", "v17",
    "v18", "v19", "v30", "v31", "memory", "cc"
542 543 544 545 546 547 548 549
  );
}
#endif //HAS_SCALEROWDOWN38_NEON

// 16x2 -> 16x1
void ScaleFilterRows_NEON(uint8* dst_ptr,
                          const uint8* src_ptr, ptrdiff_t src_stride,
                          int dst_width, int source_y_fraction) {
550
    int y_fraction = 256 - source_y_fraction;
551 552
  asm volatile (
    "cmp          %4, #0                       \n"
553
    "b.eq         100f                         \n"
554
    "add          %2, %2, %1                   \n"
555
    "cmp          %4, #64                      \n"
556
    "b.eq         75f                          \n"
557
    "cmp          %4, #128                     \n"
558
    "b.eq         50f                          \n"
559
    "cmp          %4, #192                     \n"
560
    "b.eq         25f                          \n"
561

562 563
    "dup          v5.8b, %w4                   \n"
    "dup          v4.8b, %w5                   \n"
564 565 566
    // General purpose row blend.
  "1:                                          \n"
    MEMACCESS(1)
567
    "ld1          {v0.16b}, [%1], #16          \n"
568
    MEMACCESS(2)
569
    "ld1          {v1.16b}, [%2], #16          \n"
570
    "subs         %3, %3, #16                  \n"
571 572 573 574 575 576
    "umull        v6.8h, v0.8b, v4.8b          \n"
    "umull2       v7.8h, v0.16b, v4.16b        \n"
    "umlal        v6.8h, v1.8b, v5.8b          \n"
    "umlal2       v7.8h, v1.16b, v5.16b        \n"
    "rshrn        v0.8b, v6.8h, #8             \n"
    "rshrn2       v0.16b, v7.8h, #8            \n"
577
    MEMACCESS(0)
578
    "st1          {v0.16b}, [%0], #16          \n"
579
    "b.gt         1b                           \n"
580 581 582 583 584
    "b            99f                          \n"

    // Blend 25 / 75.
  "25:                                         \n"
    MEMACCESS(1)
585
    "ld1          {v0.16b}, [%1], #16          \n"
586
    MEMACCESS(2)
587
    "ld1          {v1.16b}, [%2], #16          \n"
588
    "subs         %3, %3, #16                  \n"
589 590
    "urhadd       v0.16b, v0.16b, v1.16b       \n"
    "urhadd       v0.16b, v0.16b, v1.16b       \n"
591
    MEMACCESS(0)
592
    "st1          {v0.16b}, [%0], #16          \n"
593
    "b.gt         25b                          \n"
594 595 596 597 598
    "b            99f                          \n"

    // Blend 50 / 50.
  "50:                                         \n"
    MEMACCESS(1)
599
    "ld1          {v0.16b}, [%1], #16          \n"
600
    MEMACCESS(2)
601
    "ld1          {v1.16b}, [%2], #16          \n"
602
    "subs         %3, %3, #16                  \n"
603
    "urhadd       v0.16b, v0.16b, v1.16b       \n"
604
    MEMACCESS(0)
605
    "st1          {v0.16b}, [%0], #16          \n"
606
    "b.gt         50b                          \n"
607 608 609 610 611
    "b            99f                          \n"

    // Blend 75 / 25.
  "75:                                         \n"
    MEMACCESS(1)
612
    "ld1          {v1.16b}, [%1], #16          \n"
613
    MEMACCESS(2)
614
    "ld1          {v0.16b}, [%2], #16          \n"
615
    "subs         %3, %3, #16                  \n"
616 617
    "urhadd       v0.16b, v0.16b, v1.16b       \n"
    "urhadd       v0.16b, v0.16b, v1.16b       \n"
618
    MEMACCESS(0)
619
    "st1          {v0.16b}, [%0], #16          \n"
620
    "b.gt         75b                          \n"
621 622 623 624 625
    "b            99f                          \n"

    // Blend 100 / 0 - Copy row unchanged.
  "100:                                        \n"
    MEMACCESS(1)
626
    "ld1          {v0.16b}, [%1], #16          \n"
627 628
    "subs         %3, %3, #16                  \n"
    MEMACCESS(0)
629
    "st1          {v0.16b}, [%0], #16          \n"
630
    "b.gt         100b                         \n"
631 632 633

  "99:                                         \n"
    MEMACCESS(0)
634
    "st1          {v0.b}[15], [%0]             \n"
635 636 637 638
  : "+r"(dst_ptr),          // %0
    "+r"(src_ptr),          // %1
    "+r"(src_stride),       // %2
    "+r"(dst_width),        // %3
639 640
    "+r"(source_y_fraction),// %4
    "+r"(y_fraction)        // %5
641
  :
642
  : "v0", "v1", "v2", "v3", "v4", "v5", "v6", "v7", "memory", "cc"
643 644 645 646 647 648 649 650 651
  );
}

#ifdef HAS_SCALEARGBROWDOWN2_NEON
void ScaleARGBRowDown2_NEON(const uint8* src_ptr, ptrdiff_t src_stride,
                            uint8* dst, int dst_width) {
  asm volatile (
  "1:                                          \n"
    // load even pixels into q0, odd into q1
652 653 654 655
    MEMACCESS (0)
    "ld2        {v0.4s, v1.4s}, [%0], #32      \n"
    MEMACCESS (0)
    "ld2        {v2.4s, v3.4s}, [%0], #32      \n"
656
    "subs       %2, %2, #8                     \n"  // 8 processed per loop
657 658 659 660
    MEMACCESS (1)
    "st1        {v1.16b}, [%1], #16            \n"  // store odd pixels
    MEMACCESS (1)
    "st1        {v3.16b}, [%1], #16            \n"
661
    "b.gt       1b                             \n"
662 663 664
  : "+r" (src_ptr),          // %0
    "+r" (dst),              // %1
    "+r" (dst_width)         // %2
665
  :
666
  : "memory", "cc", "v0", "v1", "v2", "v3"  // Clobber List
667 668 669 670 671 672 673 674 675 676 677
  );
}
#endif //HAS_SCALEARGBROWDOWN2_NEON

#ifdef HAS_SCALEARGBROWDOWN2_NEON
void ScaleARGBRowDown2Box_NEON(const uint8* src_ptr, ptrdiff_t src_stride,
                               uint8* dst, int dst_width) {
  asm volatile (
    // change the stride to row 2 pointer
    "add        %1, %1, %0                     \n"
  "1:                                          \n"
678
    MEMACCESS (0)
679
    "ld4        {v0.16b,v1.16b,v2.16b,v3.16b}, [%0], #64   \n"  // load 8 ARGB pixels.
680
    "subs       %3, %3, #8                     \n"  // 8 processed per loop.
681 682 683 684 685
    "uaddlp     v0.8h, v0.16b                  \n"  // B 16 bytes -> 8 shorts.
    "uaddlp     v1.8h, v1.16b                  \n"  // G 16 bytes -> 8 shorts.
    "uaddlp     v2.8h, v2.16b                  \n"  // R 16 bytes -> 8 shorts.
    "uaddlp     v3.8h, v3.16b                  \n"  // A 16 bytes -> 8 shorts.
    MEMACCESS (1)
686
    "ld4        {v16.16b,v17.16b,v18.16b,v19.16b}, [%1], #64 \n"  // load 8 more ARGB pixels.
687 688 689 690 691 692 693 694 695
    "uadalp     v0.8h, v16.16b                 \n"  // B 16 bytes -> 8 shorts.
    "uadalp     v1.8h, v17.16b                 \n"  // G 16 bytes -> 8 shorts.
    "uadalp     v2.8h, v18.16b                 \n"  // R 16 bytes -> 8 shorts.
    "uadalp     v3.8h, v19.16b                 \n"  // A 16 bytes -> 8 shorts.
    "rshrn      v0.8b, v0.8h, #2               \n"  // downshift, round and pack
    "rshrn      v1.8b, v1.8h, #2               \n"
    "rshrn      v2.8b, v2.8h, #2               \n"
    "rshrn      v3.8b, v3.8h, #2               \n"
    MEMACCESS (2)
696
    "st4        {v0.8b,v1.8b,v2.8b,v3.8b}, [%2], #32     \n"
697
    "b.gt       1b                             \n"
698 699 700 701
  : "+r" (src_ptr),          // %0
    "+r" (src_stride),       // %1
    "+r" (dst),              // %2
    "+r" (dst_width)         // %3
702
  :
703
  : "memory", "cc", "v0", "v1", "v2", "v3", "v16", "v17", "v18", "v19"
704 705 706 707 708 709 710 711 712 713 714 715
  );
}
#endif //HAS_SCALEARGBROWDOWN2_NEON

#ifdef HAS_SCALEARGBROWDOWNEVEN_NEON
// Reads 4 pixels at a time.
// Alignment requirement: src_argb 4 byte aligned.
void ScaleARGBRowDownEven_NEON(const uint8* src_argb,  ptrdiff_t src_stride,
                               int src_stepx, uint8* dst_argb, int dst_width) {
  asm volatile (
  "1:                                          \n"
    MEMACCESS(0)
716
    "ld1        {v0.s}[0], [%0], %3            \n"
717
    MEMACCESS(0)
718
    "ld1        {v0.s}[1], [%0], %3            \n"
719
    MEMACCESS(0)
720
    "ld1        {v0.s}[2], [%0], %3            \n"
721
    MEMACCESS(0)
722
    "ld1        {v0.s}[3], [%0], %3            \n"
723 724
    "subs       %2, %2, #4                     \n"  // 4 pixels per loop.
    MEMACCESS(1)
725
    "st1        {v0.16b}, [%1], #16            \n"
726
    "b.gt       1b                             \n"
727 728 729
  : "+r"(src_argb),    // %0
    "+r"(dst_argb),    // %1
    "+r"(dst_width)    // %2
730 731
  : "r"(src_stepx * 4) // %3
  : "memory", "cc", "v0"
732 733 734 735 736 737 738
  );
}
#endif //HAS_SCALEARGBROWDOWNEVEN_NEON

#ifdef HAS_SCALEARGBROWDOWNEVEN_NEON
// Reads 4 pixels at a time.
// Alignment requirement: src_argb 4 byte aligned.
739 740
// TODO, might be worth another optimization pass in future.
// It could be upgraded to 8 pixels at a time to start with.
741 742 743 744 745 746 747
void ScaleARGBRowDownEvenBox_NEON(const uint8* src_argb, ptrdiff_t src_stride,
                                  int src_stepx,
                                  uint8* dst_argb, int dst_width) {
  asm volatile (
    "add        %1, %1, %0                     \n"
  "1:                                          \n"
    MEMACCESS(0)
748
    "ld1     {v0.8b}, [%0], %4                 \n"  // Read 4 2x2 blocks -> 2x1
749
    MEMACCESS(1)
750
    "ld1     {v1.8b}, [%1], %4                 \n"
751
    MEMACCESS(0)
752
    "ld1     {v2.8b}, [%0], %4                 \n"
753
    MEMACCESS(1)
754
    "ld1     {v3.8b}, [%1], %4                 \n"
755
    MEMACCESS(0)
756
    "ld1     {v4.8b}, [%0], %4                 \n"
757
    MEMACCESS(1)
758
    "ld1     {v5.8b}, [%1], %4                 \n"
759
    MEMACCESS(0)
760
    "ld1     {v6.8b}, [%0], %4                 \n"
761
    MEMACCESS(1)
762 763 764 765 766 767 768 769 770 771 772 773 774 775 776
    "ld1     {v7.8b}, [%1], %4                 \n"
    "uaddl   v0.8h, v0.8b, v1.8b               \n"
    "uaddl   v2.8h, v2.8b, v3.8b               \n"
    "uaddl   v4.8h, v4.8b, v5.8b               \n"
    "uaddl   v6.8h, v6.8b, v7.8b               \n"
    "mov     v16.d[1], v0.d[1]                 \n"  // ab_cd -> ac_bd
    "mov     v0.d[1], v2.d[0]                  \n"
    "mov     v2.d[0], v16.d[1]                 \n"
    "mov     v16.d[1], v4.d[1]                 \n"  // ef_gh -> eg_fh
    "mov     v4.d[1], v6.d[0]                  \n"
    "mov     v6.d[0], v16.d[1]                 \n"
    "add     v0.8h, v0.8h, v2.8h               \n"  // (a+b)_(c+d)
    "add     v4.8h, v4.8h, v6.8h               \n"  // (e+f)_(g+h)
    "rshrn   v0.8b, v0.8h, #2                  \n"  // first 2 pixels.
    "rshrn2  v0.16b, v4.8h, #2                 \n"  // next 2 pixels.
777 778
    "subs       %3, %3, #4                     \n"  // 4 pixels per loop.
    MEMACCESS(2)
779
    "st1     {v0.16b}, [%2], #16               \n"
780
    "b.gt       1b                             \n"
781 782 783 784
  : "+r"(src_argb),    // %0
    "+r"(src_stride),  // %1
    "+r"(dst_argb),    // %2
    "+r"(dst_width)    // %3
785 786
  : "r"(src_stepx * 4) // %4
  : "memory", "cc", "v0", "v1", "v2", "v3", "v4", "v5", "v6", "v7", "v16"
787 788 789
  );
}
#endif  // HAS_SCALEARGBROWDOWNEVEN_NEON
790
#endif  // !defined(LIBYUV_DISABLE_NEON) && defined(__aarch64__)
791 792 793 794 795

#ifdef __cplusplus
}  // extern "C"
}  // namespace libyuv
#endif