abstreet/headless/examples/cancel_experiment.py
Dustin Carlino 39f5d50fcd The grand country split. #326
City names are now disambiguated by a two-letter country code. This
commit handles almost everything needed to make this transition. Main
next steps are fixing up map edits automatically and making the city
picker UI understand the extra level of hierarchy.

A little bit of fallout: lakeslice gridlocks again; this regression is
actually from the recent traffic signal changes, but I'm just now
regenerating everything. Will fix soon.
2021-02-13 15:45:59 -08:00

65 lines
2.4 KiB
Python
Executable File

#!/usr/bin/python3
# This example runs the same scenario repeatedly, each time cancelling a
# different number of trips uniformly at random. The eventual goal is to
# quantify how many trips need to be cancelled to substantially speed up
# remaining ones.
#
# Before running this script, start the API server:
#
# > cargo run --release --bin headless -- --port=1234 --alerts=silence
#
# You may need to install https://requests.readthedocs.io
# Keep this script formatted with autopep8 -i
import abst_helpers
import argparse
import sys
import time
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--api', default='http://localhost:1234')
parser.add_argument('--country_code', default='us')
parser.add_argument('--city_name', default='seattle')
parser.add_argument('--map_name', default='montlake')
parser.add_argument('--hours', type=int, default=24)
parser.add_argument('--cmp1', type=int, default=None)
parser.add_argument('--cmp2', type=int, default=None)
args = parser.parse_args()
if args.cmp1 and args.cmp2 and args.cmp1 > args.cmp2:
sys.exit(
'--cmp1={} --cmp2={} invalid, --cmp1 is the baseline'.format(args.cmp1, args.cmp2))
print('Simulating {} hours of data/system/{}/{}/scenarios/{}/weekday.bin'.format(
args.hours, args.country_code, args.city_name, args.map_name))
print('')
num_succeeded_last = 0
results2 = None
for pct in range(100, 0, -10):
start = time.time()
results = abst_helpers.run_sim(args, modifiers=[{"ChangeMode": {
"pct_ppl": pct,
"departure_filter": [0.0, 86400.0],
"from_modes":["Walk", "Bike", "Transit", "Drive"],
"to_mode":None}}])
print('{}% of people cancelled: {:,} trips cancelled, {:,} trips succeeded. Simulation took {:.1f}s'.format(
pct, results.num_cancelled, len(results.trip_times), time.time() - start))
if len(results.trip_times) < num_succeeded_last:
print('--> less trips succeeded this round, so likely hit gridlock')
break
num_succeeded_last = len(results.trip_times)
if args.cmp2 == pct:
results2 = results
if args.cmp1 == pct:
print('')
print('Baseline cancelled {}%, experimental cancelled {}%'.format(
args.cmp1, args.cmp2))
results.compare(results, results2)
print('')
if __name__ == '__main__':
main()