CylinderZone.py

"""CylinderZone definition file."""

from parapy.core import Input, Part
from parapy.geom import Cylinder, Position

from model.Zone import Zone


class CylinderZone(Zone):
    """Represents a zone in the shape of a cylinder.

    Builds upon the Zone superclass. Defines the shape of the zone. All other functionalities are in the Zone class.
    """

    radius: float = Input(100)
    height: float = Input(100)

    @radius.on_slot_change
    def radius_on_change(self) -> None:
        """Update the nodes inside the zone when the radius slot is changed."""
        self.update_nodes_inside_zone()

    @height.on_slot_change
    def height_on_change(self) -> None:
        """Update the nodes inside the zone when the height slot is changed."""
        self.update_nodes_inside_zone()

    @Part
    def shape(self) -> Cylinder:
        """Define the shape of the Zone."""
        return Cylinder(
            radius=self.radius,
            height=self.height,
            position=Position(
                location=self.position.location, orientation=self.orientation
            ),
            centered=True,
            color=self.color,
            transparency=self.transparency,
        )