Support serialization in SkRecord-backed SkPictures.

dm/DMSKPTask.cpp

-#include "DMRecordTask.h"
 #include "DMSKPTask.h"
 #include "DMUtil.h"
 #include "DMWriteTask.h"
 
+#include "SkCommandLineFlags.h"
+#include "SkPictureRecorder.h"
+
+DECLARE_bool(skr);  // in DMReplayTask.cpp
+
 namespace DM {
 
-SKPTask::SKPTask(Reporter* r, TaskRunner* tr, SkPicture* pic, SkString filename)
+// Test that an SkPicture plays back the same when re-recorded into an
+// SkPicture backed by SkRecord.
+class SkrComparisonTask : public CpuTask {
+public:

[robertphillips, 2014/06/24 18:34:46]

const SkBitmap& reference ?

[mtklein, 2014/06/24 19:17:03]

This one's tricky.  I've always found it more helpful to see that we're
certainly going to copy the SkBitmap as part of the constructor API.  We'll copy
it exactly once either way, but this way the creator of SkrComparisonTask knows
this too, so they don't have to feel at all unsure about whether or not they
have to keep the SkBitmap alive for SkrComparisonTask.

The rest of DM works with SkBitmaps by value, so I'm going to leave it for
consistency.  I could be convinced to convert it so the parameters pass by
const& if it's really important to you.

+    SkrComparisonTask(const Task& parent, const SkPicture* picture, SkBitmap reference)
+        : CpuTask(parent)
+        , fPicture(picture)
+        , fReference(reference)
+        , fName(UnderJoin(parent.name().c_str(), "skr")) {}
+
+    virtual bool shouldSkip() const SK_OVERRIDE { return !FLAGS_skr; }
+    virtual SkString name() const SK_OVERRIDE { return fName; }
+
+    virtual void draw() SK_OVERRIDE {
+        SkPictureRecorder recorder;
+        fPicture->draw(recorder.EXPERIMENTAL_beginRecording(fPicture->width(), fPicture->height()));
+        SkAutoTDelete<const SkPicture> skrPicture(recorder.endRecording());
+
+        SkBitmap bitmap;
+        AllocatePixels(kN32_SkColorType, fPicture->width(), fPicture->height(), &bitmap);
+        DrawPicture(*skrPicture, &bitmap);
+
+        if (!BitmapsEqual(fReference, bitmap)) {
+            this->fail();
+            this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap)));
+        }
+    }
+
+private:
+    SkAutoTUnref<const SkPicture> fPicture;
+    const SkBitmap fReference;
+    const SkString fName;
+};
+
+
+SKPTask::SKPTask(Reporter* r, TaskRunner* tr, const SkPicture* pic, SkString filename)
     : CpuTask(r, tr), fPicture(SkRef(pic)), fName(FileToTaskName(filename)) {}
 
 void SKPTask::draw() {
     SkBitmap bitmap;
     AllocatePixels(kN32_SkColorType, fPicture->width(), fPicture->height(), &bitmap);
-    DrawPicture(fPicture, &bitmap);
+    DrawPicture(*fPicture, &bitmap);
 
-    this->spawnChild(SkNEW_ARGS(RecordTask,
-                                (*this, fPicture, bitmap, RecordTask::kNoOptimize_Mode)));
-    this->spawnChild(SkNEW_ARGS(RecordTask,
-                                (*this, fPicture, bitmap, RecordTask::kOptimize_Mode)));
     this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap)));
+    this->spawnChild(SkNEW_ARGS(SkrComparisonTask, (*this, fPicture.get(), bitmap)));
 }
 
 }  // namespace DM