[Yt-svn] yt-commit r1426 - trunk/yt/extensions

[britton at wrangler.dreamhost.com]

Author: britton
Date: Thu Sep  3 08:54:35 2009
New Revision: 1426
URL: http://yt.spacepope.org/changeset/1426

Log:
Added get_time_outputs and get_redshift_outputs (both bool, default: True) to 
disable getting time or redshift outputs.


Modified:
   trunk/yt/extensions/EnzoSimulation.py

Modified: trunk/yt/extensions/EnzoSimulation.py
==============================================================================
--- trunk/yt/extensions/EnzoSimulation.py	(original)
+++ trunk/yt/extensions/EnzoSimulation.py	Thu Sep  3 08:54:35 2009
@@ -9,8 +9,8 @@
     Super class for performing the same operation over all data dumps in 
     a simulation from one redshift to another.
     """
-    def __init__(self,EnzoParameterFile,initial_time=None,final_time=None,initial_redshift=None,final_redshift=None,
-                 links=False,enzo_parameters={}):
+    def __init__(self, EnzoParameterFile, initial_time=None, final_time=None, initial_redshift=None, final_redshift=None,
+                 links=False, enzo_parameters={}, get_time_outputs=True, get_redshift_outputs=False):
         self.EnzoParameterFile = EnzoParameterFile
         self.enzoParameters = {}
         self.redshiftOutputs = []
@@ -82,11 +82,11 @@
             self.SimulationInitialTime = 0.0
 
         # Calculate redshifts for dt data dumps.
-        if self.enzoParameters.has_key('dtDataDump'):
+        if get_time_outputs and self.enzoParameters.has_key('dtDataDump'):
             self._CalculateTimeDumps()
 
         # Calculate times for redshift dumps.
-        if self.enzoParameters['ComovingCoordinates']:
+        if get_redshift_outputs and self.enzoParameters['ComovingCoordinates']:
             self._CalculateRedshiftDumpTimes()
 
         # Combine all data dumps.