| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright 2011 The LibYuv Project Authors. All rights reserved. | 2 * Copyright 2011 The LibYuv Project Authors. All rights reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| (...skipping 210 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 221 src += src_stride_0 + src_stride_1; | 221 src += src_stride_0 + src_stride_1; |
| 222 dst += dst_stride * 2; | 222 dst += dst_stride * 2; |
| 223 } | 223 } |
| 224 if (height & 1) { | 224 if (height & 1) { |
| 225 CopyRow(src, dst, width); | 225 CopyRow(src, dst, width); |
| 226 } | 226 } |
| 227 } | 227 } |
| 228 | 228 |
| 229 // Support function for NV12 etc UV channels. | 229 // Support function for NV12 etc UV channels. |
| 230 // Width and height are plane sizes (typically half pixel width). | 230 // Width and height are plane sizes (typically half pixel width). |
| 231 static void SplitPlane(const uint8* src_uv, int src_stride_uv, | 231 static void SplitUVPlane(const uint8* src_uv, int src_stride_uv, |
| 232 uint8* dst_u, int dst_stride_u, | 232 uint8* dst_u, int dst_stride_u, |
| 233 uint8* dst_v, int dst_stride_v, | 233 uint8* dst_v, int dst_stride_v, |
| 234 int width, int height) { | 234 int width, int height) { |
| 235 int y; | 235 int y; |
| 236 void (*SplitUVRow)(const uint8* src_uv, uint8* dst_u, uint8* dst_v, | 236 void (*SplitUVRow)(const uint8* src_uv, uint8* dst_u, uint8* dst_v, |
| 237 int width) = SplitUVRow_C; | 237 int width) = SplitUVRow_C; |
| 238 // Negative height means invert the image. | 238 // Negative height means invert the image. |
| 239 if (height < 0) { | 239 if (height < 0) { |
| 240 height = -height; | 240 height = -height; |
| 241 dst_u = dst_u + (height - 1) * dst_stride_u; | 241 dst_u = dst_u + (height - 1) * dst_stride_u; |
| 242 dst_v = dst_v + (height - 1) * dst_stride_v; | 242 dst_v = dst_v + (height - 1) * dst_stride_v; |
| 243 dst_stride_u = -dst_stride_u; | 243 dst_stride_u = -dst_stride_u; |
| 244 dst_stride_v = -dst_stride_v; | 244 dst_stride_v = -dst_stride_v; |
| (...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 304 // src_stride_m420 is row planar. Normally this will be the width in pixels. | 304 // src_stride_m420 is row planar. Normally this will be the width in pixels. |
| 305 // The UV plane is half width, but 2 values, so src_stride_m420 applies to | 305 // The UV plane is half width, but 2 values, so src_stride_m420 applies to |
| 306 // this as well as the two Y planes. | 306 // this as well as the two Y planes. |
| 307 static int X420ToI420(const uint8* src_y, | 307 static int X420ToI420(const uint8* src_y, |
| 308 int src_stride_y0, int src_stride_y1, | 308 int src_stride_y0, int src_stride_y1, |
| 309 const uint8* src_uv, int src_stride_uv, | 309 const uint8* src_uv, int src_stride_uv, |
| 310 uint8* dst_y, int dst_stride_y, | 310 uint8* dst_y, int dst_stride_y, |
| 311 uint8* dst_u, int dst_stride_u, | 311 uint8* dst_u, int dst_stride_u, |
| 312 uint8* dst_v, int dst_stride_v, | 312 uint8* dst_v, int dst_stride_v, |
| 313 int width, int height) { | 313 int width, int height) { |
| 314 int y; | |
| 315 int halfwidth = (width + 1) >> 1; | 314 int halfwidth = (width + 1) >> 1; |
| 316 int halfheight = (height + 1) >> 1; | 315 int halfheight = (height + 1) >> 1; |
| 317 void (*SplitUVRow)(const uint8* src_uv, uint8* dst_u, uint8* dst_v, | |
| 318 int width) = SplitUVRow_C; | |
| 319 if (!src_y || !src_uv || | 316 if (!src_y || !src_uv || |
| 320 !dst_y || !dst_u || !dst_v || | 317 !dst_y || !dst_u || !dst_v || |
| 321 width <= 0 || height == 0) { | 318 width <= 0 || height == 0) { |
| 322 return -1; | 319 return -1; |
| 323 } | 320 } |
| 324 // Negative height means invert the image. | 321 // Negative height means invert the image. |
| 325 if (height < 0) { | 322 if (height < 0) { |
| 326 height = -height; | 323 height = -height; |
| 327 halfheight = (height + 1) >> 1; | 324 halfheight = (height + 1) >> 1; |
| 328 dst_y = dst_y + (height - 1) * dst_stride_y; | 325 dst_y = dst_y + (height - 1) * dst_stride_y; |
| (...skipping 12 matching lines...) Expand all Loading... |
| 341 src_stride_y0 = src_stride_y1 = dst_stride_y = 0; | 338 src_stride_y0 = src_stride_y1 = dst_stride_y = 0; |
| 342 } | 339 } |
| 343 // Coalesce rows. | 340 // Coalesce rows. |
| 344 if (src_stride_uv == halfwidth * 2 && | 341 if (src_stride_uv == halfwidth * 2 && |
| 345 dst_stride_u == halfwidth && | 342 dst_stride_u == halfwidth && |
| 346 dst_stride_v == halfwidth) { | 343 dst_stride_v == halfwidth) { |
| 347 halfwidth *= halfheight; | 344 halfwidth *= halfheight; |
| 348 halfheight = 1; | 345 halfheight = 1; |
| 349 src_stride_uv = dst_stride_u = dst_stride_v = 0; | 346 src_stride_uv = dst_stride_u = dst_stride_v = 0; |
| 350 } | 347 } |
| 351 #if defined(HAS_SPLITUVROW_SSE2) | |
| 352 if (TestCpuFlag(kCpuHasSSE2)) { | |
| 353 SplitUVRow = SplitUVRow_Any_SSE2; | |
| 354 if (IS_ALIGNED(halfwidth, 16)) { | |
| 355 SplitUVRow = SplitUVRow_SSE2; | |
| 356 } | |
| 357 } | |
| 358 #endif | |
| 359 #if defined(HAS_SPLITUVROW_AVX2) | |
| 360 if (TestCpuFlag(kCpuHasAVX2)) { | |
| 361 SplitUVRow = SplitUVRow_Any_AVX2; | |
| 362 if (IS_ALIGNED(halfwidth, 32)) { | |
| 363 SplitUVRow = SplitUVRow_AVX2; | |
| 364 } | |
| 365 } | |
| 366 #endif | |
| 367 #if defined(HAS_SPLITUVROW_NEON) | |
| 368 if (TestCpuFlag(kCpuHasNEON)) { | |
| 369 SplitUVRow = SplitUVRow_Any_NEON; | |
| 370 if (IS_ALIGNED(halfwidth, 16)) { | |
| 371 SplitUVRow = SplitUVRow_NEON; | |
| 372 } | |
| 373 } | |
| 374 #endif | |
| 375 #if defined(HAS_SPLITUVROW_DSPR2) | |
| 376 if (TestCpuFlag(kCpuHasDSPR2) && | |
| 377 IS_ALIGNED(src_uv, 4) && IS_ALIGNED(src_stride_uv, 4) && | |
| 378 IS_ALIGNED(dst_u, 4) && IS_ALIGNED(dst_stride_u, 4) && | |
| 379 IS_ALIGNED(dst_v, 4) && IS_ALIGNED(dst_stride_v, 4)) { | |
| 380 SplitUVRow = SplitUVRow_Any_DSPR2; | |
| 381 if (IS_ALIGNED(halfwidth, 16)) { | |
| 382 SplitUVRow = SplitUVRow_DSPR2; | |
| 383 } | |
| 384 } | |
| 385 #endif | |
| 386 | 348 |
| 387 if (dst_y) { | 349 if (dst_y) { |
| 388 if (src_stride_y0 == src_stride_y1) { | 350 if (src_stride_y0 == src_stride_y1) { |
| 389 CopyPlane(src_y, src_stride_y0, dst_y, dst_stride_y, width, height); | 351 CopyPlane(src_y, src_stride_y0, dst_y, dst_stride_y, width, height); |
| 390 } else { | 352 } else { |
| 391 CopyPlane2(src_y, src_stride_y0, src_stride_y1, dst_y, dst_stride_y, | 353 CopyPlane2(src_y, src_stride_y0, src_stride_y1, dst_y, dst_stride_y, |
| 392 width, height); | 354 width, height); |
| 393 } | 355 } |
| 394 } | 356 } |
| 395 | 357 |
| 396 for (y = 0; y < halfheight; ++y) { | 358 // Split UV plane - NV12 / NV21 |
| 397 // Copy a row of UV. | 359 SplitUVPlane(src_uv, src_stride_uv, dst_u, dst_stride_u, dst_v, dst_stride_v, |
| 398 SplitUVRow(src_uv, dst_u, dst_v, halfwidth); | 360 halfwidth, halfheight); |
| 399 dst_u += dst_stride_u; | 361 |
| 400 dst_v += dst_stride_v; | |
| 401 src_uv += src_stride_uv; | |
| 402 } | |
| 403 return 0; | 362 return 0; |
| 404 } | 363 } |
| 405 | 364 |
| 406 // Convert NV12 to I420. | 365 // Convert NV12 to I420. |
| 407 LIBYUV_API | 366 LIBYUV_API |
| 408 int NV12ToI420(const uint8* src_y, int src_stride_y, | 367 int NV12ToI420(const uint8* src_y, int src_stride_y, |
| 409 const uint8* src_uv, int src_stride_uv, | 368 const uint8* src_uv, int src_stride_uv, |
| 410 uint8* dst_y, int dst_stride_y, | 369 uint8* dst_y, int dst_stride_y, |
| 411 uint8* dst_u, int dst_stride_u, | 370 uint8* dst_u, int dst_stride_u, |
| 412 uint8* dst_v, int dst_stride_v, | 371 uint8* dst_v, int dst_stride_v, |
| (...skipping 1085 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1498 } | 1457 } |
| 1499 | 1458 |
| 1500 // Copy UV planes as is - I420 | 1459 // Copy UV planes as is - I420 |
| 1501 if (src_pixel_stride_uv == 1) { | 1460 if (src_pixel_stride_uv == 1) { |
| 1502 CopyPlane(src_u, src_stride_u, dst_u, dst_stride_u, halfwidth, halfheight); | 1461 CopyPlane(src_u, src_stride_u, dst_u, dst_stride_u, halfwidth, halfheight); |
| 1503 CopyPlane(src_v, src_stride_v, dst_v, dst_stride_v, halfwidth, halfheight); | 1462 CopyPlane(src_v, src_stride_v, dst_v, dst_stride_v, halfwidth, halfheight); |
| 1504 return 0; | 1463 return 0; |
| 1505 // Split UV planes - NV21 | 1464 // Split UV planes - NV21 |
| 1506 } else if (src_pixel_stride_uv == 2 && vu_off == -1 && | 1465 } else if (src_pixel_stride_uv == 2 && vu_off == -1 && |
| 1507 src_stride_u == src_stride_v) { | 1466 src_stride_u == src_stride_v) { |
| 1508 SplitPlane(src_v, src_stride_v, dst_v, dst_stride_v, dst_u, dst_stride_u, | 1467 SplitUVPlane(src_v, src_stride_v, dst_v, dst_stride_v, dst_u, dst_stride_u, |
| 1509 halfwidth, halfheight); | 1468 halfwidth, halfheight); |
| 1510 return 0; | 1469 return 0; |
| 1511 // Split UV planes - NV12 | 1470 // Split UV planes - NV12 |
| 1512 } else if (src_pixel_stride_uv == 2 && vu_off == 1 && | 1471 } else if (src_pixel_stride_uv == 2 && vu_off == 1 && |
| 1513 src_stride_u == src_stride_v) { | 1472 src_stride_u == src_stride_v) { |
| 1514 SplitPlane(src_u, src_stride_u, dst_u, dst_stride_u, dst_v, dst_stride_v, | 1473 SplitUVPlane(src_u, src_stride_u, dst_u, dst_stride_u, dst_v, dst_stride_v, |
| 1515 halfwidth, halfheight); | 1474 halfwidth, halfheight); |
| 1516 return 0; | 1475 return 0; |
| 1517 } | 1476 } |
| 1518 | 1477 |
| 1519 for (y = 0; y < halfheight; ++y) { | 1478 for (y = 0; y < halfheight; ++y) { |
| 1520 SplitPixels(src_u, src_pixel_stride_uv, dst_u, halfwidth); | 1479 SplitPixels(src_u, src_pixel_stride_uv, dst_u, halfwidth); |
| 1521 SplitPixels(src_v, src_pixel_stride_uv, dst_v, halfwidth); | 1480 SplitPixels(src_v, src_pixel_stride_uv, dst_v, halfwidth); |
| 1522 src_u += src_stride_u; | 1481 src_u += src_stride_u; |
| 1523 src_v += src_stride_v; | 1482 src_v += src_stride_v; |
| 1524 dst_u += dst_stride_u; | 1483 dst_u += dst_stride_u; |
| 1525 dst_v += dst_stride_v; | 1484 dst_v += dst_stride_v; |
| 1526 } | 1485 } |
| 1527 return 0; | 1486 return 0; |
| 1528 } | 1487 } |
| 1529 | 1488 |
| 1530 #ifdef __cplusplus | 1489 #ifdef __cplusplus |
| 1531 } // extern "C" | 1490 } // extern "C" |
| 1532 } // namespace libyuv | 1491 } // namespace libyuv |
| 1533 #endif | 1492 #endif |
| OLD | NEW |