Files
GDMC-2024/main.py
2024-06-16 00:24:39 +02:00

469 lines
17 KiB
Python

from networks.roads_2.Roads import Road
import json
import random
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
from gdpc import Block, Editor, geometry
from PIL import Image, ImageDraw
import networks.geometry.curve_tools as curve_tools
import networks.geometry.segment_tools as segment_tools
import networks.geometry.Strip as Strip
import networks.roads.lanes.Lane as Lane
import networks.roads.lines.Line as Line
from buildings.Building import Building
from Enums import LINE_OVERLAP, LINE_THICKNESS_MODE, ROTATION
from networks.geometry.Circle import Circle
from networks.geometry.Point2D import Point2D
from networks.geometry.Point3D import Point3D
from networks.geometry.point_tools import (
curved_corner_by_curvature,
curved_corner_by_distance,
)
from networks.geometry.Polyline import Polyline
from networks.geometry.Segment2D import Segment2D
from networks.geometry.Segment3D import Segment3D
# from networks.roads import Road as Road
from networks.roads.intersections import Intersection as Intersection
matplotlib.use('Agg')
editor = Editor(buffering=True)
# f = open('buildings\shapes.json')
# shapes = json.load(f)
# # F = Foundations((0,0), (20,20), shapes[0]['matrice'])
# # F.polygon.fill_polygon(editor, "stone", -60)
# geometry.placeCuboid(editor, (-10, -60, -10), (85, -55, 85), Block("air"))
# B = Building((0, 0), (75, 75), shapes[7]['matrice'])
# B.foundations.polygon.fill_vertice(editor, "pink_wool", -60)
# for collumn in B.foundations.collumns:
# collumn.fill(editor, "white_concrete", -60, -55)
# B.foundations.polygon.fill_polygon(editor, "white_concrete", -60)
y = 25
block_list = ["blue_concrete", "red_concrete", "green_concrete",
"yellow_concrete", "purple_concrete", "pink_concrete"]
# Over the hill
# coordinates = [(-854, 87+y, -210), (-770, 99+y, -207), (-736, 85+y, -184)]
# # Along the river
# # coordinates = [(-456, 69, -283), (-588, 106, -374), (-720, 71, -384), (-775, 67, -289), (-822, 84, -265), (-868, 77, -188), (-927, 96, -127),
# # (-926, 65, -29), (-906, 98, 42), (-902, 137, 2), (-909, 115, -62), (-924, 76, -6), (-985, 76, 37), (-1043, 76, 28), (-1102, 66, 63)]
# # Though the loop
# # coordinates = [(-1005, 113+y, -19), (-896, 113+y, 7),
# # (-807, 76+y, 54), (-738, 76+y, -10), (-678, 76+y, -86)]
# # Second zone
# coordinates = [(-805, 78, 128), (-881, 91, 104), (-950, 119, 69), (-1005, 114, 58), (-1052, 86, 30),
# (-1075, 83, 40), (-1104, 77, 63), (-1161, 69, 157), (-1144, 62, 226), (-1189, 76, 265), (-1210, 79, 329)]
# resolution, distance = curve.resolution_distance(coordinates, 6)
# curve_points = curve.curve(coordinates, resolution)
# curve_surface = CurveSurface.CurveSurface(coordinates)
# curve_surface.compute_curvature()
# curvature = []
# for i in range(len(curve_surface.curvature)):
# curvature.append((0, 1, 0))
# # Perpendicular
# curve_surface.compute_surface_perpendicular(10, curvature)
# for i in range(len(curve_surface.surface)):
# for j in range(len(curve_surface.surface[i])):
# # block = random.choice(block_list)
# for k in range(len(curve_surface.surface[i][j])):
# if k-16 < len(block_list) and k-16 >= 0:
# editor.placeBlock(
# curve_surface.surface[i][j][k], Block(block_list[k-16]))
# else:
# editor.placeBlock(
# curve_surface.surface[i][j][k], Block("stone"))
# offset = curve.offset(curve_surface.curve, -9, curvature)
# for i in range(len(offset)-1):
# line = segment.discrete_segment(offset[i], offset[i+1])
# for coordinate in line:
# editor.placeBlock(coordinate, Block("white_concrete"))
# offset = curve.offset(curve_surface.curve, 9, curvature)
# for i in range(len(offset)-1):
# line = segment.discrete_segment(offset[i], offset[i+1])
# for coordinate in line:
# editor.placeBlock(coordinate, Block("white_concrete"))
# # for coordinate in curve_surface.surface:
# # editor.placeBlock(coordinate, Block("black_concrete"))
# # for coordinate in curve_surface.curve:
# # editor.placeBlock(coordinate, Block("red_concrete"))
# # # Parallel
# # curve_surface.compute_surface_parallel(0, 10, 8, curvature)
# # for current_range in range(len(curve_surface.left_side)):
# # for coordinate in curve_surface.left_side[current_range]:
# # editor.placeBlock(coordinate, Block("yellow_concrete"))
# ---
# coordinates = [(0, 0, 0), (0, 0, 10), (0, 0, 20)]
# with open('networks/roads/lines/lines.json') as f:
# lines_type = json.load(f)
# l = Line.Line(coordinates, lines_type.get('solid_white'))
# print(l.get_surface())
# with open('networks/roads/lanes/lanes.json') as f:
# lanes_type = json.load(f)
# l = Lane.Lane(coordinates, lanes_type.get('classic_lane'), 5)
# print(l.get_surface())
# circle = curved_corner(
# ((-1365, 520), (-1326, 523)), ((-1344, 496), (-1336, 535)), 10, angle_adaptation=False, output_only_points=False)
# for coordinate in circle[0]:
# editor.placeBlock(
# (round(coordinate[0]), 125, round(coordinate[1])), Block("green_concrete"))
# ---
# r1 = Road.Road((-1341, 100, 439), "None")
# r2 = Road.Road((-1378, 100, 415), "None")
# i = Intersection.Intersection(
# (-1352, 100, 405), [(-1345, 100, 426), (-1369, 100, 412)], [r1, r2])
# ---
# r1 = Road.Road((-1337, 71, 472), "None")
# r2 = Road.Road((-1269, 80, 574), "None")
# r3 = Road.Road((-1392, 79, 527), "None")
# i = Intersection.Intersection(
# (-1327, 71, 533), [(-1335, 71, 494), (-1298, 75, 553), (-1366, 78, 530)], [r1, r2, r3])
# ---
# y = 150
# r1 = Road.Road((-1337, y, 472), "None")
# r2 = Road.Road((-1269, y, 574), "None")
# r3 = Road.Road((-1392, y, 527), "None")
# i = Intersection.Intersection(
# (-1327, y, 533), [(-1335, y, 494), (-1298, y, 553), (-1366, y, 530)], [r1, r2, r3])
# ---
# y = 100
# x = -200
# r1 = Road.Road((-1380+x, 75, 406), "None")
# r2 = Road.Road((-1365+x, 75, 468), "None")
# r3 = Road.Road((-1411+x, 75, 501), "None")
# r4 = Road.Road((-1451+x, 75, 449), "None")
# r5 = Road.Road((-1432+x, 75, 423), "None")
# i = Intersection.Intersection(
# (-1411+x, 75, 461), [(-1392+x, 75, 427), (-1385+x, 75, 465), (-1411+x, 75, 487), (-1435+x, 75, 454), (-1426+x, 75, 435)], [r1, r2, r3, r4, r5])
# i.compute_curved_corner()
# for j in range(len(i.orthogonal_delimitations)):
# coordinates = segment_tools.discrete_segment(
# i.orthogonal_delimitations[j][0][0], i.orthogonal_delimitations[j][0][1])
# for coordinate in coordinates:
# editor.placeBlock(coordinate, Block("purple_concrete"))
# coordinates = segment_tools.discrete_segment(
# i.orthogonal_delimitations[j][1][0], i.orthogonal_delimitations[j][1][1])
# for coordinate in coordinates:
# editor.placeBlock(coordinate, Block("pink_concrete"))
# coordinates = segment_tools.discrete_segment(
# i.parallel_delimitations[j][0][0], i.parallel_delimitations[j][0][1])
# for coordinate in coordinates:
# editor.placeBlock(coordinate, Block("orange_concrete"))
# coordinates = segment_tools.discrete_segment(
# i.parallel_delimitations[j][1][0], i.parallel_delimitations[j][1][1])
# for coordinate in coordinates:
# editor.placeBlock(coordinate, Block("yellow_concrete"))
# for coordinate in i.intersections:
# if coordinate != None:
# editor.placeBlock(coordinate, Block("black_concrete"))
# for k in range(len(i.intersections_curved)):
# for coordinate in i.intersections_curved[k][0]:
# if coordinate != None:
# if k >= 0:
# editor.placeBlock(
# (coordinate[0], 75, coordinate[1]), Block("gray_concrete"))
# editor.placeBlock(
# (i.intersections_curved[k][1][0], 76, i.intersections_curved[k][1][1]), Block("black_concrete"))
# coordinates = segment_tools.discrete_segment(
# i.intersections_curved[k][-1][0], i.intersections_curved[k][-1][1])
# for coordinate in coordinates:
# editor.placeBlock(coordinate, Block("lime_concrete"))
# coordinates = segment_tools.discrete_segment(
# i.intersections_curved[k][-2][0], i.intersections_curved[k][-2][1])
# for coordinate in coordinates:
# editor.placeBlock(coordinate, Block("green_concrete"))
# ---
# intersection = (-1510, 94, 455)
# xyz0 = (-1545, 90, 537)
# xyz1 = (-1535, 162, 459)
# circle = curved_corner_by_distance(
# intersection, xyz0, xyz1, 25, 0)
# line0 = segment_tools.discrete_segment(intersection, xyz0)
# line1 = segment_tools.discrete_segment(intersection, xyz1, pixel_perfect=False)
# editor.placeBlock(
# circle[1], Block("black_concrete"))
# editor.placeBlock(
# circle[3], Block("gray_concrete"))
# print(circle[3], "center")
# print(circle[4], "center")
# for coordinate in circle[0]:
# editor.placeBlock(
# coordinate, Block("white_concrete"))
# print(coordinate)
# for coordinate in line0:
# editor.placeBlock(
# coordinate, Block("blue_concrete"))
# for coordinate in line1:
# editor.placeBlock(
# coordinate, Block("red_concrete"))
# ---
# r = Road.Road(((-1829, 141, 553), (-1830, 110, 621), (-1711, 69, 625), (-1662,
# 65, 627), (-1667, 65, 761), (-1683, 70, 800), (-1721, 70, 834)), "None")
# r.place_roads()
# polyline = Polyline((Point2D(0, 0), Point2D(0, 10),
# Point2D(50, 10), Point2D(20, 20)))
# # print(polyline.radius_balance(2))
# # polyline._alpha_assign(1, polyline.length_polyline-1)
# print(polyline.alpha_radii)
# p = Polyline((Point2D(0, 0), Point2D(8, 0), Point2D(
# 16, 8), Point2D(24, 0)))
# p = Polyline((Point2D(-1420, 867), Point2D(-1362, 738),
# Point2D(-1157, 717), Point2D(-1099, 843)))
# p = Polyline((Point2D(-1183, 528), Point2D(-1138, 481),
# Point2D(-1188, 451), Point2D(-1152, 416)))
# p = Polyline((Point2D(-1225, 468), Point2D(-1138, 481),
# Point2D(-1188, 451), Point2D(-1176, 409), Point2D(-1179, 399)))
w = 100
n_points = 8
min_val, max_val = -w, w
random_points = [Point2D(random.randint(min_val, max_val), random.randint(
min_val, max_val)) for _ in range(n_points)]
# random_points = (Point2D(-75, -75), Point2D(0, -75), Point2D(75, 75),
# Point2D(75, -50), Point2D(-50, 50), Point2D(0, 0))
# random_points = random_points[0].optimized_path(random_points)
# random_points = [Point2D(-40, -56), Point2D(-94, 92), Point2D(19, -47), Point2D(
# 100, 59), Point2D(-85, -73), Point2D(-33, -9), Point2D(57, -25), Point2D(51, -34)]
random_points = random_points[0].optimized_path(random_points)
print(random_points)
# random_points = [Point2D(94, 71), Point2D(-12, 54), Point2D(-28, 10), Point2D(
# 0, -33), Point2D(80, -50), Point2D(73, -89), Point2D(-86, -3), Point2D(-82, 92)]
# random_points = [Point2D(-59, -21), Point2D(-43, -19), Point2D(-61, 19), Point2D(
# 45, 19), Point2D(80, -4), Point2D(99, 2), Point2D(47, 63), Point2D(100, -91)]
p = Polyline(random_points)
radius = p.get_radii()
center = p.get_centers()
y = 200
ww = 15
width, height = 2*w, 2*w
image = Image.new('RGB', (width, height), 'black')
draw = ImageDraw.Draw(image)
for i in range(len(p.output_points)-1):
if p.output_points[i] != 0:
s = Segment2D(Point2D(p.output_points[i].x, p.output_points[i].y), Point2D(
p.output_points[i+1].x, p.output_points[i+1].y))
s.segment_thick(ww, LINE_THICKNESS_MODE.MIDDLE)
for j in range(len(s.points_thick)-1):
# editor.placeBlock(
# s.coordinates[j].coordinate, Block("cyan_concrete"))
draw.point((s.points_thick[j].x+w,
w-s.points_thick[j].y), fill='grey')
# for i in range(2, len(p.get_arcs_intersections())-2):
# s = Segment2D(Point2D(p.acrs_intersections[i][0].x, p.acrs_intersections[i][0].y), Point2D(
# p.acrs_intersections[i-1][-1].x, p.acrs_intersections[i-1][-1].y))
# s.segment_thick(ww, LINE_THICKNESS_MODE.MIDDLE)
# for j in range(len(s.points_thick)-1):
# # editor.placeBlock(
# # s.coordinates[j].coordinate, Block("cyan_concrete"))
# draw.point((s.points_thick[j].x+w,
# w-s.points_thick[j].y), fill='green')
# draw.point((p.acrs_intersections[i][0].x+w,
# w-p.acrs_intersections[i][0].y), fill='green')
# draw.point((p.acrs_intersections[i][-1].x+w,
# w-p.acrs_intersections[i][-1].y), fill='green')
# for i in range(len(center)):
# if center[i]:
# circle = Circle(center[i])
# circle.circle_thick(round(radius[i]-ww/2), round(radius[i]+ww/2))
# for j in range(len(circle.points_thick)-1):
# if circle.points_thick[j].is_in_triangle(p.acrs_intersections[i][0], p.acrs_intersections[i][1], p.acrs_intersections[i][2]):
# # editor.placeBlock(
# # (circle.coordinates[j].x, y, circle.coordinates[j].y), Block("white_concrete"))
# draw.point((circle.points_thick[j].x+w,
# w-circle.points_thick[j].y), fill='green')
# circle.circle(radius[i])
# for j in range(len(circle.points)-1):
# if circle.points[j].is_in_triangle(p.acrs_intersections[i][0], p.acrs_intersections[i][1], p.acrs_intersections[i][2]):
# # editor.placeBlock(
# # (circle.coordinates[j].x, y, circle.coordinates[j].y), Block("white_concrete"))
# draw.point(
# (circle.points[j].x+w, w-circle.points[j].y), fill='green')
s1 = Segment2D(Point2D(p.acrs_intersections[1][0].x, p.acrs_intersections[1][0].y), Point2D(
p.output_points[0].x, p.output_points[0].y))
s1.segment_thick(ww, LINE_THICKNESS_MODE.MIDDLE)
for j in range(len(s1.points_thick)-1):
draw.point((s1.points_thick[j].x+w,
w-s1.points_thick[j].y), fill='grey')
s1 = Segment2D(Point2D(p.acrs_intersections[-2][2].x, p.acrs_intersections[-2][2].y), Point2D(
p.output_points[-1].x, p.output_points[-1].y))
s1.segment_thick(ww, LINE_THICKNESS_MODE.MIDDLE)
for j in range(len(s1.points_thick)-1):
draw.point((s1.points_thick[j].x+w,
w-s1.points_thick[j].y), fill='grey')
# for i in range(0, len(p.arcs)):
# for j in range(len(p.arcs[i])):
# draw.point((p.arcs[i][j].x+w, w-p.arcs[i][j].y), fill='green')
for i in range(1, len(p.segments)-1):
for j in range(len(p.segments[i].segment_thick(5, LINE_THICKNESS_MODE.MIDDLE))):
draw.point((p.segments[i].points_thick[j].x+w,
w-p.segments[i].points_thick[j].y), fill='white')
def get_color(i):
color = (50+round(i), 50+round(i/5), 50+round(i/3))
return color
for i in range(1, len(p.centers)-1):
draw.point((p.centers[i].x+w, w-p.centers[i].y), fill='red')
draw.point((p.acrs_intersections[i][0].x+w,
w-p.acrs_intersections[i][0].y), fill='blue')
draw.point((p.acrs_intersections[i][1].x+w,
w-p.acrs_intersections[i][1].y), fill='purple')
draw.point((p.acrs_intersections[i][2].x+w,
w-p.acrs_intersections[i][2].y), fill='blue')
for i in range(len(p.total_line_output)):
draw.point((p.total_line_output[i].x+w,
w-p.total_line_output[i].y), fill=get_color(i))
image.save('output_image.png')
# road = Road([Point3D(-1201, 75, 705), Point3D(-1162, 69, 687),
# Point3D(-1149, 71, 647), Point3D(-1191, 68, 611)], 5)
# road = Road([Point3D(-1359, 75, 696), Point3D(-1389, 126, 697),
# Point3D(-1401, 126, 714), Point3D(-1426, 126, 707), Point3D(-1452, 126, 714), Point3D(-1430, 126, 765)], 5)
# road = Road([Point3D(-1203, 73, 718), Point3D(-1157, 76, 719),
# Point3D(-1119, 76, 763), Point3D(-1101, 76, 827), Point3D(-1088, 76, 879), Point3D(-1095, 76, 944)], 10)
# road = Road([Point3D(-986, 83, 602), Point3D(-1000, 83, 647),
# Point3D(-993, 83, 680), Point3D(-965, 83, 712)], 10)
# road = Road([Point3D(-984, 97, 811), Point3D(-984, 97, 847),
# Point3D(-962, 97, 860), Point3D(-970, 97, 900), Point3D(-953, 97, 920)], 10)
# road = Road([Point3D(-1024, 106, 1000), Point3D(-1024, 101, 972),
# Point3D(-1001, 100, 966), Point3D(-977, 98, 984), Point3D(-966, 102, 1011), Point3D(-905, 97, 1013), Point3D(-774, 99, 998), Point3D(-694, 99, 1047)], 9)
# road = Road([Point3D(-745, 125, 899), Point3D(-744, 117, 944),
# Point3D(-696, 112, 941), Point3D(-645, 112, 979)], 9)
# road = Road([Point3D(-454, 130, 1046), Point3D(-497, 127, 1070),
# Point3D(-545, 85, 1019), Point3D(-545, 85, 970), Point3D(-457, 87, 865)], 9)
road = Road(Point3D.insert_3d(random_points, 'y', [random.randint(
200, 250) for _ in range(n_points)]), 15)
road.place()
# s = Segment2D(Point2D(-88, -12), Point2D(9, 75))
# s.segment_thick(3, LINE_THICKNESS_MODE.MIDDLE)
# print(s.points)
# s = Segment2D(Point2D(0, 0), Point2D(10, 10)).perpendicular(10)
# print(s)
# Note: passer parrallel dans Segment2D pour pouvoir calculer l'intersection entre deux segments
# de la Polyline pour trouver le centre du cercle. Faire l'arc de cercle en utilise is_in_triangle
# Okay mb, l'article scientifique explique une procédure sans doute plus efficace.
# alpha n'est pas un angle.