ndelement 0.4.0

Finite element definition library.
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

use ndelement::ciarlet::lagrange;
use ndelement::{
    traits::FiniteElement,
    types::{Continuity, ReferenceCellType},
};
use rlst::{DynArray, rlst_dynamic_array};

fn main() {
    // Create a P2 element on a triangle
    let element = lagrange::create::<f64, f64>(
        ReferenceCellType::Triangle,
        2,
        Continuity::Standard,
        lagrange::Variant::Equispaced,
    );

    println!("This element has {} basis functions.", element.dim());

    // Create an array to store the basis function values
    let mut basis_values = DynArray::<f64, 4>::from_shape(element.tabulate_array_shape(0, 1));
    // Create array containing the point [1/3, 1/3]
    let mut points = rlst_dynamic_array!(f64, [2, 1]);
    points[[0, 0]] = 1.0 / 3.0;
    points[[1, 0]] = 1.0 / 3.0;
    // Tabulate the element's basis functions at the point
    element.tabulate(&points, 0, &mut basis_values);
    println!(
        "The values of the basis functions at the point (1/3, 1/3) are: {:?}",
        basis_values.data().unwrap()
    );

    // Set point to [1, 0]
    points[[0, 0]] = 1.0;
    points[[1, 0]] = 0.0;
    // Tabulate the element's basis functions at the point
    element.tabulate(&points, 0, &mut basis_values);
    println!(
        "The values of the basis functions at the point (1, 0) are: {:?}",
        basis_values.data().unwrap()
    );
}