added title to hist

glamp · glamp · commit ce863ce0809e · 2013-01-27T01:36:50.000-05:00
diff --git a/bin/hist.py b/bin/hist.py
@@ -22,7 +22,7 @@ def read_numbers(numbers):
         for n in open(numbers):
             yield float(n.strip())
 
-def plot_hist(f, height=20, bincount=None, pch="o", colour="white"):
+def plot_hist(f, height=20, bincount=None, pch="o", colour="white", title=""):
     "plot a histogram given a file of numbers"
     #first apss
     if pch is None:
@@ -58,7 +58,8 @@ def plot_hist(f, height=20, bincount=None, pch="o", colour="white"):
 
     nlen = max(len(str(min_y)), len(str(max_y))) + 1
 
-
+    print title.center(len(hist) + nlen + 1)
+    print
     for y in ys:
         ylab = str(y)
         ylab += " "*(nlen - len(ylab)) + "|"
@@ -94,6 +95,8 @@ def plot_hist(f, height=20, bincount=None, pch="o", colour="white"):
     parser = optparse.OptionParser()
     parser.add_option('-f', '--file', help='a file containing a column of numbers',
                       default=None, dest='f')
+    parser.add_option('-t', '--title', help='title for the chart',
+                      default="", dest='t')
     parser.add_option('-b', '--bins', help='number of bins in the histogram',
                       default=None, dest='b')
     parser.add_option('-s', '--height', help='height of the histogram (in lines)',
@@ -107,5 +110,5 @@ def plot_hist(f, height=20, bincount=None, pch="o", colour="white"):
     if opts.f is None:
         opts.f = args[0]
     
-    plot_hist(opts.f, opts.h, opts.b, opts.p, opts.colour)
+    plot_hist(opts.f, opts.h, opts.b, opts.p, opts.colour, opts.t)
 
diff --git a/bin/utils/helpers.py b/bin/utils/helpers.py
@@ -17,13 +17,19 @@ def get_colour(colour):
 
 def printcolor(txt, sameline=False, color=get_colour("white")):
     if sameline:
-        print color + txt + bcolours["ENDC"],
+        if color=='\033[97m':
+            print txt,
+        else:
+            print color + txt + bcolours["ENDC"],
     else:
-        print color + txt + bcolours["ENDC"]
+        if color=='\033[97m':
+            print txt
+        else:
+            print color + txt + bcolours["ENDC"]
 
 def drange(start, stop, step=1.0):
     "generate between 2 numbers w/ optional step"
     r = start
     while r < stop:
         yield r
-        r += step
+        r += step
diff --git a/bin/utils/helpers.pyc b/bin/utils/helpers.pyc
