File size: 1,519 Bytes
269d09f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | from __future__ import annotations
from contourpy._contourpy import FillType, LineType, ZInterp
def as_fill_type(fill_type: FillType | str) -> FillType:
"""Coerce a FillType or string value to a FillType.
Args:
fill_type (FillType or str): Value to convert.
Return:
FillType: Converted value.
"""
if isinstance(fill_type, str):
try:
return FillType.__members__[fill_type]
except KeyError as e:
raise ValueError(f"'{fill_type}' is not a valid FillType") from e
else:
return fill_type
def as_line_type(line_type: LineType | str) -> LineType:
"""Coerce a LineType or string value to a LineType.
Args:
line_type (LineType or str): Value to convert.
Return:
LineType: Converted value.
"""
if isinstance(line_type, str):
try:
return LineType.__members__[line_type]
except KeyError as e:
raise ValueError(f"'{line_type}' is not a valid LineType") from e
else:
return line_type
def as_z_interp(z_interp: ZInterp | str) -> ZInterp:
"""Coerce a ZInterp or string value to a ZInterp.
Args:
z_interp (ZInterp or str): Value to convert.
Return:
ZInterp: Converted value.
"""
if isinstance(z_interp, str):
try:
return ZInterp.__members__[z_interp]
except KeyError as e:
raise ValueError(f"'{z_interp}' is not a valid ZInterp") from e
else:
return z_interp
|