damage_system.rs 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. extern crate specs;
  2. use specs::prelude::*;
  3. use super::{CombatStats, SufferDamage, Player};
  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 entities = ecs.entities();
  24. for (entity, stats) in (&entities, &combat_stats).join() {
  25. if stats.hp < 1 {
  26. let player = players.get(entity);
  27. match player {
  28. None => dead.push(entity),
  29. Some(_) => console::log("You are dead")
  30. }
  31. }
  32. }
  33. }
  34. for victim in dead {
  35. ecs.delete_entity(victim).expect("Unable to delete");
  36. }
  37. }