ws_simu3.rb

require 'csv'
require './restroom'

max_frequency = 5 # 従業員は最大5回トイレにいく
facilities_per_restroom = 3
max_use_duration = 1
population_range = 10..600
max_num_of_restrooms = 1..4

max_num_of_restrooms.each do |num_of_restrooms|
    puts("start #{num_of_restrooms}")

    data = {}
    population_range.step(10).each do |population_size|
        Person.population.clear
        population_size.times {Person.population << Person.new(rand(max_frequency)+1, rand(max_use_duration)+1)}
        data[population_size] = []
        restrooms = []
        num_of_restrooms.times{restrooms << Restroom.new(facilities_per_restroom)}

        DURATION.times do |t|
            restroom_shortest_queue = restrooms.min {|n,m| n.queue.size <=> m.queue.size}
            data[population_size] << restroom_shortest_queue.queue.size

            restrooms.each { |restroom|
                queue = restroom.queue.clone
                restroom.queue.clear

                until queue.empty?
                    restroom.enter queue.shift
                end
            }

            Person.population.each do |person|
                person.frequency = (t < 270 and t < 390) ? 12 : rand(max_frequency) + 1
                if person.need_to_go?
                    restroom = restrooms.min {|a,b| a.queue.size <=> b.queue.size}
                    restroom.enter person
                end
            end
            restrooms.each {|restroom| restroom.tick}
        end
        puts("now : population_size = #{population_size}")
    end

    CSV.open("result_ws_simu3_#{num_of_restrooms}.csv", 'w') do |csv|
        lbl = []
        population_range.step(10).each {|population_size| lbl << population_size}
        csv << lbl

        DURATION.times do |t|
            row = []
            population_range.step(10).each do |population_size|
                row << data[population_size][t]
            end
            csv << row
        end

    end

end