X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fappletgui%2FTreePanel.java;h=103c96ba54cce439e0912ce20d2f0b3b86be0e6e;hb=c95df1d1e8746c6fa31b94c2bf8505f25a30c168;hp=89ff20fc2254536d5e1e4b2a22d6c0fe5cf0af12;hpb=ec955aa655d8320258b91eb079bc57f688cd0a07;p=jalview.git

diff --git a/src/jalview/appletgui/TreePanel.java b/src/jalview/appletgui/TreePanel.java
index 89ff20f..103c96b 100755
--- a/src/jalview/appletgui/TreePanel.java
+++ b/src/jalview/appletgui/TreePanel.java
@@ -1,101 +1,314 @@
+/*
+ * Jalview - A Sequence Alignment Editor and Viewer
+ * Copyright (C) 2006 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
+ */
+
 package jalview.appletgui;
 
-import jalview.datamodel.*;
-import jalview.analysis.*;
-import jalview.jbappletgui.GTreePanel;
-import java.awt.event.*;
 import java.util.*;
 
+import java.awt.event.*;
+import java.awt.*;
 
-public class TreePanel extends GTreePanel
+import jalview.analysis.*;
+import jalview.datamodel.*;
+import jalview.io.*;
+
+public class TreePanel extends Frame implements ActionListener, ItemListener
 {
   SequenceI[] seq;
   String type;
   String pwtype;
-  AlignViewport av;
   int start;
   int end;
   TreeCanvas treeCanvas;
   NJTree tree;
+  AlignViewport av;
 
   public NJTree getTree()
   {
     return tree;
   }
 
-  public TreePanel(AlignViewport av, Vector seqVector, String type, String pwtype, int s, int e)
+  /**
+   * Creates a new TreePanel object.
+   *
+   * @param av DOCUMENT ME!
+   * @param seqVector DOCUMENT ME!
+   * @param type DOCUMENT ME!
+   * @param pwtype DOCUMENT ME!
+   * @param s DOCUMENT ME!
+   * @param e DOCUMENT ME!
+   */
+  public TreePanel(AlignViewport av, String type, String pwtype)
+  {
+    try
+    {
+      jbInit();
+      this.setMenuBar(jMenuBar1);
+    }
+    catch (Exception ex)
+    {
+      ex.printStackTrace();
+    }
+
+    initTreePanel(av, type, pwtype, null);
+  }
+
+  /**
+   * Creates a new TreePanel object.
+   *
+   * @param av DOCUMENT ME!
+   * @param seqVector DOCUMENT ME!
+   * @param newtree DOCUMENT ME!
+   * @param type DOCUMENT ME!
+   * @param pwtype DOCUMENT ME!
+   */
+  public TreePanel(AlignViewport av,
+                   String type,
+                   String pwtype,
+                   NewickFile newtree)
   {
-    super();
+    try
+    {
+      jbInit();
+      this.setMenuBar(jMenuBar1);
+    }
+    catch (Exception e)
+    {
+      e.printStackTrace();
+    }
 
+    initTreePanel(av, type, pwtype, newtree);
+  }
+
+
+  void initTreePanel(AlignViewport av,
+                     String type,
+                     String pwtype,
+                     NewickFile newTree)
+  {
+
+    this.av = av;
     this.type = type;
     this.pwtype = pwtype;
 
-    start = s;
-    end = e;
+    treeCanvas = new TreeCanvas(av, scrollPane);
+    scrollPane.add(treeCanvas);
+
+
+    TreeLoader tl = new TreeLoader(newTree);
+    tl.start();
 
-    String longestName = "";
-    seq = new Sequence [seqVector.size()];
-    for (int i=0;i < seqVector.size();i++)
+  }
+
+  void showOriginalData()
+  {
+    // decide if av alignment is sufficiently different to original data to warrant a new window to be created
+    // create new alignmnt window with hidden regions (unhiding hidden regions yields unaligned seqs)
+    // or create a selection box around columns in alignment view
+    // test Alignment(SeqCigar[])
+    if(tree.seqData!=null)
     {
-      seq[i] = (Sequence) seqVector.elementAt(i);
-      if(seq[i].getName().length()>longestName.length())
-        longestName = seq[i].getName();
+      Object[] alAndColsel = tree.seqData.getAlignmentAndColumnSelection(av.
+          getGapCharacter());
+
+      if (alAndColsel != null && alAndColsel[0] != null)
+      {
+        Alignment al = new Alignment( (SequenceI[]) alAndColsel[0]);
+        AlignFrame af = new AlignFrame(al,
+                                       av.applet,
+                                       "Original Data for Tree",
+                                       false);
+
+        af.viewport.setHiddenColumns( (ColumnSelection) alAndColsel[1]);
+      }
     }
+    else
+      System.out.println("Original Tree Data not available");
+  }
 
-    tree = new NJTree(seq, type, pwtype, start, end);
+  class TreeLoader extends Thread
+  {
+    NewickFile newtree;
+    jalview.datamodel.AlignmentView odata=null;
 
-    treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName);
+    public TreeLoader(NewickFile newtree)
+    {
+      this.newtree = newtree;
+    }
 
-    tree.reCount(tree.getTopNode());
-    tree.findHeight(tree.getTopNode());
-    scrollPane.add(treeCanvas);
+    public void run()
+    {
+      if(newtree!=null)
+      {
+        if (odata == null)
+          tree = new NJTree(av.alignment.getSequencesArray(),
+                            newtree);
+        else
+          tree = new NJTree(av.alignment.getSequencesArray(), odata, newtree);
 
-  }
+        }
+        else
+        {
+          int start, end;
+          SequenceI [] seqs;
+          AlignmentView seqStrings = av.getAlignmentView(av.getSelectionGroup()!=null);
+          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);
+        }
 
-public String getText(String format) {
-  return null;
-}
+      tree.reCount(tree.getTopNode());
+      tree.findHeight(tree.getTopNode());
+      treeCanvas.setTree(tree);
+      if(newtree!=null)
+      {
+        distanceMenu.setState(newtree.HasDistances());
+        bootstrapMenu.setState(newtree.HasBootstrap());
+        treeCanvas.setShowBootstrap(newtree.HasBootstrap());
+        treeCanvas.setShowDistances(newtree.HasDistances());
+      }
 
-  protected void fitToWindow_actionPerformed(ActionEvent e)
+
+      treeCanvas.repaint();
+
+      av.setCurrentTree(tree);
+
+    }
+  }
+
+  public void actionPerformed(ActionEvent evt)
   {
-    treeCanvas.fitToWindow = fitToWindow.getState();
-    repaint();
+    if(evt.getSource()==newickOutput)
+      newickOutput_actionPerformed();
+    else if(evt.getSource()==fontSize)
+      fontSize_actionPerformed();
+    else if(evt.getSource()==inputData)
+      showOriginalData();
   }
 
+  public void itemStateChanged(ItemEvent evt)
+  {
+    if(evt.getSource()==fitToWindow)
+      treeCanvas.fitToWindow = fitToWindow.getState();
+
+    else if(evt.getSource()==distanceMenu)
+      treeCanvas.setShowDistances(distanceMenu.getState());
 
-  protected void fontSize_actionPerformed(ActionEvent e)
+    else if(evt.getSource()==bootstrapMenu)
+      treeCanvas.setShowBootstrap(bootstrapMenu.getState());
+
+    else if(evt.getSource()==placeholdersMenu)
+      treeCanvas.setMarkPlaceholders(placeholdersMenu.getState());
+
+    treeCanvas.repaint();
+  }
+
+
+  public void newickOutput_actionPerformed()
+  {
+    jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
+    String output = fout.print(false, true);
+    CutAndPasteTransfer cap = new CutAndPasteTransfer(false, null);
+    cap.setText(output);
+    java.awt.Frame frame = new java.awt.Frame();
+    frame.add(cap);
+    jalview.bin.JalviewLite.addFrame(frame, type + " " + pwtype, 500, 100);
+  }
+
+  public java.awt.Font getTreeFont()
   {
- /*   if( treeCanvas==null )
-     return;
-
-   String size = fontSize.getLabel().substring( fontSize.getLabel().indexOf("-")+1);
-
-   Object selection = JOptionPane.showInternalInputDialog(Desktop.desktop,
-                               "Select font size",
-                               "Font size",
-                               JOptionPane.QUESTION_MESSAGE,
-                               null, new String[]{"1","2","4","6","8","10","12","14","16","18","20"}
-                               ,"Font Size - "+size);
-   if(selection!=null)
-   {
-     fontSize.setText("Font Size - " + selection);
-
-     int i = Integer.parseInt(selection.toString());
-     treeCanvas.setFontSize(i);
-   }
-    scrollPane.setViewportView(treeCanvas);
-*/
+    return treeCanvas.font;
   }
 
-  protected void distanceMenu_actionPerformed(ActionEvent e)
+  public void setTreeFont(java.awt.Font font)
   {
-    treeCanvas.setShowDistances(distanceMenu.getState());
+    treeCanvas.font = font;
+    treeCanvas.repaint();
   }
 
-  protected void bootstrapMenu_actionPerformed(ActionEvent e)
+  protected void fontSize_actionPerformed()
   {
-    treeCanvas.setShowBootstrap(bootstrapMenu.getState());
+     if( treeCanvas==null )
+        return;
+
+    new FontChooser(this);
+  }
+
+
+
+
+  BorderLayout borderLayout1 = new BorderLayout();
+  protected ScrollPane scrollPane = new ScrollPane();
+  MenuBar jMenuBar1 = new MenuBar();
+  Menu jMenu2 = new Menu();
+  protected MenuItem fontSize = new MenuItem();
+  protected CheckboxMenuItem bootstrapMenu = new CheckboxMenuItem();
+  protected CheckboxMenuItem distanceMenu = new CheckboxMenuItem();
+  protected CheckboxMenuItem placeholdersMenu = new CheckboxMenuItem();
+  protected CheckboxMenuItem fitToWindow = new CheckboxMenuItem();
+  Menu fileMenu = new Menu();
+  MenuItem newickOutput = new MenuItem();
+  MenuItem inputData = new MenuItem();
+
+  private void jbInit() throws Exception {
+      setLayout(borderLayout1);
+      this.setBackground(Color.white);
+      this.setFont(new java.awt.Font("Verdana", 0, 12));
+      jMenu2.setLabel("View");
+      fontSize.setLabel("Font...");
+      fontSize.addActionListener(this);
+      bootstrapMenu.setLabel("Show Bootstrap Values");
+      bootstrapMenu.addItemListener(this);
+      distanceMenu.setLabel("Show Distances");
+      distanceMenu.addItemListener(this);
+      placeholdersMenu.setLabel("Mark Unassociated Leaves");
+      placeholdersMenu.addItemListener(this);
+      fitToWindow.setState(true);
+      fitToWindow.setLabel("Fit To Window");
+      fitToWindow.addItemListener(this);
+      fileMenu.setLabel("File");
+      newickOutput.setLabel("Newick Format");
+      newickOutput.addActionListener(this);
+    inputData.setLabel("Input Data...");
+
+    add(scrollPane, BorderLayout.CENTER);
+      jMenuBar1.add(fileMenu);
+      jMenuBar1.add(jMenu2);
+      jMenu2.add(fitToWindow);
+      jMenu2.add(fontSize);
+      jMenu2.add(distanceMenu);
+      jMenu2.add(bootstrapMenu);
+      jMenu2.add(placeholdersMenu);
+    fileMenu.add(newickOutput);
+    fileMenu.add(inputData);
+    inputData.addActionListener(this);
   }
 
 }