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

[mturk at wrangler.dreamhost.com]

Author: mturk
Date: Fri Jan 18 17:01:21 2008
New Revision: 365
URL: http://yt.spacepope.org/changeset/365

Log:
Important fix for situations where the weighting field is a very small number
(much less than one.)



Modified:
   trunk/yt/lagos/Profiles.py

Modified: trunk/yt/lagos/Profiles.py
==============================================================================
--- trunk/yt/lagos/Profiles.py	(original)
+++ trunk/yt/lagos/Profiles.py	Fri Jan 18 17:01:21 2008
@@ -78,9 +78,10 @@
                     weight_data[field] += w
                     used = (used | u)
                 grid.clear_data()
+            ub = na.where(used)
             for field in fields:
                 if weight:
-                    data[field] /= weight_data[field]
+                    data[field][ub] /= weight_data[field][ub]
                 self[field] = data[field]
             self["UsedBins"] = used
 
@@ -200,7 +201,7 @@
             if weight: weight_data = source[weight].ravel()
         self.total_stuff = source_data.sum()
         binned_field = self._get_empty_field()
-        weight_field = na.ones(binned_field.shape,dtype='float64')
+        weight_field = self._get_empty_field()
         used_field = self._get_empty_field()
         bin_indices_x = args[0].ravel()
         bin_indices_y = args[1].ravel()