mirror of
https://github.com/red-data-tools/YouPlot.git
synced 2025-05-06 07:10:10 +08:00
111 lines
2.6 KiB
Ruby
111 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative 'dsv'
|
|
require_relative 'command/parser'
|
|
|
|
# FIXME
|
|
require_relative 'backends/unicode_plot_backend'
|
|
|
|
module YouPlot
|
|
Data = Struct.new(:headers, :series)
|
|
|
|
class Command
|
|
attr_accessor :command, :params, :options
|
|
attr_reader :data, :parser
|
|
|
|
def initialize(argv = ARGV)
|
|
@argv = argv
|
|
@parser = Parser.new
|
|
@command = nil
|
|
@params = nil
|
|
@options = nil
|
|
@backend = YouPlot::Backends::UnicodePlotBackend
|
|
end
|
|
|
|
def run
|
|
parser.parse_options(@argv)
|
|
@command ||= parser.command
|
|
@options ||= parser.options
|
|
@params ||= parser.params
|
|
|
|
if %i[colors color colours colour].include? @command
|
|
plot = create_plot
|
|
output_plot(plot)
|
|
else
|
|
# Sometimes the input file does not end with a newline code.
|
|
while (input = Kernel.gets(nil))
|
|
main(input)
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def main(input)
|
|
output_data(input)
|
|
|
|
@data = read_dsv(input)
|
|
|
|
pp @data if options[:debug]
|
|
|
|
plot = create_plot
|
|
output_plot(plot)
|
|
end
|
|
|
|
def read_dsv(input)
|
|
input = input.dup.force_encoding(options[:encoding]).encode('utf-8') if options[:encoding]
|
|
DSV.parse(input, options[:delimiter], options[:headers], options[:transpose])
|
|
end
|
|
|
|
def create_plot
|
|
case command
|
|
when :bar, :barplot
|
|
@backend.barplot(data, params, options[:fmt])
|
|
when :count, :c
|
|
@backend.barplot(data, params, count: true)
|
|
when :hist, :histogram
|
|
@backend.histogram(data, params)
|
|
when :line, :lineplot
|
|
@backend.line(data, params, options[:fmt])
|
|
when :lines, :lineplots
|
|
@backend.lines(data, params, options[:fmt])
|
|
when :scatter, :s
|
|
@backend.scatter(data, params, options[:fmt])
|
|
when :density, :d
|
|
@backend.density(data, params, options[:fmt])
|
|
when :box, :boxplot
|
|
@backend.boxplot(data, params)
|
|
when :colors, :color, :colours, :colour
|
|
@backend.colors(options[:color_names])
|
|
else
|
|
raise "unrecognized plot_type: #{command}"
|
|
end
|
|
end
|
|
|
|
def output_data(input)
|
|
# Pass the input to subsequent pipelines
|
|
case options[:pass]
|
|
when IO
|
|
options[:pass].print(input)
|
|
else
|
|
if options[:pass]
|
|
File.open(options[:pass], 'w') do |f|
|
|
f.print(input)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def output_plot(plot)
|
|
case options[:output]
|
|
when IO
|
|
plot.render(options[:output])
|
|
else
|
|
File.open(options[:output], 'w') do |f|
|
|
plot.render(f)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|