2 * Jalview - A Sequence Alignment Editor and Viewer
\r
3 * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle
\r
5 * This program is free software; you can redistribute it and/or
\r
6 * modify it under the terms of the GNU General Public License
\r
7 * as published by the Free Software Foundation; either version 2
\r
8 * of the License, or (at your option) any later version.
\r
10 * This program is distributed in the hope that it will be useful,
\r
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
\r
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
\r
13 * GNU General Public License for more details.
\r
15 * You should have received a copy of the GNU General Public License
\r
16 * along with this program; if not, write to the Free Software
\r
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
\r
19 package jalview.gui;
\r
21 import jalview.analysis.*;
\r
23 import jalview.datamodel.*;
\r
25 import jalview.io.*;
\r
27 import jalview.jbgui.*;
\r
29 import org.jibble.epsgraphics.*;
\r
32 import java.awt.event.*;
\r
33 import java.awt.image.*;
\r
39 import javax.imageio.*;
\r
41 import java.beans.PropertyChangeEvent;
\r
48 * @version $Revision$
\r
50 public class TreePanel extends GTreePanel
\r
54 TreeCanvas treeCanvas;
\r
59 * Creates a new TreePanel object.
\r
61 * @param av DOCUMENT ME!
\r
62 * @param seqVector DOCUMENT ME!
\r
63 * @param type DOCUMENT ME!
\r
64 * @param pwtype DOCUMENT ME!
\r
65 * @param s DOCUMENT ME!
\r
66 * @param e DOCUMENT ME!
\r
68 public TreePanel(AlignViewport av, String type, String pwtype)
\r
71 initTreePanel(av, type, pwtype, null);
\r
73 // We know this tree has distances. JBPNote TODO: prolly should add this as a userdefined default
\r
74 // showDistances(true);
\r
78 * Creates a new TreePanel object.
\r
80 * @param av DOCUMENT ME!
\r
81 * @param seqVector DOCUMENT ME!
\r
82 * @param newtree DOCUMENT ME!
\r
83 * @param type DOCUMENT ME!
\r
84 * @param pwtype DOCUMENT ME!
\r
86 public TreePanel(AlignViewport av,
\r
92 initTreePanel(av, type, pwtype, newtree);
\r
95 public AlignmentI getAlignment()
\r
97 return treeCanvas.av.getAlignment();
\r
101 void initTreePanel(AlignViewport av, String type, String pwtype,
\r
102 NewickFile newTree)
\r
106 this.pwtype = pwtype;
\r
108 treeCanvas = new TreeCanvas(av, scrollPane);
\r
109 scrollPane.setViewportView(treeCanvas);
\r
111 av.addPropertyChangeListener(new java.beans.PropertyChangeListener()
\r
113 public void propertyChange(PropertyChangeEvent evt)
\r
115 if (evt.getPropertyName().equals("alignment"))
\r
118 System.out.println("tree is null");
\r
119 if(evt.getNewValue()==null)
\r
120 System.out.println("new value is null");
\r
122 tree.UpdatePlaceHolders( (Vector) evt.getNewValue());
\r
132 TreeLoader tl = new TreeLoader(newTree);
\r
137 class TreeLoader extends Thread
\r
139 NewickFile newtree;
\r
140 jalview.datamodel.CigarArray odata=null;
\r
141 public TreeLoader(NewickFile newtree)
\r
143 this.newtree = newtree;
\r
144 if (newtree != null)
\r
146 // Must be outside run(), as Jalview2XML tries to
\r
147 // update distance/bootstrap visibility at the same time
\r
148 showBootstrap(newtree.HasBootstrap());
\r
149 showDistances(newtree.HasDistances());
\r
156 if(newtree!=null) {
\r
159 tree = new NJTree(av.alignment.getSequencesArray(),
\r
162 tree = new NJTree(av.alignment.getSequencesArray(), odata, newtree);
\r
164 if (!tree.hasOriginalSequenceData())
\r
165 allowOriginalSeqData(false);
\r
171 CigarArray seqStrings = av.getViewAsCigars(av.getSelectionGroup()!=null);
\r
172 if(av.getSelectionGroup()==null)
\r
175 end = av.alignment.getWidth();
\r
176 seqs = av.alignment.getSequencesArray();
\r
180 start = av.getSelectionGroup().getStartRes();
\r
181 end = av.getSelectionGroup().getEndRes()+1;
\r
182 seqs = av.getSelectionGroup().getSequencesInOrder(av.alignment);
\r
185 tree = new NJTree(seqs, seqStrings, type, pwtype, start, end);
\r
186 showDistances(true);
\r
190 tree.reCount(tree.getTopNode());
\r
191 tree.findHeight(tree.getTopNode());
\r
192 treeCanvas.setTree(tree);
\r
193 treeCanvas.repaint();
\r
194 av.setCurrentTree(tree);
\r
199 public void showDistances(boolean b)
\r
201 treeCanvas.setShowDistances(b);
\r
202 distanceMenu.setSelected(b);
\r
205 public void showBootstrap(boolean b)
\r
207 treeCanvas.setShowBootstrap(b);
\r
208 bootstrapMenu.setSelected(b);
\r
211 public void showPlaceholders(boolean b)
\r
213 placeholdersMenu.setState(b);
\r
214 treeCanvas.setMarkPlaceholders(b);
\r
216 private void allowOriginalSeqData(boolean b) {
\r
217 originalSeqData.setVisible(b);
\r
225 * @return DOCUMENT ME!
\r
227 public NJTree getTree()
\r
236 * @param e DOCUMENT ME!
\r
238 public void textbox_actionPerformed(ActionEvent e)
\r
240 CutAndPasteTransfer cap = new CutAndPasteTransfer();
\r
242 StringBuffer buffer = new StringBuffer();
\r
244 if (type.equals("AV"))
\r
246 buffer.append("Average distance tree using ");
\r
250 buffer.append("Neighbour joining tree using ");
\r
253 if (pwtype.equals("BL"))
\r
255 buffer.append("BLOSUM62");
\r
259 buffer.append("PID");
\r
262 Desktop.addInternalFrame(cap, buffer.toString(), 500, 100);
\r
264 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
\r
265 cap.setText(fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance()));
\r
271 * @param e DOCUMENT ME!
\r
273 public void saveAsNewick_actionPerformed(ActionEvent e)
\r
275 JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache.getProperty(
\r
276 "LAST_DIRECTORY"));
\r
277 chooser.setFileView(new JalviewFileView());
\r
278 chooser.setDialogTitle("Save tree as newick file");
\r
279 chooser.setToolTipText("Save");
\r
281 int value = chooser.showSaveDialog(null);
\r
283 if (value == JalviewFileChooser.APPROVE_OPTION)
\r
285 String choice = chooser.getSelectedFile().getPath();
\r
286 jalview.bin.Cache.setProperty("LAST_DIRECTORY",
\r
287 chooser.getSelectedFile().getParent());
\r
291 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
\r
292 String output = fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance());
\r
293 java.io.PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter(
\r
295 out.println(output);
\r
298 catch (Exception ex)
\r
300 ex.printStackTrace();
\r
308 * @param e DOCUMENT ME!
\r
310 public void printMenu_actionPerformed(ActionEvent e)
\r
312 //Putting in a thread avoids Swing painting problems
\r
313 treeCanvas.startPrinting();
\r
317 public void originalSeqData_actionPerformed(ActionEvent e)
\r
319 String originalData = tree.printOriginalSequenceData(av.getGapCharacter());
\r
320 if (originalData!=null) {
\r
321 CutAndPasteTransfer cap = new CutAndPasteTransfer();
\r
322 cap.setText(originalData);
\r
323 Desktop.addInternalFrame(cap, "Original Data",
\r
326 System.err.println("IMPLEMENTATION BUG! originalSeqData is not available.");
\r
335 * @param e DOCUMENT ME!
\r
337 public void fitToWindow_actionPerformed(ActionEvent e)
\r
339 treeCanvas.fitToWindow = fitToWindow.isSelected();
\r
346 * @param e DOCUMENT ME!
\r
348 public void font_actionPerformed(ActionEvent e)
\r
350 if (treeCanvas == null)
\r
355 new FontChooser(this);
\r
358 public Font getTreeFont()
\r
360 return treeCanvas.font;
\r
363 public void setTreeFont(Font font)
\r
365 if(treeCanvas!=null)
\r
366 treeCanvas.setFont(font);
\r
372 * @param e DOCUMENT ME!
\r
374 public void distanceMenu_actionPerformed(ActionEvent e)
\r
376 treeCanvas.setShowDistances(distanceMenu.isSelected());
\r
382 * @param e DOCUMENT ME!
\r
384 public void bootstrapMenu_actionPerformed(ActionEvent e)
\r
386 treeCanvas.setShowBootstrap(bootstrapMenu.isSelected());
\r
392 * @param e DOCUMENT ME!
\r
394 public void placeholdersMenu_actionPerformed(ActionEvent e)
\r
396 treeCanvas.setMarkPlaceholders(placeholdersMenu.isSelected());
\r
402 * @param e DOCUMENT ME!
\r
404 public void epsTree_actionPerformed(ActionEvent e)
\r
406 boolean accurateText = true;
\r
408 String renderStyle = jalview.bin.Cache.getDefault("EPS_RENDERING",
\r
409 "Prompt each time");
\r
411 // If we need to prompt, and if the GUI is visible then
\r
412 // Prompt for EPS rendering style
\r
413 if (renderStyle.equalsIgnoreCase("Prompt each time")
\r
415 (System.getProperty("java.awt.headless") != null
\r
416 && System.getProperty("java.awt.headless").equals("true")))
\r
418 EPSOptions eps = new EPSOptions();
\r
419 renderStyle = eps.getValue();
\r
421 if (renderStyle==null || eps.cancelled)
\r
427 if (renderStyle.equalsIgnoreCase("text"))
\r
429 accurateText = false;
\r
432 int width = treeCanvas.getWidth();
\r
433 int height = treeCanvas.getHeight();
\r
437 jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(
\r
438 "LAST_DIRECTORY"), new String[] { "eps" },
\r
439 new String[] { "Encapsulated Postscript" },
\r
440 "Encapsulated Postscript");
\r
441 chooser.setFileView(new jalview.io.JalviewFileView());
\r
442 chooser.setDialogTitle("Create EPS file from tree");
\r
443 chooser.setToolTipText("Save");
\r
445 int value = chooser.showSaveDialog(this);
\r
447 if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)
\r
452 jalview.bin.Cache.setProperty("LAST_DIRECTORY",
\r
453 chooser.getSelectedFile().getParent());
\r
455 FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());
\r
456 EpsGraphics2D pg = new EpsGraphics2D("Tree", out, 0, 0, width,
\r
459 pg.setAccurateTextMode(accurateText);
\r
461 treeCanvas.draw(pg, width, height);
\r
466 catch (Exception ex)
\r
468 ex.printStackTrace();
\r
475 * @param e DOCUMENT ME!
\r
477 public void pngTree_actionPerformed(ActionEvent e)
\r
479 int width = treeCanvas.getWidth();
\r
480 int height = treeCanvas.getHeight();
\r
484 jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(
\r
485 "LAST_DIRECTORY"), new String[] { "png" },
\r
486 new String[] { "Portable network graphics" },
\r
487 "Portable network graphics");
\r
489 chooser.setFileView(new jalview.io.JalviewFileView());
\r
490 chooser.setDialogTitle("Create PNG image from tree");
\r
491 chooser.setToolTipText("Save");
\r
493 int value = chooser.showSaveDialog(this);
\r
495 if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)
\r
500 jalview.bin.Cache.setProperty("LAST_DIRECTORY",
\r
501 chooser.getSelectedFile().getParent());
\r
503 FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());
\r
505 BufferedImage bi = new BufferedImage(width, height,
\r
506 BufferedImage.TYPE_INT_RGB);
\r
507 Graphics png = bi.getGraphics();
\r
509 treeCanvas.draw(png, width, height);
\r
511 ImageIO.write(bi, "png", out);
\r
514 catch (Exception ex)
\r
516 ex.printStackTrace();
\r