Make a bitmaprenderer canvas an ImageSource

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

 /*
  * Copyright (C) 2004, 2006, 2007 Apple Inc. All rights reserved.
  * Copyright (C) 2007 Alp Toker <alp@atoker.com>
  * Copyright (C) 2010 Torch Mobile (Beijing) Co. Ltd. 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.
@@ skipping 1106 matching lines @@
 }
 
 PassRefPtr<Image> HTMLCanvasElement::copiedImage(
     SourceDrawingBuffer sourceBuffer,
     AccelerationHint hint) const {
   if (!isPaintable())
     return nullptr;
   if (!m_context)
     return createTransparentImage(size());
 
+  if (m_context->getContextType() ==
+      CanvasRenderingContext::ContextImageBitmap) {
+    RefPtr<Image> image =
+        m_context->getImage(hint, SnapshotReasonGetCopiedImage);
+    if (image)
+      return m_context->getImage(hint, SnapshotReasonGetCopiedImage);
+    // Special case: transferFromImageBitmap is not yet called.
+    sk_sp<SkSurface> surface =
+        SkSurface::MakeRasterN32Premul(width(), height());
+    return StaticBitmapImage::create(surface->makeImageSnapshot());
+  }
+
   bool needToUpdate = !m_copiedImage;
   // The concept of SourceDrawingBuffer is valid on only WebGL.
   if (m_context->is3d())
     needToUpdate |= m_context->paintRenderingResultsToCanvas(sourceBuffer);
   if (needToUpdate && buffer()) {
     m_copiedImage =
         buffer()->newImageSnapshot(hint, SnapshotReasonGetCopiedImage);
     updateExternallyAllocatedMemory();
   }
   return m_copiedImage;
@@ skipping 61 matching lines @@
   if (!isPaintable()) {
     *status = InvalidSourceImageStatus;
     return nullptr;
   }
 
   if (!m_context) {
     *status = NormalSourceImageStatus;
     return createTransparentImage(size());
   }
 
+  if (m_context->getContextType() == CanvasRenderingContext::ContextImageBitmap)
+    return m_context->getImage(hint, reason);
+
   sk_sp<SkImage> skImage;
   if (m_context->is3d()) {
     // Because WebGL sources always require making a copy of the back buffer, we
     // use paintRenderingResultsToCanvas instead of getImage in order to keep a
     // cached copy of the backing in the canvas's ImageBuffer.
     renderingContext()->paintRenderingResultsToCanvas(BackBuffer);
     skImage = hasImageBuffer()
                   ? buffer()->newSkImageSnapshot(hint, reason)
                   : createTransparentImage(size())->imageForCurrentFrame();
   } else {
@@ skipping 15 matching lines @@
 
   *status = InvalidSourceImageStatus;
   return nullptr;
 }
 
 bool HTMLCanvasElement::wouldTaintOrigin(SecurityOrigin*) const {
   return !originClean();
 }
 
 FloatSize HTMLCanvasElement::elementSize(const FloatSize&) const {
+  if (m_context &&
+      m_context->getContextType() ==
+          CanvasRenderingContext::ContextImageBitmap) {
+    RefPtr<Image> image =
+        m_context->getImage(PreferNoAcceleration, SnapshotReasonDrawImage);
+    if (image)
+      return FloatSize(image->width(), image->height());
+    return FloatSize(0, 0);
+  }
   return FloatSize(width(), height());
 }
 
 IntSize HTMLCanvasElement::bitmapSourceSize() const {
   return IntSize(width(), height());
 }
 
 ScriptPromise HTMLCanvasElement::createImageBitmap(
     ScriptState* scriptState,
     EventTarget& eventTarget,
@@ skipping 104 matching lines @@
   mojom::blink::OffscreenCanvasSurfacePtr service;
   Platform::current()->interfaceProvider()->getInterface(
       mojo::GetProxy(&service));
   m_surfaceLayerBridge =
       wrapUnique(new CanvasSurfaceLayerBridge(std::move(service)));
   return m_surfaceLayerBridge->createSurfaceLayer(this->width(),
                                                   this->height());
 }
 
 }  // namespace blink