2022-01-26 15:42:42 +01:00
|
|
|
use std::cmp::{max, min};
|
|
|
|
|
2022-01-27 18:28:40 +01:00
|
|
|
use rltk::{Point, Rltk, VirtualKeyCode};
|
2022-01-26 15:42:42 +01:00
|
|
|
use specs::prelude::*;
|
|
|
|
|
2022-01-27 22:38:44 +01:00
|
|
|
use crate::gamelog::GameLog;
|
2022-01-25 14:41:24 +01:00
|
|
|
use crate::{
|
2022-01-25 22:18:17 +01:00
|
|
|
components::{CombatStats, Player, Position, Viewshed, WantsToMelee},
|
2022-01-27 22:38:44 +01:00
|
|
|
map::Map,
|
|
|
|
Item, RunState, State, WantsToPickupItem,
|
2022-01-25 14:41:24 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
pub fn try_move_player(delta_x: i32, delta_y: i32, ecs: &mut World) {
|
|
|
|
let mut positions = ecs.write_storage::<Position>();
|
|
|
|
let mut players = ecs.write_storage::<Player>();
|
2022-01-25 22:18:17 +01:00
|
|
|
let mut viewsheds = ecs.write_storage::<Viewshed>();
|
|
|
|
let combat_stats = ecs.read_storage::<CombatStats>();
|
|
|
|
let entities = ecs.entities();
|
|
|
|
let mut wants_to_melee = ecs.write_storage::<WantsToMelee>();
|
|
|
|
let map = ecs.fetch::<Map>();
|
2022-01-25 14:41:24 +01:00
|
|
|
|
2022-01-25 22:18:17 +01:00
|
|
|
for (entity, _player, pos, viewshed) in
|
2022-01-27 22:38:44 +01:00
|
|
|
(&entities, &mut players, &mut positions, &mut viewsheds).join()
|
2022-01-25 22:18:17 +01:00
|
|
|
{
|
2022-01-27 22:38:44 +01:00
|
|
|
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;
|
|
|
|
}
|
2022-01-25 22:18:17 +01:00
|
|
|
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");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if !map.blocked[destination_idx] {
|
2022-01-25 14:41:24 +01:00
|
|
|
pos.x = min(79, max(0, pos.x + delta_x));
|
|
|
|
pos.y = min(49, max(0, pos.y + delta_y));
|
2022-01-25 22:18:17 +01:00
|
|
|
|
|
|
|
viewshed.dirty = true;
|
|
|
|
let mut ppos = ecs.write_resource::<Point>();
|
|
|
|
ppos.x = pos.x;
|
|
|
|
ppos.y = pos.y;
|
2022-01-25 14:41:24 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-26 15:42:42 +01:00
|
|
|
fn get_item(ecs: &mut World) {
|
|
|
|
let player_pos = ecs.fetch::<Point>();
|
|
|
|
let player_entity = ecs.fetch::<Entity>();
|
|
|
|
let entities = ecs.entities();
|
|
|
|
let items = ecs.read_storage::<Item>();
|
|
|
|
let positions = ecs.read_storage::<Position>();
|
|
|
|
let mut gamelog = ecs.fetch_mut::<GameLog>();
|
|
|
|
|
|
|
|
let mut target_item: Option<Entity> = 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::<WantsToPickupItem>();
|
|
|
|
pickup
|
|
|
|
.insert(
|
|
|
|
*player_entity,
|
|
|
|
WantsToPickupItem {
|
|
|
|
collected_by: *player_entity,
|
|
|
|
item,
|
|
|
|
},
|
|
|
|
)
|
|
|
|
.expect("Unable to insert want to pickup");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-01-25 22:18:17 +01:00
|
|
|
pub fn player_input(gs: &mut State, ctx: &mut Rltk) -> RunState {
|
2022-01-25 14:41:24 +01:00
|
|
|
match ctx.key {
|
2022-01-25 22:18:17 +01:00
|
|
|
None => return RunState::AwaitingInput,
|
2022-01-25 14:41:24 +01:00
|
|
|
Some(key) => match key {
|
2022-01-25 22:18:17 +01:00
|
|
|
VirtualKeyCode::Left | VirtualKeyCode::H => try_move_player(-1, 0, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::Right | VirtualKeyCode::L => try_move_player(1, 0, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::Up | VirtualKeyCode::K => try_move_player(0, -1, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::Down | VirtualKeyCode::J => try_move_player(0, 1, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::Y => try_move_player(1, -1, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::U => try_move_player(-1, -1, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::N => try_move_player(1, 1, &mut gs.ecs),
|
|
|
|
VirtualKeyCode::B => try_move_player(-1, 1, &mut gs.ecs),
|
2022-01-26 15:42:42 +01:00
|
|
|
VirtualKeyCode::G => get_item(&mut gs.ecs),
|
|
|
|
VirtualKeyCode::I => return RunState::ShowInventory,
|
|
|
|
VirtualKeyCode::D => return RunState::ShowDropItem,
|
2022-01-27 18:28:40 +01:00
|
|
|
VirtualKeyCode::Escape => return RunState::SaveGame,
|
2022-01-25 22:18:17 +01:00
|
|
|
_ => return RunState::AwaitingInput,
|
2022-01-25 14:41:24 +01:00
|
|
|
},
|
|
|
|
}
|
2022-01-25 22:18:17 +01:00
|
|
|
|
|
|
|
RunState::PlayerTurn
|
2022-01-25 14:41:24 +01:00
|
|
|
}
|