1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright (C) 2025 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
#include "lightmapimageprovider.h"
#include "lightmapviewerhelpers.h"
#include <QUrlQuery>
#include <QtQuick3DRuntimeRender/private/qssglightmapio_p.h>
LightmapImageProvider::LightmapImageProvider() : QQuickImageProvider(QQuickImageProvider::Image)
{
// Generate error image
constexpr int width = 100;
m_errorImage = QImage(QSize(width, width), QImage::Format::Format_RGB888);
m_errorImage.fill(QColorConstants::White);
for (int i = 0; i < width; i++) {
m_errorImage.setPixelColor(i, i, QColorConstants::Red);
m_errorImage.setPixelColor(width - i - 1, i, QColorConstants::Red);
}
}
enum class LightmapDataType { Unset, F32, U32 };
QImage LightmapImageProvider::requestImage(const QString &id, QSize *size, const QSize & /*requestedSize*/)
{
if (size)
*size = QSize(100, 100);
const QUrl url = QUrl("lightmap://?" + id, QUrl::StrictMode);
const QUrlQuery query(url);
const QString keyWithTag = query.queryItemValue("key");
QString key = keyWithTag;
QSSGLightmapIODataTag tag = QSSGLightmapIODataTag::Unset;
if (int sep = keyWithTag.indexOf('$'); sep >= 0) {
key = keyWithTag.left(sep);
const QString tagString = keyWithTag.mid(sep + 1);
tag = LightmapViewerHelpers::stringToLightmapTag(tagString);
}
const QUrl filePath = query.queryItemValue("file");
const bool useAlpha = query.queryItemValue("alpha") == QStringLiteral("true");
LightmapDataType dataType = LightmapDataType::Unset;
switch (tag) {
case QSSGLightmapIODataTag::Mask:
dataType = LightmapDataType::U32;
break;
case QSSGLightmapIODataTag::Texture_Final:
case QSSGLightmapIODataTag::Texture_Direct:
case QSSGLightmapIODataTag::Texture_Indirect:
dataType = LightmapDataType::F32;
break;
case QSSGLightmapIODataTag::Metadata:
case QSSGLightmapIODataTag::Mesh:
case QSSGLightmapIODataTag::Unset:
default:
break;
}
QSharedPointer<QSSGLightmapLoader> loader = QSSGLightmapLoader::open(filePath.toLocalFile());
if (!loader)
return m_errorImage;
QVariantMap metadata = loader->readMetadata(key);
if (metadata.isEmpty())
return m_errorImage;
bool ok = false;
const int width = metadata[QStringLiteral("width")].toInt(&ok);
if (!ok)
return m_errorImage;
const int height = metadata[QStringLiteral("height")].toInt(&ok);
if (!ok)
return m_errorImage;
QImage::Format format = QImage::Format_Invalid;
QByteArray array;
if (dataType == LightmapDataType::U32) {
array = loader->readU32Image(key, tag);
if (array.size() != qsizetype(sizeof(quint32) * width * height))
return m_errorImage;
format = QImage::Format_RGBA8888;
LightmapViewerHelpers::maskToBBGRColor(array, useAlpha);
} else if (dataType == LightmapDataType::F32) {
array = loader->readF32Image(key, tag);
if (array.size() != qsizetype(4 * sizeof(float) * width * height))
return m_errorImage;
if (!useAlpha) {
std::array<float, 4> *rgbas = reinterpret_cast<std::array<float, 4> *>(array.data());
for (int i = 0, n = array.size() / (4 * sizeof(float)); i < n; ++i)
rgbas[i][3] = 1.0f;
}
format = QImage::Format_RGBA32FPx4;
} else {
return m_errorImage;
}
// Everything should be OK here
if (size)
*size = QSize(width, height);
QImage result(width, height, format);
memcpy(result.bits(), array.data(), array.size());
return result;
}
|
