diff --git a/src/jax_fdm/visualization/artists/network_artist.py b/src/jax_fdm/visualization/artists/network_artist.py index afbe941..f16b119 100644 --- 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 index 613a52d..ba6b3ac 100644 --- 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