[Yt-svn] yt-commit r1647 - branches/yt-1.6/yt/raven

[joishi at wrangler.dreamhost.com]

Author: joishi
Date: Thu Feb 25 16:04:46 2010
New Revision: 1647
URL: http://yt.enzotools.org/changeset/1647

Log:
forgot to add the save_to_pdf bit to PlotTypes for new save_book.


Modified:
   branches/yt-1.6/yt/raven/PlotTypes.py

Modified: branches/yt-1.6/yt/raven/PlotTypes.py
==============================================================================
--- branches/yt-1.6/yt/raven/PlotTypes.py	(original)
+++ branches/yt-1.6/yt/raven/PlotTypes.py	Thu Feb 25 16:04:46 2010
@@ -39,6 +39,12 @@
     else:
         from matplotlib.backends.backend_agg \
                 import FigureCanvasAgg as FigureCanvas
+    try:
+        from matplotlib.backends.backend_pdf \
+                import FigureCanvasPdf as FigureCanvasPDF
+        engineVals["canvas_pdf"] = FigureCanvasPDF
+    except ImportError:
+        pass
     engineVals["canvas"] = FigureCanvas
     return
 
@@ -125,6 +131,23 @@
         self["GeneratedAt"] = self.data.pf["CurrentTimeIdentifier"]
         return fn
 
+    def save_to_pdf(self, f):
+        self._redraw_image()
+        canvas = engineVals["canvas_pdf"](self._figure)
+        original_figure_alpha = self._figure.patch.get_alpha()
+        self._figure.patch.set_alpha(0.0)
+        original_axes_alpha = []
+        for ax in self._figure.axes:
+            patch = ax.patch
+            original_axes_alpha.append(patch.get_alpha())
+            patch.set_alpha(0.0)
+
+        canvas.print_pdf(f)
+
+        self._figure.set_alpha(original_figure_alpha)
+        for ax, alpha in zip(self._figure.axes,original_axes_alpha):
+            ax.patch.set_alpha(alpha)
+
     def _redraw_image(self):
         pass