Changeset 2427

Timestamp:

Mar 11, 2010 10:06:30 PM (9 years ago)

Author:

jwatson

Message:

Added logic to automatically disable and re-enable garbage collection at various sane points in the PySP extensive form writer, in order to (noticeably) improve performance.

File:

• coopr.pysp/trunk/coopr/pysp/ef.py (modified) (3 diffs)

coopr.pysp/trunk/coopr/pysp/ef.py

@@ -5 +5 @@
 import traceback
 import copy
+import gc
 
 from coopr.pysp.scenariotree import *
@@ -575 +576 @@
    # construct instances for each scenario
    #
+
+   # the construction of instances takes little overhead in terms of
+   # memory potentially lost in the garbage-collection sense (mainly
+   # only that due to parsing and instance simplification/prep-processing).
+   # to speed things along, disable garbage collection if it enabled in
+   # the first place through the instance construction process.
+   # IDEA: If this becomes too much for truly large numbers of scenarios,
+   #       we could manually collect every time X instances have been created.
+
+   re_enable_gc = False
+   if gc.isenabled() is True:
+      re_enable_gc = True
+      gc.disable()
 
    scenario_instances = {}
@@ -617 +631 @@
       scenario_instances[scenario._name] = scenario_instance
 
+   if re_enable_gc is True:
+      gc.enable()
+
    print ""