1 /* 2 * Copyright 2015 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 "GrNinePatch.h" 9 10 #include "GrBatchFlushState.h" 11 #include "GrDefaultGeoProcFactory.h" 12 #include "GrResourceProvider.h" 13 #include "GrVertexBatch.h" 14 #include "SkBitmap.h" 15 #include "SkNinePatchIter.h" 16 #include "SkRect.h" 17 18 static const GrGeometryProcessor* create_gp(bool readsCoverage) { 19 using namespace GrDefaultGeoProcFactory; 20 Color color(Color::kAttribute_Type); 21 Coverage coverage(readsCoverage ? Coverage::kSolid_Type : Coverage::kNone_Type); 22 LocalCoords localCoords(LocalCoords::kHasExplicit_Type); 23 return GrDefaultGeoProcFactory::Create(color, coverage, localCoords, SkMatrix::I()); 24 } 25 26 class GrNonAANinePatchBatch : public GrVertexBatch { 27 public: 28 DEFINE_BATCH_CLASS_ID 29 30 static const int kVertsPerRect = 4; 31 static const int kIndicesPerRect = 6; 32 static const int kRectsPerInstance = 9; // We could skip empty rects 33 34 struct Geometry { 35 SkMatrix fViewMatrix; 36 SkIRect fCenter; 37 SkRect fDst; 38 GrColor fColor; 39 }; 40 41 GrNonAANinePatchBatch(GrColor color, const SkMatrix& viewMatrix, int imageWidth, 42 int imageHeight, const SkIRect& center, const SkRect &dst) 43 : INHERITED(ClassID()) { 44 Geometry& geo = fGeoData.push_back(); 45 geo.fViewMatrix = viewMatrix; 46 geo.fColor = color; 47 geo.fCenter = center; 48 geo.fDst = dst; 49 50 fImageWidth = imageWidth; 51 fImageHeight = imageHeight; 52 53 // setup bounds 54 geo.fViewMatrix.mapRect(&fBounds, geo.fDst); 55 } 56 57 const char* name() const override { return "NonAANinePatchBatch"; } 58 59 SkString dumpInfo() const override { 60 SkString str; 61 62 for (int i = 0; i < fGeoData.count(); ++i) { 63 str.appendf("%d: Color: 0x%08x Center [L: %d, T: %d, R: %d, B: %d], " 64 "Dst [L: %.2f, T: %.2f, R: %.2f, B: %.2f]\n", 65 i, 66 fGeoData[i].fColor, 67 fGeoData[i].fCenter.fLeft, fGeoData[i].fCenter.fTop, 68 fGeoData[i].fCenter.fRight, fGeoData[i].fCenter.fBottom, 69 fGeoData[i].fDst.fLeft, fGeoData[i].fDst.fTop, 70 fGeoData[i].fDst.fRight, fGeoData[i].fDst.fBottom); 71 } 72 73 str.append(INHERITED::dumpInfo()); 74 return str; 75 } 76 77 void computePipelineOptimizations(GrInitInvariantOutput* color, 78 GrInitInvariantOutput* coverage, 79 GrBatchToXPOverrides* overrides) const override { 80 color->setUnknownFourComponents(); 81 coverage->setKnownSingleComponent(0xff); 82 } 83 84 SkSTArray<1, Geometry, true>* geoData() { return &fGeoData; } 85 86 private: 87 void onPrepareDraws(Target* target) const override { 88 SkAutoTUnref<const GrGeometryProcessor> gp(create_gp(fOverrides.readsCoverage())); 89 if (!gp) { 90 SkDebugf("Couldn't create GrGeometryProcessor\n"); 91 return; 92 } 93 94 target->initDraw(gp, this->pipeline()); 95 96 size_t vertexStride = gp->getVertexStride(); 97 int instanceCount = fGeoData.count(); 98 99 SkAutoTUnref<const GrIndexBuffer> indexBuffer( 100 target->resourceProvider()->refQuadIndexBuffer()); 101 InstancedHelper helper; 102 void* vertices = helper.init(target, kTriangles_GrPrimitiveType, vertexStride, 103 indexBuffer, kVertsPerRect, 104 kIndicesPerRect, instanceCount * kRectsPerInstance); 105 if (!vertices || !indexBuffer) { 106 SkDebugf("Could not allocate vertices\n"); 107 return; 108 } 109 110 for (int i = 0; i < instanceCount; i++) { 111 intptr_t verts = reinterpret_cast<intptr_t>(vertices) + 112 i * kRectsPerInstance * kVertsPerRect * vertexStride; 113 114 const Geometry& geo = fGeoData[i]; 115 SkNinePatchIter iter(fImageWidth, fImageHeight, geo.fCenter, geo.fDst); 116 117 SkRect srcR, dstR; 118 while (iter.next(&srcR, &dstR)) { 119 SkPoint* positions = reinterpret_cast<SkPoint*>(verts); 120 121 positions->setRectFan(dstR.fLeft, dstR.fTop, 122 dstR.fRight, dstR.fBottom, vertexStride); 123 124 SkASSERT(!geo.fViewMatrix.hasPerspective()); 125 geo.fViewMatrix.mapPointsWithStride(positions, vertexStride, kVertsPerRect); 126 127 // Setup local coords 128 static const int kLocalOffset = sizeof(SkPoint) + sizeof(GrColor); 129 SkPoint* coords = reinterpret_cast<SkPoint*>(verts + kLocalOffset); 130 coords->setRectFan(srcR.fLeft, srcR.fTop, srcR.fRight, srcR.fBottom, vertexStride); 131 132 static const int kColorOffset = sizeof(SkPoint); 133 GrColor* vertColor = reinterpret_cast<GrColor*>(verts + kColorOffset); 134 for (int j = 0; j < 4; ++j) { 135 *vertColor = geo.fColor; 136 vertColor = (GrColor*) ((intptr_t) vertColor + vertexStride); 137 } 138 verts += kVertsPerRect * vertexStride; 139 } 140 } 141 helper.recordDraw(target); 142 } 143 144 void initBatchTracker(const GrXPOverridesForBatch& overrides) override { 145 overrides.getOverrideColorIfSet(&fGeoData[0].fColor); 146 fOverrides = overrides; 147 } 148 149 bool onCombineIfPossible(GrBatch* t, const GrCaps& caps) override { 150 GrNonAANinePatchBatch* that = t->cast<GrNonAANinePatchBatch>(); 151 if (!GrPipeline::CanCombine(*this->pipeline(), this->bounds(), *that->pipeline(), 152 that->bounds(), caps)) { 153 return false; 154 } 155 156 SkASSERT(this->fImageWidth == that->fImageWidth && 157 this->fImageHeight == that->fImageHeight); 158 159 // In the event of two batches, one who can tweak, one who cannot, we just fall back to 160 // not tweaking 161 if (fOverrides.canTweakAlphaForCoverage() && !that->fOverrides.canTweakAlphaForCoverage()) { 162 fOverrides = that->fOverrides; 163 } 164 165 fGeoData.push_back_n(that->geoData()->count(), that->geoData()->begin()); 166 this->joinBounds(that->bounds()); 167 return true; 168 } 169 170 GrXPOverridesForBatch fOverrides; 171 int fImageWidth; 172 int fImageHeight; 173 SkSTArray<1, Geometry, true> fGeoData; 174 175 typedef GrVertexBatch INHERITED; 176 }; 177 178 namespace GrNinePatch { 179 GrDrawBatch* CreateNonAA(GrColor color, const SkMatrix& viewMatrix, int imageWidth, int imageHeight, 180 const SkIRect& center, const SkRect& dst) { 181 return new GrNonAANinePatchBatch(color, viewMatrix, imageWidth, imageHeight, center, dst); 182 } 183 }; 184
