Revision control
++ // Copyright (C) 2021 The Qt Company Ltd.
++ // SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
++ // Qt-Security score:significant reason:default
++
++
++ #include "qquick3dparticleutils_p.h"
++ #include <QtQuick3D/private/qquick3dnode_p.h>
++ #include <QtQuick3D/private/qquick3dmodel_p.h>
++
++ #include <QtQuick3DRuntimeRender/private/qssgrenderbuffermanager_p.h>
++
++ #include <QtQuick3D/QQuick3DGeometry>
++ #include <QtGui/qquaternion.h>
++ #include <QtCore/qdir.h>
++
++ #include <iterator>
++
++ QT_BEGIN_NAMESPACE
++
++ Q_CONSTINIT const float qt_quick3d_sine_table[] = {
++ float(0.0),
++ float(0.024541228522912288),
++ float(0.049067674327418015),
++ float(0.073564563599667426),
++ float(0.098017140329560604),
++ float(0.1224106751992162),
++ float(0.14673047445536175),
++ float(0.17096188876030122),
++ float(0.19509032201612825),
++ float(0.2191012401568698),
++ float(0.24298017990326387),
++ float(0.26671275747489837),
++ float(0.29028467725446233),
++ float(0.31368174039889152),
++ float(0.33688985339222005),
++ float(0.35989503653498811),
++ float(0.38268343236508978),
++ float(0.40524131400498986),
++ float(0.42755509343028208),
++ float(0.44961132965460654),
++ float(0.47139673682599764),
++ float(0.49289819222978404),
++ float(0.51410274419322166),
++ float(0.53499761988709715),
++ float(0.55557023301960218),
++ float(0.57580819141784534),
++ float(0.59569930449243336),
++ float(0.61523159058062682),
++ float(0.63439328416364549),
++ float(0.65317284295377676),
++ float(0.67155895484701833),
++ float(0.68954054473706683),
++ float(0.70710678118654746),
++ float(0.72424708295146689),
++ float(0.74095112535495911),
++ float(0.75720884650648446),
++ float(0.77301045336273699),
++ float(0.78834642762660623),
++ float(0.80320753148064483),
++ float(0.81758481315158371),
++ float(0.83146961230254524),
++ float(0.84485356524970701),
++ float(0.85772861000027212),
++ float(0.87008699110871135),
++ float(0.88192126434835494),
++ float(0.89322430119551532),
++ float(0.90398929312344334),
++ float(0.91420975570353069),
++ float(0.92387953251128674),
++ float(0.93299279883473885),
++ float(0.94154406518302081),
++ float(0.94952818059303667),
++ float(0.95694033573220894),
++ float(0.96377606579543984),
++ float(0.97003125319454397),
++ float(0.97570213003852857),
++ float(0.98078528040323043),
++ float(0.98527764238894122),
++ float(0.98917650996478101),
++ float(0.99247953459870997),
++ float(0.99518472667219682),
++ float(0.99729045667869021),
++ float(0.99879545620517241),
++ float(0.99969881869620425),
++ float(1.0),
++ float(0.99969881869620425),
++ float(0.99879545620517241),
++ float(0.99729045667869021),
++ float(0.99518472667219693),
++ float(0.99247953459870997),
++ float(0.98917650996478101),
++ float(0.98527764238894122),
++ float(0.98078528040323043),
++ float(0.97570213003852857),
++ float(0.97003125319454397),
++ float(0.96377606579543984),
++ float(0.95694033573220894),
++ float(0.94952818059303667),
++ float(0.94154406518302081),
++ float(0.93299279883473885),
++ float(0.92387953251128674),
++ float(0.91420975570353069),
++ float(0.90398929312344345),
++ float(0.89322430119551521),
++ float(0.88192126434835505),
++ float(0.87008699110871146),
++ float(0.85772861000027212),
++ float(0.84485356524970723),
++ float(0.83146961230254546),
++ float(0.81758481315158371),
++ float(0.80320753148064494),
++ float(0.78834642762660634),
++ float(0.7730104533627371),
++ float(0.75720884650648468),
++ float(0.74095112535495899),
++ float(0.72424708295146689),
++ float(0.70710678118654757),
++ float(0.68954054473706705),
++ float(0.67155895484701855),
++ float(0.65317284295377664),
++ float(0.63439328416364549),
++ float(0.61523159058062693),
++ float(0.59569930449243347),
++ float(0.57580819141784545),
++ float(0.55557023301960218),
++ float(0.53499761988709715),
++ float(0.51410274419322177),
++ float(0.49289819222978415),
++ float(0.47139673682599786),
++ float(0.44961132965460687),
++ float(0.42755509343028203),
++ float(0.40524131400498992),
++ float(0.38268343236508989),
++ float(0.35989503653498833),
++ float(0.33688985339222033),
++ float(0.31368174039889141),
++ float(0.29028467725446239),
++ float(0.26671275747489848),
++ float(0.24298017990326407),
++ float(0.21910124015687005),
++ float(0.19509032201612861),
++ float(0.17096188876030122),
++ float(0.1467304744553618),
++ float(0.12241067519921635),
++ float(0.098017140329560826),
++ float(0.073564563599667732),
++ float(0.049067674327417966),
++ float(0.024541228522912326),
++ float(0.0),
++ float(-0.02454122852291208),
++ float(-0.049067674327417724),
++ float(-0.073564563599667496),
++ float(-0.09801714032956059),
++ float(-0.1224106751992161),
++ float(-0.14673047445536158),
++ float(-0.17096188876030097),
++ float(-0.19509032201612836),
++ float(-0.2191012401568698),
++ float(-0.24298017990326382),
++ float(-0.26671275747489825),
++ float(-0.29028467725446211),
++ float(-0.31368174039889118),
++ float(-0.33688985339222011),
++ float(-0.35989503653498811),
++ float(-0.38268343236508967),
++ float(-0.40524131400498969),
++ float(-0.42755509343028181),
++ float(-0.44961132965460665),
++ float(-0.47139673682599764),
++ float(-0.49289819222978393),
++ float(-0.51410274419322155),
++ float(-0.53499761988709693),
++ float(-0.55557023301960196),
++ float(-0.57580819141784534),
++ float(-0.59569930449243325),
++ float(-0.61523159058062671),
++ float(-0.63439328416364527),
++ float(-0.65317284295377653),
++ float(-0.67155895484701844),
++ float(-0.68954054473706683),
++ float(-0.70710678118654746),
++ float(-0.72424708295146678),
++ float(-0.74095112535495888),
++ float(-0.75720884650648423),
++ float(-0.77301045336273666),
++ float(-0.78834642762660589),
++ float(-0.80320753148064505),
++ float(-0.81758481315158382),
++ float(-0.83146961230254524),
++ float(-0.84485356524970701),
++ float(-0.85772861000027201),
++ float(-0.87008699110871135),
++ float(-0.88192126434835494),
++ float(-0.89322430119551521),
++ float(-0.90398929312344312),
++ float(-0.91420975570353047),
++ float(-0.92387953251128652),
++ float(-0.93299279883473896),
++ float(-0.94154406518302081),
++ float(-0.94952818059303667),
++ float(-0.95694033573220882),
++ float(-0.96377606579543984),
++ float(-0.97003125319454397),
++ float(-0.97570213003852846),
++ float(-0.98078528040323032),
++ float(-0.98527764238894111),
++ float(-0.9891765099647809),
++ float(-0.99247953459871008),
++ float(-0.99518472667219693),
++ float(-0.99729045667869021),
++ float(-0.99879545620517241),
++ float(-0.99969881869620425),
++ float(-1.0),
++ float(-0.99969881869620425),
++ float(-0.99879545620517241),
++ float(-0.99729045667869021),
++ float(-0.99518472667219693),
++ float(-0.99247953459871008),
++ float(-0.9891765099647809),
++ float(-0.98527764238894122),
++ float(-0.98078528040323043),
++ float(-0.97570213003852857),
++ float(-0.97003125319454397),
++ float(-0.96377606579543995),
++ float(-0.95694033573220894),
++ float(-0.94952818059303679),
++ float(-0.94154406518302092),
++ float(-0.93299279883473907),
++ float(-0.92387953251128663),
++ float(-0.91420975570353058),
++ float(-0.90398929312344334),
++ float(-0.89322430119551532),
++ float(-0.88192126434835505),
++ float(-0.87008699110871146),
++ float(-0.85772861000027223),
++ float(-0.84485356524970723),
++ float(-0.83146961230254546),
++ float(-0.81758481315158404),
++ float(-0.80320753148064528),
++ float(-0.78834642762660612),
++ float(-0.77301045336273688),
++ float(-0.75720884650648457),
++ float(-0.74095112535495911),
++ float(-0.724247082951467),
++ float(-0.70710678118654768),
++ float(-0.68954054473706716),
++ float(-0.67155895484701866),
++ float(-0.65317284295377709),
++ float(-0.63439328416364593),
++ float(-0.61523159058062737),
++ float(-0.59569930449243325),
++ float(-0.57580819141784523),
++ float(-0.55557023301960218),
++ float(-0.53499761988709726),
++ float(-0.51410274419322188),
++ float(-0.49289819222978426),
++ float(-0.47139673682599792),
++ float(-0.44961132965460698),
++ float(-0.42755509343028253),
++ float(-0.40524131400499042),
++ float(-0.38268343236509039),
++ float(-0.359895036534988),
++ float(-0.33688985339222),
++ float(-0.31368174039889152),
++ float(-0.2902846772544625),
++ float(-0.26671275747489859),
++ float(-0.24298017990326418),
++ float(-0.21910124015687016),
++ float(-0.19509032201612872),
++ float(-0.17096188876030177),
++ float(-0.14673047445536239),
++ float(-0.12241067519921603),
++ float(-0.098017140329560506),
++ float(-0.073564563599667412),
++ float(-0.049067674327418091),
++ float(-0.024541228522912448)
++ };
++ static_assert(std::size(qt_quick3d_sine_table) == QT_QUICK3D_SINE_TABLE_SIZE);
++
++ QQuick3DNode *getSharedParentNode(QQuick3DNode *node, QQuick3DNode *system) {
++ QQuick3DNode *systemSharedParent = nullptr;
++ if (node && system) {
++ QVector<QQuick3DNode *> parents;
++ QQuick3DNode *parent = node->parentNode();
++ while (parent) {
++ parents.append(parent);
++ parent = parent->parentNode();
++ }
++
++ parent = system;
++ while (parent) {
++ if (parents.contains(parent)) {
++ systemSharedParent = parent;
++ break;
++ }
++ parent = parent->parentNode();
++ }
++ }
++ return systemSharedParent;
++ }
++
++ QMatrix4x4 calculateParticleTransform(const QQuick3DNode *parent, const QQuick3DNode *systemSharedParent)
++ {
++ QMatrix4x4 transform = parent->sceneTransform();
++ if (systemSharedParent)
++ transform = systemSharedParent->sceneTransform().inverted() * transform;
++ return transform;
++ }
++
++ QQuaternion calculateParticleRotation(const QQuick3DNode *parent, const QQuick3DNode *systemSharedParent)
++ {
++ QQuaternion rotation = parent->sceneRotation();
++ if (systemSharedParent)
++ rotation = systemSharedParent->sceneRotation().inverted() * rotation;
++ return rotation;
++ }
++
++ static QSSGMesh::Mesh loadModelShapeMesh(const QString &source)
++ {
++ QString src = source;
++ if (source.startsWith(QLatin1Char('#'))) {
++ src = QSSGBufferManager::primitivePath(source);
++ src.prepend(QLatin1String(":/"));
++ }
++ src = QDir::cleanPath(src);
++ if (src.startsWith(QLatin1String("qrc:/")))
++ src = src.mid(3);
++ QSSGMesh::Mesh mesh;
++ QFileInfo fileInfo = QFileInfo(src);
++ if (fileInfo.exists()) {
++ QFile file(fileInfo.absoluteFilePath());
++ if (!file.open(QFile::ReadOnly))
++ return {};
++ mesh = QSSGMesh::Mesh::loadMesh(&file);
++ }
++ return mesh;
++ }
++
++ QList<QVector3D> positionsFromModel(QQuick3DModel *model, const QMatrix4x4 *matrix, QQmlContext *context)
++ {
++ QVector<QVector3D> indicedPositions;
++ QVector<QVector3D> positions;
++ if (QQuick3DGeometry *geometry = model->geometry()) {
++ bool hasIndexBuffer = false;
++ QQuick3DGeometry::Attribute::ComponentType indexBufferFormat;
++ int posOffset = 0;
++ QQuick3DGeometry::Attribute::ComponentType posType = QQuick3DGeometry::Attribute::U16Type;
++ for (int i = 0; i < geometry->attributeCount(); ++i) {
++ auto attribute = geometry->attribute(i);
++ if (attribute.semantic == QQuick3DGeometry::Attribute::PositionSemantic) {
++ posOffset = attribute.offset;
++ posType = attribute.componentType;
++ } else if (attribute.semantic == QQuick3DGeometry::Attribute::IndexSemantic) {
++ hasIndexBuffer = true;
++ indexBufferFormat = attribute.componentType;
++ }
++ }
++ if (posType == QQuick3DGeometry::Attribute::F32Type) {
++ const auto &data = geometry->vertexData();
++ int stride = geometry->stride();
++ for (int i = 0; i < data.size(); i += stride) {
++ float v[3];
++ memcpy(v, data + posOffset + i, sizeof(v));
++ if (matrix)
++ positions.append(matrix->mapVector(QVector3D(v[0], v[1], v[2])));
++ else
++ positions.append(QVector3D(v[0], v[1], v[2]));
++ }
++ if (hasIndexBuffer) {
++ const auto &data = geometry->indexData();
++ int indexSize = 4;
++ if (indexBufferFormat == QQuick3DGeometry::Attribute::U16Type)
++ indexSize = 2;
++ for (int i = 0; i < data.size(); i += indexSize) {
++ qsizetype index = 0;
++ memcpy(&index, data + i, indexSize);
++ if (positions.size() > index)
++ indicedPositions.append(positions[index]);
++ }
++ }
++ }
++ } else {
++ QString src = model->source().toString();
++ if (context && !src.startsWith(QLatin1Char('#'))) {
++ if (src.contains(QStringLiteral("#")))
++ src = src.right(src.length() - src.lastIndexOf(QStringLiteral("#")));
++ else
++ src = QQmlFile::urlToLocalFileOrQrc(context->resolvedUrl(model->source()));
++ }
++ QSSGMesh::Mesh mesh = loadModelShapeMesh(src);
++ if (mesh.isValid() && mesh.drawMode() == QSSGMesh::Mesh::DrawMode::Triangles) {
++ auto entries = mesh.vertexBuffer().entries;
++ int posOffset = 0;
++ int posCount = 0;
++ // Just set 'posType' to something to avoid invalid 'maybe-uninitialized' warning
++ QSSGMesh::Mesh::ComponentType posType = QSSGMesh::Mesh::ComponentType::UnsignedInt8;
++ for (int i = 0; i < entries.size(); ++i) {
++ const char *nameStr = entries[i].name.constData();
++ if (!strcmp(nameStr, QSSGMesh::MeshInternal::getPositionAttrName())) {
++ posOffset = entries[i].offset;
++ posCount = entries[i].componentCount;
++ posType = entries[i].componentType;
++ break;
++ }
++ }
++ if (posCount == 3 && posType == QSSGMesh::Mesh::ComponentType::Float32) {
++ const auto &data = mesh.vertexBuffer().data;
++ int stride = mesh.vertexBuffer().stride;
++ for (int i = 0; i < data.size(); i += stride) {
++ float v[3];
++ memcpy(v, data + posOffset + i, sizeof(v));
++ if (matrix)
++ positions.append(matrix->map(QVector3D(v[0], v[1], v[2])));
++ else
++ positions.append(QVector3D(v[0], v[1], v[2]));
++ }
++ const auto &indexData = mesh.indexBuffer().data;
++ int indexSize = QSSGMesh::MeshInternal::byteSizeForComponentType(mesh.indexBuffer().componentType);
++ for (int i = 0; i < indexData.size(); i += indexSize) {
++ qsizetype index = 0;
++ memcpy(&index, indexData + i, indexSize);
++ if (positions.size() > index)
++ indicedPositions.append(positions[index]);
++ }
++ }
++ }
++ }
++ if (!indicedPositions.empty())
++ return indicedPositions;
++ return positions;
++ }
++
++ QT_END_NAMESPACE