use crate::components::{ CombatStats, Item, Monster, Player, Position, Viewshed, WantsToMelee, WantsToPickupItem, }; use crate::{game_log::GameLog, Map, RunState, State, TileType}; use rltk::{Point, Rltk, VirtualKeyCode}; use specs::prelude::*; use std::cmp::{max, min}; pub fn try_move_player(delta_x: i32, delta_y: i32, ecs: &mut World) { let mut positions = ecs.write_storage::(); let players = ecs.read_storage::(); let mut viewsheds = ecs.write_storage::(); let entities = ecs.entities(); let combat_stats = ecs.read_storage::(); let map = ecs.fetch::(); let mut wants_to_melee = ecs.write_storage::(); for (entity, _player, pos, viewshed) in (&entities, &players, &mut positions, &mut viewsheds).join() { if pos.x + delta_x < 1 || pos.x + delta_x > map.width - 1 || pos.y + delta_y < 1 || pos.y + delta_y > map.height - 1 { return; } let destination_idx = map.xy_idx(pos.x + delta_x, pos.y + delta_y); for potential_target in map.tile_content[destination_idx].iter() { let target = combat_stats.get(*potential_target); if let Some(_target) = target { wants_to_melee .insert( entity, WantsToMelee { target: *potential_target, }, ) .expect("Add target failed"); return; } } if !map.blocked[destination_idx] { pos.x = min(79, max(0, pos.x + delta_x)); pos.y = min(49, max(0, pos.y + delta_y)); viewshed.dirty = true; let mut ppos = ecs.write_resource::(); ppos.x = pos.x; ppos.y = pos.y; } } } pub fn player_input(gs: &mut State, ctx: &mut Rltk) -> RunState { // Player movement match ctx.key { None => return RunState::AwaitingInput, // Nothing happened Some(key) => match key { // Cardinal directions VirtualKeyCode::Left | VirtualKeyCode::Numpad4 | VirtualKeyCode::H | VirtualKeyCode::Key4 => try_move_player(-1, 0, &mut gs.ecs), VirtualKeyCode::Right | VirtualKeyCode::Numpad6 | VirtualKeyCode::L | VirtualKeyCode::Key6 => try_move_player(1, 0, &mut gs.ecs), VirtualKeyCode::Up | VirtualKeyCode::Numpad8 | VirtualKeyCode::K | VirtualKeyCode::Key8 => try_move_player(0, -1, &mut gs.ecs), VirtualKeyCode::Down | VirtualKeyCode::Numpad2 | VirtualKeyCode::J | VirtualKeyCode::S | VirtualKeyCode::Key2 => try_move_player(0, 1, &mut gs.ecs), // Diagonals VirtualKeyCode::Numpad9 | VirtualKeyCode::Y | VirtualKeyCode::Key9 => { try_move_player(1, -1, &mut gs.ecs) } VirtualKeyCode::Numpad7 | VirtualKeyCode::U | VirtualKeyCode::Key7 => { try_move_player(-1, -1, &mut gs.ecs) } VirtualKeyCode::Numpad3 | VirtualKeyCode::N | VirtualKeyCode::Key3 => { try_move_player(1, 1, &mut gs.ecs) } VirtualKeyCode::Numpad1 | VirtualKeyCode::B | VirtualKeyCode::Key1 => { try_move_player(-1, 1, &mut gs.ecs) } // Pick up item VirtualKeyCode::G => get_item(&mut gs.ecs), // Show inventory VirtualKeyCode::I => return RunState::ShowInventory, // Show item drop screen VirtualKeyCode::D => return RunState::ShowDropItem, // Save and Quit VirtualKeyCode::Escape => return RunState::SaveGame, // Level changes VirtualKeyCode::Period => { if try_next_level(&mut gs.ecs) { return RunState::NextLevel; } } // Skip Turn VirtualKeyCode::Numpad5 | VirtualKeyCode::Space => return skip_turn(&mut gs.ecs), // Remove item VirtualKeyCode::R => return RunState::ShowRemoveItem, _ => return RunState::AwaitingInput, }, } RunState::PlayerTurn } fn get_item(ecs: &mut World) { let player_pos = ecs.fetch::(); let player_entity = ecs.fetch::(); let entities = ecs.entities(); let items = ecs.read_storage::(); let positions = ecs.read_storage::(); let mut gamelog = ecs.fetch_mut::(); let mut target_item: Option = None; for (item_entity, _item, position) in (&entities, &items, &positions).join() { if position.x == player_pos.x && position.y == player_pos.y { target_item = Some(item_entity); } } match target_item { None => gamelog .entries .push("There is nothing here to pick up.".to_string()), Some(item) => { let mut pickup = ecs.write_storage::(); pickup .insert( *player_entity, WantsToPickupItem { collected_by: *player_entity, item, }, ) .expect("Unable to pick up item?!"); } } } pub fn try_next_level(ecs: &mut World) -> bool { let player_pos = ecs.fetch::(); let map = ecs.fetch::(); let player_idx = map.xy_idx(player_pos.x, player_pos.y); if map.tiles[player_idx] == TileType::DownStairs { true } else { let mut gamelog = ecs.fetch_mut::(); gamelog .entries .push("There is no way down from here.".to_string()); false } } fn skip_turn(ecs: &mut World) -> RunState { let player_entity = ecs.fetch::(); let viewshed_components = ecs.read_storage::(); let monsters = ecs.read_storage::(); let worldmap_resource = ecs.fetch::(); let mut can_heal = true; let viewshed = viewshed_components.get(*player_entity).unwrap(); for tile in viewshed.visible_tiles.iter() { let idx = worldmap_resource.xy_idx(tile.x, tile.y); for entity_id in worldmap_resource.tile_content[idx].iter() { match monsters.get(*entity_id) { None => {} Some(_) => { can_heal = false; } } } } if can_heal { let mut health_components = ecs.write_storage::(); let player_hp = health_components.get_mut(*player_entity).unwrap(); player_hp.hp = i32::min(player_hp.hp + 1, player_hp.max_hp); } RunState::PlayerTurn }