parse-monitors 0.4.0

Giant Magellan Telescope Computation Fluid Dynamics Parser
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

//! Pressure maps
//!
//! Plot the OPD maps

use glob::glob;
use npyz::npz::NpzArchive;
use parse_monitors::{
    cfd::{self, BaselineTrait},
    CFD_YEAR,
};
use rayon::prelude::*;
use std::{error::Error, time::Instant};

fn main() -> Result<(), Box<dyn Error>> {
    let pattern = "optvol_optvol*.npz";

    cfd::Baseline::<CFD_YEAR>::default()
        //.extras()
        .into_iter()
        .skip(20)
        .take(20)
        .collect::<Vec<cfd::CfdCase<CFD_YEAR>>>()
        .into_par_iter()
        .for_each(|cfd_case| {
            let now = Instant::now();
            let case_path = cfd::Baseline::<CFD_YEAR>::path()
                .expect("undefined path")
                .join(cfd_case.to_string());
            let files: Vec<_> = glob(case_path.join("optvol").join(pattern).to_str().unwrap())
                .unwrap()
                .collect();

            let file = files.last().unwrap();
            let opd: Vec<f64> = NpzArchive::open(file.as_ref().unwrap())
                .map(|mut npz| {
                    npz.by_name("opd")
                        .unwrap()
                        .map(|npy| npy.into_vec::<f64>().unwrap())
                })
                .unwrap()
                .unwrap()
                .into_iter()
                .map(|x| x * 1e6)
                .collect();

            let path = case_path.join("report").join("opd_map.png");
            let filename = format!("{}", path.as_path().display());
            let _: complot::Heatmap = (
                (opd.as_slice(), (512, 512)),
                complot::complot!(filename, xlabel = "WFE [micron]"),
            )
                .into();

            println!("{:?}: {:>8}s", file, now.elapsed().as_secs());
        });
    Ok(())
}