andam 0.1.2

A comprehensive Rust library for cosmological calculations, visualizations and simulations
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

//! Interactive plotting using Plotly

use plotly::{
    common::{Mode, Title},
    layout::{Axis, Layout},
    Plot, Scatter,
};
use std::error::Error;

/// Create an interactive line plot
pub fn create_interactive_plot(
    filename: &str,
    datasets: Vec<(&[(f64, f64)], &str)>,
    title: &str,
    x_label: &str,
    y_label: &str,
) -> Result<(), Box<dyn Error>> {
    let mut plot = Plot::new();

    for (data, name) in datasets {
        let x: Vec<f64> = data.iter().map(|(x, _)| *x).collect();
        let y: Vec<f64> = data.iter().map(|(_, y)| *y).collect();

        let trace = Scatter::new(x, y)
            .mode(Mode::Lines)
            .name(name);

        plot.add_trace(trace);
    }

    let layout = Layout::new()
        .title(Title::new(title))
        .x_axis(Axis::new().title(Title::new(x_label)))
        .y_axis(Axis::new().title(Title::new(y_label)));

    plot.set_layout(layout);
    plot.write_html(filename);

    Ok(())
}

/// Create a log-log plot
pub fn create_loglog_interactive(
    filename: &str,
    datasets: Vec<(&[(f64, f64)], &str)>,
    title: &str,
    x_label: &str,
    y_label: &str,
) -> Result<(), Box<dyn Error>> {
    let mut plot = Plot::new();

    for (data, name) in datasets {
        let x: Vec<f64> = data.iter().map(|(x, _)| *x).collect();
        let y: Vec<f64> = data.iter().map(|(_, y)| *y).collect();

        let trace = Scatter::new(x, y)
            .mode(Mode::Lines)
            .name(name);

        plot.add_trace(trace);
    }

    let layout = Layout::new()
        .title(Title::new(title))
        .x_axis(Axis::new()
            .title(Title::new(x_label))
            .type_(plotly::layout::AxisType::Log))
        .y_axis(Axis::new()
            .title(Title::new(y_label))
            .type_(plotly::layout::AxisType::Log));

    plot.set_layout(layout);
    plot.write_html(filename);

    Ok(())
}