X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FTreeCanvas.java;h=eea181f476701e4d2a58abcf7b5068100d337692;hb=eee02848424734a9466de35ff19636377652e59b;hp=f9cb39c590c83302c652ba4944540490f819efa6;hpb=8d1b94fdb228e81f3deb79bff3c34f5f68db5f07;p=jalview.git diff --git a/src/jalview/gui/TreeCanvas.java b/src/jalview/gui/TreeCanvas.java index f9cb39c..eea181f 100755 --- a/src/jalview/gui/TreeCanvas.java +++ b/src/jalview/gui/TreeCanvas.java @@ -1,506 +1,816 @@ +/* + * Jalview - A Sequence Alignment Editor and Viewer + * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA + */ package jalview.gui; - import jalview.analysis.*; + import jalview.datamodel.*; -import jalview.util.*; + import jalview.schemes.*; -import javax.swing.*; + +import jalview.util.*; + import java.awt.*; import java.awt.event.*; -import java.util.*; import java.awt.print.*; -public class TreeCanvas extends JPanel implements MouseListener, Runnable, Printable +import java.util.*; + +import javax.swing.*; + + +/** + * DOCUMENT ME! + * + * @author $author$ + * @version $Revision$ + */ +public class TreeCanvas extends JPanel implements MouseListener, Runnable, + Printable { - NJTree tree; - JScrollPane scrollPane; - AlignViewport av; - public static final String PLACEHOLDER=" * "; - Font font; - int fontSize = 12; - - boolean fitToWindow = true; - boolean showDistances = false; - boolean showBootstrap = false; - boolean markPlaceholders = false; - - int offx = 20; - int offy = 20; - - float threshold; - - String longestName; - int labelLength=-1; - - //RubberbandRectangle rubberband; - - Vector listeners; - - Hashtable nameHash = new Hashtable(); - Hashtable nodeHash = new Hashtable(); - - public TreeCanvas(AlignViewport av, NJTree tree, JScrollPane scroller, String label) - { - this.av = av; - this.tree = tree; - scrollPane = scroller; - addMouseListener(this); - tree.findHeight(tree.getTopNode()); - longestName = label; - - PaintRefresher.Register(this); - } - public void TreeSelectionChanged(Sequence sequence) - { - SequenceGroup selected = av.getSelectionGroup(); - if(selected == null) + /** DOCUMENT ME!! */ + public static final String PLACEHOLDER = " * "; + NJTree tree; + JScrollPane scrollPane; + AlignViewport av; + Font font; + int fontSize; + boolean fitToWindow = true; + boolean showDistances = false; + boolean showBootstrap = false; + boolean markPlaceholders = false; + int offx = 20; + int offy = 20; + float threshold; + String longestName; + int labelLength = -1; + + //RubberbandRectangle rubberband; + Vector listeners; + Hashtable nameHash = new Hashtable(); + Hashtable nodeHash = new Hashtable(); + + /** + * Creates a new TreeCanvas object. + * + * @param av DOCUMENT ME! + * @param tree DOCUMENT ME! + * @param scroller DOCUMENT ME! + * @param label DOCUMENT ME! + */ + public TreeCanvas(AlignViewport av, JScrollPane scroller) { - selected = new SequenceGroup(); - av.setSelectionGroup(selected); + this.av = av; + font = av.getFont(); + scrollPane = scroller; + addMouseListener(this); + PaintRefresher.Register(this, av.alignment); } - selected.setEndRes(av.alignment.getWidth()); - selected.addOrRemove(sequence); + /** + * DOCUMENT ME! + * + * @param sequence DOCUMENT ME! + */ + public void TreeSelectionChanged(Sequence sequence) + { + SequenceGroup selected = av.getSelectionGroup(); - PaintRefresher.Refresh(this); - repaint(); - } + if (selected == null) + { + selected = new SequenceGroup(); + av.setSelectionGroup(selected); + } + selected.setEndRes(av.alignment.getWidth()-1); + selected.addOrRemove(sequence, true); + PaintRefresher.Refresh(this, av.alignment); + repaint(); + } - public void setTree(NJTree tree) { - this.tree = tree; - tree.findHeight(tree.getTopNode()); - } + /** + * DOCUMENT ME! + * + * @param tree DOCUMENT ME! + */ + public void setTree(NJTree tree) + { + this.tree = tree; + tree.findHeight(tree.getTopNode()); - public void drawNode(Graphics g,SequenceNode node, float chunk, float scale, int width,int offx, int offy) { - if (node == null) { - return; - } + // Now have to calculate longest name based on the leaves + Vector leaves = tree.findLeaves(tree.getTopNode(), new Vector()); + boolean has_placeholders = false; + longestName = ""; - if (node.left() == null && node.right() == null) { - // Drawing leaf node + for (int i = 0; i < leaves.size(); i++) + { + SequenceNode lf = (SequenceNode) leaves.elementAt(i); - float height = node.height; - float dist = node.dist; + if (lf.isPlaceholder()) + { + has_placeholders = true; + } - int xstart = (int)((height-dist)*scale) + offx; - int xend = (int)(height*scale) + offx; + if (longestName.length() < ( (Sequence) lf.element()).getName() + .length()) + { + longestName = TreeCanvas.PLACEHOLDER + + ( (Sequence) lf.element()).getName(); + } + } - int ypos = (int)(node.ycount * chunk) + offy; + setMarkPlaceholders(has_placeholders); + } - if (node.element() instanceof SequenceI) - { - if ( ( (SequenceI) ( (SequenceNode) node).element()).getColor() == Color.white) - { - g.setColor(Color.black); - } - else - g.setColor( ( (SequenceI) ( (SequenceNode) node).element()).getColor(). - darker()); + /** + * 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; + } - } - else - g.setColor(Color.black); + 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) + { + if (((SequenceI) ((SequenceNode) node).element()).getColor() == Color.white) + { + g.setColor(Color.black); + } + else + { + g.setColor(((SequenceI) ((SequenceNode) node).element()).getColor() + .darker()); + } + } + else + { + g.setColor(Color.black); + } + // Draw horizontal line + g.drawLine(xstart, ypos, xend, ypos); - // Draw horizontal line - g.drawLine(xstart,ypos,xend,ypos); + String nodeLabel = ""; - String nodeLabel = ""; - if (showDistances && node.dist > 0) { - nodeLabel = new Format("%5.2f").form(node.dist); - } - if (showBootstrap) { - if (showDistances) { - nodeLabel = nodeLabel + " : "; - } - nodeLabel = nodeLabel + String.valueOf(node.getBootstrap()); - } - if (! nodeLabel.equals("")) { - g.drawString(nodeLabel,xstart,ypos - 10); - } + if (showDistances && (node.dist > 0)) + { + nodeLabel = new Format("%5.2f").form(node.dist); + } - String name = (markPlaceholders && node.isPlaceholder()) ? (PLACEHOLDER+node.getName()) : node.getName(); - FontMetrics fm = g.getFontMetrics(font); - int charWidth = fm.stringWidth(name) + 3; - int charHeight = fm.getHeight(); + if (showBootstrap) + { + if (showDistances) + { + nodeLabel = nodeLabel + " : "; + } - Rectangle rect = new Rectangle(xend+20,ypos-charHeight, - charWidth,charHeight); + nodeLabel = nodeLabel + String.valueOf(node.getBootstrap()); + } - nameHash.put((SequenceI)node.element(),rect); + if (!nodeLabel.equals("")) + { + g.drawString(nodeLabel, xstart, ypos - 10); + } - // Colour selected leaves differently - SequenceGroup selected = av.getSelectionGroup(); - if (selected!=null && selected.sequences.contains((SequenceI)node.element())) { - g.setColor(Color.gray); + String name = (markPlaceholders && node.isPlaceholder()) + ? (PLACEHOLDER + node.getName()) : node.getName(); + FontMetrics fm = g.getFontMetrics(font); + int charWidth = fm.stringWidth(name) + 3; + int charHeight = fm.getHeight(); - g.fillRect(xend + 10, ypos - charHeight + 3,charWidth,charHeight); - g.setColor(Color.white); - } - g.drawString(name,xend+10,ypos); - 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); + Rectangle rect = new Rectangle(xend + 20, ypos - charHeight, + charWidth, charHeight); - float height = node.height; - float dist = node.dist; + nameHash.put((SequenceI) node.element(), rect); - int xstart = (int)((height-dist)*scale) + offx; - int xend = (int)(height *scale) + offx; - int ypos = (int)(node.ycount *chunk) + offy; + // Colour selected leaves differently + SequenceGroup selected = av.getSelectionGroup(); - g.setColor(((SequenceNode)node).color.darker()); + if ((selected != null) && + selected.sequences.contains((SequenceI) node.element())) + { + g.setColor(Color.gray); - // Draw horizontal line - g.drawLine(xstart,ypos,xend,ypos); - g.fillRect(xend-2, ypos-2, 4,4); + g.fillRect(xend + 10, ypos - charHeight + 3, charWidth, + charHeight); + g.setColor(Color.white); + } - int ystart = (int)(((SequenceNode)node.left()) .ycount * chunk) + offy; - int yend = (int)(((SequenceNode)node.right()).ycount * chunk) + offy; + g.drawString(name, xend + 10, ypos); + 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); - Rectangle pos = new Rectangle(xend-2,ypos-2,5,5); - nodeHash.put(node,pos); + float height = node.height; + float dist = node.dist; - g.drawLine((int)(height*scale) + offx, ystart, - (int)(height*scale) + offx, yend); + int xstart = (int) ((height - dist) * scale) + offx; + int xend = (int) (height * scale) + offx; + int ypos = (int) (node.ycount * chunk) + offy; - if (showDistances && node.dist > 0) { - g.drawString(new Format("%5.2f").form(node.dist),xstart,ypos - 5); - } + g.setColor(((SequenceNode) node).color.darker()); - } - } - public Object findElement(int x, int y) { - Enumeration keys = nameHash.keys(); + // Draw horizontal line + g.drawLine(xstart, ypos, xend, ypos); + g.fillRect(xend - 2, ypos - 2, 4, 4); - while (keys.hasMoreElements()) { - Object ob = keys.nextElement(); - Rectangle rect = (Rectangle)nameHash.get(ob); + int ystart = (int) (((SequenceNode) node.left()).ycount * chunk) + + offy; + int yend = (int) (((SequenceNode) node.right()).ycount * chunk) + + offy; - if (x >= rect.x && x <= (rect.x + rect.width) && - y >= rect.y && y <= (rect.y + rect.height)) { - return ob; - } - } - keys = nodeHash.keys(); + Rectangle pos = new Rectangle(xend - 2, ypos - 2, 5, 5); + nodeHash.put(node, pos); - while (keys.hasMoreElements()) { - Object ob = keys.nextElement(); - Rectangle rect = (Rectangle)nodeHash.get(ob); + g.drawLine((int) (height * scale) + offx, ystart, + (int) (height * scale) + offx, yend); - if (x >= rect.x && x <= (rect.x + rect.width) && - y >= rect.y && y <= (rect.y + rect.height)) { - return ob; + if (showDistances && (node.dist > 0)) + { + g.drawString(new Format("%5.2f").form(node.dist), xstart, + ypos - 5); } + } } - return null; - } + /** + * DOCUMENT ME! + * + * @param x DOCUMENT ME! + * @param y DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public Object findElement(int x, int y) + { + Enumeration keys = nameHash.keys(); - public void pickNodes(Rectangle pickBox) { - int width = getWidth(); - int height = getHeight(); + while (keys.hasMoreElements()) + { + Object ob = keys.nextElement(); + Rectangle rect = (Rectangle) nameHash.get(ob); - SequenceNode top = tree.getTopNode(); + if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) && + (y <= (rect.y + rect.height))) + { + return ob; + } + } - float wscale = (float)(width*.8-offx*2)/tree.getMaxHeight() -; - if (top.count == 0) { - top.count = ((SequenceNode)top.left()).count + ((SequenceNode)top.right()).count ; - } - float chunk = (float)(height-offy*2)/top.count; + keys = nodeHash.keys(); - pickNode(pickBox,top,chunk,wscale,width,offx,offy); - } + while (keys.hasMoreElements()) + { + Object ob = keys.nextElement(); + Rectangle rect = (Rectangle) nodeHash.get(ob); - public void pickNode(Rectangle pickBox, SequenceNode node, float chunk, float scale, int width,int offx, int offy) { - if (node == null) { - return; + if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) && + (y <= (rect.y + rect.height))) + { + return ob; + } + } + + return null; } - if (node.left() == null && node.right() == null) { - float height = node.height; - float dist = node.dist; + /** + * DOCUMENT ME! + * + * @param pickBox DOCUMENT ME! + */ + public void pickNodes(Rectangle pickBox) + { + int width = getWidth(); + int height = getHeight(); - int xstart = (int)((height-dist)*scale) + offx; - int xend = (int)(height*scale) + offx; + SequenceNode top = tree.getTopNode(); - int ypos = (int)(node.ycount * chunk) + offy; + float wscale = (float) ((width * .8) - (offx * 2)) / tree.getMaxHeight(); - 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); + if (top.count == 0) + { + top.count = ((SequenceNode) top.left()).count + + ((SequenceNode) top.right()).count; } - } - } else { - pickNode(pickBox,(SequenceNode)node.left(), chunk,scale,width,offx,offy); - pickNode(pickBox,(SequenceNode)node.right(),chunk,scale,width,offx,offy); - } - } - - public void setColor(SequenceNode node, Color c) - { - if (node == null) - return; - if (node.left() == null && node.right() == null) - { - node.color = c; + float chunk = (float) (height - (offy * 2)) / top.count; - if (node.element() instanceof SequenceI) - ((SequenceI)node.element()).setColor(c); - } else - { - node.color = c; - setColor((SequenceNode)node.left(),c); - setColor((SequenceNode)node.right(),c); + pickNode(pickBox, top, chunk, wscale, width, offx, offy); } - } - - 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); - if (printJob.printDialog()) - { - try - { - printJob.print(); - } - catch (Exception PrintException) - { - PrintException.printStackTrace(); - } - } - } + /** + * 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; + } - public int print(Graphics pg, PageFormat pf, int pi) throws PrinterException - { + if ((node.left() == null) && (node.right() == null)) + { + float height = node.height; + float dist = node.dist; - pg.setFont(font); - pg.translate((int)pf.getImageableX(), (int)pf.getImageableY()); - int pwidth = (int) pf.getImageableWidth(); - int pheight = (int) pf.getImageableHeight(); + int xstart = (int) ((height - dist) * scale) + offx; + int xend = (int) (height * scale) + offx; - int noPages = getHeight() / pheight; - if(pi>noPages) - return Printable.NO_SUCH_PAGE; + 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 (pwidth > getWidth()) - pwidth = getWidth(); + 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(fitToWindow) + /** + * DOCUMENT ME! + * + * @param node DOCUMENT ME! + * @param c DOCUMENT ME! + */ + public void setColor(SequenceNode node, Color c) { - if (pheight > getHeight()) - pheight = getHeight(); + if (node == null) + { + return; + } - noPages = 0; - } - else - { + if ((node.left() == null) && (node.right() == null)) + { + node.color = c; - 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; + if (node.element() instanceof SequenceI) + { + ((SequenceI) node.element()).setColor(c); + } + } + else + { + node.color = c; + setColor((SequenceNode) node.left(), c); + setColor((SequenceNode) node.right(), c); + } } - draw(pg, pwidth, pheight); - - return Printable.PAGE_EXISTS; - - } - - public void paintComponent(Graphics g) - { + /** + * DOCUMENT ME! + */ + void startPrinting() + { + Thread thread = new Thread(this); + thread.start(); + } - font = new Font("Verdana",Font.PLAIN,fontSize); - g.setFont(font); + // put printing in a thread to avoid painting problems + public void run() + { + PrinterJob printJob = PrinterJob.getPrinterJob(); + PageFormat pf = printJob.pageDialog(printJob.defaultPage()); - FontMetrics fm = g.getFontMetrics(font); + printJob.setPrintable(this, pf); - if(nameHash.size()==0) - repaint(); + if (printJob.printDialog()) + { + try + { + printJob.print(); + } + catch (Exception PrintException) + { + PrintException.printStackTrace(); + } + } + } + /** + * 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()); - 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()); - } + int pwidth = (int) pf.getImageableWidth(); + int pheight = (int) pf.getImageableHeight(); - scrollPane.revalidate(); - } - public int getFontSize() { - return fontSize; - } - public void setFontSize(int fontSize) { - this.fontSize = fontSize; - repaint(); - } - public void draw(Graphics g1, int width, int height) { + int noPages = getHeight() / pheight; - Graphics2D g2 = (Graphics2D)g1; - g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON); - g2.setColor(Color.white); - g2.fillRect(0,0,width,height); + if (pi > noPages) + { + return Printable.NO_SUCH_PAGE; + } + if (pwidth > getWidth()) + { + pwidth = getWidth(); + } - labelLength = g2.getFontMetrics(font).stringWidth(longestName)+ 20;//20 allows for scrollbar + if (fitToWindow) + { + if (pheight > getHeight()) + { + pheight = getHeight(); + } - float wscale =(float)(width - labelLength -offx*2)/tree.getMaxHeight(); + 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; + } - SequenceNode top = tree.getTopNode(); + draw(pg, pwidth, pheight); - if (top.count == 0) { - top.count = ((SequenceNode)top.left()).count + ((SequenceNode)top.right()).count ; - } - float chunk = (float)(height-offy*2)/top.count ; + return Printable.PAGE_EXISTS; + } - drawNode(g2,tree.getTopNode(),chunk,wscale,width,offx,offy); + /** + * DOCUMENT ME! + * + * @param g DOCUMENT ME! + */ + public void paintComponent(Graphics g) + { + super.paintComponent(g); + g.setFont(font); - if (threshold != 0) - { - if(av.getCurrentTree() == tree) - g2.setColor(Color.red); + if(tree==null) + { + g.drawString("Calculating tree....", 20, getHeight()/2); + } else - g2.setColor(Color.gray); - - int x = (int)( threshold * (float)(getWidth()-labelLength - 2*offx) +offx ) ; + { + FontMetrics fm = g.getFontMetrics(font); - g2.drawLine(x,0,x,getHeight()); - } + 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()); + } - public void mouseReleased(MouseEvent e) { } - public void mouseEntered(MouseEvent e) { } - public void mouseExited(MouseEvent e) { } - public void mouseClicked(MouseEvent e) { - } + scrollPane.revalidate(); + } + } - public void mousePressed(MouseEvent e) { + /** + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public int getFontSize() + { + return fontSize; + } - av.setCurrentTree(tree); + /** + * DOCUMENT ME! + * + * @param fontSize DOCUMENT ME! + */ + public void setFontSize(int fontSize) + { + this.fontSize = fontSize; + font = new Font(font.getName(), font.getStyle(), fontSize); + repaint(); + } - int x = e.getX(); - int y = e.getY(); + /** + * 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); - Object ob = findElement(x,y); + g2.setFont(font); - if (ob instanceof SequenceI) - { - TreeSelectionChanged((Sequence)ob); - repaint(); - return; + labelLength = g2.getFontMetrics(font).stringWidth(longestName) + 20; //20 allows for scrollbar - } else if (ob instanceof SequenceNode) { - SequenceNode tmpnode = (SequenceNode)ob; - tree.swapNodes(tmpnode); - tree.reCount(tree.getTopNode()); - tree.findHeight(tree.getTopNode()); - } else { - // Find threshold + float wscale = (float) (width - labelLength - (offx * 2)) / tree.getMaxHeight(); - if (tree.getMaxHeight() != 0) { - threshold = (float)(x - offx)/(float)(getWidth()-labelLength - 2*offx); + SequenceNode top = tree.getTopNode(); - tree.getGroups().removeAllElements(); - tree.groupNodes(tree.getTopNode(),threshold); - setColor(tree.getTopNode(),Color.black); + if (top.count == 0) + { + top.count = ((SequenceNode) top.left()).count + + ((SequenceNode) top.right()).count; + } - av.setSelectionGroup(null); - av.alignment.deleteAllGroups(); + float chunk = (float) (height - (offy * 2)) / top.count; - for (int i=0; i < tree.getGroups().size(); i++) - { + drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy); - Color col = new Color((int)(Math.random()*255), - (int)(Math.random()*255), - (int)(Math.random()*255)); - setColor((SequenceNode)tree.getGroups().elementAt(i),col.brighter()); + if (threshold != 0) + { + if (av.getCurrentTree() == tree) + { + g2.setColor(Color.red); + } + else + { + g2.setColor(Color.gray); + } - Vector l = tree.findLeaves((SequenceNode)tree.getGroups().elementAt(i),new Vector()); - SequenceGroup sg = null; - for (int j = 0; j < l.size(); j++) - { - SequenceNode sn = (SequenceNode) l.elementAt(j); - if(sg==null) - sg = new SequenceGroup("TreeGroup", av.getGlobalColourScheme(), true, true,false,0,av.alignment.getWidth()); + int x = (int) ((threshold * (float) (getWidth() - labelLength - + (2 * offx))) + offx); - sg.addSequence( (Sequence) sn.element()); - } + g2.drawLine(x, 0, x, getHeight()); + } + } - if (av.getGlobalColourScheme() instanceof ConservationColourScheme) - { - ConservationColourScheme ccs = (ConservationColourScheme) av.getGlobalColourScheme(); - Conservation c = new Conservation("Group", - ResidueProperties.propHash, 3, - sg.sequences, sg.getStartRes(), - sg.getEndRes()); + /** + * DOCUMENT ME! + * + * @param e DOCUMENT ME! + */ + public void mouseReleased(MouseEvent e) + { + } - c.calculate(); - c.verdict(false, av.ConsPercGaps); - ccs = new ConservationColourScheme(c, ccs.cs); + /** + * DOCUMENT ME! + * + * @param e DOCUMENT ME! + */ + public void mouseEntered(MouseEvent e) + { + } - sg.cs = ccs; + /** + * DOCUMENT ME! + * + * @param e DOCUMENT ME! + */ + public void mouseExited(MouseEvent e) + { + } - } + /** + * DOCUMENT ME! + * + * @param e DOCUMENT ME! + */ + public void mouseClicked(MouseEvent e) + { + } + /** + * DOCUMENT ME! + * + * @param e DOCUMENT ME! + */ + public void mousePressed(MouseEvent e) + { + av.setCurrentTree(tree); + int x = e.getX(); + int y = e.getY(); - av.alignment.addGroup(sg); + Object ob = findElement(x, y); - } - } - } + if (ob instanceof SequenceI) + { + TreeSelectionChanged((Sequence) ob); + repaint(); - PaintRefresher.Refresh(this); - repaint(); + return; + } + else if (ob instanceof SequenceNode) + { + SequenceNode tmpnode = (SequenceNode) ob; + tree.swapNodes(tmpnode); + tree.reCount(tree.getTopNode()); + tree.findHeight(tree.getTopNode()); + } + else + { + // Find threshold + if (tree.getMaxHeight() != 0) + { + threshold = (float) (x - offx) / (float) (getWidth() - + labelLength - (2 * offx)); + + tree.getGroups().removeAllElements(); + tree.groupNodes(tree.getTopNode(), threshold); + setColor(tree.getTopNode(), Color.black); + + av.setSelectionGroup(null); + av.alignment.deleteAllGroups(); + + 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()); + + Vector l = tree.findLeaves((SequenceNode) tree.getGroups() + .elementAt(i), + new Vector()); + + Vector sequences = new Vector(); + + for (int j = 0; j < l.size(); j++) + { + SequenceI s1 = (SequenceI) ((SequenceNode) l.elementAt(j)).element(); + + if (!sequences.contains(s1)) + { + sequences.addElement(s1); + } + } + + ColourSchemeI cs = null; + + if (av.getGlobalColourScheme() != null) + { + 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.setThreshold(av.getGlobalColourScheme().getThreshold(), + av.getIgnoreGapsConsensus()); + } + + + SequenceGroup sg = new SequenceGroup(sequences, + "TreeGroup", cs, true, true, false, 0, + av.alignment.getWidth()); + + + if ( av.getGlobalColourScheme()!=null + && av.getGlobalColourScheme().conservationApplied()) + { + Conservation c = new Conservation("Group", + ResidueProperties.propHash, 3, sg.sequences, + sg.getStartRes(), sg.getEndRes()); + + c.calculate(); + c.verdict(false, av.ConsPercGaps); + sg.cs.setConservation(c); + } + + av.alignment.addGroup(sg); + } + } + } - } + PaintRefresher.Refresh(this, av.alignment); + repaint(); + } - public void setShowDistances(boolean state) { + /** + * DOCUMENT ME! + * + * @param state DOCUMENT ME! + */ + public void setShowDistances(boolean state) + { this.showDistances = state; repaint(); } - public void setShowBootstrap(boolean state) { - this.showBootstrap = state; - repaint(); - } - public void setMarkPlaceholders(boolean state) { - this.markPlaceholders = 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(); + } } -