Spaces:
Running
Running
"""Routines for calculating bounding boxes, point in rectangle calculations and | |
so on. | |
""" | |
from fontTools.misc.roundTools import otRound | |
from fontTools.misc.vector import Vector as _Vector | |
import math | |
import warnings | |
def calcBounds(array): | |
"""Calculate the bounding rectangle of a 2D points array. | |
Args: | |
array: A sequence of 2D tuples. | |
Returns: | |
A four-item tuple representing the bounding rectangle ``(xMin, yMin, xMax, yMax)``. | |
""" | |
if not array: | |
return 0, 0, 0, 0 | |
xs = [x for x, y in array] | |
ys = [y for x, y in array] | |
return min(xs), min(ys), max(xs), max(ys) | |
def calcIntBounds(array, round=otRound): | |
"""Calculate the integer bounding rectangle of a 2D points array. | |
Values are rounded to closest integer towards ``+Infinity`` using the | |
:func:`fontTools.misc.fixedTools.otRound` function by default, unless | |
an optional ``round`` function is passed. | |
Args: | |
array: A sequence of 2D tuples. | |
round: A rounding function of type ``f(x: float) -> int``. | |
Returns: | |
A four-item tuple of integers representing the bounding rectangle: | |
``(xMin, yMin, xMax, yMax)``. | |
""" | |
return tuple(round(v) for v in calcBounds(array)) | |
def updateBounds(bounds, p, min=min, max=max): | |
"""Add a point to a bounding rectangle. | |
Args: | |
bounds: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax), or None``. | |
p: A 2D tuple representing a point. | |
min,max: functions to compute the minimum and maximum. | |
Returns: | |
The updated bounding rectangle ``(xMin, yMin, xMax, yMax)``. | |
""" | |
(x, y) = p | |
if bounds is None: | |
return x, y, x, y | |
xMin, yMin, xMax, yMax = bounds | |
return min(xMin, x), min(yMin, y), max(xMax, x), max(yMax, y) | |
def pointInRect(p, rect): | |
"""Test if a point is inside a bounding rectangle. | |
Args: | |
p: A 2D tuple representing a point. | |
rect: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax)``. | |
Returns: | |
``True`` if the point is inside the rectangle, ``False`` otherwise. | |
""" | |
(x, y) = p | |
xMin, yMin, xMax, yMax = rect | |
return (xMin <= x <= xMax) and (yMin <= y <= yMax) | |
def pointsInRect(array, rect): | |
"""Determine which points are inside a bounding rectangle. | |
Args: | |
array: A sequence of 2D tuples. | |
rect: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax)``. | |
Returns: | |
A list containing the points inside the rectangle. | |
""" | |
if len(array) < 1: | |
return [] | |
xMin, yMin, xMax, yMax = rect | |
return [(xMin <= x <= xMax) and (yMin <= y <= yMax) for x, y in array] | |
def vectorLength(vector): | |
"""Calculate the length of the given vector. | |
Args: | |
vector: A 2D tuple. | |
Returns: | |
The Euclidean length of the vector. | |
""" | |
x, y = vector | |
return math.sqrt(x**2 + y**2) | |
def asInt16(array): | |
"""Round a list of floats to 16-bit signed integers. | |
Args: | |
array: List of float values. | |
Returns: | |
A list of rounded integers. | |
""" | |
return [int(math.floor(i + 0.5)) for i in array] | |
def normRect(rect): | |
"""Normalize a bounding box rectangle. | |
This function "turns the rectangle the right way up", so that the following | |
holds:: | |
xMin <= xMax and yMin <= yMax | |
Args: | |
rect: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax)``. | |
Returns: | |
A normalized bounding rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
return min(xMin, xMax), min(yMin, yMax), max(xMin, xMax), max(yMin, yMax) | |
def scaleRect(rect, x, y): | |
"""Scale a bounding box rectangle. | |
Args: | |
rect: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax)``. | |
x: Factor to scale the rectangle along the X axis. | |
Y: Factor to scale the rectangle along the Y axis. | |
Returns: | |
A scaled bounding rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
return xMin * x, yMin * y, xMax * x, yMax * y | |
def offsetRect(rect, dx, dy): | |
"""Offset a bounding box rectangle. | |
Args: | |
rect: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax)``. | |
dx: Amount to offset the rectangle along the X axis. | |
dY: Amount to offset the rectangle along the Y axis. | |
Returns: | |
An offset bounding rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
return xMin + dx, yMin + dy, xMax + dx, yMax + dy | |
def insetRect(rect, dx, dy): | |
"""Inset a bounding box rectangle on all sides. | |
Args: | |
rect: A bounding rectangle expressed as a tuple | |
``(xMin, yMin, xMax, yMax)``. | |
dx: Amount to inset the rectangle along the X axis. | |
dY: Amount to inset the rectangle along the Y axis. | |
Returns: | |
An inset bounding rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
return xMin + dx, yMin + dy, xMax - dx, yMax - dy | |
def sectRect(rect1, rect2): | |
"""Test for rectangle-rectangle intersection. | |
Args: | |
rect1: First bounding rectangle, expressed as tuples | |
``(xMin, yMin, xMax, yMax)``. | |
rect2: Second bounding rectangle. | |
Returns: | |
A boolean and a rectangle. | |
If the input rectangles intersect, returns ``True`` and the intersecting | |
rectangle. Returns ``False`` and ``(0, 0, 0, 0)`` if the input | |
rectangles don't intersect. | |
""" | |
(xMin1, yMin1, xMax1, yMax1) = rect1 | |
(xMin2, yMin2, xMax2, yMax2) = rect2 | |
xMin, yMin, xMax, yMax = ( | |
max(xMin1, xMin2), | |
max(yMin1, yMin2), | |
min(xMax1, xMax2), | |
min(yMax1, yMax2), | |
) | |
if xMin >= xMax or yMin >= yMax: | |
return False, (0, 0, 0, 0) | |
return True, (xMin, yMin, xMax, yMax) | |
def unionRect(rect1, rect2): | |
"""Determine union of bounding rectangles. | |
Args: | |
rect1: First bounding rectangle, expressed as tuples | |
``(xMin, yMin, xMax, yMax)``. | |
rect2: Second bounding rectangle. | |
Returns: | |
The smallest rectangle in which both input rectangles are fully | |
enclosed. | |
""" | |
(xMin1, yMin1, xMax1, yMax1) = rect1 | |
(xMin2, yMin2, xMax2, yMax2) = rect2 | |
xMin, yMin, xMax, yMax = ( | |
min(xMin1, xMin2), | |
min(yMin1, yMin2), | |
max(xMax1, xMax2), | |
max(yMax1, yMax2), | |
) | |
return (xMin, yMin, xMax, yMax) | |
def rectCenter(rect): | |
"""Determine rectangle center. | |
Args: | |
rect: Bounding rectangle, expressed as tuples | |
``(xMin, yMin, xMax, yMax)``. | |
Returns: | |
A 2D tuple representing the point at the center of the rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
return (xMin + xMax) / 2, (yMin + yMax) / 2 | |
def rectArea(rect): | |
"""Determine rectangle area. | |
Args: | |
rect: Bounding rectangle, expressed as tuples | |
``(xMin, yMin, xMax, yMax)``. | |
Returns: | |
The area of the rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
return (yMax - yMin) * (xMax - xMin) | |
def intRect(rect): | |
"""Round a rectangle to integer values. | |
Guarantees that the resulting rectangle is NOT smaller than the original. | |
Args: | |
rect: Bounding rectangle, expressed as tuples | |
``(xMin, yMin, xMax, yMax)``. | |
Returns: | |
A rounded bounding rectangle. | |
""" | |
(xMin, yMin, xMax, yMax) = rect | |
xMin = int(math.floor(xMin)) | |
yMin = int(math.floor(yMin)) | |
xMax = int(math.ceil(xMax)) | |
yMax = int(math.ceil(yMax)) | |
return (xMin, yMin, xMax, yMax) | |
def quantizeRect(rect, factor=1): | |
""" | |
>>> bounds = (72.3, -218.4, 1201.3, 919.1) | |
>>> quantizeRect(bounds) | |
(72, -219, 1202, 920) | |
>>> quantizeRect(bounds, factor=10) | |
(70, -220, 1210, 920) | |
>>> quantizeRect(bounds, factor=100) | |
(0, -300, 1300, 1000) | |
""" | |
if factor < 1: | |
raise ValueError(f"Expected quantization factor >= 1, found: {factor!r}") | |
xMin, yMin, xMax, yMax = normRect(rect) | |
return ( | |
int(math.floor(xMin / factor) * factor), | |
int(math.floor(yMin / factor) * factor), | |
int(math.ceil(xMax / factor) * factor), | |
int(math.ceil(yMax / factor) * factor), | |
) | |
class Vector(_Vector): | |
def __init__(self, *args, **kwargs): | |
warnings.warn( | |
"fontTools.misc.arrayTools.Vector has been deprecated, please use " | |
"fontTools.misc.vector.Vector instead.", | |
DeprecationWarning, | |
) | |
def pairwise(iterable, reverse=False): | |
"""Iterate over current and next items in iterable. | |
Args: | |
iterable: An iterable | |
reverse: If true, iterate in reverse order. | |
Returns: | |
A iterable yielding two elements per iteration. | |
Example: | |
>>> tuple(pairwise([])) | |
() | |
>>> tuple(pairwise([], reverse=True)) | |
() | |
>>> tuple(pairwise([0])) | |
((0, 0),) | |
>>> tuple(pairwise([0], reverse=True)) | |
((0, 0),) | |
>>> tuple(pairwise([0, 1])) | |
((0, 1), (1, 0)) | |
>>> tuple(pairwise([0, 1], reverse=True)) | |
((1, 0), (0, 1)) | |
>>> tuple(pairwise([0, 1, 2])) | |
((0, 1), (1, 2), (2, 0)) | |
>>> tuple(pairwise([0, 1, 2], reverse=True)) | |
((2, 1), (1, 0), (0, 2)) | |
>>> tuple(pairwise(['a', 'b', 'c', 'd'])) | |
(('a', 'b'), ('b', 'c'), ('c', 'd'), ('d', 'a')) | |
>>> tuple(pairwise(['a', 'b', 'c', 'd'], reverse=True)) | |
(('d', 'c'), ('c', 'b'), ('b', 'a'), ('a', 'd')) | |
""" | |
if not iterable: | |
return | |
if reverse: | |
it = reversed(iterable) | |
else: | |
it = iter(iterable) | |
first = next(it, None) | |
a = first | |
for b in it: | |
yield (a, b) | |
a = b | |
yield (a, first) | |
def _test(): | |
""" | |
>>> import math | |
>>> calcBounds([]) | |
(0, 0, 0, 0) | |
>>> calcBounds([(0, 40), (0, 100), (50, 50), (80, 10)]) | |
(0, 10, 80, 100) | |
>>> updateBounds((0, 0, 0, 0), (100, 100)) | |
(0, 0, 100, 100) | |
>>> pointInRect((50, 50), (0, 0, 100, 100)) | |
True | |
>>> pointInRect((0, 0), (0, 0, 100, 100)) | |
True | |
>>> pointInRect((100, 100), (0, 0, 100, 100)) | |
True | |
>>> not pointInRect((101, 100), (0, 0, 100, 100)) | |
True | |
>>> list(pointsInRect([(50, 50), (0, 0), (100, 100), (101, 100)], (0, 0, 100, 100))) | |
[True, True, True, False] | |
>>> vectorLength((3, 4)) | |
5.0 | |
>>> vectorLength((1, 1)) == math.sqrt(2) | |
True | |
>>> list(asInt16([0, 0.1, 0.5, 0.9])) | |
[0, 0, 1, 1] | |
>>> normRect((0, 10, 100, 200)) | |
(0, 10, 100, 200) | |
>>> normRect((100, 200, 0, 10)) | |
(0, 10, 100, 200) | |
>>> scaleRect((10, 20, 50, 150), 1.5, 2) | |
(15.0, 40, 75.0, 300) | |
>>> offsetRect((10, 20, 30, 40), 5, 6) | |
(15, 26, 35, 46) | |
>>> insetRect((10, 20, 50, 60), 5, 10) | |
(15, 30, 45, 50) | |
>>> insetRect((10, 20, 50, 60), -5, -10) | |
(5, 10, 55, 70) | |
>>> intersects, rect = sectRect((0, 10, 20, 30), (0, 40, 20, 50)) | |
>>> not intersects | |
True | |
>>> intersects, rect = sectRect((0, 10, 20, 30), (5, 20, 35, 50)) | |
>>> intersects | |
1 | |
>>> rect | |
(5, 20, 20, 30) | |
>>> unionRect((0, 10, 20, 30), (0, 40, 20, 50)) | |
(0, 10, 20, 50) | |
>>> rectCenter((0, 0, 100, 200)) | |
(50.0, 100.0) | |
>>> rectCenter((0, 0, 100, 199.0)) | |
(50.0, 99.5) | |
>>> intRect((0.9, 2.9, 3.1, 4.1)) | |
(0, 2, 4, 5) | |
""" | |
if __name__ == "__main__": | |
import sys | |
import doctest | |
sys.exit(doctest.testmod().failed) | |