sapor 0.1b1

data/lib/sapor/proportional.rb

@@ -0,0 +1,128 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2014 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+module Sapor
+  #
+  # Class representing a proportional electoral system.
+  #
+  class Proportional
+    def initialize(last_election_result, last_detailed_election_result,
+                   seat_distribution, denominators_class, threshold = 0)
+      @last_election_result = last_election_result
+      @last_detailed_election_result = last_detailed_election_result
+      @seat_distribution = seat_distribution
+      @denominators_class = denominators_class
+      @threshold = threshold
+    end
+
+    def project(simulation)
+      multiplicators = calculate_multiplicators(simulation)
+      result = create_empty_result(simulation)
+      @last_detailed_election_result.each_pair do |name, local_last_result|
+        no_of_seats = @seat_distribution[name]
+        seats = local_seats(no_of_seats, local_last_result, multiplicators)
+        add_seats_to_result(result, seats)
+      end
+      result
+    end
+
+    private
+
+    def add_seats_to_result(result, seats)
+      seats.each do |seat|
+        if result.key?(seat)
+          result[seat] += 1
+        else
+          result[seat] = 1
+        end
+      end
+    end
+
+    def calculate_multiplicators(simulation)
+      simulation_sum = simulation.values.inject(:+)
+      last_election_sum = @last_election_result.values.inject(:+)
+      multiplicators = {}
+      simulation.each_key do |choice|
+        new_fraction = simulation[choice].to_f / simulation_sum
+        last_fraction = @last_election_result[choice].to_f / last_election_sum
+        multiplicators[choice] = new_fraction / last_fraction
+      end
+      multiplicators
+    end
+
+    def create_empty_result(simulation)
+      result = {}
+      simulation.each_key do |choice|
+        result[choice] = 0
+      end
+      result[OTHER] = 0
+      result
+    end
+
+    def local_votes(local_last_result, multiplicators)
+      local_votes = {}
+      local_last_result.each_pair do |choice, votes|
+        if multiplicators.key?(choice)
+          local_votes[choice] = votes * multiplicators[choice]
+        else
+          local_votes[choice] = votes
+        end
+      end
+      local_votes
+    end
+
+    def local_quotients(local_votes, local_threshold, no_of_seats)
+      local_quotients = []
+      local_votes.each_pair do |choice, new_value|
+        next if new_value < local_threshold
+        @denominators_class.get(no_of_seats).each do |d|
+          local_quotients << [choice, new_value.to_f / d]
+        end
+      end
+      local_quotients
+    end
+
+    def local_seats(no_of_seats, local_last_result, multiplicators)
+      local_votes = local_votes(local_last_result, multiplicators)
+      local_threshold = local_votes.values.inject(:+).to_f * @threshold
+      local_quotients = local_quotients(local_votes, local_threshold,
+                                        no_of_seats)
+      sorted_quotients = local_quotients.sort { |a, b| b.last <=> a.last }
+      sorted_quotients.map(&:first).slice(0, no_of_seats)
+    end
+  end
+
+  #
+  # Class building the denominators for D'Hondt.
+  #
+  class DhondtDenominators
+    def self.get(size)
+      Range.new(1, size)
+    end
+  end
+
+  #
+  # Class building the denominators for modified Sainte-Lague.
+  #
+  class SainteLague14Denominators
+    def self.get(size)
+      Range.new(1, size).map { |a| a.equal?(1) ? 1.4 : a * 2 - 1 }
+    end
+  end
+end

data/lib/sapor/pseudorandom_multirange_enumerator.rb

@@ -0,0 +1,87 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2014 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+require 'prime'
+
+module Sapor
+  #
+  # A pseudo-random multi-range enumerator. It uses prime numbers to iterate
+  # in a pseudo-random way through the hyper-rectangle with dimensions
+  # provided as the constructor parameter. If a suitable set of primes can't
+  # be found, the constructor will raise an exception.
+  #
+  class PseudoRandomMultiRangeEnumerator
+    def initialize(dimensions)
+      @dimensions = dimensions
+      @incrementers = calculate_incrementers
+    end
+
+    def each
+      return enum_for { @dimensions.inject(:*) } unless block_given?
+      counters = Array.new(@dimensions.size, 0)
+      loop do
+        yield(counters.dup)
+        @incrementers.each_with_index do |incrementer, i|
+          counters[i] = (counters[i] + incrementer).modulo(@dimensions[i])
+        end
+        break if all_zeros?(counters)
+      end
+    end
+
+    private
+
+    def calculate_incrementers
+      incrementers = []
+      incrementer = 0
+      calculate_incrementer_indexes.each do |i|
+        incrementer = next_incrementer(incrementer)
+        if @dimensions[i] < incrementer
+          fail(ArgumentError, 'Could not construct suitable incrementers.')
+        end
+        incrementers[i] = incrementer
+      end
+      incrementers
+    end
+
+    def calculate_incrementer_indexes
+      @dimensions.each_with_index.sort.map(&:last)
+    end
+
+    def next_incrementer(incrementer)
+      incrementer += 1
+      while incrementer > 1 &&
+            (!prime?(incrementer) || !relative_prime?(incrementer))
+        incrementer += 1
+      end
+      incrementer
+    end
+
+    def prime?(number)
+      Prime.prime?(number)
+    end
+
+    def relative_prime?(incrementer)
+      @dimensions.map { |dimension| dimension.gcd(incrementer) }.max == 1
+    end
+
+    def all_zeros?(counters)
+      counters.max == 0
+    end
+  end
+end

data/lib/sapor/regional_data/area.rb

@@ -0,0 +1,80 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2014 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+module Sapor
+  #
+  # Superclass for any area (country, region, city, etc...) for which
+  # polls can be registered.
+  #
+  class Area
+    include Singleton
+
+    def lines_to_election_results(lines)
+      results = {}
+      lines.each_line { |line| add_line_to_election_results(results, line) }
+      results
+    end
+
+    def load_election_results(filename)
+      dirname = File.dirname(File.expand_path(__FILE__))
+      full_filename = File.expand_path(filename, dirname)
+      lines = File.read(full_filename)
+      lines_to_election_results(lines)
+    end
+
+    def summarize_election_results(election_results)
+      summary = {}
+      election_results.each_value do |local_results|
+        add_local_results_to_summary(summary, local_results)
+      end
+      summary
+    end
+
+    private
+
+    def add_line_to_election_results(results, line)
+      return if line.strip.empty? || line.strip.start_with?('#')
+      values = line.split('|')
+      fail(ArgumentError, "Broken line: #{line}.") unless values.size.equal?(3)
+      add_line_values_to_election_results(results, values)
+    end
+
+    def add_line_values_to_election_results(results, values)
+      constituency = values[0].strip
+      results[constituency] = {} unless results.key?(constituency)
+      choice = values[1].strip
+      votes = values[2].gsub(',', '').to_i
+      if results[constituency].key?(choice)
+        fail(ArgumentError, "Choice #{choice} appeared twice in " \
+                            "constituency #{constituency}.")
+      end
+      results[constituency][choice] = votes
+    end
+
+    def add_local_results_to_summary(summary, local_results)
+      local_results.each_pair do |choice, votes|
+        if summary.key?(choice)
+          summary[choice] += votes
+        else
+          summary[choice] = votes
+        end
+      end
+    end
+  end
+end

data/lib/sapor/regional_data/catalonia-2012-2015.psv

@@ -0,0 +1,100 @@
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2014 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+# Source: Argos, Archivo historico electoral, retrieved on 23 May 2015.
+# URL: http://www.argos.gva.es/ahe/index.html
+
+# Adapted from catalonia-2012.psv for the 2015 elections using the following
+# transformation rules:
+# c = Junts pel Sí / (Convergència i Unió + Esquerra Republicana de Catalunya – Catalunya Sí)
+#   = 39.2% / (30.7% + 13.7%)
+#   = 0.882882883
+# Junts pel Sí = c × (Convergència i Unió
+#                     + Esquerra Republicana de Catalunya – Catalunya Sí)
+# Unió Democràtica de Catalunya = (1 - c) × (Convergència i Unió
+#                                            + Esquerra Republicana de Catalunya
+#                                              – Catalunya Sí) 
+# Catalunya Sí que es Pot = Iniciativa per Catalunya Verds – Esquerra Unida i
+#                           Alternativa
+
+Barcelona | Junts pel Sí                                           | 979,373
+Barcelona | Partit dels Socialistes de Catalunya                   | 418,847
+Barcelona | Partit Popular de Catalunya                            | 361,656
+Barcelona | Catalunya Sí que es Pot                                | 303,625
+Barcelona | Ciutadans-Partido de la Ciudadanía                     | 229,746
+Barcelona | Unió Democràtica de Catalunya                          | 129,917
+Barcelona | Candidatura d'Unitat Popular – Alternativa d'Esquerres |  92,794
+Barcelona | Plataforma per Catalunya                               |  51,403
+Barcelona | Solidaritat Catalana per la Independència              |  32,296
+Barcelona | Escons en Blanc                                        |  22,817
+Barcelona | Partido Animalista Contra el Maltrato Animal           |  16,479
+Barcelona | Pirates de Catalunya                                   |  15,241
+Barcelona | Unió, Progrés i Democràcia                             |  12,147
+Barcelona | Farts.cat                                              |   9,185
+Barcelona | Via Democràtica                                        |   5,984
+Barcelona | Unificació Comunista d'Espanya                         |   1,998
+
+Girona    | Junts pel Sí                                           | 185,048
+Girona    | Partit dels Socialistes de Catalunya                   |  34,688
+Girona    | Partit Popular de Catalunya                            |  33,096
+Girona    | Unió Democràtica de Catalunya                          |  24,547
+Girona    | Catalunya Sí que es Pot                                |  20,397
+Girona    | Candidatura d'Unitat Popular – Alternativa d'Esquerres |  14,499
+Girona    | Ciutadans-Partido de la Ciudadanía                     |  12,341
+Girona    | Solidaritat Catalana per la Independència              |   6,011
+Girona    | Plataforma per Catalunya                               |   3,377
+Girona    | Escons en Blanc                                        |   1,946
+Girona    | Partido Animalista Contra el Maltrato Animal           |   1,565
+Girona    | Pirates de Catalunya                                   |   1,137
+Girona    | Farts.cat                                              |     718
+Girona    | Unió, Progrés i Democràcia                             |     701
+Girona    | Partit Republicà d'Esquerra – Izquierda Republicana    |     525
+Girona    | Unificació Comunista d'Espanya                         |     201
+
+Lleida    | Junts pel Sí                                           | 110,401
+Lleida    | Partit Popular de Catalunya                            |  23,338
+Lleida    | Partit dels Socialistes de Catalunya                   |  21,598
+Lleida    | Unió Democràtica de Catalunya                          |  14,645
+Lleida    | Catalunya Sí que es Pot                                |  11,145
+Lleida    | Ciutadans-Partido de la Ciudadanía                     |   6,881
+Lleida    | Candidatura d'Unitat Popular – Alternativa d'Esquerres |   6,302
+Lleida    | Solidaritat Catalana per la Independència              |   3,038
+Lleida    | Escons en Blanc                                        |   1,417
+Lleida    | Plataforma per Catalunya                               |   1,224
+Lleida    | Partido Animalista Contra el Maltrato Animal           |     752
+Lleida    | Pirates de Catalunya                                   |     642
+Lleida    | Farts.cat                                              |     476
+Lleida    | Unió, Progrés i Democràcia                             |     392
+Lleida    | Partit Republicà d'Esquerra – Izquierda Republicana    |     301
+Lleida    | Unificació Comunista d'Espanya                         |      90
+
+Tarragona | Junts pel Sí                                           | 148,104
+Tarragona | Partit Popular de Catalunya                            |  53,591
+Tarragona | Partit dels Socialistes de Catalunya                   |  48,642
+Tarragona | Ciutadans-Partido de la Ciudadanía                     |  26,039
+Tarragona | Catalunya Sí que es Pot                                |  24,538
+Tarragona | Unió Democràtica de Catalunya                          |  19,646
+Tarragona | Candidatura d'Unitat Popular – Alternativa d'Esquerres |  12,840
+Tarragona | Solidaritat Catalana per la Independència              |   5,493
+Tarragona | Plataforma per Catalunya                               |   4,103
+Tarragona | Escons en Blanc                                        |   2,108
+Tarragona | Partido Animalista Contra el Maltrato Animal           |   2,065
+Tarragona | Unió, Progrés i Democràcia                             |   1,373
+Tarragona | Farts.cat                                              |   1,323
+Tarragona | Pirates de Catalunya                                   |   1,199
+Tarragona | Socialistes i Republicans                              |     333
+Tarragona | Unificació Comunista d'Espanya                         |     287

data/lib/sapor/regional_data/catalonia-2012.psv

@@ -0,0 +1,87 @@
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2014 Filip van Laenen <f.a.vanlaenen@ieee.org>
+#
+# This file is part of SAPoR.
+#
+# SAPoR is free software: you can redistribute it and/or modify it under the
+# terms of the GNU General Public License as published by the Free Software
+# Foundation, either version 3 of the License, or (at your option) any later
+# version.
+#
+# SAPoR is distributed in the hope that it will be useful, but WITHOUT ANY
+# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+# A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+#
+# You can find a copy of the GNU General Public License in /doc/gpl.txt
+#
+
+# Source: Argos, Archivo historico electoral, retrieved on 23 May 2015.
+# URL: http://www.argos.gva.es/ahe/index.html
+
+Barcelona | Convergència i Unió                                           | 762,628
+Barcelona | Partit dels Socialistes de Catalunya                          | 418,847
+Barcelona | Partit Popular de Catalunya                                   | 361,656
+Barcelona | Esquerra Republicana de Catalunya – Catalunya Sí              | 346,662
+Barcelona | Iniciativa per Catalunya Verds – Esquerra Unida i Alternativa | 303,625
+Barcelona | Ciutadans-Partido de la Ciudadanía                            | 229,746
+Barcelona | Candidatura d'Unitat Popular – Alternativa d'Esquerres        |  92,794
+Barcelona | Plataforma per Catalunya                                      |  51,403
+Barcelona | Solidaritat Catalana per la Independència                     |  32,296
+Barcelona | Escons en Blanc                                               |  22,817
+Barcelona | Partido Animalista Contra el Maltrato Animal                  |  16,479
+Barcelona | Pirates de Catalunya                                          |  15,241
+Barcelona | Unió, Progrés i Democràcia                                    |  12,147
+Barcelona | Farts.cat                                                     |   9,185
+Barcelona | Via Democràtica                                               |   5,984
+Barcelona | Unificació Comunista d'Espanya                                |   1,998
+
+Girona    | Convergència i Unió                                           | 148,237
+Girona    | Esquerra Republicana de Catalunya – Catalunya Sí              |  61,358
+Girona    | Partit dels Socialistes de Catalunya                          |  34,688
+Girona    | Partit Popular de Catalunya                                   |  33,096
+Girona    | Iniciativa per Catalunya Verds – Esquerra Unida i Alternativa |  20,397
+Girona    | Candidatura d'Unitat Popular – Alternativa d'Esquerres        |  14,499
+Girona    | Ciutadans-Partido de la Ciudadanía                            |  12,341
+Girona    | Solidaritat Catalana per la Independència                     |   6,011
+Girona    | Plataforma per Catalunya                                      |   3,377
+Girona    | Escons en Blanc                                               |   1,946
+Girona    | Partido Animalista Contra el Maltrato Animal                  |   1,565
+Girona    | Pirates de Catalunya                                          |   1,137
+Girona    | Farts.cat                                                     |     718
+Girona    | Unió, Progrés i Democràcia                                    |     701
+Girona    | Partit Republicà d'Esquerra – Izquierda Republicana           |     525
+Girona    | Unificació Comunista d'Espanya                                |     201
+      
+Lleida    | Convergència i Unió                                           |  89,035
+Lleida    | Esquerra Republicana de Catalunya – Catalunya Sí              |  36,011
+Lleida    | Partit Popular de Catalunya                                   |  23,338
+Lleida    | Partit dels Socialistes de Catalunya                          |  21,598
+Lleida    | Iniciativa per Catalunya Verds – Esquerra Unida i Alternativa |  11,145
+Lleida    | Ciutadans-Partido de la Ciudadanía                            |   6,881
+Lleida    | Candidatura d'Unitat Popular – Alternativa d'Esquerres        |   6,302
+Lleida    | Solidaritat Catalana per la Independència                     |   3,038
+Lleida    | Escons en Blanc                                               |   1,417
+Lleida    | Plataforma per Catalunya                                      |   1,224
+Lleida    | Partido Animalista Contra el Maltrato Animal                  |     752
+Lleida    | Pirates de Catalunya                                          |     642
+Lleida    | Farts.cat                                                     |     476
+Lleida    | Unió, Progrés i Democràcia                                    |     392
+Lleida    | Partit Republicà d'Esquerra – Izquierda Republicana           |     301
+Lleida    | Unificació Comunista d'Espanya                                |      90
+
+Tarragona | Convergència i Unió                                           | 113,657
+Tarragona | Esquerra Republicana de Catalunya – Catalunya Sí              |  54,093
+Tarragona | Partit Popular de Catalunya                                   |  53,591
+Tarragona | Partit dels Socialistes de Catalunya                          |  48,642
+Tarragona | Ciutadans-Partido de la Ciudadanía                            |  26,039
+Tarragona | Iniciativa per Catalunya Verds – Esquerra Unida i Alternativa |  24,538
+Tarragona | Candidatura d'Unitat Popular – Alternativa d'Esquerres        |  12,840
+Tarragona | Solidaritat Catalana per la Independència                     |   5,493
+Tarragona | Plataforma per Catalunya                                      |   4,103
+Tarragona | Escons en Blanc                                               |   2,108
+Tarragona | Partido Animalista Contra el Maltrato Animal                  |   2,065
+Tarragona | Unió, Progrés i Democràcia                                    |   1,373
+Tarragona | Farts.cat                                                     |   1,323
+Tarragona | Pirates de Catalunya                                          |   1,199
+Tarragona | Socialistes i Republicans                                     |     333
+Tarragona | Unificació Comunista d'Espanya                                |     287