adding text alignment option

naturalneuralnet · naturalneuralnet · commit 229f84e6232a · 2023-11-12T20:01:52.000Z
diff --git a/bashplotlib/scatterplot.py b/bashplotlib/scatterplot.py
@@ -28,11 +28,11 @@ def get_scale(series, is_y=False, steps=20):
     return scaled_series
 
 
-def _plot_scatter(xs, ys, size, pch, colour, title, cs):
+def _plot_scatter(xs, ys, size, pch, colour, title, cs, title_align):
     plotted = set()
 
     if title:
-        print(box_text(title, 2 * (len(get_scale(xs, False, size)) + 1)))
+        print(box_text(title, 2 * (len(get_scale(xs, False, size)) + 1), title_align))
 
     print("-" * (2 * (len(get_scale(xs, False, size)) + 2)))
     for y in get_scale(ys, True, size):
@@ -49,7 +49,7 @@ def _plot_scatter(xs, ys, size, pch, colour, title, cs):
         print(" |")
     print("-" * (2 * (len(get_scale(xs, False, size)) + 2)))
 
-def plot_scatter(f, xs, ys, size, pch, colour, title):
+def plot_scatter(f, xs, ys, size, pch, colour, title, title_align):
     """
     Form a complex number.
 
@@ -81,7 +81,7 @@ def plot_scatter(f, xs, ys, size, pch, colour, title):
         with open(ys) as fh:
             ys = [float(str(row).strip()) for row in fh]
 
-    _plot_scatter(xs, ys, size, pch, colour, title, cs)
+    _plot_scatter(xs, ys, size, pch, colour, title, cs, title_align)
     
 
 
diff --git a/bashplotlib/utils/helpers.py b/bashplotlib/utils/helpers.py
@@ -76,11 +76,21 @@ def abbreviate(labels, rfill=' '):
     return abbrev
 
 
-def box_text(text, width, offset=0):
+def box_text(text, width, title_align, offset=0 ):
     """
     Return text inside an ascii textbox
     """
+    
+    aligned_text = text.center(width)
+    if title_align == "center":
+        aligned_text = text.center(width)
+    elif title_align == "left":
+        aligned_text = text.ljust(width)
+    elif title_align == "right":
+        aligned_text = text.rjust(width)
+    
+    
     box = " " * offset + "+" + "-" * (width) + "+" + "\n"
-    box += " " * offset + "|" + text.center(width) + "|" + "\n"
+    box += " " * offset + "|" + aligned_text + "|" + "\n"
     box += " " * offset +  "+" + "-" * (width) + "+" 
     return box
