[Yt-svn] yt-commit r570 - trunk/yt/lagos

[mturk at wrangler.dreamhost.com]

Author: mturk
Date: Tue Jun 17 11:33:22 2008
New Revision: 570
URL: http://yt.spacepope.org/changeset/570

Log:
Added 'TotalMass' quantity.



Modified:
   trunk/yt/lagos/DerivedQuantities.py

Modified: trunk/yt/lagos/DerivedQuantities.py
==============================================================================
--- trunk/yt/lagos/DerivedQuantities.py	(original)
+++ trunk/yt/lagos/DerivedQuantities.py	Tue Jun 17 11:33:22 2008
@@ -99,6 +99,15 @@
     def keys(self):
         return self.functions.keys()
 
+def _TotalMass(data):
+    baryon_mass = data["CellMassMsun"].sum()
+    particle_mass = data["ParticleMassMsun"].sum()
+    return baryon_mass, particle_mass
+def _combTotalMass(data, baryon_mass, particle_mass):
+    return baryon_mass.sum() + particle_mass.sum()
+add_quantity("TotalMass", function=_TotalMass,
+             combine_function=_combTotalMass, n_ret = 2)
+
 def _CenterOfMass(data):
     x = (data["x"] * data["CellMassMsun"]).sum()
     y = (data["y"] * data["CellMassMsun"]).sum()