space-crush/space-crush-app/src/debug/fleets.rs

use space_crush_common::{
    components::{Fleet, Orbit, Position},
    continue_if_none,
    misc::LogResult,
};
use specs::{prelude::*, ReadExpect, ReadStorage, System, World};

use crate::{
    misc::{HorizontalAlign, Text, TextManager, VerticalAlign},
    resources::{Camera, GameState},
    Error,
};

pub struct Fleets {
    text: Text,
}

#[derive(SystemData)]
pub struct FleetData<'a> {
    game_state: ReadExpect<'a, GameState>,
    camera: ReadExpect<'a, Camera>,
    positions: ReadStorage<'a, Position>,
    orbits: ReadStorage<'a, Orbit>,
    fleets: ReadStorage<'a, Fleet>,
}

impl Fleets {
    pub fn new(manager: &TextManager) -> Result<Self, Error> {
        let cache = manager.create_cache()?;
        let text = cache
            .new_text()
            .scale(20.0)
            .font("resources/fonts/DroidSansMono.ttf")
            .color(0.7, 0.7, 0.7, 1.0)
            .vert_align(VerticalAlign::Center)
            .horz_align(HorizontalAlign::Center)
            .text("")
            .build()?;

        Ok(Self { text })
    }
}

impl<'a> System<'a> for Fleets {
    type SystemData = FleetData<'a>;

    fn run(&mut self, data: Self::SystemData) {
        let FleetData {
            game_state,
            camera,
            positions,
            orbits,
            fleets,
        } = data;

        let player_index = game_state.player_index();

        gl::enable(gl::BLEND);

        for (position, orbit) in (&positions, &orbits).join() {
            let fleet_id = continue_if_none!(orbit.fleet(player_index));
            let fleet = continue_if_none!(fleets.get(fleet_id));

            gl::blend_func(gl::SRC_ALPHA, gl::ONE_MINUS_SRC_ALPHA);
            gl::blend_equation(gl::FUNC_SUBTRACT);
            self.text
                .update(
                    0,
                    format!(
                        "F:{}\nB:{}\nT:{}",
                        fleet.count().fighter,
                        fleet.count().bomber,
                        fleet.count().transporter
                    ),
                )
                .panic("Unable to update text")
                .render_offset(&camera.world_to_window(*position.pos()));
        }

        gl::blend_equation(gl::FUNC_ADD);
        gl::disable(gl::BLEND);
    }
}