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

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