DMSerializeTask.cpp revision 5b39f5ba9c339d1e4dae391fee9ec1396feec180 
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, false, "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