2 * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3 * Copyright (C) $$Year-Rel$$ 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.TreeModel;
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.UserColourScheme;
33 import jalview.structure.SelectionSource;
34 import jalview.util.Format;
35 import jalview.util.MessageManager;
37 import java.awt.Color;
38 import java.awt.Dimension;
40 import java.awt.FontMetrics;
41 import java.awt.Graphics;
42 import java.awt.Graphics2D;
43 import java.awt.Point;
44 import java.awt.Rectangle;
45 import java.awt.RenderingHints;
46 import java.awt.event.MouseEvent;
47 import java.awt.event.MouseListener;
48 import java.awt.event.MouseMotionListener;
49 import java.awt.print.PageFormat;
50 import java.awt.print.Printable;
51 import java.awt.print.PrinterException;
52 import java.awt.print.PrinterJob;
53 import java.util.Enumeration;
54 import java.util.Hashtable;
55 import java.util.List;
56 import java.util.Vector;
58 import javax.swing.JColorChooser;
59 import javax.swing.JPanel;
60 import javax.swing.JScrollPane;
61 import javax.swing.SwingUtilities;
62 import javax.swing.ToolTipManager;
70 public class TreeCanvas extends JPanel implements MouseListener, Runnable,
71 Printable, MouseMotionListener, SelectionSource
74 public static final String PLACEHOLDER = " * ";
78 JScrollPane scrollPane;
90 boolean fitToWindow = true;
92 boolean showDistances = false;
94 boolean showBootstrap = false;
96 boolean markPlaceholders = false;
106 int labelLength = -1;
108 Hashtable nameHash = new Hashtable();
110 Hashtable nodeHash = new Hashtable();
112 SequenceNode highlightNode;
114 boolean applyToAllViews = false;
117 * Creates a new TreeCanvas object.
128 public TreeCanvas(TreePanel tp, AlignmentPanel ap, JScrollPane scroller)
134 scrollPane = scroller;
135 addMouseListener(this);
136 addMouseMotionListener(this);
137 ToolTipManager.sharedInstance().registerComponent(this);
146 public void treeSelectionChanged(SequenceI sequence)
148 AlignmentPanel[] aps = getAssociatedPanels();
150 for (int a = 0; a < aps.length; a++)
152 SequenceGroup selected = aps[a].av.getSelectionGroup();
154 if (selected == null)
156 selected = new SequenceGroup();
157 aps[a].av.setSelectionGroup(selected);
160 selected.setEndRes(aps[a].av.getAlignment().getWidth() - 1);
161 selected.addOrRemove(sequence, true);
171 public void setTree(TreeModel tree)
174 tree.findHeight(tree.getTopNode());
176 // Now have to calculate longest name based on the leaves
177 Vector<SequenceNode> leaves = tree.findLeaves(tree.getTopNode());
178 boolean has_placeholders = false;
181 for (int i = 0; i < leaves.size(); i++)
183 SequenceNode lf = leaves.elementAt(i);
185 if (lf.isPlaceholder())
187 has_placeholders = true;
190 if (longestName.length() < ((Sequence) lf.element()).getName()
193 longestName = TreeCanvas.PLACEHOLDER
194 + ((Sequence) lf.element()).getName();
198 setMarkPlaceholders(has_placeholders);
219 public void drawNode(Graphics g, SequenceNode node, float chunk,
220 double wscale, int width, int offx, int offy)
227 if ((node.left() == null) && (node.right() == null))
230 double height = node.height;
231 double dist = node.dist;
233 int xstart = (int) ((height - dist) * wscale) + offx;
234 int xend = (int) (height * wscale) + offx;
236 int ypos = (int) (node.ycount * chunk) + offy;
238 if (node.element() instanceof SequenceI)
240 SequenceI seq = (SequenceI) node.element();
242 if (av.getSequenceColour(seq) == Color.white)
244 g.setColor(Color.black);
248 g.setColor(av.getSequenceColour(seq).darker());
253 g.setColor(Color.black);
256 // Draw horizontal line
257 g.drawLine(xstart, ypos, xend, ypos);
259 String nodeLabel = "";
261 if (showDistances && (node.dist > 0))
263 nodeLabel = new Format("%-.2f").form(node.dist);
266 if (showBootstrap && node.bootstrap > -1)
270 nodeLabel = nodeLabel + " : ";
273 nodeLabel = nodeLabel + String.valueOf(node.bootstrap);
276 if (!nodeLabel.equals(""))
278 g.drawString(nodeLabel, xstart + 2, ypos - 2);
281 String name = (markPlaceholders && node.isPlaceholder())
282 ? (PLACEHOLDER + 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, wscale, width, offx,
312 drawNode(g, (SequenceNode) node.right(), chunk, wscale, width, offx,
315 double height = node.height;
316 double dist = node.dist;
318 int xstart = (int) ((height - dist) * wscale) + offx;
319 int xend = (int) (height * wscale) + 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 = (node.left() == null ? 0
336 : (int) (((SequenceNode) node.left()).ycount * chunk)) + offy;
337 int yend = (node.right() == null ? 0
338 : (int) (((SequenceNode) node.right()).ycount * chunk))
341 Rectangle pos = new Rectangle(xend - 2, ypos - 2, 5, 5);
342 nodeHash.put(node, pos);
344 g.drawLine((int) (height * wscale) + offx, ystart,
345 (int) (height * wscale) + offx, yend);
347 String nodeLabel = "";
349 if (showDistances && (node.dist > 0))
351 nodeLabel = new Format("%-.2f").form(node.dist);
354 if (showBootstrap && node.bootstrap > -1)
358 nodeLabel = nodeLabel + " : ";
361 nodeLabel = nodeLabel + String.valueOf(node.bootstrap);
364 if (!nodeLabel.equals(""))
366 g.drawString(nodeLabel, xstart + 2, ypos - 2);
379 * @return DOCUMENT ME!
381 public Object findElement(int x, int y)
383 Enumeration keys = nameHash.keys();
385 while (keys.hasMoreElements())
387 Object ob = keys.nextElement();
388 Rectangle rect = (Rectangle) nameHash.get(ob);
390 if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y)
391 && (y <= (rect.y + rect.height)))
397 keys = nodeHash.keys();
399 while (keys.hasMoreElements())
401 Object ob = keys.nextElement();
402 Rectangle rect = (Rectangle) nodeHash.get(ob);
404 if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y)
405 && (y <= (rect.y + rect.height)))
420 public void pickNodes(Rectangle pickBox)
422 int width = getWidth();
423 int height = getHeight();
425 SequenceNode top = tree.getTopNode();
427 double wscale = ((width * .8) - (offx * 2)) / 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 double wscale, int width, int offx, int offy)
466 if ((node.left() == null) && (node.right() == null))
468 double height = node.height;
469 double dist = node.dist;
471 int xstart = (int) ((height - dist) * wscale) + offx;
472 int xend = (int) (height * wscale) + 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, wscale, width,
494 pickNode(pickBox, (SequenceNode) node.right(), chunk, wscale, width,
507 public void setColor(SequenceNode node, Color c)
516 if ((node.left() != null) || (node.right() != null)) // TODO: internal node
518 setColor((SequenceNode) node.left(), c);
519 setColor((SequenceNode) node.right(), c);
528 Thread thread = new Thread(this);
532 // put printing in a thread to avoid painting problems
536 PrinterJob printJob = PrinterJob.getPrinterJob();
537 PageFormat defaultPage = printJob.defaultPage();
538 PageFormat pf = printJob.pageDialog(defaultPage);
540 if (defaultPage == pf)
548 printJob.setPrintable(this, pf);
550 if (printJob.printDialog())
555 } catch (Exception PrintException)
557 PrintException.printStackTrace();
572 * @return DOCUMENT ME!
574 * @throws PrinterException
578 public int print(Graphics pg, PageFormat pf, int pi)
579 throws PrinterException
582 pg.translate((int) pf.getImageableX(), (int) pf.getImageableY());
584 int pwidth = (int) pf.getImageableWidth();
585 int pheight = (int) pf.getImageableHeight();
587 int noPages = getHeight() / pheight;
591 return Printable.NO_SUCH_PAGE;
594 if (pwidth > getWidth())
601 if (pheight > getHeight())
603 pheight = getHeight();
610 FontMetrics fm = pg.getFontMetrics(font);
611 int height = fm.getHeight() * nameHash.size();
612 pg.translate(0, -pi * pheight);
613 pg.setClip(0, pi * pheight, pwidth, (pi * pheight) + pheight);
615 // translate number of pages,
616 // height is screen size as this is the
617 // non overlapping text size
621 draw(pg, pwidth, pheight);
623 return Printable.PAGE_EXISTS;
633 public void paintComponent(Graphics g)
635 super.paintComponent(g);
641 MessageManager.getString("label.calculating_tree") + "....",
642 20, getHeight() / 2);
646 fm = g.getFontMetrics(font);
648 if (nameHash.size() == 0)
653 if (fitToWindow || (!fitToWindow && (scrollPane
654 .getHeight() > ((fm.getHeight() * nameHash.size()) + offy))))
656 draw(g, scrollPane.getWidth(), scrollPane.getHeight());
657 setPreferredSize(null);
661 setPreferredSize(new Dimension(scrollPane.getWidth(),
662 fm.getHeight() * nameHash.size()));
663 draw(g, scrollPane.getWidth(), fm.getHeight() * nameHash.size());
666 scrollPane.revalidate();
677 public void setFont(Font font)
693 public void draw(Graphics g1, int width, int height)
695 Graphics2D g2 = (Graphics2D) g1;
696 g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
697 RenderingHints.VALUE_ANTIALIAS_ON);
698 g2.setColor(Color.white);
699 g2.fillRect(0, 0, width, height);
702 if (longestName == null || tree == null)
704 g2.drawString("Calculating tree.", 20, 20);
706 offy = font.getSize() + 10;
708 fm = g2.getFontMetrics(font);
710 labelLength = fm.stringWidth(longestName) + 20; // 20 allows for scrollbar
712 double wscale = (width - labelLength - (offx * 2))
713 / tree.getMaxHeight();
715 SequenceNode top = tree.getTopNode();
719 top.count = ((SequenceNode) top.left()).count
720 + ((SequenceNode) top.right()).count;
723 float chunk = (float) (height - (offy)) / top.count;
725 drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy);
729 if (av.getCurrentTree() == tree)
731 g2.setColor(Color.red);
735 g2.setColor(Color.gray);
738 int x = (int) ((threshold * (getWidth() - labelLength - (2 * offx)))
741 g2.drawLine(x, 0, x, getHeight());
746 * Empty method to satisfy the MouseListener interface
751 public void mouseReleased(MouseEvent e)
754 * isPopupTrigger is set on mouseReleased on Windows
756 if (e.isPopupTrigger())
758 chooseSubtreeColour();
759 e.consume(); // prevent mouseClicked happening
764 * Empty method to satisfy the MouseListener interface
769 public void mouseEntered(MouseEvent e)
774 * Empty method to satisfy the MouseListener interface
779 public void mouseExited(MouseEvent e)
784 * Handles a mouse click on a tree node (clicks elsewhere are handled in
785 * mousePressed). Click selects the sub-tree, double-click swaps leaf nodes
786 * order, right-click opens a dialogue to choose colour for the sub-tree.
791 public void mouseClicked(MouseEvent evt)
793 if (highlightNode == null)
798 if (evt.getClickCount() > 1)
800 tree.swapNodes(highlightNode);
801 tree.reCount(tree.getTopNode());
802 tree.findHeight(tree.getTopNode());
806 Vector<SequenceNode> leaves = tree.findLeaves(highlightNode);
808 for (int i = 0; i < leaves.size(); i++)
810 SequenceI seq = (SequenceI) leaves.elementAt(i).element();
811 treeSelectionChanged(seq);
816 PaintRefresher.Refresh(tp, av.getSequenceSetId());
821 * Offer the user the option to choose a colour for the highlighted node and
822 * its children; this colour is also applied to the corresponding sequence ids
825 void chooseSubtreeColour()
827 Color col = JColorChooser.showDialog(this,
828 MessageManager.getString("label.select_subtree_colour"),
829 highlightNode.color);
832 setColor(highlightNode, col);
833 PaintRefresher.Refresh(tp, ap.av.getSequenceSetId());
839 public void mouseMoved(MouseEvent evt)
841 av.setCurrentTree(tree);
843 Object ob = findElement(evt.getX(), evt.getY());
845 if (ob instanceof SequenceNode)
847 highlightNode = (SequenceNode) ob;
849 "<html>" + MessageManager.getString("label.highlightnode"));
855 if (highlightNode != null)
857 highlightNode = null;
858 setToolTipText(null);
865 public void mouseDragged(MouseEvent ect)
870 * Handles a mouse press on a sequence name or the tree background canvas
871 * (click on a node is handled in mouseClicked). The action is to create
872 * groups by partitioning the tree at the mouse position. Colours for the
873 * groups (and sequence names) are generated randomly.
878 public void mousePressed(MouseEvent e)
880 av.setCurrentTree(tree);
883 * isPopupTrigger is set for mousePressed (Mac)
884 * or mouseReleased (Windows)
886 if (e.isPopupTrigger())
888 if (highlightNode != null)
890 chooseSubtreeColour();
896 * defer right-click handling on Windows to
897 * mouseClicked; note isRightMouseButton
898 * also matches Cmd-click on Mac which should do
901 if (SwingUtilities.isRightMouseButton(e))
909 Object ob = findElement(x, y);
911 if (ob instanceof SequenceI)
913 treeSelectionChanged((Sequence) ob);
914 PaintRefresher.Refresh(tp, ap.av.getSequenceSetId());
919 else if (!(ob instanceof SequenceNode))
922 if (tree.getMaxHeight() != 0)
924 threshold = (float) (x - offx)
925 / (float) (getWidth() - labelLength - (2 * offx));
927 List<SequenceNode> groups = tree.groupNodes(threshold);
928 setColor(tree.getTopNode(), Color.black);
930 AlignmentPanel[] aps = getAssociatedPanels();
932 // TODO push calls below into a single AlignViewportI method?
933 // see also AlignViewController.deleteGroups
934 for (int a = 0; a < aps.length; a++)
936 aps[a].av.setSelectionGroup(null);
937 aps[a].av.getAlignment().deleteAllGroups();
938 aps[a].av.clearSequenceColours();
939 if (aps[a].av.getCodingComplement() != null)
941 aps[a].av.getCodingComplement().setSelectionGroup(null);
942 aps[a].av.getCodingComplement().getAlignment()
944 aps[a].av.getCodingComplement().clearSequenceColours();
947 colourGroups(groups);
950 * clear partition (don't show vertical line) if
951 * it is to the right of all nodes
953 if (groups.isEmpty())
959 PaintRefresher.Refresh(tp, ap.av.getSequenceSetId());
965 void colourGroups(List<SequenceNode> groups)
967 AlignmentPanel[] aps = getAssociatedPanels();
968 for (int i = 0; i < groups.size(); i++)
970 Color col = new Color((int) (Math.random() * 255),
971 (int) (Math.random() * 255), (int) (Math.random() * 255));
972 setColor(groups.get(i), col.brighter());
974 Vector<SequenceNode> l = tree.findLeaves(groups.get(i));
976 Vector<SequenceI> sequences = new Vector<>();
978 for (int j = 0; j < l.size(); j++)
980 SequenceI s1 = (SequenceI) l.elementAt(j).element();
982 if (!sequences.contains(s1))
984 sequences.addElement(s1);
988 ColourSchemeI cs = null;
989 SequenceGroup sg = new SequenceGroup(sequences, null, cs, true, true,
990 false, 0, av.getAlignment().getWidth() - 1);
992 if (av.getGlobalColourScheme() != null)
994 if (av.getGlobalColourScheme() instanceof UserColourScheme)
996 cs = new UserColourScheme(
997 ((UserColourScheme) av.getGlobalColourScheme())
1003 cs = ColourSchemeProperty.getColourScheme(av, sg,
1004 ColourSchemeProperty
1005 .getColourName(av.getGlobalColourScheme()));
1007 // cs is null if shading is an annotationColourGradient
1010 // cs.setThreshold(av.getViewportColourScheme().getThreshold(),
1011 // av.isIgnoreGapsConsensus());
1014 sg.setColourScheme(cs);
1015 sg.getGroupColourScheme().setThreshold(
1016 av.getResidueShading().getThreshold(),
1017 av.isIgnoreGapsConsensus());
1018 // sg.recalcConservation();
1019 sg.setName("JTreeGroup:" + sg.hashCode());
1020 sg.setIdColour(col);
1022 for (int a = 0; a < aps.length; a++)
1024 if (aps[a].av.getGlobalColourScheme() != null
1025 && aps[a].av.getResidueShading().conservationApplied())
1027 Conservation c = new Conservation("Group", sg.getSequences(null),
1028 sg.getStartRes(), sg.getEndRes());
1030 c.verdict(false, aps[a].av.getConsPercGaps());
1031 sg.cs.setConservation(c);
1033 // indicate that associated structure views will need an update
1034 aps[a].av.setUpdateStructures(true);
1035 // propagate structure view update and sequence group to complement view
1036 aps[a].av.addSequenceGroup(new SequenceGroup(sg));
1040 // notify the panel(s) to redo any group specific stuff
1041 // also updates structure views if necessary
1042 for (int a = 0; a < aps.length; a++)
1044 aps[a].updateAnnotation();
1045 final AlignViewportI codingComplement = aps[a].av
1046 .getCodingComplement();
1047 if (codingComplement != null)
1049 ((AlignViewport) codingComplement).getAlignPanel()
1050 .updateAnnotation();
1061 public void setShowDistances(boolean state)
1063 this.showDistances = state;
1073 public void setShowBootstrap(boolean state)
1075 this.showBootstrap = state;
1085 public void setMarkPlaceholders(boolean state)
1087 this.markPlaceholders = state;
1091 AlignmentPanel[] getAssociatedPanels()
1093 if (applyToAllViews)
1095 return PaintRefresher.getAssociatedPanels(av.getSequenceSetId());
1099 return new AlignmentPanel[] { ap };