ColorTransform, transforms colors from one color space to another

ui/gfx/color_transform_unittest.cc

Index: ui/gfx/color_transform_unittest.cc
diff --git a/ui/gfx/color_transform_unittest.cc b/ui/gfx/color_transform_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..0d686e08f6efbcef333f3e2765215b6fb9e34416
--- /dev/null
+++ b/ui/gfx/color_transform_unittest.cc
@@ -0,0 +1,210 @@
+// Copyright (c) 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/gfx/color_space.h"
+#include "ui/gfx/color_transform.h"
+#include "ui/gfx/transform.h"
+
+namespace gfx {
+
+// Internal functions, exposted for testing.
+GFX_EXPORT Transform GetPrimaryMatrix(ColorSpace::PrimaryID id);
+GFX_EXPORT Transform GetTransferMatrix(ColorSpace::MatrixID id);
+GFX_EXPORT float ToLinear(ColorSpace::TransferID id, float v);
+GFX_EXPORT float FromLinear(ColorSpace::TransferID id, float v);
+
+ColorSpace::PrimaryID all_primaries[] = {
+    ColorSpace::PrimaryID::BT709,        ColorSpace::PrimaryID::BT470M,
+    ColorSpace::PrimaryID::BT470BG,      ColorSpace::PrimaryID::SMPTE170M,
+    ColorSpace::PrimaryID::SMPTE240M,    ColorSpace::PrimaryID::FILM,
+    ColorSpace::PrimaryID::BT2020,       ColorSpace::PrimaryID::SMPTEST428_1,
+    ColorSpace::PrimaryID::SMPTEST431_2, ColorSpace::PrimaryID::SMPTEST432_1,
+};
+
+ColorSpace::TransferID all_transfers[] = {
+    ColorSpace::TransferID::BT709,        ColorSpace::TransferID::GAMMA22,
+    ColorSpace::TransferID::GAMMA28,      ColorSpace::TransferID::SMPTE170M,
+    ColorSpace::TransferID::SMPTE240M,    ColorSpace::TransferID::LINEAR,
+    ColorSpace::TransferID::LOG,          ColorSpace::TransferID::LOG_SQRT,
+    ColorSpace::TransferID::IEC61966_2_4, ColorSpace::TransferID::BT1361_ECG,
+    ColorSpace::TransferID::IEC61966_2_1, ColorSpace::TransferID::BT2020_10,
+    ColorSpace::TransferID::BT2020_12,    ColorSpace::TransferID::SMPTEST2084,
+    // This one is weird as the non-linear numbers are not between 0 and 1.
+    // TODO(hubbe): Test this separately.
+    //  ColorSpace::TransferID::SMPTEST428_1,
+};
+
+ColorSpace::MatrixID all_matrices[] = {
+    ColorSpace::MatrixID::RGB, ColorSpace::MatrixID::BT709,
+    ColorSpace::MatrixID::UNSPECIFIED, ColorSpace::MatrixID::RESERVED,
+    ColorSpace::MatrixID::FCC, ColorSpace::MatrixID::BT470BG,
+    ColorSpace::MatrixID::SMPTE170M, ColorSpace::MatrixID::SMPTE240M,
+
+    // YCOCG produces lots of negative values which isn't compatible with many
+    // transfer functions.
+    // TODO(hubbe): Test this separately.
+    // ColorSpace::MatrixID::YCOCG,
+    ColorSpace::MatrixID::BT2020_NCL, ColorSpace::MatrixID::BT2020_CL,
+    ColorSpace::MatrixID::YDZDX,
+};
+
+ColorSpace::RangeID all_ranges[] = {ColorSpace::RangeID::FULL,
+                                    ColorSpace::RangeID::LIMITED};
+
+TEST(SimpleColorSpace, BT709toSRGB) {
+  ColorSpace bt709 = ColorSpace::CreateREC709();
+  ColorSpace sRGB = ColorSpace::CreateSRGB();
+  std::unique_ptr<ColorTransform> t(ColorTransform::NewColorTransform(
+      bt709, sRGB, ColorTransform::Intent::ABSOLUTE));
+
+  ColorTransform::TriStim tmp(16.0f / 255.0f, 0.5f, 0.5f);
+  t->transform(&tmp, 1);
+  EXPECT_NEAR(tmp.x(), 0.0f, 0.001f);
+  EXPECT_NEAR(tmp.y(), 0.0f, 0.001f);
+  EXPECT_NEAR(tmp.z(), 0.0f, 0.001f);
+
+  tmp = ColorTransform::TriStim(235.0f / 255.0f, 0.5f, 0.5f);
+  t->transform(&tmp, 1);
+  EXPECT_NEAR(tmp.x(), 1.0f, 0.001f);
+  EXPECT_NEAR(tmp.y(), 1.0f, 0.001f);
+  EXPECT_NEAR(tmp.z(), 1.0f, 0.001f);
+
+  // Test a blue color
+  tmp = ColorTransform::TriStim(128.0f / 255.0f, 240.0f / 255.0f, 0.5f);
+  t->transform(&tmp, 1);
+  EXPECT_GT(tmp.z(), tmp.x());
+  EXPECT_GT(tmp.z(), tmp.y());
+}
+
+TEST(SimpleColorSpace, UnknownToSRGB) {
+  ColorSpace unknown;
+  ColorSpace sRGB = ColorSpace::CreateSRGB();
+  std::unique_ptr<ColorTransform> t(ColorTransform::NewColorTransform(
+      unknown, sRGB, ColorTransform::Intent::PERCEPTUAL));
+
+  ColorTransform::TriStim tmp(16.0f / 255.0f, 0.5f, 0.5f);
+  t->transform(&tmp, 1);
+  EXPECT_NEAR(tmp.x(), 0.0f, 0.001f);
+  EXPECT_NEAR(tmp.y(), 0.0f, 0.001f);
+  EXPECT_NEAR(tmp.z(), 0.0f, 0.001f);
+
+  tmp = ColorTransform::TriStim(235.0f / 255.0f, 0.5f, 0.5f);
+  t->transform(&tmp, 1);
+  EXPECT_NEAR(tmp.x(), 1.0f, 0.001f);
+  EXPECT_NEAR(tmp.y(), 1.0f, 0.001f);
+  EXPECT_NEAR(tmp.z(), 1.0f, 0.001f);
+
+  // Test a blue color
+  tmp = ColorTransform::TriStim(128.0f / 255.0f, 240.0f / 255.0f, 0.5f);
+  t->transform(&tmp, 1);
+  EXPECT_GT(tmp.z(), tmp.x());
+  EXPECT_GT(tmp.z(), tmp.y());
+}
+
+class PrimaryTest : public testing::TestWithParam<ColorSpace::PrimaryID> {};
+
+TEST_P(PrimaryTest, checkInvertible) {
+  EXPECT_EQ(GetPrimaryMatrix(GetParam()).matrix().get(3, 3), 1.0f);
+  // Check that all primary matrices are invertable.
+  EXPECT_TRUE(GetPrimaryMatrix(GetParam()).IsInvertible());
+}
+
+INSTANTIATE_TEST_CASE_P(ColorSpace,
+                        PrimaryTest,
+                        testing::ValuesIn(all_primaries));
+
+class MatrixTest : public testing::TestWithParam<ColorSpace::MatrixID> {};
+
+TEST_P(MatrixTest, checkInvertible) {
+  EXPECT_EQ(GetTransferMatrix(GetParam()).matrix().get(3, 3), 1.0f);
+  // Check that all transfer matrices are invertable.
+  EXPECT_TRUE(GetTransferMatrix(GetParam()).IsInvertible());
+};
+
+INSTANTIATE_TEST_CASE_P(ColorSpace,
+                        MatrixTest,
+                        testing::ValuesIn(all_matrices));
+
+class TransferTest : public testing::TestWithParam<ColorSpace::TransferID> {};
+
+TEST_P(TransferTest, basicTest) {
+  for (float x = 0.0f; x <= 1.0f; x += 1.0f / 128.0f) {
+    float linear = ToLinear(GetParam(), x);
+    float x2 = FromLinear(GetParam(), linear);
+    EXPECT_NEAR(x, x2, 0.001f);
+  }
+}
+
+INSTANTIATE_TEST_CASE_P(ColorSpace,
+                        TransferTest,
+                        testing::ValuesIn(all_transfers));
+
+typedef std::tr1::tuple<ColorSpace::PrimaryID,
+                        ColorSpace::TransferID,
+                        ColorSpace::MatrixID,
+                        ColorSpace::RangeID>
+    ColorSpaceTestData;
+
+class ColorSpaceTest : public testing::TestWithParam<ColorSpaceTestData> {
+ public:
+  ColorSpaceTest()
+      : color_space_(std::tr1::get<0>(GetParam()),
+                     std::tr1::get<1>(GetParam()),
+                     std::tr1::get<2>(GetParam()),
+                     std::tr1::get<3>(GetParam())) {}
+
+ protected:
+  ColorSpace color_space_;
+};
+
+TEST_P(ColorSpaceTest, testNullTransform) {
+  std::unique_ptr<ColorTransform> t(ColorTransform::NewColorTransform(
+      color_space_, color_space_, ColorTransform::Intent::ABSOLUTE));
+  ColorTransform::TriStim tristim(0.4f, 0.5f, 0.6f);
+  t->transform(&tristim, 1);
+  EXPECT_NEAR(tristim.x(), 0.4f, 0.001f);
+  EXPECT_NEAR(tristim.y(), 0.5f, 0.001f);
+  EXPECT_NEAR(tristim.z(), 0.6f, 0.001f);
+}
+
+TEST_P(ColorSpaceTest, toXYZandBack) {
+  std::unique_ptr<ColorTransform> t1(ColorTransform::NewColorTransform(
+      color_space_, ColorSpace::CreateXYZD50(),
+      ColorTransform::Intent::ABSOLUTE));
+  std::unique_ptr<ColorTransform> t2(ColorTransform::NewColorTransform(
+      ColorSpace::CreateXYZD50(), color_space_,
+      ColorTransform::Intent::ABSOLUTE));
+  ColorTransform::TriStim tristim(0.4f, 0.5f, 0.6f);
+  t1->transform(&tristim, 1);
+  t2->transform(&tristim, 1);
+  EXPECT_NEAR(tristim.x(), 0.4f, 0.001f);
+  EXPECT_NEAR(tristim.y(), 0.5f, 0.001f);
+  EXPECT_NEAR(tristim.z(), 0.6f, 0.001f);
+}
+
+INSTANTIATE_TEST_CASE_P(
+    A,
+    ColorSpaceTest,
+    testing::Combine(testing::ValuesIn(all_primaries),
+                     testing::ValuesIn(all_transfers),
+                     testing::Values(ColorSpace::MatrixID::BT709),
+                     testing::Values(ColorSpace::RangeID::LIMITED)));
+
+INSTANTIATE_TEST_CASE_P(
+    B,
+    ColorSpaceTest,
+    testing::Combine(testing::Values(ColorSpace::PrimaryID::BT709),
+                     testing::ValuesIn(all_transfers),
+                     testing::ValuesIn(all_matrices),
+                     testing::ValuesIn(all_ranges)));
+
+INSTANTIATE_TEST_CASE_P(
+    C,
+    ColorSpaceTest,
+    testing::Combine(testing::ValuesIn(all_primaries),
+                     testing::Values(ColorSpace::TransferID::BT709),
+                     testing::ValuesIn(all_matrices),
+                     testing::ValuesIn(all_ranges)));
+}  // namespace