[Yt-svn] yt-commit r974 - trunk/yt/raven

[mturk at wrangler.dreamhost.com]

Author: mturk
Date: Wed Dec  3 13:41:32 2008
New Revision: 974
URL: http://yt.spacepope.org/changeset/974

Log:
Now you can feed data_source to add_projection and it will respect that.



Modified:
   trunk/yt/raven/PlotCollection.py

Modified: trunk/yt/raven/PlotCollection.py
==============================================================================
--- trunk/yt/raven/PlotCollection.py	(original)
+++ trunk/yt/raven/PlotCollection.py	Wed Dec  3 13:41:32 2008
@@ -265,12 +265,13 @@
     def _add_projection(self, ptype, field, axis, weight_field=None,
                       center=None, use_colorbar=True,
                       figure = None, axes = None, fig_size=None,
-                      periodic = False, **kwargs):
+                      periodic = False, data_source = None, **kwargs):
         if center == None:
             center = self.c
-        proj = self.pf.hierarchy.proj(axis, field, weight_field, center=center,
-                                      **kwargs)
-        p = self._add_plot(ptype(proj, field,
+        if data_source is None:
+            data_source = self.pf.hierarchy.proj(axis, field, weight_field,
+                                center=center, **kwargs)
+        p = self._add_plot(ptype(data_source, field,
                          use_colorbar=use_colorbar, axes=axes, figure=figure,
                          size=fig_size, periodic=periodic))
         p["Axis"] = lagos.axis_names[axis]
@@ -435,6 +436,7 @@
         retval._redraw_image()
         retval._fig_num = new_fig.number
         self.pylab.show()
+        self.pylab.draw()
         return retval
     return pylabify
 
@@ -442,7 +444,11 @@
     @wraps(func)
     def pylabify(self, *args, **kwargs):
         retval = func(self, *args, **kwargs)
-        self.pylab.show()
+        fig_num = self.pylab.gcf().number
+        for p in self.plots:
+            self.pylab.figure(p._fig_num)
+            self.pylab.draw()
+        self.pylab.figure(fig_num)
         return retval
     return pylabify