LucasJavaudin
/
metrosim
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

do not cut trips

This commit is contained in:
LucasJavaudin 2021-08-18 10:18:12 +02:00
parent c28a09010b
commit 4640069698
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
metrosim/models/pt_network.py
View File

@ -117,12 +117,14 @@ class PTNetwork(nx.MultiDiGraph):
# Sort stop_times by increasing stop_sequence.
stop_times = sorted(
stop_times, key=lambda x: int(x['stop_sequence']))
first_departure = _parse_time(stop_times[0]['departure_time'])
last_arrival = _parse_time(stop_times[-1]['arrival_time'])
if first_departure > to_time or last_arrival < from_time:
# This trip is not in the desired time window.
continue
for prev_stop, next_stop in zip(stop_times[:-1], stop_times[1:]):
dep_time = _parse_time(prev_stop['departure_time'])
arr_time = _parse_time(next_stop['arrival_time'])
if dep_time < from_time or arr_time > to_time:
# This trip stop is not in the desired time window.
continue
route_id = self.trips[trip_id]['route_id']
self.add_edge(
prev_stop['stop_id'],