2 * Jalview - A Sequence Alignment Editor and Viewer (Version 2.9.0b1)
3 * Copyright (C) 2015 The Jalview Authors
5 * This file is part of Jalview.
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.
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.
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.
23 import jalview.analysis.Conservation;
24 import jalview.analysis.NJTree;
25 import jalview.api.AlignViewportI;
26 import jalview.datamodel.Sequence;
27 import jalview.datamodel.SequenceGroup;
28 import jalview.datamodel.SequenceI;
29 import jalview.datamodel.SequenceNode;
30 import jalview.schemes.ColourSchemeI;
31 import jalview.schemes.ColourSchemeProperty;
32 import jalview.schemes.ResidueProperties;
33 import jalview.schemes.UserColourScheme;
34 import jalview.structure.SelectionSource;
35 import jalview.util.Format;
36 import jalview.util.MappingUtils;
37 import jalview.util.MessageManager;
39 import java.awt.Color;
40 import java.awt.Dimension;
42 import java.awt.FontMetrics;
43 import java.awt.Graphics;
44 import java.awt.Graphics2D;
45 import java.awt.Point;
46 import java.awt.Rectangle;
47 import java.awt.RenderingHints;
48 import java.awt.event.MouseEvent;
49 import java.awt.event.MouseListener;
50 import java.awt.event.MouseMotionListener;
51 import java.awt.print.PageFormat;
52 import java.awt.print.Printable;
53 import java.awt.print.PrinterException;
54 import java.awt.print.PrinterJob;
55 import java.util.Enumeration;
56 import java.util.Hashtable;
57 import java.util.Vector;
59 import javax.swing.JColorChooser;
60 import javax.swing.JPanel;
61 import javax.swing.JScrollPane;
62 import javax.swing.SwingUtilities;
63 import javax.swing.ToolTipManager;
71 public class TreeCanvas extends JPanel implements MouseListener, Runnable,
72 Printable, MouseMotionListener, SelectionSource
75 public static final String PLACEHOLDER = " * ";
79 JScrollPane scrollPane;
91 boolean fitToWindow = true;
93 boolean showDistances = false;
95 boolean showBootstrap = false;
97 boolean markPlaceholders = false;
107 int labelLength = -1;
109 Hashtable nameHash = new Hashtable();
111 Hashtable nodeHash = new Hashtable();
113 SequenceNode highlightNode;
115 boolean applyToAllViews = false;
118 * Creates a new TreeCanvas object.
129 public TreeCanvas(TreePanel tp, AlignmentPanel ap, JScrollPane scroller)
135 scrollPane = scroller;
136 addMouseListener(this);
137 addMouseMotionListener(this);
138 ToolTipManager.sharedInstance().registerComponent(this);
147 public void treeSelectionChanged(SequenceI sequence)
149 AlignmentPanel[] aps = getAssociatedPanels();
151 for (int a = 0; a < aps.length; a++)
153 SequenceGroup selected = aps[a].av.getSelectionGroup();
155 if (selected == null)
157 selected = new SequenceGroup();
158 aps[a].av.setSelectionGroup(selected);
161 selected.setEndRes(aps[a].av.getAlignment().getWidth() - 1);
162 selected.addOrRemove(sequence, true);
172 public void setTree(NJTree tree)
175 tree.findHeight(tree.getTopNode());
177 // Now have to calculate longest name based on the leaves
178 Vector leaves = tree.findLeaves(tree.getTopNode(), new Vector());
179 boolean has_placeholders = false;
182 for (int i = 0; i < leaves.size(); i++)
184 SequenceNode lf = (SequenceNode) leaves.elementAt(i);
186 if (lf.isPlaceholder())
188 has_placeholders = true;
191 if (longestName.length() < ((Sequence) lf.element()).getName()
194 longestName = TreeCanvas.PLACEHOLDER
195 + ((Sequence) lf.element()).getName();
199 setMarkPlaceholders(has_placeholders);
220 public void drawNode(Graphics g, SequenceNode node, float chunk,
221 float scale, int width, int offx, int offy)
228 if ((node.left() == null) && (node.right() == null))
231 float height = node.height;
232 float dist = node.dist;
234 int xstart = (int) ((height - dist) * scale) + offx;
235 int xend = (int) (height * scale) + offx;
237 int ypos = (int) (node.ycount * chunk) + offy;
239 if (node.element() instanceof SequenceI)
241 SequenceI seq = (SequenceI) node.element();
243 if (av.getSequenceColour(seq) == Color.white)
245 g.setColor(Color.black);
249 g.setColor(av.getSequenceColour(seq).darker());
254 g.setColor(Color.black);
257 // Draw horizontal line
258 g.drawLine(xstart, ypos, xend, ypos);
260 String nodeLabel = "";
262 if (showDistances && (node.dist > 0))
264 nodeLabel = new Format("%-.2f").form(node.dist);
267 if (showBootstrap && node.bootstrap > -1)
271 nodeLabel = nodeLabel + " : ";
274 nodeLabel = nodeLabel + String.valueOf(node.bootstrap);
277 if (!nodeLabel.equals(""))
279 g.drawString(nodeLabel, xstart + 2, ypos - 2);
282 String name = (markPlaceholders && node.isPlaceholder()) ? (PLACEHOLDER + node
283 .getName()) : node.getName();
285 int charWidth = fm.stringWidth(name) + 3;
286 int charHeight = font.getSize();
288 Rectangle rect = new Rectangle(xend + 10, ypos - charHeight / 2,
289 charWidth, charHeight);
291 nameHash.put(node.element(), rect);
293 // Colour selected leaves differently
294 SequenceGroup selected = av.getSelectionGroup();
296 if ((selected != null)
297 && selected.getSequences(null).contains(node.element()))
299 g.setColor(Color.gray);
301 g.fillRect(xend + 10, ypos - charHeight / 2, charWidth, charHeight);
302 g.setColor(Color.white);
305 g.drawString(name, xend + 10, ypos + fm.getDescent());
306 g.setColor(Color.black);
310 drawNode(g, (SequenceNode) node.left(), chunk, scale, width, offx,
312 drawNode(g, (SequenceNode) node.right(), chunk, scale, width, offx,
315 float height = node.height;
316 float dist = node.dist;
318 int xstart = (int) ((height - dist) * scale) + offx;
319 int xend = (int) (height * scale) + offx;
320 int ypos = (int) (node.ycount * chunk) + offy;
322 g.setColor(node.color.darker());
324 // Draw horizontal line
325 g.drawLine(xstart, ypos, xend, ypos);
326 if (node == highlightNode)
328 g.fillRect(xend - 3, ypos - 3, 6, 6);
332 g.fillRect(xend - 2, ypos - 2, 4, 4);
335 int ystart = (int) (((SequenceNode) node.left()).ycount * chunk)
337 int yend = (int) (((SequenceNode) node.right()).ycount * chunk)
340 Rectangle pos = new Rectangle(xend - 2, ypos - 2, 5, 5);
341 nodeHash.put(node, pos);
343 g.drawLine((int) (height * scale) + offx, ystart,
344 (int) (height * scale) + offx, yend);
346 String nodeLabel = "";
348 if (showDistances && (node.dist > 0))
350 nodeLabel = new Format("%-.2f").form(node.dist);
353 if (showBootstrap && node.bootstrap > -1)
357 nodeLabel = nodeLabel + " : ";
360 nodeLabel = nodeLabel + String.valueOf(node.bootstrap);
363 if (!nodeLabel.equals(""))
365 g.drawString(nodeLabel, xstart + 2, ypos - 2);
378 * @return DOCUMENT ME!
380 public Object findElement(int x, int y)
382 Enumeration keys = nameHash.keys();
384 while (keys.hasMoreElements())
386 Object ob = keys.nextElement();
387 Rectangle rect = (Rectangle) nameHash.get(ob);
389 if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y)
390 && (y <= (rect.y + rect.height)))
396 keys = nodeHash.keys();
398 while (keys.hasMoreElements())
400 Object ob = keys.nextElement();
401 Rectangle rect = (Rectangle) nodeHash.get(ob);
403 if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y)
404 && (y <= (rect.y + rect.height)))
419 public void pickNodes(Rectangle pickBox)
421 int width = getWidth();
422 int height = getHeight();
424 SequenceNode top = tree.getTopNode();
426 float wscale = (float) ((width * .8) - (offx * 2))
427 / tree.getMaxHeight();
431 top.count = ((SequenceNode) top.left()).count
432 + ((SequenceNode) top.right()).count;
435 float chunk = (float) (height - (offy)) / top.count;
437 pickNode(pickBox, top, chunk, wscale, width, offx, offy);
458 public void pickNode(Rectangle pickBox, SequenceNode node, float chunk,
459 float scale, int width, int offx, int offy)
466 if ((node.left() == null) && (node.right() == null))
468 float height = node.height;
469 float dist = node.dist;
471 int xstart = (int) ((height - dist) * scale) + offx;
472 int xend = (int) (height * scale) + offx;
474 int ypos = (int) (node.ycount * chunk) + offy;
476 if (pickBox.contains(new Point(xend, ypos)))
478 if (node.element() instanceof SequenceI)
480 SequenceI seq = (SequenceI) node.element();
481 SequenceGroup sg = av.getSelectionGroup();
485 sg.addOrRemove(seq, true);
492 pickNode(pickBox, (SequenceNode) node.left(), chunk, scale, width,
494 pickNode(pickBox, (SequenceNode) node.right(), chunk, scale, width,
507 public void setColor(SequenceNode node, Color c)
514 if ((node.left() == null) && (node.right() == null)) // TODO: internal node
518 if (node.element() instanceof SequenceI)
520 AlignmentPanel[] aps = getAssociatedPanels();
523 for (int a = 0; a < aps.length; a++)
525 final SequenceI seq = (SequenceI) node.element();
526 aps[a].av.setSequenceColour(seq, c);
534 setColor((SequenceNode) node.left(), c);
535 setColor((SequenceNode) node.right(), c);
544 Thread thread = new Thread(this);
548 // put printing in a thread to avoid painting problems
551 PrinterJob printJob = PrinterJob.getPrinterJob();
552 PageFormat pf = printJob.pageDialog(printJob.defaultPage());
554 printJob.setPrintable(this, pf);
556 if (printJob.printDialog())
561 } catch (Exception PrintException)
563 PrintException.printStackTrace();
578 * @return DOCUMENT ME!
580 * @throws PrinterException
583 public int print(Graphics pg, PageFormat pf, int pi)
584 throws PrinterException
587 pg.translate((int) pf.getImageableX(), (int) pf.getImageableY());
589 int pwidth = (int) pf.getImageableWidth();
590 int pheight = (int) pf.getImageableHeight();
592 int noPages = getHeight() / pheight;
596 return Printable.NO_SUCH_PAGE;
599 if (pwidth > getWidth())
606 if (pheight > getHeight())
608 pheight = getHeight();
615 FontMetrics fm = pg.getFontMetrics(font);
616 int height = fm.getHeight() * nameHash.size();
617 pg.translate(0, -pi * pheight);
618 pg.setClip(0, pi * pheight, pwidth, (pi * pheight) + pheight);
620 // translate number of pages,
621 // height is screen size as this is the
622 // non overlapping text size
626 draw(pg, pwidth, pheight);
628 return Printable.PAGE_EXISTS;
637 public void paintComponent(Graphics g)
639 super.paintComponent(g);
644 g.drawString(MessageManager.getString("label.calculating_tree")
645 + "....", 20, getHeight() / 2);
649 fm = g.getFontMetrics(font);
651 if (nameHash.size() == 0)
657 || (!fitToWindow && (scrollPane.getHeight() > ((fm
658 .getHeight() * nameHash.size()) + offy))))
660 draw(g, scrollPane.getWidth(), scrollPane.getHeight());
661 setPreferredSize(null);
665 setPreferredSize(new Dimension(scrollPane.getWidth(),
666 fm.getHeight() * nameHash.size()));
667 draw(g, scrollPane.getWidth(), fm.getHeight() * nameHash.size());
670 scrollPane.revalidate();
680 public void setFont(Font font)
696 public void draw(Graphics g1, int width, int height)
698 Graphics2D g2 = (Graphics2D) g1;
699 g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
700 RenderingHints.VALUE_ANTIALIAS_ON);
701 g2.setColor(Color.white);
702 g2.fillRect(0, 0, width, height);
705 if (longestName == null || tree == null)
707 g2.drawString("Calculating tree.", 20, 20);
709 offy = font.getSize() + 10;
711 fm = g2.getFontMetrics(font);
713 labelLength = fm.stringWidth(longestName) + 20; // 20 allows for scrollbar
715 float wscale = (width - labelLength - (offx * 2)) / tree.getMaxHeight();
717 SequenceNode top = tree.getTopNode();
721 top.count = ((SequenceNode) top.left()).count
722 + ((SequenceNode) top.right()).count;
725 float chunk = (float) (height - (offy)) / top.count;
727 drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy);
731 if (av.getCurrentTree() == tree)
733 g2.setColor(Color.red);
737 g2.setColor(Color.gray);
740 int x = (int) ((threshold * (getWidth() - labelLength - (2 * offx))) + offx);
742 g2.drawLine(x, 0, x, getHeight());
752 public void mouseReleased(MouseEvent e)
762 public void mouseEntered(MouseEvent e)
772 public void mouseExited(MouseEvent e)
782 public void mouseClicked(MouseEvent evt)
784 if (highlightNode != null)
786 if (SwingUtilities.isRightMouseButton(evt))
788 Color col = JColorChooser.showDialog(this,
789 MessageManager.getString("label.select_subtree_colour"),
790 highlightNode.color);
793 setColor(highlightNode, col);
796 else if (evt.getClickCount() > 1)
798 tree.swapNodes(highlightNode);
799 tree.reCount(tree.getTopNode());
800 tree.findHeight(tree.getTopNode());
804 Vector leaves = new Vector();
805 tree.findLeaves(highlightNode, leaves);
807 for (int i = 0; i < leaves.size(); i++)
809 SequenceI seq = (SequenceI) ((SequenceNode) leaves.elementAt(i))
811 treeSelectionChanged(seq);
816 PaintRefresher.Refresh(tp, av.getSequenceSetId());
821 public void mouseMoved(MouseEvent evt)
823 av.setCurrentTree(tree);
825 Object ob = findElement(evt.getX(), evt.getY());
827 if (ob instanceof SequenceNode)
829 highlightNode = (SequenceNode) ob;
830 this.setToolTipText("<html>"
831 + MessageManager.getString("label.highlightnode"));
837 if (highlightNode != null)
839 highlightNode = null;
840 setToolTipText(null);
846 public void mouseDragged(MouseEvent ect)
856 public void mousePressed(MouseEvent e)
858 av.setCurrentTree(tree);
863 Object ob = findElement(x, y);
865 if (ob instanceof SequenceI)
867 treeSelectionChanged((Sequence) ob);
868 PaintRefresher.Refresh(tp, ap.av.getSequenceSetId());
873 else if (!(ob instanceof SequenceNode))
876 if (tree.getMaxHeight() != 0)
878 threshold = (float) (x - offx)
879 / (float) (getWidth() - labelLength - (2 * offx));
881 tree.getGroups().removeAllElements();
882 tree.groupNodes(tree.getTopNode(), threshold);
883 setColor(tree.getTopNode(), Color.black);
885 AlignmentPanel[] aps = getAssociatedPanels();
887 // TODO push calls below into a single AlignViewportI method?
888 // see also AlignViewController.deleteGroups
889 for (int a = 0; a < aps.length; a++)
891 aps[a].av.setSelectionGroup(null);
892 aps[a].av.getAlignment().deleteAllGroups();
893 aps[a].av.clearSequenceColours();
895 if (av.getCodingComplement() != null)
897 av.getCodingComplement().setSelectionGroup(null);
898 av.getCodingComplement().getAlignment().deleteAllGroups();
899 av.getCodingComplement().clearSequenceColours();
904 PaintRefresher.Refresh(tp, ap.av.getSequenceSetId());
912 AlignmentPanel[] aps = getAssociatedPanels();
913 for (int i = 0; i < tree.getGroups().size(); i++)
915 Color col = new Color((int) (Math.random() * 255),
916 (int) (Math.random() * 255), (int) (Math.random() * 255));
917 setColor((SequenceNode) tree.getGroups().elementAt(i), col.brighter());
919 Vector l = tree.findLeaves(
920 (SequenceNode) tree.getGroups().elementAt(i), new Vector());
922 Vector sequences = new Vector();
924 for (int j = 0; j < l.size(); j++)
926 SequenceI s1 = (SequenceI) ((SequenceNode) l.elementAt(j))
929 if (!sequences.contains(s1))
931 sequences.addElement(s1);
935 ColourSchemeI cs = null;
936 SequenceGroup sg = new SequenceGroup(sequences, null, cs, true, true,
937 false, 0, av.getAlignment().getWidth() - 1);
939 if (av.getGlobalColourScheme() != null)
941 if (av.getGlobalColourScheme() instanceof UserColourScheme)
943 cs = new UserColourScheme(
944 ((UserColourScheme) av.getGlobalColourScheme())
950 cs = ColourSchemeProperty.getColour(sg, ColourSchemeProperty
951 .getColourName(av.getGlobalColourScheme()));
953 // cs is null if shading is an annotationColourGradient
956 cs.setThreshold(av.getGlobalColourScheme().getThreshold(),
957 av.isIgnoreGapsConsensus());
961 // sg.recalcConservation();
962 sg.setName("JTreeGroup:" + sg.hashCode());
965 for (int a = 0; a < aps.length; a++)
967 if (aps[a].av.getGlobalColourScheme() != null
968 && aps[a].av.getGlobalColourScheme().conservationApplied())
970 Conservation c = new Conservation("Group",
971 ResidueProperties.propHash, 3, sg.getSequences(null),
972 sg.getStartRes(), sg.getEndRes());
975 c.verdict(false, aps[a].av.getConsPercGaps());
976 sg.cs.setConservation(c);
979 aps[a].av.getAlignment().addGroup(new SequenceGroup(sg));
982 // TODO can we push all of the below into AlignViewportI?
983 av.getAlignment().addGroup(sg);
984 final AlignViewportI codingComplement = av.getCodingComplement();
985 if (codingComplement != null)
987 SequenceGroup mappedGroup = MappingUtils.mapSequenceGroup(sg, av,
989 if (mappedGroup.getSequences().size() > 0)
991 codingComplement.getAlignment().addGroup(mappedGroup);
992 for (SequenceI seq : mappedGroup.getSequences())
994 codingComplement.setSequenceColour(seq, col.brighter());
1000 // notify the panel to redo any group specific stuff.
1001 for (int a = 0; a < aps.length; a++)
1003 aps[a].updateAnnotation();
1004 // TODO: JAL-868 - need to ensure view colour change message is broadcast
1005 // to any Jmols listening in
1008 if (av.getCodingComplement() != null)
1010 ((AlignViewport) av.getCodingComplement()).getAlignPanel()
1011 .updateAnnotation();
1013 * idPanel. repaint ()
1024 public void setShowDistances(boolean state)
1026 this.showDistances = state;
1036 public void setShowBootstrap(boolean state)
1038 this.showBootstrap = state;
1048 public void setMarkPlaceholders(boolean state)
1050 this.markPlaceholders = state;
1054 AlignmentPanel[] getAssociatedPanels()
1056 if (applyToAllViews)
1058 return PaintRefresher.getAssociatedPanels(av.getSequenceSetId());
1062 return new AlignmentPanel[] { ap };