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

[britton at wrangler.dreamhost.com]

Author: britton
Date: Wed Sep  9 18:21:00 2009
New Revision: 1441
URL: http://yt.spacepope.org/changeset/1441

Log:
Added field, star_density, to get density of star particles using cic_deposit.


Modified:
   trunk/yt/lagos/EnzoFields.py

Modified: trunk/yt/lagos/EnzoFields.py
==============================================================================
--- trunk/yt/lagos/EnzoFields.py	(original)
+++ trunk/yt/lagos/EnzoFields.py	Wed Sep  9 18:21:00 2009
@@ -210,6 +210,20 @@
 add_field("particle_density", function=_pdensity,
           validators=[ValidateSpatial(0)], convert_function=_convertDensity)
 
+def _spdensity(field, data):
+    blank = na.zeros(data.ActiveDimensions, dtype='float32', order="FORTRAN")
+    if data.NumberOfParticles == 0: return blank
+    filter = data['creation_time'] > 0.0
+    if not filter.any(): return blank
+    cic_deposit.cic_deposit(data["particle_position_x"][filter],
+                            data["particle_position_y"][filter],
+                            data["particle_position_z"][filter], 3,
+                            data["particle_mass"][filter],
+                            blank, data.LeftEdge, data['dx'])
+    return blank
+add_field("star_density", function=_spdensity,
+          validators=[ValidateSpatial(0)], convert_function=_convertDensity)
+
 EnzoFieldInfo["Temperature"].units = r"K"
 
 #