optimize

MMT/__init__.py

@@ -16,7 +16,7 @@
     "original author": "DarkStarSword",
     "description": "Special fork version of DarkStarSword's blender_3dmigoto.py",
     "blender": (3, 6, 0),
-    "version": (1, 5, 3),
+    "version": (1, 5, 4),
     "location": "View3D",
     "warning": "",
     "category": "Generic"
@@ -47,6 +47,7 @@
     MMTCancelAutoSmooth,
     MMTShowIndexedVertices,
     MMTSetAutoSmooth89,
+    SplitMeshByCommonVertexGroup,
 
     # MMT的一键导入导出
     MMTImportAllTextModel,

MMT/mesh_operator.py

@@ -26,7 +26,7 @@ def remove_unused_vertex_group(self, context):
 
 class RemoveUnusedVertexGroupOperator(bpy.types.Operator):
     bl_idname = "object.remove_unused_vertex_group"
-    bl_label = "Remove Unused Vertex Group"
+    bl_label = "移除未使用的空顶点组"
 
     def execute(self, context):
         return remove_unused_vertex_group(self, context)
@@ -109,7 +109,7 @@ class Fatal(Exception):
 
 class MergeVertexGroupsWithSameNumber(bpy.types.Operator):
     bl_idname = "object.merge_vertex_group_with_same_number"
-    bl_label = "Merge Vertex Groups"
+    bl_label = "合并具有相同数字前缀名称的顶点组"
 
     def execute(self, context):
         return merge_vertex_group_with_same_number(self, context)
@@ -147,7 +147,7 @@ def fill_vertex_group_gaps(self, context):
 
 class FillVertexGroupGaps(bpy.types.Operator):
     bl_idname = "object.fill_vertex_group_gaps"
-    bl_label = "Fill Vertex Group Gaps"
+    bl_label = "填充数字顶点组的间隙"
 
     def execute(self, context):
         return fill_vertex_group_gaps(self, context)
@@ -199,7 +199,7 @@ def add_bone_from_vertex_group(self, context):
 
 class AddBoneFromVertexGroup(bpy.types.Operator):
     bl_idname = "object.add_bone_from_vertex_group"
-    bl_label = "Add Bone From Vertex Group"
+    bl_label = "根据顶点组自动生成骨骼"
 
     def execute(self, context):
         return add_bone_from_vertex_group(self, context)
@@ -217,7 +217,7 @@ def remove_not_number_vertex_group(self, context):
 
 class RemoveNotNumberVertexGroup(bpy.types.Operator):
     bl_idname = "object.remove_not_number_vertex_group"
-    bl_label = "Remove Not Number Vertex Group"
+    bl_label = "移除非数字名称的顶点组"
 
     def execute(self, context):
         return remove_not_number_vertex_group(self, context)
@@ -286,7 +286,7 @@ def convert_to_fragment(self, context):
 
 class ConvertToFragmentOperator(bpy.types.Operator):
     bl_idname = "object.convert_to_fragment"
-    bl_label = "Convert To Fragment"
+    bl_label = "转换为一个3Dmigoto碎片用于合并"
 
     def execute(self, context):
         return convert_to_fragment(self, context)
@@ -311,7 +311,7 @@ def delete_loose(self, context):
 
 class MMTDeleteLoose(bpy.types.Operator):
     bl_idname = "object.mmt_delete_loose"
-    bl_label = "Delete Mesh's Loose Vertex"
+    bl_label = "删除物体的松散点"
 
     def execute(self, context):
         return delete_loose(self, context)
@@ -333,7 +333,7 @@ def mmt_reset_rotation(self, context):
 
 class MMTResetRotation(bpy.types.Operator):
     bl_idname = "object.mmt_reset_rotation"
-    bl_label = "Reset x,y,z rotation number to 0 (UE Model)"
+    bl_label = "重置x,y,z的旋转角度为0 (UE Model)"
 
     def execute(self, context):
         return mmt_reset_rotation(self, context)
@@ -350,7 +350,7 @@ def mmt_cancel_auto_smooth(self, context):
 
 class MMTCancelAutoSmooth(bpy.types.Operator):
     bl_idname = "object.mmt_cancel_auto_smooth"
-    bl_label = "Cancel Auto Smooth for NORMAL (UE Model)"
+    bl_label = "取消自动平滑 (UE Model)"
 
     def execute(self, context):
         return mmt_cancel_auto_smooth(self, context)
@@ -369,7 +369,7 @@ def mmt_set_auto_smooth_89(self, context):
 
 class MMTSetAutoSmooth89(bpy.types.Operator):
     bl_idname = "object.mmt_set_auto_smooth_89"
-    bl_label = "Set Auto Smooth to 89° (Unity)"
+    bl_label = "设置Normal的自动平滑为89° (Unity)"
 
     def execute(self, context):
         return mmt_set_auto_smooth_89(self, context)
@@ -432,12 +432,69 @@ def show_indexed_vertices(self, context):
 
 class MMTShowIndexedVertices(bpy.types.Operator):
     bl_idname = "object.mmt_show_indexed_vertices"
-    bl_label = "Show Indexed Vertices and Indexes Number"
+    bl_label = "展示Indexed Vertices和Indexes Number"
 
     def execute(self, context):
         return show_indexed_vertices(self, context)
 
 
+def split_mesh_by_common_vertex_group(self, context):
+    for obj in bpy.context.selected_objects:
+        origin_name = obj.name
+        keys = obj.vertex_groups.keys()
+        real_keys = []
+        for gr in keys:
+            bpy.ops.object.mode_set(mode="EDIT")
+            # Set the vertex group as active
+            obj.vertex_groups.active_index = obj.vertex_groups[gr].index
+
+            # Deselect all verts and select only current VG
+            bpy.ops.mesh.select_all(action='DESELECT')
+            bpy.ops.object.vertex_group_select()
+
+            try:
+                bpy.ops.mesh.separate(type="SELECTED")
+                real_keys.append(gr)
+            except:
+                pass
+
+        for i in range(1, len(real_keys) + 1):
+            new_obj = bpy.data.objects['{}.{:03d}'.format(origin_name, i)]
+            new_obj.name = '{}.{}'.format(origin_name, real_keys[i - 1])
+    # for obj in bpy.context.selected_objects:
+    #     origin_name = obj.name
+    #     keys = obj.vertex_groups.keys()
+    #     real_keys = []
+    #     for gr in keys:
+    #         bpy.ops.object.mode_set(mode="EDIT")
+    #         # Set the vertex group as active
+    #         bpy.ops.object.vertex_group_set_active(group=gr)
+    #
+    #         # Deselect all verts and select only current VG
+    #         bpy.ops.mesh.select_all(action='DESELECT')
+    #         bpy.ops.object.vertex_group_select()
+    #         # bpy.ops.mesh.select_all(action='INVERT')
+    #         try:
+    #             bpy.ops.mesh.separate(type="SELECTED")
+    #             real_keys.append(gr)
+    #         except:
+    #             pass
+    #     for i in range(1, len(real_keys) + 1):
+    #         bpy.data.objects['{}.{:03d}'.format(origin_name, i)].name = '{}.{}'.format(
+    #             origin_name, real_keys[i - 1])
+
+    return {'FINISHED'}
+
+
+class SplitMeshByCommonVertexGroup(bpy.types.Operator):
+    bl_idname = "object.split_mesh_by_common_vertex_group"
+    bl_label = "根据相同的顶点组分割物体"
+
+    def execute(self, context):
+        return split_mesh_by_common_vertex_group(self, context)
+
+
+
 # -----------------------------------这个属于右键菜单注册，单独的函数要往上面放---------------------------------------
 class MigotoRightClickMenu(bpy.types.Menu):
     bl_idname = "VIEW3D_MT_object_3Dmigoto"
@@ -456,6 +513,7 @@ def draw(self, context):
         layout.operator("object.mmt_cancel_auto_smooth")
         layout.operator("object.mmt_set_auto_smooth_89")
         layout.operator("object.mmt_show_indexed_vertices")
+        layout.operator("object.split_mesh_by_common_vertex_group")
 
 
 # 定义菜单项的注册函数
@@ -486,6 +544,7 @@ def execute(self, context):
 
         output_folder_path = mmt_path + "Games\\" + current_game + "\\3Dmigoto\\Mods\\output\\"
 
+        # 这里是根据Config.json中的DrawIB来决定导入时导入具体哪个IB
         import_folder_path_list = []
         for ib_config in game_config_json:
             draw_ib = ib_config["DrawIB"]
@@ -514,6 +573,10 @@ def execute(self, context):
             # 使用 glob.glob 获取匹配的文件列表
             txt_file_list = glob(file_pattern)
             for txt_file_path in txt_file_list:
+                # 如果文件名不包含-则属于我们自动导出的文件名，则不计入统计
+                if os.path.basename(txt_file_path).find("-") == -1:
+                    continue
+
                 # self.report({'INFO'}, "txt file: " + txt_file_path)
                 txt_file_splits = os.path.basename(txt_file_path).split("-")
                 ib_file_name = txt_file_splits[0] + "-" + txt_file_splits[1]
@@ -582,9 +645,11 @@ def execute(self, context):
         selected_collection = bpy.context.collection
 
         # 遍历选中的对象
+        export_time = 0
         for obj in selected_collection.objects:
             # 判断对象是否为网格对象
             if obj.type == 'MESH':
+                export_time = export_time + 1
                 bpy.context.view_layer.objects.active = obj
                 mesh = obj.data  # 获取网格数据
 
@@ -610,7 +675,9 @@ def execute(self, context):
                 # FIXME: ExportHelper will check for overwriting vb_path, but not ib_path
 
                 export_3dmigoto(self, context, vb_path, ib_path, fmt_path)
-
-        self.report({'INFO'}, "Export Success!")
+        if export_time == 0:
+            self.report({'ERROR'}, "导出失败！请选择一个集合后再点一键导出！")
+        else:
+            self.report({'INFO'}, "一键导出成功！成功导出的部位数量：" + str(export_time))
         return {'FINISHED'}
 

MMT/migoto_format.py

@@ -1,4 +1,6 @@
 # This migoto_format.py is only used in 3dmigoto import and export options.
+import os.path
+
 from .panel import *
 
 # 这里使用type关键字创建了一个类，类名是DummyIOOBJOrientationHelper，(object,)表示继承自object对象，{}表示没定义属性和方法
@@ -882,6 +884,9 @@ def create_material_with_texture(obj, mesh_name, directory):
     material_name = f"{mesh_name}_Material"
     texture_name = f"{mesh_name}-DiffuseMap.jpg"
 
+    texture_prefix = str(mesh_name).split("-")[0] # Hash值
+    texture_suffix = "-DiffuseMap.jpg"
+
     # Создание нового материала (Create new materials)
     material = bpy.data.materials.new(name=material_name)
     material.use_nodes = True
@@ -893,7 +898,7 @@ def create_material_with_texture(obj, mesh_name, directory):
 
     if bsdf:
         # Поиск текстуры (Search for textures)
-        texture_path = find_texture(texture_name, directory, False)
+        texture_path = find_texture(texture_prefix, texture_suffix, directory)
         if texture_path:
             tex_image = material.node_tree.nodes.new('ShaderNodeTexImage')
             tex_image.image = bpy.data.images.load(texture_path)
@@ -906,22 +911,12 @@ def create_material_with_texture(obj, mesh_name, directory):
             obj.data.materials.append(material)
 
 
-def find_texture(texture_name, directory, debug=False):
-    if debug:
-        print(f"Ищем текстуру (Searching for textures): {texture_name} в директории (catalogue): {directory}")
+def find_texture(texture_prefix, texture_suffix, directory):
     for root, dirs, files in os.walk(directory):
-        if debug:
-            print(f"Проверяем директорию (check directories): {root}")
         for file in files:
-            if debug:
-                print(f"Найден файл (file found): {file}")
-            if file.endswith(texture_name):
+            if file.endswith(texture_suffix) and file.startswith(texture_prefix):
                 texture_path = os.path.join(root, file)
-                if debug:
-                    print(f"Найдена текстура (Find texture): {texture_path}")
                 return texture_path
-    if debug:
-        print(f"Текстура не найдена (Unable to find texture): {texture_name}")
     return None
 
 
@@ -981,10 +976,10 @@ def import_3dmigoto_vb_ib(operator, context, paths, flip_texcoord_v=True, axis_f
     bm.from_mesh(mesh)
 
     # 删除松散点 delete loose before get this
-    bm.verts.ensure_lookup_table()
-    for v in bm.verts:
-        if not v.link_faces:
-            bm.verts.remove(v)
+    # bm.verts.ensure_lookup_table()
+    # for v in bm.verts:
+    #     if not v.link_faces:
+    #         bm.verts.remove(v)
 
     # 将 BMesh 更新回原始网格
     bm.to_mesh(mesh)
@@ -1410,7 +1405,9 @@ def execute(self, context):
         migoto_raw_import_options = self.as_keywords(ignore=('filepath', 'files', 'filter_glob'))
 
         # 我们需要添加到一个新建的集合里，方便后续操作
-        collection = bpy.data.collections.new("MMT-Import")
+        # 这里集合的名称需要为当前文件夹的名称
+        collection_name = os.path.basename(os.path.dirname(self.filepath))
+        collection = bpy.data.collections.new(collection_name)
         bpy.context.scene.collection.children.link(collection)
 
         done = set()