op

Author: NicoMico

README.md

@@ -7,6 +7,9 @@
 
 有无法解决的问题可以联系我或者提交issue。
 
+DBMT的Blender插件就在DBMT解压缩后文件夹下的Plugins目录下，以blender开头V10XX结尾，V后面是版本号。
+尽量不要使用正在开发中的代码，避免无法适配。
+
 # 开发环境
 - Windows11
 - Visual Studio Code

blender/__init__.py

@@ -20,7 +20,7 @@
     "name": "DBMT-Blender-Plugin",
     "description": "DBMT的Blender插件 目前只支持3.6LTS版本",
     "blender": (3, 6, 0),
-    "version": (1, 0, 1, 8),
+    "version": (1, 0, 1, 9),
     "location": "View3D",
     "category": "Generic"
 }
@@ -64,7 +64,11 @@
     SplitMeshByCommonVertexGroup,
 
     # MMD类型动画Mod支持
-    MMDModIniGenerator
+    MMDModIniGenerator,
+
+    # Collection's right click menu item.
+    DBMT_MarkCollection_Switch,
+    DBMT_MarkCollection_Toggle
 )
 
 
@@ -84,6 +88,8 @@ def register():
     # 右键菜单
     bpy.types.VIEW3D_MT_object_context_menu.append(menu_func_migoto_right_click)
 
+    bpy.types.OUTLINER_MT_collection.append(menu_dbmt_mark_collection_switch)
+
     # 在Blender退出前保存选择的MMT的路径
     bpy.app.handlers.depsgraph_update_post.append(save_mmt_path)
 
@@ -99,6 +105,9 @@ def unregister():
     # 退出时移除右键菜单
     bpy.types.VIEW3D_MT_object_context_menu.remove(menu_func_migoto_right_click)
 
+
+    bpy.types.OUTLINER_MT_collection.remove(menu_dbmt_mark_collection_switch)
+
     # 退出注册时删除MMT的MMD变量
     del bpy.types.Scene.mmt_mmd_animation_mod_start_frame
     del bpy.types.Scene.mmt_mmd_animation_mod_end_frame

blender/rightclick_menu/mesh_operator.py

@@ -137,3 +137,37 @@ def draw(self, context):
 # 定义菜单项的注册函数
 def menu_func_migoto_right_click(self, context):
     self.layout.menu(MigotoRightClickMenu.bl_idname)
+
+
+
+# --------------------------
+# Right click menu for collection.
+class DBMT_MarkCollection_Switch(bpy.types.Operator):
+    # bl_idname必须小写，中间可以下划线分割，不然报错无法加载
+    bl_idname = "object.mark_collection_switch"
+    bl_label = "Mark Collection Switch"
+    bl_options = {'UNDO'}
+
+    def execute(self, context):
+        # 示例：打印当前选中的集合名称
+        if context.collection:
+            context.collection.color_tag = "COLOR_03"
+        return {'FINISHED'}
+
+class DBMT_MarkCollection_Toggle(bpy.types.Operator):
+    # bl_idname必须小写，中间可以下划线分割，不然报错无法加载
+    bl_idname = "object.mark_collection_toggle"
+    bl_label = "Mark Collection Toggle"
+    bl_options = {'UNDO'}
+
+    def execute(self, context):
+        # 示例：打印当前选中的集合名称
+        if context.collection:
+            context.collection.color_tag = "COLOR_04"
+        return {'FINISHED'}
+
+
+def menu_dbmt_mark_collection_switch(self, context):
+    self.layout.separator()
+    self.layout.operator(DBMT_MarkCollection_Switch.bl_idname, text="分支:标记为按键开关类型")
+    self.layout.operator(DBMT_MarkCollection_Toggle.bl_idname, text="分支:标记为按键切换类型")

blender/unity/migoto_export.py

@@ -369,43 +369,62 @@ def execute(self, context):
         if len(selected_collection.children) == 0:
             self.report({'ERROR'},"当前选中集合不是一个标准的分支模型集合，请检查您是否以分支集合方式导入了模型。")
             return {'FINISHED'}
+        
         # 构建一个export.json，记录当前集合所有object层级关系
-        # part_name_collection_list 装当前集合下所有的集合名称
-        # 然后是一个键值对，键是集合名称，值是一个列表，装了集合中所有mesh名称，隐藏的mesh不计入在内。
         export_json = {}
-        for child_collection in selected_collection.children:
-
-            export_part_name = child_collection.name
+        for part_collection in selected_collection.children:
+            # 从集合名称中获取导出后部位的名称，如果有.001这种自动添加的后缀则去除掉
+            export_part_name = part_collection.name
             if "." in export_part_name:
                 export_part_name = export_part_name[0:len(export_part_name) - 4]
 
-            collection_obj_name_list = []
-            for obj in child_collection.objects:
-                # 判断对象是否为网格对象，并且不是隐藏状态
-                if obj.type == 'MESH' and obj.hide_get() == False:
-                    collection_obj_name_list.append("export-" + obj.data.name)
-            export_json[export_part_name] = collection_obj_name_list
+            part_collection_json = {}
+            for model_collection in part_collection.children:
+                # 声明一个model_collection对象
+                model_collection_json = {}
+
+                # 先根据颜色确定是什么类型的集合 03是开关 04是分支
+                model_collection_type = "default"
+                if model_collection.color_tag == "COLOR_03":
+                    model_collection_type = "switch"
+                elif model_collection.color_tag == "COLOR_04":
+                    model_collection_type = "toggle"
+                model_collection_json["type"] = model_collection_type
+
+                # 集合中的模型列表
+                model_collection_obj_name_list = []
+                for obj in model_collection.objects:
+                    # 判断对象是否为网格对象，并且不是隐藏状态
+                    if obj.type == 'MESH' and obj.hide_get() == False:
+                        model_collection_obj_name_list.append("export-" + obj.data.name)
+                model_collection_json["model"] = model_collection_obj_name_list
+
+                # 集合的名称后面用作注释标记到ini文件中
+                part_collection_json[model_collection.name] = model_collection_json
+
+            export_json[export_part_name] = part_collection_json
 
         # 将字典转换为 JSON 格式的字符串
         json_string = json.dumps(export_json, ensure_ascii=False, indent=4)
-
         # 将 JSON 字符串写入文件
         with open(output_folder_path + draw_ib + "\\" + 'export.json', 'w', encoding='utf-8') as f:
             f.write(json_string)
 
         # 随后直接导出所有模型
         export_time = 0
-        for child_collection in selected_collection.children:
-            for obj in child_collection.objects:
-                # 判断对象是否为网格对象
-                if obj.type == 'MESH' and obj.hide_get() == False:
-                    export_time = export_time + 1
-                    bpy.context.view_layer.objects.active = obj
-                    vb_path = output_folder_path + draw_ib + "\\" + "export-" + obj.data.name + ".vb"
-                    ib_path = os.path.splitext(vb_path)[0] + '.ib'
-                    fmt_path = os.path.splitext(vb_path)[0] + '.fmt'
-                    
-                    export_3dmigoto(self, context, vb_path, ib_path, fmt_path)
+        for child_part_collection in selected_collection.children:
+            for model_collection in child_part_collection.children:
+                for obj in model_collection.objects:
+                    # 判断对象是否为网格对象
+                    if obj.type == 'MESH' and obj.hide_get() == False:
+                        bpy.context.view_layer.objects.active = obj
+                        vb_path = output_folder_path + draw_ib + "\\" + "export-" + obj.data.name + ".vb"
+                        ib_path = os.path.splitext(vb_path)[0] + '.ib'
+                        fmt_path = os.path.splitext(vb_path)[0] + '.fmt'
+                        
+                        export_3dmigoto(self, context, vb_path, ib_path, fmt_path)
+
+                        export_time = export_time + 1
 
         if export_time == 0:
                 self.report({'ERROR'}, "导出失败！请选择一个集合后再点一键导出！")

blender/unity/migoto_import.py

@@ -493,13 +493,23 @@ def execute(self, context):
 
             # create a new collection.
             collection = bpy.data.collections.new(folder_draw_ib_name)
+            collection.color_tag = "COLOR_07" #红色
 
             # link to scene.collection.
             bpy.context.scene.collection.children.link(collection)
 
+            part_count = 1
             for prefix in import_prefix_list:
+                component_name = "Component " + str(part_count)
                 # Create a child collection for every part in a single drawib.
-                child_collection = bpy.data.collections.new(prefix)
+                child_collection = bpy.data.collections.new(component_name)
+                child_collection.color_tag = "COLOR_05" #蓝色
+                child_collection.tag = True
+
+
+                defualt_collection = bpy.data.collections.new("default")
+                defualt_collection.color_tag = "COLOR_04" #绿色
+                defualt_collection.tag = False
 
                 # combine and verify if path exists.
                 vb_bin_path = import_folder_path + "\\" + prefix + '.vb'
@@ -522,17 +532,21 @@ def execute(self, context):
                     if fmt_path is not None:
                         obj_result = import_3dmigoto_raw_buffers(self, context, fmt_path=fmt_path, vb_path=vb_bin_path,
                                                                   ib_path=ib_bin_path, **migoto_raw_import_options)
-                        child_collection.objects.link(obj_result)
+                        defualt_collection.objects.link(obj_result)
 
                     else:
                         self.report({'ERROR'}, "Can't find .fmt file!")
 
+                    child_collection.children.link(defualt_collection)
+                    
                     # bind to parent collection
                     collection.children.link(child_collection)
 
 
                 except Fatal as e:
                     self.report({'ERROR'}, str(e))
+
+                part_count = part_count + 1
             # Select all objects under collection.
             select_collection_objects(collection)