add AxisArtist.invert_ticklabel_direction and fix axis label padding in axes_grid

leejjoon · leejjoon · commit 43f8a743385d · 2010-03-02T01:47:54.000Z
svn path=/trunk/matplotlib/; revision=8167
diff --git a/lib/mpl_toolkits/axes_grid/axis_artist.py b/lib/mpl_toolkits/axes_grid/axis_artist.py
@@ -389,6 +389,13 @@ def _set_offset_radius(self, offset_radius):
     def _get_offset_radius(self):
         return self._offset_radius
 
+
+    _get_opposite_direction = {"left":"right",
+                               "right":"left",
+                               "top":"bottom",
+                               "bottom":"top"}.__getitem__
+
+
     def _update(self, renderer):
         pass
 
@@ -665,6 +672,11 @@ def set_axis_direction(self, label_direction):
         self.set_default_alignment(label_direction)
         self.set_default_angle(label_direction)
 
+
+    def invert_axis_direction(self):
+        label_direction = self._get_opposite_direction(self._axis_direction)
+        self.set_axis_direction(label_direction)
+
     def _get_ticklabels_offsets(self, renderer, label_direction):
         """
         Calculates the offsets of the ticklabels from the tick and
@@ -1021,6 +1033,15 @@ def set_ticklabel_direction(self, tick_direction):
         else:
             self._ticklabel_add_angle = 0
 
+    def invert_ticklabel_direction(self):
+        self._ticklabel_add_angle = (self._ticklabel_add_angle + 180) % 360
+        self.major_ticklabels.invert_axis_direction()
+        self.minor_ticklabels.invert_axis_direction()
+
+    # def invert_ticks_direction(self):
+    #     self.major_ticks.set_tick_out(not self.major_ticks.get_tick_out())
+    #     self.minor_ticks.set_tick_out(not self.minor_ticks.get_tick_out())
+
     def set_axislabel_direction(self, label_direction):
         """
         Adjust the direction of the axislabel.
@@ -1190,9 +1211,10 @@ def _draw_ticks(self, renderer):
         # set extra pad for major and minor ticklabels:
         # use ticksize of majorticks even for minor ticks. not clear what is best.
 
+        dpi_cor = renderer.points_to_pixels(1.)
         if self.major_ticks.get_visible() and self.major_ticks.get_tick_out():
-            self.major_ticklabels._set_external_pad(self.major_ticks._ticksize)
-            self.minor_ticklabels._set_external_pad(self.major_ticks._ticksize)
+            self.major_ticklabels._set_external_pad(self.major_ticks._ticksize*dpi_cor)
+            self.minor_ticklabels._set_external_pad(self.major_ticks._ticksize*dpi_cor)
         else:
             self.major_ticklabels._set_external_pad(0)
             self.minor_ticklabels._set_external_pad(0)
@@ -1301,8 +1323,19 @@ def _draw_label(self, renderer):
 
         #pad_points = self.major_tick_pad
 
-        axislabel_pad = max([self.major_ticklabels._axislabel_pad,
-                             self.minor_ticklabels._axislabel_pad])
+        #print self._ticklabel_add_angle - self._axislabel_add_angle
+        #if abs(self._ticklabel_add_angle - self._axislabel_add_angle)%360 > 90:
+        if self._ticklabel_add_angle !=  self._axislabel_add_angle:
+            if (self.major_ticks.get_visible() and not self.major_ticks.get_tick_out()) \
+               or \
+               (self.minor_ticks.get_visible() and not self.major_ticks.get_tick_out()):
+                axislabel_pad = self.major_ticks._ticksize
+            else:
+                axislabel_pad = 0
+        else:
+            axislabel_pad = max([self.major_ticklabels._axislabel_pad,
+                                 self.minor_ticklabels._axislabel_pad])
+
 
         #label_offset =  axislabel_pad + self.LABELPAD
 
@@ -1341,6 +1374,7 @@ def draw(self, renderer):
 
 
         self._draw_line(renderer)
+
         self._draw_ticks(renderer)
 
         #self._draw_offsetText(renderer)
diff --git a/lib/mpl_toolkits/axes_grid/axislines.py b/lib/mpl_toolkits/axes_grid/axislines.py
@@ -336,10 +336,15 @@ def get_axislabel_pos_angle(self, axes):
             get_label_transform() returns a transform of (transAxes+offset)
             """
             loc = self._axis_direction
-            angle = dict(left=0,
-                         right=0,
-                         bottom=.5*np.pi,
-                         top=.5*np.pi)[loc]
+            #angle = dict(left=0,
+            #             right=0,
+            #             bottom=.5*np.pi,
+            #             top=.5*np.pi)[loc]
+
+            if self.nth_coord == 0:
+                angle = 0
+            else:
+                angle = 90
 
             _verts = [0.5, 0.5]
 
@@ -369,6 +374,11 @@ def get_tick_iterators(self, axes):
             else:
                 angle_normal, angle_tangent = 0, 90
 
+            if self.nth_coord == 0:
+                angle_normal, angle_tangent = 90, 0
+            else:
+                angle_normal, angle_tangent = 0, 90
+                
             #angle = 90 - 90 * self.nth_coord
 
             major = self.axis.major
