cmd_plot 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 1bdcbb0c83f03b105322d8dfe801fdf13b855ef6
+  data.tar.gz: 3d78c7759fa07b81841606523a16ad5922b843d7
+SHA512:
+  metadata.gz: abcc75b12bb93b8a52cfc57dda9f7afde74cf21f5eebc54ce6631b17a9cedad643985a0b9357d14704616abdf18fe08e8bd920e09417e013568bc37f966b9356
+  data.tar.gz: cfc61bb3f28c11fe686c3ff4df966dacc1215f205801beb299c7e29c1b7fea899f76078c2418f478be25d7f426a88d35c46984e47cbec163ed9fceea8008ccc1

data/lib/cmd_plot.rb

@@ -0,0 +1,9 @@
+class CmdPlot
+
+end
+
+require 'cmd_plot/basic_data'
+
+require 'cmd_plot/plot'
+require 'cmd_plot/bar'
+require 'cmd_plot/pcolor'

data/lib/cmd_plot/bar.rb

@@ -0,0 +1,30 @@
+class CmdPlot::Bar
+    def initialize(width = 140, height = 40, symbol = '°')
+        @width = width
+        @height = height
+        @symbol = symbol
+    end
+
+    def bar(height, labels)
+        # Calculate scaling of data to plot
+        max_height = height.max.to_f
+        max_label_length = labels.map { |l| l.length }.max
+        max_val_length = height.map { |l| l.to_s.length }.max
+        max_bar_length = (@width - max_label_length - max_val_length - 3)
+        # Plot each bar
+        for data in height.zip(labels)
+            label_fill_space = ' ' * (max_label_length - data[1].length)
+            value_fill_space = ' ' * (max_val_length - data[0].to_s.length)
+            bar_plot = @symbol * ((data[0] / max_height) * max_bar_length).round
+            puts "#{label_fill_space}#{data[1]}: #{value_fill_space}#{data[0]} #{bar_plot}"
+        end
+        return nil
+    end
+
+    def histogram(data, bins = 20)
+        # Turn data into bins with counts
+        counts, bin_centers = BasicData.build_hist(data, bins)
+        # Plot as bar chart
+        bar(counts, bin_centers.map { |p| p.round(1).to_s })
+    end
+end

data/lib/cmd_plot/basic_data.rb

@@ -0,0 +1,33 @@
+class BasicData
+
+    def self.linspace(min, max, n)
+        return Array.new(n) { |i| min.to_f + i.to_f * (max.to_f - min.to_f) / (n - 1).to_f }
+    end
+
+    def self.build_hist(data, bin_spec, maxcap = nil)
+        min = data.min
+        max = data.max
+        if bin_spec.kind_of?(Array)
+            limits = bin_spec
+            nr_bins = limits.length - 1
+        else
+            limits = BasicData.linspace(min, max, bin_spec + 1)
+            nr_bins = bin_spec
+        end
+        counts = Array.new(nr_bins, 0)
+        for d in data
+            bin = limits.map { |x| d < x }.index(true)
+            if bin.nil?
+                bin = counts.length
+            end
+            if bin == 0
+                bin = 1
+            end
+            counts[bin - 1] += 1
+        end
+        counts = counts.map { |v| [v, maxcap].min } unless maxcap.nil?
+        bin_centers = limits.each_cons(2).to_a.map { |a| (a[0] + a[1]) / 2.0 }
+        return counts, bin_centers
+    end
+
+end

data/lib/cmd_plot/pcolor.rb

@@ -0,0 +1,73 @@
+class CmdPlot::Pcolor
+
+    #red, green, brown, blue, magenta, cyan, gray
+    TEXT_COLORS = [31, 32, 33, 34, 35, 36, 37]
+    FILL_SYMBOLS = ['░', '▒', '▓', '█']
+    MAX_SHADES = TEXT_COLORS.length * FILL_SYMBOLS.length
+
+    def initialize(width = 140, height = 40)
+        @width = width
+        @height = height
+    end
+
+    def pcolor(data, shades = nil)
+        # Calculate scaling of data to picture
+        shades = data.flatten.uniq.length if shades.nil?
+        shades = MAX_SHADES if shades > MAX_SHADES or shades <= 0
+        x_scale = data[0].length / @width.to_f
+        y_scale = data.length / @height.to_f
+        z_max = data.map { |r| r.max }.max
+        z_min = data.map { |r| r.min }.min
+        puts "x_scale: #{x_scale}, y_scale: #{y_scale}, z_min: #{z_min}, z_max: #{z_max}"
+
+        # Calculcate scaled data
+        scaled_data = Array.new(@height) { Array.new(@width) { 0.0 } }
+        for row in 0..(scaled_data.length - 1)
+            for col in 0..(scaled_data[row].length - 1)
+                n = 0.0
+                lower_row_idx = (row * y_scale).floor
+                upper_row_idx = [lower_row_idx, ((row + 1) * y_scale).floor - 1].max
+                lower_col_idx = (col * x_scale).floor
+                upper_col_idx = [lower_col_idx, ((col + 1) * x_scale).floor - 1].max
+                for d_row in lower_row_idx..upper_row_idx
+                    for d_col in lower_col_idx..upper_col_idx
+                        scaled_data[row][col] += data[d_row][d_col]
+                        n += 1.0
+                    end
+                end
+                if(n > 0)
+                    scaled_data[row][col] /= n
+                else
+                    scaled_data[row][col] = nil
+                end
+            end
+        end
+
+        # Full picture starts out empty
+        pic = Array.new(@height) { Array.new(@width) { nil } }
+        # Run through data and fill in the plotting symbol/color
+        for row in 0..(scaled_data.length - 1)
+            for col in 0..(scaled_data[row].length - 1)
+                val = scaled_data[row][col]
+                next if val.nil?
+                fill = ((val - z_min) / (z_max - z_min).to_f * (shades - 1)).round
+                pic[row][col] = fill
+            end
+        end
+
+        # Plot the picture
+        for row in pic
+            for col in row
+                if col.nil?
+                    print ' '
+                else
+                    color = TEXT_COLORS[col % TEXT_COLORS.length]
+                    sym = FILL_SYMBOLS[col / TEXT_COLORS.length]
+                    print "\e[#{color}m#{sym}\e[0m"
+                end
+            end
+            print "\n"
+        end
+        return nil
+    end
+end

data/lib/cmd_plot/plot.rb

@@ -0,0 +1,116 @@
+class CmdPlot::Plot
+    def initialize(width = 140, height = 40, symbol = '°')
+        @width = width
+        @height = height
+        @symbol = symbol
+    end
+
+    def plot(x, y = nil)
+        # If only one parameter is given treat it as the y-axis data and fill up x automatically
+        if y.nil?
+            y = x
+            x = BasicData::linspace(0.0, y.length, y.length)
+        end
+        # Scale x axis to the desired console resolution
+        x_max = x.compact.max
+        x_min = x.compact.min
+        x = x.map { |v| v.nil? ? nil : ((v - x_min) / (x_max - x_min).to_f * (@width - 1)).round }
+        # Scale y axis to the desired console resolution
+        y_max = y.compact.max
+        y_min = y.compact.min
+        y = y.map { |v| v.nil? ? nil : ((v - y_min) / (y_max - y_min).to_f * (@height - 1)).round }
+
+        # Full plot-picture starts out empty
+        picture = Array.new(@height) { Array.new(@width) {' '} }
+        # Run through all the points of the curve and fill in the plotting symbol
+        for point in x.zip(y)
+            next if point[1].nil? || point[0].nil?
+            picture[@height - 1 - point[1]][point[0]] = @symbol
+        end
+
+        # X-axis
+        x_ticks = BasicData::linspace(x_min, x_max, [2, @width / 15].max).map { |v|
+            tick_to_display(v)
+        }
+        # Y-axis
+        y_ticks = BasicData::linspace(y_max, y_min, [2, @height / 10].max).map { |v|
+            tick_to_display(v)
+        }
+        y_tick_width = y_ticks.map { |e| e.length }.max
+        y_tick_spacing = (@height - y_ticks.length) / (y_ticks.length - 1)
+        missing = @height - (y_tick_spacing * (y_ticks.length - 1) + y_ticks.length)
+        missing_dist = missing == 0 ? @height : y_ticks.length / missing.to_f
+
+        # Print the y-axis and plot picture
+        row = 0
+        next_tick = 0
+        tick_offset = 0
+        for line in picture
+            if row == (tick_offset + next_tick * (y_tick_spacing + 1))
+                print ' ' * (y_tick_width - y_ticks[next_tick].length)
+                print y_ticks[next_tick] + ' ├ '
+                next_tick += 1
+                tick_offset += 1 if ((tick_offset + 1) * missing_dist) < (next_tick + 2)
+            else
+                print ' ' * y_tick_width + ' │ '
+            end
+            puts line.join
+            row += 1
+        end
+        # Print x-axis
+        x_tick_spacing = (@width - x_ticks.length) / (x_ticks.length - 1)
+        missing = @width - (x_tick_spacing * (x_ticks.length - 1) + x_ticks.length)
+        missing_dist = missing == 0 ? @width : x_ticks.length / missing.to_f
+        print ' ' * y_tick_width + ' └─'
+        next_tick = 0
+        added_space = 0
+        for tick in x_ticks
+            if tick == x_ticks[-1]
+                puts '┴'
+            else
+                print '┴'
+                print '─' * x_tick_spacing
+                next_tick += 1
+                if ((added_space + 1) * missing_dist) < (next_tick + 2)
+                    added_space += 1
+                    print '─'
+                end
+            end
+        end
+        # Print x labels
+        total_label_length = x_ticks.map(&:length).inject(:+)
+        x_tick_label_spacing = (@width - total_label_length) / (x_ticks.length - 1)
+        missing = @width - (x_tick_label_spacing * (x_ticks.length - 1) + total_label_length)
+        missing_dist = missing == 0 ? @width : x_ticks.length / missing.to_f
+        print ' ' * y_tick_width + '   '
+        next_tick = 0
+        added_space = 0
+        for tick in x_ticks
+            if tick == x_ticks[-1]
+                puts tick
+            else
+                print tick
+                print ' ' * x_tick_label_spacing
+                next_tick += 1
+                if ((added_space + 1) * missing_dist) < (next_tick + 2)
+                    added_space += 1
+                    print ' '
+                end
+            end
+        end
+        return nil
+    end
+
+    def tick_to_display(v)
+        s = v.to_s
+        dot = s.index('.')
+        dot = s.length - 1 if dot.nil?
+        if dot == 1 && s[0] == '0'
+            for dot in 1..(s.length - 1)
+                next if s[dot + 1] == '0' || s[dot + 1] == '.'
+                break
+            end
+        end
+        s[0..(dot+2)]
+    end
+end

metadata

@@ -0,0 +1,50 @@
+--- !ruby/object:Gem::Specification
+name: cmd_plot
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Andreas Buschermoehle
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2018-06-11 00:00:00.000000000 Z
+dependencies: []
+description: This is an interface to quickly visualize data on the command line. It
+  supports various kinds of plots (curves, bar diagram, histogram, 2D color, etc)
+  that are presented as text output on the console.
+email: andreas@buschermoehle.org
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/cmd_plot.rb
+- lib/cmd_plot/bar.rb
+- lib/cmd_plot/basic_data.rb
+- lib/cmd_plot/pcolor.rb
+- lib/cmd_plot/plot.rb
+homepage: http://rubygems.org/gems/cmd_plot
+licenses:
+- GPL-2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.12
+signing_key: 
+specification_version: 4
+summary: Command line plotting for quick visualization of data.
+test_files: []