From 60b5e8886b4d45352a877f37c80f8348fdb9bb85 Mon Sep 17 00:00:00 2001
From: "Lin, Yong Xiang" <r888800009@gmail.com>
Date: Wed, 4 Sep 2024 16:48:29 +0800
Subject: [PATCH] feat: show/hide Mesh Group Child Nodes Bounds

The border and origin will now be displayed by default. Users can switch between these options in the editor. If the meshgroup does not have Drawables, a pop-up window will remind the user.
---
 .../viewport/common/mesheditor/package.d      | 10 ++++
 source/creator/viewport/vertex/package.d      | 52 +++++++++++++++++++
 2 files changed, 62 insertions(+)

diff --git a/source/creator/viewport/common/mesheditor/package.d b/source/creator/viewport/common/mesheditor/package.d
index 8fd0b1a1d..73bcb96a9 100644
--- a/source/creator/viewport/common/mesheditor/package.d
+++ b/source/creator/viewport/common/mesheditor/package.d
@@ -40,8 +40,11 @@ private:
     bool mirrorVert = false;
     VertexToolMode toolMode = VertexToolMode.Points;
 
+    bool showMeshgroupNodeChild = true;
+
 public:
     bool deformOnly;
+    bool hasPopupMeshgroupNoDrawables = false;
 
     this(bool deformOnly) {
         this.deformOnly = deformOnly;
@@ -326,5 +329,12 @@ public:
         return true;
     }
 
+    void setShowMeshgroupNodeChild(bool show) {
+        this.showMeshgroupNodeChild = show;
+    }
+
+    bool getShowMeshgroupNodeChild() {
+        return showMeshgroupNodeChild;
+    }
 }
 
diff --git a/source/creator/viewport/vertex/package.d b/source/creator/viewport/vertex/package.d
index f47536685..dde85758d 100644
--- a/source/creator/viewport/vertex/package.d
+++ b/source/creator/viewport/vertex/package.d
@@ -37,9 +37,41 @@ void incViewportVertexTools() {
     editor.viewportTools();
 }
 
+class MeshGroupNodeDrawer {
+    Node node;
+    this(Node node) {
+        this.node = node;
+    }
+
+    void draw() {
+        import inochi2d.core.dbg;
+
+        if (!editor.getShowMeshgroupNodeChild())
+            return;
+
+        node.drawBounds();
+
+        auto trans = node.transform.matrix();
+        inDbgSetBuffer([vec3(0, 0, 0)], [0]);
+        inDbgPointsSize(10);
+        inDbgDrawPoints(vec4(0, 0, 0, 1), trans);
+        inDbgPointsSize(6);
+        inDbgDrawPoints(vec4(0.5, 0.8, 0.5, 1), trans);
+    }
+}
+
 void incViewportVertexOptions() {
     igPushStyleVar(ImGuiStyleVar.ItemSpacing, ImVec2(0, 0));
     igPushStyleVar(ImGuiStyleVar.WindowPadding, ImVec2(4, 4));
+        igBeginGroup();
+            if (incButtonColored("\ue3eb##mesh-group-view-node", ImVec2(0, 0), editor.getShowMeshgroupNodeChild() ? ImVec4.init : ImVec4(0.6, 0.6, 0.6, 1))) {
+                editor.setShowMeshgroupNodeChild(!editor.getShowMeshgroupNodeChild());
+            }
+            incTooltip(_("Show Mesh Group Child Nodes Bounds"));
+        igEndGroup();
+
+        igSameLine(0, 4);
+
         igBeginGroup();
             if (igButton("")) {
                 foreach (d; incSelectedNodes) {
@@ -206,6 +238,7 @@ void incViewportVertexDraw(Camera camera) {
                 mat4 transform = mgroup.transform.matrix.inverse;
                 mgroup.setOneTimeTransform(&transform);
                 Node[] subParts;
+                Node[] subNodes;
                 void findSubDrawable(Node n) {
                     if (auto m = cast(MeshGroup)n) {
                         foreach (child; n.children)
@@ -218,6 +251,8 @@ void incViewportVertexDraw(Camera camera) {
                         subParts ~= d;
                         foreach (child; n.children)
                             findSubDrawable(child);
+                    } else {
+                        subNodes ~= n;
                     }
                 }
                 findSubDrawable(mgroup);
@@ -228,9 +263,26 @@ void incViewportVertexDraw(Camera camera) {
                     a.zSort, 
                     b.zSort) > 0, SwapStrategy.stable)(subParts);
 
+                foreach (part; subNodes) {
+                    auto nd = new MeshGroupNodeDrawer(part);
+                    nd.draw();
+                }
+
                 foreach (part; subParts) {
                     part.drawOne();
                 }
+
+                if (subParts.length == 0 && !editor.hasPopupMeshgroupNoDrawables) {
+                    import creator.widgets.dialog;
+                    editor.hasPopupMeshgroupNoDrawables = true;
+                    incDialog(
+                        "NO_DRAWABLES",
+                        __("No Drawables"),
+                        _("This meshgroup has no drawables to deform. To deform, replace the child with a meshgroup or other type."),
+                        DialogLevel.Info, DialogButtons.OK
+                    );
+                }
+
                 mgroup.setOneTimeTransform(null);
             }
         }