qgsmetalroughmaterial.h

Go to the documentation of this file.

/***************************************************************************
  qgsmetalroughmaterial.h
  --------------------------------------
  Date                 : December 2023
  Copyright            : (C) 2023 by Nyall Dawson
  Email                : nyall dot dawson at gmail dot com
 ***************************************************************************
 *                                                                         *
 *   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 QGSMETALROUGHMATERIAL_H
#define QGSMETALROUGHMATERIAL_H
 
#include "qgis.h"
#include "qgis_3d.h"
#include "qgsmaterial.h"
 
#include <QObject>
 
#define SIP_NO_FILE
 
class QMatrix4x4;
 
 
// adapted from Qt's qmetalroughmaterial.h
namespace Qt3DRender
{
 
  class QFilterKey;
  class QEffect;
  class QAbstractTexture;
  class QTechnique;
  class QParameter;
  class QShaderProgram;
  class QShaderProgramBuilder;
  class QRenderPass;
 
} // namespace Qt3DRender


class _3D_EXPORT QgsMetalRoughMaterial : public QgsMaterial
{
    Q_OBJECT
  public:
    explicit QgsMetalRoughMaterial( Qt3DCore::QNode *parent = nullptr );
    ~QgsMetalRoughMaterial() override;
 
  public slots:
    void setBaseColor( const QColor &baseColor );

    void setBaseColorTexture( Qt3DRender::QAbstractTexture *baseColor );

    void setMetalness( float metalness );

    void setMetalnessTexture( Qt3DRender::QAbstractTexture *metalness );

    void setRoughness( float roughness );

    void setRoughnessTexture( Qt3DRender::QAbstractTexture *roughness );

    void setReflectance( float reflectance );

    void setAnisotropy( float anisotropy );

    void setAnisotropyRotation( float rotation );

    void setAmbientOcclusionTexture( Qt3DRender::QAbstractTexture *ambientOcclusion );

    void setNormalTexture( Qt3DRender::QAbstractTexture *normal );

    void setHeightTexture( Qt3DRender::QAbstractTexture *height );
    void setParallaxScale( double scale );

    void setEmissionTexture( Qt3DRender::QAbstractTexture *emission );

    void setEmissionColor( const QColor &color );

    void setEmissionFactor( double factor );

    void setClearCoatFactor( float factor );

    void setClearCoatRoughness( float roughness );

    void setInstancingEnabled( bool enabled, Qgis::InstancedMaterialFlags flags );

    void setInstancingMeshTransform( const QMatrix4x4 &transform );
 
    void setTextureScale( float textureScale );
    void setTextureRotation( float textureRotation );
    void setTextureOffset( float textureOffsetX, float textureOffsetY );
    void setFlatShadingEnabled( bool enabled );
 
    void setOpacity( float opacity );

    void setDataDefinedEnabled( bool enabled );

    void setDataDefinedTextureTransformEnabled( bool enabled );

    void setEnvironmentalLightingEnabled( bool enabled );
 
  private:
    void init();
 
    void updateShaders();
 
    Qt3DRender::QParameter *mBaseColorParameter = nullptr;
    Qt3DRender::QParameter *mMetalnessParameter = nullptr;
    Qt3DRender::QParameter *mRoughnessParameter = nullptr;
    Qt3DRender::QParameter *mReflectanceParameter = nullptr;
    Qt3DRender::QParameter *mAnisotropyParameter = nullptr;
    Qt3DRender::QParameter *mAnisotropyRotationParameter = nullptr;
    Qt3DRender::QParameter *mBaseColorMapParameter = nullptr;
    Qt3DRender::QParameter *mMetalnessMapParameter = nullptr;
    Qt3DRender::QParameter *mRoughnessMapParameter = nullptr;
    Qt3DRender::QParameter *mAmbientOcclusionMapParameter = nullptr;
    Qt3DRender::QParameter *mNormalMapParameter = nullptr;
    Qt3DRender::QParameter *mHeightMapParameter = nullptr;
    Qt3DRender::QParameter *mParallaxScaleParameter = nullptr;
    Qt3DRender::QParameter *mEmissionMapParameter = nullptr;
    Qt3DRender::QParameter *mEmissiveColorParameter = nullptr;
    Qt3DRender::QParameter *mEmissionFactorParameter = nullptr;
    Qt3DRender::QParameter *mClearCoatFactorParameter = nullptr;
    Qt3DRender::QParameter *mClearCoatRoughnessParameter = nullptr;
    Qt3DRender::QParameter *mTextureScaleParameter = nullptr;
    Qt3DRender::QParameter *mTextureRotationParameter = nullptr;
    Qt3DRender::QParameter *mTextureOffsetParameter = nullptr;
    Qt3DRender::QParameter *mOpacityParameter = nullptr;
    Qt3DRender::QEffect *mMetalRoughEffect = nullptr;
    Qt3DRender::QTechnique *mMetalRoughGL3Technique = nullptr;
    Qt3DRender::QRenderPass *mMetalRoughGL3RenderPass = nullptr;
    Qt3DRender::QShaderProgram *mMetalRoughGL3Shader = nullptr;
    Qt3DRender::QFilterKey *mFilterKey = nullptr;
    bool mUsingBaseColorMap = false;
    bool mUsingMetalnessMap = false;
    bool mUsingRoughnessMap = false;
    bool mUsingAmbientOcclusionMap = false;
    bool mUsingNormalMap = false;
    bool mUsingHeightMap = false;
    bool mUsingEmissionMap = false;
    bool mFlatShading = false;
    bool mInstanced = false;
    Qgis::InstancedMaterialFlags mInstanceFlags;
    Qt3DRender::QParameter *mTransformParameter = nullptr;
    Qt3DRender::QParameter *mNormalTransformParameter = nullptr;
 
    bool mDataDefinedEnabled = false;
    bool mDataDefinedTextureTransformEnabled = false;
    bool mEnableEnvironmentalLighting = false;
 
    friend class TestQgsGltf3DUtils;
};

 
#endif // QGSMETALROUGHMATERIAL_H