def test_ogr_factory_2(): src_wkt = 'MULTIPOLYGON (((0 0,100 0,100 100,0 0)))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt() src_wkt = 'MULTISURFACE (((0 0,100 0,100 100,0 0)))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt() src_wkt = 'CURVEPOLYGON ((0 0,100 0,100 100,0 0))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt() src_wkt = 'CURVEPOLYGON (CIRCULARSTRING(0 0,0 1,0 2,1 2,2 2,2 1,2 0,1 0,0 0))' exp_wkt = 'POLYGON ((0 0,0 1,0 2,1 2,2 2,2 1,2 0,1 0,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt()
def get_lonlat_intersection(geometry1, geometry2): """ gets the intersect in lonlat space. geometry1 is split at the antimeridian (i.e. the 180 degree dateline) Parameters ---------- geometry1 : OGRGeometry polygon geometry object in lonlat space is split by the antimeridian geometry2 : OGRGeometry geometry object should be the large one Returns ------- boolean does geometry1 intersect with geometry2? """ geometry1c = geometry1.Clone() geometry2c = geometry2.Clone() geometry1 = None geometry2 = None if geometry1c.GetGeometryName() == 'MULTIPOLYGON': geometry1c = ogr.ForceToPolygon(geometry1c) print( 'Warning: get_lonlat_intersection(): Take care: Multipolygon is forced to Polygon!' ) polygons = split_polygon_by_antimeridian(geometry1c) return polygons.Intersection(geometry2c)
def test_ogr_factory_6(): src_wkt_list = [ None, 'POINT EMPTY', 'LINESTRING EMPTY', 'POLYGON EMPTY', 'MULTIPOINT EMPTY', 'MULTILINESTRING EMPTY', 'MULTIPOLYGON EMPTY', 'GEOMETRYCOLLECTION EMPTY', 'POINT(0 0)', 'LINESTRING(0 0)', 'POLYGON((0 0))', 'POLYGON(EMPTY,(0 0),EMPTY,(1 1))', 'MULTIPOINT(EMPTY,(0 0),EMPTY,(1 1))', 'MULTILINESTRING(EMPTY,(0 0),EMPTY,(1 1))', 'MULTIPOLYGON(((0 0),EMPTY,(1 1)),EMPTY,((2 2)))', 'GEOMETRYCOLLECTION(POINT EMPTY)', 'GEOMETRYCOLLECTION(LINESTRING EMPTY)', 'GEOMETRYCOLLECTION(POLYGON EMPTY)', 'GEOMETRYCOLLECTION(MULTIPOINT EMPTY)', 'GEOMETRYCOLLECTION(MULTILINESTRING EMPTY)', 'GEOMETRYCOLLECTION(MULTIPOLYGON EMPTY)', 'GEOMETRYCOLLECTION(GEOMETRYCOLLECTION EMPTY)', 'GEOMETRYCOLLECTION(POINT(0 0))', 'GEOMETRYCOLLECTION(LINESTRING(0 0),LINESTRING(1 1))', 'GEOMETRYCOLLECTION(POLYGON((0 0),EMPTY,(2 2)), POLYGON((1 1)))', 'CURVEPOLYGON EMPTY', 'CURVEPOLYGON ((0 0,0 1,1 1,1 0,0 0))', 'CURVEPOLYGON (CIRCULARSTRING(0 0,1 0,0 0))', 'COMPOUNDCURVE EMPTY', 'COMPOUNDCURVE ((0 0,0 1,1 1,1 0,0 0))', 'COMPOUNDCURVE (CIRCULARSTRING(0 0,1 0,0 0))', 'CIRCULARSTRING EMPTY', 'CIRCULARSTRING (0 0,1 0,0 0)', 'MULTISURFACE EMPTY', 'MULTISURFACE (((0 0,0 1,1 1,1 0,0 0)))', 'MULTISURFACE (CURVEPOLYGON((0 0,0 1,1 1,1 0,0 0)))', 'MULTICURVE EMPTY', 'MULTICURVE ((0 0,0 1))', 'MULTICURVE (COMPOUNDCURVE((0 0,0 1)))', 'MULTICURVE (CIRCULARSTRING (0 0,1 0,0 0))', ] for src_wkt in src_wkt_list: if src_wkt is None: src_geom = None else: src_geom = ogr.CreateGeometryFromWkt(src_wkt) ogr.ForceToPolygon(src_geom) ogr.ForceToMultiPolygon(src_geom) ogr.ForceToMultiPoint(src_geom) ogr.ForceToMultiLineString(src_geom) ogr.ForceToLineString(src_geom) for target_type in range(ogr.wkbMultiSurface): gdal.PushErrorHandler('CPLQuietErrorHandler') ogr.ForceTo(src_geom, 1 + target_type) gdal.PopErrorHandler()
def ogr_factory_2(): src_wkt = 'MULTIPOLYGON (((0 0,100 0,100 100,0 0)))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' src_wkt = 'MULTISURFACE (((0 0,100 0,100 100,0 0)))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' src_wkt = 'CURVEPOLYGON ((0 0,100 0,100 100,0 0))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' src_wkt = 'CURVEPOLYGON (CIRCULARSTRING(0 0,0 1,0 2,1 2,2 2,2 1,2 0,1 0,0 0))' exp_wkt = 'POLYGON ((0 0,0 1,0 2,1 2,2 2,2 1,2 0,1 0,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' return 'success'
def ogr_factory_2(): src_wkt = 'MULTIPOLYGON (((0 0,100 0,100 100,0 0)))' exp_wkt = 'POLYGON((0 0,100 0,100 100,0 0))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToPolygon(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' return 'success'
def create_geometry(entity_geometry): assert entity_geometry is not None # Get geometry from GEOJSON or WKTstring (GraphQLStreaming) geometry_wkt = shape(entity_geometry).wkt if isinstance( entity_geometry, dict) else entity_geometry # Add geometrie poly = ogr.CreateGeometryFromWkt(geometry_wkt) # Force geometriecollection to polygon if poly and poly.GetGeometryType() == ogr.wkbGeometryCollection: poly = ogr.ForceToPolygon(poly) return poly
def loadGeometry(poLyr): poGeom = None poFeat = poLyr.GetNextFeature() ifeaCount = 0 while poFeat is not None: posrcGeom = poFeat.GetGeometryRef() posrcGeom = ogr.ForceToPolygon(posrcGeom) if ifeaCount == 0: pResGeo = posrcGeom.Clone() if ifeaCount == 3: print("error") if posrcGeom is not None: eType = wkbFlatten(posrcGeom.GetGeometryType()) # if poGeom is None: # poGeom = ogr.Geometry(ogr.wkbMultiPolygon) if eType == ogr.wkbPolygon: # poGeom.AddGeometry(posrcGeom) pResGeo = pResGeo.Union(posrcGeom) elif eType == ogr.wkbMultiPolygon: if posrcGeom is not None: for iGeom in range(posrcGeom.GetGeometryCount()): # poGeom.AddGeometry(posrcGeom.GetGeometryRef(iGeom)) pResGeo = pResGeo.Union(posrcGeom) else: print("ERROR: Geometry not of polygon type.") return None print(ifeaCount) poFeat = poLyr.GetNextFeature() ifeaCount += 1 # pGeo = poGeom.UnionCascaded() return ogr.ForceToPolygon(pResGeo)
def ogr_factory_6(): src_wkt_list = [ None, 'POINT EMPTY', 'LINESTRING EMPTY', 'POLYGON EMPTY', 'MULTIPOINT EMPTY', 'MULTILINESTRING EMPTY', 'MULTIPOLYGON EMPTY', 'GEOMETRYCOLLECTION EMPTY', 'POINT(0 0)', 'LINESTRING(0 0)', 'POLYGON((0 0))', 'POLYGON(EMPTY,(0 0),EMPTY,(1 1))', 'MULTIPOINT(EMPTY,(0 0),EMPTY,(1 1))', 'MULTILINESTRING(EMPTY,(0 0),EMPTY,(1 1))', 'MULTIPOLYGON(((0 0),EMPTY,(1 1)),EMPTY,((2 2)))', 'GEOMETRYCOLLECTION(POINT EMPTY)', 'GEOMETRYCOLLECTION(LINESTRING EMPTY)', 'GEOMETRYCOLLECTION(POLYGON EMPTY)', 'GEOMETRYCOLLECTION(MULTIPOINT EMPTY)', 'GEOMETRYCOLLECTION(MULTILINESTRING EMPTY)', 'GEOMETRYCOLLECTION(MULTIPOLYGON EMPTY)', 'GEOMETRYCOLLECTION(GEOMETRYCOLLECTION EMPTY)', 'GEOMETRYCOLLECTION(POINT(0 0))', 'GEOMETRYCOLLECTION(LINESTRING(0 0),LINESTRING(1 1))', 'GEOMETRYCOLLECTION(POLYGON((0 0),EMPTY,(2 2)), POLYGON((1 1)))', ] for src_wkt in src_wkt_list: if src_wkt is None: src_geom = None else: src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom1 = ogr.ForceToPolygon(src_geom) dst_geom2 = ogr.ForceToMultiPolygon(src_geom) dst_geom3 = ogr.ForceToMultiPoint(src_geom) dst_geom4 = ogr.ForceToMultiLineString(src_geom) dst_geom5 = ogr.ForceToLineString(src_geom) #print(src_geom.ExportToWkt(), dst_geom1.ExportToWkt(), dst_geom2.ExportToWkt(), dst_geom3.ExportToWkt(), dst_geom4.ExportToWkt()) return 'success'
def read_from_file(self, filename): """Read and unpack vector data. It is assumed that the file contains only one layer with the pertinent features. Further it is assumed for the moment that all geometries are points. * A feature is a geometry and a set of attributes. * A geometry refers to location and can be point, line, polygon or combinations thereof. * The attributes or obtained through GetField() The full OGR architecture is documented at * http://www.gdal.org/ogr/ogr_arch.html * http://www.gdal.org/ogr/ogr_apitut.html Examples are at * danieljlewis.org/files/2010/09/basicpythonmap.pdf * http://invisibleroads.com/tutorials/gdal-shapefile-points-save.html * http://www.packtpub.com/article/geospatial-data-python-geometry Limitation of the Shapefile are documented in http://resources.esri.com/help/9.3/ArcGISDesktop/com/Gp_ToolRef/ geoprocessing_tool_reference/ geoprocessing_considerations_for_shapefile_output.htm :param filename: a fully qualified location to the file :type filename: str :raises: ReadLayerError """ base_name = os.path.splitext(filename)[0] # Look for any keywords self.keywords = read_keywords(base_name + '.keywords') # FIXME (Ole): Should also look for style file to populate style_info # Determine name if 'title' in self.keywords: title = self.keywords['title'] # Lookup internationalised title if available title = safe_tr(title) vector_name = title else: # Use base_name without leading directories as name vector_name = os.path.split(base_name)[-1] if self.name is None: self.name = vector_name self.filename = filename self.geometry_type = None # In case there are no features fid = ogr.Open(filename) if fid is None: msg = 'Could not open %s' % filename raise ReadLayerError(msg) # Assume that file contains all data in one layer msg = 'Only one vector layer currently allowed' if fid.GetLayerCount() > 1 and self.sublayer is None: msg = ('WARNING: Number of layers in %s are %i. ' 'Only the first layer will currently be ' 'used. Specify sublayer when creating ' 'the Vector if you wish to use a different layer.' % (filename, fid.GetLayerCount())) LOGGER.warn(msg) # Why do we raise an exception if it is only a warning? TS raise ReadLayerError(msg) if self.sublayer is not None: layer = fid.GetLayerByName(self.sublayer) else: layer = fid.GetLayerByIndex(0) # Get spatial extent self.extent = layer.GetExtent() # Get projection p = layer.GetSpatialRef() self.projection = Projection(p) layer.ResetReading() # Extract coordinates and attributes for all features geometry = [] data = [] # Use feature iterator for feature in layer: # Record coordinates ordered as Longitude, Latitude G = feature.GetGeometryRef() if G is None: msg = ('Geometry was None in filename %s ' % filename) raise ReadLayerError(msg) else: self.geometry_type = G.GetGeometryType() if self.is_point_data: geometry.append((G.GetX(), G.GetY())) elif self.is_line_data: ring = get_ring_data(G) geometry.append(ring) elif self.is_polygon_data: polygon = get_polygon_data(G) geometry.append(polygon) elif self.is_multi_polygon_data: try: G = ogr.ForceToPolygon(G) except: msg = ('Got geometry type Multipolygon (%s) for ' 'filename %s and could not convert it to ' 'singlepart. However, you can use QGIS ' 'functionality to convert multipart vector ' 'data to singlepart (Vector -> Geometry Tools ' '-> Multipart to Singleparts and use the ' 'resulting dataset.' % (ogr.wkbMultiPolygon, filename)) raise ReadLayerError(msg) else: # Read polygon data as single part self.geometry_type = ogr.wkbPolygon polygon = get_polygon_data(G) geometry.append(polygon) else: msg = ('Only point, line and polygon geometries are ' 'supported. ' 'Geometry type in filename %s ' 'was %s.' % (filename, self.geometry_type)) raise ReadLayerError(msg) # Record attributes by name number_of_fields = feature.GetFieldCount() fields = {} for j in range(number_of_fields): name = feature.GetFieldDefnRef(j).GetName() # FIXME (Ole): Ascertain the type of each field? # We need to cast each appropriately? # This is issue #66 # (https://github.com/AIFDR/riab/issues/66) #feature_type = feature.GetFieldDefnRef(j).GetType() fields[name] = feature.GetField(j) # We do this because there is NaN problem on windows # NaN value must be converted to _pseudo_in to solve the # problem. But, when InaSAFE read the file, it'll be # converted back to NaN value, so that NaN in InaSAFE is a # numpy.nan # please check https://github.com/AIFDR/inasafe/issues/269 # for more information if fields[name] == _pseudo_inf: fields[name] = float('nan') #print 'Field', name, feature_type, j, fields[name] data.append(fields) # Store geometry coordinates as a compact numeric array self.geometry = geometry self.data = data
def openGeoJson(self, check_field, filename): driver = ogr.GetDriverByName("GeoJSON") dataSource = driver.Open(filename, 0) layer = dataSource.GetLayer() wfs_result = dict() for feat in layer: #create geometry object geom = feat.GetGeometryRef() if geom is not None: sr = osr.SpatialReference() sr.ImportFromEPSG(3857) geom_type = geom.GetGeometryType() #say to Dima if self.ForceToMultiGeom: if geom_type == ogr.wkbLineString: mercator_geom = ogr.ForceToLineString(geom) elif geom_type == ogr.wkbPolygon: mercator_geom = ogr.ForceToPolygon(geom) elif geom_type == ogr.wkbPoint: mercator_geom = ogr.ForceToMultiPoint(geom) elif geom_type == ogr.wkbMultiPolygon: mercator_geom = ogr.ForceToMultiPolygon(geom) elif geom_type == ogr.wkbMultiPoint: mercator_geom = ogr.ForceToMultiPoint(geom) elif geom_type == ogr.wkbMultiLineString: mercator_geom = ogr.ForceToMultiPolygon(geom) else: mercator_geom = geom else: mercator_geom = geom else: continue #Read broker fields feat_defn = layer.GetLayerDefn() wfs_fields = dict() for i in range(feat_defn.GetFieldCount()): field_defn = feat_defn.GetFieldDefn(i) #if field_defn.GetName() == 'gml_id': # continue #Compare by one control field if field_defn.GetName() == check_field: check_field_val = feat.GetFieldAsString(i).decode( 'utf-8') #GetFieldAsInteger64(i) #Read fields if field_defn.GetType( ) == ogr.OFTInteger: #or field_defn.GetType() == ogr.OFTInteger64: wfs_fields[field_defn.GetName()] = feat.GetFieldAsInteger( i) #GetFieldAsInteger64(i) # print "%s = %d" % (field_defn.GetName(), feat.GetFieldAsInteger64(i)) elif field_defn.GetType() == ogr.OFTReal: wfs_fields[field_defn.GetName()] = feat.GetFieldAsDouble(i) # print "%s = %.3f" % (field_defn.GetName(), feat.GetFieldAsDouble(i)) elif field_defn.GetType() == ogr.OFTString: # print "%s = %s" % (field_defn.GetName(), feat.GetFieldAsString(i)) wfs_fields[field_defn.GetName()] = feat.GetFieldAsString( i).decode('utf-8') else: # print "%s = %s" % (field_defn.GetName(), feat.GetFieldAsString(i)) wfs_fields[field_defn.GetName()] = feat.GetFieldAsString( i).decode('utf-8') #Object with keys - as values of one control field wfs_result[check_field_val] = dict() wfs_result[check_field_val]['id'] = check_field_val wfs_result[check_field_val]['fields'] = wfs_fields wfs_result[check_field_val]['geom'] = mercator_geom.Clone() layer_result_sorted = dict() for key in sorted(wfs_result): layer_result_sorted[key] = wfs_result[key] return layer_result_sorted
def kopieren(self, dsOut, lyrIn, name, zieltyp, codierung=''): try: #srs = osr.SpatialReference() #srs.ImportFromEPSG( 31254 ) ly = None # unser zukünftiges Layerobjekt # lyrIn (das Eingangsshape) hat bereits eine SRS (original oder vom kopierprogramm vorher zugewiesen) # im Modul verglRef überprüft un bei Bedarf neu zugewiesen! srs = lyrIn.GetSpatialRef() if not dsOut == None: driverName = dsOut.GetDriver().GetName() else: if zieltyp == 'file': driverName = "ESRI Shapefile" elif zieltyp == 'sqlite': drivername = "SQLite" elif zieltyp == 'postgres': driverName == "PostgreSQL" else: raise Exception #Das Ziel des Kopierprozesses #wird anhand des Treibernamens identifiziert: ACHTUNG: Ist der Workspace #leer, dann ist dsOut None (z.B. kein Shape im Verzeichnis) if driverName == "ESRI Shapefile": ly = dsOut.CopyLayer(lyrIn, name) #das Kopieren, unabhängig if ly == None: #vom OGR Treiber! raise Exception #arcgis Attributindex Dateien #glob liefert eine liste mit Pfadnamen zurück # 1.PROBLEM: Sind quasi Doppelsuffix # 2.PROBLEM: Gross Kleinschreibung sollte einheitlich sein mit dem # anderen Kopierprozesse!! atxes = glob.glob(self.dsIn.GetName() + '/' + lyrIn.GetName() + ".*.atx") for atx in atxes: if os.path.exists(atx): #Wegen der gleichen Gross/Kleinschreibung wie in # outputname angegeben!!!!!!!! atx_suffix = os.path.splitext(atx) atx_suffix = os.path.splitext( atx_suffix[0])[1] + atx_suffix[1] shutil.copyfile( atx, dsOut.GetName() + '/' + name + atx_suffix) elif driverName == "PostgreSQL" or driverName == "SQLite": #falls noch ein temp Layer existiert #(Beim Aktualisieren ist der name ja XXX_tmp_aktual!) if string.count(name, 'temp_aktual') > 0: if dsOut.GetLayerByName(name) > 0: dsOut.DeleteLayer(name) # Create a memory OGR datasource to put results in. Von frank W. Die Doku ist einfach schwach # Deshalb wirds hier zu Dokumentationszwecken dringelassen: # Der Memory Layer ist deutlich performanter bei CreateFeature als wenn es direkt auf eine # Spatial Lite DB angewandt wird mem_drv = ogr.GetDriverByName('Memory') mem_ds = mem_drv.CreateDataSource('out') ## if lyrIn.GetGeomType() == 100 and driverName == "SQLite": #Also keine Geometrie vorhanden ## if driverName == "PostgreSQL": ## ly = dsOut.CopyLayer(lyrIn,name, ['SPATIAL_INDEX=no','PRECISION=no','GEOMETRY_NAME=the_geom', 'LAUNDER=no']) ## elif driverName == "SQLite": ## ly = dsOut.CopyLayer(lyrIn,name, ['GEOMETRY_NAME=the_geom', 'LAUNDER=no']) if driverName == "SQLite": if lyrIn.GetGeomType( ) == 100: # also KEINE Geometrie Tabelle ly = dsOut.CopyLayer( lyrIn, name, ['GEOMETRY_NAME=the_geom', 'LAUNDER=no']) else: memme = mem_ds.CreateLayer(name, srs, lyrIn.GetGeomType()) i = 0 feldanzahl = lyrIn.GetLayerDefn().GetFieldCount() while i < feldanzahl: Fdefn = lyrIn.GetLayerDefn().GetFieldDefn(i) memme.CreateField(Fdefn) i = i + 1 # Unbedingt die Geometrie vereinheitlichen, im Shape ist das anscheinend # nicht immer sauber und würde sonst nicht in die DB übernommen werden!!! # Mehr Unterscheidungen (der Geometrie) gibt es derzeit noch nicht, also Single point oder line for fea in lyrIn: # None kann vom Shape weg vorkommen, deshalb die Prüfung # Gibts ein Feature ohne Geometrie (das kann beim Shape vorkommen), wirds # ignoriert. if not fea.GetGeometryRef() == None: gemi = fea.GetGeometryRef().Clone() else: next if lyrIn.GetGeomType() == ogr.wkbMultiPoint: gemi = ogr.ForceToMultiPoint(gemi) elif lyrIn.GetGeomType() == ogr.wkbMultiLineString: gemi = ogr.ForceToMultiLineString(gemi) elif lyrIn.GetGeomType() == ogr.wkbMultiPolygon: gemi = ogr.ForceToMultiPolygon(gemi) elif lyrIn.GetGeomType() == ogr.wkbPolygon: gemi = ogr.ForceToPolygon(gemi) else: gemi = gemi.Clone() fea.SetGeometry(gemi) memme.CreateFeature(fea) if memme == None: raise Exception dsOut.CopyLayer(memme, name, [ 'SPATIAL_INDEX=no', 'GEOM_TYPE=geometry', 'GEOMETRY_NAME=the_geom', 'LAUNDER=no' ]) # Einen leeren Layer in der Postgis erzeugen # ACHTUNG: Beim Shape wird zwischen Linestring und Multilinestring etc. # nicht unterschieden, es können also in einem Shape beide Arten vorkommen! # Es muss deshalb per Default immer die Multi- Version erzeugt werden! elif driverName == "PostgreSQL": if lyrIn.GetGeomType() == 100: # Keine Geometrie ly = mem_ds.CreateLayer(name, None, ogr.wkbNone) i = 0 feldanzahl = lyrIn.GetLayerDefn().GetFieldCount() while i < feldanzahl: Fdefn = lyrIn.GetLayerDefn().GetFieldDefn(i) Fdefn.SetName(string.lower(Fdefn.GetName())) ly.CreateField(Fdefn) i = i + 1 # Der gesmate Layer muss neu geschrieben werden # damit Probleme beim Laden in die Datenbank # möglichst ausgeschlossen werden for fea in lyrIn: fea_tmp = ogr.Feature(fea.GetDefnRef()) # sachinformation einfügen i = 0 while i < feldanzahl: if not fea.GetField(i) == None: if fea.GetFieldDefnRef( i).GetType() == 4: # Textfeld if codierung == 'nein' or '': fea_tmp.SetField( i, fea.GetFieldAsString(i)) else: # bei Bedarf umcodieren kasperle = fea.GetFieldAsString( i).decode( codierung, 'replace').encode( 'utf8', 'replace') fea_tmp.SetField(i, kasperle) else: # numerisches Feld fea_tmp.SetField(i, fea.GetField(i)) i = i + 1 # ein neues (konvertiertes) Feature im Memory Layer erzeugen fea_tmp.SetFID(-1) ly.CreateFeature(fea_tmp) ######################################### # Alle Formen von Punktgeometrien ######################################### #Point oder Multipoint elif lyrIn.GetGeomType( ) == ogr.wkbPoint or lyrIn.GetGeomType( ) == ogr.wkbMultiPoint: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPoint) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPoint, 'point', lyrIn, codierung, ly): ly = None #Point oder Multipoint mit M elif lyrIn.GetGeomType( ) == ogr.wkbPointM or lyrIn.GetGeomType( ) == ogr.wkbMultiPointM: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPointM) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPointM, 'point', lyrIn, codierung, ly): ly = None #Point oder Multipoint 3D mit M elif lyrIn.GetGeomType( ) == ogr.wkbPointZM or lyrIn.GetGeomType( ) == ogr.wkbMultiPointZM: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPointZM) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPointZM, 'point', lyrIn, codierung, ly): ly = None #Point oder Multipoint 3D elif lyrIn.GetGeomType( ) == ogr.wkbPoint25D or lyrIn.GetGeomType( ) == ogr.wkbMultiPoint25D: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPoint25D) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPoint25D, 'point', lyrIn, codierung, ly): ly = None ######################################### # Alle Formen von Liniengeometrien ######################################### # Line oder Multiline elif lyrIn.GetGeomType( ) == ogr.wkbLineString or lyrIn.GetGeomType( ) == ogr.wkbMultiLineString: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiLineString) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiLineString, 'line', lyrIn, codierung, ly): ly = None # Line oder Multiline mit M elif lyrIn.GetGeomType( ) == ogr.wkbLineStringM or lyrIn.GetGeomType( ) == ogr.wkbMultiLineStringM: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiLineStringM) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiLineStringM, 'line', lyrIn, codierung, ly): ly = None # Line oder Multiline 3D mit M elif lyrIn.GetGeomType( ) == ogr.wkbLineStringZM or lyrIn.GetGeomType( ) == ogr.wkbMultiLineStringZM: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiLineStringZM) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiLineStringZM, 'line', lyrIn, codierung, ly): ly = None # Linien mit 3D elif lyrIn.GetGeomType( ) == ogr.wkbMultiLineString25D or lyrIn.GetGeomType( ) == ogr.wkbLineString25D: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiLineString25D) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiLineString25D, 'line', lyrIn, codierung, ly): ly = None ######################################### # Alle Formen von Polygongeometrien ######################################### #Polygon oder Multipolygon elif lyrIn.GetGeomType( ) == ogr.wkbPolygon or lyrIn.GetGeomType( ) == ogr.wkbMultiPolygon: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPolygon) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPolygon, 'polygon', lyrIn, codierung, ly): ly = None #Polygon oder Multipolygon mit M elif lyrIn.GetGeomType( ) == ogr.wkbPolygonM or lyrIn.GetGeomType( ) == ogr.wkbMultiPolygonM: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPolygonM) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPolygonM, 'polygon', lyrIn, codierung, ly): ly = None #Polygon oder Multipolygon 3D mit M elif lyrIn.GetGeomType( ) == ogr.wkbPolygonZM or lyrIn.GetGeomType( ) == ogr.wkbMultiPolygonZM: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPolygonZM) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPolygonZM, 'polygon', lyrIn, codierung, ly): ly = None # Polygone mit 3D elif lyrIn.GetGeomType( ) == ogr.wkbMultiPolygon25D or lyrIn.GetGeomType( ) == ogr.wkbPolygon25D: ly = mem_ds.CreateLayer(name, srs, ogr.wkbMultiPolygon25D) # ACHTUNG: return hat mit dem im sub erzeugten Layer Objekt nicht funktioniert (crash) # dh wird das Lyerobjekt ly bereits vorher erzeugt und im Sub (Pointer!) verändert. # zurückgegeben wird nur das fehlerstatement if not self.__create_and_populate_layer( name, srs, ogr.wkbMultiPolygon25D, 'polygon', lyrIn, codierung, ly): ly = None # alles andere else: dsOut.CopyLayer(lyrIn, name, [ 'SPATIAL_INDEX=no', 'PRECISION=NO', 'GEOM_TYPE=geometry', 'GEOMETRY_NAME=the_geom', 'LAUNDER=yes' ]) # Fertig: der Memorylayer mit dem gesäuberten Inhalt des Shapes kann in die Datenbank kopiert werden if not ly == None: dsOut.CopyLayer(ly, name, [ 'SPATIAL_INDEX=no', 'PRECISION=NO', 'GEOM_TYPE=geometry', 'GEOMETRY_NAME=the_geom', 'LAUNDER=yes' ]) else: return [ False, 'Geometrietyp nicht unterstuetzt oder Layer Nachbesserung fehlgeschlagen' ] return [True] # Wir geben die Exception auch mit zurück - damit die fehlermeldung informativer ist! except Exception as e: print str(e) return [False, e]
def clip_gpkg(input, clip): start = time.time() driver = ogr.GetDriverByName("GPKG") inDataSource = driver.Open(input, 0) inLayer = inDataSource.GetLayerByName("一张图") print(inLayer.GetFeatureCount()) ## Clip inClipSource = driver.Open(clip, 0) inClipLayer = inClipSource.GetLayerByName("建设用地") outdriver = ogr.GetDriverByName('FileGDB') output_path = "res/res.gdb" if os.path.exists(output_path): outDataSource = outdriver.Open(output_path, 1) logging.info("文件数据库已存在,在已有数据库基础上创建图层.") else: outDataSource = outdriver.CreateDataSource(output_path) outLayer = outDataSource.CreateLayer("clip", geom_type=ogr.wkbPolygon) # pUnionGeo = loadGeometry(inClipLayer) i = 0 pFeat = inLayer.GetNextFeature() while pFeat is not None: pGeom = pFeat.GetGeometryRef() # inClipLayer.ResetReading() inClipLayer.SetSpatialFilter(pGeom) # pClipGeo = pGeom.Intersection(poClipGeo) multi = ogr.Geometry(ogr.wkbMultiPolygon) for feature in inClipLayer: # pGeo = feature.GetGeometryRef() # pResGeo = pUnionGeo.Intersection(pGeo) # if pResGeo is not None: # print("{}:{}".format(i, pResGeo.GetGeometryCount())) # else: # print(i) pGeo = feature.GetGeometryRef() if pGeo.GetGeometryType() == ogr.wkbMultiPolygon: multi.AddGeometry(ogr.ForceToPolygon(pGeo)) else: multi.AddGeometry(pGeo) pGeo = multi.UnionCascaded() if pGeo is not None: pResGeo = pGeom.Intersection(pGeo) if pResGeo is not None: print("{}:{}".format(i, pResGeo.GetGeometryCount())) else: print(i) pFeat = inLayer.GetNextFeature() i += 1 inDataSource.Destroy() inClipSource.Destroy() end = time.time() print("clip操作总共耗时 " + str(end - start) + "\n")