better unreal : collisions, scale & orientation

README.md

@@ -1,30 +1,328 @@
-# blender-one-key-export
-Blender plugin (operators) to export selection (and children) as fbx.
-Those operators are designed with a very specific worlflow in mind.
+bl_info = {
+    "name": "One Key Export",
+    "version": (0, 4, 4),
+    "blender": (2, 80, 0),
+    "category": "Export",
+    "author": "Jérémy Crombez",
+}
 
+import bpy
+import os, time, re
 
-## How to use
 
-Press "F3" and type "One Key Exporter" then you can either use the "Snowdrop", "Substaince Painter" or "Unreal" export.
+# --- --- --- #
 
-### Snowdrop
 
-Every object with no parent in your selection get exported with all its children in a dedicated .fbx file named after the root object and placed next to the .blend file.
+class Snowdrop_OneKeyExport(bpy.types.Operator):
+    """One Key Export - Snowdrop""" # Use this as a tooltip for menu items and buttons.
+    bl_idname = "object.one_key_export__snowdrop" # Unique identifier for buttons and menu items to reference.
+    bl_label = "One Key Export - Snowdrop" # Display name in the interface.
+    bl_options = {'REGISTER', 'UNDO'} # Enable undo for the operator.
+
+    def execute(self, context):
+        initially_seleted_objects = context.selected_objects
+
+        selected_root_objects = []
 
-All their children, even if not selected or hidden, will be in their respective FBX file and forced visible.
+        for selected_object in context.selected_objects:
+            selected_object.select_set(False)
+            
+            if selected_object.parent is None:
+                selected_root_objects.append(selected_object)
+                
+        for root in selected_root_objects:
+            root.select_set(True)
 
-### Substance Painter
+            previously_hidden_objects = unhide_hierarchy(root)
+            select_hierarchy(root)
+            
+            # Saving initial location
+            initial_root_location_x = root.location.x
+            initial_root_location_y = root.location.y
+            initial_root_location_z = root.location.z
+            
+            # Resetting location
+            root.location.x = 0
+            root.location.y = 0
+            root.location.z = 0
+            
+            # FBX Export
+            
+            blend_file_full_path = bpy.data.filepath
+            blend_file_path = os.path.dirname(blend_file_full_path)
+            fbx_file_name = root.name + '.fbx'
+            fbx_file_full_path = os.path.join(blend_file_path, fbx_file_name)
 
-Every selected object and their children will be exported as one .fbx file in a "workfiles" directory next to the .blend file.
+            bpy.ops.export_scene.fbx(
+                filepath=fbx_file_full_path,
+                use_selection=True,
+            )
 
-Any object containing "_col", "_scol" or "_mcol" (not case sensitive) will be ignored.
+            # Ugly workaround to wait for the export
+            # TODO: wait for the file update date to change instead
+            time.sleep(0.2)
 
-Every object with more than one UV will only keep its second UV in the export.
+            # Hide back what was hidden
+            for object in previously_hidden_objects:
+                object.hide_set(True)
+                
+            # Move back to origin location
+            root.location.x = initial_root_location_x
+            root.location.y = initial_root_location_y
+            root.location.z = initial_root_location_z
+            
+            bpy.ops.object.select_all(action='DESELECT')
+        
+        # Select back what was selected        
+        for object in initially_seleted_objects:
+            object.select_set(True)
+            
+        self.report({'INFO'}, "Export done :-)")
 
-The .fbx file will be named after the first selected object with no parent, and if there is more than one, any suffix in its name starting with "_" will be stripped from the name.
+        return {'FINISHED'}
 
-For example if you selected three root objects, "PROP_42_A", "PROP_42_B" and "PROP_42_C", the exported file will be named "PROP_42".
 
-### Unreal
+# --- --- --- #
 
-// TODO: doc
+
+class SubstancePainter_OneKeyExport(bpy.types.Operator):
+    """One Key Export - Substance Painter""" # Use this as a tooltip for menu items and buttons.
+    bl_idname = "object.one_key_export__substance_painter" # Unique identifier for buttons and menu items to reference.
+    bl_label = "One Key Export - Substance Painter" # Display name in the interface.
+    bl_options = {'REGISTER', 'UNDO'} # Enable undo for the operator.
+
+
+    def execute(self, context):
+        initially_seleted_objects = context.selected_objects
+
+        selected_root_objects = []
+
+        for selected_object in context.selected_objects:
+            selected_object.select_set(False)
+
+            if selected_object.parent is None:
+                selected_root_objects.append(selected_object)
+
+        for root in selected_root_objects:
+            root.select_set(True)
+
+            previously_hidden_objects = unhide_hierarchy(root)
+            select_hierarchy(root)
+
+            # Saving initial location
+            initial_root_location_x = root.location.x
+            initial_root_location_y = root.location.y
+            initial_root_location_z = root.location.z
+
+            # Resetting location
+            root.location.x = 0
+            root.location.y = 0
+            root.location.z = 0
+
+            # FBX Export
+
+            blend_file_full_path = bpy.data.filepath
+            blend_file_path = os.path.dirname(blend_file_full_path)
+            fbx_file_name = root.name + '.fbx'
+            fbx_file_full_path = os.path.join(blend_file_path, fbx_file_name)
+
+            bpy.ops.export_scene.fbx(
+                filepath=fbx_file_full_path,
+                use_selection=True,
+            )
+
+            # Ugly workaround to wait for the export
+            # TODO: wait for the file update date to change instead
+            time.sleep(0.2)
+
+            # Hide back what was hidden
+            for object in previously_hidden_objects:
+                object.hide_set(True)
+                
+            # Move back to origin location
+            root.location.x = initial_root_location_x
+            root.location.y = initial_root_location_y
+            root.location.z = initial_root_location_z
+            
+            bpy.ops.object.select_all(action='DESELECT')
+        
+        # Select back what was selected        
+        for object in initially_seleted_objects:
+            object.select_set(True)
+            
+        self.report({'INFO'}, "Export done :-)")
+
+        return {'FINISHED'}
+
+
+# --- --- --- #
+
+
+class UnrealEngine_OneKeyExport(bpy.types.Operator):
+    """One Key Export - Unreal Engine""" # Use this as a tooltip for menu items and buttons.
+    bl_idname = "object.one_key_export__unreal_engine" # Unique identifier for buttons and menu items to reference.
+    bl_label = "One Key Export - Unreal Engine" # Display name in the interface.
+    bl_options = {'REGISTER', 'UNDO'} # Enable undo for the operator.
+
+
+    def execute(self, context):
+        initially_seleted_objects = context.selected_objects
+
+        selected_root_objects = []
+
+        for selected_object in context.selected_objects:
+            selected_object.select_set(False)
+
+            if selected_object.parent is None:
+                selected_root_objects.append(selected_object)
+
+        for root in selected_root_objects:
+            root.select_set(True)
+
+            previously_hidden_objects = unhide_hierarchy(root)
+            select_hierarchy(root)
+            
+            # Saving initial location
+            initial_root_location_x = root.location.x
+            initial_root_location_y = root.location.y
+            initial_root_location_z = root.location.z
+            
+            # Resetting location
+            root.location.x = 0
+            root.location.y = 0
+            root.location.z = 0
+            
+            # FBX Export
+            blend_file_full_path = bpy.data.filepath
+            blend_file_path = os.path.dirname(blend_file_full_path)
+            fbx_file_name = root.name + '.fbx'
+            fbx_file_full_path = os.path.join(blend_file_path, fbx_file_name)
+
+            # Parcours de tous les objets de la scène
+            # for obj in bpy.context.scene.objects:
+            #     # Vérification si le nom de l'objet commence par "UCX_"
+            #     if obj.name.startswith("UCX_"):
+            #         # Sélection de l'objet
+            #         obj.select_set(True)
+
+            bpy.ops.export_scene.fbx(
+                filepath=fbx_file_full_path,
+                use_selection=True,
+                mesh_smooth_type='FACE',
+                use_mesh_modifiers=True,
+                use_armature_deform_only=True,
+                add_leaf_bones=False,
+                bake_anim=True,
+            )
+
+            # Ugly workaround to wait for the export
+            # TODO: wait for the file update date to change instead
+            time.sleep(0.2)
+
+            # Hide back what was hidden
+            for object in previously_hidden_objects:
+                object.hide_set(True)
+                
+            # Move back to origin location
+            root.location.x = initial_root_location_x
+            root.location.y = initial_root_location_y
+            root.location.z = initial_root_location_z
+            
+            bpy.ops.object.select_all(action='DESELECT')
+        
+        # Select back what was selected        
+        for object in initially_seleted_objects:
+            object.select_set(True)
+            
+        self.report({'INFO'}, "Export done :-)")
+
+        return {'FINISHED'}
+
+
+# --- --- --- #
+
+
+def unhide_hierarchy(object, exclude_collision = False):
+    if exclude_collision and is_collision(object):
+        return []
+
+    previously_hidden_objects = []
+
+    if object.hide_get():
+        previously_hidden_objects.append(object)
+        object.hide_set(False)
+
+    for child in object.children:
+        previously_hidden_objects += unhide_hierarchy(child)
+
+    return previously_hidden_objects
+
+
+def select_hierarchy(object, ignore_collision = False):    
+    if ignore_collision and is_collision(object):
+        object.select_set(False)
+        return
+
+    object.select_set(True)
+
+    for child in object.children:
+        select_hierarchy(child, ignore_collision)
+
+
+def only_keep_uv2(object):        
+    # Ignore objects (like Empties) who have no UVs
+    if not hasattr(object.data, 'uv_layers'):
+        return
+
+    uvs = object.data.uv_layers
+
+    # Ignore objects with only 1 uv
+    if len(uvs) == 1:
+        return
+
+    # Set UV2 as the active one
+    uvs.active_index = 1
+
+    # Delete not active UVs
+
+    not_active_uvs = [uv for uv in uvs if uv != uvs.active]
+
+    while not_active_uvs:
+        uvs.remove(not_active_uvs.pop())
+
+
+def is_collision(object):
+    return re.search('_col|_scol|_mcol', object.name, re.IGNORECASE) is not None
+
+
+# --- --- --- #
+
+
+def menu_func(self, context):
+    self.layout.operator(Snowdrop_OneKeyExport.bl_idname)
+    self.layout.operator(SubstancePainter_OneKeyExport.bl_idname)
+    self.layout.operator(UnrealEngine_OneKeyExport.bl_idname)
+
+def register():
+    bpy.utils.register_class(Snowdrop_OneKeyExport)
+    bpy.utils.register_class(SubstancePainter_OneKeyExport)
+    bpy.utils.register_class(UnrealEngine_OneKeyExport)
+
+    # Avoid adding the operator in the if it's already there.
+    if hasattr(bpy.types.VIEW3D_MT_object.draw, '_draw_funcs'):
+        if menu_func.__name__ not in (f.__name__ for f in bpy.types.VIEW3D_MT_object.draw._draw_funcs):
+            bpy.types.VIEW3D_MT_object.prepend(menu_func)
+    else:
+        bpy.types.VIEW3D_MT_object.prepend(menu_func)
+
+def unregister():
+    bpy.utils.unregister_class(Snowdrop_OneKeyExport)
+    bpy.utils.unregister_class(SubstancePainter_OneKeyExport)
+    bpy.utils.unregister_class(UnrealEngine_OneKeyExport)
+    bpy.types.VIEW3D_MT_object.remove(menu_func)
+
+
+# This allows you to run the script directly from Blender's Text editor
+# to test the add-on without having to install it.
+if __name__ == "__main__":
+    register()