OSDN Git Service

[mk_rate]
authorbeatles <beatles@b8c68f68-1e22-0410-b08e-880e1f8202b4>
Sat, 27 Dec 2008 12:27:40 +0000 (12:27 +0000)
committerbeatles <beatles@b8c68f68-1e22-0410-b08e-880e1f8202b4>
Sat, 27 Dec 2008 12:27:40 +0000 (12:27 +0000)
- Check kifu files more carefully so that files that contain
  incomplete records are skipped.
- Optimized for speed and memory size so that mk_rate might fail
  to alloc due to out of memory.

changelog
mk_rate

index 9a4f330..b331917 100644 (file)
--- a/changelog
+++ b/changelog
@@ -1,3 +1,11 @@
+2008-12-27 Daigo Moriwaki <daigo at debian dot org>
+
+       * [mk_rate]
+         - Check kifu files more carefully so that files that contain 
+           incomplete records are skipped.
+         - Optimized for speed and memory size so that mk_rate might fail
+           to alloc due to out of memory.
+
 2008-12-13 Daigo Moriwaki <daigo at debian dot org>
 
        * [shogi-server]
diff --git a/mk_rate b/mk_rate
index a035783..458e49a 100755 (executable)
--- a/mk_rate
+++ b/mk_rate
@@ -89,6 +89,7 @@
 
 require 'yaml'
 require 'time'
+require 'stringio'
 require 'getoptlong'
 require 'gsl'
 require 'rubygems'
@@ -435,7 +436,11 @@ class WinLossMatrix
 
   def self.mk_win_loss_matrix(players)
     obj = mk_matrix(players)
-    return obj.filter
+    while true do
+      obj, ret = obj.filter
+      break if ret
+    end
+    return obj
   end
 
   ##################
@@ -466,23 +471,37 @@ class WinLossMatrix
   end
 
   ##
-  # Removes a delete_index'th player and returns a new object.
+  # Removes the delete_index'th player and returns a new object.
+  #
+  #  [ 1 0 0   * [ 1 2 3   * [ 1 0   = [ 1 3
+  #    0 0 1 ]     4 5 6       0 0       7 9 ]
+  #                7 8 9 ]     0 1 ]
   #
   def delete_row(delete_index)
-    copied_cols = []
-    (0...size).each do |i|
-      next if i == delete_index
-      row = @matrix.row(i).clone
-      row.delete_at(delete_index)
-      copied_cols << row
-    end
-    if copied_cols.size == 0
+    new_matrix = nil
+    if delete_index == 0 && @matrix.size1 == 1 && @matrix.size2 == 1
       new_matrix = GSL::Matrix.new
     else
-      new_matrix = GSL::Matrix[*copied_cols]
+      a1 = []
+      size.times do |i|
+        next if i == delete_index
+        v = Array.new(size, 0)
+        v[i] = 1
+        a1 << v
+      end
+      m1 = GSL::Matrix[*a1]
+      a1 = nil
+      m2 = m1.transpose
+      new_matrix = m1 * @matrix * m2
     end
+
     new_keys = @keys.clone
     new_keys.delete_at(delete_index)
+
+    unless new_matrix
+      raise "The new matrix is null!"
+    end
+
     return WinLossMatrix.new(new_keys, new_matrix)
   end
 
@@ -516,10 +535,10 @@ class WinLossMatrix
     end
 
     # The recursion ends if there is nothing to delete
-    return self if delete.empty?
+    return [self, true] if delete.empty?
 
     new_obj = delete_rows(delete)
-    new_obj.filter
+    return [new_obj, false]
   end
 
   ##
@@ -626,30 +645,42 @@ def identify_id(id)
 end
 
 def grep(file)
-  str = File.open(file).read
-
-  if /^N\+(.*)$/ =~ str then black_name = $1.strip end
-  if /^N\-(.*)$/ =~ str then white_name = $1.strip end
-
-  if /^'summary:(.*)$/ =~ str
-    state, p1, p2 = $1.split(":").map {|a| a.strip}    
-    return if state == "abnormal"
-    p1_name, p1_mark = p1.split(" ")
-    p2_name, p2_mark = p2.split(" ")
-    if p1_name == black_name
-      black_name, black_mark = p1_name, p1_mark
-      white_name, white_mark = p2_name, p2_mark
-    elsif p2_name == black_name
-      black_name, black_mark = p2_name, p2_mark
-      white_name, white_mark = p1_name, p1_mark
-    else
-      raise "Never reach!: #{black} #{white} #{p3} #{p2}"
-    end
-  end
-  if /^'\$END_TIME:(.*)$/ =~ str
-    time = Time.parse($1.strip)
-  end
-  if /^'rating:(.*)$/ =~ str
+  black_name = nil
+  white_name = nil
+  black_mark = nil
+  white_mark = nil
+  black_id   = nil
+  white_id   = nil
+  time       = nil
+
+  lines = StringIO.new File.open(file).read
+
+  lines.each_line do |str|
+    case str
+    when /^N\+(.*)$/
+      black_name = $1.strip 
+    when /^N\-(.*)$/
+      white_name = $1.strip 
+    when /^'summary:(.*)$/
+      state, p1, p2 = $1.split(":").map {|a| a.strip}    
+      return if state == "abnormal"
+      p1_name, p1_mark = p1.split(" ")
+      p2_name, p2_mark = p2.split(" ")
+      if p1_name == black_name
+        black_name, black_mark = p1_name, p1_mark
+        white_name, white_mark = p2_name, p2_mark
+      elsif p2_name == black_name
+        black_name, black_mark = p2_name, p2_mark
+        white_name, white_mark = p1_name, p1_mark
+      else
+        raise "Never reach!: #{black} #{white} #{p3} #{p2}"
+      end
+    when /^'\$END_TIME:(.*)$/
+      time = Time.parse($1.strip)
+    end # case
+  end # each line
+
+  if /^'rating:(.*)$/ =~ lines.string
     black_id, white_id = $1.split(":").map {|a| a.strip}
     black_id = identify_id(black_id)
     white_id = identify_id(white_id)