russell_lab 1.15.0

Scientific laboratory for linear algebra and numerical mathematics
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

use plotpy::{Curve, Plot};
use russell_lab::math::GOLDEN_RATIO;
use russell_lab::*;

const OUT_DIR: &str = "/tmp/russell_lab/";

fn main() -> Result<(), StrError> {
    // values
    let np = 201;
    let xx = Vector::linspace(-2.0, 3.0, np)?;
    let mut bb = xx.get_mapped(|x| math::beta(0.5, x));
    let mut lb = xx.get_mapped(|x| math::ln_beta(0.5, x));
    let mut gg = xx.get_mapped(|x| math::gamma(x));
    let mut lg = xx.get_mapped(|x| math::ln_gamma(x).0);

    // replace ±Inf with NaN (ok in Matplotlib)
    bb.as_mut_data().iter_mut().for_each(|y| {
        if !y.is_finite() {
            *y = f64::NAN;
        }
    });
    lb.as_mut_data().iter_mut().for_each(|y| {
        if !y.is_finite() {
            *y = f64::NAN;
        }
    });
    gg.as_mut_data().iter_mut().for_each(|y| {
        if !y.is_finite() {
            *y = f64::NAN;
        }
    });
    lg.as_mut_data().iter_mut().for_each(|y| {
        if !y.is_finite() {
            *y = f64::NAN;
        }
    });

    // plot beta
    let mut curve = Curve::new();
    curve.set_line_color("#C23B23").set_line_width(2.5);
    curve.draw(xx.as_data(), bb.as_data());
    let path = format!("{}/math_beta_gamma_functions_bb.svg", OUT_DIR);
    let mut plot = Plot::new();
    plot.add(&curve)
        .set_yrange(-10.0, 10.0)
        .grid_labels_legend("$x$", "$B\\left(\\frac{1}{2},x\\right)$")
        .set_figure_size_points(GOLDEN_RATIO * 280.0, 280.0)
        .save(&path)?;

    // plot ln_beta
    let mut curve = Curve::new();
    curve.set_line_color("#976ED7").set_line_width(2.5);
    curve.draw(xx.as_data(), lb.as_data());
    let path = format!("{}/math_beta_gamma_functions_lb.svg", OUT_DIR);
    let mut plot = Plot::new();
    plot.add(&curve)
        .grid_labels_legend(
            "$x$",
            "${\\mathrm{Real}}\\left[\\log_e(B\\left(\\frac{1}{2},x\\right)\\right]$",
        )
        .set_figure_size_points(GOLDEN_RATIO * 280.0, 280.0)
        .save(&path)?;

    // plot gamma
    let mut curve = Curve::new();
    curve.set_line_color("#579ABE").set_line_width(2.5);
    curve.draw(xx.as_data(), gg.as_data());
    let path = format!("{}/math_beta_gamma_functions_gg.svg", OUT_DIR);
    let mut plot = Plot::new();
    plot.add(&curve)
        .set_yrange(-20.0, 20.0)
        .grid_labels_legend("$x$", "$\\Gamma(x)$")
        .set_figure_size_points(GOLDEN_RATIO * 280.0, 280.0)
        .save(&path)?;

    // plot ln_gamma
    let mut curve = Curve::new();
    curve.set_line_color("#03C03C").set_line_width(2.5);
    curve.draw(xx.as_data(), lg.as_data());
    let path = format!("{}/math_beta_gamma_functions_lg.svg", OUT_DIR);
    let mut plot = Plot::new();
    plot.add(&curve)
        // .set_yrange(-2.0, 3.0)
        .grid_labels_legend("$x$", "${\\mathrm{Real}}\\left[\\log_e(\\Gamma(x)\\right]$")
        .set_figure_size_points(GOLDEN_RATIO * 280.0, 280.0)
        .save(&path)?;
    Ok(())
}