incorporating horizontal conduit

Author: juney-lee

plugin/RV_tna_horizontal.py

@@ -8,28 +8,95 @@
 from compas.geometry import Box
 from compas.geometry import bounding_box
 from compas_rv.session import RVSession
+from compas_rv.conduits import HorizontalConduit
 from compas_tna.equilibrium import horizontal_nodal
 
 
 def RunCommand():
+
+    def redraw (k, xy, edges):
+        if k % conduit.refreshrate:
+            return
+        print(k)
+        conduit.lines = [[[xy[i][1], -xy[i][0]], [xy[j][1], -xy[j][0]]] for i, j in edges]
+        conduit.redraw()
+
+    # =============================================================================
+    # Check scene
+    # =============================================================================
+
     session = RVSession()
 
     form = session.find_formdiagram()
     if not form:
+        print("There is no FormDiagram in the scene.")
         return
 
     force = session.find_forcediagram()
     if not force:
+        print("There is no ForceDiagram in the scene.")
         return
 
     # =============================================================================
-    # Compute horizontal
+    # Options prompt
     # =============================================================================
 
     kmax = session.settings.tna.horizontal_kmax
     alpha = session.settings.tna.horizontal_alpha
+    refresh = session.settings.tna.horizontal_refreshrate
+
+    options = ['Alpha', 'Iterations', 'RefreshRate']
+
+    while True:
+        option = rs.GetString('Press Enter to run or ESC to exit.', strings=options)
+
+        if option is None:
+            print("Horizontal equilibrium aborted!")
+            return
+
+        if not option:
+            break
+
+        if option == 'Alpha':
+            alpha_options = ['form{}'.format(int(i * 10)) for i in range(11)]
+            alpha_default = 0
+            for i in range(11):
+                if alpha == i * 10:
+                    alpha_default = i
+                    break
+            temp = rs.GetString('Select parallelisation weight', alpha_options[alpha_default], alpha_options)
+            if not temp:
+                alpha = 100
+            else:
+                alpha = int(temp[4:])
+
+        elif option == 'Iterations':
+            new_kmax = rs.GetInteger('Enter number of iterations', kmax, 1, 10000)
+            if new_kmax or new_kmax is not None:
+                kmax = new_kmax
+
+        elif option == 'RefreshRate':
+            new_refresh = rs.GetInteger('Refresh rate for dynamic visualisation', refresh, 0, 1000)
+            if new_refresh or new_refresh is not None:
+                refresh = new_refresh
+
+    if refresh > kmax:
+        refresh = 0
+
+    session.settings.tna.horizontal_kmax = kmax
+    session.settings.tna.horizontal_alpha = alpha
+    session.settings.tna.horizontal_refreshrate = refresh
 
-    horizontal_nodal(form.diagram, force.diagram, kmax=kmax, alpha=alpha)
+    # =============================================================================
+    # Compute horizontal
+    # =============================================================================
+
+    if refresh > 0:
+        conduit = HorizontalConduit([], refreshrate=refresh)
+        with conduit.enabled():
+            horizontal_nodal(form.diagram, force.diagram, kmax=kmax, alpha=alpha, callback=redraw)
+    else:
+        horizontal_nodal(form.diagram, force.diagram, kmax=kmax, alpha=alpha)
 
     bbox_form = Box.from_bounding_box(bounding_box(form.diagram.vertices_attributes("xyz")))
     bbox_force = Box.from_bounding_box(bounding_box(force.diagram.vertices_attributes("xyz")))
@@ -49,9 +116,12 @@ def RunCommand():
     tol_angles = session.settings.tna.horizontal_max_angle
 
     if max_angle < tol_angles:
-        print(f"Horizontal equilibrium found!\nMaximum angle deviation: {max_angle} < {tol_angles}")
+        print('Horizontal equilibrium found!')
+        print('Maximum angle deviation: {} < {}'.format(max_angle, tol_angles))
+
     else:
-        rs.MessageBox(f"Horizontal equilibrium NOT found! Consider running more iterations.\nMaximum angle deviation: {max_angle} > {tol_angles}")
+        print('Horizontal equilibrium NOT found! Consider running more iterations.')
+        print('Maximum angle deviation: {} > {}'.format(max_angle, tol_angles))
 
     rs.UnselectAllObjects()