pmx_parser 0.2.0

load and save .pmx model file.
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

use std::fmt::{Debug, Formatter};
use std::io::{Read, Write};

use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};

use crate::error::PmxError;
use crate::header::Header;
use crate::kits::{read_f32x3, read_vec, write_f32x3};
use crate::RigidBodyIndex;

#[derive(Default, Debug, Clone, PartialEq)]
pub struct Joints {
    pub joints: Vec<Joint>,
}

impl Joints {
    pub fn count(&self) -> u32 {
        self.joints.len() as u32
    }
    pub fn read<R: Read>(header: &Header, read: &mut R) -> Result<Self, PmxError> {
        Ok(Self {
            joints: read_vec(read, |read| Joint::read(header, read))?,
        })
    }
    pub fn write<W: Write>(&self, header: &Header, write: &mut W) -> Result<(), PmxError> {
        write.write_u32::<LittleEndian>(self.count())?;
        for i in &self.joints {
            i.write(header, write)?;
        }
        Ok(())
    }
}

#[derive(Clone, PartialEq)]
pub struct Joint {
    pub name: String,
    pub name_en: String,
    pub joint_type: JointType,
    pub a_rigid_index: RigidBodyIndex,
    pub b_rigid_index: RigidBodyIndex,
    pub position: [f32; 3],
    pub rotation: [f32; 3],
    pub move_limit_down: [f32; 3],
    pub move_limit_up: [f32; 3],
    pub rotation_limit_down: [f32; 3],
    pub rotation_limit_up: [f32; 3],
    pub spring_const_move: [f32; 3],
    pub spring_const_rotation: [f32; 3],
}

impl Debug for Joint {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        let mut s = f.debug_struct("Joint");
        s.field("name", &self.name);
        s.finish()
    }
}

impl Joint {
    pub fn read<R: Read>(header: &Header, read: &mut R) -> Result<Self, PmxError> {
        Ok(Self {
            name: header.encoding.read(read)?,
            name_en: header.encoding.read(read)?,
            joint_type: JointType::try_from(read.read_u8()?)?,
            a_rigid_index: header.rigid_body_index.read(read)?,
            b_rigid_index: header.rigid_body_index.read(read)?,
            position: read_f32x3(read)?,
            rotation: read_f32x3(read)?,
            move_limit_down: read_f32x3(read)?,
            move_limit_up: read_f32x3(read)?,
            rotation_limit_down: read_f32x3(read)?,
            rotation_limit_up: read_f32x3(read)?,
            spring_const_move: read_f32x3(read)?,
            spring_const_rotation: read_f32x3(read)?,
        })
    }
    pub fn write<W: Write>(&self, header: &Header, write: &mut W) -> Result<(), PmxError> {
        header.encoding.write(write, self.name.as_str())?;
        header.encoding.write(write, self.name_en.as_str())?;
        write.write_u8(self.joint_type as u8)?;
        header.rigid_body_index.write(write, self.a_rigid_index)?;
        header.rigid_body_index.write(write, self.b_rigid_index)?;
        write_f32x3(write, self.position)?;
        write_f32x3(write, self.rotation)?;
        write_f32x3(write, self.move_limit_down)?;
        write_f32x3(write, self.move_limit_up)?;
        write_f32x3(write, self.rotation_limit_down)?;
        write_f32x3(write, self.rotation_limit_up)?;
        write_f32x3(write, self.spring_const_move)?;
        write_f32x3(write, self.spring_const_rotation)?;
        Ok(())
    }
}

#[derive(Copy, Clone, Debug, PartialEq, Eq)]
#[repr(u8)]
pub enum JointType {
    Spring6DOF = 0x00,
    SixDof = 0x01,
    P2P = 0x02,
    ConeTwist = 0x03,
    Slider = 0x04,
    Hinge = 0x05,
}

impl TryFrom<u8> for JointType {
    type Error = PmxError;

    fn try_from(value: u8) -> Result<Self, Self::Error> {
        match value {
            0x00 => Ok(Self::Spring6DOF),
            0x01 => Ok(Self::SixDof),
            0x02 => Ok(Self::P2P),
            0x03 => Ok(Self::ConeTwist),
            0x04 => Ok(Self::Slider),
            0x05 => Ok(Self::Hinge),
            _ => Err(PmxError::JointTypeError),
        }
    }
}