[poc] curve-filter

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

 /*
  * Copyright (C) 2007, 2008, 2009, 2010 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
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 23 matching lines @@
 #include "core/dom/Fullscreen.h"
 #include "core/dom/shadow/ShadowRoot.h"
 #include "core/frame/ImageBitmap.h"
 #include "core/frame/Settings.h"
 #include "core/html/parser/HTMLParserIdioms.h"
 #include "core/imagebitmap/ImageBitmapOptions.h"
 #include "core/layout/LayoutImage.h"
 #include "core/layout/LayoutVideo.h"
 #include "platform/RuntimeEnabledFeatures.h"
 #include "platform/UserGestureIndicator.h"
+#include "platform/graphics/ColorSpaceFilter.h"
+#include "platform/graphics/ColorSpaceProfile.h"
 #include "platform/graphics/GraphicsContext.h"
+#include "platform/graphics/GraphicsScreen.h"
 #include "platform/graphics/ImageBuffer.h"
 #include "platform/graphics/gpu/Extensions3DUtil.h"
 #include "public/platform/WebCanvas.h"
 #include "public/platform/WebGraphicsContext3D.h"
 
 namespace blink {
 
 using namespace HTMLNames;
 
 inline HTMLVideoElement::HTMLVideoElement(Document& document)
@@ skipping 132 matching lines @@
 }
 
 void HTMLVideoElement::updateDisplayState()
 {
     if (posterImageURL().isEmpty())
         setDisplayMode(Video);
     else if (displayMode() < Poster)
         setDisplayMode(Poster);
 }
 
-void HTMLVideoElement::paintCurrentFrame(SkCanvas* canvas, const IntRect& destRect, const SkPaint* paint) const
+void HTMLVideoElement::paintCurrentFrame(SkCanvas* canvas, const IntRect& destRect, const SkPaint* paint, ColorSpaceProfile* target) const
 {
     if (!webMediaPlayer())
         return;
 
     SkXfermode::Mode mode;
     if (!paint || !SkXfermode::AsMode(paint->getXfermode(), &mode))
         mode = SkXfermode::kSrcOver_Mode;
 
+    // FIXME: read the video color space from the video frame and use it as the source color profile
+    // rather than assuming an sRGB source as is done here.
+    RefPtr<SkColorFilter> colorTransform;
+    if (RefPtr<ColorSpaceProfile> targetProfile = target) {
+        RefPtr<ColorSpaceProfile> sourceProfile = screenColorProfile(ScreenDevice::sRGB);
+        colorTransform = createColorSpaceFilter(sourceProfile.get(), targetProfile.get());
+    }
+
+    SkPictureRecorder recorder;
+    IntRect sourceRect = IntRect(0, 0, videoWidth(), videoHeight());
+    SkCanvas* recordingCanvas = recorder.beginRecording(sourceRect);
     // TODO(junov, philipj): crbug.com/456529 Pass the whole SkPaint instead of only alpha and xfermode
-    webMediaPlayer()->paint(canvas, destRect, paint ? paint->getAlpha() : 0xFF, mode);
+    // and maybe correctly handle the video source color profile as well.
+    webMediaPlayer()->paint(recordingCanvas, sourceRect, 0xFF, SkXfermode::kSrc_Mode);
+    RefPtr<SkPicture> recorded = adoptRef(recorder.endRecording());
+
+    SkPaint transformPaint;
+    transformPaint.setColorFilter(colorTransform.get());
+    transformPaint.setAlpha(paint ? paint->getAlpha() : 0xFF);
+    transformPaint.setXfermodeMode(mode);
+    transformPaint.setFilterQuality(kLow_SkFilterQuality);
+
+    SkISize size = SkISize::Make(videoWidth(), videoHeight());
+    RefPtr<SkImage> image = adoptRef(SkImage::NewFromPicture(recorded.get(), size, nullptr, nullptr));
+    canvas->drawImageRect(image.get(), SkIRect::MakeSize(size), destRect, &transformPaint);
 }
 
 bool HTMLVideoElement::copyVideoTextureToPlatformTexture(WebGraphicsContext3D* context, Platform3DObject texture, GLenum internalFormat, GLenum type, bool premultiplyAlpha, bool flipY)
 {
     if (!webMediaPlayer())
         return false;
 
     ASSERT(Extensions3DUtil::canUseCopyTextureCHROMIUM(GL_TEXTURE_2D, internalFormat, type, 0));
     return webMediaPlayer()->copyVideoTextureToPlatformTexture(context, texture, internalFormat, type, premultiplyAlpha, flipY);
 }
@@ skipping 75 matching lines @@
     }
 
     IntSize intrinsicSize(videoWidth(), videoHeight());
     // FIXME: Not sure if we dhould we be doing anything with the AccelerationHint argument here?
     OwnPtr<ImageBuffer> imageBuffer = ImageBuffer::create(intrinsicSize);
     if (!imageBuffer) {
         *status = InvalidSourceImageStatus;
         return nullptr;
     }
 
-    paintCurrentFrame(imageBuffer->canvas(), IntRect(IntPoint(0, 0), intrinsicSize), nullptr);
+    paintCurrentFrame(imageBuffer->canvas(), IntRect(IntPoint(0, 0), intrinsicSize), nullptr, nullptr);
     RefPtr<Image> snapshot = imageBuffer->newImageSnapshot();
     if (!snapshot) {
         *status = InvalidSourceImageStatus;
         return nullptr;
     }
 
     *status = NormalSourceImageStatus;
     return snapshot.release();
 }
 
@@ skipping 24 matching lines @@
         return ScriptPromise();
     }
     if (!sw || !sh) {
         exceptionState.throwDOMException(IndexSizeError, String::format("The source %s provided is 0.", sw ? "height" : "width"));
         return ScriptPromise();
     }
     return ImageBitmapSource::fulfillImageBitmap(scriptState, ImageBitmap::create(this, IntRect(sx, sy, sw, sh), eventTarget.toDOMWindow()->document(), options));
 }
 
 } // namespace blink