commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 1 | #include "DMQuiltTask.h" |
| 2 | #include "DMUtil.h" |
| 3 | #include "DMWriteTask.h" |
| 4 | |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 5 | #include "SkBBHFactory.h" |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 6 | #include "SkCommandLineFlags.h" |
| 7 | #include "SkPicture.h" |
mtklein | 406654b | 2014-09-03 15:34:37 -0700 | [diff] [blame] | 8 | #include "SkTaskGroup.h" |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 9 | |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 10 | DEFINE_bool(quilt, true, "If true, draw GM via a picture into a quilt of small tiles and compare."); |
mtklein | 7cdc1ee | 2014-07-07 10:41:04 -0700 | [diff] [blame] | 11 | DEFINE_int32(quiltTile, 256, "Dimension of (square) quilt tile."); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 12 | |
| 13 | namespace DM { |
| 14 | |
mtklein | a9ceaf5 | 2014-09-29 08:44:46 -0700 | [diff] [blame] | 15 | static const char* kBBHs[] = { "nobbh", "rtree", "tilegrid" }; |
| 16 | QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference, QuiltTask::BBH bbh) |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 17 | : CpuTask(parent) |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 18 | , fBBH(bbh) |
mtklein | a9ceaf5 | 2014-09-29 08:44:46 -0700 | [diff] [blame] | 19 | , fName(UnderJoin(parent.name().c_str(), kBBHs[bbh])) |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 20 | , fGM(gm) |
| 21 | , fReference(reference) |
| 22 | {} |
| 23 | |
| 24 | static int tiles_needed(int fullDimension, int tileDimension) { |
| 25 | return (fullDimension + tileDimension - 1) / tileDimension; |
| 26 | } |
| 27 | |
mtklein | e71cd54 | 2014-10-29 14:17:13 -0700 | [diff] [blame] | 28 | struct DrawTileArgs { |
| 29 | int x, y; |
| 30 | const SkPicture* picture; |
| 31 | SkBitmap* quilt; |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 32 | }; |
| 33 | |
mtklein | e71cd54 | 2014-10-29 14:17:13 -0700 | [diff] [blame] | 34 | static void draw_tile(DrawTileArgs* arg) { |
| 35 | const DrawTileArgs& a = *arg; |
| 36 | SkBitmap tile; |
| 37 | a.quilt->extractSubset(&tile, SkIRect::MakeXYWH(a.x, a.y, FLAGS_quiltTile, FLAGS_quiltTile)); |
| 38 | SkCanvas tileCanvas(tile); |
| 39 | tileCanvas.translate(SkIntToScalar(-a.x), SkIntToScalar(-a.y)); |
| 40 | a.picture->playback(&tileCanvas); |
| 41 | tileCanvas.flush(); |
| 42 | } |
| 43 | |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 44 | void QuiltTask::draw() { |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 45 | SkAutoTDelete<SkBBHFactory> factory; |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 46 | switch (fBBH) { |
| 47 | case kNone_BBH: break; |
| 48 | case kRTree_BBH: |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 49 | factory.reset(SkNEW(SkRTreeFactory)); |
| 50 | break; |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 51 | case kTileGrid_BBH: { |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 52 | const SkTileGridFactory::TileGridInfo tiles = { |
| 53 | { FLAGS_quiltTile, FLAGS_quiltTile }, |
| 54 | /*overlap: */{0, 0}, |
| 55 | /*offset: */{0, 0}, |
| 56 | }; |
| 57 | factory.reset(SkNEW_ARGS(SkTileGridFactory, (tiles))); |
| 58 | break; |
| 59 | } |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 60 | } |
| 61 | |
| 62 | // A couple GMs draw wrong when using a bounding box hierarchy. |
| 63 | // This almost certainly means we have a bug to fix, but for now |
| 64 | // just draw without a bounding box hierarchy. |
| 65 | if (fGM->getFlags() & skiagm::GM::kNoBBH_Flag) { |
| 66 | factory.reset(NULL); |
| 67 | } |
| 68 | |
mtklein | a9ceaf5 | 2014-09-29 08:44:46 -0700 | [diff] [blame] | 69 | SkAutoTUnref<const SkPicture> recorded(RecordPicture(fGM.get(), factory.get())); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 70 | |
| 71 | SkBitmap full; |
commit-bot@chromium.org | 2664207 | 2014-05-15 17:33:31 +0000 | [diff] [blame] | 72 | AllocatePixels(fReference, &full); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 73 | |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 74 | if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { |
| 75 | // Some GMs don't draw exactly the same when tiled. Draw them in one go. |
| 76 | SkCanvas canvas(full); |
robertphillips | c5ba71d | 2014-09-04 08:42:50 -0700 | [diff] [blame] | 77 | recorded->playback(&canvas); |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 78 | canvas.flush(); |
| 79 | } else { |
| 80 | // Draw tiles in parallel into the same bitmap, simulating aggressive impl-side painting. |
mtklein | e71cd54 | 2014-10-29 14:17:13 -0700 | [diff] [blame] | 81 | int xTiles = tiles_needed(full.width(), FLAGS_quiltTile), |
| 82 | yTiles = tiles_needed(full.height(), FLAGS_quiltTile); |
| 83 | SkTDArray<DrawTileArgs> args; |
| 84 | args.setCount(xTiles*yTiles); |
| 85 | for (int y = 0; y < yTiles; y++) { |
| 86 | for (int x = 0; x < xTiles; x++) { |
| 87 | DrawTileArgs arg = { x*FLAGS_quiltTile, y*FLAGS_quiltTile, recorded, &full }; |
| 88 | args[y*xTiles + x] = arg; |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 89 | } |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 90 | } |
mtklein | e71cd54 | 2014-10-29 14:17:13 -0700 | [diff] [blame] | 91 | SkTaskGroup().batch(draw_tile, args.begin(), args.count()); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 92 | } |
| 93 | |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 94 | if (!BitmapsEqual(full, fReference)) { |
| 95 | this->fail(); |
mtklein | ea65bfa | 2014-09-09 07:59:46 -0700 | [diff] [blame] | 96 | this->spawnChild(SkNEW_ARGS(WriteTask, (*this, "GM", full))); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 97 | } |
| 98 | } |
| 99 | |
| 100 | bool QuiltTask::shouldSkip() const { |
| 101 | if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { |
| 102 | return true; |
| 103 | } |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 104 | return !FLAGS_quilt; |
| 105 | } |
| 106 | |
| 107 | } // namespace DM |