implement xtick rotation_mode

kyracho · kyracho · commit 935680863f16 · 2024-10-13T02:40: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,12 @@ 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'], \
+            f"Invalid alignment at angle {angle:.1f}: {alignment}"
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"), rotation_mode=m)
         self._rotation_mode = m
         self.stale = True
 
@@ -454,6 +454,9 @@ 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)
             # compute the text location in display coords and the offsets
             # necessary to align the bbox with that location
             if halign == 'center':
@@ -1380,6 +1383,18 @@ 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'
+
 
 class OffsetFrom:
     """Callable helper class for working with `Annotation`."""
@@ -2031,5 +2046,4 @@ def get_tightbbox(self, renderer=None):
             return Bbox.null()
         return super().get_tightbbox(renderer)
 
-
 _docstring.interpd.register(Annotation=Annotation.__init__.__doc__)
