2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

create GraphArea.check_connection

Browse source
This commit is contained in:
Laura Klünder 2016-12-20 22:34:40 +01:00
parent 0b44acc8b0
commit e3990fcaf0
1 changed files with 51 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

102
src/c3nav/routing/area.py
View file

@ -30,69 +30,69 @@ class GraphArea():
def build_connections(self): def build_connections(self):
for point1, point2 in combinations(self._built_points, 2): for point1, point2 in combinations(self._built_points, 2):
path = Path(np.vstack((point1.xy, point2.xy)))
# lies within room there, back = self.check_connection(point1, point2)
if self.mpl_clear.intersects_path(path):
if there is not None:
point1.connect_to(point2, ctype=there)
if back is not None:
point2.connect_to(point1, ctype=back)
def check_connection(self, point1, point2):
path = Path(np.vstack((point1.xy, point2.xy)))
# lies within room
if self.mpl_clear.intersects_path(path):
return None, None
# stair checker
angle = coord_angle(point1.xy, point2.xy)
stair_direction_up = None
for stair_path, stair_angle in self.mpl_stairs:
if not path.intersects_path(stair_path):
continue continue
# stair checker angle_diff = ((stair_angle - angle + 180) % 360) - 180
angle = coord_angle(point1.xy, point2.xy)
valid = True
stair_direction_up = None
for stair_path, stair_angle in self.mpl_stairs:
if not path.intersects_path(stair_path):
continue
angle_diff = ((stair_angle - angle + 180) % 360) - 180 new_direction_up = (angle_diff > 0)
if stair_direction_up is None:
stair_direction_up = new_direction_up
elif stair_direction_up != new_direction_up:
return None, None
new_direction_up = (angle_diff > 0) if not (40 < abs(angle_diff) < 150):
if stair_direction_up is None: return None, None
stair_direction_up = new_direction_up
elif stair_direction_up != new_direction_up:
valid = False
break
if not (40 < abs(angle_diff) < 150): # escalator checker
valid = False angle = coord_angle(point1.xy, point2.xy)
break escalator_direction_up = None
escalator_swap_direction = False
if not valid: for escalator in self.escalators:
if not escalator.mpl_geom.intersects_path(path, filled=True):
continue continue
# escalator checker if escalator_direction_up is not None:
angle = coord_angle(point1.xy, point2.xy) # only one escalator per connection
valid = True return None, None
escalator_direction_up = None
escalator_swap_direction = False
for escalator in self.escalators:
if not escalator.mpl_geom.intersects_path(path, filled=True):
continue
if escalator_direction_up is not None: angle_diff = ((escalator.angle - angle + 180) % 360) - 180
# only one escalator per connection
valid = False
break
angle_diff = ((escalator.angle - angle + 180) % 360) - 180 escalator_direction_up = (angle_diff > 0)
escalator_swap_direction = (escalator_direction_up != escalator.direction_up)
escalator_direction_up = (angle_diff > 0) if stair_direction_up is not None:
escalator_swap_direction = (escalator_direction_up != escalator.direction_up) return (
('stairs_up' if stair_direction_up else 'stairs_down'),
if not valid: ('stairs_down' if stair_direction_up else 'stairs_up'),
continue )
elif escalator_direction_up is not None:
if stair_direction_up is not None: if not escalator_swap_direction:
point1.connect_to(point2, ctype=('stairs_up' if stair_direction_up else 'stairs_down')) return ('escalator_up' if escalator_direction_up else 'escalator_down'), None
point2.connect_to(point1, ctype=('stairs_down' if stair_direction_up else 'stairs_up'))
elif escalator_direction_up is not None:
if not escalator_swap_direction:
point1.connect_to(point2, ctype=('escalator_up' if escalator_direction_up else 'escalator_down'))
else:
point2.connect_to(point1, ctype=('escalator_down' if escalator_direction_up else 'escalator_up'))
else: else:
point1.connect_to(point2) return None, ('escalator_down' if escalator_direction_up else 'escalator_up')
point2.connect_to(point1)
return '', ''
def add_point(self, point): def add_point(self, point):
if not self.mpl_clear.contains_point(point.xy): if not self.mpl_clear.contains_point(point.xy):