animgraph 0.1.0

Animation data flow library using hierarchical state machines
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

use serde_derive::{Deserialize, Serialize};

use crate::{
    io::Resource, BlendSampleId, BlendTree, BoneGroup, Graph, GraphNodeConstructor, PoseNode,
    PoseParent,
};

use super::{GraphNode, GraphVisitor};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PoseMaskNode {
    pub bone_group: Resource<BoneGroup>,
}

impl GraphNodeConstructor for PoseMaskNode {
    fn identity() -> &'static str {
        "pose_mask"
    }

    fn construct_entry(
        self,
        metrics: &crate::GraphMetrics,
    ) -> anyhow::Result<crate::GraphNodeEntry> {
        metrics.validate_resource(&self.bone_group, Self::identity())?;
        Ok(crate::GraphNodeEntry::PoseParent(Box::new(self)))
    }
}

impl GraphNode for PoseMaskNode {
    fn visit(&self, _visitor: &mut GraphVisitor) {}
}

impl PoseParent for PoseMaskNode {
    fn apply_parent(
        &self,
        tasks: &mut BlendTree,
        child_task: BlendSampleId,
        graph: &Graph,
    ) -> anyhow::Result<BlendSampleId> {
        if let Some(res) = self.bone_group.get(graph) {
            Ok(tasks.apply_mask(child_task, res.group))
        } else {
            Ok(child_task)
        }
    }

    fn as_pose(&self) -> Option<&dyn PoseNode> {
        None
    }
}

#[cfg(feature = "compiler")]
pub mod compile {
    use serde_json::Value;

    use crate::{
        compiler::{
            context::NodeCompiler,
            prelude::{
                Extras, IOType, Node, NodeCompilationError, NodeSerializationContext, NodeSettings,
            },
        },
        io::Resource,
        model::{IOSlot, DEFAULT_INPUT_NAME},
        BoneGroup, GraphNodeConstructor,
    };

    pub use super::PoseMaskNode;

    #[derive(Default, Debug, Clone)]
    pub struct PoseMaskNodeSettings;

    impl NodeSettings for PoseMaskNodeSettings {
        fn name() -> &'static str {
            PoseMaskNode::identity()
        }

        fn input() -> &'static [(&'static str, IOType)] {
            &[(DEFAULT_INPUT_NAME, BoneGroup::IO_TYPE)]
        }

        fn output() -> &'static [(&'static str, IOType)] {
            &[]
        }

        fn build(self) -> anyhow::Result<Extras> {
            Ok(Extras::default())
        }
    }

    impl NodeCompiler for PoseMaskNode {
        type Settings = PoseMaskNodeSettings;

        fn build<'a>(
            context: &NodeSerializationContext<'a>,
        ) -> Result<Value, NodeCompilationError> {
            let bone_group = context.input_resource(0)?;
            context.serialize_node(PoseMaskNode { bone_group })
        }
    }

    pub fn pose_mask(bone_group: impl IOSlot<Resource<BoneGroup>>) -> Node {
        let input = vec![bone_group.into_slot(DEFAULT_INPUT_NAME)];
        Node::new(input, vec![], PoseMaskNodeSettings).expect("Valid")
    }
}