qgstiledscenewireframerenderer.cpp

Go to the documentation of this file.

/***************************************************************************
                         qgstiledscenewireframerenderer.h
                         --------------------
    begin                : August 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.                                   *
 *                                                                         *
 ***************************************************************************/
 
#include "qgstiledscenewireframerenderer.h"
 
#include "qgsfillsymbol.h"
#include "qgslinesymbol.h"
#include "qgssymbollayerutils.h"
 
#include <QString>
 
using namespace Qt::StringLiterals;
 
QgsTiledSceneWireframeRenderer::QgsTiledSceneWireframeRenderer()
{
  mFillSymbol = createDefaultFillSymbol();
  mLineSymbol = createDefaultLineSymbol();
}
 
QgsTiledSceneWireframeRenderer::~QgsTiledSceneWireframeRenderer() = default;
 
QString QgsTiledSceneWireframeRenderer::type() const
{
  return u"wireframe"_s;
}
 
QgsTiledSceneRenderer *QgsTiledSceneWireframeRenderer::clone() const
{
  auto res = std::make_unique< QgsTiledSceneWireframeRenderer >();
 
  res->setFillSymbol( mFillSymbol->clone() );
  res->setLineSymbol( mLineSymbol->clone() );
  res->setUseTextureColors( mUseTextureColors );
 
  copyCommonProperties( res.get() );
 
  return res.release();
}
 
QgsTiledSceneRenderer *QgsTiledSceneWireframeRenderer::create( QDomElement &element, const QgsReadWriteContext &context )
{
  auto r = std::make_unique< QgsTiledSceneWireframeRenderer >();
  {
    const QDomElement fillSymbolElem = element.firstChildElement( u"fillSymbol"_s );
    if ( !fillSymbolElem.isNull() )
    {
      const QDomElement symbolElem = fillSymbolElem.firstChildElement( u"symbol"_s );
      std::unique_ptr< QgsFillSymbol > fillSymbol( QgsSymbolLayerUtils::loadSymbol<QgsFillSymbol>( symbolElem, context ) );
      if ( fillSymbol )
        r->mFillSymbol = std::move( fillSymbol );
    }
  }
  {
    const QDomElement lineSymbolElem = element.firstChildElement( u"lineSymbol"_s );
    if ( !lineSymbolElem.isNull() )
    {
      const QDomElement symbolElem = lineSymbolElem.firstChildElement( u"symbol"_s );
      std::unique_ptr< QgsLineSymbol > lineSymbol( QgsSymbolLayerUtils::loadSymbol<QgsLineSymbol>( symbolElem, context ) );
      if ( lineSymbol )
        r->mLineSymbol = std::move( lineSymbol );
    }
  }
 
  r->setUseTextureColors( element.attribute( u"useTextureColors"_s, u"0"_s ).toInt() );
 
  r->restoreCommonProperties( element, context );
  return r.release();
}
 
std::unique_ptr< QgsFillSymbol > QgsTiledSceneWireframeRenderer::createDefaultFillSymbol()
{
  QVariantMap properties;
  properties.insert( u"color"_s, u"white"_s );
  properties.insert( u"style"_s, u"solid"_s );
  properties.insert( u"style_border"_s, u"solid"_s );
  properties.insert( u"color_border"_s, u"black"_s );
  properties.insert( u"width_border"_s, u"0.3"_s );
  properties.insert( u"joinstyle"_s, u"miter"_s );
 
  return QgsFillSymbol::createSimple( properties );
}
 
QgsFillSymbol *QgsTiledSceneWireframeRenderer::fillSymbol() const
{
  return mFillSymbol.get();
}
 
void QgsTiledSceneWireframeRenderer::setFillSymbol( QgsFillSymbol *symbol )
{
  mFillSymbol.reset( symbol );
}
 
std::unique_ptr<QgsLineSymbol> QgsTiledSceneWireframeRenderer::createDefaultLineSymbol()
{
  QVariantMap properties;
  properties.insert( u"color"_s, u"red"_s );
 
  return QgsLineSymbol::createSimple( properties );
}
 
QgsLineSymbol *QgsTiledSceneWireframeRenderer::lineSymbol() const
{
  return mLineSymbol.get();
}
 
void QgsTiledSceneWireframeRenderer::setLineSymbol( QgsLineSymbol *symbol )
{
  mLineSymbol.reset( symbol );
}
 
bool QgsTiledSceneWireframeRenderer::useTextureColors() const
{
  return mUseTextureColors;
}
 
void QgsTiledSceneWireframeRenderer::setUseTextureColors( bool newUseTextureColors )
{
  mUseTextureColors = newUseTextureColors;
}
 
QDomElement QgsTiledSceneWireframeRenderer::save( QDomDocument &doc, const QgsReadWriteContext &context ) const
{
  QDomElement rendererElem = doc.createElement( u"renderer"_s );
 
  rendererElem.setAttribute( u"type"_s, u"wireframe"_s );
  rendererElem.setAttribute( u"useTextureColors"_s, mUseTextureColors ? u"1"_s : u"0"_s );
 
  {
    QDomElement fillSymbolElem = doc.createElement( u"fillSymbol"_s );
    const QDomElement symbolElement = QgsSymbolLayerUtils::saveSymbol( QString(),
                                      mFillSymbol.get(),
                                      doc,
                                      context );
    fillSymbolElem.appendChild( symbolElement );
    rendererElem.appendChild( fillSymbolElem );
  }
  {
    QDomElement lineSymbolElem = doc.createElement( u"lineSymbol"_s );
    const QDomElement symbolElement = QgsSymbolLayerUtils::saveSymbol( QString(),
                                      mLineSymbol.get(),
                                      doc,
                                      context );
    lineSymbolElem.appendChild( symbolElement );
    rendererElem.appendChild( lineSymbolElem );
  }
  saveCommonProperties( rendererElem, context );
 
  return rendererElem;
}
 
void QgsTiledSceneWireframeRenderer::renderTriangle( QgsTiledSceneRenderContext &context, const QPolygonF &triangle )
{
  if ( mUseTextureColors )
  {
    std::unique_ptr< QgsFillSymbol > s( mFillSymbol->clone() );
    const QImage textureImage = context.textureImage();
    if ( !textureImage.isNull() )
    {
      float textureX1;
      float textureY1;
      float textureX2;
      float textureY2;
      float textureX3;
      float textureY3;
      context.textureCoordinates( textureX1, textureY1, textureX2, textureY2, textureX3, textureY3 );
 
      const QColor centerColor( textureImage.pixelColor(
                                  static_cast<int>( ( ( textureX1 + textureX2 + textureX3 ) / 3 ) * ( textureImage.width() - 1 ) ),
                                  static_cast< int >( ( ( textureY1 + textureY2 + textureY3 ) / 3 ) * ( textureImage.height() - 1 ) ) )
                              );
      s->setColor( centerColor );
    }
    s->startRender( context.renderContext() );
    s->renderPolygon( triangle, nullptr, nullptr, context.renderContext() );
    s->stopRender( context.renderContext() );
  }
  else
  {
    mFillSymbol->renderPolygon( triangle, nullptr, nullptr, context.renderContext() );
  }
}
 
void QgsTiledSceneWireframeRenderer::renderLine( QgsTiledSceneRenderContext &context, const QPolygonF &line )
{
  mLineSymbol->renderPolyline( line, nullptr, context.renderContext() );
}
 
void QgsTiledSceneWireframeRenderer::startRender( QgsTiledSceneRenderContext &context )
{
  QgsTiledSceneRenderer::startRender( context );
 
  if ( !mUseTextureColors )
    mFillSymbol->startRender( context.renderContext() );
 
  mLineSymbol->startRender( context.renderContext() );
}
 
void QgsTiledSceneWireframeRenderer::stopRender( QgsTiledSceneRenderContext &context )
{
  if ( !mUseTextureColors )
    mFillSymbol->stopRender( context.renderContext() );
 
  mLineSymbol->stopRender( context.renderContext() );
 
  QgsTiledSceneRenderer::stopRender( context );
}
 
Qgis::TiledSceneRendererFlags QgsTiledSceneWireframeRenderer::flags() const
{
  Qgis::TiledSceneRendererFlags flags = Qgis::TiledSceneRendererFlag::RendersTriangles |
                                        Qgis::TiledSceneRendererFlag::RendersLines;
 
  if ( mUseTextureColors )
    flags |= Qgis::TiledSceneRendererFlag::RequiresTextures;
 
  return flags;
}