X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fjalview%2Fgui%2FTreePanel.java;h=2a8c9b44ae5a1d0d5d4601b29336bf838109d6e1;hb=1d6ae39a6442c455d362b933f1da89b0a481e5f5;hp=047c1d4cc749ca1bd2fafe27f12c3a4f0e599d30;hpb=9ec724cefe52ca1e3fa72d726557b0108a4b043a;p=jalview.git diff --git a/src/jalview/gui/TreePanel.java b/src/jalview/gui/TreePanel.java index 047c1d4..2a8c9b4 100755 --- a/src/jalview/gui/TreePanel.java +++ b/src/jalview/gui/TreePanel.java @@ -1,21 +1,40 @@ +/* + * Jalview - A Sequence Alignment Editor and Viewer + * Copyright (C) 2005 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.gui; -import jalview.datamodel.*; -import jalview.analysis.*; -import jalview.jbgui.GTreePanel; -import jalview.io.*; -import java.awt.event.*; -import java.util.*; import java.io.*; -import javax.swing.*; -import java.awt.*; -import java.awt.image.*; -import org.jibble.epsgraphics.*; +import java.util.*; import javax.imageio.*; +import java.awt.*; +import java.awt.event.*; +import java.awt.image.*; +import javax.swing.*; +import org.jibble.epsgraphics.*; +import jalview.analysis.*; +import jalview.datamodel.*; +import jalview.io.*; +import jalview.jbgui.*; -public class TreePanel extends GTreePanel +public class TreePanel + extends GTreePanel { SequenceI[] seq; String type; @@ -26,12 +45,8 @@ public class TreePanel extends GTreePanel TreeCanvas treeCanvas; NJTree tree; - public NJTree getTree() - { - return tree; - } - - public TreePanel(AlignViewport av, Vector seqVector, String type, String pwtype, int s, int e) + public TreePanel(AlignViewport av, Vector seqVector, String type, + String pwtype, int s, int e) { super(); @@ -42,109 +57,165 @@ public class TreePanel extends GTreePanel end = e; String longestName = ""; - seq = new Sequence [seqVector.size()]; - for (int i=0;i < seqVector.size();i++) + seq = new Sequence[seqVector.size()]; + + for (int i = 0; i < seqVector.size(); i++) { seq[i] = (Sequence) seqVector.elementAt(i); - if(seq[i].getName().length()>longestName.length()) + + if (seq[i].getName().length() > longestName.length()) + { longestName = seq[i].getName(); + } } tree = new NJTree(seq, type, pwtype, start, end); treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName); - + treeCanvas.setShowDistances(true); // We know this tree has distances. JBPNote TODO: prolly should add this as a userdefined default + setViewStateFromTreeCanvas(treeCanvas); tree.reCount(tree.getTopNode()); tree.findHeight(tree.getTopNode()); scrollPane.setViewportView(treeCanvas); - } + public TreePanel(AlignViewport av, Vector seqVector, NewickFile newtree, String type, String pwtype) + { + super(); + + // These are probably only arbitrary strings reflecting source of tree + this.type = type; + this.pwtype = pwtype; + + start = 0; + end = seqVector.size(); + + String longestName = ""; + seq = new Sequence[seqVector.size()]; + + for (int i = 0; i < seqVector.size(); i++) { - super(); - // These are probably only arbitrary strings reflecting source of tree - this.type = type; - this.pwtype = pwtype; + seq[i] = (Sequence) seqVector.elementAt(i); + } + + // This constructor matches sequence names to treenodes and sets up the tree layouts. + tree = new NJTree(seq, newtree); - start = 0; - end = seqVector.size(); + // Now have to calculate longest name based on the leaves + Vector leaves = tree.findLeaves(tree.getTopNode(), new Vector()); + boolean has_placeholders = false; - String longestName = ""; - seq = new Sequence [seqVector.size()]; - for (int i=0;i < seqVector.size();i++) + for (int i = 0; i < leaves.size(); i++) + { + SequenceNode lf = (SequenceNode) leaves.elementAt(i); + + if (lf.isPlaceholder()) { - seq[i] = (Sequence) seqVector.elementAt(i); + has_placeholders = true; } - // This constructor matches sequence names to treenodes and sets up the tree layouts. - tree = new NJTree(seq, newtree); - // Now have to calculate longest name based on the leaves - Vector leaves = tree.findLeaves(tree.getTopNode(),new Vector()); - - for (int i=0;i < leaves.size();i++) - if (longestName.length()<((Sequence) ((SequenceNode) leaves.elementAt(i)).element()).getName().length()) - longestName = TreeCanvas.PLACEHOLDER+((Sequence) ((SequenceNode) leaves.elementAt(i)).element()).getName(); - - tree.reCount(tree.getTopNode()); - tree.findHeight(tree.getTopNode()); - - treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName); - treeCanvas.setShowBootstrap(newtree.HasBootstrap()); - treeCanvas.setShowDistances(newtree.HasDistances()); - scrollPane.setViewportView(treeCanvas); - distanceMenu.setSelected(newtree.HasDistances()); - bootstrapMenu.setSelected(newtree.HasBootstrap()); - if (true) // JBPNote TODO: preference for always marking placeholders in new associated tree - placeholdersMenu.setSelected(treeCanvas.markPlaceholders); + if (longestName.length() < ( (Sequence) lf.element()).getName() + .length()) + { + longestName = TreeCanvas.PLACEHOLDER + + ( (Sequence) lf.element()).getName(); + } } + tree.reCount(tree.getTopNode()); + tree.findHeight(tree.getTopNode()); - public void textbox_actionPerformed(ActionEvent e) - { - CutAndPasteTransfer cap = new CutAndPasteTransfer(); - JInternalFrame frame = new JInternalFrame(); - frame.setContentPane(cap); + treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName); + treeCanvas.setShowBootstrap(newtree.HasBootstrap()); + treeCanvas.setShowDistances(newtree.HasDistances()); + scrollPane.setViewportView(treeCanvas); + treeCanvas.setMarkPlaceholders(has_placeholders); + setViewStateFromTreeCanvas(treeCanvas); - StringBuffer buffer = new StringBuffer(); + // JBPNote TODO: preference for always marking placeholders in new associated tree + } - if(type.equals("AV")) - buffer.append("Average distance tree using "); - else - buffer.append("Neighbour joining tree using "); + public NJTree getTree() + { + return tree; + } - if(pwtype.equals("BL")) - buffer.append("BLOSUM62"); - else - buffer.append("PID"); + private void setViewStateFromTreeCanvas(TreeCanvas tree) + { + // update view menu state from treeCanvas render state + distanceMenu.setSelected(tree.showDistances); + bootstrapMenu.setSelected(tree.showBootstrap); + placeholdersMenu.setSelected(tree.markPlaceholders); + } - Desktop.addInternalFrame(frame,buffer.toString(), 500, 100); - jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode()); - cap.setText( fout.print(false,true) ); + private void setTreeCanvasFromViewState(TreeCanvas tree) + { + // update view menu state from treeCanvas render state + tree.showDistances = distanceMenu.isSelected(); + tree.showBootstrap = bootstrapMenu.isSelected(); + tree.markPlaceholders = placeholdersMenu.isSelected(); + } - } + public void textbox_actionPerformed(ActionEvent e) + { + CutAndPasteTransfer cap = new CutAndPasteTransfer(); + StringBuffer buffer = new StringBuffer(); - public void saveAsNewick_actionPerformed(ActionEvent e) + if (type.equals("AV")) { + buffer.append("Average distance tree using "); + } + else + { + buffer.append("Neighbour joining tree using "); + } - JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache. - getProperty("LAST_DIRECTORY")); + if (pwtype.equals("BL")) + { + buffer.append("BLOSUM62"); + } + else + { + buffer.append("PID"); + } + + Desktop.addInternalFrame(cap, buffer.toString(), 500, 100); + + jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode()); + cap.setText(fout.print(false, true)); + } + + public void saveAsNewick_actionPerformed(ActionEvent e) + { + JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache. + getProperty( + "LAST_DIRECTORY")); chooser.setFileView(new JalviewFileView()); chooser.setDialogTitle("Save tree as newick file"); chooser.setToolTipText("Save"); + int value = chooser.showSaveDialog(null); - if (value == JalviewFileChooser.APPROVE_OPTION) { - String choice = chooser.getSelectedFile().getPath(); - jalview.bin.Cache.setProperty("LAST_DIRECTORY", chooser.getSelectedFile().getParent()); - try{ + + if (value == JalviewFileChooser.APPROVE_OPTION) + { + String choice = chooser.getSelectedFile().getPath(); + jalview.bin.Cache.setProperty("LAST_DIRECTORY", + chooser.getSelectedFile().getParent()); + + try + { jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode()); - String output = fout.print(false,true); // distances only - java.io.PrintWriter out = new java.io.PrintWriter( new java.io.FileWriter( choice ) ); + String output = fout.print(false, true); // distances only + java.io.PrintWriter out = new java.io.PrintWriter(new java.io. + FileWriter( + choice)); out.println(output); out.close(); } - catch (Exception ex) { + catch (Exception ex) + { ex.printStackTrace(); } } @@ -152,7 +223,6 @@ public class TreePanel extends GTreePanel protected void printMenu_actionPerformed(ActionEvent e) { - //Putting in a thread avoids Swing painting problems treeCanvas.startPrinting(); } @@ -163,29 +233,33 @@ public class TreePanel extends GTreePanel repaint(); } - protected void fontSize_actionPerformed(ActionEvent e) { - if( treeCanvas==null ) - return; - - String size = fontSize.getText().substring( fontSize.getText().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); + if (treeCanvas == null) + { + return; + } + + String size = fontSize.getText().substring(fontSize.getText().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); } protected void distanceMenu_actionPerformed(ActionEvent e) @@ -199,37 +273,46 @@ public class TreePanel extends GTreePanel } protected void placeholdersMenu_actionPerformed(ActionEvent e) - { - treeCanvas.setMarkPlaceholders(placeholdersMenu.isSelected()); + { + treeCanvas.setMarkPlaceholders(placeholdersMenu.isSelected()); } protected void epsTree_actionPerformed(ActionEvent e) { - int width = treeCanvas.getWidth(), height=treeCanvas.getHeight(); + int width = treeCanvas.getWidth(); + int height = treeCanvas.getHeight(); + try { - jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty( - "LAST_DIRECTORY"), - new String[]{"eps"}, - new String[]{"Encapsulated Postscript"}, - "Encapsulated Postscript"); + jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser( + jalview.bin.Cache.getProperty( + "LAST_DIRECTORY"), new String[] + {"eps"}, + new String[] + {"Encapsulated Postscript"}, + "Encapsulated Postscript"); chooser.setFileView(new jalview.io.JalviewFileView()); chooser.setDialogTitle("Create EPS file from tree"); chooser.setToolTipText("Save"); int value = chooser.showSaveDialog(this); + if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION) + { return; + } - jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getParent()); - FileOutputStream out = new FileOutputStream(chooser.getSelectedFile()); - EpsGraphics2D pg = new EpsGraphics2D("Tree", out, 0, 0, width, height); + jalview.bin.Cache.setProperty("LAST_DIRECTORY", + chooser.getSelectedFile().getParent()); + FileOutputStream out = new FileOutputStream(chooser.getSelectedFile()); + EpsGraphics2D pg = new EpsGraphics2D("Tree", out, 0, 0, width, + height); treeCanvas.draw(pg, width, height); - pg.flush(); - pg.close(); + pg.flush(); + pg.close(); } catch (Exception ex) { @@ -238,14 +321,18 @@ public class TreePanel extends GTreePanel } protected void pngTree_actionPerformed(ActionEvent e) - { - int width = treeCanvas.getWidth(), height = treeCanvas.getHeight(); + { + int width = treeCanvas.getWidth(); + int height = treeCanvas.getHeight(); + try { jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser( - jalview.bin.Cache.getProperty("LAST_DIRECTORY"), - new String[] {"png"}, - new String[] {"Portable network graphics"}, + jalview.bin.Cache.getProperty( + "LAST_DIRECTORY"), new String[] + {"png"}, + new String[] + {"Portable network graphics"}, "Portable network graphics"); chooser.setFileView(new jalview.io.JalviewFileView()); @@ -253,24 +340,29 @@ public class TreePanel extends GTreePanel chooser.setToolTipText("Save"); int value = chooser.showSaveDialog(this); + if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION) + { return; + } + + jalview.bin.Cache.setProperty("LAST_DIRECTORY", + chooser.getSelectedFile().getParent()); - jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getParent()); FileOutputStream out = new FileOutputStream(chooser.getSelectedFile()); - BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); + BufferedImage bi = new BufferedImage(width, height, + BufferedImage.TYPE_INT_RGB); Graphics png = bi.getGraphics(); treeCanvas.draw(png, width, height); - ImageIO.write(bi, "png", out); - out.close(); + ImageIO.write(bi, "png", out); + out.close(); } catch (Exception ex) { ex.printStackTrace(); } } - }