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
23 import javax.imageio.*;
\r
26 import java.awt.event.*;
\r
27 import java.awt.image.*;
\r
28 import javax.swing.*;
\r
30 import org.jibble.epsgraphics.*;
\r
31 import jalview.analysis.*;
\r
32 import jalview.datamodel.*;
\r
33 import jalview.io.*;
\r
34 import jalview.jbgui.*;
\r
36 public class TreePanel
\r
45 TreeCanvas treeCanvas;
\r
48 public TreePanel(AlignViewport av, Vector seqVector, String type,
\r
49 String pwtype, int s, int e)
\r
54 this.pwtype = pwtype;
\r
59 String longestName = "";
\r
60 seq = new Sequence[seqVector.size()];
\r
62 for (int i = 0; i < seqVector.size(); i++)
\r
64 seq[i] = (Sequence) seqVector.elementAt(i);
\r
66 if (seq[i].getName().length() > longestName.length())
\r
68 longestName = seq[i].getName();
\r
72 tree = new NJTree(seq, type, pwtype, start, end);
\r
74 treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName);
\r
75 treeCanvas.setShowDistances(true); // We know this tree has distances. JBPNote TODO: prolly should add this as a userdefined default
\r
76 setViewStateFromTreeCanvas(treeCanvas);
\r
77 tree.reCount(tree.getTopNode());
\r
78 tree.findHeight(tree.getTopNode());
\r
79 scrollPane.setViewportView(treeCanvas);
\r
82 public TreePanel(AlignViewport av, Vector seqVector, NewickFile newtree,
\r
83 String type, String pwtype)
\r
87 // These are probably only arbitrary strings reflecting source of tree
\r
89 this.pwtype = pwtype;
\r
92 end = seqVector.size();
\r
94 String longestName = "";
\r
95 seq = new Sequence[seqVector.size()];
\r
97 for (int i = 0; i < seqVector.size(); i++)
\r
99 seq[i] = (Sequence) seqVector.elementAt(i);
\r
102 // This constructor matches sequence names to treenodes and sets up the tree layouts.
\r
103 tree = new NJTree(seq, newtree);
\r
105 // Now have to calculate longest name based on the leaves
\r
106 Vector leaves = tree.findLeaves(tree.getTopNode(), new Vector());
\r
107 boolean has_placeholders = false;
\r
109 for (int i = 0; i < leaves.size(); i++)
\r
111 SequenceNode lf = (SequenceNode) leaves.elementAt(i);
\r
113 if (lf.isPlaceholder())
\r
115 has_placeholders = true;
\r
118 if (longestName.length() < ( (Sequence) lf.element()).getName()
\r
121 longestName = TreeCanvas.PLACEHOLDER +
\r
122 ( (Sequence) lf.element()).getName();
\r
126 tree.reCount(tree.getTopNode());
\r
127 tree.findHeight(tree.getTopNode());
\r
129 treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName);
\r
130 treeCanvas.setShowBootstrap(newtree.HasBootstrap());
\r
131 treeCanvas.setShowDistances(newtree.HasDistances());
\r
132 scrollPane.setViewportView(treeCanvas);
\r
133 treeCanvas.setMarkPlaceholders(has_placeholders);
\r
134 setViewStateFromTreeCanvas(treeCanvas);
\r
136 // JBPNote TODO: preference for always marking placeholders in new associated tree
\r
139 public NJTree getTree()
\r
144 private void setViewStateFromTreeCanvas(TreeCanvas tree)
\r
146 // update view menu state from treeCanvas render state
\r
147 distanceMenu.setSelected(tree.showDistances);
\r
148 bootstrapMenu.setSelected(tree.showBootstrap);
\r
149 placeholdersMenu.setSelected(tree.markPlaceholders);
\r
152 private void setTreeCanvasFromViewState(TreeCanvas tree)
\r
154 // update view menu state from treeCanvas render state
\r
155 tree.showDistances = distanceMenu.isSelected();
\r
156 tree.showBootstrap = bootstrapMenu.isSelected();
\r
157 tree.markPlaceholders = placeholdersMenu.isSelected();
\r
160 public void textbox_actionPerformed(ActionEvent e)
\r
162 CutAndPasteTransfer cap = new CutAndPasteTransfer();
\r
164 StringBuffer buffer = new StringBuffer();
\r
166 if (type.equals("AV"))
\r
168 buffer.append("Average distance tree using ");
\r
172 buffer.append("Neighbour joining tree using ");
\r
175 if (pwtype.equals("BL"))
\r
177 buffer.append("BLOSUM62");
\r
181 buffer.append("PID");
\r
184 Desktop.addInternalFrame(cap, buffer.toString(), 500, 100);
\r
186 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
\r
187 cap.setText(fout.print(false, true));
\r
190 public void saveAsNewick_actionPerformed(ActionEvent e)
\r
192 JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache.
\r
194 "LAST_DIRECTORY"));
\r
195 chooser.setFileView(new JalviewFileView());
\r
196 chooser.setDialogTitle("Save tree as newick file");
\r
197 chooser.setToolTipText("Save");
\r
199 int value = chooser.showSaveDialog(null);
\r
201 if (value == JalviewFileChooser.APPROVE_OPTION)
\r
203 String choice = chooser.getSelectedFile().getPath();
\r
204 jalview.bin.Cache.setProperty("LAST_DIRECTORY",
\r
205 chooser.getSelectedFile().getParent());
\r
209 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
\r
210 String output = fout.print(false, true); // distances only
\r
211 java.io.PrintWriter out = new java.io.PrintWriter(new java.io.
\r
214 out.println(output);
\r
217 catch (Exception ex)
\r
219 ex.printStackTrace();
\r
224 protected void printMenu_actionPerformed(ActionEvent e)
\r
226 //Putting in a thread avoids Swing painting problems
\r
227 treeCanvas.startPrinting();
\r
230 protected void fitToWindow_actionPerformed(ActionEvent e)
\r
232 treeCanvas.fitToWindow = fitToWindow.isSelected();
\r
236 protected void fontSize_actionPerformed(ActionEvent e)
\r
238 if (treeCanvas == null)
\r
243 String size = fontSize.getText().substring(fontSize.getText().indexOf("-") +
\r
246 Object selection = JOptionPane.showInternalInputDialog(Desktop.desktop,
\r
247 "Select font size", "Font size", JOptionPane.QUESTION_MESSAGE,
\r
251 "1", "2", "4", "6", "8", "10", "12", "14", "16", "18", "20"
\r
252 }, "Font Size - " + size);
\r
254 if (selection != null)
\r
256 fontSize.setText("Font Size - " + selection);
\r
258 int i = Integer.parseInt(selection.toString());
\r
259 treeCanvas.setFontSize(i);
\r
262 scrollPane.setViewportView(treeCanvas);
\r
265 protected void distanceMenu_actionPerformed(ActionEvent e)
\r
267 treeCanvas.setShowDistances(distanceMenu.isSelected());
\r
270 protected void bootstrapMenu_actionPerformed(ActionEvent e)
\r
272 treeCanvas.setShowBootstrap(bootstrapMenu.isSelected());
\r
275 protected void placeholdersMenu_actionPerformed(ActionEvent e)
\r
277 treeCanvas.setMarkPlaceholders(placeholdersMenu.isSelected());
\r
280 protected void epsTree_actionPerformed(ActionEvent e)
\r
282 int width = treeCanvas.getWidth();
\r
283 int height = treeCanvas.getHeight();
\r
287 jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(
\r
288 jalview.bin.Cache.getProperty(
\r
289 "LAST_DIRECTORY"), new String[]
\r
292 {"Encapsulated Postscript"},
\r
293 "Encapsulated Postscript");
\r
294 chooser.setFileView(new jalview.io.JalviewFileView());
\r
295 chooser.setDialogTitle("Create EPS file from tree");
\r
296 chooser.setToolTipText("Save");
\r
298 int value = chooser.showSaveDialog(this);
\r
300 if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)
\r
305 jalview.bin.Cache.setProperty("LAST_DIRECTORY",
\r
306 chooser.getSelectedFile().getParent());
\r
308 FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());
\r
309 EpsGraphics2D pg = new EpsGraphics2D("Tree", out, 0, 0, width,
\r
312 treeCanvas.draw(pg, width, height);
\r
317 catch (Exception ex)
\r
319 ex.printStackTrace();
\r
323 protected void pngTree_actionPerformed(ActionEvent e)
\r
325 int width = treeCanvas.getWidth();
\r
326 int height = treeCanvas.getHeight();
\r
330 jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(
\r
331 jalview.bin.Cache.getProperty(
\r
332 "LAST_DIRECTORY"), new String[]
\r
335 {"Portable network graphics"},
\r
336 "Portable network graphics");
\r
338 chooser.setFileView(new jalview.io.JalviewFileView());
\r
339 chooser.setDialogTitle("Create PNG image from tree");
\r
340 chooser.setToolTipText("Save");
\r
342 int value = chooser.showSaveDialog(this);
\r
344 if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)
\r
349 jalview.bin.Cache.setProperty("LAST_DIRECTORY",
\r
350 chooser.getSelectedFile().getParent());
\r
352 FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());
\r
354 BufferedImage bi = new BufferedImage(width, height,
\r
355 BufferedImage.TYPE_INT_RGB);
\r
356 Graphics png = bi.getGraphics();
\r
358 treeCanvas.draw(png, width, height);
\r
360 ImageIO.write(bi, "png", out);
\r
363 catch (Exception ex)
\r
365 ex.printStackTrace();
\r