qgs3dsceneexporter.h

Go to the documentation of this file.

/***************************************************************************
  qgs3dsceneexporter.h
  --------------------------------------
  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.                                   *
 *                                                                         *
 ***************************************************************************/
 
#ifndef QGS3DSCENEEXPORTER_H
#define QGS3DSCENEEXPORTER_H
 
#include "qgis_3d.h"
#include "qgs3dexportobject.h"
#include "qgsfeatureid.h"
 
#include <QFile>
#include <QMap>
#include <QMatrix4x4>
#include <QVector3D>
#include <Qt3DCore/QEntity>
#include <Qt3DExtras/QPlaneGeometry>
#include <Qt3DRender/QMesh>
#include <Qt3DRender/QSceneLoader>
 
#define SIP_NO_FILE
 
class QgsTessellatedPolygonGeometry;
class QgsTerrainTileEntity;
class QgsTerrainEntity;
class Qgs3DMapSettings;
class QgsFlatTerrainGenerator;
class QgsDemTerrainGenerator;
class QgsChunkNode;
class Qgs3DExportObject;
class QgsTerrainTextureGenerator;
class QgsVector3D;
class QgsVectorLayer;
class QgsPolygon3DSymbol;
class QgsLine3DSymbol;
class QgsPoint3DSymbol;
class QgsMeshEntity;
class TestQgs3DExporter;
 

class _3D_EXPORT Qgs3DSceneExporter : public Qt3DCore::QEntity
{
    Q_OBJECT
 
  public:
    Qgs3DSceneExporter() {}
 
    ~Qgs3DSceneExporter() override
    {
      for ( Qgs3DExportObject *obj : mObjects )
        delete obj;
    }

    bool parseVectorLayerEntity( Qt3DCore::QEntity *entity, QgsVectorLayer *layer );

    void parseTerrain( QgsTerrainEntity *terrain, const QString &layer );

    bool save( QString sceneName, QString sceneFolderPath, const Qgis::Export3DSceneFormat &exportFormat = Qgis::Export3DSceneFormat::Obj, int precision = 6 ) const;

    void setSmoothEdges( bool smoothEdges ) { mSmoothEdges = smoothEdges; }
    bool smoothEdges() const { return mSmoothEdges; }

    void setExportNormals( bool exportNormals ) { mExportNormals = exportNormals; }
    bool exportNormals() const { return mExportNormals; }

    void setExportTextures( bool exportTextures ) { mExportTextures = exportTextures; }
    bool exportTextures() const { return mExportTextures; }

    void setTerrainResolution( int resolution ) { mTerrainResolution = resolution; }
    int terrainResolution() const { return mTerrainResolution; }

    void setTerrainTextureResolution( int resolution ) { mTerrainTextureResolution = resolution; }
    int terrainTextureResolution() const { return mTerrainTextureResolution; }
    void setScale( float scale ) { mScale = scale; }
    float scale() const { return mScale; }

    bool terrainExportEnabled() const { return mTerrainExportEnabled; }

    void setTerrainExportEnabled( bool enabled ) { mTerrainExportEnabled = enabled; }
 
  private:
    QVector<Qgs3DExportObject *> processInstancedPointGeometry( Qt3DCore::QEntity *entity, const QString &objectNamePrefix );
    QVector<Qgs3DExportObject *> processSceneLoaderGeometries( Qt3DRender::QSceneLoader *sceneLoader, const QString &objectNamePrefix );
    Qgs3DExportObject *processGeometryRenderer( Qt3DRender::QGeometryRenderer *mesh, const QString &objectNamePrefix, const QMatrix4x4 &sceneTransform = QMatrix4x4() );
    void processEntityMaterial( Qt3DCore::QEntity *entity, Qgs3DExportObject *object ) const;
    QVector<Qgs3DExportObject *> processLines( Qt3DCore::QEntity *entity, const QString &objectNamePrefix );
    Qgs3DExportObject *processPoints( Qt3DCore::QEntity *entity, const QString &objectNamePrefix );

    QgsTerrainTileEntity *getFlatTerrainEntity( QgsTerrainEntity *terrain, QgsChunkNode *node, const QgsVector3D &mapOrigin );
    QgsTerrainTileEntity *getDemTerrainEntity( QgsTerrainEntity *terrain, QgsChunkNode *node, const QgsVector3D &mapOrigin );
    QgsTerrainTileEntity *getMeshTerrainEntity( QgsTerrainEntity *terrain, QgsChunkNode *node, const QgsVector3D &mapOrigin );

    void parseDemTile( QgsTerrainTileEntity *tileEntity, const QString &layerName );
    void parseFlatTile( QgsTerrainTileEntity *tileEntity, const QString &layerName );
    void parseMeshTile( QgsTerrainTileEntity *meshEntity, const QString &layerName );
 
    QString getObjectName( const QString &name );

    void getSceneCenterAndScale( QVector3D &center, float &scale ) const;

    bool saveObj( QString sceneName, QString sceneFolderPath, int precision ) const;

    bool saveStl( QString sceneName, QString sceneFolderPath, int precision ) const;
 
  private:
    QMap<QString, int> mUsedObjectNamesCounter;
    QVector<Qgs3DExportObject *> mObjects;
 
    bool mSmoothEdges = false;
    int mTerrainResolution = 128;
    bool mExportNormals = true;
    bool mExportTextures = false;
    int mTerrainTextureResolution = 512;
    float mScale = 1.0f;
    bool mTerrainExportEnabled = true;
 
    QSet<QgsFeatureId> mExportedFeatureIds;
 
    friend QgsPolygon3DSymbol;
    friend QgsLine3DSymbol;
    friend QgsPoint3DSymbol;
    friend TestQgs3DExporter;
};
 
#endif // QGS3DSCENEEXPORTER_H