viz resultants nodes edges

Author: tomvanmele

src/compas_assembly/viewer/app.py

@@ -2,6 +2,7 @@
 
 import os
 from compas.colors import Color
+from compas.geometry import Line
 from compas.utilities import remap_values
 from compas_assembly.datastructures import Interface
 
@@ -16,6 +17,16 @@
 
 
 class DEMController(Controller):
+    def show_nodes(self, *args, **kwargs):
+        state = args[0]
+        self.app.nodes.is_visible = state
+        self.app.view.update()
+
+    def show_edges(self, *args, **kwargs):
+        state = args[0]
+        self.app.edges.is_visible = state
+        self.app.view.update()
+
     def show_interfaces(self, *args, **kwargs):
         state = args[0]
         self.app.interfaces.is_visible = state
@@ -32,6 +43,11 @@ def show_normalforces(self, *args, **kwargs):
         self.app.tension.is_visible = state
         self.app.view.update()
 
+    def show_resultants(self, *args, **kwargs):
+        state = args[0]
+        self.app.resultants.is_visible = state
+        self.app.view.update()
+
     def scale_contactforces(self, *args, **kwargs):
         value = args[0]
         if value <= 50:
@@ -46,22 +62,32 @@ def scale_contactforces(self, *args, **kwargs):
 
         if self.app._compression:
             for o, line in zip(self.app.compression._objects, self.app._compression):
-                o._data.start = line.midpoint - line.vector * 0.5 * scale
-                o._data.end = line.midpoint + line.vector * 0.5 * scale
+                if line.length:
+                    o._data.start = line.midpoint - line.vector * 0.5 * scale
+                    o._data.end = line.midpoint + line.vector * 0.5 * scale
             self.app.compression.update()
 
         if self.app._tension:
             for o, line in zip(self.app.tension._objects, self.app._tension):
-                o._data.start = line.midpoint - line.vector * 0.5 * scale
-                o._data.end = line.midpoint + line.vector * 0.5 * scale
+                if line.length:
+                    o._data.start = line.midpoint - line.vector * 0.5 * scale
+                    o._data.end = line.midpoint + line.vector * 0.5 * scale
             self.app.tension.update()
 
         if self.app._friction:
             for o, line in zip(self.app.friction._objects, self.app._friction):
-                o._data.start = line.midpoint - line.vector * 0.5 * scale
-                o._data.end = line.midpoint + line.vector * 0.5 * scale
+                if line.length:
+                    o._data.start = line.midpoint - line.vector * 0.5 * scale
+                    o._data.end = line.midpoint + line.vector * 0.5 * scale
             self.app.friction.update()
 
+        if self.app._resultants:
+            for o, line in zip(self.app.resultants._objects, self.app._resultants):
+                if line.length:
+                    o._data.start = line.midpoint - line.vector * 0.5 * scale
+                    o._data.end = line.midpoint + line.vector * 0.5 * scale
+            self.app.resultants.update()
+
         self.app.view.update()
 
 
@@ -102,6 +128,58 @@ def add_assembly(
         color_support=Color.red(),
         color_block=Color.grey(),
     ):
+        node_point = {}
+        nodes = []
+        properties = []
+        for node in assembly.graph.nodes():
+            block = assembly.graph.node_attribute(node, "block")
+            is_support = assembly.graph.node_attribute(node, "is_support")
+            color = color_support if is_support else color_block
+            point = block.center()
+            nodes.append(point)
+            node_point[node] = point
+            properties.append(
+                {
+                    "pointcolor": color,
+                    "pointsize": 10,
+                }
+            )
+        self.nodes = self.add(Collection(nodes, properties))
+
+        edges = []
+        properties = []
+        for u, v in assembly.graph.edges():
+            a = node_point[u]
+            b = node_point[v]
+            color = Color.black()
+            edges.append(Line(a, b))
+            properties.append(
+                {
+                    "linecolor": color,
+                    "linewidth": 3,
+                }
+            )
+        self.edges = self.add(Collection(edges, properties))
+
+        blocks = []
+        properties = []
+        for node in assembly.graph.nodes():
+            block = assembly.graph.node_attribute(node, "block")
+            is_support = assembly.graph.node_attribute(node, "is_support")
+            color = color_support if is_support else color_block
+            blocks.append(block)
+            properties.append(
+                {
+                    "facecolor": color.lightened(50),
+                    "linecolor": color,
+                    "linewidth": linewidth,
+                    "show_faces": show_faces,
+                    "show_lines": show_lines,
+                    "show_points": show_points,
+                }
+            )
+        self.blocks = self.add(Collection(blocks, properties), opacity=opacity)
+
         blocks = []
         properties = []
         for node in assembly.graph.nodes():
@@ -134,6 +212,7 @@ def add_assembly(
         compression = []
         tension = []
         friction = []
+        resultants = []
         for edge in assembly.graph.edges():
             interfaces: List[Interface] = assembly.graph.edge_attribute(
                 edge, "interfaces"
@@ -142,6 +221,7 @@ def add_assembly(
                 compression += interface.compressionforces
                 tension += interface.tensionforces
                 friction += interface.frictionforces
+                resultants += interface.resultantforce
 
         self._compression = [line.copy() for line in compression]
         self.compression = self.add(
@@ -164,6 +244,13 @@ def add_assembly(
             linecolor=Color.cyan(),
         )
 
+        self._resultants = [line.copy() for line in resultants]
+        self.resultants = self.add(
+            Collection(resultants),
+            linewidth=3,
+            linecolor=Color.green(),
+        )
+
     def _add_sidebar_items(self, items, *args, **kwargs):
         for item in items:
             if item["type"] == "radio":

src/compas_assembly/viewer/config.json

@@ -96,6 +96,18 @@
     ],
     "toolbar": [],
     "sidebar": [
+        {
+            "type": "checkbox",
+            "text": "Show Nodes",
+            "action": "show_nodes",
+            "checked": true
+        },
+        {
+            "type": "checkbox",
+            "text": "Show Edges",
+            "action": "show_edges",
+            "checked": true
+        },
         {
             "type": "checkbox",
             "text": "Show Interfaces",
@@ -114,6 +126,12 @@
             "action": "show_normalforces",
             "checked": true
         },
+        {
+            "type": "checkbox",
+            "text": "Show Resultants",
+            "action": "show_resultants",
+            "checked": true
+        },
         {
             "type": "slider",
             "annotation": "Scale Contact Forces",