Changeset 2403

Timestamp:

Feb 25, 2010 11:42:02 AM (10 years ago)

Author:

jwatson

Message:

Added a utility method to compute the aggregate cost for a scenario, by looping over the stage variables and querying instance variable values.

File:

• coopr.pysp/trunk/coopr/pysp/scenariotree.py (modified) (2 diffs)

coopr.pysp/trunk/coopr/pysp/scenariotree.py

@@ -355 +355 @@
    def get_scenario(self, name):
       return self._scenario_map[name]
+
+   #
+   # compute the scenario cost for the input instance, i.e.,
+   # the sum of all stage cost variables.
+   #
+   def compute_scenario_cost(self, instance):
+      aggregate_cost = 0.0
+      for stage in self._stages:
+         instance_cost_variable = instance.active_components(Var)[stage._cost_variable[0].name][stage._cost_variable[1]]()
+         print "\tStage=%20s     Cost=%10.4f" % (stage._name, instance_cost_variable)
+         aggregate_cost += instance_cost_variable
+      return aggregate_cost
 
    #
@@ -649 +661 @@
             print "\t\t" + tree_node._name
 
-         aggregate_cost = 0.0
-         for stage in self._stages:
-            instance_cost_variable = instance.active_components(Var)[stage._cost_variable[0].name][stage._cost_variable[1]]()
-            print "\tStage=%20s     Cost=%10.4f" % (stage._name, instance_cost_variable)
-            aggregate_cost += instance_cost_variable
+         aggregate_cost = self.compute_scenario_cost(instance)
          print "\tTotal scenario cost=%10.4f" % aggregate_cost
          print ""