src/third_party/skia/dm/DMReplayTask.cpp

   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