YUVTest.cpp 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. /*
  2. * Copyright 2013 Google Inc.
  3. *
  4. * Use of this source code is governed by a BSD-style license that can be
  5. * found in the LICENSE file.
  6. */
  7. #include "include/codec/SkCodec.h"
  8. #include "include/core/SkStream.h"
  9. #include "include/core/SkYUVASizeInfo.h"
  10. #include "include/private/SkTemplates.h"
  11. #include "src/core/SkAutoMalloc.h"
  12. #include "tests/Test.h"
  13. #include "tools/Resources.h"
  14. static void codec_yuv(skiatest::Reporter* reporter,
  15. const char path[],
  16. SkISize expectedSizes[4]) {
  17. std::unique_ptr<SkStream> stream(GetResourceAsStream(path));
  18. if (!stream) {
  19. return;
  20. }
  21. std::unique_ptr<SkCodec> codec(SkCodec::MakeFromStream(std::move(stream)));
  22. REPORTER_ASSERT(reporter, codec);
  23. if (!codec) {
  24. return;
  25. }
  26. // Test queryYUV8()
  27. SkYUVASizeInfo info;
  28. {
  29. bool success = codec->queryYUV8(nullptr, nullptr);
  30. REPORTER_ASSERT(reporter, !success);
  31. success = codec->queryYUV8(&info, nullptr);
  32. REPORTER_ASSERT(reporter, (expectedSizes == nullptr) == !success);
  33. if (!success) {
  34. return;
  35. }
  36. for (int i = 0; i < SkYUVASizeInfo::kMaxCount; ++i) {
  37. REPORTER_ASSERT(reporter, info.fSizes[i] == expectedSizes[i]);
  38. REPORTER_ASSERT(reporter,
  39. info.fWidthBytes[i] == (uint32_t) SkAlign8(info.fSizes[i].width()));
  40. }
  41. }
  42. {
  43. SkYUVColorSpace colorSpace;
  44. bool success = codec->queryYUV8(&info, &colorSpace);
  45. REPORTER_ASSERT(reporter, (expectedSizes == nullptr) == !success);
  46. if (!success) {
  47. return;
  48. }
  49. for (int i = 0; i < SkYUVASizeInfo::kMaxCount; ++i) {
  50. REPORTER_ASSERT(reporter, info.fSizes[i] == expectedSizes[i]);
  51. REPORTER_ASSERT(reporter,
  52. info.fWidthBytes[i] == (uint32_t) SkAlign8(info.fSizes[i].width()));
  53. }
  54. REPORTER_ASSERT(reporter, kJPEG_SkYUVColorSpace == colorSpace);
  55. }
  56. // Allocate the memory for the YUV decode
  57. size_t totalBytes = info.computeTotalBytes();
  58. SkAutoMalloc storage(totalBytes);
  59. void* planes[SkYUVASizeInfo::kMaxCount];
  60. info.computePlanes(storage.get(), planes);
  61. // Test getYUV8Planes()
  62. REPORTER_ASSERT(reporter, SkCodec::kInvalidInput == codec->getYUV8Planes(info, nullptr));
  63. REPORTER_ASSERT(reporter, SkCodec::kSuccess == codec->getYUV8Planes(info, planes));
  64. }
  65. DEF_TEST(Jpeg_YUV_Codec, r) {
  66. SkISize sizes[4];
  67. sizes[0].set(128, 128);
  68. sizes[1].set(64, 64);
  69. sizes[2].set(64, 64);
  70. sizes[3].set(0, 0);
  71. codec_yuv(r, "images/color_wheel.jpg", sizes);
  72. // H2V2
  73. sizes[0].set(512, 512);
  74. sizes[1].set(256, 256);
  75. sizes[2].set(256, 256);
  76. codec_yuv(r, "images/mandrill_512_q075.jpg", sizes);
  77. // H1V1
  78. sizes[1].set(512, 512);
  79. sizes[2].set(512, 512);
  80. codec_yuv(r, "images/mandrill_h1v1.jpg", sizes);
  81. // H2V1
  82. sizes[1].set(256, 512);
  83. sizes[2].set(256, 512);
  84. codec_yuv(r, "images/mandrill_h2v1.jpg", sizes);
  85. // Non-power of two dimensions
  86. sizes[0].set(439, 154);
  87. sizes[1].set(220, 77);
  88. sizes[2].set(220, 77);
  89. codec_yuv(r, "images/cropped_mandrill.jpg", sizes);
  90. sizes[0].set(8, 8);
  91. sizes[1].set(4, 4);
  92. sizes[2].set(4, 4);
  93. codec_yuv(r, "images/randPixels.jpg", sizes);
  94. // Progressive images
  95. sizes[0].set(512, 512);
  96. sizes[1].set(512, 512);
  97. sizes[2].set(512, 512);
  98. codec_yuv(r, "images/brickwork-texture.jpg", sizes);
  99. codec_yuv(r, "images/brickwork_normal-map.jpg", sizes);
  100. // A CMYK encoded image should fail.
  101. codec_yuv(r, "images/CMYK.jpg", nullptr);
  102. // A grayscale encoded image should fail.
  103. codec_yuv(r, "images/grayscale.jpg", nullptr);
  104. // A PNG should fail.
  105. codec_yuv(r, "images/arrow.png", nullptr);
  106. }
  107. #include "include/effects/SkColorMatrix.h"
  108. #include "src/core/SkYUVMath.h"
  109. // Be sure that the two matrices are inverses of each other
  110. // (i.e. rgb2yuv and yuv2rgb
  111. DEF_TEST(YUVMath, reporter) {
  112. const SkYUVColorSpace spaces[] = {
  113. kJPEG_SkYUVColorSpace,
  114. kRec601_SkYUVColorSpace,
  115. kRec709_SkYUVColorSpace,
  116. kIdentity_SkYUVColorSpace,
  117. };
  118. // Not sure what the theoretical precision we can hope for is, so pick a big value that
  119. // passes (when I think we're correct).
  120. const float tolerance = 1.0f/(1 << 18);
  121. for (auto cs : spaces) {
  122. float r2y[20], y2r[20];
  123. SkColorMatrix_RGB2YUV(cs, r2y);
  124. SkColorMatrix_YUV2RGB(cs, y2r);
  125. SkColorMatrix r2ym, y2rm;
  126. r2ym.setRowMajor(r2y);
  127. y2rm.setRowMajor(y2r);
  128. r2ym.postConcat(y2rm);
  129. float tmp[20];
  130. r2ym.getRowMajor(tmp);
  131. for (int i = 0; i < 20; ++i) {
  132. float expected = 0;
  133. if (i % 6 == 0) { // diagonal
  134. expected = 1;
  135. }
  136. REPORTER_ASSERT(reporter, SkScalarNearlyEqual(tmp[i], expected, tolerance));
  137. }
  138. }
  139. }