Skip to content

Commit

Permalink
Moved Anchor and Direction into ImageFont
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed Jan 26, 2025
1 parent 0e0515c commit eed5926
Show file tree
Hide file tree
Showing 8 changed files with 68 additions and 60 deletions.
8 changes: 4 additions & 4 deletions Tests/test_imagefont.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
from packaging.version import parse as parse_version

from PIL import Image, ImageDraw, ImageFont, features
from PIL._typing import Anchor, StrOrBytesPath
from PIL._typing import StrOrBytesPath

from .helper import (
assert_image_equal,
Expand Down Expand Up @@ -795,7 +795,7 @@ def test_variation_set_by_axes(font: ImageFont.FreeTypeFont) -> None:
ids=("ls", "ms", "rs", "ma", "mt", "mm", "mb", "md"),
)
def test_anchor(
layout_engine: ImageFont.Layout, anchor: Anchor, left: int, top: int
layout_engine: ImageFont.Layout, anchor: ImageFont.Anchor, left: int, top: int
) -> None:
name, text = "quick", "Quick"
path = f"Tests/images/test_anchor_{name}_{anchor}.png"
Expand Down Expand Up @@ -842,7 +842,7 @@ def test_anchor(
),
)
def test_anchor_multiline(
layout_engine: ImageFont.Layout, anchor: Anchor, align: ImageDraw.Align
layout_engine: ImageFont.Layout, anchor: ImageFont.Anchor, align: ImageDraw.Align
) -> None:
target = f"Tests/images/test_anchor_multiline_{anchor}_{align}.png"
text = "a\nlong\ntext sample"
Expand Down Expand Up @@ -880,7 +880,7 @@ def test_anchor_invalid(font: ImageFont.FreeTypeFont) -> None:
with pytest.raises(ValueError):
d.multiline_textbbox((0, 0), "foo\nbar", anchor=anchor) # type: ignore[arg-type]

anchors: list[Anchor] = ["lt", "lb"]
anchors: list[ImageFont.Anchor] = ["lt", "lb"]
for anchor2 in anchors:
with pytest.raises(ValueError):
d.multiline_text((0, 0), "foo\nbar", anchor=anchor2)
Expand Down
15 changes: 10 additions & 5 deletions Tests/test_imagefontctl.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import pytest

from PIL import Image, ImageDraw, ImageFont
from PIL._typing import Anchor, Direction

from .helper import assert_image_similar_tofile, skip_unless_feature

Expand Down Expand Up @@ -233,7 +232,9 @@ def test_getlength(
("i" + ("\u030C" * 15) + "i", "i" + "\u032C" * 15 + "i", "\u035Cii", "i\u0305i"),
ids=("caron-above", "caron-below", "double-breve", "overline"),
)
def test_getlength_combine(mode: str, direction: Direction, text: str) -> None:
def test_getlength_combine(
mode: str, direction: ImageFont.Direction, text: str
) -> None:
if text == "i\u0305i" and direction == "ttb":
pytest.skip("fails with this font")

Expand All @@ -253,7 +254,7 @@ def test_getlength_combine(mode: str, direction: Direction, text: str) -> None:


@pytest.mark.parametrize("anchor", ("lt", "mm", "rb", "sm"))
def test_anchor_ttb(anchor: Anchor) -> None:
def test_anchor_ttb(anchor: ImageFont.Anchor) -> None:
text = "f"
path = f"Tests/images/test_anchor_ttb_{text}_{anchor}.png"
f = ImageFont.truetype("Tests/fonts/NotoSans-Regular.ttf", 120)
Expand Down Expand Up @@ -310,7 +311,11 @@ def test_anchor_ttb(anchor: Anchor) -> None:
"name, text, anchor, dir, epsilon", combine_tests, ids=[r[0] for r in combine_tests]
)
def test_combine(
name: str, text: str, anchor: Anchor | None, dir: Direction | None, epsilon: float
name: str,
text: str,
anchor: ImageFont.Anchor | None,
dir: ImageFont.Direction | None,
epsilon: float,
) -> None:
path = f"Tests/images/test_combine_{name}.png"
f = ImageFont.truetype("Tests/fonts/NotoSans-Regular.ttf", 48)
Expand Down Expand Up @@ -342,7 +347,7 @@ def test_combine(
("rm", "right"), # pass with getsize
),
)
def test_combine_multiline(anchor: Anchor, align: ImageDraw.Align) -> None:
def test_combine_multiline(anchor: ImageFont.Anchor, align: ImageDraw.Align) -> None:
# test that multiline text uses getlength, not getsize or getbbox

path = f"Tests/images/test_combine_multiline_{anchor}_{align}.png"
Expand Down
5 changes: 2 additions & 3 deletions docs/example/anchors.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,11 @@
from __future__ import annotations

from PIL import Image, ImageDraw, ImageFont
from PIL._typing import Anchor

font = ImageFont.truetype("Tests/fonts/NotoSans-Regular.ttf", 16)


def test(anchor: Anchor) -> Image.Image:
def test(anchor: ImageFont.Anchor) -> Image.Image:
im = Image.new("RGBA", (200, 100), "white")
d = ImageDraw.Draw(im)
d.line(((100, 0), (100, 100)), "gray")
Expand All @@ -18,7 +17,7 @@ def test(anchor: Anchor) -> Image.Image:
if __name__ == "__main__":
im = Image.new("RGBA", (600, 300), "white")
d = ImageDraw.Draw(im)
anchors: list[list[Anchor]] = [
anchors: list[list[ImageFont.Anchor]] = [
["ma", "mt", "mm"],
["ms", "mb", "md"],
["ls", "ms", "rs"],
Expand Down
5 changes: 5 additions & 0 deletions docs/reference/ImageFont.rst
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,11 @@ Constants
raise a :py:exc:`ValueError` if the number of characters is over this limit. The
check can be disabled by setting ``ImageFont.MAX_STRING_LENGTH = None``.

.. class:: Anchor

Type hint literal with the possible anchor values. See :ref:`text-anchors` for
details.

Dictionaries
------------

Expand Down
20 changes: 10 additions & 10 deletions src/PIL/ImageDraw.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@

from . import Image, ImageColor
from ._deprecate import deprecate
from ._typing import Anchor, Coords, Direction
from ._typing import Coords

# experimental access to the outline API
Outline: Callable[[], Image.core._Outline] | None
Expand Down Expand Up @@ -585,10 +585,10 @@ def text(
| ImageFont.TransposedFont
| None
) = None,
anchor: Anchor | None = None,
anchor: ImageFont.Anchor | None = None,
spacing: float = 4,
align: Align = "left",
direction: Direction | None = None,
direction: ImageFont.Direction | None = None,
features: list[str] | None = None,
language: str | None = None,
stroke_width: float = 0,
Expand Down Expand Up @@ -710,10 +710,10 @@ def multiline_text(
| ImageFont.TransposedFont
| None
) = None,
anchor: Anchor | None = None,
anchor: ImageFont.Anchor | None = None,
spacing: float = 4,
align: Align = "left",
direction: Direction | None = None,
direction: ImageFont.Direction | None = None,
features: list[str] | None = None,
language: str | None = None,
stroke_width: float = 0,
Expand Down Expand Up @@ -800,7 +800,7 @@ def textlength(
| ImageFont.TransposedFont
| None
) = None,
direction: Direction | None = None,
direction: ImageFont.Direction | None = None,
features: list[str] | None = None,
language: str | None = None,
embedded_color: bool = False,
Expand Down Expand Up @@ -830,10 +830,10 @@ def textbbox(
| ImageFont.TransposedFont
| None
) = None,
anchor: Anchor | None = None,
anchor: ImageFont.Anchor | None = None,
spacing: float = 4,
align: Align = "left",
direction: Direction | None = None,
direction: ImageFont.Direction | None = None,
features: list[str] | None = None,
language: str | None = None,
stroke_width: float = 0,
Expand Down Expand Up @@ -880,10 +880,10 @@ def multiline_textbbox(
| ImageFont.TransposedFont
| None
) = None,
anchor: Anchor | None = None,
anchor: ImageFont.Anchor | None = None,
spacing: float = 4,
align: Align = "left",
direction: Direction | None = None,
direction: ImageFont.Direction | None = None,
features: list[str] | None = None,
language: str | None = None,
stroke_width: float = 0,
Expand Down
33 changes: 31 additions & 2 deletions src/PIL/ImageFont.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,17 +34,46 @@
from enum import IntEnum
from io import BytesIO
from types import ModuleType
from typing import IO, TYPE_CHECKING, Any, BinaryIO, TypedDict, cast
from typing import IO, TYPE_CHECKING, Any, BinaryIO, Literal, TypedDict, cast

from . import Image, features
from ._typing import Anchor, Direction, StrOrBytesPath
from ._typing import StrOrBytesPath
from ._util import DeferredError, is_path

if TYPE_CHECKING:
from . import ImageFile
from ._imaging import ImagingFont
from ._imagingft import Font

Anchor = Literal[
"la",
"lt",
"lm",
"ls",
"lb",
"ld",
"ma",
"mt",
"mm",
"ms",
"mb",
"md",
"ra",
"rt",
"rm",
"rs",
"rb",
"rd",
"sa",
"st",
"sm",
"ss",
"sb",
"sd",
]

Direction = Literal["rtl", "ltr", "ttb"]


class Axis(TypedDict):
minimum: int | None
Expand Down
11 changes: 5 additions & 6 deletions src/PIL/_imagingft.pyi
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
from typing import Any, Callable

from . import ImageFont, _imaging
from ._typing import Anchor, Direction

class Font:
@property
Expand All @@ -25,11 +24,11 @@ class Font:
string: str | bytes,
fill: Callable[[int, int], _imaging.ImagingCore],
mode: str,
dir: Direction | None,
dir: ImageFont.Direction | None,
features: list[str] | None,
lang: str | None,
stroke_width: float,
anchor: Anchor | None,
anchor: ImageFont.Anchor | None,
foreground_ink_long: int,
x_start: float,
y_start: float,
Expand All @@ -39,17 +38,17 @@ class Font:
self,
string: str | bytes | bytearray,
mode: str,
dir: Direction | None,
dir: ImageFont.Direction | None,
features: list[str] | None,
lang: str | None,
anchor: Anchor | None,
anchor: ImageFont.Anchor | None,
/,
) -> tuple[tuple[int, int], tuple[int, int]]: ...
def getlength(
self,
string: str | bytes,
mode: str,
dir: Direction | None,
dir: ImageFont.Direction | None,
features: list[str] | None,
lang: str | None,
/,
Expand Down
31 changes: 1 addition & 30 deletions src/PIL/_typing.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import os
import sys
from collections.abc import Sequence
from typing import TYPE_CHECKING, Any, Literal, Protocol, TypeVar, Union
from typing import TYPE_CHECKING, Any, Protocol, TypeVar, Union

if TYPE_CHECKING:
from numbers import _IntegralLike as IntegralLike
Expand Down Expand Up @@ -49,33 +49,4 @@ def read(self, length: int = ..., /) -> _T_co: ...

StrOrBytesPath = Union[str, bytes, os.PathLike[str], os.PathLike[bytes]]

Anchor = Literal[
"la",
"lt",
"lm",
"ls",
"lb",
"ld",
"ma",
"mt",
"mm",
"ms",
"mb",
"md",
"ra",
"rt",
"rm",
"rs",
"rb",
"rd",
"sa",
"st",
"sm",
"ss",
"sb",
"sd",
]

Direction = Literal["rtl", "ltr", "ttb"]

__all__ = ["Buffer", "IntegralLike", "StrOrBytesPath", "SupportsRead", "TypeGuard"]

0 comments on commit eed5926

Please sign in to comment.