Refactor ImageBuffer to make OffscreenCanvas match HTMLCanvasElement

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

 /*
  * Copyright (C) 2004, 2006, 2009, 2010 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 122 matching lines @@
   // Used for canvas capture.
   void AddListener(CanvasDrawListener*);
   void RemoveListener(CanvasDrawListener*);
 
   // Used for rendering
   void DidDraw(const FloatRect&) override;
   void DidDraw() override;
 
   void Paint(GraphicsContext&, const LayoutRect&);
 
-  PaintCanvas* DrawingCanvas() const;
-  void DisableDeferral(DisableDeferralReason) const;
+  PaintCanvas* DrawingCanvas();
+  void DisableDeferral(DisableDeferralReason);
   PaintCanvas* ExistingDrawingCanvas() const;
 
   CanvasRenderingContext* RenderingContext() const { return context_.Get(); }
 
   void EnsureUnacceleratedImageBuffer();
-  ImageBuffer* Buffer() const;
   PassRefPtr<Image> CopiedImage(SourceDrawingBuffer,
                                 AccelerationHint,
-                                SnapshotReason) const;
+                                SnapshotReason);
   void ClearCopiedImage();
 
   bool OriginClean() const;
   void SetOriginTainted() { origin_clean_ = false; }
 
   AffineTransform BaseTransform() const;
 
   bool Is3d() const;
   bool Is2d() const;
   bool IsAnimated2d() const;
 
-  bool HasImageBuffer() const { return image_buffer_.get(); }
-  void DiscardImageBuffer();
+  void DiscardImageBuffer() override;
+  ImageBuffer* GetImageBuffer() const override { return image_buffer_.get(); }
+  ImageBuffer* GetOrCreateImageBuffer() override;
 
   bool ShouldBeDirectComposited() const;
 
-  void PrepareSurfaceForPaintingIfNeeded() const;
+  void PrepareSurfaceForPaintingIfNeeded();
 
   const AtomicString ImageSourceURL() const override;
 
   InsertionNotificationRequest InsertedInto(ContainerNode*) override;
 
   bool IsDirty() { return !dirty_rect_.IsEmpty(); }
 
   void DoDeferredPaintInvalidation();
 
   void FinalizeFrame() override;
 
   // ContextLifecycleObserver and PageVisibilityObserver implementation
   void ContextDestroyed(ExecutionContext*) override;
 
   // PageVisibilityObserver implementation
   void PageVisibilityChanged() override;
 
   // CanvasImageSource implementation
   PassRefPtr<Image> GetSourceImageForCanvas(SourceImageStatus*,
                                             AccelerationHint,
                                             SnapshotReason,
-                                            const FloatSize&) const override;
+                                            const FloatSize&) override;
   bool WouldTaintOrigin(SecurityOrigin*) const override;
   FloatSize ElementSize(const FloatSize&) const override;
   bool IsCanvasElement() const override { return true; }
   bool IsOpaque() const override;
   bool IsAccelerated() const override;
   int SourceWidth() override { return size_.Width(); }
   int SourceHeight() override { return size_.Height(); }
 
   // CanvasSurfaceLayerBridgeObserver implementation
   void OnWebLayerReplaced() override;
@@ skipping 133 matching lines @@
   std::unique_ptr<CanvasSurfaceLayerBridge> surface_layer_bridge_;
 
   int num_frames_since_last_rendering_mode_switch_;
   bool pending_rendering_mode_switch_;
   bool did_notify_listeners_for_current_frame_ = false;
 };
 
 }  // namespace blink
 
 #endif  // HTMLCanvasElement_h