qgssimplelinematerial3dhandler.cpp

Go to the documentation of this file.

/***************************************************************************
  qgssimplelinematerial3dhandler.cpp
  --------------------------------------
  Date                 : August 2020
  Copyright            : (C) 2020 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.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgssimplelinematerial3dhandler.h"
 
#include "qgs3dutils.h"
#include "qgslinematerial_p.h"
#include "qgslinestring.h"
#include "qgslinevertexdata_p.h"
#include "qgssimplelinematerialsettings.h"
 
#include <QMap>
#include <QString>
#include <Qt3DCore/QAttribute>
#include <Qt3DCore/QBuffer>
#include <Qt3DCore/QGeometry>
#include <Qt3DExtras/Qt3DWindow>
#include <Qt3DRender/QEffect>
#include <Qt3DRender/QParameter>
#include <Qt3DRender/QTexture>
 
using namespace Qt::StringLiterals;
 
 
QgsMaterial *QgsSimpleLineMaterial3DHandler::toMaterial( const QgsAbstractMaterialSettings *settings, Qgis::MaterialRenderingTechnique technique, const QgsMaterialContext &context ) const
{
  const QgsSimpleLineMaterialSettings *lineSettings = dynamic_cast< const QgsSimpleLineMaterialSettings * >( settings );
  Q_ASSERT( lineSettings );
 
  switch ( technique )
  {
    case Qgis::MaterialRenderingTechnique::Triangles:
    case Qgis::MaterialRenderingTechnique::InstancedPoints:
    case Qgis::MaterialRenderingTechnique::Points:
    case Qgis::MaterialRenderingTechnique::TrianglesWithFixedTexture:
    case Qgis::MaterialRenderingTechnique::TrianglesFromModel:
    case Qgis::MaterialRenderingTechnique::TrianglesDataDefined:
    case Qgis::MaterialRenderingTechnique::Billboards:
      return nullptr;
 
    case Qgis::MaterialRenderingTechnique::Lines:
    {
      if ( context.isHighlighted() )
      {
        // QgsHighlightMaterial does not support lines
        return nullptr;
      }
 
      QgsLineMaterial *mat = new QgsLineMaterial;
      if ( !context.isSelected() )
      {
        mat->setLineColor( lineSettings->ambient() );
        mat->setUseVertexColors( lineSettings->dataDefinedProperties().isActive( QgsAbstractMaterialSettings::Property::Ambient ) );
      }
      else
      {
        // update the material with selection colors
        mat->setLineColor( context.selectionColor() );
        mat->setUseVertexColors( false );
      }
      return mat;
    }
  }
  return nullptr;
}
 
QMap<QString, QString> QgsSimpleLineMaterial3DHandler::toExportParameters( const QgsAbstractMaterialSettings *settings ) const
{
  const QgsSimpleLineMaterialSettings *lineSettings = dynamic_cast< const QgsSimpleLineMaterialSettings * >( settings );
  Q_ASSERT( lineSettings );
 
  QMap<QString, QString> parameters;
  parameters[u"Ka"_s] = u"%1 %2 %3"_s.arg( lineSettings->ambient().redF() ).arg( lineSettings->ambient().greenF() ).arg( lineSettings->ambient().blueF() );
  return parameters;
}
 
QByteArray QgsSimpleLineMaterial3DHandler::dataDefinedVertexColorsAsByte( const QgsAbstractMaterialSettings *settings, const QgsExpressionContext &expressionContext ) const
{
  const QgsSimpleLineMaterialSettings *lineSettings = dynamic_cast< const QgsSimpleLineMaterialSettings * >( settings );
  Q_ASSERT( lineSettings );
 
  const QColor ambient = Qgs3DUtils::srgbToLinear( lineSettings->dataDefinedProperties().valueAsColor( QgsAbstractMaterialSettings::Property::Ambient, expressionContext, lineSettings->ambient() ) );
 
  QByteArray array;
  array.resize( sizeof( unsigned char ) * 3 );
  unsigned char *fptr = reinterpret_cast<unsigned char *>( array.data() );
 
  *fptr++ = static_cast<unsigned char>( ambient.red() );
  *fptr++ = static_cast<unsigned char>( ambient.green() );
  *fptr++ = static_cast<unsigned char>( ambient.blue() );
  return array;
}
 
void QgsSimpleLineMaterial3DHandler::applyDataDefinedToGeometry( const QgsAbstractMaterialSettings *, Qt3DCore::QGeometry *geometry, int instanceCount, const QByteArray &data ) const
{
  Qt3DCore::QBuffer *dataBuffer = new Qt3DCore::QBuffer( geometry );
 
  Qt3DCore::QAttribute *colorAttribute = new Qt3DCore::QAttribute( geometry );
  colorAttribute->setName( u"dataDefinedColor"_s );
  colorAttribute->setVertexBaseType( Qt3DCore::QAttribute::UnsignedByte );
  colorAttribute->setVertexSize( 3 );
  colorAttribute->setAttributeType( Qt3DCore::QAttribute::VertexAttribute );
  colorAttribute->setBuffer( dataBuffer );
  colorAttribute->setByteStride( 3 * sizeof( unsigned char ) );
  colorAttribute->setByteOffset( 0 );
  colorAttribute->setCount( instanceCount );
  colorAttribute->setDivisor( 1 );
  geometry->addAttribute( colorAttribute );
 
  dataBuffer->setData( data );
}
 
QList<QgsAbstractMaterial3DHandler::PreviewMeshType> QgsSimpleLineMaterial3DHandler::previewMeshTypes() const
{
  PreviewMeshType lines;
  lines.type = u"lines"_s;
  lines.displayName = QObject::tr( "Lines" );
  return { lines };
}
 
Qt3DCore::QEntity *QgsSimpleLineMaterial3DHandler::createPreviewMesh( const QString &, Qt3DCore::QEntity *parent ) const
{
  QgsLineVertexData lineVertexData;
  constexpr double s = 1.0;
  // just a boring old flat square
  lineVertexData.addLineString( QgsLineString( { -s, s, s, -s, -s }, { -s, -s, s, s, -s }, { 0, 0, 0, 0, 0 } ) );
 
  Qt3DCore::QEntity *entity = lineVertexData.createSegmentEntity( nullptr );
  entity->setParent( parent );
 
  if ( Qt3DCore::QEntity *joinEntity = lineVertexData.createJoinEntity( nullptr ) )
    joinEntity->setParent( entity );
 
  return entity;
}
 
Qt3DCore::QEntity *QgsSimpleLineMaterial3DHandler::createPreviewScene(
  const QgsAbstractMaterialSettings *settings, const QString &type, const QgsMaterialContext &context, QWindow *window, Qt3DCore::QEntity *parent
) const
{
  Qt3DCore::QEntity *root = new Qt3DCore::QEntity( parent );
  Qt3DCore::QEntity *mesh = createPreviewMesh( type, root );
 
  QgsMaterial *mat = toMaterial( settings, Qgis::MaterialRenderingTechnique::Lines, context );
  QgsLineMaterial *lineMaterial = qobject_cast<QgsLineMaterial *>( mat );
  Q_ASSERT( lineMaterial );
  lineMaterial->setLineWidth( 2 );
 
  mat->setParent( mesh );
  mesh->addComponent( mat );
 
  QgsLineMaterial *joinMaterial = nullptr;
  const QList<QgsLineMaterial *> childLineMaterials = mesh->findChildren<QgsLineMaterial *>();
  for ( QgsLineMaterial *candidate : childLineMaterials )
  {
    if ( candidate->part() == QgsLineMaterial::LinePart::Join )
    {
      joinMaterial = candidate;
      break;
    }
  }
  if ( joinMaterial )
    lineMaterial->copyLineParametersTo( joinMaterial );
 
  if ( window )
  {
    // ensure viewport size is updated if preview widget window size changes
    auto updateViewport = [lineMaterial, joinMaterial, window]() {
      const QSizeF size( window->width(), window->height() );
      lineMaterial->setViewportSize( size );
      if ( joinMaterial )
        joinMaterial->setViewportSize( size );
    };
    QObject::connect( window, &Qt3DExtras::Qt3DWindow::widthChanged, lineMaterial, updateViewport );
    QObject::connect( window, &Qt3DExtras::Qt3DWindow::heightChanged, lineMaterial, updateViewport );
    updateViewport();
  }
 
  return root;
}
 
bool QgsSimpleLineMaterial3DHandler::updatePreviewScene( Qt3DCore::QEntity *sceneRoot, const QgsAbstractMaterialSettings *settings, const QgsMaterialContext & ) const
{
  const QgsSimpleLineMaterialSettings *lineSettings = qgis::down_cast< const QgsSimpleLineMaterialSettings * >( settings );
 
  const QList<QgsLineMaterial *> materials = sceneRoot->findChildren<QgsLineMaterial *>();
  for ( QgsLineMaterial *material : materials )
    material->setLineColor( lineSettings->ambient() );
 
  return true;
}