RatRod-rs 0.2.1

An FEM implementation written in Rust
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169

pub mod node;
pub mod model;
pub mod elements;
pub mod material;
pub mod section;

#[cfg(test)]
mod tests {
    use sparse_matrix::vector::Vector;

    use crate::{elements::ElementType, model::Model, node::Node, material::Material, section::Section};

    use more_asserts::assert_le;

    #[test]
    fn spring_in_tension() {
        let mut model = Model::new(2, 0.00000001);
        model.add_node(Node {
            x : 0.,
            y : 0.,
            z : 0.
        }).add_node(Node {
                x : 1.,
                y : 0.,
                z : 0.
            }).add_material(Material {
                e : 1.
            }).add_section(Section {
                s : 1.,
                i : 1.
            }).add_element(ElementType::Truss, vec![0,1], 0, 0)
            .add_u_boundary_condition(0, 0, 0.)
            .add_u_boundary_condition(0, 1, 0.)
            .set_force(1, 0, 1.);

        model.solve();

        let u = model.u;
        let f = model.f;
        assert_eq!(u, Vector { values: vec![0.0, 0.0, 0.0, 1.0, 0.0, 0.0] });
        assert_eq!(f, Vector { values: vec![-1.0, 0.0, 0.0, 1.0, 0.0, 0.0] });
    }

    #[test]
    fn beam_in_flexion() {
        let mut model = Model::new(2, 0.00000001);
        model.add_node(Node {
            x : 0.,
            y : 0.,
            z : 0.
        }).add_node(Node {
                x : 1.,
                y : 0.,
                z : 0.
            }).add_material(Material {
                e : 1.
            }).add_section(Section {
                s : 1.,
                i : 1.
            }).add_element(ElementType::Beam, vec![0,1], 0, 0)
            .add_u_boundary_condition(0, 0, 0.)
            .add_u_boundary_condition(0, 1, 0.)
            .add_u_boundary_condition(0, 2, 0.)
            .set_force(1, 1, 3.);

        model.solve();

        let u = model.u;
        let f = model.f;
        assert_le!((&(&u - &Vector { values: vec![0.0, 0.0, 0.0, 0.0, 1.0, 1.5] }).unwrap() * &(&u - &Vector { values: vec![0.0, 0.0, 0.0, 0.0, 1.0, 1.5] }).unwrap()).unwrap().sqrt(), 0.01);
        assert_le!((&(&f - &Vector { values: vec![0.0, -3.0, -3.0, 0.0, 3.0, 0.0] }).unwrap() * &(&u - &Vector { values: vec![0.0, -3.0, -3.0, 0.0, 3.0, 0.0] }).unwrap()).unwrap().sqrt(), 0.01);
    }

    #[test]
    fn multielement_beam_in_flexion() {
        let mut model = Model::new(2, 0.00000001);
        model.add_node(Node {
            x : 0.,
            y : 0.,
            z : 0.
        }).add_node(Node {
                x : 0.5,
                y : 0.,
                z : 0.
            }).add_node(Node {
                x : 1.,
                y : 0.,
                z : 0.
            }).add_material(Material {
                e : 1.
            }).add_section(Section {
                s : 1.,
                i : 1.
            }).add_element(ElementType::Beam, vec![0,1], 0, 0)
            .add_element(ElementType::Beam, vec![1,2], 0, 0)
            .add_u_boundary_condition(0, 0, 0.)
            .add_u_boundary_condition(0, 1, 0.)
            .add_u_boundary_condition(0, 2, 0.)
            .set_force(2, 1, 3.);

        model.solve();

        let u = model.u;
        let f = model.f;
        assert_le!((&(&u - &Vector { values: vec![0.0, 0.0, 0.0, 0.0, 15./48., 9./8., 0.0, 1.0, 1.5] }).unwrap() * &(&u - &Vector { values: vec![0.0, 0.0, 0.0, 0.0, 15./48., 9./8., 0.0, 1.0, 1.5] }).unwrap()).unwrap().sqrt(), 0.01);
        assert_le!((&(&f - &Vector { values: vec![0.0, -3.0, -3.0, 0.0, 0.0, 0.0, 0.0, 3.0, 0.0] }).unwrap() * &(&u - &Vector { values: vec![0.0, -3.0, -3.0, 0.0, 0.0, 0.0, 0.0, 3.0, 0.0] }).unwrap()).unwrap().sqrt(), 0.01);
    }
    #[test]
    fn save_model() {
        let mut model = Model::new(2, 0.00000001);
        model.add_node(Node {
            x : 0.,
            y : 0.,
            z : 0.
        }).add_node(Node {
                x : 0.5,
                y : 0.,
                z : 0.
            }).add_node(Node {
                x : 1.,
                y : 0.,
                z : 0.
            }).add_material(Material {
                e : 1.
            }).add_section(Section {
                s : 1.,
                i : 1.
            }).add_element(ElementType::Beam, vec![0,1], 0, 0)
            .add_element(ElementType::Beam, vec![1,2], 0, 0)
            .add_u_boundary_condition(0, 0, 0.)
            .add_u_boundary_condition(0, 1, 0.)
            .add_u_boundary_condition(0, 2, 0.)
            .set_force(2, 1, 3.);
        model.save("model_test.json");
    }

    #[test]
    fn load_model() {
        let mut model = Model::new(2, 0.00000001);
        model.add_node(Node {
            x : 0.,
            y : 0.,
            z : 0.
        }).add_node(Node {
                x : 0.5,
                y : 0.,
                z : 0.
            }).add_node(Node {
                x : 1.,
                y : 0.,
                z : 0.
            }).add_material(Material {
                e : 1.
            }).add_section(Section {
                s : 1.,
                i : 1.
            }).add_element(ElementType::Beam, vec![0,1], 0, 0)
            .add_element(ElementType::Beam, vec![1,2], 0, 0)
            .add_u_boundary_condition(0, 0, 0.)
            .add_u_boundary_condition(0, 1, 0.)
            .add_u_boundary_condition(0, 2, 0.)
            .set_force(2, 1, 3.);
        model.save("model_test.json");
        let test = Model::load("model_test.json");
        assert_eq!(model, test);
    }
}