diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /dom/canvas/WebGLShader.cpp | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'dom/canvas/WebGLShader.cpp')
-rw-r--r-- | dom/canvas/WebGLShader.cpp | 190 |
1 files changed, 190 insertions, 0 deletions
diff --git a/dom/canvas/WebGLShader.cpp b/dom/canvas/WebGLShader.cpp new file mode 100644 index 0000000000..033e97fe2b --- /dev/null +++ b/dom/canvas/WebGLShader.cpp @@ -0,0 +1,190 @@ +/* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#include "WebGLShader.h" + +#include "GLSLANG/ShaderLang.h" +#include "GLContext.h" +#include "mozilla/dom/WebGLRenderingContextBinding.h" +#include "mozilla/MemoryReporting.h" +#include "nsPrintfCString.h" +#include "nsString.h" +#include "prenv.h" +#include "WebGLContext.h" +#include "WebGLObjectModel.h" +#include "WebGLShaderValidator.h" +#include "WebGLValidateStrings.h" + +namespace mozilla { + +static void PrintLongString(const char* const begin, const size_t len) { + // Wow - Roll Your Own Foreach-Lines because printf_stderr has a hard-coded + // internal size, so long strings are truncated. + + const size_t chunkSize = 1000; + auto buf = std::vector<char>(chunkSize + 1); // +1 for null-term + const auto bufBegin = buf.data(); + + auto chunkBegin = begin; + const auto end = begin + len; + while (chunkBegin + chunkSize < end) { + memcpy(bufBegin, chunkBegin, chunkSize); + printf_stderr("%s", bufBegin); + chunkBegin += chunkSize; + } + printf_stderr("%s", chunkBegin); +} + +template <size_t N> +static bool SubstringStartsWith(const std::string& testStr, size_t offset, + const char (&refStr)[N]) { + for (size_t i = 0; i < N - 1; i++) { + if (testStr[offset + i] != refStr[i]) return false; + } + return true; +} + +static void GetCompilationStatusAndLog(gl::GLContext* gl, GLuint shader, + bool* const out_success, + std::string* const out_log) { + GLint compileStatus = LOCAL_GL_FALSE; + gl->fGetShaderiv(shader, LOCAL_GL_COMPILE_STATUS, &compileStatus); + + // It's simpler if we always get the log. + GLint lenWithNull = 0; + gl->fGetShaderiv(shader, LOCAL_GL_INFO_LOG_LENGTH, &lenWithNull); + if (lenWithNull < 1) { + lenWithNull = 1; + } + std::vector<char> buffer(lenWithNull); + gl->fGetShaderInfoLog(shader, buffer.size(), nullptr, buffer.data()); + *out_log = buffer.data(); + + *out_success = (compileStatus == LOCAL_GL_TRUE); +} + +//////////////////////////////////////////////////////////////////////////////// + +WebGLShader::WebGLShader(WebGLContext* webgl, GLenum type) + : WebGLContextBoundObject(webgl), + mGLName(webgl->gl->fCreateShader(type)), + mType(type) { + mCompileResults = std::make_unique<webgl::ShaderValidatorResults>(); +} + +WebGLShader::~WebGLShader() { + if (!mContext) return; + mContext->gl->fDeleteShader(mGLName); +} + +void WebGLShader::ShaderSource(const std::string& u8) { + mSource = CrushGlslToAscii(u8); +} + +void WebGLShader::CompileShader() { + mCompilationSuccessful = false; + + gl::GLContext* gl = mContext->gl; + + static const bool kDumpShaders = PR_GetEnv("MOZ_WEBGL_DUMP_SHADERS"); + if (MOZ_UNLIKELY(kDumpShaders)) { + printf_stderr("==== begin MOZ_WEBGL_DUMP_SHADERS ====\n"); + PrintLongString(mSource.c_str(), mSource.size()); + } + + { + const auto validator = mContext->CreateShaderValidator(mType); + MOZ_ASSERT(validator); + + mCompileResults = validator->ValidateAndTranslate(mSource.c_str()); + } + + mCompilationLog = mCompileResults->mInfoLog.c_str(); + const auto& success = mCompileResults->mValid; + + if (MOZ_UNLIKELY(kDumpShaders)) { + printf_stderr("\n==== \\/ \\/ \\/ ====\n"); + if (success) { + const auto& translated = mCompileResults->mObjectCode; + PrintLongString(translated.data(), translated.size()); + } else { + printf_stderr("Validation failed:\n%s", + mCompileResults->mInfoLog.c_str()); + } + printf_stderr("\n==== end ====\n"); + } + + if (!success) return; + + const std::array<const char*, 1> parts = { + mCompileResults->mObjectCode.c_str()}; + gl->fShaderSource(mGLName, parts.size(), parts.data(), nullptr); + + gl->fCompileShader(mGLName); + + GetCompilationStatusAndLog(gl, mGLName, &mCompilationSuccessful, + &mCompilationLog); +} + +//////////////////////////////////////////////////////////////////////////////// + +size_t WebGLShader::CalcNumSamplerUniforms() const { + size_t accum = 0; + for (const auto& cur : mCompileResults->mUniforms) { + const auto& type = cur.type; + if (type == LOCAL_GL_SAMPLER_2D || type == LOCAL_GL_SAMPLER_CUBE) { + accum += cur.getArraySizeProduct(); + } + } + return accum; +} + +size_t WebGLShader::NumAttributes() const { + return mCompileResults->mAttributes.size(); +} + +void WebGLShader::BindAttribLocation(GLuint prog, const std::string& userName, + GLuint index) const { + for (const auto& attrib : mCompileResults->mAttributes) { + if (attrib.name == userName) { + mContext->gl->fBindAttribLocation(prog, index, attrib.mappedName.c_str()); + return; + } + } +} + +void WebGLShader::MapTransformFeedbackVaryings( + const std::vector<std::string>& varyings, + std::vector<std::string>* out_mappedVaryings) const { + MOZ_ASSERT(mType == LOCAL_GL_VERTEX_SHADER); + MOZ_ASSERT(out_mappedVaryings); + + out_mappedVaryings->clear(); + out_mappedVaryings->reserve(varyings.size()); + + const auto& shaderVaryings = mCompileResults->mVaryings; + + for (const auto& userName : varyings) { + const auto* mappedName = &userName; + for (const auto& shaderVarying : shaderVaryings) { + if (shaderVarying.name == userName) { + mappedName = &shaderVarying.mappedName; + break; + } + } + out_mappedVaryings->push_back(*mappedName); + } +} + +//////////////////////////////////////////////////////////////////////////////// +// Boilerplate + +size_t WebGLShader::SizeOfIncludingThis(MallocSizeOf mallocSizeOf) const { + return mallocSizeOf(this) + mSource.size() + 1 + + mCompileResults->SizeOfIncludingThis(mallocSizeOf) + + mCompilationLog.size() + 1; +} + +} // namespace mozilla |