damage_system.rs 2.1 KB

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