use std::collections::BTreeSet;
use maplit::btreeset;
use rand::seq::SliceRandom;
use rand::SeedableRng;
use rand_xorshift::XorShiftRng;
use geom::Duration;
use map_gui::tools::{grey_out_map, nice_map_name, open_browser, PopupMsg};
use map_model::AreaType;
use sim::{AgentType, PersonID, TripEndpoint, TripID};
use widgetry::{
lctrl, ControlState, EventCtx, GfxCtx, HorizontalAlignment, Key, Line, Outcome, Panel,
SimpleState, Text, TextExt, Toggle, VerticalAlignment, Widget,
};
use crate::app::{App, Transition};
use crate::common::jump_to_time_upon_startup;
use crate::edit::EditMode;
use crate::info::{OpenTrip, Tab};
use crate::sandbox::gameplay::{GameplayMode, GameplayState};
use crate::sandbox::{Actions, SandboxControls, SandboxMode, SpeedSetting};
pub struct Actdev {
top_right: Panel,
scenario_name: String,
bg_traffic: bool,
once: bool,
}
impl Actdev {
pub fn new_state(
ctx: &mut EventCtx,
scenario_name: String,
bg_traffic: bool,
) -> Box<dyn GameplayState> {
Box::new(Actdev {
top_right: Panel::empty(ctx),
scenario_name,
bg_traffic,
once: true,
})
}
}
impl GameplayState for Actdev {
fn event(
&mut self,
ctx: &mut EventCtx,
app: &mut App,
controls: &mut SandboxControls,
actions: &mut Actions,
) -> Option<Transition> {
if self.once {
self.once = false;
if self.bg_traffic {
let mut highlight = BTreeSet::new();
let study_area = &app
.primary
.map
.all_areas()
.iter()
.find(|a| a.area_type == AreaType::StudyArea)
.unwrap()
.polygon;
for person in app.primary.sim.get_all_people() {
if let TripEndpoint::Bldg(b) = app.primary.sim.trip_info(person.trips[0]).start
{
if study_area.contains_pt(app.primary.map.get_b(b).polygon.center()) {
highlight.insert(person.id);
}
}
}
app.primary.sim.set_highlighted_people(highlight);
}
controls.time_panel.as_mut().unwrap().override_height =
Some(self.top_right.panel_dims().height);
controls
.time_panel
.as_mut()
.unwrap()
.resume(ctx, app, SpeedSetting::Faster);
}
match self.top_right.event(ctx) {
Outcome::Clicked(x) => match x.as_ref() {
"change scenario" => {
let scenario = if self.scenario_name == "base" {
"go_active"
} else {
"base"
};
return Some(Transition::Replace(SandboxMode::async_new(
app,
GameplayMode::Actdev(
app.primary.map.get_name().clone(),
scenario.to_string(),
self.bg_traffic,
),
jump_to_time_upon_startup(Duration::hours(8)),
)));
}
"Edit map" => Some(Transition::Push(EditMode::new_state(
ctx,
app,
GameplayMode::Actdev(
app.primary.map.get_name().clone(),
self.scenario_name.clone(),
self.bg_traffic,
),
))),
"about A/B Street" => {
let panel = Panel::new_builder(Widget::col(vec![
Widget::row(vec![
Line("About A/B Street").small_heading().into_widget(ctx),
ctx.style().btn_close_widget(ctx),
]),
Line("Created by Dustin Carlino, Yuwen Li, & Michael Kirk")
.small()
.into_widget(ctx),
Text::from(
"A/B Street is a traffic simulation game based on OpenStreetMap. You \
can modify roads and intersections, measure the effects on different \
groups, and advocate for your proposal.",
)
.wrap_to_pct(ctx, 50)
.into_widget(ctx),
"This is a simplified version. Check out the full version below."
.text_widget(ctx),
ctx.style().btn_outline.text("abstreet.org").build_def(ctx),
]))
.build(ctx);
Some(Transition::Push(<dyn SimpleState<_>>::new_state(
panel,
Box::new(About),
)))
}
"Follow someone" => {
if let Some((person, trip)) = find_active_trip(app) {
app.primary.layer = None;
ctx.canvas.cam_zoom = 40.0;
controls.common.as_mut().unwrap().launch_info_panel(
ctx,
app,
Tab::PersonTrips(person, OpenTrip::single(trip)),
actions,
);
None
} else {
return Some(Transition::Push(PopupMsg::new_state(
ctx,
"Nobody's around...",
vec!["There are no active trips right now"],
)));
}
}
"Cycling" => {
app.primary.layer =
Some(Box::new(crate::layer::map::BikeActivity::new(ctx, app)));
None
}
"Walking" => {
app.primary.layer = Some(Box::new(crate::layer::traffic::Throughput::new(
ctx,
app,
btreeset! { AgentType::Pedestrian },
)));
None
}
_ => unreachable!(),
},
Outcome::Changed(_) => {
return Some(Transition::Replace(SandboxMode::async_new(
app,
GameplayMode::Actdev(
app.primary.map.get_name().clone(),
self.scenario_name.clone(),
!self.bg_traffic,
),
jump_to_time_upon_startup(Duration::hours(8)),
)));
}
_ => None,
}
}
fn draw(&self, g: &mut GfxCtx, _: &App) {
self.top_right.draw(g);
}
fn recreate_panels(&mut self, ctx: &mut EventCtx, app: &App) {
let col = Widget::col(vec![
Widget::row(vec![
ctx.style()
.btn_plain
.btn()
.image_path("system/assets/pregame/logo.svg")
.image_dims(50.0)
.build_widget(ctx, "about A/B Street"),
Line(nice_map_name(app.primary.map.get_name()))
.small_heading()
.into_widget(ctx),
ctx.style()
.btn_outline
.icon_text("system/assets/tools/pencil.svg", "Edit map")
.hotkey(lctrl(Key::E))
.build_def(ctx),
])
.centered(),
Widget::row(vec![
ctx.style()
.btn_popup_icon_text("system/assets/tools/calendar.svg", "scenario")
.label_styled_text(
match self.scenario_name.as_ref() {
"base" => Text::from_all(vec![
Line("Baseline / "),
Line("Go Active").secondary(),
]),
"go_active" => Text::from_all(vec![
Line("Baseline").secondary(),
Line(" / Go Active"),
]),
_ => unreachable!(),
},
ControlState::Default,
)
.build_widget(ctx, "change scenario"),
Toggle::checkbox(ctx, "background traffic", None, self.bg_traffic),
]),
Widget::row(vec![
ctx.style()
.btn_plain
.icon_text("system/assets/tools/location.svg", "Follow someone")
.build_def(ctx),
ctx.style()
.btn_plain
.icon_text("system/assets/meters/pedestrian.svg", "Walking")
.build_def(ctx),
ctx.style()
.btn_plain
.icon_text("system/assets/meters/bike.svg", "Cycling")
.build_def(ctx),
]),
]);
self.top_right = Panel::new_builder(col)
.aligned(HorizontalAlignment::Right, VerticalAlignment::Top)
.build(ctx);
}
fn has_tool_panel(&self) -> bool {
false
}
}
struct About;
impl SimpleState<App> for About {
fn on_click(&mut self, _: &mut EventCtx, _: &mut App, x: &str, _: &Panel) -> Transition {
if x == "close" {
return Transition::Pop;
} else if x == "abstreet.org" {
open_browser("https://abstreet.org");
}
Transition::Keep
}
fn draw(&self, g: &mut GfxCtx, app: &App) {
grey_out_map(g, app);
}
}
fn find_active_trip(app: &App) -> Option<(PersonID, TripID)> {
let mut all = Vec::new();
for agent in app.primary.sim.active_agents() {
if let Some(trip) = app.primary.sim.agent_to_trip(agent) {
if let Some(person) = app.primary.sim.trip_to_person(trip) {
all.push((person, trip));
}
}
}
all.choose(&mut XorShiftRng::from_entropy()).cloned()
}