Simplify VMobject.set_rgba_array

This commit is contained in:
Grant Sanderson 2023-02-02 18:23:41 -08:00
parent 4629e08769
commit 3d0fe27c55

View file

@ -166,19 +166,11 @@ class VMobject(Mobject):
def set_rgba_array(
self,
rgba_array: Vect4Array,
name: str | None = None,
name: str = "stroke_rgba",
recurse: bool = False
) -> Self:
if name is None:
names = ["fill_rgba", "stroke_rgba"]
else:
names = [name]
for name in names:
super().set_rgba_array(rgba_array, name, recurse)
if name == "fill_rgba":
self.note_changed_fill()
elif name == "stroke_rgba":
self.note_changed_stroke()
return self