nullgeo-cli 0.2.0

Command-line ray tracer for null geodesics in arbitrary spacetimes
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
134
135
136

use assert_cmd::Command;

#[test]
fn render_command_writes_a_valid_png() {
    let dir = std::env::temp_dir().join(format!("nullgeo_render_test_{}", std::process::id()));
    std::fs::create_dir_all(&dir).unwrap();
    let out = dir.join("out.png");
    let scene = format!(
        r#"
[metric]
kind = "schwarzschild"

[camera]
position = [-15.0, 0.0, 0.0]
fov_deg = 60.0
width = 12
height = 8

[sky]
checker_deg = 20.0

[integrator]
tol = 1e-8
max_steps = 20000

[[output]]
path = "{}"
"#,
        out.display()
    );
    let scene_path = dir.join("scene.toml");
    std::fs::write(&scene_path, scene).unwrap();

    Command::cargo_bin("nullgeo")
        .unwrap()
        .args(["render", scene_path.to_str().unwrap()])
        .assert()
        .success();

    let img = image::open(&out).unwrap();
    assert_eq!((img.width(), img.height()), (12, 8));
    std::fs::remove_dir_all(&dir).ok();
}

#[test]
fn render_command_writes_maps_from_one_trace() {
    let dir = std::env::temp_dir().join(format!("nullgeo_maps_test_{}", std::process::id()));
    std::fs::create_dir_all(&dir).unwrap();
    let scene = format!(
        r#"
[metric]
kind = "schwarzschild"

[camera]
position = [-15.0, 0.0, 0.0]
fov_deg = 60.0
width = 10
height = 6

[sky]
graticule_deg = 15.0

[integrator]
tol = 1e-8
max_steps = 20000

[[output]]
path = "{dir}/beauty.png"

[[output]]
path = "{dir}/class.png"
kind = "classification"

[[output]]
path = "{dir}/min_r.csv"
kind = "min-radius"

[[output]]
path = "{dir}/min_r.pfm"
kind = "min-radius"

[[output]]
path = "{dir}/order.ppm"
kind = "image-order"

[[output]]
path = "{dir}/beauty16.png"
tone = "aces"
bit_depth = 16
"#,
        dir = dir.display()
    );
    let scene_path = dir.join("scene.toml");
    std::fs::write(&scene_path, scene).unwrap();

    Command::cargo_bin("nullgeo")
        .unwrap()
        .args(["render", scene_path.to_str().unwrap()])
        .assert()
        .success();

    let img = image::open(dir.join("class.png")).unwrap();
    assert_eq!((img.width(), img.height()), (10, 6));

    let csv = std::fs::read_to_string(dir.join("min_r.csv")).unwrap();
    let rows: Vec<&str> = csv.lines().collect();
    assert_eq!(rows.len(), 6);
    assert!(rows.iter().all(|r| r.split(',').count() == 10));

    let pfm = std::fs::read(dir.join("min_r.pfm")).unwrap();
    assert!(pfm.starts_with(b"Pf\n10 6\n-1.0\n"));
    assert_eq!(pfm.len(), b"Pf\n10 6\n-1.0\n".len() + 10 * 6 * 4);

    let ppm = std::fs::read(dir.join("order.ppm")).unwrap();
    assert!(ppm.starts_with(b"P6\n10 6\n255\n"));

    let deep = image::open(dir.join("beauty16.png")).unwrap();
    assert_eq!((deep.width(), deep.height()), (10, 6));
    assert_eq!(deep.color(), image::ColorType::Rgb16);
    std::fs::remove_dir_all(&dir).ok();
}

#[test]
fn render_command_rejects_invalid_scene() {
    let dir = std::env::temp_dir().join(format!("nullgeo_badscene_test_{}", std::process::id()));
    std::fs::create_dir_all(&dir).unwrap();
    let scene_path = dir.join("scene.toml");
    std::fs::write(&scene_path, "[metric]\nkind = \"warp-drive\"\n").unwrap();

    Command::cargo_bin("nullgeo")
        .unwrap()
        .args(["render", scene_path.to_str().unwrap()])
        .assert()
        .failure();
    std::fs::remove_dir_all(&dir).ok();
}