Implement color management for ImageData

third_party/WebKit/Source/core/html/ImageData.cpp

 /*
  * Copyright (C) 2008 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  * 2.  Redistributions in binary form must reproduce the above copyright
@@ skipping 81 matching lines @@
       DOMUint8ClampedArray::createOrNull(dataSize.ValueOrDie());
   if (!byteArray) {
     exceptionState.throwDOMException(V8Error,
                                      "Out of memory at ImageData creation");
     return nullptr;
   }
 
   return new ImageData(IntSize(width, height), byteArray);
 }
 
+ImageData* ImageData::create(unsigned width,
+                             unsigned height,
+                             String colorSpace,
+                             ExceptionState& exceptionState) {
+  if (!width || !height) {
+    exceptionState.throwDOMException(
+        IndexSizeError, String::format("The source %s is zero or not a number.",
+                                       width ? "height" : "width"));
+    return nullptr;
+  }
+
+  CheckedNumeric<unsigned> dataSize = 4;
+  dataSize *= width;
+  dataSize *= height;
+  if (!dataSize.IsValid() || static_cast<int>(width) < 0 ||
+      static_cast<int>(height) < 0) {
+    exceptionState.throwDOMException(
+        IndexSizeError,
+        "The requested image size exceeds the supported range.");
+    return nullptr;
+  }
+
+  DOMUint8ClampedArray* byteArray =
+      DOMUint8ClampedArray::createOrNull(dataSize.ValueOrDie());
+  if (!byteArray) {
+    exceptionState.throwDOMException(V8Error,

[Justin Novosad, 2016/12/08 20:58:59]

this should be V8RangeError

[zakerinasab, 2016/12/08 22:13:52]

Fixed for this and similar occurrences.

+                                     "Out of memory at ImageData creation");
+    return nullptr;
+  }
+
+  return new ImageData(IntSize(width, height), byteArray, colorSpace);
+}
+
 bool ImageData::validateConstructorArguments(DOMUint8ClampedArray* data,
                                              unsigned width,
                                              unsigned& lengthInPixels,
                                              ExceptionState& exceptionState) {
   if (!width) {
     exceptionState.throwDOMException(
         IndexSizeError, "The source width is zero or not a number.");
     return false;
   }
   DCHECK(data);
@@ skipping 12 matching lines @@
   if (length % width) {
     exceptionState.throwDOMException(
         IndexSizeError,
         "The input data byte length is not a multiple of (4 * width).");
     return false;
   }
   lengthInPixels = length;
   return true;
 }
 
+bool ImageData::validateConstructorArguments(DOMUint8ClampedArray* data,
+                                             unsigned width,
+                                             unsigned& lengthInPixels,
+                                             String colorSpace,
+                                             ExceptionState& exceptionState) {
+  if (colorSpace != kLegacyImageDataColorSpaceName &&
+      colorSpace != kSRGBImageDataColorSpaceName) {
+    exceptionState.throwDOMException(NotSupportedError,
+                                     "The input color space is not supported "
+                                     "in Uint8ClampedArraye-backed image "
+                                     "data.");
+    return false;
+  }
+  return validateConstructorArguments(data, width, lengthInPixels,
+                                      exceptionState);
+}
+
 ImageData* ImageData::create(DOMUint8ClampedArray* data,
                              unsigned width,
                              ExceptionState& exceptionState) {
   unsigned lengthInPixels = 0;
   if (!validateConstructorArguments(data, width, lengthInPixels,
                                     exceptionState)) {
     DCHECK(exceptionState.hadException());
     return nullptr;
   }
   DCHECK_GT(lengthInPixels, 0u);
@@ skipping 16 matching lines @@
   DCHECK_GT(width, 0u);
   if (height != lengthInPixels / width) {
     exceptionState.throwDOMException(
         IndexSizeError,
         "The input data byte length is not equal to (4 * width * height).");
     return nullptr;
   }
   return new ImageData(IntSize(width, height), data);
 }
 
+ImageData* ImageData::create(DOMUint8ClampedArray* data,
+                             unsigned width,
+                             unsigned height,
+                             String colorSpace,
+                             ExceptionState& exceptionState) {
+  unsigned lengthInPixels = 0;
+  if (!validateConstructorArguments(data, width, lengthInPixels, colorSpace,
+                                    exceptionState)) {
+    DCHECK(exceptionState.hadException());
+    return nullptr;
+  }
+  DCHECK_GT(lengthInPixels, 0u);
+  DCHECK_GT(width, 0u);
+  if (height != lengthInPixels / width) {
+    exceptionState.throwDOMException(
+        IndexSizeError,
+        "The input data byte length is not equal to (4 * width * height).");
+    return nullptr;
+  }
+  return new ImageData(IntSize(width, height), data, colorSpace);
+}
+
+ImageDataColorSpace ImageData::getImageDataColorSpace(String colorSpaceName) {
+  if (colorSpaceName == kLegacyImageDataColorSpaceName)
+    return kLegacyImageDataColorSpace;
+  if (colorSpaceName == kSRGBImageDataColorSpaceName)
+    return kSRGBImageDataColorSpace;
+  if (colorSpaceName == kLinearRGBImageDataColorSpaceName)
+    return kLinearRGBImageDataColorSpace;
+  NOTREACHED();
+  return kLegacyImageDataColorSpace;
+}
+
+String ImageData::getImageDataColorSpaceName(ImageDataColorSpace colorSpace) {
+  switch (colorSpace) {
+    case kLegacyImageDataColorSpace:
+      return kLegacyImageDataColorSpaceName;
+    case kSRGBImageDataColorSpace:
+      return kSRGBImageDataColorSpaceName;
+    case kLinearRGBImageDataColorSpace:
+      return kLinearRGBImageDataColorSpaceName;
+  }
+  NOTREACHED();
+  return String();
+}
+
+// TODO(zakerinasab): Fix this when ImageBitmap color correction code is landed.
+// Tip: If the source Image Data has a color space, createImageBitmap must
+// respect this color space even when no color space tag is passed to it.
 ScriptPromise ImageData::createImageBitmap(ScriptState* scriptState,
                                            EventTarget& eventTarget,
                                            Optional<IntRect> cropRect,
                                            const ImageBitmapOptions& options,
                                            ExceptionState& exceptionState) {
   if ((cropRect &&
        !ImageBitmap::isSourceSizeValid(cropRect->width(), cropRect->height(),
                                        exceptionState)) ||
       !ImageBitmap::isSourceSizeValid(bitmapSourceSize().width(),
                                       bitmapSourceSize().height(),
@@ skipping 24 matching lines @@
     v8::Local<v8::Value> pixelArray = toV8(m_data.get(), wrapper, isolate);
     if (pixelArray.IsEmpty() ||
         !v8CallBoolean(wrapper->DefineOwnProperty(
             isolate->GetCurrentContext(), v8AtomicString(isolate, "data"),
             pixelArray, v8::ReadOnly)))
       return v8::Local<v8::Object>();
   }
   return wrapper;
 }
 
-ImageData::ImageData(const IntSize& size, DOMUint8ClampedArray* byteArray)
-    : m_size(size), m_data(byteArray) {
+ImageData::ImageData(const IntSize& size,
+                     DOMUint8ClampedArray* byteArray,
+                     String colorSpaceName)
+    : m_size(size),
+      m_colorSpace(getImageDataColorSpace(colorSpaceName)),
+      m_data(byteArray) {
   DCHECK_GE(size.width(), 0);
   DCHECK_GE(size.height(), 0);
   SECURITY_CHECK(static_cast<unsigned>(size.width() * size.height() * 4) <=
                  m_data->length());
 }
 
 }  // namespace blink