Fixed bug http://dragonfly.compbio.dundee.ac.uk/mantis/view.php?id=9326 for viewing...
authorjprocter <Jim Procter>
Fri, 18 Nov 2005 16:07:41 +0000 (16:07 +0000)
committerjprocter <Jim Procter>
Fri, 18 Nov 2005 16:07:41 +0000 (16:07 +0000)
src/jalview/analysis/NJTree.java
src/jalview/gui/TreePanel.java
src/jalview/io/NewickFile.java

index 46a2ced..99592f2 100755 (executable)
 */\r
 package jalview.analysis;\r
 \r
-import jalview.datamodel.*;\r
-\r
-import jalview.io.NewickFile;\r
-\r
-import jalview.schemes.ResidueProperties;\r
+import java.util.*;\r
 \r
+import jalview.datamodel.*;\r
+import jalview.io.*;\r
+import jalview.schemes.*;\r
 import jalview.util.*;\r
 \r
-import java.util.*;\r
-\r
 \r
 /**\r
  * DOCUMENT ME!\r
@@ -60,6 +57,10 @@ public class NJTree
     Object leaves = null;\r
     int start;\r
     int end;\r
+    boolean hasDistances = true; // normal case for jalview trees\r
+    boolean hasBootstrap = false; // normal case for jalview trees\r
+\r
+  private boolean hasRootDistance = true;\r
 \r
     /**\r
      * Creates a new NJTree object.\r
@@ -81,6 +82,11 @@ public class NJTree
     public NJTree(SequenceI[] seqs, NewickFile treefile)\r
     {\r
         top = treefile.getTree();\r
+\r
+        hasDistances = treefile.HasDistances();\r
+        hasBootstrap = treefile.HasBootstrap();\r
+        hasRootDistance = treefile.HasRootDistance();\r
+\r
         maxheight = findHeight(top);\r
 \r
         SequenceIdMatcher algnIds = new SequenceIdMatcher(seqs);\r
@@ -375,7 +381,7 @@ public class NJTree
     public void findNewNJDistances(SequenceNode tmpi, SequenceNode tmpj,\r
         float dist)\r
     {\r
-     \r
+\r
         tmpi.dist = ((dist + ri) - rj) / 2;\r
         tmpj.dist = (dist - tmpi.dist);\r
 \r
@@ -1134,6 +1140,27 @@ public class NJTree
     {\r
         return top;\r
     }\r
+    /**\r
+     *\r
+     * @return true if tree has real distances\r
+     */\r
+    public boolean isHasDistances() {\r
+      return hasDistances;\r
+    }\r
+\r
+    /**\r
+     *\r
+     * @return true if tree has real bootstrap values\r
+     */\r
+    public boolean isHasBootstrap() {\r
+      return hasBootstrap;\r
+    }\r
+\r
+  public boolean isHasRootDistance()\r
+  {\r
+    return hasRootDistance;\r
+  }\r
+\r
 }\r
 \r
 \r
index 04be0fd..9c1e42e 100755 (executable)
@@ -236,7 +236,7 @@ public class TreePanel extends GTreePanel
         Desktop.addInternalFrame(cap, buffer.toString(), 500, 100);\r
 \r
         jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());\r
-        cap.setText(fout.print(false, true));\r
+        cap.setText(fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance()));\r
     }\r
 \r
     /**\r
@@ -263,7 +263,7 @@ public class TreePanel extends GTreePanel
             try\r
             {\r
                 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());\r
-                String output = fout.print(false, true); // distances only\r
+                String output = fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance());\r
                 java.io.PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter(\r
                             choice));\r
                 out.println(output);\r
index d696605..077faa0 100755 (executable)
@@ -1,20 +1,20 @@
-/*
-* Jalview - A Sequence Alignment Editor and Viewer
-* Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle
-*
-* This program is free software; you can redistribute it and/or
-* modify it under the terms of the GNU General Public License
-* as published by the Free Software Foundation; either version 2
-* of the License, or (at your option) any later version.
-*
-* This program is distributed in the hope that it will be useful,
-* but WITHOUT ANY WARRANTY; without even the implied warranty of
-* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-* GNU General Public License for more details.
-*
-* You should have received a copy of the GNU General Public License
-* along with this program; if not, write to the Free Software
-* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
+/*\r
+* Jalview - A Sequence Alignment Editor and Viewer\r
+* Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
+*\r
+* This program is free software; you can redistribute it and/or\r
+* modify it under the terms of the GNU General Public License\r
+* as published by the Free Software Foundation; either version 2\r
+* of the License, or (at your option) any later version.\r
+*\r
+* This program is distributed in the hope that it will be useful,\r
+* but WITHOUT ANY WARRANTY; without even the implied warranty of\r
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
+* GNU General Public License for more details.\r
+*\r
+* You should have received a copy of the GNU General Public License\r
+* along with this program; if not, write to the Free Software\r
+* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA\r
 */\r
 \r
 // NewickFile.java\r
@@ -166,6 +166,10 @@ public class NewickFile extends FileParse
         return HasDistances;\r
     }\r
 \r
+    public boolean HasRootDistance()\r
+    {\r
+        return RootHasDistance;\r
+    }\r
     /**\r
      * DOCUMENT ME!\r
      *\r
@@ -199,7 +203,7 @@ public class NewickFile extends FileParse
         String Error = null;\r
         String nodename = null;\r
 \r
-        float DefDistance = (float) 0.00001; // @param Default distance for a node - very very small\r
+        float DefDistance = (float) 0.001; // @param Default distance for a node - very very small\r
         int DefBootstrap = 0; // @param Default bootstrap for a node\r
 \r
         float distance = DefDistance;\r
@@ -323,7 +327,7 @@ public class NewickFile extends FileParse
                 com.stevesoft.pat.Regex nbootstrap = new com.stevesoft.pat.Regex(\r
                         "\\S+([0-9+]+)\\S*:");\r
                 com.stevesoft.pat.Regex ndist = new com.stevesoft.pat.Regex(\r
-                        ":([-0-9.+]+)");\r
+                        ":([-0-9Ee.+]+)");\r
 \r
                 if (uqnodename.search(fstring) &&\r
                         ((uqnodename.matchedFrom(1) == 0) ||\r
@@ -388,12 +392,13 @@ public class NewickFile extends FileParse
                 {\r
                     // Write node info here\r
                     c.setName(nodename);\r
-                    c.dist = (HasDistances) ? distance : 0;\r
-                    c.setBootstrap((HasBootstrap) ? bootstrap : 0);\r
-\r
+                    // Trees without distances still need a render distance\r
+                    c.dist = (HasDistances) ? distance : DefDistance;\r
+                    // be consistent for internal bootstrap defaults too\r
+                    c.setBootstrap((HasBootstrap) ? bootstrap : DefBootstrap);\r
                     if (c == realroot)\r
                     {\r
-                        RootHasDistance = nodehasdistance; // JBPNote This is really UGLY!!!\r
+                        RootHasDistance = nodehasdistance; // JBPNote This is really UGLY!!! Ensure root node gets its given distance\r
                     }\r
                 }\r
                 else\r
@@ -416,8 +421,9 @@ public class NewickFile extends FileParse
                         else\r
                         {\r
                             // Insert a dummy node for polytomy\r
+                            // dummy nodes have distances\r
                             SequenceNode newdummy = new SequenceNode(null, c,\r
-                                    null, 0, 0, true);\r
+                                    null, (HasDistances ? 0 : DefDistance), 0, true);\r
                             newdummy.SetChildren(c.left(), newnode);\r
                             c.setLeft(newdummy);\r
                         }\r
@@ -481,7 +487,7 @@ public class NewickFile extends FileParse
 \r
         if (!RootHasDistance)\r
         {\r
-            root.dist = 0;\r
+            root.dist = (HasDistances) ? 0 : DefDistance;\r
         }\r
     }\r
 \r
@@ -496,9 +502,10 @@ public class NewickFile extends FileParse
     }\r
 \r
     /**\r
-     * DOCUMENT ME!\r
+     * Generate a newick format tree according to internal flags\r
+     * for bootstraps, distances and root distances.\r
      *\r
-     * @return DOCUMENT ME!\r
+     * @return new hampshire tree in a single line\r
      */\r
     public String print()\r
     {\r
@@ -512,11 +519,14 @@ public class NewickFile extends FileParse
     }\r
 \r
     /**\r
-     * DOCUMENT ME!\r
      *\r
-     * @param withbootstraps DOCUMENT ME!\r
      *\r
-     * @return DOCUMENT ME!\r
+     * Generate a newick format tree according to internal flags\r
+     * for distances and root distances and user specificied writing of\r
+     * bootstraps.\r
+     * @param withbootstraps controls if bootstrap values are explicitly written.\r
+     *\r
+     * @return new hampshire tree in a single line\r
      */\r
     public String print(boolean withbootstraps)\r
     {\r
@@ -533,12 +543,14 @@ public class NewickFile extends FileParse
     }\r
 \r
     /**\r
-     * DOCUMENT ME!\r
      *\r
-     * @param withbootstraps DOCUMENT ME!\r
-     * @param withdists DOCUMENT ME!\r
+     * Generate newick format tree according to internal flags\r
+     * for writing root node distances.\r
      *\r
-     * @return DOCUMENT ME!\r
+     * @param withbootstraps explicitly write bootstrap values\r
+     * @param withdists explicitly write distances\r
+     *\r
+     * @return new hampshire tree in a single line\r
      */\r
     public String print(boolean withbootstraps, boolean withdists)\r
     {\r
@@ -555,13 +567,13 @@ public class NewickFile extends FileParse
     }\r
 \r
     /**\r
-     * DOCUMENT ME!\r
+     * Generate newick format tree according to user specified flags\r
      *\r
-     * @param withbootstraps DOCUMENT ME!\r
-     * @param withdists DOCUMENT ME!\r
-     * @param printRootInfo DOCUMENT ME!\r
+     * @param withbootstraps explicitly write bootstrap values\r
+     * @param withdists explicitly write distances\r
+     * @param printRootInfo explicitly write root distance\r
      *\r
-     * @return DOCUMENT ME!\r
+     * @return new hampshire tree in a single line\r
      */\r
     public String print(boolean withbootstraps, boolean withdists,\r
         boolean printRootInfo)\r
@@ -699,8 +711,12 @@ public class NewickFile extends FileParse
             {\r
                 if (c.isDummy())\r
                 {\r
-                    _print(tf, (SequenceNode) c.right());\r
                     _print(tf, (SequenceNode) c.left());\r
+                    if (c.left() != null)\r
+                    {\r
+                      tf.append(",");\r
+                    }\r
+                    _print(tf, (SequenceNode) c.right());\r
                 }\r
                 else\r
                 {\r
@@ -724,6 +740,11 @@ public class NewickFile extends FileParse
     {\r
         try\r
         {\r
+            if (args==null || args.length!=1) {\r
+              System.err.println("Takes one argument - file name of a newick tree file.");\r
+              System.exit(0);\r
+            }\r
+\r
             File fn = new File(args[0]);\r
 \r
             StringBuffer newickfile = new StringBuffer();\r