def segments_from_splits(route, splits): segments = [ ] start_split = False segment_start = route[0] for tid in route: if start_split: segment_start = tid start_split = False if tid in splits: segments.append(Segment.from_route(route, segment_start, tid)) start_split = True return segments
def segment_from_name(name, rooms, route): start_transition_name, end_transition_name = name.split(' to ') start = transition_from_name(start_transition_name, rooms, route) end = transition_from_name(end_transition_name, rooms, route) return Segment.from_route(route, start, end)