wizardwerdna-pokerstats 0.4.0

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 @@
+0.4.0

data/lib/checkps

@@ -0,0 +1,183 @@
+#!/usr/bin/ruby
+require "getoptlong"
+# ENV['RAILS_ENV'] = ENV['RAILS_ENV'] || 'development'  
+# require File.expand_path(File.dirname(__FILE__) + "/../config/environment")
+require File.expand_path(File.dirname(__FILE__) + "/pokerstats/pokerstars_file")
+require File.expand_path(File.dirname(__FILE__) + '/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 = PlayerStatistics.new
+  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]
+)
+
+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 "Judi's Awesome Poker Program -- for Absolute Poker, version 0.6\n"
+  when "--user"
+    user = arg unless arg.empty?
+  end
+end
+
+if $*.empty?
+  display_recent_pokerstars_results user
+else
+  display_poker_edge_results
+end

data/lib/pokerstats/hand_constants.rb

@@ -0,0 +1,12 @@
+module HandConstants
+  HAND_INFORMATION_KEYS = [:session_filename, :starting_at, :name, :description, :sb, :bb, :board, :total_pot, :rake, :played_at, :tournament]
+
+  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

data/lib/pokerstats/hand_history.rb

@@ -0,0 +1,35 @@
+require File.expand_path(File.dirname(__FILE__) + "/hand_constants")
+require File.expand_path(File.dirname(__FILE__) + "/pokerstars_hand_history_parser")
+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

data/lib/pokerstats/hand_statistics.rb

@@ -0,0 +1,223 @@
+require 'rubygems'
+require 'pluggable'
+require File.expand_path(File.dirname(__FILE__) + '/hand_constants')
+require File.expand_path(File.dirname(__FILE__) + '/hand_statistics_api')
+
+class HandStatistics
+  include Pluggable
+  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
+  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.rails_migration_for_player_data
+    prefix = <<-PREFIX
+    class AddHandStatisticsForPlayer < ActiveRecord::Migration
+      def self.up
+        create_table :player_hand_statistics do |t|
+          t.integer :hand_id
+    PREFIX
+    middle = plugin_factory.inject(""){|string, each| string + each.rails_migration_segment_for_player_data}
+    suffix = <<-SUFFIX
+        end
+      end
+      def self.down
+       drop_table :player_hand_statistics
+      end
+    end
+    SUFFIX
+    return prefix + middle + suffix
+  end
+  
+  private
+  def method_missing symbol, *args
+    plugins.send symbol, *args
+  end
+end
+
+# Load Plugins and Delegate non-api public methods to plugins
+Dir[File.dirname(__FILE__) + "/plugins/*_statistics.rb"].each {|filename| require File.expand_path(filename)}
+HandStatistics.delegate_plugin_public_methods_except HandStatisticsAPI.public_methods
+
+puts HandStatistics.rails_migration_for_player_data