Merge branch 'wdoor-stable'

[shogi-server/shogi-server.git] / utils / csa-filter.rb

diff --git a/utils/csa-filter.rb b/utils/csa-filter.rb
index f575ee2..72a8662 100755 (executable)
--- a/utils/csa-filter.rb
+++ b/utils/csa-filter.rb
@@ -5,7 +5,7 @@
 # you will see such files under the some_dir directory.
 #
 # Author::    Daigo Moriwaki <daigo at debian dot org>
 # you will see such files under the some_dir directory.
 #
 # Author::    Daigo Moriwaki <daigo at debian dot org>

-# Copyright:: Copyright (C) 2006-2008  Daigo Moriwaki <daigo at debian dot org>
+# Copyright:: Copyright (C) 2006-2012 Daigo Moriwaki <daigo at debian dot org>

 #
 # $Id$
 #
 #
 # $Id$
 #


@@ -28,6 +28,7 @@
 require 'time'
 require 'pathname'
 require 'getoptlong'
 require 'time'
 require 'pathname'
 require 'getoptlong'

+require 'nkf'

 
 class CsaFileReader
   WIN_MARK  = "win"
 
 class CsaFileReader
   WIN_MARK  = "win"


@@ -35,24 +36,29 @@ class CsaFileReader
   DRAW_MARK = "draw"
 
   attr_reader :file_name
   DRAW_MARK = "draw"
 
   attr_reader :file_name

+  attr_reader :str

   attr_reader :black_name, :white_name
   attr_reader :black_id, :white_id
   attr_reader :winner, :loser
   attr_reader :black_name, :white_name
   attr_reader :black_id, :white_id
   attr_reader :winner, :loser

+  attr_reader :state

   attr_reader :start_time, :end_time
   attr_reader :start_time, :end_time

+  attr_reader :ply

 
 

-  def initialize(file_name)
+  def initialize(file_name, encoding="Shift_JIS:EUC-JP")

     @file_name = file_name
     @file_name = file_name

+    @encoding = encoding
+    @ply = 0

     grep
   end
 
   def grep
     grep
   end
 
   def grep

-    str = File.open(@file_name).read
+    @str = File.open(@file_name, "r:#{@encoding}").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"
+
+    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}    

       p1_name, p1_mark = p1.split(" ")
       p2_name, p2_mark = p2.split(" ")
       if p1_name == @black_name
       p1_name, p1_mark = p1.split(" ")
       p2_name, p2_mark = p2.split(" ")
       if p1_name == @black_name


@@ -65,13 +71,13 @@ class CsaFileReader
         raise "Never reach!: #{black} #{white} #{p3} #{p2}"
       end
     end
         raise "Never reach!: #{black} #{white} #{p3} #{p2}"
       end
     end

-    if /^\$START_TIME:(.*)$/ =~ str
+    if /^\$START_TIME:(.*)$/ =~ @str

       @start_time = Time.parse($1.strip)
     end
       @start_time = Time.parse($1.strip)
     end

-    if /^'\$END_TIME:(.*)$/ =~ str
+    if /^'\$END_TIME:(.*)$/ =~ @str

       @end_time = Time.parse($1.strip)
     end
       @end_time = Time.parse($1.strip)
     end

-    if /^'rating:(.*)$/ =~ str
+    if /^'rating:(.*)$/ =~ @str

       black_id, white_id = $1.split(":").map {|a| a.strip}
       @black_id = identify_id(black_id)
       @white_id = identify_id(white_id)
       black_id, white_id = $1.split(":").map {|a| a.strip}
       @black_id = identify_id(black_id)
       @white_id = identify_id(white_id)


@@ -88,6 +94,20 @@ class CsaFileReader
         end
       end
     end
         end
       end
     end

+
+    @str.each_line do |line|
+      if /^[\+\-]\d{4}[A-Z]{2}/ =~ line
+        @ply += 1
+      end
+    end
+  end
+
+  def movetimes
+    ret = []
+    @str.gsub(%r!^T(\d+)!) do |match|
+      ret << $1.to_i
+    end
+    return ret

   end
 
   def to_s
   end
 
   def to_s


@@ -95,7 +115,8 @@ class CsaFileReader
            "BlackName #{@black_name}, WhiteName #{@white_name}\n" +
            "BlackId #{@black_id}, WhiteId #{@white_id}\n" +
            "Winner #{@winner}, Loser #{@loser}\n"    +
            "BlackName #{@black_name}, WhiteName #{@white_name}\n" +
            "BlackId #{@black_id}, WhiteId #{@white_id}\n" +
            "Winner #{@winner}, Loser #{@loser}\n"    +

-           "Start #{@start_time}, End #{@end_time}\n"
+           "Start #{@start_time}, End #{@end_time}\n" +
+           "Ply #{@ply}"

   end
 
   def identify_id(id)
   end
 
   def identify_id(id)


@@ -129,7 +150,6 @@ if $0 == __FILE__
       eval "$OPT_#{name.sub(/^--/, '').gsub(/-/, '_').upcase} = '#{arg}'"
     end
   rescue
       eval "$OPT_#{name.sub(/^--/, '').gsub(/-/, '_').upcase} = '#{arg}'"
     end
   rescue

-    puts "hoge"

     usage
   end
   
     usage
   end
   


@@ -141,19 +161,19 @@ if $0 == __FILE__
 
       if $OPT_PLAYERS
         players = $OPT_PLAYERS.split(",")
 
       if $OPT_PLAYERS
         players = $OPT_PLAYERS.split(",")

-        unless (csa.black_id.index(players[0]) == 0 &&
-                csa.white_id.index(players[1]) == 0) ||
-               (csa.black_id.index(players[1]) == 0 &&
-                csa.white_id.index(players[0]) == 0)
+        unless (csa.black_id.downcase.index(players[0].downcase) == 0 &&
+                csa.white_id.downcase.index(players[1].downcase) == 0) ||
+               (csa.black_id.downcase.index(players[1].downcase) == 0 &&
+                csa.white_id.downcase.index(players[0].downcase) == 0)

           next
         end
       end
       
       if $OPT_BLACK
           next
         end
       end
       
       if $OPT_BLACK

-        next unless csa.black_id.index($OPT_BLACK) == 0
+        next unless csa.black_id.downcase.index($OPT_BLACK.downcase) == 0

       end
       if $OPT_WHITE
       end
       if $OPT_WHITE

-        next unless csa.white_id.index($OPT_WHITE) == 0
+        next unless csa.white_id.downcase.index($OPT_WHITE.downcase) == 0

       end
       puts csa.file_name
     end
       end
       puts csa.file_name
     end