Replace SkPictureReplacementPlayback with GrRecordReplaceDraw

tests/RecordReplaceDrawTest.cpp

Index: tests/RecordReplaceDrawTest.cpp
diff --git a/tests/RecordReplaceDrawTest.cpp b/tests/RecordReplaceDrawTest.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..6893876402efb3ae25c1c3a7b67f21e8681a6d3a
--- /dev/null
+++ b/tests/RecordReplaceDrawTest.cpp
@@ -0,0 +1,125 @@
+/*
+ * Copyright 2014 Google Inc.
+ *
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ */
+
+#if SK_SUPPORT_GPU
+
+#include "Test.h"
+#include "RecordTestUtils.h"
+
+#include "SkBBHFactory.h"
+#include "SkRecordDraw.h"
+#include "SkRecorder.h"
+#include "SkUtils.h"
+#include "GrRecordReplaceDraw.h"
+
+static const int kWidth = 100;
+static const int kHeight = 100;
+
+class JustOneDraw : public SkDrawPictureCallback {
+public:
+    JustOneDraw() : fCalls(0) {}
+
+    virtual bool abortDrawing() SK_OVERRIDE { return fCalls++ > 0; }
+private:
+    int fCalls;
+};
+
+// Make sure the abort callback works
+DEF_TEST(RecordReplaceDraw_Abort, r) {
+    // Record two commands.
+    SkRecord record;
+    SkRecorder recorder(&record, kWidth, kHeight);
+    recorder.drawRect(SkRect::MakeWH(SkIntToScalar(kWidth), SkIntToScalar(kHeight)), SkPaint());
+    recorder.clipRect(SkRect::MakeWH(SkIntToScalar(kWidth), SkIntToScalar(kHeight)));
+
+    SkRecord rerecord;
+    SkRecorder canvas(&rerecord, kWidth, kHeight);
+
+    GrReplacements replacements;
+    JustOneDraw callback;
+    GrRecordReplaceDraw(record, &canvas, NULL/*bbh*/, &replacements, &callback);
+
+    REPORTER_ASSERT(r, 3 == rerecord.count());
+    assert_type<SkRecords::Save>(r, rerecord, 0);
+    assert_type<SkRecords::DrawRect>(r, rerecord, 1);
+    assert_type<SkRecords::Restore>(r, rerecord, 2);
+}
+
+// Make sure GrRecordReplaceDraw balances unbalanced saves
+DEF_TEST(RecordReplaceDraw_Unbalanced, r) {
+    SkRecord record;
+    SkRecorder recorder(&record, kWidth, kHeight);
+    recorder.save();  // We won't balance this, but GrRecordReplaceDraw will for us.
+
+    SkRecord rerecord;
+    SkRecorder canvas(&rerecord, kWidth, kHeight);
+
+    GrReplacements replacements;
+    GrRecordReplaceDraw(record, &canvas, NULL/*bbh*/, &replacements, NULL/*callback*/);
+
+    REPORTER_ASSERT(r, 4 == rerecord.count());
+    assert_type<SkRecords::Save>(r, rerecord, 0);
+    assert_type<SkRecords::Save>(r, rerecord, 1);
+    assert_type<SkRecords::Restore>(r, rerecord, 2);
+    assert_type<SkRecords::Restore>(r, rerecord, 3);
+}
+
+static SkImage* make_image(SkColor color) {
+    const SkPMColor pmcolor = SkPreMultiplyColor(color);
+    const SkImageInfo info = SkImageInfo::MakeN32Premul(kWidth, kHeight);
+    const size_t rowBytes = info.minRowBytes();
+    const size_t size = rowBytes * info.fHeight;
+
+    SkAutoMalloc addr(size);
+    sk_memset32((SkPMColor*)addr.get(), pmcolor, SkToInt(size >> 2));
+
+    return SkImage::NewRasterCopy(info, addr.get(), rowBytes);
+}
+
+// Test out the layer replacement functionality with and w/o a BBH
+void test_replacements(skiatest::Reporter* r, bool useBBH) {
+    SkRecord record;
+    SkRecorder recorder(&record, kWidth, kHeight);
+    SkAutoTDelete<SkPaint> paint(SkNEW(SkPaint));
+    recorder.saveLayer(NULL, paint);
+    recorder.clear(SK_ColorRED);
+    recorder.restore();
+    recorder.drawRect(SkRect::MakeWH(SkIntToScalar(kWidth/2), SkIntToScalar(kHeight/2)), 
+                      SkPaint());
+
+    GrReplacements replacements;
+    GrReplacements::ReplacementInfo* ri = replacements.push();
+    ri->fStart = 0;
+    ri->fStop = 2;
+    ri->fPos.set(0, 0);
+    ri->fImage = make_image(SK_ColorRED);
+    ri->fPaint = paint;
+    ri->fSrcRect = SkIRect::MakeWH(kWidth, kHeight);
+
+    SkAutoTUnref<SkBBoxHierarchy> bbh;
+
+    if (useBBH) {
+        SkRTreeFactory factory;
+        bbh.reset((factory)(kWidth, kHeight));
+        SkRecordFillBounds(record, bbh);
+    }
+
+    SkRecord rerecord;
+    SkRecorder canvas(&rerecord, kWidth, kHeight);
+    GrRecordReplaceDraw(record, &canvas, bbh, &replacements, NULL/*callback*/);
+
+    REPORTER_ASSERT(r, 4 == rerecord.count());
+    assert_type<SkRecords::Save>(r, rerecord, 0);
+    assert_type<SkRecords::DrawBitmapRectToRect>(r, rerecord, 1);
+    assert_type<SkRecords::DrawRect>(r, rerecord, 2);
+    assert_type<SkRecords::Restore>(r, rerecord, 3);
+}
+
+DEF_TEST(RecordReplaceDraw_Replace, r)        { test_replacements(r, false); }
+DEF_TEST(RecordReplaceDraw_ReplaceWithBBH, r) { test_replacements(r, true); }
+
+#endif