use std::collections::BTreeMap;
use rand::seq::SliceRandom;
use rand::{Rng, SeedableRng};
use rand_xorshift::XorShiftRng;
use geom::{Angle, Duration, Time};
use map_model::Map;
use sim::{
AgentID, CarID, ParkingSpot, PedestrianID, Person, PersonID, PersonState, TripEndpoint, TripID,
TripMode, TripResult, VehicleType,
};
use widgetry::{
Btn, Color, EdgeInsets, EventCtx, GeomBatch, Key, Line, RewriteColor, Text, TextExt, TextSpan,
Widget,
};
use crate::app::App;
use crate::info::{building, header_btns, make_table, make_tabs, trip, Details, OpenTrip, Tab};
pub fn trips(
ctx: &mut EventCtx,
app: &App,
details: &mut Details,
id: PersonID,
open_trips: &mut BTreeMap<TripID, OpenTrip>,
is_paused: bool,
) -> Vec<Widget> {
let mut rows = header(
ctx,
app,
details,
id,
Tab::PersonTrips(id, open_trips.clone()),
is_paused,
);
let map = &app.primary.map;
let sim = &app.primary.sim;
let person = sim.get_person(id);
if !open_trips.is_empty() {
details.unzoomed.push(
app.cs.fade_map_dark,
app.primary.map.get_boundary_polygon().clone(),
);
}
let mut wheres_waldo = true;
for (idx, t) in person.trips.iter().enumerate() {
let (trip_status, color, maybe_info) = match sim.trip_to_agent(*t) {
TripResult::TripNotStarted => {
if wheres_waldo {
wheres_waldo = false;
rows.push(current_status(ctx, person, map));
}
if sim.time() > sim.trip_info(*t).departure {
(
"delayed start",
Color::YELLOW,
open_trips
.get_mut(t)
.map(|open_trip| trip::future(ctx, app, *t, open_trip, details)),
)
} else {
(
"future",
Color::hex("#4CA7E9"),
open_trips
.get_mut(t)
.map(|open_trip| trip::future(ctx, app, *t, open_trip, details)),
)
}
}
TripResult::Ok(a) => {
assert!(wheres_waldo);
wheres_waldo = false;
(
"ongoing",
Color::hex("#7FFA4D"),
open_trips
.get_mut(t)
.map(|open_trip| trip::ongoing(ctx, app, *t, a, open_trip, details)),
)
}
TripResult::ModeChange => {
assert!(wheres_waldo);
wheres_waldo = false;
(
"ongoing",
Color::hex("#7FFA4D"),
open_trips.get(t).map(|_| Widget::nothing()),
)
}
TripResult::TripDone => {
assert!(wheres_waldo);
(
"finished",
Color::hex("#A3A3A3"),
if open_trips.contains_key(t) {
Some(trip::finished(ctx, app, id, open_trips, *t, details))
} else {
None
},
)
}
TripResult::TripCancelled => {
(
"cancelled",
Color::hex("#EB3223"),
open_trips
.get_mut(t)
.map(|open_trip| trip::cancelled(ctx, app, *t, open_trip, details)),
)
}
TripResult::TripDoesntExist => unreachable!(),
};
let trip = sim.trip_info(*t);
let (row_btn, hitbox) = Widget::custom_row(vec![
format!("Trip {} ", idx + 1)
.batch_text(ctx)
.centered_vert()
.margin_right(21),
Widget::row(vec![
GeomBatch::load_svg(
ctx.prerender,
match trip.mode {
TripMode::Walk => "system/assets/meters/pedestrian.svg",
TripMode::Bike => "system/assets/meters/bike.svg",
TripMode::Drive => "system/assets/meters/car.svg",
TripMode::Transit => "system/assets/meters/bus.svg",
},
)
.scale(0.75)
.autocrop()
.color(RewriteColor::ChangeAll(color))
.batch(),
Line(trip_status)
.small()
.fg(color)
.batch(ctx)
.container()
.padding_bottom(2),
])
.centered()
.fully_rounded()
.outline(1.0, color)
.bg(color.alpha(0.2))
.padding(EdgeInsets {
top: 5.0,
bottom: 5.0,
left: 10.0,
right: 10.0,
})
.margin_right(21),
if trip.modified {
Line("modified").batch(ctx).centered_vert().margin_right(15)
} else {
Widget::nothing()
},
if trip.capped {
Line("capped").batch(ctx).centered_vert().margin_right(15)
} else {
Widget::nothing()
},
if trip_status == "finished" {
if let Some(before) = app
.has_prebaked()
.and_then(|_| app.prebaked().finished_trip_time(*t))
{
let (after, _) = app.primary.sim.finished_trip_time(*t).unwrap();
Text::from(cmp_duration_shorter(after, before))
.batch(ctx)
.centered_vert()
} else {
Widget::nothing()
}
} else {
Widget::nothing()
},
{
let mut icon = GeomBatch::load_svg(
ctx.prerender,
"system/assets/widgetry/arrow_drop_down.svg",
)
.autocrop()
.color(RewriteColor::ChangeAll(Color::WHITE))
.scale(1.5);
if !open_trips.contains_key(t) {
icon = icon.rotate(Angle::degrees(180.0));
}
icon.batch().container().align_right().margin_right(10)
},
])
.centered()
.outline(2.0, Color::WHITE)
.padding(16)
.bg(app.cs.inner_panel)
.to_geom(ctx, Some(0.3));
rows.push(
Btn::custom(
row_btn.clone(),
row_btn.color(RewriteColor::Change(app.cs.inner_panel, app.cs.hovering)),
hitbox,
None,
)
.build(
ctx,
format!(
"{} {}",
if open_trips.contains_key(t) {
"hide"
} else {
"show"
},
t
),
None,
)
.margin_above(if idx == 0 { 0 } else { 16 }),
);
if let Some(info) = maybe_info {
rows.push(
info.outline(2.0, Color::WHITE)
.bg(app.cs.inner_panel)
.padding(16),
);
let mut new_trips = open_trips.clone();
new_trips.remove(t);
details
.hyperlinks
.insert(format!("hide {}", t), Tab::PersonTrips(id, new_trips));
} else {
let mut new_trips = open_trips.clone();
new_trips.insert(*t, OpenTrip::new());
details
.hyperlinks
.insert(format!("show {}", t), Tab::PersonTrips(id, new_trips));
}
}
if wheres_waldo {
rows.push(current_status(ctx, person, map));
}
rows
}
pub fn bio(
ctx: &mut EventCtx,
app: &App,
details: &mut Details,
id: PersonID,
is_paused: bool,
) -> Vec<Widget> {
let mut rows = header(ctx, app, details, id, Tab::PersonBio(id), is_paused);
let person = app.primary.sim.get_person(id);
let mut rng = XorShiftRng::seed_from_u64(id.0 as u64);
let mut svg_data = Vec::new();
svg_face::generate_face(&mut svg_data, &mut rng).unwrap();
let batch = GeomBatch::from_svg_contents(svg_data).autocrop();
let dims = batch.get_dims();
let batch = batch.scale((200.0 / dims.width).min(200.0 / dims.height));
rows.push(Widget::draw_batch(ctx, batch).centered_horiz());
let nickname = petname::Petnames::default().generate(&mut rng, 2, " ");
let age = rng.gen_range(5, 100);
let mut table = vec![("Nickname", nickname), ("Age", age.to_string())];
if app.opts.dev {
table.push(("Debug ID", format!("{:?}", person.orig_id)));
}
rows.extend(make_table(ctx, table));
if let Some(p) = app.primary.sim.get_pandemic_model() {
let status = if p.is_sane(id) {
"Susceptible".to_string()
} else if p.is_exposed(id) {
format!("Exposed at {}", p.get_time(id).unwrap().ampm_tostring())
} else if p.is_infectious(id) {
format!("Infected at {}", p.get_time(id).unwrap().ampm_tostring())
} else if p.is_recovered(id) {
format!("Recovered at {}", p.get_time(id).unwrap().ampm_tostring())
} else if p.is_dead(id) {
format!("Dead at {}", p.get_time(id).unwrap().ampm_tostring())
} else {
"Other (hospitalized or quarantined)".to_string()
};
rows.push(
Text::from_all(vec![
Line("Pandemic model state: ").secondary(),
Line(status),
])
.draw(ctx),
);
}
let mut has_bike = false;
for v in &person.vehicles {
if v.vehicle_type == VehicleType::Bike {
has_bike = true;
} else {
if app.primary.sim.lookup_parked_car(v.id).is_some() {
rows.push(
Btn::text_bg2(format!("Owner of {} (parked)", v.id)).build_def(ctx, None),
);
details
.hyperlinks
.insert(format!("Owner of {} (parked)", v.id), Tab::ParkedCar(v.id));
} else if let PersonState::Trip(t) = person.state {
match app.primary.sim.trip_to_agent(t) {
TripResult::Ok(AgentID::Car(x)) if x == v.id => {
rows.push(format!("Owner of {} (currently driving)", v.id).draw_text(ctx));
}
_ => {
rows.push(format!("Owner of {} (off-map)", v.id).draw_text(ctx));
}
}
} else {
rows.push(format!("Owner of {} (off-map)", v.id).draw_text(ctx));
}
}
}
if has_bike {
rows.push("Owns a bike".draw_text(ctx));
}
rows
}
pub fn schedule(
ctx: &mut EventCtx,
app: &App,
details: &mut Details,
id: PersonID,
is_paused: bool,
) -> Vec<Widget> {
let mut rows = header(ctx, app, details, id, Tab::PersonSchedule(id), is_paused);
let person = app.primary.sim.get_person(id);
let mut rng = XorShiftRng::seed_from_u64(id.0 as u64);
let mut last_t = Time::START_OF_DAY;
for t in &person.trips {
let trip = app.primary.sim.trip_info(*t);
let at = match trip.start {
TripEndpoint::Bldg(b) => {
let b = app.primary.map.get_b(b);
if b.amenities.is_empty() {
b.address.clone()
} else {
let list = b
.amenities
.iter()
.map(|a| a.names.get(app.opts.language.as_ref()))
.collect::<Vec<_>>();
format!("{} (at {})", list.choose(&mut rng).unwrap(), b.address)
}
}
TripEndpoint::Border(_) => "off-map".to_string(),
TripEndpoint::SuddenlyAppear(_) => "suddenly appear".to_string(),
};
rows.push(
Text::from(Line(format!(
" Spends {} at {}",
trip.departure - last_t,
at
)))
.draw(ctx),
);
last_t = trip.departure;
}
let last_trip = app.primary.sim.trip_info(*person.trips.last().unwrap());
let at = match last_trip.end {
TripEndpoint::Bldg(b) => {
let b = app.primary.map.get_b(b);
if b.amenities.is_empty() {
b.address.clone()
} else {
let list = b
.amenities
.iter()
.map(|a| a.names.get(app.opts.language.as_ref()))
.collect::<Vec<_>>();
format!("{} (at {})", list.choose(&mut rng).unwrap(), b.address)
}
}
TripEndpoint::Border(_) => "off-map".to_string(),
TripEndpoint::SuddenlyAppear(_) => "suddenly disappear".to_string(),
};
rows.push(
Text::from(Line(format!(
" Spends {} at {}",
app.primary.sim.get_end_of_day() - last_trip.departure,
at
)))
.draw(ctx),
);
rows
}
pub fn crowd(
ctx: &EventCtx,
app: &App,
details: &mut Details,
members: &Vec<PedestrianID>,
) -> Vec<Widget> {
let mut rows = vec![];
rows.push(Widget::row(vec![
Line("Pedestrian crowd").small_heading().draw(ctx),
header_btns(ctx),
]));
for (idx, id) in members.into_iter().enumerate() {
let person = app
.primary
.sim
.agent_to_person(AgentID::Pedestrian(*id))
.unwrap();
rows.push(Widget::row(vec![
format!("{})", idx + 1).draw_text(ctx).centered_vert(),
Btn::text_fg(person.to_string()).build_def(ctx, None),
]));
details.hyperlinks.insert(
person.to_string(),
Tab::PersonTrips(
person,
OpenTrip::single(
app.primary
.sim
.agent_to_trip(AgentID::Pedestrian(*id))
.unwrap(),
),
),
);
}
rows
}
pub fn parked_car(
ctx: &mut EventCtx,
app: &App,
details: &mut Details,
id: CarID,
is_paused: bool,
) -> Vec<Widget> {
let mut rows = vec![];
rows.push(Widget::row(vec![
Line(format!("Parked car #{}", id.0))
.small_heading()
.draw(ctx),
Widget::row(vec![
if is_paused {
Btn::svg_def("system/assets/tools/location.svg").build(
ctx,
"follow (run the simulation)",
Key::F,
)
} else {
GeomBatch::load_svg(ctx.prerender, "system/assets/tools/location.svg")
.color(RewriteColor::ChangeAll(Color::hex("#7FFA4D")))
.to_btn(ctx)
.build(ctx, "unfollow (pause the simulation)", Key::F)
},
Btn::close(ctx),
])
.align_right(),
]));
let p = app.primary.sim.get_owner_of_car(id).unwrap();
rows.push(Btn::text_bg2(format!("Owned by {}", p)).build_def(ctx, None));
details.hyperlinks.insert(
format!("Owned by {}", p),
Tab::PersonTrips(p, BTreeMap::new()),
);
if let Some(p) = app.primary.sim.lookup_parked_car(id) {
match p.spot {
ParkingSpot::Onstreet(_, _) | ParkingSpot::Lot(_, _) => {
ctx.canvas.center_on_map_pt(
app.primary
.sim
.canonical_pt_for_agent(AgentID::Car(id), &app.primary.map)
.unwrap(),
);
}
ParkingSpot::Offstreet(b, _) => {
ctx.canvas
.center_on_map_pt(app.primary.map.get_b(b).polygon.center());
rows.push(
format!("Parked inside {}", app.primary.map.get_b(b).address).draw_text(ctx),
);
}
}
rows.push(
format!(
"Parked here for {}",
app.primary.sim.time() - p.parked_since
)
.draw_text(ctx),
);
} else {
rows.push("No longer parked".draw_text(ctx));
}
rows
}
fn header(
ctx: &mut EventCtx,
app: &App,
details: &mut Details,
id: PersonID,
tab: Tab,
is_paused: bool,
) -> Vec<Widget> {
let mut rows = vec![];
let (current_trip, (descr, maybe_icon)) = match app.primary.sim.get_person(id).state {
PersonState::Inside(b) => {
ctx.canvas
.center_on_map_pt(app.primary.map.get_b(b).label_center);
building::draw_occupants(details, app, b, Some(id));
(None, ("indoors", Some("system/assets/tools/home.svg")))
}
PersonState::Trip(t) => (
Some(t),
if let Some(a) = app.primary.sim.trip_to_agent(t).ok() {
if let Some(pt) = app.primary.sim.canonical_pt_for_agent(a, &app.primary.map) {
ctx.canvas.center_on_map_pt(pt);
}
match a {
AgentID::Pedestrian(_) => {
("walking", Some("system/assets/meters/pedestrian.svg"))
}
AgentID::Car(c) => match c.1 {
VehicleType::Car => ("driving", Some("system/assets/meters/car.svg")),
VehicleType::Bike => ("biking", Some("system/assets/meters/bike.svg")),
VehicleType::Bus | VehicleType::Train => unreachable!(),
},
AgentID::BusPassenger(_, _) => {
("riding a bus", Some("system/assets/meters/bus.svg"))
}
}
} else {
("...", None)
},
),
PersonState::OffMap => (None, ("off map", None)),
};
rows.push(Widget::custom_row(vec![
Line(format!("{}", id)).small_heading().draw(ctx),
if let Some(icon) = maybe_icon {
Widget::draw_svg_transform(ctx, icon, RewriteColor::ChangeAll(Color::hex("#A3A3A3")))
.margin_left(28)
} else {
Widget::nothing()
},
Line(format!("{}", descr))
.small_heading()
.fg(Color::hex("#A3A3A3"))
.draw(ctx)
.margin_horiz(10),
Widget::row(vec![
if is_paused {
Btn::svg_def("system/assets/tools/location.svg").build(
ctx,
"follow (run the simulation)",
Key::F,
)
} else {
GeomBatch::load_svg(ctx.prerender, "system/assets/tools/location.svg")
.color(RewriteColor::ChangeAll(Color::hex("#7FFA4D")))
.to_btn(ctx)
.build(ctx, "unfollow (pause the simulation)", Key::F)
},
Btn::close(ctx),
])
.align_right(),
]));
let open_trips = if let Some(t) = current_trip {
OpenTrip::single(t)
} else {
BTreeMap::new()
};
let mut tabs = vec![
("Trips", Tab::PersonTrips(id, open_trips)),
("Bio", Tab::PersonBio(id)),
];
if app.opts.dev {
tabs.push(("Schedule", Tab::PersonSchedule(id)));
}
rows.push(make_tabs(ctx, &mut details.hyperlinks, tab, tabs));
rows
}
fn current_status(ctx: &EventCtx, person: &Person, map: &Map) -> Widget {
(match person.state {
PersonState::Inside(b) => {
format!("Currently inside {}", map.get_b(b).address).draw_text(ctx)
}
PersonState::Trip(_) => unreachable!(),
PersonState::OffMap => "Currently outside the map boundaries".draw_text(ctx),
})
.margin_vert(16)
}
fn cmp_duration_shorter(after: Duration, before: Duration) -> TextSpan {
if after.epsilon_eq(before) {
Line("no change").small()
} else if after < before {
Line(format!("{} faster", before - after))
.small()
.fg(Color::GREEN)
} else if after > before {
Line(format!("{} slower", after - before))
.small()
.fg(Color::RED)
} else {
unreachable!()
}
}