OLD | NEW |
1 #include "DMQuiltTask.h" | 1 #include "DMQuiltTask.h" |
2 #include "DMUtil.h" | 2 #include "DMUtil.h" |
3 #include "DMWriteTask.h" | 3 #include "DMWriteTask.h" |
4 | 4 |
5 #include "SkCommandLineFlags.h" | 5 #include "SkCommandLineFlags.h" |
6 #include "SkPicture.h" | 6 #include "SkPicture.h" |
7 #include "SkThreadPool.h" | 7 #include "ThreadPool.h" |
8 | 8 |
9 DEFINE_bool(quilt, true, "If true, draw into a quilt of small tiles and compare.
"); | 9 DEFINE_bool(quilt, true, "If true, draw into a quilt of small tiles and compare.
"); |
10 DEFINE_int32(quiltTile, 16, "Dimension of (square) quilt tile."); | 10 DEFINE_int32(quiltTile, 16, "Dimension of (square) quilt tile."); |
11 DEFINE_bool(quiltThreaded, false, "If true, draw quilt tiles with multiple threa
ds."); | 11 DEFINE_bool(quiltThreaded, false, "If true, draw quilt tiles with multiple threa
ds."); |
12 | 12 |
13 namespace DM { | 13 namespace DM { |
14 | 14 |
15 QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference) | 15 QuiltTask::QuiltTask(const Task& parent, skiagm::GM* gm, SkBitmap reference) |
16 : CpuTask(parent) | 16 : CpuTask(parent) |
17 , fName(UnderJoin(parent.name().c_str(), "quilt")) | 17 , fName(UnderJoin(parent.name().c_str(), "quilt")) |
18 , fGM(gm) | 18 , fGM(gm) |
19 , fReference(reference) | 19 , fReference(reference) |
20 {} | 20 {} |
21 | 21 |
22 static int tiles_needed(int fullDimension, int tileDimension) { | 22 static int tiles_needed(int fullDimension, int tileDimension) { |
23 return (fullDimension + tileDimension - 1) / tileDimension; | 23 return (fullDimension + tileDimension - 1) / tileDimension; |
24 } | 24 } |
25 | 25 |
26 class Tile : public SkRunnable { | 26 class Tile : public Runnable { |
27 public: | 27 public: |
28 Tile(int x, int y, SkColorType colorType, | 28 Tile(int x, int y, SkColorType colorType, |
29 const SkPicture& picture, SkCanvas* canvas, SkMutex* mutex) | 29 const SkPicture& picture, SkCanvas* canvas, SkMutex* mutex) |
30 : fX(x) | 30 : fX(x) |
31 , fY(y) | 31 , fY(y) |
32 , fColorType(colorType) | 32 , fColorType(colorType) |
33 , fPicture(picture) | 33 , fPicture(picture) |
34 , fCanvas(canvas) | 34 , fCanvas(canvas) |
35 , fMutex(mutex) {} | 35 , fMutex(mutex) {} |
36 | 36 |
(...skipping 26 matching lines...) Expand all Loading... |
63 }; | 63 }; |
64 | 64 |
65 void QuiltTask::draw() { | 65 void QuiltTask::draw() { |
66 SkAutoTUnref<SkPicture> recorded(RecordPicture(fGM.get())); | 66 SkAutoTUnref<SkPicture> recorded(RecordPicture(fGM.get())); |
67 | 67 |
68 SkBitmap full; | 68 SkBitmap full; |
69 AllocatePixels(fReference, &full); | 69 AllocatePixels(fReference, &full); |
70 SkCanvas fullCanvas(full); | 70 SkCanvas fullCanvas(full); |
71 SkMutex mutex; // Guards fullCanvas. | 71 SkMutex mutex; // Guards fullCanvas. |
72 | 72 |
73 SkThreadPool pool(FLAGS_quiltThreaded ? SkThreadPool::kThreadPerCore : 0); | 73 ThreadPool pool(FLAGS_quiltThreaded ? ThreadPool::kThreadPerCore : 0); |
74 | 74 |
75 for (int y = 0; y < tiles_needed(full.height(), FLAGS_quiltTile); y++) { | 75 for (int y = 0; y < tiles_needed(full.height(), FLAGS_quiltTile); y++) { |
76 for (int x = 0; x < tiles_needed(full.width(), FLAGS_quiltTile); x++) { | 76 for (int x = 0; x < tiles_needed(full.width(), FLAGS_quiltTile); x++) { |
77 // Deletes itself when done. | 77 // Deletes itself when done. |
78 pool.add(new Tile(x, y, fReference.colorType(), *recorded, &fullCanv
as, &mutex)); | 78 pool.add(new Tile(x, y, fReference.colorType(), *recorded, &fullCanv
as, &mutex)); |
79 } | 79 } |
80 } | 80 } |
81 | 81 |
82 pool.wait(); | 82 pool.wait(); |
83 fullCanvas.flush(); | 83 fullCanvas.flush(); |
84 | 84 |
85 if (!BitmapsEqual(full, fReference)) { | 85 if (!BitmapsEqual(full, fReference)) { |
86 this->fail(); | 86 this->fail(); |
87 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, full))); | 87 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, full))); |
88 } | 88 } |
89 } | 89 } |
90 | 90 |
91 bool QuiltTask::shouldSkip() const { | 91 bool QuiltTask::shouldSkip() const { |
92 if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { | 92 if (fGM->getFlags() & skiagm::GM::kSkipPicture_Flag) { |
93 return true; | 93 return true; |
94 } | 94 } |
95 if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { | 95 if (fGM->getFlags() & skiagm::GM::kSkipTiled_Flag) { |
96 return true; | 96 return true; |
97 } | 97 } |
98 return !FLAGS_quilt; | 98 return !FLAGS_quilt; |
99 } | 99 } |
100 | 100 |
101 } // namespace DM | 101 } // namespace DM |
OLD | NEW |