armory/blender/arm/proxy.py

from typing import Any, Dict

import bpy

def proxy_sync_loc(self, context):
    if context.object == None or context.object.proxy == None:
        return
    if context.object.arm_proxy_sync_loc:
        sync_location(context.object)

def proxy_sync_rot(self, context):
    if context.object == None or context.object.proxy == None:
        return
    if context.object.arm_proxy_sync_rot:
        sync_rotation(context.object)

def proxy_sync_scale(self, context):
    if context.object == None or context.object.proxy == None:
        return
    if context.object.arm_proxy_sync_scale:
        sync_scale(context.object)

def proxy_sync_materials(self, context):
    if context.object == None or context.object.proxy == None:
        return
    if context.object.arm_proxy_sync_materials:
        sync_materials(context.object)

def proxy_sync_modifiers(self, context):
    if context.object == None or context.object.proxy == None:
        return
    if context.object.arm_proxy_sync_modifiers:
        sync_modifiers(context.object)

def proxy_sync_traits(self, context):
    if context.object == None or context.object.proxy == None:
        return
    if context.object.arm_proxy_sync_traits:
        sync_traits(context.object)

def make(obj):
    traverse(obj, is_parent=True)

def traverse(obj, is_parent=False):
    if obj == None or obj.library == None or obj.proxy != None:
        return

    # Make proxy for all linked children
    for c in obj.children:
        traverse(c)

    override = bpy.context.copy()
    override['object'] = obj
    bpy.context.view_layer.objects.active = obj
    bpy.ops.object.proxy_make(override)

    # Reparent created proxies
    for c in obj.children:
        if c.proxy != None:
            c.parent = bpy.context.view_layer.objects.active
            c.matrix_parent_inverse = bpy.context.view_layer.objects.active.matrix_world.inverted()

    active = bpy.context.view_layer.objects.active
    sync_modifiers(active)
    # No transform sync for parent
    if is_parent:
        active.arm_proxy_sync_loc = False
        active.arm_proxy_sync_rot = False
        active.arm_proxy_sync_scale = False

def sync_location(obj):
    obj.location = obj.proxy.location

def sync_rotation(obj):
    obj.rotation_euler = obj.proxy.rotation_euler

def sync_scale(obj):
    obj.scale = obj.proxy.scale

# https://blender.stackexchange.com/questions/4878
def sync_modifiers(obj):
    proxy = obj.proxy
    obj.modifiers.clear()
    for mSrc in obj.proxy.modifiers:
        mDst = obj.modifiers.get(mSrc.name, None)
        if not mDst:
            mDst = obj.modifiers.new(mSrc.name, mSrc.type)

        # collect names of writable properties
        properties = [p.identifier for p in mSrc.bl_rna.properties
                      if not p.is_readonly]

        # copy those properties
        for prop in properties:
            setattr(mDst, prop, getattr(mSrc, prop))

def sync_collection(col_src, col_dst, clear_dst=True):
    """Synchronizes collection properties. If `clear_dst` is False, the
    destination collection is not cleared before syncing."""
    if clear_dst:
        col_dst.clear()

    for m_src in col_src:
        m_dst = col_dst.add()

        # collect names of writable properties
        properties = [p.identifier for p in m_src.bl_rna.properties
                      if not p.is_readonly]

        # copy those properties
        for prop in properties:
            setattr(m_dst, prop, getattr(m_src, prop))

def sync_traits(obj: bpy.types.Object):
    """Synchronizes the traits of the given object with the traits of
    its proxy.
    If `arm.proxy_sync_trait_props` is `False`, the values of the trait
    properties are kept where possible.
    """
    # (Optionally) keep the old property values
    values: Dict[bpy.types.Object, Dict[str, Dict[str, Any]]] = {}
    for i in range(len(obj.arm_traitlist)):
        if not obj.arm_proxy_sync_trait_props:
            for prop in obj.arm_traitlist[i].arm_traitpropslist:
                values[obj][obj.arm_traitlist[i].name][prop.name] = prop.get_value()

    sync_collection(obj.proxy.arm_traitlist, obj.arm_traitlist)

    for i in range(len(obj.arm_traitlist)):
        sync_collection(obj.proxy.arm_traitlist[i].arm_traitpropslist, obj.arm_traitlist[i].arm_traitpropslist)

        # Set stored property values
        if not obj.arm_proxy_sync_trait_props:
            if values.get(obj) is None:
                continue

            value = values[obj].get(obj.arm_traitlist[i].name)
            if value is None:
                continue

            for prop in obj.arm_traitlist[i].arm_traitpropslist:

                value = values[obj].get(prop.name)
                if value is not None:
                    prop.set_value(value)


def sync_materials(obj):
    # Blender likes to crash here:(
    pass
    # proxy_mats = []
    # for slot in obj.proxy.material_slots:
    #     proxy_mats.append(slot.name)
    # override = bpy.context.copy()
    # override['object'] = obj
    # obj.active_material_index = 0
    # for i in range(len(obj.material_slots)):
    #     bpy.ops.object.material_slot_remove(override)
    # for slot in proxy_mats:
    #     bpy.ops.object.material_slot_add(override)
    #     obj.material_slots[-1].material = bpy.data.materials[slot]