JAL-1953 2.11.2 with Archeopteryx!

[jalview.git] / src / jalview / ext / archaeopteryx / TreeNode.java

diff --git a/src/jalview/ext/archaeopteryx/TreeNode.java b/src/jalview/ext/archaeopteryx/TreeNode.java
index 9e1b0d1..0844171 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,18 +20,23 @@ 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;
-    if (!aptxNode.getNodeData().getSequence().isEmpty())
+    if (aptxNode.getNodeData().getSequence() != null)
     {
     nodeSeq = DataConversions
-              .createJalviewSequence(aptxNode.getNodeData().getSequence());
+              .createJalviewSequence(aptxNode);
     }
     originalNodes.put(aptxNode, this);
     wrappedNodes.put(this, 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);
 
   }
 
@@ -243,4 +247,10 @@ public class TreeNode implements TreeNodeI
     return node.getYcoord();
   }
 
+  @Override
+  public boolean isCollapsed()
+  {
+    return node.isCollapse();
+  }
+
 }