xref: /aosp_15_r20/external/skia/src/gpu/ganesh/glsl/GrGLSLProgramDataManager.cpp (revision c8dee2aa9b3f27cf6c858bd81872bdeb2c07ed17)
1 /*
2  * Copyright 2016 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 #include "src/gpu/ganesh/glsl/GrGLSLProgramDataManager.h"
9 
10 #include "include/core/SkMatrix.h"
11 #include "include/private/base/SkTemplates.h"
12 #include "src/core/SkMatrixPriv.h"
13 
14 #include <cstddef>
15 
setSkMatrix(UniformHandle u,const SkMatrix & matrix) const16 void GrGLSLProgramDataManager::setSkMatrix(UniformHandle u, const SkMatrix& matrix) const {
17     float mt[] = {
18         matrix.get(SkMatrix::kMScaleX),
19         matrix.get(SkMatrix::kMSkewY),
20         matrix.get(SkMatrix::kMPersp0),
21         matrix.get(SkMatrix::kMSkewX),
22         matrix.get(SkMatrix::kMScaleY),
23         matrix.get(SkMatrix::kMPersp1),
24         matrix.get(SkMatrix::kMTransX),
25         matrix.get(SkMatrix::kMTransY),
26         matrix.get(SkMatrix::kMPersp2),
27     };
28     this->setMatrix3f(u, mt);
29 }
30 
setSkM44(UniformHandle u,const SkM44 & matrix) const31 void GrGLSLProgramDataManager::setSkM44(UniformHandle u, const SkM44& matrix) const {
32     this->setMatrix4f(u, SkMatrixPriv::M44ColMajor(matrix));
33 }
34 
setRuntimeEffectUniforms(SkSpan<const SkRuntimeEffect::Uniform> uniforms,SkSpan<const UniformHandle> handles,SkSpan<const Specialized> specialized,const void * src) const35 void GrGLSLProgramDataManager::setRuntimeEffectUniforms(
36         SkSpan<const SkRuntimeEffect::Uniform> uniforms,
37         SkSpan<const UniformHandle> handles,
38         SkSpan<const Specialized> specialized,
39         const void* src) const {
40     SkASSERT(uniforms.empty() || src);
41     SkASSERT(specialized.empty() || specialized.size() == uniforms.size());
42     SkASSERT(!specialized.empty() || handles.size() == uniforms.size());
43 
44     using Type = SkRuntimeEffect::Uniform::Type;
45     size_t handleIdx = 0;
46     for (size_t uniformIdx = 0; uniformIdx < uniforms.size(); ++uniformIdx) {
47         const auto& u = uniforms[uniformIdx];
48         auto floatData = [=] { return SkTAddOffset<const float>(src, u.offset); };
49         auto intData   = [=] { return SkTAddOffset<const int  >(src, u.offset); };
50         if (!specialized.empty() && specialized[uniformIdx] == Specialized::kYes) {
51             continue;
52         }
53         const auto h = handles[handleIdx++];
54         switch (u.type) {
55             case Type::kFloat:  this->set1fv(h, u.count, floatData()); break;
56             case Type::kFloat2: this->set2fv(h, u.count, floatData()); break;
57             case Type::kFloat3: this->set3fv(h, u.count, floatData()); break;
58             case Type::kFloat4: this->set4fv(h, u.count, floatData()); break;
59 
60             case Type::kFloat2x2: this->setMatrix2fv(h, u.count, floatData()); break;
61             case Type::kFloat3x3: this->setMatrix3fv(h, u.count, floatData()); break;
62             case Type::kFloat4x4: this->setMatrix4fv(h, u.count, floatData()); break;
63 
64             case Type::kInt:  this->set1iv(h, u.count, intData()); break;
65             case Type::kInt2: this->set2iv(h, u.count, intData()); break;
66             case Type::kInt3: this->set3iv(h, u.count, intData()); break;
67             case Type::kInt4: this->set4iv(h, u.count, intData()); break;
68 
69             default:
70                 SkDEBUGFAIL("Unsupported uniform type");
71                 break;
72         }
73     }
74     SkASSERT(handleIdx == handles.size());
75 }
76