Source code for modsim_package.python.rectangle_compression_cartesian_product

"""Parameter sets and schemas for the rectangle compression simulation."""




def parameter_schema(
    width: list[float] | tuple[float, ...] = (1.0, 1.1),
    height: list[float] | tuple[float, ...] = (1.0, 1.1),
    global_seed: list[float] | tuple[float, ...] = (1.0,),
    displacement: list[float] | tuple[float, ...] = (-0.01,),
) -> dict[str, list[float]]:
    """Return WAVES CartesianProduct parameter schema.

    :param width: The rectangle width
    :param height: The rectangle height
    :param global_seed: The global mesh seed size
    :param displacement: The rectangle top surface displacement

    :returns: WAVES CartesianProduct parameter schema
    """
    schema = {
        "width": list(width),
        "height": list(height),
        "global_seed": list(global_seed),
        "displacement": list(displacement),
    }
    return schema