Merge Justin

[selectricity-live] / lib / rubyvote / condorcet.rb

diff --git a/lib/rubyvote/condorcet.rb b/lib/rubyvote/condorcet.rb
index aaa504477373e07735b6eab9b7c847db0c68232a..715399566e270db367d311c9a26d2f7c08fd1bcf 100644 (file)
--- a/lib/rubyvote/condorcet.rb
+++ b/lib/rubyvote/condorcet.rb
@@ -32,7 +32,7 @@
 ## the CloneproofSSDVote classes but should not be used directly.
 
 class CondorcetVote < ElectionVote
 ## the CloneproofSSDVote classes but should not be used directly.
 
 class CondorcetVote < ElectionVote

-
+  

   attr_accessor :results
 
   def initialize(votes=nil)
   attr_accessor :results
 
   def initialize(votes=nil)


@@ -144,15 +144,39 @@ end
 ## directly.
 
 class CondorcetResult < ElectionResult
 ## directly.
 
 class CondorcetResult < ElectionResult

+  attr_reader :matrix
+  

   def initialize(voteobj=nil)
     unless voteobj and voteobj.kind_of?( CondorcetVote )
       raise ArgumentError, "You must pass a CondorcetVote array.", caller
     end
     super(voteobj)
   def initialize(voteobj=nil)
     unless voteobj and voteobj.kind_of?( CondorcetVote )
       raise ArgumentError, "You must pass a CondorcetVote array.", caller
     end
     super(voteobj)

+    @matrix = voteobj.votes

   end
   end

-
+  
+  def list_defeats
+    victors = Array.new
+    ties = Array.new
+    candidates = @matrix.keys.sort
+    
+    candidates.each do |candidate|
+      candidates.each do |challenger|
+        next if candidate == challenger
+        diff = @matrix[candidate][challenger] - @matrix[challenger][candidate]
+        if diff > 0 
+          victors << [candidate, challenger, diff]
+        elsif diff == 0 && ties.include?([challenger, candidate]) == false
+          ties << [candidate, challenger] 
+        end
+      end
+    end
+    
+    victories = victors.sort {|a,b| b[2] <=> a[2]}
+    
+    return victories, ties    
+  end
+        

   protected
   protected

-

   def defeats(candidates=nil, votes=nil)
     candidates = @election.candidates unless candidates
     votes = @election.votes unless votes
   def defeats(candidates=nil, votes=nil)
     candidates = @election.candidates unless candidates
     votes = @election.votes unless votes


@@ -180,7 +204,7 @@ class PureCondorcetResult < CondorcetResult
   end
 
   protected
   end
 
   protected

-
+  

   def condorcet
     votes = @election.votes
     candidates = @election.candidates
   def condorcet
     votes = @election.votes
     candidates = @election.candidates