Use SkImage as a backing store for copying 2d Contexts to ImageBitmaps.

Source/core/page/ImageBitmap.cpp

 // Copyright 2013 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 "config.h"
 #include "core/page/ImageBitmap.h"
 
 #include "core/html/HTMLCanvasElement.h"
 #include "core/html/HTMLImageElement.h"
 #include "core/html/HTMLVideoElement.h"
@@ skipping 17 matching lines @@
 static inline PassRefPtr<Image> cropImage(Image* image, const IntRect& cropRect)
 {
     SkBitmap cropped;
     image->nativeImageForCurrentFrame()->bitmap().extractSubset(&cropped, cropRect);
     return BitmapImage::create(NativeImageSkia::create(cropped));
 }
 
 ImageBitmap::ImageBitmap(HTMLImageElement* image, const IntRect& cropRect)
     : m_cropRect(cropRect)
     , m_imageElement(image)
+    , m_bitmap(0)
+    , m_derivedFromCanvas(false)
 {
     m_imageElement->addClient(this);
 
     IntRect srcRect = intersection(cropRect, IntRect(0, 0, image->width(), image->height()));
     m_bitmapRect = IntRect(IntPoint(max(0, -cropRect.x()), max(0, -cropRect.y())), srcRect.size());
     m_bitmapOffset = srcRect.location();
 
     ScriptWrappable::init(this);
 }
 
 ImageBitmap::ImageBitmap(HTMLVideoElement* video, const IntRect& cropRect)
     : m_cropRect(cropRect)
     , m_imageElement(0)
     , m_bitmapOffset(IntPoint())
+    , m_derivedFromCanvas(false)
 {
     IntRect videoRect = IntRect(IntPoint(), video->player()->naturalSize());
     IntRect srcRect = intersection(cropRect, videoRect);
     IntRect dstRect(IntPoint(), srcRect.size());
 
     m_buffer = ImageBuffer::create(videoRect.size());
     GraphicsContext* c = m_buffer->context();
     c->clip(dstRect);
     c->translate(-srcRect.x(), -srcRect.y());
     video->paintCurrentFrameInContext(c, videoRect);
     m_bitmap = m_buffer->copyImage(DontCopyBackingStore);
     m_bitmapRect = IntRect(IntPoint(max(0, -cropRect.x()), max(0, -cropRect.y())), srcRect.size());
 
     ScriptWrappable::init(this);
 }
 
 ImageBitmap::ImageBitmap(HTMLCanvasElement* canvas, const IntRect& cropRect)
     : m_cropRect(cropRect)
     , m_imageElement(0)
+    , m_bitmap(canvas->buffer()->imageSnapshot())
     , m_bitmapOffset(IntPoint())
+    , m_derivedFromCanvas(true)
 {
-    IntSize canvasSize = canvas->size();
-    IntRect srcRect = intersection(cropRect, IntRect(IntPoint(), canvasSize));
-    IntRect dstRect(IntPoint(), srcRect.size());
-
-    m_buffer = ImageBuffer::create(canvasSize);
-    m_buffer->context()->drawImageBuffer(canvas->buffer(), dstRect, srcRect);
-    m_bitmap = m_buffer->copyImage(DontCopyBackingStore);
+    IntRect srcRect = intersection(cropRect, IntRect(IntPoint(), canvas->size()));
     m_bitmapRect = IntRect(IntPoint(max(0, -cropRect.x()), max(0, -cropRect.y())), srcRect.size());
+    m_bitmapOffset = srcRect.location();
 
     ScriptWrappable::init(this);
 }
 
 ImageBitmap::ImageBitmap(ImageData* data, const IntRect& cropRect)
     : m_cropRect(cropRect)
     , m_imageElement(0)
     , m_bitmapOffset(IntPoint())
+    , m_derivedFromCanvas(false)
 {
     IntRect srcRect = intersection(cropRect, IntRect(IntPoint(), data->size()));
 
     m_buffer = ImageBuffer::create(data->size());
     if (srcRect.width() > 0 && srcRect.height() > 0)
         m_buffer->putByteArray(Unmultiplied, data->data(), data->size(), srcRect, IntPoint(min(0, -cropRect.x()), min(0, -cropRect.y())));
 
     m_bitmap = m_buffer->copyImage(DontCopyBackingStore);
     m_bitmapRect = IntRect(IntPoint(max(0, -cropRect.x()), max(0, -cropRect.y())),  srcRect.size());
 
     ScriptWrappable::init(this);
 }
 
 ImageBitmap::ImageBitmap(ImageBitmap* bitmap, const IntRect& cropRect)
     : m_cropRect(cropRect)
     , m_imageElement(bitmap->imageElement())
     , m_bitmapOffset(IntPoint())
+    , m_derivedFromCanvas(bitmap->derivedFromCanvas())
 {
     IntRect oldBitmapRect = bitmap->bitmapRect();
     IntRect srcRect = intersection(cropRect, oldBitmapRect);
     m_bitmapRect = IntRect(IntPoint(max(0, oldBitmapRect.x() - cropRect.x()), max(0, oldBitmapRect.y() - cropRect.y())), srcRect.size());
 
     if (m_imageElement) {
+        m_bitmap = 0;
         m_imageElement->addClient(this);
-        m_bitmap = 0;
+        m_bitmapOffset = srcRect.location();
+    } else if (m_derivedFromCanvas) {
+        m_bitmap = bitmap->bitmapImage();
         m_bitmapOffset = srcRect.location();
     } else {
-        IntRect adjustedCropRect(IntPoint(cropRect.x() -oldBitmapRect.x(), cropRect.y() - oldBitmapRect.y()), cropRect.size());
+        IntRect adjustedCropRect(IntPoint(cropRect.x() - oldBitmapRect.x(), cropRect.y() - oldBitmapRect.y()), cropRect.size());
         m_bitmap = cropImage(bitmap->bitmapImage().get(), adjustedCropRect);
     }
+
     ScriptWrappable::init(this);
 }
 
 ImageBitmap::~ImageBitmap()
 {
     if (m_imageElement)
         m_imageElement->removeClient(this);
 }
 
 PassRefPtr<ImageBitmap> ImageBitmap::create(HTMLImageElement* image, const IntRect& cropRect)
@@ skipping 34 matching lines @@
 void ImageBitmap::notifyImageSourceChanged()
 {
     m_bitmap = cropImage(m_imageElement->cachedImage()->image(), m_cropRect);
     m_bitmapOffset = IntPoint();
     m_imageElement = 0;
 }
 
 PassRefPtr<Image> ImageBitmap::bitmapImage() const
 {
     ASSERT((m_imageElement || m_bitmap) && (!m_imageElement || !m_bitmap));
+    ASSERT((m_imageElement || m_derivedFromCanvas) && (!m_imageElement || !m_derivedFromCanvas));
+
     if (m_imageElement)
         return m_imageElement->cachedImage()->image();
     return m_bitmap;
 }
 
 }