1 #include "DMQuiltTask.h" 2 #include "DMUtil.h" 3 #include "DMWriteTask.h" 4 5 #include "SkCommandLineFlags.h" 6 #include "SkPicture.h" 7 8 DEFINE_bool(quilt, true, "If true, draw into a quilt of small tiles and compare."); 9 DEFINE_int32(quiltTile, 16, "Dimension of (square) quilt tile."); 10 11 namespace DM { 12 13 QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference) 14 : CpuTask(parent) 15 , fName(UnderJoin(parent.name().c_str(), "quilt")) 16 , fGM(gm) 17 , fReference(reference) 18 {} 19 20 static int tiles_needed(int fullDimension, int tileDimension) { 21 return (fullDimension + tileDimension - 1) / tileDimension; 22 } 23 24 void QuiltTask::draw() { 25 SkAutoTUnref<SkPicture> recorded(RecordPicture(fGM.get())); 26 27 SkBitmap full; 28 AllocatePixels(fReference, &full); 29 SkCanvas fullCanvas(full); 30 31 SkBitmap tile; 32 tile.allocPixels(SkImageInfo::Make(FLAGS_quiltTile, FLAGS_quiltTile, 33 fReference.colorType(), kPremul_SkAlphaType)); 34 SkCanvas tileCanvas(tile); 35 36 for (int y = 0; y < tiles_needed(full.height(), tile.height()); y++) { 37 for (int x = 0; x < tiles_needed(full.width(), tile.width()); x++) { 38 SkAutoCanvasRestore ar(&tileCanvas, true/*also save now*/); 39 40 const SkScalar xOffset = SkIntToScalar(x * tile.width()), 41 yOffset = SkIntToScalar(y * tile.height()); 42 SkMatrix matrix = tileCanvas.getTotalMatrix(); 43 matrix.postTranslate(-xOffset, -yOffset); 44 tileCanvas.setMatrix(matrix); 45 46 recorded->draw(&tileCanvas); 47 tileCanvas.flush(); 48 fullCanvas.drawBitmap(tile, xOffset, yOffset, NULL); 49 } 50 } 51 52 if (!BitmapsEqual(full, fReference)) { 53 this->fail(); 54 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, full))); 55 } 56 } 57 58 bool QuiltTask::shouldSkip() const { 59 if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { 60 return true; 61 } 62 if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { 63 return true; 64 } 65 return !FLAGS_quilt; 66 } 67 68 } // namespace DM 69