[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/core/SkImageGenerator.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()) {
         return BitmapImage::create();
     }
 
@@ skipping 45 matching lines @@
     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)
@@ skipping 138 matching lines @@
 bool BitmapImage::hasColorProfile() const
 {
     return m_source.hasColorProfile();
 }
 
 String BitmapImage::filenameExtension() const
 {
     return m_source.filenameExtension();
 }
 
+PassRefPtr<SkImage> BitmapImage::pictureForCurrentFrame()
+{
+    RELEASE_ASSERT(currentFrame() < m_frames.size() && m_frames[currentFrame()].m_frame);
+
+    FrameData& frame = m_frames[currentFrame()];
+
+    RELEASE_ASSERT(currentScreenId()); // There should be an active graphics screen.
+
+    RefPtr<SkColorFilter> colorTransform;
+    if (imageColorProfilesEnabled()) {
+        RefPtr<ColorSpaceProfile> source = colorProfile();
+        if (!source)
+            source = screenColorProfile(ScreenDevice::sRGB);
+        colorTransform = createColorSpaceFilter(source.get(), screenColorProfile(currentScreenId()).get());
+    }
+
+    if (!colorTransform)
+        return frame.m_frame;
+
+    RefPtr<SkImage> cachedImage = frame.find(colorTransform.get());
+    if (cachedImage)
+        return cachedImage;
+
+    RefPtr<SkImage> skImage = frame.m_frame;
+    SkPictureRecorder recorder;
+    SkRect bounds = SkRect::MakeIWH(skImage->width(), skImage->height());
+    SkCanvas* canvas = recorder.beginRecording(bounds, nullptr, 0);
+    SkPaint paint;
+    paint.setColorFilter(colorTransform.get());
+    canvas->drawImage(skImage.get(), 0, 0, &paint);
+
+    RefPtr<SkPicture> picture = adoptRef(recorder.endRecording());
+    if (!picture)
+        return skImage;
+
+    SkISize skISize = SkISize::Make(skImage->width(), skImage->height());
+    skImage = adoptRef(SkImage::NewFromPicture(picture.get(), skISize, nullptr, nullptr));
+
+    // FIXME: Should we not cache if (currentScreenId() == ScreenDevice::sRGBCanvas)?
+    frame.cache(colorTransform.get(), skImage.get());
+    return skImage;
+}
+
 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.
+
+    skImage = pictureForCurrentFrame();
+    if (!skImage)
+        return;
 
     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));
+    canvas->drawImageRect(skImage.get(), adjustedSrcRect, adjustedDstRect, &paint, srcRectConstraint);
+
     canvas->restoreToCount(initialSaveCount);
-
-    if (currentFrameIsLazyDecoded())
-        PlatformInstrumentation::didDrawLazyPixelRef(image->uniqueID());
+    if (skImage->isLazyGenerated())
+        PlatformInstrumentation::didDrawLazyPixelRef(skImage->uniqueID());
 
     if (ImageObserver* observer = imageObserver())
         observer->didDraw(this);
 
     startAnimation();
 }
 
 size_t BitmapImage::frameCount()
 {
     if (!m_haveFrameCount) {
@@ skipping 71 matching lines @@
 {
     if (frameCount() > 1) {
         RefPtr<SkImage> firstFrame = frameAtIndex(0);
         if (firstFrame)
             return StaticBitmapImage::create(firstFrame);
     }
 
     return Image::imageForDefaultFrame();
 }
 
+PassRefPtr<SkColorFilter> BitmapImage::imageColorTransform()
+{
+    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);
 }
@@ skipping 231 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