minor tweaks on axes_grid1 to improve the usability

leejjoon · leejjoon · commit 5d628a6a38e3 · 2010-04-20T16:43:42.000Z
svn path=/trunk/matplotlib/; revision=8246
diff --git a/lib/mpl_toolkits/axes_grid1/__init__.py b/lib/mpl_toolkits/axes_grid1/__init__.py
@@ -3,3 +3,5 @@
      make_axes_locatable
 from axes_grid import Grid, ImageGrid, AxesGrid
 #from axes_divider import make_axes_locatable
+
+from parasite_axes import host_subplot, host_axes
diff --git a/lib/mpl_toolkits/axes_grid1/axes_divider.py b/lib/mpl_toolkits/axes_grid1/axes_divider.py
@@ -501,7 +501,8 @@ def new_vertical(self, size, pad=None, pack_start=False, **kwargs):
         return ax
 
 
-    def append_axes(self, position, size, pad=None, **kwargs):
+    def append_axes(self, position, size, pad=None, add_to_figure=True,
+                    **kwargs):
         """
         create an axes at the given *position* with the same height
         (or width) of the main axes.
@@ -523,7 +524,8 @@ def append_axes(self, position, size, pad=None, **kwargs):
         else:
             raise ValueError("the position must be one of left, right, bottom, or top")
 
-        self._fig.add_axes(ax)
+        if add_to_figure:
+            self._fig.add_axes(ax)
         return ax
 
     def get_aspect(self):
diff --git a/lib/mpl_toolkits/axes_grid1/parasite_axes.py b/lib/mpl_toolkits/axes_grid1/parasite_axes.py
@@ -505,11 +505,30 @@ def _get_base_axes_attr(self, attrname):
     return new_class
 
 def host_subplot_class_factory(axes_class):
-    host_axes = host_axes_class_factory(axes_class=Axes)
-    subplot_host = subplot_class_factory(HostAxes)
-    return subplot_host
+    host_axes_class = host_axes_class_factory(axes_class=axes_class)
+    subplot_host_class = subplot_class_factory(host_axes_class)
+    return subplot_host_class
 
 HostAxes = host_axes_class_factory(axes_class=Axes)
 SubplotHost = subplot_class_factory(HostAxes)
 
 
+def host_axes(*args, **kwargs):
+    import matplotlib.pyplot as plt
+    axes_class = kwargs.pop("axes_class", None)
+    host_axes_class = host_axes_class_factory(axes_class)
+    fig = plt.gcf()
+    ax = host_axes_class(fig, *args, **kwargs)
+    fig.add_axes(ax)
+    plt.draw_if_interactive()
+    return ax
+
+def host_subplot(*args, **kwargs):
+    import matplotlib.pyplot as plt
+    axes_class = kwargs.pop("axes_class", None)
+    host_subplot_class = host_subplot_class_factory(axes_class)
+    fig = plt.gcf()
+    ax = host_subplot_class(fig, *args, **kwargs)
+    fig.add_subplot(ax)
+    plt.draw_if_interactive()
+    return ax
diff --git a/lib/mpl_toolkits/axisartist/__init__.py b/lib/mpl_toolkits/axisartist/__init__.py
@@ -19,3 +19,4 @@
 
 SubplotHost = subplot_class_factory(HostAxes)
 
+
diff --git a/lib/mpl_toolkits/axisartist/axislines.py b/lib/mpl_toolkits/axisartist/axislines.py
@@ -459,7 +459,7 @@ def new_fixed_axis(self, loc,
                        offset=None,
                        axes=None,
                        ):
-
+ 
         if axes is None:
             warnings.warn("'new_fixed_axis' explicitly requires the axes keyword.")
             axes = self.axes
@@ -697,6 +697,17 @@ def invalidate_grid_helper(self):
         self._grid_helper.invalidate()
 
 
+    def new_fixed_axis(self, loc, offset=None):
+        gh = self.get_grid_helper()
+        axis = gh.new_fixed_axis(loc,
+                                 nth_coord=None,
+                                 axis_direction=None,
+                                 offset=offset,
+                                 axes=self,
+                                 )
+        return axis
+
+
     def new_floating_axis(self, nth_coord, value,
                           axis_direction="bottom",
                           ):
@@ -758,7 +769,6 @@ def get_tightbbox(self, renderer):
 
 Subplot = maxes.subplot_class_factory(Axes)
 
-
 class AxesZero(Axes):
     def __init__(self, *kl, **kw):
 

Original file line number	Diff line number	Diff line change
`@@ -19,3 +19,4 @@`
`19`	`19`
`20`	`20`	`SubplotHost = subplot_class_factory(HostAxes)`
`21`	`21`
	`22`	`+`