work for adafruit#59 to remove changing types of objects when reassigning

Author: matt-land

adafruit_imageload/bmp/indexed.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 #
 # SPDX-License-Identifier: MIT
 
@@ -43,8 +43,8 @@ def load(
     color_depth: int,
     compression: int,
     *,
-    bitmap: BitmapConstructor = None,
-    palette: PaletteConstructor = None,
+    bitmap: Optional[BitmapConstructor] = None,
+    palette: Optional[PaletteConstructor] = None,
 ) -> Tuple[Bitmap, Optional[Palette]]:
     """Loads indexed bitmap data into bitmap and palette objects.
 
@@ -54,19 +54,24 @@ def load(
     :param int data_start: Byte location where the data starts (after headers)
     :param int colors: Number of distinct colors in the image
     :param int color_depth: Number of bits used to store a value
-    :param int compression: 0 - none, 1 - 8bit RLE, 2 - 4bit RLE"""
+    :param int compression: 0 - none, 1 - 8bit RLE, 2 - 4bit RLE
+    :param BitmapConstructor bitmap: a function that returns a displayio.Bitmap
+    :param PaletteConstructor palette: a function that returns a displayio.Palette
+    """
     # pylint: disable=too-many-arguments,too-many-locals,too-many-branches
+    palette_obj: Optional[Palette] = None
     if palette:
-        palette = palette(colors)  # type: Palette
+        palette_obj = palette(colors)
 
         file.seek(data_start - colors * 4)
         for value in range(colors):
             c_bytes = file.read(4)
             # Need to swap red & blue bytes (bytes 0 and 2)
-            palette[value] = bytes(
+            palette_obj[value] = bytes(
                 b"".join([c_bytes[2:3], c_bytes[1:2], c_bytes[0:1], c_bytes[3:1]])
             )
 
+    bitmap_obj: Optional[Bitmap] = None
     if bitmap:
         minimum_color_depth = 1
         while colors > 2**minimum_color_depth:
@@ -78,7 +83,7 @@ def load(
 
             # convert unsigned int to signed int when height is negative
             height = negative_height_check(height)
-        bitmap = bitmap(width, abs(height), colors)  # type: Bitmap
+        bitmap_obj = bitmap(width, abs(height), colors)
         file.seek(data_start)
         line_size = width // (8 // color_depth)
         if width % (8 // color_depth) != 0:
@@ -97,10 +102,9 @@ def load(
             range3 = 1
 
         if compression == 0:
-
             if _bitmap_readinto:
                 _bitmap_readinto(
-                    bitmap,
+                    bitmap_obj,
                     file,
                     bits_per_pixel=color_depth,
                     element_size=4,
@@ -120,17 +124,17 @@ def load(
                         pixel = (
                             chunk[i] >> (8 - color_depth * (x % pixels_per_byte + 1))
                         ) & mask
-                        bitmap[offset + x] = pixel
+                        bitmap_obj[offset + x] = pixel
         elif compression in (1, 2):
             decode_rle(
-                bitmap=bitmap,
+                bitmap=bitmap_obj,
                 file=file,
                 compression=compression,
                 y_range=(range1, range2, range3),
                 width=width,
             )
 
-    return bitmap, palette
+    return bitmap_obj, palette_obj
 
 
 def decode_rle(

adafruit_imageload/gif.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2019 Radomir Dopieralski for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 #
 # SPDX-License-Identifier: MIT
 
@@ -32,7 +32,7 @@ def load(
     file: BufferedReader,
     *,
     bitmap: BitmapConstructor,
-    palette: PaletteConstructor = None
+    palette: Optional[PaletteConstructor] = None
 ) -> Tuple[Bitmap, Optional[Palette]]:
     """Loads a GIF image from the open ``file``.
 

adafruit_imageload/png.py

@@ -1,4 +1,5 @@
 # SPDX-FileCopyrightText: 2022 Radomir Dopieralski
+# SPDX-FileCopyrightText: 2023 Matt Land
 #
 # SPDX-License-Identifier: MIT
 
@@ -9,13 +10,14 @@
 Load pixel values (indices or colors) into a bitmap and colors into a palette
 from a PNG file.
 
-* Author(s): Radomir Dopieralski
+* Author(s): Radomir Dopieralski, Matt Land
 
 """
 
 try:
-    # pylint: disable=unused-import
-    import typing
+    from io import BufferedReader
+    from typing import Optional
+    from displayio import Palette, Bitmap
     from .displayio_types import PaletteConstructor, BitmapConstructor
 except ImportError:
     pass
@@ -28,17 +30,23 @@
 
 
 def load(
-    file: str, *, bitmap: BitmapConstructor, palette: PaletteConstructor = None
-):  # pylint: disable=too-many-locals,too-many-branches
-    """Loads a PNG image from the open ``file``.
+    file: BufferedReader,
+    *,
+    bitmap: BitmapConstructor,
+    palette: Optional[PaletteConstructor] = None
+) -> tuple[Bitmap, Optional[Palette]]:
+    """
+    Loads a PNG image from the open ``file``.
 
     Returns tuple of bitmap object and palette object.
 
     :param file: The *.png file being loaded
     :param object bitmap: Type to store bitmap data. Must have API similar to
       `displayio.Bitmap`.
     :param object palette: Type to store the palette. Must have API similar to
-      `displayio.Palette`. Will be skipped if None"""
+      `displayio.Palette`. Will be skipped if None
+    """
+    # pylint: disable=too-many-locals,too-many-branches
     header = file.read(8)
     if header != b"\x89PNG\r\n\x1a\n":
         raise ValueError("Not a PNG file")
@@ -47,6 +55,8 @@ def load(
     pal = None
     mode = None
     depth = None
+    width = 0
+    height = 0
     while True:
         size, chunk = struct.unpack(">I4s", file.read(8))
         if chunk == b"IHDR":

adafruit_imageload/pnm/__init__.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -40,8 +40,8 @@ def load(
     file: BufferedReader,
     header: bytes,
     *,
-    bitmap: BitmapConstructor = None,
-    palette: PaletteConstructor = None
+    bitmap: Optional[BitmapConstructor] = None,
+    palette: Optional[PaletteConstructor] = None
 ) -> Tuple[Optional[Bitmap], Optional[Palette]]:
     """
     Scan for netpbm format info, skip over comments, and delegate to a submodule
@@ -92,19 +92,20 @@ def load(
                 )
 
         if len(pnm_header) == 2 and magic_number in [b"P1", b"P4"]:
-            bitmap = bitmap(pnm_header[0], pnm_header[1], 1)  # type: Bitmap
+            bitmap_obj = bitmap(pnm_header[0], pnm_header[1], 1)
+            palette_obj: Optional[Palette] = None
             if palette:
-                palette = palette(1)  # type: Palette
-                palette[0] = b"\xFF\xFF\xFF"
+                palette_obj = palette(1)
+                palette_obj[0] = b"\xFF\xFF\xFF"
             if magic_number.startswith(b"P1"):
                 from . import pbm_ascii
 
                 return pbm_ascii.load(
                     file,
                     pnm_header[0],
                     pnm_header[1],
-                    bitmap=bitmap,
-                    palette=palette,
+                    bitmap=bitmap_obj,
+                    palette=palette_obj,
                 )
 
             from . import pbm_binary
@@ -113,7 +114,7 @@ def load(
                 file,
                 pnm_header[0],
                 pnm_header[1],
-                bitmap=bitmap,
+                bitmap=bitmap_obj,
                 palette=palette,
             )
 

adafruit_imageload/pnm/pbm_ascii.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -32,7 +32,7 @@ def load(
     width: int,
     height: int,
     bitmap: Bitmap,
-    palette: Palette = None,
+    palette: Optional[Palette] = None,
 ) -> Tuple[Bitmap, Optional[Palette]]:
     """
     Load a P1 'PBM' ascii image into the displayio.Bitmap

adafruit_imageload/pnm/pbm_binary.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -31,7 +31,7 @@ def load(
     width: int,
     height: int,
     bitmap: Bitmap,
-    palette: Palette = None,
+    palette: Optional[Palette] = None,
 ) -> Tuple[Bitmap, Optional[Palette]]:
     """
     Load a P4 'PBM' binary image into the Bitmap

adafruit_imageload/pnm/pgm/__init__.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -29,8 +29,8 @@ def load(
     magic_number: bytes,
     header: List[int],
     *,
-    bitmap: BitmapConstructor = None,
-    palette: PaletteConstructor = None
+    bitmap: Optional[BitmapConstructor] = None,
+    palette: Optional[PaletteConstructor] = None
 ) -> Tuple[Optional[Bitmap], Optional[Palette]]:
     """
     Perform the load of Netpbm greyscale images (P2, P5)

adafruit_imageload/pnm/pgm/ascii.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -27,8 +27,8 @@ def load(
     file: BufferedReader,
     width: int,
     height: int,
-    bitmap: BitmapConstructor = None,
-    palette: PaletteConstructor = None,
+    bitmap: Optional[BitmapConstructor] = None,
+    palette: Optional[PaletteConstructor] = None,
 ) -> Tuple[Optional[Bitmap], Optional[Palette]]:
     """
     Load a PGM ascii file (P2)
@@ -46,10 +46,12 @@ def load(
             _palette_colors.add(int_pixel)
             pixel = bytearray()
         pixel += byte
+    actual_palette: Optional[Palette] = None
     if palette:
-        palette = build_palette(palette, _palette_colors)  # type: Palette
+        actual_palette = build_palette(palette, _palette_colors)
+    actual_bitmap: Optional[Bitmap] = None
     if bitmap:
-        bitmap = bitmap(width, height, len(_palette_colors))  # type: Bitmap
+        actual_bitmap = bitmap(width, height, len(_palette_colors))
         _palette_colors = list(_palette_colors)
         file.seek(data_start)
         for y in range(height):
@@ -61,8 +63,8 @@ def load(
                         break
                     pixel += byte
                 int_pixel = int("".join(["%c" % char for char in pixel]))
-                bitmap[x, y] = _palette_colors.index(int_pixel)
-    return bitmap, palette
+                actual_bitmap[x, y] = _palette_colors.index(int_pixel)
+    return actual_bitmap, actual_palette
 
 
 def build_palette(

adafruit_imageload/pnm/pgm/binary.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -27,8 +27,8 @@ def load(
     file: BufferedReader,
     width: int,
     height: int,
-    bitmap: BitmapConstructor = None,
-    palette: PaletteConstructor = None,
+    bitmap: Optional[BitmapConstructor] = None,
+    palette: Optional[PaletteConstructor] = None,
 ) -> Tuple[Optional[Bitmap], Optional[Palette]]:
     """
     Load a P5 format file (binary), handle PGM (greyscale)
@@ -40,17 +40,19 @@ def load(
         for pixel in data_line:
             palette_colors.add(pixel)
 
+    actual_palette: Optional[Palette] = None
     if palette:
-        palette = build_palette(palette, palette_colors)  # type: Palette
+        actual_palette = build_palette(palette, palette_colors)
+    actual_bitmap: Optional[Bitmap] = None
     if bitmap:
-        bitmap = bitmap(width, height, len(palette_colors))  # type: Bitmap
+        actual_bitmap = bitmap(width, height, len(palette_colors))
         palette_colors = list(palette_colors)  # type: List[int]
         file.seek(data_start)
         for y in range(height):
             data_line = iter(bytes(file.read(width)))
             for x, pixel in enumerate(data_line):
-                bitmap[x, y] = palette_colors.index(pixel)
-    return bitmap, palette
+                actual_bitmap[x, y] = palette_colors.index(pixel)
+    return actual_bitmap, actual_palette
 
 
 def build_palette(

adafruit_imageload/pnm/ppm_ascii.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: 2018 Scott Shawcroft for Adafruit Industries
-# SPDX-FileCopyrightText: 2022 Matt Land
+# SPDX-FileCopyrightText: 2022-2023 Matt Land
 # SPDX-FileCopyrightText: Brooke Storm
 # SPDX-FileCopyrightText: Sam McGahan
 #
@@ -38,8 +38,8 @@ def load(
     file: BufferedReader,
     width: int,
     height: int,
-    bitmap: BitmapConstructor = None,
-    palette: PaletteConstructor = None,
+    bitmap: Optional[BitmapConstructor] = None,
+    palette: Optional[PaletteConstructor] = None,
 ) -> Tuple[Optional[Bitmap], Optional[Palette]]:
     """
     :param stream file: infile with the position set at start of data
@@ -55,20 +55,22 @@ def load(
     for triplet in read_three_colors(file):
         palette_colors.add(triplet)
 
+    actual_palette: Optional[Palette] = None
     if palette:
-        palette = palette(len(palette_colors))  # type: Palette
+        actual_palette = palette(len(palette_colors))
         for counter, color in enumerate(palette_colors):
-            palette[counter] = color
+            actual_palette[counter] = color
+    actual_bitmap: Optional[Bitmap] = None
     if bitmap:
         file.seek(data_start)
-        bitmap = bitmap(width, height, len(palette_colors))  # type: Bitmap
+        actual_bitmap = bitmap(width, height, len(palette_colors))
         palette_colors = list(palette_colors)  # type: List[bytes]
         for y in range(height):
             for x in range(width):
                 for color in read_three_colors(file):
-                    bitmap[x, y] = palette_colors.index(color)
+                    actual_bitmap[x, y] = palette_colors.index(color)
                     break  # exit the inner generator
-    return bitmap, palette
+    return actual_bitmap, actual_palette
 
 
 def read_three_colors(file: BufferedReader) -> Iterator[bytes]: