abstreet/editor/src/plugins/view/debug_objects.rs

use crate::objects::{Ctx, DEBUG, ID};
use crate::plugins::{Plugin, PluginCtx};
use ezgui::{Color, GfxCtx, Text, TEXT_FG_COLOR};
use piston::input::Key;

pub struct DebugObjectsState {
    tooltip_key_held: bool,
    selected: Option<ID>,

    tooltip_key: Key,
    debug_key: Key,
}

impl DebugObjectsState {
    pub fn new(tooltip_key: Key, debug_key: Key) -> DebugObjectsState {
        DebugObjectsState {
            tooltip_key_held: false,
            selected: None,
            tooltip_key,
            debug_key,
        }
    }
}

impl Plugin for DebugObjectsState {
    fn ambient_event(&mut self, ctx: &mut PluginCtx) {
        self.selected = ctx.primary.current_selection;
        if self.tooltip_key_held {
            self.tooltip_key_held = !ctx.input.key_released(self.tooltip_key);
        } else {
            // TODO Can't really display an OSD action if we're not currently selecting something.
            // Could only activate sometimes, but that seems a bit harder to use.
            self.tooltip_key_held = ctx.input.unimportant_key_pressed(
                self.tooltip_key,
                DEBUG,
                "hold Ctrl to show tooltips",
            );
        }

        if let Some(id) = self.selected {
            if ctx.input.key_pressed(self.debug_key, "debug") {
                id.debug(
                    &ctx.primary.map,
                    &mut ctx.primary.sim,
                    &ctx.primary.draw_map,
                );
            }
        }
    }

    fn draw(&self, g: &mut GfxCtx, ctx: &Ctx) {
        if self.tooltip_key_held {
            if let Some(id) = self.selected {
                ctx.canvas.draw_mouse_tooltip(g, tooltip_lines(id, ctx));
            }
        }
    }
}

fn tooltip_lines(obj: ID, ctx: &Ctx) -> Text {
    let (map, sim, draw_map) = (&ctx.map, &ctx.sim, &ctx.draw_map);
    let mut txt = Text::new();
    match obj {
        ID::Lane(id) => {
            let l = map.get_l(id);
            let r = map.get_r(l.parent);
            let i1 = map.get_source_intersection(id);
            let i2 = map.get_destination_intersection(id);

            txt.add_line(format!(
                "{} is {}",
                l.id,
                r.osm_tags.get("name").unwrap_or(&"???".to_string())
            ));
            txt.add_line(format!("From OSM way {}", r.osm_way_id));
            txt.add_line(format!("Parent {} points to {}", r.id, r.dst_i));
            txt.add_line(format!(
                "Lane goes from {} to {}",
                i1.elevation, i2.elevation
            ));
            txt.add_line(format!(
                "Lane is {} long, parent {} is {} long",
                l.length(),
                r.id,
                r.center_pts.length()
            ));
            for (k, v) in &r.osm_tags {
                txt.add_line(format!("{} = {}", k, v));
            }
            if l.is_parking() {
                txt.add_line(format!("Has {} parking spots", l.number_parking_spots()));
            }
        }
        ID::Intersection(id) => {
            txt.add_line(id.to_string());
            txt.add_line(format!("Roads: {:?}", map.get_i(id).roads));
        }
        ID::Turn(id) => {
            let t = map.get_t(id);
            txt.add_line(format!("{}", id));
            txt.add_line(format!("{:?}", t.turn_type));
        }
        ID::Building(id) => {
            let b = map.get_b(id);
            txt.add_line(format!(
                "Building #{:?} (from OSM way {})",
                id, b.osm_way_id
            ));
            for (k, v) in &b.osm_tags {
                txt.add_styled_line(k.to_string(), Color::RED, None);
                txt.append(" = ".to_string(), TEXT_FG_COLOR, None);
                txt.append(v.to_string(), Color::BLUE, None);
            }
        }
        ID::Car(id) => {
            for line in sim.car_tooltip(id) {
                txt.add_wrapped_line(&ctx.canvas, line);
            }
        }
        ID::Pedestrian(id) => {
            for line in sim.ped_tooltip(id) {
                txt.add_wrapped_line(&ctx.canvas, line);
            }
        }
        ID::ExtraShape(id) => {
            for (k, v) in &draw_map.get_es(id).attributes {
                txt.add_styled_line(k.to_string(), Color::RED, None);
                txt.append(" = ".to_string(), TEXT_FG_COLOR, None);
                txt.append(v.to_string(), Color::BLUE, None);
            }
        }
        ID::Parcel(id) => {
            txt.add_line(id.to_string());
        }
        ID::BusStop(id) => {
            txt.add_line(id.to_string());
            for r in map.get_all_bus_routes() {
                if r.stops.contains(&id) {
                    txt.add_line(format!("- Route {}", r.name));
                }
            }
        }
        ID::Area(id) => {
            let a = map.get_a(id);
            txt.add_line(format!("{} (from OSM way {})", id, a.osm_way_id));
            for (k, v) in &a.osm_tags {
                txt.add_line(format!("{} = {}", k, v));
            }
        }
        ID::Trip(_) => {}
    };
    txt
}