| OLD | NEW |
| 1 #include "DMSerializeTask.h" | 1 #include "DMSerializeTask.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 "SkPixelRef.h" | 7 #include "SkPixelRef.h" |
| 8 | 8 |
| 9 DEFINE_bool(serialize, true, "If true, run picture serialization tests."); | 9 DEFINE_bool(serialize, true, "If true, run picture serialization tests."); |
| 10 | 10 |
| 11 namespace DM { | 11 namespace DM { |
| 12 | 12 |
| 13 SerializeTask::SerializeTask(const Task& parent, | 13 SerializeTask::SerializeTask(const Task& parent, |
| 14 skiagm::GM* gm, | 14 skiagm::GM* gm, |
| 15 SkBitmap reference) | 15 SkBitmap reference) |
| 16 : Task(parent) | 16 : Task(parent) |
| 17 , fName(UnderJoin(parent.name().c_str(), "serialize")) | 17 , fName(UnderJoin(parent.name().c_str(), "serialize")) |
| 18 , fGM(gm) | 18 , fGM(gm) |
| 19 , fReference(reference) | 19 , fReference(reference) |
| 20 {} | 20 {} |
| 21 | 21 |
| 22 static SkData* trivial_bitmap_encoder(size_t* pixelRefOffset, const SkBitmap& bi
tmap) { | |
| 23 if (NULL == bitmap.pixelRef()) { | |
| 24 return NULL; | |
| 25 } | |
| 26 SkData* data = bitmap.pixelRef()->refEncodedData(); | |
| 27 *pixelRefOffset = bitmap.pixelRefOffset(); | |
| 28 return data; | |
| 29 } | |
| 30 | |
| 31 void SerializeTask::draw() { | 22 void SerializeTask::draw() { |
| 32 SkPicture recorded; | 23 SkPicture recorded; |
| 33 RecordPicture(fGM.get(), &recorded); | 24 RecordPicture(fGM.get(), &recorded); |
| 34 | 25 |
| 35 SkDynamicMemoryWStream wStream; | 26 SkDynamicMemoryWStream wStream; |
| 36 recorded.serialize(&wStream, &trivial_bitmap_encoder); | 27 recorded.serialize(&wStream, NULL); |
| 37 SkAutoTUnref<SkStream> rStream(wStream.detachAsStream()); | 28 SkAutoTUnref<SkStream> rStream(wStream.detachAsStream()); |
| 38 SkAutoTUnref<SkPicture> reconstructed(SkPicture::CreateFromStream(rStream)); | 29 SkAutoTUnref<SkPicture> reconstructed(SkPicture::CreateFromStream(rStream)); |
| 39 | 30 |
| 40 SkBitmap bitmap; | 31 SkBitmap bitmap; |
| 41 SetupBitmap(fReference.config(), fGM.get(), &bitmap); | 32 SetupBitmap(fReference.config(), fGM.get(), &bitmap); |
| 42 DrawPicture(reconstructed, &bitmap); | 33 DrawPicture(reconstructed, &bitmap); |
| 43 if (!BitmapsEqual(bitmap, fReference)) { | 34 if (!BitmapsEqual(bitmap, fReference)) { |
| 44 this->fail(); | 35 this->fail(); |
| 45 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap))); | 36 this->spawnChild(SkNEW_ARGS(WriteTask, (*this, bitmap))); |
| 46 } | 37 } |
| 47 } | 38 } |
| 48 | 39 |
| 49 bool SerializeTask::shouldSkip() const { | 40 bool SerializeTask::shouldSkip() const { |
| 50 return !FLAGS_serialize || fGM->getFlags() & skiagm::GM::kSkipPicture_Flag; | 41 return !FLAGS_serialize || fGM->getFlags() & skiagm::GM::kSkipPicture_Flag; |
| 51 } | 42 } |
| 52 | 43 |
| 53 } // namespace DM | 44 } // namespace DM |
| OLD | NEW |