-#!/usr/bin/env perl
+#!/usr/bin/env ruby
-use warnings;
-use strict;
+# Copyright (C) 2007-2013 Martin A. Hansen.
-use Maasha::Biotools;
+# This program is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public License
+# as published by the Free Software Foundation; either version 2
+# of the License, or (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+
+# http://www.gnu.org/copyleft/gpl.html
+
+# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
+
+# This program is part of the Biopieces framework (www.biopieces.org).
+
+# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> DESCRIPTION <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
+
+# Write tabular output from the stream.
+
+# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
+
+class Numeric
+ def commify
+ self.to_s.gsub(/(^[-+]?\d+?(?=(?>(?:\d{3})+)(?!\d))|\G\d{3}(?=\d))/, '\1,')
+ end
+end
+
+require 'terminal-table'
+require 'maasha/biopieces'
+require 'maasha/filesys'
+
+casts = []
+casts << {long: 'no_stream', short: 'x', type: 'flag', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'pretty', short: 'p', type: 'flag', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'comment', short: 'c', type: 'flag', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'commify', short: 'C', type: 'flag', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'delimit', short: 'd', type: 'string', mandatory: false, default: "\t", allowed: nil, disallowed: nil}
+casts << {long: 'keys', short: 'k', type: 'list', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'no_keys', short: 'K', type: 'list', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'data_out', short: 'o', type: 'file', mandatory: false, default: nil, allowed: nil, disallowed: nil}
+casts << {long: 'compress', short: 'Z', type: 'string', mandatory: false, default: nil, allowed: "gzip,bzip,bzip2", disallowed: nil}
+
+options = Biopieces.options_parse(ARGV, casts)
+
+compress = options[:compress] ? options[:compress].to_sym : nil
+
+raise "--data_out is mandatory for compressed output" if compress and not options[:data_out]
+
+rows = []
+headings = nil
+comment = true if options[:comment]
+no_keys = options[:no_keys].each_with_object({}) { |i, h| h[i.to_sym] = true } if options[:no_keys]
+
+def columns_order(record)
+ record.keys.each do |key|
+ unless key.match(/^V\d+$/)
+ return record.keys
+ end
+ end
+
+ record.keys.sort { |a, b| a.to_s[1 .. a.to_s.size].to_i <=> b.to_s[1 .. a.to_s.size].to_i }
+end
+
+tab_out = options[:data_out] ? Filesys.open(options[:data_out], 'w', compress: compress) : STDOUT
+
+Biopieces.open(options[:stream_in], options[:stream_out]) do |input, output|
+ input.each do |record|
+ unless headings
+ if options[:keys]
+ headings = options[:keys].map { |k| k.to_sym }
+ else
+ headings = columns_order(record)
+ end
+
+ headings.reject! {|r| no_keys[r] } if options[:no_keys]
+ end
+
+ row = record.values_at(*headings)
+
+ if options[:pretty]
+ rows << row
+ else
+ if comment
+ tab_out.puts "#" + headings.join(options[:delimit]) unless headings.empty?
+ comment = false
+ end
+
+ tab_out.puts row.join(options[:delimit]) unless row.empty?
+ end
+
+ output.puts record unless options[:no_stream]
+ end
+end
+
+if options[:pretty]
+ table = Terminal::Table.new
+ table.headings = headings if options[:comment]
+
+ first_row = rows.first.dup
+
+ if options[:commify]
+ rows.each do |row|
+ row.each_with_index do |cell, i|
+ begin Integer(cell)
+ row[i] = cell.to_i.commify
+ rescue
+ begin Float(cell)
+ row[i] = cell.to_f.commify
+ rescue
+ end
+ end
+ end
+ end
+ end
+
+ table.rows = rows
+
+ first_row.each_with_index do |cell, i|
+ begin Float(cell)
+ table.align_column(i, :right)
+ rescue
+ end
+ end
+
+ tab_out.puts table
+end
+
+tab_out.close
+
+
+# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
+
+
+__END__