damage_system.rs 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. extern crate specs;
  2. use super::{CombatStats, Player, SufferDamage};
  3. use rltk::console;
  4. use specs::prelude::*;
  5. pub struct DamageSystem {}
  6. impl<'a> System<'a> for DamageSystem {
  7. type SystemData = (
  8. WriteStorage<'a, CombatStats>,
  9. WriteStorage<'a, SufferDamage>,
  10. );
  11. fn run(&mut self, data: Self::SystemData) {
  12. let (mut stats, mut damage) = data;
  13. for (mut stats, damage) in (&mut stats, &damage).join() {
  14. stats.hp -= damage.amount;
  15. }
  16. damage.clear();
  17. }
  18. }
  19. pub fn delete_the_dead(ecs: &mut World) {
  20. let mut dead: Vec<Entity> = Vec::new();
  21. // Using a scope to make the borrow checker happy
  22. {
  23. let combat_stats = ecs.read_storage::<CombatStats>();
  24. let players = ecs.read_storage::<Player>();
  25. let entities = ecs.entities();
  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 => dead.push(entity),
  31. Some(_) => console::log("You are dead"),
  32. }
  33. }
  34. }
  35. }
  36. for victim in dead {
  37. ecs.delete_entity(victim).expect("Unable to delete");
  38. }
  39. }