chore: update type hint of SVGMobject

This commit is contained in:
TonyCrane
2022-02-16 20:37:07 +08:00
parent 37b548395c
commit 05bee011d2

View File

@ -169,9 +169,9 @@ class SVGMobject(VMobject):
mob.shift(vec) mob.shift(vec)
return mob return mob
def get_mobject_from(self, shape: se.Shape | se.Text) -> VMobject | None: def get_mobject_from(self, shape: se.GraphicObject) -> VMobject | None:
shape_class_to_func_map: dict[ shape_class_to_func_map: dict[
type, Callable[[se.Shape | se.Text], VMobject] type, Callable[[se.GraphicObject], VMobject]
] = { ] = {
se.Path: self.path_to_mobject, se.Path: self.path_to_mobject,
se.SimpleLine: self.line_to_mobject, se.SimpleLine: self.line_to_mobject,
@ -196,7 +196,7 @@ class SVGMobject(VMobject):
@staticmethod @staticmethod
def apply_style_to_mobject( def apply_style_to_mobject(
mob: VMobject, mob: VMobject,
shape: se.Shape | se.Text shape: se.GraphicObject
) -> VMobject: ) -> VMobject:
mob.set_style( mob.set_style(
stroke_width=shape.stroke_width, stroke_width=shape.stroke_width,
@ -216,7 +216,7 @@ class SVGMobject(VMobject):
end=_convert_point_to_3d(line.x2, line.y2) end=_convert_point_to_3d(line.x2, line.y2)
) )
def rect_to_mobject(self, rect: se.Rect) -> Rectangle | RoundedRectangle: def rect_to_mobject(self, rect: se.Rect) -> Rectangle:
if rect.rx == 0 or rect.ry == 0: if rect.rx == 0 or rect.ry == 0:
mob = Rectangle( mob = Rectangle(
width=rect.width, width=rect.width,