Add support for more advanced color correction

ash/display/display_color_manager_chromeos.cc

Index: ash/display/display_color_manager_chromeos.cc
diff --git a/ash/display/display_color_manager_chromeos.cc b/ash/display/display_color_manager_chromeos.cc
index e5559198c5a08970543401947a6ba70b36ef7ced..eb347449f0c3bb61c29fffa9db04e4aa9d234890 100644
--- a/ash/display/display_color_manager_chromeos.cc
+++ b/ash/display/display_color_manager_chromeos.cc
@@ -23,7 +23,11 @@ namespace ash {
 namespace {
 
 scoped_ptr<DisplayColorManager::ColorCalibrationData> ParseDisplayProfile(
-    const base::FilePath& path) {
+    const base::FilePath& path,
+    bool has_color_correction_matrix) {
+  VLOG(1) << "Trying ICC file " << path.value()
+          << " has_color_correction_matrix: "
+          << (has_color_correction_matrix ? "true" : "false");
   qcms_profile* display_profile = qcms_profile_from_path(path.value().c_str());
   if (!display_profile) {
     LOG(WARNING) << "Unable to load ICC file: " << path.value();
@@ -32,30 +36,111 @@ scoped_ptr<DisplayColorManager::ColorCalibrationData> ParseDisplayProfile(
 
   size_t vcgt_channel_length =
       qcms_profile_get_vcgt_channel_length(display_profile);
-  if (!vcgt_channel_length) {
-    LOG(WARNING) << "No vcgt table in ICC file: " << path.value();
-    qcms_profile_release(display_profile);
-    return nullptr;
-  }
 
-  std::vector<uint16_t> vcgt_data;
-  vcgt_data.resize(vcgt_channel_length * 3);
-  if (!qcms_profile_get_vcgt_rgb_channels(display_profile, &vcgt_data[0])) {
-    LOG(WARNING) << "Unable to get vcgt data";
+  if (!has_color_correction_matrix && !vcgt_channel_length) {
+    LOG(WARNING) << "No vcgt table in ICC file: " << path.value();
     qcms_profile_release(display_profile);
     return nullptr;
   }
 
   scoped_ptr<DisplayColorManager::ColorCalibrationData> data(
       new DisplayColorManager::ColorCalibrationData());
-  data->lut.resize(vcgt_channel_length);
-  for (size_t i = 0; i < vcgt_channel_length; ++i) {
-    data->lut[i].r = vcgt_data[i];
-    data->lut[i].g = vcgt_data[vcgt_channel_length + i];
-    data->lut[i].b = vcgt_data[(vcgt_channel_length * 2) + i];
+  if (vcgt_channel_length) {
+    VLOG_IF(1, has_color_correction_matrix)
+        << "Using VCGT data on CTM enabled platform.";
+
+    std::vector<uint16_t> vcgt_data;
+    vcgt_data.resize(vcgt_channel_length * 3);
+    if (!qcms_profile_get_vcgt_rgb_channels(display_profile, &vcgt_data[0])) {
+      LOG(WARNING) << "Unable to get vcgt data";
+      qcms_profile_release(display_profile);
+      return nullptr;
+    }
+
+    data->gamma_lut.resize(vcgt_channel_length);
+    for (size_t i = 0; i < vcgt_channel_length; ++i) {
+      data->gamma_lut[i].r = vcgt_data[i];
+      data->gamma_lut[i].g = vcgt_data[vcgt_channel_length + i];
+      data->gamma_lut[i].b = vcgt_data[(vcgt_channel_length * 2) + i];
+    }
+  } else {
+    VLOG(1) << "Using full degamma/gamma/CTM from profile.";
+    qcms_profile* srgb_profile = qcms_profile_sRGB();
+
+    qcms_transform* transform =
+        qcms_transform_create(srgb_profile, QCMS_DATA_RGB_8, display_profile,
+                              QCMS_DATA_RGB_8, QCMS_INTENT_PERCEPTUAL);
+
+    if (!transform) {
+      LOG(WARNING)
+          << "Unable to create transformation from sRGB to display profile.";
+
+      qcms_profile_release(display_profile);
+      qcms_profile_release(srgb_profile);
+      return nullptr;
+    }
+
+    if (!qcms_transform_is_matrix(transform)) {
+      LOG(WARNING) << "No transformation matrix available";
+
+      qcms_transform_release(transform);
+      qcms_profile_release(display_profile);
+      qcms_profile_release(srgb_profile);
+      return nullptr;
+    }
+
+    size_t degamma_size = qcms_transform_get_input_trc_rgba(
+        transform, srgb_profile, QCMS_TRC_USHORT, NULL);
+    size_t gamma_size = qcms_transform_get_output_trc_rgba(
+        transform, display_profile, QCMS_TRC_USHORT, NULL);
+
+    if (degamma_size == 0 || gamma_size == 0) {
+      LOG(WARNING)
+          << "Invalid number of elements in gamma tables: degamma size = "
+          << degamma_size << " gamma size = " << gamma_size;
+
+      qcms_transform_release(transform);
+      qcms_profile_release(display_profile);
+      qcms_profile_release(srgb_profile);
+      return nullptr;
+    }
+
+    std::vector<uint16_t> degamma_data;
+    std::vector<uint16_t> gamma_data;
+    degamma_data.resize(degamma_size * 4);
+    gamma_data.resize(gamma_size * 4);
+
+    qcms_transform_get_input_trc_rgba(transform, srgb_profile, QCMS_TRC_USHORT,
+                                      &degamma_data[0]);
+    qcms_transform_get_output_trc_rgba(transform, display_profile,
+                                       QCMS_TRC_USHORT, &gamma_data[0]);
+
+    data->degamma_lut.resize(degamma_size);
+    for (size_t i = 0; i < degamma_size; ++i) {
+      data->degamma_lut[i].r = degamma_data[i * 4];
+      data->degamma_lut[i].g = degamma_data[(i * 4) + 1];
+      data->degamma_lut[i].b = degamma_data[(i * 4) + 2];
+    }
+
+    data->gamma_lut.resize(gamma_size);
+    for (size_t i = 0; i < gamma_size; ++i) {
+      data->gamma_lut[i].r = gamma_data[i * 4];
+      data->gamma_lut[i].g = gamma_data[(i * 4) + 1];
+      data->gamma_lut[i].b = gamma_data[(i * 4) + 2];
+    }
+
+    data->correction_matrix.resize(9);
+    for (int i = 0; i < 9; ++i) {
+      data->correction_matrix[i] =
+          qcms_transform_get_matrix(transform, i / 3, i % 3);
+    }
+
+    qcms_transform_release(transform);
+    qcms_profile_release(srgb_profile);
   }
+
+  VLOG(1) << "ICC file successfully parsed";
   qcms_profile_release(display_profile);
-  VLOG(1) << "Gamma data successfully read from icc file";
   return data;
 }
 
@@ -90,9 +175,11 @@ void DisplayColorManager::OnDisplayModeChanged(
 void DisplayColorManager::ApplyDisplayColorCalibration(int64_t display_id,
                                                        int64_t product_id) {
   if (calibration_map_.find(product_id) != calibration_map_.end()) {
-    ColorCalibrationData* ramp = calibration_map_[product_id];
-    if (!configurator_->SetGammaRamp(display_id, ramp->lut))
-      LOG(WARNING) << "Error applying gamma ramp";
+    ColorCalibrationData* data = calibration_map_[product_id];
+    if (!configurator_->SetColorCorrection(display_id, data->degamma_lut,
+                                           data->gamma_lut,
+                                           data->correction_matrix))
+      LOG(WARNING) << "Error applying color correction data";
   }
 }
 
@@ -108,12 +195,14 @@ void DisplayColorManager::LoadCalibrationForDisplay(
       display->product_id(),
       base::Bind(&DisplayColorManager::FinishLoadCalibrationForDisplay,
                  weak_ptr_factory_.GetWeakPtr(), display->display_id(),
-                 display->product_id(), display->type()));
+                 display->product_id(), display->has_color_correction_matrix(),
+                 display->type()));
 }
 
 void DisplayColorManager::FinishLoadCalibrationForDisplay(
     int64_t display_id,
     int64_t product_id,
+    bool has_color_correction_matrix,
     ui::DisplayConnectionType type,
     const base::FilePath& path,
     bool file_downloaded) {
@@ -137,7 +226,8 @@ void DisplayColorManager::FinishLoadCalibrationForDisplay(
           << " with product id: " << product_string;
 
   base::PostTaskAndReplyWithResult(
-      blocking_pool_, FROM_HERE, base::Bind(&ParseDisplayProfile, path),
+      blocking_pool_, FROM_HERE,
+      base::Bind(&ParseDisplayProfile, path, has_color_correction_matrix),
       base::Bind(&DisplayColorManager::UpdateCalibrationData,
                  weak_ptr_factory_.GetWeakPtr(), display_id, product_id));
 }