Example Composable Panel Down

"""Auto-generated example: Composable Panel: ComposeType.down."""

import math
import numpy as np
from PIL import Image, ImageDraw
import bencher as bn


def _polygon_points(radius, sides, start_angle=0.0):
    points = []
    for ang in np.linspace(0, 360, sides + 1):
        angle = math.radians(start_angle + ang)
        points.append([math.sin(angle) * radius, math.cos(angle) * radius])
    return points


def _draw_polygon_image(points, color, linewidth, size=200):
    img = Image.new("RGBA", (size, size), (0, 0, 0, 0))
    draw = ImageDraw.Draw(img)
    scaled = [((p[0] + 1) * size / 2, (1 - p[1]) * size / 2) for p in points]
    draw.line(scaled, fill=color, width=int(linewidth))
    return img

class BenchableImageResult(bn.ParametrizedSweep):
    """Lightweight polygon renderer for composable container demos."""

    sides = bn.IntSweep(default=3, bounds=(3, 7), doc="Number of polygon sides")
    radius = bn.FloatSweep(default=0.6, bounds=(0.2, 1.0), doc="Polygon radius")
    color = bn.StringSweep(["red", "green", "blue"], doc="Line color")

    polygon = bn.ResultImage(doc="Rendered polygon image")
    area = bn.ResultFloat("u^2", doc="Polygon area")

    def benchmark(self):
        points = _polygon_points(self.radius, self.sides)
        img = _draw_polygon_image(points, self.color, linewidth=3)
        filepath = bn.gen_image_path("polygon")
        img.save(filepath, "PNG")
        self.polygon = str(filepath)
        self.area = (
            self.sides * (2 * self.radius * math.sin(math.pi / self.sides)) ** 2
        ) / (4 * math.tan(math.pi / self.sides))


class _PanelComposeDemo(BenchableImageResult):
    """Compose polygon images into a Panel layout."""

    result_image = bn.ResultImage(doc='Composed panel image')

    def benchmark(self):
        points = _polygon_points(self.radius, self.sides)
        img = _draw_polygon_image(points, self.color, linewidth=3)
        filepath = bn.gen_image_path("panel_compose")
        img.save(filepath, "PNG")
        self.result_image = str(filepath)
        return self.get_results_values_as_dict()


def example_composable_panel_down(run_cfg: bn.BenchRunCfg | None = None) -> bn.Bench:
    """Composable Panel: ComposeType.down."""
    bench = _PanelComposeDemo().to_bench(run_cfg)
    bench.plot_sweep(
        title="Panel Layout: ComposeType.down",
        input_vars=["sides", "color"],
        result_vars=["result_image"],
    )

    return bench


if __name__ == "__main__":
    bn.run(example_composable_panel_down, subsampling_divisions=2)