| /* | |
| * 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 "GrDefaultGeoProcFactory.h" | |
| #include "GrInvariantOutput.h" | |
| #include "SkRefCnt.h" | |
| #include "glsl/GrGLSLFragmentShaderBuilder.h" | |
| #include "glsl/GrGLSLGeometryProcessor.h" | |
| #include "glsl/GrGLSLVertexShaderBuilder.h" | |
| #include "glsl/GrGLSLVarying.h" | |
| #include "glsl/GrGLSLUniformHandler.h" | |
| #include "glsl/GrGLSLUtil.h" | |
| /* | |
| * The default Geometry Processor simply takes position and multiplies it by the uniform view | |
| * matrix. It also leaves coverage untouched. Behind the scenes, we may add per vertex color or | |
| * local coords. | |
| */ | |
| enum GPFlag { | |
| kColor_GPFlag = 0x1, | |
| kLocalCoord_GPFlag = 0x2, | |
| kCoverage_GPFlag= 0x4, | |
| kTransformedLocalCoord_GPFlag = 0x8, | |
| }; | |
| class DefaultGeoProc : public GrGeometryProcessor { | |
| public: | |
| static sk_sp<GrGeometryProcessor> Make(uint32_t gpTypeFlags, | |
| GrColor color, | |
| const SkMatrix& viewMatrix, | |
| const SkMatrix& localMatrix, | |
| bool localCoordsWillBeRead, | |
| bool coverageWillBeIgnored, | |
| uint8_t coverage) { | |
| return sk_sp<GrGeometryProcessor>(new DefaultGeoProc( | |
| gpTypeFlags, color, viewMatrix, localMatrix, coverage, | |
| localCoordsWillBeRead, coverageWillBeIgnored)); | |
| } | |
| const char* name() const override { return "DefaultGeometryProcessor"; } | |
| const Attribute* inPosition() const { return fInPosition; } | |
| const Attribute* inColor() const { return fInColor; } | |
| const Attribute* inLocalCoords() const { return fInLocalCoords; } | |
| const Attribute* inCoverage() const { return fInCoverage; } | |
| GrColor color() const { return fColor; } | |
| bool colorIgnored() const { return GrColor_ILLEGAL == fColor; } | |
| bool hasVertexColor() const { return SkToBool(fInColor); } | |
| const SkMatrix& viewMatrix() const { return fViewMatrix; } | |
| const SkMatrix& localMatrix() const { return fLocalMatrix; } | |
| bool localCoordsWillBeRead() const { return fLocalCoordsWillBeRead; } | |
| uint8_t coverage() const { return fCoverage; } | |
| bool coverageWillBeIgnored() const { return fCoverageWillBeIgnored; } | |
| bool hasVertexCoverage() const { return SkToBool(fInCoverage); } | |
| class GLSLProcessor : public GrGLSLGeometryProcessor { | |
| public: | |
| GLSLProcessor() | |
| : fViewMatrix(SkMatrix::InvalidMatrix()), fColor(GrColor_ILLEGAL), fCoverage(0xff) {} | |
| void onEmitCode(EmitArgs& args, GrGPArgs* gpArgs) override { | |
| const DefaultGeoProc& gp = args.fGP.cast<DefaultGeoProc>(); | |
| GrGLSLVertexBuilder* vertBuilder = args.fVertBuilder; | |
| GrGLSLPPFragmentBuilder* fragBuilder = args.fFragBuilder; | |
| GrGLSLVaryingHandler* varyingHandler = args.fVaryingHandler; | |
| GrGLSLUniformHandler* uniformHandler = args.fUniformHandler; | |
| // emit attributes | |
| varyingHandler->emitAttributes(gp); | |
| // Setup pass through color | |
| if (!gp.colorIgnored()) { | |
| if (gp.hasVertexColor()) { | |
| varyingHandler->addPassThroughAttribute(gp.inColor(), args.fOutputColor); | |
| } else { | |
| this->setupUniformColor(fragBuilder, uniformHandler, args.fOutputColor, | |
| &fColorUniform); | |
| } | |
| } | |
| // Setup position | |
| this->setupPosition(vertBuilder, | |
| uniformHandler, | |
| gpArgs, | |
| gp.inPosition()->fName, | |
| gp.viewMatrix(), | |
| &fViewMatrixUniform); | |
| if (gp.hasExplicitLocalCoords()) { | |
| // emit transforms with explicit local coords | |
| this->emitTransforms(vertBuilder, | |
| varyingHandler, | |
| uniformHandler, | |
| gpArgs->fPositionVar, | |
| gp.inLocalCoords()->fName, | |
| gp.localMatrix(), | |
| args.fTransformsIn, | |
| args.fTransformsOut); | |
| } else if(gp.hasTransformedLocalCoords()) { | |
| // transforms have already been applied to vertex attributes on the cpu | |
| this->emitTransforms(vertBuilder, | |
| varyingHandler, | |
| gp.inLocalCoords()->fName, | |
| args.fTransformsIn, | |
| args.fTransformsOut); | |
| } else { | |
| // emit transforms with position | |
| this->emitTransforms(vertBuilder, | |
| varyingHandler, | |
| uniformHandler, | |
| gpArgs->fPositionVar, | |
| gp.inPosition()->fName, | |
| gp.localMatrix(), | |
| args.fTransformsIn, | |
| args.fTransformsOut); | |
| } | |
| // Setup coverage as pass through | |
| if (!gp.coverageWillBeIgnored()) { | |
| if (gp.hasVertexCoverage()) { | |
| fragBuilder->codeAppendf("float alpha = 1.0;"); | |
| varyingHandler->addPassThroughAttribute(gp.inCoverage(), "alpha"); | |
| fragBuilder->codeAppendf("%s = vec4(alpha);", args.fOutputCoverage); | |
| } else if (gp.coverage() == 0xff) { | |
| fragBuilder->codeAppendf("%s = vec4(1);", args.fOutputCoverage); | |
| } else { | |
| const char* fragCoverage; | |
| fCoverageUniform = uniformHandler->addUniform(kFragment_GrShaderFlag, | |
| kFloat_GrSLType, | |
| kDefault_GrSLPrecision, | |
| "Coverage", | |
| &fragCoverage); | |
| fragBuilder->codeAppendf("%s = vec4(%s);", args.fOutputCoverage, fragCoverage); | |
| } | |
| } | |
| } | |
| static inline void GenKey(const GrGeometryProcessor& gp, | |
| const GrGLSLCaps&, | |
| GrProcessorKeyBuilder* b) { | |
| const DefaultGeoProc& def = gp.cast<DefaultGeoProc>(); | |
| uint32_t key = def.fFlags; | |
| key |= def.colorIgnored() << 8; | |
| key |= def.coverageWillBeIgnored() << 9; | |
| key |= def.hasVertexColor() << 10; | |
| key |= def.hasVertexCoverage() << 11; | |
| key |= def.coverage() == 0xff ? 0x1 << 12 : 0; | |
| key |= def.localCoordsWillBeRead() && def.localMatrix().hasPerspective() ? 0x1 << 24 : | |
| 0x0; | |
| key |= ComputePosKey(def.viewMatrix()) << 25; | |
| b->add32(key); | |
| } | |
| void setData(const GrGLSLProgramDataManager& pdman, | |
| const GrPrimitiveProcessor& gp) override { | |
| const DefaultGeoProc& dgp = gp.cast<DefaultGeoProc>(); | |
| if (!dgp.viewMatrix().isIdentity() && !fViewMatrix.cheapEqualTo(dgp.viewMatrix())) { | |
| fViewMatrix = dgp.viewMatrix(); | |
| float viewMatrix[3 * 3]; | |
| GrGLSLGetMatrix<3>(viewMatrix, fViewMatrix); | |
| pdman.setMatrix3f(fViewMatrixUniform, viewMatrix); | |
| } | |
| if (dgp.color() != fColor && !dgp.hasVertexColor()) { | |
| float c[4]; | |
| GrColorToRGBAFloat(dgp.color(), c); | |
| pdman.set4fv(fColorUniform, 1, c); | |
| fColor = dgp.color(); | |
| } | |
| if (!dgp.coverageWillBeIgnored() && | |
| dgp.coverage() != fCoverage && !dgp.hasVertexCoverage()) { | |
| pdman.set1f(fCoverageUniform, GrNormalizeByteToFloat(dgp.coverage())); | |
| fCoverage = dgp.coverage(); | |
| } | |
| } | |
| void setTransformData(const GrPrimitiveProcessor& primProc, | |
| const GrGLSLProgramDataManager& pdman, | |
| int index, | |
| const SkTArray<const GrCoordTransform*, true>& transforms) override { | |
| this->setTransformDataHelper(primProc.cast<DefaultGeoProc>().fLocalMatrix, pdman, index, | |
| transforms); | |
| } | |
| private: | |
| SkMatrix fViewMatrix; | |
| GrColor fColor; | |
| uint8_t fCoverage; | |
| UniformHandle fViewMatrixUniform; | |
| UniformHandle fColorUniform; | |
| UniformHandle fCoverageUniform; | |
| typedef GrGLSLGeometryProcessor INHERITED; | |
| }; | |
| void getGLSLProcessorKey(const GrGLSLCaps& caps, GrProcessorKeyBuilder* b) const override { | |
| GLSLProcessor::GenKey(*this, caps, b); | |
| } | |
| GrGLSLPrimitiveProcessor* createGLSLInstance(const GrGLSLCaps&) const override { | |
| return new GLSLProcessor(); | |
| } | |
| private: | |
| DefaultGeoProc(uint32_t gpTypeFlags, | |
| GrColor color, | |
| const SkMatrix& viewMatrix, | |
| const SkMatrix& localMatrix, | |
| uint8_t coverage, | |
| bool localCoordsWillBeRead, | |
| bool coverageWillBeIgnored) | |
| : fInPosition(nullptr) | |
| , fInColor(nullptr) | |
| , fInLocalCoords(nullptr) | |
| , fInCoverage(nullptr) | |
| , fColor(color) | |
| , fViewMatrix(viewMatrix) | |
| , fLocalMatrix(localMatrix) | |
| , fCoverage(coverage) | |
| , fFlags(gpTypeFlags) | |
| , fLocalCoordsWillBeRead(localCoordsWillBeRead) | |
| , fCoverageWillBeIgnored(coverageWillBeIgnored) { | |
| this->initClassID<DefaultGeoProc>(); | |
| bool hasColor = SkToBool(gpTypeFlags & kColor_GPFlag); | |
| bool hasExplicitLocalCoords = SkToBool(gpTypeFlags & kLocalCoord_GPFlag); | |
| bool hasTransformedLocalCoords = SkToBool(gpTypeFlags & kTransformedLocalCoord_GPFlag); | |
| bool hasLocalCoord = hasExplicitLocalCoords || hasTransformedLocalCoords; | |
| bool hasCoverage = SkToBool(gpTypeFlags & kCoverage_GPFlag); | |
| fInPosition = &this->addVertexAttrib("inPosition", kVec2f_GrVertexAttribType, | |
| kHigh_GrSLPrecision); | |
| if (hasColor) { | |
| fInColor = &this->addVertexAttrib("inColor", kVec4ub_GrVertexAttribType); | |
| } | |
| if (hasLocalCoord) { | |
| fInLocalCoords = &this->addVertexAttrib("inLocalCoord", kVec2f_GrVertexAttribType); | |
| if (hasExplicitLocalCoords) { | |
| this->setHasExplicitLocalCoords(); | |
| } else { | |
| SkASSERT(hasTransformedLocalCoords); | |
| this->setHasTransformedLocalCoords(); | |
| } | |
| } | |
| if (hasCoverage) { | |
| fInCoverage = &this->addVertexAttrib("inCoverage", kFloat_GrVertexAttribType); | |
| } | |
| } | |
| const Attribute* fInPosition; | |
| const Attribute* fInColor; | |
| const Attribute* fInLocalCoords; | |
| const Attribute* fInCoverage; | |
| GrColor fColor; | |
| SkMatrix fViewMatrix; | |
| SkMatrix fLocalMatrix; | |
| uint8_t fCoverage; | |
| uint32_t fFlags; | |
| bool fLocalCoordsWillBeRead; | |
| bool fCoverageWillBeIgnored; | |
| GR_DECLARE_GEOMETRY_PROCESSOR_TEST; | |
| typedef GrGeometryProcessor INHERITED; | |
| }; | |
| GR_DEFINE_GEOMETRY_PROCESSOR_TEST(DefaultGeoProc); | |
| sk_sp<GrGeometryProcessor> DefaultGeoProc::TestCreate(GrProcessorTestData* d) { | |
| uint32_t flags = 0; | |
| if (d->fRandom->nextBool()) { | |
| flags |= kColor_GPFlag; | |
| } | |
| if (d->fRandom->nextBool()) { | |
| flags |= kCoverage_GPFlag; | |
| } | |
| if (d->fRandom->nextBool()) { | |
| flags |= kLocalCoord_GPFlag; | |
| } | |
| if (d->fRandom->nextBool()) { | |
| flags |= kTransformedLocalCoord_GPFlag; | |
| } | |
| return DefaultGeoProc::Make(flags, | |
| GrRandomColor(d->fRandom), | |
| GrTest::TestMatrix(d->fRandom), | |
| GrTest::TestMatrix(d->fRandom), | |
| d->fRandom->nextBool(), | |
| d->fRandom->nextBool(), | |
| GrRandomCoverage(d->fRandom)); | |
| } | |
| sk_sp<GrGeometryProcessor> GrDefaultGeoProcFactory::Make(const Color& color, | |
| const Coverage& coverage, | |
| const LocalCoords& localCoords, | |
| const SkMatrix& viewMatrix) { | |
| uint32_t flags = 0; | |
| flags |= color.fType == Color::kAttribute_Type ? kColor_GPFlag : 0; | |
| flags |= coverage.fType == Coverage::kAttribute_Type ? kCoverage_GPFlag : 0; | |
| flags |= localCoords.fType == LocalCoords::kHasExplicit_Type ? kLocalCoord_GPFlag : 0; | |
| flags |= localCoords.fType == LocalCoords::kHasTransformed_Type ? | |
| kTransformedLocalCoord_GPFlag : 0; | |
| uint8_t inCoverage = coverage.fCoverage; | |
| bool coverageWillBeIgnored = coverage.fType == Coverage::kNone_Type; | |
| bool localCoordsWillBeRead = localCoords.fType != LocalCoords::kUnused_Type; | |
| GrColor inColor = color.fColor; | |
| return DefaultGeoProc::Make(flags, | |
| inColor, | |
| viewMatrix, | |
| localCoords.fMatrix ? *localCoords.fMatrix : SkMatrix::I(), | |
| localCoordsWillBeRead, | |
| coverageWillBeIgnored, | |
| inCoverage); | |
| } | |
| sk_sp<GrGeometryProcessor> GrDefaultGeoProcFactory::MakeForDeviceSpace( | |
| const Color& color, | |
| const Coverage& coverage, | |
| const LocalCoords& localCoords, | |
| const SkMatrix& viewMatrix) { | |
| SkMatrix invert = SkMatrix::I(); | |
| if (LocalCoords::kUnused_Type != localCoords.fType) { | |
| SkASSERT(LocalCoords::kUsePosition_Type == localCoords.fType); | |
| if (!viewMatrix.isIdentity() && !viewMatrix.invert(&invert)) { | |
| SkDebugf("Could not invert\n"); | |
| return nullptr; | |
| } | |
| if (localCoords.hasLocalMatrix()) { | |
| invert.preConcat(*localCoords.fMatrix); | |
| } | |
| } | |
| LocalCoords inverted(LocalCoords::kUsePosition_Type, &invert); | |
| return Make(color, coverage, inverted, SkMatrix::I()); | |
| } |