scirs2-integrate 0.4.0

Numerical integration module for SciRS2 (scirs2-integrate)
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

//! Fluid state representations for 2D and 3D computational fluid dynamics.

use scirs2_core::ndarray::{Array2, Array3};

/// Fluid state representation for 2D fluid dynamics
#[derive(Debug, Clone)]
pub struct FluidState {
    /// Velocity field (u, v) for 2D
    pub velocity: Vec<Array2<f64>>,
    /// Pressure field
    pub pressure: Array2<f64>,
    /// Temperature field (optional)
    pub temperature: Option<Array2<f64>>,
    /// Time
    pub time: f64,
    /// Grid spacing
    pub dx: f64,
    pub dy: f64,
}

/// 3D fluid state representation for 3D fluid dynamics
#[derive(Debug, Clone)]
pub struct FluidState3D {
    /// Velocity field (u, v, w)
    pub velocity: Vec<Array3<f64>>,
    /// Pressure field
    pub pressure: Array3<f64>,
    /// Temperature field (optional)
    pub temperature: Option<Array3<f64>>,
    /// Time
    pub time: f64,
    /// Grid spacing
    pub dx: f64,
    pub dy: f64,
    pub dz: f64,
}

impl FluidState {
    /// Create a new 2D fluid state with the given dimensions and grid spacing
    pub fn new(nx: usize, ny: usize, dx: f64, dy: f64) -> Self {
        Self {
            velocity: vec![
                Array2::zeros((nx, ny)), // u-component
                Array2::zeros((nx, ny)), // v-component
            ],
            pressure: Array2::zeros((nx, ny)),
            temperature: None,
            time: 0.0,
            dx,
            dy,
        }
    }

    /// Create a new 2D fluid state with temperature field
    pub fn new_with_temperature(nx: usize, ny: usize, dx: f64, dy: f64) -> Self {
        Self {
            velocity: vec![
                Array2::zeros((nx, ny)), // u-component
                Array2::zeros((nx, ny)), // v-component
            ],
            pressure: Array2::zeros((nx, ny)),
            temperature: Some(Array2::zeros((nx, ny))),
            time: 0.0,
            dx,
            dy,
        }
    }

    /// Get the grid dimensions (nx, ny)
    pub fn dimensions(&self) -> (usize, usize) {
        let shape = self.pressure.raw_dim();
        (shape[0], shape[1])
    }

    /// Check if the state has temperature field
    pub fn has_temperature(&self) -> bool {
        self.temperature.is_some()
    }
}

impl FluidState3D {
    /// Create a new 3D fluid state with the given dimensions and grid spacing
    pub fn new(nx: usize, ny: usize, nz: usize, dx: f64, dy: f64, dz: f64) -> Self {
        Self {
            velocity: vec![
                Array3::zeros((nx, ny, nz)), // u-component
                Array3::zeros((nx, ny, nz)), // v-component
                Array3::zeros((nx, ny, nz)), // w-component
            ],
            pressure: Array3::zeros((nx, ny, nz)),
            temperature: None,
            time: 0.0,
            dx,
            dy,
            dz,
        }
    }

    /// Create a new 3D fluid state with temperature field
    pub fn new_with_temperature(
        nx: usize,
        ny: usize,
        nz: usize,
        dx: f64,
        dy: f64,
        dz: f64,
    ) -> Self {
        Self {
            velocity: vec![
                Array3::zeros((nx, ny, nz)), // u-component
                Array3::zeros((nx, ny, nz)), // v-component
                Array3::zeros((nx, ny, nz)), // w-component
            ],
            pressure: Array3::zeros((nx, ny, nz)),
            temperature: Some(Array3::zeros((nx, ny, nz))),
            time: 0.0,
            dx,
            dy,
            dz,
        }
    }

    /// Get the grid dimensions (nx, ny, nz)
    pub fn dimensions(&self) -> (usize, usize, usize) {
        let shape = self.pressure.raw_dim();
        (shape[0], shape[1], shape[2])
    }

    /// Check if the state has temperature field
    pub fn has_temperature(&self) -> bool {
        self.temperature.is_some()
    }
}