qgsterraintexturegenerator_p.cpp

Go to the documentation of this file.

/***************************************************************************
  qgsterraintexturegenerator_p.cpp
  --------------------------------------
  Date                 : July 2017
  Copyright            : (C) 2017 by Martin Dobias
  Email                : wonder dot sk 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 "qgsterraintexturegenerator_p.h"
 
#include "qgs3dmapsettings.h"
#include "qgsabstractterrainsettings.h"
#include "qgseventtracing.h"
#include "qgsfutureutils.h"
#include "qgsmaprenderersequentialjob.h"
#include "qgsmapsettings.h"
#include "qgsmapthemecollection.h"
#include "qgsproject.h"
 
#include <QString>
 
#include "moc_qgsterraintexturegenerator_p.cpp"
 
using namespace Qt::StringLiterals;

 
QgsTerrainTextureGenerator::QgsTerrainTextureGenerator( const Qgs3DMapSettings &map )
  : mMap( map )
  , mTextureSize( QSize( mMap.terrainSettings()->mapTileResolution(), mMap.terrainSettings()->mapTileResolution() ) )
{}
 
QFuture<QImage> QgsTerrainTextureGenerator::render( const QgsRectangle &extent, QgsChunkNodeId tileId, const QString &debugText )
{
  QgsMapSettings mapSettings( baseMapSettings() );
  mapSettings.setExtent( extent );
  QSize size = QSize( mTextureSize );
 
  QgsRectangle clippedExtent = extent;
  if ( mMap.sceneMode() == Qgis::SceneMode::Local && mMap.terrainGenerator()->type() == QgsTerrainGenerator::Flat )
  {
    // The flat terrain generator might have non-square tiles, clipped at the scene's extent.
    // We need to produce non-square textures for those cases.
    clippedExtent = extent.intersect( mMap.extent() );
    mapSettings.setExtent( clippedExtent );
  }
  if ( !qgsDoubleNear( clippedExtent.width(), clippedExtent.height() ) )
  {
    if ( clippedExtent.height() > clippedExtent.width() )
      size.setWidth( static_cast< int >( std::round( size.width() * clippedExtent.width() / clippedExtent.height() ) ) );
    else if ( clippedExtent.height() < clippedExtent.width() )
      size.setHeight( static_cast< int >( std::round( size.height() * clippedExtent.height() / clippedExtent.width() ) ) );
  }
  mapSettings.setOutputSize( size );
 
  QgsEventTracing::addEvent( QgsEventTracing::AsyncBegin, u"3D"_s, u"Texture"_s, tileId.text() );
 
  QgsMapRendererSequentialJob *job = new QgsMapRendererSequentialJob( mapSettings );
  connect( job, &QgsMapRendererJob::finished, this, &QgsTerrainTextureGenerator::onRenderingFinished );
 
  auto jobData = std::make_unique<JobData>();
  jobData->tileId = tileId;
  jobData->job = job;
  jobData->extent = extent;
  jobData->debugText = debugText;
 
  connect( &jobData->watcher, &QFutureWatcher<QImage>::canceled, this, [this, &jobData = *jobData]() {
    disconnect( jobData.job, &QgsMapRendererJob::finished, this, &QgsTerrainTextureGenerator::onRenderingFinished );
    connect( jobData.job, &QgsMapRendererJob::finished, jobData.job, &QgsMapRendererSequentialJob::deleteLater );
    jobData.job->cancelWithoutBlocking();
    jobData.promise.finish();
    mJobs.erase( jobData.job );
  } );
  jobData->watcher.setFuture( jobData->promise.future() );
 
  mJobs[job] = std::move( jobData ); //store job data just before launching the job
  job->start();
 
  // QgsDebugMsgLevel( u"added job: %1 .... in queue: %2"_s.arg( jobData.jobId ).arg( jobs.count() ), 2);
  return mJobs[job]->promise.future();
}
 
void QgsTerrainTextureGenerator::onRenderingFinished()
{
  QgsMapRendererSequentialJob *mapJob = static_cast<QgsMapRendererSequentialJob *>( sender() );
 
  Q_ASSERT( mJobs.contains( mapJob ) );
  JobData &jobData = *mJobs[mapJob];
 
  QImage img = mapJob->renderedImage();
 
  if ( mMap.debugFlags().testFlag( Qgis::Map3DDebugFlag::ShowTerrainTileInfo ) )
  {
    // extra tile information for debugging
    QPainter p( &img );
    p.setPen( Qt::red );
    p.setBackgroundMode( Qt::OpaqueMode );
    QFont font = p.font();
    font.setPixelSize( std::max( 30, mMap.terrainSettings()->mapTileResolution() / 6 ) );
    p.setFont( font );
    p.drawRect( 0, 0, img.width() - 1, img.height() - 1 );
    p.drawText( img.rect(), jobData.debugText, QTextOption( Qt::AlignCenter ) );
    p.end();
  }
 
  mapJob->deleteLater();
  jobData.promise.addResult( img );
  jobData.promise.finish();
  QgsEventTracing::addEvent( QgsEventTracing::AsyncEnd, u"3D"_s, u"Texture"_s, jobData.tileId.text() );
  mJobs.erase( mapJob );
}
 
QgsMapSettings QgsTerrainTextureGenerator::baseMapSettings()
{
  QgsMapSettings mapSettings;
 
  mapSettings.setOutputSize( mTextureSize );
  mapSettings.setDestinationCrs( mMap.sceneMode() == Qgis::SceneMode::Globe ? mMap.crs().toGeographicCrs() : mMap.crs() );
  mapSettings.setBackgroundColor( mMap.backgroundColor() );
  mapSettings.setFlag( Qgis::MapSettingsFlag::DrawLabeling, mMap.showLabels() );
  mapSettings.setFlag( Qgis::MapSettingsFlag::Render3DMap );
  mapSettings.setTransformContext( mMap.transformContext() );
  mapSettings.setPathResolver( mMap.pathResolver() );
  mapSettings.setRendererUsage( mMap.rendererUsage() );
 
  QList<QgsMapLayer *> layers;
  QgsMapThemeCollection *mapThemes = mMap.mapThemeCollection();
  QString mapThemeName = mMap.terrainMapTheme();
  if ( mapThemeName.isEmpty() || !mapThemes || !mapThemes->hasMapTheme( mapThemeName ) )
  {
    layers = mMap.layers();
  }
  else
  {
    layers = mapThemes->mapThemeVisibleLayers( mapThemeName );
    mapSettings.setLayerStyleOverrides( mapThemes->mapThemeStyleOverrides( mapThemeName ) );
  }
  layers.erase( std::remove_if( layers.begin(), layers.end(), []( const QgsMapLayer *layer ) { return layer->renderer3D(); } ), layers.end() );
  mapSettings.setLayers( layers );
 
  return mapSettings;
}