set formdiagram supports

Author: tomvanmele

commands/TNA_formdiagram.py

@@ -103,7 +103,12 @@ def RunCommand():
             if not n:
                 return
 
-            formdiagram = FormDiagram.create_cross(x_span=x_span, y_span=y_span, n=n)
+            formdiagram = FormDiagram.create_cross(
+                x_span=x_span,
+                y_span=y_span,
+                n=n,
+                supports=None,  # type: ignore
+            )
 
             session["params"]["formdiagram"] = "cross"
             session["params"]["x_span"] = x_span
@@ -133,7 +138,13 @@ def RunCommand():
             if not n_hoops:
                 return
 
-            formdiagram = FormDiagram.create_fan(x_span=x_span, y_span=y_span, n_fans=n_fans, n_hoops=n_hoops)
+            formdiagram = FormDiagram.create_fan(
+                x_span=x_span,
+                y_span=y_span,
+                n_fans=n_fans,
+                n_hoops=n_hoops,
+                supports=None,  # type: ignore
+            )
 
             session["params"]["formdiagram"] = "fan"
             session["params"]["x_span"] = x_span
@@ -164,7 +175,13 @@ def RunCommand():
             if not ny:
                 return
 
-            formdiagram = FormDiagram.create_ortho(x_span=x_span, y_span=y_span, nx=nx, ny=ny)
+            formdiagram = FormDiagram.create_ortho(
+                x_span=x_span,
+                y_span=y_span,
+                nx=nx,
+                ny=ny,
+                supports=None,  # type: ignore
+            )
 
             session["params"]["formdiagram"] = "ortho"
             session["params"]["x_span"] = x_span

commands/TNA_supports.py

@@ -0,0 +1,84 @@
+#! python3
+# venv: brg-csd
+# r: compas_masonry
+
+import rhinoscriptsyntax as rs  # type: ignore
+
+from compas_masonry.scene import RhinoFormDiagramObject
+from compas_masonry.session import MasonrySession as Session
+from compas_rui import feedback
+from compas_tna.diagrams import FormDiagram
+
+
+def RunCommand():
+    session = Session()
+
+    formdiagram = session["formdiagram"]
+
+    if not formdiagram:
+        feedback.warn("There is no FormDiagram. Please create one first.")
+        return
+
+    formobject: RhinoFormDiagramObject = session.scene.find_by_itemtype(FormDiagram)  # type: ignore
+    if not formobject:
+        session.scene.add(formdiagram, name="FormDiagram", layer="Masonry::TNA::FormDiagram")  # type: ignore
+
+    formobject.redraw()
+
+    # =============================================================================
+    # Update the supports
+    # =============================================================================
+
+    rs.UnselectAllObjects()
+
+    options = ["Add", "Remove"]
+    option = rs.GetString("Add or Remove supports", strings=options)
+    if not option:
+        return
+
+    if option == "Add":
+        formobject.show_vertices = list(formobject.vertices())  # type: ignore
+        formobject.show_edges = list(formobject.edges())  # type: ignore
+        formobject.redraw()
+
+        selected = formobject.select_vertices()
+
+        if selected:
+            formobject.mesh.vertices_attribute(name="is_support", value=True, keys=selected)
+
+    elif option == "Remove":
+        formobject.show_vertices = list(formobject.vertices())  # type: ignore
+        formobject.show_edges = list(formobject.edges())  # type: ignore
+        formobject.redraw()
+
+        selected = formobject.select_vertices()
+
+        if selected:
+            formobject.mesh.vertices_attribute(name="is_support", value=False, keys=selected)
+
+    # =============================================================================
+    # Update scene
+    # =============================================================================
+
+    rs.UnselectAllObjects()
+
+    formobject.show_edges = False  # type: ignore
+    formobject.show_faces = True  # type: ignore
+    formobject.redraw()
+
+    rs.Redraw()
+
+    # =============================================================================
+    # Save
+    # =============================================================================
+
+    # if session.settings.autosave:
+    #     session.record(name="Analysis")
+
+
+# =============================================================================
+# Run as main
+# =============================================================================
+
+if __name__ == "__main__":
+    RunCommand()