ColorSpaceTest.cpp 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336
  1. /*
  2. * Copyright 2016 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/SkColorSpace.h"
  9. #include "include/core/SkData.h"
  10. #include "include/core/SkImageInfo.h"
  11. #include "include/core/SkRefCnt.h"
  12. #include "include/core/SkStream.h"
  13. #include "include/core/SkTypes.h"
  14. #include "include/third_party/skcms/skcms.h"
  15. #include "src/core/SkColorSpacePriv.h"
  16. #include "tests/Test.h"
  17. #include "tools/Resources.h"
  18. #include "png.h"
  19. #include <string.h>
  20. #include <memory>
  21. #include <utility>
  22. static bool almost_equal(float a, float b) {
  23. return SkTAbs(a - b) < 0.001f;
  24. }
  25. static void test_space(skiatest::Reporter* r, SkColorSpace* space,
  26. const float red[], const float green[], const float blue[],
  27. bool expectSRGB = false) {
  28. REPORTER_ASSERT(r, nullptr != space);
  29. REPORTER_ASSERT(r, expectSRGB == space->gammaCloseToSRGB());
  30. skcms_Matrix3x3 mat;
  31. space->toXYZD50(&mat);
  32. const float* ref[3] = { red, green, blue };
  33. for (int i = 0; i < 3; ++i) {
  34. REPORTER_ASSERT(r, almost_equal(ref[i][0], mat.vals[0][i]));
  35. REPORTER_ASSERT(r, almost_equal(ref[i][1], mat.vals[1][i]));
  36. REPORTER_ASSERT(r, almost_equal(ref[i][2], mat.vals[2][i]));
  37. }
  38. }
  39. static void test_path(skiatest::Reporter* r, const char* path,
  40. const float red[], const float green[], const float blue[],
  41. bool expectSRGB = false) {
  42. std::unique_ptr<SkStream> stream(GetResourceAsStream(path));
  43. REPORTER_ASSERT(r, nullptr != stream);
  44. if (!stream) {
  45. return;
  46. }
  47. std::unique_ptr<SkCodec> codec(SkCodec::MakeFromStream(std::move(stream)));
  48. REPORTER_ASSERT(r, nullptr != codec);
  49. if (!codec) {
  50. return;
  51. }
  52. auto colorSpace = codec->getInfo().refColorSpace();
  53. test_space(r, colorSpace.get(), red, green, blue, expectSRGB);
  54. }
  55. static constexpr float g_sRGB_R[]{ 0.4358f, 0.2224f, 0.0139f };
  56. static constexpr float g_sRGB_G[]{ 0.3853f, 0.7170f, 0.0971f };
  57. static constexpr float g_sRGB_B[]{ 0.1430f, 0.0606f, 0.7139f };
  58. DEF_TEST(ColorSpace_sRGB, r) {
  59. test_space(r, sk_srgb_singleton(), g_sRGB_R, g_sRGB_G, g_sRGB_B, true);
  60. }
  61. DEF_TEST(ColorSpaceParseICCProfiles, r) {
  62. #if (PNG_LIBPNG_VER_MAJOR > 1) || (PNG_LIBPNG_VER_MAJOR == 1 && PNG_LIBPNG_VER_MINOR >= 6)
  63. test_path(r, "images/color_wheel_with_profile.png", g_sRGB_R, g_sRGB_G, g_sRGB_B, true);
  64. #endif
  65. const float red[] = { 0.385117f, 0.716904f, 0.0970612f };
  66. const float green[] = { 0.143051f, 0.0606079f, 0.713913f };
  67. const float blue[] = { 0.436035f, 0.222488f, 0.013916f };
  68. test_path(r, "images/icc-v2-gbr.jpg", red, green, blue);
  69. test_path(r, "images/webp-color-profile-crash.webp",
  70. red, green, blue);
  71. test_path(r, "images/webp-color-profile-lossless.webp",
  72. red, green, blue);
  73. test_path(r, "images/webp-color-profile-lossy.webp",
  74. red, green, blue);
  75. test_path(r, "images/webp-color-profile-lossy-alpha.webp",
  76. red, green, blue);
  77. }
  78. static void test_serialize(skiatest::Reporter* r, sk_sp<SkColorSpace> space, bool isNamed) {
  79. sk_sp<SkData> data1 = space->serialize();
  80. size_t bytes = space->writeToMemory(nullptr);
  81. sk_sp<SkData> data2 = SkData::MakeUninitialized(bytes);
  82. space->writeToMemory(data2->writable_data());
  83. sk_sp<SkColorSpace> newSpace1 = SkColorSpace::Deserialize(data1->data(), data1->size());
  84. sk_sp<SkColorSpace> newSpace2 = SkColorSpace::Deserialize(data2->data(), data2->size());
  85. if (isNamed) {
  86. REPORTER_ASSERT(r, space.get() == newSpace1.get());
  87. REPORTER_ASSERT(r, space.get() == newSpace2.get());
  88. } else {
  89. REPORTER_ASSERT(r, SkColorSpace::Equals(space.get(), newSpace1.get()));
  90. REPORTER_ASSERT(r, SkColorSpace::Equals(space.get(), newSpace2.get()));
  91. }
  92. }
  93. DEF_TEST(ColorSpace_Serialize, r) {
  94. test_serialize(r, SkColorSpace::MakeSRGB(), true);
  95. test_serialize(r, SkColorSpace::MakeSRGBLinear(), true);
  96. auto test = [&](const char* path) {
  97. sk_sp<SkData> data = GetResourceAsData(path);
  98. skcms_ICCProfile profile;
  99. REPORTER_ASSERT(r, skcms_Parse(data->data(), data->size(), &profile));
  100. sk_sp<SkColorSpace> space = SkColorSpace::Make(profile);
  101. REPORTER_ASSERT(r, space);
  102. test_serialize(r, space, false);
  103. };
  104. test("icc_profiles/HP_ZR30w.icc");
  105. test("icc_profiles/HP_Z32x.icc");
  106. skcms_TransferFunction fn;
  107. fn.a = 1.0f;
  108. fn.b = 0.0f;
  109. fn.c = 1.0f;
  110. fn.d = 0.5f;
  111. fn.e = 0.0f;
  112. fn.f = 0.0f;
  113. fn.g = 1.0f;
  114. skcms_Matrix3x3 toXYZ = {{
  115. { 1, 0, 0 },
  116. { 0, 1, 0 },
  117. { 0, 0, 1 },
  118. }};
  119. test_serialize(r, SkColorSpace::MakeRGB(fn, toXYZ), false);
  120. }
  121. DEF_TEST(ColorSpace_Equals, r) {
  122. sk_sp<SkColorSpace> srgb = SkColorSpace::MakeSRGB();
  123. auto parse = [&](const char* path) {
  124. sk_sp<SkData> data = GetResourceAsData(path);
  125. skcms_ICCProfile profile;
  126. REPORTER_ASSERT(r, skcms_Parse(data->data(), data->size(), &profile));
  127. sk_sp<SkColorSpace> space = SkColorSpace::Make(profile);
  128. REPORTER_ASSERT(r, space);
  129. return space;
  130. };
  131. sk_sp<SkColorSpace> z30 = parse("icc_profiles/HP_ZR30w.icc");
  132. sk_sp<SkColorSpace> z32 = parse("icc_profiles/HP_Z32x.icc");
  133. skcms_TransferFunction fn;
  134. fn.a = 1.0f;
  135. fn.b = 0.0f;
  136. fn.c = 1.0f;
  137. fn.d = 0.5f;
  138. fn.e = 0.0f;
  139. fn.f = 0.0f;
  140. fn.g = 1.0f;
  141. skcms_Matrix3x3 toXYZ = {{
  142. { 1, 0, 0 },
  143. { 0, 1, 0 },
  144. { 0, 0, 1 },
  145. }};
  146. sk_sp<SkColorSpace> rgb4 = SkColorSpace::MakeRGB(fn, toXYZ);
  147. REPORTER_ASSERT(r, SkColorSpace::Equals(nullptr, nullptr));
  148. REPORTER_ASSERT(r, SkColorSpace::Equals(srgb.get(), srgb.get()));
  149. REPORTER_ASSERT(r, SkColorSpace::Equals(z30.get(), z30.get()));
  150. REPORTER_ASSERT(r, SkColorSpace::Equals(z32.get(), z32.get()));
  151. REPORTER_ASSERT(r, SkColorSpace::Equals(rgb4.get(), rgb4.get()));
  152. REPORTER_ASSERT(r, !SkColorSpace::Equals(nullptr, srgb.get()));
  153. REPORTER_ASSERT(r, !SkColorSpace::Equals(srgb.get(), nullptr));
  154. REPORTER_ASSERT(r, !SkColorSpace::Equals(z30.get(), srgb.get()));
  155. REPORTER_ASSERT(r, !SkColorSpace::Equals(z32.get(), z30.get()));
  156. REPORTER_ASSERT(r, !SkColorSpace::Equals(z30.get(), rgb4.get()));
  157. REPORTER_ASSERT(r, !SkColorSpace::Equals(srgb.get(), rgb4.get()));
  158. }
  159. static inline bool matrix_almost_equal(const skcms_Matrix3x3& a, const skcms_Matrix3x3& b) {
  160. for (int r = 0; r < 3; ++r) {
  161. for (int c = 0; c < 3; ++c) {
  162. if (!almost_equal(a.vals[r][c], b.vals[r][c])) {
  163. return false;
  164. }
  165. }
  166. }
  167. return true;
  168. }
  169. static inline void check_primaries(skiatest::Reporter* r, const SkColorSpacePrimaries& primaries,
  170. const skcms_Matrix3x3& reference) {
  171. skcms_Matrix3x3 toXYZ;
  172. bool result = primaries.toXYZD50(&toXYZ);
  173. REPORTER_ASSERT(r, result);
  174. REPORTER_ASSERT(r, matrix_almost_equal(toXYZ, reference));
  175. }
  176. DEF_TEST(ColorSpace_Primaries, r) {
  177. // sRGB primaries (D65)
  178. skcms_Matrix3x3 srgbToXYZ;
  179. bool result = skcms_PrimariesToXYZD50(
  180. 0.64f, 0.33f,
  181. 0.30f, 0.60f,
  182. 0.15f, 0.06f,
  183. 0.3127f, 0.3290f,
  184. &srgbToXYZ);
  185. REPORTER_ASSERT(r, result);
  186. sk_sp<SkColorSpace> space = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB, srgbToXYZ);
  187. REPORTER_ASSERT(r, SkColorSpace::MakeSRGB() == space);
  188. // ProPhoto (D50)
  189. SkColorSpacePrimaries proPhoto;
  190. proPhoto.fRX = 0.7347f;
  191. proPhoto.fRY = 0.2653f;
  192. proPhoto.fGX = 0.1596f;
  193. proPhoto.fGY = 0.8404f;
  194. proPhoto.fBX = 0.0366f;
  195. proPhoto.fBY = 0.0001f;
  196. proPhoto.fWX = 0.34567f;
  197. proPhoto.fWY = 0.35850f;
  198. skcms_Matrix3x3 proToXYZ = {{
  199. { 0.7976749f, 0.1351917f, 0.0313534f },
  200. { 0.2880402f, 0.7118741f, 0.0000857f },
  201. { 0.0000000f, 0.0000000f, 0.8252100f },
  202. }};
  203. check_primaries(r, proPhoto, proToXYZ);
  204. // NTSC (C)
  205. SkColorSpacePrimaries ntsc;
  206. ntsc.fRX = 0.67f;
  207. ntsc.fRY = 0.33f;
  208. ntsc.fGX = 0.21f;
  209. ntsc.fGY = 0.71f;
  210. ntsc.fBX = 0.14f;
  211. ntsc.fBY = 0.08f;
  212. ntsc.fWX = 0.31006f;
  213. ntsc.fWY = 0.31616f;
  214. skcms_Matrix3x3 ntscToXYZ = {{
  215. { 0.6343706f, 0.1852204f, 0.1446290f },
  216. { 0.3109496f, 0.5915984f, 0.0974520f },
  217. { -0.0011817f, 0.0555518f, 0.7708399f }
  218. }};
  219. check_primaries(r, ntsc, ntscToXYZ);
  220. // DCI P3 (D65)
  221. SkColorSpacePrimaries p3;
  222. p3.fRX = 0.680f;
  223. p3.fRY = 0.320f;
  224. p3.fGX = 0.265f;
  225. p3.fGY = 0.690f;
  226. p3.fBX = 0.150f;
  227. p3.fBY = 0.060f;
  228. p3.fWX = 0.3127f;
  229. p3.fWY = 0.3290f;
  230. space = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB, SkNamedGamut::kDCIP3);
  231. skcms_Matrix3x3 reference;
  232. SkAssertResult(space->toXYZD50(&reference));
  233. check_primaries(r, p3, reference);
  234. // Rec 2020 (D65)
  235. SkColorSpacePrimaries rec2020;
  236. rec2020.fRX = 0.708f;
  237. rec2020.fRY = 0.292f;
  238. rec2020.fGX = 0.170f;
  239. rec2020.fGY = 0.797f;
  240. rec2020.fBX = 0.131f;
  241. rec2020.fBY = 0.046f;
  242. rec2020.fWX = 0.3127f;
  243. rec2020.fWY = 0.3290f;
  244. space = SkColorSpace::MakeRGB(SkNamedTransferFn::kSRGB, SkNamedGamut::kRec2020);
  245. SkAssertResult(space->toXYZD50(&reference));
  246. check_primaries(r, rec2020, reference);
  247. }
  248. DEF_TEST(ColorSpace_MatrixHash, r) {
  249. sk_sp<SkColorSpace> srgb = SkColorSpace::MakeSRGB();
  250. skcms_TransferFunction fn;
  251. fn.a = 1.0f;
  252. fn.b = 0.0f;
  253. fn.c = 0.0f;
  254. fn.d = 0.0f;
  255. fn.e = 0.0f;
  256. fn.f = 0.0f;
  257. fn.g = 3.0f;
  258. sk_sp<SkColorSpace> strange = SkColorSpace::MakeRGB(fn, SkNamedGamut::kSRGB);
  259. REPORTER_ASSERT(r, srgb->toXYZD50Hash() == strange->toXYZD50Hash());
  260. }
  261. DEF_TEST(ColorSpace_IsSRGB, r) {
  262. sk_sp<SkColorSpace> srgb0 = SkColorSpace::MakeSRGB();
  263. skcms_TransferFunction fn;
  264. fn.a = 1.0f;
  265. fn.b = 0.0f;
  266. fn.c = 0.0f;
  267. fn.d = 0.0f;
  268. fn.e = 0.0f;
  269. fn.f = 0.0f;
  270. fn.g = 2.2f;
  271. sk_sp<SkColorSpace> twoDotTwo = SkColorSpace::MakeRGB(fn, SkNamedGamut::kSRGB);
  272. REPORTER_ASSERT(r, srgb0->isSRGB());
  273. REPORTER_ASSERT(r, !twoDotTwo->isSRGB());
  274. }
  275. DEF_TEST(ColorSpace_skcms_IsSRGB, r) {
  276. sk_sp<SkColorSpace> srgb = SkColorSpace::Make(*skcms_sRGB_profile());
  277. REPORTER_ASSERT(r, srgb->isSRGB());
  278. }
  279. DEF_TEST(ColorSpace_skcms_sRGB_exact, r) {
  280. skcms_ICCProfile profile;
  281. sk_srgb_singleton()->toProfile(&profile);
  282. REPORTER_ASSERT(r, 0 == memcmp(&profile, skcms_sRGB_profile(), sizeof(skcms_ICCProfile)));
  283. }