0
@@ -186,19 +186,36 @@ module Stats
0
     def []=(i,v)
0
       @counts[i] = v
0
     end
0
-    
0
+
0
     def to_s(options = {})
0
-      if options[:max_width]
0
+      if options[:orientation] == :vertical
0
+        plot_vertically options
0
+      else
0
+        plot_horizontally options
0
+      end
0
+    end
0
+
0
+    private
0
+
0
+    def plot_vertically(options = {})
0
+      scale = scale_to options[:max_height]
0
+      height = (@counts.max * scale).ceil
0
+      rows = (0..height).to_a.reverse.map do |h|
0
+        @counts.map{|c| (c * scale).ceil >= h ? "*" : " " }.join('  ')
0
+      end.join("\n") + "\n" + ("_" * ((@counts.size * 3) - 2))
0
+    end
0
+
0
+    def plot_horizontally(options = {})
0
+      @counts.map {|c| "| "+"*"*(c*scale_to(options[:max_width])).ceil}.join("\n")
0
+    end
0
+
0
+    def scale_to(val)
0
+      if val
0
         max = @counts.max
0
-        if max > options[:max_width]
0
-          scale = options[:max_width] / max.to_f
0
-        else
0
-          scale = 1
0
-        end
0
+        max > val ? options[:max_width] / max.to_f : 1
0
       else
0
-        scale = 1
0
+        1
0
       end
0
-      @counts.map {|c| "| "+"*"*(c*scale).ceil}.join("\n")
0
     end
0
   end
0