damage_system.rs 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. extern crate specs;
  2. use specs::prelude::*;
  3. use super::{CombatStats, SufferDamage, Player, Name, gamelog::GameLog};
  4. use rltk::console;
  5. pub struct DamageSystem {}
  6. impl<'a> System<'a> for DamageSystem {
  7. type SystemData = ( WriteStorage<'a, CombatStats>,
  8. WriteStorage<'a, SufferDamage> );
  9. fn run(&mut self, data : Self::SystemData) {
  10. let (mut stats, mut damage) = data;
  11. for (mut stats, damage) in (&mut stats, &damage).join() {
  12. stats.hp -= damage.amount;
  13. }
  14. damage.clear();
  15. }
  16. }
  17. pub fn delete_the_dead(ecs : &mut World) {
  18. let mut dead : Vec<Entity> = Vec::new();
  19. // Using a scope to make the borrow checker happy
  20. {
  21. let combat_stats = ecs.read_storage::<CombatStats>();
  22. let players = ecs.read_storage::<Player>();
  23. let names = ecs.read_storage::<Name>();
  24. let entities = ecs.entities();
  25. let mut log = ecs.write_resource::<GameLog>();
  26. for (entity, stats) in (&entities, &combat_stats).join() {
  27. if stats.hp < 1 {
  28. let player = players.get(entity);
  29. match player {
  30. None => {
  31. let victim_name = names.get(entity);
  32. if let Some(victim_name) = victim_name {
  33. log.entries.insert(0, format!("{} is dead", &victim_name.name));
  34. }
  35. dead.push(entity)
  36. }
  37. Some(_) => console::log("You are dead")
  38. }
  39. }
  40. }
  41. }
  42. for victim in dead {
  43. ecs.delete_entity(victim).expect("Unable to delete");
  44. }
  45. }