epanet-rs 0.2.2

A fast, modern and safe re-implementation of the EPANET2 hydraulic solver, written in Rust.
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

//! Container for per-report-step solver results (flows, heads, demands) in user units.

use crate::model::units::{Cfs, FlowUnits, Ft, UnitSystem};
use crate::solver::state::SolverState;
use serde::Serialize;

#[derive(Serialize, Debug)]
pub struct SolverResult {
    pub flows: Vec<Vec<Cfs>>,
    pub heads: Vec<Vec<Ft>>,
    pub demands: Vec<Vec<Cfs>>,
}

impl SolverResult {
    pub fn new(n_links: usize, n_nodes: usize, n_steps: usize) -> Self {
        Self {
            flows: vec![vec![0.0; n_links]; n_steps],
            heads: vec![vec![0.0; n_nodes]; n_steps],
            demands: vec![vec![0.0; n_nodes]; n_steps],
        }
    }

    pub fn append(&mut self, state: &SolverState, step: usize) {
        self.flows[step] = state.flows.clone();
        self.heads[step] = state.heads.clone();
        self.demands[step] = state.demands.clone();
    }

    // convert the solver units back to the original units
    pub fn convert_units(&mut self, to_flow_units: &FlowUnits, to_unit_system: &UnitSystem) {
        let flow_scale = to_flow_units.per_cfs();
        let head_scale = to_unit_system.per_feet();

        self.flows
            .iter_mut()
            .flatten()
            .for_each(|flow| *flow *= flow_scale);
        self.heads
            .iter_mut()
            .flatten()
            .for_each(|head| *head *= head_scale);
        self.demands
            .iter_mut()
            .flatten()
            .for_each(|demand| *demand *= flow_scale);
    }
}