Add snapshot entry point to SkPictureRecorder

tests/PictureTest.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 "SkBitmapDevice.h"
 #include "SkCanvas.h"
 #include "SkColorPriv.h"
@@ skipping 683 matching lines @@
     }
 }
 
 static void set_canvas_to_save_count_4(SkCanvas* canvas) {
     canvas->restoreToCount(1);
     canvas->save();
     canvas->save();
     canvas->save();
 }
 
+// Test out SkPictureRecorder::snapshot
+static void test_snapshot(skiatest::Reporter* reporter) {
+    SkPictureRecorder recorder;
+
+    // Calling snapshot when not recording will return NULL
+    REPORTER_ASSERT(reporter, NULL == recorder.snapshot());
+
+    SkCanvas* canvas = recorder.beginRecording(10, 10, NULL, 0);
+
+    SkRect r = SkRect::MakeWH(5, 5);
+    SkPaint p;
+
+    canvas->drawRect(r, p);
+
+    SkAutoTUnref<SkPicture> snapshot(recorder.snapshot());
+    REPORTER_ASSERT(reporter, !snapshot->willPlayBackBitmaps());
+
+    SkBitmap bm;
+    make_bm(&bm, 10, 10, SK_ColorRED, true);
+
+    r.offset(5.0f, 5.0f);
+    canvas->drawBitmapRectToRect(bm, NULL, r);
+
+    SkAutoTUnref<SkPicture> final(recorder.endRecording());
+    REPORTER_ASSERT(reporter, final->willPlayBackBitmaps());
+
+    REPORTER_ASSERT(reporter, snapshot->uniqueID() != final->uniqueID());
+
+    // The snapshot shouldn't pick up any operations added after it was made
+    REPORTER_ASSERT(reporter, !snapshot->willPlayBackBitmaps());
+}
+
 static void test_unbalanced_save_restores(skiatest::Reporter* reporter) {
     SkCanvas testCanvas(100, 100);
     set_canvas_to_save_count_4(&testCanvas);
 
     REPORTER_ASSERT(reporter, 4 == testCanvas.getSaveCount());
 
     SkPaint paint;
     SkRect rect = SkRect::MakeLTRB(-10000000, -10000000, 10000000, 10000000);
 
     SkPictureRecorder recorder;
@@ skipping 474 matching lines @@
     }
 }
 
 DEF_TEST(Picture, reporter) {
 #ifdef SK_DEBUG
     test_deleting_empty_playback();
     test_serializing_empty_picture();
 #else
     test_bad_bitmap();
 #endif
+    test_snapshot(reporter);
     test_unbalanced_save_restores(reporter);
     test_peephole();
     test_gatherpixelrefs(reporter);
     test_gatherpixelrefsandrects(reporter);
     test_bitmap_with_encoded_data(reporter);
     test_clone_empty(reporter);
     test_draw_empty(reporter);
     test_clip_bound_opt(reporter);
     test_clip_expansion(reporter);
     test_hierarchical(reporter);
@@ skipping 26 matching lines @@
     SkAutoTUnref<SkPicture> picture(recorder.endRecording());
 }
 
 DEF_TEST(Canvas_EmptyBitmap, r) {
     SkBitmap dst;
     dst.allocN32Pixels(10, 10);
     SkCanvas canvas(dst);
 
     test_draw_bitmaps(&canvas);
 }