handles cases when no armature in the scene

Author: nkx

io_mesh_w3d/common/utils/animation_export.py

@@ -118,6 +118,9 @@ def retrieve_channels(obj, hierarchy, timecoded, name=None):
 
 
 def retrieve_animation(context, animation_name, hierarchy, rig, timecoded):
+    if hierarchy is None:
+        return None
+    
     channels = []
 
     for mesh in get_objects('MESH'):

io_mesh_w3d/common/utils/hierarchy_export.py

@@ -24,7 +24,8 @@ def retrieve_hierarchy(context, container_name):
     if len(rigs) == 0:
         hierarchy.header.name = container_name
         hierarchy.header.center_pos = Vector()
-        context.warning('scene does not contain an armature object!')
+        context.warning('Scene does not contain an armature object! Can only export meshes')
+        return None, None
 
     if len(rigs) > 0:
         rig = rigs[0]

io_mesh_w3d/common/utils/mesh_export.py

@@ -87,7 +87,7 @@ def retrieve_meshes(context, hierarchy, rig, container_name, force_vertex_materi
             matrix = Matrix.Identity(4)
             matrix_2 = Matrix.Identity(4)
 
-            if vertex.groups:
+            if vertex.groups and hierarchy:
                 vert_inf = VertexInfluence()
                 vert_inf.bone_idx = find_bone_index(hierarchy, mesh_object, vertex.groups[0].group)
                 vert_inf.bone_inf = vertex.groups[0].weight
@@ -130,7 +130,7 @@ def retrieve_meshes(context, hierarchy, rig, container_name, force_vertex_materi
                 else:
                     matrix_2 = matrix
 
-            elif is_skinned:
+            elif is_skinned and hierarchy:
                 unskinned_vertices_error = True
                 context.error(f'skinned mesh \'{mesh_object.name}\' vertex {i} is not rigged to any bone!')
 

io_mesh_w3d/export_utils.py

@@ -40,11 +40,15 @@ def retrieve_data(context, export_settings):
         context.error(f'Filename is longer than {STRING_LENGTH} characters, aborting export!')
         return None
 
-    bpy.ops.object.mode_set(mode='OBJECT')
+    try:
+        bpy.ops.object.mode_set(mode='OBJECT')
+    except:
+        pass
     hierarchy, rig, hlod = None, None, None
 
     hierarchy, rig = retrieve_hierarchy(context, container_name)
-    hlod = create_hlod(hierarchy, container_name)
+    if hierarchy:
+        hlod = create_hlod(hierarchy, container_name)
 
     data_context = DataContext(
         container_name=container_name,
@@ -72,12 +76,14 @@ def retrieve_data(context, export_settings):
                 context.error('aborting export!')
                 return None
 
-    if 'H' in export_mode and not hierarchy.validate(context):
-        context.error('aborting export!')
-        return None
+    if 'H' in export_mode :
+        if hierarchy is not None:
+            if not hierarchy.validate(context):
+                context.error('aborting export!')
+                return None
 
     if export_mode in ['HM', 'HAM']:
-        if not data_context.hlod.validate(context):
+        if data_context.hlod and not data_context.hlod.validate(context):
             context.error('aborting export!')
             return None
 
@@ -89,7 +95,7 @@ def retrieve_data(context, export_settings):
     if 'A' in export_mode:
         timecoded = export_settings['compression'] == 'TC'
         data_context.animation = retrieve_animation(context, container_name, hierarchy, rig, timecoded)
-        if not data_context.animation.validate(context):
+        if data_context.animation and not data_context.animation.validate(context):
             context.error('aborting export!')
             return None
     return data_context

io_mesh_w3d/w3x/export_w3x.py

@@ -24,7 +24,7 @@ def save(context, export_settings, data_context):
 
     # directory = os.path.dirname(context.filepath) + os.path.sep
 
-    if 'H' in export_mode:
+    if 'H' in export_mode and data_context.hierarchy:
         if export_settings['individual_files']:
             path = filepath_pure + "_SKL" + context.filename_ext
             context.info('Saving file :' + path)
@@ -50,14 +50,15 @@ def save(context, export_settings, data_context):
             else:
                 mesh.create(root)
         # w3dcontainer
-        if export_settings['individual_files']:
-                path = filepath_pure + "_SKN" + context.filename_ext
-                context.info('Saving file :' + path)
-                write_struct(data_context.hlod, path)
-        else:
-            data_context.hlod.create(root)
+        if data_context.hlod:
+            if export_settings['individual_files']:
+                    path = filepath_pure + "_SKN" + context.filename_ext
+                    context.info('Saving file :' + path)
+                    write_struct(data_context.hlod, path)
+            else:
+                data_context.hlod.create(root)
 
-    if "A" in export_mode:
+    if "A" in export_mode and data_context.animation:
         if export_settings['individual_files']:
             path = directory + data_context.animation.name() + context.filename_ext
             context.info('Saving file :' + path)
@@ -78,5 +79,7 @@ def save(context, export_settings, data_context):
         context.info('Saving file :' + filepath)
         write(root, filepath)
 
+    if not data_context.hierarchy:
+        context.warning('Export incomplete')
 
     return {'FINISHED'}