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