a-b-street/abstreet
1
1
Fork 0
mirror of https://github.com/a-b-street/abstreet.git synced 2024-12-24 23:15:24 +03:00
Code Issues Projects Releases Wiki Activity

Cap all roads in the Python experiment. Right now, this'll unintentionally merge adjacent zones with the same cap into zones.

Browse Source
This commit is contained in:
Dustin Carlino 2020-10-01 15:13:18 -07:00
parent c2ef091fb4
commit f951b296f7
1 changed files with 38 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
headless/examples/cap_experiment.py
View File

@ -24,6 +24,7 @@ def main():
parser.add_argument('--hours', type=int, default=24)
parser.add_argument('--cap_pct', type=int, default=80)
parser.add_argument('--rounds', type=int, default=10)
parser.add_argument('--cap_all_roads', type=bool, default=True)
args = parser.parse_args()
print('Simulating {} hours of data/system/scenarios/{}/weekday.bin'.format(args.hours, args.map_name))
@ -32,20 +33,27 @@ def main():
for _ in range(args.rounds):
print('')
print('{} roads have a {}% cap'.format(
len(edits['commands']), args.cap_pct))
print('{} roads have a cap'.format(len(edits['commands'])))
experiment = abst_helpers.run_sim(args, edits=edits)
baseline.compare(experiment)
print('{:,} trips changed due to the caps'.format(
len(experiment.capped_trips)))
# Cap the busiest road
busiest_road, thruput = find_busiest_road(args)
cmd = get(args, '/map/get-edit-road-command',
params={'id': busiest_road}).json()
cmd['ChangeRoad']['new']['access_restrictions']['cap_vehicles_per_hour'] = int(
(args.cap_pct / 100.0) * thruput)
edits['commands'].append(cmd)
if args.cap_all_roads:
edits['commands'] = cap_all_roads(args)
else:
# Cap the busiest road
busiest_road, thruput = find_busiest_road(args)
cmd = get(args, '/map/get-edit-road-command',
params={'id': busiest_road}).json()
cmd['ChangeRoad']['new']['access_restrictions']['cap_vehicles_per_hour'] = int(
(args.cap_pct / 100.0) * thruput)
edits['commands'].append(cmd)
# Write the final edits
f = open('cap_edits.json', 'w')
f.write(get(args, '/map/get-edits').text)
f.close()
# Find the road with the most car traffic in any one hour period
@ -64,5 +72,26 @@ def find_busiest_road(args):
return (max_key[0], max_value)
# Cap all roads to some percent of their max throughput over any one hour period
def cap_all_roads(args):
thruput = get(
args, '/data/get-road-thruput').json()['counts']
max_per_road = {}
for r, agent, hr, count in thruput:
if agent == 'Car':
if r not in max_per_road or count > max_per_road[r]:
max_per_road[r] = count
commands = []
for r, count in max_per_road.items():
cap = int((args.cap_pct / 100.0) * count)
# Don't go too low
if cap > 10:
cmd = get(args, '/map/get-edit-road-command',
params={'id': r}).json()
cmd['ChangeRoad']['new']['access_restrictions']['cap_vehicles_per_hour'] = cap
commands.append(cmd)
return commands
if __name__ == '__main__':
main()