Allow rotation of tick labels

Author: mlq

matplotlib2tikz/axes.py

@@ -198,6 +198,35 @@ def __init__(self, data, obj):
                     assert direction == 'inout'
                     self.axis_options.append('tick align=center')
 
+        # Set each rotation for every label
+        x_tick_labels_rotation = \
+            [label.get_rotation() for label in obj.xaxis.get_majorticklabels()]
+        if any(x_tick_labels_rotation) != 0:
+            # check if every value is the same
+            if len(set(x_tick_labels_rotation)) == 1:
+                self.axis_options.append('xticklabel style = {rotate=%d}' %
+                                         x_tick_labels_rotation[0])
+            else:
+                value = 'every x tick label/.style = {\n' \
+                    'rotate={,%s}[\\ticknum]\n' \
+                    '}' % ','.join(str(x) for x in x_tick_labels_rotation)
+
+                self.axis_options.append(value)
+
+        y_tick_labels_rotation = \
+            [label.get_rotation() for label in obj.yaxis.get_majorticklabels()]
+        if any(y_tick_labels_rotation) != 0:
+            # check if every value is the same
+            if len(set(y_tick_labels_rotation)) == 1:
+                self.axis_options.append('yticklabel style = {rotate=%d}' %
+                                         y_tick_labels_rotation[0])
+            else:
+                value = 'every y tick label/.style = {\n' \
+                    'rotate={,%s}[\\ticknum]\n' \
+                    '}' % ','.join(str(x) for x in y_tick_labels_rotation)
+
+                self.axis_options.append(value)
+
         # Don't use get_{x,y}gridlines for gridlines; see discussion on
         # <http://sourceforge.net/p/matplotlib/mailman/message/25169234/>
         # Coordinate of the lines are entirely meaningless, but styles

test/testfunctions/__init__.py

@@ -25,6 +25,7 @@
   'noise2',
   'patches',
   'quadmesh',
+  'rotated_labels',
   'scatter',
   'scatter_colormap',
   'sharex_and_y',

test/testfunctions/rotated_labels.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+
+desc = 'Rotated labels'
+phash = 'a91fa323753d8d03'
+
+
+def plot():
+    from matplotlib import pyplot as plt
+
+    x = [1, 2, 3, 4]
+    y = [1, 4, 9, 6]
+
+    fig = plt.figure()
+
+    plt.subplot(611)
+    plt.plot(x, y, 'ro')
+    plt.xticks(x, rotation='horizontal')
+
+    plt.subplot(612)
+    plt.plot(x, y, 'ro')
+    plt.xticks(x, rotation='vertical')
+
+    ax = plt.subplot(613)
+    ax.plot(x, y, 'ro')
+    plt.xticks(y, rotation=-25)
+
+    ax = plt.subplot(614)
+    ax.plot(x, y, 'ro')
+    plt.yticks(y, rotation=-25)
+
+    ax = plt.subplot(615)
+    ax.plot(x, y, 'ro')
+
+    for idx, label in enumerate(ax.get_xticklabels()):
+        label.set_rotation(45 if idx % 2 == 0 else 0)
+
+    ax = plt.subplot(616)
+    ax.plot(x, y, 'ro')
+
+    for idx, label in enumerate(ax.get_yticklabels()):
+        label.set_rotation(90 if idx % 2 == 0 else 0)
+
+    return fig