cli/
import_grid2demand.rs1use anyhow::{anyhow, bail, Result};
2use serde::Deserialize;
3
4use abstutil::{prettyprint_usize, Timer};
5use geom::{Duration, LonLat, Time};
6use map_model::Map;
7use synthpop::{
8 ExternalPerson, ExternalTrip, ExternalTripEndpoint, Scenario, TripMode, TripPurpose,
9};
10
11pub fn run(csv_path: String, map: String) -> Result<()> {
12 let mut timer = Timer::new("import grid2demand");
13 timer.start("parse CSV");
14 let people = parse_trips(csv_path)?;
15 timer.stop("parse CSV");
16 let map = Map::load_synchronously(map, &mut timer);
17
18 let mut s = Scenario::empty(&map, "grid2demand");
19 s.only_seed_buses = None;
21 let orig_num = people.len();
22 let skip_problems = true;
23 s.people = ExternalPerson::import(&map, people, skip_problems)?;
24 s = s.remove_weird_schedules(true);
26 println!(
27 "Imported {}/{} people",
28 prettyprint_usize(s.people.len()),
29 prettyprint_usize(orig_num)
30 );
31 s.save();
32
33 Ok(())
34}
35
36fn parse_trips(csv_path: String) -> Result<Vec<ExternalPerson>> {
37 let mut people = Vec::new();
38 for rec in csv::Reader::from_reader(fs_err::File::open(csv_path)?).deserialize() {
39 let rec: Record = rec?;
40 let mode = match rec.agent_type.as_ref() {
41 "v" => TripMode::Drive,
42 "b" => TripMode::Bike,
43 "p" => TripMode::Walk,
44 "t" => TripMode::Transit,
45 x => bail!("Unknown agent_type {}", x),
46 };
47 let (origin, destination) = parse_linestring(&rec.geometry)
48 .ok_or_else(|| anyhow!("didn't parse geometry {}", rec.geometry))?;
49 let departure = parse_time(rec.departure_time)?;
50 people.push(ExternalPerson {
53 trips: vec![ExternalTrip {
54 departure,
55 origin: ExternalTripEndpoint::Position(origin),
56 destination: ExternalTripEndpoint::Position(destination),
57 mode,
58 purpose: TripPurpose::Work,
59 }],
60 });
61 }
62 Ok(people)
63}
64
65fn parse_linestring(input: &str) -> Option<(LonLat, LonLat)> {
66 let mut nums = Vec::new();
68 for x in input
69 .strip_prefix("LINESTRING(")?
70 .strip_suffix(')')?
71 .split(&[' ', ','][..])
72 {
73 nums.push(x.parse::<f64>().ok()?);
74 }
75 if nums.len() != 4 {
76 return None;
77 }
78 Some((LonLat::new(nums[0], nums[1]), LonLat::new(nums[2], nums[3])))
79}
80
81fn parse_time(input: String) -> Result<Time> {
82 let hours = input[0..2].parse::<usize>()?;
84 let mins = input[2..].parse::<usize>()?;
85 Ok(Time::START_OF_DAY + Duration::hours(hours) + Duration::minutes(mins))
86}
87
88#[derive(Debug, Deserialize)]
89struct Record {
90 agent_type: String,
91 geometry: String,
92 departure_time: String,
93}