3b1b-manim/manimlib/utils/family_ops.py

from __future__ import annotations

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from typing import Iterable

    from manimlib.mobject.mobject import Mobject


def extract_mobject_family_members(
    mobject_list: Iterable[Mobject],
    exclude_pointless: bool = False
) -> list[Mobject]:
    return [
        sm
        for mob in mobject_list
        for sm in mob.get_family()
        if (not exclude_pointless) or sm.has_points()
    ]
chore: add type hints to manimlib.utils 2022-02-12 23:47:23 +08:00			`from __future__ import annotations`

chore: only import some classes when type checking 2022-02-14 21:34:56 +08:00			`from typing import TYPE_CHECKING`
style: insert an empty line after import 2022-02-16 21:08:25 +08:00
chore: only import some classes when type checking 2022-02-14 21:34:56 +08:00			`if TYPE_CHECKING:`
Sort imports 2022-04-12 19:19:59 +08:00			`from typing import Iterable`

chore: only import some classes when type checking 2022-02-14 21:34:56 +08:00			`from manimlib.mobject.mobject import Mobject`
Created and moved extract_mobject_family_members and restructure_list_to_exclude_certain_family_members functions 2020-01-15 18:30:58 -08:00

chore: add type hints to manimlib.utils 2022-02-12 23:47:23 +08:00			`def extract_mobject_family_members(`
			`mobject_list: Iterable[Mobject],`
Cleanup extract_mobject_family_members 2022-04-20 21:51:56 -07:00			`exclude_pointless: bool = False`
chore: add type hints to manimlib.utils 2022-02-12 23:47:23 +08:00			`) -> list[Mobject]:`
Cleanup extract_mobject_family_members 2022-04-20 21:51:56 -07:00			`return [`
			`sm`
Forget about using remove_list_redundancies 2020-02-14 15:29:35 -08:00			`for mob in mobject_list`
Cleanup extract_mobject_family_members 2022-04-20 21:51:56 -07:00			`for sm in mob.get_family()`
			`if (not exclude_pointless) or sm.has_points()`
			`]`