| 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
|
|
|