add connector customization

3 files changed, 23 insertions, 5 deletions

diff --git a/src/chart.ml b/src/chart.ml
index 194e3ba..39f97f9 100644
--- a/src/chart.ml
+++ b/src/chart.ml
@@ -29,9 +29,8 @@ let pp formatter { vals : int list; opts : Options.t } =
     match Stdlib.compare s e with
     | 0 -> make_marker axis_spc s (height - s) "-"
     | x ->
-        let ltr_chr, rtl_chr =
-          match x with 1 -> ("└", "┐") | _ -> ("┘", "┌")
-        in
+        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
diff --git a/src/connector.ml b/src/connector.ml
new file mode 100644
index 0000000..d00d036
--- /dev/null
+++ b/src/connector.ml
@@ -0,0 +1,6 @@
+type t = Squared | Rounded
+
+let of_kind (k : t) =
+  match k with
+  | Squared -> ("└", "┐", "┘", "┌")
+  | Rounded -> ("╰", "╮", "╯", "╭")
diff --git a/src/options.ml b/src/options.ml
index 341c0bb..fa49ef1 100644
--- a/src/options.ml
+++ b/src/options.ml
@@ -1,7 +1,20 @@
-type t = { height : int; ascii : bool; marker : Marker.t }
+type t = {
+  height : int;
+  ascii : bool;
+  marker : Marker.t;
+  connector : Connector.t;
+}
 
-let default = { height = 14; ascii = false; marker = Marker.Circle }
+let default =
+  {
+    height = 14;
+    ascii = false;
+    marker = Marker.Circle;
+    connector = Connector.Rounded;
+  }
 
 let with_height v c = { c with height = v }
 
 let with_marker v c = { c with marker = v }
+
+let with_connector v c = { c with connector = v }