aeon-tk 0.3.3

Toolkit for running finite difference simulations with adaptive mesh refinement
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

use crate::geometry::IndexSpace;

pub trait Support<const N: usize> {
    fn num_points(&self) -> usize;
    fn point(&self, i: usize) -> [f64; N];
}

/// A [-1, 1]ᴺ hypercube with uniformly placed support points along each axis.
pub struct Uniform<const N: usize>([usize; N]);

impl<const N: usize> Uniform<N> {
    pub fn new(width: [usize; N]) -> Self {
        Self(width)
    }
}

impl<const N: usize> Support<N> for Uniform<N> {
    fn num_points(&self) -> usize {
        self.0.iter().product()
    }

    fn point(&self, i: usize) -> [f64; N] {
        let spacing: [f64; N] = std::array::from_fn(|i| 2.0 / (self.0[i] - 1) as f64);
        let cartesian = IndexSpace::new(self.0).cartesian_from_linear(i);

        std::array::from_fn(|i| spacing[i] * cartesian[i] as f64 - 1.0)
    }
}