from pathlib import Path from test.conftest import assert_pdf_equal import pytest from fpdf import FPDF from fpdf.drawing import ( Arc, BezierCurve, BoundingBox, Ellipse, HorizontalLine, Line, PaintedPath, Point, QuadraticBezierCurve, Rectangle, RelativeArc, RelativeBezierCurve, RelativeHorizontalLine, RelativeLine, RelativeQuadraticBezierCurve, RelativeVerticalLine, RoundedRectangle, Transform, VerticalLine, ) from fpdf.enums import PathPaintRule HERE = Path(__file__).resolve().parent def rotate_around_center(bbox: BoundingBox) -> Transform: """ Create a transformation that rotates the bounding box around its center. """ center_x = (bbox.x0 + bbox.x1) / 2 center_y = (bbox.y0 + bbox.y1) / 2 return ( Transform.translation(center_x, center_y) .rotate(45) .translate(-center_x + 150, -center_y) ) TRANSFORMS = { "identity": Transform.identity(), "scaled": Transform.scaling(2, 3), "rotated": rotate_around_center, "translated": Transform.translation(50, 100), } @pytest.mark.parametrize( "shape, start, expected_bbox", [ ( Line(pt=Point(50, 20)), Point(10, 20), BoundingBox(10, 20, 50, 20), ), ( Line(pt=Point(30, 80)), Point(30, 10), BoundingBox(30, 10, 30, 80), ), ( Line(pt=Point(100, 100)), Point(50, 50), BoundingBox(50, 50, 100, 100), ), ( Line(pt=Point(50, 50)), Point(100, 100), BoundingBox(50, 50, 100, 100), ), ( Line(pt=Point(5, 5)), Point(-10, -5), BoundingBox(-10, -5, 5, 5), ), (Line(pt=Point(42, 42)), Point(42, 42), BoundingBox(42, 42, 42, 42)), ( RelativeLine(pt=Point(40, 0)), Point(10, 20), BoundingBox(10, 20, 50, 20), ), ( RelativeLine(pt=Point(0, 60)), Point(30, 10), BoundingBox(30, 10, 30, 70), ), ( RelativeLine(pt=Point(100, 100)), Point(0, 0), BoundingBox(0, 0, 100, 100), ), ( RelativeLine(pt=Point(-100, -100)), Point(100, 100), BoundingBox(0, 0, 100, 100), ), ( RelativeLine(pt=Point(15, 10)), Point(-10, -5), BoundingBox(-10, -5, 5, 5), ), ( RelativeLine(pt=Point(0, 0)), Point(42, 42), BoundingBox(42, 42, 42, 42), ), (HorizontalLine(x=50), Point(10, 20), BoundingBox(10, 20, 50, 20)), (HorizontalLine(x=20), Point(50, 15), BoundingBox(20, 15, 50, 15)), ( HorizontalLine(x=42), Point(42, 42), BoundingBox(42, 42, 42, 42), ), ( RelativeHorizontalLine(x=40), Point(10, 20), BoundingBox(10, 20, 50, 20), ), ( RelativeHorizontalLine(x=-30), Point(20, 15), BoundingBox(-10, 15, 20, 15), ), ( RelativeHorizontalLine(x=0), Point(42, 42), BoundingBox(42, 42, 42, 42), ), (VerticalLine(y=80), Point(30, 10), BoundingBox(30, 10, 30, 80)), (VerticalLine(y=20), Point(20, 50), BoundingBox(20, 20, 20, 50)), (VerticalLine(y=42), Point(42, 42), BoundingBox(42, 42, 42, 42)), (RelativeVerticalLine(y=50), Point(30, 10), BoundingBox(30, 10, 30, 60)), ( RelativeVerticalLine(y=-30), Point(20, 50), BoundingBox(20, 20, 20, 50), ), ( RelativeVerticalLine(y=0), Point(42, 42), BoundingBox(42, 42, 42, 42), ), ( BezierCurve(c1=Point(30, 0), c2=Point(70, 0), end=Point(100, 0)), Point(0, 0), BoundingBox(0, 0, 100, 0), ), ( BezierCurve(c1=Point(0, 30), c2=Point(0, 70), end=Point(0, 100)), Point(0, 0), BoundingBox(0, 0, 0, 100), ), ( BezierCurve(c1=Point(100, 0), c2=Point(0, 100), end=Point(100, 100)), Point(0, 0), BoundingBox(0, 0, 100, 100), ), ( BezierCurve(c1=Point(70, 100), c2=Point(30, 100), end=Point(0, 100)), Point(100, 100), BoundingBox(0, 100, 100, 100), ), ( BezierCurve(c1=Point(150, 0), c2=Point(-150, 100), end=Point(100, 100)), Point(0, 0), BoundingBox(-3.808522398518157, 0.0, 100.0, 100.0), ), ( BezierCurve(c1=Point(25, 25), c2=Point(50, 50), end=Point(75, 75)), Point(0, 0), BoundingBox(0, 0, 75, 75), ), ( BezierCurve(c1=Point(42, 42), c2=Point(42, 42), end=Point(42, 42)), Point(42, 42), BoundingBox(42, 42, 42, 42), ), ( RelativeBezierCurve( c1=Point(30, 60), c2=Point(60, 90), end=Point(100, 100) ), Point(0, 0), BoundingBox(0, 0, 100, 100), ), ( RelativeBezierCurve( c1=Point(20, -20), c2=Point(-30, 30), end=Point(-50, -50) ), Point(25, 25), BoundingBox(-25.0, -25.0, 29.63395588138013, 25.0), ), ( RelativeBezierCurve(c1=Point(0, 0), c2=Point(0, 0), end=Point(50, 50)), Point(0, 0), BoundingBox(0, 0, 50, 50), ), ( RelativeBezierCurve(c1=Point(0, 0), c2=Point(0, 0), end=Point(0, 0)), Point(10, 10), BoundingBox(10, 10, 10, 10), ), ( RelativeBezierCurve(c1=Point(50, 1), c2=Point(100, 2), end=Point(150, 1)), Point(0, 0), BoundingBox(0.0, 0.0, 150.0, 1.4142135623730951), ), ( RelativeBezierCurve(c1=Point(1, 50), c2=Point(2, 100), end=Point(1, 150)), Point(0, 0), BoundingBox(0.0, 0.0, 1.4142135623730951, 150.0), ), ( QuadraticBezierCurve(ctrl=Point(50, 100), end=Point(100, 0)), Point(0, 0), BoundingBox(0, 0, 100, 50), ), ( QuadraticBezierCurve(ctrl=Point(50, 100), end=Point(100, 0)), Point(0, 0), BoundingBox(0, 0, 100, 50), ), ( QuadraticBezierCurve(ctrl=Point(50, 0), end=Point(100, 0)), Point(0, 0), BoundingBox(0, 0, 100, 0), ), ( QuadraticBezierCurve(ctrl=Point(50, 50), end=Point(100, 100)), Point(0, 0), BoundingBox(0, 0, 100, 100), ), ( QuadraticBezierCurve(ctrl=Point(25, -100), end=Point(50, 0)), Point(0, 0), BoundingBox(0, -50, 50, 0), ), ( QuadraticBezierCurve(ctrl=Point(42, 42), end=Point(42, 42)), Point(42, 42), BoundingBox(42, 42, 42, 42), ), ( RelativeQuadraticBezierCurve(ctrl=Point(50, 100), end=Point(100, 0)), Point(0, 0), BoundingBox(0, 0, 100, 50), ), ( RelativeQuadraticBezierCurve(ctrl=Point(50, 0), end=Point(100, 0)), Point(0, 0), BoundingBox(0, 0, 100, 0), ), ( RelativeQuadraticBezierCurve(ctrl=Point(50, 50), end=Point(100, 100)), Point(0, 0), BoundingBox(0, 0, 100, 100), ), ( RelativeQuadraticBezierCurve(ctrl=Point(25, -100), end=Point(50, 0)), Point(0, 0), BoundingBox(0, -50, 50, 0), ), ( RelativeQuadraticBezierCurve(ctrl=Point(0, 0), end=Point(0, 0)), Point(42, 42), BoundingBox(42, 42, 42, 42), ), ( RelativeQuadraticBezierCurve(ctrl=Point(-25, 25), end=Point(-50, -50)), Point(100, 100), BoundingBox(50, 50, 100, 106.25), ), ( Arc( radii=Point(50, 50), rotation=0, large=False, sweep=True, end=Point(50, 50), ), Point(0, 0), BoundingBox(0, 0, 50, 50), ), ( Arc( radii=Point(50, 50), rotation=0, large=True, sweep=False, end=Point(-50, 0), ), Point(50, 0), BoundingBox(-50, -50, 50, 0), ), ( Arc( radii=Point(50, 30), rotation=45, large=False, sweep=True, end=Point(100, 0), ), Point(0, 0), BoundingBox(-5.405509724430832, -68.61529643011123, 99.99999989849465, 0), ), ( Arc( radii=Point(0.1, 0.1), rotation=0, large=False, sweep=True, end=Point(0.1, 0), ), Point(0, 0), BoundingBox(0, -0.013397459621556158, 0.1, 0), ), ( Arc( radii=Point(25, 40), rotation=0, large=False, sweep=False, end=Point(-40, 25), ), Point(0, 0), BoundingBox( -39.99999998902608, -8.416886842006127, 0.0, 24.999999894774596 ), ), ( RelativeArc( radii=Point(50, 50), rotation=0, large=False, sweep=True, end=Point(50, 50), ), Point(0, 0), BoundingBox(0, 0, 50, 50), ), ( RelativeArc( radii=Point(50, 50), rotation=0, large=True, sweep=False, end=Point(-100, 0), ), Point(50, 0), BoundingBox(-50.0, -50.0, 50.0, 0.0), ), ( RelativeArc( radii=Point(50, 30), rotation=45, large=False, sweep=True, end=Point(100, 0), ), Point(0, 0), BoundingBox(-5.405509724430832, -68.61529643011123, 99.99999989849465, 0), ), ( RelativeArc( radii=Point(0.1, 0.1), rotation=0, large=False, sweep=True, end=Point(0.1, 0), ), Point(0, 0), BoundingBox(0, -0.013397459621556158, 0.1, 0), ), ( RelativeArc( radii=Point(25, 40), rotation=0, large=False, sweep=False, end=Point(-40, 25), ), Point(0, 0), BoundingBox(-39.99999998902608, -8.416886842006127, 0, 24.999999894774596), ), ( Rectangle(org=Point(10, 20), size=Point(80, 60)), None, BoundingBox(10, 20, 90, 80), ), ( Rectangle(org=Point(0, 0), size=Point(100, 100)), None, BoundingBox(0, 0, 100, 100), ), ( Rectangle(org=Point(-50, -25), size=Point(30, 45)), None, BoundingBox(-50, -25, -20, 20), ), ( Rectangle(org=Point(42, 42), size=Point(0, 0)), None, BoundingBox(42, 42, 42, 42), ), ( Rectangle(org=Point(5, 10), size=Point(100, 0.1)), None, BoundingBox(5, 10, 105, 10.1), ), ( Rectangle(org=Point(5, 10), size=Point(0.1, 100)), None, BoundingBox(5, 10, 5.1, 110), ), ( RoundedRectangle( org=Point(0, 0), size=Point(100, 50), corner_radii=Point(10, 10) ), None, BoundingBox(0, 0, 100, 50), ), ( RoundedRectangle( org=Point(10, 20), size=Point(30, 40), corner_radii=Point(0, 0) ), None, BoundingBox(10, 20, 40, 60), ), ( RoundedRectangle( org=Point(50, 50), size=Point(-30, -20), corner_radii=Point(5, 5) ), None, BoundingBox(20, 30, 50, 50), ), ( RoundedRectangle( org=Point(0, 0), size=Point(20, 10), corner_radii=Point(30, 30) ), None, BoundingBox(0, 0, 20, 10), ), ( RoundedRectangle( org=Point(0, 0), size=Point(0, 50), corner_radii=Point(10, 10) ), None, BoundingBox(0, 0, 0, 50), ), ( RoundedRectangle( org=Point(0, 0), size=Point(80, 0), corner_radii=Point(10, 10) ), None, BoundingBox(0, 0, 80, 0), ), ( RoundedRectangle( org=Point(30, 30), size=Point(-10, -10), corner_radii=Point(0, 0) ), None, BoundingBox(20, 20, 30, 30), ), ( Ellipse(radii=Point(10, 10), center=Point(0, 0)), None, BoundingBox(-10, -10, 10, 10), ), ( Ellipse(radii=Point(10, 10), center=Point(100, 200)), None, BoundingBox(90, 190, 110, 210), ), ( Ellipse(radii=Point(25, 10), center=Point(50, 75)), None, BoundingBox(25, 65, 75, 85), ), ( Ellipse(radii=Point(0, 20), center=Point(50, 50)), None, BoundingBox.empty(), ), ( Ellipse(radii=Point(15, 0), center=Point(0, 0)), None, BoundingBox.empty(), ), ], ) def test_shape_bounding_box(shape, start, expected_bbox): bbox, _ = shape.bounding_box(start) if not bbox.is_valid(): assert not expected_bbox.is_valid() else: assert bbox == expected_bbox for _, tf in TRANSFORMS.items(): transform = tf(expected_bbox) if callable(tf) else tf start = start if start else Point(0, 0) path = PaintedPath() path.move_to(start.x, start.y) path.add_path_element(shape) path.style.stroke_color = "#000000" path.style.paint_rule = PathPaintRule.FILL_NONZERO path.transform = transform bbox, _ = path.bounding_box(start=start) expected_scaled_bbox = expected_bbox.transformed(transform) if not bbox.is_valid(): assert not expected_scaled_bbox.is_valid() return assert bbox == expected_scaled_bbox def simple_path_absolute_relative(): path = PaintedPath() path.move_to(10, 10) path.line_to(100, 10) path.line_relative(0, 40) path.line_to(10, 50) path.close() return path def circle_and_rectangle(): path = PaintedPath() path.circle(60, 60, 30) path.rectangle(20, 20, 40, 40) return path def mixed_curves_and_lines(): path = PaintedPath() path.move_to(0, 0) path.curve_to(30, 60, 60, 90, 100, 100) path.quadratic_curve_relative(20, -100, 50, 0) path.line_to(0, 0) path.close() return path def arc_and_line(): path = PaintedPath() path.move_to(50, 50) path.arc_relative(25, 25, 0, False, True, 50, 50) path.line_to(25, 75) return path def rect_base(): p = PaintedPath() p.style.fill_color = "#88aaee" p.rectangle(10, 10, 90, 40) # bbox: (10,10)-(100,50) return p def fill_only_rect_nonzero(): p = rect_base() p.style.fill_color = "#88aaee" p.style.stroke_color = None p.style.paint_rule = PathPaintRule.FILL_NONZERO return p # FILL_NONZERO, no stroke expansion def stroke_only_rect_w4(): p = rect_base() p.style.fill_color = None p.style.stroke_color = "#000000" p.style.stroke_width = 4 # expand by 2 in x/y return p # STROKE def stroke_fill_rect_w2(): p = rect_base() p.style.fill_color = "#cccccc" p.style.stroke_color = "#000000" p.style.stroke_width = 2 # expand by 1 in x/y p.style.paint_rule = PathPaintRule.STROKE_FILL_NONZERO return p # STROKE_FILL_NONZERO def evenodd_fill_with_hole(): p = PaintedPath() # outer rect (20,20)-(90,90) p.rectangle(20, 20, 70, 70) # inner rect (40,40)-(70,70) => hole p.rectangle(40, 40, 30, 30) p.style.fill_color = "#88aaee" p.style.stroke_color = None p.style.paint_rule = PathPaintRule.FILL_EVENODD return p # FILL_EVENODD, bbox is outer rect only def scaled_stroke_rect_w2(): p = rect_base() p.transform = Transform.scaling(2.0, 0.5) p.style.fill_color = None p.style.stroke_color = "#000000" p.style.stroke_width = 2 p.style.paint_rule = PathPaintRule.STROKE return p # Geom scales to (20,5)-(200,25); row_norms=(2,0.5); pad=(2,0.5) def translated_fill_only(): p = rect_base() p.transform = Transform.translation(15, -5) p.style.fill_color = "#88aaee" p.style.stroke_color = None p.style.paint_rule = PathPaintRule.FILL_NONZERO return p # FILL_NONZERO, just shifted def nested_scaled_circle_stroke2(): p = PaintedPath() p.transform = Transform.scaling(1.5, 1.5) p.circle(20, 20, 10) # base bbox (10,10)-(30,30) -> scaled to (15,15)-(45,45) p.style.fill_color = None p.style.stroke_color = "#000000" p.style.stroke_width = 2 # row_norms ~ (1.5,1.5); pad=(1.5,1.5) return p def hairline_stroke_rect(): p = rect_base() p.style.fill_color = None p.style.stroke_color = "#000000" p.style.stroke_width = 0 # hairline: no bbox expansion in user space return p def zero_opacity_stroke_rect(): p = rect_base() p.style.fill_color = None p.style.stroke_color = "#000000" p.style.stroke_width = 6 p.style.stroke_opacity = 0.0 # invisible stroke => no expansion return p def dont_paint_rect(): p = rect_base() p.style.paint_rule = PathPaintRule.DONT_PAINT # bbox should remain purely geometric return p @pytest.mark.parametrize( "test_id, path_builder, expected_bbox", [ ( "simple_path_absolute_relative", simple_path_absolute_relative, BoundingBox(9.5, 9.5, 100.5, 50.5), ), ( "circle_and_rectangle", circle_and_rectangle, BoundingBox(19.5, 19.5, 90.5, 90.5), ), ( "mixed_curves_and_lines", mixed_curves_and_lines, BoundingBox(-0.5, -0.5, 150.5, 100.5), ), ( "arc_and_line", arc_and_line, BoundingBox( 24.5, 39.14466094591647, 110.85533905932736, 100.50000004195078 ), ), # paint-rule variations (no transforms) ( "fill_only_nonzero", fill_only_rect_nonzero, BoundingBox(10.0, 10.0, 100.0, 50.0), ), ("stroke_only_w4", stroke_only_rect_w4, BoundingBox(8.0, 8.0, 102.0, 52.0)), ("stroke_fill_w2", stroke_fill_rect_w2, BoundingBox(9.0, 9.0, 101.0, 51.0)), ( "evenodd_fill_with_hole", evenodd_fill_with_hole, BoundingBox(20.0, 20.0, 90.0, 90.0), ), # transforms ( "scaled_stroke_rect_w2", scaled_stroke_rect_w2, BoundingBox(18.0, 4.5, 202.0, 25.5), ), ( "translated_fill_only", translated_fill_only, BoundingBox(25.0, 5.0, 115.0, 45.0), ), ( "nested_scaled_circle_stroke2", nested_scaled_circle_stroke2, BoundingBox(13.5, 13.5, 46.5, 46.5), ), # special stroke cases ( "hairline_stroke_rect", hairline_stroke_rect, BoundingBox(10.0, 10.0, 100.0, 50.0), ), ( "zero_opacity_stroke_rect", zero_opacity_stroke_rect, BoundingBox(10.0, 10.0, 100.0, 50.0), ), # explicit DONT_PAINT (no stroke expansion) ("dont_paint_rect", dont_paint_rect, BoundingBox(10.0, 10.0, 100.0, 50.0)), ], ) def test_composite_painted_path(test_id, path_builder, expected_bbox, tmp_path): path = path_builder() bbox, _ = path.bounding_box(Point(0, 0)) print(bbox, expected_bbox) assert bbox == expected_bbox pdf = FPDF() pdf.add_page() with pdf.drawing_context() as gc: gc.add_item(path) with pdf.drawing_context() as gc: # Draw bounding box in red bbox_path = PaintedPath() bbox_path.rectangle(bbox.x0, bbox.y0, bbox.x1 - bbox.x0, bbox.y1 - bbox.y0) bbox_path.style.stroke_color = "#ff0000" bbox_path.style.stroke_width = 0.1 bbox_path.style.fill_color = None gc.add_item(bbox_path) assert_pdf_equal( pdf, HERE / f"generated_pdf/bounding_box_painted_path_{test_id}.pdf", tmp_path, )