47 lines
1.8 KiB
Python
47 lines
1.8 KiB
Python
|
import json
|
||
|
|
from shapely.geometry import shape, mapping, MultiPolygon, Polygon
|
||
|
|
from shapely.ops import unary_union
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
def enforce_right_hand_rule(geojson):
|
||
|
|
for feature in geojson['features']:
|
||
|
|
geometry = shape(feature['geometry'])
|
||
|
|
if isinstance(geometry, Polygon):
|
||
|
|
if not geometry.exterior.is_ccw:
|
||
|
|
geometry = Polygon(geometry.exterior.coords[::-1], [interior.coords[::-1] for interior in geometry.interiors])
|
||
|
|
elif isinstance(geometry, MultiPolygon):
|
||
|
|
new_polygons = []
|
||
|
|
for polygon in geometry.geoms: # Use .geoms to iterate over the individual polygons
|
||
|
|
if not polygon.exterior.is_ccw:
|
||
|
|
polygon = Polygon(polygon.exterior.coords[::-1], [interior.coords[::-1] for interior in polygon.interiors])
|
||
|
|
new_polygons.append(polygon)
|
||
|
|
geometry = MultiPolygon(new_polygons)
|
||
|
|
feature['geometry'] = mapping(geometry)
|
||
|
|
return geojson
|
||
|
|
|
||
|
|
data_path = Path(__file__).parent.parent / 'hub/data' # Adjust this path as needed
|
||
|
|
# Load the GeoJSON file
|
||
|
|
input_filepath = data_path / 'test.geojson'
|
||
|
|
output_filepath = data_path / 'test_updated.geojson'
|
||
|
|
|
||
|
|
with open(input_filepath, 'r') as f:
|
||
|
|
geojson_data = json.load(f)
|
||
|
|
|
||
|
|
# Iterate through the features and convert MultiPolygons to Polygons
|
||
|
|
for feature in geojson_data['features']:
|
||
|
|
geom = shape(feature['geometry'])
|
||
|
|
if isinstance(geom, MultiPolygon):
|
||
|
|
# Merge the polygons into a single polygon
|
||
|
|
merged_polygon = unary_union(geom)
|
||
|
|
# Convert the merged polygon back to GeoJSON format
|
||
|
|
feature['geometry'] = mapping(merged_polygon)
|
||
|
|
|
||
|
|
# Enforce the right-hand rule
|
||
|
|
geojson_data = enforce_right_hand_rule(geojson_data)
|
||
|
|
|
||
|
|
# Save the updated GeoJSON file
|
||
|
|
with open(output_filepath, 'w') as f:
|
||
|
|
json.dump(geojson_data, f)
|
||
|
|
|
||
|
|
|