Class: ERV::DiscreteUniformDistribution

Inherits:
Distribution show all
Defined in:
lib/erv/discrete_uniform_distribution.rb

Instance Method Summary collapse

Methods inherited from Distribution

#sample

Constructor Details

#initialize(opts) ⇒ DiscreteUniformDistribution

Returns a new instance of DiscreteUniformDistribution.

Raises:

  • (ArgumentError) 


13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/erv/discrete_uniform_distribution.rb', line 13

def initialize(opts)
  super(opts)

  raise ArgumentError unless opts[:max_value]
  max = opts[:max_value].to_i
  min = opts[:min_value].try(:to_i) || 0

  if RUBY_PLATFORM == 'java'
    # create distribution
    d = UniformIntegerDistribution.new(@rng, min, max)
    # setup sampling function
    @func = Proc.new { d.sample }
  else
    # setup sampling function
    @func = Proc.new { min + @rng.uniform_int(max-min) }
  end
end