Esempio n. 1
0
 def processAlgorithm(self, progress):
     useSelection = self.getParameterValue(Intersection.USE_SELECTED)
     useSelection2 = self.getParameterValue(Intersection.USE_SELECTED2)
     vlayerA = QGisLayers.getObjectFromUri(
         self.getParameterValue(Intersection.INPUT))
     vlayerB = QGisLayers.getObjectFromUri(
         self.getParameterValue(Intersection.INPUT2))
     GEOS_EXCEPT = True
     FEATURE_EXCEPT = True
     vproviderA = vlayerA.dataProvider()
     allAttrsA = vproviderA.attributeIndexes()
     vproviderA.select(allAttrsA)
     vproviderB = vlayerB.dataProvider()
     allAttrsB = vproviderB.attributeIndexes()
     vproviderB.select(allAttrsB)
     # check for crs compatibility
     crsA = vproviderA.crs()
     crsB = vproviderB.crs()
     if not crsA.isValid() or not crsB.isValid():
         SextanteLog.addToLog(
             SextanteLog.LOG_WARNING,
             "Intersection. Invalid CRS. Results might be unexpected")
     else:
         if not crsA != crsB:
             SextanteLog.addToLog(
                 SextanteLog.LOG_WARNING,
                 "Intersection. Non-matching CRSs. Results might be unexpected"
             )
     fields = ftools_utils.combineVectorFields(vlayerA, vlayerB)
     longNames = ftools_utils.checkFieldNameLength(fields)
     if not longNames.isEmpty():
         raise GeoAlgorithmExecutionException(
             "Following field names are longer than 10 characters:\n" +
             longNames.join('\n'))
     writer = self.getOutputFromName(Intersection.OUTPUT).getVectorWriter(
         fields, vproviderA.geometryType(), vproviderA.crs())
     inFeatA = QgsFeature()
     inFeatB = QgsFeature()
     outFeat = QgsFeature()
     index = ftools_utils.createIndex(vproviderB)
     nElement = 0
     # there is selection in input layer
     if useSelection:
         nFeat = vlayerA.selectedFeatureCount()
         selectionA = vlayerA.selectedFeatures()
         # we have selection in overlay layer
         if useSelection2:
             selectionB = vlayerB.selectedFeaturesIds()
             for inFeatA in selectionA:
                 nElement += 1
                 progress.setPercentage(int(nElement / nFeat * 100))
                 geom = QgsGeometry(inFeatA.geometry())
                 atMapA = inFeatA.attributeMap()
                 intersects = index.intersects(geom.boundingBox())
                 for id in intersects:
                     if id in selectionB:
                         vproviderB.featureAtId(int(id), inFeatB, True,
                                                allAttrsB)
                         tmpGeom = QgsGeometry(inFeatB.geometry())
                         try:
                             if geom.intersects(tmpGeom):
                                 atMapB = inFeatB.attributeMap()
                                 int_geom = QgsGeometry(
                                     geom.intersection(tmpGeom))
                                 if int_geom.wkbType() == 7:
                                     int_com = geom.combine(tmpGeom)
                                     int_sym = geom.symDifference(tmpGeom)
                                     int_geom = QgsGeometry(
                                         int_com.difference(int_sym))
                                 try:
                                     outFeat.setGeometry(int_geom)
                                     outFeat.setAttributeMap(
                                         ftools_utils.
                                         combineVectorAttributes(
                                             atMapA, atMapB))
                                     writer.addFeature(outFeat)
                                 except:
                                     FEATURE_EXCEPT = False
                                     continue
                         except:
                             GEOS_EXCEPT = False
                             break
         # we don't have selection in overlay layer
         else:
             for inFeatA in selectionA:
                 nElement += 1
                 progress.setPercentage(int(nElement / nFeat * 100))
                 geom = QgsGeometry(inFeatA.geometry())
                 atMapA = inFeatA.attributeMap()
                 intersects = index.intersects(geom.boundingBox())
                 for id in intersects:
                     vproviderB.featureAtId(int(id), inFeatB, True,
                                            allAttrsB)
                     tmpGeom = QgsGeometry(inFeatB.geometry())
                     try:
                         if geom.intersects(tmpGeom):
                             atMapB = inFeatB.attributeMap()
                             int_geom = QgsGeometry(
                                 geom.intersection(tmpGeom))
                             if int_geom.wkbType() == 7:
                                 int_com = geom.combine(tmpGeom)
                                 int_sym = geom.symDifference(tmpGeom)
                                 int_geom = QgsGeometry(
                                     int_com.difference(int_sym))
                             try:
                                 outFeat.setGeometry(int_geom)
                                 outFeat.setAttributeMap(
                                     ftools_utils.combineVectorAttributes(
                                         atMapA, atMapB))
                                 writer.addFeature(outFeat)
                             except:
                                 EATURE_EXCEPT = False
                                 continue
                     except:
                         GEOS_EXCEPT = False
                         break
     # there is no selection in input layer
     else:
         nFeat = vproviderA.featureCount()
         vproviderA.rewind()
         # we have selection in overlay layer
         if useSelection2:
             selectionB = vlayerB.selectedFeaturesIds()
             while vproviderA.nextFeature(inFeatA):
                 nElement += 1
                 progress.setPercentage(int(nElement / nFeat * 100))
                 geom = QgsGeometry(inFeatA.geometry())
                 atMapA = inFeatA.attributeMap()
                 intersects = index.intersects(geom.boundingBox())
                 for id in intersects:
                     if id in selectionB:
                         vproviderB.featureAtId(int(id), inFeatB, True,
                                                allAttrsB)
                         tmpGeom = QgsGeometry(inFeatB.geometry())
                         try:
                             if geom.intersects(tmpGeom):
                                 atMapB = inFeatB.attributeMap()
                                 int_geom = QgsGeometry(
                                     geom.intersection(tmpGeom))
                                 if int_geom.wkbType() == 7:
                                     int_com = geom.combine(tmpGeom)
                                     int_sym = geom.symDifference(tmpGeom)
                                     int_geom = QgsGeometry(
                                         int_com.difference(int_sym))
                                 try:
                                     outFeat.setGeometry(int_geom)
                                     outFeat.setAttributeMap(
                                         ftools_utils.
                                         combineVectorAttributes(
                                             atMapA, atMapB))
                                     writer.addFeature(outFeat)
                                 except:
                                     FEATURE_EXCEPT = False
                                     continue
                         except:
                             GEOS_EXCEPT = False
                             break
         # we have no selection in overlay layer
         else:
             while vproviderA.nextFeature(inFeatA):
                 nElement += 1
                 progress.setPercentage(int(nElement / nFeat * 100))
                 geom = QgsGeometry(inFeatA.geometry())
                 atMapA = inFeatA.attributeMap()
                 intersects = index.intersects(geom.boundingBox())
                 for id in intersects:
                     vproviderB.featureAtId(int(id), inFeatB, True,
                                            allAttrsB)
                     tmpGeom = QgsGeometry(inFeatB.geometry())
                     try:
                         if geom.intersects(tmpGeom):
                             atMapB = inFeatB.attributeMap()
                             int_geom = QgsGeometry(
                                 geom.intersection(tmpGeom))
                             if int_geom.wkbType() == 7:
                                 int_com = geom.combine(tmpGeom)
                                 int_sym = geom.symDifference(tmpGeom)
                                 int_geom = QgsGeometry(
                                     int_com.difference(int_sym))
                             try:
                                 outFeat.setGeometry(int_geom)
                                 outFeat.setAttributeMap(
                                     ftools_utils.combineVectorAttributes(
                                         atMapA, atMapB))
                                 writer.addFeature(outFeat)
                             except:
                                 FEATURE_EXCEPT = False
                                 continue
                     except:
                         GEOS_EXCEPT = False
                         break
     del writer
     if not GEOS_EXCEPT:
         SextanteLog.addToLog(
             SextanteLog.LOG_WARNING,
             "Geometry exception while computing intersection")
     if not FEATURE_EXCEPT:
         SextanteLog.addToLog(
             SextanteLog.LOG_WARNING,
             "Feature exception while computing interesection")
Esempio n. 2
0
 def processAlgorithm(self, progress):
     useSelection = self.getParameterValue(Intersection.USE_SELECTED)
     useSelection2 = self.getParameterValue(Intersection.USE_SELECTED2)
     vlayerA = QGisLayers.getObjectFromUri(self.getParameterValue(Intersection.INPUT))
     vlayerB = QGisLayers.getObjectFromUri(self.getParameterValue(Intersection.INPUT2))
     GEOS_EXCEPT = True
     FEATURE_EXCEPT = True
     vproviderA = vlayerA.dataProvider()
     allAttrsA = vproviderA.attributeIndexes()
     vproviderA.select( allAttrsA )
     vproviderB = vlayerB.dataProvider()
     allAttrsB = vproviderB.attributeIndexes()
     vproviderB.select( allAttrsB )
     # check for crs compatibility
     crsA = vproviderA.crs()
     crsB = vproviderB.crs()
     if not crsA.isValid() or not crsB.isValid():
         SextanteLog.addToLog(SextanteLog.LOG_WARNING, "Intersection. Invalid CRS. Results might be unexpected")
     else:
         if not crsA != crsB:
             SextanteLog.addToLog(SextanteLog.LOG_WARNING, "Intersection. Non-matching CRSs. Results might be unexpected")
     fields = ftools_utils.combineVectorFields(vlayerA, vlayerB)
     longNames = ftools_utils.checkFieldNameLength( fields )
     if not longNames.isEmpty():
         raise GeoAlgorithmExecutionException("Following field names are longer than 10 characters:\n" +  longNames.join('\n') )
     writer = self.getOutputFromName(Intersection.OUTPUT).getVectorWriter(fields, vproviderA.geometryType(), vproviderA.crs() )
     inFeatA = QgsFeature()
     inFeatB = QgsFeature()
     outFeat = QgsFeature()
     index = ftools_utils.createIndex( vproviderB )
     nElement = 0
     # there is selection in input layer
     if useSelection:
       nFeat = vlayerA.selectedFeatureCount()
       selectionA = vlayerA.selectedFeatures()
       # we have selection in overlay layer
       if useSelection2:
         selectionB = vlayerB.selectedFeaturesIds()
         for inFeatA in selectionA:
           nElement += 1
           progress.setPercentage(int(nElement/nFeat * 100))
           geom = QgsGeometry( inFeatA.geometry() )
           atMapA = inFeatA.attributeMap()
           intersects = index.intersects( geom.boundingBox() )
           for id in intersects:
             if id in selectionB:
               vproviderB.featureAtId( int( id ), inFeatB , True, allAttrsB )
               tmpGeom = QgsGeometry( inFeatB.geometry() )
               try:
                 if geom.intersects( tmpGeom ):
                   atMapB = inFeatB.attributeMap()
                   int_geom = QgsGeometry( geom.intersection( tmpGeom ) )
                   if int_geom.wkbType() == 7:
                     int_com = geom.combine( tmpGeom )
                     int_sym = geom.symDifference( tmpGeom )
                     int_geom = QgsGeometry( int_com.difference( int_sym ) )
                   try:
                     outFeat.setGeometry( int_geom )
                     outFeat.setAttributeMap( ftools_utils.combineVectorAttributes( atMapA, atMapB ) )
                     writer.addFeature( outFeat )
                   except:
                     FEATURE_EXCEPT = False
                     continue
               except:
                 GEOS_EXCEPT = False
                 break
       # we don't have selection in overlay layer
       else:
         for inFeatA in selectionA:
           nElement += 1
           progress.setPercentage(int(nElement/nFeat * 100))
           geom = QgsGeometry( inFeatA.geometry() )
           atMapA = inFeatA.attributeMap()
           intersects = index.intersects( geom.boundingBox() )
           for id in intersects:
             vproviderB.featureAtId( int( id ), inFeatB , True, allAttrsB )
             tmpGeom = QgsGeometry( inFeatB.geometry() )
             try:
               if geom.intersects( tmpGeom ):
                 atMapB = inFeatB.attributeMap()
                 int_geom = QgsGeometry( geom.intersection( tmpGeom ) )
                 if int_geom.wkbType() == 7:
                   int_com = geom.combine( tmpGeom )
                   int_sym = geom.symDifference( tmpGeom )
                   int_geom = QgsGeometry( int_com.difference( int_sym ) )
                 try:
                   outFeat.setGeometry( int_geom )
                   outFeat.setAttributeMap( ftools_utils.combineVectorAttributes( atMapA, atMapB ) )
                   writer.addFeature( outFeat )
                 except:
                   EATURE_EXCEPT = False
                   continue
             except:
               GEOS_EXCEPT = False
               break
     # there is no selection in input layer
     else:
       nFeat = vproviderA.featureCount()
       vproviderA.rewind()
       # we have selection in overlay layer
       if useSelection2:
         selectionB = vlayerB.selectedFeaturesIds()
         while vproviderA.nextFeature( inFeatA ):
           nElement += 1
           progress.setPercentage(int(nElement/nFeat * 100))
           geom = QgsGeometry( inFeatA.geometry() )
           atMapA = inFeatA.attributeMap()
           intersects = index.intersects( geom.boundingBox() )
           for id in intersects:
             if id in selectionB:
               vproviderB.featureAtId( int( id ), inFeatB , True, allAttrsB )
               tmpGeom = QgsGeometry( inFeatB.geometry() )
               try:
                 if geom.intersects( tmpGeom ):
                   atMapB = inFeatB.attributeMap()
                   int_geom = QgsGeometry( geom.intersection( tmpGeom ) )
                   if int_geom.wkbType() == 7:
                     int_com = geom.combine( tmpGeom )
                     int_sym = geom.symDifference( tmpGeom )
                     int_geom = QgsGeometry( int_com.difference( int_sym ) )
                   try:
                     outFeat.setGeometry( int_geom )
                     outFeat.setAttributeMap( ftools_utils.combineVectorAttributes( atMapA, atMapB ) )
                     writer.addFeature( outFeat )
                   except:
                     FEATURE_EXCEPT = False
                     continue
               except:
                 GEOS_EXCEPT = False
                 break
       # we have no selection in overlay layer
       else:
         while vproviderA.nextFeature( inFeatA ):
           nElement += 1
           progress.setPercentage(int(nElement/nFeat * 100))
           geom = QgsGeometry( inFeatA.geometry() )
           atMapA = inFeatA.attributeMap()
           intersects = index.intersects( geom.boundingBox() )
           for id in intersects:
             vproviderB.featureAtId( int( id ), inFeatB , True, allAttrsB )
             tmpGeom = QgsGeometry( inFeatB.geometry() )
             try:
               if geom.intersects( tmpGeom ):
                 atMapB = inFeatB.attributeMap()
                 int_geom = QgsGeometry( geom.intersection( tmpGeom ) )
                 if int_geom.wkbType() == 7:
                   int_com = geom.combine( tmpGeom )
                   int_sym = geom.symDifference( tmpGeom )
                   int_geom = QgsGeometry( int_com.difference( int_sym ) )
                 try:
                   outFeat.setGeometry( int_geom )
                   outFeat.setAttributeMap( ftools_utils.combineVectorAttributes( atMapA, atMapB ) )
                   writer.addFeature( outFeat )
                 except:
                   FEATURE_EXCEPT = False
                   continue
             except:
               GEOS_EXCEPT = False
               break
     del writer
     if not GEOS_EXCEPT:
         SextanteLog.addToLog(SextanteLog.LOG_WARNING, "Geometry exception while computing intersection")
     if not FEATURE_EXCEPT:
         SextanteLog.addToLog(SextanteLog.LOG_WARNING, "Feature exception while computing interesection")
Esempio n. 3
0
    def processAlgorithm(self, progress):
        vlayerA = QGisLayers.getObjectFromUri(self.getParameterValue(Union.INPUT))
        vlayerB = QGisLayers.getObjectFromUri(self.getParameterValue(Union.INPUT2))
        GEOS_EXCEPT = True
        FEATURE_EXCEPT = True
        vproviderA = vlayerA.dataProvider()
        allAttrsA = vproviderA.attributeIndexes()
        vproviderA.select( allAttrsA )
        vproviderB = vlayerB.dataProvider()
        allAttrsB = vproviderB.attributeIndexes()
        vproviderB.select( allAttrsB )

        # check for crs compatibility
        crsA = vproviderA.crs()
        crsB = vproviderB.crs()
        if not crsA.isValid() or not crsB.isValid():
            SextanteLog.addToLog(SextanteLog.LOG_WARNING, "Union. Invalid CRS. Results might be unexpected")
        else:
            if not crsA != crsB:
                SextanteLog.addToLog(SextanteLog.LOG_WARNING, "Union. Non-matching CRSs. Results might be unexpected")
        fields = ftools_utils.combineVectorFields(vlayerA, vlayerB )
        longNames = ftools_utils.checkFieldNameLength( fields )
        if not longNames.isEmpty():
            raise GeoAlgorithmExecutionException("Following field names are longer than 10 characters:\n" +  longNames.join('\n') )
        writer = self.getOutputFromName(Union.OUTPUT).getVectorWriter(fields, vproviderA.geometryType(), vproviderA.crs() )
        inFeatA = QgsFeature()
        inFeatB = QgsFeature()
        outFeat = QgsFeature()
        indexA = ftools_utils.createIndex( vproviderB )
        indexB = ftools_utils.createIndex( vproviderA )
        nFeat = vproviderA.featureCount() * vproviderB.featureCount()
        vproviderA.rewind()
        count = 0
        nElement = 0

        while vproviderA.nextFeature( inFeatA ):
          progress.setPercentage(int(nElement/nFeat * 100))
          nElement += 1
          found = False
          geom = QgsGeometry( inFeatA.geometry() )
          diff_geom = QgsGeometry( geom )
          atMapA = inFeatA.attributeMap()
          intersects = indexA.intersects( geom.boundingBox() )
          if len( intersects ) < 1:
            try:
              outFeat.setGeometry( geom )
              outFeat.setAttributeMap( atMapA )
              writer.addFeature( outFeat )
            except:
              # this really shouldn't happen, as we
              # haven't edited the input geom at all
              FEATURE_EXCEPT = False
          else:
            for id in intersects:
              count += 1
              vproviderB.featureAtId( int( id ), inFeatB , True, allAttrsB )
              atMapB = inFeatB.attributeMap()
              tmpGeom = QgsGeometry( inFeatB.geometry() )
              try:
                if geom.intersects( tmpGeom ):
                  found = True
                  int_geom = geom.intersection( tmpGeom )

                  if int_geom is None:
                    # There was a problem creating the intersection
                    GEOS_EXCEPT = False
                    int_geom = QgsGeometry()
                  else:
                    int_geom = QgsGeometry(int_geom)

                  if diff_geom.intersects( tmpGeom ):
                    diff_geom = diff_geom.difference( tmpGeom )
                    if diff_geom is None:
                      # It's possible there was an error here?
                      diff_geom = QgsGeometry()
                    else:
                      diff_geom = QgsGeometry(diff_geom)

                  if int_geom.wkbType() == 0:
                    # intersection produced different geomety types
                    temp_list = int_geom.asGeometryCollection()
                    for i in temp_list:
                      if i.type() == geom.type():
                          int_geom = QgsGeometry( i )
                  try:
                    outFeat.setGeometry( int_geom )
                    outFeat.setAttributeMap( ftools_utils.combineVectorAttributes( atMapA, atMapB ) )
                    writer.addFeature( outFeat )
                  except Exception, err:
                    FEATURE_EXCEPT = False
                else:
                  # this only happends if the bounding box
                  # intersects, but the geometry doesn't
                  try:
                    outFeat.setGeometry( geom )
                    outFeat.setAttributeMap( atMapA )
                    writer.addFeature( outFeat )
                  except:
                    # also shoudn't ever happen
                    FEATURE_EXCEPT = False
              except Exception, err:
                GEOS_EXCEPT = False
                found = False