mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
37 lines
1.4 KiB
Python
37 lines
1.4 KiB
Python
import itertools as it
|
|
|
|
|
|
def extract_mobject_family_members(mobject_list, only_those_with_points=False):
|
|
result = list(it.chain(*[
|
|
mob.get_family()
|
|
for mob in mobject_list
|
|
]))
|
|
if only_those_with_points:
|
|
result = [mob for mob in result if mob.has_points()]
|
|
return result
|
|
|
|
|
|
def restructure_list_to_exclude_certain_family_members(mobject_list, to_remove):
|
|
"""
|
|
Removes anything in to_remove from mobject_list, but in the event that one of
|
|
the items to be removed is a member of the family of an item in mobject_list,
|
|
the other family members are added back into the list.
|
|
|
|
This is useful in cases where a scene contains a group, e.g. Group(m1, m2, m3),
|
|
but one of its submobjects is removed, e.g. scene.remove(m1), it's useful
|
|
for the list of mobject_list to be edited to contain other submobjects, but not m1.
|
|
"""
|
|
new_list = []
|
|
to_remove = extract_mobject_family_members(to_remove)
|
|
|
|
def add_safe_mobjects_from_list(list_to_examine, set_to_remove):
|
|
for mob in list_to_examine:
|
|
if mob in set_to_remove:
|
|
continue
|
|
intersect = set_to_remove.intersection(mob.get_family())
|
|
if intersect:
|
|
add_safe_mobjects_from_list(mob.submobjects, intersect)
|
|
else:
|
|
new_list.append(mob)
|
|
add_safe_mobjects_from_list(mobject_list, set(to_remove))
|
|
return new_list
|