Create SkColorSpaceXform to handle color conversions

dm/DMSrcSink.cpp

 /*
  * Copyright 2015 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "DMSrcSink.h"
+#include "Resources.h"
 #include "SkAndroidCodec.h"
 #include "SkCodec.h"
 #include "SkCodecImageGenerator.h"
 #include "SkColorSpace.h"
 #include "SkColorSpace_Base.h"
+#include "SkColorSpaceXform.h"
 #include "SkCommonFlags.h"
 #include "SkData.h"
 #include "SkDocument.h"
 #include "SkError.h"
 #include "SkImageGenerator.h"
 #include "SkImageGeneratorCG.h"
 #include "SkImageGeneratorWIC.h"
 #include "SkMallocPixelRef.h"
 #include "SkMultiPictureDraw.h"
 #include "SkNullCanvas.h"
 #include "SkOSFile.h"
 #include "SkOpts.h"
 #include "SkPictureData.h"
 #include "SkPictureRecorder.h"
 #include "SkRandom.h"
 #include "SkRecordDraw.h"
 #include "SkRecorder.h"
 #include "SkSVGCanvas.h"
 #include "SkStream.h"
 #include "SkTLogic.h"
 #include "SkXMLWriter.h"
 #include "SkSwizzler.h"
 #include <functional>
 
 #if defined(SK_BUILD_FOR_WIN)
     #include "SkAutoCoInitialize.h"
 #endif
 
+#include "qcms.h"
+
 DEFINE_bool(multiPage, false, "For document-type backends, render the source"
             " into multiple pages");
 DEFINE_bool(RAW_threading, true, "Allow RAW decodes to run on multiple threads?");
 
 using sk_gpu_test::GrContextFactory;
 
 namespace DM {
 
 GMSrc::GMSrc(skiagm::GMRegistry::Factory factory) : fFactory(factory) {}
 
@@ skipping 768 matching lines @@
     }
     return codec->getInfo().dimensions();
 }
 
 Name ImageGenSrc::name() const {
     return SkOSPath::Basename(fPath.c_str());
 }
 
 /*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
 
-ColorCodecSrc::ColorCodecSrc(Path path, Mode mode, sk_sp<SkColorSpace> dstSpace)
+ColorCodecSrc::ColorCodecSrc(Path path, Mode mode)
     : fPath(path)
     , fMode(mode)
-    , fDstSpace(dstSpace)
 {}
 
 bool ColorCodecSrc::veto(SinkFlags flags) const {
     // Test to direct raster backends (8888 and 565).
     return flags.type != SinkFlags::kRaster || flags.approach != SinkFlags::kDirect;
 }
 
-static uint8_t clampFloatToByte(float v) {
-    v = v * 255.0f;
-    if (v > 255.0f) {
-        return 255;
-    } else if (v < 0.0f) {
-        return 0;
-    } else {
-        return (uint8_t) (v + 0.5f);
-    }
-}
-
 Error ColorCodecSrc::draw(SkCanvas* canvas) const {
     if (kRGB_565_SkColorType == canvas->imageInfo().colorType()) {
         return Error::Nonfatal("No need to test color correction to 565 backend.");
     }
 
     SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
     if (!encoded) {
         return SkStringPrintf("Couldn't read %s.", fPath.c_str());
     }
 
@@ skipping 12 matching lines @@
     SkImageInfo decodeInfo = info;
     if (kBaseline_Mode != fMode) {
         decodeInfo = decodeInfo.makeColorType(kRGBA_8888_SkColorType);
     }
 
     SkCodec::Result r = codec->getPixels(decodeInfo, bitmap.getPixels(), bitmap.rowBytes());
     if (SkCodec::kSuccess != r) {
         return SkStringPrintf("Couldn't getPixels %s. Error code %d", fPath.c_str(), r);
     }
 
+    // Load the dst ICC profile.  This particular dst is fairly similar to Adobe RGB.
+    sk_sp<SkData> dstData = SkData::MakeFromFileName(
+            GetResourcePath("monitor_profiles/HP_ZR30w.icc").c_str());
+
     switch (fMode) {
         case kBaseline_Mode:
             canvas->drawBitmap(bitmap, 0, 0);
             break;
         case kDst_HPZR30w_Mode: {
             sk_sp<SkColorSpace> srcSpace = sk_ref_sp(codec->getColorSpace());
-            if (!srcSpace) {
-                return SkStringPrintf("Cannot test color correction without a src profile.");
-            } else if (!as_CSB(srcSpace)->gammas()->isValues()) {
+            sk_sp<SkColorSpace> dstSpace = SkColorSpace::NewICC(dstData->data(), dstData->size());

[scroggo, 2016/05/31 21:13:34]

Should we fail gracefully if dstData is null?

[msarett, 2016/05/31 22:02:19]

Yes I think so.  Let's fail, but not crash, if the resource path isn't set
properly.

+            SkASSERT(dstSpace);
+
+            std::unique_ptr<SkColorSpaceXform> xform = SkColorSpaceXform::New(srcSpace, dstSpace);
+            if (!xform) {
                 // FIXME (msarett):
-                // The conversion here doesn't cover all of the images that I've uploaded for
+                // I haven't implemented conversions for all of the images that I've uploaded for
                 // testing.  Once we support all of them, this should be a fatal error.
-                return Error::Nonfatal("Unimplemented gamma conversion.");
+                return Error::Nonfatal("Unimplemented color conversion.");
             }
 
-            // Build a matrix to transform to dst gamut.
-            // srcToDst = inverse(dstToXYZ) * srcToXYZ
-            const SkMatrix44& srcToXYZ = srcSpace->xyz();
-            const SkMatrix44& dstToXYZ = fDstSpace->xyz();
-            SkMatrix44 srcToDst(SkMatrix44::kUninitialized_Constructor);
-            dstToXYZ.invert(&srcToDst);
-            srcToDst.postConcat(srcToXYZ);
-
+            uint32_t* row = (uint32_t*) bitmap.getPixels();
             for (int y = 0; y < info.height(); y++) {
-                for (int x = 0; x < info.width(); x++) {
-                    // Extract floats.
-                    uint32_t* pixelPtr = (uint32_t*) bitmap.getAddr(x, y);
-                    float src[3];
-                    src[0] = ((*pixelPtr >>  0) & 0xFF) / 255.0f;
-                    src[1] = ((*pixelPtr >>  8) & 0xFF) / 255.0f;
-                    src[2] = ((*pixelPtr >> 16) & 0xFF) / 255.0f;
-
-                    // Convert to linear.
-                    src[0] = pow(src[0], as_CSB(srcSpace)->gammas()->fRed.fValue);
-                    src[1] = pow(src[1], as_CSB(srcSpace)->gammas()->fGreen.fValue);
-                    src[2] = pow(src[2], as_CSB(srcSpace)->gammas()->fBlue.fValue);
-
-                    // Convert to dst gamut.
-                    float dst[3];
-                    dst[0] = src[0]*srcToDst.getFloat(0, 0) + src[1]*srcToDst.getFloat(1, 0) +
-                             src[2]*srcToDst.getFloat(2, 0) + srcToDst.getFloat(3, 0);
-                    dst[1] = src[0]*srcToDst.getFloat(0, 1) + src[1]*srcToDst.getFloat(1, 1) +
-                             src[2]*srcToDst.getFloat(2, 1) + srcToDst.getFloat(3, 1);
-                    dst[2] = src[0]*srcToDst.getFloat(0, 2) + src[1]*srcToDst.getFloat(1, 2) +
-                             src[2]*srcToDst.getFloat(2, 2) + srcToDst.getFloat(3, 2);
-
-                    // Convert to dst gamma.
-                    dst[0] = pow(dst[0], 1.0f / as_CSB(fDstSpace)->gammas()->fRed.fValue);
-                    dst[1] = pow(dst[1], 1.0f / as_CSB(fDstSpace)->gammas()->fGreen.fValue);
-                    dst[2] = pow(dst[2], 1.0f / as_CSB(fDstSpace)->gammas()->fBlue.fValue);
-
-                    *pixelPtr = SkPackARGB32NoCheck(((*pixelPtr >> 24) & 0xFF),
-                                                    clampFloatToByte(dst[0]),
-                                                    clampFloatToByte(dst[1]),
-                                                    clampFloatToByte(dst[2]));
-                }
+                xform->xform_RGBA_8888(row, row, info.width());
+                row = SkTAddOffset<uint32_t>(row, bitmap.rowBytes());
             }
 
             canvas->drawBitmap(bitmap, 0, 0);
             break;
         }
+        case kQCMS_HPZR30w_Mode: {
+            sk_sp<SkData> srcData = codec->getICCData();
+            qcms_profile* srcSpace = qcms_profile_from_memory(srcData->data(), srcData->size());
+            if (!srcSpace) {
+                return Error::Nonfatal(SkStringPrintf("QCMS cannot create profile for %s.\n",
+                                       fPath.c_str()));
+            }
+
+            qcms_profile* dstSpace = qcms_profile_from_memory(dstData->data(), dstData->size());
+            SkASSERT(dstSpace);
+            qcms_transform* transform = qcms_transform_create(srcSpace, QCMS_DATA_RGBA_8, dstSpace,
+                                                              QCMS_DATA_RGBA_8,
+                                                              QCMS_INTENT_PERCEPTUAL);
+            if (!transform) {
+                qcms_profile_release(srcSpace);

[scroggo, 2016/05/31 21:13:34]

Can you put these in an auto deleter?

[msarett, 2016/05/31 22:02:19]

Done.

+                qcms_profile_release(dstSpace);
+                return SkStringPrintf("QCMS cannot create transform for %s.\n", fPath.c_str());
+            }
+
+#ifdef SK_PMCOLOR_IS_RGBA
+            qcms_output_type outType = QCMS_OUTPUT_RGBX;
+#else
+            qcms_output_type outType = QCMS_OUTPUT_BGRX;
+#endif
+
+            // Perform color correction.
+            uint32_t* row = (uint32_t*) bitmap.getPixels();
+            for (int y = 0; y < info.height(); y++) {
+                qcms_transform_data_type(transform, row, row, info.width(), outType);
+                row = SkTAddOffset<uint32_t>(row, bitmap.rowBytes());
+            }
+
+            qcms_profile_release(srcSpace);
+            qcms_profile_release(dstSpace);
+            qcms_transform_release(transform);
+            canvas->drawBitmap(bitmap, 0, 0);
+            break;
+        }
         default:
             SkASSERT(false);
             return "Invalid fMode";
     }
     return "";
 }
 
 SkISize ColorCodecSrc::size() const {
     SkAutoTUnref<SkData> encoded(SkData::NewFromFileName(fPath.c_str()));
     SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(encoded));
@@ skipping 575 matching lines @@
             skr.visit(i, drawsAsSingletonPictures);
         }
         sk_sp<SkPicture> macroPic(macroRec.finishRecordingAsPicture());
 
         canvas->drawPicture(macroPic);
         return check_against_reference(bitmap, src, fSink);
     });
 }
 
 }  // namespace DM