scripts-blender/addons/asset_pipeline_2/core.py

@@ -2,12 +2,27 @@ import bpy
 
 from . import asset_suffix, transferable_data
 from pathlib import Path
-
+from typing import Dict
 from .asset_mapping import AssetTransferMapping
 
 from . import constants
 
 
+def get_ownership(local_col: str, task_layer_name: str):
+    new_transfer_data: Dict[bpy.props.CollectionProperty, str, str, str, str] = {}
+    for obj in local_col.all_objects:
+        transferable_data.get_vertex_groups(obj, task_layer_name, new_transfer_data)
+        transferable_data.get_material_slots(obj, task_layer_name, new_transfer_data)
+        transferable_data.get_modifiers(obj, task_layer_name, new_transfer_data)
+    return new_transfer_data
+
+
+def set_ownership(new_transfer_data):
+    for data in new_transfer_data:
+        item = new_transfer_data[data]
+        transferable_data.transfer_data_add_entry(item[0], item[1], item[2], item[3])
+
+
 def remap_user(source_datablock: bpy.data, target_datablock: bpy.data):
     """Remap datablock and append name to datablock that has been remapped"""
     print(f"REMAPPING {source_datablock.name} to {target_datablock.name}")

scripts-blender/addons/asset_pipeline_2/ops.py

@@ -21,14 +21,19 @@ class ASSETPIPE_OT_update_ownership(bpy.types.Operator):
     bl_label = 'Update Ownership'
 
     def execute(self, context):
-        obj = context.active_object
+        # TODO LOOP OVER ALl COllECTION OBJECTS
+        # TODO CREATE UI TO CONFIRM NEW OWNERSHIP ITEMS
+        local_col = bpy.data.collections.get(get_parent_col_name())
+        if not local_col:
+            self.report({'ERROR'}, "Top level collection could not be found")
+            return {'CANCELLED'}
         task_layer_name = get_task_layer_name_from_file()
         if not task_layer_name:
             self.report({'ERROR'}, "Current File Name doesn't contain valid task layer")
             return {'CANCELLED'}
-        transferable_data.vertex_groups_update(obj, task_layer_name)
-        transferable_data.modifiers_update(obj, task_layer_name)
-        transferable_data.material_slot_update(obj, task_layer_name)
+
+        new_data = core.get_ownership(local_col, task_layer_name)
+        core.set_ownership(new_data)
         return {'FINISHED'}
 
 

scripts-blender/addons/asset_pipeline_2/transferable_data.py

@@ -1,9 +1,12 @@
 import bpy
 
 from . import transfer_functions, asset_suffix
+from typing import Dict
 
 
 # TODO Consider using transfer mapping here instead of matching suffixes
+
+
 def update_transfer_data_ownership(transfer_data_item, target_obj: bpy.types.Object):
     ownership = target_obj.transfer_data_ownership
     transfer_items_names = [item.name for item in ownership]
@@ -57,7 +60,7 @@ def transfer_data_add_entry(ownership, name, td_type, task_layer_name):
 
 ## FUNCTIONS SPECFIC TO TRANSFER DATA TYPES
 # VERTEX GROUPS
-def vertex_groups_update(obj, task_layer_name):
+def get_vertex_groups(obj, task_layer_name, new_transfer_data):
     ownership = obj.transfer_data_ownership
     for vertex_group in obj.vertex_groups:
         # Only add new ownership item if vertex group doesn't have an owner
@@ -65,26 +68,41 @@ def vertex_groups_update(obj, task_layer_name):
             ownership, vertex_group.name, "VERTEX_GROUP"
         )
         if len(matches) == 0:
-            transfer_data_add_entry(
-                ownership, vertex_group.name, "VERTEX_GROUP", task_layer_name
+            # NEED UNIQUE NAME INCASE OF DUPLICATES
+            name = vertex_group.name + '_' + obj.name
+            new_transfer_data[name] = (
+                ownership,
+                vertex_group.name,
+                "VERTEX_GROUP",
+                task_layer_name,
             )
 
 
 # MODIFIERS
-def modifiers_update(obj, task_layer_name):
+def get_modifiers(obj, task_layer_name, new_transfer_data):
     ownership = obj.transfer_data_ownership
     for mod in obj.modifiers:
         matches = check_transfer_data_entry(ownership, mod.name, "MODIFIER")
         if len(matches) == 0:
-            transfer_data_add_entry(ownership, mod.name, "MODIFIER", task_layer_name)
+            name = mod.name + '_' + obj.name
+            new_transfer_data[name] = (
+                ownership,
+                mod.name,
+                "MODIFIER",
+                task_layer_name,
+            )
 
 
 # MATERIAL SLOT
-def material_slot_update(obj, task_layer_name):
+def get_material_slots(obj, task_layer_name, new_transfer_data):
     ownership = obj.transfer_data_ownership
     for slot in obj.material_slots:
         matches = check_transfer_data_entry(ownership, slot.name, "MATERIAL_SLOT")
         if len(matches) == 0:
-            transfer_data_add_entry(
-                ownership, slot.name, "MATERIAL_SLOT", task_layer_name
+            name = slot.name + '_' + obj.name
+            new_transfer_data[name] = (
+                ownership,
+                slot.name,
+                "MATERIAL_SLOT",
+                task_layer_name,
             )