manimgeo.utils.utils 源代码

from __future__ import annotations

from typing import TYPE_CHECKING, Iterable

if TYPE_CHECKING:
    from ..components import BaseGeometry



def flatten(iterable: Iterable):
    """展平对象"""
    for item in iterable:
        if isinstance(item, list):
            yield from flatten(item)
        else:
            yield item




def print_dependencies(root: "BaseGeometry", depth: int = 0, max_depth: int = 20):
    """绘制依赖关系"""
    from ..utils.output import color_text, generate_color_from_id
    
    if root is None:
        print("  "*depth + "· None")
        return
        
    if depth > max_depth:
        print("  "*depth + "· ... (max depth reached)")
        return
    
    name_str = f" - ({root.name})" if hasattr(root, 'name') and root.name else ""
    print("  "*depth + f"· {color_text(type(root).__name__, *generate_color_from_id(root))}{name_str}")
    
    if not hasattr(root, 'dependents'):
        return
        
    for dep in root.dependents:
        print_dependencies(dep, depth+1, max_depth)