JAL-2844 branch colour setter improved

diff --git a/src/jalview/ext/archaeopteryx/TreeNode.java b/src/jalview/ext/archaeopteryx/TreeNode.java
index b43e9b4..752826b 100644 (file)
--- a/src/jalview/ext/archaeopteryx/TreeNode.java
+++ b/src/jalview/ext/archaeopteryx/TreeNode.java
@@ -12,7 +12,6 @@ import java.util.Map;
 
 import org.forester.phylogeny.PhylogenyMethods;
 import org.forester.phylogeny.PhylogenyNode;
-import org.forester.phylogeny.data.BranchColor;
 
 public class TreeNode implements TreeNodeI
 {
@@ -21,11 +20,16 @@ public class TreeNode implements TreeNodeI
   private SequenceI nodeSeq;
 
   private static Map<PhylogenyNode, TreeNodeI> originalNodes = new HashMap<>(
-          500); // prolly make this size dynamic
+          500); // prolly make this size dynamic?
 
   private static Map<TreeNodeI, PhylogenyNode> wrappedNodes = new HashMap<>(
           500);
 
+  /**
+   * Please don't use me directly.
+   * 
+   * @param aptxNode
+   */
   private TreeNode(PhylogenyNode aptxNode)
   {
     node = aptxNode;
@@ -117,7 +121,7 @@ public class TreeNode implements TreeNodeI
   @Override
   public void setBranchColor(Color branchColor)
   {
-    node.getBranchData().setBranchColor(new BranchColor(branchColor));
+    PhylogenyMethods.setBranchColorValue(node, branchColor);
 
   }