tb 0.8 → 0.9

data/sample/poi-xls2csv.rb

@@ -3,11 +3,11 @@
 # sample/poi-xls2csv.rb - XLS to CSV convert using Apache POI with JRuby.
 #
 # Copyright (C) 2011-2012 Tanaka Akira  <akr@fsij.org>
-# 
+#
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
 # are met:
-# 
+#
 #  1. Redistributions of source code must retain the above copyright
 #     notice, this list of conditions and the following disclaimer.
 #  2. Redistributions in binary form must reproduce the above
@@ -17,7 +17,7 @@
 #  3. The name of the author may not be used to endorse or promote
 #     products derived from this software without specific prior
 #     written permission.
-# 
+#
 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
@@ -85,7 +85,7 @@ def convert_single_cell(cell)
        ExcelDateStylesHash[style.getDataFormat]
       d = cell.getDateCellValue
       val = "%d-%02d-%02d %02d:%02d:%02d" % [
-	d.getYear+1900, d.getMonth+1, d.getDate, d.getHours, d.getMinutes, d.getSeconds
+        d.getYear+1900, d.getMonth+1, d.getDate, d.getHours, d.getMinutes, d.getSeconds
       ]
       val = val + ":date" if $opt_type
     else
@@ -119,9 +119,9 @@ def convert_cell(sheet, merged, row, x, y)
     topleft_cell = sheet.getRow(y1).getCell(x1)
     if $opt_mergecells == 'topleft'
       if x == x1 && y == y1
-	val = convert_single_cell(topleft_cell)
+        val = convert_single_cell(topleft_cell)
       else
-	val = nil
+        val = nil
       end
     else
       val = convert_single_cell(topleft_cell)
@@ -146,7 +146,7 @@ def get_merged_regions(sheet)
     rid = [x1, y1, x2, y2]
     y1.upto(y2) {|y|
       x1.upto(x2) {|x|
-	merged[[x,y]] = rid
+        merged[[x,y]] = rid
       }
     }
   }
@@ -205,66 +205,66 @@ def convert_horizontal_borders(sheet, merged, upper_y, min_firstcol)
       left_x = right_x - 1
       upper_line = lower_line = left_line = right_line = false
       if upper_row
-	if !merged[[left_x, upper_y]] || !merged[[right_x, upper_y]] ||
-	   merged[[left_x, upper_y]] != merged[[right_x, upper_y]]
-	  if !upper_line && upper_cellrange.include?(left_x) &&
-	     (upperleft_cell = upper_row.getCell(left_x)) &&
-	     upperleft_cell.getCellStyle.getBorderRight != Java::HSSFCellStyle::BORDER_NONE
-	    upper_line = true
-	  end
-	  if !upper_line && upper_cellrange.include?(right_x) &&
-	     (upperright_cell = upper_row.getCell(right_x)) &&
-	     upperright_cell.getCellStyle.getBorderLeft != Java::HSSFCellStyle::BORDER_NONE
-	    upper_line = true
-	  end
-	end
-	if !merged[[left_x, upper_y]] || !merged[[left_x, lower_y]] ||
-	   merged[[left_x, upper_y]] != merged[[left_x, lower_y]]
-	  if !left_line && upper_cellrange.include?(left_x) &&
-	     (upperleft_cell = upper_row.getCell(left_x)) &&
-	     upperleft_cell.getCellStyle.getBorderBottom != Java::HSSFCellStyle::BORDER_NONE
-	    left_line = true
-	  end
-	end
-	if !merged[[right_x, upper_y]] || !merged[[right_x, lower_y]] ||
-	   merged[[right_x, upper_y]] != merged[[right_x, lower_y]]
-	  if !right_line && upper_cellrange.include?(right_x) &&
-	     (upperright_cell = upper_row.getCell(right_x)) &&
-	     upperright_cell.getCellStyle.getBorderBottom != Java::HSSFCellStyle::BORDER_NONE
-	    right_line = true
-	  end
-	end
+        if !merged[[left_x, upper_y]] || !merged[[right_x, upper_y]] ||
+           merged[[left_x, upper_y]] != merged[[right_x, upper_y]]
+          if !upper_line && upper_cellrange.include?(left_x) &&
+             (upperleft_cell = upper_row.getCell(left_x)) &&
+             upperleft_cell.getCellStyle.getBorderRight != Java::HSSFCellStyle::BORDER_NONE
+            upper_line = true
+          end
+          if !upper_line && upper_cellrange.include?(right_x) &&
+             (upperright_cell = upper_row.getCell(right_x)) &&
+             upperright_cell.getCellStyle.getBorderLeft != Java::HSSFCellStyle::BORDER_NONE
+            upper_line = true
+          end
+        end
+        if !merged[[left_x, upper_y]] || !merged[[left_x, lower_y]] ||
+           merged[[left_x, upper_y]] != merged[[left_x, lower_y]]
+          if !left_line && upper_cellrange.include?(left_x) &&
+             (upperleft_cell = upper_row.getCell(left_x)) &&
+             upperleft_cell.getCellStyle.getBorderBottom != Java::HSSFCellStyle::BORDER_NONE
+            left_line = true
+          end
+        end
+        if !merged[[right_x, upper_y]] || !merged[[right_x, lower_y]] ||
+           merged[[right_x, upper_y]] != merged[[right_x, lower_y]]
+          if !right_line && upper_cellrange.include?(right_x) &&
+             (upperright_cell = upper_row.getCell(right_x)) &&
+             upperright_cell.getCellStyle.getBorderBottom != Java::HSSFCellStyle::BORDER_NONE
+            right_line = true
+          end
+        end
       end
       if lower_row
-	if !merged[[left_x, lower_y]] || !merged[[right_x, lower_y]] ||
-	   merged[[left_x, lower_y]] != merged[[right_x, lower_y]]
-	  if !lower_line && lower_cellrange.include?(left_x) &&
-	     (lowerleft_cell = lower_row.getCell(left_x)) &&
-	     lowerleft_cell.getCellStyle.getBorderRight != Java::HSSFCellStyle::BORDER_NONE
-	    lower_line = true
-	  end
-	  if !lower_line && lower_cellrange.include?(right_x) &&
-	     (lowerright_cell = lower_row.getCell(right_x)) &&
-	     lowerright_cell.getCellStyle.getBorderLeft != Java::HSSFCellStyle::BORDER_NONE
-	    lower_line = true
-	  end
-	end
-	if !merged[[left_x, upper_y]] || !merged[[left_x, lower_y]] ||
-	   merged[[left_x, upper_y]] != merged[[left_x, lower_y]]
-	  if !left_line && lower_cellrange.include?(left_x) &&
-	     (lowerleft_cell = lower_row.getCell(left_x)) &&
-	     lowerleft_cell.getCellStyle.getBorderTop != Java::HSSFCellStyle::BORDER_NONE
-	    left_line = true
-	  end
-	end
-	if !merged[[right_x, upper_y]] || !merged[[right_x, lower_y]] ||
-	   merged[[right_x, upper_y]] != merged[[right_x, lower_y]]
-	  if !right_line && lower_cellrange.include?(right_x) &&
-	     (lowerright_cell = lower_row.getCell(right_x)) &&
-	     lowerright_cell.getCellStyle.getBorderTop != Java::HSSFCellStyle::BORDER_NONE
-	    right_line = true
-	  end
-	end
+        if !merged[[left_x, lower_y]] || !merged[[right_x, lower_y]] ||
+           merged[[left_x, lower_y]] != merged[[right_x, lower_y]]
+          if !lower_line && lower_cellrange.include?(left_x) &&
+             (lowerleft_cell = lower_row.getCell(left_x)) &&
+             lowerleft_cell.getCellStyle.getBorderRight != Java::HSSFCellStyle::BORDER_NONE
+            lower_line = true
+          end
+          if !lower_line && lower_cellrange.include?(right_x) &&
+             (lowerright_cell = lower_row.getCell(right_x)) &&
+             lowerright_cell.getCellStyle.getBorderLeft != Java::HSSFCellStyle::BORDER_NONE
+            lower_line = true
+          end
+        end
+        if !merged[[left_x, upper_y]] || !merged[[left_x, lower_y]] ||
+           merged[[left_x, upper_y]] != merged[[left_x, lower_y]]
+          if !left_line && lower_cellrange.include?(left_x) &&
+             (lowerleft_cell = lower_row.getCell(left_x)) &&
+             lowerleft_cell.getCellStyle.getBorderTop != Java::HSSFCellStyle::BORDER_NONE
+            left_line = true
+          end
+        end
+        if !merged[[right_x, upper_y]] || !merged[[right_x, lower_y]] ||
+           merged[[right_x, upper_y]] != merged[[right_x, lower_y]]
+          if !right_line && lower_cellrange.include?(right_x) &&
+             (lowerright_cell = lower_row.getCell(right_x)) &&
+             lowerright_cell.getCellStyle.getBorderTop != Java::HSSFCellStyle::BORDER_NONE
+            right_line = true
+          end
+        end
       end
       if upper_line && lower_line && !left_line && !right_line
         joint = '|'
@@ -283,19 +283,19 @@ def convert_horizontal_borders(sheet, merged, upper_y, min_firstcol)
       hborder = nil
       cell_x = min_firstcol + i / 2
       if !merged[[cell_x, upper_y]] || !merged[[cell_x, lower_y]] ||
-	 merged[[cell_x, upper_y]] != merged[[cell_x, lower_y]]
-	if !hborder && upper_row && upper_cellrange.include?(cell_x) &&
-	   (upper_cell = upper_row.getCell(cell_x)) &&
-	   upper_cell.getCellStyle.getBorderBottom != Java::HSSFCellStyle::BORDER_NONE
-	  hborder = '-'
-	  hborder = hborder + ":#{bordertype(upper_cell.getCellStyle.getBorderBottom)}" if $opt_type
-	end
-	if !hborder && lower_row && lower_cellrange.include?(cell_x) &&
-	   (lower_cell = lower_row.getCell(cell_x)) &&
-	   lower_cell.getCellStyle.getBorderTop != Java::HSSFCellStyle::BORDER_NONE
-	  hborder = '-'
-	  hborder = hborder + ":#{bordertype(lower_cell.getCellStyle.getBorderTop)}" if $opt_type
-	end
+         merged[[cell_x, upper_y]] != merged[[cell_x, lower_y]]
+        if !hborder && upper_row && upper_cellrange.include?(cell_x) &&
+           (upper_cell = upper_row.getCell(cell_x)) &&
+           upper_cell.getCellStyle.getBorderBottom != Java::HSSFCellStyle::BORDER_NONE
+          hborder = '-'
+          hborder = hborder + ":#{bordertype(upper_cell.getCellStyle.getBorderBottom)}" if $opt_type
+        end
+        if !hborder && lower_row && lower_cellrange.include?(cell_x) &&
+           (lower_cell = lower_row.getCell(cell_x)) &&
+           lower_cell.getCellStyle.getBorderTop != Java::HSSFCellStyle::BORDER_NONE
+          hborder = '-'
+          hborder = hborder + ":#{bordertype(lower_cell.getCellStyle.getBorderTop)}" if $opt_type
+        end
       end
       #hborder ||= ' '
       ary << hborder

data/sample/poi-xls2csv.sh

@@ -3,11 +3,11 @@
 # sample/poi-xls2csv.sh - script to run sample/poi-xls2csv.rb
 #
 # Copyright (C) 2011 Tanaka Akira  <akr@fsij.org>
-# 
+#
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
 # are met:
-# 
+#
 #  1. Redistributions of source code must retain the above copyright
 #     notice, this list of conditions and the following disclaimer.
 #  2. Redistributions in binary form must reproduce the above
@@ -17,7 +17,7 @@
 #  3. The name of the author may not be used to endorse or promote
 #     products derived from this software without specific prior
 #     written permission.
-# 
+#
 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

data/sample/tbplot

@@ -1,11 +1,11 @@
 #!/usr/bin/env ruby
 #
-# Copyright (C) 2011-2012 Tanaka Akira  <akr@fsij.org>
-# 
+# Copyright (C) 2011-2013 Tanaka Akira  <akr@fsij.org>
+#
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
 # are met:
-# 
+#
 #  1. Redistributions of source code must retain the above copyright
 #     notice, this list of conditions and the following disclaimer.
 #  2. Redistributions in binary form must reproduce the above
@@ -15,7 +15,7 @@
 #  3. The name of the author may not be used to endorse or promote
 #     products derived from this software without specific prior
 #     written permission.
-# 
+#
 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
@@ -31,12 +31,51 @@
 # plot a graph using gnuplot.
 
 # usage:
-#   tbplot [-x field,...] [-y field,...] [--shape field] [--color field] [--size field] [--facet-x field] [--facet-y field] filename 
-
-$:.unshift '/home/akr/ruby/tb/lib'
+#   tbplot [options] table-filename [image-filename]
 
 require 'tb/cmdtop'
 
+$x_field = nil
+$y_field = nil
+$shape_field = nil
+$color_field = nil
+$size_field = nil
+$facet_x_field = nil
+$facet_y_field = nil
+$max_size = nil
+$logx = nil
+$logy = nil
+$prompt = nil
+$gnuplot_terminal = nil
+
+def process_options(argv)
+  op = OptionParser.new
+  op.def_option('-x FIELD,...', 'x-fields') {|f| $x_field = split_field_list_argument(f).map {|ff| AxisScale.new(ff) } }
+  op.def_option('-y FIELD,...', 'y-fields') {|f| $y_field = split_field_list_argument(f).map {|ff| AxisScale.new(ff) } }
+  op.def_option('--shape=FIELD', 'shape-field') {|f| $shape_field = f }
+  op.def_option('--color=FIELD', 'color-field') {|f| $color_field = f }
+  op.def_option('--size=FIELD', 'size-field') {|f| $size_field = f }
+  op.def_option('--facet-x=FIELD', 'facet-x-field') {|f| $facet_x_field = f }
+  op.def_option('--facet-y=FIELD', 'facet-y-field') {|f| $facet_y_field = f }
+  op.def_option('--shapecolor=FIELD', 'shape-field and color-field') {|f| $shape_field = $color_field = f }
+  op.def_option('--max-size=MAX-SIZE', 'maximum point size') {|v| $max_size = v.to_f }
+  op.def_option('--loglog', 'use full log scale') {|v| $logx = $logy = true }
+  op.def_option('--logx', 'use log scale for x-axis') {|v| $logx = true }
+  op.def_option('--logy', 'use log scale for y-axis') {|v| $logy = true }
+  op.def_option('--prompt', 'enter gnuplot interactive mode') {|v| $prompt = true }
+  op.def_option('--gnuplot-terminal=TERMINAL', 'gnuplot terminal.  e.g. "png size 500,500"') {|v| $gnuplot_terminal = v }
+
+  op.parse!(argv)
+
+  if !$max_size
+    if $size_field
+      $max_size = 100.0
+    else
+      $max_size = 1.0
+    end
+  end
+end
+
 def gnuplot_escape_string(string)
   string = string.dup.force_encoding("ascii-8bit") if string.respond_to? :force_encoding
   '"' + string.gsub(/[^A-Za-z]/) {|c| sprintf("\\%03o", c.ord) } + '"'
@@ -70,14 +109,25 @@ class ValueChecker
         @numeric_min = val if val < @numeric_min
         @numeric_max = val if @numeric_max < val
       end
-    elsif /\A\s*-?(\d+(\.\d*)?|\.\d+)([eE][-+]?\d+)?\s*\z/ =~ val
+    elsif /\A\s*-?(\d+)\s*\z/ =~ val
+      v = val.to_i
       @numeric += 1
       if @numeric == 1
-        @numeric_min = @numeric_max = val
+        @numeric_min = @numeric_max = v
       else
-        @numeric_min = val if val < @numeric_min
-        @numeric_max = val if @numeric_max < val
+        @numeric_min = v if v < @numeric_min
+        @numeric_max = v if @numeric_max < v
       end
+    elsif /\A\s*-?(\d+\.\d*|\.\d+)([eE][-+]?\d+)?\s*\z/ =~ val
+      v = val.to_f
+      @numeric += 1
+      if @numeric == 1
+        @numeric_min = @numeric_max = v
+      else
+        @numeric_min = v if v < @numeric_min
+        @numeric_max = v if @numeric_max < v
+      end
+
     elsif /\A\s*-?\d+-\d\d(-\d\d(T\d\d(:\d\d(:\d\d(\.\d+)?(Z|[+-]\d\d:?\d\d)?)?)?)?)?\s*\z/ =~ val ||
        /\A\s*-?\d+\d\d\d\dT\d\d(\d\d(\d\d(\.\d+)?(Z|[+-]\d\d:?\d\d)?)?)?\s*\z/ =~ val
       @time += 1
@@ -123,10 +173,10 @@ class AxisScale
   # create an AxisScale object.
   #
   #   AxisScale.new("fieldname")
-  #   AxisScale.new("logscale(fieldname)")
+  #   AxisScale.new("log(fieldname)")
   def initialize(desc)
     @desc = desc
-    if /\Alogscale\((.*)\)\z/ =~ desc
+    if /\Alog\((.*)\)\z/ =~ desc
       @field = $1
       @logscale = true
     else
@@ -134,117 +184,144 @@ class AxisScale
       @logscale = false
     end
   end
-  attr_reader :desc, :field, :logscale
+  attr_reader :desc, :field
+  attr_accessor :logscale
 end
 
-$x_field = nil
-$y_field = nil
-$shape_field = nil
-$color_field = nil
-$size_field = nil
-$facet_x_field = nil
-$facet_y_field = nil
-$max_size = nil
-
-def main_body(argv)
-  op = OptionParser.new
-  op.def_option('-x FIELD,...', 'x-fields') {|f| $x_field = split_field_list_argument(f).map {|ff| AxisScale.new(ff) } }
-  op.def_option('-y FIELD,...', 'y-fields') {|f| $y_field = split_field_list_argument(f).map {|ff| AxisScale.new(ff) } }
-  op.def_option('--shape=FIELD', 'shape-field') {|f| $shape_field = f }
-  op.def_option('--color=FIELD', 'color-field') {|f| $color_field = f }
-  op.def_option('--size=FIELD', 'size-field') {|f| $size_field = f }
-  op.def_option('--facet-x=FIELD', 'facet-x-field') {|f| $facet_x_field = f }
-  op.def_option('--facet-y=FIELD', 'facet-y-field') {|f| $facet_y_field = f }
-  op.def_option('--shapecolor=FIELD', 'shape-field and color-field') {|f| $shape_field = $color_field = f }
-  op.def_option('--max-size=MAX-SIZE', 'maximum point size') {|v| $max_size = v.to_f }
-
-  op.parse!(argv)
-
-  if !$max_size
-    if $size_field
-      $max_size = 100.0
-    else
-      $max_size = 1.0
+def bind_fields(header0, checkers)
+  header = header0.dup
+  fields = [$shape_field, $color_field, $size_field, $facet_x_field, $facet_y_field]
+  fields.concat $x_field.map {|as| as.field } if $x_field
+  fields.concat $y_field.map {|as| as.field } if $y_field
+  fields.each {|f|
+    if f && !header.include?(f)
+      err("field not found: #{f.inspect}")
+    end
+  }
+  header -= fields.compact
+  if !$x_field
+    if header.empty?
+      err("x-field not found")
+    end
+    $x_field = [AxisScale.new(header.shift)]
+  end
+  if !$y_field
+    if header.empty?
+      err("y-field not found")
     end
+    $y_field = [AxisScale.new(header.shift)]
   end
+  if !$shape_field && !$color_field && !header.empty?
+    $shape_field = $color_field = header.shift
+  end
+  if $logx
+    $x_field.each {|axis| axis.logscale = true }
+  end
+  if $logy
+    $y_field.each {|axis| axis.logscale = true }
+  end
+  uniq_fields = [$shape_field, $color_field, $size_field, $facet_x_field, $facet_y_field].compact.uniq
+end
 
+def input_table(table_filename)
   tmps = {}
+  checkers = {}
 
-  argv = ARGV.empty? ? ['-'] : ARGV
-  r = Tb::CatReader.open(argv)
+  r = Tb.open_reader(table_filename).to_fileenumerator
 
-  checkers = {}
-  uniq_fields = nil
-  r.with_header {|header0|
-    header = header0.dup
-    fields = [$shape_field, $color_field, $size_field, $facet_x_field, $facet_y_field]
-    fields.concat $x_field.map {|as| as.field } if $x_field
-    fields.concat $y_field.map {|as| as.field } if $y_field
-    fields.each {|f|
-      if f && !header.include?(f)
-        err("field not found: #{f.inspect}")
-      end
+  r.use {
+    r.with_header {|header|
+      header.each {|f|
+        checkers[f] ||= ValueChecker.new
+      }
+    }.each {|pairs|
+      pairs.each {|f, v|
+        checkers[f].check(v)
+      }
     }
-    header -= fields.compact
-    if !$x_field
-      if header.empty?
-        err("x-field not found")
-      end
-      $x_field = [AxisScale.new(header.shift)]
-    end
-    if !$y_field
-      if header.empty?
-        err("y-field not found")
-      end
-      $y_field = [AxisScale.new(header.shift)]
-    end
-    uniq_fields = [$shape_field, $color_field, $size_field, $facet_x_field, $facet_y_field].compact.uniq
-  }.each {|pairs|
-    $x_field.each {|x_scale|
-      x_field = x_scale.field
-      $y_field.each {|y_scale|
-        y_field = y_scale.field
-        next if uniq_fields.any? {|f| pairs[f].nil? }
-        next if pairs[x_field].nil?
-        vs = {}
-        checkers[x_field] ||= ValueChecker.new
-        vs[x_field] = checkers[x_field].check(pairs[x_field])
-        checkers[y_field] ||= ValueChecker.new
-        vs[y_field] = checkers[y_field].check(pairs[y_field])
-        uniq_fields.each {|f|
-          checkers[f] ||= ValueChecker.new
-          vs[f] = checkers[f].check(pairs[f])
+
+    uniq_fields = nil
+    r.with_header {|header0|
+      uniq_fields = bind_fields(header0, checkers)
+    }.each {|pairs|
+      $x_field.each {|x_scale|
+        x_field = x_scale.field
+        $y_field.each {|y_scale|
+          y_field = y_scale.field
+          next if uniq_fields.any? {|f| pairs[f].nil? }
+          next if pairs[x_field].nil?
+          vs = {}
+          vs[x_field] = checkers[x_field].check(pairs[x_field])
+          vs[y_field] = checkers[y_field].check(pairs[y_field])
+          uniq_fields.each {|f|
+            vs[f] = checkers[f].check(pairs[f])
+          }
+          x = vs[x_field]
+          y = vs[y_field]
+
+          size = 1
+          if $shape_field
+            shape = vs[$shape_field]
+          end
+          if $color_field
+            color = vs[$color_field]
+          end
+          if $size_field
+            size = vs[$size_field]
+          end
+          if $facet_x_field
+            facet_x = vs[$facet_x_field]
+          end
+          if $facet_y_field
+            facet_y = vs[$facet_y_field]
+          end
+          key1 = [facet_x, facet_y, x_scale, y_scale]
+          key2 = [shape, color]
+          tmps[key1] ||= {}
+          tmps[key1][key2] ||= Tempfile.new('tbplot')
+          tmps[key1][key2].puts "#{x} #{y} #{size}"
         }
-        x = vs[x_field]
-        y = vs[y_field]
-
-        size = 1
-        if $shape_field
-          shape = vs[$shape_field]
-        end
-        if $color_field
-          color = vs[$color_field]
-        end
-        if $size_field
-          size = vs[$size_field]
-        end
-        if $facet_x_field
-          facet_x = vs[$facet_x_field]
-        end
-        if $facet_y_field
-          facet_y = vs[$facet_y_field]
-        end
-        key1 = [facet_x, facet_y, x_scale, y_scale]
-        key2 = [shape, color]
-        tmps[key1] ||= {}
-        tmps[key1][key2] ||= Tempfile.new('tbplot')
-        tmps[key1][key2].puts "#{x} #{y} #{size}"
       }
     }
   }
+  return checkers, tmps
+end
 
+def output_gnuplot(checkers, tmps, table_filename, image_filename)
   tmps.each {|k1, h| h.each {|k2, v| v.close } }
   gnuplot_command = ''
+
+  if image_filename
+    if $gnuplot_terminal
+      gnuplot_command << "set term #{$gnuplot_terminal}\n"
+    else
+      case image_filename
+      when /\.png\z/
+        gnuplot_command << "set term png\n"
+      when /\.eps\z/
+        gnuplot_command << "set term epscairo\n"
+      when /\.ps\z/
+        gnuplot_command << "set term postscript\n"
+      when /\.pdf\z/
+        gnuplot_command << "set term pdfcairo\n"
+      when /\.svg\z/
+        gnuplot_command << "set term svg\n"
+      when /\.jpeg\z|\.jpg\z/
+        gnuplot_command << "set term jpeg\n"
+      else
+        err("unknown image filename extension: #{image_filename.inspect}")
+      end
+    end
+    gnuplot_command << "set output #{gnuplot_escape_string image_filename}\n"
+  else
+    if $gnuplot_terminal
+      gnuplot_command << "set term #{$gnuplot_terminal}\n"
+    end
+  end
+
+  whole_title = File.basename(table_filename).sub(/\.(csv|json|ltsv|tsv)\z/, '')
+  gnuplot_command << "set title #{gnuplot_escape_string(whole_title)}\n"
+
   gnuplot_command << 'set timefmt "%Y-%m-%dT%H:%M:%SZ"' << "\n"
   use_multiplot = false
   if $facet_x_field || $facet_y_field || 1 < $x_field.length || 1 < $y_field.length
@@ -277,6 +354,16 @@ def main_body(argv)
       gnuplot_command << "set pointsize #{pointsize}\n"
     end
 
+    gnuplot_command << "set key outside\n"
+    #gnuplot_command << "unset key\n"
+    key_title = []
+    key_title << $shape_field if $shape_field
+    key_title << $color_field if $color_field
+    if !key_title.empty?
+      key_title = key_title.uniq.join(' ')
+      gnuplot_command << "set key title #{gnuplot_escape_string(key_title)}\n"
+    end
+
     if use_multiplot
       x_graphindex = 0.0
       if $facet_x_field
@@ -306,11 +393,12 @@ def main_body(argv)
       gnuplot_command << ' using 1:2:3 '
       title = []
       if shape_value
-        title << " #{$shape_field}=#{shape_value.to_s}"
+        title << " #{shape_value.to_s}"
       end
       if color_value
-        title << " #{$color_field}=#{color_value.to_s}"
+        title << " #{color_value.to_s}"
       end
+
       title = title.uniq.join(' ')
       gnuplot_command << ' title ' << gnuplot_escape_string(title)
       gnuplot_command << ' with points'
@@ -332,14 +420,28 @@ def main_body(argv)
   if use_multiplot
     gnuplot_command << "unset multiplot\n"
   end
-  gnuplot_command << "pause mouse any\n"
+  if !image_filename && !$prompt
+    gnuplot_command << "pause mouse any\n"
+  end
   tmp_gnuplot_command = Tempfile.new(['tbplot', '.gp'])
   tmp_gnuplot_command << gnuplot_command
   tmp_gnuplot_command.close
   #puts gnuplot_command; sleep 100
-  system('gnuplot', tmp_gnuplot_command.path)
+  args = ['gnuplot', tmp_gnuplot_command.path]
+  if $prompt
+    args << '-' # enter interactive mode.
+  end
+  system(*args)
 end
 
+def main_body(argv)
+  process_options(argv)
+  argv = ['-'] if argv.empty?
+  table_filename = argv.shift
+  image_filename = argv.shift
+  checkers, tmps = input_table(table_filename)
+  output_gnuplot(checkers, tmps, table_filename, image_filename)
+end
 
 def main(argv)
   main_body(argv)