conspire/domain/fem/block/element/linear/wedge/
mod.rs1#[cfg(test)]
2mod test;
3
4use crate::{
5 fem::block::element::{
6 FRAC_1_SQRT_3, FiniteElement, ParametricCoordinate, ParametricCoordinates,
7 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 shape_functions(parametric_coordinate: ParametricCoordinate<M>) -> ShapeFunctions<N> {
49 let [xi_1, xi_2, xi_3] = parametric_coordinate.into();
50 let xi_0 = 1.0 - xi_1 - xi_2;
51 [
52 0.5 * xi_0 * (1.0 - xi_3),
53 0.5 * xi_1 * (1.0 - xi_3),
54 0.5 * xi_2 * (1.0 - xi_3),
55 0.5 * xi_0 * (1.0 + xi_3),
56 0.5 * xi_1 * (1.0 + xi_3),
57 0.5 * xi_2 * (1.0 + xi_3),
58 ]
59 .into()
60 }
61 fn shape_functions_gradients(
62 parametric_coordinate: ParametricCoordinate<M>,
63 ) -> ShapeFunctionsGradients<M, N> {
64 let [xi_1, xi_2, xi_3] = parametric_coordinate.into();
65 let xi_0 = 1.0 - xi_1 - xi_2;
66 [
67 [-0.5 * (1.0 - xi_3), -0.5 * (1.0 - xi_3), -0.5 * xi_0],
68 [0.5 * (1.0 - xi_3), 0.0, -0.5 * xi_1],
69 [0.0, 0.5 * (1.0 - xi_3), -0.5 * xi_2],
70 [-0.5 * (1.0 + xi_3), -0.5 * (1.0 + xi_3), 0.5 * xi_0],
71 [0.5 * (1.0 + xi_3), 0.0, 0.5 * xi_1],
72 [0.0, 0.5 * (1.0 + xi_3), 0.5 * xi_2],
73 ]
74 .into()
75 }
76}
77
78impl LinearFiniteElement<G, N> for Wedge {}