[poc] curve-filter

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

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "platform/graphics/RecordingImageBufferSurface.h"
 
 #include "platform/graphics/CanvasMetrics.h"
 #include "platform/graphics/ExpensiveCanvasHeuristicParameters.h"
 #include "platform/graphics/GraphicsContext.h"
 #include "platform/graphics/ImageBuffer.h"
+#include "platform/graphics/StaticBitmapImage.h"
 #include "public/platform/Platform.h"
 #include "third_party/skia/include/core/SkCanvas.h"
+#include "third_party/skia/include/core/SkColorFilter.h"
 #include "third_party/skia/include/core/SkPictureRecorder.h"
 #include "wtf/PassOwnPtr.h"
 #include "wtf/PassRefPtr.h"
 
 namespace blink {
 
 RecordingImageBufferSurface::RecordingImageBufferSurface(const IntSize& size, PassOwnPtr<RecordingImageBufferFallbackSurfaceFactory> fallbackFactory, OpacityMode opacityMode)
     : ImageBufferSurface(size, opacityMode)
     , m_imageBuffer(0)
     , m_currentFramePixelCount(0)
@@ skipping 173 matching lines @@
     m_previousFrame = adoptRef(m_currentFrame->endRecording());
     m_previousFrameHasExpensiveOp = m_currentFrameHasExpensiveOp;
     m_previousFramePixelCount = m_currentFramePixelCount;
     if (!initializeCurrentFrame())
         return false;
 
     m_frameWasCleared = false;
     return true;
 }
 
-void RecordingImageBufferSurface::draw(GraphicsContext& context, const FloatRect& destRect, const FloatRect& srcRect, SkXfermode::Mode op)
+void RecordingImageBufferSurface::draw(GraphicsContext& context, const FloatRect& destRect, const FloatRect& srcRect, SkXfermode::Mode op, SkColorFilter* transform)
 {
     if (m_fallbackSurface) {
-        m_fallbackSurface->draw(context, destRect, srcRect, op);
+        m_fallbackSurface->draw(context, destRect, srcRect, op, transform);
         return;
     }
 
     RefPtr<SkPicture> picture = getPicture();
-    if (picture) {
-        context.compositePicture(picture.get(), destRect, srcRect, op);
-    } else {
-        ImageBufferSurface::draw(context, destRect, srcRect, op);
+    if (!picture) {
+        ImageBufferSurface::draw(context, destRect, srcRect, op, transform);
+        return;
     }
+
+    const SkISize size = SkISize::Make(width(), height());
+    RefPtr<SkImage> snapshot = adoptRef(SkImage::NewFromPicture(picture.get(), size, nullptr, nullptr));
+
+    // FIXME: why was this alternative path being used to draw the picture? Does it draw
+    // differently to the context.drawImage path that other surfaces use to draw?
+    // context.compositePicture(picture.get(), destRect, srcRect, op);
+
+    if (!transform) {
+        RefPtr<Image> image = StaticBitmapImage::create(snapshot.release());
+        context.drawImage(image.get(), destRect, srcRect, op);
+        return;
+    }
+
+    SkPaint transformPaint;
+    transformPaint.setColorFilter(transform);
+    transformPaint.setXfermodeMode(SkXfermode::kSrc_Mode);
+
+    SkPictureRecorder recorder;
+    FloatRect sourceRect = IntRect(0, 0, width(), height());
+    SkCanvas* canvas = recorder.beginRecording(sourceRect);
+    canvas->drawImage(snapshot.get(), 0, 0, &transformPaint);
+    RefPtr<SkPicture> recorded = adoptRef(recorder.endRecordingAsPicture());
+
+    snapshot = adoptRef(SkImage::NewFromPicture(recorded.get(), size, nullptr, nullptr));
+
+    RefPtr<Image> image = StaticBitmapImage::create(snapshot.release());
+    context.drawImage(image.get(), destRect, srcRect, op);
 }
 
 bool RecordingImageBufferSurface::isExpensiveToPaint()
 {
     if (m_fallbackSurface)
         return m_fallbackSurface->isExpensiveToPaint();
 
     if (m_didRecordDrawCommandsInCurrentFrame) {
         if (m_currentFrameHasExpensiveOp)
             return true;
@@ skipping 41 matching lines @@
 
 void RecordingImageBufferSurface::setIsHidden(bool hidden)
 {
     if (m_fallbackSurface)
         m_fallbackSurface->setIsHidden(hidden);
     else
         ImageBufferSurface::setIsHidden(hidden);
 }
 
 } // namespace blink