fbx-dom 0.1.0

Document Model for Reading in FBX
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

//! FBX `Deformer` / `BlendShape` — Assimp [`BlendShape`](https://github.com/assimp/assimp/blob/master/code/AssetLib/FBX/FBXDocument.h).

use std::collections::HashMap;
use std::convert::TryFrom;

use crate::{OwnedDocument, OwnedObject, Property};

use super::{BlendShapeChannel, FbxObjectTag, FbxTypeMismatch, fbx_object_tag};

#[derive(Debug, PartialEq)]
pub struct BlendShape(pub OwnedObject);

impl BlendShape {
    pub fn inner(&self) -> &OwnedObject {
        &self.0
    }

    pub fn into_inner(self) -> OwnedObject {
        self.0
    }

    pub fn properties(&self) -> &HashMap<String, Property> {
        &self.0.properties
    }

    pub fn property(&self, name: &str) -> Option<&Property> {
        self.0.properties.get(name)
    }

    /// Resolve `BlendShapeChannel -> BlendShape` links via owned OO connections.
    pub fn get_blend_shape_channels<'a>(
        &'a self,
        document: &'a OwnedDocument,
    ) -> Vec<&'a BlendShapeChannel> {
        let blend_shape_id = self.inner().object_index;
        document
            .blend_shape_channels
            .iter()
            .filter(|channel| {
                channel
                    .inner()
                    .connected_object_ids
                    .contains(&blend_shape_id)
            })
            .collect()
    }
}

impl TryFrom<OwnedObject> for BlendShape {
    type Error = FbxTypeMismatch;

    fn try_from(o: OwnedObject) -> Result<Self, Self::Error> {
        match fbx_object_tag(&o) {
            FbxObjectTag::BlendShape => Ok(BlendShape(o)),
            _ => Err(FbxTypeMismatch::wrong_object_kind(
                o,
                "BlendShape".to_string(),
            )),
        }
    }
}

#[cfg(test)]
mod tests {
    use std::collections::HashMap;
    use std::convert::TryFrom;

    use crate::objects::{
        BlendShapeChannel, DEFORMER_BLEND_SHAPE_CHANNEL_CLASS_NAME,
        DEFORMER_BLEND_SHAPE_CLASS_NAME, DEFORMER_TYPE_NAME,
    };
    use crate::{OwnedDocument, OwnedObject, Property};

    use super::BlendShape;

    #[test]
    fn property_accessors() {
        let mut properties = HashMap::new();
        properties.insert("Foo".into(), Property::Int(7));
        let o = OwnedObject {
            object_index: 15,
            name: "BlendShape::A".into(),
            type_name: DEFORMER_TYPE_NAME.into(),
            class_name: DEFORMER_BLEND_SHAPE_CLASS_NAME.into(),
            properties,
            attributes: HashMap::new(),
            connected_object_ids: vec![],
            object_property_targets: vec![],
            pp_property_targets: HashMap::new(),
        };
        let b = BlendShape::try_from(o).unwrap();
        assert_eq!(b.property("Foo"), Some(&Property::Int(7)));
    }

    #[test]
    fn resolves_blend_shape_channel_connections() {
        let blend_shape = BlendShape::try_from(OwnedObject {
            object_index: 50,
            name: "BlendShape::B".into(),
            type_name: DEFORMER_TYPE_NAME.into(),
            class_name: DEFORMER_BLEND_SHAPE_CLASS_NAME.into(),
            properties: HashMap::new(),
            attributes: HashMap::new(),
            connected_object_ids: vec![],
            object_property_targets: vec![],
            pp_property_targets: HashMap::new(),
        })
        .unwrap();
        let channel = BlendShapeChannel::try_from(OwnedObject {
            object_index: 51,
            name: "BlendShapeChannel::B".into(),
            type_name: DEFORMER_TYPE_NAME.into(),
            class_name: DEFORMER_BLEND_SHAPE_CHANNEL_CLASS_NAME.into(),
            properties: HashMap::new(),
            attributes: HashMap::new(),
            connected_object_ids: vec![50],
            object_property_targets: vec![],
            pp_property_targets: HashMap::new(),
        })
        .unwrap();

        let mut owned = OwnedDocument::default();
        owned.blend_shape_channels = vec![channel];
        let linked = blend_shape.get_blend_shape_channels(&owned);
        assert_eq!(linked.len(), 1);
        assert_eq!(linked[0].inner().object_index, 51);
    }
}