rust/lib-hwengine-future/src/ai/mod.rs
author unC0Rr
Sun, 27 Oct 2024 17:00:17 +0100
branchtransitional_engine
changeset 16039 a236360669cc
child 16040 6c5b3c576fc6
permissions -rw-r--r--
Start on AI implementation in rust
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
16039
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     1
use std::collections::HashMap;
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     2
use integral_geometry::Point;
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     3
use crate::GameField;
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     4
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     5
pub struct Target {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     6
    point: Point,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     7
    health: i32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     8
    radius: u32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
     9
    density: f32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    10
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    11
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    12
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    13
pub struct Hedgehog {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    14
    pub(crate) x: f32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    15
    pub(crate) y: f32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    16
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    17
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    18
pub struct AI<'a> {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    19
    game_field: &'a GameField,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    20
    targets: Vec<Target>,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    21
    team: Vec<Hedgehog>,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    22
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    23
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    24
#[derive(Clone)]
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    25
enum Direction {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    26
    Left,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    27
    Right
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    28
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    29
#[derive(Clone)]
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    30
enum Action {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    31
    Walk(Direction),
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    32
    LongJump,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    33
    HighJump(usize)
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    34
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    35
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    36
#[derive(Clone)]
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    37
struct Waypoint {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    38
    x: f32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    39
    y: f32,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    40
    ticks: usize,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    41
    damage: usize,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    42
    previous_point: Option<(usize, Action)>,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    43
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    44
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    45
#[derive(Default)]
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    46
pub struct Waypoints {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    47
    key_points: Vec<Waypoint>,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    48
    points: HashMap<Point, Waypoint>,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    49
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    50
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    51
impl Waypoints {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    52
    fn add_keypoint(&mut self, waypoint: Waypoint) {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    53
        let [x, y] = [waypoint.x, waypoint.y].map(|i| i as i32);
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    54
        let point = Point::new(x, y);
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    55
        self.key_points.push(waypoint.clone());
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    56
        self.points.insert(point, waypoint);
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    57
    }
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    58
}
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    59
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    60
impl<'a> AI<'a> {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    61
    pub fn new(game_field: &'a GameField) -> AI<'a> {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    62
        Self {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    63
            game_field,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    64
            targets: vec![],
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    65
            team: vec![],
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    66
        }
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    67
    }
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    68
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    69
    pub fn get_team_mut(&mut self) -> &mut Vec<Hedgehog> {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    70
        &mut self.team
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    71
    }
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    72
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    73
    pub fn walk(hedgehog: &Hedgehog) {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    74
        let mut stack = Vec::<usize>::new();
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    75
        let mut waypoints = Waypoints::default();
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    76
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    77
        waypoints.add_keypoint(Waypoint{
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    78
            x: hedgehog.x,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    79
            y: hedgehog.y,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    80
            ticks: 0,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    81
            damage: 0,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    82
            previous_point: None,
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    83
        });
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    84
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    85
        while let Some(wp) = stack.pop() {
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    86
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    87
        }
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    88
    }
a236360669cc Start on AI implementation in rust
unC0Rr
parents:
diff changeset
    89
}