Qt
Internal/Contributor docs for the Qt SDK. Note: These are NOT official API docs; those are found at https://doc.qt.io/
Loading...
Searching...
No Matches
qssgrenderreflectionmap.cpp
Go to the documentation of this file.
1// Copyright (C) 2021 The Qt Company Ltd.
2// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
3// Qt-Security score:significant reason:default
4
5
6#include <QtQuick3DRuntimeRender/private/qssgrenderreflectionprobe_p.h>
7#include <QtQuick3DRuntimeRender/private/qssgrenderlayer_p.h>
8#include <QtQuick3DRuntimeRender/private/qssglayerrenderdata_p.h>
9#include "qssgrendercontextcore.h"
10
11QT_BEGIN_NAMESPACE
12
13const int prefilterSampleCount = 16;
14
15QSSGRenderReflectionMap::QSSGRenderReflectionMap(const QSSGRenderContextInterface &inContext)
16 : m_context(inContext)
17{
18}
19
20QSSGRenderReflectionMap::~QSSGRenderReflectionMap()
21{
22 releaseCachedResources();
23}
24
25void QSSGRenderReflectionMap::releaseCachedResources()
26{
27 for (QSSGReflectionMapEntry &entry : m_reflectionMapList)
28 entry.destroyRhiResources();
29
30 m_reflectionMapList.clear();
31}
32
33static QRhiTexture *allocateRhiReflectionTexture(QRhi *rhi,
34 QRhiTexture::Format format,
35 const QSize &size,
36 QRhiTexture::Flags flags = {})
37{
38 auto texture = rhi->newTexture(format, size, 1, flags);
39 if (!texture->create())
40 qWarning("Failed to create reflection map texture of size %dx%d", size.width(), size.height());
41 return texture;
42}
43
44static QRhiRenderBuffer *allocateRhiReflectionRenderBuffer(QRhi *rhi,
45 QRhiRenderBuffer::Type type,
46 const QSize &size)
47{
48 auto renderBuffer = rhi->newRenderBuffer(type, size, 1);
49 if (!renderBuffer->create())
50 qWarning("Failed to build depth-stencil buffer of size %dx%d", size.width(), size.height());
51 return renderBuffer;
52}
53
54
55void QSSGRenderReflectionMap::addReflectionMapEntry(qint32 probeIdx, const QSSGRenderReflectionProbe &probe)
56{
57 QRhi *rhi = m_context.rhiContext()->rhi();
58 // Bail out if there is no QRhi, since we can't add entries without it
59 if (!rhi)
60 return;
61
62 QRhiTexture::Format rhiFormat = QRhiTexture::RGBA16F;
63
64 const QByteArray rtName = probe.debugObjectName.toLatin1();
65
66 const int mapRes = 1 << probe.reflectionMapRes;
67 QSize pixelSize(mapRes, mapRes);
68 QSSGReflectionMapEntry *pEntry = reflectionMapEntry(probeIdx);
69
70 if (!pEntry) {
71 QRhiRenderBuffer *depthStencil = allocateRhiReflectionRenderBuffer(rhi, QRhiRenderBuffer::DepthStencil, pixelSize);
72 QRhiTexture *map = allocateRhiReflectionTexture(rhi, rhiFormat, pixelSize, QRhiTexture::RenderTarget | QRhiTexture::CubeMap
73 | QRhiTexture::MipMapped | QRhiTexture::UsedWithGenerateMips);
74 QRhiTexture *prefiltered = allocateRhiReflectionTexture(rhi, rhiFormat, pixelSize, QRhiTexture::RenderTarget | QRhiTexture::CubeMap
75 | QRhiTexture::MipMapped | QRhiTexture::UsedWithGenerateMips);
76 m_reflectionMapList.push_back(QSSGReflectionMapEntry::withRhiCubeMap(probeIdx, map, prefiltered, depthStencil));
77
78 pEntry = &m_reflectionMapList.back();
79 }
80
81 if (pEntry) {
82 pEntry->m_needsRender = true;
83
84 if (probe.hasScheduledUpdate)
85 pEntry->m_rendered = false;
86
87 if (!pEntry->m_rhiDepthStencil || mapRes != pEntry->m_rhiCube->pixelSize().width()) {
88 pEntry->destroyRhiResources();
89 pEntry->m_rhiDepthStencil = allocateRhiReflectionRenderBuffer(rhi, QRhiRenderBuffer::DepthStencil, pixelSize);
90 pEntry->m_rhiCube = allocateRhiReflectionTexture(rhi, rhiFormat, pixelSize, QRhiTexture::RenderTarget | QRhiTexture::CubeMap
91 | QRhiTexture::MipMapped | QRhiTexture::UsedWithGenerateMips);
92 pEntry->m_rhiPrefilteredCube = allocateRhiReflectionTexture(rhi, rhiFormat, pixelSize, QRhiTexture::RenderTarget | QRhiTexture::CubeMap
93 | QRhiTexture::MipMapped | QRhiTexture::UsedWithGenerateMips);
94 }
95
96 // Additional graphics resources: samplers, render targets.
97 if (pEntry->m_rhiRenderTargets.isEmpty()) {
98 pEntry->m_rhiRenderTargets.resize(6);
99 for (int i = 0; i < 6; ++i)
100 pEntry->m_rhiRenderTargets[i] = nullptr;
101 }
102 Q_ASSERT(pEntry->m_rhiRenderTargets.size() == 6);
103
104 if (pEntry->m_skyBoxSrbs.isEmpty()) {
105 pEntry->m_skyBoxSrbs.resize(6);
106 for (int i = 0; i < 6; ++i)
107 pEntry->m_skyBoxSrbs[i] = nullptr;
108 }
109
110
111 for (const auto face : QSSGRenderTextureCubeFaces) {
112 QRhiTextureRenderTarget *&rt(pEntry->m_rhiRenderTargets[quint8(face)]);
113 if (!rt) {
114 QRhiColorAttachment att(pEntry->m_rhiCube);
115 att.setLayer(quint8(face)); // 6 render targets, each referencing one face of the cubemap
116 QRhiTextureRenderTargetDescription rtDesc;
117 rtDesc.setColorAttachments({ att });
118 rtDesc.setDepthStencilBuffer(pEntry->m_rhiDepthStencil);
119 rt = rhi->newTextureRenderTarget(rtDesc);
120 rt->setDescription(rtDesc);
121 if (!pEntry->m_rhiRenderPassDesc)
122 pEntry->m_rhiRenderPassDesc = rt->newCompatibleRenderPassDescriptor();
123 rt->setRenderPassDescriptor(pEntry->m_rhiRenderPassDesc);
124 if (!rt->create())
125 qWarning("Failed to build reflection map render target");
126 }
127 rt->setName(rtName + QByteArrayLiteral(" reflection cube face: ") + QSSGBaseTypeHelpers::displayName(face));
128 }
129
130 if (!pEntry->m_prefilterPipeline) {
131 const QSize mapSize = pEntry->m_rhiCube->pixelSize();
132
133 int mipmapCount = rhi->mipLevelsForSize(mapSize);
134 mipmapCount = qMin(mipmapCount, 6); // don't create more than 6 mip levels
135
136 // Create a renderbuffer for each mip level
137 for (int mipLevel = 0; mipLevel < mipmapCount; ++mipLevel) {
138 const QSize levelSize = QSize(mapSize.width() * std::pow(0.5, mipLevel),
139 mapSize.height() * std::pow(0.5, mipLevel));
140 pEntry->m_prefilterMipLevelSizes.insert(mipLevel, levelSize);
141 // Setup Render targets (6 * mipmapCount)
142 QVarLengthArray<QRhiTextureRenderTarget *, 6> renderTargets;
143 for (const auto face : QSSGRenderTextureCubeFaces) {
144 QRhiColorAttachment att(pEntry->m_rhiPrefilteredCube);
145 att.setLayer(quint8(face));
146 att.setLevel(mipLevel);
147 QRhiTextureRenderTargetDescription rtDesc;
148 rtDesc.setColorAttachments({att});
149 auto renderTarget = rhi->newTextureRenderTarget(rtDesc);
150 renderTarget->setName(rtName + QByteArrayLiteral(" reflection prefilter mip/face ")
151 + QByteArray::number(mipLevel) + QByteArrayLiteral("/") + QSSGBaseTypeHelpers::displayName(face));
152 renderTarget->setDescription(rtDesc);
153 if (!pEntry->m_rhiPrefilterRenderPassDesc)
154 pEntry->m_rhiPrefilterRenderPassDesc = renderTarget->newCompatibleRenderPassDescriptor();
155 renderTarget->setRenderPassDescriptor(pEntry->m_rhiPrefilterRenderPassDesc);
156 if (!renderTarget->create())
157 qWarning("Failed to build prefilter cube map render target");
158 renderTargets << renderTarget;
159 }
160 pEntry->m_rhiPrefilterRenderTargetsMap.insert(mipLevel, renderTargets);
161 }
162
163 const auto &prefilterShaderStages = m_context.shaderCache()->getBuiltInRhiShaders().getRhiReflectionprobePreFilterShader();
164
165 const QSSGRhiSamplerDescription samplerMipMapDesc {
166 QRhiSampler::Linear,
167 QRhiSampler::Linear,
168 QRhiSampler::Linear,
169 QRhiSampler::ClampToEdge,
170 QRhiSampler::ClampToEdge,
171 QRhiSampler::Repeat
172 };
173
174 QRhiSampler *cubeSampler = m_context.rhiContext()->sampler(samplerMipMapDesc);
175
176 QRhiVertexInputLayout inputLayout;
177 inputLayout.setBindings({
178 { 3 * sizeof(float) }
179 });
180 inputLayout.setAttributes({
181 { 0, 0, QRhiVertexInputAttribute::Float3, 0 }
182 });
183
184 int ubufElementSize = rhi->ubufAligned(128);
185 pEntry->m_prefilterVertBuffer = rhi->newBuffer(QRhiBuffer::Dynamic, QRhiBuffer::UniformBuffer, ubufElementSize * 6);
186 pEntry->m_prefilterVertBuffer->create();
187
188 const int uBufSamplesSize = 16 * prefilterSampleCount + 8;
189 int uBufSamplesElementSize = rhi->ubufAligned(uBufSamplesSize);
190 pEntry->m_prefilterFragBuffer = rhi->newBuffer(QRhiBuffer::Dynamic, QRhiBuffer::UniformBuffer, uBufSamplesElementSize * mipmapCount);
191 pEntry->m_prefilterFragBuffer->create();
192
193 pEntry->m_prefilterPipeline = rhi->newGraphicsPipeline();
194 pEntry->m_prefilterPipeline->setCullMode(QRhiGraphicsPipeline::Front);
195 pEntry->m_prefilterPipeline->setFrontFace(QRhiGraphicsPipeline::CCW);
196 pEntry->m_prefilterPipeline->setDepthOp(QRhiGraphicsPipeline::LessOrEqual);
197 pEntry->m_prefilterPipeline->setShaderStages({
198 *prefilterShaderStages->vertexStage(),
199 *prefilterShaderStages->fragmentStage()
200 });
201
202 pEntry->m_prefilterSrb = rhi->newShaderResourceBindings();
203 pEntry->m_prefilterSrb->setBindings({
204 QRhiShaderResourceBinding::uniformBufferWithDynamicOffset(0, QRhiShaderResourceBinding::VertexStage, pEntry->m_prefilterVertBuffer, 128),
205 QRhiShaderResourceBinding::uniformBufferWithDynamicOffset(2, QRhiShaderResourceBinding::FragmentStage, pEntry->m_prefilterFragBuffer, uBufSamplesSize),
206 QRhiShaderResourceBinding::sampledTexture(1, QRhiShaderResourceBinding::FragmentStage, pEntry->m_rhiCube, cubeSampler)
207 });
208 pEntry->m_prefilterSrb->create();
209
210 pEntry->m_prefilterPipeline->setVertexInputLayout(inputLayout);
211 pEntry->m_prefilterPipeline->setShaderResourceBindings(pEntry->m_prefilterSrb);
212 pEntry->m_prefilterPipeline->setRenderPassDescriptor(pEntry->m_rhiPrefilterRenderPassDesc);
213 if (!pEntry->m_prefilterPipeline->create())
214 qWarning("failed to create pre-filter reflection map pipeline state");
215
216 const auto &irradianceShaderStages = m_context.shaderCache()->getBuiltInRhiShaders().getRhienvironmentmapPreFilterShader(false /* isRGBE */);
217
218 pEntry->m_irradiancePipeline = rhi->newGraphicsPipeline();
219 pEntry->m_irradiancePipeline->setCullMode(QRhiGraphicsPipeline::Front);
220 pEntry->m_irradiancePipeline->setFrontFace(QRhiGraphicsPipeline::CCW);
221 pEntry->m_irradiancePipeline->setDepthOp(QRhiGraphicsPipeline::LessOrEqual);
222 pEntry->m_irradiancePipeline->setShaderStages({
223 *irradianceShaderStages->vertexStage(),
224 *irradianceShaderStages->fragmentStage()
225 });
226
227 int ubufIrradianceSize = rhi->ubufAligned(20);
228 pEntry->m_irradianceFragBuffer = rhi->newBuffer(QRhiBuffer::Dynamic, QRhiBuffer::UniformBuffer, ubufIrradianceSize);
229 pEntry->m_irradianceFragBuffer->create();
230
231 pEntry->m_irradianceSrb = rhi->newShaderResourceBindings();
232 pEntry->m_irradianceSrb->setBindings({
233 QRhiShaderResourceBinding::uniformBufferWithDynamicOffset(0, QRhiShaderResourceBinding::VertexStage, pEntry->m_prefilterVertBuffer, 128),
234 QRhiShaderResourceBinding::uniformBufferWithDynamicOffset(2, QRhiShaderResourceBinding::FragmentStage, pEntry->m_irradianceFragBuffer, 20),
235 QRhiShaderResourceBinding::sampledTexture(1, QRhiShaderResourceBinding::FragmentStage, pEntry->m_rhiCube, cubeSampler)
236 });
237 pEntry->m_irradianceSrb->create();
238
239 pEntry->m_irradiancePipeline->setShaderResourceBindings(pEntry->m_irradianceSrb);
240 pEntry->m_irradiancePipeline->setVertexInputLayout(inputLayout);
241 pEntry->m_irradiancePipeline->setRenderPassDescriptor(pEntry->m_rhiPrefilterRenderPassDesc);
242 if (!pEntry->m_irradiancePipeline->create())
243 qWarning("failed to create irradiance reflection map pipeline state");
244 }
245
246 pEntry->m_timeSlicing = probe.timeSlicing;
247 pEntry->m_probeIndex = probeIdx;
248 Q_QUICK3D_PROFILE_ASSIGN_ID(&probe, pEntry);
249 }
250}
251
252void QSSGRenderReflectionMap::addTexturedReflectionMapEntry(qint32 probeIdx, const QSSGRenderReflectionProbe &probe)
253{
254 QSSGReflectionMapEntry *pEntry = reflectionMapEntry(probeIdx);
255 const QSSGRenderImageTexture probeTexture = m_context.bufferManager()->loadRenderImage(probe.texture, QSSGBufferManager::MipModeFollowRenderImage);
256 if (!pEntry) {
257 if (probeTexture.m_texture)
258 m_reflectionMapList.push_back(QSSGReflectionMapEntry::withRhiTexturedCubeMap(probeIdx, probeTexture.m_texture));
259 else
260 addReflectionMapEntry(probeIdx, probe);
261 } else {
262 if (pEntry->m_rhiDepthStencil)
263 pEntry->destroyRhiResources();
264 if (probeTexture.m_texture)
265 pEntry->m_rhiPrefilteredCube = probeTexture.m_texture;
266 }
267}
268
269QSSGReflectionMapEntry *QSSGRenderReflectionMap::reflectionMapEntry(int probeIdx)
270{
271 Q_ASSERT(probeIdx >= 0);
272
273 for (int i = 0; i < m_reflectionMapList.size(); i++) {
274 QSSGReflectionMapEntry *pEntry = &m_reflectionMapList[i];
275 if (pEntry->m_probeIndex == quint32(probeIdx))
276 return pEntry;
277 }
278
279 return nullptr;
280}
281
282QSSGReflectionMapEntry::QSSGReflectionMapEntry()
283 : m_probeIndex(std::numeric_limits<quint32>::max())
284{
285}
286
287// Vertex data for rendering reflection cube map
288static const float cube[] = {
289 -1.0f,-1.0f,-1.0f, // -X side
290 -1.0f,-1.0f, 1.0f,
291 -1.0f, 1.0f, 1.0f,
292 -1.0f, 1.0f, 1.0f,
293 -1.0f, 1.0f,-1.0f,
294 -1.0f,-1.0f,-1.0f,
295
296 -1.0f,-1.0f,-1.0f, // -Z side
297 1.0f, 1.0f,-1.0f,
298 1.0f,-1.0f,-1.0f,
299 -1.0f,-1.0f,-1.0f,
300 -1.0f, 1.0f,-1.0f,
301 1.0f, 1.0f,-1.0f,
302
303 -1.0f,-1.0f,-1.0f, // -Y side
304 1.0f,-1.0f,-1.0f,
305 1.0f,-1.0f, 1.0f,
306 -1.0f,-1.0f,-1.0f,
307 1.0f,-1.0f, 1.0f,
308 -1.0f,-1.0f, 1.0f,
309
310 -1.0f, 1.0f,-1.0f, // +Y side
311 -1.0f, 1.0f, 1.0f,
312 1.0f, 1.0f, 1.0f,
313 -1.0f, 1.0f,-1.0f,
314 1.0f, 1.0f, 1.0f,
315 1.0f, 1.0f,-1.0f,
316
317 1.0f, 1.0f,-1.0f, // +X side
318 1.0f, 1.0f, 1.0f,
319 1.0f,-1.0f, 1.0f,
320 1.0f,-1.0f, 1.0f,
321 1.0f,-1.0f,-1.0f,
322 1.0f, 1.0f,-1.0f,
323
324 -1.0f, 1.0f, 1.0f, // +Z side
325 -1.0f,-1.0f, 1.0f,
326 1.0f, 1.0f, 1.0f,
327 -1.0f,-1.0f, 1.0f,
328 1.0f,-1.0f, 1.0f,
329 1.0f, 1.0f, 1.0f,
330
331 0.0f, 1.0f, // -X side
332 1.0f, 1.0f,
333 1.0f, 0.0f,
334 1.0f, 0.0f,
335 0.0f, 0.0f,
336 0.0f, 1.0f,
337
338 1.0f, 1.0f, // -Z side
339 0.0f, 0.0f,
340 0.0f, 1.0f,
341 1.0f, 1.0f,
342 1.0f, 0.0f,
343 0.0f, 0.0f,
344
345 1.0f, 0.0f, // -Y side
346 1.0f, 1.0f,
347 0.0f, 1.0f,
348 1.0f, 0.0f,
349 0.0f, 1.0f,
350 0.0f, 0.0f,
351
352 1.0f, 0.0f, // +Y side
353 0.0f, 0.0f,
354 0.0f, 1.0f,
355 1.0f, 0.0f,
356 0.0f, 1.0f,
357 1.0f, 1.0f,
358
359 1.0f, 0.0f, // +X side
360 0.0f, 0.0f,
361 0.0f, 1.0f,
362 0.0f, 1.0f,
363 1.0f, 1.0f,
364 1.0f, 0.0f,
365
366 0.0f, 0.0f, // +Z side
367 0.0f, 1.0f,
368 1.0f, 0.0f,
369 0.0f, 1.0f,
370 1.0f, 1.0f,
371 1.0f, 0.0f,
372};
373
374float radicalInverseVdC(uint bits)
375{
376 bits = (bits << 16u) | (bits >> 16u);
377 bits = ((bits & 0x55555555u) << 1u) | ((bits & 0xAAAAAAAAu) >> 1u);
378 bits = ((bits & 0x33333333u) << 2u) | ((bits & 0xCCCCCCCCu) >> 2u);
379 bits = ((bits & 0x0F0F0F0Fu) << 4u) | ((bits & 0xF0F0F0F0u) >> 4u);
380 bits = ((bits & 0x00FF00FFu) << 8u) | ((bits & 0xFF00FF00u) >> 8u);
381 return float(bits) * 2.3283064365386963e-10; // / 0x100000000
382}
383
384QVector2D hammersley(uint i, uint N)
385{
386 return QVector2D(float(i) / float(N), radicalInverseVdC(i));
387}
388
389QVector3D importanceSampleGGX(QVector2D xi, float roughness)
390{
391 float a = roughness*roughness;
392
393 float phi = 2.0f * M_PI * xi.x();
394 float cosTheta = sqrt((1.0f - xi.y()) / (1.0f + (a*a - 1.0f) * xi.y()));
395 float sinTheta = sqrt(1.0f - cosTheta * cosTheta);
396
397 // from spherical coordinates to cartesian coordinates
398 return QVector3D(cos(phi) * sinTheta, sin(phi) * sinTheta, cosTheta);
399}
400
401float distributionGGX(float nDotH, float roughness)
402{
403 float a = roughness * roughness;
404 float a2 = a * a;
405 float nDotH2 = nDotH * nDotH;
406
407 float nom = a2;
408 float denom = nDotH2 * (a2 - 1.0f) + 1.0f;
409 denom = M_PI * denom * denom;
410
411 return nom / denom;
412}
413
414void fillPrefilterValues(float roughness, float resolution,
415 QVarLengthArray<QVector4D, prefilterSampleCount> &sampleDirections,
416 float &invTotalWeight, uint &sampleCount)
417{
418 for (int i = 0; i < prefilterSampleCount * 8; ++i) {
419 const QVector2D xi = hammersley(i, prefilterSampleCount);
420 const QVector3D half = importanceSampleGGX(xi, roughness);
421 QVector3D light = 2.0f * half.z() * half - QVector3D(0, 0, 1);
422 light.normalize();
423 const float D = distributionGGX(half.z(), roughness);
424 const float pdf = D * half.z() / (4.0f * half.z()) + 0.0001f;
425 const float saTexel = 4.0f * M_PI / (6.0f * resolution * resolution);
426 const float saSample = 1.0f / (float(prefilterSampleCount) * pdf + 0.0001f);
427 float mipLevel = roughness == 0.0f ? 0.0f : 0.5f * log2(saSample / saTexel);
428 if (light.z() > 0) {
429 sampleDirections.append(QVector4D(light, mipLevel));
430 invTotalWeight += light.z();
431 sampleCount++;
432 if (sampleCount >= prefilterSampleCount)
433 break;
434 }
435 }
436 invTotalWeight = 1.0f / invTotalWeight;
437}
438
439void QSSGReflectionMapEntry::renderMips(QSSGRhiContext *rhiCtx)
440{
441 auto *rhi = rhiCtx->rhi();
442 auto *cb = rhiCtx->commandBuffer();
443
444 auto *rub = rhi->nextResourceUpdateBatch();
445 rub->generateMips(m_rhiCube);
446 QRhiBuffer *vertexBuffer = rhi->newBuffer(QRhiBuffer::Immutable, QRhiBuffer::VertexBuffer, sizeof(cube));
447 vertexBuffer->create();
448 vertexBuffer->deleteLater();
449 rub->uploadStaticBuffer(vertexBuffer, cube);
450 cb->resourceUpdate(rub);
451
452 const QRhiCommandBuffer::VertexInput vbufBinding(vertexBuffer, 0);
453
454 int ubufElementSize = rhi->ubufAligned(128);
455
456 const int uBufSamplesSize = 16 * prefilterSampleCount + 8;
457 int uBufSamplesElementSize = rhi->ubufAligned(uBufSamplesSize);
458 int uBufIrradianceElementSize = rhi->ubufAligned(20);
459
460 // Uniform Data
461 QMatrix4x4 mvp = rhi->clipSpaceCorrMatrix();
462 mvp.perspective(90.0f, 1.0f, 0.1f, 10.0f);
463
464 auto lookAt = [](const QVector3D &eye, const QVector3D &center, const QVector3D &up) {
465 QMatrix4x4 viewMatrix;
466 viewMatrix.lookAt(eye, center, up);
467 return viewMatrix;
468 };
469 QVarLengthArray<QMatrix4x4, 6> views;
470 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(1.0, 0.0, 0.0), QVector3D(0.0f, -1.0f, 0.0f)));
471 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(-1.0, 0.0, 0.0), QVector3D(0.0f, -1.0f, 0.0f)));
472 if (rhi->isYUpInFramebuffer()) {
473 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(0.0, 1.0, 0.0), QVector3D(0.0f, 0.0f, 1.0f)));
474 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(0.0, -1.0, 0.0), QVector3D(0.0f, 0.0f, -1.0f)));
475 } else {
476 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(0.0, -1.0, 0.0), QVector3D(0.0f, 0.0f, -1.0f)));
477 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(0.0, 1.0, 0.0), QVector3D(0.0f, 0.0f, 1.0f)));
478 }
479 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(0.0, 0.0, 1.0), QVector3D(0.0f, -1.0f, 0.0f)));
480 views.append(lookAt(QVector3D(0.0f, 0.0f, 0.0f), QVector3D(0.0, 0.0, -1.0), QVector3D(0.0f, -1.0f, 0.0f)));
481
482 rub = rhi->nextResourceUpdateBatch();
483 for (const auto face : QSSGRenderTextureCubeFaces) {
484 rub->updateDynamicBuffer(m_prefilterVertBuffer, quint8(face) * ubufElementSize, 64, mvp.constData());
485 rub->updateDynamicBuffer(m_prefilterVertBuffer, quint8(face) * ubufElementSize + 64, 64, views[quint8(face)].constData());
486 }
487
488 const QSize mapSize = m_rhiCube->pixelSize();
489
490 int mipmapCount = rhi->mipLevelsForSize(mapSize);
491 mipmapCount = qMin(mipmapCount, 6);
492
493 const float resolution = mapSize.width();
494 QVarLengthArray<QVector4D, prefilterSampleCount> sampleDirections;
495
496 // set the samples uniform buffer data
497 for (int mipLevel = 0; mipLevel < mipmapCount - 1; ++mipLevel) {
498 Q_ASSERT(mipmapCount - 2);
499 const float roughness = float(mipLevel) / float(mipmapCount - 2);
500 float invTotalWeight = 0.0f;
501 uint sampleCount = 0;
502
503 sampleDirections.clear();
504 fillPrefilterValues(roughness, resolution, sampleDirections, invTotalWeight, sampleCount);
505
506 rub->updateDynamicBuffer(m_prefilterFragBuffer, mipLevel * uBufSamplesElementSize, 16 * prefilterSampleCount, sampleDirections.constData());
507 rub->updateDynamicBuffer(m_prefilterFragBuffer, mipLevel * uBufSamplesElementSize + 16 * prefilterSampleCount, 4, &invTotalWeight);
508 rub->updateDynamicBuffer(m_prefilterFragBuffer, mipLevel * uBufSamplesElementSize + 16 * prefilterSampleCount + 4, 4, &sampleCount);
509 }
510 {
511 const float roughness = 0.0f; // doesn't matter for irradiance
512 const float lodBias = 0.0f;
513 const int distribution = 0;
514 const int sampleCount = resolution / 4;
515
516 rub->updateDynamicBuffer(m_irradianceFragBuffer, 0, 4, &roughness);
517 rub->updateDynamicBuffer(m_irradianceFragBuffer, 4, 4, &resolution);
518 rub->updateDynamicBuffer(m_irradianceFragBuffer, 4 + 4, 4, &lodBias);
519 rub->updateDynamicBuffer(m_irradianceFragBuffer, 4 + 4 + 4, 4, &sampleCount);
520 rub->updateDynamicBuffer(m_irradianceFragBuffer, 4 + 4 + 4 + 4, 4, &distribution);
521 }
522
523 cb->resourceUpdate(rub);
524
525 // Render
526 for (int mipLevel = 0; mipLevel < mipmapCount; ++mipLevel) {
527 if (mipLevel > 0 && m_timeSlicing == QSSGRenderReflectionProbe::ReflectionTimeSlicing::AllFacesAtOnce)
528 mipLevel = m_timeSliceFrame;
529
530 for (auto face : QSSGRenderTextureCubeFaces) {
531 if (m_timeSlicing == QSSGRenderReflectionProbe::ReflectionTimeSlicing::IndividualFaces)
532 face = m_timeSliceFace;
533
534 cb->beginPass(m_rhiPrefilterRenderTargetsMap[mipLevel][quint8(face)], QColor(0, 0, 0, 1), { 1.0f, 0 }, nullptr, rhiCtx->commonPassFlags());
535 QSSGRHICTX_STAT(rhiCtx, beginRenderPass(m_rhiPrefilterRenderTargetsMap[mipLevel][quint8(face)]));
536 Q_QUICK3D_PROFILE_START(QQuick3DProfiler::Quick3DRenderPass);
537 if (mipLevel < mipmapCount - 1) {
538 // Specular pre-filtered Cube Map levels
539 cb->setGraphicsPipeline(m_prefilterPipeline);
540 cb->setVertexInput(0, 1, &vbufBinding);
541 cb->setViewport(QRhiViewport(0, 0, m_prefilterMipLevelSizes[mipLevel].width(), m_prefilterMipLevelSizes[mipLevel].height()));
542 QVector<QPair<int, quint32>> dynamicOffsets = {
543 { 0, quint32(ubufElementSize * quint8(face)) },
544 { 2, quint32(uBufSamplesElementSize * mipLevel) }
545 };
546 cb->setShaderResources(m_prefilterSrb, 2, dynamicOffsets.constData());
547 } else {
548 // Diffuse Irradiance
549 cb->setGraphicsPipeline(m_irradiancePipeline);
550 cb->setVertexInput(0, 1, &vbufBinding);
551 cb->setViewport(QRhiViewport(0, 0, m_prefilterMipLevelSizes[mipLevel].width(), m_prefilterMipLevelSizes[mipLevel].height()));
552 QVector<QPair<int, quint32>> dynamicOffsets = {
553 { 0, quint32(ubufElementSize * quint8(face)) },
554 { 2, quint32(uBufIrradianceElementSize) }
555 };
556 cb->setShaderResources(m_irradianceSrb, 1, dynamicOffsets.constData());
557 }
558 Q_QUICK3D_PROFILE_START(QQuick3DProfiler::Quick3DRenderCall);
559 cb->draw(36);
560 QSSGRHICTX_STAT(rhiCtx, draw(36, 1));
561 Q_QUICK3D_PROFILE_END_WITH_ID(QQuick3DProfiler::Quick3DRenderCall, 36llu | (1llu << 32), profilingId);
562 cb->endPass();
563 QSSGRHICTX_STAT(rhiCtx, endRenderPass());
564 Q_QUICK3D_PROFILE_END_WITH_STRING(QQuick3DProfiler::Quick3DRenderPass, 0, QSSG_RENDERPASS_NAME("reflection_map", mipLevel, face));
565
566 if (m_timeSlicing == QSSGRenderReflectionProbe::ReflectionTimeSlicing::IndividualFaces)
567 break;
568 }
569
570 if (mipLevel > 0 && m_timeSlicing == QSSGRenderReflectionProbe::ReflectionTimeSlicing::AllFacesAtOnce) {
571 m_timeSliceFrame++;
572 if (m_timeSliceFrame >= mipmapCount)
573 m_timeSliceFrame = 1;
574 break;
575 }
576 }
577 cb->debugMarkEnd();
578}
579
580QSSGReflectionMapEntry QSSGReflectionMapEntry::withRhiTexturedCubeMap(quint32 probeIdx, QRhiTexture *prefiltered)
581{
582 QSSGReflectionMapEntry e;
583 e.m_probeIndex = probeIdx;
584 e.m_rhiPrefilteredCube = prefiltered;
585 return e;
586}
587
588QSSGReflectionMapEntry QSSGReflectionMapEntry::withRhiCubeMap(quint32 probeIdx,
589 QRhiTexture *cube,
590 QRhiTexture *prefiltered,
591 QRhiRenderBuffer *depthStencil)
592{
593 QSSGReflectionMapEntry e;
594 e.m_probeIndex = probeIdx;
595 e.m_rhiCube = cube;
596 e.m_rhiPrefilteredCube = prefiltered;
597 e.m_rhiDepthStencil = depthStencil;
598 return e;
599}
600
601void QSSGReflectionMapEntry::destroyRhiResources()
602{
603 delete m_rhiCube;
604 m_rhiCube = nullptr;
605 // Without depth stencil the prefiltered cubemap is assumed to be not owned here and shouldn't be deleted
606 if (m_rhiDepthStencil)
607 delete m_rhiPrefilteredCube;
608 m_rhiPrefilteredCube = nullptr;
609 delete m_rhiDepthStencil;
610 m_rhiDepthStencil = nullptr;
611
612 qDeleteAll(m_rhiRenderTargets);
613 m_rhiRenderTargets.clear();
614 delete m_rhiRenderPassDesc;
615 m_rhiRenderPassDesc = nullptr;
616
617 delete m_prefilterPipeline;
618 m_prefilterPipeline = nullptr;
619 delete m_irradiancePipeline;
620 m_irradiancePipeline = nullptr;
621 delete m_prefilterSrb;
622 m_prefilterSrb = nullptr;
623 delete m_irradianceSrb;
624 m_irradianceSrb = nullptr;
625 delete m_prefilterVertBuffer;
626 m_prefilterVertBuffer = nullptr;
627 delete m_prefilterFragBuffer;
628 m_prefilterFragBuffer = nullptr;
629 delete m_irradianceFragBuffer;
630 m_irradianceFragBuffer = nullptr;
631 delete m_rhiPrefilterRenderPassDesc;
632 m_rhiPrefilterRenderPassDesc = nullptr;
633 for (const auto &e : std::as_const(m_rhiPrefilterRenderTargetsMap))
634 qDeleteAll(e);
635 m_rhiPrefilterRenderTargetsMap.clear();
636 m_prefilterMipLevelSizes.clear();
637}
638
639QT_END_NAMESPACE
QT_BEGIN_NAMESPACE
Combined button and popup list for selecting options.
Definition qsequentialanimationgroup.cpp:47
distributionGGX
float distributionGGX(float nDotH, float roughness)
Definition qssgrenderreflectionmap.cpp:401
allocateRhiReflectionRenderBuffer
static QRhiRenderBuffer * allocateRhiReflectionRenderBuffer(QRhi *rhi, QRhiRenderBuffer::Type type, const QSize &size)
Definition qssgrenderreflectionmap.cpp:44
prefilterSampleCount
QT_BEGIN_NAMESPACE const int prefilterSampleCount
Definition qssgrenderreflectionmap.cpp:13
allocateRhiReflectionTexture
static QRhiTexture * allocateRhiReflectionTexture(QRhi *rhi, QRhiTexture::Format format, const QSize &size, QRhiTexture::Flags flags={})
Definition qssgrenderreflectionmap.cpp:33
fillPrefilterValues
void fillPrefilterValues(float roughness, float resolution, QVarLengthArray< QVector4D, prefilterSampleCount > &sampleDirections, float &invTotalWeight, uint &sampleCount)
Definition qssgrenderreflectionmap.cpp:414
radicalInverseVdC
float radicalInverseVdC(uint bits)
Definition qssgrenderreflectionmap.cpp:374
cube
static const float cube[]
Definition qssgrenderreflectionmap.cpp:288
importanceSampleGGX
QVector3D importanceSampleGGX(QVector2D xi, float roughness)
Definition qssgrenderreflectionmap.cpp:389
hammersley
QVector2D hammersley(uint i, uint N)
Definition qssgrenderreflectionmap.cpp:384