plotpy 1.21.0

Rust plotting library using Python (Matplotlib)
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
120
121
122
123
124
125
126
127
128
129

use plotpy::{Image, Plot, StrError};
use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::Path;

const OUT_DIR: &str = "/tmp/plotpy/integ_tests";

#[test]
fn test_image_1() -> Result<(), StrError> {
    // data
    let data = [
        [0.8, 2.4, 2.5, 3.9, 0.0, 4.0, 0.0],
        [2.4, 0.0, 4.0, 1.0, 2.7, 0.0, 0.0],
        [1.1, 2.4, 0.8, 4.3, 1.9, 4.4, 0.0],
        [0.6, 0.0, 0.3, 0.0, 3.1, 0.0, 0.0],
        [0.7, 1.7, 0.6, 2.6, 2.2, 6.2, 0.0],
        [1.3, 1.2, 0.0, 0.0, 0.0, 3.2, 5.1],
        [0.1, 2.0, 0.0, 1.4, 0.0, 1.9, 6.3],
    ];

    // image plot and options
    let mut img = Image::new();
    img.set_colormap_name("terrain").set_extra("alpha=0.8").draw(&data);

    let mut plot = Plot::new();
    plot.add(&img);

    // save figure
    let path = Path::new(OUT_DIR).join("integ_image_1.svg");
    plot.set_show_errors(true).save(&path)?;

    // check number of lines
    let file = File::open(path).map_err(|_| "cannot open file")?;
    let buffered = BufReader::new(file);
    let lines_iter = buffered.lines();
    let c = lines_iter.count();
    assert!(c > 420 && c < 500);
    Ok(())
}

#[test]
fn test_image_with_rgb() -> Result<(), StrError> {
    let data = vec![
        // --- Row 0 ---
        vec![
            vec![1.0, 0.0, 0.0], // Pixel 0,0: Red
            vec![0.0, 1.0, 0.0], // Pixel 0,1: Green
            vec![0.0, 0.0, 1.0], // Pixel 0,2: Blue
        ],
        // --- Row 1 ---
        vec![
            vec![1.0, 1.0, 0.0], // Pixel 1,0: Yellow
            vec![1.0, 0.0, 1.0], // Pixel 1,1: Magenta
            vec![0.0, 1.0, 1.0], // Pixel 1,2: Cyan
        ],
        // --- Row 2 ---
        vec![
            vec![0.5, 0.5, 0.5], // Pixel 2,0: Gray
            vec![1.0, 1.0, 1.0], // Pixel 2,1: White
            vec![0.0, 0.0, 0.0], // Pixel 2,2: Black
        ],
    ];

    // image plot and options
    let mut img = Image::new();
    img.set_colormap_name("terrain")
        .set_extra("alpha=0.8")
        .draw_rgb_or_rgba(&data);

    let mut plot = Plot::new();
    plot.add(&img);

    // save figure
    let path = Path::new(OUT_DIR).join("integ_image_with_rgb.svg");
    plot.set_show_errors(true).save(&path)?;

    // check number of lines
    let file = File::open(path).map_err(|_| "cannot open file")?;
    let buffered = BufReader::new(file);
    let lines_iter = buffered.lines();
    let c = lines_iter.count();
    assert!(c > 400 && c < 430);
    Ok(())
}

#[test]
fn test_image_with_rgba() -> Result<(), StrError> {
    let data = vec![
        // --- Row 0 ---
        vec![
            vec![1.0, 0.0, 0.0, 0.5], // Pixel 0,0: Red
            vec![0.0, 1.0, 0.0, 0.5], // Pixel 0,1: Green
            vec![0.0, 0.0, 1.0, 0.5], // Pixel 0,2: Blue
        ],
        // --- Row 1 ---
        vec![
            vec![1.0, 1.0, 0.0, 0.8], // Pixel 1,0: Yellow
            vec![1.0, 0.0, 1.0, 0.8], // Pixel 1,1: Magenta
            vec![0.0, 1.0, 1.0, 0.8], // Pixel 1,2: Cyan
        ],
        // --- Row 2 ---
        vec![
            vec![0.5, 0.5, 0.5, 0.2], // Pixel 2,0: Gray
            vec![1.0, 1.0, 1.0, 0.2], // Pixel 2,1: White
            vec![0.0, 0.0, 0.0, 0.2], // Pixel 2,2: Black
        ],
    ];

    // image plot and options
    let mut img = Image::new();
    img.set_colormap_name("terrain")
        .set_extra("alpha=0.8")
        .draw_rgb_or_rgba(&data);

    let mut plot = Plot::new();
    plot.add(&img);

    // save figure
    let path = Path::new(OUT_DIR).join("integ_image_with_rgba.svg");
    plot.set_show_errors(true).save(&path)?;

    // check number of lines
    let file = File::open(path).map_err(|_| "cannot open file")?;
    let buffered = BufReader::new(file);
    let lines_iter = buffered.lines();
    let c = lines_iter.count();
    assert!(c > 400 && c < 430);
    Ok(())
}