Update main2.py

main2.py

@@ -0,0 +1,214 @@
+import random
+from math import exp, sqrt
+
+from gdpc import Editor, Block, geometry, Transform
+
+from House import *
+from networks.geometry.Point3D import Point3D
+from networks.roads_2.Road import Road
+from world_maker.data_analysis import transpose_form_heightmap
+from world_maker.District import Road as Road_grid
+from world_maker.Skeleton import Skeleton, simplify_coordinates
+from world_maker.terraforming import remove_trees, smooth_terrain
+from world_maker.world_maker import world_maker
+from networks.geometry.Point3D import Point3D
+from networks.geometry.Point2D import Point2D
+from networks.geometry.Circle import Circle
+
+from PIL import Image as img
+from PIL.Image import Image
+from utils.JsonReader import JsonReader
+from utils.YamlReader import YamlReader
+from buildings.Building import Building
+
+from utils.functions import *
+from utils.Enums import DIRECTION
+import time
+
+
+def main():
+    start_time_all = time.time()
+    start_time = time.time()
+    rectangle_house_mountain, rectangle_building, skeleton_highway, skeleton_mountain, road_grid = world_maker()
+    time_world_maker = time.time() - start_time
+    print(f"[TIME] World_maker {time_world_maker}")
+    editor = Editor(host="http://gdmc.ale-pri.com:9000", buffering=True)
+    buildArea = editor.getBuildArea()
+    origin = ((buildArea.begin).x, (buildArea.begin).z)
+    center = (abs(buildArea.begin.x - buildArea.end.x) / 2,
+              abs(buildArea.begin.z - buildArea.end.z) / 2)
+    length_world = sqrt((center[0]*2) ** 2 + (center[1]*2) ** 2)
+
+    start_time = time.time()
+    remove_trees('./world_maker/data/heightmap.png', './world_maker/data/treemap.png',
+                 './world_maker/data/smooth_sobel_watermap.png')
+    time_remove_tree = time.time() - start_time
+    print(f"[TIME] Remove tree {time_remove_tree}")
+
+    """
+    start_time = time.time()
+    smooth_terrain('./world_maker/data/heightmap.png',
+                   './world_maker/data/heightmap_smooth.png', './world_maker/data/smooth_sobel_watermap.png')
+    time_smooth_terrain = time.time() - start_time
+    print(f"[TIME] Smooth terrain {time_smooth_terrain}")
+
+    start_time = time.time()
+    set_roads(skeleton_highway, origin)
+    set_roads(skeleton_mountain, origin)
+    time_roads = time.time() - start_time
+    print(f"[TIME] Roads {time_roads}")
+    # set_roads_grids(road_grid, origin)
+    # roads.setRoads(skeleton_mountain)
+    # roads.setRoads(skeleton_highway)
+
+    blocks = {
+        "wall": "blackstone",
+        "roof": "blackstone",
+        "roof_slab": "blackstone_slab",
+        "door": "oak_door",
+        "window": "glass_pane",
+        "entrance": "oak_door",
+        "stairs": "quartz_stairs",
+        "stairs_slab": "quartz_slab",
+        "celling": "quartz_block",
+        "floor": "quartz_block",
+        "celling_slab": "quartz_slab",
+        "garden_outline": "oak_leaves",
+        "garden_floor": "grass_block"
+    }
+
+    entranceDirection = ["N", "S", "E", "W"]
+
+    # get every differents buildings shapes
+    f = JsonReader('./buildings/shapes.json')
+    shapes = f.data
+    baseShape = shapes[0]['matrice']
+
+    # get the random data for the buildings
+    y = YamlReader('params.yml')
+    random_data = y.data
+    # create a building at the relative position 0,0 with 20 blocks length and 20 blocks width, with a normal shape and 10 floors
+
+    # build it with your custom materials
+
+    start_time = time.time()
+    for buildings in rectangle_building:
+        height = get_height_building_from_center(
+            center, (buildings[0][0], buildings[0][2]), length_world)
+        start = (min(buildings[0][0], buildings[1][0]) + origin[0], buildings[0]
+                 [1], min(buildings[0][2], buildings[1][2]) + origin[1])
+        end = (max(buildings[0][0], buildings[1][0]) + origin[0],
+               buildings[1]
+               [1]+height, max(buildings[0][2], buildings[1][2]) + origin[1])
+
+        building = Building(random_data["buildings"], [
+            start, end], baseShape, DIRECTION.EAST)
+        building.build(editor, ["stone_bricks", "glass_pane", "glass", "cobblestone_wall", "stone_brick_stairs",
+                                "oak_planks", "white_concrete", "cobblestone", "stone_brick_slab", "iron_bars"])
+
+    time_buildings = time.time() - start_time
+    print(f"[TIME] Buildings {time_buildings}")
+
+    start_time = time.time()
+    for buildings in rectangle_house_mountain:
+        start = (buildings[0][0] + origin[0], buildings[0]
+                 [1], buildings[0][2] + origin[1])
+        end = (buildings[1][0] + origin[0], buildings[1]
+               [1], buildings[1][2] + origin[1])
+        house = House(editor, start, end,
+                      entranceDirection[random.randint(0, 3)], blocks)
+        house.build()
+    time_houses = time.time() - start_time
+    print(f"[TIME] Houses {time_houses}") """
+
+    print("[GDMC] Done!\n\n")
+
+    print(f"[TIME] Total {time.time() - start_time_all}")
+    print(f"[TIME] World_maker {time_world_maker}")
+    print(f"[TIME] Remove tree {time_remove_tree}")
+    #print(f"[TIME] Smooth terrain {time_smooth_terrain}")
+    #print(f"[TIME] Roads {time_roads}")
+    #print(f"[TIME] Buildings {time_buildings}")
+    #print(f"[TIME] Houses {time_houses}")
+
+
+def get_height_building_from_center(center, position, length_world):
+    length = abs(
+        sqrt(((center[0] - position[0]) ** 2 + (center[1] - position[1]) ** 2)))
+    print(length, length_world)
+    return int(exp(-(length / (length_world / 4)) ** 2) * 75 + 30)
+
+
+def set_roads_grids(road_grid: Road_grid, origin):
+    for i in range(len(road_grid)):
+        if road_grid[i].border:
+            for j in range(len(road_grid)):
+                # Same line
+                if (road_grid[i].position.x == road_grid[j].position.x and road_grid[i].position.y != road_grid[
+                    j].position.y) or (
+                        road_grid[i].position.x != road_grid[j].position.x and road_grid[i].position.y == road_grid[
+                            j].position.y):
+                    point_1 = transpose_form_heightmap(
+                        './world_maker/data/heightmap.png', (road_grid[i].position.x, road_grid[i].position.y), origin)
+                    point_2 = transpose_form_heightmap(
+                        './world_maker/data/heightmap.png', (road_grid[j].position.x, road_grid[j].position.y), origin)
+                    Road(
+                        [Point3D(point_1[0], point_1[1], point_1[2]), Point3D(point_2[0], point_2[1], point_2[2])], 9)
+
+
+def set_roads(skeleton: Skeleton, origin):
+    # Parsing
+    print("[Roads] Start parsing...")
+    for i in range(len(skeleton.lines)):
+        print(f"[Roads] Parsing skeleton {i + 1}/{len(skeleton.lines)}.")
+        for j in range(len(skeleton.lines[i])):
+            xyz = transpose_form_heightmap('./world_maker/data/heightmap.png',
+                                           skeleton.coordinates[skeleton.lines[i][j]], origin)
+            heightmap_smooth = img.open(
+                './world_maker/data/road_heightmap.png')
+            skeleton.lines[i][j] = [xyz[0], heightmap_smooth.getpixel(
+                (skeleton.coordinates[skeleton.lines[i][j]][0], skeleton.coordinates[skeleton.lines[i][j]][-1])), xyz[2]]
+
+    print("[Roads] Start simplification...")
+    # Simplification
+    for i in range(len(skeleton.lines)):
+        print(f"[Roads] Simplify skelton {i + 1}/{len(skeleton.lines)}")
+        print(f"[Roads] Number of points: {len(skeleton.lines[i])}")
+        skeleton.lines[i] = simplify_coordinates(skeleton.lines[i], 20)
+        j = 0
+        while j < len(skeleton.lines[i])-1:
+            print(f"[Distance] {Point3D(skeleton.lines[i][j][0], skeleton.lines[i][j][1], skeleton.lines[i][j][2]).distance(Point3D(skeleton.lines[i][j+1][0], skeleton.lines[i][j+1][1], skeleton.lines[i][j+1][2]))}")
+            if Point3D(skeleton.lines[i][j][0], skeleton.lines[i][j][1], skeleton.lines[i][j][2]).distance(Point3D(skeleton.lines[i][j+1][0], skeleton.lines[i][j+1][1], skeleton.lines[i][j+1][2])) <= 20:
+                print(skeleton.lines[i][j+1], skeleton.lines[i][j])
+                del skeleton.lines[i][j+1]
+                print("[Roads] Delete point to close")
+            j += 1
+        j = 0
+        while j < len(skeleton.lines[i])-1:
+            print(f"[Distance] {Point3D(skeleton.lines[i][j][0], skeleton.lines[i][j][1], skeleton.lines[i][j][2]).distance(Point3D(skeleton.lines[i][j+1][0], skeleton.lines[i][j+1][1], skeleton.lines[i][j+1][2]))}")
+            if Point3D(skeleton.lines[i][j][0], skeleton.lines[i][j][1], skeleton.lines[i][j][2]).distance(Point3D(skeleton.lines[i][j+1][0], skeleton.lines[i][j+1][1], skeleton.lines[i][j+1][2])) <= 10:
+                print(skeleton.lines[i][j+1], skeleton.lines[i][j])
+                del skeleton.lines[i][j+1]
+                print("[Roads] Delete point to close")
+            j += 1
+        j = 0
+        while j < len(skeleton.lines[i])-1:
+            print(f"[Distance] {Point3D(skeleton.lines[i][j][0], skeleton.lines[i][j][1], skeleton.lines[i][j][2]).distance(Point3D(skeleton.lines[i][j+1][0], skeleton.lines[i][j+1][1], skeleton.lines[i][j+1][2]))}")
+            if Point3D(skeleton.lines[i][j][0], skeleton.lines[i][j][1], skeleton.lines[i][j][2]).distance(Point3D(skeleton.lines[i][j+1][0], skeleton.lines[i][j+1][1], skeleton.lines[i][j+1][2])) <= 10:
+                print(skeleton.lines[i][j+1], skeleton.lines[i][j])
+                del skeleton.lines[i][j+1]
+                print("[Roads] Delete point to close")
+            j += 1
+        print(
+            f"[Roads] Number of points after simplification: {len(skeleton.lines[i])}")
+
+    print("[Roads] Start generation...")
+    for i in range(len(skeleton.lines)):
+        print(f"[Roads] Generating roads {i + 1}/{len(skeleton.lines)}.")
+        if len(skeleton.lines[i]) >= 4:
+            Road(Point3D.from_arrays(skeleton.lines[i]), 9)
+            print(f"[ROAD] Points: {skeleton.lines[i]}")
+
+
+if __name__ == '__main__':
+    main()