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 |