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

[skillman at wrangler.dreamhost.com]

Author: skillman
Date: Fri Jul  3 15:58:30 2009
New Revision: 1361
URL: http://yt.spacepope.org/changeset/1361

Log:
In case the user wants to do additional actions on the hierachy not in
round_robin mode, save_data function is reset to round_robin=False
after halo projections are made.  Also, a definition of
_finalize_parallel was moved back to the correct location so that all
the virial quantities are written out.  --SWS 7/3/09



Modified:
   trunk/yt/extensions/HaloProfiler.py

Modified: trunk/yt/extensions/HaloProfiler.py
==============================================================================
--- trunk/yt/extensions/HaloProfiler.py	(original)
+++ trunk/yt/extensions/HaloProfiler.py	Fri Jul  3 15:58:30 2009
@@ -96,9 +96,6 @@
 
     def makeProfiles(self):
         "Make radial profiles for all halos on the list."
-        def _finalize_parallel(self):
-            self.virialQuantities = self._mpi_catlist(self.virialQuantities)
-            self.virialQuantities.sort(key = lambda a:a['id'])
 
         # Get halo(s).
         if self.halos is 'single':
@@ -183,6 +180,10 @@
 
         self._WriteVirialQuantities()
 
+    def _finalize_parallel(self):
+        self.virialQuantities = self._mpi_catlist(self.virialQuantities)
+        self.virialQuantities.sort(key = lambda a:a['id'])
+
     @lagos.parallel_root_only
     def __check_directory(self, outputDir):
         if (os.path.exists(outputDir)):
@@ -289,6 +290,7 @@
                 pc.clear_plots()
             del region
         del pc
+        self.pf.hierarchy._reset_save_data(round_robin=False)
 
     @lagos.parallel_root_only
     def _WriteVirialQuantities(self):