43 lines
2.2 KiB
Python
43 lines
2.2 KiB
Python
from networks.geometry.segment_tools import parallel, orthogonal
|
|
from networks.geometry.point_tools import sort_by_clockwise, segments_intersection, curved_corner_by_distance
|
|
from networks.roads import Road
|
|
|
|
|
|
class Intersection:
|
|
def __init__(self, center, coordinates, Roads):
|
|
self.center = center
|
|
self.coordinates = coordinates
|
|
self.Roads = Roads
|
|
self.parallel_delimitations = []
|
|
self.orthogonal_delimitations = []
|
|
self.intersections = []
|
|
self.intersections_curved = []
|
|
|
|
def compute_curved_corner(self):
|
|
# Necessary to test nearby intersection
|
|
self.coordinates = sort_by_clockwise(self.coordinates)
|
|
|
|
for i, coordinate in enumerate(self.coordinates):
|
|
right_side, left_side = parallel((coordinate, self.center), self.Roads[i].width), parallel(
|
|
(coordinate, self.center), -self.Roads[i].width)
|
|
self.parallel_delimitations.append((right_side, left_side))
|
|
self.orthogonal_delimitations.append(
|
|
((right_side[0], left_side[0]), (right_side[-1], left_side[-1])))
|
|
|
|
for j in range(len(self.Roads)):
|
|
self.intersections.append(segments_intersection(
|
|
self.parallel_delimitations[j][1], self.parallel_delimitations[(j+1) % len(self.Roads)][0], full_line=False))
|
|
|
|
next_parallel = tuple(self.parallel_delimitations[(
|
|
j+1) % len(self.Roads)][0][0]), tuple(self.parallel_delimitations[(j+1) % len(self.Roads)][0][1])
|
|
current_parallel = tuple(self.parallel_delimitations[j][1][0]), tuple(
|
|
self.parallel_delimitations[j][1][1])
|
|
|
|
intersection2d = segments_intersection(((current_parallel[0][0], current_parallel[0][-1]), (current_parallel[1][0], current_parallel[1][-1])), ((
|
|
next_parallel[0][0], next_parallel[0][-1]), (next_parallel[1][0], next_parallel[1][-1])), full_line=False)
|
|
|
|
intersection = (
|
|
round(intersection2d[0]), 100, round(intersection2d[1]))
|
|
self.intersections_curved.append(curved_corner_by_distance(
|
|
intersection, current_parallel[0], next_parallel[0], 10, 0, full_line=True))
|