tb 0.2 → 0.3

data/lib/tb/json.rb

@@ -0,0 +1,50 @@
+# Copyright (C) 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
+#     copyright notice, this list of conditions and the following
+#     disclaimer in the documentation and/or other materials provided
+#     with the distribution.
+#  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
+# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
+# GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
+# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+require 'json'
+
+class Tb
+  class JSONReader
+    include Tb::Enum
+
+    def initialize(string)
+      @ary = JSON.parse(string)
+    end
+
+    def header_and_each(header_proc)
+      header_proc.call(nil) if header_proc
+      @ary.each {|obj|
+        yield obj
+      }
+    end
+
+    def each(&block)
+      header_and_each(nil, &block)
+    end
+  end
+end

data/lib/tb/pager.rb

@@ -14,11 +14,11 @@ class Tb::Pager
   end
 
   def initialize
-    if STDOUT.tty?
+    if $stdout.tty?
       @io = nil
       @buf = ''
     else
-      @io = STDOUT
+      @io = $stdout
       @buf = nil
     end
   end
@@ -81,8 +81,8 @@ class Tb::Pager
   DEFAULT_COLUMNS = 80
 
   def winsize
-    if STDOUT.respond_to? :winsize
-      lines, columns = STDOUT.winsize
+    if $stdout.respond_to? :winsize
+      lines, columns = $stdout.winsize
       return [lines, columns] if lines != 0 && columns != 0
     end
     [DEFAULT_LINES, DEFAULT_COLUMNS]
@@ -122,10 +122,10 @@ class Tb::Pager
 
   def close
     if !@io
-      STDOUT.print @buf
+      $stdout.print @buf
     else
       # don't need to ouput @buf because @buf is nil.
-      @io.close if @io != STDOUT
+      @io.close if @io != $stdout
     end
     nil
   end

data/lib/tb/pairs.rb

@@ -0,0 +1,227 @@
+# Copyright (C) 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
+#     copyright notice, this list of conditions and the following
+#     disclaimer in the documentation and/or other materials provided
+#     with the distribution.
+#  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
+# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
+# GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
+# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+require 'weakref'
+
+class Tb::Pairs
+  def self.get_key2index(keys)
+    wm = (Thread.current[:tb_pairs_frozen_info] ||= {})
+    w = wm[keys]
+    if w
+      begin
+        return w.__getobj__
+      rescue WeakRef::RefError
+      end
+    end
+    keys = keys.dup.freeze
+    k2i = {}
+    keys.each_with_index {|k, i|
+      k2i[k] = i
+    }
+    k2i.freeze
+    info = [keys, k2i]
+    w = WeakRef.new(info)
+    wm[keys.dup] = w
+    info
+  end
+
+  include Enumerable
+
+  def initialize(pairs)
+    keys = []
+    vals = []
+    pairs.each {|k, v|
+      keys << k
+      vals << v
+    }
+    keys, k2i = Tb::Pairs.get_key2index(keys)
+    @keys = keys
+    @k2i = k2i
+    @vals = vals
+  end
+
+  def pretty_print(q) # :nodoc:
+    q.object_group(self) {
+      fs = @keys
+      unless fs.empty?
+        q.text ':'
+        q.breakable
+      end
+      q.seplist(fs, nil, :each) {|f|
+        v = self[f]
+        q.group {
+          q.pp f
+          q.text '=>'
+          q.group(1) {
+            q.breakable ''
+            q.pp v
+          }
+        }
+      }
+    }
+  end
+  alias inspect pretty_print_inspect # :nodoc:
+
+  def [](key)
+    i = @k2i.fetch(key) {
+      return nil
+    }
+    @vals[i]
+  end
+
+  def each
+    @keys.each_index {|i|
+      yield [@keys[i], @vals[i]]
+    }
+  end
+
+  def each_key(&b)
+    @keys.each(&b)
+  end
+
+  def each_value(&b)
+    @vals.each(&b)
+  end
+
+  def to_h
+    h = {}
+    @keys.each_with_index {|k, i|
+      v = @vals[i]
+      h[k] = v
+    }
+    h
+  end
+
+  def empty?
+    @keys.empty?
+  end
+
+  if defined?(::KeyError)
+    KeyError = ::KeyError
+  else
+    KeyError = IndexError
+  end
+
+  def fetch(key, *rest)
+    if 1 < rest.length
+      raise ArgumentError, "wrong number of arguments (#{1+rest.length} for 1..2)"
+    elsif block_given?
+      i = @k2i.fetch(key) {
+        return yield(key)
+      }
+      @vals[i]
+    elsif !rest.empty?
+      i = @k2i.fetch(key) {
+        return rest[0]
+      }
+      @vals[i]
+    else
+      i = @k2i.fetch(key)
+      @vals[i]
+    end
+  end
+
+  def has_key?(key)
+    @k2i.has_key?(key)
+  end
+  alias include? has_key?
+  alias key? has_key?
+  alias member? has_key?
+
+  def has_value?(value)
+    i = index(value)
+    if i
+      true
+    else
+      false
+    end
+  end
+  alias value? has_value?
+
+  def index(value)
+    @vals.each_with_index {|v, i|
+      if v == value
+        return @keys[i]
+      end
+    }
+    nil
+  end
+  alias key index
+
+  def invert
+    Tb::Pairs.new(self.map {|k, v| [v, k] })
+  end
+
+  def keys
+    @keys.dup
+  end
+
+  def length
+    @keys.length
+  end
+  alias size length
+
+  def merge(other)
+    pairs = []
+    self.each {|k, v|
+      if other.has_key? k
+        if block_given?
+          v = yield(k, v, other[k])
+        else
+          v = other[k]
+        end
+      end
+      pairs << [k, v]
+    }
+    other.each {|k, v|
+      next if self.has_key? k
+      pairs << [k, v]
+    }
+    Tb::Pairs.new(pairs)
+  end
+
+  def reject
+    pairs = []
+    self.each {|kv|
+      unless yield kv
+        pairs << kv
+      end
+    }
+    Tb::Pairs.new(pairs)
+  end
+
+  def values
+    @vals.dup
+  end
+
+  def values_at(*keys)
+    keys.map {|k|
+      self[k]
+    }
+  end
+end

data/lib/tb/pnm.rb

@@ -1,28 +1,32 @@
 # lib/tb/pnm.rb - tools for (very small) PNM images.
 #
-# Copyright (C) 2010-2011 Tanaka Akira  <akr@fsij.org>
+# Copyright (C) 2010-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:
+# 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 copyright notice,
-#     this list of conditions and the following disclaimer in the documentation
-#     and/or other materials provided with the distribution.
-#  3. The name of the author may not be used to endorse or promote products
-#     derived from this software without specific prior written permission.
+#  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
+#     copyright notice, this list of conditions and the following
+#     disclaimer in the documentation and/or other materials provided
+#     with the distribution.
+#  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 ARE DISCLAIMED. IN NO
-# EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
-# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
-# OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
-# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
-# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
-# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
-# OF SUCH DAMAGE.
+# 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
+# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
+# GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
+# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 class Tb
   # :call-seq:
@@ -175,9 +179,6 @@ class Tb
       each {|ary| result << ary }
       result
     end
-
-    def close
-    end
   end
 
   # :call-seq:

data/lib/tb/reader.rb

@@ -1,43 +1,48 @@
 # lib/tb/reader.rb - Tb::Reader class
 #
-# Copyright (C) 2011 Tanaka Akira  <akr@fsij.org>
+# 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:
+# 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 copyright notice,
-#     this list of conditions and the following disclaimer in the documentation
-#     and/or other materials provided with the distribution.
-#  3. The name of the author may not be used to endorse or promote products
-#     derived from this software without specific prior written permission.
+#  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
+#     copyright notice, this list of conditions and the following
+#     disclaimer in the documentation and/or other materials provided
+#     with the distribution.
+#  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 ARE DISCLAIMED. IN NO
-# EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
-# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
-# OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
-# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
-# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING
-# IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
-# OF SUCH DAMAGE.
+# 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
+# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+# DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
+# GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+# WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
+# OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
+# EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 class Tb::Reader
-  def initialize(rawreader, opts={})
+  include Tb::Enum
+
+  def initialize(opts={}, &rawreader_open)
     @opt_n = opts[:numeric]
-    @opt_close = opts[:close]
-    @reader = rawreader
+    @reader_open = rawreader_open
     @fieldset = nil
   end
 
-  def header
+  def internal_header(rawreader)
     return @fieldset.header if @fieldset
     if @opt_n
       @fieldset = Tb::FieldSet.new
     else
-      while ary = @reader.shift
+      while ary = rawreader.shift
         if ary.all? {|elt| elt.nil? || elt == '' }
           next
         else
@@ -51,53 +56,51 @@ class Tb::Reader
   end
 
   def index_from_field_ex(f)
-    self.header
+    raise TypeError if !@fieldset
     @fieldset.index_from_field_ex(f)
   end
 
   def index_from_field(f)
-    self.header
+    raise TypeError if !@fieldset
     @fieldset.index_from_field(f)
   end
 
   def field_from_index_ex(i)
+    raise TypeError if !@fieldset
     raise ArgumentError, "negative index: #{i}" if i < 0
-    self.header
     @fieldset.field_from_index_ex(i)
   end
 
   def field_from_index(i)
+    raise TypeError if !@fieldset
     raise ArgumentError, "negative index: #{i}" if i < 0
-    self.header
     @fieldset.field_from_index(i)
   end
 
-  def shift
-    header
-    ary = @reader.shift
+  def internal_shift(rawreader)
+    raise TypeError if !@fieldset
+    ary = rawreader.shift
     field_from_index_ex(ary.length-1) if ary && !ary.empty?
     ary
   end
 
-  def each
-    while ary = self.shift
-      yield ary
-    end
-    nil
-  end
-
-  def read_all
-    result = []
-    while ary = self.shift
-      result << ary
-    end
-    result
+  def header_and_each(header_proc)
+    body = lambda {|rawreader|
+      h = self.internal_header(rawreader)
+      header_proc.call(h) if header_proc
+      while ary = self.internal_shift(rawreader)
+        pairs = []
+        ary.each_with_index {|v, i|
+          f = field_from_index_ex(i)
+          pairs << [f, v]
+        }
+        yield Tb::Pairs.new(pairs)
+      end
+    }
+    @reader_open.call(body)
   end
 
-  def close
-    @reader.close
-    if @opt_close
-      @opt_close.close
-    end
+  def each(&block)
+    header_and_each(nil, &block)
   end
 end