utils/players_graph.rb

   1 #!/usr/bin/ruby
   2 #    This generates graphs of evaluation values from comments in CSA files.
   3 #    Ruby libraries that are required:
   4 #      - RubyGems: http://rubyforge.org/projects/rubygems/
   5 #      - rgplot:   http://rubyforge.org/projects/rgplot/
   6 #    OS librariles that is required:
   7 #      - Gnuplot:  http://www.gnuplot.info/
   8 #                  On Debian, $ sudo apt-get install gnuplot
   9 #
  10 #    Copyright (C) 2008  Daigo Moriwaki <daigo@debian.org>
  11 #
  12 #    Version: $Id$
  13 #
  14 #    This program is free software; you can redistribute it and/or modify
  15 #    it under the terms of the GNU General Public License as published by
  16 #    the Free Software Foundation; either version 2 of the License, or
  17 #    (at your option) any later version.
  18 #
  19 #    This program is distributed in the hope that it will be useful,
  20 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  21 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22 #    GNU General Public License for more details.
  23 #
  24 #    You should have received a copy of the GNU General Public License
  25 #    along with this program; if not, write to the Free Software
  26 #    Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  27
  28 require 'pathname'
  29 require 'getoptlong'
  30 require 'yaml'
  31 require 'date'
  32 require 'set'
  33 require 'rubygems'
  34 require 'gnuplot'
  35
  36 $players = {}
  37
  38 class Format
  39   attr_reader :ext, :size
  40   def initialize(root)
  41     @root = root
  42     @size = "1,1"
  43   end
  44 end
  45
  46 class LargeFormat < Format
  47   def initialize(root)
  48     super
  49   end
  50
  51   def apply(plot)
  52     plot.terminal @ext
  53     plot.size     @size
  54     plot.format 'x "%y/%m/%d"'
  55     plot.ytics  "100"
  56     plot.mytics "5"
  57     plot.xlabel  "Date"
  58     plot.ylabel  "Rate"
  59   end
  60 end
  61
  62 class LargePngFormat < LargeFormat
  63   def initialize(root)
  64     super
  65     @ext = "png"
  66   end
  67
  68   def to_image_file(name)
  69     return File.join(@root, "#{name}-large.#{ext}")
  70   end
  71 end
  72
  73 class SmallPngFormat < Format
  74   def initialize(root)
  75     super
  76     @ext = "png"
  77   end
  78
  79   def to_image_file(name)
  80     return File.join(@root, "#{name}-small.#{ext}")
  81   end
  82
  83   def apply(plot)
  84     plot.terminal "png size 320,200 crop"
  85     plot.format   'x "%b"'
  86     plot.ytics    "200"
  87     plot.mytics   "2"
  88   end
  89 end
  90
  91 class SvgFormat < LargeFormat
  92   def initialize(root)
  93     super
  94     @ext = "svg"
  95   end
  96
  97   def to_image_file(name)
  98     return File.join(@root, "#{name}.#{ext}")
  99   end
 100 end
 101
 102 def plot(format, name, dates, rates, rdates, rrates)
 103   Gnuplot.open do |gp|
 104     Gnuplot::Plot.new( gp ) do |plot|
 105       format.apply(plot)
 106       plot.title   name
 107       plot.output format.to_image_file(name)
 108
 109       #plot.size    "ratio #{1/1.618}"
 110       plot.xdata   "time"
 111       plot.timefmt '"%Y-%m-%d"'
 112       plot.xrange  "[\"%s\":\"%s\"]" %
 113                     [dates.first.strftime("%Y-%m-%d"),
 114                      dates.last.strftime("%Y-%m-%d")]
 115       ymin = ((rates + rrates).min/50) * 50
 116       ymax = ((rates + rrates).max/50 + 1) * 50
 117       plot.yrange "[%s:%s]" % [ymin, ymax]
 118       plot.grid
 119       data = []
 120       data << Gnuplot::DataSet.new([dates, rates]) do |ds|
 121                 ds.using = "1:2"
 122                 ds.with  = "lines"
 123                 ds.title = "original"
 124               end
 125       if !rdates.empty?
 126         data << Gnuplot::DataSet.new([rdates, rrates]) do |ds|
 127                   ds.using = "1:2"
 128                   ds.with  = "lines"
 129                   ds.title = "relative (rate24)"
 130                 end
 131       end
 132       plot.data = data
 133     end
 134   end
 135 end
 136
 137 def load_file(file_name)
 138   if /^.*-(\d{8}).yaml$/ =~ file_name
 139     date = Date::parse($1)
 140   else
 141     return
 142   end
 143   db = YAML::load_file(file_name)
 144   return unless db['players'][0]
 145   db['players'][0].each do |name, hash|
 146     $players[name] ||= {}
 147     $players[name][date] = hash['rate'].to_i
 148   end
 149 end
 150
 151 if $0 == __FILE__
 152   def usage
 153     puts "Usage: #{$0} [--output-dir dir] <players_yaml_files>..."
 154     puts "Options:"
 155     puts "  --output-dir dir  Images will be located in the dir."
 156     exit 1
 157   end
 158
 159   usage if ARGV.empty?
 160
 161   parser = GetoptLong.new
 162   parser.set_options(['--output-dir', '-o', GetoptLong::REQUIRED_ARGUMENT])
 163   begin
 164     parser.each_option do |name, arg|
 165       eval "$OPT_#{name.sub(/^--/, '').gsub(/-/, '_').upcase} = '#{arg}'"
 166     end
 167   rescue
 168     usage
 169   end
 170
 171   while file = ARGV.shift
 172     load_file(file)
 173   end
 174
 175   formats = [LargePngFormat.new($OPT_OUTPUT_DIR),
 176              SmallPngFormat.new($OPT_OUTPUT_DIR),
 177              SvgFormat.new($OPT_OUTPUT_DIR)]
 178
 179   $players.each do |name, hash|
 180     dates, rates = hash.sort.transpose
 181     rdates = dates.find_all do |date|
 182       $players["YSS+707d4f98d9d2620cdaab58f19d02a2e4"] &&
 183       $players["YSS+707d4f98d9d2620cdaab58f19d02a2e4"][date]
 184     end
 185     rrates = rdates.map do |date|
 186       2300 - $players["YSS+707d4f98d9d2620cdaab58f19d02a2e4"][date] + hash[date]
 187     end
 188     formats.each do |format|
 189       plot(format, name, dates, rates, rdates, rrates)
 190     end
 191   end
 192 end
 193