Skip to content

Commit

Permalink
Mesh editing fixes (#45598)
Browse files Browse the repository at this point in the history
* fix QgsMeshVerticesElevationDataset

* disable mesh simplification in edit mode

* fix export process algorithm

do not run some algs if layer is in edit mode
build new triangular mesh to not interact with layer triangular mesh

* disable mesh calculator in edit mode
  • Loading branch information
vcloarec committed Oct 21, 2021
1 parent a7db0f2 commit b4741b6
Show file tree
Hide file tree
Showing 5 changed files with 29 additions and 17 deletions.
25 changes: 13 additions & 12 deletions src/analysis/processing/qgsalgorithmexportmesh.cpp
Expand Up @@ -258,6 +258,9 @@ bool QgsExportMeshOnElement::prepareAlgorithm( const QVariantMap &parameters, Qg
if ( !meshLayer || !meshLayer->isValid() )
return false;

if ( meshLayer->isEditable() )
throw QgsProcessingException( QObject::tr( "Input mesh layer in edit mode is not supported" ) );

QgsCoordinateReferenceSystem outputCrs = parameterAsCrs( parameters, QStringLiteral( "CRS_OUTPUT" ), context );
if ( !outputCrs.isValid() )
outputCrs = meshLayer->crs();
Expand Down Expand Up @@ -505,7 +508,6 @@ void QgsExportMeshOnGridAlgorithm::initAlgorithm( const QVariantMap &configurati

static void extractDatasetValues( const QList<int> &datasetGroups,
QgsMeshLayer *meshLayer,
const QgsTriangularMesh &triangularMesh,
const QgsMesh &nativeMesh,
const QgsInterval &relativeTime,
const QSet<int> supportedDataType,
Expand All @@ -522,7 +524,7 @@ static void extractDatasetValues( const QList<int> &datasetGroups,
if ( supportedDataType.contains( dataGroup.metadata.dataType() ) )
{
int valueCount = dataGroup.metadata.dataType() == QgsMeshDatasetGroupMetadata::DataOnVertices ?
triangularMesh.vertices().count() : nativeMesh.faceCount();
nativeMesh.vertices.count() : nativeMesh.faceCount();
dataGroup.datasetValues = meshLayer->datasetValues( datasetIndex, 0, valueCount );
dataGroup.activeFaces = meshLayer->areFacesActive( datasetIndex, 0, nativeMesh.faceCount() );
if ( dataGroup.metadata.dataType() == QgsMeshDatasetGroupMetadata::DataOnVolumes )
Expand Down Expand Up @@ -550,7 +552,6 @@ bool QgsExportMeshOnGridAlgorithm::prepareAlgorithm( const QVariantMap &paramete
if ( !meshLayer->nativeMesh() )
meshLayer->updateTriangularMesh( mTransform ); //necessary to load the native mesh

mTriangularMesh = *meshLayer->triangularMesh();
const QgsMesh &nativeMesh = *meshLayer->nativeMesh();

QList<int> datasetGroups =
Expand All @@ -565,7 +566,8 @@ bool QgsExportMeshOnGridAlgorithm::prepareAlgorithm( const QVariantMap &paramete
QVariant parameterTimeVariant = parameters.value( QStringLiteral( "DATASET_TIME" ) );
QgsInterval relativeTime = datasetRelativetime( parameterTimeVariant, meshLayer, context );

extractDatasetValues( datasetGroups, meshLayer, mTriangularMesh, nativeMesh, relativeTime, supportedDataType(), mDataPerGroup, feedback );
extractDatasetValues( datasetGroups, meshLayer, nativeMesh, relativeTime, supportedDataType(), mDataPerGroup, feedback );
mTriangularMesh.update( meshLayer->nativeMesh(), mTransform );

mExportVectorOption = parameterAsInt( parameters, QStringLiteral( "VECTOR_OPTION" ), context );

Expand Down Expand Up @@ -770,7 +772,7 @@ bool QgsMeshRasterizeAlgorithm::prepareAlgorithm( const QVariantMap &parameters,
if ( !meshLayer->nativeMesh() )
meshLayer->updateTriangularMesh( mTransform ); //necessary to load the native mesh

mTriangularMesh = *meshLayer->triangularMesh();
mTriangularMesh.update( meshLayer->nativeMesh(), mTransform );

QList<int> datasetGroups =
QgsProcessingParameterMeshDatasetGroups::valueAsDatasetGroup( parameters.value( QStringLiteral( "DATASET_GROUPS" ) ) );
Expand All @@ -784,7 +786,7 @@ bool QgsMeshRasterizeAlgorithm::prepareAlgorithm( const QVariantMap &parameters,
QVariant parameterTimeVariant = parameters.value( QStringLiteral( "DATASET_TIME" ) );
QgsInterval relativeTime = datasetRelativetime( parameterTimeVariant, meshLayer, context );

extractDatasetValues( datasetGroups, meshLayer, mTriangularMesh, *meshLayer->nativeMesh(), relativeTime, supportedDataType(), mDataPerGroup, feedback );
extractDatasetValues( datasetGroups, meshLayer, *meshLayer->nativeMesh(), relativeTime, supportedDataType(), mDataPerGroup, feedback );

mLayerRendererSettings = meshLayer->rendererSettings();

Expand Down Expand Up @@ -962,7 +964,7 @@ bool QgsMeshContoursAlgorithm::prepareAlgorithm( const QVariantMap &parameters,
if ( !meshLayer->nativeMesh() )
meshLayer->updateTriangularMesh( mTransform ); //necessary to load the native mesh

mTriangularMesh = *meshLayer->triangularMesh();
mTriangularMesh.update( meshLayer->nativeMesh(), mTransform );
mNativeMesh = *meshLayer->nativeMesh();

// Prepare levels
Expand Down Expand Up @@ -1029,7 +1031,7 @@ bool QgsMeshContoursAlgorithm::prepareAlgorithm( const QVariantMap &parameters,

mDateTimeString = meshLayer->formatTime( relativeTime.hours() );

extractDatasetValues( datasetGroups, meshLayer, mTriangularMesh, mNativeMesh, relativeTime, supportedDataType(), mDataPerGroup, feedback );
extractDatasetValues( datasetGroups, meshLayer, mNativeMesh, relativeTime, supportedDataType(), mDataPerGroup, feedback );

mLayerRendererSettings = meshLayer->rendererSettings();

Expand Down Expand Up @@ -1237,8 +1239,7 @@ bool QgsMeshExportCrossSection::prepareAlgorithm( const QVariantMap &parameters,
return false;

mMeshLayerCrs = meshLayer->crs();
mTriangularMesh = *meshLayer->triangularMesh();

mTriangularMesh.update( meshLayer->nativeMesh() );
QList<int> datasetGroups =
QgsProcessingParameterMeshDatasetGroups::valueAsDatasetGroup( parameters.value( QStringLiteral( "DATASET_GROUPS" ) ) );

Expand All @@ -1251,7 +1252,7 @@ bool QgsMeshExportCrossSection::prepareAlgorithm( const QVariantMap &parameters,
QVariant parameterTimeVariant = parameters.value( QStringLiteral( "DATASET_TIME" ) );
QgsInterval relativeTime = datasetRelativetime( parameterTimeVariant, meshLayer, context );

extractDatasetValues( datasetGroups, meshLayer, mTriangularMesh, *meshLayer->nativeMesh(), relativeTime, supportedDataType(), mDataPerGroup, feedback );
extractDatasetValues( datasetGroups, meshLayer, *meshLayer->nativeMesh(), relativeTime, supportedDataType(), mDataPerGroup, feedback );

mLayerRendererSettings = meshLayer->rendererSettings();

Expand Down Expand Up @@ -1455,7 +1456,7 @@ bool QgsMeshExportTimeSeries::prepareAlgorithm( const QVariantMap &parameters, Q
return false;

mMeshLayerCrs = meshLayer->crs();
mTriangularMesh = *meshLayer->triangularMesh();
mTriangularMesh.update( meshLayer->nativeMesh() );

QList<int> datasetGroups =
QgsProcessingParameterMeshDatasetGroups::valueAsDatasetGroup( parameters.value( QStringLiteral( "DATASET_GROUPS" ) ) );
Expand Down
8 changes: 7 additions & 1 deletion src/app/qgisapp.cpp
Expand Up @@ -6888,7 +6888,13 @@ void QgisApp::showRasterCalculator()

void QgisApp::showMeshCalculator()
{
QgsMeshCalculatorDialog d( qobject_cast<QgsMeshLayer *>( activeLayer() ), this );
QgsMeshLayer *meshLayer = qobject_cast<QgsMeshLayer *>( activeLayer() );
if ( meshLayer && meshLayer->isEditable() )
{
QMessageBox::information( this, tr( "Mesh Calculator" ), tr( "Mesh calculator with mesh layer in edit mode is not supported." ) );
return;
}
QgsMeshCalculatorDialog d( meshLayer, this );
if ( d.exec() == QDialog::Accepted )
{
//invoke analysis library
Expand Down
7 changes: 4 additions & 3 deletions src/core/mesh/qgsmeshdataset.cpp
Expand Up @@ -1111,9 +1111,10 @@ QgsMeshDataBlock QgsMeshVerticesElevationDataset::datasetValues( bool isScalar,
return QgsMeshDataBlock();

QgsMeshDataBlock block( QgsMeshDataBlock::ScalarDouble, count );
QVector<double> values( std::min( count, ( mMesh->vertexCount() - valueIndex ) ) );
for ( int i = 0; i < values.count(); ++i )
values[i] = mMesh->vertex( i ).z();
int effectiveValueCount = std::min( count, ( mMesh->vertexCount() - valueIndex ) );
QVector<double> values( effectiveValueCount );
for ( int i = valueIndex; i < effectiveValueCount; ++i )
values[i] = mMesh->vertex( i - valueIndex ).z();
block.setValues( values );
block.setValid( true );
return block;
Expand Down
2 changes: 2 additions & 0 deletions src/core/mesh/qgsmeshlayer.cpp
Expand Up @@ -964,6 +964,8 @@ bool QgsMeshLayer::startFrameEditing( const QgsCoordinateTransform &transform )
return false;
}

mSimplificationSettings.setEnabled( false );

updateTriangularMesh( transform );

mMeshEditor = new QgsMeshEditor( this );
Expand Down
4 changes: 3 additions & 1 deletion src/gui/mesh/qgsmeshlayerproperties.cpp
Expand Up @@ -216,8 +216,10 @@ void QgsMeshLayerProperties::syncToLayer()
w->syncToLayer( mMeshLayer );

QgsDebugMsgLevel( QStringLiteral( "populate rendering tab" ), 4 );
QgsMeshSimplificationSettings simplifySettings = mMeshLayer->meshSimplificationSettings();
if ( mMeshLayer->isEditable() )
mSimplifyMeshGroupBox->setEnabled( false );

QgsMeshSimplificationSettings simplifySettings = mMeshLayer->meshSimplificationSettings();
mSimplifyMeshGroupBox->setChecked( simplifySettings.isEnabled() );
mSimplifyReductionFactorSpinBox->setValue( simplifySettings.reductionFactor() );
mSimplifyMeshResolutionSpinBox->setValue( simplifySettings.meshResolution() );
Expand Down

0 comments on commit b4741b6

Please sign in to comment.