1 /* 2 * Copyright 2014 Google Inc. 3 * 4 * Use of this source code is governed by a BSD-style license that can be 5 * found in the LICENSE file. 6 */ 7 8 #include "SkDeque.h" 9 #include "SkLayerRasterizer.h" 10 #include "SkPaint.h" 11 #include "SkRasterizer.h" 12 #include "Test.h" 13 14 class SkReadBuffer; 15 16 // Dummy class to place on a paint just to ensure the paint's destructor 17 // is called. 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 { 21 public: 22 DummyRasterizer() 23 : INHERITED() 24 { 25 // Not threadsafe. Only used in one thread. 26 gCount++; 27 } 28 29 ~DummyRasterizer() { 30 // Not threadsafe. Only used in one thread. 31 gCount--; 32 } 33 34 static int GetCount() { return gCount; } 35 36 SK_DECLARE_NOT_FLATTENABLE_PROCS(DummyRasterizer) 37 38 private: 39 static int gCount; 40 41 typedef SkRasterizer INHERITED; 42 }; 43 44 int DummyRasterizer::gCount; 45 46 // Check to make sure that the SkPaint in the layer has its destructor called. 47 DEF_TEST(LayerRasterizer_destructor, reporter) { 48 { 49 SkPaint paint; 50 paint.setRasterizer(SkNEW(DummyRasterizer))->unref(); 51 REPORTER_ASSERT(reporter, DummyRasterizer::GetCount() == 1); 52 53 SkLayerRasterizer::Builder builder; 54 builder.addLayer(paint); 55 } 56 REPORTER_ASSERT(reporter, DummyRasterizer::GetCount() == 0); 57 } 58 59 class LayerRasterizerTester { 60 public: 61 static int CountLayers(const SkLayerRasterizer& layerRasterizer) { 62 return layerRasterizer.fLayers->count(); 63 } 64 65 static const SkDeque& GetLayers(const SkLayerRasterizer& layerRasterizer) { 66 return *layerRasterizer.fLayers; 67 } 68 }; 69 70 // MUST stay in sync with definition of SkLayerRasterizer_Rec in SkLayerRasterizer.cpp. 71 struct SkLayerRasterizer_Rec { 72 SkPaint fPaint; 73 SkVector fOffset; 74 }; 75 76 static bool equals(const SkLayerRasterizer_Rec& rec1, const SkLayerRasterizer_Rec& rec2) { 77 return rec1.fPaint == rec2.fPaint && rec1.fOffset == rec2.fOffset; 78 } 79 80 DEF_TEST(LayerRasterizer_copy, reporter) { 81 SkLayerRasterizer::Builder builder; 82 REPORTER_ASSERT(reporter, NULL == builder.snapshotRasterizer()); 83 SkPaint paint; 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)); 88 } 89 90 // Create a layer rasterizer with all the existing layers. 91 SkAutoTUnref<SkLayerRasterizer> firstCopy(builder.snapshotRasterizer()); 92 93 // Add one more layer. 94 paint.setFlags(SkPaint::kAllFlags); 95 builder.addLayer(paint); 96 97 SkAutoTUnref<SkLayerRasterizer> oneLarger(builder.snapshotRasterizer()); 98 SkAutoTUnref<SkLayerRasterizer> detached(builder.detachRasterizer()); 99 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); 105 106 const SkLayerRasterizer_Rec* recFirstCopy = NULL; 107 const SkLayerRasterizer_Rec* recOneLarger = NULL; 108 const SkLayerRasterizer_Rec* recDetached = NULL; 109 110 const SkDeque& layersFirstCopy = LayerRasterizerTester::GetLayers(*firstCopy.get()); 111 const SkDeque& layersOneLarger = LayerRasterizerTester::GetLayers(*oneLarger.get()); 112 const SkDeque& layersDetached = LayerRasterizerTester::GetLayers(*detached.get()); 113 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)); 120 121 SkDeque::F2BIter iterFirstCopy(layersFirstCopy); 122 SkDeque::F2BIter iterOneLarger(layersOneLarger); 123 SkDeque::F2BIter iterDetached(layersDetached); 124 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()); 129 130 REPORTER_ASSERT(reporter, equals(*recOneLarger, *recDetached)); 131 if (smallerCount == i) { 132 REPORTER_ASSERT(reporter, recFirstCopy == NULL); 133 } else { 134 REPORTER_ASSERT(reporter, equals(*recFirstCopy, *recOneLarger)); 135 } 136 } 137 } 138 139 DEF_TEST(LayerRasterizer_detachEmpty, reporter) { 140 SkLayerRasterizer::Builder builder; 141 REPORTER_ASSERT(reporter, NULL == builder.detachRasterizer()); 142 } 143