mirror of
https://github.com/red-data-tools/YouPlot.git
synced 2025-05-06 15:11:12 +08:00
238 lines
8.2 KiB
Ruby
238 lines
8.2 KiB
Ruby
# 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('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
|
|
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|
|
|
# Usage and help messages
|
|
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)
|
|
MSG
|
|
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
|
|
when nil
|
|
warn main_parser.help
|
|
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
|