plot3d 0.1.11

Utilities for reading, writing, and manipulating NASA PLOT3D structured grids.
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

use plot3d::{block_face_functions, connectivity_fast, read_plot3d_ascii, FaceRecord};
#[test]
fn test_connectivity() {
    // download mesh
    let url = "https://nasa-public-data.s3.amazonaws.com/plot3d_utilities/VSPT_ASCII.xyz";
    let ascii_path = "VSPT_ASCII.xyz";
    if !std::path::Path::new(ascii_path).exists() {
        let bytes = reqwest::blocking::get(url).unwrap().bytes().unwrap();
        std::fs::write(ascii_path, &bytes).unwrap();
    }

    // read ASCII
    let blocks = read_plot3d_ascii(ascii_path).unwrap();
    assert!(blocks.len() == 2);
    let (i, j, k) = (8, 8, 8);

    println!("Block 0 printing at {i}, {j}, {k}");
    blocks[0].print_xyz(i, j, k);

    let factor = 4;
    let reduced_blocks = block_face_functions::reduce_blocks(&blocks, factor);
    println!("Reduced block 0 printing at {i}, {j}, {k}");
    reduced_blocks[0].print_xyz(i / factor, j / factor, k / factor);

    // verify all nodes align between original and reduced block
    let original = &blocks[0];
    let reduced = &reduced_blocks[0];
    for (ri, i_idx) in (0..original.imax).step_by(factor).enumerate() {
        for (rj, j_idx) in (0..original.jmax).step_by(factor).enumerate() {
            for (rk, k_idx) in (0..original.kmax).step_by(factor).enumerate() {
                let (ox, oy, oz) = original.xyz(i_idx, j_idx, k_idx);
                let (rx, ry, rz) = reduced.xyz(ri, rj, rk);
                assert!((ox - rx).abs() < 1e-9);
                assert!((oy - ry).abs() < 1e-9);
                assert!((oz - rz).abs() < 1e-9);
            }
        }
    }

    // // Find Connectivity
    let (matches, outer_faces) = connectivity_fast(&blocks);

    fn face_summary(
        face: &FaceRecord,
    ) -> (
        usize,
        usize,
        usize,
        usize,
        usize,
        usize,
        usize,
        Option<usize>,
    ) {
        (
            face.block_index,
            face.il,
            face.jl,
            face.kl,
            face.ih,
            face.jh,
            face.kh,
            face.id,
        )
    }

    let matches_summary: Vec<_> = matches
        .iter()
        .map(|m| {
            (
                face_summary(&m.block1),
                face_summary(&m.block2),
                m.points.len(),
            )
        })
        .collect();

    // Expected: 2 matches, not 3. A 3rd match previously emitted by
    // Phase 3 as blk0 [0..128, 0..100, 32] <-> blk1 [40..168, 0..100, 0]
    // (52 points) was a wrap-around duplicate of the 858-point Phase 2
    // match on the SAME block pair and the SAME block1 region
    // [40..168, 0..100, 0]. `phase3_overlaps_existing` now drops it,
    // matching the Python plot3d reference implementation. See
    // `docs/plot3d_phase3_overlap_guard.md` in tgs-py-grc for the
    // root-cause writeup.
    let expected_matches = vec![
        (
            (0, 128, 0, 32, 256, 100, 32, None),
            (1, 40, 0, 0, 168, 100, 0, None),
            858,
        ),
        (
            (0, 0, 0, 0, 0, 100, 32, None),
            (0, 256, 0, 0, 256, 100, 32, None),
            0,
        ),
    ];
    assert_eq!(matches_summary, expected_matches);

    let outer_faces_summary: Vec<_> = outer_faces.iter().map(face_summary).collect();
    // The blk0 K=32 face region [0..128, 0..100, 32] is now reported
    // as an outer face because the spurious Phase 3 match that used
    // to claim it was dropped as a duplicate. This matches the Python
    // plot3d reference, which lists the same region in outer_faces.
    let expected_outer_faces = vec![
        (0, 0, 0, 0, 256, 0, 32, Some(1)),
        (0, 0, 100, 0, 256, 100, 32, Some(2)),
        (0, 0, 0, 0, 256, 100, 0, Some(3)),
        (0, 0, 0, 32, 128, 100, 32, Some(4)),
        (1, 0, 0, 0, 0, 100, 52, Some(5)),
        (1, 268, 0, 0, 268, 100, 52, Some(6)),
        (1, 0, 0, 0, 268, 0, 52, Some(7)),
        (1, 0, 100, 0, 268, 100, 52, Some(8)),
        (1, 0, 0, 52, 268, 100, 52, Some(9)),
        (1, 0, 0, 0, 40, 100, 0, Some(10)),
        (1, 168, 0, 0, 268, 100, 0, Some(11)),
    ];
    assert_eq!(outer_faces_summary, expected_outer_faces);
}