Integrating SkSurface with SkDeferredCanvas

src/image/SkSurface_Raster.cpp

 /*
  * Copyright 2012 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "SkSurface_Base.h"
 #include "SkImagePriv.h"
 #include "SkCanvas.h"
 #include "SkDevice.h"
 #include "SkMallocPixelRef.h"
 
 static const size_t kIgnoreRowBytesValue = (size_t)~0;
 
 class SkSurface_Raster : public SkSurface_Base {
 public:
     static bool Valid(const SkImage::Info&, size_t rb = kIgnoreRowBytesValue);
 
     SkSurface_Raster(const SkImage::Info&, void*, size_t rb);
     SkSurface_Raster(const SkImage::Info&, SkPixelRef*, size_t rb);
 
     virtual SkCanvas* onNewCanvas() SK_OVERRIDE;
     virtual SkSurface* onNewSurface(const SkImage::Info&) SK_OVERRIDE;
     virtual SkImage* onNewImageShapshot() SK_OVERRIDE;
     virtual void onDraw(SkCanvas*, SkScalar x, SkScalar y,
                         const SkPaint*) SK_OVERRIDE;
-    virtual void onCopyOnWrite(SkImage*, SkCanvas*) SK_OVERRIDE;
+    virtual void onCopyOnWrite(SkImage*, SkCanvas*, bool overwrite) SK_OVERRIDE;
 
 private:
     SkBitmap    fBitmap;
     bool        fWeOwnThePixels;
 
     typedef SkSurface_Base INHERITED;
 };
 
 ///////////////////////////////////////////////////////////////////////////////
 
@@ skipping 78 matching lines @@
 
 void SkSurface_Raster::onDraw(SkCanvas* canvas, SkScalar x, SkScalar y,
                               const SkPaint* paint) {
     canvas->drawBitmap(fBitmap, x, y, paint);
 }
 
 SkImage* SkSurface_Raster::onNewImageShapshot() {
     return SkNewImageFromBitmap(fBitmap, fWeOwnThePixels);
 }
 
-void SkSurface_Raster::onCopyOnWrite(SkImage* image, SkCanvas* canvas) {
+void SkSurface_Raster::onCopyOnWrite(SkImage* image, SkCanvas* canvas, bool overwrite) {
     // are we sharing pixelrefs with the image?
     if (SkBitmapImageGetPixelRef(image) == fBitmap.pixelRef()) {
         SkASSERT(fWeOwnThePixels);
-        SkBitmap prev(fBitmap);
-        prev.deepCopyTo(&fBitmap, prev.config());
+        if (!overwrite) {
+            SkBitmap prev(fBitmap);
+            prev.deepCopyTo(&fBitmap, prev.config());
+        } else {
+            SkBitmap newBitmap;
+            newBitmap.setConfig(fBitmap.getConfig(), fBitmap.width(), fBitmap.height(),
+                                fBitmap.rowBytes());
+            newBitmap.allocPixels();
+            fBitmap.swap(newBitmap);
+        }
         // Now fBitmap is a deep copy of itself (and therefore different from

[Stephen White, 2013/03/26 20:51:44]

Nit:  this comment is not strictly true anymore, since it may not be a deep
copy.

         // what is being used by the image. Next we update the canvas to use
         // this as its backend, so we can't modify the image's pixels anymore.
         canvas->getDevice()->replaceBitmapBackendForRasterSurface(fBitmap);
     }
 }
 
 ///////////////////////////////////////////////////////////////////////////////
 
 SkSurface* SkSurface::NewRasterDirect(const SkImage::Info& info, void* pixels, size_t rowBytes) {
     if (!SkSurface_Raster::Valid(info, rowBytes)) {
@@ skipping 20 matching lines @@
 
     size_t size = (size_t)size64;
     void* pixels = sk_malloc_throw(size);
     if (NULL == pixels) {
         return NULL;
     }
 
     SkAutoTUnref<SkPixelRef> pr(SkNEW_ARGS(SkMallocPixelRef, (pixels, size, NULL, true)));
     return SkNEW_ARGS(SkSurface_Raster, (info, pr, rowBytes));
 }