From 62295b5c80d1c2f4778eac51de6e6abdc083002d Mon Sep 17 00:00:00 2001 From: changliao1025 Date: Fri, 28 Jul 2023 17:26:32 -0700 Subject: [PATCH] fix a function error gdal geometry type --- .../external/pyearth/gis/gdal/gdal_functions.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/pyflowline/external/pyearth/gis/gdal/gdal_functions.py b/pyflowline/external/pyearth/gis/gdal/gdal_functions.py index 9a25cff..85143de 100644 --- a/pyflowline/external/pyearth/gis/gdal/gdal_functions.py +++ b/pyflowline/external/pyearth/gis/gdal/gdal_functions.py @@ -470,12 +470,14 @@ def read_mesh_boundary(sFilename_boundary_in): def get_geometry_coords(geometry): - if geometry.GetGeometryType() == ogr.wkbPolygon: - return get_polygon_exterior_coords(geometry) - elif geometry.GetGeometryType() == ogr.wkbLineString: - return get_linestring_coords(geometry) - elif geometry.GetGeometryType() == ogr.wkbPoint: + + sGeometry_type = geometry.GetGeometryName() + if sGeometry_type =='POINT': return get_point_coords(geometry) + elif sGeometry_type == 'LINESTRING': + return get_linestring_coords(geometry) + elif sGeometry_type =='POLYGON': + return get_polygon_exterior_coords(geometry) else: raise ValueError("Unsupported geometry type.")