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
106
107
108
109
110
|
// 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 QUrlQuery query(QUrl("lightmap://?" + id));
const QString key = query.queryItemValue("key");
const QUrl filePath = query.queryItemValue("file");
const bool useAlpha = query.queryItemValue("alpha") == QStringLiteral("true");
const QString meshPrefix = QStringLiteral("_mesh");
const QString maskSuffix = QStringLiteral("_mask");
const QString finalSuffix = QStringLiteral("_final");
const QString directSuffix = QStringLiteral("_direct");
const QString indirectSuffix = QStringLiteral("_indirect");
const QString metadataSuffix = QStringLiteral("_metadata");
QString baseKey = key;
LightmapDataType dataType = LightmapDataType::Unset;
if (key.endsWith(maskSuffix)) {
baseKey = key.chopped(maskSuffix.length());
dataType = LightmapDataType::U32;
} else if (key.endsWith(directSuffix)) {
baseKey = key.chopped(directSuffix.length());
dataType = LightmapDataType::F32;
} else if (key.endsWith(indirectSuffix)) {
baseKey = key.chopped(indirectSuffix.length());
dataType = LightmapDataType::F32;
} else if (key.endsWith(finalSuffix)) {
baseKey = key.chopped(finalSuffix.length());
dataType = LightmapDataType::F32;
} else if (key.endsWith(metadataSuffix)) {
return m_errorImage;
} else if (key.startsWith(meshPrefix)) {
return m_errorImage;
} else {
// assume f32 image
baseKey = key;
dataType = LightmapDataType::F32;
}
QSharedPointer<QSSGLightmapLoader> loader = QSSGLightmapLoader::open(filePath.toLocalFile());
if (!loader)
return m_errorImage;
QVariantMap metadata = loader->readMetadata(baseKey);
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);
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);
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;
}
|
