fixed active bones

Roberto De Ioris · Roberto De Ioris · commit 449356fd887a · 2017-09-27T16:40:35.000+02:00
diff --git a/tutorials/FixingMixamoRootMotionWithPython_Assets/mixamo.py b/tutorials/FixingMixamoRootMotionWithPython_Assets/mixamo.py
@@ -68,7 +68,7 @@ def get_updated_bone_index(self, old_skeleton, new_skeleton, old_bone_map, new_b
         return len(new_bone_map)-1
 
     def fix_bones_influences(self, mesh, old_skeleton):
-        required_bones = []
+        active_bones = []
         for section in range(0, mesh.skeletal_mesh_sections_num()):
             vertices = mesh.skeletal_mesh_get_soft_vertices(0, section)
             ue.log_warning(len(vertices))
@@ -81,8 +81,8 @@ def fix_bones_influences(self, mesh, old_skeleton):
                 for index, bone_id in enumerate(bone_ids):
                     if vertex.influence_weights[index] > 0:
                         bone_ids[index] = self.get_updated_bone_index(old_skeleton, mesh.Skeleton, old_bone_map, new_bone_map, bone_id)
-                        if new_bone_map[bone_ids[index]] not in required_bones:
-                            required_bones.append(new_bone_map[bone_ids[index]])
+                        if new_bone_map[bone_ids[index]] not in active_bones:
+                            active_bones.append(new_bone_map[bone_ids[index]])
                 vertex.influence_bones = bone_ids
                 new_vertices.append(vertex)
 
@@ -92,10 +92,9 @@ def fix_bones_influences(self, mesh, old_skeleton):
             mesh.skeletal_mesh_set_bone_map(new_bone_map, 0, section)
 
         # specify which bones are active and required (ensure root is added to required bones)
-        mesh.skeletal_mesh_set_active_bone_indices(required_bones)
-        if 0 not in required_bones:
-            required_bones += [0]
-        mesh.skeletal_mesh_set_required_bones(required_bones)
+        mesh.skeletal_mesh_set_active_bone_indices(active_bones)
+        # mark all the bones as required (eventually you can be more selective)
+        mesh.skeletal_mesh_set_required_bones(list(range(0, mesh.Skeleton.skeleton_bones_get_num())))
 
     def set_skeleton(self, asset_data):
         self.choosen_skeleton = asset_data.get_asset()
