OLD | NEW |
(Empty) | |
| 1 #include "DMQuiltTask.h" |
| 2 #include "DMUtil.h" |
| 3 #include "DMWriteTask.h" |
| 4 |
| 5 #include "SkBBHFactory.h" |
| 6 #include "SkCommandLineFlags.h" |
| 7 #include "SkPicture.h" |
| 8 #include "SkTaskGroup.h" |
| 9 |
| 10 DEFINE_bool(quilt, true, "If true, draw GM via a picture into a quilt of small t
iles and compare."); |
| 11 DEFINE_int32(quiltTile, 256, "Dimension of (square) quilt tile."); |
| 12 |
| 13 namespace DM { |
| 14 |
| 15 static const char* kBBHs[] = { "nobbh", "rtree", "tilegrid" }; |
| 16 QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference, Qui
ltTask::BBH bbh) |
| 17 : CpuTask(parent) |
| 18 , fBBH(bbh) |
| 19 , fName(UnderJoin(parent.name().c_str(), kBBHs[bbh])) |
| 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 |
| 28 struct DrawTileArgs { |
| 29 int x, y; |
| 30 const SkPicture* picture; |
| 31 SkBitmap* quilt; |
| 32 }; |
| 33 |
| 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, F
LAGS_quiltTile)); |
| 38 SkCanvas tileCanvas(tile); |
| 39 tileCanvas.translate(SkIntToScalar(-a.x), SkIntToScalar(-a.y)); |
| 40 a.picture->playback(&tileCanvas); |
| 41 tileCanvas.flush(); |
| 42 } |
| 43 |
| 44 void QuiltTask::draw() { |
| 45 SkAutoTDelete<SkBBHFactory> factory; |
| 46 switch (fBBH) { |
| 47 case kNone_BBH: break; |
| 48 case kRTree_BBH: |
| 49 factory.reset(SkNEW(SkRTreeFactory)); |
| 50 break; |
| 51 } |
| 52 |
| 53 // A couple GMs draw wrong when using a bounding box hierarchy. |
| 54 // This almost certainly means we have a bug to fix, but for now |
| 55 // just draw without a bounding box hierarchy. |
| 56 if (fGM->getFlags() & skiagm::GM::kNoBBH_Flag) { |
| 57 factory.reset(NULL); |
| 58 } |
| 59 |
| 60 SkAutoTUnref<const SkPicture> recorded(RecordPicture(fGM.get(), factory.get(
))); |
| 61 |
| 62 SkBitmap full; |
| 63 AllocatePixels(fReference, &full); |
| 64 |
| 65 if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { |
| 66 // Some GMs don't draw exactly the same when tiled. Draw them in one go
. |
| 67 SkCanvas canvas(full); |
| 68 recorded->playback(&canvas); |
| 69 canvas.flush(); |
| 70 } else { |
| 71 // Draw tiles in parallel into the same bitmap, simulating aggressive im
pl-side painting. |
| 72 int xTiles = tiles_needed(full.width(), FLAGS_quiltTile), |
| 73 yTiles = tiles_needed(full.height(), FLAGS_quiltTile); |
| 74 SkTDArray<DrawTileArgs> args; |
| 75 args.setCount(xTiles*yTiles); |
| 76 for (int y = 0; y < yTiles; y++) { |
| 77 for (int x = 0; x < xTiles; x++) { |
| 78 DrawTileArgs arg = { x*FLAGS_quiltTile, y*FLAGS_quiltTile, recor
ded, &full }; |
| 79 args[y*xTiles + x] = arg; |
| 80 } |
| 81 } |
| 82 SkTaskGroup().batch(draw_tile, args.begin(), args.count()); |
| 83 } |
| 84 |
| 85 if (!BitmapsEqual(full, fReference)) { |
| 86 this->fail(); |
| 87 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, "GM", full))); |
| 88 } |
| 89 } |
| 90 |
| 91 bool QuiltTask::shouldSkip() const { |
| 92 if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { |
| 93 return true; |
| 94 } |
| 95 return !FLAGS_quilt; |
| 96 } |
| 97 |
| 98 } // namespace DM |
OLD | NEW |