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,20 +166,12 @@ class VMobject(Mobject):
def set_rgba_array( def set_rgba_array(
self, self,
rgba_array: Vect4Array, rgba_array: Vect4Array,
name: str | None = None, name: str = "stroke_rgba",
recurse: bool = False recurse: bool = False
) -> Self: ) -> Self:
if name is None: super().set_rgba_array(rgba_array, name, recurse)
names = ["fill_rgba", "stroke_rgba"] self.note_changed_fill()
else: self.note_changed_stroke()
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 return self
def set_fill( def set_fill(