Skip to content

Commit

Permalink
Added ImageType as Image.Image TypeAlias
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed Dec 25, 2023
1 parent df83428 commit 425fc3f
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 24 deletions.
3 changes: 3 additions & 0 deletions src/PIL/Image.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@
from collections.abc import Callable, MutableMapping
from enum import IntEnum
from pathlib import Path
from typing import TypeAlias

try:
from defusedxml import ElementTree
Expand All @@ -60,6 +61,8 @@
from ._binary import i32le, o32be, o32le
from ._util import DeferredError, is_path

ImageType: TypeAlias = Image.Image

logger = logging.getLogger(__name__)


Expand Down
60 changes: 36 additions & 24 deletions src/PIL/ImageChops.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
from . import Image


def constant(image: Image.Image, value: int) -> Image.Image:
def constant(image: Image.ImageType, value: int) -> Image.ImageType:
"""Fill a channel with a given gray level.
:rtype: :py:class:`~PIL.Image.Image`
Expand All @@ -29,7 +29,7 @@ def constant(image: Image.Image, value: int) -> Image.Image:
return Image.new("L", image.size, value)


def duplicate(image: Image.Image) -> Image.Image:
def duplicate(image: Image.ImageType) -> Image.ImageType:
"""Copy a channel. Alias for :py:meth:`PIL.Image.Image.copy`.
:rtype: :py:class:`~PIL.Image.Image`
Expand All @@ -38,7 +38,7 @@ def duplicate(image: Image.Image) -> Image.Image:
return image.copy()


def invert(image: Image.Image) -> Image.Image:
def invert(image: Image.ImageType) -> Image.ImageType:
"""
Invert an image (channel). ::
Expand All @@ -51,7 +51,7 @@ def invert(image: Image.Image) -> Image.Image:
return image._new(image.im.chop_invert())


def lighter(image1: Image.Image, image2: Image.Image) -> Image.Image:
def lighter(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Compares the two images, pixel by pixel, and returns a new image containing
the lighter values. ::
Expand All @@ -66,7 +66,7 @@ def lighter(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_lighter(image2.im))


def darker(image1: Image.Image, image2: Image.Image) -> Image.Image:
def darker(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Compares the two images, pixel by pixel, and returns a new image containing
the darker values. ::
Expand All @@ -81,7 +81,7 @@ def darker(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_darker(image2.im))


def difference(image1: Image.Image, image2: Image.Image) -> Image.Image:
def difference(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Returns the absolute value of the pixel-by-pixel difference between the two
images. ::
Expand All @@ -96,7 +96,7 @@ def difference(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_difference(image2.im))


def multiply(image1: Image.Image, image2: Image.Image) -> Image.Image:
def multiply(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Superimposes two images on top of each other.
Expand All @@ -113,7 +113,7 @@ def multiply(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_multiply(image2.im))


def screen(image1: Image.Image, image2: Image.Image) -> Image.Image:
def screen(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Superimposes two inverted images on top of each other. ::
Expand All @@ -127,7 +127,7 @@ def screen(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_screen(image2.im))


def soft_light(image1: Image.Image, image2: Image.Image) -> Image.Image:
def soft_light(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Superimposes two images on top of each other using the Soft Light algorithm
Expand All @@ -139,7 +139,7 @@ def soft_light(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_soft_light(image2.im))


def hard_light(image1: Image.Image, image2: Image.Image) -> Image.Image:
def hard_light(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Superimposes two images on top of each other using the Hard Light algorithm
Expand All @@ -151,7 +151,7 @@ def hard_light(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_hard_light(image2.im))


def overlay(image1: Image.Image, image2: Image.Image) -> Image.Image:
def overlay(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""
Superimposes two images on top of each other using the Overlay algorithm
Expand All @@ -164,8 +164,11 @@ def overlay(image1: Image.Image, image2: Image.Image) -> Image.Image:


def add(
image1: Image.Image, image2: Image.Image, scale: float = 1.0, offset: float = 0
) -> Image.Image:
image1: Image.ImageType,
image2: Image.ImageType,
scale: float = 1.0,
offset: float = 0,
) -> Image.ImageType:
"""
Adds two images, dividing the result by scale and adding the
offset. If omitted, scale defaults to 1.0, and offset to 0.0. ::
Expand All @@ -181,8 +184,11 @@ def add(


def subtract(
image1: Image.Image, image2: Image.Image, scale: float = 1.0, offset: float = 0
) -> Image.Image:
image1: Image.ImageType,
image2: Image.ImageType,
scale: float = 1.0,
offset: float = 0,
) -> Image.ImageType:
"""
Subtracts two images, dividing the result by scale and adding the offset.
If omitted, scale defaults to 1.0, and offset to 0.0. ::
Expand All @@ -197,7 +203,7 @@ def subtract(
return image1._new(image1.im.chop_subtract(image2.im, scale, offset))


def add_modulo(image1: Image.Image, image2: Image.Image) -> Image.Image:
def add_modulo(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""Add two images, without clipping the result. ::
out = ((image1 + image2) % MAX)
Expand All @@ -210,7 +216,9 @@ def add_modulo(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_add_modulo(image2.im))


def subtract_modulo(image1: Image.Image, image2: Image.Image) -> Image.Image:
def subtract_modulo(
image1: Image.ImageType, image2: Image.ImageType
) -> Image.ImageType:
"""Subtract two images, without clipping the result. ::
out = ((image1 - image2) % MAX)
Expand All @@ -223,7 +231,7 @@ def subtract_modulo(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_subtract_modulo(image2.im))


def logical_and(image1: Image.Image, image2: Image.Image) -> Image.Image:
def logical_and(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""Logical AND between two images.
Both of the images must have mode "1". If you would like to perform a
Expand All @@ -241,7 +249,7 @@ def logical_and(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_and(image2.im))


def logical_or(image1: Image.Image, image2: Image.Image) -> Image.Image:
def logical_or(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""Logical OR between two images.
Both of the images must have mode "1". ::
Expand All @@ -256,7 +264,7 @@ def logical_or(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_or(image2.im))


def logical_xor(image1: Image.Image, image2: Image.Image) -> Image.Image:
def logical_xor(image1: Image.ImageType, image2: Image.ImageType) -> Image.ImageType:
"""Logical XOR between two images.
Both of the images must have mode "1". ::
Expand All @@ -271,7 +279,9 @@ def logical_xor(image1: Image.Image, image2: Image.Image) -> Image.Image:
return image1._new(image1.im.chop_xor(image2.im))


def blend(image1: Image.Image, image2: Image.Image, alpha: float) -> Image.Image:
def blend(
image1: Image.ImageType, image2: Image.ImageType, alpha: float
) -> Image.ImageType:
"""Blend images using constant transparency weight. Alias for
:py:func:`PIL.Image.blend`.
Expand All @@ -282,8 +292,8 @@ def blend(image1: Image.Image, image2: Image.Image, alpha: float) -> Image.Image


def composite(
image1: Image.Image, image2: Image.Image, mask: Image.Image
) -> Image.Image:
image1: Image.ImageType, image2: Image.ImageType, mask: Image.ImageType
) -> Image.ImageType:
"""Create composite using transparency mask. Alias for
:py:func:`PIL.Image.composite`.
Expand All @@ -293,7 +303,9 @@ def composite(
return Image.composite(image1, image2, mask)


def offset(image: Image.Image, xoffset: int, yoffset: int | None = None) -> Image.Image:
def offset(
image: Image.ImageType, xoffset: int, yoffset: int | None = None
) -> Image.ImageType:
"""Returns a copy of the image where data has been offset by the given
distances. Data wraps around the edges. If ``yoffset`` is omitted, it
is assumed to be equal to ``xoffset``.
Expand Down

0 comments on commit 425fc3f

Please sign in to comment.