Skip to content

image_tools

image_tools

Utils for image operations.

replace_color 

replace_color(
    input_image: Image.Image,
    original: tuple[int, ...] | int,
    replacement: tuple[int, ...] | int,
    *,
    output_image_mode: str = "RGBA"
) -> Image.Image

Replace original color from input image with replacement color.

Source code in src/pygerber/backend/rasterized_2d/image_tools.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def replace_color(
    input_image: Image.Image,
    original: tuple[int, ...] | int,
    replacement: tuple[int, ...] | int,
    *,
    output_image_mode: str = "RGBA",
) -> Image.Image:
    """Replace `original` color from input image with `replacement` color."""
    if input_image.mode != output_image_mode:
        output_image = input_image.convert(output_image_mode)
    else:
        output_image = input_image.copy()

    for x in range(input_image.width):
        for y in range(input_image.height):
            if input_image.getpixel((x, y)) == original:
                output_image.putpixel((x, y), replacement)

    return output_image