X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FTreeCanvas.java;h=c5650f5a973013c65c436c11991d76b7bd60a390;hb=aced09c4feeaf3406269442c14e54abeeb4cad81;hp=dc08b9ad084f3a6cf2b41150dd4f32661eea0bf3;hpb=1b032d3acf34d0c4166c84366f89e29ff4c784eb;p=jalview.git diff --git a/src/jalview/gui/TreeCanvas.java b/src/jalview/gui/TreeCanvas.java index dc08b9a..c5650f5 100755 --- a/src/jalview/gui/TreeCanvas.java +++ b/src/jalview/gui/TreeCanvas.java @@ -1,881 +1,1001 @@ /* - * 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 2.8.2) + * Copyright (C) 2014 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.util.*; import java.awt.*; import java.awt.event.*; import java.awt.print.*; - -import java.util.*; - import javax.swing.*; +import jalview.analysis.*; +import jalview.datamodel.*; +import jalview.schemes.*; +import jalview.util.*; /** * DOCUMENT ME! - * + * * @author $author$ * @version $Revision$ */ public class TreeCanvas extends JPanel implements MouseListener, Runnable, - Printable, MouseMotionListener + Printable, MouseMotionListener { - /** DOCUMENT ME!! */ - public static final String PLACEHOLDER = " * "; - NJTree tree; - JScrollPane scrollPane; - AlignViewport av; - 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; - String longestName; - int labelLength = -1; - - Hashtable nameHash = new Hashtable(); - Hashtable nodeHash = new Hashtable(); - SequenceNode highlightNode; - - /** - * Creates a new TreeCanvas object. - * - * @param av DOCUMENT ME! - * @param tree DOCUMENT ME! - * @param scroller DOCUMENT ME! - * @param label DOCUMENT ME! - */ - public TreeCanvas(AlignmentPanel ap, JScrollPane scroller) - { - this.av = ap.av; - this.ap = ap; - font = av.getFont(); - scrollPane = scroller; - addMouseListener(this); - addMouseMotionListener(this); - PaintRefresher.Register(this, ap.av.getSequenceSetId()); - ToolTipManager.sharedInstance().registerComponent(this); - } - - - /** - * DOCUMENT ME! - * - * @param sequence DOCUMENT ME! - */ - public void treeSelectionChanged(SequenceI sequence) - { - SequenceGroup selected = av.getSelectionGroup(); - - if (selected == null) - { - selected = new SequenceGroup(); - av.setSelectionGroup(selected); - } + /** DOCUMENT ME!! */ + public static final String PLACEHOLDER = " * "; - selected.setEndRes(av.alignment.getWidth()-1); - selected.addOrRemove(sequence, true); - } + NJTree tree; - /** - * DOCUMENT ME! - * - * @param tree DOCUMENT ME! - */ - public void setTree(NJTree tree) - { - this.tree = tree; - tree.findHeight(tree.getTopNode()); + JScrollPane scrollPane; - // Now have to calculate longest name based on the leaves - Vector leaves = tree.findLeaves(tree.getTopNode(), new Vector()); - boolean has_placeholders = false; - longestName = ""; + TreePanel tp; - for (int i = 0; i < leaves.size(); i++) - { - SequenceNode lf = (SequenceNode) leaves.elementAt(i); + AlignViewport av; - if (lf.isPlaceholder()) - { - has_placeholders = true; - } + AlignmentPanel ap; - if (longestName.length() < ( (Sequence) lf.element()).getName() - .length()) - { - longestName = TreeCanvas.PLACEHOLDER + - ( (Sequence) lf.element()).getName(); - } - } + Font font; - setMarkPlaceholders(has_placeholders); - } + FontMetrics fm; - /** - * 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! - */ - public void drawNode(Graphics g, SequenceNode node, float chunk, - float scale, int width, int offx, int offy) - { - if (node == null) - { - return; - } + boolean fitToWindow = true; - if ((node.left() == null) && (node.right() == null)) - { - // Drawing leaf node - float height = node.height; - float dist = node.dist; - - int xstart = (int) ((height - dist) * scale) + offx; - int xend = (int) (height * scale) + offx; - - int ypos = (int) (node.ycount * chunk) + offy; - - if (node.element() instanceof SequenceI) - { - SequenceI seq = (SequenceI)((SequenceNode) node).element(); - - if (av.getSequenceColour(seq) == Color.white) - { - g.setColor(Color.black); - } - else - { - g.setColor(av.getSequenceColour(seq).darker()); - } - } - else - { - g.setColor(Color.black); - } - - // Draw horizontal line - g.drawLine(xstart, ypos, xend, ypos); - - String nodeLabel = ""; - - if (showDistances && (node.dist > 0)) - { - nodeLabel = new Format("%-.2f").form(node.dist); - } - - if (showBootstrap) - { - if (showDistances) - { - nodeLabel = nodeLabel + " : "; - } - - nodeLabel = nodeLabel + String.valueOf(node.getBootstrap()); - } - - if (!nodeLabel.equals("")) - { - g.drawString(nodeLabel, xstart+2, ypos - 2); - } - - String name = (markPlaceholders && node.isPlaceholder()) - ? (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); - - nameHash.put((SequenceI) node.element(), rect); - - // Colour selected leaves differently - SequenceGroup selected = av.getSelectionGroup(); - - if ((selected != null) && - selected.getSequences(false).contains((SequenceI) node.element())) - { - g.setColor(Color.gray); - - g.fillRect(xend + 10, ypos-charHeight/2, charWidth, - charHeight); - g.setColor(Color.white); - } - - g.drawString(name, xend + 10, ypos+fm.getDescent()); - g.setColor(Color.black); - } - else - { - drawNode(g, (SequenceNode) node.left(), chunk, scale, width, offx, - offy); - drawNode(g, (SequenceNode) node.right(), chunk, scale, width, offx, - offy); - - float height = node.height; - float dist = node.dist; - - int xstart = (int) ((height - dist) * scale) + offx; - int xend = (int) (height * scale) + offx; - int ypos = (int) (node.ycount * chunk) + offy; - - g.setColor(((SequenceNode) 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; - - 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); - - if (showDistances && (node.dist > 0)) - { - g.drawString(new Format("%-.2f").form(node.dist).trim(), xstart+2, - ypos - 2); - } - } - } + boolean showDistances = false; - /** - * DOCUMENT ME! - * - * @param x DOCUMENT ME! - * @param y DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public Object findElement(int x, int y) - { - Enumeration keys = nameHash.keys(); + boolean showBootstrap = false; - while (keys.hasMoreElements()) - { - Object ob = keys.nextElement(); - Rectangle rect = (Rectangle) nameHash.get(ob); - - if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) && - (y <= (rect.y + rect.height))) - { - return ob; - } - } + boolean markPlaceholders = false; - keys = nodeHash.keys(); + int offx = 20; - while (keys.hasMoreElements()) - { - Object ob = keys.nextElement(); - Rectangle rect = (Rectangle) nodeHash.get(ob); - - if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) && - (y <= (rect.y + rect.height))) - { - return ob; - } - } + int offy; - return null; - } + float threshold; - /** - * DOCUMENT ME! - * - * @param pickBox DOCUMENT ME! - */ - public void pickNodes(Rectangle pickBox) - { - int width = getWidth(); - int height = getHeight(); + String longestName; - SequenceNode top = tree.getTopNode(); + int labelLength = -1; - float wscale = (float) ((width * .8) - (offx * 2)) / tree.getMaxHeight(); + Hashtable nameHash = new Hashtable(); - if (top.count == 0) - { - top.count = ((SequenceNode) top.left()).count + - ((SequenceNode) top.right()).count; - } + Hashtable nodeHash = new Hashtable(); - float chunk = (float) (height - (offy)) / top.count; + SequenceNode highlightNode; - pickNode(pickBox, top, chunk, wscale, width, offx, offy); - } + boolean applyToAllViews = false; + + /** + * Creates a new TreeCanvas object. + * + * @param av + * DOCUMENT ME! + * @param tree + * DOCUMENT ME! + * @param scroller + * DOCUMENT ME! + * @param label + * DOCUMENT ME! + */ + public TreeCanvas(TreePanel tp, AlignmentPanel ap, JScrollPane scroller) + { + this.tp = tp; + this.av = ap.av; + this.ap = ap; + font = av.getFont(); + scrollPane = scroller; + addMouseListener(this); + addMouseMotionListener(this); + ToolTipManager.sharedInstance().registerComponent(this); + } - /** - * 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! - */ - public void pickNode(Rectangle pickBox, SequenceNode node, float chunk, - float scale, int width, int offx, int offy) + /** + * DOCUMENT ME! + * + * @param sequence + * DOCUMENT ME! + */ + public void treeSelectionChanged(SequenceI sequence) + { + AlignmentPanel[] aps = getAssociatedPanels(); + + for (int a = 0; a < aps.length; a++) { - if (node == null) - { - return; - } + SequenceGroup selected = aps[a].av.getSelectionGroup(); - 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; - - int ypos = (int) (node.ycount * chunk) + offy; - - if (pickBox.contains(new Point(xend, ypos))) - { - if (node.element() instanceof SequenceI) - { - SequenceI seq = (SequenceI) node.element(); - SequenceGroup sg = av.getSelectionGroup(); - - if (sg != null) - { - sg.addOrRemove(seq, true); - } - } - } - } - else - { - pickNode(pickBox, (SequenceNode) node.left(), chunk, scale, width, - offx, offy); - pickNode(pickBox, (SequenceNode) node.right(), chunk, scale, width, - offx, offy); - } + if (selected == null) + { + selected = new SequenceGroup(); + aps[a].av.setSelectionGroup(selected); + } + + selected.setEndRes(aps[a].av.getAlignment().getWidth() - 1); + selected.addOrRemove(sequence, true); } + } - /** - * DOCUMENT ME! - * - * @param node DOCUMENT ME! - * @param c DOCUMENT ME! - */ - public void setColor(SequenceNode node, Color c) + /** + * DOCUMENT ME! + * + * @param tree + * DOCUMENT ME! + */ + public void setTree(NJTree 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()); + boolean has_placeholders = false; + longestName = ""; + + for (int i = 0; i < leaves.size(); i++) { - if (node == null) - { - return; - } + SequenceNode lf = (SequenceNode) leaves.elementAt(i); - if ((node.left() == null) && (node.right() == null)) - { - node.color = c; + if (lf.isPlaceholder()) + { + has_placeholders = true; + } - if (node.element() instanceof SequenceI) - { - av.setSequenceColour((SequenceI) node.element(), c); - } - } - else - { - node.color = c; - setColor((SequenceNode) node.left(), c); - setColor((SequenceNode) node.right(), c); - } + if (longestName.length() < ((Sequence) lf.element()).getName() + .length()) + { + longestName = TreeCanvas.PLACEHOLDER + + ((Sequence) lf.element()).getName(); + } } - /** - * DOCUMENT ME! - */ - void startPrinting() + setMarkPlaceholders(has_placeholders); + } + + /** + * 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! + */ + public void drawNode(Graphics g, SequenceNode node, float chunk, + float scale, int width, int offx, int offy) + { + if (node == null) { - Thread thread = new Thread(this); - thread.start(); + return; } - // put printing in a thread to avoid painting problems - public void run() + if ((node.left() == null) && (node.right() == null)) { - PrinterJob printJob = PrinterJob.getPrinterJob(); - PageFormat pf = printJob.pageDialog(printJob.defaultPage()); + // Drawing leaf node + float height = node.height; + float dist = node.dist; + + int xstart = (int) ((height - dist) * scale) + offx; + int xend = (int) (height * scale) + offx; - printJob.setPrintable(this, pf); + int ypos = (int) (node.ycount * chunk) + offy; + + if (node.element() instanceof SequenceI) + { + SequenceI seq = (SequenceI) ((SequenceNode) node).element(); - if (printJob.printDialog()) + if (av.getSequenceColour(seq) == Color.white) { - try - { - printJob.print(); - } - catch (Exception PrintException) - { - PrintException.printStackTrace(); - } + g.setColor(Color.black); } - } + else + { + g.setColor(av.getSequenceColour(seq).darker()); + } + } + else + { + g.setColor(Color.black); + } - /** - * DOCUMENT ME! - * - * @param pg DOCUMENT ME! - * @param pf DOCUMENT ME! - * @param pi DOCUMENT ME! - * - * @return DOCUMENT ME! - * - * @throws PrinterException DOCUMENT ME! - */ - public int print(Graphics pg, PageFormat pf, int pi) - throws PrinterException - { - pg.setFont(font); - pg.translate((int) pf.getImageableX(), (int) pf.getImageableY()); + // Draw horizontal line + g.drawLine(xstart, ypos, xend, ypos); - int pwidth = (int) pf.getImageableWidth(); - int pheight = (int) pf.getImageableHeight(); + String nodeLabel = ""; - int noPages = getHeight() / pheight; + if (showDistances && (node.dist > 0)) + { + nodeLabel = new Format("%-.2f").form(node.dist); + } - if (pi > noPages) + if (showBootstrap && node.bootstrap > -1) + { + if (showDistances) { - return Printable.NO_SUCH_PAGE; + nodeLabel = nodeLabel + " : "; } - if (pwidth > getWidth()) - { - pwidth = getWidth(); - } + nodeLabel = nodeLabel + String.valueOf(node.bootstrap); + } - if (fitToWindow) - { - if (pheight > getHeight()) - { - pheight = getHeight(); - } + if (!nodeLabel.equals("")) + { + g.drawString(nodeLabel, xstart + 2, ypos - 2); + } - noPages = 0; - } - else - { - FontMetrics fm = pg.getFontMetrics(font); - int height = fm.getHeight() * nameHash.size(); - pg.translate(0, -pi * pheight); - pg.setClip(0, pi * pheight, pwidth, (pi * pheight) + pheight); - - // translate number of pages, - // height is screen size as this is the - // non overlapping text size - pheight = height; - } + String name = (markPlaceholders && node.isPlaceholder()) ? (PLACEHOLDER + node + .getName()) : node.getName(); - draw(pg, pwidth, pheight); + int charWidth = fm.stringWidth(name) + 3; + int charHeight = font.getSize(); - return Printable.PAGE_EXISTS; - } + Rectangle rect = new Rectangle(xend + 10, ypos - charHeight / 2, + charWidth, charHeight); + + nameHash.put((SequenceI) node.element(), rect); + + // Colour selected leaves differently + SequenceGroup selected = av.getSelectionGroup(); - /** - * DOCUMENT ME! - * - * @param g DOCUMENT ME! - */ - public void paintComponent(Graphics g) + if ((selected != null) + && selected.getSequences(null).contains( + (SequenceI) node.element())) + { + g.setColor(Color.gray); + + g.fillRect(xend + 10, ypos - charHeight / 2, charWidth, charHeight); + g.setColor(Color.white); + } + + g.drawString(name, xend + 10, ypos + fm.getDescent()); + g.setColor(Color.black); + } + else { - super.paintComponent(g); - g.setFont(font); + drawNode(g, (SequenceNode) node.left(), chunk, scale, width, offx, + offy); + drawNode(g, (SequenceNode) node.right(), chunk, scale, width, offx, + offy); + + float height = node.height; + float dist = node.dist; + + int xstart = (int) ((height - dist) * scale) + offx; + int xend = (int) (height * scale) + offx; + int ypos = (int) (node.ycount * chunk) + offy; + + g.setColor(((SequenceNode) 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); + } - if(tree==null) + int ystart = (int) (((SequenceNode) node.left()).ycount * chunk) + + offy; + int yend = (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); + + String nodeLabel = ""; + + if (showDistances && (node.dist > 0)) + { + nodeLabel = new Format("%-.2f").form(node.dist); + } + + if (showBootstrap && node.bootstrap > -1) + { + if (showDistances) { - g.drawString("Calculating tree....", 20, getHeight()/2); + nodeLabel = nodeLabel + " : "; } - else - { - fm = g.getFontMetrics(font); - if (nameHash.size() == 0) - { - repaint(); - } + nodeLabel = nodeLabel + String.valueOf(node.bootstrap); + } - if (fitToWindow || - (!fitToWindow && - (scrollPane.getHeight() > ( (fm.getHeight() * nameHash.size()) + - offy)))) - { - draw(g, scrollPane.getWidth(), scrollPane.getHeight()); - setPreferredSize(null); - } - else - { - setPreferredSize(new Dimension(scrollPane.getWidth(), - fm.getHeight() * nameHash.size())); - draw(g, scrollPane.getWidth(), fm.getHeight() * nameHash.size()); - } + if (!nodeLabel.equals("")) + { + g.drawString(nodeLabel, xstart + 2, ypos - 2); + } + } + } - scrollPane.revalidate(); - } + /** + * 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()) + { + Object ob = keys.nextElement(); + Rectangle rect = (Rectangle) nameHash.get(ob); + + if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) + && (y <= (rect.y + rect.height))) + { + return ob; + } } + keys = nodeHash.keys(); - /** - * DOCUMENT ME! - * - * @param fontSize DOCUMENT ME! - */ - public void setFont(Font font) + while (keys.hasMoreElements()) { - this.font = font; - repaint(); + Object ob = keys.nextElement(); + Rectangle rect = (Rectangle) nodeHash.get(ob); + + if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) + && (y <= (rect.y + rect.height))) + { + return ob; + } } - /** - * DOCUMENT ME! - * - * @param g1 DOCUMENT ME! - * @param width DOCUMENT ME! - * @param height DOCUMENT ME! - */ - public void draw(Graphics g1, int width, int height) + return null; + } + + /** + * DOCUMENT ME! + * + * @param pickBox + * DOCUMENT ME! + */ + public void pickNodes(Rectangle pickBox) + { + int width = getWidth(); + int height = getHeight(); + + SequenceNode top = tree.getTopNode(); + + float wscale = (float) ((width * .8) - (offx * 2)) + / tree.getMaxHeight(); + + if (top.count == 0) { - Graphics2D g2 = (Graphics2D) g1; - g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, - RenderingHints.VALUE_ANTIALIAS_ON); - g2.setColor(Color.white); - g2.fillRect(0, 0, width, height); + top.count = ((SequenceNode) top.left()).count + + ((SequenceNode) top.right()).count; + } - g2.setFont(font); + float chunk = (float) (height - (offy)) / top.count; - offy = font.getSize()+10; + pickNode(pickBox, top, chunk, wscale, width, offx, offy); + } - fm = g2.getFontMetrics(font); + /** + * 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! + */ + public void pickNode(Rectangle pickBox, SequenceNode node, float chunk, + float scale, int width, int offx, int offy) + { + if (node == null) + { + return; + } - labelLength = fm.stringWidth(longestName) + 20; //20 allows for scrollbar + if ((node.left() == null) && (node.right() == null)) + { + float height = node.height; + float dist = node.dist; - float wscale = (float) (width - labelLength - (offx * 2)) / tree.getMaxHeight(); + int xstart = (int) ((height - dist) * scale) + offx; + int xend = (int) (height * scale) + offx; - SequenceNode top = tree.getTopNode(); + int ypos = (int) (node.ycount * chunk) + offy; - if (top.count == 0) + if (pickBox.contains(new Point(xend, ypos))) + { + if (node.element() instanceof SequenceI) { - top.count = ((SequenceNode) top.left()).count + - ((SequenceNode) top.right()).count; + SequenceI seq = (SequenceI) node.element(); + SequenceGroup sg = av.getSelectionGroup(); + + if (sg != null) + { + sg.addOrRemove(seq, true); + } } + } + } + else + { + pickNode(pickBox, (SequenceNode) node.left(), chunk, scale, width, + offx, offy); + pickNode(pickBox, (SequenceNode) node.right(), chunk, scale, width, + offx, offy); + } + } - float chunk = (float) (height - (offy)) / top.count; + /** + * DOCUMENT ME! + * + * @param node + * DOCUMENT ME! + * @param c + * DOCUMENT ME! + */ + public void setColor(SequenceNode node, Color c) + { + if (node == null) + { + return; + } - drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy); + if ((node.left() == null) && (node.right() == null)) // TODO: internal node + { + node.color = c; - if (threshold != 0) + if (node.element() instanceof SequenceI) + { + AlignmentPanel[] aps = getAssociatedPanels(); + if (aps != null) { - if (av.getCurrentTree() == tree) - { - g2.setColor(Color.red); - } - else - { - g2.setColor(Color.gray); - } - - int x = (int) ((threshold * (float) (getWidth() - labelLength - - (2 * offx))) + offx); - - g2.drawLine(x, 0, x, getHeight()); + for (int a = 0; a < aps.length; a++) + { + aps[a].av.setSequenceColour((SequenceI) node.element(), c); + } } + } + } + else + { + node.color = c; + setColor((SequenceNode) node.left(), c); + setColor((SequenceNode) node.right(), c); } + } + + /** + * DOCUMENT ME! + */ + void startPrinting() + { + Thread thread = new Thread(this); + thread.start(); + } + + // put printing in a thread to avoid painting problems + public void run() + { + PrinterJob printJob = PrinterJob.getPrinterJob(); + PageFormat pf = printJob.pageDialog(printJob.defaultPage()); + + printJob.setPrintable(this, pf); - /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! - */ - public void mouseReleased(MouseEvent e) + if (printJob.printDialog()) { + try + { + printJob.print(); + } catch (Exception PrintException) + { + PrintException.printStackTrace(); + } } + } - /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! - */ - public void mouseEntered(MouseEvent e) + /** + * DOCUMENT ME! + * + * @param pg + * DOCUMENT ME! + * @param pf + * DOCUMENT ME! + * @param pi + * DOCUMENT ME! + * + * @return DOCUMENT ME! + * + * @throws PrinterException + * DOCUMENT ME! + */ + public int print(Graphics pg, PageFormat pf, int pi) + throws PrinterException + { + pg.setFont(font); + pg.translate((int) pf.getImageableX(), (int) pf.getImageableY()); + + int pwidth = (int) pf.getImageableWidth(); + int pheight = (int) pf.getImageableHeight(); + + int noPages = getHeight() / pheight; + + if (pi > noPages) { + return Printable.NO_SUCH_PAGE; } - /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! - */ - public void mouseExited(MouseEvent e) + if (pwidth > getWidth()) { + pwidth = getWidth(); } - /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! - */ - public void mouseClicked(MouseEvent evt) + if (fitToWindow) { - if(highlightNode!=null) + if (pheight > getHeight()) { - if (SwingUtilities.isRightMouseButton(evt)) - { - Color col = JColorChooser.showDialog(this, "Select Sub-Tree Colour", - highlightNode.color); + pheight = getHeight(); + } - setColor(highlightNode, col); - } - else - if(evt.getClickCount()>1) - { - tree.swapNodes(highlightNode); - tree.reCount(tree.getTopNode()); - tree.findHeight(tree.getTopNode()); - } - else - { - Vector leaves = new Vector(); - tree.findLeaves(highlightNode, leaves); + noPages = 0; + } + else + { + FontMetrics fm = pg.getFontMetrics(font); + int height = fm.getHeight() * nameHash.size(); + pg.translate(0, -pi * pheight); + pg.setClip(0, pi * pheight, pwidth, (pi * pheight) + pheight); + + // translate number of pages, + // height is screen size as this is the + // non overlapping text size + pheight = height; + } - for (int i = 0; i < leaves.size(); i++) - { - SequenceI seq = - (SequenceI) ( (SequenceNode) leaves.elementAt(i)).element(); - treeSelectionChanged(seq); - } - } + draw(pg, pwidth, pheight); + + return Printable.PAGE_EXISTS; + } + + /** + * DOCUMENT ME! + * + * @param g + * DOCUMENT ME! + */ + public void paintComponent(Graphics g) + { + super.paintComponent(g); + g.setFont(font); + + if (tree == null) + { + g.drawString(MessageManager.getString("label.calculating_tree") + + "....", 20, getHeight() / 2); + } + else + { + fm = g.getFontMetrics(font); - PaintRefresher.Refresh(this, av.getSequenceSetId()); + if (nameHash.size() == 0) + { repaint(); } + + if (fitToWindow + || (!fitToWindow && (scrollPane.getHeight() > ((fm + .getHeight() * nameHash.size()) + offy)))) + { + draw(g, scrollPane.getWidth(), scrollPane.getHeight()); + setPreferredSize(null); + } + else + { + setPreferredSize(new Dimension(scrollPane.getWidth(), + fm.getHeight() * nameHash.size())); + draw(g, scrollPane.getWidth(), fm.getHeight() * nameHash.size()); + } + + scrollPane.revalidate(); } + } + /** + * DOCUMENT ME! + * + * @param fontSize + * DOCUMENT ME! + */ + public void setFont(Font font) + { + this.font = font; + repaint(); + } + + /** + * 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); + 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 + float wscale = (float) (width - labelLength - (offx * 2)) + / tree.getMaxHeight(); - public void mouseMoved(MouseEvent evt) + SequenceNode top = tree.getTopNode(); + + if (top.count == 0) { - av.setCurrentTree(tree); + top.count = ((SequenceNode) top.left()).count + + ((SequenceNode) top.right()).count; + } - Object ob = findElement(evt.getX(), evt.getY()); + float chunk = (float) (height - (offy)) / top.count; - if (ob instanceof SequenceNode) + drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy); + + if (threshold != 0) + { + if (av.getCurrentTree() == tree) { - highlightNode = (SequenceNode) ob; - this.setToolTipText( - "Left click to select leaves" - + "
Double-click to invert leaves" - + "
Right click to change colour"); - repaint(); + g2.setColor(Color.red); + } + else + { + g2.setColor(Color.gray); + } + + int x = (int) ((threshold * (float) (getWidth() - labelLength - (2 * offx))) + offx); + + g2.drawLine(x, 0, x, getHeight()); + } + } + + /** + * DOCUMENT ME! + * + * @param e + * DOCUMENT ME! + */ + public void mouseReleased(MouseEvent e) + { + } + + /** + * DOCUMENT ME! + * + * @param e + * DOCUMENT ME! + */ + public void mouseEntered(MouseEvent e) + { + } + + /** + * DOCUMENT ME! + * + * @param e + * DOCUMENT ME! + */ + public void mouseExited(MouseEvent e) + { + } + /** + * DOCUMENT ME! + * + * @param e + * DOCUMENT ME! + */ + public void mouseClicked(MouseEvent evt) + { + if (highlightNode != null) + { + if (SwingUtilities.isRightMouseButton(evt)) + { + Color col = JColorChooser.showDialog(this, + MessageManager.getString("label.select_subtree_colour"), highlightNode.color); + if (col != null) + { + setColor(highlightNode, col); + } + } + else if (evt.getClickCount() > 1) + { + tree.swapNodes(highlightNode); + tree.reCount(tree.getTopNode()); + tree.findHeight(tree.getTopNode()); } else { - if (highlightNode != null) + Vector leaves = new Vector(); + tree.findLeaves(highlightNode, leaves); + + for (int i = 0; i < leaves.size(); i++) { - highlightNode = null; - setToolTipText(null); - repaint(); + SequenceI seq = (SequenceI) ((SequenceNode) leaves.elementAt(i)) + .element(); + treeSelectionChanged(seq); } + av.sendSelection(); } + + PaintRefresher.Refresh(tp, av.getSequenceSetId()); + repaint(); + } } - public void mouseDragged(MouseEvent ect) - {} + public void mouseMoved(MouseEvent evt) + { + av.setCurrentTree(tree); - /** - * DOCUMENT ME! - * - * @param e DOCUMENT ME! - */ - public void mousePressed(MouseEvent e) + Object ob = findElement(evt.getX(), evt.getY()); + + if (ob instanceof SequenceNode) { - av.setCurrentTree(tree); + highlightNode = (SequenceNode) ob; + this.setToolTipText("" + + MessageManager.getString("label.highlightnode")); + repaint(); - int x = e.getX(); - int y = e.getY(); + } + else + { + if (highlightNode != null) + { + highlightNode = null; + setToolTipText(null); + repaint(); + } + } + } - Object ob = findElement(x, y); + public void mouseDragged(MouseEvent ect) + { + } - if (ob instanceof SequenceI) - { - treeSelectionChanged( (Sequence) ob); - PaintRefresher.Refresh(this, ap.av.getSequenceSetId()); - repaint(); - return; - } - else if( !(ob instanceof SequenceNode) ) - { - // Find threshold - if (tree.getMaxHeight() != 0) - { - threshold = (float) (x - offx) / (float) (getWidth() - - labelLength - (2 * offx)); + /** + * DOCUMENT ME! + * + * @param e + * DOCUMENT ME! + */ + public void mousePressed(MouseEvent e) + { + av.setCurrentTree(tree); - tree.getGroups().removeAllElements(); - tree.groupNodes(tree.getTopNode(), threshold); - setColor(tree.getTopNode(), Color.black); + int x = e.getX(); + int y = e.getY(); - av.setSelectionGroup(null); - av.alignment.deleteAllGroups(); - av.sequenceColours=null; + Object ob = findElement(x, y); - colourGroups(); - } + if (ob instanceof SequenceI) + { + treeSelectionChanged((Sequence) ob); + PaintRefresher.Refresh(tp, ap.av.getSequenceSetId()); + repaint(); + av.sendSelection(); + return; + } + else if (!(ob instanceof SequenceNode)) + { + // Find threshold + if (tree.getMaxHeight() != 0) + { + threshold = (float) (x - offx) + / (float) (getWidth() - labelLength - (2 * offx)); - PaintRefresher.Refresh(this, ap.av.getSequenceSetId()); - repaint(); - } + tree.getGroups().removeAllElements(); + tree.groupNodes(tree.getTopNode(), threshold); + setColor(tree.getTopNode(), Color.black); + AlignmentPanel[] aps = getAssociatedPanels(); + + for (int a = 0; a < aps.length; a++) + { + aps[a].av.setSelectionGroup(null); + aps[a].av.getAlignment().deleteAllGroups(); + aps[a].av.clearSequenceColours(); + } + colourGroups(); + } + PaintRefresher.Refresh(tp, ap.av.getSequenceSetId()); + repaint(); } - void colourGroups() + } + + void colourGroups() + { + AlignmentPanel[] aps = getAssociatedPanels(); + for (int i = 0; i < tree.getGroups().size(); i++) { - for (int i = 0; i < tree.getGroups().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((SequenceNode) tree.getGroups().elementAt(i), col.brighter()); - Vector l = tree.findLeaves( (SequenceNode) tree.getGroups() - .elementAt(i), - new Vector()); + Vector l = tree.findLeaves( + (SequenceNode) tree.getGroups().elementAt(i), new Vector()); - Vector sequences = new Vector(); + Vector sequences = new Vector(); - for (int j = 0; j < l.size(); j++) - { - SequenceI s1 = (SequenceI) ( (SequenceNode) l.elementAt(j)).element(); + for (int j = 0; j < l.size(); j++) + { + SequenceI s1 = (SequenceI) ((SequenceNode) l.elementAt(j)) + .element(); - if (!sequences.contains(s1)) - { - sequences.addElement(s1); - } + if (!sequences.contains(s1)) + { + sequences.addElement(s1); } + } - ColourSchemeI cs = null; + ColourSchemeI cs = null; + SequenceGroup sg = new SequenceGroup(sequences, null, cs, true, true, + false, 0, av.getAlignment().getWidth() - 1); - if (av.getGlobalColourScheme() != null) + if (av.getGlobalColourScheme() != null) + { + if (av.getGlobalColourScheme() instanceof UserColourScheme) { - if (av.getGlobalColourScheme() instanceof UserColourScheme) - { - cs = new UserColourScheme( - ( (UserColourScheme) av.getGlobalColourScheme()).getColours()); - - } - else - cs = ColourSchemeProperty.getColour(sequences, - av.alignment.getWidth(), - ColourSchemeProperty. - getColourName( - av.getGlobalColourScheme())); + cs = new UserColourScheme( + ((UserColourScheme) av.getGlobalColourScheme()) + .getColours()); + } + else + { + cs = ColourSchemeProperty.getColour(sg, ColourSchemeProperty + .getColourName(av.getGlobalColourScheme())); + } + // cs is null if shading is an annotationColourGradient + if (cs != null) + { cs.setThreshold(av.getGlobalColourScheme().getThreshold(), - av.getIgnoreGapsConsensus()); + av.getIgnoreGapsConsensus()); } - - SequenceGroup sg = new SequenceGroup(sequences, - "TreeGroup", cs, true, true, false, - 0, - av.alignment.getWidth() - 1); - - if (av.getGlobalColourScheme() != null - && av.getGlobalColourScheme().conservationApplied()) + } + sg.cs = cs; + // sg.recalcConservation(); + sg.setName("JTreeGroup:" + sg.hashCode()); + sg.setIdColour(col); + for (int a = 0; a < aps.length; a++) + { + 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()); + ResidueProperties.propHash, 3, sg.getSequences(null), + sg.getStartRes(), sg.getEndRes()); c.calculate(); - c.verdict(false, av.ConsPercGaps); + c.verdict(false, aps[a].av.getConsPercGaps()); sg.cs.setConservation(c); } - av.alignment.addGroup(sg); + aps[a].av.getAlignment().addGroup(new SequenceGroup(sg)); } - } - - /** - * DOCUMENT ME! - * - * @param state DOCUMENT ME! - */ - public void setShowDistances(boolean state) + // notify the panel to redo any group specific stuff. + for (int a = 0; a < aps.length; a++) { - this.showDistances = state; - repaint(); + aps[a].updateAnnotation(); + // TODO: JAL-868 - need to ensure view colour change message is broadcast + // to any Jmols listening in } - /** - * DOCUMENT ME! - * - * @param state DOCUMENT ME! - */ - public void setShowBootstrap(boolean state) + } + + /** + * DOCUMENT ME! + * + * @param state + * DOCUMENT ME! + */ + public void setShowDistances(boolean state) + { + this.showDistances = state; + repaint(); + } + + /** + * DOCUMENT ME! + * + * @param state + * DOCUMENT ME! + */ + public void setShowBootstrap(boolean state) + { + this.showBootstrap = state; + repaint(); + } + + /** + * DOCUMENT ME! + * + * @param state + * DOCUMENT ME! + */ + public void setMarkPlaceholders(boolean state) + { + this.markPlaceholders = state; + repaint(); + } + + AlignmentPanel[] getAssociatedPanels() + { + if (applyToAllViews) { - this.showBootstrap = state; - repaint(); + return PaintRefresher.getAssociatedPanels(av.getSequenceSetId()); } - - /** - * DOCUMENT ME! - * - * @param state DOCUMENT ME! - */ - public void setMarkPlaceholders(boolean state) + else { - this.markPlaceholders = state; - repaint(); + return new AlignmentPanel[] + { ap }; } + } }