Remove more of PIL

[hackapet/Adafruit_Blinka_Displayio.git] / displayio / _palette.py

diff --git a/displayio/_palette.py b/displayio/_palette.py
index 969c769b9ae6738cbce64e806c184d8805110976..63d98eec3385ceea9e19ed6c556565abfb4f84c4 100644 (file)
--- a/displayio/_palette.py
+++ b/displayio/_palette.py
@@ -18,7 +18,10 @@ displayio for Blinka
 """
 
 from typing import Optional, Union, Tuple
 """
 
 from typing import Optional, Union, Tuple

-import circuitpython_typing
+from circuitpython_typing import ReadableBuffer
+from ._colorconverter import ColorConverter
+from ._colorspace import Colorspace
+from ._structs import InputPixelStruct, OutputPixelStruct, ColorStruct

 
 __version__ = "0.0.0+auto.0"
 __repo__ = "https://github.com/adafruit/Adafruit_Blinka_displayio.git"
 
 __version__ = "0.0.0+auto.0"
 __repo__ = "https://github.com/adafruit/Adafruit_Blinka_displayio.git"


@@ -29,31 +32,18 @@ class Palette:
     format internally to save memory.
     """
 
     format internally to save memory.
     """
 

-    def __init__(self, color_count: int):
+    def __init__(self, color_count: int, dither: bool = False):

         """Create a Palette object to store a set number of colors."""
         self._needs_refresh = False
         """Create a Palette object to store a set number of colors."""
         self._needs_refresh = False

+        self._dither = dither

 
         self._colors = []
         for _ in range(color_count):
             self._colors.append(self._make_color(0))
 
         self._colors = []
         for _ in range(color_count):
             self._colors.append(self._make_color(0))

-            self._update_rgba(len(self._colors) - 1)
-
-    def _update_rgba(self, index):
-        color = self._colors[index]["rgb888"]
-        transparent = self._colors[index]["transparent"]
-        self._colors[index]["rgba"] = (
-            color >> 16,
-            (color >> 8) & 0xFF,
-            color & 0xFF,
-            0 if transparent else 0xFF,
-        )

 
     def _make_color(self, value, transparent=False):
 
     def _make_color(self, value, transparent=False):

-        color = {
-            "transparent": transparent,
-            "rgb888": 0,
-            "rgba": (0, 0, 0, 255),
-        }
+        color = ColorStruct(transparent=transparent)
+

         if isinstance(value, (tuple, list, bytes, bytearray)):
             value = (value[0] & 0xFF) << 16 | (value[1] & 0xFF) << 8 | value[2] & 0xFF
         elif isinstance(value, int):
         if isinstance(value, (tuple, list, bytes, bytearray)):
             value = (value[0] & 0xFF) << 16 | (value[1] & 0xFF) << 8 | value[2] & 0xFF
         elif isinstance(value, int):


@@ -61,7 +51,7 @@ class Palette:
                 raise ValueError("Color must be between 0x000000 and 0xFFFFFF")
         else:
             raise TypeError("Color buffer must be a buffer, tuple, list, or int")
                 raise ValueError("Color must be between 0x000000 and 0xFFFFFF")
         else:
             raise TypeError("Color buffer must be a buffer, tuple, list, or int")

-        color["rgb888"] = value
+        color.rgb888 = value

         self._needs_refresh = True
 
         return color
         self._needs_refresh = True
 
         return color


@@ -73,7 +63,7 @@ class Palette:
     def __setitem__(
         self,
         index: int,
     def __setitem__(
         self,
         index: int,

-        value: Union[int, circuitpython_typing.ReadableBuffer, Tuple[int, int, int]],
+        value: Union[int, ReadableBuffer, Tuple[int, int, int]],

     ) -> None:
         """Sets the pixel color at the given index. The index should be
         an integer in the range 0 to color_count-1.
     ) -> None:
         """Sets the pixel color at the given index. The index should be
         an integer in the range 0 to color_count-1.


@@ -82,30 +72,27 @@ class Palette:
         (to represent an RGB value). Value can be an int, bytes (3 bytes (RGB) or
         4 bytes (RGB + pad byte)), bytearray, or a tuple or list of 3 integers.
         """
         (to represent an RGB value). Value can be an int, bytes (3 bytes (RGB) or
         4 bytes (RGB + pad byte)), bytearray, or a tuple or list of 3 integers.
         """

-        if self._colors[index]["rgb888"] != value:
+        if self._colors[index].rgb888 != value:

             self._colors[index] = self._make_color(value)
             self._colors[index] = self._make_color(value)

-            self._update_rgba(index)

 
     def __getitem__(self, index: int) -> Optional[int]:
         if not 0 <= index < len(self._colors):
             raise ValueError("Palette index out of range")
 
     def __getitem__(self, index: int) -> Optional[int]:
         if not 0 <= index < len(self._colors):
             raise ValueError("Palette index out of range")

-        return self._colors[index]["rgb888"]
+        return self._colors[index].rgb888

 
     def make_transparent(self, palette_index: int) -> None:
         """Set the palette index to be a transparent color"""
 
     def make_transparent(self, palette_index: int) -> None:
         """Set the palette index to be a transparent color"""

-        self._colors[palette_index]["transparent"] = True
-        self._update_rgba(palette_index)
+        self._colors[palette_index].transparent = True

 
     def make_opaque(self, palette_index: int) -> None:
         """Set the palette index to be an opaque color"""
 
     def make_opaque(self, palette_index: int) -> None:
         """Set the palette index to be an opaque color"""

-        self._colors[palette_index]["transparent"] = False
-        self._update_rgba(palette_index)
+        self._colors[palette_index].transparent = False

 
     def _get_palette(self):
         """Generate a palette for use with PIL"""
         palette = []
         for color in self._colors:
 
     def _get_palette(self):
         """Generate a palette for use with PIL"""
         palette = []
         for color in self._colors:

-            palette += color["rgba"][0:3]
+            palette += color.rgba()[0:3]

         return palette
 
     def _get_alpha_palette(self):
         return palette
 
     def _get_alpha_palette(self):


@@ -114,9 +101,56 @@ class Palette:
         palette = []
         for color in self._colors:
             for _ in range(3):
         palette = []
         for color in self._colors:
             for _ in range(3):

-                palette += [0 if color["transparent"] else 255]
+                palette += [0 if color.transparent else 0xFF]

         return palette
 
         return palette
 

+    def _get_color(
+        self,
+        colorspace: Colorspace,
+        input_pixel: InputPixelStruct,
+        output_color: OutputPixelStruct,
+    ):
+        palette_index = input_pixel.pixel
+        if palette_index > len(self._colors) or self._colors[palette_index].transparent:
+            output_color.opaque = False
+            return
+
+        color = self._colors[palette_index]
+        if (
+            not self._dither
+            and color.cached_colorspace == colorspace
+            and color.cached_colorspace_grayscale_bit == colorspace.grayscale_bit
+            and color.cached_colorspace_grayscale == colorspace.grayscale
+        ):
+            output_color.pixel = self._colors[palette_index].cached_color
+            return
+
+        rgb888_pixel = input_pixel
+        ColorConverter._convert_color(  # pylint: disable=protected-access
+            colorspace, self._dither, rgb888_pixel, output_color
+        )
+        if not self._dither:
+            color.cached_colorspace = colorspace
+            color.cached_color = output_color.pixel
+            color.cached_colorspace_grayscale = colorspace.grayscale
+            color.cached_colorspace_grayscale_bit = colorspace.grayscale_bit
+

     def is_transparent(self, palette_index: int) -> bool:
         """Returns True if the palette index is transparent. Returns False if opaque."""
     def is_transparent(self, palette_index: int) -> bool:
         """Returns True if the palette index is transparent. Returns False if opaque."""

-        return self._colors[palette_index]["transparent"]
+        return self._colors[palette_index].transparent
+
+    def _finish_refresh(self):
+        pass
+
+    @property
+    def dither(self) -> bool:
+        """When true the palette dithers the output by adding
+        random noise when truncating to display bitdepth
+        """
+        return self._dither
+
+    @dither.setter
+    def dither(self, value: bool):
+        if not isinstance(value, bool):
+            raise ValueError("Value should be boolean")
+        self._dither = value