roguelike-game/src/main.rs

218 lines
6.4 KiB
Rust
Raw Normal View History

use rltk::{GameState, Point, Rltk};
2021-10-21 12:54:39 -04:00
use specs::prelude::*;
2021-10-20 12:01:15 -04:00
2021-10-22 10:05:06 -04:00
mod components;
mod damage_system;
mod gamelog;
mod gui;
mod inventory_system;
2021-10-22 10:05:06 -04:00
mod map;
mod map_indexing_system;
mod melee_combat_system;
mod monster_ai_system;
2021-10-22 10:05:06 -04:00
mod player;
mod rect;
mod spawner;
2021-10-25 15:26:39 -04:00
mod visibility_system;
pub use components::*;
2021-10-29 15:15:22 -04:00
use damage_system::DamageSystem;
pub use gamelog::GameLog;
2021-11-04 09:40:58 -04:00
use inventory_system::{ItemCollectionSystem, ItemDropSystem, PotionUseSystem};
pub use map::*;
use map_indexing_system::MapIndexingSystem;
use melee_combat_system::MeleeCombatSystem;
use monster_ai_system::MonsterAI;
use player::*;
pub use rect::Rect;
use visibility_system::VisibilitySystem;
2021-10-25 15:26:39 -04:00
2021-10-29 11:11:17 -04:00
pub const MAP_SIZE: usize = 80 * 50;
2021-10-21 12:54:39 -04:00
2021-10-26 15:43:59 -04:00
#[derive(PartialEq, Copy, Clone)]
pub enum RunState {
2021-10-29 15:15:22 -04:00
AwaitingInput,
PreRun,
PlayerTurn,
MonsterTurn,
2021-11-03 15:11:19 -04:00
ShowInventory,
2021-11-04 09:40:58 -04:00
ShowDropItem,
2021-10-26 15:43:59 -04:00
}
2021-10-22 10:05:06 -04:00
pub struct State {
2021-10-26 15:43:59 -04:00
pub ecs: World,
2021-10-21 12:54:39 -04:00
}
2021-10-20 12:01:15 -04:00
2021-10-25 14:23:19 -04:00
impl State {
fn run_systems(&mut self) {
2021-10-25 15:26:39 -04:00
let mut vis = VisibilitySystem {};
vis.run_now(&self.ecs);
2021-10-26 15:43:59 -04:00
let mut mob = MonsterAI {};
mob.run_now(&self.ecs);
2021-10-29 11:11:17 -04:00
let mut mapindex = MapIndexingSystem {};
mapindex.run_now(&self.ecs);
2021-10-29 15:15:22 -04:00
let mut melee = MeleeCombatSystem {};
melee.run_now(&self.ecs);
let mut damage = DamageSystem {};
damage.run_now(&self.ecs);
2021-11-03 15:11:19 -04:00
let mut pickup = ItemCollectionSystem {};
pickup.run_now(&self.ecs);
2021-11-03 15:59:23 -04:00
let mut potions = PotionUseSystem {};
potions.run_now(&self.ecs);
2021-11-04 09:40:58 -04:00
let mut drop_items = ItemDropSystem {};
drop_items.run_now(&self.ecs);
2021-10-25 14:23:19 -04:00
self.ecs.maintain();
}
}
2021-10-20 12:01:15 -04:00
impl GameState for State {
fn tick(&mut self, ctx: &mut Rltk) {
ctx.cls();
2021-10-29 15:15:22 -04:00
let mut newrunstate;
{
let runstate = self.ecs.fetch::<RunState>();
newrunstate = *runstate;
}
match newrunstate {
RunState::PreRun => {
self.run_systems();
2021-11-03 15:59:23 -04:00
self.ecs.maintain();
2021-10-29 15:15:22 -04:00
newrunstate = RunState::AwaitingInput;
}
RunState::AwaitingInput => {
newrunstate = player_input(self, ctx);
}
RunState::PlayerTurn => {
self.run_systems();
2021-11-03 15:59:23 -04:00
self.ecs.maintain();
2021-10-29 15:15:22 -04:00
newrunstate = RunState::MonsterTurn;
}
RunState::MonsterTurn => {
self.run_systems();
2021-11-03 15:59:23 -04:00
self.ecs.maintain();
2021-10-29 15:15:22 -04:00
newrunstate = RunState::AwaitingInput;
}
2021-11-03 15:11:19 -04:00
RunState::ShowInventory => {
2021-11-03 15:59:23 -04:00
let result = gui::show_inventory(self, ctx);
match result.0 {
gui::ItemMenuResult::Cancel => newrunstate = RunState::AwaitingInput,
gui::ItemMenuResult::NoResponse => {}
gui::ItemMenuResult::Selected => {
let item_entity = result.1.unwrap();
let mut intent = self.ecs.write_storage::<WantsToDrinkPotion>();
intent
.insert(
*self.ecs.fetch::<Entity>(),
WantsToDrinkPotion {
potion: item_entity,
},
)
.expect("failed to add intent to drink potion");
newrunstate = RunState::PlayerTurn;
}
2021-11-03 15:11:19 -04:00
}
}
2021-11-04 09:40:58 -04:00
RunState::ShowDropItem => {
let result = gui::drop_item_menu(self, ctx);
match result.0 {
gui::ItemMenuResult::Cancel => newrunstate = RunState::AwaitingInput,
gui::ItemMenuResult::NoResponse => {}
gui::ItemMenuResult::Selected => {
let item_entity = result.1.unwrap();
let mut intent = self.ecs.write_storage::<WantsToDropItem>();
intent
.insert(
*self.ecs.fetch::<Entity>(),
WantsToDropItem { item: item_entity },
)
.expect("failed to add intent to drop item");
newrunstate = RunState::PlayerTurn;
}
}
}
2021-10-29 15:15:22 -04:00
}
2021-10-21 14:24:40 -04:00
2021-10-29 15:15:22 -04:00
{
let mut runwriter = self.ecs.write_resource::<RunState>();
*runwriter = newrunstate;
2021-10-26 15:43:59 -04:00
}
2021-10-21 14:24:40 -04:00
2021-10-29 15:15:22 -04:00
damage_system::delete_the_dead(&mut self.ecs);
2021-10-25 14:23:19 -04:00
draw_map(&self.ecs, ctx);
2021-10-21 14:52:03 -04:00
2021-10-21 14:24:40 -04:00
let positions = self.ecs.read_storage::<Position>();
let renderables = self.ecs.read_storage::<Renderable>();
2021-10-26 14:38:30 -04:00
let map = self.ecs.fetch::<Map>();
2021-10-21 14:24:40 -04:00
for (pos, render) in (&positions, &renderables).join() {
2021-10-26 14:38:30 -04:00
let idx = map.xy_idx(pos.x, pos.y);
2021-10-29 15:15:22 -04:00
2021-10-26 14:38:30 -04:00
if map.visible_tiles[idx] {
2021-10-29 15:15:22 -04:00
ctx.set(pos.x, pos.y, render.fg, render.bg, render.glyph)
2021-10-26 14:38:30 -04:00
}
2021-10-21 14:24:40 -04:00
}
gui::draw_ui(&self.ecs, ctx);
2021-10-21 14:24:40 -04:00
}
}
2021-10-20 12:01:15 -04:00
fn main() -> rltk::BError {
use rltk::RltkBuilder;
let context = RltkBuilder::simple80x50()
.with_title("Roguelike Tutorial")
.build()?;
2021-10-29 15:15:22 -04:00
let mut gs = State { ecs: World::new() };
2021-10-21 12:54:39 -04:00
gs.ecs.register::<Position>();
gs.ecs.register::<Renderable>();
2021-10-21 14:24:40 -04:00
gs.ecs.register::<Player>();
2021-10-25 15:26:39 -04:00
gs.ecs.register::<Viewshed>();
2021-10-26 15:43:59 -04:00
gs.ecs.register::<Monster>();
gs.ecs.register::<Name>();
2021-10-29 11:11:17 -04:00
gs.ecs.register::<BlocksTile>();
2021-10-29 15:15:22 -04:00
gs.ecs.register::<CombatStats>();
gs.ecs.register::<WantsToMelee>();
gs.ecs.register::<SufferDamage>();
2021-11-03 15:11:19 -04:00
gs.ecs.register::<Item>();
gs.ecs.register::<Potion>();
gs.ecs.register::<InBackpack>();
gs.ecs.register::<WantsToPickupItem>();
2021-11-03 15:59:23 -04:00
gs.ecs.register::<WantsToDrinkPotion>();
2021-11-04 09:40:58 -04:00
gs.ecs.register::<WantsToDropItem>();
2021-10-21 12:54:39 -04:00
2021-10-25 14:23:19 -04:00
let map: Map = Map::new_map_rooms_and_corridors();
let (player_x, player_y) = map.rooms[0].center();
2021-10-22 14:50:04 -04:00
let player_entity = spawner::player(&mut gs.ecs, player_x, player_y);
2021-10-26 15:43:59 -04:00
gs.ecs.insert(rltk::RandomNumberGenerator::new());
for room in map.rooms.iter().skip(1) {
spawner::spawn_room(&mut gs.ecs, room);
2021-10-26 14:38:30 -04:00
}
2021-10-22 14:50:04 -04:00
2021-10-26 14:38:30 -04:00
gs.ecs.insert(map);
2021-10-26 15:43:59 -04:00
gs.ecs.insert(Point::new(player_x, player_y));
2021-10-29 15:15:22 -04:00
gs.ecs.insert(player_entity);
gs.ecs.insert(RunState::PreRun);
gs.ecs.insert(gamelog::GameLog {
entries: vec!["Welcome to Rusty Roguelike".to_string()],
});
2021-10-26 14:38:30 -04:00
2021-10-20 12:01:15 -04:00
rltk::main_loop(context, gs)
2021-10-21 12:54:39 -04:00
}