Index: src/gpu/GrRODrawState.h |
diff --git a/src/gpu/GrRODrawState.h b/src/gpu/GrRODrawState.h |
index 0abd4a6aa54cce8169f5f9a16c93983a747a74d3..1cbc65bdd3f69662ae11e34f6797112c1c1ab2a5 100644 |
--- a/src/gpu/GrRODrawState.h |
+++ b/src/gpu/GrRODrawState.h |
@@ -121,8 +121,12 @@ public: |
int numColorStages() const { return fColorStages.count(); } |
int numCoverageStages() const { return fCoverageStages.count(); } |
- int numTotalStages() const { return this->numColorStages() + this->numCoverageStages(); } |
+ int numTotalStages() const { |
+ return this->numColorStages() + this->numCoverageStages() + |
+ (this->getGeometryProcessor() ? 1 : 0); |
+ } |
+ const GrEffectStage* getGeometryProcessor() const { return fGeometryProcessor.get(); } |
const GrEffectStage& getColorStage(int stageIdx) const { return fColorStages[stageIdx]; } |
const GrEffectStage& getCoverageStage(int stageIdx) const { return fCoverageStages[stageIdx]; } |
@@ -359,6 +363,7 @@ protected: |
GrBlendCoeff fDstBlend; |
typedef SkSTArray<4, GrEffectStage> EffectStageArray; |
+ SkAutoTUnref<GrEffectStage> fGeometryProcessor; |
EffectStageArray fColorStages; |
EffectStageArray fCoverageStages; |