JAL-2805 expanded tree and treenode interfaces

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

diff --git a/src/jalview/ext/archaeopteryx/Tree.java b/src/jalview/ext/archaeopteryx/Tree.java
index de5f32a..884a359 100644 (file)
--- a/src/jalview/ext/archaeopteryx/Tree.java
+++ b/src/jalview/ext/archaeopteryx/Tree.java
@@ -1,11 +1,11 @@
 package jalview.ext.archaeopteryx;
 
-import jalview.ext.treeviewer.ExternalTree;
-import jalview.ext.treeviewer.ExternalTreeNode;
+import jalview.ext.treeviewer.ExternalTreeI;
+import jalview.ext.treeviewer.ExternalTreeNodeI;
 
 import org.forester.phylogeny.Phylogeny;
 
-public class Tree implements ExternalTree
+public class Tree implements ExternalTreeI
 {
   private final Phylogeny tree;
 
@@ -18,12 +18,13 @@ public class Tree implements ExternalTree
   public Phylogeny getOriginalTree()
   {
     return tree;
+
   }
 
   @Override
-  public ExternalTreeNode getRoot()
+  public ExternalTreeNodeI getRoot()
   {
-    ExternalTreeNode root = new TreeNode(tree.getRoot());
+    ExternalTreeNodeI root = new TreeNode(tree.getRoot());
     return root;
   }
 
@@ -33,4 +34,17 @@ public class Tree implements ExternalTree
     tree.setName(name);
 
   }
+
+  @Override
+  public ExternalTreeNodeI getNodeWithName(String name)
+  {
+    return new TreeNode(tree.getNode(name));
+
+  }
+
+  @Override
+  public String[] getAllLeafNames()
+  {
+    return tree.getAllExternalNodeNames();
+  }
 }