mirror of
https://github.com/red-data-tools/YouPlot.git
synced 2025-05-06 07:10:10 +08:00
Separate the Preprocess module into new files
This commit is contained in:
parent
02857a46a3
commit
c426378bda
65
lib/preprocess.rb
Normal file
65
lib/preprocess.rb
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
module Uplot
|
||||||
|
module Preprocess
|
||||||
|
module_function
|
||||||
|
|
||||||
|
def input(input, delimiter, headers, transpose)
|
||||||
|
arr = read_csv(input, delimiter)
|
||||||
|
headers = get_headers(arr, headers, transpose)
|
||||||
|
series = get_series(arr, headers, transpose)
|
||||||
|
Data.new(headers, series)
|
||||||
|
end
|
||||||
|
|
||||||
|
def read_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
|
@ -1,5 +1,6 @@
|
|||||||
require 'unicode_plot'
|
require 'unicode_plot'
|
||||||
require 'uplot/version'
|
require 'uplot/version'
|
||||||
|
require 'uplot/preprocess'
|
||||||
require 'uplot/command.rb'
|
require 'uplot/command.rb'
|
||||||
|
|
||||||
module Uplot
|
module Uplot
|
||||||
|
@ -5,70 +5,6 @@ require 'unicode_plot'
|
|||||||
module Uplot
|
module Uplot
|
||||||
Data = Struct.new(:headers, :series)
|
Data = Struct.new(:headers, :series)
|
||||||
|
|
||||||
module Preprocess
|
|
||||||
module_function
|
|
||||||
|
|
||||||
def input(input, delimiter, headers, transpose)
|
|
||||||
arr = read_csv(input, delimiter)
|
|
||||||
headers = get_headers(arr, headers, transpose)
|
|
||||||
series = get_series(arr, headers, transpose)
|
|
||||||
Data.new(headers, series)
|
|
||||||
end
|
|
||||||
|
|
||||||
def read_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
|
|
||||||
|
|
||||||
class Command
|
class Command
|
||||||
Params = Struct.new(
|
Params = Struct.new(
|
||||||
# Sort me!
|
# Sort me!
|
||||||
|
Loading…
Reference in New Issue
Block a user