[Visualization] Fixed bug in reaction forces coloring

arpastrana · Sep 20, 2023 · b350021 · b350021
1 parent f8434d8
commit b350021
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/jax_fdm/visualization/artists/network_artist.py b/src/jax_fdm/visualization/artists/network_artist.py
@@ -81,6 +81,9 @@ def __init__(self,
         self.edge_color = edgecolor
 
         self.load_color = loadcolor or self.default_loadcolor
+
+        if edgecolor == "force":
+            self.default_reactioncolor = Color.from_rgb255(0, 150, 10)
         self.reaction_color = reactioncolor or self.default_reactioncolor
 
         self.load_scale = loadscale or self.default_loadscale

diff --git a/src/jax_fdm/visualization/viewers/network_artist.py b/src/jax_fdm/visualization/viewers/network_artist.py
@@ -318,7 +318,7 @@ def add_reactions(self):
         reactions = {}
 
         for node, arrow in self.collection_reactions.items():
-            obj = self.add_reaction(arrow, self.default_reactioncolor)
+            obj = self.add_reaction(arrow, self.reaction_color)
             reactions[node] = obj
 
         return reactions