fix tight layout and rotation bugs

Author: lionel42

pygame_matplotlib/backend_pygame.py

@@ -181,7 +181,7 @@ def draw_path(self, gc, path, transform, rgbFace=None):
         poly_points = []
 
         for point, code in path.iter_segments(transform):
-            logger.debug(point, code)
+            logger.debug(f"{point=}, {code=}")
             if code == Path.LINETO:
                 draw_func(self.surface, color, previous_point, point)
                 previous_point = point
@@ -239,6 +239,11 @@ def draw_image(self, gc, x, y, im):
         )
 
     def draw_text(self, gc, x, y, s, prop, angle, ismath=False, mtext=None):
+
+        logger.info(
+            f"Drawing text: {s=} at ({x=}, {y=}) with {angle=} and ismath={ismath} "
+            f"{mtext=} {prop=} {gc=}"
+        )
         # make sure font module is initialized
         if not pygame.font.get_init():
             pygame.font.init()
@@ -249,6 +254,9 @@ def draw_text(self, gc, x, y, s, prop, angle, ismath=False, mtext=None):
         font_surface = myfont.render(
             s, gc.get_antialiased(), [val * 255 for val in gc.get_rgb()]
         )
+        if angle:
+            font_surface = pygame.transform.rotate(font_surface, angle)
+        
         # Get the expected size of the font
         width, height = myfont.size(s)
         # Tuple for the position of the font
@@ -311,6 +319,8 @@ def points_to_pixels(self, points):
         # return points/72.0 * self.dpi.get()
 
     def clear(self):
+        if not hasattr(self, "surface"):
+            return
         self.surface.fill("white")
 
     def copy_from_bbox(self, bbox):

tests/test_mpl_functionalities.py

@@ -0,0 +1,25 @@
+import matplotlib.pyplot as plt
+import matplotlib
+
+matplotlib.use("pygame")
+
+
+def test_save_jpg():
+    plt.figure()
+    plt.plot([1, 2], [1, 2], color="green")
+    plt.text(1.5, 1.5, "2", size=50)
+
+def test_tight_layout():
+    plt.figure()
+    plt.plot([1, 2], [1, 2], color="green")
+    plt.text(1.5, 1.5, "2", size=50)
+    plt.tight_layout()
+
+
+def test_ylabel():
+    # Issue that the ylabel was not reotated, sadly I cannot  visually test this
+    plt.figure()
+    plt.plot([1, 2], [1, 2], color="green")
+    plt.text(1.5, 1.5, "2", size=50)
+    plt.ylabel("swag")
+    plt.tight_layout()