128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
from District import District
|
|
from Position import Position
|
|
from PIL import Image
|
|
import random
|
|
|
|
|
|
class City:
|
|
"""
|
|
Attributes:
|
|
districts (list): The list of districts in the city.
|
|
map_data (list): The 2D list representing the map of the city.
|
|
height_map (list): The 2D list representing the height map of the city.
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""
|
|
The constructor for the City class.
|
|
"""
|
|
self.districts = []
|
|
self.map_data = []
|
|
self.height_map = []
|
|
self.init_maps()
|
|
|
|
def init_maps(self):
|
|
"""
|
|
Initialize the maps of the city. It reads the heightmap and watermap images and converts them into 2D lists.
|
|
"""
|
|
heightmap = Image.open('./data/heightmap.png').convert('L')
|
|
watermap = Image.open('./data/watermap.png').convert('L')
|
|
width, height = heightmap.size
|
|
self.map_data = [[-1 if watermap.getpixel((x, y)) > 0 else 0 for x in range(width)] for y in range(height)]
|
|
self.height_map = [[heightmap.getpixel((x, y)) for x in range(width)] for y in range(height)]
|
|
watermap.close()
|
|
heightmap.close()
|
|
|
|
def add_district(self, center: Position, district_type: str = ""):
|
|
"""
|
|
Add a new district to the city.
|
|
|
|
:param district_type:
|
|
:param center: The center position of the new district.
|
|
"""
|
|
self.districts.append(District(len(self.districts) + 1, center, district_type))
|
|
self.map_data[center.y][center.x] = len(self.districts)
|
|
|
|
def is_expend_finished(self):
|
|
"""
|
|
Check if the expansion of all districts in the city is finished.
|
|
|
|
:return: True if the expansion is finished, False otherwise.
|
|
"""
|
|
for district in self.districts:
|
|
if len(district.area_expend_from_point) > 0:
|
|
return False
|
|
return True
|
|
|
|
def choose_expend_point(self, point: Position, index_district: int):
|
|
"""
|
|
Choose a point to expand a district based on the distance between the center of the district and the point itself.
|
|
|
|
:param point: The point to be expanded.
|
|
:param index_district: The index of the district to be expanded.
|
|
"""
|
|
min_distance = point.distance_to(self.districts[index_district].center_expend)
|
|
index_district_chosen = index_district
|
|
for i in range(index_district + 1, len(self.districts)):
|
|
if point in self.districts[i].area_expend:
|
|
distance = point.distance_to(self.districts[i].center_expend)
|
|
if distance < min_distance:
|
|
min_distance = distance
|
|
self.districts[index_district_chosen].area_expend.remove(point)
|
|
index_district_chosen = i
|
|
else:
|
|
self.districts[i].area_expend.remove(point)
|
|
self.districts[index_district_chosen].area.append(point)
|
|
self.districts[index_district_chosen].area_expend_from_point.append(point)
|
|
self.districts[index_district_chosen].area_expend.remove(point)
|
|
self.map_data[point.y][point.x] = index_district_chosen + 1
|
|
|
|
def update_expend_district(self):
|
|
"""
|
|
Update the expansion points of all districts in the city.
|
|
"""
|
|
|
|
for district in self.districts:
|
|
if len(district.area_expend_from_point) > 0:
|
|
district.update_expend_points(district.area_expend_from_point[0], self.map_data, self.height_map)
|
|
for district in self.districts:
|
|
for point in district.area_expend:
|
|
self.choose_expend_point(point, district.tile_id - 1)
|
|
|
|
def loop_expend_district(self):
|
|
"""
|
|
Loop the expansion of all districts in the city until all districts are fully expanded.
|
|
"""
|
|
loop_count = 0
|
|
while not self.is_expend_finished():
|
|
self.update_expend_district()
|
|
loop_count += 1
|
|
if loop_count % 100 == 0:
|
|
print("[City] Loop count: ", loop_count)
|
|
|
|
def custom_district_draw_map(self):
|
|
"""
|
|
Draw the map of the city with different colors for each district.
|
|
"""
|
|
width, height = len(self.map_data[0]), len(self.map_data)
|
|
img = Image.new('RGB', (width, height))
|
|
colors = {i: (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255))
|
|
for i in range(1, len(self.districts) + 1)}
|
|
|
|
for y in range(height):
|
|
for x in range(width):
|
|
if self.map_data[y][x] <= 0:
|
|
img.putpixel((x, y), (0, 0, 0))
|
|
else:
|
|
img.putpixel((x, y), colors[self.map_data[y][x]])
|
|
|
|
img.save('./data/custom_district.png')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
city = City()
|
|
for i in range(10):
|
|
city.add_district(Position(random.randint(0, 400), random.randint(0, 400)))
|
|
city.loop_expend_district()
|
|
city.custom_district_draw_map()
|