def test_ogr_factory_5(): src_wkt = 'LINESTRING(2 5,10 20)' exp_wkt = 'MULTILINESTRING((2 5,10 20))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt() src_wkt = 'GEOMETRYCOLLECTION(LINESTRING(2 5,10 20),LINESTRING(0 0,10 10))' exp_wkt = 'MULTILINESTRING((2 5,10 20),(0 0,10 10))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt() src_wkt = 'POLYGON((2 5,10 20),(0 0,10 10))' exp_wkt = 'MULTILINESTRING((2 5,10 20),(0 0,10 10))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt() src_wkt = 'MULTIPOLYGON(((2 5,10 20),(0 0,10 10)),((2 5,10 20)))' exp_wkt = 'MULTILINESTRING((2 5,10 20),(0 0,10 10),(2 5,10 20))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) assert not ogrtest.check_feature_geometry(dst_geom, exp_wkt), dst_geom.ExportToWkt()
def features(self, data): for objekt in data['geometricke_objekty']: f = ogr.Feature(feature_def=self.vrstva.GetLayerDefn()) atributy = data['atributy'] atributy.update(objekt.get('atributy', {})) for atribut, hodnota in atributy.iteritems(): if isinstance(hodnota, tuple): # datetime hodnota f.SetField(atribut, *hodnota) else: #SM >>> # if isinstance(hodnota, basestring): # f.SetField(atribut, hodnota.encode(self.kodovanie)) # else: #SM <<< f.SetField(atribut, hodnota) wkt = objekt['wkt'] geom = ogr.CreateGeometryFromWkt(wkt) if geom.GetGeometryType() == ogr.wkbLineString: geom = ogr.ForceToMultiLineString(geom) elif geom.GetGeometryType() == ogr.wkbPolygon: geom = ogr.ForceToMultiPolygon(geom) f.SetGeometryDirectly(geom) yield f
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 __init__(self, geom_in): if type(geom_in) is str: self._geom = ogr.CreateGeometryFromWkt(geom_in) elif type(geom_in) is ogr.Geometry: self._geom = geom_in else: raise RuntimeError("Unknown geometry Syntax!") if self._geom.GetGeometryType() != 3: raise RuntimeError("Geometry is not a POLYGON!") # initialize attributes self._area = self._geom.Area() self._perim = ogr.ForceToMultiLineString(self._geom).Length() xmin, xmax, ymin, ymax = self._geom.GetEnvelope() dx = xmax - xmin dy = ymax - ymin self._bbperm = 2.0*(dx + dy) self._bbarea = dx * dy #self._circle = getCircle(self._geom.GetGeometryRef(0).GetPoints()) # we hit python recursion limits on big polygons, so # changed this to use the convex hull of the polygon which # will give the same result with far less points # when computing the minimum bounding circle of the polygon chull = self._geom.ConvexHull() chullref = chull.GetGeometryRef(0) chullpts = chullref.GetPoints() self._circle = getCircle(chullpts)
def __create_and_populate_layer(self, name, srs, ogr_type, geom_type, lyrIn, codierung, ly_tmp): try: i = 0 feldanzahl = lyrIn.GetLayerDefn().GetFieldCount() while i < feldanzahl: Fdefn = lyrIn.GetLayerDefn().GetFieldDefn(i) Fdefn.SetName(string.lower(Fdefn.GetName())) ly_tmp.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: # Umwandeln der Geometrie in ein MultiLinestring gemi = fea.GetGeometryRef() if geom_type == 'point': gemi = ogr.ForceToMultiPoint(gemi) if geom_type == 'line': gemi = ogr.ForceToMultiLineString(gemi) if geom_type == 'polygon': gemi = ogr.ForceToMultiPolygon(gemi) fea_tmp = ogr.Feature(fea.GetDefnRef()) # gemoetrie einfügen (die geänderte) fea_tmp.SetGeometry(gemi) # 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_tmp.CreateFeature(fea_tmp) #Layer erfolgreich erstellt return True # Layer nicht erfolgreich erstellt! except: return False
def _perimeter(geometry): #### with holes? """Compute the perimeter of the polygon. >>> g = ogr.CreateGeometryFromWkt('POLYGON ((0 0,1 0,1 1,0 1,0 0))') >>> _perimeter(g) 4.0 """ return ogr.ForceToMultiLineString(geometry).Length()
def forceOgrMulti(output, geom_type): if geom_type == "Point": output = ogr.ForceToMultiPolygon(output) elif geom_type == "LineString": output = ogr.ForceToMultiLineString(output) elif geom_type == "Polygon": output = ogr.ForceToMultiPolygon(output) return output
def geometry_force_multi(ogr_geom): geom_type = ogr_geom.GetGeometryType() if geom_type == ogr.wkbPoint: return ogr.ForceToMultiPoint(ogr_geom) if geom_type == ogr.wkbLineString: return ogr.ForceToMultiLineString(ogr_geom) if geom_type == ogr.wkbPolygon: return ogr.ForceToMultiPolygon(ogr_geom) return ogr_geom
def points2ogr_polygon(points): """Construct a OGR polygon from an input point list (not closed)""" # input an iterable of 2d 'points', slow interface for large collections... s = ogr.Geometry(ogr.wkbLineString) for p in points: s.AddPoint_2D(p[0], p[1]) s.AddPoint_2D(points[0][0], points[0][1]) # close p = ogr.BuildPolygonFromEdges(ogr.ForceToMultiLineString(s)) return p
def ogr_factory_5(): src_wkt = 'LINESTRING(2 5,10 20)' exp_wkt = 'MULTILINESTRING((2 5,10 20))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' src_wkt = 'GEOMETRYCOLLECTION(LINESTRING(2 5,10 20),LINESTRING(0 0,10 10))' exp_wkt = 'MULTILINESTRING((2 5,10 20),(0 0,10 10))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' src_wkt = 'POLYGON((2 5,10 20),(0 0,10 10))' exp_wkt = 'MULTILINESTRING((2 5,10 20),(0 0,10 10))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' src_wkt = 'MULTIPOLYGON(((2 5,10 20),(0 0,10 10)),((2 5,10 20)))' exp_wkt = 'MULTILINESTRING((2 5,10 20),(0 0,10 10),(2 5,10 20))' src_geom = ogr.CreateGeometryFromWkt(src_wkt) dst_geom = ogr.ForceToMultiLineString(src_geom) if ogrtest.check_feature_geometry(dst_geom, exp_wkt): print(dst_geom.ExportToWkt()) return 'fail' return 'success'
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 load_from_ogr(self, ogrlayer, strdecode): source_osr = ogrlayer.GetSpatialRef() target_osr = osr.SpatialReference() target_osr.ImportFromEPSG(self.srs_id) transform = osr.CoordinateTransformation(source_osr, target_osr) feature = ogrlayer.GetNextFeature() fid = 0 while feature: fid += 1 geom = feature.GetGeometryRef() if geom.GetGeometryType() == ogr.wkbPoint: geom = ogr.ForceToMultiPoint(geom) elif geom.GetGeometryType() == ogr.wkbLineString: geom = ogr.ForceToMultiLineString(geom) elif geom.GetGeometryType() == ogr.wkbPolygon: geom = ogr.ForceToMultiPolygon(geom) geom.Transform(transform) fld_values = dict() for i in range(feature.GetFieldCount()): fld_type = feature.GetFieldDefnRef(i).GetType() fld_value = None if fld_type == ogr.OFTInteger: fld_value = feature.GetFieldAsInteger(i) elif fld_type == ogr.OFTReal: fld_value = feature.GetFieldAsDouble(i) elif fld_type == ogr.OFTString: fld_value = strdecode(feature.GetFieldAsString(i)) fld_values[self[feature.GetFieldDefnRef( i).GetNameRef()].key] = fld_value obj = self.model(fid=fid, geom=ga.WKTSpatialElement(str(geom), self.srs_id), **fld_values) DBSession.add(obj) feature = ogrlayer.GetNextFeature()
def _create_polygon(self, geom, i, z): """Create polygon from closed linestring geometry. If input geometry is not closed linestring that PolygonError is raised. :param geom: input linestring geometry :param line_id: ID of an input geometry :return: polygon geometry """ if geom == []: pass else: ring = ogr.ForceToMultiLineString(geom) poly = ogr.BuildPolygonFromEdges(ring, 0, 1) # Add polygon to polygon_list new_poly = MyPolygon(i, z, poly) self.polygon_list.append(new_poly)
def getPoints(self): """Return a list of points from a polygon geometry""" gg = ogr.ForceToMultiLineString(self._geom) return gg.GetGeometryRef(0).GetPoints()
line = ogr.Geometry(ogr.wkbLineString) layer = dataSourceIn.GetLayer('loc') #nom de la table for feature in layer: line.AddPoint(feature.GetField("LONG"), feature.GetField("LAT"), feature.GetField("ALT")) print line.ExportToWkt() #exporter line en gpx -> OK #multilinestring https://pcjericks.github.io/py-gdalogr-cookbook/geometry.html#create-a-multilinestring if dataSourceOut is None: print 'Could not get datasourceout' else: print 'datasourceout ok' multiline = ogr.ForceToMultiLineString(line) print multiline.ExportToWkt() outLayer = dataSourceOut.CreateLayer("un_nom", geom_type=ogr.wkbMultiLineString) outLayerDefn = outLayer.GetLayerDefn() outFeature = ogr.Feature(outLayerDefn) outFeature.SetGeometry(multiline) outLayer.CreateFeature(outFeature) outFeature = None dataSourceIn = None dataSourceOut = None
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 load_from_ogr(self, ogrlayer, strdecode): source_osr = ogrlayer.GetSpatialRef() target_osr = osr.SpatialReference() target_osr.ImportFromEPSG(self.srs_id) transform = osr.CoordinateTransformation(source_osr, target_osr) is_multi = self.geometry_type in GEOM_TYPE.is_multi has_z = self.geometry_type in GEOM_TYPE.has_z for fid, feature in enumerate(ogrlayer): geom = feature.GetGeometryRef() gtype = geom.GetGeometryType() if gtype not in GEOM_TYPE_OGR: raise ValidationError( _("Unknown geometry type: %d (%s).") % (gtype, ogr.GeometryTypeToName(gtype))) geom.Transform(transform) # Force single geometries to multi if is_multi: if gtype in (ogr.wkbPoint, ogr.wkbPoint25D): geom = ogr.ForceToMultiPoint(geom) elif gtype in (ogr.wkbLineString, ogr.wkbLineString25D): geom = ogr.ForceToMultiLineString(geom) elif gtype in (ogr.wkbPolygon, ogr.wkbPolygon25D): geom = ogr.ForceToMultiPolygon(geom) # Force Z if has_z and not geom.Is3D(): geom.Set3D(True) # Check geometry type again gtype = geom.GetGeometryType() if gtype not in GEOM_TYPE_OGR or _GEOM_OGR_2_TYPE[ gtype] != self.geometry_type: raise ValidationError( _("Unknown geometry type: %d (%s).") % (gtype, ogr.GeometryTypeToName(gtype))) fld_values = dict() for i in range(feature.GetFieldCount()): fld_type = feature.GetFieldDefnRef(i).GetType() if (not feature.IsFieldSet(i) or feature.IsFieldNull(i)): fld_value = None elif fld_type == ogr.OFTInteger: fld_value = feature.GetFieldAsInteger(i) elif fld_type == ogr.OFTInteger64: fld_value = feature.GetFieldAsInteger64(i) elif fld_type == ogr.OFTReal: fld_value = feature.GetFieldAsDouble(i) elif fld_type == ogr.OFTDate: year, month, day = feature.GetFieldAsDateTime(i)[0:3] fld_value = date(year, month, day) elif fld_type == ogr.OFTTime: hour, minute, second = feature.GetFieldAsDateTime(i)[3:6] fld_value = time(hour, minute, int(second)) elif fld_type == ogr.OFTDateTime: year, month, day, hour, minute, second, tz = \ feature.GetFieldAsDateTime(i) fld_value = datetime(year, month, day, hour, minute, int(second)) elif fld_type == ogr.OFTIntegerList: fld_value = json.dumps(feature.GetFieldAsIntegerList(i)) elif fld_type == ogr.OFTInteger64List: fld_value = json.dumps(feature.GetFieldAsInteger64List(i)) elif fld_type == ogr.OFTRealList: fld_value = json.dumps(feature.GetFieldAsDoubleList(i)) elif fld_type == ogr.OFTStringList: # TODO: encoding fld_value = json.dumps(feature.GetFieldAsStringList(i)) elif fld_type == ogr.OFTString: try: fld_value = strdecode(feature.GetFieldAsString(i)) except UnicodeDecodeError: raise ValidationError( _("It seems like declared and actual attributes " "encodings do not match. Unable to decode " "attribute #%(attr)d of feature #%(feat)d. " "Try declaring different encoding.") % dict(feat=fid, attr=i)) fld_name = strdecode(feature.GetFieldDefnRef(i).GetNameRef()) fld_values[self[fld_name].key] = fld_value obj = self.model(fid=fid, geom=ga.elements.WKBElement(bytearray( geom.ExportToWkb()), srid=self.srs_id), **fld_values) DBSession.add(obj)
def load_from_ogr(self, ogrlayer, strdecode): source_osr = ogrlayer.GetSpatialRef() target_osr = osr.SpatialReference() target_osr.ImportFromEPSG(self.srs_id) ltype = ogrlayer.GetGeomType() & (~ogr.wkb25DBit) transform = osr.CoordinateTransformation(source_osr, target_osr) for fid, feature in enumerate(ogrlayer): geom = feature.GetGeometryRef() # Bring 25D geometries to 2D if geom.GetGeometryType() & ogr.wkb25DBit: geom.FlattenTo2D() gtype = geom.GetGeometryType() if gtype not in self.accepted_gtype: raise ValidationError( _("Geometry type (%s) does not match column type (%s).") % (GEOM_TYPE_DISPLAY[gtype - 1], GEOM_TYPE_DISPLAY[ltype - 1])) geom.Transform(transform) if (self.geometry_type in (GEOM_TYPE.MULTIPOINT, GEOM_TYPE.MULTILINESTRING, GEOM_TYPE.MULTIPOLYGON)): if gtype == ogr.wkbPoint: geom = ogr.ForceToMultiPoint(geom) elif gtype == ogr.wkbLineString: geom = ogr.ForceToMultiLineString(geom) elif gtype == ogr.wkbPolygon: geom = ogr.ForceToMultiPolygon(geom) fld_values = dict() for i in range(feature.GetFieldCount()): fld_type = feature.GetFieldDefnRef(i).GetType() if (not feature.IsFieldSet(i) or (gdal_gt_22 and feature.IsFieldNull(i))): fld_value = None elif fld_type == ogr.OFTInteger: fld_value = feature.GetFieldAsInteger(i) elif gdal_gt_20 and fld_type == ogr.OFTInteger64: fld_value = feature.GetFieldAsInteger64(i) elif fld_type == ogr.OFTReal: fld_value = feature.GetFieldAsDouble(i) elif fld_type == ogr.OFTDate: year, month, day = feature.GetFieldAsDateTime(i)[0:3] fld_value = date(year, month, day) elif fld_type == ogr.OFTTime: hour, minute, second = feature.GetFieldAsDateTime(i)[3:6] fld_value = time(hour, minute, int(second)) elif fld_type == ogr.OFTDateTime: year, month, day, hour, minute, second, tz = \ feature.GetFieldAsDateTime(i) fld_value = datetime(year, month, day, hour, minute, int(second)) elif fld_type == ogr.OFTIntegerList: fld_value = json.dumps(feature.GetFieldAsIntegerList(i)) elif gdal_gt_20 and fld_type == ogr.OFTInteger64List: fld_value = json.dumps(feature.GetFieldAsInteger64List(i)) elif fld_type == ogr.OFTRealList: fld_value = json.dumps(feature.GetFieldAsDoubleList(i)) elif fld_type == ogr.OFTStringList: # TODO: encoding fld_value = json.dumps(feature.GetFieldAsStringList(i)) elif fld_type == ogr.OFTString: try: fld_value = strdecode(feature.GetFieldAsString(i)) except UnicodeDecodeError: raise ValidationError( _("It seems like declared and actual attributes " "encodings do not match. Unable to decode " "attribute #%(attr)d of feature #%(feat)d. " "Try declaring different encoding.") % dict(feat=fid, attr=i)) fld_name = strdecode(feature.GetFieldDefnRef(i).GetNameRef()) fld_values[self[fld_name].key] = fld_value obj = self.model(fid=fid, geom=ga.elements.WKTElement(str(geom), srid=self.srs_id), **fld_values) DBSession.add(obj)