cornerstone 0.3.1

A behavior tree library
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

use serde_json::json;

use crate::{node::strip_ref_tag, NodeStatus, TreeNode, TreeNodeWrapper};

use super::DataProxy;

pub trait ActionNodeImpl: Send {
    fn tick_status(&mut self, data_proxy: &mut DataProxy) -> NodeStatus;

    fn node_info(&self) -> String {
        format!("{}", std::any::type_name::<Self>())
    }

    fn halt(&mut self) {}
}

pub struct ActionWrapper {
    pub data_proxy: DataProxy,
    node: Box<dyn ActionNodeImpl>,
}

impl TreeNode for ActionWrapper {
    fn tick(&mut self) -> NodeStatus {
        self.node.tick_status(&mut self.data_proxy)
    }

    fn halt(&mut self) {
        tracing::debug!("halt action: {}", std::any::type_name::<Self>());

        self.node.halt();
    }
}

impl ActionWrapper {
    pub fn new(data_proxy: DataProxy, node: Box<dyn ActionNodeImpl>) -> Self {
        Self { data_proxy, node }
    }
}

#[derive(Default)]
pub struct SetBlackboard;

impl ActionNodeImpl for SetBlackboard {
    fn tick_status(&mut self, data_proxy: &mut DataProxy) -> NodeStatus {
        let Some(output_key) = data_proxy.get_input::<String>("output_key") else {
            return NodeStatus::Failure;
        };

        let Some(value) = data_proxy.get_input::<String>("value") else {
            return NodeStatus::Failure;
        };

        data_proxy.blackboard().set(output_key, json!(value));

        NodeStatus::Success
    }
}