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

split route into connections

Browse source
This commit is contained in:
Laura Klünder 2016-12-17 14:46:15 +01:00
parent 988c7b1d11
commit 4f847f4e49
5 changed files with 87 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

79
src/c3nav/routing/route.py
View file

@ -1,22 +1,23 @@
from abc import ABC
from abc import ABC, abstractmethod
import numpy as np
from django.utils.functional import cached_property
class RouteSegment(ABC):
def __init__(self, router, from_point, to_point):
def __init__(self, routers, router, from_point, to_point):
"""
:param router: a Router (RoomRouter, GraphRouter, …)
:param from_point: in-router index of first point
:param to_point: in-router index of last point
"""
self.routers = routers
self.router = router
self.from_point = int(from_point)
self.to_point = int(to_point)
def as_route(self):
return Route([self])
return SegmentRoute([self])
def _get_points(self):
points = [self.to_point]
@ -27,6 +28,10 @@ class RouteSegment(ABC):
points.append(current)
return tuple(reversed(points))
@abstractmethod
def get_connections(self):
pass
@cached_property
def distance(self):
return self.router.shortest_paths[self.from_point, self.to_point]
@ -37,15 +42,17 @@ class RoomRouteSegment(RouteSegment):
"""
Route segment within a Room
:param room: GraphRoom
:param router: RoomRouter
:param from_point: in-room index of first point
:param to_point: in-room index of last point
"""
super().__init__(routers[room], from_point, to_point)
super().__init__(routers, routers[room], from_point, to_point)
self.room = room
self.global_from_point = room.points[from_point]
self.global_to_point = room.points[to_point]
def get_connections(self):
points = self._get_points()
return tuple(self.room.get_connection(from_point, to_point)
for from_point, to_point in zip(points[:-1], points[1:]))
def __repr__(self):
return ('<RoomRouteSegment in %r from points %d to %d with distance %f>' %
(self.room, self.from_point, self.to_point, self.distance))
@ -57,11 +64,26 @@ class LevelRouteSegment(RouteSegment):
Route segment within a Level (from room transfer point to room transfer point)
:param level: GraphLevel
"""
super().__init__(routers[level], from_point, to_point)
super().__init__(routers, routers[level], from_point, to_point)
self.level = level
self.global_from_point = level.room_transfer_points[from_point]
self.global_to_point = level.room_transfer_points[to_point]
def split(self):
segments = []
points = self._get_points()
for from_point, to_point in zip(points[:-1], points[1:]):
room = self.level.rooms[self.router.room_transfers[from_point, to_point]]
global_from_point = self.level.room_transfer_points[from_point]
global_to_point = self.level.room_transfer_points[to_point]
segments.append(RoomRouteSegment(room, self.routers,
from_point=room.points.index(global_from_point),
to_point=room.points.index(global_to_point)))
return tuple(segments)
def get_connections(self):
return sum((segment.get_connections() for segment in self.split()), ())
def __repr__(self):
return ('<LevelRouteSegment in %r from points %d to %d with distance %f>' %
(self.level, self.from_point, self.to_point, self.distance))
@ -73,24 +95,57 @@ class GraphRouteSegment(RouteSegment):
Route segment within a Graph (from level transfer point to level transfer point)
:param graph: Graph
"""
super().__init__(routers[graph], from_point, to_point)
super().__init__(routers, routers[graph], from_point, to_point)
self.graph = graph
self.global_from_point = graph.level_transfer_points[from_point]
self.global_to_point = graph.level_transfer_points[to_point]
def split(self):
segments = []
points = self._get_points()
for from_point, to_point in zip(points[:-1], points[1:]):
level = self.graph.levels[self.router.level_transfers[from_point, to_point]]
global_from_point = self.graph.level_transfer_points[from_point]
global_to_point = self.graph.level_transfer_points[to_point]
segments.append(LevelRouteSegment(level, self.routers,
from_point=level.room_transfer_points.index(global_from_point),
to_point=level.room_transfer_points.index(global_to_point)))
return tuple(segments)
def get_connections(self):
return sum((segment.get_connections() for segment in self.split()), ())
def __repr__(self):
return ('<GraphRouteSegment in %r from points %d to %d with distance %f>' %
(self.graph, self.from_point, self.to_point, self.distance))
class Route:
class SegmentRoute:
def __init__(self, segments, distance=None):
self.segments = sum(((item.segments if isinstance(item, Route) else (item, )) for item in segments), ())
self.segments = sum(((item.segments if isinstance(item, SegmentRoute) else (item,))
for item in segments if item.from_point != item.to_point), ())
self.distance = sum(segment.distance for segment in self.segments)
self.from_point = segments[0].global_from_point
self.to_point = segments[-1].global_to_point
def __repr__(self):
return ('<SegmentedRoute (\n %s\n) distance=%f>' %
('\n '.join(repr(segment) for segment in self.segments), self.distance))
def split(self):
return Route(sum((segment.get_connections() for segment in self.segments), ()))
class Route:
def __init__(self, connections, distance=None):
self.connections = tuple(connections)
self.distance = sum(connection.distance for connection in self.connections)
self.from_point = connections[0].from_point
self.to_point = connections[-1].to_point
def __repr__(self):
return ('<Route (\n %s\n) distance=%f>' %
('\n '.join(repr(segment) for segment in self.segments), self.distance))
('\n '.join(repr(connection) for connection in self.connections), self.distance))
class NoRoute: