1 files changed, 10 insertions, 3 deletions
diff --git a/dm/DMRecordTask.h b/dm/DMRecordTask.h
index 94ac412e42..1420724bb0 100644
--- a/dm/DMRecordTask.h
+++ b/dm/DMRecordTask.h
@@ -3,28 +3,35 @@
 
 #include "DMTask.h"
 #include "SkBitmap.h"
+#include "SkPicture.h"
 #include "SkString.h"
 #include "SkTemplates.h"
 #include "gm.h"
 
-// Records a GM through an SkRecord, draws it, and compares against the reference bitmap.
+// Records a GM or SKP through an SkRecord, draws it, and compares against the reference bitmap.
 
 namespace DM {
 
 class RecordTask : public CpuTask {
 
 public:
-    RecordTask(const Task& parent, skiagm::GM*, SkBitmap reference, bool optimize);
+    enum Mode {
+        kNoOptimize_Mode,
+        kOptimize_Mode,
+    };
+    RecordTask(const Task& parent, skiagm::GM*, SkBitmap reference, Mode);
+    RecordTask(const Task& parent, SkPicture*,  SkBitmap reference, Mode);
 
     virtual void draw() SK_OVERRIDE;
     virtual bool shouldSkip() const SK_OVERRIDE;
     virtual SkString name() const SK_OVERRIDE { return fName; }
 
 private:
+    bool fOptimize;
     const SkString fName;
+    SkAutoTUnref<SkPicture> fPicture;
     SkAutoTDelete<skiagm::GM> fGM;
     const SkBitmap fReference;
-    bool fOptimize;
 };
 
 }  // namespace DM