egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 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 | #ifndef GrGLSLProgramBuilder_DEFINED |
| 9 | #define GrGLSLProgramBuilder_DEFINED |
| 10 | |
| 11 | #include "GrGeometryProcessor.h" |
| 12 | #include "GrGpu.h" |
egdaniel | 2d721d3 | 2015-11-11 13:06:05 -0800 | [diff] [blame] | 13 | #include "glsl/GrGLSLFragmentShaderBuilder.h" |
| 14 | #include "glsl/GrGLSLGeometryShaderBuilder.h" |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 15 | #include "glsl/GrGLSLPrimitiveProcessor.h" |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 16 | #include "glsl/GrGLSLProgramDataManager.h" |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 17 | #include "glsl/GrGLSLUniformHandler.h" |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 18 | #include "glsl/GrGLSLTextureSampler.h" |
egdaniel | 2d721d3 | 2015-11-11 13:06:05 -0800 | [diff] [blame] | 19 | #include "glsl/GrGLSLVertexShaderBuilder.h" |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 20 | #include "glsl/GrGLSLXferProcessor.h" |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 21 | |
| 22 | class GrGLSLCaps; |
| 23 | class GrGLSLShaderVar; |
egdaniel | 0eafe79 | 2015-11-20 14:01:22 -0800 | [diff] [blame] | 24 | class GrGLSLVaryingHandler; |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 25 | |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 26 | typedef SkSTArray<8, GrGLSLFragmentProcessor*, true> GrGLSLFragProcs; |
| 27 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 28 | class GrGLSLProgramBuilder { |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 29 | public: |
| 30 | typedef GrGpu::DrawArgs DrawArgs; |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 31 | typedef GrGLSLUniformHandler::UniformHandle UniformHandle; |
| 32 | |
| 33 | virtual ~GrGLSLProgramBuilder() {} |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 34 | |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 35 | virtual const GrCaps* caps() const = 0; |
egdaniel | a2e3e0f | 2015-11-19 07:23:45 -0800 | [diff] [blame] | 36 | virtual const GrGLSLCaps* glslCaps() const = 0; |
| 37 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 38 | const GrPrimitiveProcessor& primitiveProcessor() const { return *fArgs.fPrimitiveProcessor; } |
| 39 | const GrPipeline& pipeline() const { return *fArgs.fPipeline; } |
| 40 | const GrProgramDesc& desc() const { return *fArgs.fDesc; } |
| 41 | const GrProgramDesc::KeyHeader& header() const { return fArgs.fDesc->header(); } |
| 42 | |
cdalton | 5e58cee | 2016-02-11 12:49:47 -0800 | [diff] [blame^] | 43 | void appendUniformDecls(GrShaderFlags visibility, SkString*) const; |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 44 | |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 45 | // Handles for program uniforms (other than per-effect uniforms) |
| 46 | struct BuiltinUniformHandles { |
| 47 | UniformHandle fRTAdjustmentUni; |
| 48 | |
| 49 | // We use the render target height to provide a y-down frag coord when specifying |
| 50 | // origin_upper_left is not supported. |
| 51 | UniformHandle fRTHeightUni; |
| 52 | }; |
| 53 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 54 | // Used to add a uniform in the vertex shader for transforming into normalized device space. |
| 55 | void addRTAdjustmentUniform(GrSLPrecision precision, const char* name, const char** outName); |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 56 | const char* rtAdjustment() const { return "rtAdjustment"; } |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 57 | |
| 58 | // Used to add a uniform for the RenderTarget height (used for frag position) without mangling |
| 59 | // the name of the uniform inside of a stage. |
| 60 | void addRTHeightUniform(const char* name, const char** outName); |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 61 | |
| 62 | // Generates a name for a variable. The generated string will be name prefixed by the prefix |
| 63 | // char (unless the prefix is '\0'). It also will mangle the name to be stage-specific unless |
| 64 | // explicitly asked not to. |
| 65 | void nameVariable(SkString* out, char prefix, const char* name, bool mangle = true); |
| 66 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 67 | virtual GrGLSLUniformHandler* uniformHandler() = 0; |
| 68 | virtual const GrGLSLUniformHandler* uniformHandler() const = 0; |
egdaniel | 0eafe79 | 2015-11-20 14:01:22 -0800 | [diff] [blame] | 69 | virtual GrGLSLVaryingHandler* varyingHandler() = 0; |
| 70 | |
egdaniel | b80ec8b | 2016-02-09 09:54:43 -0800 | [diff] [blame] | 71 | // Used for backend customization of the output color and secondary color variables from the |
| 72 | // fragment processor. Only used if the outputs are explicitly declared in the shaders |
| 73 | virtual void finalizeFragmentOutputColor(GrGLSLShaderVar& outputColor) {} |
| 74 | virtual void finalizeFragmentSecondaryColor(GrGLSLShaderVar& outputColor) {} |
| 75 | |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 76 | // number of each input/output type in a single allocation block, used by many builders |
| 77 | static const int kVarsPerBlock; |
| 78 | |
egdaniel | 0eafe79 | 2015-11-20 14:01:22 -0800 | [diff] [blame] | 79 | GrGLSLVertexBuilder fVS; |
| 80 | GrGLSLGeometryBuilder fGS; |
egdaniel | 2d721d3 | 2015-11-11 13:06:05 -0800 | [diff] [blame] | 81 | GrGLSLFragmentShaderBuilder fFS; |
egdaniel | 0eafe79 | 2015-11-20 14:01:22 -0800 | [diff] [blame] | 82 | |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 83 | int fStageIndex; |
| 84 | |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 85 | const DrawArgs& fArgs; |
| 86 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 87 | BuiltinUniformHandles fUniformHandles; |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 88 | |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 89 | GrGLSLPrimitiveProcessor* fGeometryProcessor; |
| 90 | GrGLSLXferProcessor* fXferProcessor; |
| 91 | GrGLSLFragProcs fFragmentProcessors; |
| 92 | |
egdaniel | 7ea439b | 2015-12-03 09:20:44 -0800 | [diff] [blame] | 93 | protected: |
| 94 | explicit GrGLSLProgramBuilder(const DrawArgs& args); |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 95 | |
| 96 | bool emitAndInstallProcs(GrGLSLExpr4* inputColor, GrGLSLExpr4* inputCoverage, int maxTextures); |
| 97 | |
| 98 | void cleanupFragmentProcessors(); |
| 99 | |
egdaniel | 9f1d415 | 2016-02-10 09:50:38 -0800 | [diff] [blame] | 100 | void finalizeShaders(); |
| 101 | |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 102 | private: |
| 103 | // reset is called by program creator between each processor's emit code. It increments the |
| 104 | // stage offset for variable name mangling, and also ensures verfication variables in the |
| 105 | // fragment shader are cleared. |
| 106 | void reset() { |
| 107 | this->addStage(); |
| 108 | fFS.reset(); |
| 109 | } |
| 110 | void addStage() { fStageIndex++; } |
| 111 | |
| 112 | class AutoStageAdvance { |
| 113 | public: |
| 114 | AutoStageAdvance(GrGLSLProgramBuilder* pb) |
| 115 | : fPB(pb) { |
| 116 | fPB->reset(); |
| 117 | // Each output to the fragment processor gets its own code section |
| 118 | fPB->fFS.nextStage(); |
| 119 | } |
| 120 | ~AutoStageAdvance() {} |
| 121 | private: |
| 122 | GrGLSLProgramBuilder* fPB; |
| 123 | }; |
| 124 | |
| 125 | // Generates a possibly mangled name for a stage variable and writes it to the fragment shader. |
| 126 | // If GrGLSLExpr4 has a valid name then it will use that instead |
| 127 | void nameExpression(GrGLSLExpr4*, const char* baseName); |
| 128 | |
| 129 | void emitAndInstallPrimProc(const GrPrimitiveProcessor&, |
| 130 | GrGLSLExpr4* outputColor, |
| 131 | GrGLSLExpr4* outputCoverage); |
| 132 | void emitAndInstallFragProcs(int procOffset, int numProcs, GrGLSLExpr4* inOut); |
| 133 | void emitAndInstallFragProc(const GrFragmentProcessor&, |
| 134 | int index, |
| 135 | const GrGLSLExpr4& input, |
| 136 | GrGLSLExpr4* output); |
| 137 | void emitAndInstallXferProc(const GrXferProcessor&, |
| 138 | const GrGLSLExpr4& colorIn, |
| 139 | const GrGLSLExpr4& coverageIn, |
ethannicholas | 2279325 | 2016-01-30 09:59:10 -0800 | [diff] [blame] | 140 | bool ignoresCoverage, |
| 141 | GrPixelLocalStorageState plsState); |
egdaniel | fa89632 | 2016-01-13 12:19:30 -0800 | [diff] [blame] | 142 | void emitFSOutputSwizzle(bool hasSecondaryOutput); |
| 143 | |
| 144 | void verify(const GrPrimitiveProcessor&); |
| 145 | void verify(const GrXferProcessor&); |
| 146 | void verify(const GrFragmentProcessor&); |
| 147 | |
| 148 | virtual void emitSamplers(const GrProcessor& processor, |
| 149 | GrGLSLTextureSampler::TextureSamplerArray* outSamplers) = 0; |
| 150 | |
| 151 | GrGLSLPrimitiveProcessor::TransformsIn fCoordTransforms; |
| 152 | GrGLSLPrimitiveProcessor::TransformsOut fOutCoords; |
egdaniel | 8dcdedc | 2015-11-11 06:27:20 -0800 | [diff] [blame] | 153 | }; |
| 154 | |
| 155 | #endif |