lib/rubyonacid/factory.rb in rubyonacid-0.3.1 vs lib/rubyonacid/factory.rb in rubyonacid-0.4.0

@@ -2,14 +2,27 @@
 
 #The parent class for all other Factories.
 #Should not normally be instantiated directly.
 class Factory
 
-  def initialize(*args)
+  #An array of factories to be queried by get_unit and have their return values averaged.
+  attr_accessor :source_factories
+
+  #Takes a hash with these keys and defaults:
+  #  :source_factories => []
+  def initialize(options = {})
     @minimums = {}
     @maximums = {}
+    @source_factories = options[:source_factories] || []
   end
+  
+  #Calls #get_unit(key) on each source factory and averages results.
+  def get_unit(key)
+    return nil if source_factories.empty?
+    values = source_factories.map{|factory| factory.get_unit(key)}
+    average = values.inject(0.0){|sum, value| sum += value} / source_factories.size
+  end
 
   #Calls get_unit with key to get value between 0.0 and 1.0, then converts that value to be between given minimum and maximum.
   def within(key, minimum, maximum)
     get_unit(key) * (maximum - minimum) + minimum
   end
@@ -32,10 +45,9 @@
     all_choices = choices.flatten
     index = (get_unit(key) * all_choices.length).floor
     index = all_choices.length - 1 if index > all_choices.length - 1
     all_choices[index]
   end
-  
   
 end
 
 end
\ No newline at end of file