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

use sparse_matrix::matrix::coo_mat::CooMat;

use self::{truss::Truss, beam::Beam};

pub mod truss;
pub mod beam;

use serde::{Serialize, Deserialize};

#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
pub enum Element {
    Truss(Truss),
    Beam(Beam),
}

pub enum ElementType {
    Truss,
    Beam
}

pub trait ElementTrait {
    fn get_matrix(&self, dimension : u8, mat_size : usize, nodes : &Vec<usize>) -> CooMat;
}

impl ElementTrait for Element {
    fn get_matrix(&self, dimension : u8, mat_size : usize, nodes : &Vec<usize>) -> CooMat {
        match self {
            Element::Truss(truss) => truss.get_matrix(dimension, mat_size, nodes[0], nodes[1]),
            Element::Beam(beam) => beam.get_matrix(dimension, mat_size, nodes[0], nodes[1]),
        }
    }
}