swak 0.1.3 → 0.1.4

data/lib/swak.rb

@@ -1,157 +1,5 @@
+require File.join(File.dirname(__FILE__), 'swak', 'toplevel')
 require File.join(File.dirname(__FILE__), 'swak', 'logger')
 require File.join(File.dirname(__FILE__), 'swak', 'interval')
 require File.join(File.dirname(__FILE__), 'swak', 'crypt')
-
-class String
-  AnsiMap = {"k" => 0,
-             "r" => 1,
-             "g" => 2,
-             "y" => 3,
-             "b" => 4,
-             "m" => 5,
-             "c" => 6,
-             "w" => 7,
-             "d" => 9}
-
-  def color(color, fg=true)
-    raise "Illegal color" unless AnsiMap.include?(color)
-    color_int = AnsiMap[color]
-    return "[#{fg ? 3 : 4}#{color_int}m#{self}[#{fg ? 3 : 4}9m"
-  end
-
-  # Wraps long string by lines, using whitespace boundaries
-  def wrap(max_chars_per_line=80)
-    return gsub(/\n/," ").scan(/\S.{0,#{max_chars_per_line-2}}\S(?=\s|$)|\S+/).join("\n")
-  end
-
-end
-
-#######################################################
-#######################################################
-
-class Object
-  def is_i?
-    true if Integer(self) rescue false
-  end
-
-  def is_f?
-    true if Float(self) rescue false
-  end
-
-  def to_i_strict
-    if is_i?
-      return to_i
-    else
-      raise "String '#{self}' cannot be converted to an int"
-    end
-      
-  end
-
-  def to_f_strict
-    if is_f?
-      return to_f
-    else
-      raise "String '#{self}' cannot be converted to a float"
-    end
-  end
-end
-
-class NilClass
-  undef_method(:is_i?)
-  undef_method(:is_f?)
-  undef_method(:to_i_strict)
-  undef_method(:to_f_strict)
-  undef_method(:to_f)
-  undef_method(:to_i)
-end
-
-class Float
-  def is_i?
-    return self == self.to_i
-  end
-end
-
-
-#######################################################
-#######################################################
-
-
-class Array
-
-	def hist
-		h = {}
-		for item in self
-			h[item] ||= 0
-			h[item] += 1
-		end
-		return h
-	end
-
-	def shuffle!
-		sort_by {rand}
-		return self
-	end
-	
-	def shuffle
-		return dup().sort_by {rand}
-	end
-
-	def sum
-		total = 0
-		for item in self
-			total += item
-		end
-		return total
-	end
-
-	def avg
-		return sum.to_f / self.length
-	end
-
-	# Sample variance
-	def variance
-		average = avg()
-		variance = 0
-		for item in self
-			diff = item - average
-			variance += diff * diff
-		end
-		return variance / (self.length - 1)
-	end
-
-	def stddev
-		return Math.sqrt(self.variance())
-	end
-
-	def mean
-		return self.avg
-	end
-
-  def argmax
-    max_i = 0
-    max_val = self[max_i]
-
-    self.each_with_index do |x, i|
-      if x > max_val
-        max_val = x
-        max_i = i
-      end
-    end
-
-    return max_i
-  end
-
-  def argmin
-    min_i = 0
-    min_val = self[min_i]
-
-    self.each_with_index do |x, i|
-      if x < min_val
-        min_val = x
-        min_i = i
-      end
-    end
-
-    return min_i
-  end
-end 
+require File.join(File.dirname(__FILE__), 'swak', 'table')

data/lib/swak/table.rb

@@ -0,0 +1,96 @@
+module Swak
+  module Table
+    def Table.read(io_or_fn, opts={:has_header => false, :type => String, :delim => "\t"})
+      if io_or_fn.is_a?(String)
+        io = File.new(io_or_fn)
+      else
+        io = io_or_fn
+      end
+      if opts[:has_header]
+        header = io.gets 
+        opts[:header] = header.chomp
+      end
+      
+      table = []
+      delim = opts[:delim] || "\t"
+
+      if (opts[:type] == Integer)
+        for line in io
+          line.chomp!
+          f = line.split(delim)
+          table << f.map{|x| x.to_i_strict}
+        end
+      elsif (opts[:type] == Float)
+        for line in io
+          line.chomp!
+          f = line.chomp.split(delim)
+          table << f.map{|x| x.to_f_strict}
+        end
+      else # Assume String
+        for line in io
+          line.chomp!
+          table << line.split(delim)
+        end
+      end
+
+      return table
+    end
+
+    def Table.write(table, io_or_fn, opts={:header => nil, :fmt => nil, :delim => "\t"})
+      raise "Illegal fmt in Swak::Table.write" if !opts[:fmt].nil? && !opts[:fmt].is_a?(String)
+
+      delim = opts[:delim] || "\t"
+
+      header = opts[:header]
+      header = header.join(delim) if header.is_a?(Array) && header.size > 0
+
+      if io_or_fn.is_a?(String)
+        io = File.new(io_or_fn, "w")
+      else
+        io = io_or_fn
+      end
+
+      io.puts header if !header.nil?
+
+      if opts[:fmt]
+        fmt = opts[:fmt]
+        for row in table
+          io.puts row.map{|str| fmt % str }.join(delim)
+        end
+      else
+        for row in table
+          io.puts row.join(delim)
+        end
+      end
+
+      io.close if io_or_fn.is_a?(String)
+    end
+
+    def Table.transpose(table)
+      raise "Error in Swak::Table.transpose(): Cannot transpose nil table" if table.nil?
+      return [] if table.size == 0
+
+      num_rows = table.size
+      num_cols = table[0].size
+
+      out_mat = Array.new(num_cols)
+
+      for row in table
+        raise "Error in Swak::Table.transpose(): Doesn't support jagged table" if row.size != num_cols
+      end
+
+      c = 0
+      while c < num_cols
+        out_mat[c] = Array.new(num_rows)
+        r = 0
+        while r < num_rows
+          out_mat[c][r] = table[r][c]
+          r += 1
+        end
+        c += 1
+      end
+
+      return out_mat
+    end
+  end
+end

data/lib/swak/toplevel.rb

@@ -0,0 +1,153 @@
+class String
+  AnsiMap = {"k" => 0,
+             "r" => 1,
+             "g" => 2,
+             "y" => 3,
+             "b" => 4,
+             "m" => 5,
+             "c" => 6,
+             "w" => 7,
+             "d" => 9}
+
+  def color(color, fg=true)
+    raise "Illegal color" unless AnsiMap.include?(color)
+    color_int = AnsiMap[color]
+    return "[#{fg ? 3 : 4}#{color_int}m#{self}[#{fg ? 3 : 4}9m"
+  end
+
+  # Wraps long string by lines, using whitespace boundaries
+  def wrap(max_chars_per_line=80)
+    return gsub(/\n/," ").scan(/\S.{0,#{max_chars_per_line-2}}\S(?=\s|$)|\S+/).join("\n")
+  end
+
+end
+
+#######################################################
+#######################################################
+
+class Object
+  def is_i?
+    true if Integer(self) rescue false
+  end
+
+  def is_f?
+    true if Float(self) rescue false
+  end
+
+  def to_i_strict
+    if is_i?
+      return to_i
+    else
+      raise "String '#{self}' cannot be converted to an int"
+    end
+      
+  end
+
+  def to_f_strict
+    if is_f?
+      return to_f
+    else
+      raise "String '#{self}' cannot be converted to a float"
+    end
+  end
+end
+
+class NilClass
+  undef_method(:is_i?)
+  undef_method(:is_f?)
+  undef_method(:to_i_strict)
+  undef_method(:to_f_strict)
+  undef_method(:to_f)
+  undef_method(:to_i)
+end
+
+class Float
+  def is_i?
+    return self == self.to_i
+  end
+end
+
+
+#######################################################
+#######################################################
+
+
+class Array
+
+	def hist
+		h = {}
+		for item in self
+			h[item] ||= 0
+			h[item] += 1
+		end
+		return h
+	end
+
+	def shuffle!
+		sort_by {rand}
+		return self
+	end
+	
+	def shuffle
+		return dup().sort_by {rand}
+	end
+
+	def sum
+		total = 0
+		for item in self
+			total += item
+		end
+		return total
+	end
+
+	def avg
+		return sum.to_f / self.length
+	end
+
+	# Sample variance
+	def variance
+		average = avg()
+		variance = 0
+		for item in self
+			diff = item - average
+			variance += diff * diff
+		end
+		return variance / (self.length - 1)
+	end
+
+	def stddev
+		return Math.sqrt(self.variance())
+	end
+
+	def mean
+		return self.avg
+	end
+
+  def argmax
+    max_i = 0
+    max_val = self[max_i]
+
+    self.each_with_index do |x, i|
+      if x > max_val
+        max_val = x
+        max_i = i
+      end
+    end
+
+    return max_i
+  end
+
+  def argmin
+    min_i = 0
+    min_val = self[min_i]
+
+    self.each_with_index do |x, i|
+      if x < min_val
+        min_val = x
+        min_i = i
+      end
+    end
+
+    return min_i
+  end
+end 

metadata

@@ -2,7 +2,7 @@
 name: swak
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors: 
 - Jesse Rodriguez
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-07-10 00:00:00 Z
+date: 2012-07-16 00:00:00 Z
 dependencies: []
 
 description: Random tools and mixins
@@ -24,7 +24,9 @@ extra_rdoc_files: []
 files: 
 - lib/swak.rb
 - lib/swak/crypt.rb
+- lib/swak/toplevel.rb
 - lib/swak/interval.rb
+- lib/swak/table.rb
 - lib/swak/logger.rb
 homepage: http://rubygems.org/gems/swak
 licenses: []