paint_image_unittest.cc 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "cc/paint/paint_image.h"
  5. #include <utility>
  6. #include "base/test/gtest_util.h"
  7. #include "cc/paint/paint_image_builder.h"
  8. #include "cc/test/fake_paint_image_generator.h"
  9. #include "cc/test/skia_common.h"
  10. #include "cc/test/test_paint_worklet_input.h"
  11. #include "testing/gtest/include/gtest/gtest.h"
  12. #include "third_party/skia/include/core/SkColorSpace.h"
  13. namespace cc {
  14. TEST(PaintImageTest, DecodesCorrectFrames) {
  15. std::vector<FrameMetadata> frames = {
  16. FrameMetadata(true, base::Milliseconds(2)),
  17. FrameMetadata(true, base::Milliseconds(3))};
  18. sk_sp<FakePaintImageGenerator> generator =
  19. sk_make_sp<FakePaintImageGenerator>(SkImageInfo::MakeN32Premul(10, 10),
  20. frames);
  21. PaintImage image = PaintImageBuilder::WithDefault()
  22. .set_id(PaintImage::GetNextId())
  23. .set_paint_image_generator(generator)
  24. .TakePaintImage();
  25. // When there's no decoded SkImage the color usage defaults to SRGB.
  26. EXPECT_EQ(image.GetContentColorUsage(), gfx::ContentColorUsage::kSRGB);
  27. // The recorded index is 0u but ask for 1u frame.
  28. SkImageInfo info = SkImageInfo::MakeN32Premul(10, 10);
  29. std::vector<size_t> memory(info.computeMinByteSize());
  30. image.Decode(memory.data(), &info, nullptr, 1u,
  31. PaintImage::GetNextGeneratorClientId());
  32. ASSERT_EQ(generator->frames_decoded().size(), 1u);
  33. EXPECT_EQ(generator->frames_decoded().count(1u), 1u);
  34. generator->reset_frames_decoded();
  35. // Not N32 color type.
  36. info.makeColorType(kRGB_565_SkColorType);
  37. memory = std::vector<size_t>(info.computeMinByteSize());
  38. image.Decode(memory.data(), &info, nullptr, 1u,
  39. PaintImage::GetNextGeneratorClientId());
  40. ASSERT_EQ(generator->frames_decoded().size(), 1u);
  41. EXPECT_EQ(generator->frames_decoded().count(1u), 1u);
  42. generator->reset_frames_decoded();
  43. }
  44. TEST(PaintImageTest, SupportedDecodeSize) {
  45. SkISize full_size = SkISize::Make(10, 10);
  46. std::vector<SkISize> supported_sizes = {SkISize::Make(5, 5)};
  47. std::vector<FrameMetadata> frames = {FrameMetadata()};
  48. sk_sp<FakePaintImageGenerator> generator =
  49. sk_make_sp<FakePaintImageGenerator>(
  50. SkImageInfo::MakeN32Premul(full_size.width(), full_size.height()),
  51. frames, true, supported_sizes);
  52. PaintImage image = PaintImageBuilder::WithDefault()
  53. .set_id(PaintImage::GetNextId())
  54. .set_paint_image_generator(generator)
  55. .TakePaintImage();
  56. EXPECT_EQ(image.GetSupportedDecodeSize(supported_sizes[0]),
  57. supported_sizes[0]);
  58. }
  59. TEST(PaintImageTest, GetSkImageForFrameNotGeneratorBacked) {
  60. PaintImage image = CreateBitmapImage(gfx::Size(10, 10));
  61. EXPECT_EQ(image.GetSwSkImage(),
  62. image.GetSkImageForFrame(PaintImage::kDefaultFrameIndex,
  63. PaintImage::GetNextGeneratorClientId()));
  64. }
  65. TEST(PaintImageTest, DecodeToYuv420NoAlpha) {
  66. const SkISize full_size = SkISize::Make(10, 10);
  67. SkYUVAInfo yuva_info(full_size, SkYUVAInfo::PlaneConfig::kY_U_V,
  68. SkYUVAInfo::Subsampling::k420,
  69. kJPEG_Full_SkYUVColorSpace);
  70. SkYUVAPixmapInfo yuva_pixmap_info(yuva_info,
  71. SkYUVAPixmapInfo::DataType::kUnorm8,
  72. /*row bytes*/ nullptr);
  73. sk_sp<FakePaintImageGenerator> yuv_generator =
  74. sk_make_sp<FakePaintImageGenerator>(SkImageInfo::MakeN32Premul(full_size),
  75. yuva_pixmap_info);
  76. PaintImage image = PaintImageBuilder::WithDefault()
  77. .set_id(PaintImage::GetNextId())
  78. .set_paint_image_generator(yuv_generator)
  79. .TakePaintImage();
  80. std::vector<uint8_t> memory(yuva_pixmap_info.computeTotalBytes());
  81. auto pixmaps =
  82. SkYUVAPixmaps::FromExternalMemory(yuva_pixmap_info, memory.data());
  83. SkYUVAPixmapInfo image_yuva_pixmap_info;
  84. ASSERT_TRUE(image.IsYuv(SkYUVAPixmapInfo::SupportedDataTypes::All(),
  85. &image_yuva_pixmap_info));
  86. ASSERT_EQ(yuva_pixmap_info, image_yuva_pixmap_info);
  87. image.DecodeYuv(pixmaps, 1u /* frame_index */,
  88. PaintImage::GetNextGeneratorClientId());
  89. ASSERT_EQ(yuv_generator->frames_decoded().size(), 1u);
  90. EXPECT_EQ(yuv_generator->frames_decoded().count(1u), 1u);
  91. yuv_generator->reset_frames_decoded();
  92. }
  93. TEST(PaintImageTest, BuildPaintWorkletImage) {
  94. gfx::SizeF size(100, 50);
  95. scoped_refptr<TestPaintWorkletInput> input =
  96. base::MakeRefCounted<TestPaintWorkletInput>(size);
  97. PaintImage paint_image = PaintImageBuilder::WithDefault()
  98. .set_id(1)
  99. .set_paint_worklet_input(std::move(input))
  100. .TakePaintImage();
  101. EXPECT_TRUE(paint_image.paint_worklet_input());
  102. EXPECT_EQ(paint_image.width(), size.width());
  103. EXPECT_EQ(paint_image.height(), size.height());
  104. EXPECT_EQ(paint_image.GetContentColorUsage(), gfx::ContentColorUsage::kSRGB);
  105. }
  106. TEST(PaintImageTest, SrgbImage) {
  107. auto generator = sk_make_sp<FakePaintImageGenerator>(
  108. SkImageInfo::Make(10, 10, kRGBA_F16_SkColorType, kUnknown_SkAlphaType,
  109. gfx::ColorSpace::CreateSRGB().ToSkColorSpace()));
  110. PaintImage image = PaintImageBuilder::WithDefault()
  111. .set_id(PaintImage::GetNextId())
  112. .set_paint_image_generator(generator)
  113. .set_is_high_bit_depth(true)
  114. .TakePaintImage();
  115. EXPECT_TRUE(image.is_high_bit_depth());
  116. EXPECT_EQ(image.GetContentColorUsage(), gfx::ContentColorUsage::kSRGB);
  117. }
  118. TEST(PaintImageTest, HbdImage) {
  119. auto generator = sk_make_sp<FakePaintImageGenerator>(SkImageInfo::Make(
  120. 10, 10, kRGBA_F16_SkColorType, kUnknown_SkAlphaType,
  121. gfx::ColorSpace::CreateDisplayP3D65().ToSkColorSpace()));
  122. PaintImage image = PaintImageBuilder::WithDefault()
  123. .set_id(PaintImage::GetNextId())
  124. .set_paint_image_generator(generator)
  125. .set_is_high_bit_depth(true)
  126. .TakePaintImage();
  127. EXPECT_TRUE(image.is_high_bit_depth());
  128. EXPECT_EQ(image.GetContentColorUsage(),
  129. gfx::ContentColorUsage::kWideColorGamut);
  130. }
  131. TEST(PaintImageTest, PqHdrImage) {
  132. auto generator = sk_make_sp<FakePaintImageGenerator>(
  133. SkImageInfo::Make(10, 10, kRGBA_F16_SkColorType, kUnknown_SkAlphaType,
  134. gfx::ColorSpace::CreateHDR10().ToSkColorSpace()));
  135. PaintImage image = PaintImageBuilder::WithDefault()
  136. .set_id(PaintImage::GetNextId())
  137. .set_paint_image_generator(generator)
  138. .set_is_high_bit_depth(true)
  139. .TakePaintImage();
  140. EXPECT_TRUE(image.is_high_bit_depth());
  141. EXPECT_EQ(image.GetContentColorUsage(), gfx::ContentColorUsage::kHDR);
  142. }
  143. TEST(PaintImageTest, HlgHdrImage) {
  144. auto generator = sk_make_sp<FakePaintImageGenerator>(
  145. SkImageInfo::Make(10, 10, kRGBA_F16_SkColorType, kUnknown_SkAlphaType,
  146. gfx::ColorSpace::CreateHLG().ToSkColorSpace()));
  147. PaintImage image = PaintImageBuilder::WithDefault()
  148. .set_id(PaintImage::GetNextId())
  149. .set_paint_image_generator(generator)
  150. .set_is_high_bit_depth(true)
  151. .TakePaintImage();
  152. EXPECT_TRUE(image.is_high_bit_depth());
  153. EXPECT_EQ(image.GetContentColorUsage(), gfx::ContentColorUsage::kHDR);
  154. }
  155. } // namespace cc