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