Use vertexless shaders when NVpr is available

src/gpu/gl/GrGLProgramEffects.cpp

 /*
  * Copyright 2013 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "GrGLProgramEffects.h"
 #include "GrDrawEffect.h"
 #include "gl/GrGLEffect.h"
@@ skipping 19 matching lines @@
 
 /**
  * The key for an individual coord transform is made up of a matrix type and a bit that
  * indicates the source of the input coords.
  */
 enum {
     kMatrixTypeKeyBits   = 2,
     kMatrixTypeKeyMask   = (1 << kMatrixTypeKeyBits) - 1,
     kPositionCoords_Flag = (1 << kMatrixTypeKeyBits),
     kTransformKeyBits    = kMatrixTypeKeyBits + 1,
-    kTransformKeyMask    = (1 << kTransformKeyBits) - 1,
 };
 
 namespace {
 
 /**
  * Do we need to either map r,g,b->a or a->r. configComponentMask indicates which channels are
  * present in the texture's config. swizzleComponentMask indicates the channels present in the
  * shader swizzle.
  */
 inline bool swizzle_requires_alpha_remapping(const GrGLCaps& caps,
@@ skipping 11 matching lines @@
         }
         if (kRGB_GrColorComponentFlags & swizzleComponentMask) {
             // The 'r', 'g', and/or 'b's must be mapped to 'a' according to our semantics that
             // alpha-only textures smear alpha across all four channels when read.
             return true;
         }
     }
     return false;
 }
 
+/**
+ * Retrieves the matrix type from transformKey for the transform at transformIdx.
+ */
+MatrixType get_matrix_type(EffectKey transformKey, int transformIdx) {
+    return static_cast<MatrixType>(
+               (transformKey >> (kTransformKeyBits * transformIdx)) & kMatrixTypeKeyMask);
+}
+
+/**
+ * Retrieves the source coords from transformKey for the transform at transformIdx. It may not be
+ * the same coordinate set as the original GrCoordTransform if the position and local coords are
+ * identical for this program.
+ */
+GrCoordSet get_source_coords(EffectKey transformKey, int transformIdx) {
+    return (transformKey >> (kTransformKeyBits * transformIdx)) & kPositionCoords_Flag ?
+               kPosition_GrCoordSet :
+               kLocal_GrCoordSet;
+}
+
+/**
+ * Retrieves the final translation that a transform needs to apply to its source coords (and
+ * verifies that a translation is all it needs).
+ */
+void get_transform_translation(const GrDrawEffect& drawEffect,
+                               int transformIdx,
+                               GrGLfloat* tx,
+                               GrGLfloat* ty) {
+    const GrCoordTransform& coordTransform = (*drawEffect.effect())->coordTransform(transformIdx);
+    SkASSERT(!coordTransform.reverseY());
+    const SkMatrix& matrix = coordTransform.getMatrix();
+    if (kLocal_GrCoordSet == coordTransform.sourceCoords() &&
+        !drawEffect.programHasExplicitLocalCoords()) {
+        const SkMatrix& coordChangeMatrix = drawEffect.getCoordChangeMatrix();
+        SkASSERT(SkMatrix::kTranslate_Mask == (matrix.getType() | coordChangeMatrix.getType()));
+        *tx = SkScalarToFloat(matrix[SkMatrix::kMTransX] + coordChangeMatrix[SkMatrix::kMTransX]);
+        *ty = SkScalarToFloat(matrix[SkMatrix::kMTransY] + coordChangeMatrix[SkMatrix::kMTransY]);
+    } else {
+        SkASSERT(SkMatrix::kTranslate_Mask == matrix.getType());
+        *tx = SkScalarToFloat(matrix[SkMatrix::kMTransX]);
+        *ty = SkScalarToFloat(matrix[SkMatrix::kMTransY]);
+    }
+}
+
+/**
+ * Retrieves the final matrix that a transform needs to apply to its source coords.
+ */
+SkMatrix get_transform_matrix(const GrDrawEffect& drawEffect, int transformIdx) {
+    const GrCoordTransform& coordTransform = (*drawEffect.effect())->coordTransform(transformIdx);
+    SkMatrix combined;
+    if (kLocal_GrCoordSet == coordTransform.sourceCoords() &&
+        !drawEffect.programHasExplicitLocalCoords()) {
+        combined.setConcat(coordTransform.getMatrix(), drawEffect.getCoordChangeMatrix());
+    } else {
+        combined = coordTransform.getMatrix();
+    }
+    if (coordTransform.reverseY()) {
+        // combined.postScale(1,-1);
+        // combined.postTranslate(0,1);
+        combined.set(SkMatrix::kMSkewY,
+            combined[SkMatrix::kMPersp0] - combined[SkMatrix::kMSkewY]);
+        combined.set(SkMatrix::kMScaleY,
+            combined[SkMatrix::kMPersp1] - combined[SkMatrix::kMScaleY]);
+        combined.set(SkMatrix::kMTransY,
+            combined[SkMatrix::kMPersp2] - combined[SkMatrix::kMTransY]);
+    }
+    return combined;
+}
+
 }
 
 EffectKey GrGLProgramEffects::GenAttribKey(const GrDrawEffect& drawEffect) {
     EffectKey key = 0;
     int numAttributes = drawEffect.getVertexAttribIndexCount();
     SkASSERT(numAttributes <= 2);
     const int* attributeIndices = drawEffect.getVertexAttribIndices();
     for (int a = 0; a < numAttributes; ++a) {
         EffectKey value = attributeIndices[a] << 3 * a;
         SkASSERT(0 == (value & key)); // keys for each attribute ought not to overlap
@@ skipping 90 matching lines @@
     }
 }
 
 void GrGLVertexProgramEffects::setTransformData(const GrGLUniformManager& uniformManager,
                                                 const GrDrawEffect& drawEffect,
                                                 int effectIdx) {
     SkTArray<Transform, true>& transforms = fTransforms[effectIdx];
     int numTransforms = transforms.count();
     SkASSERT(numTransforms == (*drawEffect.effect())->numTransforms());
     for (int t = 0; t < numTransforms; ++t) {
-        const GrCoordTransform& coordTransform = (*drawEffect.effect())->coordTransform(t);
-        const SkMatrix& matrix = coordTransform.getMatrix();
-        const SkMatrix& coordChangeMatrix = kLocal_GrCoordSet == coordTransform.sourceCoords() ?
-                                                drawEffect.getCoordChangeMatrix() :
-                                                SkMatrix::I();
         SkASSERT(transforms[t].fHandle.isValid() != (kVoid_GrSLType == transforms[t].fType));
         switch (transforms[t].fType) {
             case kVoid_GrSLType:
-                SkASSERT(matrix.isIdentity());
-                SkASSERT(coordChangeMatrix.isIdentity());
-                SkASSERT(!coordTransform.reverseY());
+                SkASSERT(get_transform_matrix(drawEffect, t).isIdentity());
                 return;
             case kVec2f_GrSLType: {
-                SkASSERT(SkMatrix::kTranslate_Mask == (matrix.getType() | coordChangeMatrix.getType()));
-                SkASSERT(!coordTransform.reverseY());
-                SkScalar tx = matrix[SkMatrix::kMTransX] + (coordChangeMatrix)[SkMatrix::kMTransX];
-                SkScalar ty = matrix[SkMatrix::kMTransY] + (coordChangeMatrix)[SkMatrix::kMTransY];
+                GrGLfloat tx, ty;
+                get_transform_translation(drawEffect, t, &tx, &ty);
                 if (transforms[t].fCurrentValue.get(SkMatrix::kMTransX) != tx ||
                     transforms[t].fCurrentValue.get(SkMatrix::kMTransY) != ty) {
                     uniformManager.set2f(transforms[t].fHandle, tx, ty);
                     transforms[t].fCurrentValue.set(SkMatrix::kMTransX, tx);
                     transforms[t].fCurrentValue.set(SkMatrix::kMTransY, ty);
                 }
                 break;
             }
             case kMat33f_GrSLType: {
-                SkMatrix combined;
-                combined.setConcat(matrix, coordChangeMatrix);
-                if (coordTransform.reverseY()) {
-                    // combined.postScale(1,-1);
-                    // combined.postTranslate(0,1);
-                    combined.set(SkMatrix::kMSkewY,
-                        combined[SkMatrix::kMPersp0] - combined[SkMatrix::kMSkewY]);
-                    combined.set(SkMatrix::kMScaleY,
-                        combined[SkMatrix::kMPersp1] - combined[SkMatrix::kMScaleY]);
-                    combined.set(SkMatrix::kMTransY,
-                        combined[SkMatrix::kMPersp2] - combined[SkMatrix::kMTransY]);
-                }
-                if (!transforms[t].fCurrentValue.cheapEqualTo(combined)) {
-                    uniformManager.setSkMatrix(transforms[t].fHandle, combined);
-                    transforms[t].fCurrentValue = combined;
+                const SkMatrix& matrix = get_transform_matrix(drawEffect, t);
+                if (!transforms[t].fCurrentValue.cheapEqualTo(matrix)) {
+                    uniformManager.setSkMatrix(transforms[t].fHandle, matrix);
+                    transforms[t].fCurrentValue = matrix;
                 }
                 break;
             }
             default:
                 GrCrash("Unexpected uniform type.");
         }
     }
 }
 
+void GrGLTexGenProgramEffects::setData(GrGpuGL* gpu,
+                                       const GrGLUniformManager& uniformManager,
+                                       const GrEffectStage* effectStages[]) {
+    int numEffects = fGLEffects.count();
+    SkASSERT(numEffects == fTransformKeys.count());
+    SkASSERT(numEffects == fSamplers.count());
+    int texCoordIdx = fFirstTexCoordIdx;
+    SkASSERT(texCoordIdx >= 0);
+    for (int e = 0; e < numEffects; ++e) {
+        GrDrawEffect drawEffect(*effectStages[e], false);
+        fGLEffects[e]->setData(uniformManager, drawEffect);
+        this->setTexGenState(gpu, drawEffect, &texCoordIdx, e);
+        this->bindTextures(gpu, *drawEffect.effect(), e);
+    }
+    SkASSERT(fNumTexCoordSets == texCoordIdx - fFirstTexCoordIdx);
+}
+
+void GrGLTexGenProgramEffects::setTexGenState(GrGpuGL* gpu,
+                                              const GrDrawEffect& drawEffect,
+                                              int* texCoordIdx,
+                                              int effectIdx) {
+    EffectKey transformKey = fTransformKeys[effectIdx];
+    int numTransforms = (*drawEffect.effect())->numTransforms();
+    for (int t = 0; t < numTransforms; ++t) {
+        switch (get_matrix_type(transformKey, t)) {
+            case kIdentity_MatrixType: {
+                SkASSERT(get_transform_matrix(drawEffect, t).isIdentity());
+                GrGLfloat identity[] = {1, 0, 0,
+                                        0, 1, 0};
+                gpu->enableTexGen((*texCoordIdx)++, GrGpuGL::kST_TexGenComponents, identity);
+                break;
+            }
+            case kTrans_MatrixType: {
+                GrGLfloat tx, ty;
+                get_transform_translation(drawEffect, t, &tx, &ty);
+                GrGLfloat translate[] = {1, 0, tx,
+                                         0, 1, ty};
+                gpu->enableTexGen((*texCoordIdx)++, GrGpuGL::kST_TexGenComponents, translate);
+                break;
+            }
+            case kNoPersp_MatrixType: {
+                const SkMatrix& transform = get_transform_matrix(drawEffect, t);
+                gpu->enableTexGen((*texCoordIdx)++, GrGpuGL::kST_TexGenComponents, transform);
+                break;
+            }
+            case kGeneral_MatrixType: {
+                const SkMatrix& transform = get_transform_matrix(drawEffect, t);
+                gpu->enableTexGen((*texCoordIdx)++, GrGpuGL::kSTR_TexGenComponents, transform);
+                break;
+            }
+            default:
+                GrCrash("Unexpected matrixs type.");
+        }
+    }
+}
+
 void GrGLProgramEffects::bindTextures(GrGpuGL* gpu, const GrEffectRef& effect, int effectIdx) {
     const SkTArray<Sampler, true>& samplers = fSamplers[effectIdx];
     int numSamplers = samplers.count();
     SkASSERT(numSamplers == effect->numTextures());
     for (int s = 0; s < numSamplers; ++s) {
         SkASSERT(samplers[s].fTextureUnit >= 0);
         const GrTextureAccess& textureAccess = effect->textureAccess(s);
         gpu->bindTexture(samplers[s].fTextureUnit,
                          textureAccess.getParams(),
                          static_cast<GrGLTexture*>(textureAccess.getTexture()));
@@ skipping 56 matching lines @@
                                      (*stage.getEffect())->vertexAttribType(a),
                                      attributeName);
     }
 }
 
 void GrGLVertexProgramEffectsBuilder::emitTransforms(const GrEffectRef& effect,
                                                      EffectKey effectKey,
                                                      TransformedCoordsArray* outCoords) {
     typedef GrGLVertexProgramEffects::Transform Transform;
     SkTArray<Transform, true>& transforms = fProgramEffects->fTransforms.push_back();
-    EffectKey totalKey = GrBackendEffectFactory::GetTransformKey(effectKey);
+    EffectKey transformKey = GrBackendEffectFactory::GetTransformKey(effectKey);
     int numTransforms = effect->numTransforms();
     transforms.push_back_n(numTransforms);
     for (int t = 0; t < numTransforms; t++) {
-        EffectKey key = (totalKey >> (kTransformKeyBits * t)) & kTransformKeyMask;
         GrSLType varyingType = kVoid_GrSLType;
         const char* uniName;
-        switch (key & kMatrixTypeKeyMask) {
+        switch (get_matrix_type(transformKey, t)) {
             case kIdentity_MatrixType:
                 transforms[t].fType = kVoid_GrSLType;
                 uniName = NULL;
                 varyingType = kVec2f_GrSLType;
                 break;
             case kTrans_MatrixType:
                 transforms[t].fType = kVec2f_GrSLType;
                 uniName = "StageTranslate";
                 varyingType = kVec2f_GrSLType;
                 break;
@@ skipping 27 matching lines @@
         SkString suffixedVaryingName;
         if (0 != t) {
             suffixedVaryingName.append(varyingName);
             suffixedVaryingName.appendf("_%i", t);
             varyingName = suffixedVaryingName.c_str();
         }
         const char* vsVaryingName;
         const char* fsVaryingName;
         fBuilder->addVarying(varyingType, varyingName, &vsVaryingName, &fsVaryingName);
 
-        const GrGLShaderVar& coords = (kPositionCoords_Flag & key) ?
+        const GrGLShaderVar& coords = kPosition_GrCoordSet == get_source_coords(transformKey, t) ?
                                           fBuilder->positionAttribute() :
                                           fBuilder->localCoordsAttribute();
         // varying = matrix * coords (logically)
         switch (transforms[t].fType) {
             case kVoid_GrSLType:
                 SkASSERT(kVec2f_GrSLType == varyingType);
                 fBuilder->vsCodeAppendf("\t%s = %s;\n", vsVaryingName, coords.c_str());
                 break;
             case kVec2f_GrSLType:
                 SkASSERT(kVec2f_GrSLType == varyingType);
                 fBuilder->vsCodeAppendf("\t%s = %s + %s;\n",
                                         vsVaryingName, uniName, coords.c_str());
                 break;
             case kMat33f_GrSLType: {
                 SkASSERT(kVec2f_GrSLType == varyingType || kVec3f_GrSLType == varyingType);
                 if (kVec2f_GrSLType == varyingType) {
                     fBuilder->vsCodeAppendf("\t%s = (%s * vec3(%s, 1)).xy;\n",
                                             vsVaryingName, uniName, coords.c_str());
                 } else {
                     fBuilder->vsCodeAppendf("\t%s = %s * vec3(%s, 1);\n",
                                             vsVaryingName, uniName, coords.c_str());
                 }
                 break;
             }
             default:
                 GrCrash("Unexpected uniform type.");
         }
-        SkNEW_APPEND_TO_TARRAY(outCoords, TransformedCoords, (fsVaryingName, varyingType));
+        SkNEW_APPEND_TO_TARRAY(outCoords, TransformedCoords,
+                               (SkString(fsVaryingName), varyingType));
     }
 }
 
+GrGLTexGenProgramEffectsBuilder::GrGLTexGenProgramEffectsBuilder(
+        GrGLFragmentOnlyShaderBuilder* builder,
+        int reserveCount)
+    : fBuilder(builder)
+    , fProgramEffects(SkNEW_ARGS(GrGLTexGenProgramEffects, (reserveCount))) {
+}
+
+void GrGLTexGenProgramEffectsBuilder::emitEffect(const GrEffectStage& stage,
+                                                 EffectKey key,
+                                                 const char* outColor,
+                                                 const char* inColor,
+                                                 int stageIndex) {
+    SkASSERT(NULL != fProgramEffects.get());
+
+    GrDrawEffect drawEffect(stage, false);
+    const GrEffectRef& effect = *stage.getEffect();
+    SkSTArray<2, TransformedCoords> coords(effect->numTransforms());
+    SkSTArray<4, TextureSampler> samplers(effect->numTextures());
+
+    SkASSERT(0 == stage.getVertexAttribIndexCount());
+    this->setupTexGen(effect, key, &coords);
+    INHERITED::emitSamplers(fBuilder, fProgramEffects.get(), effect, &samplers);
+
+    GrGLEffect* glEffect = effect->getFactory().createGLInstance(drawEffect);
+    fProgramEffects->fGLEffects.push_back(glEffect);
+
+    // Enclose custom code in a block to avoid namespace conflicts
+    SkString openBrace;
+    openBrace.printf("\t{ // Stage %d: %s\n", stageIndex, glEffect->name());
+    fBuilder->fsCodeAppend(openBrace.c_str());
+
+    SkASSERT(!glEffect->isVertexEffect());
+    glEffect->emitCode(fBuilder, drawEffect, key, outColor, inColor, coords, samplers);
+
+    fBuilder->fsCodeAppend("\t}\n");
+}
+
+void GrGLTexGenProgramEffectsBuilder::setupTexGen(const GrEffectRef& effect,
+                                                  EffectKey effectKey,
+                                                  TransformedCoordsArray* outCoords) {
+    int numTransforms = effect->numTransforms();
+    SkDEBUGCODE(fProgramEffects->fNumTexCoordSets += numTransforms;)
+    int texCoordIdx = fBuilder->addTexCoordSets(numTransforms);
+    EffectKey transformKey = GrBackendEffectFactory::GetTransformKey(effectKey);
+    fProgramEffects->fTransformKeys.push_back(transformKey);
+    SkString name;
+    for (int t = 0; t < numTransforms; ++t) {
+        GrSLType type = kGeneral_MatrixType == get_matrix_type(transformKey, t) ?
+                            kVec3f_GrSLType :
+                            kVec2f_GrSLType;
+        name.printf("%s(gl_TexCoord[%i])", GrGLSLTypeString(type), texCoordIdx++);
+        SkNEW_APPEND_TO_TARRAY(outCoords, TransformedCoords, (name, type));
+    }
+}
+
 void GrGLProgramEffectsBuilder::emitSamplers(GrGLShaderBuilder* builder,
                                              GrGLProgramEffects* programEffects,
                                              const GrEffectRef& effect,
                                              TextureSamplerArray* outSamplers) {
     typedef GrGLProgramEffects::Sampler Sampler;
     SkTArray<Sampler, true>& samplers = programEffects->fSamplers.push_back();
     int numTextures = effect->numTextures();
     samplers.push_back_n(numTextures);
     SkString name;
     for (int t = 0; t < numTextures; ++t) {
         name.printf("Sampler%d", t);
         samplers[t].fUniform = builder->addUniform(GrGLShaderBuilder::kFragment_Visibility,
                                                    kSampler2D_GrSLType,
                                                    name.c_str());
         SkNEW_APPEND_TO_TARRAY(outSamplers, TextureSampler,
                                (samplers[t].fUniform, effect->textureAccess(t)));
     }
 }