Calculate stage stats in GrOptDrawState

src/gpu/GrOptDrawState.cpp

 /*
  * Copyright 2014 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "GrOptDrawState.h"
 
 #include "GrDrawState.h"
@@ skipping 26 matching lines @@
 
     if (drawState.hasGeometryProcessor()) {
         fGeometryProcessor.reset(SkNEW_ARGS(GrEffectStage, (*drawState.getGeometryProcessor())));
     } else {
         fGeometryProcessor.reset(NULL);
     }
 
     this->copyEffectiveColorStages(drawState);
     this->copyEffectiveCoverageStages(drawState);
     this->adjustFromBlendOpts();
+    this->getStageStats();
 };
 
 void GrOptDrawState::adjustFromBlendOpts() {
 
     switch (fBlendOptFlags) {
         case kNone_BlendOpt:
         case kSkipDraw_BlendOptFlag:
             break;
         case kCoverageAsAlpha_BlendOptFlag:
             fFlagBits |= kCoverageDrawing_StateBit;
@@ skipping 115 matching lines @@
     }
 #endif
     if (ds.numCoverageStages() > 0) {
         fCoverageStages.reset(&ds.getCoverageStage(firstCoverageStage),
                               ds.numCoverageStages() - firstCoverageStage);
     } else {
         fCoverageStages.reset();
     }
 }
 
+static void get_stage_stats(const GrEffectStage& stage, bool* readsDst,
+                            bool* readsFragPosition, bool* requiresVertexShader) {
+    if (stage.getEffect()->willReadDstColor()) {
+        *readsDst = true;
+    }
+    if (stage.getEffect()->willReadFragmentPosition()) {
+        *readsFragPosition = true;
+    }
+    if (stage.getEffect()->requiresVertexShader()) {
+        *requiresVertexShader = true;
+    }
+}
+void GrOptDrawState::getStageStats() {
+    fReadsDst = false;
+    fReadsFragPosition = false;
+    fRequiresVertexShader = false;
+
+    for (int s = 0; s < this->numColorStages(); ++s) {
+        const GrEffectStage& stage = this->getColorStage(s);
+        get_stage_stats(stage, &fReadsDst, &fReadsFragPosition, &fRequiresVertexShader);
+    }
+    for (int s = 0; s < this->numCoverageStages(); ++s) {
+        const GrEffectStage& stage = this->getCoverageStage(s);
+        get_stage_stats(stage, &fReadsDst, &fReadsFragPosition, &fRequiresVertexShader);
+    }
+    // TODO: Once non Geom Processors can't be vertex effects we can remove this assert
+    SkASSERT(!fRequiresVertexShader);
+    if (this->hasGeometryProcessor()) {
+        const GrEffectStage& stage = *this->getGeometryProcessor();
+        get_stage_stats(stage, &fReadsDst, &fReadsFragPosition, &fRequiresVertexShader);
+        SkASSERT(fRequiresVertexShader);
+    }
+    // TODO: Once we have actually separate GP's from other effects we can just set
+    // fRequiresVertexShader to be hasGeometryProcessor (or actaully just drop the bool all
+    // together and just check for GP when needed.
+    SkASSERT(this->hasGeometryProcessor() == fRequiresVertexShader);
+}
+
 bool GrOptDrawState::operator== (const GrOptDrawState& that) const {
     return this->isEqual(that);
 }