blob: 551851c10e8e8565c2c8d8bd42121ee7e88b9a25 [file] [log] [blame]
egdaniel7ea439b2015-12-03 09:20:44 -08001/*
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
Mike Kleinc0bd9f92019-04-23 12:05:21 -05008#include "src/gpu/gl/GrGLUniformHandler.h"
egdaniel7ea439b2015-12-03 09:20:44 -08009
Mike Kleinc0bd9f92019-04-23 12:05:21 -050010#include "src/gpu/GrTexturePriv.h"
11#include "src/gpu/gl/GrGLCaps.h"
12#include "src/gpu/gl/GrGLGpu.h"
13#include "src/gpu/gl/builders/GrGLProgramBuilder.h"
14#include "src/sksl/SkSLCompiler.h"
egdaniel7ea439b2015-12-03 09:20:44 -080015
16#define GL_CALL(X) GR_GL_CALL(this->glGpu()->glInterface(), X)
17#define GL_CALL_RET(R, X) GR_GL_CALL_RET(this->glGpu()->glInterface(), R, X)
18
Robert Phillipsfe8da172018-01-24 14:52:02 +000019bool valid_name(const char* name) {
20 // disallow unknown names that start with "sk_"
21 if (!strncmp(name, GR_NO_MANGLE_PREFIX, strlen(GR_NO_MANGLE_PREFIX))) {
22 return !strcmp(name, SkSL::Compiler::RTADJUST_NAME);
23 }
24 return true;
25}
26
egdaniel7ea439b2015-12-03 09:20:44 -080027GrGLSLUniformHandler::UniformHandle GrGLUniformHandler::internalAddUniformArray(
28 uint32_t visibility,
29 GrSLType type,
egdaniel7ea439b2015-12-03 09:20:44 -080030 const char* name,
31 bool mangleName,
32 int arrayCount,
33 const char** outName) {
34 SkASSERT(name && strlen(name));
Robert Phillipsfe8da172018-01-24 14:52:02 +000035 SkASSERT(valid_name(name));
egdaniel7ea439b2015-12-03 09:20:44 -080036 SkASSERT(0 != visibility);
egdaniel7ea439b2015-12-03 09:20:44 -080037
38 UniformInfo& uni = fUniforms.push_back();
39 uni.fVariable.setType(type);
Brian Salomon99938a82016-11-21 13:41:08 -050040 uni.fVariable.setTypeModifier(GrShaderVar::kUniform_TypeModifier);
egdaniel7ea439b2015-12-03 09:20:44 -080041 // TODO this is a bit hacky, lets think of a better way. Basically we need to be able to use
42 // the uniform view matrix name in the GP, and the GP is immutable so it has to tell the PB
43 // exactly what name it wants to use for the uniform view matrix. If we prefix anythings, then
44 // the names will mismatch. I think the correct solution is to have all GPs which need the
45 // uniform view matrix, they should upload the view matrix in their setData along with regular
46 // uniforms.
47 char prefix = 'u';
Robert Phillipsfe8da172018-01-24 14:52:02 +000048 if ('u' == name[0] || !strncmp(name, GR_NO_MANGLE_PREFIX, strlen(GR_NO_MANGLE_PREFIX))) {
egdaniel7ea439b2015-12-03 09:20:44 -080049 prefix = '\0';
50 }
51 fProgramBuilder->nameVariable(uni.fVariable.accessName(), prefix, name, mangleName);
52 uni.fVariable.setArrayCount(arrayCount);
53 uni.fVisibility = visibility;
Brian Salomon101b8442016-11-18 11:58:54 -050054 uni.fLocation = -1;
egdaniel7ea439b2015-12-03 09:20:44 -080055
56 if (outName) {
57 *outName = uni.fVariable.c_str();
58 }
59 return GrGLSLUniformHandler::UniformHandle(fUniforms.count() - 1);
60}
61
Greg Daniel9a51a862018-11-30 10:18:14 -050062GrGLSLUniformHandler::SamplerHandle GrGLUniformHandler::addSampler(const GrTexture* texture,
63 const GrSamplerState&,
Greg Daniel2c3398d2019-06-19 11:58:01 -040064 const GrSwizzle& swizzle,
Greg Daniel9a51a862018-11-30 10:18:14 -050065 const char* name,
66 const GrShaderCaps* shaderCaps) {
egdaniel09aa1fc2016-04-20 07:09:46 -070067 SkASSERT(name && strlen(name));
Brian Salomon101b8442016-11-18 11:58:54 -050068
egdaniel09aa1fc2016-04-20 07:09:46 -070069 SkString mangleName;
70 char prefix = 'u';
71 fProgramBuilder->nameVariable(&mangleName, prefix, name, true);
Brian Salomon101b8442016-11-18 11:58:54 -050072
Greg Daniel9a51a862018-11-30 10:18:14 -050073 GrTextureType type = texture->texturePriv().textureType();
74
Brian Salomon101b8442016-11-18 11:58:54 -050075 UniformInfo& sampler = fSamplers.push_back();
Brian Salomon60dd8c72018-07-30 10:24:13 -040076 sampler.fVariable.setType(GrSLCombinedSamplerTypeForTextureType(type));
Brian Salomon99938a82016-11-21 13:41:08 -050077 sampler.fVariable.setTypeModifier(GrShaderVar::kUniform_TypeModifier);
Brian Salomon101b8442016-11-18 11:58:54 -050078 sampler.fVariable.setName(mangleName);
79 sampler.fLocation = -1;
Greg Daniel0f70be82018-10-08 17:35:08 +000080 sampler.fVisibility = kFragment_GrShaderFlag;
Brian Salomon68ba1172019-06-05 11:15:08 -040081 if (shaderCaps->textureSwizzleAppliedInShader()) {
82 fSamplerSwizzles.push_back(swizzle);
83 SkASSERT(fSamplers.count() == fSamplerSwizzles.count());
84 }
egdaniel09aa1fc2016-04-20 07:09:46 -070085 return GrGLSLUniformHandler::SamplerHandle(fSamplers.count() - 1);
86}
87
cdalton5e58cee2016-02-11 12:49:47 -080088void GrGLUniformHandler::appendUniformDecls(GrShaderFlags visibility, SkString* out) const {
egdaniel7ea439b2015-12-03 09:20:44 -080089 for (int i = 0; i < fUniforms.count(); ++i) {
90 if (fUniforms[i].fVisibility & visibility) {
Brian Salomon94efbf52016-11-29 13:43:05 -050091 fUniforms[i].fVariable.appendDecl(fProgramBuilder->shaderCaps(), out);
Brian Salomonf9f45122016-11-29 11:59:17 -050092 out->append(";");
egdaniel7ea439b2015-12-03 09:20:44 -080093 }
94 }
egdaniel09aa1fc2016-04-20 07:09:46 -070095 for (int i = 0; i < fSamplers.count(); ++i) {
Brian Salomon101b8442016-11-18 11:58:54 -050096 if (fSamplers[i].fVisibility & visibility) {
Brian Salomon94efbf52016-11-29 13:43:05 -050097 fSamplers[i].fVariable.appendDecl(fProgramBuilder->shaderCaps(), out);
egdaniel09aa1fc2016-04-20 07:09:46 -070098 out->append(";\n");
99 }
100 }
egdaniel7ea439b2015-12-03 09:20:44 -0800101}
102
103void GrGLUniformHandler::bindUniformLocations(GrGLuint programID, const GrGLCaps& caps) {
104 if (caps.bindUniformLocationSupport()) {
Brian Salomonf9f45122016-11-29 11:59:17 -0500105 int currUniform = 0;
106 for (int i = 0; i < fUniforms.count(); ++i, ++currUniform) {
107 GL_CALL(BindUniformLocation(programID, currUniform, fUniforms[i].fVariable.c_str()));
108 fUniforms[i].fLocation = currUniform;
egdaniel7ea439b2015-12-03 09:20:44 -0800109 }
Brian Salomonf9f45122016-11-29 11:59:17 -0500110 for (int i = 0; i < fSamplers.count(); ++i, ++currUniform) {
111 GL_CALL(BindUniformLocation(programID, currUniform, fSamplers[i].fVariable.c_str()));
112 fSamplers[i].fLocation = currUniform;
113 }
egdaniel7ea439b2015-12-03 09:20:44 -0800114 }
115}
116
Brian Osman2c60a382019-06-26 15:19:30 -0400117void GrGLUniformHandler::getUniformLocations(GrGLuint programID, const GrGLCaps& caps, bool force) {
118 if (!caps.bindUniformLocationSupport() || force) {
egdaniel7ea439b2015-12-03 09:20:44 -0800119 int count = fUniforms.count();
120 for (int i = 0; i < count; ++i) {
121 GrGLint location;
122 GL_CALL_RET(location, GetUniformLocation(programID, fUniforms[i].fVariable.c_str()));
123 fUniforms[i].fLocation = location;
124 }
egdaniel09aa1fc2016-04-20 07:09:46 -0700125 for (int i = 0; i < fSamplers.count(); ++i) {
126 GrGLint location;
Brian Salomon101b8442016-11-18 11:58:54 -0500127 GL_CALL_RET(location, GetUniformLocation(programID, fSamplers[i].fVariable.c_str()));
egdaniel09aa1fc2016-04-20 07:09:46 -0700128 fSamplers[i].fLocation = location;
129 }
egdaniel7ea439b2015-12-03 09:20:44 -0800130 }
131}
132
133const GrGLGpu* GrGLUniformHandler::glGpu() const {
134 GrGLProgramBuilder* glPB = (GrGLProgramBuilder*) fProgramBuilder;
135 return glPB->gpu();
136}