forked from s_ranjbar/city_retrofit
35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
|
import json
|
||
|
|
from shapely import Polygon
|
||
|
|
from shapely import Point
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
|
||
|
|
def process_geojson(x, y, diff):
|
||
|
|
selection_box = Polygon([[x + diff, y - diff],
|
||
|
|
[x - diff, y - diff],
|
||
|
|
[x - diff, y + diff],
|
||
|
|
[x + diff, y + diff]])
|
||
|
|
geojson_file = Path('./data/collinear_clean 2.geojson').resolve()
|
||
|
|
output_file = Path('./input_files/output_buildings.geojson').resolve()
|
||
|
|
buildings_in_region = []
|
||
|
|
|
||
|
|
with open(geojson_file, 'r') as file:
|
||
|
|
city = json.load(file)
|
||
|
|
buildings = city['features']
|
||
|
|
|
||
|
|
for building in buildings:
|
||
|
|
coordinates = building['geometry']['coordinates'][0]
|
||
|
|
building_polygon = Polygon(coordinates)
|
||
|
|
centroid = Point(building_polygon.centroid)
|
||
|
|
|
||
|
|
if centroid.within(selection_box):
|
||
|
|
buildings_in_region.append(building)
|
||
|
|
|
||
|
|
output_region = {"type": "FeatureCollection",
|
||
|
|
"features": buildings_in_region}
|
||
|
|
|
||
|
|
with open(output_file, 'w') as file:
|
||
|
|
file.write(json.dumps(output_region, indent=2))
|
||
|
|
|
||
|
|
return output_file
|