Skip to main content

conspire/domain/fem/block/element/linear/wedge/
mod.rs

1#[cfg(test)]
2mod test;
3
4use crate::{
5    fem::block::element::{
6        FRAC_1_SQRT_3, FiniteElement, IntegrationWeights, ParametricCoordinate,
7        ParametricCoordinates, ParametricReference, ShapeFunctions, ShapeFunctionsGradients,
8        linear::{LinearElement, LinearFiniteElement, M},
9    },
10    math::ScalarList,
11    units::Volume,
12};
13
14const G: usize = 6;
15const N: usize = 6;
16const P: usize = N;
17
18pub type Wedge = LinearElement<G, N>;
19
20impl FiniteElement<G, M, N, P> for Wedge {
21    fn integration_points() -> ParametricCoordinates<G, M> {
22        [
23            [2.0 / 3.0, 1.0 / 6.0, -FRAC_1_SQRT_3],
24            [1.0 / 6.0, 2.0 / 3.0, -FRAC_1_SQRT_3],
25            [1.0 / 6.0, 1.0 / 6.0, -FRAC_1_SQRT_3],
26            [2.0 / 3.0, 1.0 / 6.0, FRAC_1_SQRT_3],
27            [1.0 / 6.0, 2.0 / 3.0, FRAC_1_SQRT_3],
28            [1.0 / 6.0, 1.0 / 6.0, FRAC_1_SQRT_3],
29        ]
30        .into()
31    }
32    fn integration_weights(&self) -> &IntegrationWeights<G, Volume> {
33        &self.integration_weights
34    }
35    fn parametric_reference() -> ParametricReference<M, N> {
36        [
37            [0.0, 0.0, -1.0],
38            [1.0, 0.0, -1.0],
39            [0.0, 1.0, -1.0],
40            [0.0, 0.0, 1.0],
41            [1.0, 0.0, 1.0],
42            [0.0, 1.0, 1.0],
43        ]
44        .into()
45    }
46    fn parametric_weights() -> ScalarList<G> {
47        [1.0 / 6.0; G].into()
48    }
49    fn shape_functions(parametric_coordinate: ParametricCoordinate<M>) -> ShapeFunctions<N> {
50        let [xi_1, xi_2, xi_3] = parametric_coordinate.into();
51        let xi_0 = 1.0 - xi_1 - xi_2;
52        [
53            0.5 * xi_0 * (1.0 - xi_3),
54            0.5 * xi_1 * (1.0 - xi_3),
55            0.5 * xi_2 * (1.0 - xi_3),
56            0.5 * xi_0 * (1.0 + xi_3),
57            0.5 * xi_1 * (1.0 + xi_3),
58            0.5 * xi_2 * (1.0 + xi_3),
59        ]
60        .into()
61    }
62    fn shape_functions_gradients(
63        parametric_coordinate: ParametricCoordinate<M>,
64    ) -> ShapeFunctionsGradients<M, N> {
65        let [xi_1, xi_2, xi_3] = parametric_coordinate.into();
66        let xi_0 = 1.0 - xi_1 - xi_2;
67        [
68            [-0.5 * (1.0 - xi_3), -0.5 * (1.0 - xi_3), -0.5 * xi_0],
69            [0.5 * (1.0 - xi_3), 0.0, -0.5 * xi_1],
70            [0.0, 0.5 * (1.0 - xi_3), -0.5 * xi_2],
71            [-0.5 * (1.0 + xi_3), -0.5 * (1.0 + xi_3), 0.5 * xi_0],
72            [0.5 * (1.0 + xi_3), 0.0, 0.5 * xi_1],
73            [0.0, 0.5 * (1.0 + xi_3), 0.5 * xi_2],
74        ]
75        .into()
76    }
77}
78
79impl LinearFiniteElement<G, N> for Wedge {}