X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FTreePanel.java;h=f636257634463dad9d239241173bcdc313a9b0a9;hb=3d32993c37812a293d61b8cb746ac8a4f246bed0;hp=fea470428a31c7fd549e5ccca435ed0adae848b0;hpb=cd092aaf2104ed3651aa9ee80d0c24bb57dfae27;p=jalview.git

diff --git a/src/jalview/gui/TreePanel.java b/src/jalview/gui/TreePanel.java
index fea4704..f636257 100755
--- a/src/jalview/gui/TreePanel.java
+++ b/src/jalview/gui/TreePanel.java
@@ -38,7 +38,6 @@ import java.util.*;
 
 import javax.imageio.*;
 
-import javax.swing.*;
 import java.beans.PropertyChangeEvent;
 
 
@@ -50,13 +49,11 @@ import java.beans.PropertyChangeEvent;
  */
 public class TreePanel extends GTreePanel
 {
-    SequenceI[] seq;
     String type;
     String pwtype;
-    int start;
-    int end;
     TreeCanvas treeCanvas;
     NJTree tree;
+    AlignViewport av;
 
     /**
      * Creates a new TreePanel object.
@@ -68,11 +65,10 @@ public class TreePanel extends GTreePanel
      * @param s DOCUMENT ME!
      * @param e DOCUMENT ME!
      */
-    public TreePanel(AlignViewport av, Vector seqVector, String type,
-                     String pwtype, int s, int e)
+    public TreePanel(AlignViewport av, String type, String pwtype)
     {
       super();
-      initTreePanel(av, seqVector, type, pwtype, s, e, null);
+      initTreePanel(av, type, pwtype, null);
 
       // We know this tree has distances. JBPNote TODO: prolly should add this as a userdefined default
       showDistances(true);
@@ -87,28 +83,28 @@ public class TreePanel extends GTreePanel
      * @param type DOCUMENT ME!
      * @param pwtype DOCUMENT ME!
      */
-    public TreePanel(AlignViewport av, Vector seqVector, NewickFile newtree,
-                     String type, String pwtype)
+    public TreePanel(AlignViewport av,
+                     String type,
+                     String pwtype,
+                     NewickFile newtree)
     {
       super();
-      initTreePanel(av, seqVector, type, pwtype, 0, seqVector.size(), newtree);
+      initTreePanel(av, type, pwtype, newtree);
     }
 
+    public AlignmentI getAlignment()
+    {
+      return treeCanvas.av.getAlignment();
+    }
 
-    void initTreePanel(AlignViewport av, Vector seqVector, String type,
-                       String pwtype, int s, int e, NewickFile newTree)
+
+    void initTreePanel(AlignViewport av, String type,  String pwtype,
+                       NewickFile newTree)
     {
 
       this.type = type;
       this.pwtype = pwtype;
 
-      start = s;
-      end = e;
-
-      seq = new Sequence[seqVector.size()];
-      seqVector.toArray(seq);
-
-
       treeCanvas = new TreeCanvas(av, scrollPane);
       scrollPane.setViewportView(treeCanvas);
 
@@ -130,6 +126,8 @@ public class TreePanel extends GTreePanel
         }
       });
 
+      this.av = av;
+
 
       TreeLoader tl = new TreeLoader(newTree);
       tl.start();
@@ -154,10 +152,36 @@ public class TreePanel extends GTreePanel
 
       public void run()
       {
+
         if(newtree!=null)
-          tree = new NJTree(seq, newtree);
+          tree = new NJTree(av.alignment.getSequencesArray(),
+                            newtree);
         else
-          tree = new NJTree(seq, type, pwtype, start, end);
+        {
+          int start, end;
+          SequenceI [] seqs;
+          String [] seqStrings = null;
+          if (av.hasHiddenColumns)
+          {
+            seqStrings = av.getViewAsString(true);
+          }
+
+          if(av.getSelectionGroup()==null)
+          {
+            start = 0;
+            end = av.alignment.getWidth();
+            seqs = av.alignment.getSequencesArray();
+          }
+          else
+          {
+            start = av.getSelectionGroup().getStartRes();
+            end = av.getSelectionGroup().getEndRes()+1;
+            seqs = av.getSelectionGroup().getSequencesInOrder(av.alignment);
+          }
+
+          tree = new NJTree(seqs, seqStrings, type, pwtype, start, end);
+        }
+
 
         tree.reCount(tree.getTopNode());
         tree.findHeight(tree.getTopNode());
@@ -165,6 +189,8 @@ public class TreePanel extends GTreePanel
 
         treeCanvas.repaint();
 
+        av.setCurrentTree(tree);
+
       }
     }
 
@@ -232,7 +258,7 @@ public class TreePanel extends GTreePanel
         Desktop.addInternalFrame(cap, buffer.toString(), 500, 100);
 
         jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
-        cap.setText(fout.print(false, true));
+        cap.setText(fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance()));
     }
 
     /**
@@ -259,7 +285,7 @@ public class TreePanel extends GTreePanel
             try
             {
                 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
-                String output = fout.print(false, true); // distances only
+                String output = fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance());
                 java.io.PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter(
                             choice));
                 out.println(output);
@@ -283,6 +309,16 @@ public class TreePanel extends GTreePanel
         treeCanvas.startPrinting();
     }
 
+
+    public void originalSeqData_actionPerformed(ActionEvent e)
+    {
+      CutAndPasteTransfer cap = new CutAndPasteTransfer();
+      cap.setText(tree.printOriginalSequenceData());
+      Desktop.addInternalFrame(cap, "Original Data",
+          400, 400);
+    }
+
+
     /**
      * DOCUMENT ME!
      *