src/jalview/appletgui/TreePanel.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
   4  *
   5  * This file is part of Jalview.
   6  *
   7  * Jalview is free software: you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation, either version 3
  10  * of the License, or (at your option) any later version.
  11  *
  12  * Jalview is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty
  14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15  * PURPOSE.  See the GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  19  * The Jalview Authors are detailed in the 'AUTHORS' file.
  20  */
  21 package jalview.appletgui;
  22
  23 import jalview.analysis.NJTree;
  24 import jalview.api.analysis.ScoreModelI;
  25 import jalview.api.analysis.ViewBasedAnalysisI;
  26 import jalview.datamodel.Alignment;
  27 import jalview.datamodel.AlignmentView;
  28 import jalview.datamodel.HiddenColumns;
  29 import jalview.datamodel.SequenceI;
  30 import jalview.io.NewickFile;
  31 import jalview.schemes.ResidueProperties;
  32 import jalview.util.MessageManager;
  33
  34 import java.awt.BorderLayout;
  35 import java.awt.CheckboxMenuItem;
  36 import java.awt.Color;
  37 import java.awt.Menu;
  38 import java.awt.MenuBar;
  39 import java.awt.MenuItem;
  40 import java.awt.ScrollPane;
  41 import java.awt.event.ActionEvent;
  42 import java.awt.event.ActionListener;
  43 import java.awt.event.ItemEvent;
  44 import java.awt.event.ItemListener;
  45
  46 public class TreePanel extends EmbmenuFrame implements ActionListener,
  47         ItemListener
  48 {
  49   SequenceI[] seq;
  50
  51   String type;
  52
  53   String pwtype;
  54
  55   int start;
  56
  57   int end;
  58
  59   TreeCanvas treeCanvas;
  60
  61   NJTree tree;
  62
  63   AlignmentPanel ap;
  64
  65   AlignViewport av;
  66
  67   public NJTree getTree()
  68   {
  69     return tree;
  70   }
  71
  72   @Override
  73   public void finalize() throws Throwable
  74   {
  75     ap = null;
  76     av = null;
  77     super.finalize();
  78   }
  79
  80   /**
  81    * Creates a new TreePanel object.
  82    *
  83    * @param av
  84    *          DOCUMENT ME!
  85    * @param seqVector
  86    *          DOCUMENT ME!
  87    * @param type
  88    *          DOCUMENT ME!
  89    * @param pwtype
  90    *          DOCUMENT ME!
  91    * @param s
  92    *          DOCUMENT ME!
  93    * @param e
  94    *          DOCUMENT ME!
  95    */
  96   public TreePanel(AlignmentPanel ap, String type, String pwtype)
  97   {
  98     try
  99     {
 100       jbInit();
 101       this.setMenuBar(jMenuBar1);
 102     } catch (Exception ex)
 103     {
 104       ex.printStackTrace();
 105     }
 106
 107     initTreePanel(ap, type, pwtype, null);
 108   }
 109
 110   /**
 111    * Creates a new TreePanel object.
 112    *
 113    * @param av
 114    *          DOCUMENT ME!
 115    * @param seqVector
 116    *          DOCUMENT ME!
 117    * @param newtree
 118    *          DOCUMENT ME!
 119    * @param type
 120    *          DOCUMENT ME!
 121    * @param pwtype
 122    *          DOCUMENT ME!
 123    */
 124   public TreePanel(AlignmentPanel ap, String type, String pwtype,
 125           NewickFile newtree)
 126   {
 127     try
 128     {
 129       jbInit();
 130       this.setMenuBar(jMenuBar1);
 131     } catch (Exception e)
 132     {
 133       e.printStackTrace();
 134     }
 135
 136     initTreePanel(ap, type, pwtype, newtree);
 137   }
 138
 139   void initTreePanel(AlignmentPanel ap, String type, String pwtype,
 140           NewickFile newTree)
 141   {
 142
 143     this.ap = ap;
 144     this.av = ap.av;
 145     this.type = type;
 146     this.pwtype = pwtype;
 147
 148     treeCanvas = new TreeCanvas(ap, scrollPane);
 149     TreeLoader tl = new TreeLoader(newTree);
 150     tl.start();
 151     embedMenuIfNeeded(treeCanvas);
 152     scrollPane.add(treeCanvas, BorderLayout.CENTER);
 153   }
 154
 155   void showOriginalData()
 156   {
 157     // decide if av alignment is sufficiently different to original data to
 158     // warrant a new window to be created
 159     // create new alignmnt window with hidden regions (unhiding hidden regions
 160     // yields unaligned seqs)
 161     // or create a selection box around columns in alignment view
 162     // test Alignment(SeqCigar[])
 163     if (tree.seqData != null)
 164     {
 165       char gc = '-';
 166       try
 167       {
 168         // we try to get the associated view's gap character
 169         // but this may fail if the view was closed...
 170         gc = av.getGapCharacter();
 171       } catch (Exception ex)
 172       {
 173       }
 174       ;
 175       Object[] alAndColsel = tree.seqData
 176               .getAlignmentAndHiddenColumns(gc);
 177
 178       if (alAndColsel != null && alAndColsel[0] != null)
 179       {
 180         Alignment al = new Alignment((SequenceI[]) alAndColsel[0]);
 181         AlignFrame af = new AlignFrame(al, av.applet,
 182                 "Original Data for Tree", false);
 183
 184         af.viewport.getAlignment().setHiddenColumns(
 185                 (HiddenColumns) alAndColsel[1]);
 186       }
 187     }
 188     else
 189     {
 190       System.out.println("Original Tree Data not available");
 191     }
 192   }
 193
 194   class TreeLoader extends Thread
 195   {
 196     NewickFile newtree;
 197
 198     jalview.datamodel.AlignmentView odata = null;
 199
 200     public TreeLoader(NewickFile newtree)
 201     {
 202       this.newtree = newtree;
 203     }
 204
 205     @Override
 206     public void run()
 207     {
 208       if (newtree != null)
 209       {
 210         if (odata == null)
 211         {
 212           tree = new NJTree(av.getAlignment().getSequencesArray(), newtree);
 213         }
 214         else
 215         {
 216           tree = new NJTree(av.getAlignment().getSequencesArray(), odata,
 217                   newtree);
 218         }
 219
 220       }
 221       else
 222       {
 223         int start, end;
 224         SequenceI[] seqs;
 225         boolean selview = av.getSelectionGroup() != null
 226                 && av.getSelectionGroup().getSize() > 1;
 227         AlignmentView seqStrings = av.getAlignmentView(selview);
 228         if (!selview)
 229         {
 230           start = 0;
 231           end = av.getAlignment().getWidth();
 232           seqs = av.getAlignment().getSequencesArray();
 233         }
 234         else
 235         {
 236           start = av.getSelectionGroup().getStartRes();
 237           end = av.getSelectionGroup().getEndRes() + 1;
 238           seqs = av.getSelectionGroup().getSequencesInOrder(
 239                   av.getAlignment());
 240         }
 241         ScoreModelI sm = ResidueProperties.getScoreModel(pwtype);
 242         if (sm instanceof ViewBasedAnalysisI)
 243         {
 244           try
 245           {
 246             sm = sm.getClass().newInstance();
 247             ((ViewBasedAnalysisI) sm)
 248                     .configureFromAlignmentView(treeCanvas.ap);
 249           } catch (Exception q)
 250           {
 251             System.err.println("Couldn't create a scoremodel instance for "
 252                     + sm.getName());
 253             q.printStackTrace();
 254           }
 255           tree = new NJTree(seqs, seqStrings, type, pwtype, sm, start, end);
 256         }
 257         else
 258         {
 259           tree = new NJTree(seqs, seqStrings, type, pwtype, null, start,
 260                   end);
 261         }
 262       }
 263
 264       tree.reCount(tree.getTopNode());
 265       tree.findHeight(tree.getTopNode());
 266       treeCanvas.setTree(tree);
 267       if (newtree != null)
 268       {
 269         // Set default view, paying lip service to any overriding tree view
 270         // parameter settings
 271         boolean showDist = newtree.HasDistances()
 272                 && av.applet.getDefaultParameter("showTreeDistances",
 273                         newtree.HasDistances());
 274         boolean showBoots = newtree.HasBootstrap()
 275                 && av.applet.getDefaultParameter("showTreeBootstraps",
 276                         newtree.HasBootstrap());
 277         distanceMenu.setState(showDist);
 278         bootstrapMenu.setState(showBoots);
 279         treeCanvas.setShowBootstrap(showBoots);
 280         treeCanvas.setShowDistances(showDist);
 281         treeCanvas.setMarkPlaceholders(av.applet.getDefaultParameter(
 282                 "showUnlinkedTreeNodes", false));
 283       }
 284
 285       treeCanvas.repaint();
 286
 287       av.setCurrentTree(tree);
 288
 289     }
 290   }
 291
 292   @Override
 293   public void actionPerformed(ActionEvent evt)
 294   {
 295     if (evt.getSource() == newickOutput)
 296     {
 297       newickOutput_actionPerformed();
 298     }
 299     else if (evt.getSource() == fontSize)
 300     {
 301       fontSize_actionPerformed();
 302     }
 303     else if (evt.getSource() == inputData)
 304     {
 305       showOriginalData();
 306     }
 307   }
 308
 309   @Override
 310   public void itemStateChanged(ItemEvent evt)
 311   {
 312     if (evt.getSource() == fitToWindow)
 313     {
 314       treeCanvas.fitToWindow = fitToWindow.getState();
 315     }
 316
 317     else if (evt.getSource() == distanceMenu)
 318     {
 319       treeCanvas.setShowDistances(distanceMenu.getState());
 320     }
 321
 322     else if (evt.getSource() == bootstrapMenu)
 323     {
 324       treeCanvas.setShowBootstrap(bootstrapMenu.getState());
 325     }
 326
 327     else if (evt.getSource() == placeholdersMenu)
 328     {
 329       treeCanvas.setMarkPlaceholders(placeholdersMenu.getState());
 330     }
 331
 332     treeCanvas.repaint();
 333   }
 334
 335   public void newickOutput_actionPerformed()
 336   {
 337     jalview.io.NewickFile fout = new jalview.io.NewickFile(
 338             tree.getTopNode());
 339     String output = fout.print(false, true);
 340     CutAndPasteTransfer cap = new CutAndPasteTransfer(false, null);
 341     cap.setText(output);
 342     java.awt.Frame frame = new java.awt.Frame();
 343     frame.add(cap);
 344     jalview.bin.JalviewLite.addFrame(frame, type + " " + pwtype, 500, 100);
 345   }
 346
 347   public java.awt.Font getTreeFont()
 348   {
 349     return treeCanvas.font;
 350   }
 351
 352   public void setTreeFont(java.awt.Font font)
 353   {
 354     treeCanvas.font = font;
 355     treeCanvas.repaint();
 356   }
 357
 358   protected void fontSize_actionPerformed()
 359   {
 360     if (treeCanvas == null)
 361     {
 362       return;
 363     }
 364
 365     new FontChooser(this);
 366   }
 367
 368   BorderLayout borderLayout1 = new BorderLayout();
 369
 370   protected ScrollPane scrollPane = new ScrollPane();
 371
 372   MenuBar jMenuBar1 = new MenuBar();
 373
 374   Menu jMenu2 = new Menu();
 375
 376   protected MenuItem fontSize = new MenuItem();
 377
 378   protected CheckboxMenuItem bootstrapMenu = new CheckboxMenuItem();
 379
 380   protected CheckboxMenuItem distanceMenu = new CheckboxMenuItem();
 381
 382   protected CheckboxMenuItem placeholdersMenu = new CheckboxMenuItem();
 383
 384   protected CheckboxMenuItem fitToWindow = new CheckboxMenuItem();
 385
 386   Menu fileMenu = new Menu();
 387
 388   MenuItem newickOutput = new MenuItem();
 389
 390   MenuItem inputData = new MenuItem();
 391
 392   private void jbInit() throws Exception
 393   {
 394     setLayout(borderLayout1);
 395     this.setBackground(Color.white);
 396     this.setFont(new java.awt.Font("Verdana", 0, 12));
 397     jMenu2.setLabel(MessageManager.getString("action.view"));
 398     fontSize.setLabel(MessageManager.getString("action.font"));
 399     fontSize.addActionListener(this);
 400     bootstrapMenu.setLabel(MessageManager
 401             .getString("label.show_bootstrap_values"));
 402     bootstrapMenu.addItemListener(this);
 403     distanceMenu.setLabel(MessageManager.getString("label.show_distances"));
 404     distanceMenu.addItemListener(this);
 405     placeholdersMenu.setLabel(MessageManager
 406             .getString("label.mark_unassociated_leaves"));
 407     placeholdersMenu.addItemListener(this);
 408     fitToWindow.setState(true);
 409     fitToWindow.setLabel(MessageManager.getString("label.fit_to_window"));
 410     fitToWindow.addItemListener(this);
 411     fileMenu.setLabel(MessageManager.getString("action.file"));
 412     newickOutput.setLabel(MessageManager.getString("label.newick_format"));
 413     newickOutput.addActionListener(this);
 414     inputData.setLabel(MessageManager.getString("label.input_data"));
 415
 416     add(scrollPane, BorderLayout.CENTER);
 417     jMenuBar1.add(fileMenu);
 418     jMenuBar1.add(jMenu2);
 419     jMenu2.add(fitToWindow);
 420     jMenu2.add(fontSize);
 421     jMenu2.add(distanceMenu);
 422     jMenu2.add(bootstrapMenu);
 423     jMenu2.add(placeholdersMenu);
 424     fileMenu.add(newickOutput);
 425     fileMenu.add(inputData);
 426     inputData.addActionListener(this);
 427   }
 428
 429 }