From: foamyguy Date: Tue, 29 Apr 2025 01:17:30 +0000 (-0500) Subject: implement boundary_fill() and format X-Git-Tag: 2.2.0^2~5 X-Git-Url: https://git.ayoreis.com/hackapet/Adafruit_Blinka_Displayio.git/commitdiff_plain/50c817f6739915e81a9000124cf75218d2c826a9 implement boundary_fill() and format --- diff --git a/bitmaptools/__init__.py b/bitmaptools/__init__.py index e9ca3bb..cc83549 100644 --- a/bitmaptools/__init__.py +++ b/bitmaptools/__init__.py @@ -59,10 +59,13 @@ def draw_circle(dest_bitmap: Bitmap, x: int, y: int, radius: int, value: int): xb += 1 -def draw_polygon(dest_bitmap: Bitmap, - xs: circuitpython_typing.ReadableBuffer, - ys: circuitpython_typing.ReadableBuffer, - value: int, close: bool = True): +def draw_polygon( + dest_bitmap: Bitmap, + xs: circuitpython_typing.ReadableBuffer, + ys: circuitpython_typing.ReadableBuffer, + value: int, + close: bool = True, +): if len(xs) != len(ys): raise ValueError("Length of xs and ys must be equal.") @@ -70,25 +73,40 @@ def draw_polygon(dest_bitmap: Bitmap, cur_point = (xs[i], ys[i]) next_point = (xs[i + 1], ys[i + 1]) print(f"cur: {cur_point}, next: {next_point}") - draw_line(dest_bitmap=dest_bitmap, - x1=cur_point[0], y1=cur_point[1], - x2=next_point[0], y2=next_point[1], - value=value) + draw_line( + dest_bitmap=dest_bitmap, + x1=cur_point[0], + y1=cur_point[1], + x2=next_point[0], + y2=next_point[1], + value=value, + ) if close: print(f"close: {(xs[0], ys[0])} - {(xs[-1], ys[-1])}") - draw_line(dest_bitmap=dest_bitmap, - x1=xs[0], y1=ys[0], - x2=xs[-1], y2=ys[-1], - value=value) - - -def blit(dest_bitmap: Bitmap, source_bitmap: Bitmap, - x: int, y: int, *, - x1: int = 0, y1: int = 0, - x2: int | None = None, y2: int | None = None, - skip_source_index: int | None = None, - skip_dest_index: int | None = None): + draw_line( + dest_bitmap=dest_bitmap, + x1=xs[0], + y1=ys[0], + x2=xs[-1], + y2=ys[-1], + value=value, + ) + + +def blit( + dest_bitmap: Bitmap, + source_bitmap: Bitmap, + x: int, + y: int, + *, + x1: int = 0, + y1: int = 0, + x2: int | None = None, + y2: int | None = None, + skip_source_index: int | None = None, + skip_dest_index: int | None = None, +): """Inserts the source_bitmap region defined by rectangular boundaries""" # pylint: disable=invalid-name if x2 is None: @@ -112,49 +130,52 @@ def blit(dest_bitmap: Bitmap, source_bitmap: Bitmap, y_placement = y + y_count if (dest_bitmap.width > x_placement >= 0) and ( - dest_bitmap.height > y_placement >= 0 + dest_bitmap.height > y_placement >= 0 ): # ensure placement is within target bitmap # get the palette index from the source bitmap this_pixel_color = source_bitmap[ y1 + (y_count * source_bitmap.width) + x1 + x_count - ] + ] - if (skip_source_index is None) or (this_pixel_color != skip_source_index): + if (skip_source_index is None) or ( + this_pixel_color != skip_source_index + ): if (skip_dest_index is None) or ( - dest_bitmap[y_placement * dest_bitmap.width + x_placement] != skip_dest_index): + dest_bitmap[y_placement * dest_bitmap.width + x_placement] + != skip_dest_index + ): dest_bitmap[ # Direct index into a bitmap array is speedier than [x,y] tuple y_placement * dest_bitmap.width + x_placement - ] = this_pixel_color + ] = this_pixel_color elif y_placement > dest_bitmap.height: break def rotozoom( - dest_bitmap: Bitmap, - source_bitmap: Bitmap, - *, - ox: Optional[int] = None, - oy: Optional[int] = None, - dest_clip0: Optional[Tuple[int, int]] = None, - dest_clip1: Optional[Tuple[int, int]] = None, - px: Optional[int] = None, - py: Optional[int] = None, - source_clip0: Optional[Tuple[int, int]] = None, - source_clip1: Optional[Tuple[int, int]] = None, - angle: Optional[float] = None, - scale: Optional[float] = None, - skip_index: Optional[int] = None, + dest_bitmap: Bitmap, + source_bitmap: Bitmap, + *, + ox: Optional[int] = None, + oy: Optional[int] = None, + dest_clip0: Optional[Tuple[int, int]] = None, + dest_clip1: Optional[Tuple[int, int]] = None, + px: Optional[int] = None, + py: Optional[int] = None, + source_clip0: Optional[Tuple[int, int]] = None, + source_clip1: Optional[Tuple[int, int]] = None, + angle: Optional[float] = None, + scale: Optional[float] = None, + skip_index: Optional[int] = None, ): - if ox is None: ox = dest_bitmap.width // 2 if oy in None: oy = dest_bitmap.height // 2 if dest_clip0 is None: - dest_clip0 = (0,0) + dest_clip0 = (0, 0) if dest_clip1 is None: - dest_clip1 = (dest_bitmap.width,dest_bitmap.height) + dest_clip1 = (dest_bitmap.width, dest_bitmap.height) if px is None: px = source_bitmap.width // 2 @@ -162,9 +183,9 @@ def rotozoom( py = source_bitmap.height // 2 if source_clip0 is None: - source_clip0 = (0,0) + source_clip0 = (0, 0) if source_clip1 is None: - source_clip1 = (source_bitmap.width,source_bitmap.height) + source_clip1 = (source_bitmap.width, source_bitmap.height) if angle is None: angle = 0.0 @@ -235,7 +256,9 @@ def rotozoom( u = rowu + minx * du_row v = rowv + minx * dv_row for x in range(minx, maxx + 1): - if (source_clip0_x <= u < source_clip1_x) and (source_clip0_y <= v < source_clip1_y): + if (source_clip0_x <= u < source_clip1_x) and ( + source_clip0_y <= v < source_clip1_y + ): c = source_bitmap[int(u), int(v)] if skip_index is None or c != skip_index: dest_bitmap[x, y] = c @@ -246,17 +269,20 @@ def rotozoom( def arrayblit( - bitmap: Bitmap, - data: circuitpython_typing.ReadableBuffer, - x1: int = 0, y1: int = 0, - x2: Optional[int] = None, y2: Optional[int] = None, - skip_index: Optional[int] = None): + bitmap: Bitmap, + data: circuitpython_typing.ReadableBuffer, + x1: int = 0, + y1: int = 0, + x2: Optional[int] = None, + y2: Optional[int] = None, + skip_index: Optional[int] = None, +): if x2 is None: x2 = bitmap.width if y2 is None: y2 = bitmap.height - _value_count = 2 ** bitmap._bits_per_value + _value_count = 2**bitmap._bits_per_value for y in range(y1, y2): for x in range(x1, x2): i = y * (x2 - x1) + x @@ -265,19 +291,23 @@ def arrayblit( bitmap[x, y] = value -def readinto(bitmap: Bitmap, - file: BinaryIO, - bits_per_pixel: int, - element_size: int = 1, - reverse_pixels_in_element: bool = False, - swap_bytes: bool = False, - reverse_rows: bool = False): +def readinto( + bitmap: Bitmap, + file: BinaryIO, + bits_per_pixel: int, + element_size: int = 1, + reverse_pixels_in_element: bool = False, + swap_bytes: bool = False, + reverse_rows: bool = False, +): width = bitmap.width height = bitmap.height bits_per_value = bitmap._bits_per_value mask = (1 << bits_per_value) - 1 - elements_per_row = (width * bits_per_pixel + element_size * 8 - 1) // (element_size * 8) + elements_per_row = (width * bits_per_pixel + element_size * 8 - 1) // ( + element_size * 8 + ) rowsize = element_size * elements_per_row for y in range(height): @@ -291,15 +321,15 @@ def readinto(bitmap: Bitmap, if swap_bytes: if element_size == 2: rowdata = bytearray( - b''.join( - struct.pack('H', rowdata[i:i + 2])[0]) + b"".join( + struct.pack("H", rowdata[i : i + 2])[0]) for i in range(0, len(rowdata), 2) ) ) elif element_size == 4: rowdata = bytearray( - b''.join( - struct.pack('I', rowdata[i:i + 4])[0]) + b"".join( + struct.pack("I", rowdata[i : i + 4])[0]) for i in range(0, len(rowdata), 4) ) ) @@ -325,32 +355,46 @@ def readinto(bitmap: Bitmap, elif bits_per_pixel == 8: value = rowdata[x] elif bits_per_pixel == 16: - value = struct.unpack_from(' maxx: + maxx = cur_x + if cur_y > maxy: + maxy = cur_y + + dest_bitmap[cur_x, cur_y] = fill_color_value + + above_point = (cur_x, cur_y - 1) + below_point = (cur_x, cur_y + 1) + left_point = (cur_x - 1, cur_y) + right_point = (cur_x + 1, cur_y) + + if ( + above_point[1] >= 0 + and above_point not in seen_points + and above_point not in fill_points + ): + fill_points.append(above_point) + if ( + below_point[1] < dest_bitmap.height + and below_point not in seen_points + and below_point not in fill_points + ): + fill_points.append(below_point) + if ( + left_point[0] >= 0 + and left_point not in seen_points + and left_point not in fill_points + ): + fill_points.append(left_point) + if ( + right_point[0] < dest_bitmap.width + and right_point not in seen_points + and right_point not in fill_points + ): + fill_points.append(right_point)