make backends registerable

leejjoon · leejjoon · commit a8b8c27d7e31 · 2010-02-03T19:41:53.000Z
svn path=/trunk/matplotlib/; revision=8108
diff --git a/lib/matplotlib/backend_bases.py b/lib/matplotlib/backend_bases.py
@@ -40,6 +40,15 @@
 import matplotlib.textpath as textpath
 from matplotlib.path import Path
 
+
+
+_backend_d = {}
+
+def register_backend(format, backend_class):
+    _backend_d[format] = backend_class
+
+
+
 class RendererBase:
     """An abstract base class to handle drawing/rendering operations.
 
@@ -1518,6 +1527,33 @@ def get_supported_filetypes_grouped(self):
             groupings[name].sort()
         return groupings
 
+
+    def _get_print_method(self, format):
+        method_name = 'print_%s' % format
+
+        # check for registered backends
+        if format in _backend_d:
+            backend_class = _backend_d[format]
+            
+            def _print_method(*args, **kwargs):
+                backend = self.switch_backends(backend_class)
+                print_method = getattr(backend, method_name)
+                return print_method(*args, **kwargs)
+
+            return _print_method
+            
+        if (format not in self.filetypes or
+            not hasattr(self, method_name)):
+            formats = self.filetypes.keys()
+            formats.sort()
+            raise ValueError(
+                'Format "%s" is not supported.\n'
+                'Supported formats: '
+                '%s.' % (format, ', '.join(formats)))
+
+        return getattr(self, method_name)
+
+
     def print_figure(self, filename, dpi=None, facecolor='w', edgecolor='w',
                      orientation='portrait', format=None, **kwargs):
         """
@@ -1573,16 +1609,8 @@ def print_figure(self, filename, dpi=None, facecolor='w', edgecolor='w',
                     filename = filename.rstrip('.') + '.' + format
         format = format.lower()
 
-        method_name = 'print_%s' % format
-        if (format not in self.filetypes or
-            not hasattr(self, method_name)):
-            formats = self.filetypes.keys()
-            formats.sort()
-            raise ValueError(
-                'Format "%s" is not supported.\n'
-                'Supported formats: '
-                '%s.' % (format, ', '.join(formats)))
-
+        print_method = self._get_print_method(format)
+        
         if dpi is None:
             dpi = rcParams['savefig.dpi']
 
@@ -1609,7 +1637,8 @@ def print_figure(self, filename, dpi=None, facecolor='w', edgecolor='w',
                 # the backend to support file-like object, i'm going
                 # to leave it as it is. However, a better solution
                 # than stringIO seems to be needed. -JJL
-                result = getattr(self, method_name)(
+                #result = getattr(self, method_name)(
+                result = print_method(
                     cStringIO.StringIO(),
                     dpi=dpi,
                     facecolor=facecolor,
@@ -1642,7 +1671,8 @@ def print_figure(self, filename, dpi=None, facecolor='w', edgecolor='w',
             _bbox_inches_restore = None
 
         try:
-            result = getattr(self, method_name)(
+            #result = getattr(self, method_name)(
+            result = print_method(
                 filename,
                 dpi=dpi,
                 facecolor=facecolor,
diff --git a/lib/matplotlib/tight_bbox.py b/lib/matplotlib/tight_bbox.py
@@ -6,6 +6,7 @@
 from matplotlib.transforms import Bbox, TransformedBbox, Affine2D
 
 
+
 def adjust_bbox(fig, format, bbox_inches):
     """
     Temporarily adjust the figure so that only the specified area
@@ -46,11 +47,9 @@ def restore_bbox():
         fig.transFigure.invalidate()
         fig.patch.set_bounds(0, 0, 1, 1)
 
-    if format in ["png", "raw", "rgba"]:
-        adjust_bbox_png(fig, bbox_inches)
-        return restore_bbox
-    elif format in ["pdf", "eps", "svg", "svgz"]:
-        adjust_bbox_pdf(fig, bbox_inches)
+    adjust_bbox_handler = _adjust_bbox_handler_d.get(format)
+    if adjust_bbox_handler is not None:
+        adjust_bbox_handler(fig, bbox_inches)
         return restore_bbox
     else:
         warnings.warn("bbox_inches option for %s backend is not implemented yet." % (format))
@@ -125,3 +124,8 @@ def process_figure_for_rasterizing(figure,
     return bbox_inches, r
 
 
+_adjust_bbox_handler_d = {}
+for format in ["png", "raw", "rgba"]:
+    _adjust_bbox_handler_d[format] = adjust_bbox_png
+for format in ["pdf", "eps", "svg", "svgz"]:
+    _adjust_bbox_handler_d[format] = adjust_bbox_pdf
