form

Author: tomvanmele

plugin/RV_form.py

@@ -2,57 +2,60 @@
 # venv: rhinovault
 # r: compas>=2.5, compas_rui>=0.3.1, compas_session>=0.4.1, compas_tna>=0.5
 
-
 import rhinoscriptsyntax as rs  # type: ignore
 
 from compas.geometry import Box
 from compas.geometry import bounding_box
 from compas.geometry import scale_vector
 from compas.geometry import sum_vectors
 from compas_rv.datastructures import FormDiagram
-from compas_rv.datastructures import Pattern
-from compas_rv.scene import RhinoPatternObject
+from compas_rv.datastructures import ThrustDiagram
 from compas_rv.session import RVSession
 
 
 def RunCommand():
     session = RVSession()
 
-    pattern: RhinoPatternObject = session.scene.find_by_itemtype(Pattern)
+    pattern = session.find_pattern()
     if not pattern:
         return
 
+    session.clear_all_diagrams()
+
     # =============================================================================
     # Init the form diagram
     # =============================================================================
 
     rs.UnselectAllObjects()
 
-    form = FormDiagram.from_pattern(pattern.mesh)
-    form.name = "FormDiagram"
+    formdiagram = FormDiagram.from_pattern(pattern.mesh)
+    formdiagram.name = "FormDiagram"
 
-    form.vertices_attribute(name="z", value=0)
+    formdiagram.vertices_attribute(name="z", value=0)
 
-    normals = [form.face_normal(face) for face in form.faces_where(_is_loaded=True)]
+    normals = [formdiagram.face_normal(face) for face in formdiagram.faces_where(_is_loaded=True)]
     scale = 1 / len(normals)
     normal = scale_vector(sum_vectors(normals), scale)
     if normal[2] < 0:
-        form.flip_cycles()
+        formdiagram.flip_cycles()
 
-    form.vertices_attribute("is_fixed", False)
+    formdiagram.vertices_attribute("is_fixed", False)
 
     fixed = list(pattern.mesh.vertices_where(is_fixed=True))
 
     if fixed:
         for vertex in fixed:
-            if form.has_vertex(vertex):
-                form.vertex_attribute(vertex, "is_support", True)
+            if formdiagram.has_vertex(vertex):
+                formdiagram.vertex_attribute(vertex, "is_support", True)
 
-    bbox = Box.from_bounding_box(bounding_box(form.vertices_attributes("xyz")))
+    thrustdiagram: ThrustDiagram = formdiagram.copy(cls=ThrustDiagram)
+    thrustdiagram.name = "ThrustDiagram"
+
+    # set an initial value for zmax
+    bbox = Box.from_bounding_box(bounding_box(formdiagram.vertices_attributes("xyz")))
     diagonal = bbox.points[2] - bbox.points[0]
     zmax = 0.25 * diagonal.length
-
-    session.settings.tna.vertical.zmax = zmax
+    session.settings.tna.vertical_zmax = zmax
 
     # =============================================================================
     # Update scene
@@ -62,15 +65,13 @@ def RunCommand():
 
     pattern.show = False
 
-    session.scene.add(form, name=form.name)
+    session.scene.add(formdiagram, name=formdiagram.name)
+    session.scene.add(thrustdiagram, name=thrustdiagram.name, show=False)
+
     session.scene.redraw()
 
     rs.Redraw()
 
-    # =============================================================================
-    # Save session
-    # =============================================================================
-
     if session.settings.autosave:
         session.record(name="Init Form Diagram")
 

plugin/RV_form_modify.py

@@ -2,23 +2,20 @@
 # venv: rhinovault
 # r: compas>=2.5, compas_rui>=0.3.1, compas_session>=0.4.1, compas_tna>=0.5
 
-
 import rhinoscriptsyntax as rs  # type: ignore
 
-from compas_rv.datastructures import FormDiagram
-from compas_rv.scene import RhinoFormObject
 from compas_rv.session import RVSession
 
 
 def RunCommand():
     session = RVSession()
 
-    form: RhinoFormObject = session.scene.find_by_itemtype(FormDiagram)
+    form = session.find_formdiagram()
     if not form:
         return
 
     # =============================================================================
-    # Modify pattern vertices
+    # Modify form vertices
     # =============================================================================
 
     rs.UnselectAllObjects()
@@ -29,14 +26,20 @@ def RunCommand():
         return
 
     if option == "VertexAttributes":
-        selectable = list(form.mesh.vertices())
-        selected = form.select_vertices(selectable)
+        form.show_vertices = list(form.diagram.vertices())
+        form.redraw_vertices()
+
+        selected = form.select_vertices()
+
         if selected:
             form.update_vertex_attributes(selected)
 
     elif option == "EdgeAttributes":
-        selectable = list(form.mesh.edges_where(_is_edge=True))
-        selected = form.select_edges(selectable)
+        form.show_edges = list(form.diagram.edges_where(_is_edge=True))
+        form.redraw_edges()
+
+        selected = form.select_edges()
+
         if selected:
             form.update_edge_attributes(selected)
 
@@ -54,14 +57,8 @@ def RunCommand():
     form.show_fixed = True
     form.show_supports = True
     form.show_edges = True
-    form.clear()
-    form.draw()
-
-    rs.Redraw()
 
-    # =============================================================================
-    # Save session
-    # =============================================================================
+    form.redraw()
 
     if session.settings.autosave:
         session.record(name="Modify Form Diagram")

plugin/RV_form_relax.py

@@ -0,0 +1,48 @@
+#! python3
+# venv: rhinovault
+# r: compas>=2.5, compas_rui>=0.3.1, compas_session>=0.4.1, compas_tna>=0.5
+
+import rhinoscriptsyntax as rs  # type: ignore
+
+from compas_rv.session import RVSession
+
+
+def RunCommand():
+    session = RVSession()
+
+    form = session.find_formdiagram()
+    if not form:
+        return
+
+    # =============================================================================
+    # Pattern relax
+    # =============================================================================
+
+    rs.UnselectAllObjects()
+
+    form.mesh.relax()
+
+    # =============================================================================
+    # Update scene
+    # =============================================================================
+
+    rs.UnselectAllObjects()
+
+    form.show_vertices = True
+    form.show_free = False
+    form.show_fixed = True
+    form.show_supports = True
+    form.show_edges = True
+
+    form.redraw()
+
+    if session.settings.autosave:
+        session.record(name="Relax the FormDiagram")
+
+
+# =============================================================================
+# Run as main
+# =============================================================================
+
+if __name__ == "__main__":
+    RunCommand()

plugin/RV_form_smooth.py

@@ -0,0 +1,65 @@
+#! python3
+# venv: rhinovault
+# r: compas>=2.5, compas_rui>=0.3.1, compas_session>=0.4.1, compas_tna>=0.5
+
+import rhinoscriptsyntax as rs  # type: ignore
+
+from compas.itertools import flatten
+from compas_rv.session import RVSession
+
+
+def RunCommand():
+    session = RVSession()
+
+    form = session.find_formdiagram()
+    if not form:
+        return
+
+    # =============================================================================
+    # Pattern relax
+    # =============================================================================
+
+    rs.UnselectAllObjects()
+
+    anchors = list(form.diagram.vertices_where(is_support=True))
+    fixed = list(form.diagram.vertices_where(is_fixed=True))
+    fixed = anchors + fixed
+
+    options = ["True", "False"]
+    option = rs.GetString("Press Enter to smooth or ESC to exit. Keep all boundaries fixed?", strings=options)
+    if option is None:
+        return
+
+    if option == "True":
+        # perhaps just redefine "vertices_on_boundaries"?
+        fixed += list(flatten(form.diagram.vertices_on_boundaries()))
+        fixed += list(flatten([form.diagram.face_vertices(face) for face in form.diagram.faces_where(_is_loaded=False)]))
+
+    fixed = list(set(fixed))
+
+    form.diagram.smooth_area(fixed=fixed)
+
+    # =============================================================================
+    # Update scene
+    # =============================================================================
+
+    rs.UnselectAllObjects()
+
+    form.show_vertices = True
+    form.show_free = False
+    form.show_fixed = True
+    form.show_supports = True
+    form.show_edges = True
+
+    form.redraw()
+
+    if session.settings.autosave:
+        session.record(name="Smooth the FormDiagram")
+
+
+# =============================================================================
+# Run as main
+# =============================================================================
+
+if __name__ == "__main__":
+    RunCommand()