2025-12-01

extensions/blender_org/govietools/Functions/functions.py

@@ -0,0 +1,190 @@
+import os
+import signal
+import subprocess
+import sys
+from pathlib import Path
+
+import bpy  # type: ignore
+
+addon_dir = ""
+server_pid = None
+
+
+def get_addon_dir(file):
+    global addon_dir
+    addon_dir = os.path.dirname(file)
+
+
+def select_object(self, obj):
+    try:
+        bpy.ops.object.select_all(action="DESELECT")
+        bpy.context.view_layer.objects.active = obj
+        obj.select_set(True)
+    except RuntimeError:
+        self.report({"INFO"}, "Object {} not in View Layer".format(obj.name))
+
+
+def select_object_by_mat(self, mat):
+    for obj in bpy.context.scene.objects:
+        if obj.type != "MESH":
+            continue
+        object_materials = [slot.material for slot in obj.material_slots]
+        if mat in object_materials:
+            select_object(self, obj)
+
+
+def start_server(glb_file_path, port):
+    global server_pid
+
+    if server_pid is not None:
+        stop_server()
+
+    server_file_path = Path(addon_dir, "Server", "server.py")
+
+    sub_args = [str(Path(sys.executable))]
+
+    if hasattr(bpy.app, "python_args"):
+        for a in bpy.app.python_args:
+            sub_args.append(a)
+    else:
+        sub_args.append("-I")
+
+    sub_args.append(str(server_file_path))
+    sub_args.append(glb_file_path)
+    sub_args.append(str(port))
+
+    server_process = subprocess.Popen(sub_args)
+
+    server_pid = server_process.pid
+
+
+def stop_server():
+    global server_pid
+
+    if server_pid is None:
+        print("No process id for the server. Has it been started?")
+        return
+
+    try:
+        os.kill(server_pid, signal.SIGTERM)
+        print("Closed process " + str(server_pid))
+    except OSError:
+        print("Could not kill server process")
+
+
+def convert_umlaut(str):
+    spcial_char_map = {ord("ä"): "ae", ord("ü"): "ue", ord("ö"): "oe", ord("ß"): "ss"}
+    return str.translate(spcial_char_map)
+
+
+def rename_annotation():
+    for o in bpy.context.scene.objects:
+        if o.name.startswith("Under"):
+            o.data.name = o.name
+
+
+def get_or_create_joined_collection(scene):
+    joined_collection = scene.collection.children.get("JoinedObjects")
+    if not joined_collection:
+        joined_collection = bpy.data.collections.new(name="JoinedObjects")
+        scene.collection.children.link(joined_collection)
+    return joined_collection
+
+
+def has_material_variants(meshdata):
+    if not hasattr(meshdata, "gltf2_variant_mesh_data"):
+        return False
+
+    if len(meshdata.gltf2_variant_mesh_data) == 0:
+        return False
+
+    return True
+
+
+def join_objects_without_visibility_property(export_only_visible, export_selected):
+    current_scene = bpy.context.scene
+
+    # If set, only visible objects get exported
+    if export_only_visible:
+        invisible_objects = []
+
+        for o in current_scene.objects:
+            if not o.visible_get():
+                invisible_objects.append(o)
+
+        for o in invisible_objects:
+            bpy.data.objects.remove(o)
+
+    # If set, export only objects that are selected
+    if export_selected:
+        not_selected_objects = []
+        for o in current_scene.objects:
+            if not o.select_get():
+                not_selected_objects.append(o)
+
+        for o in not_selected_objects:
+            bpy.data.objects.remove(o)
+
+    # Select only objects that can be joined
+    bpy.ops.object.select_all(action="SELECT")
+
+    for o in current_scene.objects:
+        if "visibility" in o.keys():
+            o.select_set(False)
+            # also deselect all children
+            for child in o.children_recursive:
+                child.select_set(False)
+            continue
+
+        if "clickablePart" in o.keys():
+            o.select_set(False)
+            # also deselect all children
+            for child in o.children_recursive:
+                child.select_set(False)
+            continue
+
+        if o.animation_data is not None:
+            o.select_set(False)
+            continue
+
+        if o.type != "MESH":
+            o.select_set(False)
+            continue
+
+        if has_material_variants(o.data):
+            o.select_set(False)
+            continue
+
+    if len(current_scene.objects) == 0:
+        print("No objects available to join")
+        return False
+
+    # make the first selected object active
+    for o in current_scene.objects:
+        if o.select_get():
+            current_scene.view_layers[0].objects.active = o
+            break
+
+    bpy.ops.object.join()
+    bpy.ops.object.select_all(action="SELECT")
+    return True
+
+
+def optimize_scene(gltf_export_param):
+    bpy.ops.wm.save_mainfile()
+    # Work on a copy of the scene
+    bpy.ops.scene.new(type="FULL_COPY")
+    use_visible = False
+    use_selection = False
+    if "use_visible" in gltf_export_param:
+        use_visible = gltf_export_param["use_visible"]
+    if "use_selection" in gltf_export_param:
+        use_selection = gltf_export_param["use_selection"]
+
+    join_objects_without_visibility_property(
+        export_only_visible=use_visible, export_selected=use_selection
+    )
+
+    bpy.ops.export_scene.gltf(**gltf_export_param)
+    bpy.ops.scene.delete()
+    bpy.ops.wm.open_mainfile(filepath=bpy.data.filepath)

extensions/blender_org/govietools/Functions/gui_functions.py

@@ -0,0 +1,25 @@
+import bpy  # type: ignore
+
+from . import functions
+
+
+def update_sel_item(self, value):
+    scene = self
+    list_object = scene.objects[scene.object_index]
+    functions.select_object(self, list_object)
+
+
+def remap_vis_prop(self, value):
+    for obj in bpy.context.scene.objects:
+        if obj.get("visibility") is not None:
+            obj["visibility"] = int(obj.visibility_bool)
+
+
+def headline(layout, *valueList):
+    box = layout.box()
+    row = box.row()
+
+    split = row.split()
+    for pair in valueList:
+        split = split.split(factor=pair[0])
+        split.label(text=pair[1])