1 | // Copyright (C) 2017 Klaralvdalens Datakonsult AB (KDAB). |
2 | // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only |
3 | |
4 | #ifndef QT3DRENDER_RENDER_OPENGL_GLSHADER_P_H |
5 | #define QT3DRENDER_RENDER_OPENGL_GLSHADER_P_H |
6 | |
7 | // |
8 | // W A R N I N G |
9 | // ------------- |
10 | // |
11 | // This file is not part of the Qt API. It exists for the convenience |
12 | // of other Qt classes. This header file may change from version to |
13 | // version without notice, or even be removed. |
14 | // |
15 | // We mean it. |
16 | // |
17 | |
18 | |
19 | #include <shadervariables_p.h> |
20 | #include <shaderparameterpack_p.h> |
21 | #include <Qt3DRender/qshaderprogram.h> |
22 | #include <QMutex> |
23 | |
24 | #ifdef QT_BUILD_INTERNAL |
25 | class tst_BenchShaderParameterPack; |
26 | #endif |
27 | |
28 | QT_BEGIN_NAMESPACE |
29 | |
30 | class QOpenGLShaderProgram; |
31 | |
32 | namespace Qt3DRender { |
33 | |
34 | namespace Render { |
35 | |
36 | namespace OpenGL { |
37 | |
38 | class Q_AUTOTEST_EXPORT GLShader |
39 | { |
40 | public: |
41 | GLShader(); |
42 | ~GLShader(); |
43 | |
44 | void setGraphicsContext(GraphicsContext *context); |
45 | GraphicsContext *graphicsContext(); |
46 | |
47 | bool isLoaded() const { return m_isLoaded; } |
48 | void setLoaded(bool loaded) { m_isLoaded = loaded; } |
49 | |
50 | void prepareUniforms(ShaderParameterPack &pack); |
51 | |
52 | void setFragOutputs(const QHash<QString, int> &fragOutputs); |
53 | const QHash<QString, int> fragOutputs() const; |
54 | |
55 | inline const std::vector<int> &uniformsNamesIds() const { return m_uniformsNamesIds; } |
56 | inline const std::vector<int> &lightUniformsNamesIds() const { return m_lightUniformsNamesIds; } |
57 | inline const std::vector<int> &standardUniformNameIds() const { return m_standardUniformNamesIds; } |
58 | inline const std::vector<int> &uniformBlockNamesIds() const { return m_uniformBlockNamesIds; } |
59 | inline const std::vector<int> &storageBlockNamesIds() const { return m_shaderStorageBlockNamesIds; } |
60 | inline const std::vector<int> &attributeNamesIds() const { return m_attributeNamesIds; } |
61 | |
62 | const std::vector<QString> &uniformsNames() const; |
63 | const std::vector<QString> &attributesNames() const; |
64 | const std::vector<QString> &uniformBlockNames() const; |
65 | const std::vector<QString> &storageBlockNames() const; |
66 | |
67 | inline const std::vector<ShaderUniform> &uniforms() const { return m_uniforms; } |
68 | inline const std::vector<ShaderAttribute> &attributes() const { return m_attributes; } |
69 | inline const std::vector<ShaderUniformBlock> &uniformBlocks() const { return m_uniformBlocks; } |
70 | inline const std::vector<ShaderStorageBlock> &storageBlocks() const { return m_shaderStorageBlocks; } |
71 | |
72 | QHash<QString, ShaderUniform> activeUniformsForUniformBlock(int blockIndex) const; |
73 | |
74 | ShaderUniformBlock uniformBlockForBlockIndex(int blockNameId) const noexcept; |
75 | ShaderUniformBlock uniformBlockForBlockNameId(int blockIndex) const noexcept; |
76 | ShaderUniformBlock uniformBlockForBlockName(const QString &blockName) const noexcept; |
77 | |
78 | ShaderStorageBlock storageBlockForBlockIndex(int blockIndex) const noexcept; |
79 | ShaderStorageBlock storageBlockForBlockNameId(int blockNameId) const noexcept; |
80 | ShaderStorageBlock storageBlockForBlockName(const QString &blockName) const noexcept; |
81 | |
82 | enum ParameterKind { |
83 | Uniform, |
84 | UBO, |
85 | SSBO, |
86 | Struct |
87 | }; |
88 | ParameterKind categorizeVariable(int nameId) const noexcept; |
89 | |
90 | bool hasUniform(int nameId) const noexcept; |
91 | inline bool hasActiveVariables() const noexcept { return m_hasActiveVariables; } |
92 | inline int parameterPackSize() const noexcept { return m_parameterPackSize; } |
93 | |
94 | QOpenGLShaderProgram *shaderProgram() { return &m_shader; } |
95 | |
96 | void setShaderCode(const std::vector<QByteArray> shaderCode) { m_shaderCode = shaderCode; } |
97 | const std::vector<QByteArray> &shaderCode() const; |
98 | |
99 | private: |
100 | bool m_isLoaded; |
101 | QOpenGLShaderProgram m_shader; |
102 | GraphicsContext *m_graphicsContext; |
103 | |
104 | std::vector<QString> m_uniformsNames; |
105 | std::vector<int> m_uniformsNamesIds; |
106 | std::vector<int> m_lightUniformsNamesIds; |
107 | std::vector<int> m_standardUniformNamesIds; |
108 | std::vector<ShaderUniform> m_uniforms; |
109 | |
110 | std::vector<QString> m_attributesNames; |
111 | std::vector<int> m_attributeNamesIds; |
112 | std::vector<ShaderAttribute> m_attributes; |
113 | |
114 | std::vector<QString> m_uniformBlockNames; |
115 | std::vector<int> m_uniformBlockNamesIds; |
116 | std::vector<ShaderUniformBlock> m_uniformBlocks; |
117 | QHash<int, QHash<QString, ShaderUniform> > m_uniformBlockIndexToShaderUniforms; |
118 | |
119 | std::vector<QString> m_shaderStorageBlockNames; |
120 | std::vector<int> m_shaderStorageBlockNamesIds; |
121 | std::vector<ShaderStorageBlock> m_shaderStorageBlocks; |
122 | |
123 | QHash<QString, int> m_fragOutputs; |
124 | std::vector<QByteArray> m_shaderCode; |
125 | |
126 | int m_parameterPackSize; |
127 | int m_hasActiveVariables; |
128 | |
129 | // Private so that only GraphicContext can call it |
130 | void initializeUniforms(const std::vector<ShaderUniform> &uniformsDescription); |
131 | void initializeAttributes(const std::vector<ShaderAttribute> &attributesDescription); |
132 | void initializeUniformBlocks(const std::vector<ShaderUniformBlock> &uniformBlockDescription); |
133 | void initializeShaderStorageBlocks(const std::vector<ShaderStorageBlock> &shaderStorageBlockDescription); |
134 | |
135 | friend class GraphicsContext; |
136 | #ifdef QT_BUILD_INTERNAL |
137 | friend class ::tst_BenchShaderParameterPack; |
138 | #endif |
139 | |
140 | mutable QMutex m_mutex; |
141 | QMetaObject::Connection m_contextConnection; |
142 | }; |
143 | |
144 | } // OpenGL |
145 | |
146 | } // Render |
147 | |
148 | } // Qt3DRender |
149 | |
150 | QT_END_NAMESPACE |
151 | |
152 | #endif // QT3DRENDER_RENDER_OPENGL_GLSHADER_P_H |
153 | |