pokerstats 2.0.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,5 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Andrew C. Greenberg
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.rdoc

@@ -0,0 +1,18 @@
+= pokerstats
+
+A library for extracting, computing and reporting statistics of poker hands parsed from hand history files.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but
+   bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2009 Andrew C. Greenberg. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,50 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "pokerstats"
+    gem.summary = %Q{poker hand history statistics library}
+    gem.description = %Q{a library for extracting, computing and reporting statistics of poker hands parsed from hand history files}
+    gem.email = "wizardwerdna@gmail.com"
+    gem.homepage = "http://github.com/wizardwerdna/pokerstats"
+    gem.authors = ["Andrew C. Greenberg"]
+    gem.add_development_dependency "rspec"
+    gem.add_development_dependency "wizardwerdna-pluggable"
+    gem.add_dependency "wizardwerdna-pluggable"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION')
+    version = File.read('VERSION')
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "pokerstats #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+2.0.1

data/bin/checkps

@@ -0,0 +1,190 @@
+#!/usr/bin/ruby
+require "getoptlong"
+require "rubygems"
+require "activesupport"
+require "open-uri"
+# ENV['RAILS_ENV'] = ENV['RAILS_ENV'] || 'development'  
+# require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
+require File.expand_path(File.dirname(__FILE__) + "/../lib/pokerstats/pokerstars_file")
+require File.expand_path(File.dirname(__FILE__) + '/../lib/pokerstats/player_statistics')
+# require 'pokerstats/pokerstars_file'
+# require 'pokerstats/player_statistics'
+
+def search_poker_edge(playername, players_shown = {})
+  return if players_shown[playername]
+  escaped_playername = URI.escape(playername).gsub(/["'\[\]]/,'\\\\\&').gsub(/[\[\]]/,'\\\\\\\\\&')
+  result = `curl -s http://www.poker-edge.com/whoami.php?site=Stars\\&name=#{escaped_playername}`
+  if result =~ /(Pre-Flop Tend.*\n)/
+    verbose = $1.gsub(/<\/?[^>]*>/, "")
+    if verbose =~ /Pre-Flop Tendency: ([^-]*) -/
+      preflop = $1
+    else
+      preflop = "N/A"
+    end
+  else
+    preflop = "N/A (data error)"
+  end
+  if result =~ /(Player Type.*\n)/
+    verbose = $1.gsub(/<\/?[^>]*>/, "")
+    if verbose =~ /[Yy]ou are a ([^(]* \(.*\))/
+      player_type = $1
+    else
+      player_type = ""
+    end
+  else
+    player_type = ""
+  end
+  players_shown[playername] = preflop
+  players_shown[playername] += " " + player_type unless player_type.empty?
+end
+
+def display_ratio numerator, denominator
+  if numerator.nil? or denominator.nil?
+    return "***"
+  elsif denominator < 9
+    return "#{numerator}/#{denominator}"
+  else
+    return "#{(100.0 * numerator / denominator).to_i}%"
+  end
+end
+
+def dopsfile(file, players_shown)
+  return if File.directory?(file)
+  players = {}
+  last = nil
+  statistics = Pokerstats::PlayerStatistics.new
+  Pokerstats::PokerstarsFile.open(file).each do |handrecord|
+    begin
+      handrecord.parse
+      statistics.record(handrecord)
+      last = handrecord
+    rescue Exception => e
+      puts e.message
+    end
+  end
+  return if last.nil?
+  players = last.stats.players
+  puts
+  puts "=" * file.size
+  puts file
+  puts "=" * file.size
+  STDOUT.sync = true
+  printf("Searching Poker-Edge: ")
+  players.each {|each| printf("%s ", each); search_poker_edge(each, players_shown)}
+  printf("\n")
+  STDOUT.sync = false
+  puts "=" * file.size
+  reports = statistics.reports
+  printf "%-20s %3s %4s %4s %5s %5s %5s %5s %s\n", "Screen Name", "Num", "VP$%", "PFR%", "Pre/Pos", "BAtt%", "BDef%", "CBet%", "Poker-Edge Description"
+  printf "%-20s %-39s %s\n", "-"*20, "-"*39, "-"*47
+  players.each do |each|
+    report = reports[each]
+    # puts report.to_yaml
+    t_hands = report[:t_hands]
+    vpi_p = display_ratio report[:t_vpip], report[:t_hands]
+    pfr_p = display_ratio report[:t_pfr_opportunity_taken], report[:t_pfr_opportunity]
+    prefa = report[:t_preflop_passive]. zero? ? 0.0 : 1.0 * report[:t_preflop_aggressive] / report[:t_preflop_passive]
+    posfa = report[:t_postflop_passive]. zero? ? 0.0 : 1.0 * report[:t_postflop_aggressive] / report[:t_postflop_passive]
+    batt_p = display_ratio report[:t_blind_attack_opportunity_taken], report[:t_blind_attack_opportunity]
+    bdef_p = display_ratio report[:t_blind_defense_opportunity_taken], report[:t_blind_defense_opportunity]
+    cbet_p = display_ratio report[:t_cbet_opportunity_taken], report[:t_cbet_opportunity]
+    description = players_shown[each][/\(.*\)/]
+    description ||= ""
+    description.gsub!("Passive", "P")
+    description.gsub!("Aggressive", "A")
+    description.gsub!("Tight", "T")
+    description.gsub!("Loose", "L")
+    players_shown[each].gsub!(/\(.*\)/, description)
+    printf "%-20s %3d %4s %4s %2.1f/%2.1f %5s %5s %5s %s\n", each, t_hands, vpi_p, pfr_p, prefa, posfa, batt_p, bdef_p, cbet_p, players_shown[each]
+  end
+  puts "=" * file.size
+  GC.start  
+#   puts last.reports.keys.inspect
+#   puts
+#   # puts
+#   # puts "=" * 90
+#   # puts last.path
+#   # players.each {|each| display(each, players_shown)}
+#   # puts
+#   # puts "=" * 90
+#   # puts "PLAYERS NOW AT THIS TABLE"
+#   # puts "=" * 90
+#   # printf "%-20s %3s %4s %4s %5s %s\n", "Screen Name", "Num", "VP$%", "PFR%", "Pre/Pos", "Poker-Edge Description"
+#   # printf "%-20s %-14s %s\n", "-"*20, "-"*21, "-"*47
+#   # players.each do |each|
+#   #   description = players_shown[each][/\(.*\)/]
+#   #   description ||= ""
+#   #   description.gsub!("Passive", "P")
+#   #   description.gsub!("Aggressive", "A")
+#   #   description.gsub!("Tight", "T")
+#   #   description.gsub!("Loose", "L")
+#   #   players_shown[each].gsub!(/\(.*\)/, description)
+#   #   printf "%-20s %3d %3d%% %3d%% %2.1f/%2.1f %s\n", each, 
+#   #     hands[each], (100.0 * vpip[each])/hands[each], (100.0 * pfr[each])/hands[each],
+#   #     preflop_passive[each].zero? ? 0.0 : (1.0 * preflop_aggressive[each]) / preflop_passive[each],
+#   #     postflop_passive[each].zero? ? 0.0 : (1.0 * postflop_aggressive[each]) / postflop_passive[each],
+#   #     players_shown[each]
+#   # end
+#   # puts "=" * 90
+#   # puts "information on #{hands.size} players collected"
+#   # hands = vpip = pfr = sawflop = preflop_aggressive = preflop_passive = nil
+
+#   # puts
+end
+
+def newpsfiles(user, time)
+  Dir["/Users/#{user}/Library/Application Support/PokerStars/HandHistory/**/*"].select{|each| File.mtime(each) > time}
+end
+
+def getpsdata(user, time, players_shown)
+  puts "Loading PokerStars HandHistories that have changed since #{time}"
+  while (files = newpsfiles(user, time)).empty?
+    sleep 1
+  end
+  puts files.inspect
+  files.each {|each| dopsfile(each, players_shown)}
+end
+
+def display_recent_pokerstars_results user
+  players_shown = {}
+  getpsdata(user, Time.now - 3000, players_shown)
+  loop {getpsdata(user, Time.now, players_shown)}
+end
+
+def display_poker_edge_results
+  players_shown = {}
+  $*.each do |playername|
+    puts "Poker Edge Search for #{playername}"
+    search_poker_edge(playername, players_shown)
+    puts "="*80
+    printf "%-20s %s\n", playername, players_shown[playername]
+    puts "="*80
+  end
+end
+
+opts = GetoptLong.new(
+  [ "--help",     "-h",   GetoptLong::NO_ARGUMENT],
+  [ "--version",  "-v",   GetoptLong::NO_ARGUMENT],
+  [ "--user",     "-u",   GetoptLong::OPTIONAL_ARGUMENT]
+)
+
+version_file = File.expand_path(File.dirname(__FILE__) + "/../VERSION")
+print "Pokerstars HandHistory Statistics, v#{File.read(version_file).chop}\n"
+print "Copyright (c) 2009 Andrew C. Greenberg All Rights Reserved\n"
+user = `whoami`.chop
+opts.each do |opt, arg|
+  case opt
+  when "--help", "--usage"
+    print "#{$0} playername {--user username} {--player playername} {--help } {--version}\n"
+  when "--version"
+    print "Pokerstars version  #{File.read(version_file).chop}\n"
+  when "--user"
+    user = arg unless arg.empty?
+  end
+end
+
+if $*.empty?
+  display_recent_pokerstars_results user
+else
+  display_poker_edge_results
+end

data/generators/pokerstats/USAGE

@@ -0,0 +1 @@
+foo bie doobie

data/generators/pokerstats/pokerstats_generator.rb

@@ -0,0 +1,7 @@
+class PokerstatsGenerator < Rails::Generator::Base  
+  def manifest
+    record do |m|
+      m.migration_template "create_pokerstats.rhtml",   "db/migrate", :migration_file_name => "create_pokerstats"
+    end
+  end
+end

data/generators/pokerstats/templates/create_pokerstats.rhtml

@@ -0,0 +1,28 @@
+<% 
+require 'rubygems'
+require 'pokerstats/hand_statistics'
+%>
+
+class CreatePokerstats < ActiveRecord::Migration
+  def self.up
+	  create_table :hand_statistics do |t|
+		<%= Pokerstats::HandStatistics.hand_statistics_migration_data -%>
+	    t.timestamps
+	  end
+	  create_table :player_statistics do |t|
+		t.integer :hand_statistic_id
+		t.integer :player_id
+		<%= Pokerstats::HandStatistics.player_statistics_migration_data -%>		
+	    t.timestamps
+	  end
+	  create_table :players do |t|
+	    t.string :name
+	  end
+  end
+
+  def self.down
+	drop_table :players
+    drop_table :player_statistics
+	drop_table :hand_statistics
+  end
+end

data/lib/pokerstats.rb

@@ -0,0 +1,2 @@
+module Pokerstats
+end

data/lib/pokerstats/hand_constants.rb

@@ -0,0 +1,27 @@
+module Pokerstats
+  module HandConstants
+    HAND_REPORT_SPECIFICATION = [
+      # [key,   sql_type,   function]
+      [:session_filename, 'string'],
+      [:starting_at, 'datetime'],
+      [:name, 'string'], 
+      [:description, 'string'], 
+      [:sb, 'decimal'], 
+      [:bb, 'decimal'], 
+      [:board, 'string'], 
+      [:total_pot, 'decimal'], 
+      [:rake, 'decimal'], 
+      [:played_at, 'datetime'], 
+      [:tournament, 'string']
+    ]
+    HAND_INFORMATION_KEYS = HAND_REPORT_SPECIFICATION.map{|each| each.first}
+    HAND_RECORD_INCOMPLETE_MESSAGE = "hand record is incomplete"
+    PLAYER_RECORDS_NO_PLAYER_REGISTERED = "no players have been registered"
+    PLAYER_RECORDS_DUPLICATE_PLAYER_NAME = "player screen_name has been registered twice"
+    PLAYER_RECORDS_NO_BUTTON_REGISTERED = "no button has been registered"
+    PLAYER_RECORDS_UNREGISTERED_PLAYER = "player has not been registered"
+    PLAYER_RECORDS_OUT_OF_BALANCE = "hand record is out of balance"                
+
+    MAX_SEATS = 12
+  end
+end

data/lib/pokerstats/hand_history.rb

@@ -0,0 +1,38 @@
+require File.expand_path(File.dirname(__FILE__) + "/hand_constants")
+require File.expand_path(File.dirname(__FILE__) + "/pokerstars_hand_history_parser")
+
+module Pokerstats
+  class HandHistory
+    attr_accessor :lines, :source, :position, :stats
+    def initialize lines, source, position, parser_class = PokerstarsHandHistoryParser
+      @lines = lines
+      @source = source
+      @position = position
+      @parsed = false
+      @parser_class = parser_class
+      @stats = HandStatistics.new
+    end
+  
+    def parsed?
+      @parsed
+    end
+  
+    def parse
+      @parser = @parser_class.new(@stats)
+      @lines.each do |each_line| 
+        begin
+          @parser.parse(each_line)
+        rescue => e
+          raise "#{@source}:#{position}: #{e.message}"
+        end
+      end
+      @stats.update_hand :session_filename => source, :starting_at => position
+      @parsed = true
+    end
+  
+    def reports
+      parse unless parsed?
+      @stats.reports
+    end
+  end
+end

data/lib/pokerstats/hand_statistics.rb

@@ -0,0 +1,225 @@
+require 'rubygems'
+require 'pluggable'
+require 'bigdecimal'
+require 'bigdecimal/util'
+require 'activesupport'
+require File.expand_path(File.dirname(__FILE__) + '/hand_constants')
+require File.expand_path(File.dirname(__FILE__) + '/hand_statistics_api')
+module Pokerstats
+  class HandStatistics
+    include Pluggable
+    include HandConstants
+    plugin_include_module HandStatisticsAPI
+    def initialize
+      install_plugins self
+      @hand_information = {}
+      @player_hashes = []
+      @button_player_index = nil
+      @cached_player_position = nil
+      @street_state = nil
+      street_transition(:prelude)
+    end
+
+    ##
+    # Hand Information
+    ##
+  
+    def hand_record
+      raise "#{HAND_RECORD_INCOMPLETE_MESSAGE}: #{(HAND_INFORMATION_KEYS - @hand_information.keys).inspect}" unless (HAND_INFORMATION_KEYS - @hand_information.keys).empty?
+      HAND_INFORMATION_KEYS.inject({}) do |hash, key| 
+        hash.merge!(key => @hand_information[key])
+      end
+    end
+
+    def update_hand update
+      street_transition(update[:street]) unless update[:street] == @street_state
+      @hand_information.update(update)
+      self
+    end
+  
+    ##
+    # Player Information
+    ##
+  
+    def player_records_without_validation
+      @player_hashes
+    end
+
+    def player_records
+      raise PLAYER_RECORDS_NO_PLAYER_REGISTERED if players.empty?
+      raise PLAYER_RECORDS_NO_BUTTON_REGISTERED if button.nil?
+      raise PLAYER_RECORDS_OUT_OF_BALANCE if out_of_balance
+      self.player_records_without_validation
+    end
+
+    def players
+      @player_hashes.sort{|a, b| a[:seat] <=> b[:seat]}.collect{|each| each[:screen_name]}
+    end
+  
+    def number_players
+      @player_hashes.size
+    end
+  
+    def register_player player
+      screen_name = player[:screen_name]
+      raise "#{PLAYER_RECORDS_DUPLICATE_PLAYER_NAME}: #{screen_name.inspect}" if players.member?(screen_name)
+      @cached_player_position = nil
+      @player_hashes << player
+      plugins.each{|each| each.register_player(screen_name, @street_state)}  #why the second parameter?
+      street_transition_for_player(@street_state, screen_name)
+    end
+  
+    ###
+    # Street state information
+    ##
+  
+    def street
+      @street_state
+    end
+  
+    def street_transition street
+      @street_state = street
+      plugins.each{|each| each.street_transition(street)}
+      players.each {|player| street_transition_for_player(street, player)}
+    end
+  
+    def street_transition_for_player street, screen_name
+      plugins.each{|each| each.street_transition_for_player(street, screen_name)}
+    end
+  
+    ##
+    # Button and Position Information
+    ##
+  
+    def register_button button_index
+      @cached_player_position = nil
+      @button_player_index = button_index
+    end
+  
+    def button
+      @button_player_index
+    end
+  
+    def button_relative_seat(player_hash)
+      (player_hash[:seat] + MAX_SEATS - @button_player_index) % MAX_SEATS
+    end
+
+    # long computation is cached, which cache is cleared every time a new player is registered
+    def calculate_player_position screen_name
+      @cached_player_position = {}
+      @player_hashes.sort!{|a,b| button_relative_seat(a) <=> button_relative_seat(b)}
+      @player_hashes = [@player_hashes.pop] + @player_hashes unless @player_hashes.first[:seat] == @button_player_index
+      @player_hashes.each_with_index{|player, index| player[:position] = index, @cached_player_position[player[:screen_name]] = index}
+      @cached_player_position[screen_name]
+    end
+  
+    def position screen_name
+      (@cached_player_position && @cached_player_position[screen_name]) || calculate_player_position(screen_name)
+    end
+  
+    def button?(screen_name)
+      position(screen_name) && position(screen_name).zero?
+    end
+  
+    # The cutoff position is defined as the player to the left of the button if there are three players, otherwise nil
+    def cutoff_position
+      # formerly: (number_players > 3) && (-1 % number_players)
+      -1 % number_players if number_players > 3
+    end
+  
+    def cutoff?(screen_name)
+      position(screen_name) == cutoff_position
+    end
+  
+    def blind?(screen_name)
+      (sbpos?(screen_name) || bbpos?(screen_name)) and !posted(screen_name).zero?
+    end
+  
+    def sbpos?(screen_name)
+      (number_players > 2) && position(screen_name) == 1
+    end
+  
+    def bbpos?(screen_name)
+      (number_players > 2) && position(screen_name) == 2
+    end
+
+    def attacker?(screen_name)
+      (number_players > 2) && (button?(screen_name) || cutoff?(screen_name))
+    end
+  
+  
+    ##
+    # Action Information
+    ##
+    def aggression(description)
+      case description
+      when /call/
+        :passive
+      when /raise/
+        :aggressive
+      when /bet/
+        :aggressive
+      when /fold/
+        :fold
+      when /check/
+        :check
+      else
+        :neutral
+      end
+    end
+  
+    def register_action(screen_name, description, options={})
+      raise "#{PLAYER_RECORDS_UNREGISTERED_PLAYER}: #{screen_name.inspect}" unless players.member?(screen_name)
+      plugins.each do |each|
+        each.apply_action(
+          {:screen_name => screen_name, :description => description, :aggression => aggression(description)}.update(options), 
+          @street_state)
+      end
+    end
+  
+    ##
+    # Reporting Information
+    ##
+  
+    def report_player(player)
+      result = {}
+      plugins.each {|each| result.merge!(each.report(player))}
+      result
+    end
+    
+    def reports
+      result = {}
+      players.each{|each| result[each] = report_player(each)}
+      result
+    end
+  
+    def report_hand_information
+      @hand_information
+    end
+  
+    def self.hand_statistics_migration_data
+      HAND_REPORT_SPECIFICATION.inject("") do |string, each|
+        string + "t.#{each[1]}\t#{each[0].inspect}\n"
+      end
+    end
+  
+    def self.player_statistics_migration_data
+      plugin_factory.inject("") do |str, each_plugin|
+        tstr = "# from #{each_plugin.inspect}\n"
+        each_plugin.report_specification.each do |each_datum|
+          tstr += "t.#{each_datum[1]}\t#{each_datum[0].inspect}\n"
+        end
+        str = str + tstr
+      end
+    end
+    private
+    def method_missing symbol, *args
+      plugins.send symbol, *args
+    end
+  end
+  Dir[File.dirname(__FILE__) + "/plugins/*_statistics.rb"].each {|filename| require File.expand_path(filename)}
+  HandStatistics.delegate_plugin_public_methods_except HandStatisticsAPI.public_methods
+end
+# Load Plugins and Delegate non-api public methods to plugins
+# puts Pokerstats::HandStatistics.hand_statistics_migration_data
+# puts Pokerstats::HandStatistics.player_statistics_migration_data