qgs3dsceneexporter.cpp

Go to the documentation of this file.

/***************************************************************************
  qgs3dsceneexporter.cpp
  --------------------------------------
  Date                 : June 2020
  Copyright            : (C) 2020 by Belgacem Nedjima
  Email                : gb underscore nedjima at esi dot dz
 ***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgs3dsceneexporter.h"
 
#include "qgs3dexportobject.h"
#include "qgs3dmapsettings.h"
#include "qgs3dutils.h"
#include "qgsabstract3drenderer.h"
#include "qgsbillboardgeometry.h"
#include "qgschunknode.h"
#include "qgsdemterraingenerator.h"
#include "qgsdemterraintilegeometry_p.h"
#include "qgsdemterraintileloader_p.h"
#include "qgsflatterraingenerator.h"
#include "qgsfutureutils.h"
#include "qgsgeotransform.h"
#include "qgsimagetexture.h"
#include "qgsmeshterraingenerator.h"
#include "qgsterrainentity.h"
#include "qgsterraingenerator.h"
#include "qgsterraintexturegenerator_p.h"
#include "qgsterraintextureimage_p.h"
#include "qgsterraintileentity_p.h"
#include "qgstessellatedpolygongeometry.h"
#include "qgsvectorlayer.h"
#include "qgsvectorlayer3drenderer.h"
 
#include <QByteArray>
#include <QFile>
#include <QString>
#include <QTextStream>
#include <QVector>
#include <Qt3DCore/QAttribute>
#include <Qt3DCore/QBuffer>
#include <Qt3DCore/QComponent>
#include <Qt3DCore/QEntity>
#include <Qt3DCore/QGeometry>
#include <Qt3DCore/QNode>
#include <Qt3DCore/QTransform>
#include <Qt3DExtras/QConeGeometry>
#include <Qt3DExtras/QCuboidGeometry>
#include <Qt3DExtras/QCylinderGeometry>
#include <Qt3DExtras/QDiffuseSpecularMaterial>
#include <Qt3DExtras/QExtrudedTextMesh>
#include <Qt3DExtras/QPhongMaterial>
#include <Qt3DExtras/QPlaneGeometry>
#include <Qt3DExtras/QSphereGeometry>
#include <Qt3DExtras/QTextureMaterial>
#include <Qt3DExtras/QTorusGeometry>
#include <Qt3DRender/QAbstractTexture>
#include <Qt3DRender/QAbstractTextureImage>
#include <Qt3DRender/QGeometryRenderer>
#include <Qt3DRender/QMesh>
#include <Qt3DRender/QParameter>
#include <Qt3DRender/QSceneLoader>
#include <Qt3DRender/QTexture>
#include <Qt3DRender/QTextureImage>
 
#include "moc_qgs3dsceneexporter.cpp"
 
using namespace Qt::StringLiterals;
 
template<typename T> QVector<T> getAttributeData( Qt3DCore::QAttribute *attribute, const QByteArray &data )
{
  const uint bytesOffset = attribute->byteOffset();
  const uint bytesStride = attribute->byteStride();
  const uint vertexSize = attribute->vertexSize();
  const uint dataSize = static_cast<uint>( data.size() );
  QVector<T> result;
 
  if ( bytesStride == 0 )
  {
    QgsDebugError( "bytesStride==0, the attribute probably was not set properly" );
    return result;
  }
 
  const char *pData = data.constData();
  for ( unsigned int i = bytesOffset; i < dataSize; i += bytesStride )
  {
    for ( unsigned int j = 0; j < vertexSize * sizeof( T ); j += sizeof( T ) )
    {
      T v;
      memcpy( &v, pData + i + j, sizeof( T ) );
      result.push_back( v );
    }
  }
  return result;
}
 
template<typename T> QVector<uint> _getIndexDataImplementation( const QByteArray &data )
{
  QVector<uint> result;
  const char *pData = data.constData();
  for ( int i = 0; i < data.size(); i += sizeof( T ) )
  {
    T v;
    memcpy( &v, pData + i, sizeof( T ) );
    result.push_back( ( uint ) v );
  }
  return result;
}
 
QVector<uint> getIndexData( Qt3DCore::QAttribute *indexAttribute, const QByteArray &data )
{
  switch ( indexAttribute->vertexBaseType() )
  {
    case Qt3DCore::QAttribute::VertexBaseType::Int:
      return _getIndexDataImplementation<int>( data );
    case Qt3DCore::QAttribute::VertexBaseType::UnsignedInt:
      return _getIndexDataImplementation<uint>( data );
    case Qt3DCore::QAttribute::VertexBaseType::Short:
      return _getIndexDataImplementation<short>( data );
    case Qt3DCore::QAttribute::VertexBaseType::UnsignedShort:
      return _getIndexDataImplementation<ushort>( data );
    case Qt3DCore::QAttribute::VertexBaseType::Byte:
      return _getIndexDataImplementation<char>( data );
    case Qt3DCore::QAttribute::VertexBaseType::UnsignedByte:
      return _getIndexDataImplementation<uchar>( data );
    default:
      QgsDebugError( "Probably trying to get index data using an attribute that has vertex data" );
      break;
  }
  return QVector<uint>();
}
 
QByteArray getData( Qt3DCore::QBuffer *buffer )
{
  QByteArray bytes = buffer->data();
  if ( bytes.isNull() )
  {
    QgsDebugError( "QBuffer is null" );
  }
  return bytes;
}
 
Qt3DCore::QAttribute *findAttribute( Qt3DCore::QGeometry *geometry, const QString &name, Qt3DCore::QAttribute::AttributeType type )
{
  QVector<Qt3DCore::QAttribute *> attributes = geometry->attributes();
  for ( Qt3DCore::QAttribute *attribute : attributes )
  {
    if ( attribute->attributeType() != type )
      continue;
    if ( name.isEmpty() || attribute->name() == name )
      return attribute;
  }
  return nullptr;
}
 
template<typename Component> Component *findTypedComponent( Qt3DCore::QEntity *entity )
{
  if ( !entity )
    return nullptr;
  QVector<Qt3DCore::QComponent *> components = entity->components();
  for ( Qt3DCore::QComponent *component : components )
  {
    Component *typedComponent = qobject_cast<Component *>( component );
    if ( typedComponent )
      return typedComponent;
  }
  return nullptr;
}
 
bool Qgs3DSceneExporter::parseVectorLayerEntity( Qt3DCore::QEntity *entity, QgsVectorLayer *layer )
{
  QgsAbstract3DRenderer *abstractRenderer = layer->renderer3D();
  const QString rendererType = abstractRenderer->type();
 
  if ( rendererType == "rulebased" )
  {
    int prevSize = mObjects.size();
    // Potential bug: meshes loaded using Qt3DRender::QSceneLoader will probably have wrong scale and translation
    const QList<Qt3DRender::QGeometryRenderer *> renderers = entity->findChildren<Qt3DRender::QGeometryRenderer *>();
    for ( Qt3DRender::QGeometryRenderer *renderer : renderers )
    {
      Qt3DCore::QEntity *parentEntity = qobject_cast<Qt3DCore::QEntity *>( renderer->parent() );
      if ( !parentEntity )
        continue;
      Qgs3DExportObject *object = processGeometryRenderer( renderer, layer->name() + u"_"_s );
      if ( !object )
        continue;
      if ( mExportTextures )
        processEntityMaterial( parentEntity, object );
      mObjects.push_back( object );
    }
 
    mObjects << processLines( entity, layer->name() + u"_"_s );
 
    return mObjects.size() > prevSize;
  }
 
  else if ( rendererType == "vector" )
  {
    QgsVectorLayer3DRenderer *vectorLayerRenderer = dynamic_cast<QgsVectorLayer3DRenderer *>( abstractRenderer );
    if ( vectorLayerRenderer )
    {
      const QgsAbstract3DSymbol *symbol = vectorLayerRenderer->symbol();
      return symbol->exportGeometries( this, entity, layer->name() + u"_"_s );
    }
    else
      return false;
  }
 
  else
  {
    // TODO: handle pointcloud/mesh/etc. layers
    QgsDebugMsgLevel( u"Type '%1' of layer '%2' is not exportable."_s.arg( layer->name(), rendererType ), 2 );
    return false;
  }
 
  return false;
}
 
void Qgs3DSceneExporter::processEntityMaterial( Qt3DCore::QEntity *entity, Qgs3DExportObject *object ) const
{
  Qt3DExtras::QPhongMaterial *phongMaterial = findTypedComponent<Qt3DExtras::QPhongMaterial>( entity );
  if ( phongMaterial )
  {
    QgsPhongMaterialSettings material = Qgs3DUtils::phongMaterialFromQt3DComponent( phongMaterial );
    object->setupMaterial( &material );
  }
 
  Qt3DExtras::QDiffuseSpecularMaterial *diffuseMapMaterial = findTypedComponent<Qt3DExtras::QDiffuseSpecularMaterial>( entity );
  if ( diffuseMapMaterial )
  {
    const Qt3DRender::QTexture2D *diffuseTexture = diffuseMapMaterial->diffuse().value<Qt3DRender::QTexture2D *>();
    if ( diffuseTexture )
    {
      const QVector<Qt3DRender::QAbstractTextureImage *> textureImages = diffuseTexture->textureImages();
      for ( const Qt3DRender::QAbstractTextureImage *tex : textureImages )
      {
        const QgsImageTexture *imageTexture = dynamic_cast<const QgsImageTexture *>( tex );
        if ( imageTexture )
        {
          const QImage image = imageTexture->getImage();
          object->setTextureImage( image );
          break;
        }
      }
    }
  }
}
 
void Qgs3DSceneExporter::parseTerrain( QgsTerrainEntity *terrain, const QString &layerName )
{
  Qgs3DMapSettings *settings = terrain->mapSettings();
  if ( !settings->terrainRenderingEnabled() || !mTerrainExportEnabled )
    return;
 
  QgsChunkNode *node = terrain->rootNode();
 
  QgsTerrainGenerator *generator = settings->terrainGenerator();
  if ( !generator )
    return;
  QgsTerrainTileEntity *terrainTile = nullptr;
  QgsTerrainTextureGenerator *textureGenerator = terrain->textureGenerator();
  const QSize oldResolution = textureGenerator->textureSize();
  textureGenerator->setTextureSize( QSize( mTerrainTextureResolution, mTerrainTextureResolution ) );
  switch ( generator->type() )
  {
    case QgsTerrainGenerator::Dem:
      terrainTile = getDemTerrainEntity( terrain, node, settings->origin() );
      parseDemTile( terrainTile, layerName + u"_"_s );
      break;
    case QgsTerrainGenerator::Flat:
      terrainTile = getFlatTerrainEntity( terrain, node, settings->origin() );
      parseFlatTile( terrainTile, layerName + u"_"_s );
      break;
    case QgsTerrainGenerator::Mesh:
      terrainTile = getMeshTerrainEntity( terrain, node, settings->origin() );
      parseMeshTile( terrainTile, layerName + u"_"_s );
      break;
    // TODO: implement other terrain types
    case QgsTerrainGenerator::Online:
    case QgsTerrainGenerator::QuantizedMesh:
      break;
  }
  textureGenerator->setTextureSize( oldResolution );
}
 
QgsTerrainTileEntity *Qgs3DSceneExporter::getFlatTerrainEntity( QgsTerrainEntity *terrain, QgsChunkNode *node, const QgsVector3D &mapOrigin )
{
  QgsFlatTerrainGenerator *generator = qgis::down_cast<QgsFlatTerrainGenerator *>( terrain->mapSettings()->terrainGenerator() );
  QFuture<QgsChunkLoaderResult> resultFuture = generator->loadChunk( node );
  QgsFutureUtils::waitForFinished( resultFuture );
  // the entity we created will be deallocated once the scene exporter is deallocated
  Qt3DCore::QEntity *entity = resultFuture.takeResult().createEntity( this );
  QgsTerrainTileEntity *tileEntity = qobject_cast<QgsTerrainTileEntity *>( entity );
 
  const QList<QgsGeoTransform *> transforms = entity->findChildren<QgsGeoTransform *>();
  for ( QgsGeoTransform *transform : transforms )
  {
    transform->setOrigin( mapOrigin );
  }
 
  return tileEntity;
}
 
QgsTerrainTileEntity *Qgs3DSceneExporter::getDemTerrainEntity( QgsTerrainEntity *terrain, QgsChunkNode *node, const QgsVector3D &mapOrigin )
{
  // Just create a new tile (we don't need to export exact level of details as in the scene)
  // create the entity synchronously and then it will be deleted once our scene exporter instance is deallocated
  QgsDemTerrainGenerator *generator = qgis::down_cast<QgsDemTerrainGenerator *>( terrain->mapSettings()->terrainGenerator()->clone() );
  generator->setResolution( mTerrainResolution );
  QFuture<QgsChunkLoaderResult> resultFuture = generator->loadChunk( node );
  QgsFutureUtils::waitForFinished( resultFuture );
  QgsTerrainTileEntity *tileEntity = qobject_cast<QgsTerrainTileEntity *>( resultFuture.takeResult().createEntity( this ) );
 
  const QList<QgsGeoTransform *> transforms = tileEntity->findChildren<QgsGeoTransform *>();
  for ( QgsGeoTransform *transform : transforms )
  {
    transform->setOrigin( mapOrigin );
  }
 
  delete generator;
  return tileEntity;
}
 
QgsTerrainTileEntity *Qgs3DSceneExporter::getMeshTerrainEntity( QgsTerrainEntity *terrain, QgsChunkNode *node, const QgsVector3D &mapOrigin )
{
  QgsMeshTerrainGenerator *generator = qgis::down_cast<QgsMeshTerrainGenerator *>( terrain->mapSettings()->terrainGenerator() );
  QFuture<QgsChunkLoaderResult> resultFuture = generator->loadChunk( node );
  QgsFutureUtils::waitForFinished( resultFuture );
  QgsTerrainTileEntity *tileEntity = qobject_cast<QgsTerrainTileEntity *>( resultFuture.takeResult().createEntity( this ) );
 
  const QList<QgsGeoTransform *> transforms = tileEntity->findChildren<QgsGeoTransform *>();
  for ( QgsGeoTransform *transform : transforms )
  {
    transform->setOrigin( mapOrigin );
  }
 
  return tileEntity;
}
 
void Qgs3DSceneExporter::parseFlatTile( QgsTerrainTileEntity *tileEntity, const QString &layerName )
{
  Qt3DRender::QGeometryRenderer *mesh = findTypedComponent<Qt3DRender::QGeometryRenderer>( tileEntity );
  Qt3DCore::QTransform *transform = findTypedComponent<Qt3DCore::QTransform>( tileEntity );
 
  Qt3DCore::QGeometry *geometry = mesh->geometry();
  Qt3DExtras::QPlaneGeometry *tileGeometry = qobject_cast<Qt3DExtras::QPlaneGeometry *>( geometry );
  if ( !tileGeometry )
  {
    QgsDebugError( "Qt3DExtras::QPlaneGeometry* is expected but something else was given" );
    return;
  }
 
  // Generate vertice data
  Qt3DCore::QAttribute *positionAttribute = tileGeometry->positionAttribute();
  if ( !positionAttribute )
  {
    QgsDebugError( QString( "Cannot export '%1' - geometry has no position attribute!" ).arg( layerName ) );
    return;
  }
  const QByteArray verticesBytes = getData( positionAttribute->buffer() );
  if ( verticesBytes.isNull() )
  {
    QgsDebugError( QString( "Geometry for '%1' has position attribute with empty data!" ).arg( layerName ) );
    return;
  }
  const QVector<float> positionBuffer = getAttributeData<float>( positionAttribute, verticesBytes );
 
  // Generate index data
  Qt3DCore::QAttribute *indexAttribute = tileGeometry->indexAttribute();
  if ( !indexAttribute )
  {
    QgsDebugError( QString( "Cannot export '%1' - geometry has no index attribute!" ).arg( layerName ) );
    return;
  }
  const QByteArray indexBytes = getData( indexAttribute->buffer() );
  if ( indexBytes.isNull() )
  {
    QgsDebugError( QString( "Geometry for '%1' has index attribute with empty data!" ).arg( layerName ) );
    return;
  }
  const QVector<uint> indexesBuffer = getIndexData( indexAttribute, indexBytes );
 
  QString objectNamePrefix = layerName;
  if ( objectNamePrefix != QString() )
    objectNamePrefix += QString();
 
  Qgs3DExportObject *object = new Qgs3DExportObject( getObjectName( objectNamePrefix + u"Flat_tile"_s ) );
  mObjects.push_back( object );
 
  object->setSmoothEdges( mSmoothEdges );
  object->setupTriangle( positionBuffer, indexesBuffer, transform->matrix() );
 
  if ( mExportNormals )
  {
    // Everts
    QVector<float> normalsBuffer;
    for ( int i = 0; i < positionBuffer.size(); i += 3 )
      normalsBuffer << 0.0f << 1.0f << 0.0f;
    object->setupNormalCoordinates( normalsBuffer, transform->matrix() );
  }
 
  Qt3DCore::QAttribute *texCoordsAttribute = tileGeometry->texCoordAttribute();
  if ( mExportTextures && texCoordsAttribute )
  {
    // Reuse vertex buffer data for texture coordinates
    const QVector<float> texCoords = getAttributeData<float>( texCoordsAttribute, verticesBytes );
    object->setupTextureCoordinates( texCoords );
 
    QgsTerrainTextureImage *textureImage = tileEntity->textureImage();
    const QImage img = textureImage->getImage();
    object->setTextureImage( img );
  }
}
 
void Qgs3DSceneExporter::parseDemTile( QgsTerrainTileEntity *tileEntity, const QString &layerName )
{
  Qt3DRender::QGeometryRenderer *mesh = findTypedComponent<Qt3DRender::QGeometryRenderer>( tileEntity );
  Qt3DCore::QTransform *transform = findTypedComponent<Qt3DCore::QTransform>( tileEntity );
 
  Qt3DCore::QGeometry *geometry = mesh->geometry();
  DemTerrainTileGeometry *tileGeometry = qobject_cast<DemTerrainTileGeometry *>( geometry );
  if ( !tileGeometry )
  {
    QgsDebugError( "DemTerrainTileGeometry* is expected but something else was given" );
    return;
  }
 
  Qt3DCore::QAttribute *positionAttribute = tileGeometry->positionAttribute();
  const QByteArray positionBytes = positionAttribute->buffer()->data();
  const QVector<float> positionBuffer = getAttributeData<float>( positionAttribute, positionBytes );
 
  Qt3DCore::QAttribute *indexAttribute = tileGeometry->indexAttribute();
  const QByteArray indexBytes = indexAttribute->buffer()->data();
  const QVector<unsigned int> indexBuffer = getIndexData( indexAttribute, indexBytes );
 
  Qgs3DExportObject *object = new Qgs3DExportObject( getObjectName( layerName + u"DEM_tile"_s ) );
  mObjects.push_back( object );
 
  object->setSmoothEdges( mSmoothEdges );
  object->setupTriangle( positionBuffer, indexBuffer, transform->matrix() );
 
  Qt3DCore::QAttribute *normalsAttributes = tileGeometry->normalAttribute();
  if ( mExportNormals && normalsAttributes )
  {
    const QByteArray normalsBytes = normalsAttributes->buffer()->data();
    const QVector<float> normalsBuffer = getAttributeData<float>( normalsAttributes, normalsBytes );
    object->setupNormalCoordinates( normalsBuffer, transform->matrix() );
  }
 
  Qt3DCore::QAttribute *texCoordsAttribute = tileGeometry->texCoordsAttribute();
  if ( mExportTextures && texCoordsAttribute )
  {
    const QByteArray texCoordsBytes = texCoordsAttribute->buffer()->data();
    const QVector<float> texCoordsBuffer = getAttributeData<float>( texCoordsAttribute, texCoordsBytes );
    object->setupTextureCoordinates( texCoordsBuffer );
 
    QgsTerrainTextureImage *textureImage = tileEntity->textureImage();
    const QImage img = textureImage->getImage();
    object->setTextureImage( img );
  }
}
 
void Qgs3DSceneExporter::parseMeshTile( QgsTerrainTileEntity *tileEntity, const QString &layerName )
{
  QString objectNamePrefix = layerName;
  if ( objectNamePrefix != QString() )
    objectNamePrefix += '_'_L1;
 
  const QList<Qt3DRender::QGeometryRenderer *> renderers = tileEntity->findChildren<Qt3DRender::QGeometryRenderer *>();
  for ( Qt3DRender::QGeometryRenderer *renderer : renderers )
  {
    Qgs3DExportObject *obj = processGeometryRenderer( renderer, objectNamePrefix );
    if ( !obj )
      continue;
    mObjects << obj;
  }
}
 
QVector<Qgs3DExportObject *> Qgs3DSceneExporter::processInstancedPointGeometry( Qt3DCore::QEntity *entity, const QString &objectNamePrefix )
{
  // built-in Qt3D geometries (e.g. cylinder, plane, ...) assume Y axis going "up",
  // They are rotated to have their Z axis goes "up", like the rest of the scene
  // Retrieve the rotation matrix
  const QMatrix4x4 instanceMaterialTransform( 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, -1.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0 );
 
  QVector3D symbolScale;
  QVector4D symbolRotation;
  const QList<Qt3DRender::QMaterial *> materials = entity->findChildren<Qt3DRender::QMaterial *>();
  for ( Qt3DRender::QMaterial *material : materials )
  {
    for ( const Qt3DRender::QParameter *parameter : material->parameters() )
    {
      if ( parameter->name() == "symbolScale"_L1 )
      {
        symbolScale = parameter->value().value<QVector3D>();
      }
      else if ( parameter->name() == "symbolRotation"_L1 )
      {
        symbolRotation = parameter->value().value<QVector4D>();
      }
    }
  }
 
  QQuaternion rotationQuat( symbolRotation.w(), symbolRotation.x(), symbolRotation.y(), symbolRotation.z() );
 
  QVector<Qgs3DExportObject *> objects;
  const QList<Qt3DCore::QGeometry *> geometriesList = entity->findChildren<Qt3DCore::QGeometry *>();
  for ( Qt3DCore::QGeometry *geometry : geometriesList )
  {
    Qt3DCore::QAttribute *positionAttribute = findAttribute( geometry, Qt3DCore::QAttribute::defaultPositionAttributeName(), Qt3DCore::QAttribute::VertexAttribute );
    Qt3DCore::QAttribute *indexAttribute = findAttribute( geometry, QString(), Qt3DCore::QAttribute::IndexAttribute );
    if ( !positionAttribute || !indexAttribute )
    {
      QgsDebugError( QString( "Cannot export '%1' - geometry has no position or index attribute!" ).arg( objectNamePrefix ) );
      continue;
    }
 
    Qt3DCore::QAttribute *instanceScaleAttribute = findAttribute( geometry, u"instanceScale"_s, Qt3DCore::QAttribute::VertexAttribute );
    Qt3DCore::QAttribute *instanceRotationAttribute = findAttribute( geometry, u"instanceRotation"_s, Qt3DCore::QAttribute::VertexAttribute );
 
    const QByteArray vertexBytes = positionAttribute->buffer()->data();
    const QByteArray indexBytes = indexAttribute->buffer()->data();
    if ( vertexBytes.isNull() || indexBytes.isNull() )
    {
      QgsDebugError( QString( "Geometry for '%1' has position or index attribute with empty data!" ).arg( objectNamePrefix ) );
      continue;
    }
 
    const QVector<float> positionData = getAttributeData<float>( positionAttribute, vertexBytes );
    const QVector<uint> indexData = getIndexData( indexAttribute, indexBytes );
 
    const QVector<QVector3D> instanceScaleData = instanceScaleAttribute ? getAttributeData<QVector3D>( instanceScaleAttribute, instanceScaleAttribute->buffer()->data() ) : QVector<QVector3D>();
    const QVector<QVector4D> instanceRotationData = instanceRotationAttribute ? getAttributeData<QVector4D>( instanceRotationAttribute, instanceRotationAttribute->buffer()->data() )
                                                                              : QVector<QVector4D>();
 
    Qt3DCore::QAttribute *instanceDataAttribute = findAttribute( geometry, u"instanceTranslation"_s, Qt3DCore::QAttribute::VertexAttribute );
    if ( !instanceDataAttribute )
    {
      QgsDebugError( QString( "Cannot export '%1' - geometry has no instanceTranslation attribute!" ).arg( objectNamePrefix ) );
      continue;
    }
    const QByteArray instancePositionBytes = getData( instanceDataAttribute->buffer() );
    if ( instancePositionBytes.isNull() )
    {
      QgsDebugError( QString( "Geometry for '%1' has instanceTranslation attribute with empty data!" ).arg( objectNamePrefix ) );
      continue;
    }
    QVector<float> instancePosition = getAttributeData<float>( instanceDataAttribute, instancePositionBytes );
 
    int instanceIndex = 0;
    for ( int i = 0; i < instancePosition.size(); i += 3, instanceIndex++ )
    {
      Qgs3DExportObject *object = new Qgs3DExportObject( getObjectName( objectNamePrefix + u"instance_point"_s ) );
      objects.push_back( object );
      QMatrix4x4 instanceTransform;
      instanceTransform.translate( instancePosition[i], instancePosition[i + 1], instancePosition[i + 2] );
 
      QQuaternion instanceRotation = rotationQuat;
      if ( instanceRotationAttribute )
      {
        const QVector4D instanceRotationVec = instanceRotationData[instanceIndex];
        instanceRotation = QQuaternion( instanceRotationVec.w(), instanceRotationVec.x(), instanceRotationVec.y(), instanceRotationVec.z() );
      }
      instanceTransform.rotate( instanceRotation );
      instanceTransform.scale( instanceScaleAttribute ? instanceScaleData[instanceIndex] : symbolScale );
      instanceTransform *= instanceMaterialTransform;
      object->setupTriangle( positionData, indexData, instanceTransform );
 
      object->setSmoothEdges( mSmoothEdges );
 
      Qt3DCore::QAttribute *normalsAttribute = findAttribute( geometry, Qt3DCore::QAttribute::defaultNormalAttributeName(), Qt3DCore::QAttribute::VertexAttribute );
      if ( mExportNormals && normalsAttribute )
      {
        // Reuse vertex bytes
        const QVector<float> normalsData = getAttributeData<float>( normalsAttribute, vertexBytes );
        object->setupNormalCoordinates( normalsData, instanceTransform );
      }
    }
  }
 
  return objects;
}
 
QVector<Qgs3DExportObject *> Qgs3DSceneExporter::processSceneLoaderGeometries( Qt3DRender::QSceneLoader *sceneLoader, const QString &objectNamePrefix )
{
  QVector<Qgs3DExportObject *> objects;
  Qt3DCore::QEntity *sceneLoaderParent = qobject_cast<Qt3DCore::QEntity *>( sceneLoader->parent() );
  Qt3DCore::QTransform *entityTransform = findTypedComponent<Qt3DCore::QTransform>( sceneLoaderParent );
  QMatrix4x4 sceneTransform;
  if ( entityTransform )
  {
    sceneTransform = entityTransform->matrix();
  }
  QStringList entityNames = sceneLoader->entityNames();
  for ( const QString &entityName : entityNames )
  {
    Qt3DRender::QGeometryRenderer *mesh = qobject_cast<Qt3DRender::QGeometryRenderer *>( sceneLoader->component( entityName, Qt3DRender::QSceneLoader::GeometryRendererComponent ) );
    Qgs3DExportObject *object = processGeometryRenderer( mesh, objectNamePrefix, sceneTransform );
    if ( !object )
      continue;
    objects.push_back( object );
  }
  return objects;
}
 
Qgs3DExportObject *Qgs3DSceneExporter::processGeometryRenderer( Qt3DRender::QGeometryRenderer *geomRenderer, const QString &objectNamePrefix, const QMatrix4x4 &sceneTransform )
{
  // We only export triangles for now
  if ( geomRenderer->primitiveType() != Qt3DRender::QGeometryRenderer::Triangles )
    return nullptr;
 
  Qt3DCore::QGeometry *geometry = geomRenderer->geometry();
  if ( !geometry )
    return nullptr;
 
  if ( findAttribute( geometry, u"pointA"_s, Qt3DCore::QAttribute::VertexAttribute ) )
    return nullptr;
 
  // === Compute triangleIndexStartingIndiceToKeep according to duplicated features
  //
  // In the case of polygons, we have multiple feature geometries within the same geometry object (QgsTessellatedPolygonGeometry).
  // The QgsTessellatedPolygonGeometry class holds the list of all feature ids included.
  // To avoid exporting the same geometry (it can be included in multiple QgsTessellatedPolygonGeometry) more than once,
  // we keep a list of already exported fid and compare with the fid of the current QgsTessellatedPolygonGeometry.
  // As we cannot retrieve the specific geometry part for a featureId from the QgsTessellatedPolygonGeometry, we only reject
  // the geometry if all the featureid are already present.
  QVector<std::pair<uint, uint>> triangleIndexStartingIndiceToKeep;
  QgsTessellatedPolygonGeometry *tessGeom = dynamic_cast<QgsTessellatedPolygonGeometry *>( geometry );
  if ( tessGeom )
  {
    QVector<QgsFeatureId> featureIds = tessGeom->featureIds();
 
    QSet<QgsFeatureId> tempFeatToAdd;
    QVector<uint> triangleIndex = tessGeom->triangleIndexStartingIndices();
 
    for ( int idx = 0; idx < featureIds.size(); idx++ )
    {
      const QgsFeatureId feat = featureIds[idx];
      if ( !mExportedFeatureIds.contains( feat ) )
      {
        // add the feature (as it was unknown) to temp set and not to the mExportedFeatureIds (as featureIds can have the same id multiple times)
        tempFeatToAdd += feat;
 
        // keep the feature triangle indexes
        const uint startIdx = triangleIndex[idx];
        const uint endIdx = idx < triangleIndex.size() - 1 ? triangleIndex[idx + 1] : std::numeric_limits<uint>::max();
 
        if ( startIdx < endIdx ) // keep only valid intervals
          triangleIndexStartingIndiceToKeep.append( std::pair<uint, uint>( startIdx, endIdx ) );
      }
    }
    mExportedFeatureIds += tempFeatToAdd;
 
    if ( triangleIndexStartingIndiceToKeep.isEmpty() ) // all featureid are already exported
    {
      return nullptr;
    }
  }
 
  // === Compute inherited scale and translation from child entity to parent
  QMatrix4x4 transformMatrix = sceneTransform;
  QObject *parent = geomRenderer->parent();
  while ( parent )
  {
    Qt3DCore::QEntity *entity = qobject_cast<Qt3DCore::QEntity *>( parent );
    Qt3DCore::QTransform *transform = findTypedComponent<Qt3DCore::QTransform>( entity );
    if ( transform )
    {
      transformMatrix = transform->matrix() * transformMatrix;
    }
    parent = parent->parent();
  }
 
  Qt3DCore::QAttribute *positionAttribute = nullptr;
  Qt3DCore::QAttribute *indexAttribute = nullptr;
  QByteArray indexBytes, vertexBytes;
  QVector<uint> indexDataTmp;
  QVector<uint> indexData;
  QVector<float> positionData;
 
  // === Extract position data
  positionAttribute = findAttribute( geometry, Qt3DCore::QAttribute::defaultPositionAttributeName(), Qt3DCore::QAttribute::VertexAttribute );
  if ( !positionAttribute )
  {
    QgsDebugError( QString( "Cannot export '%1' - geometry has no position attribute!" ).arg( objectNamePrefix ) );
    return nullptr;
  }
 
  vertexBytes = getData( positionAttribute->buffer() );
  if ( vertexBytes.isNull() )
  {
    QgsDebugError( QString( "Will not export '%1' as geometry has empty position data!" ).arg( objectNamePrefix ) );
    return nullptr;
  }
 
  positionData = getAttributeData<float>( positionAttribute, vertexBytes );
 
  // === Search for face index data
  QVector<Qt3DCore::QAttribute *> attributes = geometry->attributes();
  for ( Qt3DCore::QAttribute *attribute : attributes )
  {
    if ( attribute->attributeType() == Qt3DCore::QAttribute::IndexAttribute )
    {
      indexAttribute = attribute;
      indexBytes = getData( indexAttribute->buffer() );
      if ( indexBytes.isNull() )
      {
        QgsDebugError( QString( "Geometry for '%1' has index attribute with empty data!" ).arg( objectNamePrefix ) );
      }
      else
      {
        indexDataTmp = getIndexData( indexAttribute, indexBytes );
        break;
      }
    }
  }
 
  // for tessellated polygons that don't have index attributes, build them from positionData
  if ( !indexAttribute )
  {
    for ( uint i = 0; i < static_cast<uint>( positionData.size() / 3 ); ++i )
    {
      indexDataTmp.push_back( i );
    }
  }
 
  // === Filter face index data if needed
  if ( triangleIndexStartingIndiceToKeep.isEmpty() )
  {
    // when geometry is NOT a QgsTessellatedPolygonGeometry, no filter, take them all
    indexData.append( indexDataTmp );
  }
  else
  {
    // when geometry is a QgsTessellatedPolygonGeometry, filter according to triangleIndexStartingIndiceToKeep
    int intervalIdx = 0;
    const int triangleIndexStartingIndiceToKeepSize = triangleIndexStartingIndiceToKeep.size();
    const uint indexDataTmpSize = static_cast<uint>( indexDataTmp.size() );
    for ( uint i = 0; i + 2 < indexDataTmpSize; i += 3 )
    {
      const uint triangleIdx = i / 3;
 
      // search for valid triangle index interval
      while ( intervalIdx < triangleIndexStartingIndiceToKeepSize && triangleIdx >= triangleIndexStartingIndiceToKeep[intervalIdx].second )
      {
        intervalIdx++;
      }
 
      // keep only triangles within the triangle index interval
      if ( intervalIdx < triangleIndexStartingIndiceToKeepSize && triangleIdx >= triangleIndexStartingIndiceToKeep[intervalIdx].first && triangleIdx < triangleIndexStartingIndiceToKeep[intervalIdx].second )
      {
        indexData.push_back( indexDataTmp[static_cast<int>( i )] );
        indexData.push_back( indexDataTmp[static_cast<int>( i + 1 )] );
        indexData.push_back( indexDataTmp[static_cast<int>( i + 2 )] );
      }
    }
  }
 
  // === Create Qgs3DExportObject
  Qgs3DExportObject *object = new Qgs3DExportObject( getObjectName( objectNamePrefix + u"mesh_geometry"_s ) );
  object->setupTriangle( positionData, indexData, transformMatrix );
 
  Qt3DCore::QAttribute *normalsAttribute = findAttribute( geometry, Qt3DCore::QAttribute::defaultNormalAttributeName(), Qt3DCore::QAttribute::VertexAttribute );
  if ( mExportNormals && normalsAttribute )
  {
    // Reuse vertex bytes
    const QVector<float> normalsData = getAttributeData<float>( normalsAttribute, vertexBytes );
    object->setupNormalCoordinates( normalsData, transformMatrix );
  }
 
  Qt3DCore::QAttribute *texCoordsAttribute = findAttribute( geometry, Qt3DCore::QAttribute::defaultTextureCoordinateAttributeName(), Qt3DCore::QAttribute::VertexAttribute );
  if ( mExportTextures && texCoordsAttribute )
  {
    // Reuse vertex bytes
    const QVector<float> texCoordsData = getAttributeData<float>( texCoordsAttribute, vertexBytes );
    object->setupTextureCoordinates( texCoordsData );
  }
 
  return object;
}
 
QVector<Qgs3DExportObject *> Qgs3DSceneExporter::processLines( Qt3DCore::QEntity *entity, const QString &objectNamePrefix )
{
  QVector<Qgs3DExportObject *> objs;
  const QList<Qt3DRender::QGeometryRenderer *> renderers = entity->findChildren<Qt3DRender::QGeometryRenderer *>();
  for ( Qt3DRender::QGeometryRenderer *renderer : renderers )
  {
    Qt3DCore::QGeometry *geom = renderer->geometry();
    Qt3DCore::QAttribute *pointAAttribute = findAttribute( geom, u"pointA"_s, Qt3DCore::QAttribute::VertexAttribute );
    Qt3DCore::QAttribute *pointBAttribute = findAttribute( geom, u"pointB"_s, Qt3DCore::QAttribute::VertexAttribute );
    Qt3DCore::QAttribute *pointCAttribute = findAttribute( geom, u"pointC"_s, Qt3DCore::QAttribute::VertexAttribute );
    if ( !pointAAttribute || !pointBAttribute || pointCAttribute )
      continue;
 
    const QByteArray pointABytes = getData( pointAAttribute->buffer() );
    const QByteArray pointBBytes = getData( pointBAttribute->buffer() );
    if ( pointABytes.isNull() || pointBBytes.isNull() )
    {
      QgsDebugError( QString( "Geometry for '%1' has pointA or pointB attribute with empty data!" ).arg( objectNamePrefix ) );
      continue;
    }
    const QVector<float> pointAData = getAttributeData<float>( pointAAttribute, pointABytes );
    const QVector<float> pointBData = getAttributeData<float>( pointBAttribute, pointBBytes );
 
    QVector<float> positionData;
    positionData.reserve( pointAData.size() + pointBData.size() );
    for ( int i = 0; i + 2 < pointAData.size(); i += 3 )
    {
      positionData << pointAData[i] << pointAData[i + 1] << pointAData[i + 2];
      positionData << pointBData[i] << pointBData[i + 1] << pointBData[i + 2];
    }
 
    Qgs3DExportObject *exportObject = new Qgs3DExportObject( getObjectName( objectNamePrefix + u"line"_s ) );
    exportObject->setupLine( positionData );
 
    objs.push_back( exportObject );
  }
  return objs;
}
 
Qgs3DExportObject *Qgs3DSceneExporter::processPoints( Qt3DCore::QEntity *entity, const QString &objectNamePrefix )
{
  QVector<float> points;
  const QList<Qt3DRender::QGeometryRenderer *> renderers = entity->findChildren<Qt3DRender::QGeometryRenderer *>();
  for ( Qt3DRender::QGeometryRenderer *renderer : renderers )
  {
    Qt3DCore::QGeometry *geometry = qobject_cast<QgsBillboardGeometry *>( renderer->geometry() );
    if ( !geometry )
      continue;
    Qt3DCore::QAttribute *positionAttribute = findAttribute( geometry, "instancePosition", Qt3DCore::QAttribute::VertexAttribute );
    if ( !positionAttribute )
    {
      QgsDebugError( QString( "Cannot export '%1' - geometry has no position attribute!" ).arg( objectNamePrefix ) );
      continue;
    }
    const QByteArray positionBytes = getData( positionAttribute->buffer() );
    if ( positionBytes.isNull() )
    {
      QgsDebugError( QString( "Geometry for '%1' has position attribute with empty data!" ).arg( objectNamePrefix ) );
      continue;
    }
    const QVector<float> positions = getAttributeData<float>( positionAttribute, positionBytes );
    points << positions;
  }
  Qgs3DExportObject *obj = new Qgs3DExportObject( getObjectName( objectNamePrefix + u"points"_s ) );
  obj->setupPoint( points );
  return obj;
}
 
bool Qgs3DSceneExporter::save( QString sceneName, QString sceneFolderPath, const Qgis::Export3DSceneFormat &exportFormat, int precision ) const
{
  if ( mObjects.isEmpty() )
  {
    return false;
  }
 
  switch ( exportFormat )
  {
    case Qgis::Export3DSceneFormat::Obj:
      return saveObj( sceneName, sceneFolderPath, precision );
    case Qgis::Export3DSceneFormat::StlAscii:
      return saveStl( sceneName, sceneFolderPath, precision );
  }
 
  BUILTIN_UNREACHABLE
}
 
bool Qgs3DSceneExporter::saveObj( QString sceneName, QString sceneFolderPath, int precision ) const
{
  const QString objFilePath = QDir( sceneFolderPath ).filePath( sceneName + u".obj"_s );
  const QString mtlFilePath = QDir( sceneFolderPath ).filePath( sceneName + u".mtl"_s );
 
  QFile file( objFilePath );
  if ( !file.open( QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate ) )
  {
    QgsDebugError( u"Scene can not be exported to '%1'. File access error."_s.arg( objFilePath ) );
    return false;
  }
  QFile mtlFile( mtlFilePath );
  if ( !mtlFile.open( QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate ) )
  {
    QgsDebugError( u"Scene can not be exported to '%1'. File access error."_s.arg( mtlFilePath ) );
    return false;
  }
 
  QVector3D center;
  float scale;
  getSceneCenterAndScale( center, scale );
 
  QTextStream out( &file );
  // set material library name
  const QString mtlLibName = sceneName + ".mtl";
  out << "mtllib " << mtlLibName << "\n";
 
  QTextStream mtlOut( &mtlFile );
  for ( Qgs3DExportObject *obj : std::as_const( mObjects ) )
  {
    if ( !obj )
      continue;
 
    const QString materialName = obj->saveMaterial( mtlOut, sceneFolderPath );
    obj->saveTo( out, scale, center, Qgis::Export3DSceneFormat::Obj, precision, materialName );
  }
 
  QgsDebugMsgLevel( u"Scene exported to '%1'"_s.arg( objFilePath ), 2 );
  return true;
}
 
bool Qgs3DSceneExporter::saveStl( QString sceneName, QString sceneFolderPath, int precision ) const
{
  const QString stlFilePath = QDir( sceneFolderPath ).filePath( sceneName + u".stl"_s );
 
  QFile file( stlFilePath );
  if ( !file.open( QIODevice::WriteOnly | QIODevice::Text | QIODevice::Truncate ) )
  {
    QgsDebugError( u"Scene can not be exported to '%1'. File access error."_s.arg( stlFilePath ) );
    return false;
  }
 
  QVector3D center;
  float scale;
  getSceneCenterAndScale( center, scale );
 
  QTextStream out( &file );
 
  for ( Qgs3DExportObject *object : std::as_const( mObjects ) )
  {
    if ( !object )
      continue;
 
    object->saveTo( out, scale, center, Qgis::Export3DSceneFormat::StlAscii, precision );
  }
 
  QgsDebugMsgLevel( u"Scene exported to '%1'"_s.arg( stlFilePath ), 2 );
  return true;
}
 
void Qgs3DSceneExporter::getSceneCenterAndScale( QVector3D &center, float &scale ) const
{
  const float minFloat = std::numeric_limits<float>::lowest();
  const float maxfloat = std::numeric_limits<float>::max();
 
  float minX = maxfloat;
  float minY = maxfloat;
  float minZ = maxfloat;
  float maxX = minFloat;
  float maxY = minFloat;
  float maxZ = minFloat;
  for ( Qgs3DExportObject *obj : std::as_const( mObjects ) )
  {
    obj->objectBounds( minX, minY, minZ, maxX, maxY, maxZ );
  }
 
  const float diffX = maxX - minX;
  const float diffY = maxY - minY;
  const float diffZ = maxZ - minZ;
 
  center.setX( ( minX + maxX ) / 2.0f );
  center.setY( ( minY + maxY ) / 2.0f );
  center.setZ( ( minZ + maxZ ) / 2.0f );
 
  scale = std::max( diffX, std::max( diffY, diffZ ) ) / mScale;
}
 
QString Qgs3DSceneExporter::getObjectName( const QString &name )
{
  QString ret = name;
  if ( mUsedObjectNamesCounter.contains( name ) )
  {
    ret = u"%1%2"_s.arg( name ).arg( mUsedObjectNamesCounter[name] );
    mUsedObjectNamesCounter[name]++;
  }
  else
    mUsedObjectNamesCounter[name] = 2;
  return ret;
}