v0.2.3

.github/FUNDING.yml

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

Gemfile

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

README.md

@@ -1,7 +1,9 @@
-# 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. 
 
@@ -17,7 +19,7 @@ gem install u-plot
 
 ## Usage
 
-### histogram
+**histogram**
 
 ```sh
 ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
@@ -46,10 +48,20 @@ ruby -r numo/narray -e "puts Numo::DFloat.new(1000).rand_norm.to_a" \
                                 Frequency
 ```
 
+**scatter**
+
+```sh
+wget https://raw.githubusercontent.com/uiuc-cse/data-fa14/gh-pages/data/iris.csv -qO - \
+ | cut -f1-4 -d, \
+ | uplot scatter -H -d, -t IRIS -m 10
+```
+
 ## Development
 
 ## 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).

exe/uplot

@@ -2,4 +2,4 @@
 
 require 'uplot'
 
-Uplot::Command.new(ARGV).run
+Uplot::Command.new.run

lib/uplot.rb

@@ -1,6 +1,8 @@
 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,253 +1,271 @@
 require 'optparse'
-require 'csv'
-require 'unicode_plot'
+require_relative 'preprocessing'
+require_relative 'command/params'
 
 module Uplot
+  Data = Struct.new(:headers, :series)
+
   class Command
-    def initialize(argv)
-      @params    = {}
-      @ptype     = nil
-      @headers   = nil
-      @delimiter = "\t"
-      @transpose = false
-      @output    = false
-      @count     = false
-      @fmt       = 'xyy'
-      @debug     = false
-      parse_options(argv)
+    attr_accessor :params, :command
+    attr_reader :main_parser, :sub_parsers, :raw_inputs, :data, :fmt
+
+    def initialize
+      @params = Params.new
+
+      @command    = nil
+      @headers    = nil
+      @delimiter  = "\t"
+      @transpose  = false
+      @output     = false
+      @count      = false
+      @fmt        = 'xyy'
+
+      @raw_inputs = []
+      @debug      = false
     end
 
-    def create_parser
-      OptionParser.new
-                  .on('-o', '--output', TrueClass) { |v| @output = v }
-                  .on('-d', '--delimiter VAL', String) { |v| @delimiter = v }
-                  .on('-H', '--headers', TrueClass)    { |v| @headers = v }
-                  .on('-T', '--transpose', TrueClass)  { |v| @transpose = v }
-                  .on('-t', '--title VAL', String)     { |v| @params[:title] = v }
-                  .on('-w', '--width VAL', Numeric)    { |v| @params[:width] = v }
-                  .on('-h', '--height VAL', Numeric)   { |v| @params[:height] = v }
-                  .on('-b', '--border VAL', Numeric)   { |v| @params[:border] = v }
-                  .on('-m', '--margin VAL', Numeric)   { |v| @params[:margin] = v }
-                  .on('-p', '--padding VAL', Numeric)  { |v| @params[:padding] = v }
-                  .on('-c', '--color VAL', String)     { |v| @params[:color] = v.to_sym }
-                  .on('-l', '--labels', TrueClass)     { |v| @params[:labels] = v }
-                  .on('--fmt VAL', String)             { |v| @fmt = v }
-                  .on('--debug', TrueClass)            { |v| @debug = v }
+    def create_default_parser
+      OptionParser.new do |opt|
+        opt.program_name = 'uplot'
+        opt.version = Uplot::VERSION
+        opt.on('-o', '--output', TrueClass) do |v|
+          @output = v
+        end
+           .on('-d', '--delimiter VAL', String) do |v|
+          @delimiter = v
+        end
+           .on('-H', '--headers', TrueClass) do |v|
+          @headers = v
+        end
+           .on('-T', '--transpose', TrueClass) do |v|
+          @transpose = v
+        end
+           .on('-t', '--title VAL', String) do |v|
+          params.title = v
+        end
+           .on('-w', '--width VAL', Numeric) do |v|
+          params.width = v
+        end
+           .on('-h', '--height VAL', Numeric) do |v|
+          params.height = v
+        end
+           .on('-b', '--border VAL', Numeric) do |v|
+          params.border = v
+        end
+           .on('-m', '--margin VAL', Numeric) do |v|
+          params.margin = v
+        end
+           .on('-p', '--padding VAL', Numeric) do |v|
+          params.padding = v
+        end
+           .on('-c', '--color VAL', String) do |v|
+          params.color = v =~ /\A[0-9]+\z/ ? v.to_i : v.to_sym
+        end
+           .on('-x', '--xlabel VAL', String) do |v|
+          params.xlabel = v
+        end
+           .on('-y', '--ylabel VAL', String) do |v|
+          params.ylabel = v
+        end
+           .on('-l', '--labels', TrueClass) do |v|
+          params.labels = v
+        end
+           .on('--fmt VAL', String) do |v|
+          @fmt = v
+        end
+           .on('--debug', TrueClass) do |v|
+          @debug = v
+        end
+        yield opt if block_given?
+      end
     end
 
-    def parse_options(argv)
-      main_parser          = create_parser
-      parsers              = Hash.new { |h, k| h[k] = create_parser }
-      parsers['barplot']   = parsers['bar']
-                             .on('--symbol VAL', String) { |v| @params[:symbol] = v }
-                             .on('--xscale VAL', String) { |v| @params[:xscale] = v }
-                             .on('--xlabel VAL', String) { |v| @params[:xlabel] = v }
-                             .on('--count', TrueClass)   { |v| @count = v }
-      parsers['count']     = parsers['c'] # barplot -c
-                             .on('--symbol VAL', String) { |v| @params[:symbol] = v }
-      parsers['histogram'] = parsers['hist']
-                             .on('-n', '--nbins VAL', Numeric) { |v| @params[:nbins] = v }
-                             .on('--closed VAL', String) { |v| @params[:closed] = v }
-                             .on('--symbol VAL', String) { |v| @params[:symbol] = v }
-      parsers['lineplot']  = parsers['line']
-                             .on('--canvas VAL', String) { |v| @params[:canvas] = v }
-                             .on('--xlim VAL', String)   { |v| @params[:xlim] = get_lim(v) }
-                             .on('--ylim VAL', String)   { |v| @params[:ylim] = get_lim(v) }
-      parsers['lineplots'] = parsers['lines']
-                             .on('--canvas VAL', String) { |v| @params[:canvas] = v }
-                             .on('--xlim VAL', String)   { |v| @params[:xlim] = get_lim(v) }
-                             .on('--ylim VAL', String)   { |v| @params[:ylim] = get_lim(v) }
-      parsers['scatter']   = parsers['s']
-                             .on('--canvas VAL', String) { |v| @params[:canvas] = v }
-                             .on('--xlim VAL', String)   { |v| @params[:xlim] = get_lim(v) }
-                             .on('--ylim VAL', String)   { |v| @params[:ylim] = get_lim(v) }
-      parsers['density']   = parsers['d']
-                             .on('--grid', TrueClass)    { |v| @params[:grid] = v }
-                             .on('--xlim VAL', String)   { |v| @params[:xlim] = get_lim(v) }
-                             .on('--ylim VAL', String)   { |v| @params[:ylim] = get_lim(v) }
-      parsers['boxplot']   = parsers['box']
-                             .on('--xlim VAL', String)   { |v| @params[:xlim] = get_lim(v) }
-      parsers.default      = nil
+    def create_sub_parsers
+      parsers = Hash.new { |h, k| h[k] = create_default_parser }
 
-      main_parser.banner = <<~MSG
-        Program: Uplot (Tools for plotting on the terminal)
-        Version: #{Uplot::VERSION} (using unicode_plot #{UnicodePlot::VERSION})
+      parsers[:barplot] = \
+        parsers[:bar]
+        .on('--symbol VAL', String) do |v|
+          params.symbol = v
+        end
+        .on('--xscale VAL', String) do |v|
+          params.xscale = v
+        end
+        .on('--count', TrueClass) do |v|
+          @count = v
+        end
 
-        Usage:   uplot <command> [options]
+      parsers[:count] = \
+        parsers[:c] # barplot -c
+        .on('--symbol VAL', String) do |v|
+          params.symbol = v
+        end
 
-        Command: #{parsers.keys.join(' ')}
+      parsers[:histogram] = \
+        parsers[:hist]
+        .on('-n', '--nbins VAL', Numeric) do |v|
+          params.nbins = v
+        end
+        .on('--closed VAL', String) do |v|
+          params.closed = v
+        end
+        .on('--symbol VAL', String) do |v|
+          params.symbol = v
+        end
 
-      MSG
-      main_parser.order!(argv)
-      @ptype = argv.shift
+      parsers[:lineplot] = \
+        parsers[:line]
+        .on('--canvas VAL', String) do |v|
+          params.canvas = v
+        end
+        .on('--xlim VAL', String) do |v|
+          params.xlim = get_lim(v)
+        end
+        .on('--ylim VAL', String) do |v|
+          params.ylim = get_lim(v)
+        end
 
-      unless parsers.has_key?(@ptype)
-        puts main_parser.help
-        warn "unrecognized command '#{@ptype}'"
+      parsers[:lineplots] = \
+        parsers[:lines]
+        .on('--canvas VAL', String) do |v|
+          params.canvas = v
+        end
+        .on('--xlim VAL', String) do |v|
+          params.xlim = get_lim(v)
+        end
+        .on('--ylim VAL', String) do |v|
+          params.ylim = get_lim(v)
+        end
+
+      parsers[:scatter] = \
+        parsers[:s]
+        .on('--canvas VAL', String) do |v|
+          params.canvas = v
+        end
+        .on('--xlim VAL', String) do |v|
+          params.xlim = get_lim(v)
+        end
+        .on('--ylim VAL', String) do |v|
+          params.ylim = get_lim(v)
+        end
+
+      parsers[:density] = \
+        parsers[:d]
+        .on('--grid', TrueClass) do |v|
+          params.grid = v
+        end
+        .on('--xlim VAL', String) do |v|
+          params.xlim = get_lim(v)
+        end
+        .on('--ylim VAL', String) do |v|
+          params.ylim = get_lim(v)
+        end
+
+      parsers[:boxplot] = \
+        parsers[:box]
+        .on('--xlim VAL', String) do |v|
+          params.xlim = get_lim(v)
+        end
+
+      parsers[:colors]
+        .on('-n', '--names', TrueClass) do |v|
+          @color_names = v
+        end
+
+      # Preventing the generation of new sub-commands
+      parsers.default = nil
+      parsers
+    end
+
+    def create_main_parser
+      create_default_parser do |main_parser|
+        # Usage and help messages
+        main_parser.banner = \
+          <<~MSG
+            Program: uplot (Tools for plotting on the terminal)
+            Version: #{Uplot::VERSION} (using unicode_plot #{UnicodePlot::VERSION})
+
+            Usage:   uplot <command> [options]
+
+            Command: #{sub_parsers.keys.join(' ')}
+
+            Options:
+          MSG
+      end
+    end
+
+    def parse_options(argv = ARGV)
+      @sub_parsers = create_sub_parsers
+      @main_parser = create_main_parser
+
+      begin
+        main_parser.order!(argv)
+      rescue OptionParser::ParseError => e
+        warn "uplot: #{e.message}"
         exit 1
       end
-      parser = parsers[@ptype]
-      parser.parse!(argv) unless argv.empty?
-    end
 
-    def run
-      # Sometimes the input file does not end with a newline code.
-      while input = Kernel.gets(nil)
-        input.freeze
-        data, headers = preprocess(input)
-        pp input: input, data: data, headers: headers if @debug
-        case @ptype
-        when 'bar', 'barplot'
-          barplot(data, headers)
-        when 'count', 'c'
-          @count = true
-          barplot(data, headers)
-        when 'hist', 'histogram'
-          histogram(data, headers)
-        when 'line', 'lineplot'
-          line(data, headers)
-        when 'lines', 'lineplots'
-          lines(data, headers)
-        when 'scatter', 'scatterplot'
-          scatter(data, headers)
-        when 'density'
-          density(data, headers)
-        when 'box', 'boxplot'
-          boxplot(data, headers)
-        end.render($stderr)
+      @command = argv.shift&.to_sym
 
-        print input if @output
-      end
-    end
-
-    # https://stackoverflow.com/q/26016632
-    def transpose2(arr) # Should be renamed
-      Array.new(arr.map(&:length).max) { |i| arr.map { |e| e[i] } }
-    end
-
-    def preprocess(input)
-      data = CSV.parse(input, col_sep: @delimiter)
-      data.delete([]) # Remove blank lines.
-      data.delete_if { |i| i.all? nil } # Room for improvement.
-      p parsed_csv: data if @debug
-      headers = nil
-      if @transpose
-        if @headers
-          headers = []
-          # each but destructive like map
-          data.each { |series| headers << series.shift }
+      unless sub_parsers.has_key?(command)
+        if command.nil?
+          warn main_parser.help
+        else
+          warn "uplot: unrecognized command '#{command}'"
         end
-      else
-        headers = data.shift if @headers
-        data = transpose2(data)
+        exit 1
       end
-      [data, headers]
-    end
+      parser = sub_parsers[command]
 
-    def preprocess_count(data)
-      if Enumerable.method_defined? :tally
-        data[0].tally
-      else # https://github.com/marcandre/backports tally
-        data[0].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
-
-    def barplot(data, headers)
-      data = preprocess_count(data) if @count
-      @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)
+      begin
+        parser.parse!(argv) unless argv.empty?
+      rescue OptionParser::ParseError => e
+        warn "uplot: #{e.message}"
+        exit 1
+      end
     end
 
     def get_lim(str)
       str.split(/-|:|\.\./)[0..1].map(&:to_f)
     end
 
-    def line(data, headers)
-      if data.size == 1
-        @params[:ylabel] ||= headers[0] if headers
-        y = data[0]
-        x = (1..y.size).to_a
-      else
-        @params[:xlabel] ||= headers[0] if headers
-        @params[:ylabel] ||= headers[1] if headers
-        x = data[0]
-        y = data[1]
+    def run
+      parse_options
+
+      if command == :colors
+        Plot.colors
+        exit
       end
-      x = x.map(&:to_f)
-      y = y.map(&:to_f)
-      UnicodePlot.lineplot(x, y, **@params.compact)
-    end
 
-    def get_method2(method1)
-      (method1.to_s + '!').to_sym
-    end
+      # Sometimes the input file does not end with a newline code.
+      while input = Kernel.gets(nil)
+        input.freeze
+        @raw_inputs << input
+        @data = Preprocessing.input(input, @delimiter, @headers, @transpose)
+        pp @data if @debug
+        case command
+        when :bar, :barplot
+          Plot.barplot(data, params, @count)
+        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.render($stderr)
 
-    def xyy_plot(data, headers, method1) # improve method name
-      method2 = get_method2(method1)
-      data.map! { |series| series.map(&:to_f) }
-      @params[:name] ||= headers[1] if headers
-      @params[:xlabel] ||= headers[0] if headers
-      @params[:ylim] ||= data[1..-1].flatten.minmax # need?
-      plot = UnicodePlot.public_send(method1, data[0], data[1], **@params.compact)
-      2.upto(data.size - 1) do |i|
-        UnicodePlot.public_send(method2, plot, data[0], data[i], name: headers[i])
+        print input if @output
       end
-      plot
-    end
-
-    def xyxy_plot(data, headers, method1) # improve method name
-      method2 = get_method2(method1)
-      data.map! { |series| series.map(&:to_f) }
-      data = data.each_slice(2).to_a
-      @params[:name] ||= headers[0] if headers
-      @params[:xlim] = data.map(&:first).flatten.minmax
-      @params[:ylim] = data.map(&:last).flatten.minmax
-      x1, y1 = data.shift
-      plot = UnicodePlot.public_send(method1, x1, y1, **@params.compact)
-      data.each_with_index do |(xi, yi), i|
-        UnicodePlot.public_send(method2, plot, xi, yi, name: headers[(i + 1) * 2])
-      end
-      plot
-    end
-
-    def lines(data, headers)
-      case @fmt
-      when 'xyy'
-        xyy_plot(data, headers, :lineplot)
-      when 'xyxy'
-        xyxy_plot(data, headers, :lineplot)
-      end
-    end
-
-    def scatter(data, headers)
-      case @fmt
-      when 'xyy'
-        xyy_plot(data, headers, :scatterplot)
-      when 'xyxy'
-        xyxy_plot(data, headers, :scatterplot)
-      end
-    end
-
-    def density(data, headers)
-      case @fmt
-      when 'xyy'
-        xyy_plot(data, headers, :densityplot)
-      when 'xyxy'
-        xyxy_plot(data, headers, :densityplot)
-      end
-    end
-
-    def boxplot(data, headers)
-      headers ||= (1..data.size).map(&:to_s)
-      data.map! { |series| series.map(&:to_f) }
-      UnicodePlot.boxplot(headers, data, **@params.compact)
     end
   end
 end

lib/uplot/command/params.rb

@@ -0,0 +1,30 @@
+module Uplot
+  class Command
+    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/plot.rb

@@ -0,0 +1,157 @@
+require 'unicode_plot'
+
+module Uplot
+  # plotting functions.
+  module Plot
+    module_function
+
+    def barplot(data, params, count = false)
+      headers = data.headers
+      series = data.series
+      if count
+        series = Preprocessing.count(series[0])
+        params.title = headers[0] if headers
+      end
+      params.title ||= headers[1] if headers
+      labels = series[0]
+      values = series[1].map(&:to_f)
+      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_s + '!').to_sym
+    end
+
+    def xyy_plot(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 xyxy_plot(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 lines(data, params, fmt = 'xyy')
+      check_series_size(data, fmt)
+      case fmt
+      when 'xyy'
+        xyy_plot(data, :lineplot, params)
+      when 'xyxy'
+        xyxy_plot(data, :lineplot, params)
+      end
+    end
+
+    def scatter(data, params, fmt = 'xyy')
+      check_series_size(data, fmt)
+      case fmt
+      when 'xyy'
+        xyy_plot(data, :scatterplot, params)
+      when 'xyxy'
+        xyxy_plot(data, :scatterplot, params)
+      end
+    end
+
+    def density(data, params, fmt = 'xyy')
+      check_series_size(data, fmt)
+      case fmt
+      when 'xyy'
+        xyy_plot(data, :densityplot, params)
+      when 'xyxy'
+        xyxy_plot(data, :densityplot, params)
+      end
+    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
+      UnicodePlot::StyledPrinter::TEXT_COLORS.each do |k, v|
+        print v
+        print k
+        print "\t"
+        print '  ●'
+        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 @@
+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
+      else
+        if headers
+          transpose2(arr[1..-1])
+        else
+          transpose2(arr)
+        end
+      end
+    end
+
+    def count(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,3 @@
 module Uplot
-  VERSION = '0.1.3'.freeze
+  VERSION = '0.2.3'.freeze
 end

test/test_helper.rb

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

test/uplot/command_test.rb

@@ -0,0 +1,4 @@
+require_relative '../test_helper.rb'
+
+class UplotCommandTest < Test::Unit::TestCase
+end

test/uplot/plot_test.rb

@@ -0,0 +1,4 @@
+require_relative '../test_helper.rb'
+
+class UplotPlotTest < Test::Unit::TestCase
+end

test/uplot/preprocessing_test.rb

@@ -0,0 +1,130 @@
+require_relative '../test_helper.rb'
+
+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 do
+    assert_equal([%i[a b c], [3, 2, 1]], @m.count(%i[a a a b b c]))
+    assert_equal([%i[c b a], [3, 2, 1]], @m.count(%i[a b b c c c]))
+  end
+end

uplot.gemspec

@@ -24,5 +24,6 @@ Gem::Specification.new do |spec|
   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