diff --git a/main.py b/main.py index 5b090e4..6915220 100644 --- a/main.py +++ b/main.py @@ -1,3 +1,6 @@ + +from PIL import Image, ImageDraw +import matplotlib.pyplot as plt from networks.geometry.Point3D import Point3D from networks.geometry.Segment3D import Segment3D from networks.geometry.Segment2D import Segment2D @@ -21,6 +24,11 @@ from networks.roads.intersections import Intersection as Intersection from networks.geometry.point_tools import curved_corner_by_curvature, curved_corner_by_distance + +import matplotlib +matplotlib.use('Agg') + + editor = Editor(buffering=True) # f = open('buildings\shapes.json') @@ -278,8 +286,8 @@ block_list = ["blue_concrete", "red_concrete", "green_concrete", # p = Polyline((Point2D(-1225, 468), Point2D(-1138, 481), # Point2D(-1188, 451), Point2D(-1176, 409), Point2D(-1179, 399))) -p = Polyline((Point2D(-1206, 414), Point2D(-1176, 409), - Point2D(-1188, 451), Point2D(-1138, 481), Point2D(-1225, 500))) +p = Polyline((Point2D(32, 0), Point2D(16, 16), + Point2D(16, 0), Point2D(0, 0))) # Point2D(-1156, 378), Point2D(-1220, 359), Point2D(-1265, 290) # print(p.alpha_radii) @@ -293,14 +301,11 @@ print(p.lengths) y = 160 -for i in range(len(p.coordinates)-1): - if p.coordinates[i] != None: - s = Segment3D(Point3D(p.coordinates[i].x, y, p.coordinates[i].y), Point3D( - p.coordinates[i+1].x, y, p.coordinates[i+1].y)) - for j in range(len(s.coordinates)-1): - editor.placeBlock( - s.coordinates[j].coordinate, Block("cyan_concrete")) +width, height = 100, 100 +image = Image.new('RGB', (width, height), 'white') + +draw = ImageDraw.Draw(image) for i in range(len(center)): if center[i]: @@ -308,7 +313,21 @@ for i in range(len(center)): for j in range(len(circle.coordinates)-1): editor.placeBlock( (circle.coordinates[j].x, y, circle.coordinates[j].y), Block("white_concrete")) + draw.point((circle.coordinates[j].x+25, + 50-circle.coordinates[j].y), fill='black') +for i in range(len(p.coordinates)-1): + if p.coordinates[i] != None: + s = Segment3D(Point3D(p.coordinates[i].x, y, p.coordinates[i].y), Point3D( + p.coordinates[i+1].x, y, p.coordinates[i+1].y)) + for j in range(len(s.coordinates)-1): + editor.placeBlock( + s.coordinates[j].coordinate, Block("cyan_concrete")) + draw.point((s.coordinates[j].x+25, + 50-s.coordinates[j].z), fill='red') + + +image.save('output_image.png') # s = Segment2D(Point2D(0, 0), Point2D(10, 10)).perpendicular(10) # print(s) diff --git a/output_image.png b/output_image.png new file mode 100644 index 0000000..2e233fb Binary files /dev/null and b/output_image.png differ