implement xtick rotation_mode

kyracho · kyracho · commit 03d0f37c067b · 2024-10-13T07:03:15.000Z
diff --git a/lib/matplotlib/axis.py b/lib/matplotlib/axis.py
@@ -346,6 +346,11 @@ def _apply_params(self, **kwargs):
                    if k in _gridline_param_names}
         self.gridline.set(**grid_kw)
 
+        if 'rotation_mode' in kwargs:
+            rotation_mode = kwargs.pop('rotation_mode')
+            self.label1.set_rotation_mode(rotation_mode)
+            self.label2.set_rotation_mode(rotation_mode)
+
     def update_position(self, loc):
         """Set the location of tick in data coords with scalar *loc*."""
         raise NotImplementedError('Derived must override')
@@ -1072,7 +1077,7 @@ def _translate_tick_params(kw, reverse=False):
             'tick1On', 'tick2On', 'label1On', 'label2On',
             'length', 'direction', 'left', 'bottom', 'right', 'top',
             'labelleft', 'labelbottom', 'labelright', 'labeltop',
-            'labelrotation',
+            'labelrotation', 'rotation_mode',
             *_gridline_param_names]
 
         keymap = {
diff --git a/lib/matplotlib/tests/test_text.py b/lib/matplotlib/tests/test_text.py
@@ -1135,3 +1135,19 @@ def test_font_wrap():
     plt.text(3, 4, t, family='monospace', ha='right', wrap=True)
     plt.text(-1, 0, t, fontsize=14, style='italic', ha='left', rotation=-15,
              wrap=True)
+
+
+def test_ha_for_angle():
+    text_instance = Text()
+    angles = np.arange(0, 360.1, 0.1)
+    for angle in angles:
+        alignment = text_instance.ha_for_angle(angle)
+        assert alignment in ['center', 'left', 'right']
+
+
+def test_va_for_angle():
+    text_instance = Text()
+    angles = np.arange(0, 360.1, 0.1)
+    for angle in angles:
+        alignment = text_instance.va_for_angle(angle)
+        assert alignment in ['center', 'top', 'bottom']
diff --git a/lib/matplotlib/text.py b/lib/matplotlib/text.py
@@ -310,7 +310,7 @@ def set_rotation_mode(self, m):
         if m is None:
             m = "default"
         else:
-            _api.check_in_list(("anchor", "default"), rotation_mode=m)
+            _api.check_in_list(("anchor", "default", "xtick", "ytick"), rotation_mode=m)
         self._rotation_mode = m
         self.stale = True
 
@@ -454,6 +454,12 @@ def _get_layout(self, renderer):
 
         rotation_mode = self.get_rotation_mode()
         if rotation_mode != "anchor":
+            if rotation_mode == 'xtick':
+                angle = self.get_rotation()
+                halign = self.ha_for_angle(angle)
+            elif rotation_mode == 'ytick':
+                angle = self.get_rotation()
+                valign = self.va_for_angle(angle)
             # compute the text location in display coords and the offsets
             # necessary to align the bbox with that location
             if halign == 'center':
@@ -467,6 +473,8 @@ def _get_layout(self, renderer):
                 offsety = (ymin + ymax) / 2
             elif valign == 'top':
                 offsety = ymax
+            elif valign == 'bottom':
+                offsety = ymin
             elif valign == 'baseline':
                 offsety = ymin + descent
             elif valign == 'center_baseline':
@@ -1380,6 +1388,30 @@ def set_fontname(self, fontname):
         """
         self.set_fontfamily(fontname)
 
+    def ha_for_angle(self, angle):
+        """
+        Determines horizontal alignment ('ha') based on the angle of rotation.
+        """
+        if (angle < 5 or 85 <= angle < 105 or 355 <= angle <= 360 or
+                170 <= angle < 190 or 265 <= angle < 275):
+            return 'center'
+        elif 5 <= angle < 85 or 190 <= angle < 265:
+            return 'right'
+        elif 105 <= angle < 170 or 275 <= angle < 355:
+            return 'left'
+
+    def va_for_angle(self, angle):
+        """
+        Determines vertical alignment ('va') based on the angle of rotation.
+        """
+        if (angle < 5 or 355 <= angle <= 360 or 170 <= angle < 190
+                or 85 <= angle < 105 or 265 <= angle < 275):
+            return 'center'
+        elif 190 <= angle < 265 or 5 <= angle < 85:
+            return 'top'
+        elif 105 <= angle < 170 or 275 <= angle < 355:
+            return 'bottom'
+
 
 class OffsetFrom:
     """Callable helper class for working with `Annotation`."""
diff --git a/lib/matplotlib/text.pyi b/lib/matplotlib/text.pyi
@@ -106,6 +106,8 @@ class Text(Artist):
     def set_fontname(self, fontname: str | Iterable[str]) -> None: ...
     def get_antialiased(self) -> bool: ...
     def set_antialiased(self, antialiased: bool) -> None: ...
+    def ha_for_angle(self, angle: Any) -> Literal['center', 'right', 'left'] | None: ...
+    def va_for_angle(self, angle: Any) -> Literal['center', 'top', 'bottom'] | None: ...
 
 class OffsetFrom:
     def __init__(
