use crate::app::{App, FindDelayedIntersections, ShowEverything};
use crate::common::Warping;
use crate::game::{DrawBaselayer, PopupMsg, State, Transition};
use crate::helpers::ID;
use crate::render::DrawOptions;
use crate::sandbox::{GameplayMode, SandboxMode};
use abstutil::prettyprint_usize;
use geom::{Duration, Polygon, Pt2D, Ring, Time};
use instant::Instant;
use sim::AlertLocation;
use widgetry::{
hotkey, AreaSlider, Btn, Checkbox, Choice, Color, EventCtx, GeomBatch, GfxCtx,
HorizontalAlignment, Key, Line, Outcome, Panel, PersistentSplit, RewriteColor, Text,
UpdateType, VerticalAlignment, Widget,
};
pub struct SpeedControls {
pub panel: Panel,
paused: bool,
setting: SpeedSetting,
}
#[derive(Clone, Copy, PartialEq, PartialOrd)]
enum SpeedSetting {
Realtime,
Fast,
Faster,
Fastest,
}
impl SpeedControls {
fn make_panel(ctx: &mut EventCtx, app: &App, paused: bool, setting: SpeedSetting) -> Panel {
let mut row = Vec::new();
row.push(
if paused {
Btn::svg_def("system/assets/speed/triangle.svg").build(
ctx,
"play",
hotkey(Key::Space),
)
} else {
Btn::svg_def("system/assets/speed/pause.svg").build(
ctx,
"pause",
hotkey(Key::Space),
)
}
.container()
.padding(9)
.bg(app.cs.section_bg)
.margin_right(16),
);
row.push(
Widget::custom_row(
vec![
(SpeedSetting::Realtime, "real-time speed"),
(SpeedSetting::Fast, "5x speed"),
(SpeedSetting::Faster, "30x speed"),
(SpeedSetting::Fastest, "3600x speed"),
]
.into_iter()
.map(|(s, label)| {
let mut txt = Text::from(Line(label).small());
txt.extend(Text::tooltip(ctx, hotkey(Key::LeftArrow), "slow down"));
txt.extend(Text::tooltip(ctx, hotkey(Key::RightArrow), "speed up"));
GeomBatch::load_svg(ctx.prerender, "system/assets/speed/triangle.svg")
.color(if setting >= s {
RewriteColor::NoOp
} else {
RewriteColor::ChangeAll(Color::WHITE.alpha(0.2))
})
.to_btn(ctx)
.tooltip(txt)
.build(ctx, label, None)
.margin_right(6)
})
.collect(),
)
.bg(app.cs.section_bg)
.centered()
.padding(6)
.margin_right(16),
);
row.push(
PersistentSplit::new(
ctx,
"step forwards",
app.opts.time_increment,
hotkey(Key::M),
vec![
Choice::new("+1h", Duration::hours(1)),
Choice::new("+30m", Duration::minutes(30)),
Choice::new("+10m", Duration::minutes(10)),
Choice::new("+0.1s", Duration::seconds(0.1)),
],
)
.bg(app.cs.section_bg)
.margin_right(16),
);
row.push(
Widget::custom_row(vec![
Btn::svg_def("system/assets/speed/jump_to_time.svg")
.build(ctx, "jump to specific time", hotkey(Key::B))
.container()
.padding(9),
Btn::svg_def("system/assets/speed/reset.svg")
.build(ctx, "reset to midnight", hotkey(Key::X))
.container()
.padding(9),
])
.bg(app.cs.section_bg),
);
Panel::new(Widget::custom_row(row))
.aligned(
HorizontalAlignment::Center,
VerticalAlignment::BottomAboveOSD,
)
.build(ctx)
}
pub fn new(ctx: &mut EventCtx, app: &App) -> SpeedControls {
let panel = SpeedControls::make_panel(ctx, app, false, SpeedSetting::Realtime);
SpeedControls {
panel,
paused: false,
setting: SpeedSetting::Realtime,
}
}
pub fn event(
&mut self,
ctx: &mut EventCtx,
app: &mut App,
maybe_mode: Option<&GameplayMode>,
) -> Option<Transition> {
match self.panel.event(ctx) {
Outcome::Clicked(x) => match x.as_ref() {
"real-time speed" => {
self.setting = SpeedSetting::Realtime;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
return None;
}
"5x speed" => {
self.setting = SpeedSetting::Fast;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
return None;
}
"30x speed" => {
self.setting = SpeedSetting::Faster;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
return None;
}
"3600x speed" => {
self.setting = SpeedSetting::Fastest;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
return None;
}
"play" => {
self.paused = false;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
return None;
}
"pause" => {
self.pause(ctx, app);
}
"reset to midnight" => {
if let Some(mode) = maybe_mode {
return Some(Transition::Replace(SandboxMode::new(
ctx,
app,
mode.clone(),
)));
} else {
return Some(Transition::Push(PopupMsg::new(
ctx,
"Error",
vec!["Sorry, you can't go rewind time from this mode."],
)));
}
}
"jump to specific time" => {
return Some(Transition::Push(Box::new(JumpToTime::new(
ctx,
app,
maybe_mode.cloned(),
))));
}
"step forwards" => {
let dt = self.panel.persistent_split_value("step forwards");
if dt == Duration::seconds(0.1) {
app.primary
.sim
.tiny_step(&app.primary.map, &mut app.primary.sim_cb);
app.recalculate_current_selection(ctx);
return Some(Transition::KeepWithMouseover);
}
return Some(Transition::Push(TimeWarpScreen::new(
ctx,
app,
app.primary.sim.time() + dt,
None,
)));
}
_ => unreachable!(),
},
_ => {}
}
app.opts.time_increment = self.panel.persistent_split_value("step forwards");
if ctx.input.key_pressed(Key::LeftArrow) {
match self.setting {
SpeedSetting::Realtime => self.pause(ctx, app),
SpeedSetting::Fast => {
self.setting = SpeedSetting::Realtime;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
SpeedSetting::Faster => {
self.setting = SpeedSetting::Fast;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
SpeedSetting::Fastest => {
self.setting = SpeedSetting::Faster;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
}
}
if ctx.input.key_pressed(Key::RightArrow) {
match self.setting {
SpeedSetting::Realtime => {
if self.paused {
self.paused = false;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
} else {
self.setting = SpeedSetting::Fast;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
}
SpeedSetting::Fast => {
self.setting = SpeedSetting::Faster;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
SpeedSetting::Faster => {
self.setting = SpeedSetting::Fastest;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
SpeedSetting::Fastest => {}
}
}
if !self.paused {
if let Some(real_dt) = ctx.input.nonblocking_is_update_event() {
ctx.input.use_update_event();
let multiplier = match self.setting {
SpeedSetting::Realtime => 1.0,
SpeedSetting::Fast => 5.0,
SpeedSetting::Faster => 30.0,
SpeedSetting::Fastest => 3600.0,
};
let dt = multiplier * real_dt;
app.primary.sim.time_limited_step(
&app.primary.map,
dt,
Duration::seconds(0.033),
&mut app.primary.sim_cb,
);
app.recalculate_current_selection(ctx);
}
}
let alerts = app.primary.sim.clear_alerts();
if !alerts.is_empty() {
let popup = PopupMsg::new(
ctx,
"Alerts",
alerts.iter().map(|(_, _, msg)| msg).collect(),
);
let maybe_id = match alerts[0].1 {
AlertLocation::Nil => None,
AlertLocation::Intersection(i) => Some(ID::Intersection(i)),
AlertLocation::Person(_) => None,
AlertLocation::Building(b) => Some(ID::Building(b)),
};
self.pause(ctx, app);
if let Some(id) = maybe_id {
return Some(Transition::Multi(vec![
Transition::Push(popup),
Transition::Push(Warping::new(
ctx,
id.canonical_point(&app.primary).unwrap(),
Some(10.0),
None,
&mut app.primary,
)),
]));
} else {
return Some(Transition::Push(popup));
}
}
None
}
pub fn draw(&self, g: &mut GfxCtx) {
self.panel.draw(g);
}
pub fn pause(&mut self, ctx: &mut EventCtx, app: &App) {
if !self.paused {
self.paused = true;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
}
pub fn resume_realtime(&mut self, ctx: &mut EventCtx, app: &App) {
if self.paused || self.setting != SpeedSetting::Realtime {
self.paused = false;
self.setting = SpeedSetting::Realtime;
self.panel = SpeedControls::make_panel(ctx, app, self.paused, self.setting);
}
}
pub fn is_paused(&self) -> bool {
self.paused
}
}
struct JumpToTime {
panel: Panel,
target: Time,
halt_limit: Duration,
maybe_mode: Option<GameplayMode>,
}
impl JumpToTime {
fn new(ctx: &mut EventCtx, app: &App, maybe_mode: Option<GameplayMode>) -> JumpToTime {
let target = app.primary.sim.time();
let end_of_day = app.primary.sim.get_end_of_day();
let halt_limit = app.opts.time_warp_halt_limit;
JumpToTime {
target,
halt_limit,
maybe_mode,
panel: Panel::new(Widget::col(vec![
Widget::row(vec![
Line("Jump to what time?").small_heading().draw(ctx),
Btn::plaintext("X")
.build(ctx, "close", hotkey(Key::Escape))
.align_right(),
]),
Checkbox::checkbox(
ctx,
"skip drawing (for faster simulations)",
None,
app.opts.dont_draw_time_warp,
)
.margin_above(30)
.named("don't draw"),
Widget::horiz_separator(ctx, 0.25).margin_above(10),
if app.has_prebaked().is_some() {
Widget::draw_batch(
ctx,
GeomBatch::from(vec![(
Color::WHITE.alpha(0.7),
area_under_curve(
app.prebaked().active_agents(end_of_day),
500.0,
50.0,
),
)]),
)
} else {
Widget::nothing()
},
AreaSlider::new(
ctx,
0.25 * ctx.canvas.window_width,
target.to_percent(end_of_day).min(1.0),
)
.named("time slider")
.centered_horiz()
.margin_above(16),
build_jump_to_time_btn(target, ctx),
Widget::horiz_separator(ctx, 0.25).margin_above(16),
build_jump_to_delay_picker(halt_limit, ctx).margin_above(16),
build_jump_to_delay_button(halt_limit, ctx),
]))
.build(ctx),
}
}
}
impl State for JumpToTime {
fn event(&mut self, ctx: &mut EventCtx, app: &mut App) -> Transition {
match self.panel.event(ctx) {
Outcome::Clicked(x) => match x.as_ref() {
"close" => {
return Transition::Pop;
}
"jump to time" => {
if self.target < app.primary.sim.time() {
if let Some(mode) = self.maybe_mode.take() {
return Transition::Multi(vec![
Transition::Replace(SandboxMode::new(ctx, app, mode)),
Transition::Push(TimeWarpScreen::new(ctx, app, self.target, None)),
]);
} else {
return Transition::Replace(PopupMsg::new(
ctx,
"Error",
vec!["Sorry, you can't go rewind time from this mode."],
));
}
}
return Transition::Replace(TimeWarpScreen::new(ctx, app, self.target, None));
}
"choose delay" => return Transition::Keep,
"jump to delay" => {
let halt_limit = self.panel.persistent_split_value("choose delay");
app.opts.time_warp_halt_limit = halt_limit;
return Transition::Replace(TimeWarpScreen::new(
ctx,
app,
app.primary.sim.get_end_of_day(),
Some(halt_limit),
));
}
_ => unreachable!(),
},
Outcome::Changed => {
app.opts.dont_draw_time_warp = self.panel.is_checked("don't draw");
}
_ => {}
}
let target = app
.primary
.sim
.get_end_of_day()
.percent_of(self.panel.area_slider("time slider").get_percent())
.round_seconds(600.0);
if target != self.target {
self.target = target;
self.panel
.replace(ctx, "jump to time", build_jump_to_time_btn(target, ctx));
}
let halt_limit = self.panel.persistent_split_value("choose delay");
if halt_limit != self.halt_limit {
self.halt_limit = halt_limit;
self.panel.replace(
ctx,
"jump to delay",
build_jump_to_delay_button(halt_limit, ctx),
);
}
if self.panel.clicked_outside(ctx) {
return Transition::Pop;
}
Transition::Keep
}
fn draw(&self, g: &mut GfxCtx, app: &App) {
State::grey_out_map(g, app);
self.panel.draw(g);
}
}
pub struct TimeWarpScreen {
target: Time,
wall_time_started: Instant,
sim_time_started: geom::Time,
halt_upon_delay: Option<Duration>,
panel: Panel,
}
impl TimeWarpScreen {
pub fn new(
ctx: &mut EventCtx,
app: &mut App,
target: Time,
mut halt_upon_delay: Option<Duration>,
) -> Box<dyn State> {
if let Some(halt_limit) = halt_upon_delay {
if app.primary.sim_cb.is_none() {
app.primary.sim_cb = Some(Box::new(FindDelayedIntersections {
halt_limit,
report_limit: halt_limit,
currently_delayed: Vec::new(),
}));
app.primary.sim.set_periodic_callback(Duration::minutes(1));
} else {
halt_upon_delay = None;
}
}
Box::new(TimeWarpScreen {
target,
wall_time_started: Instant::now(),
sim_time_started: app.primary.sim.time(),
halt_upon_delay,
panel: Panel::new(
Widget::col(vec![
Text::new().draw(ctx).named("text"),
Btn::text_bg2("stop now")
.build_def(ctx, hotkey(Key::Escape))
.centered_horiz(),
])
.force_width(700.0),
)
.build(ctx),
})
}
}
impl State for TimeWarpScreen {
fn event(&mut self, ctx: &mut EventCtx, app: &mut App) -> Transition {
if ctx.input.nonblocking_is_update_event().is_some() {
ctx.input.use_update_event();
app.primary.sim.time_limited_step(
&app.primary.map,
self.target - app.primary.sim.time(),
Duration::seconds(0.033),
&mut app.primary.sim_cb,
);
for (t, maybe_i, alert) in app.primary.sim.clear_alerts() {
return Transition::Replace(PopupMsg::new(
ctx,
"Alert",
vec![format!("At {}, near {:?}, {}", t, maybe_i, alert)],
));
}
if let Some(ref mut cb) = app.primary.sim_cb {
let di = cb.downcast_mut::<FindDelayedIntersections>().unwrap();
if let Some((i, t)) = di.currently_delayed.get(0) {
if app.primary.sim.time() - *t > di.halt_limit {
let id = ID::Intersection(*i);
app.layer =
Some(Box::new(crate::layer::traffic::TrafficJams::new(ctx, app)));
return Transition::Replace(Warping::new(
ctx,
id.canonical_point(&app.primary).unwrap(),
Some(10.0),
Some(id),
&mut app.primary,
));
}
}
}
let now = app.primary.sim.time();
let (finished_after, _) = app.primary.sim.num_trips();
let finished_before = if app.has_prebaked().is_some() {
let mut cnt = 0;
for (t, _, _, _) in &app.prebaked().finished_trips {
if *t > now {
break;
}
cnt += 1;
}
Some(cnt)
} else {
None
};
let elapsed_sim_time = now - self.sim_time_started;
let elapsed_wall_time = Duration::realtime_elapsed(self.wall_time_started);
let txt = Text::from_multiline(vec![
Line("Let's do the time warp again!").small_heading(),
Line(format!(
"{} / {}",
now.ampm_tostring(),
self.target.ampm_tostring()
)),
Line(format!(
"Speed: {}x",
prettyprint_usize((elapsed_sim_time / elapsed_wall_time) as usize)
)),
if let Some(n) = finished_before {
Line(format!(
"Finished trips: {} ({} compared to before \"{}\")",
prettyprint_usize(finished_after),
compare_count(finished_after, n),
app.primary.map.get_edits().edits_name,
))
} else {
Line(format!(
"Finished trips: {}",
prettyprint_usize(finished_after)
))
},
]);
self.panel.replace(ctx, "text", txt.draw(ctx).named("text"));
}
if app.primary.sim.time() == self.target {
return Transition::Pop;
}
match self.panel.event(ctx) {
Outcome::Clicked(x) => match x.as_ref() {
"stop now" => {
return Transition::Pop;
}
_ => unreachable!(),
},
_ => {}
}
if self.panel.clicked_outside(ctx) {
return Transition::Pop;
}
ctx.request_update(UpdateType::Game);
Transition::Keep
}
fn draw_baselayer(&self) -> DrawBaselayer {
DrawBaselayer::Custom
}
fn draw(&self, g: &mut GfxCtx, app: &App) {
if app.opts.dont_draw_time_warp {
g.clear(app.cs.section_bg);
} else {
app.draw(
g,
DrawOptions::new(),
&app.primary.sim,
&ShowEverything::new(),
);
State::grey_out_map(g, app);
}
self.panel.draw(g);
}
fn on_destroy(&mut self, _: &mut EventCtx, app: &mut App) {
if self.halt_upon_delay.is_some() {
assert!(app.primary.sim_cb.is_some());
app.primary.sim_cb = None;
app.primary.sim.unset_periodic_callback();
}
}
}
pub struct TimePanel {
time: Time,
pub panel: Panel,
}
impl TimePanel {
pub fn new(ctx: &mut EventCtx, app: &App) -> TimePanel {
TimePanel {
time: app.primary.sim.time(),
panel: Panel::new(Widget::col(vec![
Text::from(Line(app.primary.sim.time().ampm_tostring()).big_monospaced())
.draw(ctx)
.centered_horiz(),
{
let mut batch = GeomBatch::new();
let width = 300.0;
let height = 15.0;
let percent = app
.primary
.sim
.time()
.to_percent(app.primary.sim.get_end_of_day())
.min(1.0);
batch.push(Color::WHITE, Polygon::rectangle(width, height));
if percent != 0.0 {
batch.push(
if percent < 0.25 || percent > 0.75 {
app.cs.night_time_slider
} else {
app.cs.day_time_slider
},
Polygon::rectangle(percent * width, height),
);
}
Widget::draw_batch(ctx, batch)
},
Widget::custom_row(vec![
Line("00:00").small_monospaced().draw(ctx),
Widget::draw_svg(ctx, "system/assets/speed/sunrise.svg"),
Line("12:00").small_monospaced().draw(ctx),
Widget::draw_svg(ctx, "system/assets/speed/sunset.svg"),
Line("24:00").small_monospaced().draw(ctx),
])
.evenly_spaced(),
]))
.aligned(HorizontalAlignment::Left, VerticalAlignment::Top)
.build(ctx),
}
}
pub fn event(&mut self, ctx: &mut EventCtx, app: &mut App) {
if self.time != app.primary.sim.time() {
*self = TimePanel::new(ctx, app);
}
self.panel.event(ctx);
}
pub fn draw(&self, g: &mut GfxCtx) {
self.panel.draw(g);
}
}
fn area_under_curve(raw: Vec<(Time, usize)>, width: f64, height: f64) -> Polygon {
assert!(!raw.is_empty());
let min_x = Time::START_OF_DAY;
let min_y = 0;
let max_x = raw.last().unwrap().0;
let max_y = raw.iter().max_by_key(|(_, cnt)| *cnt).unwrap().1;
let mut pts = Vec::new();
for (t, cnt) in raw {
pts.push(lttb::DataPoint::new(
width * (t - min_x) / (max_x - min_x),
height * (1.0 - (((cnt - min_y) as f64) / ((max_y - min_y) as f64))),
));
}
let mut downsampled = Vec::new();
for pt in lttb::lttb(pts, 100) {
downsampled.push(Pt2D::new(pt.x, pt.y));
}
downsampled.push(Pt2D::new(width, height));
downsampled.push(downsampled[0]);
Ring::must_new(downsampled).to_polygon()
}
fn compare_count(after: usize, before: usize) -> String {
if after == before {
"+0".to_string()
} else if after > before {
format!("+{}", prettyprint_usize(after - before))
} else {
format!("-{}", prettyprint_usize(before - after))
}
}
fn build_jump_to_time_btn(target: Time, ctx: &EventCtx) -> Widget {
Btn::text_bg2(format!("Jump to {}", target.ampm_tostring()))
.build(ctx, "jump to time", hotkey(Key::Enter))
.named("jump to time")
.centered_horiz()
.margin_above(16)
}
fn build_jump_to_delay_button(halt_limit: Duration, ctx: &EventCtx) -> Widget {
Btn::text_bg2(format!("Jump to next {} delay", halt_limit))
.build(ctx, "jump to delay", hotkey(Key::D))
.named("jump to delay")
.centered_horiz()
.margin_above(16)
}
fn build_jump_to_delay_picker(halt_limit: Duration, ctx: &EventCtx) -> Widget {
PersistentSplit::new(
ctx,
"choose delay",
halt_limit,
None,
vec![
Choice::new("1 minute delay", Duration::minutes(1)),
Choice::new("2 minute delay", Duration::minutes(2)),
Choice::new("5 minute delay", Duration::minutes(5)),
Choice::new("10 minute delay", Duration::minutes(10)),
],
)
.outline(2.0, Color::WHITE)
.centered_horiz()
}