X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FTreeCanvas.java;h=6f143dbb5b54f24342db1cc7c355d17558822b99;hb=d1e20f6a5654ea6baf378afa3277b4e751c69a96;hp=98299ab7a2fcbd2359fc0adf0a09a0f5b2cf1674;hpb=cce9f2efe50ca646b758e4473d1ea6e52d3a5aaf;p=jalview.git diff --git a/src/jalview/gui/TreeCanvas.java b/src/jalview/gui/TreeCanvas.java index 98299ab..6f143db 100755 --- a/src/jalview/gui/TreeCanvas.java +++ b/src/jalview/gui/TreeCanvas.java @@ -1,101 +1,146 @@ /* - * 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 + * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$) + * Copyright (C) $$Year-Rel$$ The Jalview Authors + * + * This file is part of Jalview. + * + * Jalview 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 3 * 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. - * + * + * Jalview 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 + * along with Jalview. If not, see . + * The Jalview Authors are detailed in the 'AUTHORS' file. */ package jalview.gui; -import jalview.analysis.*; - -import jalview.datamodel.*; - -import jalview.schemes.*; - -import jalview.util.*; - -import java.awt.*; -import java.awt.event.*; -import java.awt.print.*; - -import java.util.*; - -import javax.swing.*; - +import java.awt.Color; +import java.awt.Dimension; +import java.awt.Font; +import java.awt.FontMetrics; +import java.awt.Graphics; +import java.awt.Graphics2D; +import java.awt.Point; +import java.awt.Rectangle; +import java.awt.RenderingHints; +import java.awt.event.MouseEvent; +import java.awt.event.MouseListener; +import java.awt.event.MouseMotionListener; +import java.awt.print.PageFormat; +import java.awt.print.Printable; +import java.awt.print.PrinterException; +import java.awt.print.PrinterJob; +import java.util.Hashtable; +import java.util.List; +import java.util.Map; +import java.util.Map.Entry; +import java.util.Vector; + +import javax.swing.JPanel; +import javax.swing.JScrollPane; +import javax.swing.SwingUtilities; +import javax.swing.ToolTipManager; + +import jalview.analysis.Conservation; +import jalview.analysis.TreeModel; +import jalview.api.AlignViewportI; +import jalview.datamodel.Sequence; +import jalview.datamodel.SequenceGroup; +import jalview.datamodel.SequenceI; +import jalview.datamodel.SequenceNode; +import jalview.gui.JalviewColourChooser.ColourChooserListener; +import jalview.schemes.ColourSchemeI; +import jalview.structure.SelectionSource; +import jalview.util.Format; +import jalview.util.MessageManager; /** * DOCUMENT ME! - * + * * @author $author$ * @version $Revision$ */ public class TreeCanvas extends JPanel implements MouseListener, Runnable, - Printable, MouseMotionListener + Printable, MouseMotionListener, SelectionSource { /** DOCUMENT ME!! */ public static final String PLACEHOLDER = " * "; - NJTree tree; + + TreeModel tree; + JScrollPane scrollPane; + TreePanel tp; - AlignViewport av; - AlignmentPanel ap; + + private AlignViewport av; + + private AlignmentPanel ap; + Font font; + FontMetrics fm; + boolean fitToWindow = true; + boolean showDistances = false; + boolean showBootstrap = false; + boolean markPlaceholders = false; + int offx = 20; + int offy; - float threshold; + + private float threshold; + String longestName; + int labelLength = -1; - Hashtable nameHash = new Hashtable(); - Hashtable nodeHash = new Hashtable(); + Map nameHash = new Hashtable<>(); + + Map nodeHash = new Hashtable<>(); + SequenceNode highlightNode; boolean applyToAllViews = false; /** * Creates a new TreeCanvas object. - * - * @param av DOCUMENT ME! - * @param tree DOCUMENT ME! - * @param scroller DOCUMENT ME! - * @param label DOCUMENT ME! + * + * @param av + * DOCUMENT ME! + * @param tree + * DOCUMENT ME! + * @param scroller + * DOCUMENT ME! + * @param label + * DOCUMENT ME! */ - public TreeCanvas(TreePanel tp, - AlignmentPanel ap, - JScrollPane scroller) + public TreeCanvas(TreePanel tp, AlignmentPanel ap, JScrollPane scroller) { this.tp = tp; this.av = ap.av; - this.ap = ap; + this.setAssociatedPanel(ap); font = av.getFont(); scrollPane = scroller; addMouseListener(this); addMouseMotionListener(this); - PaintRefresher.Register(tp, ap.av.getSequenceSetId()); ToolTipManager.sharedInstance().registerComponent(this); } /** * DOCUMENT ME! - * - * @param sequence DOCUMENT ME! + * + * @param sequence + * DOCUMENT ME! */ public void treeSelectionChanged(SequenceI sequence) { @@ -111,40 +156,41 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, aps[a].av.setSelectionGroup(selected); } - selected.setEndRes(aps[a].av.alignment.getWidth() - 1); + selected.setEndRes(aps[a].av.getAlignment().getWidth() - 1); selected.addOrRemove(sequence, true); } } /** * DOCUMENT ME! - * - * @param tree DOCUMENT ME! + * + * @param tree + * DOCUMENT ME! */ - public void setTree(NJTree tree) + public void setTree(TreeModel tree) { this.tree = tree; tree.findHeight(tree.getTopNode()); // Now have to calculate longest name based on the leaves - Vector leaves = tree.findLeaves(tree.getTopNode(), new Vector()); + Vector leaves = tree.findLeaves(tree.getTopNode()); boolean has_placeholders = false; longestName = ""; for (int i = 0; i < leaves.size(); i++) { - SequenceNode lf = (SequenceNode) leaves.elementAt(i); + SequenceNode lf = leaves.elementAt(i); if (lf.isPlaceholder()) { has_placeholders = true; } - if (longestName.length() < ( (Sequence) lf.element()).getName() - .length()) + if (longestName.length() < ((Sequence) lf.element()).getName() + .length()) { - longestName = TreeCanvas.PLACEHOLDER + - ( (Sequence) lf.element()).getName(); + longestName = TreeCanvas.PLACEHOLDER + + ((Sequence) lf.element()).getName(); } } @@ -153,37 +199,44 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param g DOCUMENT ME! - * @param node DOCUMENT ME! - * @param chunk DOCUMENT ME! - * @param scale DOCUMENT ME! - * @param width DOCUMENT ME! - * @param offx DOCUMENT ME! - * @param offy DOCUMENT ME! + * + * @param g + * DOCUMENT ME! + * @param node + * DOCUMENT ME! + * @param chunk + * DOCUMENT ME! + * @param wscale + * DOCUMENT ME! + * @param width + * DOCUMENT ME! + * @param offx + * DOCUMENT ME! + * @param offy + * DOCUMENT ME! */ public void drawNode(Graphics g, SequenceNode node, float chunk, - float scale, int width, int offx, int offy) + double wscale, int width, int offx, int offy) { if (node == null) { return; } - if ( (node.left() == null) && (node.right() == null)) + if ((node.left() == null) && (node.right() == null)) { // Drawing leaf node - float height = node.height; - float dist = node.dist; + double height = node.height; + double dist = node.dist; - int xstart = (int) ( (height - dist) * scale) + offx; - int xend = (int) (height * scale) + offx; + int xstart = (int) ((height - dist) * wscale) + offx; + int xend = (int) (height * wscale) + offx; int ypos = (int) (node.ycount * chunk) + offy; if (node.element() instanceof SequenceI) { - SequenceI seq = (SequenceI) ( (SequenceNode) node).element(); + SequenceI seq = (SequenceI) node.element(); if (av.getSequenceColour(seq) == Color.white) { @@ -206,17 +259,17 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, if (showDistances && (node.dist > 0)) { - nodeLabel = new Format("%-.2f").form(node.dist); + nodeLabel = new Format("%g").form(node.dist); } - if (showBootstrap) + if (showBootstrap && node.bootstrap > -1) { if (showDistances) { nodeLabel = nodeLabel + " : "; } - nodeLabel = nodeLabel + String.valueOf(node.getBootstrap()); + nodeLabel = nodeLabel + String.valueOf(node.bootstrap); } if (!nodeLabel.equals("")) @@ -225,26 +278,26 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, } String name = (markPlaceholders && node.isPlaceholder()) - ? (PLACEHOLDER + node.getName()) : node.getName(); + ? (PLACEHOLDER + node.getName()) + : node.getName(); int charWidth = fm.stringWidth(name) + 3; int charHeight = font.getSize(); Rectangle rect = new Rectangle(xend + 10, ypos - charHeight / 2, - charWidth, charHeight); + charWidth, charHeight); - nameHash.put( (SequenceI) node.element(), rect); + nameHash.put(node.element(), rect); // Colour selected leaves differently SequenceGroup selected = av.getSelectionGroup(); - if ( (selected != null) && - selected.getSequences(false).contains( (SequenceI) node.element())) + if ((selected != null) + && selected.getSequences(null).contains(node.element())) { g.setColor(Color.gray); - g.fillRect(xend + 10, ypos - charHeight / 2, charWidth, - charHeight); + g.fillRect(xend + 10, ypos - charHeight / 2, charWidth, charHeight); g.setColor(Color.white); } @@ -253,81 +306,98 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, } else { - drawNode(g, (SequenceNode) node.left(), chunk, scale, width, offx, - offy); - drawNode(g, (SequenceNode) node.right(), chunk, scale, width, offx, - offy); + drawNode(g, (SequenceNode) node.left(), chunk, wscale, width, offx, + offy); + drawNode(g, (SequenceNode) node.right(), chunk, wscale, width, offx, + offy); - float height = node.height; - float dist = node.dist; + double height = node.height; + double dist = node.dist; - int xstart = (int) ( (height - dist) * scale) + offx; - int xend = (int) (height * scale) + offx; + int xstart = (int) ((height - dist) * wscale) + offx; + int xend = (int) (height * wscale) + offx; int ypos = (int) (node.ycount * chunk) + offy; - g.setColor( ( (SequenceNode) node).color.darker()); + g.setColor(node.color.darker()); // Draw horizontal line g.drawLine(xstart, ypos, xend, ypos); if (node == highlightNode) + { g.fillRect(xend - 3, ypos - 3, 6, 6); + } else + { g.fillRect(xend - 2, ypos - 2, 4, 4); + } - int ystart = (int) ( ( (SequenceNode) node.left()).ycount * chunk) + - offy; - int yend = (int) ( ( (SequenceNode) node.right()).ycount * chunk) + - offy; + int ystart = (node.left() == null ? 0 + : (int) (((SequenceNode) node.left()).ycount * chunk)) + offy; + int yend = (node.right() == null ? 0 + : (int) (((SequenceNode) node.right()).ycount * chunk)) + + offy; Rectangle pos = new Rectangle(xend - 2, ypos - 2, 5, 5); nodeHash.put(node, pos); - g.drawLine( (int) (height * scale) + offx, ystart, - (int) (height * scale) + offx, yend); + g.drawLine((int) (height * wscale) + offx, ystart, + (int) (height * wscale) + offx, yend); + + String nodeLabel = ""; if (showDistances && (node.dist > 0)) { - g.drawString(new Format("%-.2f").form(node.dist).trim(), xstart + 2, - ypos - 2); + nodeLabel = new Format("%g").form(node.dist); + } + + if (showBootstrap && node.bootstrap > -1) + { + if (showDistances) + { + nodeLabel = nodeLabel + " : "; + } + + nodeLabel = nodeLabel + String.valueOf(node.bootstrap); + } + + if (!nodeLabel.equals("")) + { + g.drawString(nodeLabel, xstart + 2, ypos - 2); } } } /** * DOCUMENT ME! - * - * @param x DOCUMENT ME! - * @param y DOCUMENT ME! - * + * + * @param x + * DOCUMENT ME! + * @param y + * DOCUMENT ME! + * * @return DOCUMENT ME! */ public Object findElement(int x, int y) { - Enumeration keys = nameHash.keys(); - - while (keys.hasMoreElements()) + for (Entry entry : nameHash.entrySet()) { - Object ob = keys.nextElement(); - Rectangle rect = (Rectangle) nameHash.get(ob); + Rectangle rect = entry.getValue(); - if ( (x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) && - (y <= (rect.y + rect.height))) + if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) + && (y <= (rect.y + rect.height))) { - return ob; + return entry.getKey(); } } - keys = nodeHash.keys(); - - while (keys.hasMoreElements()) + for (Entry entry : nodeHash.entrySet()) { - Object ob = keys.nextElement(); - Rectangle rect = (Rectangle) nodeHash.get(ob); + Rectangle rect = entry.getValue(); - if ( (x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) && - (y <= (rect.y + rect.height))) + if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) + && (y <= (rect.y + rect.height))) { - return ob; + return entry.getKey(); } } @@ -336,8 +406,9 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param pickBox DOCUMENT ME! + * + * @param pickBox + * DOCUMENT ME! */ public void pickNodes(Rectangle pickBox) { @@ -346,12 +417,12 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, SequenceNode top = tree.getTopNode(); - float wscale = (float) ( (width * .8) - (offx * 2)) / tree.getMaxHeight(); + double wscale = ((width * .8) - (offx * 2)) / tree.getMaxHeight(); if (top.count == 0) { - top.count = ( (SequenceNode) top.left()).count + - ( (SequenceNode) top.right()).count; + top.count = ((SequenceNode) top.left()).count + + ((SequenceNode) top.right()).count; } float chunk = (float) (height - (offy)) / top.count; @@ -361,30 +432,36 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param pickBox DOCUMENT ME! - * @param node DOCUMENT ME! - * @param chunk DOCUMENT ME! - * @param scale DOCUMENT ME! - * @param width DOCUMENT ME! - * @param offx DOCUMENT ME! - * @param offy DOCUMENT ME! + * + * @param pickBox + * DOCUMENT ME! + * @param node + * DOCUMENT ME! + * @param chunk + * DOCUMENT ME! + * @param wscale + * DOCUMENT ME! + * @param width + * DOCUMENT ME! + * @param offx + * DOCUMENT ME! + * @param offy + * DOCUMENT ME! */ public void pickNode(Rectangle pickBox, SequenceNode node, float chunk, - float scale, int width, int offx, int offy) + double wscale, int width, int offx, int offy) { if (node == null) { return; } - if ( (node.left() == null) && (node.right() == null)) + if ((node.left() == null) && (node.right() == null)) { - float height = node.height; - float dist = node.dist; - - int xstart = (int) ( (height - dist) * scale) + offx; - int xend = (int) (height * scale) + offx; + double height = node.height; + // double dist = node.dist; + // int xstart = (int) ((height - dist) * wscale) + offx; + int xend = (int) (height * wscale) + offx; int ypos = (int) (node.ycount * chunk) + offy; @@ -404,18 +481,20 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, } else { - pickNode(pickBox, (SequenceNode) node.left(), chunk, scale, width, - offx, offy); - pickNode(pickBox, (SequenceNode) node.right(), chunk, scale, width, - offx, offy); + pickNode(pickBox, (SequenceNode) node.left(), chunk, wscale, width, + offx, offy); + pickNode(pickBox, (SequenceNode) node.right(), chunk, wscale, width, + offx, offy); } } /** * DOCUMENT ME! - * - * @param node DOCUMENT ME! - * @param c DOCUMENT ME! + * + * @param node + * DOCUMENT ME! + * @param c + * DOCUMENT ME! */ public void setColor(SequenceNode node, Color c) { @@ -424,25 +503,21 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, return; } - if ( (node.left() == null) && (node.right() == null)) + node.color = c; + if (node.element() instanceof SequenceI) { - node.color = c; - - if (node.element() instanceof SequenceI) + final SequenceI seq = (SequenceI) node.element(); + AlignmentPanel[] aps = getAssociatedPanels(); + if (aps != null) { - AlignmentPanel[] aps = getAssociatedPanels(); for (int a = 0; a < aps.length; a++) { - aps[a].av.setSequenceColour( (SequenceI) node.element(), c); + aps[a].av.setSequenceColour(seq, c); } } } - else - { - node.color = c; - setColor( (SequenceNode) node.left(), c); - setColor( (SequenceNode) node.right(), c); - } + setColor((SequenceNode) node.left(), c); + setColor((SequenceNode) node.right(), c); } /** @@ -455,10 +530,20 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, } // put printing in a thread to avoid painting problems + @Override public void run() { PrinterJob printJob = PrinterJob.getPrinterJob(); - PageFormat pf = printJob.pageDialog(printJob.defaultPage()); + PageFormat defaultPage = printJob.defaultPage(); + PageFormat pf = printJob.pageDialog(defaultPage); + + if (defaultPage == pf) + { + /* + * user cancelled + */ + return; + } printJob.setPrintable(this, pf); @@ -467,8 +552,7 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, try { printJob.print(); - } - catch (Exception PrintException) + } catch (Exception PrintException) { PrintException.printStackTrace(); } @@ -477,20 +561,25 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param pg DOCUMENT ME! - * @param pf DOCUMENT ME! - * @param pi DOCUMENT ME! - * + * + * @param pg + * DOCUMENT ME! + * @param pf + * DOCUMENT ME! + * @param pi + * DOCUMENT ME! + * * @return DOCUMENT ME! - * - * @throws PrinterException DOCUMENT ME! + * + * @throws PrinterException + * DOCUMENT ME! */ + @Override public int print(Graphics pg, PageFormat pf, int pi) - throws PrinterException + throws PrinterException { pg.setFont(font); - pg.translate( (int) pf.getImageableX(), (int) pf.getImageableY()); + pg.translate((int) pf.getImageableX(), (int) pf.getImageableY()); int pwidth = (int) pf.getImageableWidth(); int pheight = (int) pf.getImageableHeight(); @@ -536,9 +625,11 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param g DOCUMENT ME! + * + * @param g + * DOCUMENT ME! */ + @Override public void paintComponent(Graphics g) { super.paintComponent(g); @@ -546,21 +637,22 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, if (tree == null) { - g.drawString("Calculating tree....", 20, getHeight() / 2); + g.drawString( + MessageManager.getString("label.calculating_tree") + "....", + 20, getHeight() / 2); } else { fm = g.getFontMetrics(font); - if (nameHash.size() == 0) + int nameCount = nameHash.size(); + if (nameCount == 0) { repaint(); } - if (fitToWindow || - (!fitToWindow && - (scrollPane.getHeight() > ( (fm.getHeight() * nameHash.size()) + - offy)))) + if (fitToWindow || (!fitToWindow && (scrollPane + .getHeight() > ((fm.getHeight() * nameCount) + offy)))) { draw(g, scrollPane.getWidth(), scrollPane.getHeight()); setPreferredSize(null); @@ -568,8 +660,8 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, else { setPreferredSize(new Dimension(scrollPane.getWidth(), - fm.getHeight() * nameHash.size())); - draw(g, scrollPane.getWidth(), fm.getHeight() * nameHash.size()); + fm.getHeight() * nameCount)); + draw(g, scrollPane.getWidth(), fm.getHeight() * nameCount); } scrollPane.revalidate(); @@ -578,9 +670,11 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param fontSize DOCUMENT ME! + * + * @param fontSize + * DOCUMENT ME! */ + @Override public void setFont(Font font) { this.font = font; @@ -589,36 +683,42 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param g1 DOCUMENT ME! - * @param width DOCUMENT ME! - * @param height DOCUMENT ME! + * + * @param g1 + * DOCUMENT ME! + * @param width + * DOCUMENT ME! + * @param height + * DOCUMENT ME! */ public void draw(Graphics g1, int width, int height) { Graphics2D g2 = (Graphics2D) g1; g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, - RenderingHints.VALUE_ANTIALIAS_ON); + RenderingHints.VALUE_ANTIALIAS_ON); g2.setColor(Color.white); g2.fillRect(0, 0, width, height); - g2.setFont(font); + if (longestName == null || tree == null) + { + g2.drawString("Calculating tree.", 20, 20); + } offy = font.getSize() + 10; fm = g2.getFontMetrics(font); - labelLength = fm.stringWidth(longestName) + 20; //20 allows for scrollbar + labelLength = fm.stringWidth(longestName) + 20; // 20 allows for scrollbar - float wscale = (float) (width - labelLength - (offx * 2)) / - tree.getMaxHeight(); + double wscale = (width - labelLength - (offx * 2)) + / tree.getMaxHeight(); SequenceNode top = tree.getTopNode(); if (top.count == 0) { - top.count = ( (SequenceNode) top.left()).count + - ( (SequenceNode) top.right()).count; + top.count = ((SequenceNode) top.left()).count + + ((SequenceNode) top.right()).count; } float chunk = (float) (height - (offy)) / top.count; @@ -636,81 +736,111 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, g2.setColor(Color.gray); } - int x = (int) ( (threshold * (float) (getWidth() - labelLength - - (2 * offx))) + offx); + int x = (int) ((threshold * (getWidth() - labelLength - (2 * offx))) + + offx); g2.drawLine(x, 0, x, getHeight()); } } /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! + * Empty method to satisfy the MouseListener interface + * + * @param e */ + @Override public void mouseReleased(MouseEvent e) { + /* + * isPopupTrigger is set on mouseReleased on Windows + */ + if (e.isPopupTrigger()) + { + chooseSubtreeColour(); + e.consume(); // prevent mouseClicked happening + } } /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! + * Empty method to satisfy the MouseListener interface + * + * @param e */ + @Override public void mouseEntered(MouseEvent e) { } /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! + * Empty method to satisfy the MouseListener interface + * + * @param e */ + @Override public void mouseExited(MouseEvent e) { } /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! + * Handles a mouse click on a tree node (clicks elsewhere are handled in + * mousePressed). Click selects the sub-tree, double-click swaps leaf nodes + * order, right-click opens a dialogue to choose colour for the sub-tree. + * + * @param e */ + @Override public void mouseClicked(MouseEvent evt) { - if (highlightNode != null) + if (highlightNode == null) { - if (SwingUtilities.isRightMouseButton(evt)) - { - Color col = JColorChooser.showDialog(this, "Select Sub-Tree Colour", - highlightNode.color); + return; + } - setColor(highlightNode, col); - } - else - if (evt.getClickCount() > 1) + if (evt.getClickCount() > 1) + { + tree.swapNodes(highlightNode); + tree.reCount(tree.getTopNode()); + tree.findHeight(tree.getTopNode()); + } + else + { + Vector leaves = tree.findLeaves(highlightNode); + + for (int i = 0; i < leaves.size(); i++) { - tree.swapNodes(highlightNode); - tree.reCount(tree.getTopNode()); - tree.findHeight(tree.getTopNode()); + SequenceI seq = (SequenceI) leaves.elementAt(i).element(); + treeSelectionChanged(seq); } - else - { - Vector leaves = new Vector(); - tree.findLeaves(highlightNode, leaves); + av.sendSelection(); + } - for (int i = 0; i < leaves.size(); i++) - { - SequenceI seq = - (SequenceI) ( (SequenceNode) leaves.elementAt(i)).element(); - treeSelectionChanged(seq); - } - } + PaintRefresher.Refresh(tp, av.getSequenceSetId()); + repaint(); + } - PaintRefresher.Refresh(tp, av.getSequenceSetId()); - repaint(); - } + /** + * Offer the user the option to choose a colour for the highlighted node and + * its children; this colour is also applied to the corresponding sequence ids + * in the alignment + */ + void chooseSubtreeColour() + { + String ttl = MessageManager.getString("label.select_subtree_colour"); + ColourChooserListener listener = new ColourChooserListener() + { + @Override + public void colourSelected(Color c) + { + setColor(highlightNode, c); + PaintRefresher.Refresh(tp, ap.av.getSequenceSetId()); + repaint(); + } + }; + JalviewColourChooser.showColourChooser(this, ttl, highlightNode.color, + listener); } + @Override public void mouseMoved(MouseEvent evt) { av.setCurrentTree(tree); @@ -721,9 +851,7 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, { highlightNode = (SequenceNode) ob; this.setToolTipText( - "Left click to select leaves" - + "
Double-click to invert leaves" - + "
Right click to change colour"); + "" + MessageManager.getString("label.highlightnode")); repaint(); } @@ -738,18 +866,48 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, } } + @Override public void mouseDragged(MouseEvent ect) - {} + { + } /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! + * Handles a mouse press on a sequence name or the tree background canvas + * (click on a node is handled in mouseClicked). The action is to create + * groups by partitioning the tree at the mouse position. Colours for the + * groups (and sequence names) are generated randomly. + * + * @param e */ + @Override public void mousePressed(MouseEvent e) { av.setCurrentTree(tree); + /* + * isPopupTrigger is set for mousePressed (Mac) + * or mouseReleased (Windows) + */ + if (e.isPopupTrigger()) + { + if (highlightNode != null) + { + chooseSubtreeColour(); + } + return; + } + + /* + * defer right-click handling on Windows to + * mouseClicked; note isRightMouseButton + * also matches Cmd-click on Mac which should do + * nothing here + */ + if (SwingUtilities.isRightMouseButton(e)) + { + return; + } + int x = e.getX(); int y = e.getY(); @@ -757,59 +915,77 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, if (ob instanceof SequenceI) { - treeSelectionChanged( (Sequence) ob); - PaintRefresher.Refresh(tp, ap.av.getSequenceSetId()); + treeSelectionChanged((Sequence) ob); + PaintRefresher.Refresh(tp, + getAssociatedPanel().av.getSequenceSetId()); repaint(); + av.sendSelection(); return; } - else if (! (ob instanceof SequenceNode)) + else if (!(ob instanceof SequenceNode)) { // Find threshold if (tree.getMaxHeight() != 0) { - threshold = (float) (x - offx) / (float) (getWidth() - - labelLength - (2 * offx)); + threshold = (float) (x - offx) + / (float) (getWidth() - labelLength - (2 * offx)); - tree.getGroups().removeAllElements(); - tree.groupNodes(tree.getTopNode(), threshold); + List groups = tree.groupNodes(threshold); setColor(tree.getTopNode(), Color.black); AlignmentPanel[] aps = getAssociatedPanels(); + // TODO push calls below into a single AlignViewportI method? + // see also AlignViewController.deleteGroups for (int a = 0; a < aps.length; a++) { aps[a].av.setSelectionGroup(null); - aps[a].av.alignment.deleteAllGroups(); - aps[a].av.sequenceColours = null; + aps[a].av.getAlignment().deleteAllGroups(); + aps[a].av.clearSequenceColours(); + if (aps[a].av.getCodingComplement() != null) + { + aps[a].av.getCodingComplement().setSelectionGroup(null); + aps[a].av.getCodingComplement().getAlignment() + .deleteAllGroups(); + aps[a].av.getCodingComplement().clearSequenceColours(); + } + aps[a].av.setUpdateStructures(true); + } + colourGroups(groups); + + /* + * clear partition (don't show vertical line) if + * it is to the right of all nodes + */ + if (groups.isEmpty()) + { + threshold = 0f; } - colourGroups(); } - PaintRefresher.Refresh(tp, ap.av.getSequenceSetId()); + PaintRefresher.Refresh(tp, + getAssociatedPanel().av.getSequenceSetId()); repaint(); } } - void colourGroups() + void colourGroups(List groups) { - for (int i = 0; i < tree.getGroups().size(); i++) + AlignmentPanel[] aps = getAssociatedPanels(); + for (int i = 0; i < groups.size(); i++) { - Color col = new Color( (int) (Math.random() * 255), - (int) (Math.random() * 255), - (int) (Math.random() * 255)); - setColor( (SequenceNode) tree.getGroups().elementAt(i), - col.brighter()); + Color col = new Color((int) (Math.random() * 255), + (int) (Math.random() * 255), (int) (Math.random() * 255)); + setColor(groups.get(i), col.brighter()); - Vector l = tree.findLeaves( (SequenceNode) tree.getGroups() - .elementAt(i), - new Vector()); + Vector l = tree.findLeaves(groups.get(i)); - Vector sequences = new Vector(); + Vector sequences = new Vector<>(); for (int j = 0; j < l.size(); j++) { - SequenceI s1 = (SequenceI) ( (SequenceNode) l.elementAt(j)).element(); + SequenceI s1 = (SequenceI) l.elementAt(j).element(); if (!sequences.contains(s1)) { @@ -818,57 +994,63 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, } ColourSchemeI cs = null; + SequenceGroup _sg = new SequenceGroup(sequences, null, cs, true, true, + false, 0, av.getAlignment().getWidth() - 1); + + _sg.setName("JTreeGroup:" + _sg.hashCode()); + _sg.setIdColour(col); - if (av.getGlobalColourScheme() != null) + for (int a = 0; a < aps.length; a++) { - if (av.getGlobalColourScheme() instanceof UserColourScheme) + SequenceGroup sg = new SequenceGroup(_sg); + AlignViewport viewport = aps[a].av; + + // Propagate group colours in each view + if (viewport.getGlobalColourScheme() != null) { - cs = new UserColourScheme( - ( (UserColourScheme) av.getGlobalColourScheme()).getColours()); + cs = viewport.getGlobalColourScheme().getInstance(viewport, sg); + sg.setColourScheme(cs); + sg.getGroupColourScheme().setThreshold( + viewport.getResidueShading().getThreshold(), + viewport.isIgnoreGapsConsensus()); + if (viewport.getResidueShading().conservationApplied()) + { + Conservation c = new Conservation("Group", + sg.getSequences(null), sg.getStartRes(), + sg.getEndRes()); + c.calculate(); + c.verdict(false, viewport.getConsPercGaps()); + sg.cs.setConservation(c); + } } - else - cs = ColourSchemeProperty.getColour(sequences, - av.alignment.getWidth(), - ColourSchemeProperty. - getColourName( - av.getGlobalColourScheme())); - - cs.setThreshold(av.getGlobalColourScheme().getThreshold(), - av.getIgnoreGapsConsensus()); + // indicate that associated structure views will need an update + viewport.setUpdateStructures(true); + // propagate structure view update and sequence group to complement view + viewport.addSequenceGroup(sg); } + } - SequenceGroup sg = new SequenceGroup(sequences, - "TreeGroup", cs, true, true, false, - 0, - av.alignment.getWidth() - 1); - - AlignmentPanel[] aps = getAssociatedPanels(); - for (int a = 0; a < aps.length; a++) + // notify the panel(s) to redo any group specific stuff + // also updates structure views if necessary + for (int a = 0; a < aps.length; a++) + { + aps[a].updateAnnotation(); + final AlignViewportI codingComplement = aps[a].av + .getCodingComplement(); + if (codingComplement != null) { - if (aps[a].av.getGlobalColourScheme() != null - && aps[a].av.getGlobalColourScheme().conservationApplied()) - { - Conservation c = new Conservation("Group", - ResidueProperties.propHash, 3, - sg.getSequences(false), - sg.getStartRes(), sg.getEndRes()); - - c.calculate(); - c.verdict(false, aps[a].av.ConsPercGaps); - sg.cs.setConservation(c); - } - - aps[a].av.alignment.addGroup(sg); + ((AlignViewport) codingComplement).getAlignPanel() + .updateAnnotation(); } } - } /** * DOCUMENT ME! - * - * @param state DOCUMENT ME! + * + * @param state + * DOCUMENT ME! */ public void setShowDistances(boolean state) { @@ -878,8 +1060,9 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param state DOCUMENT ME! + * + * @param state + * DOCUMENT ME! */ public void setShowBootstrap(boolean state) { @@ -889,8 +1072,9 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, /** * DOCUMENT ME! - * - * @param state DOCUMENT ME! + * + * @param state + * DOCUMENT ME! */ public void setMarkPlaceholders(boolean state) { @@ -905,6 +1089,48 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable, return PaintRefresher.getAssociatedPanels(av.getSequenceSetId()); } else - return new AlignmentPanel[]{ap}; + { + return new AlignmentPanel[] { getAssociatedPanel() }; + } + } + + public AlignmentPanel getAssociatedPanel() + { + return ap; + } + + public void setAssociatedPanel(AlignmentPanel ap) + { + this.ap = ap; + } + + public AlignViewport getViewport() + { + return av; + } + + public void setViewport(AlignViewport av) + { + this.av = av; + } + + public float getThreshold() + { + return threshold; + } + + public void setThreshold(float threshold) + { + this.threshold = threshold; + } + + public boolean isApplyToAllViews() + { + return this.applyToAllViews; + } + + public void setApplyToAllViews(boolean applyToAllViews) + { + this.applyToAllViews = applyToAllViews; } }