-package jalview.gui;\r
-\r
-import jalview.datamodel.*;\r
-import jalview.analysis.*;\r
-import jalview.jbgui.GTreePanel;\r
-import jalview.io.*;\r
-import java.awt.event.*;\r
-import java.util.*;\r
-import java.io.*;\r
-import jalview.util.*;\r
-import javax.swing.*;\r
-import java.awt.print.*;\r
-import java.awt.*;\r
-import java.awt.image.*;\r
-import org.jibble.epsgraphics.*;\r
-import javax.imageio.*;\r
-\r
-\r
-\r
-\r
-public class TreePanel extends GTreePanel\r
-{\r
- SequenceI[] seq;\r
- String type;\r
- String pwtype;\r
- AlignViewport av;\r
- int start;\r
- int end;\r
- TreeCanvas treeCanvas;\r
- NJTree tree;\r
-\r
- public NJTree getTree()\r
- {\r
- return tree;\r
- }\r
-\r
- public TreePanel(AlignViewport av, Vector seqVector, String type, String pwtype, int s, int e)\r
- {\r
- super();\r
-\r
- this.type = type;\r
- this.pwtype = pwtype;\r
-\r
- start = s;\r
- end = e;\r
-\r
- String longestName = "";\r
- seq = new Sequence [seqVector.size()];\r
- for (int i=0;i < seqVector.size();i++)\r
- {\r
- seq[i] = (Sequence) seqVector.elementAt(i);\r
- if(seq[i].getName().length()>longestName.length())\r
- longestName = seq[i].getName();\r
- }\r
-\r
- tree = new NJTree(seq, type, pwtype, start, end);\r
-\r
- treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName);\r
-\r
- tree.reCount(tree.getTopNode());\r
- tree.findHeight(tree.getTopNode());\r
- scrollPane.setViewportView(treeCanvas);\r
-\r
- }\r
- public TreePanel(AlignViewport av, Vector seqVector, NewickFile newtree,\r
- String type, String pwtype)\r
- {\r
- super();\r
- // These are probably only arbitrary strings reflecting source of tree\r
- this.type = type;\r
- this.pwtype = pwtype;\r
-\r
- start = 0;\r
- end = seqVector.size();\r
-\r
- String longestName = "";\r
- seq = new Sequence [seqVector.size()];\r
- for (int i=0;i < seqVector.size();i++)\r
- {\r
- seq[i] = (Sequence) seqVector.elementAt(i);\r
- if(seq[i].getName().length()>longestName.length())\r
- longestName = seq[i].getName();\r
- }\r
- // This constructor matches sequence names to treenodes and sets up the tree layouts.\r
- tree = new NJTree(seq, newtree);\r
-\r
- tree.reCount(tree.getTopNode());\r
- tree.findHeight(tree.getTopNode());\r
-\r
- treeCanvas = new TreeCanvas(av, tree, scrollPane, longestName);\r
- treeCanvas.setShowBootstrap(newtree.HasBootstrap());\r
- treeCanvas.setShowDistances(newtree.HasDistances());\r
- scrollPane.setViewportView(treeCanvas);\r
-\r
- }\r
-\r
-\r
-\r
-public String getText(String format) {\r
- return null;\r
-}\r
-\r
-\r
- protected void saveMenu_actionPerformed(ActionEvent e)\r
- {\r
- JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache.\r
- getProperty("LAST_DIRECTORY"));\r
- chooser.setFileView(new JalviewFileView());\r
- chooser.setDialogTitle("Save tree as newick file");\r
- chooser.setToolTipText("Save");\r
- int value = chooser.showSaveDialog(null);\r
- if (value == JalviewFileChooser.APPROVE_OPTION) {\r
- String choice = chooser.getSelectedFile().getPath();\r
- jalview.bin.Cache.setProperty("LAST_DIRECTORY", choice);\r
- try{\r
- jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());\r
- String output = fout.print(false,true); // distances only\r
- java.io.PrintWriter out = new java.io.PrintWriter( new java.io.FileWriter( choice ) );\r
- out.println(output);\r
- out.close();\r
- }\r
- catch (Exception ex) {\r
- ex.printStackTrace();\r
- }\r
- }\r
- }\r
-\r
- protected void printMenu_actionPerformed(ActionEvent e)\r
- {\r
-\r
- //Putting in a thread avoids Swing painting problems\r
- treeCanvas.startPrinting();\r
- }\r
-\r
- protected void fitToWindow_actionPerformed(ActionEvent e)\r
- {\r
- treeCanvas.fitToWindow = fitToWindow.isSelected();\r
- repaint();\r
- }\r
-\r
-\r
- protected void fontSize_actionPerformed(ActionEvent e)\r
- {\r
- if( treeCanvas==null )\r
- return;\r
-\r
- String size = fontSize.getText().substring( fontSize.getText().indexOf("-")+1);\r
-\r
- Object selection = JOptionPane.showInternalInputDialog(Desktop.desktop,\r
- "Select font size",\r
- "Font size",\r
- JOptionPane.QUESTION_MESSAGE,\r
- null, new String[]{"1","2","4","6","8","10","12","14","16","18","20"}\r
- ,"Font Size - "+size);\r
- if(selection!=null)\r
- {\r
- fontSize.setText("Font Size - " + selection);\r
-\r
- int i = Integer.parseInt(selection.toString());\r
- treeCanvas.setFontSize(i);\r
- }\r
- scrollPane.setViewportView(treeCanvas);\r
-\r
- }\r
-\r
- protected void distanceMenu_actionPerformed(ActionEvent e)\r
- {\r
- treeCanvas.setShowDistances(distanceMenu.isSelected());\r
- }\r
-\r
- protected void bootstrapMenu_actionPerformed(ActionEvent e)\r
- {\r
- treeCanvas.setShowBootstrap(bootstrapMenu.isSelected());\r
- }\r
-\r
-\r
- protected void epsTree_actionPerformed(ActionEvent e)\r
- {\r
- int width = treeCanvas.getWidth(), height=treeCanvas.getHeight();\r
- try\r
- {\r
- jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(\r
- "LAST_DIRECTORY"), new String[]{"eps"}, "Encapsulated Postscript");\r
- chooser.setFileView(new jalview.io.JalviewFileView());\r
- chooser.setDialogTitle("Create EPS file from tree");\r
- chooser.setToolTipText("Save");\r
-\r
- int value = chooser.showSaveDialog(this);\r
- if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
- return;\r
-\r
- jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getPath());\r
- FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
- EpsGraphics2D pg = new EpsGraphics2D("Tree", out, 0, 0, width, height);\r
-\r
-\r
- treeCanvas.draw(pg, width, height);\r
-\r
- pg.flush();\r
- pg.close();\r
- }\r
- catch (Exception ex)\r
- {\r
- ex.printStackTrace();\r
- }\r
- }\r
-\r
- protected void pngTree_actionPerformed(ActionEvent e)\r
- {\r
- int width = treeCanvas.getWidth(), height = treeCanvas.getHeight();\r
- try\r
- {\r
- jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(\r
- "LAST_DIRECTORY"), new String[]{"png"}, "Portable network graphics");\r
- chooser.setFileView(new jalview.io.JalviewFileView());\r
- chooser.setDialogTitle("Create PNG image from tree");\r
- chooser.setToolTipText("Save");\r
-\r
- int value = chooser.showSaveDialog(this);\r
- if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
- return;\r
-\r
- jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getPath());\r
- FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
-\r
- BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);\r
- Graphics png = bi.getGraphics();\r
-\r
- treeCanvas.draw(png, width, height);\r
-\r
- ImageIO.write(bi, "png", out);\r
- out.close();\r
- }\r
- catch (Exception ex)\r
- {\r
- ex.printStackTrace();\r
- }\r
- }\r
-\r
-}\r
+/*
+ * 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.gui;
+
+import jalview.analysis.*;
+
+import jalview.datamodel.*;
+
+import jalview.io.*;
+
+import jalview.jbgui.*;
+
+import org.jibble.epsgraphics.*;
+
+import java.awt.*;
+import java.awt.event.*;
+import java.awt.image.*;
+
+import java.io.*;
+
+import java.util.*;
+
+import javax.imageio.*;
+
+import java.beans.PropertyChangeEvent;
+
+
+/**
+ * DOCUMENT ME!
+ *
+ * @author $author$
+ * @version $Revision$
+ */
+public class TreePanel extends GTreePanel
+{
+ String type;
+ String pwtype;
+ TreeCanvas treeCanvas;
+ NJTree tree;
+ AlignViewport av;
+
+ /**
+ * 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)
+ {
+ super();
+ initTreePanel(av, type, pwtype, null);
+
+ // We know this tree has distances. JBPNote TODO: prolly should add this as a userdefined default
+ // showDistances(true);
+ }
+
+ /**
+ * 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();
+ initTreePanel(av, type, pwtype, newtree);
+ }
+
+ public AlignmentI getAlignment()
+ {
+ return treeCanvas.av.getAlignment();
+ }
+
+
+ void initTreePanel(AlignViewport av, String type, String pwtype,
+ NewickFile newTree)
+ {
+
+ this.type = type;
+ this.pwtype = pwtype;
+
+ treeCanvas = new TreeCanvas(av, scrollPane);
+ scrollPane.setViewportView(treeCanvas);
+
+ av.addPropertyChangeListener(new java.beans.PropertyChangeListener()
+ {
+ public void propertyChange(PropertyChangeEvent evt)
+ {
+ if (evt.getPropertyName().equals("alignment"))
+ {
+ if(tree==null)
+ System.out.println("tree is null");
+ if(evt.getNewValue()==null)
+ System.out.println("new value is null");
+
+ tree.UpdatePlaceHolders( (Vector) evt.getNewValue());
+
+ repaint();
+ }
+ }
+ });
+
+ this.av = av;
+
+
+ TreeLoader tl = new TreeLoader(newTree);
+ tl.start();
+
+ }
+
+ class TreeLoader extends Thread
+ {
+ NewickFile newtree;
+ jalview.datamodel.AlignmentView odata=null;
+ public TreeLoader(NewickFile newtree)
+ {
+ this.newtree = newtree;
+ if (newtree != null)
+ {
+ // Must be outside run(), as Jalview2XML tries to
+ // update distance/bootstrap visibility at the same time
+ showBootstrap(newtree.HasBootstrap());
+ showDistances(newtree.HasDistances());
+ }
+ }
+
+ 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);
+ }
+ if (!tree.hasOriginalSequenceData())
+ allowOriginalSeqData(false);
+ }
+ 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);
+ showDistances(true);
+ }
+
+
+ tree.reCount(tree.getTopNode());
+ tree.findHeight(tree.getTopNode());
+ treeCanvas.setTree(tree);
+ treeCanvas.repaint();
+ av.setCurrentTree(tree);
+
+ }
+ }
+
+ public void showDistances(boolean b)
+ {
+ treeCanvas.setShowDistances(b);
+ distanceMenu.setSelected(b);
+ }
+
+ public void showBootstrap(boolean b)
+ {
+ treeCanvas.setShowBootstrap(b);
+ bootstrapMenu.setSelected(b);
+ }
+
+ public void showPlaceholders(boolean b)
+ {
+ placeholdersMenu.setState(b);
+ treeCanvas.setMarkPlaceholders(b);
+ }
+
+ private void allowOriginalSeqData(boolean b) {
+ originalSeqData.setVisible(b);
+ }
+
+
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @return DOCUMENT ME!
+ */
+ public NJTree getTree()
+ {
+ return tree;
+ }
+
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void textbox_actionPerformed(ActionEvent e)
+ {
+ CutAndPasteTransfer cap = new CutAndPasteTransfer();
+
+ StringBuffer buffer = new StringBuffer();
+
+ if (type.equals("AV"))
+ {
+ buffer.append("Average distance tree using ");
+ }
+ else
+ {
+ buffer.append("Neighbour joining tree using ");
+ }
+
+ 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(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance()));
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ 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
+ {
+ jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());
+ String output = fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance());
+ java.io.PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter(
+ choice));
+ out.println(output);
+ out.close();
+ }
+ catch (Exception ex)
+ {
+ ex.printStackTrace();
+ }
+ }
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void printMenu_actionPerformed(ActionEvent e)
+ {
+ //Putting in a thread avoids Swing painting problems
+ treeCanvas.startPrinting();
+ }
+
+
+ public void originalSeqData_actionPerformed(ActionEvent e)
+ {
+ if (!tree.hasOriginalSequenceData())
+ {
+ jalview.bin.Cache.log.info("Unexpected call to originalSeqData_actionPerformed - should have hidden this menu action.");
+ return;
+ }
+ // 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[])
+ Object[] alAndColsel = tree.seqData.getAlignmentAndColumnSelection(av.
+ getGapCharacter());
+
+
+ if (alAndColsel != null && alAndColsel[0]!=null)
+ {
+ // AlignmentOrder origorder = new AlignmentOrder(alAndColsel[0]);
+
+ Alignment al = new Alignment((SequenceI[]) alAndColsel[0]);
+ Alignment dataset = av.getAlignment().getDataset();
+ if (dataset != null)
+ {
+ al.setDataset(dataset);
+ }
+
+ if (true)
+ {
+ // make a new frame!
+ AlignFrame af = new AlignFrame(al, (ColumnSelection) alAndColsel[1]);
+
+ //>>>This is a fix for the moment, until a better solution is found!!<<<
+ // af.getFeatureRenderer().transferSettings(alignFrame.getFeatureRenderer());
+
+ // af.addSortByOrderMenuItem(ServiceName + " Ordering",
+ // msaorder);
+
+ Desktop.addInternalFrame(af, "Original Data for " + this.title,
+ AlignFrame.NEW_WINDOW_WIDTH,
+ AlignFrame.NEW_WINDOW_HEIGHT);
+ }
+ }
+ }
+
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void fitToWindow_actionPerformed(ActionEvent e)
+ {
+ treeCanvas.fitToWindow = fitToWindow.isSelected();
+ repaint();
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void font_actionPerformed(ActionEvent e)
+ {
+ if (treeCanvas == null)
+ {
+ return;
+ }
+
+ new FontChooser(this);
+ }
+
+ public Font getTreeFont()
+ {
+ return treeCanvas.font;
+ }
+
+ public void setTreeFont(Font font)
+ {
+ if(treeCanvas!=null)
+ treeCanvas.setFont(font);
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void distanceMenu_actionPerformed(ActionEvent e)
+ {
+ treeCanvas.setShowDistances(distanceMenu.isSelected());
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void bootstrapMenu_actionPerformed(ActionEvent e)
+ {
+ treeCanvas.setShowBootstrap(bootstrapMenu.isSelected());
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void placeholdersMenu_actionPerformed(ActionEvent e)
+ {
+ treeCanvas.setMarkPlaceholders(placeholdersMenu.isSelected());
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void epsTree_actionPerformed(ActionEvent e)
+ {
+ boolean accurateText = true;
+
+ String renderStyle = jalview.bin.Cache.getDefault("EPS_RENDERING",
+ "Prompt each time");
+
+ // If we need to prompt, and if the GUI is visible then
+ // Prompt for EPS rendering style
+ if (renderStyle.equalsIgnoreCase("Prompt each time")
+ && !
+ (System.getProperty("java.awt.headless") != null
+ && System.getProperty("java.awt.headless").equals("true")))
+ {
+ EPSOptions eps = new EPSOptions();
+ renderStyle = eps.getValue();
+
+ if (renderStyle==null || eps.cancelled)
+ return;
+
+
+ }
+
+ if (renderStyle.equalsIgnoreCase("text"))
+ {
+ accurateText = false;
+ }
+
+ 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");
+ 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);
+
+ pg.setAccurateTextMode(accurateText);
+
+ treeCanvas.draw(pg, width, height);
+
+ pg.flush();
+ pg.close();
+ }
+ catch (Exception ex)
+ {
+ ex.printStackTrace();
+ }
+ }
+
+ /**
+ * DOCUMENT ME!
+ *
+ * @param e DOCUMENT ME!
+ */
+ public void pngTree_actionPerformed(ActionEvent e)
+ {
+ 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" },
+ "Portable network graphics");
+
+ chooser.setFileView(new jalview.io.JalviewFileView());
+ chooser.setDialogTitle("Create PNG image 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());
+
+ 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();
+ }
+ catch (Exception ex)
+ {
+ ex.printStackTrace();
+ }
+ }
+}
git://source.jalview.org / jalview.git / blobdiff