blob: 39f97f99d19d6ed00779f0b91741c64aaf345bfb (
plain)
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
|
type t = { vals : int list; opts : Options.t }
let default = { vals = []; opts = Options.default }
let plot v c = { c with vals = v }
let with_options v c = { c with opts = v }
let pp formatter { vals : int list; opts : Options.t } =
let open Utils in
let height = opts.height in
let min_val = list_min vals in
let max_val = list_max vals in
let range = max_val - min_val in
let scale v = (v - min_val) * height / range in
let new_vals = List.map scale vals in
let rec pad left right s =
if left = 0 then s :: repeat right space
else space :: pad (left - 1) right s
in
let new_vals' = zip new_vals (List.tl new_vals) in
let make_marker prefix left right s =
prefix
::
(if left = 0 then s :: repeat right space
else space :: pad (left - 1) right s)
in
let make_connector s e =
match Stdlib.compare s e with
| 0 -> make_marker axis_spc s (height - s) "-"
| x ->
let dr, rd, ru, ur = Connector.of_kind opts.connector in
let ltr_chr, rtl_chr = match x with 1 -> (dr, rd) | _ -> (ru, ur) in
let st, en = Stdlib.(min s e, max s e) in
let l = en - st |> saturating_sub 1 in
let left, right = (repeat st space, repeat (height - l - st) space) in
List.flatten
[
[ axis_spc ]; left; [ ltr_chr ]; repeat l "│"; [ rtl_chr ]; right;
]
in
let m = Format.asprintf "%a" Marker.pp opts.marker in
let rec make_graph = function
| [] -> []
| [ (s, e) ] ->
[
make_marker axis s (height - s) m;
make_connector s e;
make_marker axis e (height - e) m;
]
| (s, e) :: tl ->
make_marker axis s (height - s) m :: make_connector s e :: make_graph tl
in
make_graph new_vals' |> rot_left
|> List.map (String.concat "")
|> String.concat "\n"
|> Format.fprintf formatter "%s"
|
