qgs3dexportobject.h

Go to the documentation of this file.

/***************************************************************************
  Qgs3DExportObject.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 Qgs3DExportObject_H
#define Qgs3DExportObject_H
 
#include "qgis.h"
#include "qgis_3d.h"
 
#include <QImage>
#include <QMap>
#include <QMatrix4x4>
#include <QObject>
#include <QTextStream>
#include <QVector>
#include <QVector3D>
 
#define SIP_NO_FILE
 
class QgsAbstractMaterialSettings;

class _3D_EXPORT Qgs3DExportObject
{
  public:
    enum ObjectType
    {
      TriangularFaces,
      LineStrip,
      Points
    };

    Qgs3DExportObject( const QString &name )
      : mName( name )
    {}

    QString name() const { return mName; }
    void setName( const QString &name ) { mName = name; }

    ObjectType type() const { return mType; }

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

    void setupLine( const QVector<float> &positionsBuffer );

    void setupTriangle( const QVector<float> &positionsBuffer, const QVector<uint> &facesIndexes, const QMatrix4x4 &transform );

    void setupPoint( const QVector<float> &positionsBuffer );

    void setupNormalCoordinates( const QVector<float> &normalsBuffer, const QMatrix4x4 &transform );
    void setupTextureCoordinates( const QVector<float> &texturesBuffer );
    void setupMaterial( QgsAbstractMaterialSettings *material );

    void setTextureImage( const QImage &image ) { this->mTextureImage = image; };
    QImage textureImage() const { return mTextureImage; }

    void objectBounds( float &minX, float &minY, float &minZ, float &maxX, float &maxY, float &maxZ ) const;

    void saveTo(
      QTextStream &out, float scale, const QVector3D &center, const Qgis::Export3DSceneFormat &exportFormat = Qgis::Export3DSceneFormat::Obj, int precision = 6, QString materialName = QString()
    ) const;

    QString saveMaterial( QTextStream &mtlOut, const QString &folder ) const;

    QVector<float> vertexPosition() const { return mVertexPosition; }

    QVector<float> normals() const { return mNormals; }

    QVector<float> texturesUV() const { return mTexturesUV; }

    QVector<unsigned int> indexes() const { return mIndexes; }
 
  private:
    void setupPositionCoordinates( const QVector<float> &positionsBuffer, const QMatrix4x4 &transform = QMatrix4x4() );

    void saveToObj( QTextStream &out, float scale, const QVector3D &center, int precision = 6, QString materialName = QString() ) const;

    void saveToStl( QTextStream &out, float scale, const QVector3D &center, int precision = 6 ) const;
 
  private:
    QString mName;
    ObjectType mType = ObjectType::TriangularFaces;
    QString mParentName;
    QVector<float> mVertexPosition;
    QVector<float> mNormals;
    QVector<float> mTexturesUV;
    QVector<unsigned int> mIndexes;
    QMap<QString, QString> mMaterialParameters;
 
    QImage mTextureImage;
 
    bool mSmoothEdges = false;
};
 
#endif // Qgs3DExportObject_H