1 #include "DMReplayTask.h" 2 #include "DMWriteTask.h" 3 #include "DMUtil.h" 4 5 #include "SkBBHFactory.h" 6 #include "SkCommandLineFlags.h" 7 #include "SkPicture.h" 8 9 DEFINE_bool(replay, true, "If true, run picture replay tests."); 10 DEFINE_bool(rtree, true, "If true, run picture replay tests with an rtree."); 11 12 namespace DM { 13 14 ReplayTask::ReplayTask(const Task& parent, 15 skiagm::GM* gm, 16 SkBitmap reference, 17 Mode mode) 18 : CpuTask(parent) 19 , fUseRTree(mode == kRTree_Mode) 20 , fName(UnderJoin(parent.name().c_str(), fUseRTree ? "rtree" : "replay")) 21 , fGM(gm) 22 , fReference(reference) 23 {} 24 25 void ReplayTask::draw() { 26 SkAutoTDelete<SkBBHFactory> factory; 27 if (fUseRTree) { 28 factory.reset(SkNEW(SkRTreeFactory)); 29 } 30 SkAutoTUnref<SkPicture> recorded(RecordPicture(fGM.get(), 0, factory.get())); 31 32 SkBitmap bitmap; 33 AllocatePixels(fReference, &bitmap); 34 DrawPicture(recorded, &bitmap); 35 if (!BitmapsEqual(bitmap, fReference)) { 36 this->fail(); 37 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap))); 38 } 39 } 40 41 bool ReplayTask::shouldSkip() const { 42 if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { 43 return true; 44 } 45 46 if (FLAGS_rtree && fUseRTree) { 47 return false; 48 } 49 if (FLAGS_replay && !fUseRTree) { 50 return false; 51 } 52 return true; 53 } 54 55 } // namespace DM 56
