lib.py

# SPDX-FileCopyrightText: 2014-2024 Mikhail Rachinskiy
# SPDX-License-Identifier: GPL-3.0-or-later

from bpy.types import AnimData, Object


def ad_check(ad: AnimData) -> bool:
    return ad and (ad.action or ad.nla_tracks)


def ad_get(ob: Object, use_ob=True, use_data=True, use_sk=True, use_mat=True) -> list[AnimData]:
    ads = []

    if use_ob and ad_check(ob.animation_data):
        ads.append(ob.animation_data)

    if ob.data:

        if use_data and ad_check(ob.data.animation_data):
            ads.append(ob.data.animation_data)

        if use_sk:
            try:
                ad = ob.data.shape_keys.animation_data
                if ad_check(ad):
                    ads.append(ad)
            except AttributeError:
                pass

    if use_mat and ob.material_slots:
        for slot in ob.material_slots:
            if slot.material is not None:
                if ad_check(slot.material.animation_data):
                    ads.append(slot.material.animation_data)
                if slot.material.node_tree and ad_check(slot.material.node_tree.animation_data):
                    ads.append(slot.material.node_tree.animation_data)

    return ads


def effector_radius(ob: Object) -> float:
    if ob.type == "EMPTY":
        return ob.empty_display_size * ob.matrix_world.to_scale().x

    return ob.dimensions.x / 2