v0.2.8

* How to show sub-coomand help

.github/FUNDING.yml

@@ -0,0 +1,2 @@
+ko_fi: kojix2
+patreon: kojix2

.gitignore

@@ -6,4 +6,5 @@
 /pkg/
 /spec/reports/
 /tmp/
+/vendor/
 *.lock

Gemfile

@@ -1,7 +1,6 @@
+# frozen_string_literal: true
+
 source 'https://rubygems.org'
 
 # Specify your gem's dependencies in uplot.gemspec
 gemspec
-
-gem 'minitest', '~> 5.0'
-gem 'rake', '~> 12.0'

README.md

@@ -1,55 +1,150 @@
-# Uplot
+# uplot
 
 [![Build Status](https://travis-ci.com/kojix2/uplot.svg?branch=master)](https://travis-ci.com/kojix2/uplot)
 [![Gem Version](https://badge.fury.io/rb/u-plot.svg)](https://badge.fury.io/rb/u-plot)
+[![Docs Latest](https://img.shields.io/badge/docs-latest-blue.svg)](https://rubydoc.info/gems/u-plot)
+[![The MIT License](https://img.shields.io/badge/license-MIT-blue.svg)](LICENSE.txt)
 
 Create ASCII charts on the terminal with data from standard streams in the pipeline. 
 
 :bar_chart: Powered by [UnicodePlot](https://github.com/red-data-tools/unicode_plot.rb)
 
-:construction: Under development
-
 ## Installation
 
 ```
 gem install u-plot
 ```
 
-## Usage
+Note: The name of the Gem is `u-plot` (not uplot).
 
-### histogram
+## Screenshots
+
+**histogram**
 
 ```sh
 ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
   | uplot hist --nbins 15
 ```
 
+<img src="https://i.imgur.com/wpsoGJq.png" width="75%" height="75%">
+
+```sh
+echo -e "from numpy import random;" \
+        "n = random.randn(10000);"  \
+        "print('\\\n'.join(str(i) for i in n))" \
+| python \
+| uplot hist --nbins 20
 ```
-                ┌                                        ┐ 
-   [-4.5, -4.0) ┤ 1                                        
-   [-4.0, -3.5) ┤ 0                                        
-   [-3.5, -3.0) ┤ 1                                        
-   [-3.0, -2.5) ┤▇▇ 9                                      
-   [-2.5, -2.0) ┤▇▇▇ 15                                    
-   [-2.0, -1.5) ┤▇▇▇▇▇▇▇▇▇ 50                              
-   [-1.5, -1.0) ┤▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇ 97                     
-   [-1.0, -0.5) ┤▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇ 154          
-   [-0.5,  0.0) ┤▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇ 193   
-   [ 0.0,  0.5) ┤▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇ 165        
-   [ 0.5,  1.0) ┤▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇ 152          
-   [ 1.0,  1.5) ┤▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇▇ 86                       
-   [ 1.5,  2.0) ┤▇▇▇▇▇▇▇▇▇ 51                              
-   [ 2.0,  2.5) ┤▇▇▇▇ 21                                   
-   [ 2.5,  3.0) ┤▇ 3                                       
-   [ 3.0,  3.5) ┤ 2                                        
-                └                                        ┘ 
-                                Frequency
+
+<img src="https://i.imgur.com/97R2MQx.png" width="75%" height="75%">
+
+**scatter**
+
+```sh
+curl -s https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv \
+| cut -f1-4 -d, \
+| uplot scatter -H -d, -t IRIS
+```
+
+<img src="https://i.imgur.com/STX7bFT.png" width="75%" height="75%">
+
+**line**
+
+```sh
+curl -s https://www.mhlw.go.jp/content/pcr_positive_daily.csv \
+| cut -f2 -d, \
+| uplot line -w 50 -h 15 -t 'PCR positive tests' --xlabel Date --ylabel number
+```
+
+<img src="https://i.imgur.com/PVl5dsa.png" width="75%" height="75%">
+
+**box**
+
+```sh
+curl -s https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv \
+| cut -f1-4 -d, \
+| uplot box -H -d, -t IRIS
+```
+
+<img src="https://i.imgur.com/sNI4SmN.png" width="75%" height="75%">
+
+**colors**
+
+```sh
+uplot colors
+```
+
+<img src="https://i.imgur.com/LxyHQsz.png">
+
+## Usage
+
+`uplot --help`
+
+```
+Program: uplot (Tools for plotting on the terminal)
+Version: 0.2.7 (using UnicodePlot 0.0.4)
+Source:  https://github.com/kojix2/uplot
+
+Usage:   uplot <command> [options] <in.tsv>
+
+Commands:
+    barplot    bar
+    histogram  hist
+    lineplot   line
+    lineplots  lines
+    scatter    s
+    density    d
+    boxplot    box
+    colors                   show the list of available colors
+
+    count      c             baplot based on the number of occurrences
+                             (slower than `sort | uniq -c | sort -n -k1`)
+
+Options:
+    -O, --pass [VAL]         file to output standard input data to [stdout]
+                             for inserting uplot in the middle of Unix pipes
+    -o, --output VAL         file to output results to [stderr]
+    -d, --delimiter VAL      use DELIM instead of TAB for field delimiter
+    -H, --headers            specify that the input has header row
+    -T, --transpose          transpose the axes of the input data
+    -t, --title VAL          print string on the top of plot
+    -x, --xlabel VAL         print string on the bottom of the plot
+    -y, --ylabel VAL         print string on the far left of the plot
+    -w, --width VAL          number of characters per row
+    -h, --height VAL         number of rows
+    -b, --border VAL         specify the style of the bounding box
+    -m, --margin VAL         number of spaces to the left of the plot
+    -p, --padding VAL        space of the left and right of the plot
+    -c, --color VAL          color of the drawing
+        --[no-]labels        hide the labels
+        --fmt VAL            xyxy : header is like x1, y1, x2, y2, x3, y3...
+                             xyy  : header is like x, y1, y2, y2, y3...
+```
+
+Use `--help` to print command-specific options.
+
+`uplot hist --help`
+
+```
+Usage: uplot histogram [options] <in.tsv>
+
+Options for histogram:
+        --symbol VAL         character to be used to plot the bars
+        --closed VAL
+    -n, --nbins VAL          approximate number of bins
+
+Options:
+...
 ```
 
 ## Development
 
+Let's keep it simple.
+
 ## Contributing
 
+Bug reports and pull requests are welcome on GitHub at [https://github.com/kojix2/uplot](https://github.com/kojix2/uplot).
+
 ## License
 
-The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+[MIT License](https://opensource.org/licenses/MIT).

Rakefile

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'bundler/gem_tasks'
 require 'rake/testtask'
 

exe/uplot

@@ -1,5 +1,6 @@
 #!/usr/bin/env ruby
+# frozen_string_literal: true
 
 require 'uplot'
 
-Uplot::Command.new(ARGV).run
+Uplot::Command.new.run

lib/uplot.rb

@@ -1,6 +1,10 @@
+# frozen_string_literal: true
+
 require 'unicode_plot'
 require 'uplot/version'
-require 'uplot/command.rb'
+require 'uplot/preprocessing'
+require 'uplot/plot'
+require 'uplot/command'
 
 module Uplot
 end

lib/uplot/command.rb

@@ -1,149 +1,79 @@
-require 'optparse'
-require 'csv'
+# frozen_string_literal: true
+
+require_relative 'preprocessing'
+require_relative 'command/parser'
 
 module Uplot
+  Data = Struct.new(:headers, :series)
+
   class Command
-    def initialize(argv)
-      @params = {}
-      @ptype = nil
-      @headers = nil
-      @delimiter = "\t"
-      @output = false
-      parse_options(argv)
-    end
+    attr_accessor :params
+    attr_reader :data, :fmt, :parser
 
-    def create_parser
-      OptionParser.new do |opt|
-        opt.on('-o', '--output', TrueClass)     { |v| @output = v }
-        opt.on('-d', '--delimiter VAL', String) { |v| @delimiter = v }
-        opt.on('-H', '--headers', TrueClass)    { |v| @headers = v }
-        opt.on('-t', '--title VAL', String)     { |v| @params[:title] = v }
-        opt.on('-w', '--width VAL', Numeric)    { |v| @params[:width] = v }
-        opt.on('-h', '--height VAL', Numeric)   { |v| @params[:height] = v }
-        opt.on('-b', '--border VAL', Numeric)   { |v| @params[:border] = v }
-        opt.on('-m', '--margin VAL', Numeric)   { |v| @params[:margin] = v }
-        opt.on('-p', '--padding VAL', Numeric)  { |v| @params[:padding] = v }
-        opt.on('-l', '--labels', TrueClass)     { |v| @params[:labels] = v }
-      end
-    end
-
-    def parse_options(argv)
-      main_parser            = create_parser
-      parsers                = Hash.new { |h, k| h[k] = create_parser }
-      parsers['hist']        .on('--nbins VAL', Numeric) { |v| @params[:nbins] = v }
-      parsers['histogram']   = parsers['hist']
-      parsers['lineplot']    = parsers['line']
-      parsers['lineplots']   = parsers['lines']
-      parsers['scatterplot'] = parsers['scatter']
-      parsers['barplot']     = parsers['bar']
-      parsers['boxplot']     = parsers['box']
-      parsers.default        = nil
-
-      main_parser.banner = <<~MSG
-        Usage:\tuplot <command> [options]
-        Command:\t#{parsers.keys.join(' ')}
-      MSG
-      main_parser.order!(argv)
-      @ptype = argv.shift
-
-      unless parsers.has_key?(@ptype)
-        warn "unrecognized command '#{@ptype}'"
-        exit 1
-      end
-      parser = parsers[@ptype]
-      parser.parse!(argv) unless argv.empty?
+    def initialize
+      @params = Params.new
+      @parser = Parser.new
     end
 
     def run
+      parser.parse_options
+      command   = parser.command
+      params    = parser.params
+      delimiter = parser.delimiter
+      transpose = parser.transpose
+      headers   = parser.headers
+      pass      = parser.pass
+      output    = parser.output
+      fmt       = parser.fmt
+      @debug    = parser.debug
+
+      if command == :colors
+        Plot.colors(parser.color_names)
+        exit
+      end
+
       # Sometimes the input file does not end with a newline code.
-      while input = Kernel.gets(nil)
+      while (input = Kernel.gets(nil))
         input.freeze
-        data, headers = preprocess(input)
-        case @ptype
-        when 'hist', 'histogram'
-          histogram(data, headers)
-        when 'line', 'lineplot'
-          line(data, headers)
-        when 'lines'
-          lines(data, headers)
-        when 'scatter', 'scatterplot'
-          scatter(data, headers)
-        when 'bar', 'barplot'
-          barplot(data, headers)
-        when 'box', 'boxplot'
-          boxplot(data, headers)
-        end.render($stderr)
+        @data = Preprocessing.input(input, delimiter, headers, transpose)
+        pp @data if @debug
+        plot = case command
+               when :bar, :barplot
+                 Plot.barplot(data, params)
+               when :count, :c
+                 Plot.barplot(data, params, count: true)
+               when :hist, :histogram
+                 Plot.histogram(data, params)
+               when :line, :lineplot
+                 Plot.line(data, params)
+               when :lines, :lineplots
+                 Plot.lines(data, params, fmt)
+               when :scatter, :s
+                 Plot.scatter(data, params, fmt)
+               when :density, :d
+                 Plot.density(data, params, fmt)
+               when :box, :boxplot
+                 Plot.boxplot(data, params)
+               else
+                 raise "unrecognized plot_type: #{command}"
+               end
 
-        print input if @output
+        if output.is_a?(IO)
+          plot.render(output)
+        else
+          File.open(output, 'w') do |f|
+            plot.render(f)
+          end
+        end
+
+        if pass.is_a?(IO)
+          print input
+        elsif pass
+          File.open(pass, 'w') do |f|
+            f.print(input)
+          end
+        end
       end
     end
-
-    def preprocess(input)
-      data = CSV.parse(input, col_sep: @delimiter)
-      if @headers
-        headers = data.shift
-        data = data.transpose
-        [data, headers]
-      else
-        data = data.transpose
-        [data, nil]
-      end
-    end
-
-    def barplot(data, headers)
-      @params[:title] ||= headers[1] if headers
-      UnicodePlot.barplot(data[0], data[1].map(&:to_f), **@params)
-    end
-
-    def histogram(data, headers)
-      @params[:title] ||= headers[0] if headers # labels?
-      series = data[0].map(&:to_f)
-      UnicodePlot.histogram(series, **@params.compact)
-    end
-
-    def line(data, headers)
-      if data.size == 1
-        @params[:name] ||= headers[0] if headers
-        y = data[0]
-        x = (1..y.size).to_a
-      else
-        @params[:name] ||= headers[1] if headers
-        x = data[0]
-        y = data[1]
-      end
-      x = x.map(&:to_f)
-      y = y.map(&:to_f)
-      UnicodePlot.lineplot(x, y, **@params.compact)
-    end
-
-    def lines(data, headers)
-      data.map! { |series| series.map(&:to_f) }
-      @params[:name] ||= headers[1] if headers
-      plot = UnicodePlot.lineplot(data[0], data[1], **@params.compact)
-      2.upto(data.size - 1) do |i|
-        UnicodePlot.lineplot!(plot, data[0], data[i], name: headers[i])
-      end
-      plot
-    end
-
-    def scatter(data, headers)
-      data.map! { |series| series.map(&:to_f) }
-      @params[:name] ||= headers[1] if headers
-      plot = UnicodePlot.scatterplot(data[0], data[1], **@params.compact)
-      2.upto(data.size - 1) do |i|
-        UnicodePlot.scatterplot!(plot, data[0], data[i], name: headers[i])
-      end
-      plot
-    end
-
-    def boxplot(data, headers)
-      headers ||= (1..data.size).to_a
-      data.map! { |series| series.map(&:to_f) }
-      plot = UnicodePlot.boxplot(headers[0], data[0], **@params.compact)
-      1.upto(data.size - 1) do |i|
-        UnicodePlot.boxplot!(plot, headers[i], data[i])
-      end
-      plot
-    end
   end
 end

lib/uplot/command/params.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Uplot
+  class Command
+    # UnicodePlot parameters.
+    # * Normally in a Ruby program, you might use hash for the parameter object.
+    # * Here, I use Struct for 2 safety reason.
+    # * The keys are static in Struct.
+    # * Struct does not conflict with keyword arguments. Hash dose.
+    Params = Struct.new(
+      # Sort me!
+      :title,
+      :width,
+      :height,
+      :border,
+      :margin,
+      :padding,
+      :color,
+      :xlabel,
+      :ylabel,
+      :labels,
+      :symbol,
+      :xscale,
+      :nbins,
+      :closed,
+      :canvas,
+      :xlim,
+      :ylim,
+      :grid,
+      :name
+    ) do
+      def to_hc
+        to_h.compact
+      end
+    end
+  end
+end

lib/uplot/command/parser.rb

@@ -0,0 +1,261 @@
+# frozen_string_literal: true
+
+require 'optparse'
+require_relative 'params'
+
+module Uplot
+  class Command
+    class Parser
+      attr_reader :command, :params,
+                  :delimiter, :transpose, :headers, :pass, :output, :fmt,
+                  :color_names, :debug
+
+      def initialize
+        @command     = nil
+        @params      = Params.new
+
+        @delimiter   = "\t"
+        @transpose   = false
+        @headers     = nil
+        @pass        = false
+        @output      = $stderr
+        @fmt         = 'xyy'
+        @debug       = false
+        @color_names = false
+      end
+
+      def create_default_parser
+        OptionParser.new do |opt|
+          opt.program_name  = 'uplot'
+          opt.version       = Uplot::VERSION
+          opt.summary_width = 24
+          opt.on_tail('') # Add a blank line at the end
+          opt.separator('')
+          opt.on('Common options:')
+          opt.on('-O', '--pass [VAL]', 'file to output standard input data to [stdout]',
+                 'for inserting uplot in the middle of Unix pipes') do |v|
+            @pass = v || $stdout
+          end
+          opt.on('-o', '--output VAL', 'file to output results to [stderr]') do |v|
+            @output = v
+          end
+          opt.on('-d', '--delimiter VAL', String, 'use DELIM instead of TAB for field delimiter') do |v|
+            @delimiter = v
+          end
+          opt.on('-H', '--headers', TrueClass, 'specify that the input has header row') do |v|
+            @headers = v
+          end
+          opt.on('-T', '--transpose', TrueClass, 'transpose the axes of the input data') do |v|
+            @transpose = v
+          end
+          opt.on('-t', '--title VAL', String, 'print string on the top of plot') do |v|
+            params.title = v
+          end
+          opt.on('-x', '--xlabel VAL', String, 'print string on the bottom of the plot') do |v|
+            params.xlabel = v
+          end
+          opt.on('-y', '--ylabel VAL', String, 'print string on the far left of the plot') do |v|
+            params.ylabel = v
+          end
+          opt.on('-w', '--width VAL', Integer, 'number of characters per row') do |v|
+            params.width = v
+          end
+          opt.on('-h', '--height VAL', Numeric, 'number of rows') do |v|
+            params.height = v
+          end
+          opt.on('-b', '--border VAL', String, 'specify the style of the bounding box') do |v|
+            params.border = v.to_sym
+          end
+          opt.on('-m', '--margin VAL', Numeric, 'number of spaces to the left of the plot') do |v|
+            params.margin = v
+          end
+          opt.on('-p', '--padding VAL', Numeric, 'space of the left and right of the plot') do |v|
+            params.padding = v
+          end
+          opt.on('-c', '--color VAL', String, 'color of the drawing') do |v|
+            params.color = v =~ /\A[0-9]+\z/ ? v.to_i : v.to_sym
+          end
+          opt.on('--[no-]labels', TrueClass, 'hide the labels') do |v|
+            params.labels = v
+          end
+          opt.on('--fmt VAL', String, 'xyxy : header is like x1, y1, x2, y2, x3, y3...', 'xyy  : header is like x, y1, y2, y2, y3...') do |v|
+            @fmt = v
+          end
+          # Optparse adds the help option, but it doesn't show up in usage.
+          # This is why you need the code below.
+          opt.on('--help', 'print sub-command help menu') do
+            puts opt.help
+            exit
+          end
+          opt.on('--debug', TrueClass, 'print preprocessed data') do |v|
+            @debug = v
+          end
+          yield opt if block_given?
+        end
+      end
+
+      def main_parser
+        @main_parser ||= create_default_parser do |main_parser|
+          # Here, help message is stored in the banner.
+          # Because help of main_parser may be referred by `sub_parser`.
+          
+          main_parser.banner = \
+            <<~MSG
+              
+              Program: uplot (Tools for plotting on the terminal)
+              Version: #{Uplot::VERSION} (using UnicodePlot #{UnicodePlot::VERSION})
+              Source:  https://github.com/kojix2/uplot
+
+              Usage:   uplot <command> [options] <in.tsv>
+
+              Commands:
+                  barplot    bar           draw a horizontal barplot
+                  histogram  hist          draw a horizontal histogram
+                  lineplot   line          draw a line chart
+                  lineplots  lines         draw a line chart with multiple series
+                  scatter    s             draw a scatter plot
+                  density    d             draw a density plot
+                  boxplot    box           draw a horizontal boxplot
+                  colors                   show the list of available colors
+
+                  count      c             draw a baplot based on the number of 
+                                           occurrences (slow)
+              
+              General options:
+                  --help                   print command specific help menu
+                  --version                print the version of uplot
+            MSG
+
+          # Actually, main_parser can take common optional arguments.
+          # However, these options dose not be shown in the help menu.
+          # I think the main help should be simple.
+          main_parser.on('--help', 'print sub-command help menu') do
+            puts main_parser.banner
+            puts
+            exit
+          end
+        end
+      end
+
+      def sub_parser
+        @sub_parser ||= create_default_parser do |parser|
+          parser.banner = <<~MSG
+
+            Usage: uplot #{command} [options] <in.tsv>
+
+            Options for #{command}:
+          MSG
+
+          case command
+
+          # If you type only `uplot` in the terminal.
+          when nil
+            warn main_parser.banner
+            warn "\n"
+            exit 1
+
+          when :barplot, :bar
+            parser.on_head('--symbol VAL', String, 'character to be used to plot the bars') do |v|
+              params.symbol = v
+            end
+            parser.on_head('--xscale VAL', String, 'axis scaling') do |v|
+              params.xscale = v
+            end
+
+          when :count, :c
+            parser.on_head('--symbol VAL', String, 'character to be used to plot the bars') do |v|
+              params.symbol = v
+            end
+
+          when :histogram, :hist
+            parser.on_head('-n', '--nbins VAL', Numeric, 'approximate number of bins') do |v|
+              params.nbins = v
+            end
+            parser.on_head('--closed VAL', String) do |v|
+              params.closed = v
+            end
+            parser.on_head('--symbol VAL', String, 'character to be used to plot the bars') do |v|
+              params.symbol = v
+            end
+
+          when :lineplot, :line
+            parser.on_head('--canvas VAL', String, 'type of canvas') do |v|
+              params.canvas = v
+            end
+            parser.on_head('--xlim VAL', Array, 'plotting range for the x coordinate') do |v|
+              params.xlim = v.take(2)
+            end
+            parser.on_head('--ylim VAL', Array, 'plotting range for the y coordinate') do |v|
+              params.ylim = v.take(2)
+            end
+
+          when :lineplots, :lines
+            parser.on_head('--canvas VAL', String) do |v|
+              params.canvas = v
+            end
+            parser.on_head('--xlim VAL', Array, 'plotting range for the x coordinate') do |v|
+              params.xlim = v.take(2)
+            end
+            parser.on_head('--ylim VAL', Array, 'plotting range for the y coordinate') do |v|
+              params.ylim = v.take(2)
+            end
+
+          when :scatter, :s
+            parser.on_head('--canvas VAL', String) do |v|
+              params.canvas = v
+            end
+            parser.on_head('--xlim VAL', Array, 'plotting range for the x coordinate') do |v|
+              params.xlim = v.take(2)
+            end
+            parser.on_head('--ylim VAL', Array, 'plotting range for the y coordinate') do |v|
+              params.ylim = v.take(2)
+            end
+
+          when :density, :d
+            parser.on_head('--grid', TrueClass) do |v|
+              params.grid = v
+            end
+            parser.on_head('--xlim VAL', Array, 'plotting range for the x coordinate') do |v|
+              params.xlim = v.take(2)
+            end
+            parser.on_head('--ylim VAL', Array, 'plotting range for the y coordinate') do |v|
+              params.ylim = v.take(2)
+            end
+
+          when :boxplot, :box
+            parser.on_head('--xlim VAL', Array, 'plotting range for the x coordinate') do |v|
+              params.xlim = v.take(2)
+            end
+
+          when :colors
+            parser.on_head('-n', '--names', 'show color names only', TrueClass) do |v|
+              @color_names = v
+            end
+
+          else
+            warn "uplot: unrecognized command '#{command}'"
+            exit 1
+          end
+        end
+      end
+
+      def parse_options(argv = ARGV)
+        begin
+          main_parser.order!(argv)
+        rescue OptionParser::ParseError => e
+          warn "uplot: #{e.message}"
+          exit 1
+        end
+
+        @command = argv.shift&.to_sym
+
+        begin
+          sub_parser.parse!(argv)
+        rescue OptionParser::ParseError => e
+          warn "uplot: #{e.message}"
+          exit 1
+        end
+      end
+    end
+  end
+end

lib/uplot/plot.rb

@@ -0,0 +1,165 @@
+# frozen_string_literal: true
+
+require 'unicode_plot'
+
+module Uplot
+  # plotting functions.
+  module Plot
+    module_function
+
+    def barplot(data, params, count: false)
+      headers = data.headers
+      series = data.series
+      # `uplot count`
+      if count
+        series = Preprocessing.count_values(series[0])
+        params.title = headers[0] if headers
+      end
+      if series.size == 1
+        # If there is only one series, use the line number for label.
+        params.title ||= headers[0] if headers
+        labels = Array.new(series[0].size) { |i| (i + 1).to_s }
+        values = series[0].map(&:to_f)
+      else
+        params.title ||= headers[1] if headers
+        labels = series[0]
+        values = series[1].map(&:to_f)
+      end
+      UnicodePlot.barplot(labels, values, **params.to_hc)
+    end
+
+    def histogram(data, params)
+      headers = data.headers
+      series = data.series
+      params.title ||= data.headers[0] if headers
+      values = series[0].map(&:to_f)
+      UnicodePlot.histogram(values, **params.to_hc)
+    end
+
+    def line(data, params)
+      headers = data.headers
+      series = data.series
+      if series.size == 1
+        # If there is only one series, it is assumed to be sequential data.
+        params.ylabel ||= headers[0] if headers
+        y = series[0].map(&:to_f)
+        UnicodePlot.lineplot(y, **params.to_hc)
+      else
+        # If there are 2 or more series,
+        # assume that the first 2 series are the x and y series respectively.
+        if headers
+          params.xlabel ||= headers[0]
+          params.ylabel ||= headers[1]
+        end
+        x = series[0].map(&:to_f)
+        y = series[1].map(&:to_f)
+        UnicodePlot.lineplot(x, y, **params.to_hc)
+      end
+    end
+
+    def get_method2(method1)
+      "#{method1}!".to_sym
+    end
+
+    def plot_xyy(data, method1, params)
+      headers = data.headers
+      series = data.series
+      method2 = get_method2(method1)
+      series.map! { |s| s.map(&:to_f) }
+      if headers
+        params.name   ||= headers[1]
+        params.xlabel ||= headers[0]
+      end
+      params.ylim ||= series[1..-1].flatten.minmax # why need?
+      plot = UnicodePlot.public_send(method1, series[0], series[1], **params.to_hc)
+      2.upto(series.size - 1) do |i|
+        UnicodePlot.public_send(method2, plot, series[0], series[i], name: headers&.[](i))
+      end
+      plot
+    end
+
+    def plot_xyxy(data, method1, params)
+      headers = data.headers
+      series = data.series
+      method2 = get_method2(method1)
+      series.map! { |s| s.map(&:to_f) }
+      series = series.each_slice(2).to_a
+      params.name ||= headers[0] if headers
+      params.xlim = series.map(&:first).flatten.minmax # why need?
+      params.ylim = series.map(&:last).flatten.minmax  # why need?
+      x1, y1 = series.shift
+      plot = UnicodePlot.public_send(method1, x1, y1, **params.to_hc)
+      series.each_with_index do |(xi, yi), i|
+        UnicodePlot.public_send(method2, plot, xi, yi, name: headers&.[]((i + 1) * 2))
+      end
+      plot
+    end
+
+    def plot_fmt(data, fmt, method1, params)
+      case fmt
+      when 'xyy'
+        plot_xyy(data, method1, params)
+      when 'xyxy'
+        plot_xyxy(data, method1, params)
+      else
+        raise "Unknown format: #{fmt}"
+      end
+    end
+
+    def lines(data, params, fmt = 'xyy')
+      check_series_size(data, fmt)
+      plot_fmt(data, fmt, :lineplot, params)
+    end
+
+    def scatter(data, params, fmt = 'xyy')
+      check_series_size(data, fmt)
+      plot_fmt(data, fmt, :scatterplot, params)
+    end
+
+    def density(data, params, fmt = 'xyy')
+      check_series_size(data, fmt)
+      plot_fmt(data, fmt, :densityplot, params)
+    end
+
+    def boxplot(data, params)
+      headers = data.headers
+      series = data.series
+      headers ||= (1..series.size).map(&:to_s)
+      series.map! { |s| s.map(&:to_f) }
+      UnicodePlot.boxplot(headers, series, **params.to_hc)
+    end
+
+    def colors(color_names = false)
+      UnicodePlot::StyledPrinter::TEXT_COLORS.each do |k, v|
+        print v
+        print k
+        unless color_names
+          print "\t"
+          print '  ●'
+        end
+        print "\033[0m"
+        print "\t"
+      end
+      puts
+    end
+
+    def check_series_size(data, fmt)
+      series = data.series
+      if series.size == 1
+        warn 'uplot: There is only one series of input data. Please check the delimiter.'
+        warn ''
+        warn "  Headers: \e[35m#{data.headers.inspect}\e[0m"
+        warn "  The first item is: \e[35m\"#{series[0][0]}\"\e[0m"
+        warn "  The last item is : \e[35m\"#{series[0][-1]}\"\e[0m"
+        exit 1
+      end
+      if fmt == 'xyxy' && series.size.odd?
+        warn 'uplot: In the xyxy format, the number of series must be even.'
+        warn ''
+        warn "  Number of series: \e[35m#{series.size}\e[0m"
+        warn "  Headers: \e[35m#{data.headers.inspect}\e[0m"
+        exit 1
+      end
+    end
+  end
+end

lib/uplot/preprocessing.rb

@@ -0,0 +1,86 @@
+# frozen_string_literal: true
+
+require 'csv'
+
+module Uplot
+  module Preprocessing
+    module_function
+
+    def input(input, delimiter, headers, transpose)
+      arr = parse_as_csv(input, delimiter)
+      headers = get_headers(arr, headers, transpose)
+      series = get_series(arr, headers, transpose)
+      if headers.nil?
+        Data.new(headers, series)
+      else
+        if headers.include?(nil)
+          warn "\e[35mHeaders contains nil in it.\e[0m"
+        elsif headers.include? ''
+          warn "\e[35mHeaders contains \"\" in it.\e[0m"
+        end
+        h_size = headers.size
+        s_size = series.size
+        if h_size == s_size
+          Data.new(headers, series)
+        elsif h_size > s_size
+          warn "\e[35mThe number of headers is greater than the number of series.\e[0m"
+          exit 1
+        elsif h_size < s_size
+          warn "\e[35mThe number of headers is less than the number of series.\e[0m"
+          exit 1
+        end
+      end
+    end
+
+    def parse_as_csv(input, delimiter)
+      CSV.parse(input, col_sep: delimiter)
+         .delete_if do |i|
+           i == [] or i.all? nil
+         end
+    end
+
+    # Transpose different sized ruby arrays
+    # https://stackoverflow.com/q/26016632
+    def transpose2(arr)
+      Array.new(arr.map(&:length).max) { |i| arr.map { |e| e[i] } }
+    end
+
+    def get_headers(arr, headers, transpose)
+      if headers
+        if transpose
+          arr.map(&:first)
+        else
+          arr[0]
+        end
+      end
+    end
+
+    def get_series(arr, headers, transpose)
+      if transpose
+        if headers
+          arr.map { |row| row[1..-1] }
+        else
+          arr
+        end
+      elsif headers
+        transpose2(arr[1..-1])
+      else
+        transpose2(arr)
+      end
+    end
+
+    def count_values(arr)
+      # tally was added in Ruby 2.7
+      if Enumerable.method_defined? :tally
+        arr.tally
+      else
+        # https://github.com/marcandre/backports
+        arr.each_with_object(Hash.new(0)) { |item, res| res[item] += 1 }
+           .tap { |h| h.default = nil }
+      end
+        .sort { |a, b| a[1] <=> b[1] }
+        .reverse
+        .transpose
+    end
+  end
+end

lib/uplot/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Uplot
-  VERSION = '0.1.2'.freeze
+  VERSION = '0.2.8'
 end

test/fixtures/iris.csv

@@ -0,0 +1,152 @@
+sepal_length,sepal_width,petal_length,petal_width,species
+5.1,3.5,1.4,0.2,Iris-setosa
+4.9,3.0,1.4,0.2,Iris-setosa
+4.7,3.2,1.3,0.2,Iris-setosa
+4.6,3.1,1.5,0.2,Iris-setosa
+5.0,3.6,1.4,0.2,Iris-setosa
+5.4,3.9,1.7,0.4,Iris-setosa
+4.6,3.4,1.4,0.3,Iris-setosa
+5.0,3.4,1.5,0.2,Iris-setosa
+4.4,2.9,1.4,0.2,Iris-setosa
+4.9,3.1,1.5,0.1,Iris-setosa
+5.4,3.7,1.5,0.2,Iris-setosa
+4.8,3.4,1.6,0.2,Iris-setosa
+4.8,3.0,1.4,0.1,Iris-setosa
+4.3,3.0,1.1,0.1,Iris-setosa
+5.8,4.0,1.2,0.2,Iris-setosa
+5.7,4.4,1.5,0.4,Iris-setosa
+5.4,3.9,1.3,0.4,Iris-setosa
+5.1,3.5,1.4,0.3,Iris-setosa
+5.7,3.8,1.7,0.3,Iris-setosa
+5.1,3.8,1.5,0.3,Iris-setosa
+5.4,3.4,1.7,0.2,Iris-setosa
+5.1,3.7,1.5,0.4,Iris-setosa
+4.6,3.6,1.0,0.2,Iris-setosa
+5.1,3.3,1.7,0.5,Iris-setosa
+4.8,3.4,1.9,0.2,Iris-setosa
+5.0,3.0,1.6,0.2,Iris-setosa
+5.0,3.4,1.6,0.4,Iris-setosa
+5.2,3.5,1.5,0.2,Iris-setosa
+5.2,3.4,1.4,0.2,Iris-setosa
+4.7,3.2,1.6,0.2,Iris-setosa
+4.8,3.1,1.6,0.2,Iris-setosa
+5.4,3.4,1.5,0.4,Iris-setosa
+5.2,4.1,1.5,0.1,Iris-setosa
+5.5,4.2,1.4,0.2,Iris-setosa
+4.9,3.1,1.5,0.1,Iris-setosa
+5.0,3.2,1.2,0.2,Iris-setosa
+5.5,3.5,1.3,0.2,Iris-setosa
+4.9,3.1,1.5,0.1,Iris-setosa
+4.4,3.0,1.3,0.2,Iris-setosa
+5.1,3.4,1.5,0.2,Iris-setosa
+5.0,3.5,1.3,0.3,Iris-setosa
+4.5,2.3,1.3,0.3,Iris-setosa
+4.4,3.2,1.3,0.2,Iris-setosa
+5.0,3.5,1.6,0.6,Iris-setosa
+5.1,3.8,1.9,0.4,Iris-setosa
+4.8,3.0,1.4,0.3,Iris-setosa
+5.1,3.8,1.6,0.2,Iris-setosa
+4.6,3.2,1.4,0.2,Iris-setosa
+5.3,3.7,1.5,0.2,Iris-setosa
+5.0,3.3,1.4,0.2,Iris-setosa
+7.0,3.2,4.7,1.4,Iris-versicolor
+6.4,3.2,4.5,1.5,Iris-versicolor
+6.9,3.1,4.9,1.5,Iris-versicolor
+5.5,2.3,4.0,1.3,Iris-versicolor
+6.5,2.8,4.6,1.5,Iris-versicolor
+5.7,2.8,4.5,1.3,Iris-versicolor
+6.3,3.3,4.7,1.6,Iris-versicolor
+4.9,2.4,3.3,1.0,Iris-versicolor
+6.6,2.9,4.6,1.3,Iris-versicolor
+5.2,2.7,3.9,1.4,Iris-versicolor
+5.0,2.0,3.5,1.0,Iris-versicolor
+5.9,3.0,4.2,1.5,Iris-versicolor
+6.0,2.2,4.0,1.0,Iris-versicolor
+6.1,2.9,4.7,1.4,Iris-versicolor
+5.6,2.9,3.6,1.3,Iris-versicolor
+6.7,3.1,4.4,1.4,Iris-versicolor
+5.6,3.0,4.5,1.5,Iris-versicolor
+5.8,2.7,4.1,1.0,Iris-versicolor
+6.2,2.2,4.5,1.5,Iris-versicolor
+5.6,2.5,3.9,1.1,Iris-versicolor
+5.9,3.2,4.8,1.8,Iris-versicolor
+6.1,2.8,4.0,1.3,Iris-versicolor
+6.3,2.5,4.9,1.5,Iris-versicolor
+6.1,2.8,4.7,1.2,Iris-versicolor
+6.4,2.9,4.3,1.3,Iris-versicolor
+6.6,3.0,4.4,1.4,Iris-versicolor
+6.8,2.8,4.8,1.4,Iris-versicolor
+6.7,3.0,5.0,1.7,Iris-versicolor
+6.0,2.9,4.5,1.5,Iris-versicolor
+5.7,2.6,3.5,1.0,Iris-versicolor
+5.5,2.4,3.8,1.1,Iris-versicolor
+5.5,2.4,3.7,1.0,Iris-versicolor
+5.8,2.7,3.9,1.2,Iris-versicolor
+6.0,2.7,5.1,1.6,Iris-versicolor
+5.4,3.0,4.5,1.5,Iris-versicolor
+6.0,3.4,4.5,1.6,Iris-versicolor
+6.7,3.1,4.7,1.5,Iris-versicolor
+6.3,2.3,4.4,1.3,Iris-versicolor
+5.6,3.0,4.1,1.3,Iris-versicolor
+5.5,2.5,4.0,1.3,Iris-versicolor
+5.5,2.6,4.4,1.2,Iris-versicolor
+6.1,3.0,4.6,1.4,Iris-versicolor
+5.8,2.6,4.0,1.2,Iris-versicolor
+5.0,2.3,3.3,1.0,Iris-versicolor
+5.6,2.7,4.2,1.3,Iris-versicolor
+5.7,3.0,4.2,1.2,Iris-versicolor
+5.7,2.9,4.2,1.3,Iris-versicolor
+6.2,2.9,4.3,1.3,Iris-versicolor
+5.1,2.5,3.0,1.1,Iris-versicolor
+5.7,2.8,4.1,1.3,Iris-versicolor
+6.3,3.3,6.0,2.5,Iris-virginica
+5.8,2.7,5.1,1.9,Iris-virginica
+7.1,3.0,5.9,2.1,Iris-virginica
+6.3,2.9,5.6,1.8,Iris-virginica
+6.5,3.0,5.8,2.2,Iris-virginica
+7.6,3.0,6.6,2.1,Iris-virginica
+4.9,2.5,4.5,1.7,Iris-virginica
+7.3,2.9,6.3,1.8,Iris-virginica
+6.7,2.5,5.8,1.8,Iris-virginica
+7.2,3.6,6.1,2.5,Iris-virginica
+6.5,3.2,5.1,2.0,Iris-virginica
+6.4,2.7,5.3,1.9,Iris-virginica
+6.8,3.0,5.5,2.1,Iris-virginica
+5.7,2.5,5.0,2.0,Iris-virginica
+5.8,2.8,5.1,2.4,Iris-virginica
+6.4,3.2,5.3,2.3,Iris-virginica
+6.5,3.0,5.5,1.8,Iris-virginica
+7.7,3.8,6.7,2.2,Iris-virginica
+7.7,2.6,6.9,2.3,Iris-virginica
+6.0,2.2,5.0,1.5,Iris-virginica
+6.9,3.2,5.7,2.3,Iris-virginica
+5.6,2.8,4.9,2.0,Iris-virginica
+7.7,2.8,6.7,2.0,Iris-virginica
+6.3,2.7,4.9,1.8,Iris-virginica
+6.7,3.3,5.7,2.1,Iris-virginica
+7.2,3.2,6.0,1.8,Iris-virginica
+6.2,2.8,4.8,1.8,Iris-virginica
+6.1,3.0,4.9,1.8,Iris-virginica
+6.4,2.8,5.6,2.1,Iris-virginica
+7.2,3.0,5.8,1.6,Iris-virginica
+7.4,2.8,6.1,1.9,Iris-virginica
+7.9,3.8,6.4,2.0,Iris-virginica
+6.4,2.8,5.6,2.2,Iris-virginica
+6.3,2.8,5.1,1.5,Iris-virginica
+6.1,2.6,5.6,1.4,Iris-virginica
+7.7,3.0,6.1,2.3,Iris-virginica
+6.3,3.4,5.6,2.4,Iris-virginica
+6.4,3.1,5.5,1.8,Iris-virginica
+6.0,3.0,4.8,1.8,Iris-virginica
+6.9,3.1,5.4,2.1,Iris-virginica
+6.7,3.1,5.6,2.4,Iris-virginica
+6.9,3.1,5.1,2.3,Iris-virginica
+5.8,2.7,5.1,1.9,Iris-virginica
+6.8,3.2,5.9,2.3,Iris-virginica
+6.7,3.3,5.7,2.5,Iris-virginica
+6.7,3.0,5.2,2.3,Iris-virginica
+6.3,2.5,5.0,1.9,Iris-virginica
+6.5,3.0,5.2,2.0,Iris-virginica
+6.2,3.4,5.4,2.3,Iris-virginica
+5.9,3.0,5.1,1.8,Iris-virginica
+

test/test_helper.rb

@@ -1,4 +1,8 @@
-$LOAD_PATH.unshift File.expand_path('../lib', __dir__)
+# frozen_string_literal: true
+
+require 'simplecov'
+SimpleCov.start
+
 require 'uplot'
 
 require 'test/unit'

test/uplot/command_test.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+
+class UplotCommandTest < Test::Unit::TestCase
+end

test/uplot/plot_test.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+
+class UplotPlotTest < Test::Unit::TestCase
+end

test/uplot/preprocessing_test.rb

@@ -0,0 +1,132 @@
+# frozen_string_literal: true
+
+require_relative '../test_helper'
+
+class UplotPreprocessingTest < Test::Unit::TestCase
+  def setup
+    @m = Uplot::Preprocessing
+  end
+
+  test :transpose2 do
+    n = nil
+
+    assert_equal([[1, 2, 3],
+                  [4, 5, 6],
+                  [7, 8, 9]], @m.transpose2([[1, 4, 7],
+                                             [2, 5, 8],
+                                             [3, 6, 9]]))
+    assert_equal([[1, 2, 3],
+                  [4, 5, n],
+                  [6, n, n]], @m.transpose2([[1, 4, 6],
+                                             [2, 5],
+                                             [3]]))
+    assert_equal([[1, 2, 3],
+                  [n, 4, 5],
+                  [n, n, 6]], @m.transpose2([[1],
+                                             [2, 4],
+                                             [3, 5, 6]]))
+  end
+
+  test :get_headers do
+    assert_equal([1, 4, 7], @m.get_headers([[1, 2, 3],
+                                            [4, 5, 6],
+                                            [7, 8, 9]], true, true))
+
+    assert_equal([1, 2, 3], @m.get_headers([[1, 4, 6],
+                                            [2, 5],
+                                            [3]], true, true))
+
+    assert_equal([1, 2, 3], @m.get_headers([[1],
+                                            [2, 4],
+                                            [3, 5, 6]], true, true))
+
+    assert_equal([1, 2, 3], @m.get_headers([[1, 2, 3],
+                                            [4, 5, 6],
+                                            [7, 8, 9]], true, false))
+
+    assert_equal([1, 4, 6], @m.get_headers([[1, 4, 6],
+                                            [2, 5],
+                                            [3]], true, false))
+
+    assert_equal([1], @m.get_headers([[1],
+                                      [2, 4],
+                                      [3, 5, 6]], true, false))
+
+    assert_equal(nil, @m.get_headers([[1, 2, 3],
+                                      [4, 5, 6],
+                                      [7, 8, 9]], false, true))
+
+    assert_equal(nil, @m.get_headers([[1, 2, 3],
+                                      [4, 5, 6],
+                                      [7, 8, 9]], false, false))
+  end
+
+  test :get_series do
+    n = nil
+
+    assert_equal([[2, 3], [5, 6], [8, 9]], @m.get_series([[1, 2, 3],
+                                                          [4, 5, 6],
+                                                          [7, 8, 9]], true, true))
+
+    assert_equal([[4, 6], [5], []], @m.get_series([[1, 4, 6],
+                                                   [2, 5],
+                                                   [3]], true, true))
+
+    assert_equal([[], [4], [5, 6]], @m.get_series([[1],
+                                                   [2, 4],
+                                                   [3, 5, 6]], true, true))
+
+    assert_equal([[4, 7], [5, 8], [6, 9]], @m.get_series([[1, 2, 3],
+                                                          [4, 5, 6],
+                                                          [7, 8, 9]], true, false))
+
+    assert_equal([[2, 3], [5, nil]], @m.get_series([[1, 4, 6],
+                                                    [2, 5],
+                                                    [3]], true, false))
+
+    assert_equal([[2, 3], [4, 5], [nil, 6]], @m.get_series([[1],
+                                                            [2, 4],
+                                                            [3, 5, 6]], true, false))
+
+    assert_equal([[1, 2, 3],
+                  [4, 5, 6],
+                  [7, 8, 9]], @m.get_series([[1, 2, 3],
+                                             [4, 5, 6],
+                                             [7, 8, 9]], false, true))
+
+    assert_equal([[1, 4, 6],
+                  [2, 5],
+                  [3]], @m.get_series([[1, 4, 6],
+                                       [2, 5],
+                                       [3]], false, true))
+
+    assert_equal([[1],
+                  [2, 4],
+                  [3, 5, 6]], @m.get_series([[1],
+                                             [2, 4],
+                                             [3, 5, 6]], false, true))
+
+    assert_equal([[1, 4, 7],
+                  [2, 5, 8],
+                  [3, 6, 9]], @m.get_series([[1, 2, 3],
+                                             [4, 5, 6],
+                                             [7, 8, 9]], false, false))
+
+    assert_equal([[1, 2, 3],
+                  [4, 5, n],
+                  [6, n, n]], @m.get_series([[1, 4, 6],
+                                             [2, 5],
+                                             [3]], false, false))
+
+    assert_equal([[1, 2, 3],
+                  [n, 4, 5],
+                  [n, n, 6]], @m.get_series([[1],
+                                             [2, 4],
+                                             [3, 5, 6]], false, false))
+  end
+
+  test :count_values do
+    assert_equal([%i[a b c], [3, 2, 1]], @m.count_values(%i[a a a b b c]))
+    assert_equal([%i[c b a], [3, 2, 1]], @m.count_values(%i[a b b c c c]))
+  end
+end

test/uplot_test.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'test_helper'
 
 class UplotTest < Test::Unit::TestCase

uplot.gemspec

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require_relative 'lib/uplot/version'
 
 Gem::Specification.new do |spec|
@@ -13,7 +15,7 @@ Gem::Specification.new do |spec|
   MSG
   spec.homepage      = 'https://github.com/kojix2/uplot'
   spec.license       = 'MIT'
-  spec.required_ruby_version = Gem::Requirement.new('>= 2.3.0')
+  spec.required_ruby_version = Gem::Requirement.new('>= 2.4.0')
 
   spec.files         = Dir['*.{md,txt}', '{lib,exe}/**/*']
   spec.bindir        = 'exe'
@@ -23,5 +25,7 @@ Gem::Specification.new do |spec|
   spec.add_runtime_dependency 'unicode_plot'
   spec.add_development_dependency 'bundler'
   spec.add_development_dependency 'rake'
+  spec.add_development_dependency 'rubocop'
+  spec.add_development_dependency 'simplecov'
   spec.add_development_dependency 'test-unit'
 end