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 | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 8 | #include "SkThreadPool.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 | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 15 | static SkString suffix(QuiltTask::Backend backend, QuiltTask::BBH bbh) { |
| 16 | static const char* kBackends[] = { "default", "skrecord" }; |
mtklein | 2a65a23 | 2014-08-26 14:07:04 -0700 | [diff] [blame] | 17 | static const char* kBBHs[] = { "nobbh", "rtree", "tilegrid" }; |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 18 | return SkStringPrintf("%s-%s", kBackends[backend], kBBHs[bbh]); |
| 19 | } |
| 20 | |
| 21 | QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference, |
| 22 | QuiltTask::BBH bbh, QuiltTask::Backend backend) |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 23 | : CpuTask(parent) |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 24 | , fBBH(bbh) |
| 25 | , fBackend(backend) |
| 26 | , fName(UnderJoin(parent.name().c_str(), suffix(backend, bbh).c_str())) |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 27 | , fGM(gm) |
| 28 | , fReference(reference) |
| 29 | {} |
| 30 | |
| 31 | static int tiles_needed(int fullDimension, int tileDimension) { |
| 32 | return (fullDimension + tileDimension - 1) / tileDimension; |
| 33 | } |
| 34 | |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 35 | class Tile : public SkRunnable { |
| 36 | public: |
mtklein | 7cdc1ee | 2014-07-07 10:41:04 -0700 | [diff] [blame] | 37 | Tile(int x, int y, const SkPicture& picture, SkBitmap* quilt) |
| 38 | : fX(x * FLAGS_quiltTile) |
| 39 | , fY(y * FLAGS_quiltTile) |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 40 | , fPicture(picture) |
mtklein | 7cdc1ee | 2014-07-07 10:41:04 -0700 | [diff] [blame] | 41 | , fQuilt(quilt) {} |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 42 | |
| 43 | virtual void run() SK_OVERRIDE { |
| 44 | SkBitmap tile; |
mtklein | 7cdc1ee | 2014-07-07 10:41:04 -0700 | [diff] [blame] | 45 | fQuilt->extractSubset(&tile, SkIRect::MakeXYWH(fX, fY, FLAGS_quiltTile, FLAGS_quiltTile)); |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 46 | SkCanvas tileCanvas(tile); |
| 47 | |
mtklein | 7cdc1ee | 2014-07-07 10:41:04 -0700 | [diff] [blame] | 48 | tileCanvas.translate(SkIntToScalar(-fX), SkIntToScalar(-fY)); |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 49 | fPicture.draw(&tileCanvas); |
| 50 | tileCanvas.flush(); |
| 51 | |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 52 | delete this; |
| 53 | } |
| 54 | |
| 55 | private: |
| 56 | const int fX, fY; |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 57 | const SkPicture& fPicture; |
mtklein | 7cdc1ee | 2014-07-07 10:41:04 -0700 | [diff] [blame] | 58 | SkBitmap* fQuilt; |
mtklein | ec2ae58 | 2014-07-01 07:46:50 -0700 | [diff] [blame] | 59 | }; |
| 60 | |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 61 | void QuiltTask::draw() { |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 62 | SkAutoTDelete<SkBBHFactory> factory; |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 63 | switch (fBBH) { |
| 64 | case kNone_BBH: break; |
| 65 | case kRTree_BBH: |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 66 | factory.reset(SkNEW(SkRTreeFactory)); |
| 67 | break; |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 68 | case kTileGrid_BBH: { |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 69 | const SkTileGridFactory::TileGridInfo tiles = { |
| 70 | { FLAGS_quiltTile, FLAGS_quiltTile }, |
| 71 | /*overlap: */{0, 0}, |
| 72 | /*offset: */{0, 0}, |
| 73 | }; |
| 74 | factory.reset(SkNEW_ARGS(SkTileGridFactory, (tiles))); |
| 75 | break; |
| 76 | } |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 77 | } |
| 78 | |
| 79 | // A couple GMs draw wrong when using a bounding box hierarchy. |
| 80 | // This almost certainly means we have a bug to fix, but for now |
| 81 | // just draw without a bounding box hierarchy. |
| 82 | if (fGM->getFlags() & skiagm::GM::kNoBBH_Flag) { |
| 83 | factory.reset(NULL); |
| 84 | } |
| 85 | |
| 86 | SkAutoTUnref<const SkPicture> recorded( |
mtklein | 5ad6ee1 | 2014-08-11 08:08:43 -0700 | [diff] [blame] | 87 | RecordPicture(fGM.get(), factory.get(), kSkRecord_Backend == fBackend)); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 88 | |
| 89 | SkBitmap full; |
commit-bot@chromium.org | 2664207 | 2014-05-15 17:33:31 +0000 | [diff] [blame] | 90 | AllocatePixels(fReference, &full); |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 91 | |
mtklein | e4636aa | 2014-07-09 13:10:58 -0700 | [diff] [blame] | 92 | if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { |
| 93 | // Some GMs don't draw exactly the same when tiled. Draw them in one go. |
| 94 | SkCanvas canvas(full); |
| 95 | recorded->draw(&canvas); |
| 96 | canvas.flush(); |
| 97 | } else { |
| 98 | // Draw tiles in parallel into the same bitmap, simulating aggressive impl-side painting. |
| 99 | SkThreadPool pool(SkThreadPool::kThreadPerCore); |
| 100 | for (int y = 0; y < tiles_needed(full.height(), FLAGS_quiltTile); y++) { |
| 101 | for (int x = 0; x < tiles_needed(full.width(), FLAGS_quiltTile); x++) { |
| 102 | // Deletes itself when done. |
| 103 | pool.add(new Tile(x, y, *recorded, &full)); |
| 104 | } |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 105 | } |
| 106 | } |
| 107 | |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 108 | if (!BitmapsEqual(full, fReference)) { |
| 109 | this->fail(); |
| 110 | this->spawnChild(SkNEW_ARGS(WriteTask, (*this, full))); |
| 111 | } |
| 112 | } |
| 113 | |
| 114 | bool QuiltTask::shouldSkip() const { |
| 115 | if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { |
| 116 | return true; |
| 117 | } |
commit-bot@chromium.org | a90c680 | 2014-04-30 13:20:45 +0000 | [diff] [blame] | 118 | return !FLAGS_quilt; |
| 119 | } |
| 120 | |
| 121 | } // namespace DM |