Index: dm/DMQuiltTask.cpp |
diff --git a/dm/DMQuiltTask.cpp b/dm/DMQuiltTask.cpp |
new file mode 100644 |
index 0000000000000000000000000000000000000000..83a989da718907cf5e070db719346bebadd641a4 |
--- /dev/null |
+++ b/dm/DMQuiltTask.cpp |
@@ -0,0 +1,98 @@ |
+#include "DMQuiltTask.h" |
+#include "DMUtil.h" |
+#include "DMWriteTask.h" |
+ |
+#include "SkBBHFactory.h" |
+#include "SkCommandLineFlags.h" |
+#include "SkPicture.h" |
+#include "SkTaskGroup.h" |
+ |
+DEFINE_bool(quilt, true, "If true, draw GM via a picture into a quilt of small tiles and compare."); |
+DEFINE_int32(quiltTile, 256, "Dimension of (square) quilt tile."); |
+ |
+namespace DM { |
+ |
+static const char* kBBHs[] = { "nobbh", "rtree", "tilegrid" }; |
+QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference, QuiltTask::BBH bbh) |
+ : CpuTask(parent) |
+ , fBBH(bbh) |
+ , fName(UnderJoin(parent.name().c_str(), kBBHs[bbh])) |
+ , fGM(gm) |
+ , fReference(reference) |
+ {} |
+ |
+static int tiles_needed(int fullDimension, int tileDimension) { |
+ return (fullDimension + tileDimension - 1) / tileDimension; |
+} |
+ |
+struct DrawTileArgs { |
+ int x, y; |
+ const SkPicture* picture; |
+ SkBitmap* quilt; |
+}; |
+ |
+static void draw_tile(DrawTileArgs* arg) { |
+ const DrawTileArgs& a = *arg; |
+ SkBitmap tile; |
+ a.quilt->extractSubset(&tile, SkIRect::MakeXYWH(a.x, a.y, FLAGS_quiltTile, FLAGS_quiltTile)); |
+ SkCanvas tileCanvas(tile); |
+ tileCanvas.translate(SkIntToScalar(-a.x), SkIntToScalar(-a.y)); |
+ a.picture->playback(&tileCanvas); |
+ tileCanvas.flush(); |
+} |
+ |
+void QuiltTask::draw() { |
+ SkAutoTDelete<SkBBHFactory> factory; |
+ switch (fBBH) { |
+ case kNone_BBH: break; |
+ case kRTree_BBH: |
+ factory.reset(SkNEW(SkRTreeFactory)); |
+ break; |
+ } |
+ |
+ // A couple GMs draw wrong when using a bounding box hierarchy. |
+ // This almost certainly means we have a bug to fix, but for now |
+ // just draw without a bounding box hierarchy. |
+ if (fGM->getFlags() & skiagm::GM::kNoBBH_Flag) { |
+ factory.reset(NULL); |
+ } |
+ |
+ SkAutoTUnref<const SkPicture> recorded(RecordPicture(fGM.get(), factory.get())); |
+ |
+ SkBitmap full; |
+ AllocatePixels(fReference, &full); |
+ |
+ if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { |
+ // Some GMs don't draw exactly the same when tiled. Draw them in one go. |
+ SkCanvas canvas(full); |
+ recorded->playback(&canvas); |
+ canvas.flush(); |
+ } else { |
+ // Draw tiles in parallel into the same bitmap, simulating aggressive impl-side painting. |
+ int xTiles = tiles_needed(full.width(), FLAGS_quiltTile), |
+ yTiles = tiles_needed(full.height(), FLAGS_quiltTile); |
+ SkTDArray<DrawTileArgs> args; |
+ args.setCount(xTiles*yTiles); |
+ for (int y = 0; y < yTiles; y++) { |
+ for (int x = 0; x < xTiles; x++) { |
+ DrawTileArgs arg = { x*FLAGS_quiltTile, y*FLAGS_quiltTile, recorded, &full }; |
+ args[y*xTiles + x] = arg; |
+ } |
+ } |
+ SkTaskGroup().batch(draw_tile, args.begin(), args.count()); |
+ } |
+ |
+ if (!BitmapsEqual(full, fReference)) { |
+ this->fail(); |
+ this->spawnChild(SkNEW_ARGS(WriteTask, (*this, "GM", full))); |
+ } |
+} |
+ |
+bool QuiltTask::shouldSkip() const { |
+ if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { |
+ return true; |
+ } |
+ return !FLAGS_quilt; |
+} |
+ |
+} // namespace DM |