340 lines
14 KiB
Rust
340 lines
14 KiB
Rust
use super::{
|
|
gamelog, rex_assets::RexAssets, CombatStats, InBackpack, Map, Name, Player, Point, Position, RunState, State,
|
|
Viewshed,
|
|
};
|
|
use rltk::{Rltk, TextBlock, VirtualKeyCode, RGB};
|
|
use specs::prelude::*;
|
|
|
|
pub fn draw_ui(ecs: &World, ctx: &mut Rltk) {
|
|
ctx.draw_box(0, 43, 79, 6, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK));
|
|
|
|
// Render stats
|
|
let combat_stats = ecs.read_storage::<CombatStats>();
|
|
let players = ecs.read_storage::<Player>();
|
|
for (_player, stats) in (&players, &combat_stats).join() {
|
|
let health = format!(" HP {}/{} ", stats.hp, stats.max_hp);
|
|
ctx.print_color_right(36, 43, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), &health);
|
|
ctx.draw_bar_horizontal(38, 43, 34, stats.hp, stats.max_hp, RGB::named(rltk::RED), RGB::named(rltk::BLACK));
|
|
}
|
|
|
|
// Render message log
|
|
let mut block = TextBlock::new(1, 44, 78, 5);
|
|
let _ = block.print(&gamelog::log_display());
|
|
block.render(&mut rltk::BACKEND_INTERNAL.lock().consoles[0].console);
|
|
|
|
// Render depth
|
|
let map = ecs.fetch::<Map>();
|
|
let depth = format!(" D{} ", map.depth);
|
|
ctx.print_color(74, 43, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), &depth);
|
|
|
|
// Render mouse cursor
|
|
let mouse_pos = ctx.mouse_pos();
|
|
ctx.set_bg(mouse_pos.0, mouse_pos.1, RGB::named(rltk::MAGENTA));
|
|
|
|
draw_tooltips(ecs, ctx);
|
|
}
|
|
|
|
fn draw_tooltips(ecs: &World, ctx: &mut Rltk) {
|
|
let map = ecs.fetch::<Map>();
|
|
let names = ecs.read_storage::<Name>();
|
|
let positions = ecs.read_storage::<Position>();
|
|
|
|
let mouse_pos = ctx.mouse_pos();
|
|
if mouse_pos.0 >= map.width || mouse_pos.1 >= map.height {
|
|
return;
|
|
}
|
|
let mut tooltip: Vec<String> = Vec::new();
|
|
for (name, position) in (&names, &positions).join() {
|
|
let idx = map.xy_idx(position.x, position.y);
|
|
if position.x == mouse_pos.0 && position.y == mouse_pos.1 && map.visible_tiles[idx] {
|
|
tooltip.push(name.name.to_string());
|
|
}
|
|
}
|
|
|
|
if !tooltip.is_empty() {
|
|
if mouse_pos.0 > 40 {
|
|
let arrow_pos = Point::new(mouse_pos.0 - 2, mouse_pos.1);
|
|
let left_x = mouse_pos.0 - 3;
|
|
let mut y = mouse_pos.1;
|
|
for s in tooltip.iter() {
|
|
for i in 0..2 {
|
|
ctx.print_color(
|
|
arrow_pos.x - i,
|
|
y,
|
|
RGB::named(rltk::WHITE),
|
|
RGB::named(rltk::GREY),
|
|
&" ".to_string(),
|
|
);
|
|
}
|
|
ctx.print_color_right(left_x, y, RGB::named(rltk::WHITE), RGB::named(rltk::GREY), s);
|
|
y += 1;
|
|
}
|
|
ctx.print_color(
|
|
arrow_pos.x,
|
|
arrow_pos.y,
|
|
RGB::named(rltk::WHITE),
|
|
RGB::named(rltk::GREY),
|
|
&"->".to_string(),
|
|
);
|
|
} else {
|
|
let arrow_pos = Point::new(mouse_pos.0 + 1, mouse_pos.1);
|
|
let left_x = mouse_pos.0 + 3;
|
|
let mut y = mouse_pos.1;
|
|
for s in tooltip.iter() {
|
|
for i in 0..2 {
|
|
ctx.print_color(
|
|
arrow_pos.x + 1 + i,
|
|
y,
|
|
RGB::named(rltk::WHITE),
|
|
RGB::named(rltk::GREY),
|
|
&" ".to_string(),
|
|
);
|
|
}
|
|
ctx.print_color(left_x + 1, y, RGB::named(rltk::WHITE), RGB::named(rltk::DARKGREY), s);
|
|
y += 1;
|
|
}
|
|
ctx.print_color(
|
|
arrow_pos.x,
|
|
arrow_pos.y,
|
|
RGB::named(rltk::WHITE),
|
|
RGB::named(rltk::GREY),
|
|
&"<-".to_string(),
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(PartialEq, Copy, Clone)]
|
|
pub enum ItemMenuResult {
|
|
Cancel,
|
|
NoResponse,
|
|
Selected,
|
|
}
|
|
|
|
pub fn show_inventory(gs: &mut State, ctx: &mut Rltk) -> (ItemMenuResult, Option<Entity>) {
|
|
let player_entity = gs.ecs.fetch::<Entity>();
|
|
let names = gs.ecs.read_storage::<Name>();
|
|
let backpack = gs.ecs.read_storage::<InBackpack>();
|
|
let entities = gs.ecs.entities();
|
|
|
|
let inventory = (&backpack, &names).join().filter(|item| item.0.owner == *player_entity);
|
|
let count = inventory.count();
|
|
|
|
let mut y = (25 - (count / 2)) as i32;
|
|
ctx.draw_box(15, y - 2, 37, (count + 3) as i32, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK));
|
|
ctx.print_color(18, y - 2, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), "Inventory");
|
|
ctx.print_color(18, y + count as i32 + 1, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), "ESC to cancel");
|
|
|
|
let mut equippable: Vec<Entity> = Vec::new();
|
|
let mut j = 0;
|
|
for (entity, _pack, name) in (&entities, &backpack, &names).join().filter(|item| item.1.owner == *player_entity) {
|
|
ctx.set(17, y, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK), rltk::to_cp437('('));
|
|
ctx.set(18, y, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), 97 + j as rltk::FontCharType);
|
|
ctx.set(19, y, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK), rltk::to_cp437(')'));
|
|
|
|
ctx.print(21, y, &name.name.to_string());
|
|
equippable.push(entity);
|
|
y += 1;
|
|
j += 1;
|
|
}
|
|
|
|
match ctx.key {
|
|
None => (ItemMenuResult::NoResponse, None),
|
|
Some(key) => match key {
|
|
VirtualKeyCode::Escape => (ItemMenuResult::Cancel, None),
|
|
_ => {
|
|
let selection = rltk::letter_to_option(key);
|
|
if selection > -1 && selection < count as i32 {
|
|
return (ItemMenuResult::Selected, Some(equippable[selection as usize]));
|
|
}
|
|
(ItemMenuResult::NoResponse, None)
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
pub fn drop_item_menu(gs: &mut State, ctx: &mut Rltk) -> (ItemMenuResult, Option<Entity>) {
|
|
let player_entity = gs.ecs.fetch::<Entity>();
|
|
let names = gs.ecs.read_storage::<Name>();
|
|
let backpack = gs.ecs.read_storage::<InBackpack>();
|
|
let entities = gs.ecs.entities();
|
|
|
|
let inventory = (&backpack, &names).join().filter(|item| item.0.owner == *player_entity);
|
|
let count = inventory.count();
|
|
|
|
let mut y = (25 - (count / 2)) as i32;
|
|
ctx.draw_box(15, y - 2, 37, (count + 3) as i32, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK));
|
|
ctx.print_color(18, y - 2, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), "Drop what?");
|
|
ctx.print_color(18, y + count as i32 + 1, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), "ESC to cancel");
|
|
|
|
let mut equippable: Vec<Entity> = Vec::new();
|
|
let mut j = 0;
|
|
for (entity, _pack, name) in (&entities, &backpack, &names).join().filter(|item| item.1.owner == *player_entity) {
|
|
ctx.set(17, y, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK), rltk::to_cp437('('));
|
|
ctx.set(18, y, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), 97 + j as rltk::FontCharType);
|
|
ctx.set(19, y, RGB::named(rltk::WHITE), RGB::named(rltk::BLACK), rltk::to_cp437(')'));
|
|
|
|
ctx.print(21, y, &name.name.to_string());
|
|
equippable.push(entity);
|
|
y += 1;
|
|
j += 1;
|
|
}
|
|
|
|
match ctx.key {
|
|
None => (ItemMenuResult::NoResponse, None),
|
|
Some(key) => match key {
|
|
VirtualKeyCode::Escape => (ItemMenuResult::Cancel, None),
|
|
_ => {
|
|
let selection = rltk::letter_to_option(key);
|
|
if selection > -1 && selection < count as i32 {
|
|
return (ItemMenuResult::Selected, Some(equippable[selection as usize]));
|
|
}
|
|
(ItemMenuResult::NoResponse, None)
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
pub fn ranged_target(gs: &mut State, ctx: &mut Rltk, range: i32, aoe: i32) -> (ItemMenuResult, Option<Point>) {
|
|
let player_entity = gs.ecs.fetch::<Entity>();
|
|
let player_pos = gs.ecs.fetch::<Point>();
|
|
let viewsheds = gs.ecs.read_storage::<Viewshed>();
|
|
|
|
ctx.print_color(5, 0, RGB::named(rltk::YELLOW), RGB::named(rltk::BLACK), "select target");
|
|
|
|
// Highlight available cells
|
|
let mut available_cells = Vec::new();
|
|
let visible = viewsheds.get(*player_entity);
|
|
if let Some(visible) = visible {
|
|
// We have a viewshed
|
|
for idx in visible.visible_tiles.iter() {
|
|
let distance = rltk::DistanceAlg::Pythagoras.distance2d(*player_pos, *idx);
|
|
if distance <= range as f32 {
|
|
ctx.set_bg(idx.x, idx.y, RGB::named(rltk::BLUE));
|
|
available_cells.push(idx);
|
|
}
|
|
}
|
|
} else {
|
|
return (ItemMenuResult::Cancel, None);
|
|
}
|
|
|
|
// Draw mouse cursor
|
|
let mouse_pos = ctx.mouse_pos();
|
|
let map = gs.ecs.fetch::<Map>();
|
|
let mut valid_target = false;
|
|
for idx in available_cells.iter() {
|
|
if idx.x == mouse_pos.0 && idx.y == mouse_pos.1 {
|
|
valid_target = true;
|
|
}
|
|
}
|
|
if valid_target {
|
|
if aoe > 0 {
|
|
let mut blast_tiles = rltk::field_of_view(Point::new(mouse_pos.0, mouse_pos.1), aoe, &*map);
|
|
blast_tiles.retain(|p| p.x > 0 && p.x < map.width - 1 && p.y > 0 && p.y < map.height - 1);
|
|
for tile in blast_tiles.iter() {
|
|
ctx.set_bg(tile.x, tile.y, RGB::named(rltk::DARKCYAN));
|
|
}
|
|
}
|
|
ctx.set_bg(mouse_pos.0, mouse_pos.1, RGB::named(rltk::CYAN));
|
|
if ctx.left_click {
|
|
return (ItemMenuResult::Selected, Some(Point::new(mouse_pos.0, mouse_pos.1)));
|
|
}
|
|
} else {
|
|
ctx.set_bg(mouse_pos.0, mouse_pos.1, RGB::named(rltk::RED));
|
|
if ctx.left_click {
|
|
return (ItemMenuResult::Cancel, None);
|
|
}
|
|
}
|
|
|
|
(ItemMenuResult::NoResponse, None)
|
|
}
|
|
|
|
#[derive(PartialEq, Copy, Clone)]
|
|
pub enum MainMenuSelection {
|
|
NewGame,
|
|
LoadGame,
|
|
Quit,
|
|
}
|
|
|
|
#[derive(PartialEq, Copy, Clone)]
|
|
pub enum MainMenuResult {
|
|
NoSelection { selected: MainMenuSelection },
|
|
Selected { selected: MainMenuSelection },
|
|
}
|
|
|
|
pub fn main_menu(gs: &mut State, ctx: &mut Rltk) -> MainMenuResult {
|
|
let save_exists = super::saveload_system::does_save_exist();
|
|
let runstate = gs.ecs.fetch::<RunState>();
|
|
let assets = gs.ecs.fetch::<RexAssets>();
|
|
|
|
ctx.render_xp_sprite(&assets.menu, 0, 0);
|
|
|
|
ctx.print_color(40, 21, RGB::named(rltk::GREEN), RGB::from_f32(0.11, 0.11, 0.11), "RUST-RL");
|
|
|
|
if let RunState::MainMenu { menu_selection: selection } = *runstate {
|
|
let mut y = 24;
|
|
if selection == MainMenuSelection::NewGame {
|
|
ctx.print_color(37, 24, RGB::named(rltk::YELLOW), RGB::from_f32(0.11, 0.11, 0.11), "[");
|
|
ctx.print_color(39, 24, RGB::named(rltk::GREEN), RGB::from_f32(0.11, 0.11, 0.11), "new game");
|
|
ctx.print_color(48, 24, RGB::named(rltk::YELLOW), RGB::from_f32(0.11, 0.11, 0.11), "]");
|
|
} else {
|
|
ctx.print_color(39, 24, RGB::named(rltk::WHITE), RGB::from_f32(0.11, 0.11, 0.11), "new game");
|
|
}
|
|
y += 2;
|
|
if save_exists {
|
|
if selection == MainMenuSelection::LoadGame {
|
|
ctx.print_color(36, y, RGB::named(rltk::YELLOW), RGB::from_f32(0.11, 0.11, 0.11), "[");
|
|
ctx.print_color(38, y, RGB::named(rltk::GREEN), RGB::from_f32(0.11, 0.11, 0.11), "load game");
|
|
ctx.print_color(48, y, RGB::named(rltk::YELLOW), RGB::from_f32(0.11, 0.11, 0.11), "]");
|
|
} else {
|
|
ctx.print_color(38, y, RGB::named(rltk::WHITE), RGB::from_f32(0.11, 0.11, 0.11), "load game");
|
|
}
|
|
y += 2;
|
|
}
|
|
if selection == MainMenuSelection::Quit {
|
|
ctx.print_color(37, y, RGB::named(rltk::YELLOW), RGB::from_f32(0.11, 0.11, 0.11), "[");
|
|
ctx.print_color(39, y, RGB::named(rltk::GREEN), RGB::from_f32(0.11, 0.11, 0.11), "goodbye!");
|
|
ctx.print_color(48, y, RGB::named(rltk::YELLOW), RGB::from_f32(0.11, 0.11, 0.11), "]");
|
|
} else {
|
|
ctx.print_color(43, y, RGB::named(rltk::WHITE), RGB::from_f32(0.11, 0.11, 0.11), "quit");
|
|
}
|
|
|
|
match ctx.key {
|
|
None => return MainMenuResult::NoSelection { selected: selection },
|
|
Some(key) => match key {
|
|
VirtualKeyCode::Escape | VirtualKeyCode::C => {
|
|
return MainMenuResult::NoSelection { selected: MainMenuSelection::Quit }
|
|
}
|
|
VirtualKeyCode::N => return MainMenuResult::NoSelection { selected: MainMenuSelection::NewGame },
|
|
VirtualKeyCode::L => return MainMenuResult::NoSelection { selected: MainMenuSelection::LoadGame },
|
|
VirtualKeyCode::Up => {
|
|
let mut new_selection;
|
|
match selection {
|
|
MainMenuSelection::NewGame => new_selection = MainMenuSelection::Quit,
|
|
MainMenuSelection::LoadGame => new_selection = MainMenuSelection::NewGame,
|
|
MainMenuSelection::Quit => new_selection = MainMenuSelection::LoadGame,
|
|
}
|
|
if new_selection == MainMenuSelection::LoadGame && !save_exists {
|
|
new_selection = MainMenuSelection::NewGame;
|
|
}
|
|
return MainMenuResult::NoSelection { selected: new_selection };
|
|
}
|
|
VirtualKeyCode::Down => {
|
|
let mut new_selection;
|
|
match selection {
|
|
MainMenuSelection::NewGame => new_selection = MainMenuSelection::LoadGame,
|
|
MainMenuSelection::LoadGame => new_selection = MainMenuSelection::Quit,
|
|
MainMenuSelection::Quit => new_selection = MainMenuSelection::NewGame,
|
|
}
|
|
if new_selection == MainMenuSelection::LoadGame && !save_exists {
|
|
new_selection = MainMenuSelection::Quit;
|
|
}
|
|
return MainMenuResult::NoSelection { selected: new_selection };
|
|
}
|
|
VirtualKeyCode::Return => return MainMenuResult::Selected { selected: selection },
|
|
_ => return MainMenuResult::NoSelection { selected: selection },
|
|
},
|
|
}
|
|
}
|
|
MainMenuResult::NoSelection { selected: MainMenuSelection::NewGame }
|
|
}
|