vtk-pure-rs 0.2.0

Pure Rust visualization toolkit — data structures, filters, I/O, rendering
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 crate::data::{AnyDataArray, DataArray, ImageData};

/// Apply a spherical region-of-interest mask to ImageData.
///
/// Voxels inside the sphere keep their values; outside voxels are set to `outside_value`.
pub fn image_sphere_roi(
    input: &ImageData, scalars: &str, center: [f64; 3], radius: f64, outside_value: f64,
) -> ImageData {
    let arr = match input.point_data().get_array(scalars) {
        Some(a) => a, None => return input.clone(),
    };

    let dims = input.dimensions();
    let nx = dims[0] as usize; let ny = dims[1] as usize; let nz = dims[2] as usize;
    let origin = input.origin(); let spacing = input.spacing();
    let r2 = radius * radius;

    let mut buf = [0.0f64];
    let mut values = Vec::with_capacity(nx*ny*nz);

    for k in 0..nz { for j in 0..ny { for i in 0..nx {
        let x = origin[0] + i as f64 * spacing[0] - center[0];
        let y = origin[1] + j as f64 * spacing[1] - center[1];
        let z = origin[2] + k as f64 * spacing[2] - center[2];
        arr.tuple_as_f64(k*ny*nx+j*nx+i, &mut buf);
        values.push(if x*x+y*y+z*z <= r2 { buf[0] } else { outside_value });
    }}}

    let mut img = input.clone();
    let mut new_attrs = crate::data::DataSetAttributes::new();
    for i in 0..input.point_data().num_arrays() {
        let a = input.point_data().get_array_by_index(i).unwrap();
        if a.name() == scalars {
            new_attrs.add_array(AnyDataArray::F64(DataArray::from_vec(scalars, values.clone(), 1)));
        } else { new_attrs.add_array(a.clone()); }
    }
    *img.point_data_mut() = new_attrs;
    img
}

/// Apply a box ROI mask to ImageData.
pub fn image_box_roi(
    input: &ImageData, scalars: &str, bounds: [f64; 6], outside_value: f64,
) -> ImageData {
    let arr = match input.point_data().get_array(scalars) {
        Some(a) => a, None => return input.clone(),
    };

    let dims = input.dimensions();
    let nx = dims[0] as usize; let ny = dims[1] as usize; let nz = dims[2] as usize;
    let origin = input.origin(); let spacing = input.spacing();

    let mut buf = [0.0f64];
    let mut values = Vec::with_capacity(nx*ny*nz);

    for k in 0..nz { for j in 0..ny { for i in 0..nx {
        let x = origin[0] + i as f64 * spacing[0];
        let y = origin[1] + j as f64 * spacing[1];
        let z = origin[2] + k as f64 * spacing[2];
        arr.tuple_as_f64(k*ny*nx+j*nx+i, &mut buf);
        let inside = x >= bounds[0] && x <= bounds[1]
            && y >= bounds[2] && y <= bounds[3]
            && z >= bounds[4] && z <= bounds[5];
        values.push(if inside { buf[0] } else { outside_value });
    }}}

    let mut img = input.clone();
    let mut new_attrs = crate::data::DataSetAttributes::new();
    for i in 0..input.point_data().num_arrays() {
        let a = input.point_data().get_array_by_index(i).unwrap();
        if a.name() == scalars {
            new_attrs.add_array(AnyDataArray::F64(DataArray::from_vec(scalars, values.clone(), 1)));
        } else { new_attrs.add_array(a.clone()); }
    }
    *img.point_data_mut() = new_attrs;
    img
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn sphere_roi() {
        let mut img = ImageData::with_dimensions(5, 5, 1);
        img.set_spacing([1.0; 3]);
        img.point_data_mut().add_array(AnyDataArray::F64(
            DataArray::from_vec("v", vec![1.0; 25], 1),
        ));

        let result = image_sphere_roi(&img, "v", [2.0, 2.0, 0.0], 1.5, 0.0);
        let arr = result.point_data().get_array("v").unwrap();
        let mut buf = [0.0f64];
        arr.tuple_as_f64(12, &mut buf); assert_eq!(buf[0], 1.0); // center inside
        arr.tuple_as_f64(0, &mut buf); assert_eq!(buf[0], 0.0); // corner outside
    }

    #[test]
    fn box_roi() {
        let mut img = ImageData::with_dimensions(5, 5, 1);
        img.set_spacing([1.0; 3]);
        img.point_data_mut().add_array(AnyDataArray::F64(
            DataArray::from_vec("v", vec![1.0; 25], 1),
        ));

        let result = image_box_roi(&img, "v", [1.0, 3.0, 1.0, 3.0, -1.0, 1.0], 0.0);
        let arr = result.point_data().get_array("v").unwrap();
        let mut buf = [0.0f64];
        arr.tuple_as_f64(12, &mut buf); assert_eq!(buf[0], 1.0); // (2,2) inside
        arr.tuple_as_f64(0, &mut buf); assert_eq!(buf[0], 0.0); // (0,0) outside
    }

    #[test]
    fn missing_array() {
        let img = ImageData::with_dimensions(3, 3, 1);
        let r = image_sphere_roi(&img, "nope", [0.0; 3], 1.0, 0.0);
        assert_eq!(r.dimensions(), [3, 3, 1]);
    }
}