mirror of
https://github.com/samsonjs/mystery7-simulator.git
synced 2026-03-25 09:25:47 +00:00
70 lines
2.1 KiB
Ruby
Executable file
70 lines
2.1 KiB
Ruby
Executable file
#!/usr/bin/env ruby
|
|
|
|
require 'rubygems'
|
|
require 'bundler/setup'
|
|
require 'trollop'
|
|
require './roulette'
|
|
|
|
def main
|
|
options = Trollop::options do
|
|
opt :style, "American or European", :type => String, :default => 'American'
|
|
opt :spins, "Number of spins", :short => 'n', :type => :int, :default => 60
|
|
opt :database, "Filename for results database", :type => String, :default => File.expand_path("~/Projects/Mystery7/results.sqlite")
|
|
opt :record, "Record results", :short => 'r'
|
|
opt :seed, "Seed for the RNG", :type => :int
|
|
opt :sessions, "Number of sessions to simulate", :short => 's', :type => :int, :default => 100
|
|
opt :misses, "Number of misses before sleeping", :short => 'm', :default => 4
|
|
opt :verbose, "Print stats after each spin", :short => 'v'
|
|
end
|
|
|
|
puts ">>> Simulating #{options[:sessions]} #{options[:style]} style games of #{options[:spins]} spins, sleeping after #{options[:misses]} miss#{options[:misses] > 0 ? 'es' : ''}..."
|
|
|
|
roulette = Roulette.new(options.dup)
|
|
|
|
# puts ">>> Seed: #{roulette.seed}"
|
|
|
|
overall_status = {
|
|
:net => 0,
|
|
:snakes => 0,
|
|
:wins => 0
|
|
}
|
|
|
|
options[:sessions].times do
|
|
|
|
roulette.simulate
|
|
|
|
status = {
|
|
:net => 0,
|
|
:snakes => 0,
|
|
:wins => 0
|
|
}
|
|
|
|
Roulette::Sets.each do |letter, set|
|
|
puts "# of #{letter.to_s.upcase}s: #{roulette.counts[letter]}" if options[:verbose]
|
|
set_status = roulette.set_status[letter]
|
|
puts "status: #{set_status.inspect}" if options[:verbose]
|
|
status[:net] += set_status[:net]
|
|
status[:snakes] += set_status[:snakes]
|
|
status[:wins] += set_status[:wins]
|
|
end
|
|
|
|
overall_status[:net] += status[:net]
|
|
overall_status[:snakes] += status[:snakes]
|
|
overall_status[:wins] += status[:wins]
|
|
|
|
if options[:verbose]
|
|
puts "Net profit: #{status[:net]}"
|
|
puts "Wins: #{status[:wins]}"
|
|
puts "Snakes: #{status[:snakes]}"
|
|
end
|
|
|
|
end
|
|
|
|
puts "Net profit: #{overall_status[:net]}"
|
|
puts "Wins: #{overall_status[:wins]}"
|
|
puts "Snakes: #{overall_status[:snakes]}"
|
|
|
|
puts ">>> Results are in #{options[:database]}." if options[:record]
|
|
end
|
|
|
|
main if __FILE__ == $0
|