Branch data     Line data    Source code

       1                 :            : /***************************************************************************
       2                 :            :     qgsgeometrymultipartcheck.cpp
       3                 :            :     ---------------------
       4                 :            :     begin                : September 2015
       5                 :            :     copyright            : (C) 2014 by Sandro Mani / Sourcepole AG
       6                 :            :     email                : smani at sourcepole dot ch
       7                 :            :  ***************************************************************************
       8                 :            :  *                                                                         *
       9                 :            :  *   This program is free software; you can redistribute it and/or modify  *
      10                 :            :  *   it under the terms of the GNU General Public License as published by  *
      11                 :            :  *   the Free Software Foundation; either version 2 of the License, or     *
      12                 :            :  *   (at your option) any later version.                                   *
      13                 :            :  *                                                                         *
      14                 :            :  ***************************************************************************/
      15                 :            : 
      16                 :            : #include "qgsgeometrycheckcontext.h"
      17                 :            : #include "qgsgeometrymultipartcheck.h"
      18                 :            : #include "qgsfeaturepool.h"
      19                 :            : 
      20                 :         41 : QList<QgsSingleGeometryCheckError *> QgsGeometryMultipartCheck::processGeometry( const QgsGeometry &geometry ) const
      21                 :            : {
      22                 :         41 :   QList<QgsSingleGeometryCheckError *> errors;
      23                 :            : 
      24                 :         41 :   const QgsAbstractGeometry *geom = geometry.constGet();
      25                 :         41 :   QgsWkbTypes::Type type = geom->wkbType();
      26                 :         41 :   if ( geom->partCount() == 1 && QgsWkbTypes::isMultiType( type ) )
      27                 :            :   {
      28                 :         32 :     errors.append( new QgsSingleGeometryCheckError( this, geometry, geometry ) );
      29                 :         32 :   }
      30                 :         41 :   return errors;
      31                 :         41 : }
      32                 :            : 
      33                 :          1 : void QgsGeometryMultipartCheck::fixError( const QMap<QString, QgsFeaturePool *> &featurePools, QgsGeometryCheckError *error, int method, const QMap<QString, int> & /*mergeAttributeIndices*/, Changes &changes ) const
      34                 :            : {
      35                 :          1 :   QgsFeaturePool *featurePool = featurePools[ error->layerId() ];
      36                 :          1 :   QgsFeature feature;
      37                 :          1 :   if ( !featurePool->getFeature( error->featureId(), feature ) )
      38                 :            :   {
      39                 :          0 :     error->setObsolete();
      40                 :          0 :     return;
      41                 :            :   }
      42                 :          1 :   QgsGeometry featureGeom = feature.geometry();
      43                 :          1 :   const QgsAbstractGeometry *geom = featureGeom.constGet();
      44                 :            : 
      45                 :            :   // Check if error still applies
      46                 :          1 :   if ( geom->partCount() > 1 || !QgsWkbTypes::isMultiType( geom->wkbType() ) )
      47                 :            :   {
      48                 :          0 :     error->setObsolete();
      49                 :          0 :     return;
      50                 :            :   }
      51                 :            : 
      52                 :            :   // Fix error
      53                 :          1 :   if ( method == NoChange )
      54                 :            :   {
      55                 :          0 :     error->setFixed( method );
      56                 :          0 :   }
      57                 :          1 :   else if ( method == ConvertToSingle )
      58                 :            :   {
      59                 :          0 :     feature.setGeometry( QgsGeometry( QgsGeometryCheckerUtils::getGeomPart( geom, 0 )->clone() ) );
      60                 :          0 :     featurePool->updateFeature( feature );
      61                 :          0 :     error->setFixed( method );
      62                 :          0 :     changes[error->layerId()][feature.id()].append( Change( ChangeFeature, ChangeChanged ) );
      63                 :          0 :   }
      64                 :          1 :   else if ( method == RemoveObject )
      65                 :            :   {
      66                 :          1 :     featurePool->deleteFeature( feature.id() );
      67                 :          1 :     error->setFixed( method );
      68                 :          1 :     changes[error->layerId()][feature.id()].append( Change( ChangeFeature, ChangeRemoved ) );
      69                 :          1 :   }
      70                 :            :   else
      71                 :            :   {
      72                 :          0 :     error->setFixFailed( tr( "Unknown method" ) );
      73                 :            :   }
      74                 :          1 : }
      75                 :            : 
      76                 :          1 : QStringList QgsGeometryMultipartCheck::resolutionMethods() const
      77                 :            : {
      78                 :          2 :   static QStringList methods = QStringList()
      79                 :          1 :                                << tr( "Convert to single part feature" )
      80                 :          1 :                                << tr( "Delete feature" )
      81                 :          1 :                                << tr( "No action" );
      82                 :          1 :   return methods;
      83                 :          0 : }