| /* |
| * Copyright 2016 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "SkCanvas.h" |
| #include "SkDrawable.h" |
| #include "SkOnce.h" |
| #include "SkPictureRecorder.h" |
| #include "SkReadBuffer.h" |
| #include "SkRect.h" |
| #include "SkStream.h" |
| #include "SkWriteBuffer.h" |
| #include "Test.h" |
| |
| class IntDrawable : public SkDrawable { |
| public: |
| IntDrawable(uint32_t a, uint32_t b, uint32_t c, uint32_t d) |
| : fA(a) |
| , fB(b) |
| , fC(c) |
| , fD(d) |
| {} |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| buffer.writeUInt(fA); |
| buffer.writeUInt(fB); |
| buffer.writeUInt(fC); |
| buffer.writeUInt(fD); |
| } |
| |
| static sk_sp<SkFlattenable> CreateProc(SkReadBuffer& buffer) { |
| uint32_t a = buffer.readUInt(); |
| uint32_t b = buffer.readUInt(); |
| uint32_t c = buffer.readUInt(); |
| uint32_t d = buffer.readUInt(); |
| return sk_sp<IntDrawable>(new IntDrawable(a, b, c, d)); |
| } |
| |
| Factory getFactory() const override { return CreateProc; } |
| |
| uint32_t a() const { return fA; } |
| uint32_t b() const { return fB; } |
| uint32_t c() const { return fC; } |
| uint32_t d() const { return fD; } |
| |
| const char* getTypeName() const override { return "IntDrawable"; } |
| |
| protected: |
| SkRect onGetBounds() override { return SkRect::MakeEmpty(); } |
| void onDraw(SkCanvas*) override {} |
| |
| private: |
| uint32_t fA; |
| uint32_t fB; |
| uint32_t fC; |
| uint32_t fD; |
| }; |
| |
| class PaintDrawable : public SkDrawable { |
| public: |
| PaintDrawable(const SkPaint& paint) |
| : fPaint(paint) |
| {} |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| buffer.writePaint(fPaint); |
| } |
| |
| static sk_sp<SkFlattenable> CreateProc(SkReadBuffer& buffer) { |
| SkPaint paint; |
| buffer.readPaint(&paint); |
| return sk_sp<PaintDrawable>(new PaintDrawable(paint)); |
| } |
| |
| Factory getFactory() const override { return CreateProc; } |
| |
| const SkPaint& paint() const { return fPaint; } |
| |
| const char* getTypeName() const override { return "PaintDrawable"; } |
| |
| protected: |
| SkRect onGetBounds() override { return SkRect::MakeEmpty(); } |
| void onDraw(SkCanvas*) override {} |
| |
| private: |
| SkPaint fPaint; |
| }; |
| |
| class CompoundDrawable : public SkDrawable { |
| public: |
| CompoundDrawable(uint32_t a, uint32_t b, uint32_t c, uint32_t d, const SkPaint& paint) |
| : fIntDrawable(new IntDrawable(a, b, c, d)) |
| , fPaintDrawable(new PaintDrawable(paint)) |
| {} |
| |
| CompoundDrawable(IntDrawable* intDrawable, PaintDrawable* paintDrawable) |
| : fIntDrawable(SkRef(intDrawable)) |
| , fPaintDrawable(SkRef(paintDrawable)) |
| {} |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| buffer.writeFlattenable(fIntDrawable.get()); |
| buffer.writeFlattenable(fPaintDrawable.get()); |
| } |
| |
| static sk_sp<SkFlattenable> CreateProc(SkReadBuffer& buffer) { |
| sk_sp<SkFlattenable> intDrawable( |
| buffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| SkASSERT(intDrawable); |
| SkASSERT(!strcmp("IntDrawable", intDrawable->getTypeName())); |
| |
| sk_sp<SkFlattenable> paintDrawable( |
| buffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| SkASSERT(paintDrawable); |
| SkASSERT(!strcmp("PaintDrawable", paintDrawable->getTypeName())); |
| |
| return sk_sp<CompoundDrawable>(new CompoundDrawable((IntDrawable*) intDrawable.get(), |
| (PaintDrawable*) paintDrawable.get())); |
| } |
| |
| Factory getFactory() const override { return CreateProc; } |
| |
| IntDrawable* intDrawable() const { return fIntDrawable.get(); } |
| PaintDrawable* paintDrawable() const { return fPaintDrawable.get(); } |
| |
| const char* getTypeName() const override { return "CompoundDrawable"; } |
| |
| protected: |
| SkRect onGetBounds() override { return SkRect::MakeEmpty(); } |
| void onDraw(SkCanvas*) override {} |
| |
| private: |
| sk_sp<IntDrawable> fIntDrawable; |
| sk_sp<PaintDrawable> fPaintDrawable; |
| }; |
| |
| class RootDrawable : public SkDrawable { |
| public: |
| RootDrawable(uint32_t a, uint32_t b, uint32_t c, uint32_t d, const SkPaint& paint, |
| uint32_t e, uint32_t f, uint32_t g, uint32_t h, SkDrawable* drawable) |
| : fCompoundDrawable(new CompoundDrawable(a, b, c, d, paint)) |
| , fIntDrawable(new IntDrawable(e, f, g, h)) |
| , fDrawable(SkRef(drawable)) |
| {} |
| |
| RootDrawable(CompoundDrawable* compoundDrawable, IntDrawable* intDrawable, |
| SkDrawable* drawable) |
| : fCompoundDrawable(SkRef(compoundDrawable)) |
| , fIntDrawable(SkRef(intDrawable)) |
| , fDrawable(SkRef(drawable)) |
| {} |
| |
| void flatten(SkWriteBuffer& buffer) const override { |
| buffer.writeFlattenable(fCompoundDrawable.get()); |
| buffer.writeFlattenable(fIntDrawable.get()); |
| buffer.writeFlattenable(fDrawable.get()); |
| } |
| |
| static sk_sp<SkFlattenable> CreateProc(SkReadBuffer& buffer) { |
| sk_sp<SkFlattenable> compoundDrawable( |
| buffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| SkASSERT(compoundDrawable); |
| SkASSERT(!strcmp("CompoundDrawable", compoundDrawable->getTypeName())); |
| |
| sk_sp<SkFlattenable> intDrawable( |
| buffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| SkASSERT(intDrawable); |
| SkASSERT(!strcmp("IntDrawable", intDrawable->getTypeName())); |
| |
| sk_sp<SkFlattenable> drawable( |
| buffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| SkASSERT(drawable); |
| |
| return sk_sp<RootDrawable>(new RootDrawable((CompoundDrawable*) compoundDrawable.get(), |
| (IntDrawable*) intDrawable.get(), |
| (SkDrawable*) drawable.get())); |
| } |
| |
| Factory getFactory() const override { return CreateProc; } |
| |
| CompoundDrawable* compoundDrawable() const { return fCompoundDrawable.get(); } |
| IntDrawable* intDrawable() const { return fIntDrawable.get(); } |
| SkDrawable* drawable() const { return fDrawable.get(); } |
| |
| const char* getTypeName() const override { return "RootDrawable"; } |
| |
| protected: |
| SkRect onGetBounds() override { return SkRect::MakeEmpty(); } |
| void onDraw(SkCanvas*) override {} |
| |
| private: |
| sk_sp<CompoundDrawable> fCompoundDrawable; |
| sk_sp<IntDrawable> fIntDrawable; |
| sk_sp<SkDrawable> fDrawable; |
| }; |
| |
| static void register_test_drawables(SkReadBuffer& buffer) { |
| buffer.setCustomFactory(SkString("IntDrawable"), IntDrawable::CreateProc); |
| buffer.setCustomFactory(SkString("PaintDrawable"), PaintDrawable::CreateProc); |
| buffer.setCustomFactory(SkString("CompoundDrawable"), CompoundDrawable::CreateProc); |
| buffer.setCustomFactory(SkString("RootDrawable"), RootDrawable::CreateProc); |
| } |
| |
| DEF_TEST(FlattenDrawable, r) { |
| // Create and serialize the test drawable |
| sk_sp<SkDrawable> drawable(new IntDrawable(1, 2, 3, 4)); |
| SkPaint paint; |
| paint.setColor(SK_ColorBLUE); |
| sk_sp<RootDrawable> root(new RootDrawable(5, 6, 7, 8, paint, 9, 10, 11, 12, drawable.get())); |
| SkBinaryWriteBuffer writeBuffer; |
| writeBuffer.writeFlattenable(root.get()); |
| |
| // Copy the contents of the write buffer into a read buffer |
| sk_sp<SkData> data = SkData::MakeUninitialized(writeBuffer.bytesWritten()); |
| writeBuffer.writeToMemory(data->writable_data()); |
| SkReadBuffer readBuffer(data->data(), data->size()); |
| register_test_drawables(readBuffer); |
| |
| // Deserialize and verify the drawable |
| sk_sp<SkDrawable> out((SkDrawable*)readBuffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| REPORTER_ASSERT(r, out); |
| REPORTER_ASSERT(r, !strcmp("RootDrawable", out->getTypeName())); |
| |
| RootDrawable* rootOut = (RootDrawable*) out.get(); |
| REPORTER_ASSERT(r, 5 == rootOut->compoundDrawable()->intDrawable()->a()); |
| REPORTER_ASSERT(r, 6 == rootOut->compoundDrawable()->intDrawable()->b()); |
| REPORTER_ASSERT(r, 7 == rootOut->compoundDrawable()->intDrawable()->c()); |
| REPORTER_ASSERT(r, 8 == rootOut->compoundDrawable()->intDrawable()->d()); |
| REPORTER_ASSERT(r, SK_ColorBLUE == |
| rootOut->compoundDrawable()->paintDrawable()->paint().getColor()); |
| REPORTER_ASSERT(r, 9 == rootOut->intDrawable()->a()); |
| REPORTER_ASSERT(r, 10 == rootOut->intDrawable()->b()); |
| REPORTER_ASSERT(r, 11 == rootOut->intDrawable()->c()); |
| REPORTER_ASSERT(r, 12 == rootOut->intDrawable()->d()); |
| |
| // Note that we can still recognize the generic drawable as an IntDrawable |
| SkDrawable* generic = rootOut->drawable(); |
| REPORTER_ASSERT(r, !strcmp("IntDrawable", generic->getTypeName())); |
| IntDrawable* integer = (IntDrawable*) generic; |
| REPORTER_ASSERT(r, 1 == integer->a()); |
| REPORTER_ASSERT(r, 2 == integer->b()); |
| REPORTER_ASSERT(r, 3 == integer->c()); |
| REPORTER_ASSERT(r, 4 == integer->d()); |
| } |
| |
| DEF_TEST(FlattenRecordedDrawable, r) { |
| // Record a set of canvas draw commands |
| SkPictureRecorder recorder; |
| SkCanvas* canvas = recorder.beginRecording(1000.0f, 1000.0f); |
| SkPaint paint; |
| paint.setColor(SK_ColorGREEN); |
| canvas->drawPoint(42.0f, 17.0f, paint); |
| paint.setColor(SK_ColorRED); |
| canvas->drawPaint(paint); |
| SkPaint textPaint; |
| textPaint.setColor(SK_ColorBLUE); |
| canvas->drawText("TEXT", 4, 467.0f, 100.0f, textPaint); |
| |
| // Draw some drawables as well |
| sk_sp<SkDrawable> drawable(new IntDrawable(1, 2, 3, 4)); |
| sk_sp<RootDrawable> root(new RootDrawable(5, 6, 7, 8, paint, 9, 10, 11, 12, drawable.get())); |
| canvas->drawDrawable(root.get(), 747.0f, 242.0f); |
| sk_sp<PaintDrawable> paintDrawable(new PaintDrawable(paint)); |
| canvas->drawDrawable(paintDrawable.get(), 500.0, 500.0f); |
| sk_sp<CompoundDrawable> comDrawable(new CompoundDrawable(13, 14, 15, 16, textPaint)); |
| canvas->drawDrawable(comDrawable.get(), 10.0f, 10.0f); |
| |
| // Serialize the recorded drawable |
| sk_sp<SkDrawable> recordedDrawable = recorder.finishRecordingAsDrawable(); |
| SkBinaryWriteBuffer writeBuffer; |
| writeBuffer.writeFlattenable(recordedDrawable.get()); |
| |
| // Copy the contents of the write buffer into a read buffer |
| sk_sp<SkData> data = SkData::MakeUninitialized(writeBuffer.bytesWritten()); |
| writeBuffer.writeToMemory(data->writable_data()); |
| SkReadBuffer readBuffer(data->data(), data->size()); |
| register_test_drawables(readBuffer); |
| |
| // Deserialize and verify the drawable |
| sk_sp<SkDrawable> out((SkDrawable*)readBuffer.readFlattenable(SkFlattenable::kSkDrawable_Type)); |
| REPORTER_ASSERT(r, out); |
| REPORTER_ASSERT(r, !strcmp("SkRecordedDrawable", out->getTypeName())); |
| } |