chore: changing building_id to building_name in dhn generator

This commit is contained in:
Majid Rezaei 2024-07-31 11:27:52 -04:00
parent d907b21a92
commit d363efd927

View File

@ -71,14 +71,14 @@ class DistrictHeatingNetworkCreator:
city = json.load(file)
self.centroids = []
self.building_ids = []
self.building_names = []
buildings = city['features']
for building in buildings:
coordinates = building['geometry']['coordinates'][0]
building_polygon = Polygon(coordinates)
centroid = building_polygon.centroid
self.centroids.append(centroid)
self.building_ids.append(building['id'])
self.building_names.append(building['id'])
# Load road data
with open(self.roads_file, 'r') as file:
@ -281,8 +281,8 @@ class DistrictHeatingNetworkCreator:
try:
for i, centroid in enumerate(self.centroids):
centroid_tuple = (centroid.x, centroid.y)
building_id = self.building_ids[i]
self.final_mst.add_node(centroid_tuple, type='building', id=building_id)
building_name = self.building_names[i]
self.final_mst.add_node(centroid_tuple, type='building', id=building_name)
nearest_point = None
min_distance = float('inf')
for node in self.final_mst.nodes():