sapor 0.3.6

data/lib/sapor/many_past_the_post.rb

@@ -0,0 +1,113 @@
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2020 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 an electoral system with more than one
+  # district where parties only have a limited number of candidates in each of
+  # the constituencies.
+  #
+  class ManyPastThePost
+    def initialize(last_election_result, last_detailed_election_result,
+                   seat_distribution, caps = {}, weights = {})
+      @last_election_result = last_election_result
+      @last_detailed_election_result = last_detailed_election_result
+      @seat_distribution = seat_distribution
+      @caps = caps
+      @weights = weights
+    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]
+        caps = @caps.empty? ? {} : @caps[name]
+        seats = local_seats(no_of_seats, local_last_result, multiplicators,
+                            caps, @weights)
+        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|
+        local_votes[choice] = if multiplicators.key?(choice)
+                                votes * multiplicators[choice]
+                              else
+                                votes
+                              end
+      end
+      local_votes
+    end
+
+    def local_quotients(local_votes, no_of_seats, caps, weights)
+      local_quotients = []
+      local_votes.each_pair do |choice, new_value|
+        actual_no_of_seats = caps.empty? ? no_of_seats : [caps[choice], no_of_seats].min
+        actual_weights = weights[actual_no_of_seats - 1]
+        actual_weights.each do |w|
+          local_quotients << [choice, new_value.to_f * w]
+        end
+      end
+      local_quotients
+    end
+
+    def local_seats(no_of_seats, local_last_result, multiplicators, caps,
+                    weights)
+      local_votes = local_votes(local_last_result, multiplicators)
+      local_quotients = local_quotients(local_votes, no_of_seats, caps, weights)
+      sorted_quotients = local_quotients.sort { |a, b| b.last <=> a.last }
+      sorted_quotients.map(&:first).slice(0, no_of_seats)
+    end
+  end
+end

data/lib/sapor/multi_district_leveled_proportional.rb

@@ -0,0 +1,64 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2020 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 multi-district proportional electoral system with a
+  # second round of leveling seats.
+  #
+  class MultiDistrictLeveledProportional
+    def initialize(last_election_result, last_detailed_election_result,
+                   seat_distribution, leveling_seats, leveling_threshold,
+                   denominators_class)
+      @proportional = MultiDistrictProportional.new(last_election_result,
+                                                    last_detailed_election_result,
+                                                    seat_distribution, denominators_class)
+      @leveling_seats = leveling_seats
+      @leveling_threshold = leveling_threshold
+      @denominators_class = denominators_class
+    end
+
+    def project(simulation)
+      result = @proportional.project(simulation)
+      threshold = @leveling_threshold * simulation.values.inject(:+)
+      quotients = []
+      simulation.each_pair do |choice, votes|
+        next if votes < threshold
+        denominators(result[choice]).each do |d|
+          quotients << [choice, votes.to_f / d]
+        end
+      end
+      seats = quotients.sort { |a, b| b.last <=> a.last }.map(&:first).slice(0, @leveling_seats)
+      seats.each do |seat|
+        if result.key?(seat)
+          result[seat] += 1
+        else
+          result[seat] = 1
+        end
+      end
+      result
+    end
+
+    private
+
+    def denominators(seats)
+      @denominators_class.get(seats + @leveling_seats).reverse_each.take(@leveling_seats)
+    end
+  end
+end

data/lib/sapor/multi_district_proportional.rb

@@ -0,0 +1,123 @@
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2020 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 with more than one
+  # district.
+  #
+  class MultiDistrictProportional
+    def initialize(last_election_result, last_detailed_election_result,
+                   seat_distribution, denominators_class, threshold = 0,
+                   national_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
+      @national_threshold = national_threshold
+    end
+
+    def project(simulation)
+      allowed_parties = []
+      unless @national_threshold.zero?
+        threshold = @national_threshold * simulation.values.inject(:+)
+        simulation.each_pair do |choice, votes|
+          allowed_parties << choice if votes >= threshold
+        end
+      end
+      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,
+                            allowed_parties)
+        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|
+        local_votes[choice] = if multiplicators.key?(choice)
+                                votes * multiplicators[choice]
+                              else
+                                votes
+                              end
+      end
+      local_votes
+    end
+
+    def local_quotients(local_votes, local_threshold, no_of_seats,
+                        allowed_parties)
+      local_quotients = []
+      local_votes.each_pair do |choice, new_value|
+        next unless allowed_parties.empty? || allowed_parties.include?(choice)
+        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,
+                    allowed_parties)
+      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, allowed_parties)
+      sorted_quotients = local_quotients.sort { |a, b| b.last <=> a.last }
+      sorted_quotients.map(&:first).slice(0, no_of_seats)
+    end
+  end
+end

data/lib/sapor/multi_district_variable_threshold_proportional.rb

@@ -0,0 +1,128 @@
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2020 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 with more than one
+  # district.
+  #
+  class MultiDistrictVariableThresholdProportional
+    def initialize(last_election_result, last_detailed_election_result,
+                   seat_distribution, denominators_class,
+                   party_list_threshold, coalition_list_threshold,
+                   coalition_lists, minority_lists)
+      @last_election_result = last_election_result
+      @last_detailed_election_result = last_detailed_election_result
+      @seat_distribution = seat_distribution
+      @denominators_class = denominators_class
+      @party_list_threshold = party_list_threshold
+      @coalition_list_threshold = coalition_list_threshold
+      @coalition_lists = coalition_lists
+      @minority_lists = minority_lists
+    end
+
+    def project(simulation)
+      allowed_parties = []
+      total_votes = simulation.values.inject(:+)
+      party_list_threshold = @party_list_threshold * total_votes
+      coalition_list_threshold = @coalition_list_threshold * total_votes
+      simulation.each_pair do |choice, votes|
+        next unless @minority_lists.include?(choice) ||
+                    @coalition_lists.include?(choice) &&
+                    votes >= coalition_list_threshold ||
+                    !@coalition_lists.include?(choice) &&
+                    votes >= party_list_threshold
+        allowed_parties << choice
+      end
+      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,
+                            allowed_parties)
+        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|
+        local_votes[choice] = if multiplicators.key?(choice)
+                                votes * multiplicators[choice]
+                              else
+                                votes
+                              end
+      end
+      local_votes
+    end
+
+    def local_quotients(local_votes, no_of_seats, allowed_parties)
+      local_quotients = []
+      local_votes.each_pair do |choice, new_value|
+        next unless allowed_parties.empty? || allowed_parties.include?(choice)
+        @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,
+                    allowed_parties)
+      local_votes = local_votes(local_last_result, multiplicators)
+      local_quotients = local_quotients(local_votes, no_of_seats,
+                                        allowed_parties)
+      sorted_quotients = local_quotients.sort { |a, b| b.last <=> a.last }
+      sorted_quotients.map(&:first).slice(0, no_of_seats)
+    end
+  end
+end

data/lib/sapor/number_formatter.rb

@@ -0,0 +1,45 @@
+# encoding: utf-8
+#
+# Statistical Analysis of Polling Results (SAPoR)
+# Copyright (C) 2020 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
+  #
+  # Module to format numbers.
+  #
+  module NumberFormatter
+    def three_digits_percentage(number)
+      if number >= 0.9995 || number < 0.000005
+        sprintf('%.0f', number * 100) + '%'
+      elsif number >= 0.09995
+        sprintf('%.1f', number * 100) + '%'
+      elsif number >= 0.009995
+        sprintf('%.2f', number * 100) + '%'
+      else
+        sprintf('%.3f', number * 100) + '%'
+      end
+    end
+
+    def six_char_percentage(number)
+      sprintf('%5.1f', number * 100) + '%'
+    end
+
+    def with_thousands_separator(integer)
+      integer.to_s.reverse.gsub(/...(?=.)/, '\&,').reverse
+    end
+  end
+end