visibility_system.rs 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. extern crate specs;
  2. use specs::prelude::*;
  3. use super::{Viewshed, Position, Map, Player, Hidden, gamelog::GameLog, Name};
  4. extern crate rltk;
  5. use rltk::{field_of_view, Point};
  6. pub struct VisibilitySystem {}
  7. impl<'a> System<'a> for VisibilitySystem {
  8. #[allow(clippy::type_complexity)]
  9. type SystemData = ( WriteExpect<'a, Map>,
  10. Entities<'a>,
  11. WriteStorage<'a, Viewshed>,
  12. ReadStorage<'a, Position>,
  13. ReadStorage<'a, Player>,
  14. WriteStorage<'a, Hidden>,
  15. WriteExpect<'a, rltk::RandomNumberGenerator>,
  16. WriteExpect<'a, GameLog>,
  17. ReadStorage<'a, Name>,);
  18. fn run(&mut self, data : Self::SystemData) {
  19. let (mut map, entities, mut viewshed, pos, player,
  20. mut hidden, mut rng, mut log, names) = data;
  21. for (ent,viewshed,pos) in (&entities, &mut viewshed, &pos).join() {
  22. if viewshed.dirty {
  23. viewshed.dirty = false;
  24. viewshed.visible_tiles = field_of_view(Point::new(pos.x, pos.y), viewshed.range, &*map);
  25. // If this is the player, reveal what they can see
  26. let _p : Option<&Player> = player.get(ent);
  27. if let Some(_p) = _p {
  28. for t in map.visible_tiles.iter_mut() { *t = false };
  29. for vis in viewshed.visible_tiles.iter() {
  30. let idx = map.xy_idx(vis.x, vis.y);
  31. map.revealed_tiles[idx] = true;
  32. map.visible_tiles[idx] = true;
  33. // Chance to reveal hidden things
  34. for e in map.tile_content[idx].iter() {
  35. let maybe_hidden = hidden.get(*e);
  36. if let Some(_maybe_hidden) = maybe_hidden {
  37. if rng.roll_dice(1,24)==1 {
  38. let name = names.get(*e);
  39. if let Some(name) = name {
  40. log.entries.insert(0, format!("You spotted a {}.", &name.name));
  41. }
  42. hidden.remove(*e);
  43. }
  44. }
  45. }
  46. }
  47. }
  48. }
  49. }
  50. }
  51. }