monster_ai_system.rs 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. extern crate specs;
  2. use specs::prelude::*;
  3. use super::{Viewshed, Monster, Name, Map, Position};
  4. extern crate rltk;
  5. use rltk::{Point};
  6. pub struct MonsterAI {}
  7. impl<'a> System<'a> for MonsterAI {
  8. #[allow(clippy::type_complexity)]
  9. type SystemData = ( WriteExpect<'a, Map>,
  10. ReadExpect<'a, Point>,
  11. WriteStorage<'a, Viewshed>,
  12. ReadStorage<'a, Monster>,
  13. ReadStorage<'a, Name>,
  14. WriteStorage<'a, Position>);
  15. fn run(&mut self, data : Self::SystemData) {
  16. let (mut map, player_pos, mut viewshed, monster, name, mut position) = data;
  17. for (mut viewshed,_monster,name,mut pos) in (&mut viewshed, &monster, &name, &mut position).join() {
  18. let distance = rltk::DistanceAlg::Pythagoras.distance2d(Point::new(pos.x, pos.y), *player_pos);
  19. if distance < 1.5 {
  20. // Attack goes here
  21. println!("{} shouts insults", name.name);
  22. return;
  23. }
  24. if viewshed.visible_tiles.contains(&*player_pos) {
  25. // Path to the player
  26. let path = rltk::a_star_search(
  27. map.xy_idx(pos.x, pos.y) as i32,
  28. map.xy_idx(player_pos.x, player_pos.y) as i32,
  29. &mut *map
  30. );
  31. if path.success && path.steps.len()>1 {
  32. let mut idx = map.xy_idx(pos.x, pos.y);
  33. map.blocked[idx] = false;
  34. pos.x = path.steps[1] % map.width;
  35. pos.y = path.steps[1] / map.width;
  36. idx = map.xy_idx(pos.x, pos.y);
  37. map.blocked[idx] = true;
  38. viewshed.dirty = true;
  39. }
  40. }
  41. }
  42. }
  43. }