[poc] curve-filter

third_party/WebKit/Source/platform/graphics/BitmapImage.cpp

 /*
  * Copyright (C) 2006 Samuel Weinig (sam.weinig@gmail.com)
  * Copyright (C) 2004, 2005, 2006, 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 13 matching lines @@
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "platform/graphics/BitmapImage.h"
 
 #include "platform/PlatformInstrumentation.h"
 #include "platform/Timer.h"
 #include "platform/TraceEvent.h"
 #include "platform/geometry/FloatRect.h"
 #include "platform/graphics/BitmapImageMetrics.h"
+#include "platform/graphics/ColorSpaceFilter.h"
+#include "platform/graphics/ColorSpaceProfile.h"
 #include "platform/graphics/DeferredImageDecoder.h"
+#include "platform/graphics/GraphicsContext.h"
+#include "platform/graphics/GraphicsScreen.h"
+#include "platform/graphics/ImageBuffer.h"
 #include "platform/graphics/ImageObserver.h"
 #include "platform/graphics/StaticBitmapImage.h"
 #include "platform/graphics/skia/SkiaUtils.h"
 #include "third_party/skia/include/core/SkCanvas.h"
+#include "third_party/skia/include/core/SkColorFilter.h"
+#include "third_party/skia/include/effects/SkColorFilterImageFilter.h"
 #include "wtf/PassRefPtr.h"
+#include "wtf/RefPtr.h"
 #include "wtf/text/WTFString.h"
 
 namespace blink {
 
 PassRefPtr<BitmapImage> BitmapImage::createWithOrientationForTesting(const SkBitmap& bitmap, ImageOrientation orientation)
 {
-    if (bitmap.isNull()) {
+    if (bitmap.isNull())
         return BitmapImage::create();
-    }
 
     RefPtr<BitmapImage> result = adoptRef(new BitmapImage(bitmap));
     result->m_frames[0].m_orientation = orientation;
     if (orientation.usesWidthAsHeight())
         result->m_sizeRespectingOrientation = result->m_size.transposedSize();
     return result.release();
 }
 
 BitmapImage::BitmapImage(ImageObserver* observer)
     : Image(observer)
     , m_currentFrame(0)
     , m_repetitionCount(cAnimationNone)
     , m_repetitionCountStatus(Unknown)
     , m_repetitionsComplete(0)
     , m_desiredFrameStartTime(0)
     , m_frameCount(0)
     , m_animationPolicy(ImageAnimationPolicyAllowed)
     , m_animationFinished(false)
     , m_allDataReceived(false)
     , m_haveSize(false)
     , m_sizeAvailable(false)
     , m_hasUniformFrameSize(true)
     , m_haveFrameCount(false)
+    , m_drawingToCanvasElement(false)
 {
 }
 
 BitmapImage::BitmapImage(const SkBitmap& bitmap, ImageObserver* observer)
     : Image(observer)
     , m_size(bitmap.width(), bitmap.height())
     , m_currentFrame(0)
     , m_repetitionCount(cAnimationNone)
     , m_repetitionCountStatus(Unknown)
     , m_repetitionsComplete(0)
     , m_frameCount(1)
     , m_animationPolicy(ImageAnimationPolicyAllowed)
     , m_animationFinished(true)
     , m_allDataReceived(true)
     , m_haveSize(true)
     , m_sizeAvailable(true)
     , m_haveFrameCount(true)
+    , m_drawingToCanvasElement(false)
 {
     // Since we don't have a decoder, we can't figure out the image orientation.
     // Set m_sizeRespectingOrientation to be the same as m_size so it's not 0x0.
     m_sizeRespectingOrientation = m_size;
 
     m_frames.grow(1);
     m_frames[0].m_hasAlpha = !bitmap.isOpaque();
     m_frames[0].m_frame = adoptRef(SkImage::NewFromBitmap(bitmap));
     m_frames[0].m_haveMetadata = true;
 }
 
 BitmapImage::~BitmapImage()
 {
     stopAnimation();
 }
 
-bool BitmapImage::isBitmapImage() const
-{
-    return true;
-}
-
-bool BitmapImage::currentFrameHasSingleSecurityOrigin() const
-{
-    return true;
-}
-
 int BitmapImage::totalFrameBytes()
 {
     const size_t numFrames = frameCount();
     size_t totalBytes = 0;
     for (size_t i = 0; i < numFrames; ++i)
         totalBytes += m_source.frameBytesAtIndex(i);
     return safeCast<int>(totalBytes);
 }
 
 void BitmapImage::destroyDecodedData(bool destroyAll)
@@ skipping 10 matching lines @@
 
 void BitmapImage::destroyDecodedDataIfNecessary()
 {
     // Animated images >5MB are considered large enough that we'll only hang on
     // to one frame at a time.
     static const size_t cLargeAnimationCutoff = 5242880;
     size_t allFrameBytes = 0;
     for (size_t i = 0; i < m_frames.size(); ++i)
         allFrameBytes += m_frames[i].m_frameBytes;
 
-    if (allFrameBytes > cLargeAnimationCutoff) {
+    if (allFrameBytes > cLargeAnimationCutoff)
         destroyDecodedData(false);
-    }
 }
 
 void BitmapImage::destroyMetadataAndNotify(size_t frameBytesCleared)
 {
     if (frameBytesCleared && imageObserver())
         imageObserver()->decodedSizeChanged(this, -safeCast<int>(frameBytesCleared));
 }
 
 void BitmapImage::cacheFrame(size_t index)
 {
     size_t numFrames = frameCount();
     if (m_frames.size() < numFrames)
         m_frames.grow(numFrames);
 
     int deltaBytes = totalFrameBytes();
 
-
     // We are caching frame snapshots.  This is OK even for partially decoded frames,
     // as they are cleared by dataChanged() when new data arrives.
     m_frames[index].m_frame = m_source.createFrameAtIndex(index);
 
     m_frames[index].m_orientation = m_source.orientationAtIndex(index);
     m_frames[index].m_haveMetadata = true;
     m_frames[index].m_isComplete = m_source.frameIsCompleteAtIndex(index);
     if (repetitionCount(false) != cAnimationNone)
         m_frames[index].m_duration = m_source.frameDurationAtIndex(index);
     m_frames[index].m_hasAlpha = m_source.frameHasAlphaAtIndex(index);
@@ skipping 87 matching lines @@
 
 String BitmapImage::filenameExtension() const
 {
     return m_source.filenameExtension();
 }
 
 void BitmapImage::draw(SkCanvas* canvas, const SkPaint& paint, const FloatRect& dstRect, const FloatRect& srcRect, RespectImageOrientationEnum shouldRespectImageOrientation, ImageClampingMode clampMode)
 {
     TRACE_EVENT0("skia", "BitmapImage::draw");
 
-    RefPtr<SkImage> image = imageForCurrentFrame();
-    if (!image)
-        return; // It's too early and we don't have an image yet.
+    RefPtr<SkImage> skImage = imageForCurrentFrame();
+    if (!skImage)
+        return; // Bail: we don't have an image yet.
 
     FloatRect adjustedSrcRect = srcRect;
-    adjustedSrcRect.intersect(FloatRect(0, 0, image->width(), image->height()));
-
+    adjustedSrcRect.intersect(FloatRect(0, 0, skImage->width(), skImage->height()));
     if (adjustedSrcRect.isEmpty() || dstRect.isEmpty())
         return; // Nothing to draw.
+    FloatRect adjustedDstRect = dstRect;
 
     ImageOrientation orientation = DefaultImageOrientation;
     if (shouldRespectImageOrientation == RespectImageOrientation)
         orientation = frameOrientationAtIndex(m_currentFrame);
 
+    SkCanvas::SrcRectConstraint srcRectConstraint = WebCoreClampingModeToSkiaRectConstraint(clampMode);
+
     int initialSaveCount = canvas->getSaveCount();
-    FloatRect adjustedDstRect = dstRect;
+
     if (orientation != DefaultImageOrientation) {
         canvas->save();
 
         // ImageOrientation expects the origin to be at (0, 0)
         canvas->translate(adjustedDstRect.x(), adjustedDstRect.y());
         adjustedDstRect.setLocation(FloatPoint());
 
         canvas->concat(affineTransformToSkMatrix(orientation.transformFromDefault(adjustedDstRect.size())));
 
         if (orientation.usesWidthAsHeight()) {
             // The destination rect will have it's width and height already reversed for the orientation of
             // the image, as it was needed for page layout, so we need to reverse it back here.
             adjustedDstRect = FloatRect(adjustedDstRect.x(), adjustedDstRect.y(), adjustedDstRect.height(), adjustedDstRect.width());
         }
     }
 
-    SkRect skSrcRect = adjustedSrcRect;
-    canvas->drawImageRect(image.get(), skSrcRect, adjustedDstRect, &paint,
-        WebCoreClampingModeToSkiaRectConstraint(clampMode));
+    RefPtr<SkColorFilter> transform;
+    if (imageColorProfilesEnabled() && hasColorProfile())
+        transform = imageColorTransform();
+
+    if (!transform) {
+        canvas->drawImageRect(skImage.get(), adjustedSrcRect, adjustedDstRect, &paint, srcRectConstraint);
+
+        canvas->restoreToCount(initialSaveCount);
+        if (skImage->isLazyGenerated())
+            PlatformInstrumentation::didDrawLazyPixelRef(skImage->uniqueID());
+        this->didDraw();
+        return;
+    }
+
+    float srcArea = adjustedSrcRect.width() * adjustedSrcRect.height();
+    float dstArea = adjustedDstRect.width() * adjustedDstRect.height();
+
+    if (dstArea <= srcArea) {
+        SkPaint transformPaint = paint;
+        transformPaint.setColorFilter(transform.get());
+        canvas->drawImageRect(skImage.get(), adjustedSrcRect, adjustedDstRect, &transformPaint, srcRectConstraint);
+
+        canvas->restoreToCount(initialSaveCount);
+        if (skImage->isLazyGenerated())
+            PlatformInstrumentation::didDrawLazyPixelRef(skImage->uniqueID());
+        this->didDraw();
+        return;
+    }
+
+    RefPtr<SkImage> image = skImage; // skImage might be indirectly drawn here (due to transforms).
+
+    if (RefPtr<SkImageFilter> imageFilter = adoptRef(SkColorFilterImageFilter::Create(transform.get())))
+        image = adoptRef(image->applyFilter(imageFilter.get(), nullptr, true));
+
+    canvas->drawImageRect(image.get(), adjustedSrcRect, adjustedDstRect, &paint, srcRectConstraint);
+
     canvas->restoreToCount(initialSaveCount);
+    if (skImage->isLazyGenerated())
+        PlatformInstrumentation::didDrawLazyPixelRef(skImage->uniqueID());
+    this->didDraw();
+    return;
+}
 
-    if (currentFrameIsLazyDecoded())
-        PlatformInstrumentation::didDrawLazyPixelRef(image->uniqueID());
-
+void BitmapImage::didDraw()
+{
     if (ImageObserver* observer = imageObserver())
         observer->didDraw(this);
 
     startAnimation();
 }
 
 size_t BitmapImage::frameCount()
 {
     if (!m_haveFrameCount) {
         m_frameCount = m_source.frameCount();
@@ skipping 69 matching lines @@
 {
     if (frameCount() > 1) {
         RefPtr<SkImage> firstFrame = frameAtIndex(0);
         if (firstFrame)
             return StaticBitmapImage::create(firstFrame);
     }
 
     return Image::imageForDefaultFrame();
 }
 
+PassRefPtr<SkColorFilter> BitmapImage::imageColorTransform()
+{
+    if (currentScreenId()) // FIXME: remove these <canvas> code asserts.
+        RELEASE_ASSERT(!drawingToCanvasElement());
+    else
+        RELEASE_ASSERT(drawingToCanvasElement()); // sRGB render target.
+
+    return createColorSpaceFilter(colorProfile().get(), screenColorProfile(currentScreenId()).get());
+}
+
 bool BitmapImage::frameHasAlphaAtIndex(size_t index)
 {
     if (m_frames.size() <= index)
         return true;
 
     if (m_frames[index].m_haveMetadata)
         return m_frames[index].m_hasAlpha;
 
     return m_source.frameHasAlphaAtIndex(index);
 }
 
 bool BitmapImage::currentFrameKnownToBeOpaque(MetadataMode metadataMode)
 {
     if (metadataMode == PreCacheMetadata) {
         // frameHasAlphaAtIndex() conservatively returns false for uncached frames. To increase the
         // chance of an accurate answer, pre-cache the current frame metadata.
         frameAtIndex(currentFrame());
     }
+
     return !frameHasAlphaAtIndex(currentFrame());
 }
 
 bool BitmapImage::currentFrameIsComplete()
 {
     return frameIsCompleteAtIndex(currentFrame());
 }
 
 bool BitmapImage::currentFrameIsLazyDecoded()
 {
@@ skipping 213 matching lines @@
     destroyDecodedDataIfNecessary();
 
     // We need to draw this frame if we advanced to it while not skipping, or if
     // while trying to skip frames we hit the last frame and thus had to stop.
     if (skippingFrames != advancedAnimation)
         imageObserver()->animationAdvanced(this);
     return advancedAnimation;
 }
 
 } // namespace blink