blob: 329fb2100b674b07d428b38839d9e74df077373d [file] [log] [blame]
joshualitt30ba4362014-08-21 20:18:45 -07001/*
2 * Copyright 2014 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
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "src/gpu/gl/builders/GrGLProgramBuilder.h"
joshualitt8072caa2015-02-12 14:20:52 -08009
Adlai Holler3d0359a2020-07-09 15:35:55 -040010#include "include/gpu/GrDirectContext.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050011#include "src/core/SkATrace.h"
12#include "src/core/SkAutoMalloc.h"
Brian Osman9e4e4c72020-06-10 07:19:34 -040013#include "src/core/SkReadBuffer.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050014#include "src/core/SkTraceEvent.h"
Brian Osman9e4e4c72020-06-10 07:19:34 -040015#include "src/core/SkWriteBuffer.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050016#include "src/gpu/GrAutoLocaleSetter.h"
Adlai Hollera0693042020-10-14 11:23:11 -040017#include "src/gpu/GrDirectContextPriv.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050018#include "src/gpu/GrPersistentCacheUtils.h"
19#include "src/gpu/GrProgramDesc.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050020#include "src/gpu/GrShaderCaps.h"
Brian Osmanac9be9d2019-05-01 10:29:34 -040021#include "src/gpu/GrShaderUtils.h"
Brian Salomon3ec1f542019-06-17 17:54:57 +000022#include "src/gpu/GrSwizzle.h"
Mike Kleinc0bd9f92019-04-23 12:05:21 -050023#include "src/gpu/gl/GrGLGpu.h"
24#include "src/gpu/gl/GrGLProgram.h"
25#include "src/gpu/gl/builders/GrGLProgramBuilder.h"
John Stilesfbd050b2020-08-03 13:21:46 -040026
27#include <memory>
Mike Kleinc0bd9f92019-04-23 12:05:21 -050028#include "src/gpu/gl/builders/GrGLShaderStringBuilder.h"
29#include "src/gpu/glsl/GrGLSLFragmentProcessor.h"
30#include "src/gpu/glsl/GrGLSLGeometryProcessor.h"
31#include "src/gpu/glsl/GrGLSLProgramDataManager.h"
32#include "src/gpu/glsl/GrGLSLXferProcessor.h"
joshualitt30ba4362014-08-21 20:18:45 -070033
joshualitt30ba4362014-08-21 20:18:45 -070034#define GL_CALL(X) GR_GL_CALL(this->gpu()->glInterface(), X)
35#define GL_CALL_RET(R, X) GR_GL_CALL_RET(this->gpu()->glInterface(), R, X)
36
Brian Osmaned58e002019-09-06 14:42:43 -040037static void cleanup_shaders(GrGLGpu* gpu, const SkTDArray<GrGLuint>& shaderIDs) {
38 for (int i = 0; i < shaderIDs.count(); ++i) {
39 GR_GL_CALL(gpu->glInterface(), DeleteShader(shaderIDs[i]));
40 }
41}
42
43static void cleanup_program(GrGLGpu* gpu, GrGLuint programID,
44 const SkTDArray<GrGLuint>& shaderIDs) {
45 GR_GL_CALL(gpu->glInterface(), DeleteProgram(programID));
46 cleanup_shaders(gpu, shaderIDs);
47}
48
Stephen Whiteb1857852020-02-07 15:33:23 +000049sk_sp<GrGLProgram> GrGLProgramBuilder::CreateProgram(
Austin Eng77fdf662020-02-07 01:26:16 +000050 GrGLGpu* gpu,
Stephen Whiteb1857852020-02-07 15:33:23 +000051 GrRenderTarget* renderTarget,
52 const GrProgramDesc& desc,
53 const GrProgramInfo& programInfo,
Brian Osmaned58e002019-09-06 14:42:43 -040054 const GrGLPrecompiledProgram* precompiledProgram) {
Stan Iliev43324e72020-01-13 16:17:48 -050055 ATRACE_ANDROID_FRAMEWORK_ALWAYS("shader_compile");
bsalomon3318ee72015-03-16 11:56:29 -070056 GrAutoLocaleSetter als("C");
57
joshualitt47bb3822014-10-07 16:43:25 -070058 // create a builder. This will be handed off to effects so they can use it to add
59 // uniforms, varyings, textures, etc
Stephen Whiteb1857852020-02-07 15:33:23 +000060 GrGLProgramBuilder builder(gpu, renderTarget, desc, programInfo);
joshualitt47bb3822014-10-07 16:43:25 -070061
Robert Phillips9da87e02019-02-04 13:26:26 -050062 auto persistentCache = gpu->getContext()->priv().getPersistentCache();
Brian Osmaned58e002019-09-06 14:42:43 -040063 if (persistentCache && !precompiledProgram) {
Stephen Whiteb1857852020-02-07 15:33:23 +000064 sk_sp<SkData> key = SkData::MakeWithoutCopy(desc.asKey(), desc.keyLength());
Robert Phillips0c4b7b12018-03-06 08:20:37 -050065 builder.fCached = persistentCache->load(*key);
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -040066 // the eventual end goal is to completely skip emitAndInstallProcs on a cache hit, but it's
67 // doing necessary setup in addition to generating the SkSL code. Currently we are only able
68 // to skip the SkSL->GLSL step on a cache hit.
69 }
Ethan Nicholas2983f402017-05-08 09:36:08 -040070 if (!builder.emitAndInstallProcs()) {
halcanary96fcdcc2015-08-27 07:41:13 -070071 return nullptr;
joshualitt6c891102015-05-13 08:51:49 -070072 }
Brian Osmaned58e002019-09-06 14:42:43 -040073 return builder.finalize(precompiledProgram);
joshualitt47bb3822014-10-07 16:43:25 -070074}
75
joshualitt47bb3822014-10-07 16:43:25 -070076/////////////////////////////////////////////////////////////////////////////
77
egdaniel0e1853c2016-03-17 11:35:45 -070078GrGLProgramBuilder::GrGLProgramBuilder(GrGLGpu* gpu,
Robert Phillipsd0fe8752019-01-31 14:13:59 -050079 GrRenderTarget* renderTarget,
Stephen Whiteb1857852020-02-07 15:33:23 +000080 const GrProgramDesc& desc,
81 const GrProgramInfo& programInfo)
82 : INHERITED(renderTarget, desc, programInfo)
Brian Salomon802cb312018-06-08 18:05:20 -040083 , fGpu(gpu)
84 , fVaryingHandler(this)
85 , fUniformHandler(this)
Brian Salomon92be2f72018-06-19 14:33:47 -040086 , fVertexAttributeCnt(0)
87 , fInstanceAttributeCnt(0)
Brian Salomon802cb312018-06-08 18:05:20 -040088 , fVertexStride(0)
89 , fInstanceStride(0) {}
joshualitt30ba4362014-08-21 20:18:45 -070090
egdanielfa896322016-01-13 12:19:30 -080091const GrCaps* GrGLProgramBuilder::caps() const {
92 return fGpu->caps();
93}
94
Brian Osmanac9be9d2019-05-01 10:29:34 -040095bool GrGLProgramBuilder::compileAndAttachShaders(const SkSL::String& glsl,
egdaniel574a4c12015-11-02 06:22:44 -080096 GrGLuint programId,
97 GrGLenum type,
Ethan Nicholas941e7e22016-12-12 15:33:30 -050098 SkTDArray<GrGLuint>* shaderIds,
Brian Osman5e7fbfd2019-05-03 13:13:35 -040099 GrContextOptions::ShaderErrorHandler* errHandler) {
egdaniel574a4c12015-11-02 06:22:44 -0800100 GrGLGpu* gpu = this->gpu();
101 GrGLuint shaderId = GrGLCompileAndAttachShader(gpu->glContext(),
102 programId,
103 type,
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400104 glsl,
Brian Osman8518f2e2019-05-01 14:13:41 -0400105 gpu->stats(),
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400106 errHandler);
egdaniel574a4c12015-11-02 06:22:44 -0800107 if (!shaderId) {
108 return false;
109 }
110
111 *shaderIds->append() = shaderId;
egdaniel574a4c12015-11-02 06:22:44 -0800112 return true;
113}
114
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400115void GrGLProgramBuilder::computeCountsAndStrides(GrGLuint programID,
116 const GrPrimitiveProcessor& primProc,
117 bool bindAttribLocations) {
118 fVertexAttributeCnt = primProc.numVertexAttributes();
119 fInstanceAttributeCnt = primProc.numInstanceAttributes();
John Stilesfbd050b2020-08-03 13:21:46 -0400120 fAttributes = std::make_unique<GrGLProgram::Attribute[]>(
121 fVertexAttributeCnt + fInstanceAttributeCnt);
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400122 auto addAttr = [&](int i, const auto& a, size_t* stride) {
Brian Osman4a3f5c82018-09-18 16:16:38 -0400123 fAttributes[i].fCPUType = a.cpuType();
124 fAttributes[i].fGPUType = a.gpuType();
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400125 fAttributes[i].fOffset = *stride;
126 *stride += a.sizeAlign4();
127 fAttributes[i].fLocation = i;
128 if (bindAttribLocations) {
129 GL_CALL(BindAttribLocation(programID, i, a.name()));
130 }
131 };
132 fVertexStride = 0;
133 int i = 0;
Brian Osmanf04fb3c2018-11-12 15:34:00 -0500134 for (const auto& attr : primProc.vertexAttributes()) {
135 addAttr(i++, attr, &fVertexStride);
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400136 }
Brian Osmanf04fb3c2018-11-12 15:34:00 -0500137 SkASSERT(fVertexStride == primProc.vertexStride());
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400138 fInstanceStride = 0;
Brian Osmanf04fb3c2018-11-12 15:34:00 -0500139 for (const auto& attr : primProc.instanceAttributes()) {
140 addAttr(i++, attr, &fInstanceStride);
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400141 }
Brian Osmanf04fb3c2018-11-12 15:34:00 -0500142 SkASSERT(fInstanceStride == primProc.instanceStride());
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400143}
144
Ethan Nicholascd700e92018-08-24 16:43:57 -0400145void GrGLProgramBuilder::addInputVars(const SkSL::Program::Inputs& inputs) {
146 if (inputs.fRTWidth) {
147 this->addRTWidthUniform(SKSL_RTWIDTH_NAME);
148 }
149 if (inputs.fRTHeight) {
150 this->addRTHeightUniform(SKSL_RTHEIGHT_NAME);
151 }
152}
153
Brian Osmana085a412019-04-25 09:44:43 -0400154static constexpr SkFourByteTag kSKSL_Tag = SkSetFourByteTag('S', 'K', 'S', 'L');
155static constexpr SkFourByteTag kGLSL_Tag = SkSetFourByteTag('G', 'L', 'S', 'L');
Brian Osmana66081d2019-09-03 14:59:26 -0400156static constexpr SkFourByteTag kGLPB_Tag = SkSetFourByteTag('G', 'L', 'P', 'B');
Brian Osmana085a412019-04-25 09:44:43 -0400157
Ethan Nicholas8d058832019-01-07 10:49:58 -0500158void GrGLProgramBuilder::storeShaderInCache(const SkSL::Program::Inputs& inputs, GrGLuint programID,
Brian Osmaned58e002019-09-06 14:42:43 -0400159 const SkSL::String shaders[], bool isSkSL,
Brian Osman4524e842019-09-24 16:03:41 -0400160 SkSL::Program::Settings* settings) {
Robert Phillips9da87e02019-02-04 13:26:26 -0500161 if (!this->gpu()->getContext()->priv().getPersistentCache()) {
Ethan Nicholas8d058832019-01-07 10:49:58 -0500162 return;
163 }
Stephen Whiteb1857852020-02-07 15:33:23 +0000164 sk_sp<SkData> key = SkData::MakeWithoutCopy(this->desc().asKey(), this->desc().keyLength());
Ethan Nicholas8d058832019-01-07 10:49:58 -0500165 if (fGpu->glCaps().programBinarySupport()) {
166 // binary cache
167 GrGLsizei length = 0;
168 GL_CALL(GetProgramiv(programID, GL_PROGRAM_BINARY_LENGTH, &length));
169 if (length > 0) {
Brian Osman9e4e4c72020-06-10 07:19:34 -0400170 SkBinaryWriteBuffer writer;
171 writer.writeInt(GrPersistentCacheUtils::kCurrentVersion);
172 writer.writeUInt(kGLPB_Tag);
Brian Osmana66081d2019-09-03 14:59:26 -0400173
Brian Osman9e4e4c72020-06-10 07:19:34 -0400174 writer.writePad32(&inputs, sizeof(inputs));
Brian Osman6b797fe2019-04-08 13:56:36 -0400175
Brian Osman9e4e4c72020-06-10 07:19:34 -0400176 SkAutoSMalloc<2048> binary(length);
Ethan Nicholas8d058832019-01-07 10:49:58 -0500177 GrGLenum binaryFormat;
Brian Osman9e4e4c72020-06-10 07:19:34 -0400178 GL_CALL(GetProgramBinary(programID, length, &length, &binaryFormat, binary.get()));
179
180 writer.writeUInt(binaryFormat);
181 writer.writeInt(length);
182 writer.writePad32(binary.get(), length);
Brian Osman6b797fe2019-04-08 13:56:36 -0400183
184 auto data = writer.snapshotAsData();
185 this->gpu()->getContext()->priv().getPersistentCache()->store(*key, *data);
Ethan Nicholas8d058832019-01-07 10:49:58 -0500186 }
187 } else {
Brian Osman4524e842019-09-24 16:03:41 -0400188 // source cache, plus metadata to allow for a complete precompile
189 GrPersistentCacheUtils::ShaderMetadata meta;
190 meta.fSettings = settings;
191 meta.fHasCustomColorOutput = fFS.hasCustomColorOutput();
192 meta.fHasSecondaryColorOutput = fFS.hasSecondaryOutput();
193 for (const auto& attr : this->primitiveProcessor().vertexAttributes()) {
194 meta.fAttributeNames.emplace_back(attr.name());
195 }
196 for (const auto& attr : this->primitiveProcessor().instanceAttributes()) {
197 meta.fAttributeNames.emplace_back(attr.name());
198 }
199
Brian Osmana085a412019-04-25 09:44:43 -0400200 auto data = GrPersistentCacheUtils::PackCachedShaders(isSkSL ? kSKSL_Tag : kGLSL_Tag,
Brian Osman4524e842019-09-24 16:03:41 -0400201 shaders, &inputs, 1, &meta);
Brian Osman6b797fe2019-04-08 13:56:36 -0400202 this->gpu()->getContext()->priv().getPersistentCache()->store(*key, *data);
Ethan Nicholas8d058832019-01-07 10:49:58 -0500203 }
204}
205
Stephen Whiteb1857852020-02-07 15:33:23 +0000206sk_sp<GrGLProgram> GrGLProgramBuilder::finalize(const GrGLPrecompiledProgram* precompiledProgram) {
Brian Salomon5f394272019-07-02 14:07:49 -0400207 TRACE_EVENT0("skia.gpu", TRACE_FUNC);
Ryan Macnak38a10ad2017-07-10 10:36:34 -0700208
joshualitt47bb3822014-10-07 16:43:25 -0700209 // verify we can get a program id
210 GrGLuint programID;
Brian Osmaned58e002019-09-06 14:42:43 -0400211 if (precompiledProgram) {
212 programID = precompiledProgram->fProgramID;
213 } else {
214 GL_CALL_RET(programID, CreateProgram());
215 }
joshualitt47bb3822014-10-07 16:43:25 -0700216 if (0 == programID) {
halcanary96fcdcc2015-08-27 07:41:13 -0700217 return nullptr;
joshualitt30ba4362014-08-21 20:18:45 -0700218 }
219
Ethan Nicholas06d55fb2017-11-08 09:48:50 -0500220 if (this->gpu()->glCaps().programBinarySupport() &&
Brian Osman064729e2019-06-18 17:22:59 -0400221 this->gpu()->glCaps().programParameterSupport() &&
Brian Osmaned58e002019-09-06 14:42:43 -0400222 this->gpu()->getContext()->priv().getPersistentCache() &&
223 !precompiledProgram) {
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400224 GL_CALL(ProgramParameteri(programID, GR_GL_PROGRAM_BINARY_RETRIEVABLE_HINT, GR_GL_TRUE));
225 }
226
egdaniel9f1d4152016-02-10 09:50:38 -0800227 this->finalizeShaders();
228
joshualitt47bb3822014-10-07 16:43:25 -0700229 // compile shaders and bind attributes / uniforms
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400230 auto errorHandler = this->gpu()->getContext()->priv().getShaderErrorHandler();
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400231 const GrPrimitiveProcessor& primProc = this->primitiveProcessor();
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500232 SkSL::Program::Settings settings;
Robert Phillipsd0fe8752019-01-31 14:13:59 -0500233 settings.fFlipY = this->origin() != kTopLeft_GrSurfaceOrigin;
Robert Phillipsc1541ae2019-02-04 12:05:37 -0500234 settings.fSharpenTextures =
Robert Phillips9da87e02019-02-04 13:26:26 -0500235 this->gpu()->getContext()->priv().options().fSharpenMipmappedTextures;
Brian Salomondc092132018-04-04 10:14:16 -0400236 settings.fFragColorIsInOut = this->fragColorIsInOut();
237
Ethan Nicholas941e7e22016-12-12 15:33:30 -0500238 SkSL::Program::Inputs inputs;
joshualitt30ba4362014-08-21 20:18:45 -0700239 SkTDArray<GrGLuint> shadersToDelete;
Brian Salomond8575452020-02-25 12:13:29 -0500240
241 bool checkLinked = !fGpu->glCaps().skipErrorChecks();
242
Ethan Nicholas8d058832019-01-07 10:49:58 -0500243 bool cached = fCached.get() != nullptr;
Brian Osman2c60a382019-06-26 15:19:30 -0400244 bool usedProgramBinaries = false;
Brian Osman6b797fe2019-04-08 13:56:36 -0400245 SkSL::String glsl[kGrShaderTypeCount];
Brian Osmancbc33b82019-04-19 14:16:19 -0400246 SkSL::String* sksl[kGrShaderTypeCount] = {
247 &fVS.fCompilerString,
248 &fGS.fCompilerString,
249 &fFS.fCompilerString,
250 };
Brian Osmancbc33b82019-04-19 14:16:19 -0400251 SkSL::String cached_sksl[kGrShaderTypeCount];
Brian Osmaned58e002019-09-06 14:42:43 -0400252 if (precompiledProgram) {
253 // This is very similar to when we get program binaries. We even set that flag, as it's
254 // used to prevent other compile work later, and to force re-querying uniform locations.
255 this->addInputVars(precompiledProgram->fInputs);
Brian Osman4524e842019-09-24 16:03:41 -0400256 this->computeCountsAndStrides(programID, primProc, false);
Brian Osmaned58e002019-09-06 14:42:43 -0400257 usedProgramBinaries = true;
258 } else if (cached) {
Stan Iliev43324e72020-01-13 16:17:48 -0500259 ATRACE_ANDROID_FRAMEWORK_ALWAYS("cache_hit");
Brian Osman9e4e4c72020-06-10 07:19:34 -0400260 SkReadBuffer reader(fCached->data(), fCached->size());
Brian Osman1facd5e2020-03-16 16:21:24 -0400261 SkFourByteTag shaderType = GrPersistentCacheUtils::GetType(&reader);
Brian Osmana66081d2019-09-03 14:59:26 -0400262
263 switch (shaderType) {
264 case kGLPB_Tag: {
265 // Program binary cache hit. We may opt not to use this if we don't trust program
266 // binaries on this driver
267 if (!fGpu->glCaps().programBinarySupport()) {
268 cached = false;
269 break;
Ethan Nicholas8d058832019-01-07 10:49:58 -0500270 }
Brian Osman9e4e4c72020-06-10 07:19:34 -0400271 reader.readPad32(&inputs, sizeof(inputs));
272 GrGLenum binaryFormat = reader.readUInt();
273 GrGLsizei length = reader.readInt();
Brian Osmana66081d2019-09-03 14:59:26 -0400274 const void* binary = reader.skip(length);
Brian Osman9e4e4c72020-06-10 07:19:34 -0400275 if (!reader.isValid()) {
276 break;
277 }
Brian Salomon24069eb2020-06-24 10:19:52 -0400278 this->gpu()->clearErrorsAndCheckForOOM();
Brian Osmana66081d2019-09-03 14:59:26 -0400279 GR_GL_CALL_NOERRCHECK(this->gpu()->glInterface(),
280 ProgramBinary(programID, binaryFormat,
281 const_cast<void*>(binary), length));
Brian Salomon24069eb2020-06-24 10:19:52 -0400282 if (this->gpu()->getErrorAndCheckForOOM() == GR_GL_NO_ERROR) {
Brian Osmana66081d2019-09-03 14:59:26 -0400283 if (checkLinked) {
284 cached = this->checkLinkStatus(programID, errorHandler, nullptr, nullptr);
285 }
286 if (cached) {
287 this->addInputVars(inputs);
288 this->computeCountsAndStrides(programID, primProc, false);
289 }
290 } else {
291 cached = false;
Ethan Nicholas8d058832019-01-07 10:49:58 -0500292 }
Brian Osmana66081d2019-09-03 14:59:26 -0400293 usedProgramBinaries = cached;
294 break;
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400295 }
Brian Osmana66081d2019-09-03 14:59:26 -0400296
297 case kGLSL_Tag:
298 // Source cache hit, we don't need to compile the SkSL->GLSL
299 GrPersistentCacheUtils::UnpackCachedShaders(&reader, glsl, &inputs, 1);
300 break;
301
302 case kSKSL_Tag:
303 // SkSL cache hit, this should only happen in tools overriding the generated SkSL
Brian Osman9e4e4c72020-06-10 07:19:34 -0400304 if (GrPersistentCacheUtils::UnpackCachedShaders(&reader, cached_sksl, &inputs, 1)) {
305 for (int i = 0; i < kGrShaderTypeCount; ++i) {
306 sksl[i] = &cached_sksl[i];
307 }
Brian Osmana085a412019-04-25 09:44:43 -0400308 }
Brian Osmana66081d2019-09-03 14:59:26 -0400309 break;
Brian Osman43f443f2020-06-05 11:11:36 -0400310
311 default:
312 // We got something invalid, so pretend it wasn't there
Brian Osman9e4e4c72020-06-10 07:19:34 -0400313 reader.validate(false);
Brian Osman43f443f2020-06-05 11:11:36 -0400314 break;
Ethan Nicholas98ad5b72018-03-13 09:53:02 -0400315 }
Brian Osman9e4e4c72020-06-10 07:19:34 -0400316 if (!reader.isValid()) {
317 cached = false;
318 }
Ethan Nicholas98ad5b72018-03-13 09:53:02 -0400319 }
Brian Osmana66081d2019-09-03 14:59:26 -0400320 if (!usedProgramBinaries) {
Stan Iliev43324e72020-01-13 16:17:48 -0500321 ATRACE_ANDROID_FRAMEWORK_ALWAYS("cache_miss");
Brian Osmanf7924452019-09-24 10:44:37 -0400322 // Either a cache miss, or we got something other than binaries from the cache
323
324 /*
325 Fragment Shader
326 */
Brian Osman6b797fe2019-04-08 13:56:36 -0400327 if (glsl[kFragment_GrShaderType].empty()) {
Brian Osmanf71b0702019-04-03 13:04:16 -0400328 // Don't have cached GLSL, need to compile SkSL->GLSL
Ethan Nicholas8d058832019-01-07 10:49:58 -0500329 if (fFS.fForceHighPrecision) {
330 settings.fForceHighPrecision = true;
331 }
Brian Osman79719262020-11-23 14:54:33 -0500332 std::unique_ptr<SkSL::Program> fs = GrSkSLtoGLSL(this->gpu(),
Brian Osmanac9be9d2019-05-01 10:29:34 -0400333 SkSL::Program::kFragment_Kind,
Brian Osmancbc33b82019-04-19 14:16:19 -0400334 *sksl[kFragment_GrShaderType],
Ethan Nicholas8d058832019-01-07 10:49:58 -0500335 settings,
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400336 &glsl[kFragment_GrShaderType],
337 errorHandler);
Ethan Nicholas8d058832019-01-07 10:49:58 -0500338 if (!fs) {
Brian Osmaned58e002019-09-06 14:42:43 -0400339 cleanup_program(fGpu, programID, shadersToDelete);
Ethan Nicholas8d058832019-01-07 10:49:58 -0500340 return nullptr;
341 }
342 inputs = fs->fInputs;
egdaniel8dcdedc2015-11-11 06:27:20 -0800343 }
Brian Osmanf7924452019-09-24 10:44:37 -0400344
345 this->addInputVars(inputs);
Brian Osmanac9be9d2019-05-01 10:29:34 -0400346 if (!this->compileAndAttachShaders(glsl[kFragment_GrShaderType], programID,
Brian Osmane11dfd32019-07-23 10:29:41 -0400347 GR_GL_FRAGMENT_SHADER, &shadersToDelete, errorHandler)) {
Brian Osmaned58e002019-09-06 14:42:43 -0400348 cleanup_program(fGpu, programID, shadersToDelete);
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400349 return nullptr;
350 }
351
Brian Osmanf7924452019-09-24 10:44:37 -0400352 /*
353 Vertex Shader
354 */
Brian Osman6b797fe2019-04-08 13:56:36 -0400355 if (glsl[kVertex_GrShaderType].empty()) {
Brian Osmanf71b0702019-04-03 13:04:16 -0400356 // Don't have cached GLSL, need to compile SkSL->GLSL
Brian Osman79719262020-11-23 14:54:33 -0500357 std::unique_ptr<SkSL::Program> vs = GrSkSLtoGLSL(this->gpu(),
Brian Osmanac9be9d2019-05-01 10:29:34 -0400358 SkSL::Program::kVertex_Kind,
Brian Osmancbc33b82019-04-19 14:16:19 -0400359 *sksl[kVertex_GrShaderType],
Brian Osmanf71b0702019-04-03 13:04:16 -0400360 settings,
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400361 &glsl[kVertex_GrShaderType],
362 errorHandler);
Brian Osmanf71b0702019-04-03 13:04:16 -0400363 if (!vs) {
Brian Osmaned58e002019-09-06 14:42:43 -0400364 cleanup_program(fGpu, programID, shadersToDelete);
Brian Osmanf71b0702019-04-03 13:04:16 -0400365 return nullptr;
366 }
367 }
Brian Osmanac9be9d2019-05-01 10:29:34 -0400368 if (!this->compileAndAttachShaders(glsl[kVertex_GrShaderType], programID,
Brian Osmane11dfd32019-07-23 10:29:41 -0400369 GR_GL_VERTEX_SHADER, &shadersToDelete, errorHandler)) {
Brian Osmaned58e002019-09-06 14:42:43 -0400370 cleanup_program(fGpu, programID, shadersToDelete);
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400371 return nullptr;
372 }
373
Brian Osmanf7924452019-09-24 10:44:37 -0400374 // This also binds vertex attribute locations. NVPR doesn't really use vertices,
375 // even though it requires a vertex shader in the program.
376 if (!primProc.isPathRendering()) {
Ethan Nicholasad77dce2018-07-11 13:59:03 -0400377 this->computeCountsAndStrides(programID, primProc, true);
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400378 }
379
Brian Osmanf7924452019-09-24 10:44:37 -0400380 /*
Chris Dalton5a2f9622019-12-27 14:56:38 -0700381 Tessellation Shaders
382 */
383 if (fProgramInfo.primProc().willUseTessellationShaders()) {
384 // Tessellation shaders are not currently supported by SkSL. So here, we temporarily
385 // generate GLSL strings directly using back door methods on GrPrimitiveProcessor, and
386 // pass those raw strings on to the driver.
387 SkString versionAndExtensionDecls;
388 versionAndExtensionDecls.appendf("%s\n", this->shaderCaps()->versionDeclString());
389 if (const char* extensionString = this->shaderCaps()->tessellationExtensionString()) {
390 versionAndExtensionDecls.appendf("#extension %s : require\n", extensionString);
391 }
392
393 SkString tessControlShader = primProc.getTessControlShaderGLSL(
Chris Dalton54c90932020-06-24 10:51:52 -0600394 fGeometryProcessor.get(), versionAndExtensionDecls.c_str(), fUniformHandler,
395 *this->shaderCaps());
Chris Dalton5a2f9622019-12-27 14:56:38 -0700396 if (!this->compileAndAttachShaders(tessControlShader.c_str(), programID,
397 GR_GL_TESS_CONTROL_SHADER, &shadersToDelete,
398 errorHandler)) {
399 cleanup_program(fGpu, programID, shadersToDelete);
400 return nullptr;
401 }
402
403 SkString tessEvaluationShader = primProc.getTessEvaluationShaderGLSL(
Chris Dalton54c90932020-06-24 10:51:52 -0600404 fGeometryProcessor.get(), versionAndExtensionDecls.c_str(), fUniformHandler,
405 *this->shaderCaps());
Chris Dalton5a2f9622019-12-27 14:56:38 -0700406 if (!this->compileAndAttachShaders(tessEvaluationShader.c_str(), programID,
407 GR_GL_TESS_EVALUATION_SHADER, &shadersToDelete,
408 errorHandler)) {
409 cleanup_program(fGpu, programID, shadersToDelete);
410 return nullptr;
411 }
412 }
413
414 /*
Brian Osmanf7924452019-09-24 10:44:37 -0400415 Geometry Shader
416 */
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400417 if (primProc.willUseGeoShader()) {
Brian Osman6b797fe2019-04-08 13:56:36 -0400418 if (glsl[kGeometry_GrShaderType].empty()) {
Brian Osmanf71b0702019-04-03 13:04:16 -0400419 // Don't have cached GLSL, need to compile SkSL->GLSL
420 std::unique_ptr<SkSL::Program> gs;
Brian Osman79719262020-11-23 14:54:33 -0500421 gs = GrSkSLtoGLSL(this->gpu(),
Brian Osmanac9be9d2019-05-01 10:29:34 -0400422 SkSL::Program::kGeometry_Kind,
Brian Osmancbc33b82019-04-19 14:16:19 -0400423 *sksl[kGeometry_GrShaderType],
Brian Osmanf71b0702019-04-03 13:04:16 -0400424 settings,
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400425 &glsl[kGeometry_GrShaderType],
426 errorHandler);
Brian Osmanf71b0702019-04-03 13:04:16 -0400427 if (!gs) {
Brian Osmaned58e002019-09-06 14:42:43 -0400428 cleanup_program(fGpu, programID, shadersToDelete);
Brian Osmanf71b0702019-04-03 13:04:16 -0400429 return nullptr;
430 }
431 }
Brian Osmanac9be9d2019-05-01 10:29:34 -0400432 if (!this->compileAndAttachShaders(glsl[kGeometry_GrShaderType], programID,
Brian Osmane11dfd32019-07-23 10:29:41 -0400433 GR_GL_GEOMETRY_SHADER, &shadersToDelete,
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400434 errorHandler)) {
Brian Osmaned58e002019-09-06 14:42:43 -0400435 cleanup_program(fGpu, programID, shadersToDelete);
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400436 return nullptr;
437 }
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400438 }
439 this->bindProgramResourceLocations(programID);
440
441 GL_CALL(LinkProgram(programID));
Ethan Nicholas5a0338c2019-01-02 11:48:56 -0500442 if (checkLinked) {
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400443 if (!this->checkLinkStatus(programID, errorHandler, sksl, glsl)) {
Brian Osman2ab2ac02019-09-09 13:54:54 -0400444 cleanup_program(fGpu, programID, shadersToDelete);
Ethan Nicholas5a0338c2019-01-02 11:48:56 -0500445 return nullptr;
Brian Salomone334c592017-05-15 11:00:58 -0400446 }
Brian Salomone334c592017-05-15 11:00:58 -0400447 }
joshualittfe1233c2014-10-07 12:16:35 -0700448 }
Brian Osman2c60a382019-06-26 15:19:30 -0400449 this->resolveProgramResourceLocations(programID, usedProgramBinaries);
joshualittdb0d3ca2014-10-07 12:42:26 -0700450
Brian Osmaned58e002019-09-06 14:42:43 -0400451 cleanup_shaders(fGpu, shadersToDelete);
Brian Osmane4c88bb2019-06-27 16:15:11 -0400452
Chris Dalton5a2f9622019-12-27 14:56:38 -0700453 // We temporarily can't cache tessellation shaders while using back door GLSL.
454 //
Brian Osmaned58e002019-09-06 14:42:43 -0400455 // We also can't cache SkSL or GLSL if we were given a precompiled program, but there's not
456 // much point in doing so.
Brian Osman0b22f3b2020-03-23 13:21:24 -0400457 if (!cached && !primProc.willUseTessellationShaders() && !precompiledProgram) {
Chris Dalton8dae7eb2019-12-27 22:20:55 -0700458 // FIXME: Remove the check for tessellation shaders in the above 'if' once the back door
459 // GLSL mechanism is removed.
460 (void)&GrPrimitiveProcessor::getTessControlShaderGLSL;
Brian Osmana085a412019-04-25 09:44:43 -0400461 bool isSkSL = false;
Brian Osmana66081d2019-09-03 14:59:26 -0400462 if (fGpu->getContext()->priv().options().fShaderCacheStrategy ==
463 GrContextOptions::ShaderCacheStrategy::kSkSL) {
Brian Osmancbc33b82019-04-19 14:16:19 -0400464 for (int i = 0; i < kGrShaderTypeCount; ++i) {
Brian Osmanac9be9d2019-05-01 10:29:34 -0400465 glsl[i] = GrShaderUtils::PrettyPrint(*sksl[i]);
Brian Osmancbc33b82019-04-19 14:16:19 -0400466 }
Brian Osmana085a412019-04-25 09:44:43 -0400467 isSkSL = true;
Brian Osmancbc33b82019-04-19 14:16:19 -0400468 }
Brian Osman4524e842019-09-24 16:03:41 -0400469 this->storeShaderInCache(inputs, programID, glsl, isSkSL, &settings);
Ethan Nicholasd1b2eec2017-11-01 15:45:43 -0400470 }
joshualitt47bb3822014-10-07 16:43:25 -0700471 return this->createProgram(programID);
472}
473
kkinnunen7aedda52015-06-29 23:01:28 -0700474void GrGLProgramBuilder::bindProgramResourceLocations(GrGLuint programID) {
egdaniel7ea439b2015-12-03 09:20:44 -0800475 fUniformHandler.bindUniformLocations(programID, fGpu->glCaps());
kkinnunen7aedda52015-06-29 23:01:28 -0700476
egdaniel8dcdedc2015-11-11 06:27:20 -0800477 const GrGLCaps& caps = this->gpu()->glCaps();
478 if (fFS.hasCustomColorOutput() && caps.bindFragDataLocationSupport()) {
479 GL_CALL(BindFragDataLocation(programID, 0,
egdaniel2d721d32015-11-11 13:06:05 -0800480 GrGLSLFragmentShaderBuilder::DeclaredColorOutputName()));
egdaniel8dcdedc2015-11-11 06:27:20 -0800481 }
Brian Salomon1edc5b92016-11-29 13:43:46 -0500482 if (fFS.hasSecondaryOutput() && caps.shaderCaps()->mustDeclareFragmentShaderOutput()) {
egdaniel8dcdedc2015-11-11 06:27:20 -0800483 GL_CALL(BindFragDataLocationIndexed(programID, 0, 1,
egdaniel2d721d32015-11-11 13:06:05 -0800484 GrGLSLFragmentShaderBuilder::DeclaredSecondaryColorOutputName()));
egdaniel8dcdedc2015-11-11 06:27:20 -0800485 }
joshualittd8dd47b2015-09-11 11:45:01 -0700486
487 // handle NVPR separable varyings
488 if (!fGpu->glCaps().shaderCaps()->pathRenderingSupport() ||
489 !fGpu->glPathRendering()->shouldBindFragmentInputs()) {
490 return;
491 }
Michael Ludwig45191342020-03-24 12:29:39 -0400492 int i = 0;
493 for (auto& varying : fVaryingHandler.fPathProcVaryingInfos.items()) {
494 GL_CALL(BindFragmentInputLocation(programID, i, varying.fVariable.c_str()));
495 varying.fLocation = i;
496 ++i;
joshualittd8dd47b2015-09-11 11:45:01 -0700497 }
joshualitt47bb3822014-10-07 16:43:25 -0700498}
499
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400500bool GrGLProgramBuilder::checkLinkStatus(GrGLuint programID,
501 GrContextOptions::ShaderErrorHandler* errorHandler,
502 SkSL::String* sksl[], const SkSL::String glsl[]) {
joshualitt47bb3822014-10-07 16:43:25 -0700503 GrGLint linked = GR_GL_INIT_ZERO;
504 GL_CALL(GetProgramiv(programID, GR_GL_LINK_STATUS, &linked));
505 if (!linked) {
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400506 SkSL::String allShaders;
507 if (sksl) {
508 allShaders.appendf("// Vertex SKSL\n%s\n", sksl[kVertex_GrShaderType]->c_str());
509 if (!sksl[kGeometry_GrShaderType]->empty()) {
510 allShaders.appendf("// Geometry SKSL\n%s\n", sksl[kGeometry_GrShaderType]->c_str());
511 }
512 allShaders.appendf("// Fragment SKSL\n%s\n", sksl[kFragment_GrShaderType]->c_str());
513 }
514 if (glsl) {
515 allShaders.appendf("// Vertex GLSL\n%s\n", glsl[kVertex_GrShaderType].c_str());
516 if (!glsl[kGeometry_GrShaderType].empty()) {
517 allShaders.appendf("// Geometry GLSL\n%s\n", glsl[kGeometry_GrShaderType].c_str());
518 }
519 allShaders.appendf("// Fragment GLSL\n%s\n", glsl[kFragment_GrShaderType].c_str());
520 }
joshualitt47bb3822014-10-07 16:43:25 -0700521 GrGLint infoLen = GR_GL_INIT_ZERO;
522 GL_CALL(GetProgramiv(programID, GR_GL_INFO_LOG_LENGTH, &infoLen));
523 SkAutoMalloc log(sizeof(char)*(infoLen+1)); // outside if for debugger
524 if (infoLen > 0) {
525 // retrieve length even though we don't need it to workaround
526 // bug in chrome cmd buffer param validation.
527 GrGLsizei length = GR_GL_INIT_ZERO;
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400528 GL_CALL(GetProgramInfoLog(programID, infoLen+1, &length, (char*)log.get()));
joshualitt47bb3822014-10-07 16:43:25 -0700529 }
Brian Osman5e7fbfd2019-05-03 13:13:35 -0400530 errorHandler->compileError(allShaders.c_str(), infoLen > 0 ? (const char*)log.get() : "");
joshualitt47bb3822014-10-07 16:43:25 -0700531 }
532 return SkToBool(linked);
533}
534
Brian Osman2c60a382019-06-26 15:19:30 -0400535void GrGLProgramBuilder::resolveProgramResourceLocations(GrGLuint programID, bool force) {
536 fUniformHandler.getUniformLocations(programID, fGpu->glCaps(), force);
joshualittd8dd47b2015-09-11 11:45:01 -0700537
538 // handle NVPR separable varyings
539 if (!fGpu->glCaps().shaderCaps()->pathRenderingSupport() ||
kkinnunen7bdb05d2016-01-25 00:47:23 -0800540 fGpu->glPathRendering()->shouldBindFragmentInputs()) {
joshualittd8dd47b2015-09-11 11:45:01 -0700541 return;
542 }
Michael Ludwig45191342020-03-24 12:29:39 -0400543 for (auto& varying : fVaryingHandler.fPathProcVaryingInfos.items()) {
joshualittd8dd47b2015-09-11 11:45:01 -0700544 GrGLint location;
egdaniel0eafe792015-11-20 14:01:22 -0800545 GL_CALL_RET(location, GetProgramResourceLocation(
546 programID,
547 GR_GL_FRAGMENT_INPUT,
Michael Ludwig45191342020-03-24 12:29:39 -0400548 varying.fVariable.c_str()));
549 varying.fLocation = location;
joshualittd8dd47b2015-09-11 11:45:01 -0700550 }
joshualitt30ba4362014-08-21 20:18:45 -0700551}
552
Stephen Whiteb1857852020-02-07 15:33:23 +0000553sk_sp<GrGLProgram> GrGLProgramBuilder::createProgram(GrGLuint programID) {
Robert Phillipsdf546db2020-05-29 09:42:54 -0400554 return GrGLProgram::Make(fGpu,
555 fUniformHandles,
556 programID,
557 fUniformHandler.fUniforms,
558 fUniformHandler.fSamplers,
559 fVaryingHandler.fPathProcVaryingInfos,
560 std::move(fGeometryProcessor),
561 std::move(fXferProcessor),
562 std::move(fFragmentProcessors),
Robert Phillipsdf546db2020-05-29 09:42:54 -0400563 std::move(fAttributes),
564 fVertexAttributeCnt,
565 fInstanceAttributeCnt,
566 fVertexStride,
567 fInstanceStride);
joshualitt47bb3822014-10-07 16:43:25 -0700568}
Brian Osmaned58e002019-09-06 14:42:43 -0400569
570bool GrGLProgramBuilder::PrecompileProgram(GrGLPrecompiledProgram* precompiledProgram,
571 GrGLGpu* gpu,
572 const SkData& cachedData) {
Brian Osman9e4e4c72020-06-10 07:19:34 -0400573 SkReadBuffer reader(cachedData.data(), cachedData.size());
Brian Osman1facd5e2020-03-16 16:21:24 -0400574 SkFourByteTag shaderType = GrPersistentCacheUtils::GetType(&reader);
Brian Osmaned58e002019-09-06 14:42:43 -0400575 if (shaderType != kSKSL_Tag) {
576 // TODO: Support GLSL, and maybe even program binaries, too?
577 return false;
578 }
579
580 const GrGLInterface* gl = gpu->glInterface();
581 auto errorHandler = gpu->getContext()->priv().getShaderErrorHandler();
Brian Osmaned58e002019-09-06 14:42:43 -0400582
583 SkSL::Program::Settings settings;
Brian Osmaned58e002019-09-06 14:42:43 -0400584 settings.fSharpenTextures = gpu->getContext()->priv().options().fSharpenMipmappedTextures;
Brian Osman4524e842019-09-24 16:03:41 -0400585 GrPersistentCacheUtils::ShaderMetadata meta;
586 meta.fSettings = &settings;
Brian Osmaned58e002019-09-06 14:42:43 -0400587
588 SkSL::String shaders[kGrShaderTypeCount];
589 SkSL::Program::Inputs inputs;
Brian Osman9e4e4c72020-06-10 07:19:34 -0400590 if (!GrPersistentCacheUtils::UnpackCachedShaders(&reader, shaders, &inputs, 1, &meta)) {
591 return false;
592 }
593
594 GrGLuint programID;
595 GR_GL_CALL_RET(gl, programID, CreateProgram());
596 if (0 == programID) {
597 return false;
598 }
599
600 SkTDArray<GrGLuint> shadersToDelete;
Brian Osmaned58e002019-09-06 14:42:43 -0400601
602 auto compileShader = [&](SkSL::Program::Kind kind, const SkSL::String& sksl, GrGLenum type) {
603 SkSL::String glsl;
Brian Osman79719262020-11-23 14:54:33 -0500604 auto program = GrSkSLtoGLSL(gpu, kind, sksl, settings, &glsl, errorHandler);
Brian Osmaned58e002019-09-06 14:42:43 -0400605 if (!program) {
606 return false;
607 }
608
609 if (GrGLuint shaderID = GrGLCompileAndAttachShader(gpu->glContext(), programID, type, glsl,
610 gpu->stats(), errorHandler)) {
611 shadersToDelete.push_back(shaderID);
612 return true;
613 } else {
614 return false;
615 }
616 };
617
618 if (!compileShader(SkSL::Program::kFragment_Kind,
619 shaders[kFragment_GrShaderType],
620 GR_GL_FRAGMENT_SHADER) ||
621 !compileShader(SkSL::Program::kVertex_Kind,
622 shaders[kVertex_GrShaderType],
623 GR_GL_VERTEX_SHADER) ||
624 (!shaders[kGeometry_GrShaderType].empty() &&
625 !compileShader(SkSL::Program::kGeometry_Kind,
626 shaders[kGeometry_GrShaderType],
627 GR_GL_GEOMETRY_SHADER))) {
628 cleanup_program(gpu, programID, shadersToDelete);
629 return false;
630 }
631
Brian Osman4524e842019-09-24 16:03:41 -0400632 for (int i = 0; i < meta.fAttributeNames.count(); ++i) {
633 GR_GL_CALL(gpu->glInterface(), BindAttribLocation(programID, i,
634 meta.fAttributeNames[i].c_str()));
635 }
636
Brian Osmand7e76592020-11-02 12:26:22 -0500637 const GrGLCaps& caps = gpu->glCaps();
Brian Osman4524e842019-09-24 16:03:41 -0400638 if (meta.fHasCustomColorOutput && caps.bindFragDataLocationSupport()) {
639 GR_GL_CALL(gpu->glInterface(), BindFragDataLocation(programID, 0,
640 GrGLSLFragmentShaderBuilder::DeclaredColorOutputName()));
641 }
642 if (meta.fHasSecondaryColorOutput && caps.shaderCaps()->mustDeclareFragmentShaderOutput()) {
643 GR_GL_CALL(gpu->glInterface(), BindFragDataLocationIndexed(programID, 0, 1,
644 GrGLSLFragmentShaderBuilder::DeclaredSecondaryColorOutputName()));
645 }
646
Brian Osmaned58e002019-09-06 14:42:43 -0400647 GR_GL_CALL(gpu->glInterface(), LinkProgram(programID));
648 GrGLint linked = GR_GL_INIT_ZERO;
649 GR_GL_CALL(gpu->glInterface(), GetProgramiv(programID, GR_GL_LINK_STATUS, &linked));
650 if (!linked) {
651 cleanup_program(gpu, programID, shadersToDelete);
652 return false;
653 }
654
655 cleanup_shaders(gpu, shadersToDelete);
656
657 precompiledProgram->fProgramID = programID;
658 precompiledProgram->fInputs = inputs;
659 return true;
660}