convert_from_argb.cc 36 KB
Newer Older
1 2 3 4 5 6
/*
 *  Copyright 2012 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
7
 *  in the file PATENTS. All contributing project authors may
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
 *  be found in the AUTHORS file in the root of the source tree.
 */

#include "libyuv/convert_from_argb.h"

#include "libyuv/basic_types.h"
#include "libyuv/cpu_id.h"
#include "libyuv/planar_functions.h"
#include "libyuv/row.h"

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

// ARGB little endian (bgra in memory) to I444
LIBYUV_API
int ARGBToI444(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               uint8* dst_u, int dst_stride_u,
               uint8* dst_v, int dst_stride_v,
               int width, int height) {
30
  int y;
31
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
32 33
      ARGBToYRow_C;
  void (*ARGBToUV444Row)(const uint8* src_argb, uint8* dst_u, uint8* dst_v,
34
      int width) = ARGBToUV444Row_C;
35 36 37 38 39 40 41 42
  if (!src_argb || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
43
  // Coalesce rows.
44 45 46 47
  if (src_stride_argb == width * 4 &&
      dst_stride_y == width &&
      dst_stride_u == width &&
      dst_stride_v == width) {
48 49 50
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_y = dst_stride_u = dst_stride_v = 0;
51
  }
52
#if defined(HAS_ARGBTOUV444ROW_SSSE3)
53
    if (TestCpuFlag(kCpuHasSSSE3)) {
54 55
      ARGBToUV444Row = ARGBToUV444Row_Any_SSSE3;
      if (IS_ALIGNED(width, 16)) {
56
        ARGBToUV444Row = ARGBToUV444Row_SSSE3;
57 58
      }
  }
59 60
#endif
#if defined(HAS_ARGBTOUV444ROW_NEON)
61
  if (TestCpuFlag(kCpuHasNEON)) {
62 63 64 65 66
    ARGBToUV444Row = ARGBToUV444Row_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToUV444Row = ARGBToUV444Row_NEON;
    }
  }
67
#endif
68
#if defined(HAS_ARGBTOYROW_SSSE3)
69
  if (TestCpuFlag(kCpuHasSSSE3)) {
70 71
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
72
      ARGBToYRow = ARGBToYRow_SSSE3;
73 74
    }
  }
75 76 77 78 79 80 81 82
#endif
#if defined(HAS_ARGBTOYROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
83 84
#endif
#if defined(HAS_ARGBTOYROW_NEON)
85
  if (TestCpuFlag(kCpuHasNEON)) {
86 87 88 89 90 91 92
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
  }
#endif

93
  for (y = 0; y < height; ++y) {
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
    ARGBToUV444Row(src_argb, dst_u, dst_v, width);
    ARGBToYRow(src_argb, dst_y, width);
    src_argb += src_stride_argb;
    dst_y += dst_stride_y;
    dst_u += dst_stride_u;
    dst_v += dst_stride_v;
  }
  return 0;
}

// ARGB little endian (bgra in memory) to I422
LIBYUV_API
int ARGBToI422(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               uint8* dst_u, int dst_stride_u,
               uint8* dst_v, int dst_stride_v,
               int width, int height) {
111 112
  int y;
  void (*ARGBToUV422Row)(const uint8* src_argb, uint8* dst_u, uint8* dst_v,
113 114
      int width) = ARGBToUV422Row_C;
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
115
      ARGBToYRow_C;
116 117 118 119 120 121 122 123
  if (!src_argb || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
124
  // Coalesce rows.
125 126 127 128
  if (src_stride_argb == width * 4 &&
      dst_stride_y == width &&
      dst_stride_u * 2 == width &&
      dst_stride_v * 2 == width) {
129 130 131
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_y = dst_stride_u = dst_stride_v = 0;
132
  }
133
#if defined(HAS_ARGBTOUV422ROW_SSSE3)
134
  if (TestCpuFlag(kCpuHasSSSE3)) {
135 136
    ARGBToUV422Row = ARGBToUV422Row_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
137
      ARGBToUV422Row = ARGBToUV422Row_SSSE3;
138 139
    }
  }
140 141
#endif
#if defined(HAS_ARGBTOUV422ROW_NEON)
142
  if (TestCpuFlag(kCpuHasNEON)) {
143 144 145 146 147
    ARGBToUV422Row = ARGBToUV422Row_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUV422Row = ARGBToUV422Row_NEON;
    }
  }
148 149
#endif
#if defined(HAS_ARGBTOYROW_SSSE3)
150
  if (TestCpuFlag(kCpuHasSSSE3)) {
151 152
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
153
      ARGBToYRow = ARGBToYRow_SSSE3;
154 155
    }
  }
156
#endif
157 158 159 160 161 162 163 164
#if defined(HAS_ARGBTOYROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
165
#if defined(HAS_ARGBTOYROW_NEON)
166
  if (TestCpuFlag(kCpuHasNEON)) {
167 168 169 170 171 172 173
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
  }
#endif

174
  for (y = 0; y < height; ++y) {
175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
    ARGBToUV422Row(src_argb, dst_u, dst_v, width);
    ARGBToYRow(src_argb, dst_y, width);
    src_argb += src_stride_argb;
    dst_y += dst_stride_y;
    dst_u += dst_stride_u;
    dst_v += dst_stride_v;
  }
  return 0;
}

// ARGB little endian (bgra in memory) to I411
LIBYUV_API
int ARGBToI411(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               uint8* dst_u, int dst_stride_u,
               uint8* dst_v, int dst_stride_v,
               int width, int height) {
192 193
  int y;
  void (*ARGBToUV411Row)(const uint8* src_argb, uint8* dst_u, uint8* dst_v,
194 195
      int width) = ARGBToUV411Row_C;
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
196
      ARGBToYRow_C;
197 198 199 200 201 202 203 204
  if (!src_argb || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
205
  // Coalesce rows.
206 207 208 209
  if (src_stride_argb == width * 4 &&
      dst_stride_y == width &&
      dst_stride_u * 4 == width &&
      dst_stride_v * 4 == width) {
210 211 212
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_y = dst_stride_u = dst_stride_v = 0;
213
  }
214
#if defined(HAS_ARGBTOYROW_SSSE3)
215
  if (TestCpuFlag(kCpuHasSSSE3)) {
216 217
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
218
      ARGBToYRow = ARGBToYRow_SSSE3;
219 220
    }
  }
221 222
#endif
#if defined(HAS_ARGBTOYROW_AVX2)
223
  if (TestCpuFlag(kCpuHasAVX2)) {
224 225 226 227 228 229 230
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
#if defined(HAS_ARGBTOYROW_NEON)
231
  if (TestCpuFlag(kCpuHasNEON)) {
232 233 234 235
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
236 237 238
  }
#endif
#if defined(HAS_ARGBTOUV411ROW_NEON)
239
  if (TestCpuFlag(kCpuHasNEON)) {
240 241 242
    ARGBToUV411Row = ARGBToUV411Row_Any_NEON;
    if (IS_ALIGNED(width, 32)) {
      ARGBToUV411Row = ARGBToUV411Row_NEON;
243 244 245 246
    }
  }
#endif

247
  for (y = 0; y < height; ++y) {
248 249 250 251 252 253 254 255 256 257
    ARGBToUV411Row(src_argb, dst_u, dst_v, width);
    ARGBToYRow(src_argb, dst_y, width);
    src_argb += src_stride_argb;
    dst_y += dst_stride_y;
    dst_u += dst_stride_u;
    dst_v += dst_stride_v;
  }
  return 0;
}

258
LIBYUV_API
259 260 261 262
int ARGBToNV12(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               uint8* dst_uv, int dst_stride_uv,
               int width, int height) {
263 264 265 266
  int y;
  int halfwidth = (width + 1) >> 1;
  void (*ARGBToUVRow)(const uint8* src_argb0, int src_stride_argb,
                      uint8* dst_u, uint8* dst_v, int width) = ARGBToUVRow_C;
267
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
268 269 270
      ARGBToYRow_C;
  void (*MergeUVRow_)(const uint8* src_u, const uint8* src_v, uint8* dst_uv,
                      int width) = MergeUVRow_C;
271 272
  if (!src_argb ||
      !dst_y || !dst_uv ||
273
      width <= 0 || height == 0) {
274 275 276 277 278 279 280 281
    return -1;
  }
  // Negative height means invert the image.
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
282
#if defined(HAS_ARGBTOYROW_SSSE3) && defined(HAS_ARGBTOUVROW_SSSE3)
283
  if (TestCpuFlag(kCpuHasSSSE3)) {
284 285 286
    ARGBToUVRow = ARGBToUVRow_Any_SSSE3;
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
287 288
      ARGBToUVRow = ARGBToUVRow_SSSE3;
      ARGBToYRow = ARGBToYRow_SSSE3;
289 290
    }
  }
291
#endif
292
#if defined(HAS_ARGBTOYROW_AVX2) && defined(HAS_ARGBTOUVROW_AVX2)
293
  if (TestCpuFlag(kCpuHasAVX2)) {
294
    ARGBToUVRow = ARGBToUVRow_Any_AVX2;
295 296
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
297
      ARGBToUVRow = ARGBToUVRow_AVX2;
298 299 300 301
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
302
#if defined(HAS_ARGBTOYROW_NEON)
303
  if (TestCpuFlag(kCpuHasNEON)) {
304 305 306 307
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
308 309 310
  }
#endif
#if defined(HAS_ARGBTOUVROW_NEON)
311
  if (TestCpuFlag(kCpuHasNEON)) {
312 313 314
    ARGBToUVRow = ARGBToUVRow_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUVRow = ARGBToUVRow_NEON;
315 316 317 318
    }
  }
#endif
#if defined(HAS_MERGEUVROW_SSE2)
319
  if (TestCpuFlag(kCpuHasSSE2)) {
320 321
    MergeUVRow_ = MergeUVRow_Any_SSE2;
    if (IS_ALIGNED(halfwidth, 16)) {
322
      MergeUVRow_ = MergeUVRow_SSE2;
323 324 325 326
    }
  }
#endif
#if defined(HAS_MERGEUVROW_AVX2)
327
  if (TestCpuFlag(kCpuHasAVX2)) {
328 329
    MergeUVRow_ = MergeUVRow_Any_AVX2;
    if (IS_ALIGNED(halfwidth, 32)) {
330
      MergeUVRow_ = MergeUVRow_AVX2;
331 332 333 334
    }
  }
#endif
#if defined(HAS_MERGEUVROW_NEON)
335
  if (TestCpuFlag(kCpuHasNEON)) {
336 337 338 339 340 341
    MergeUVRow_ = MergeUVRow_Any_NEON;
    if (IS_ALIGNED(halfwidth, 16)) {
      MergeUVRow_ = MergeUVRow_NEON;
    }
  }
#endif
342 343
  {
    // Allocate a rows of uv.
344 345
    align_buffer_64(row_u, ((halfwidth + 31) & ~31) * 2);
    uint8* row_v = row_u + ((halfwidth + 31) & ~31);
346

347 348 349 350 351 352 353 354 355 356 357 358 359 360 361
    for (y = 0; y < height - 1; y += 2) {
      ARGBToUVRow(src_argb, src_stride_argb, row_u, row_v, width);
      MergeUVRow_(row_u, row_v, dst_uv, halfwidth);
      ARGBToYRow(src_argb, dst_y, width);
      ARGBToYRow(src_argb + src_stride_argb, dst_y + dst_stride_y, width);
      src_argb += src_stride_argb * 2;
      dst_y += dst_stride_y * 2;
      dst_uv += dst_stride_uv;
    }
    if (height & 1) {
      ARGBToUVRow(src_argb, 0, row_u, row_v, width);
      MergeUVRow_(row_u, row_v, dst_uv, halfwidth);
      ARGBToYRow(src_argb, dst_y, width);
    }
    free_aligned_buffer_64(row_u);
362 363 364 365 366
  }
  return 0;
}

// Same as NV12 but U and V swapped.
367
LIBYUV_API
368 369 370 371
int ARGBToNV21(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               uint8* dst_uv, int dst_stride_uv,
               int width, int height) {
372 373 374 375
  int y;
  int halfwidth = (width + 1) >> 1;
  void (*ARGBToUVRow)(const uint8* src_argb0, int src_stride_argb,
                      uint8* dst_u, uint8* dst_v, int width) = ARGBToUVRow_C;
376
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
377 378 379
      ARGBToYRow_C;
  void (*MergeUVRow_)(const uint8* src_u, const uint8* src_v, uint8* dst_uv,
                      int width) = MergeUVRow_C;
380 381
  if (!src_argb ||
      !dst_y || !dst_uv ||
382
      width <= 0 || height == 0) {
383 384 385 386 387 388 389 390
    return -1;
  }
  // Negative height means invert the image.
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
391
#if defined(HAS_ARGBTOYROW_SSSE3) && defined(HAS_ARGBTOUVROW_SSSE3)
392
  if (TestCpuFlag(kCpuHasSSSE3)) {
393 394 395
    ARGBToUVRow = ARGBToUVRow_Any_SSSE3;
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
396 397
      ARGBToUVRow = ARGBToUVRow_SSSE3;
      ARGBToYRow = ARGBToYRow_SSSE3;
398 399
    }
  }
400
#endif
401
#if defined(HAS_ARGBTOYROW_AVX2) && defined(HAS_ARGBTOUVROW_AVX2)
402
  if (TestCpuFlag(kCpuHasAVX2)) {
403
    ARGBToUVRow = ARGBToUVRow_Any_AVX2;
404 405
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
406
      ARGBToUVRow = ARGBToUVRow_AVX2;
407 408 409 410
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
411
#if defined(HAS_ARGBTOYROW_NEON)
412
  if (TestCpuFlag(kCpuHasNEON)) {
413 414 415 416
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
417 418 419
  }
#endif
#if defined(HAS_ARGBTOUVROW_NEON)
420
  if (TestCpuFlag(kCpuHasNEON)) {
421 422 423
    ARGBToUVRow = ARGBToUVRow_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUVRow = ARGBToUVRow_NEON;
424 425 426 427
    }
  }
#endif
#if defined(HAS_MERGEUVROW_SSE2)
428
  if (TestCpuFlag(kCpuHasSSE2)) {
429 430
    MergeUVRow_ = MergeUVRow_Any_SSE2;
    if (IS_ALIGNED(halfwidth, 16)) {
431
      MergeUVRow_ = MergeUVRow_SSE2;
432 433 434 435
    }
  }
#endif
#if defined(HAS_MERGEUVROW_AVX2)
436
  if (TestCpuFlag(kCpuHasAVX2)) {
437 438
    MergeUVRow_ = MergeUVRow_Any_AVX2;
    if (IS_ALIGNED(halfwidth, 32)) {
439
      MergeUVRow_ = MergeUVRow_AVX2;
440 441 442 443
    }
  }
#endif
#if defined(HAS_MERGEUVROW_NEON)
444
  if (TestCpuFlag(kCpuHasNEON)) {
445 446 447 448 449 450
    MergeUVRow_ = MergeUVRow_Any_NEON;
    if (IS_ALIGNED(halfwidth, 16)) {
      MergeUVRow_ = MergeUVRow_NEON;
    }
  }
#endif
451 452
  {
    // Allocate a rows of uv.
453 454
    align_buffer_64(row_u, ((halfwidth + 31) & ~31) * 2);
    uint8* row_v = row_u + ((halfwidth + 31) & ~31);
455

456 457 458 459 460 461 462 463 464 465 466 467 468 469 470
    for (y = 0; y < height - 1; y += 2) {
      ARGBToUVRow(src_argb, src_stride_argb, row_u, row_v, width);
      MergeUVRow_(row_v, row_u, dst_uv, halfwidth);
      ARGBToYRow(src_argb, dst_y, width);
      ARGBToYRow(src_argb + src_stride_argb, dst_y + dst_stride_y, width);
      src_argb += src_stride_argb * 2;
      dst_y += dst_stride_y * 2;
      dst_uv += dst_stride_uv;
    }
    if (height & 1) {
      ARGBToUVRow(src_argb, 0, row_u, row_v, width);
      MergeUVRow_(row_v, row_u, dst_uv, halfwidth);
      ARGBToYRow(src_argb, dst_y, width);
    }
    free_aligned_buffer_64(row_u);
471 472 473 474 475
  }
  return 0;
}

// Convert ARGB to YUY2.
476
LIBYUV_API
477 478 479
int ARGBToYUY2(const uint8* src_argb, int src_stride_argb,
               uint8* dst_yuy2, int dst_stride_yuy2,
               int width, int height) {
480 481
  int y;
  void (*ARGBToUV422Row)(const uint8* src_argb, uint8* dst_u, uint8* dst_v,
482 483
      int width) = ARGBToUV422Row_C;
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
484 485 486 487
      ARGBToYRow_C;
  void (*I422ToYUY2Row)(const uint8* src_y, const uint8* src_u,
      const uint8* src_v, uint8* dst_yuy2, int width) = I422ToYUY2Row_C;

488
  if (!src_argb || !dst_yuy2 ||
489
      width <= 0 || height == 0) {
490 491 492 493 494 495 496 497
    return -1;
  }
  // Negative height means invert the image.
  if (height < 0) {
    height = -height;
    dst_yuy2 = dst_yuy2 + (height - 1) * dst_stride_yuy2;
    dst_stride_yuy2 = -dst_stride_yuy2;
  }
498
  // Coalesce rows.
499
  if (src_stride_argb == width * 4 &&
500
      dst_stride_yuy2 == width * 2) {
501 502 503
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_yuy2 = 0;
504
  }
505
#if defined(HAS_ARGBTOUV422ROW_SSSE3)
506
  if (TestCpuFlag(kCpuHasSSSE3)) {
507 508
    ARGBToUV422Row = ARGBToUV422Row_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
509
      ARGBToUV422Row = ARGBToUV422Row_SSSE3;
510 511
    }
  }
512 513
#endif
#if defined(HAS_ARGBTOUV422ROW_NEON)
514
  if (TestCpuFlag(kCpuHasNEON)) {
515 516 517 518 519
    ARGBToUV422Row = ARGBToUV422Row_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUV422Row = ARGBToUV422Row_NEON;
    }
  }
520 521
#endif
#if defined(HAS_ARGBTOYROW_SSSE3)
522
  if (TestCpuFlag(kCpuHasSSSE3)) {
523 524
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
525
      ARGBToYRow = ARGBToYRow_SSSE3;
526 527
    }
  }
528
#endif
529 530 531 532 533 534 535 536
#if defined(HAS_ARGBTOYROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
537
#if defined(HAS_ARGBTOYROW_NEON)
538
  if (TestCpuFlag(kCpuHasNEON)) {
539 540 541 542 543 544 545 546
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
  }
#endif

#if defined(HAS_I422TOYUY2ROW_SSE2)
547
  if (TestCpuFlag(kCpuHasSSE2)) {
548 549 550 551 552
    I422ToYUY2Row = I422ToYUY2Row_Any_SSE2;
    if (IS_ALIGNED(width, 16)) {
      I422ToYUY2Row = I422ToYUY2Row_SSE2;
    }
  }
553 554
#endif
#if defined(HAS_I422TOYUY2ROW_NEON)
555
  if (TestCpuFlag(kCpuHasNEON)) {
556 557 558 559 560 561
    I422ToYUY2Row = I422ToYUY2Row_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      I422ToYUY2Row = I422ToYUY2Row_NEON;
    }
  }
#endif
562

563 564 565 566 567
  {
    // Allocate a rows of yuv.
    align_buffer_64(row_y, ((width + 63) & ~63) * 2);
    uint8* row_u = row_y + ((width + 63) & ~63);
    uint8* row_v = row_u + ((width + 63) & ~63) / 2;
568

569 570 571 572 573 574 575
    for (y = 0; y < height; ++y) {
      ARGBToUV422Row(src_argb, row_u, row_v, width);
      ARGBToYRow(src_argb, row_y, width);
      I422ToYUY2Row(row_y, row_u, row_v, dst_yuy2, width);
      src_argb += src_stride_argb;
      dst_yuy2 += dst_stride_yuy2;
    }
576

577 578
    free_aligned_buffer_64(row_y);
  }
579 580 581 582
  return 0;
}

// Convert ARGB to UYVY.
583
LIBYUV_API
584 585 586
int ARGBToUYVY(const uint8* src_argb, int src_stride_argb,
               uint8* dst_uyvy, int dst_stride_uyvy,
               int width, int height) {
587 588
  int y;
  void (*ARGBToUV422Row)(const uint8* src_argb, uint8* dst_u, uint8* dst_v,
589 590
      int width) = ARGBToUV422Row_C;
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
591 592 593 594
      ARGBToYRow_C;
  void (*I422ToUYVYRow)(const uint8* src_y, const uint8* src_u,
      const uint8* src_v, uint8* dst_uyvy, int width) = I422ToUYVYRow_C;

595
  if (!src_argb || !dst_uyvy ||
596
      width <= 0 || height == 0) {
597 598 599 600 601 602 603 604
    return -1;
  }
  // Negative height means invert the image.
  if (height < 0) {
    height = -height;
    dst_uyvy = dst_uyvy + (height - 1) * dst_stride_uyvy;
    dst_stride_uyvy = -dst_stride_uyvy;
  }
605
  // Coalesce rows.
606
  if (src_stride_argb == width * 4 &&
607
      dst_stride_uyvy == width * 2) {
608 609 610
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_uyvy = 0;
611
  }
612
#if defined(HAS_ARGBTOUV422ROW_SSSE3)
613
  if (TestCpuFlag(kCpuHasSSSE3)) {
614 615
    ARGBToUV422Row = ARGBToUV422Row_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
616
      ARGBToUV422Row = ARGBToUV422Row_SSSE3;
617 618
    }
  }
619 620
#endif
#if defined(HAS_ARGBTOUV422ROW_NEON)
621
  if (TestCpuFlag(kCpuHasNEON)) {
622 623 624 625 626
    ARGBToUV422Row = ARGBToUV422Row_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUV422Row = ARGBToUV422Row_NEON;
    }
  }
627 628
#endif
#if defined(HAS_ARGBTOYROW_SSSE3)
629
  if (TestCpuFlag(kCpuHasSSSE3)) {
630 631
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
632
      ARGBToYRow = ARGBToYRow_SSSE3;
633 634
    }
  }
635
#endif
636 637 638 639 640 641 642 643
#if defined(HAS_ARGBTOYROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
644
#if defined(HAS_ARGBTOYROW_NEON)
645
  if (TestCpuFlag(kCpuHasNEON)) {
646 647 648 649 650 651 652 653
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
  }
#endif

#if defined(HAS_I422TOUYVYROW_SSE2)
654
  if (TestCpuFlag(kCpuHasSSE2)) {
655 656 657 658 659
    I422ToUYVYRow = I422ToUYVYRow_Any_SSE2;
    if (IS_ALIGNED(width, 16)) {
      I422ToUYVYRow = I422ToUYVYRow_SSE2;
    }
  }
660 661
#endif
#if defined(HAS_I422TOUYVYROW_NEON)
662
  if (TestCpuFlag(kCpuHasNEON)) {
663 664 665 666 667 668
    I422ToUYVYRow = I422ToUYVYRow_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      I422ToUYVYRow = I422ToUYVYRow_NEON;
    }
  }
#endif
669

670 671 672 673 674
  {
    // Allocate a rows of yuv.
    align_buffer_64(row_y, ((width + 63) & ~63) * 2);
    uint8* row_u = row_y + ((width + 63) & ~63);
    uint8* row_v = row_u + ((width + 63) & ~63) / 2;
675

676 677 678 679 680 681 682
    for (y = 0; y < height; ++y) {
      ARGBToUV422Row(src_argb, row_u, row_v, width);
      ARGBToYRow(src_argb, row_y, width);
      I422ToUYVYRow(row_y, row_u, row_v, dst_uyvy, width);
      src_argb += src_stride_argb;
      dst_uyvy += dst_stride_uyvy;
    }
683

684 685
    free_aligned_buffer_64(row_y);
  }
686 687 688 689 690 691 692 693
  return 0;
}

// Convert ARGB to I400.
LIBYUV_API
int ARGBToI400(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               int width, int height) {
694
  int y;
695
  void (*ARGBToYRow)(const uint8* src_argb, uint8* dst_y, int width) =
696
      ARGBToYRow_C;
697 698 699 700 701 702 703 704
  if (!src_argb || !dst_y || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
705
  // Coalesce rows.
706 707
  if (src_stride_argb == width * 4 &&
      dst_stride_y == width) {
708 709 710
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_y = 0;
fbarchard@google.com's avatar
fbarchard@google.com committed
711
  }
712
#if defined(HAS_ARGBTOYROW_SSSE3)
713
  if (TestCpuFlag(kCpuHasSSSE3)) {
714 715
    ARGBToYRow = ARGBToYRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
716
      ARGBToYRow = ARGBToYRow_SSSE3;
717 718
    }
  }
719 720
#endif
#if defined(HAS_ARGBTOYROW_AVX2)
721
  if (TestCpuFlag(kCpuHasAVX2)) {
722 723 724 725 726 727 728
    ARGBToYRow = ARGBToYRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYRow = ARGBToYRow_AVX2;
    }
  }
#endif
#if defined(HAS_ARGBTOYROW_NEON)
729
  if (TestCpuFlag(kCpuHasNEON)) {
730 731 732 733 734 735 736
    ARGBToYRow = ARGBToYRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYRow = ARGBToYRow_NEON;
    }
  }
#endif

737
  for (y = 0; y < height; ++y) {
738 739 740 741 742 743 744
    ARGBToYRow(src_argb, dst_y, width);
    src_argb += src_stride_argb;
    dst_y += dst_stride_y;
  }
  return 0;
}

fbarchard@google.com's avatar
fbarchard@google.com committed
745
// Shuffle table for converting ARGB to RGBA.
746
static uvec8 kShuffleMaskARGBToRGBA = {
fbarchard@google.com's avatar
fbarchard@google.com committed
747 748 749
  3u, 0u, 1u, 2u, 7u, 4u, 5u, 6u, 11u, 8u, 9u, 10u, 15u, 12u, 13u, 14u
};

750 751 752 753 754
// Convert ARGB to RGBA.
LIBYUV_API
int ARGBToRGBA(const uint8* src_argb, int src_stride_argb,
               uint8* dst_rgba, int dst_stride_rgba,
               int width, int height) {
fbarchard@google.com's avatar
fbarchard@google.com committed
755 756
  return ARGBShuffle(src_argb, src_stride_argb,
                     dst_rgba, dst_stride_rgba,
757
                     (const uint8*)(&kShuffleMaskARGBToRGBA),
fbarchard@google.com's avatar
fbarchard@google.com committed
758
                     width, height);
759 760 761 762 763 764 765
}

// Convert ARGB To RGB24.
LIBYUV_API
int ARGBToRGB24(const uint8* src_argb, int src_stride_argb,
                uint8* dst_rgb24, int dst_stride_rgb24,
                int width, int height) {
766
  int y;
767
  void (*ARGBToRGB24Row)(const uint8* src_argb, uint8* dst_rgb, int width) =
768
      ARGBToRGB24Row_C;
769 770 771 772 773 774 775 776
  if (!src_argb || !dst_rgb24 || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
777
  // Coalesce rows.
778 779
  if (src_stride_argb == width * 4 &&
      dst_stride_rgb24 == width * 3) {
780 781 782
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_rgb24 = 0;
fbarchard@google.com's avatar
fbarchard@google.com committed
783
  }
784
#if defined(HAS_ARGBTORGB24ROW_SSSE3)
785
  if (TestCpuFlag(kCpuHasSSSE3)) {
786 787 788 789 790
    ARGBToRGB24Row = ARGBToRGB24Row_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
      ARGBToRGB24Row = ARGBToRGB24Row_SSSE3;
    }
  }
791 792
#endif
#if defined(HAS_ARGBTORGB24ROW_NEON)
793
  if (TestCpuFlag(kCpuHasNEON)) {
794 795 796 797 798 799 800
    ARGBToRGB24Row = ARGBToRGB24Row_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToRGB24Row = ARGBToRGB24Row_NEON;
    }
  }
#endif

801
  for (y = 0; y < height; ++y) {
802 803 804 805 806 807 808 809 810 811 812 813
    ARGBToRGB24Row(src_argb, dst_rgb24, width);
    src_argb += src_stride_argb;
    dst_rgb24 += dst_stride_rgb24;
  }
  return 0;
}

// Convert ARGB To RAW.
LIBYUV_API
int ARGBToRAW(const uint8* src_argb, int src_stride_argb,
              uint8* dst_raw, int dst_stride_raw,
              int width, int height) {
814
  int y;
815
  void (*ARGBToRAWRow)(const uint8* src_argb, uint8* dst_rgb, int width) =
816
      ARGBToRAWRow_C;
817 818 819 820 821 822 823 824
  if (!src_argb || !dst_raw || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
825
  // Coalesce rows.
826 827
  if (src_stride_argb == width * 4 &&
      dst_stride_raw == width * 3) {
828 829 830
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_raw = 0;
fbarchard@google.com's avatar
fbarchard@google.com committed
831
  }
832
#if defined(HAS_ARGBTORAWROW_SSSE3)
833
  if (TestCpuFlag(kCpuHasSSSE3)) {
834 835 836 837 838
    ARGBToRAWRow = ARGBToRAWRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
      ARGBToRAWRow = ARGBToRAWRow_SSSE3;
    }
  }
839 840
#endif
#if defined(HAS_ARGBTORAWROW_NEON)
841
  if (TestCpuFlag(kCpuHasNEON)) {
842 843 844 845 846 847 848
    ARGBToRAWRow = ARGBToRAWRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToRAWRow = ARGBToRAWRow_NEON;
    }
  }
#endif

849
  for (y = 0; y < height; ++y) {
850 851 852 853 854 855 856
    ARGBToRAWRow(src_argb, dst_raw, width);
    src_argb += src_stride_argb;
    dst_raw += dst_stride_raw;
  }
  return 0;
}

857
// Ordered 8x8 dither for 888 to 565.  Values from 0 to 7.
858 859 860 861 862
static const uint8 kDither565_4x4[16] = {
  0, 4, 1, 5,
  6, 2, 7, 3,
  1, 5, 0, 4,
  7, 3, 6, 2,
863 864
};

865
// Convert ARGB To RGB565 with 4x4 dither matrix (16 bytes).
866 867 868
LIBYUV_API
int ARGBToRGB565Dither(const uint8* src_argb, int src_stride_argb,
                       uint8* dst_rgb565, int dst_stride_rgb565,
869
                       const uint8* dither4x4, int width, int height) {
870 871
  int y;
  void (*ARGBToRGB565DitherRow)(const uint8* src_argb, uint8* dst_rgb,
872
      const uint32 dither4, int width) = ARGBToRGB565DitherRow_C;
873 874 875 876 877 878 879 880
  if (!src_argb || !dst_rgb565 || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
881 882
  if (!dither4x4) {
    dither4x4 = kDither565_4x4;
883 884 885 886
  }
#if defined(HAS_ARGBTORGB565DITHERROW_SSE2)
  if (TestCpuFlag(kCpuHasSSE2)) {
    ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_SSE2;
887
    if (IS_ALIGNED(width, 4)) {
888 889
      ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_SSE2;
    }
890
  }
891 892 893 894 895 896 897 898
#endif
#if defined(HAS_ARGBTORGB565DITHERROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_AVX2;
    if (IS_ALIGNED(width, 8)) {
      ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_AVX2;
    }
  }
899 900 901 902 903 904 905 906
#endif
#if defined(HAS_ARGBTORGB565DITHERROW_NEON)
  if (TestCpuFlag(kCpuHasNEON)) {
    ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_NEON;
    }
  }
907
#endif
908 909
  for (y = 0; y < height; ++y) {
    ARGBToRGB565DitherRow(src_argb, dst_rgb565,
910
                          *(uint32*)(dither4x4 + ((y & 3) << 2)), width);
911 912 913 914 915 916
    src_argb += src_stride_argb;
    dst_rgb565 += dst_stride_rgb565;
  }
  return 0;
}

917
// Convert ARGB To RGB565.
918
// TODO(fbarchard): Consider using dither function low level with zeros.
919 920 921 922
LIBYUV_API
int ARGBToRGB565(const uint8* src_argb, int src_stride_argb,
                 uint8* dst_rgb565, int dst_stride_rgb565,
                 int width, int height) {
923
  int y;
924
  void (*ARGBToRGB565Row)(const uint8* src_argb, uint8* dst_rgb, int width) =
925
      ARGBToRGB565Row_C;
926 927 928 929 930 931 932 933
  if (!src_argb || !dst_rgb565 || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
934
  // Coalesce rows.
935 936
  if (src_stride_argb == width * 4 &&
      dst_stride_rgb565 == width * 2) {
937 938 939
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_rgb565 = 0;
fbarchard@google.com's avatar
fbarchard@google.com committed
940
  }
941
#if defined(HAS_ARGBTORGB565ROW_SSE2)
942
  if (TestCpuFlag(kCpuHasSSE2)) {
943 944 945 946 947
    ARGBToRGB565Row = ARGBToRGB565Row_Any_SSE2;
    if (IS_ALIGNED(width, 4)) {
      ARGBToRGB565Row = ARGBToRGB565Row_SSE2;
    }
  }
948
#endif
949 950 951 952 953 954 955 956
#if defined(HAS_ARGBTORGB565ROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToRGB565Row = ARGBToRGB565Row_Any_AVX2;
    if (IS_ALIGNED(width, 8)) {
      ARGBToRGB565Row = ARGBToRGB565Row_AVX2;
    }
  }
#endif
957
#if defined(HAS_ARGBTORGB565ROW_NEON)
958
  if (TestCpuFlag(kCpuHasNEON)) {
959 960 961 962 963 964 965
    ARGBToRGB565Row = ARGBToRGB565Row_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToRGB565Row = ARGBToRGB565Row_NEON;
    }
  }
#endif

966
  for (y = 0; y < height; ++y) {
967 968 969 970 971 972 973 974 975 976 977 978
    ARGBToRGB565Row(src_argb, dst_rgb565, width);
    src_argb += src_stride_argb;
    dst_rgb565 += dst_stride_rgb565;
  }
  return 0;
}

// Convert ARGB To ARGB1555.
LIBYUV_API
int ARGBToARGB1555(const uint8* src_argb, int src_stride_argb,
                   uint8* dst_argb1555, int dst_stride_argb1555,
                   int width, int height) {
979
  int y;
980
  void (*ARGBToARGB1555Row)(const uint8* src_argb, uint8* dst_rgb, int width) =
981
      ARGBToARGB1555Row_C;
982 983 984 985 986 987 988 989
  if (!src_argb || !dst_argb1555 || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
990
  // Coalesce rows.
991 992
  if (src_stride_argb == width * 4 &&
      dst_stride_argb1555 == width * 2) {
993 994 995
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_argb1555 = 0;
fbarchard@google.com's avatar
fbarchard@google.com committed
996
  }
997
#if defined(HAS_ARGBTOARGB1555ROW_SSE2)
998
  if (TestCpuFlag(kCpuHasSSE2)) {
999 1000 1001 1002 1003
    ARGBToARGB1555Row = ARGBToARGB1555Row_Any_SSE2;
    if (IS_ALIGNED(width, 4)) {
      ARGBToARGB1555Row = ARGBToARGB1555Row_SSE2;
    }
  }
1004
#endif
1005 1006 1007 1008 1009 1010 1011 1012
#if defined(HAS_ARGBTOARGB1555ROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToARGB1555Row = ARGBToARGB1555Row_Any_AVX2;
    if (IS_ALIGNED(width, 8)) {
      ARGBToARGB1555Row = ARGBToARGB1555Row_AVX2;
    }
  }
#endif
1013
#if defined(HAS_ARGBTOARGB1555ROW_NEON)
1014
  if (TestCpuFlag(kCpuHasNEON)) {
1015 1016 1017 1018 1019 1020 1021
    ARGBToARGB1555Row = ARGBToARGB1555Row_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToARGB1555Row = ARGBToARGB1555Row_NEON;
    }
  }
#endif

1022
  for (y = 0; y < height; ++y) {
1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034
    ARGBToARGB1555Row(src_argb, dst_argb1555, width);
    src_argb += src_stride_argb;
    dst_argb1555 += dst_stride_argb1555;
  }
  return 0;
}

// Convert ARGB To ARGB4444.
LIBYUV_API
int ARGBToARGB4444(const uint8* src_argb, int src_stride_argb,
                   uint8* dst_argb4444, int dst_stride_argb4444,
                   int width, int height) {
1035
  int y;
1036
  void (*ARGBToARGB4444Row)(const uint8* src_argb, uint8* dst_rgb, int width) =
1037
      ARGBToARGB4444Row_C;
1038 1039 1040 1041 1042 1043 1044 1045
  if (!src_argb || !dst_argb4444 || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
1046
  // Coalesce rows.
1047 1048
  if (src_stride_argb == width * 4 &&
      dst_stride_argb4444 == width * 2) {
1049 1050 1051
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_argb4444 = 0;
fbarchard@google.com's avatar
fbarchard@google.com committed
1052
  }
1053
#if defined(HAS_ARGBTOARGB4444ROW_SSE2)
1054
  if (TestCpuFlag(kCpuHasSSE2)) {
1055 1056 1057 1058 1059
    ARGBToARGB4444Row = ARGBToARGB4444Row_Any_SSE2;
    if (IS_ALIGNED(width, 4)) {
      ARGBToARGB4444Row = ARGBToARGB4444Row_SSE2;
    }
  }
1060
#endif
1061 1062 1063 1064 1065 1066 1067 1068
#if defined(HAS_ARGBTOARGB4444ROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToARGB4444Row = ARGBToARGB4444Row_Any_AVX2;
    if (IS_ALIGNED(width, 8)) {
      ARGBToARGB4444Row = ARGBToARGB4444Row_AVX2;
    }
  }
#endif
1069
#if defined(HAS_ARGBTOARGB4444ROW_NEON)
1070
  if (TestCpuFlag(kCpuHasNEON)) {
1071 1072 1073 1074 1075 1076 1077
    ARGBToARGB4444Row = ARGBToARGB4444Row_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToARGB4444Row = ARGBToARGB4444Row_NEON;
    }
  }
#endif

1078
  for (y = 0; y < height; ++y) {
1079 1080 1081 1082 1083 1084 1085
    ARGBToARGB4444Row(src_argb, dst_argb4444, width);
    src_argb += src_stride_argb;
    dst_argb4444 += dst_stride_argb4444;
  }
  return 0;
}

1086 1087 1088 1089 1090 1091 1092
// Convert ARGB to J420. (JPeg full range I420).
LIBYUV_API
int ARGBToJ420(const uint8* src_argb, int src_stride_argb,
               uint8* dst_yj, int dst_stride_yj,
               uint8* dst_u, int dst_stride_u,
               uint8* dst_v, int dst_stride_v,
               int width, int height) {
1093 1094
  int y;
  void (*ARGBToUVJRow)(const uint8* src_argb0, int src_stride_argb,
1095
                       uint8* dst_u, uint8* dst_v, int width) = ARGBToUVJRow_C;
1096
  void (*ARGBToYJRow)(const uint8* src_argb, uint8* dst_yj, int width) =
1097
      ARGBToYJRow_C;
1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
  if (!src_argb ||
      !dst_yj || !dst_u || !dst_v ||
      width <= 0 || height == 0) {
    return -1;
  }
  // Negative height means invert the image.
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
1109
#if defined(HAS_ARGBTOYJROW_SSSE3) && defined(HAS_ARGBTOUVJROW_SSSE3)
1110
  if (TestCpuFlag(kCpuHasSSSE3)) {
1111
    ARGBToUVJRow = ARGBToUVJRow_Any_SSSE3;
1112 1113
    ARGBToYJRow = ARGBToYJRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
1114 1115
      ARGBToUVJRow = ARGBToUVJRow_SSSE3;
      ARGBToYJRow = ARGBToYJRow_SSSE3;
1116 1117 1118
    }
  }
#endif
1119
#if defined(HAS_ARGBTOYJROW_AVX2)
1120
  if (TestCpuFlag(kCpuHasAVX2)) {
1121 1122 1123 1124 1125 1126
    ARGBToYJRow = ARGBToYJRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYJRow = ARGBToYJRow_AVX2;
    }
  }
#endif
1127
#if defined(HAS_ARGBTOYJROW_NEON)
1128
  if (TestCpuFlag(kCpuHasNEON)) {
1129 1130 1131 1132
    ARGBToYJRow = ARGBToYJRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYJRow = ARGBToYJRow_NEON;
    }
1133 1134 1135
  }
#endif
#if defined(HAS_ARGBTOUVJROW_NEON)
1136
  if (TestCpuFlag(kCpuHasNEON)) {
1137 1138 1139
    ARGBToUVJRow = ARGBToUVJRow_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUVJRow = ARGBToUVJRow_NEON;
1140 1141 1142 1143
    }
  }
#endif

1144
  for (y = 0; y < height - 1; y += 2) {
1145
    ARGBToUVJRow(src_argb, src_stride_argb, dst_u, dst_v, width);
1146 1147 1148 1149 1150 1151 1152 1153
    ARGBToYJRow(src_argb, dst_yj, width);
    ARGBToYJRow(src_argb + src_stride_argb, dst_yj + dst_stride_yj, width);
    src_argb += src_stride_argb * 2;
    dst_yj += dst_stride_yj * 2;
    dst_u += dst_stride_u;
    dst_v += dst_stride_v;
  }
  if (height & 1) {
1154
    ARGBToUVJRow(src_argb, 0, dst_u, dst_v, width);
1155 1156 1157 1158 1159
    ARGBToYJRow(src_argb, dst_yj, width);
  }
  return 0;
}

1160 1161 1162 1163 1164 1165 1166 1167 1168
// ARGB little endian (bgra in memory) to J422
LIBYUV_API
int ARGBToJ422(const uint8* src_argb, int src_stride_argb,
               uint8* dst_y, int dst_stride_y,
               uint8* dst_u, int dst_stride_u,
               uint8* dst_v, int dst_stride_v,
               int width, int height) {
  int y;
  void (*ARGBToUVJ422Row)(const uint8* src_argb, uint8* dst_u, uint8* dst_v,
1169 1170
      int width) = ARGBToUVJ422Row_C;
  void (*ARGBToYJRow)(const uint8* src_argb, uint8* dst_y, int width) =
1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213
      ARGBToYJRow_C;
  if (!src_argb || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
  // Coalesce rows.
  if (src_stride_argb == width * 4 &&
      dst_stride_y == width &&
      dst_stride_u * 2 == width &&
      dst_stride_v * 2 == width) {
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_y = dst_stride_u = dst_stride_v = 0;
  }
#if defined(HAS_ARGBTOUVJ422ROW_SSSE3)
  if (TestCpuFlag(kCpuHasSSSE3)) {
    ARGBToUVJ422Row = ARGBToUVJ422Row_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUVJ422Row = ARGBToUVJ422Row_SSSE3;
    }
  }
#endif
#if defined(HAS_ARGBTOUVJ422ROW_NEON)
  if (TestCpuFlag(kCpuHasNEON)) {
    ARGBToUVJ422Row = ARGBToUVJ422Row_Any_NEON;
    if (IS_ALIGNED(width, 16)) {
      ARGBToUVJ422Row = ARGBToUVJ422Row_NEON;
    }
  }
#endif

#if defined(HAS_ARGBTOYJROW_SSSE3)
  if (TestCpuFlag(kCpuHasSSSE3)) {
    ARGBToYJRow = ARGBToYJRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
      ARGBToYJRow = ARGBToYJRow_SSSE3;
    }
  }
#endif
1214 1215 1216 1217 1218 1219 1220 1221
#if defined(HAS_ARGBTOYJROW_AVX2)
  if (TestCpuFlag(kCpuHasAVX2)) {
    ARGBToYJRow = ARGBToYJRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYJRow = ARGBToYJRow_AVX2;
    }
  }
#endif
1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241
#if defined(HAS_ARGBTOYJROW_NEON)
  if (TestCpuFlag(kCpuHasNEON)) {
    ARGBToYJRow = ARGBToYJRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYJRow = ARGBToYJRow_NEON;
    }
  }
#endif

  for (y = 0; y < height; ++y) {
    ARGBToUVJ422Row(src_argb, dst_u, dst_v, width);
    ARGBToYJRow(src_argb, dst_y, width);
    src_argb += src_stride_argb;
    dst_y += dst_stride_y;
    dst_u += dst_stride_u;
    dst_v += dst_stride_v;
  }
  return 0;
}

1242 1243 1244 1245 1246
// Convert ARGB to J400.
LIBYUV_API
int ARGBToJ400(const uint8* src_argb, int src_stride_argb,
               uint8* dst_yj, int dst_stride_yj,
               int width, int height) {
1247
  int y;
1248
  void (*ARGBToYJRow)(const uint8* src_argb, uint8* dst_yj, int width) =
1249
      ARGBToYJRow_C;
1250 1251 1252 1253 1254 1255 1256 1257
  if (!src_argb || !dst_yj || width <= 0 || height == 0) {
    return -1;
  }
  if (height < 0) {
    height = -height;
    src_argb = src_argb + (height - 1) * src_stride_argb;
    src_stride_argb = -src_stride_argb;
  }
1258
  // Coalesce rows.
1259 1260
  if (src_stride_argb == width * 4 &&
      dst_stride_yj == width) {
1261 1262 1263
    width *= height;
    height = 1;
    src_stride_argb = dst_stride_yj = 0;
1264 1265
  }
#if defined(HAS_ARGBTOYJROW_SSSE3)
1266
  if (TestCpuFlag(kCpuHasSSSE3)) {
1267 1268
    ARGBToYJRow = ARGBToYJRow_Any_SSSE3;
    if (IS_ALIGNED(width, 16)) {
1269
      ARGBToYJRow = ARGBToYJRow_SSSE3;
1270 1271 1272
    }
  }
#endif
1273
#if defined(HAS_ARGBTOYJROW_AVX2)
1274
  if (TestCpuFlag(kCpuHasAVX2)) {
1275 1276 1277 1278 1279 1280
    ARGBToYJRow = ARGBToYJRow_Any_AVX2;
    if (IS_ALIGNED(width, 32)) {
      ARGBToYJRow = ARGBToYJRow_AVX2;
    }
  }
#endif
1281
#if defined(HAS_ARGBTOYJROW_NEON)
1282
  if (TestCpuFlag(kCpuHasNEON)) {
1283 1284 1285 1286 1287 1288 1289
    ARGBToYJRow = ARGBToYJRow_Any_NEON;
    if (IS_ALIGNED(width, 8)) {
      ARGBToYJRow = ARGBToYJRow_NEON;
    }
  }
#endif

1290
  for (y = 0; y < height; ++y) {
1291 1292 1293 1294 1295 1296 1297
    ARGBToYJRow(src_argb, dst_yj, width);
    src_argb += src_stride_argb;
    dst_yj += dst_stride_yj;
  }
  return 0;
}

1298 1299 1300 1301
#ifdef __cplusplus
}  // extern "C"
}  // namespace libyuv
#endif