Implemented Display rotation

diff --git a/displayio/display.py b/displayio/display.py
index 904302b7688c0ba0905ae09c95d371f4ff49d0a6..e5998e1c8a6e3361280a2fb7b40e6aa166b91dbd 100644 (file)
--- a/displayio/display.py
+++ b/displayio/display.py
@@ -217,7 +217,11 @@ class Display:
 
     def _refresh_display_area(self, rectangle):
         """Loop through dirty rectangles and redraw that area."""
 
     def _refresh_display_area(self, rectangle):
         """Loop through dirty rectangles and redraw that area."""

-        data = numpy.array(self._buffer.crop(rectangle).convert("RGB")).astype("uint16")
+
+        img = self._buffer.crop(rectangle).convert("RGB")
+        img = img.rotate(self._rotation)
+
+        data = numpy.array(img).astype("uint16")

         color = (
             ((data[:, :, 0] & 0xF8) << 8)
             | ((data[:, :, 1] & 0xFC) << 3)
         color = (
             ((data[:, :, 0] & 0xF8) << 8)
             | ((data[:, :, 1] & 0xFC) << 3)


@@ -228,20 +232,50 @@ class Display:
             numpy.dstack(((color >> 8) & 0xFF, color & 0xFF)).flatten().tolist()
         )
 
             numpy.dstack(((color >> 8) & 0xFF, color & 0xFF)).flatten().tolist()
         )
 

+        display_rectangle = self._apply_rotation(rectangle)
+

         self._write(
             self._set_column_command,
             self._encode_pos(
         self._write(
             self._set_column_command,
             self._encode_pos(

-                rectangle.x1 + self._colstart, rectangle.x2 + self._colstart - 1
+                display_rectangle.x1 + self._colstart,
+                display_rectangle.x2 + self._colstart - 1,

             ),
         )
         self._write(
             self._set_row_command,
             self._encode_pos(
             ),
         )
         self._write(
             self._set_row_command,
             self._encode_pos(

-                rectangle.y1 + self._rowstart, rectangle.y2 + self._rowstart - 1
+                display_rectangle.y1 + self._rowstart,
+                display_rectangle.y2 + self._rowstart - 1,

             ),
         )
             ),
         )

+

         self._write(self._write_ram_command, pixels)
 
         self._write(self._write_ram_command, pixels)
 

+    def _apply_rotation(self, rectangle):
+        """Adjust the rectangle coordinates based on rotation"""
+        if self._rotation == 90:
+            return Rectangle(
+                self._width - rectangle.y2,
+                rectangle.x1,
+                self._width - rectangle.y1,
+                rectangle.x2,
+            )
+        if self._rotation == 180:
+            return Rectangle(
+                self._width - rectangle.x2,
+                self._height - rectangle.y2,
+                self._width - rectangle.x1,
+                self._height - rectangle.y1,
+            )
+        if self._rotation == 270:
+            return Rectangle(
+                rectangle.y1,
+                self._height - rectangle.x2,
+                rectangle.y2,
+                self._height - rectangle.x1,
+            )
+        return rectangle
+

     def _encode_pos(self, x, y):
         """Encode a postion into bytes."""
         return struct.pack(self._bounds_encoding, x, y)
     def _encode_pos(self, x, y):
         """Encode a postion into bytes."""
         return struct.pack(self._bounds_encoding, x, y)