zhangyiss/YouPlot
1
0
Fork 0
mirror of https://github.com/red-data-tools/YouPlot.git synced 2025-05-06 07:10:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use Stract as params

Browse Source
This commit is contained in:
kojix2 2020-08-15 16:46:03 +09:00
parent 6db2372c29
commit ff1176797f
1 changed files with 76 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

124
lib/uplot/command.rb
View File

@ -4,8 +4,36 @@ require 'unicode_plot'
module Uplot module Uplot
class Command class Command
attr_accessor :params
Params = Struct.new(
:title,
:width,
:height,
:border,
:margin,
:padding,
:color,
:xlabel,
:ylabel,
:labels,
:symbol,
:xscale,
:nbins,
:closed,
:canvas,
:xlim,
:ylim,
:grid
) do
def to_hc
to_h.compact
end
end
def initialize(argv) def initialize(argv)
@params = {} @params = Params.new
@ptype = nil @ptype = nil
@headers = nil @headers = nil
@delimiter = "\t" @delimiter = "\t"
@ -25,16 +53,16 @@ module Uplot
.on('-d', '--delimiter VAL', String) { |v| @delimiter = v } .on('-d', '--delimiter VAL', String) { |v| @delimiter = v }
.on('-H', '--headers', TrueClass) { |v| @headers = v } .on('-H', '--headers', TrueClass) { |v| @headers = v }
.on('-T', '--transpose', TrueClass) { |v| @transpose = v } .on('-T', '--transpose', TrueClass) { |v| @transpose = v }
.on('-t', '--title VAL', String) { |v| @params[:title] = v } .on('-t', '--title VAL', String) { |v| params.title = v }
.on('-w', '--width VAL', Numeric) { |v| @params[:width] = v } .on('-w', '--width VAL', Numeric) { |v| params.width = v }
.on('-h', '--height VAL', Numeric) { |v| @params[:height] = v } .on('-h', '--height VAL', Numeric) { |v| params.height = v }
.on('-b', '--border VAL', Numeric) { |v| @params[:border] = v } .on('-b', '--border VAL', Numeric) { |v| params.border = v }
.on('-m', '--margin VAL', Numeric) { |v| @params[:margin] = v } .on('-m', '--margin VAL', Numeric) { |v| params.margin = v }
.on('-p', '--padding VAL', Numeric) { |v| @params[:padding] = v } .on('-p', '--padding VAL', Numeric) { |v| params.padding = v }
.on('-c', '--color VAL', String) { |v| @params[:color] = v.to_sym } .on('-c', '--color VAL', String) { |v| params.color = v.to_sym }
.on('-x', '--xlabel VAL', String) { |v| @params[:xlabel] = v } .on('-x', '--xlabel VAL', String) { |v| params.xlabel = v }
.on('-y', '--ylabel VAL', String) { |v| @params[:ylabel] = v } .on('-y', '--ylabel VAL', String) { |v| params.ylabel = v }
.on('-l', '--labels', TrueClass) { |v| @params[:labels] = v } .on('-l', '--labels', TrueClass) { |v| params.labels = v }
.on('--fmt VAL', String) { |v| @fmt = v } .on('--fmt VAL', String) { |v| @fmt = v }
.on('--debug', TrueClass) { |v| @debug = v } .on('--debug', TrueClass) { |v| @debug = v }
end end
@ -44,33 +72,33 @@ module Uplot
main_parser = create_parser main_parser = create_parser
parsers = Hash.new { |h, k| h[k] = create_parser } parsers = Hash.new { |h, k| h[k] = create_parser }
parsers['barplot'] = parsers['bar'] parsers['barplot'] = parsers['bar']
.on('--symbol VAL', String) { |v| @params[:symbol] = v } .on('--symbol VAL', String) { |v| params.symbol = v }
.on('--xscale VAL', String) { |v| @params[:xscale] = v } .on('--xscale VAL', String) { |v| params.xscale = v }
.on('--count', TrueClass) { |v| @count = v } .on('--count', TrueClass) { |v| @count = v }
parsers['count'] = parsers['c'] # barplot -c parsers['count'] = parsers['c'] # barplot -c
.on('--symbol VAL', String) { |v| @params[:symbol] = v } .on('--symbol VAL', String) { |v| params.symbol = v }
parsers['histogram'] = parsers['hist'] parsers['histogram'] = parsers['hist']
.on('-n', '--nbins VAL', Numeric) { |v| @params[:nbins] = v } .on('-n', '--nbins VAL', Numeric) { |v| params.nbins = v }
.on('--closed VAL', String) { |v| @params[:closed] = v } .on('--closed VAL', String) { |v| params.closed = v }
.on('--symbol VAL', String) { |v| @params[:symbol] = v } .on('--symbol VAL', String) { |v| params.symbol = v }
parsers['lineplot'] = parsers['line'] parsers['lineplot'] = parsers['line']
.on('--canvas VAL', String) { |v| @params[:canvas] = v } .on('--canvas VAL', String) { |v| params.canvas = v }
.on('--xlim VAL', String) { |v| @params[:xlim] = get_lim(v) } .on('--xlim VAL', String) { |v| params.xlim = get_lim(v) }
.on('--ylim VAL', String) { |v| @params[:ylim] = get_lim(v) } .on('--ylim VAL', String) { |v| params.ylim = get_lim(v) }
parsers['lineplots'] = parsers['lines'] parsers['lineplots'] = parsers['lines']
.on('--canvas VAL', String) { |v| @params[:canvas] = v } .on('--canvas VAL', String) { |v| params.canvas = v }
.on('--xlim VAL', String) { |v| @params[:xlim] = get_lim(v) } .on('--xlim VAL', String) { |v| params.xlim = get_lim(v) }
.on('--ylim VAL', String) { |v| @params[:ylim] = get_lim(v) } .on('--ylim VAL', String) { |v| params.ylim = get_lim(v) }
parsers['scatter'] = parsers['s'] parsers['scatter'] = parsers['s']
.on('--canvas VAL', String) { |v| @params[:canvas] = v } .on('--canvas VAL', String) { |v| params.canvas = v }
.on('--xlim VAL', String) { |v| @params[:xlim] = get_lim(v) } .on('--xlim VAL', String) { |v| params.xlim = get_lim(v) }
.on('--ylim VAL', String) { |v| @params[:ylim] = get_lim(v) } .on('--ylim VAL', String) { |v| params.ylim = get_lim(v) }
parsers['density'] = parsers['d'] parsers['density'] = parsers['d']
.on('--grid', TrueClass) { |v| @params[:grid] = v } .on('--grid', TrueClass) { |v| params.grid = v }
.on('--xlim VAL', String) { |v| @params[:xlim] = get_lim(v) } .on('--xlim VAL', String) { |v| params.xlim = get_lim(v) }
.on('--ylim VAL', String) { |v| @params[:ylim] = get_lim(v) } .on('--ylim VAL', String) { |v| params.ylim = get_lim(v) }
parsers['boxplot'] = parsers['box'] parsers['boxplot'] = parsers['box']
.on('--xlim VAL', String) { |v| @params[:xlim] = get_lim(v) } .on('--xlim VAL', String) { |v| params.xlim = get_lim(v) }
parsers.default = nil parsers.default = nil
main_parser.banner = <<~MSG main_parser.banner = <<~MSG
@ -199,14 +227,14 @@ module Uplot
def barplot(data, headers) def barplot(data, headers)
data = preprocess_count(data) if @count data = preprocess_count(data) if @count
@params[:title] ||= headers[1] if headers params.title ||= headers[1] if headers
UnicodePlot.barplot(data[0], data[1].map(&:to_f), **@params) UnicodePlot.barplot(data[0], data[1].map(&:to_f), **params.to_hc)
end end
def histogram(data, headers) def histogram(data, headers)
@params[:title] ||= headers[0] if headers # labels? params.title ||= headers[0] if headers # labels?
series = data[0].map(&:to_f) series = data[0].map(&:to_f)
UnicodePlot.histogram(series, **@params.compact) UnicodePlot.histogram(series, **params.to_hc)
end end
def get_lim(str) def get_lim(str)
@ -215,15 +243,15 @@ module Uplot
def line(data, headers) def line(data, headers)
if data.size == 1 if data.size == 1
@params[:ylabel] ||= headers[0] if headers params.ylabel ||= headers[0] if headers
y = data[0].map(&:to_f) y = data[0].map(&:to_f)
UnicodePlot.lineplot(y, **@params.compact) UnicodePlot.lineplot(y, **params.to_hc)
else else
@params[:xlabel] ||= headers[0] if headers params.xlabel ||= headers[0] if headers
@params[:ylabel] ||= headers[1] if headers params.ylabel ||= headers[1] if headers
x = data[0].map(&:to_f) x = data[0].map(&:to_f)
y = data[1].map(&:to_f) y = data[1].map(&:to_f)
UnicodePlot.lineplot(x, y, **@params.compact) UnicodePlot.lineplot(x, y, **params.to_hc)
end end
end end
@ -234,10 +262,10 @@ module Uplot
def xyy_plot(data, headers, method1) # improve method name def xyy_plot(data, headers, method1) # improve method name
method2 = get_method2(method1) method2 = get_method2(method1)
data.map! { |series| series.map(&:to_f) } data.map! { |series| series.map(&:to_f) }
@params[:name] ||= headers[1] if headers params.name ||= headers[1] if headers
@params[:xlabel] ||= headers[0] if headers params.xlabel ||= headers[0] if headers
@params[:ylim] ||= data[1..-1].flatten.minmax # need? params.ylim ||= data[1..-1].flatten.minmax # need?
plot = UnicodePlot.public_send(method1, data[0], data[1], **@params.compact) plot = UnicodePlot.public_send(method1, data[0], data[1], **params.to_hc)
2.upto(data.size - 1) do |i| 2.upto(data.size - 1) do |i|
UnicodePlot.public_send(method2, plot, data[0], data[i], name: headers[i]) UnicodePlot.public_send(method2, plot, data[0], data[i], name: headers[i])
end end
@ -248,11 +276,11 @@ module Uplot
method2 = get_method2(method1) method2 = get_method2(method1)
data.map! { |series| series.map(&:to_f) } data.map! { |series| series.map(&:to_f) }
data = data.each_slice(2).to_a data = data.each_slice(2).to_a
@params[:name] ||= headers[0] if headers params.name ||= headers[0] if headers
@params[:xlim] = data.map(&:first).flatten.minmax params.xlim = data.map(&:first).flatten.minmax
@params[:ylim] = data.map(&:last).flatten.minmax params.ylim = data.map(&:last).flatten.minmax
x1, y1 = data.shift x1, y1 = data.shift
plot = UnicodePlot.public_send(method1, x1, y1, **@params.compact) plot = UnicodePlot.public_send(method1, x1, y1, **params.to_hc)
data.each_with_index do |(xi, yi), i| data.each_with_index do |(xi, yi), i|
UnicodePlot.public_send(method2, plot, xi, yi, name: headers[(i + 1) * 2]) UnicodePlot.public_send(method2, plot, xi, yi, name: headers[(i + 1) * 2])
end end
@ -289,7 +317,7 @@ module Uplot
def boxplot(data, headers) def boxplot(data, headers)
headers ||= (1..data.size).map(&:to_s) headers ||= (1..data.size).map(&:to_s)
data.map! { |series| series.map(&:to_f) } data.map! { |series| series.map(&:to_f) }
UnicodePlot.boxplot(headers, data, **@params.compact) UnicodePlot.boxplot(headers, data, params.to_hc)
end end
end end
end end