PowerDistribution

Bases: MeshDistribution

Power path distribution class expression: fn(u) = u**power expression: fn(u) = u

For more information, see MeshDistribution.

Source code in pymesh/mesh/mesh_distributions.py

class PowerDistribution(MeshDistribution):
    """Power path distribution class
    expression: fn(u) = u**power
    expression: fn(u) = u

    For more information, see [MeshDistribution][pymesh.mesh.mesh_distributions.MeshDistribution].
    """

    power = AsNumber(return_type=float)

    def __init__(self, power: int | float = 1.0, flip_direction: bool = False):
        super().__init__(flip_direction=flip_direction)
        self.power = power

    def copy(self) -> Self:
        return PowerDistribution(self.power, self.flip_direction)

    def get_dist_fn(self):
        flip = True if self.flip_direction else False  # breaks ref to self

        def fn(u: int | float, flip_direction: bool = flip) -> float:
            u = self.validate_fn_input(u=u, flip_direction=flip_direction)
            exp = u**self.power
            return self.flip_exp(exp, flip_direction)

        return fn