Files
kittycad.py/kittycad/models/path_segment.py
Jess Frazelle 6b8807feea improvements
Signed-off-by: Jess Frazelle <github@jessfraz.com>
2023-11-28 14:16:05 -08:00

511 lines
15 KiB
Python

from typing import Any, Dict, List, Type, TypeVar, Union
import attr
from typing_extensions import Self
from ..models.angle import Angle
from ..models.point2d import Point2d
from ..models.point3d import Point3d
from ..types import UNSET, Unset
DQ = TypeVar("DQ", bound="line")
@attr.s(auto_attribs=True)
class line:
"""A straight line segment. Goes from the current path "pen" to the given endpoint.""" # noqa: E501
end: Union[Unset, Point3d] = UNSET
relative: Union[Unset, bool] = False
type: str = "line"
additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
def to_dict(self) -> Dict[str, Any]:
if not isinstance(self.end, Unset):
end = self.end
relative = self.relative
type = self.type
field_dict: Dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update({})
if end is not UNSET:
field_dict["end"] = end.to_dict()
if relative is not UNSET:
field_dict["relative"] = relative
field_dict["type"] = type
return field_dict
@classmethod
def from_dict(cls: Type[DQ], src_dict: Dict[str, Any]) -> DQ:
d = src_dict.copy()
_end = d.pop("end", UNSET)
end: Union[Unset, Point3d]
if isinstance(_end, Unset):
end = UNSET
else:
end = _end # type: ignore[arg-type]
relative = d.pop("relative", UNSET)
type = d.pop("type", UNSET)
line = cls(
end=end,
relative=relative,
type=type,
)
line.additional_properties = d
return line
@property
def additional_keys(self) -> List[str]:
return list(self.additional_properties.keys())
def __getitem__(self, key: str) -> Any:
return self.additional_properties[key]
def __setitem__(self, key: str, value: Any) -> None:
self.additional_properties[key] = value
def __delitem__(self, key: str) -> None:
del self.additional_properties[key]
def __contains__(self, key: str) -> bool:
return key in self.additional_properties
UY = TypeVar("UY", bound="arc")
@attr.s(auto_attribs=True)
class arc:
"""A circular arc segment.""" # noqa: E501
angle_end: Union[Unset, float] = UNSET
angle_start: Union[Unset, float] = UNSET
center: Union[Unset, Point2d] = UNSET
end: Union[Unset, Angle] = UNSET
radius: Union[Unset, float] = UNSET
relative: Union[Unset, bool] = False
start: Union[Unset, Angle] = UNSET
type: str = "arc"
additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
def to_dict(self) -> Dict[str, Any]:
angle_end = self.angle_end
angle_start = self.angle_start
if not isinstance(self.center, Unset):
center = self.center
if not isinstance(self.end, Unset):
end = self.end
radius = self.radius
relative = self.relative
if not isinstance(self.start, Unset):
start = self.start
type = self.type
field_dict: Dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update({})
if angle_end is not UNSET:
field_dict["angle_end"] = angle_end
if angle_start is not UNSET:
field_dict["angle_start"] = angle_start
if center is not UNSET:
field_dict["center"] = center.to_dict()
if end is not UNSET:
field_dict["end"] = end.to_dict()
if radius is not UNSET:
field_dict["radius"] = radius
if relative is not UNSET:
field_dict["relative"] = relative
if start is not UNSET:
field_dict["start"] = start.to_dict()
field_dict["type"] = type
return field_dict
@classmethod
def from_dict(cls: Type[UY], src_dict: Dict[str, Any]) -> UY:
d = src_dict.copy()
angle_end = d.pop("angle_end", UNSET)
angle_start = d.pop("angle_start", UNSET)
_center = d.pop("center", UNSET)
center: Union[Unset, Point2d]
if isinstance(_center, Unset):
center = UNSET
else:
center = _center # type: ignore[arg-type]
_end = d.pop("end", UNSET)
end: Union[Unset, Angle]
if isinstance(_end, Unset):
end = UNSET
else:
end = _end # type: ignore[arg-type]
radius = d.pop("radius", UNSET)
relative = d.pop("relative", UNSET)
_start = d.pop("start", UNSET)
start: Union[Unset, Angle]
if isinstance(_start, Unset):
start = UNSET
else:
start = _start # type: ignore[arg-type]
type = d.pop("type", UNSET)
arc = cls(
angle_end=angle_end,
angle_start=angle_start,
center=center,
end=end,
radius=radius,
relative=relative,
start=start,
type=type,
)
arc.additional_properties = d
return arc
@property
def additional_keys(self) -> List[str]:
return list(self.additional_properties.keys())
def __getitem__(self, key: str) -> Any:
return self.additional_properties[key]
def __setitem__(self, key: str, value: Any) -> None:
self.additional_properties[key] = value
def __delitem__(self, key: str) -> None:
del self.additional_properties[key]
def __contains__(self, key: str) -> bool:
return key in self.additional_properties
PD = TypeVar("PD", bound="bezier")
@attr.s(auto_attribs=True)
class bezier:
"""A cubic bezier curve segment. Start at the end of the current line, go through control point 1 and 2, then end at a given point.""" # noqa: E501
control1: Union[Unset, Point3d] = UNSET
control2: Union[Unset, Point3d] = UNSET
end: Union[Unset, Point3d] = UNSET
relative: Union[Unset, bool] = False
type: str = "bezier"
additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
def to_dict(self) -> Dict[str, Any]:
if not isinstance(self.control1, Unset):
control1 = self.control1
if not isinstance(self.control2, Unset):
control2 = self.control2
if not isinstance(self.end, Unset):
end = self.end
relative = self.relative
type = self.type
field_dict: Dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update({})
if control1 is not UNSET:
field_dict["control1"] = control1.to_dict()
if control2 is not UNSET:
field_dict["control2"] = control2.to_dict()
if end is not UNSET:
field_dict["end"] = end.to_dict()
if relative is not UNSET:
field_dict["relative"] = relative
field_dict["type"] = type
return field_dict
@classmethod
def from_dict(cls: Type[PD], src_dict: Dict[str, Any]) -> PD:
d = src_dict.copy()
_control1 = d.pop("control1", UNSET)
control1: Union[Unset, Point3d]
if isinstance(_control1, Unset):
control1 = UNSET
else:
control1 = _control1 # type: ignore[arg-type]
_control2 = d.pop("control2", UNSET)
control2: Union[Unset, Point3d]
if isinstance(_control2, Unset):
control2 = UNSET
else:
control2 = _control2 # type: ignore[arg-type]
_end = d.pop("end", UNSET)
end: Union[Unset, Point3d]
if isinstance(_end, Unset):
end = UNSET
else:
end = _end # type: ignore[arg-type]
relative = d.pop("relative", UNSET)
type = d.pop("type", UNSET)
bezier = cls(
control1=control1,
control2=control2,
end=end,
relative=relative,
type=type,
)
bezier.additional_properties = d
return bezier
@property
def additional_keys(self) -> List[str]:
return list(self.additional_properties.keys())
def __getitem__(self, key: str) -> Any:
return self.additional_properties[key]
def __setitem__(self, key: str, value: Any) -> None:
self.additional_properties[key] = value
def __delitem__(self, key: str) -> None:
del self.additional_properties[key]
def __contains__(self, key: str) -> bool:
return key in self.additional_properties
SM = TypeVar("SM", bound="tangential_arc")
@attr.s(auto_attribs=True)
class tangential_arc:
"""Adds a tangent arc from current pen position with the given radius and angle.""" # noqa: E501
offset: Union[Unset, Angle] = UNSET
radius: Union[Unset, float] = UNSET
type: str = "tangential_arc"
additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
def to_dict(self) -> Dict[str, Any]:
if not isinstance(self.offset, Unset):
offset = self.offset
radius = self.radius
type = self.type
field_dict: Dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update({})
if offset is not UNSET:
field_dict["offset"] = offset.to_dict()
if radius is not UNSET:
field_dict["radius"] = radius
field_dict["type"] = type
return field_dict
@classmethod
def from_dict(cls: Type[SM], src_dict: Dict[str, Any]) -> SM:
d = src_dict.copy()
_offset = d.pop("offset", UNSET)
offset: Union[Unset, Angle]
if isinstance(_offset, Unset):
offset = UNSET
else:
offset = _offset # type: ignore[arg-type]
radius = d.pop("radius", UNSET)
type = d.pop("type", UNSET)
tangential_arc = cls(
offset=offset,
radius=radius,
type=type,
)
tangential_arc.additional_properties = d
return tangential_arc
@property
def additional_keys(self) -> List[str]:
return list(self.additional_properties.keys())
def __getitem__(self, key: str) -> Any:
return self.additional_properties[key]
def __setitem__(self, key: str, value: Any) -> None:
self.additional_properties[key] = value
def __delitem__(self, key: str) -> None:
del self.additional_properties[key]
def __contains__(self, key: str) -> bool:
return key in self.additional_properties
JL = TypeVar("JL", bound="tangential_arc_to")
@attr.s(auto_attribs=True)
class tangential_arc_to:
"""Adds a tangent arc from current pen position to the new position.""" # noqa: E501
angle_snap_increment: Union[Unset, Angle] = UNSET
to: Union[Unset, Point3d] = UNSET
type: str = "tangential_arc_to"
additional_properties: Dict[str, Any] = attr.ib(init=False, factory=dict)
def to_dict(self) -> Dict[str, Any]:
if not isinstance(self.angle_snap_increment, Unset):
angle_snap_increment = self.angle_snap_increment
if not isinstance(self.to, Unset):
to = self.to
type = self.type
field_dict: Dict[str, Any] = {}
field_dict.update(self.additional_properties)
field_dict.update({})
if angle_snap_increment is not UNSET:
field_dict["angle_snap_increment"] = angle_snap_increment.to_dict()
if to is not UNSET:
field_dict["to"] = to.to_dict()
field_dict["type"] = type
return field_dict
@classmethod
def from_dict(cls: Type[JL], src_dict: Dict[str, Any]) -> JL:
d = src_dict.copy()
_angle_snap_increment = d.pop("angle_snap_increment", UNSET)
angle_snap_increment: Union[Unset, Angle]
if isinstance(_angle_snap_increment, Unset):
angle_snap_increment = UNSET
else:
angle_snap_increment = _angle_snap_increment # type: ignore[arg-type]
_to = d.pop("to", UNSET)
to: Union[Unset, Point3d]
if isinstance(_to, Unset):
to = UNSET
else:
to = _to # type: ignore[arg-type]
type = d.pop("type", UNSET)
tangential_arc_to = cls(
angle_snap_increment=angle_snap_increment,
to=to,
type=type,
)
tangential_arc_to.additional_properties = d
return tangential_arc_to
@property
def additional_keys(self) -> List[str]:
return list(self.additional_properties.keys())
def __getitem__(self, key: str) -> Any:
return self.additional_properties[key]
def __setitem__(self, key: str, value: Any) -> None:
self.additional_properties[key] = value
def __delitem__(self, key: str) -> None:
del self.additional_properties[key]
def __contains__(self, key: str) -> bool:
return key in self.additional_properties
class PathSegment:
"""A segment of a path. Paths are composed of many segments."""
type: Union[
line,
arc,
bezier,
tangential_arc,
tangential_arc_to,
] = None
def __init__(
self,
type: Union[
type(line),
type(arc),
type(bezier),
type(tangential_arc),
type(tangential_arc_to),
],
):
self.type = type
def to_dict(self) -> Dict[str, Any]:
if isinstance(self.type, line):
n: line = self.type
return n.to_dict()
elif isinstance(self.type, arc):
n: arc = self.type
return n.to_dict()
elif isinstance(self.type, bezier):
n: bezier = self.type
return n.to_dict()
elif isinstance(self.type, tangential_arc):
n: tangential_arc = self.type
return n.to_dict()
elif isinstance(self.type, tangential_arc_to):
n: tangential_arc_to = self.type
return n.to_dict()
raise Exception("Unknown type")
def from_dict(self, d) -> Self:
if d.get("type") == "line":
n: line = line()
n.from_dict(d)
self.type = n
return Self
elif d.get("type") == "arc":
n: arc = arc()
n.from_dict(d)
self.type = n
return self
elif d.get("type") == "bezier":
n: bezier = bezier()
n.from_dict(d)
self.type = n
return self
elif d.get("type") == "tangential_arc":
n: tangential_arc = tangential_arc()
n.from_dict(d)
self.type = n
return self
elif d.get("type") == "tangential_arc_to":
n: tangential_arc_to = tangential_arc_to()
n.from_dict(d)
self.type = n
return self
raise Exception("Unknown type")