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
ReplayTask(const Task & parent,skiagm::GM * gm,SkBitmap reference,Mode mode)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
draw()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
shouldSkip() const41 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