bevior_tree 0.11.0

Behavior tree plugin for Bevy.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use super::ConverterStrategy;
use super::ResultConverter;
use crate as bevior_tree;
use crate::node::prelude::*;

pub mod prelude {
    pub use super::{ForceResult, Invert};
}

#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Debug)]
struct InvertStrategy;
#[cfg_attr(feature = "serde", typetag::serde)]
impl ConverterStrategy for InvertStrategy {
    fn convert(&self, result: NodeResult) -> NodeResult {
        !result
    }
}

/// Invert the result of the child.
#[delegate_node(delegate)]
pub struct Invert {
    delegate: ResultConverter,
}
impl Invert {
    pub fn new(child: impl Node) -> Self {
        Self {
            delegate: ResultConverter::new(child, InvertStrategy),
        }
    }
}

#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[derive(Debug)]
struct ForceResultStrategy {
    result: NodeResult,
}
#[cfg_attr(feature = "serde", typetag::serde)]
impl ConverterStrategy for ForceResultStrategy {
    fn convert(&self, _result: NodeResult) -> NodeResult {
        self.result
    }
}

/// Returns the specified result whatever the child returns.
#[delegate_node(delegate)]
pub struct ForceResult {
    delegate: ResultConverter,
}
impl ForceResult {
    pub fn new(child: impl Node, result: NodeResult) -> Self {
        Self {
            delegate: ResultConverter::new(child, ForceResultStrategy { result }),
        }
    }
}

#[cfg(test)]
mod tests {
    use crate::tester_util::prelude::*;

    #[test]
    fn test_invert() {
        let mut app = App::new();
        app.add_plugins((TesterPlugin, BehaviorTreePlugin::default()));
        let task = TesterTask0::new(1, NodeResult::Success);
        let converter = Invert::new(task);
        let tree = BehaviorTree::from_node(
            converter,
            &mut app.world_mut().resource_mut::<Assets<BehaviorTreeRoot>>(),
        );
        let entity = app.world_mut().spawn(tree).id();
        app.update();
        app.update();
        let status = app.world().get::<TreeStatus>(entity);
        assert!(
            if let Some(TreeStatus(NodeStatus::Complete(result))) = status {
                result == &NodeResult::Failure
            } else {
                false
            },
            "Invert should match the result."
        );
    }

    #[test]
    fn test_force_result() {
        let mut app = App::new();
        app.add_plugins((TesterPlugin, BehaviorTreePlugin::default()));
        let task = TesterTask0::new(1, NodeResult::Success);
        let converter = ForceResult::new(task, NodeResult::Failure);
        let tree = BehaviorTree::from_node(
            converter,
            &mut app.world_mut().resource_mut::<Assets<BehaviorTreeRoot>>(),
        );
        let entity = app.world_mut().spawn(tree).id();
        app.update();
        app.update();
        let status = app.world().get::<TreeStatus>(entity);
        assert!(
            if let Some(TreeStatus(NodeStatus::Complete(result))) = status {
                result == &NodeResult::Failure
            } else {
                false
            },
            "ForceResult should match the result."
        );
    }
}