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
9 DEFINE_bool(serialize, true, "If true, run picture serialization tests.");
10
11 namespace DM {
12
SerializeTask(const Task & parent,skiagm::GM * gm,SkBitmap reference)13 SerializeTask::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
trivial_bitmap_encoder(size_t * pixelRefOffset,const SkBitmap & bitmap)22 static 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
draw()31 void 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
shouldSkip() const49 bool SerializeTask::shouldSkip() const {
50 return !FLAGS_serialize || fGM->getFlags() & skiagm::GM::kSkipPicture_Flag;
51 }
52
53 } // namespace DM
54