Changeset 2990 for coopr.pysp/trunk/coopr/pysp/ef_writer_script.py

Timestamp:

Sep 4, 2010 1:28:07 PM (10 years ago)

Author:

jwatson

Message:

Added option to PySP scripts to allow for output of the scenario tree solutions in CSV file format. Option is --solution-writer=coopr.pysp.csvsolutionwriter.py. This also required some overdue refactoring of the PySP code, plus re-work of how the post-PH extensive form solution is displayed.

File:

• coopr.pysp/trunk/coopr/pysp/ef_writer_script.py (modified) (5 diffs)

coopr.pysp/trunk/coopr/pysp/ef_writer_script.py

@@ -29 +29 @@
 from coopr.opt.parallel import SolverManagerFactory
 
+from pyutilib.component.core import ExtensionPoint
+from coopr.pysp.solutionwriter import ISolutionWriterExtension
+
 #
 # utility method to construct an option parser for ef writer arguments
@@ -105 +108 @@
                      type="float",
                      default=None)   
+   parser.add_option("--solution-writer",
+                     help="The plugin invoked to write the scenario tree solution. Defaults to the empty list.",
+                     action="append",
+                     dest="solution_writer",
+                     type="string",
+                     default = [])
    parser.add_option("--output-solver-log",
                      help="Output solver log during the extensive form solve.",
@@ -142 +151 @@
       cvar_weight = options.cvar_weight
       risk_alpha = options.risk_alpha
+
+   # validate the solution writer plugin exists, to avoid a lot of wasted work.
+   for solution_writer_name in options.solution_writer:
+      print "Trying to import solution writer="+solution_writer_name
+      __import__(solution_writer_name)
+      print "Module successfully loaded"
 
    scenario_tree, binding_instance, scenario_instances = write_ef_from_scratch(os.path.expanduser(options.model_directory),
@@ -184 +199 @@
       load_ef_solution(ef_results, binding_instance, scenario_instances)
       scenario_tree.snapshotSolutionFromInstances(scenario_instances)
+
+      # handle output of solution from the scenario tree.
       print ""
       print "Extensive form solution:"
@@ -190 +207 @@
       print "Extensive form costs:"
       scenario_tree.pprintCosts(scenario_instances)
+
+      solution_writer_plugins = ExtensionPoint(ISolutionWriterExtension)
+      for plugin in solution_writer_plugins:
+         plugin.write(scenario_tree, "ef")
 
 def run(args=None):