sigma 0.0.0 → 0.1.0

data/lib/generators/sigma/migration_generator.rb

@@ -5,6 +5,7 @@ module MigrationGenerator
     generate("migration", "add_wins_to_#{@model_name.pluralize} wins:integer")
     generate("migration", "add_losses_to_#{@model_name.pluralize} losses:integer")
     generate("migration", "add_draws_to_#{@model_name.pluralize} draws:integer")
+    generate("migration", "add_expectations_to_#{@model_name.pluralize} expectations:string")
   end
 
   def set_default_values
@@ -12,11 +13,29 @@ module MigrationGenerator
     migrations.each do |m|
       name = m.split(/^[0-9]+_/)[1]
       if name == "add_skill_to_#{@model_name.pluralize}.rb"
-        puts 'passed'
-        inject_into_file  "db/migrate/#{m}", 
-                          ", :default => #{@scale/2}", 
-                          :after => ":float"
+        update_migration(m, @scale/2, ":float")
+      elsif name == "add_doubt_to_#{@model_name.pluralize}.rb"
+        update_migration(m, @scale/6, ":float")
+      elsif name == "add_wins_to_#{@model_name.pluralize}.rb" || name == "add_losses_to_#{@model_name.pluralize}.rb" || name == "add_draws_to_#{@model_name.pluralize}.rb"
+        update_migration(m, 0, ":integer")
       end
     end
   end
+
+  def migrate
+    puts <<-EOS
+
+=======================================
+> Running rake db:migrate
+=======================================
+
+    EOS
+    rake("db:migrate")
+  end
+
+  private
+
+  def update_migration (file, default, after)
+    inject_into_file  "db/migrate/#{file}", ", :default => #{default}", :after => "#{after}"
+  end
 end

data/lib/generators/sigma/model_updater.rb

@@ -0,0 +1,15 @@
+module ModelUpdater
+  def add_methods
+    resource = File.read find_in_source_paths("resource_model.rb")
+    inject_into_file  "app/models/#{@model_name}.rb",
+                      ":skill, :doubt, :wins, :losses, :draws, :expectations, ",
+                      :after => "attr_accessible "
+    inject_into_file  "app/models/#{@model_name}.rb",
+                      "\n#{resource}\n",
+                      :after => /attr_accessible [a-z:_, ]+$/
+  end
+
+  def creating_templates
+    template "sigma.rb", "config/initializers/sigma.rb"
+  end
+end

data/lib/generators/sigma/sigma_generator.rb

@@ -0,0 +1,23 @@
+require "generators/sigma/model_updater"
+require "generators/sigma/migration_generator"
+require "generators/sigma/generator_instructions"
+
+class SigmaGenerator < Rails::Generators::Base
+  include MigrationGenerator, GeneratorInstructions, ModelUpdater
+
+  source_root File.expand_path("../../templates", __FILE__)
+  
+  desc "Setup Sigma for some resource"
+
+  def execute
+    @model_name = ask("What is your resource model? (eg. user)")
+    @scale      = ask("What will be the scale? (default is 50)").to_f
+    @scale      = 50.0 if 0
+    generate_migrations
+    set_default_values
+    creating_templates
+    add_methods
+    migrate
+    instructions
+  end
+end

data/lib/generators/templates/resource_model.rb

@@ -0,0 +1,114 @@
+  before_create "self.expectations = {
+    'win_expectation'=>{'wins'=>0, 'losses'=>0, 'draws'=>0},
+    'lost_expectation'=>{'wins'=>0, 'losses'=>0, 'draws'=>0},
+    'draw_expectation'=>{'wins'=>0, 'losses'=>0, 'draws'=>0}
+  }"
+
+  serialize :expectations
+
+  scope :ranking, select("#{self.name.pluralize.downcase}.*")
+    .select("(#{self.name.pluralize.downcase}.skill - 3 * #{self.name.pluralize.downcase}.doubt) as rating")
+    .order("rating DESC")
+
+  def rating
+    self.skill - 3 * self.doubt
+  end
+
+  def position
+    self.class.ranking.index(self) + 1
+  end
+
+  def matches
+    self.wins + self.losses + self.draws
+  end
+
+  def won(match_difficult)
+    define_match_data(match_difficult)
+
+    if @expectation == true
+      skill_won = self.skill/@alpha/Sigma::SCALE/(Sigma::SCALE+@difficult/@alpha)
+    else
+      skill_won = self.skill * @alpha
+    end
+
+    self.skill = self.skill + skill_won
+    update_sigma(true)
+    self.increment :wins
+    self.save
+  end
+
+  def lost(match_difficult)
+    define_match_data(match_difficult)
+
+    if !@expectation
+      skill_lost = self.skill/@alpha/Sigma::SCALE/(Sigma::SCALE+@difficult.abs/@alpha)
+    else
+      skill_lost = self.skill * @alpha
+    end
+    self.skill = self.skill - skill_lost
+    update_sigma(false)
+    self.increment :losses
+    self.save
+  end
+
+  def define_match_data(difficult)
+    @expectation          = (difficult == 0) ? 0 : difficult > 0
+    @resource_probability = probability(@expectation)
+
+    if difficult == 0 || difficult < 2.5*Sigma::SCALE/100 && difficult > 0
+      @difficult = 2.5*Sigma::SCALE/100
+    elsif difficult > -2.5*Sigma::SCALE/100 && difficult < 0
+      @difficult = -2.5*Sigma::SCALE/100
+    else
+      @difficult = difficult.abs
+    end
+    @alpha = (@difficult / Sigma::SCALE).abs
+  end
+
+  def update_sigma(exp)
+    exp_result    = (@expectation == true) ? 'win_expectation' : nil
+    exp_result    ||= (@expectation == false) ? 'lost_expectation' : 'draw_expectation'
+    result        = (exp == true) ? 'wins' : nil
+    result        ||= (exp == false) ? 'losses' : 'draws'
+    expectations  = self.expectations[exp_result][result]
+
+    self.expectations[exp_result][result] = expectations + 1
+    if @expectation == exp
+      salpha     = (1 - @resource_probability) * @alpha
+      self.doubt = self.doubt - self.doubt * salpha
+    else
+      salpha     = @resource_probability * @alpha
+      self.doubt = self.doubt + self.doubt * salpha
+    end
+  end
+
+  def probability(expectation)
+    result       = (expectation == true) ? 'wins' : nil
+    result       ||= (expectation == false) ? 'losses' : 'draws'
+
+    exp_result   = (expectation == true) ? :we : nil
+    exp_result   ||= (expectation == false) ? :le : :de
+
+    w = self.wins   * 100.0 / ((matches == 0) ? 1 : matches)
+    l = self.losses * 100.0 / ((matches == 0) ? 1 : matches)
+    d = self.draws  * 100.0 / ((matches == 0) ? 1 : matches)
+
+    expectations = {
+                      'wins'   =>  { we:0, le:0, de:0 },
+                      'losses' =>  { we:0, le:0, de:0 },
+                      'draws'  =>  { we:0, le:0, de:0 }
+                   }
+
+    expectations.each do |k, v|
+      mult = (k == 'wins') ? w : nil
+      mult ||= (k == 'losses') ? l : d
+
+      v[:we] = mult*(self.expectations['win_expectation'][k]*100.0/((self[k] == 0) ? 1 : self[k]))
+      v[:le] = mult*(self.expectations['lost_expectation'][k]*100.0/((self[k] == 0) ? 1 : self[k]))
+      v[:de] = mult*(self.expectations['draw_expectation'][k]*100.0/((self[k] == 0) ? 1 : self[k]))
+    end
+
+    all_probabilities = expectations[result][:we]+expectations[result][:le]+expectations[result][:de]
+    probability       = expectations[result][exp_result] / ((all_probabilities == 0) ? 1 : all_probabilities)
+    (probability == 0) ? 0.25 : probability
+  end

data/lib/generators/templates/sigma.rb

@@ -0,0 +1 @@
+Sigma::SCALE = <%= @scale %>

data/lib/sigma.rb

@@ -1,3 +1,2 @@
 class Sigma
-  require "sigma/core"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sigma
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-06 00:00:00.000000000Z
+date: 2013-05-06 00:00:00.000000000 Z
 dependencies: []
 description: ''
 email: joaomdmoura@gmail.com
@@ -20,8 +20,10 @@ files:
 - lib/sigma.rb
 - lib/generators/sigma/generator_instructions.rb
 - lib/generators/sigma/migration_generator.rb
-- lib/generators/sigma/setup_generator.rb
-- lib/sigma/core.rb
+- lib/generators/sigma/model_updater.rb
+- lib/generators/sigma/sigma_generator.rb
+- lib/generators/templates/resource_model.rb
+- lib/generators/templates/sigma.rb
 - init.rb
 homepage: http://joaomdmoura.github.com/sigma/
 licenses: []
@@ -43,7 +45,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.18
+rubygems_version: 1.8.15
 signing_key: 
 specification_version: 3
 summary: A ranking algorithm implementation for Ruby on Rails applications.

data/lib/generators/sigma/setup_generator.rb

@@ -1,22 +0,0 @@
-require "generators/sigma/migration_generator"
-require "generators/sigma/generator_instructions"
-
-class Sigma
-  class SetupGenerator < Rails::Generators::Base
-    include MigrationGenerator
-    include GeneratorInstructions
-
-    source_root File.expand_path("../../templates", __FILE__)
-    
-    desc "Setup Sigma for some resource"
-
-    def execute
-      @model_name = ask("What is your resource model? (eg. user)")
-      @scale      = ask("What will be the scale? (eg. 50)").to_i
-      generate_migrations
-      set_default_values
-      instructions
-    end
-  
-  end
-end

data/lib/sigma/core.rb

@@ -1,5 +0,0 @@
-class Sigma
-  class Core
-    #TODO
-  end
-end