2 * Copyright 2014 Google Inc.
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
9 #include "SkLayerRasterizer.h"
11 #include "SkRasterizer.h"
16 // Dummy class to place on a paint just to ensure the paint's destructor
18 // ONLY to be used by LayerRasterizer_destructor, since other tests may
19 // be run in a separate thread, and this class is not threadsafe.
20 class DummyRasterizer : public SkRasterizer {
25 // Not threadsafe. Only used in one thread.
30 // Not threadsafe. Only used in one thread.
34 static int GetCount() { return gCount; }
36 SK_DECLARE_NOT_FLATTENABLE_PROCS(DummyRasterizer)
41 typedef SkRasterizer INHERITED;
44 int DummyRasterizer::gCount;
46 // Check to make sure that the SkPaint in the layer has its destructor called.
47 DEF_TEST(LayerRasterizer_destructor, reporter) {
50 paint.setRasterizer(SkNEW(DummyRasterizer))->unref();
51 REPORTER_ASSERT(reporter, DummyRasterizer::GetCount() == 1);
53 SkLayerRasterizer::Builder builder;
54 builder.addLayer(paint);
56 REPORTER_ASSERT(reporter, DummyRasterizer::GetCount() == 0);
59 class LayerRasterizerTester {
61 static int CountLayers(const SkLayerRasterizer& layerRasterizer) {
62 return layerRasterizer.fLayers->count();
65 static const SkDeque& GetLayers(const SkLayerRasterizer& layerRasterizer) {
66 return *layerRasterizer.fLayers;
70 // MUST stay in sync with definition of SkLayerRasterizer_Rec in SkLayerRasterizer.cpp.
71 struct SkLayerRasterizer_Rec {
76 static bool equals(const SkLayerRasterizer_Rec& rec1, const SkLayerRasterizer_Rec& rec2) {
77 return rec1.fPaint == rec2.fPaint && rec1.fOffset == rec2.fOffset;
80 DEF_TEST(LayerRasterizer_copy, reporter) {
81 SkLayerRasterizer::Builder builder;
82 REPORTER_ASSERT(reporter, NULL == builder.snapshotRasterizer());
84 // Create a bunch of paints with different flags.
85 for (uint32_t flags = 0x01; flags < SkPaint::kAllFlags; flags <<= 1) {
86 paint.setFlags(flags);
87 builder.addLayer(paint, static_cast<SkScalar>(flags), static_cast<SkScalar>(flags));
90 // Create a layer rasterizer with all the existing layers.
91 SkAutoTUnref<SkLayerRasterizer> firstCopy(builder.snapshotRasterizer());
93 // Add one more layer.
94 paint.setFlags(SkPaint::kAllFlags);
95 builder.addLayer(paint);
97 SkAutoTUnref<SkLayerRasterizer> oneLarger(builder.snapshotRasterizer());
98 SkAutoTUnref<SkLayerRasterizer> detached(builder.detachRasterizer());
100 // Check the counts for consistency.
101 const int largerCount = LayerRasterizerTester::CountLayers(*oneLarger.get());
102 const int smallerCount = LayerRasterizerTester::CountLayers(*firstCopy.get());
103 REPORTER_ASSERT(reporter, largerCount == LayerRasterizerTester::CountLayers(*detached.get()));
104 REPORTER_ASSERT(reporter, smallerCount == largerCount - 1);
106 const SkLayerRasterizer_Rec* recFirstCopy = NULL;
107 const SkLayerRasterizer_Rec* recOneLarger = NULL;
108 const SkLayerRasterizer_Rec* recDetached = NULL;
110 const SkDeque& layersFirstCopy = LayerRasterizerTester::GetLayers(*firstCopy.get());
111 const SkDeque& layersOneLarger = LayerRasterizerTester::GetLayers(*oneLarger.get());
112 const SkDeque& layersDetached = LayerRasterizerTester::GetLayers(*detached.get());
114 // Ensure that our version of SkLayerRasterizer_Rec is the same as the one in
115 // SkLayerRasterizer.cpp - or at least the same size. If the order were switched, we
116 // would fail the test elsewhere.
117 REPORTER_ASSERT(reporter, layersFirstCopy.elemSize() == sizeof(SkLayerRasterizer_Rec));
118 REPORTER_ASSERT(reporter, layersOneLarger.elemSize() == sizeof(SkLayerRasterizer_Rec));
119 REPORTER_ASSERT(reporter, layersDetached.elemSize() == sizeof(SkLayerRasterizer_Rec));
121 SkDeque::F2BIter iterFirstCopy(layersFirstCopy);
122 SkDeque::F2BIter iterOneLarger(layersOneLarger);
123 SkDeque::F2BIter iterDetached(layersDetached);
125 for (int i = 0; i < largerCount; ++i) {
126 recFirstCopy = static_cast<const SkLayerRasterizer_Rec*>(iterFirstCopy.next());
127 recOneLarger = static_cast<const SkLayerRasterizer_Rec*>(iterOneLarger.next());
128 recDetached = static_cast<const SkLayerRasterizer_Rec*>(iterDetached.next());
130 REPORTER_ASSERT(reporter, equals(*recOneLarger, *recDetached));
131 if (smallerCount == i) {
132 REPORTER_ASSERT(reporter, recFirstCopy == NULL);
134 REPORTER_ASSERT(reporter, equals(*recFirstCopy, *recOneLarger));
139 DEF_TEST(LayerRasterizer_detachEmpty, reporter) {
140 SkLayerRasterizer::Builder builder;
141 REPORTER_ASSERT(reporter, NULL == builder.detachRasterizer());