shogi_server/util.rb

   1 ## $Id$
   2
   3 ## Copyright (C) 2004 NABEYA Kenichi (aka nanami@2ch)
   4 ## Copyright (C) 2007-2012 Daigo Moriwaki (daigo at debian dot org)
   5 ##
   6 ## This program is free software; you can redistribute it and/or modify
   7 ## it under the terms of the GNU General Public License as published by
   8 ## the Free Software Foundation; either version 2 of the License, or
   9 ## (at your option) any later version.
  10 ##
  11 ## This program is distributed in the hope that it will be useful,
  12 ## but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 ## GNU General Public License for more details.
  15 ##
  16 ## You should have received a copy of the GNU General Public License
  17 ## along with this program; if not, write to the Free Software
  18 ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20 require 'date'
  21 require 'fileutils'
  22 require 'pathname'
  23 require 'thread'
  24
  25 module ShogiServer
  26
  27   # Generate a random number such as i<=n<max
  28   def random(i, max)
  29     return i if i >= max
  30     return rand(max-i)+i
  31   end
  32   module_function :random
  33
  34   def shuffle(array)
  35     return if array.size < 2
  36     for i in 0...(array.size-1)
  37       r = random(i, array.size)
  38       a = array[i]
  39       array[i] = array[r]
  40       array[r] = a
  41     end
  42   end
  43   module_function :shuffle
  44
  45   def factorial(n)
  46     return 1 if n<=1
  47     ret = 1
  48     while n >= 2
  49       ret *= n
  50       n -= 1
  51     end
  52     return ret
  53   end
  54   module_function :factorial
  55
  56   def nCk(n, k)
  57     return 0 if n < k
  58     numerator   = factorial(n)
  59     denominator = factorial(k) * factorial(n - k)
  60     return numerator / denominator
  61   end
  62   module_function :nCk
  63
  64   # See if the file is writable. The file will be created if it does not exist
  65   # yet.
  66   # Return true if the file is writable, otherwise false.
  67   #
  68   def is_writable_file?(file)
  69     if String === file
  70       file = Pathname.new file
  71     end
  72     if file.exist?
  73       if file.file?
  74         return file.writable_real?
  75       else
  76         return false
  77       end
  78     end
  79
  80     begin
  81       file.open("w") {|fh| }
  82       file.delete
  83     rescue
  84       return false
  85     end
  86
  87     return true
  88   end
  89   module_function :is_writable_file?
  90
  91   # Convert a DateTime insntace to a Time instance.
  92   #
  93   def datetime2time(dt)
  94     return Time.mktime dt.year, dt.month, dt.day, dt.hour, dt.min, dt.sec
  95   end
  96   module_function :datetime2time
  97
  98   # Convert a Time instance to a DateTime instance
  99   #
 100   def time2datetime(time)
 101     return DateTime.new(time.year, time.mon, time.mday,
 102                         time.hour, time.min, time.sec)
 103   end
 104   module_function :time2datetime
 105
 106   # Parse string representing a day-of-week and return a coresponding
 107   # integer value: 1 (Monday) - 7 (Sunday)
 108   #
 109   def parse_dow(str)
 110     index = Date::DAYNAMES.index(str) || Date::ABBR_DAYNAMES.index(str)
 111     return nil if index.nil?
 112     return index == 0 ? 7 : index
 113   end
 114   module_function :parse_dow
 115
 116   # Mkdir in a thread-safe way.
 117   #
 118   class Mkdir
 119     @@mutex = Mutex.new
 120
 121     # Return true if a directory is successfully created or a directory
 122     # exists already; false otherwise.
 123     #
 124     # @param path a directory name of a path to be created. For example,
 125     # given /hoge/hoo/foo.txt, aim to create /hoge/hoo.
 126     def Mkdir.mkdir_for(path)
 127       unless FileTest.directory?(File.dirname(path))
 128         @@mutex.synchronize do
 129           unless FileTest.directory?(File.dirname(path))
 130             begin
 131               FileUtils.mkdir_p File.dirname(path)
 132             rescue
 133               return false
 134             end
 135           end
 136         end # mutex
 137       end
 138       return true
 139     end
 140   end # class Mkdir
 141 end