beamplot 0.1.1

Plotter
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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

use beambook::{geometry::{point,rectangle,ORIGIN},Color,Book,Page,Plot};
use beamplot::{
    plot::*,
    font::Font,
    homography::Homography,
    maps::{Map,LinearMap,ExponentialMap}
};

fn main() {
    let font = Font::new();
    let mut bk = Book::new();
    let mut pg = Page::new();

    let origin = ORIGIN;

    let dx = 100.0;
    // let pressures = Array1::linspace(200.0,700.0,50);
    // for &pressure0 in pressures.iter() {
    let pressure0 = 350.0;
    let mut pl = Plot::new();

    let size = 10.0;
    let p1 = point(0.0,200.0);
    let p3 = dx*point(1.0,0.0);
    let p2 = p1 + p3;
    let y0 = 0.0;
    let y1 = 1013.25;
    let x0 = -0.5;
    let x1 = 1.5;

    let x_map = LinearMap::new(0.0,p3.x,x0,x1);
    let y_map = ExponentialMap::new(0.0,p1.y,y0,y1);
    let tick_spacing = size/4.0;
    let ticks_y = try_ticks(10,&y_map,
			    |pos,y| if pos == Position::Last { format!("hPa {}",y) } else { format!("{}",y) },
			    tick_spacing).unwrap();
    println!("Ticks Y: {:?}",ticks_y);
    let ticks_x = try_ticks(5,&x_map,|_,x| format!("{:.1}",x),tick_spacing).unwrap();
    println!("Ticks X: {:?}",ticks_x);

    ruler(&font,size,origin,p1,true,false,&ticks_y).plot(&mut pl);
    ruler(&font,size,p1,p2,false,true,&ticks_x).plot(&mut pl);
    // ruler(&font,x0,x1,p1,p2,size,false,true,
    // 	  |_,y| format!("{:.4}",y),
    // 	  |x0,x1,dl| {
    // 	      let m = ((1.0/dl).floor() as usize).max(2);
    // 	      let mut ticks = Array2::zeros((m,2));
    // 	      ticks.slice_mut(s![..,0]).assign(&Array1::linspace(0.0,1.0,m));
    // 	      ticks.slice_mut(s![..,1]).assign(&Array1::linspace(x0,x1,m));
    // 	      ticks
    // 	  }).plot(&mut pl);
    // let dx = size*point(20.0,0.0);
    // ruler(&font,0.0,101325.0,ORIGIN+dx,p1+dx,size).plot(&mut pl);
    // .plot(&mut pl);

    if x0*x1 < 0.0 { pl.line(0x888,origin+x_map.inverse(0.0)*point(1.0,0.0),origin+p1+x_map.inverse(0.0)*point(1.0,0.0)); }
    pl.rectangle(0xfff,rectangle(origin,p2));
    let mut f1 = |p:f64| (-sq((p - 0.2*pressure0)/150.0)).exp() + 0.05*(p/25.0).cos();
    let mut f2 = |p:f64| (-sq((p - 1.0*pressure0)/150.0)).exp() + 0.10*(p/30.0).cos();
    let mut f3 = |p:f64| (-sq((p - 1.5*pressure0)/150.0)).exp() + 0.15*(p/35.0).cos();
    let mut g = |f:&mut dyn Fn(f64)->f64,color:Color| {
	curve(origin,p1,p2,
	      &y_map,
	      &x_map,
	      1.0,
	      false,
	      color,
	      f).plot(&mut pl)
    };
    g(&mut f1,0xf00);
    g(&mut f2,0x0f0);
    g(&mut f3,0x0ff);
    bottom_align(&text_lines(&font, size, 0xfff, &["T"]).rc()).transformed(&Homography::translation(origin)).plot(&mut pl);

    let origin = origin + point(size,0.0) + p3;
    if x0*x1 < 0.0 { pl.line(0x888,origin+x_map.inverse(0.0)*point(1.0,0.0),origin+p1+x_map.inverse(0.0)*point(1.0,0.0)); }
    pl.rectangle(0xfff,rectangle(origin,origin+p2));
    let mut f1 = |p:f64| (-sq((p - 0.2*pressure0)/10.0)).exp() + 0.05*(p/25.0).cos();
    let mut f2 = |p:f64| (-sq((p - 1.0*pressure0)/10.0)).exp() + 0.10*(p/30.0).cos();
    let mut f3 = |p:f64| (-sq((p - 1.5*pressure0)/10.0)).exp() + 0.15*(p/35.0).cos();
    let mut g = |f:&mut dyn Fn(f64)->f64,color:Color| {
	curve(origin,origin+p1,origin+p2,
	      &y_map,
	      &x_map,
	      1.0,
	      false,
	      color,
	      f).plot(&mut pl)
    };
    g(&mut f1,0xf00);
    g(&mut f2,0x0f0);
    g(&mut f3,0x0ff);
    ruler(&font,size,origin+p1,origin+p2,false,true,&ticks_x).plot(&mut pl);
    bottom_align(&text_lines(&font, size, 0xfff, &["H_2O"]).rc()).transformed(&Homography::translation(origin)).plot(&mut pl);

    let origin = origin + point(size,0.0) + p3;
    if x0*x1 < 0.0 { pl.line(0x888,origin+x_map.inverse(0.0)*point(1.0,0.0),origin+p1+x_map.inverse(0.0)*point(1.0,0.0)); }
    pl.rectangle(0xfff,rectangle(origin,origin + p2));
    let mut f1 = |p:f64| (-sq((p - 0.2*pressure0)/50.0)).exp() + 0.15*(p/15.0).cos();
    let mut f2 = |p:f64| (-sq((p - 1.0*pressure0)/50.0)).exp() + 0.05*(p/20.0).cos();
    let mut f3 = |p:f64| (-sq((p - 1.5*pressure0)/50.0)).exp() + 0.10*(p/25.0).cos();
    let mut f4 = |p:f64| (-sq((p - 1.2*pressure0)/30.0)).exp() + 0.12*(p/50.0).cos();
    let mut g = |f:&mut dyn Fn(f64)->f64,color:Color| {
	curve(origin,origin + p1,origin + p2,
	      &y_map,
	      &x_map,
	      1.0,
	      false,
	      color,
	      f).plot(&mut pl)
    };
    g(&mut f1,0xf00);
    g(&mut f2,0x0f0);
    g(&mut f3,0x0ff);
    g(&mut f4,0xff0);
    ruler(&font,size,origin+p1,origin+p2,false,true,&ticks_x).plot(&mut pl);
    bottom_align(&text_lines(&font, size, 0xfff, &["CH_4"]).rc()).transformed(&Homography::translation(origin)).plot(&mut pl);

    let subtitle =
	left_align(&top_align(&legend(&font,size,
				      &[(0xf00,"IASI"),
					(0x0f0,"TROPOMI"),
					(0x0ff,"IASI+L1(TROPOMI)"),
					(0xff0,"IASI+L2(TROPOMI)")],
				      0xfff).rc())).transformed(&Homography::translation(origin + p3));
    subtitle.plot(&mut pl);

    bottom_align(
	&text_lines(&font,
		    1.5*size,
		    0xfff,
		    &[
			&format!("Averaging kernels, p_0={:.3} hPa",
				 pressure0),
			"Note that the kernels have been normalized by pressure",
			"Error amplification: 3x,1.5x, Q=σAT^4"
		    ]).rc())
	.transformed(&Homography::translation(point(0.0,subtitle.area.a.y - 2.0*size))).plot(&mut pl);
    
    pg.plot(pl);
    // }
    bk.page(pg);
    bk.save_to_file("traj.mpk").unwrap();
}

fn sq(x:f64)->f64 { x*x }