X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FTreeCanvas.java;h=a9c3ae0153fbf8b4b910b2969b9e5ac14c260d76;hb=6a698ae1f2a42d84b287f3095a9244c711e0ce2e;hp=5ebf323be7e10d75b14ad580c163c8109fbb72e7;hpb=99c58ee0ae2a848f982552e53feaf6d5cb9925e5;p=jalview.git

diff --git a/src/jalview/gui/TreeCanvas.java b/src/jalview/gui/TreeCanvas.java
index 5ebf323..a9c3ae0 100755
--- a/src/jalview/gui/TreeCanvas.java
+++ b/src/jalview/gui/TreeCanvas.java
@@ -1,525 +1,864 @@
 /*
-* 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
-*/
-
+ * 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
+ * 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
-{
-  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;
+import java.util.*;
 
-  Hashtable nameHash = new Hashtable();
-  Hashtable nodeHash = new Hashtable();
+import javax.swing.*;
 
-  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!
+ *
+ * @author $author$
+ * @version $Revision$
+ */
+public class TreeCanvas extends JPanel implements MouseListener, Runnable,
+    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)
     {
-      selected = new SequenceGroup();
-      av.setSelectionGroup(selected);
+        this.av = ap.av;
+        this.ap = ap;
+        font = av.getFont();
+        scrollPane = scroller;
+        addMouseListener(this);
+        addMouseMotionListener(this);
+        PaintRefresher.Register(this, ap.av.getSequenceSetId());
     }
 
-    selected.setEndRes(av.alignment.getWidth());
-    selected.addOrRemove(sequence);
-
-
-    PaintRefresher.Refresh(this);
-    repaint();
- }
-
 
+    /**
+     * DOCUMENT ME!
+     *
+     * @param sequence DOCUMENT ME!
+     */
+    public void treeSelectionChanged(SequenceI sequence)
+    {
+        SequenceGroup selected = av.getSelectionGroup();
 
-  public void setTree(NJTree tree) {
-    this.tree = tree;
-    tree.findHeight(tree.getTopNode());
-  }
+        if (selected == null)
+        {
+            selected = new SequenceGroup();
+            av.setSelectionGroup(selected);
+        }
 
-    public void drawNode(Graphics g,SequenceNode node, float chunk, float scale, int width,int offx, int offy) {
-    if (node == null) {
-      return;
+        selected.setEndRes(av.alignment.getWidth()-1);
+        selected.addOrRemove(sequence, 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;
+    /**
+     * DOCUMENT ME!
+     *
+     * @param tree DOCUMENT ME!
+     */
+    public void setTree(NJTree tree)
+    {
+        this.tree = tree;
+        tree.findHeight(tree.getTopNode());
 
-      int ypos = (int)(node.ycount * chunk) + offy;
+        // 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.element() instanceof SequenceI)
-      {
-       if ( ( (SequenceI) ( (SequenceNode) node).element()).getColor() == Color.white)
-       {
-         g.setColor(Color.black);
-       }
-       else
-        g.setColor( ( (SequenceI) ( (SequenceNode) node).element()).getColor().
-                   darker());
+        for (int i = 0; i < leaves.size(); i++)
+        {
+          SequenceNode lf = (SequenceNode) leaves.elementAt(i);
 
-      }
-      else
-          g.setColor(Color.black);
+          if (lf.isPlaceholder())
+          {
+            has_placeholders = true;
+          }
 
+          if (longestName.length() < ( (Sequence) lf.element()).getName()
+              .length())
+          {
+            longestName = TreeCanvas.PLACEHOLDER +
+                ( (Sequence) lf.element()).getName();
+          }
+        }
 
-      // Draw horizontal line
-      g.drawLine(xstart,ypos,xend,ypos);
+        setMarkPlaceholders(has_placeholders);
+    }
 
-      String nodeLabel = "";
-      if (showDistances && node.dist > 0) {
-        nodeLabel = new Format("%5.2f").form(node.dist);
-      }
-      if (showBootstrap) {
-        if (showDistances) {
-          nodeLabel = nodeLabel + " : ";
+    /**
+     * 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;
         }
-        nodeLabel = nodeLabel + String.valueOf(node.getBootstrap());
-      }
-      if (! nodeLabel.equals("")) {
-      g.drawString(nodeLabel,xstart,ypos - 10);
-      }
 
-      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 ((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);
+            }
 
-      Rectangle rect = new Rectangle(xend+20,ypos-charHeight,
-                                     charWidth,charHeight);
+            // Draw horizontal line
+            g.drawLine(xstart, ypos, xend, ypos);
 
-      nameHash.put((SequenceI)node.element(),rect);
+            String nodeLabel = "";
 
-      // Colour selected leaves differently
-      SequenceGroup selected = av.getSelectionGroup();
-      if (selected!=null && selected.sequences.contains((SequenceI)node.element())) {
-        g.setColor(Color.gray);
+            if (showDistances && (node.dist > 0))
+            {
+                nodeLabel = new Format("%-.2f").form(node.dist);
+            }
 
-        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);
+            if (showBootstrap)
+            {
+                if (showDistances)
+                {
+                    nodeLabel = nodeLabel + " : ";
+                }
 
-      float height = node.height;
-      float dist   = node.dist;
+                nodeLabel = nodeLabel + String.valueOf(node.getBootstrap());
+            }
 
-      int xstart = (int)((height-dist)*scale) + offx;
-      int xend   = (int)(height       *scale) + offx;
-      int ypos   = (int)(node.ycount  *chunk) + offy;
+            if (!nodeLabel.equals(""))
+            {
+                g.drawString(nodeLabel, xstart+2, ypos - 2);
+            }
 
-      g.setColor(((SequenceNode)node).color.darker());
+            String name = (markPlaceholders && node.isPlaceholder())
+                ? (PLACEHOLDER + node.getName()) : node.getName();
 
-      // Draw horizontal line
-      g.drawLine(xstart,ypos,xend,ypos);
-      g.fillRect(xend-2, ypos-2, 4,4);
+            int charWidth = fm.stringWidth(name) + 3;
+            int charHeight = font.getSize();
 
-      int ystart = (int)(((SequenceNode)node.left()) .ycount * chunk) + offy;
-      int yend   = (int)(((SequenceNode)node.right()).ycount * chunk) + offy;
+            Rectangle rect = new Rectangle(xend+10, ypos-charHeight/2,
+                    charWidth, charHeight);
 
-      Rectangle pos = new Rectangle(xend-2,ypos-2,5,5);
-      nodeHash.put(node,pos);
+            nameHash.put((SequenceI) node.element(), rect);
 
-      g.drawLine((int)(height*scale) + offx, ystart,
-                 (int)(height*scale) + offx, yend);
+            // Colour selected leaves differently
+            SequenceGroup selected = av.getSelectionGroup();
 
-      if (showDistances && node.dist > 0) {
-        g.drawString(new Format("%5.2f").form(node.dist),xstart,ypos - 5);
-      }
+            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);
+            }
+        }
     }
-  }
-  public Object findElement(int x, int y) {
-       Enumeration keys = nameHash.keys();
 
-    while (keys.hasMoreElements()) {
+    /**
+     * 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);
+            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;
+            if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) &&
+                    (y <= (rect.y + rect.height)))
+            {
+                return ob;
             }
-  }
-  keys = nodeHash.keys();
+        }
+
+        keys = nodeHash.keys();
 
-    while (keys.hasMoreElements()) {
+        while (keys.hasMoreElements())
+        {
             Object ob = keys.nextElement();
-            Rectangle rect = (Rectangle)nodeHash.get(ob);
+            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;
+            if ((x >= rect.x) && (x <= (rect.x + rect.width)) && (y >= rect.y) &&
+                    (y <= (rect.y + rect.height)))
+            {
+                return ob;
             }
+        }
+
+        return null;
     }
-      return null;
 
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param pickBox DOCUMENT ME!
+     */
+    public void pickNodes(Rectangle pickBox)
+    {
+        int width = getWidth();
+        int height = getHeight();
 
-  public void pickNodes(Rectangle pickBox) {
-    int width  = getWidth();
-    int height = getHeight();
+        SequenceNode top = tree.getTopNode();
 
-    SequenceNode top = tree.getTopNode();
+        float wscale = (float) ((width * .8) - (offx * 2)) / tree.getMaxHeight();
 
-    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;
+        if (top.count == 0)
+        {
+            top.count = ((SequenceNode) top.left()).count +
+                ((SequenceNode) top.right()).count;
+        }
 
-    pickNode(pickBox,top,chunk,wscale,width,offx,offy);
-  }
+        float chunk = (float) (height - (offy)) / top.count;
 
-  public void pickNode(Rectangle pickBox, SequenceNode node, float chunk, float scale, int width,int offx, int offy) {
-    if (node == null) {
-      return;
+        pickNode(pickBox, top, chunk, wscale, width, offx, offy);
     }
 
-    if (node.left() == null && node.right() == null) {
-      float height = node.height;
-      float dist   = node.dist;
+    /**
+     * 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;
+        }
+
+        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 xstart = (int)((height-dist)*scale) + offx;
-      int xend   = (int)(height*scale) + offx;
+            int ypos = (int) (node.ycount * chunk) + offy;
 
-      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 (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 (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);
         }
-      }
-    } 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)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param node DOCUMENT ME!
+     * @param c DOCUMENT ME!
+     */
+    public void setColor(SequenceNode node, Color c)
     {
-      node.color = c;
+        if (node == null)
+        {
+            return;
+        }
+
+        if ((node.left() == null) && (node.right() == null))
+        {
+            node.color = c;
+
+            if (node.element() instanceof SequenceI)
+            {
+                ((SequenceI) node.element()).setColor(c);
+            }
+        }
+        else
+        {
+            node.color = c;
+            setColor((SequenceNode) node.left(), c);
+            setColor((SequenceNode) node.right(), c);
+        }
+    }
 
-      if (node.element() instanceof SequenceI)
-          ((SequenceI)node.element()).setColor(c);
-    } else
+    /**
+     * DOCUMENT ME!
+     */
+    void startPrinting()
     {
-      node.color = c;
-      setColor((SequenceNode)node.left(),c);
-      setColor((SequenceNode)node.right(),c);
+        Thread thread = new Thread(this);
+        thread.start();
     }
-  }
 
-  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());
 
-  // 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);
 
-    printJob.setPrintable(this, pf);
-    if (printJob.printDialog())
-    {
-      try
-      {
-        printJob.print();
-      }
-      catch (Exception PrintException)
-      {
-        PrintException.printStackTrace();
-      }
+        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());
 
-  public int print(Graphics pg, PageFormat pf, int pi) throws PrinterException
-  {
+        int pwidth = (int) pf.getImageableWidth();
+        int pheight = (int) pf.getImageableHeight();
 
-    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;
 
-    int noPages = getHeight() / pheight;
-    if(pi>noPages)
-      return Printable.NO_SUCH_PAGE;
+        if (pi > noPages)
+        {
+            return Printable.NO_SUCH_PAGE;
+        }
 
+        if (pwidth > getWidth())
+        {
+            pwidth = getWidth();
+        }
 
-    if (pwidth > getWidth())
-        pwidth = getWidth();
+        if (fitToWindow)
+        {
+            if (pheight > getHeight())
+            {
+                pheight = getHeight();
+            }
 
-    if(fitToWindow)
-    {
-      if (pheight > getHeight())
-        pheight = getHeight();
+            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;
+        }
 
-      noPages = 0;
+        draw(pg, pwidth, pheight);
+
+        return Printable.PAGE_EXISTS;
     }
-    else
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param g DOCUMENT ME!
+     */
+    public void paintComponent(Graphics g)
     {
+        super.paintComponent(g);
+        g.setFont(font);
+
+        if(tree==null)
+        {
+          g.drawString("Calculating tree....", 20, getHeight()/2);
+        }
+        else
+        {
+          fm = g.getFontMetrics(font);
+
+          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());
+          }
 
-        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;
+          scrollPane.revalidate();
+        }
     }
 
-    draw(pg, pwidth, pheight);
 
-    return Printable.PAGE_EXISTS;
+    /**
+     * 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);
 
-  public void paintComponent(Graphics g)
-  {
+        offy = font.getSize()+10;
 
-    font = new Font("Verdana",Font.PLAIN,fontSize);
-    g.setFont(font);
+        fm = g2.getFontMetrics(font);
 
-    FontMetrics fm = g.getFontMetrics(font);
+        labelLength = fm.stringWidth(longestName) + 20; //20 allows for scrollbar
 
-    if(nameHash.size()==0)
-      repaint();
+        float wscale = (float) (width - labelLength - (offx * 2)) / tree.getMaxHeight();
 
+        SequenceNode top = tree.getTopNode();
 
-    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 (top.count == 0)
+        {
+            top.count = ((SequenceNode) top.left()).count +
+                ((SequenceNode) top.right()).count;
+        }
 
-    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) {
+        float chunk = (float) (height - (offy)) / top.count;
 
-      Graphics2D g2 = (Graphics2D)g1;
-      g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
-      g2.setColor(Color.white);
-      g2.fillRect(0,0,width,height);
+        drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy);
 
+        if (threshold != 0)
+        {
+            if (av.getCurrentTree() == tree)
+            {
+                g2.setColor(Color.red);
+            }
+            else
+            {
+                g2.setColor(Color.gray);
+            }
 
-      labelLength = g2.getFontMetrics(font).stringWidth(longestName)+ 20;//20 allows for scrollbar
+            int x = (int) ((threshold * (float) (getWidth() - labelLength -
+                (2 * offx))) + offx);
 
-      float wscale =(float)(width - labelLength -offx*2)/tree.getMaxHeight();
+            g2.drawLine(x, 0, x, getHeight());
+        }
+    }
 
-      SequenceNode top = tree.getTopNode();
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseReleased(MouseEvent e)
+    {
+    }
 
-      if (top.count == 0) {
-          top.count = ((SequenceNode)top.left()).count + ((SequenceNode)top.right()).count ;
-      }
-      float chunk = (float)(height-offy*2)/top.count ;
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseEntered(MouseEvent e)
+    {
+    }
 
-      drawNode(g2,tree.getTopNode(),chunk,wscale,width,offx,offy);
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseExited(MouseEvent e)
+    {
+    }
 
-      if (threshold != 0)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseClicked(MouseEvent evt)
+    {
+      if(highlightNode!=null)
       {
-        if(av.getCurrentTree() == tree)
-          g2.setColor(Color.red);
+        if(evt.getClickCount()>1)
+        {
+          tree.swapNodes(highlightNode);
+          tree.reCount(tree.getTopNode());
+          tree.findHeight(tree.getTopNode());
+        }
         else
-          g2.setColor(Color.gray);
-
-          int x = (int)(    threshold * (float)(getWidth()-labelLength - 2*offx) +offx   ) ;
+        {
+          Vector leaves = new Vector();
+          tree.findLeaves(highlightNode, leaves);
+
+          for (int i = 0; i < leaves.size(); i++)
+          {
+            SequenceI seq =
+                (SequenceI) ( (SequenceNode) leaves.elementAt(i)).element();
+            treeSelectionChanged(seq);
+          }
+        }
 
-          g2.drawLine(x,0,x,getHeight());
+        PaintRefresher.Refresh(this, av.getSequenceSetId());
+        repaint();
       }
+    }
 
-  }
-
-  public void mouseReleased(MouseEvent e) { }
-  public void mouseEntered(MouseEvent e) { }
-  public void mouseExited(MouseEvent e) { }
-  public void mouseClicked(MouseEvent e) {
-  }
 
-  public void mousePressed(MouseEvent e) {
 
+    public void mouseMoved(MouseEvent evt)
+    {
       av.setCurrentTree(tree);
 
-      int x = e.getX();
-      int y = e.getY();
-
-      Object ob = findElement(x,y);
+      Object ob = findElement(evt.getX(), evt.getY());
 
-      if (ob instanceof SequenceI)
+      if (ob instanceof SequenceNode)
+      {
+        highlightNode = (SequenceNode) ob;
+        repaint();
+      }
+      else
       {
-          TreeSelectionChanged((Sequence)ob);
+        if (highlightNode != null)
+        {
+          highlightNode = null;
           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
+    public void mouseDragged(MouseEvent ect)
+    {}
 
-          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();
+        Object ob = findElement(x, y);
 
-              for (int i=0; i < tree.getGroups().size(); i++)
-              {
+        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));
+
+                tree.getGroups().removeAllElements();
+                tree.groupNodes(tree.getTopNode(), threshold);
+                setColor(tree.getTopNode(), Color.black);
+
+                av.setSelectionGroup(null);
+                av.alignment.deleteAllGroups();
+
+                colourGroups();
+              }
 
-                  Color col = new Color((int)(Math.random()*255),
-                                        (int)(Math.random()*255),
-                                        (int)(Math.random()*255));
-                  setColor((SequenceNode)tree.getGroups().elementAt(i),col.brighter());
+              PaintRefresher.Refresh(this, ap.av.getSequenceSetId());
+              repaint();
+        }
 
-                  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());
 
-                    sg.addSequence( (Sequence) sn.element());
-                  }
+    }
 
-                  if (av.getGlobalColourScheme() instanceof ConservationColourScheme)
-                  {
-                    ConservationColourScheme ccs = (ConservationColourScheme) av.getGlobalColourScheme();
-                    Conservation c = new Conservation("Group",
-                                                      ResidueProperties.propHash, 3,
-                                                      sg.sequences, sg.getStartRes(),
-                                                      sg.getEndRes());
+    void colourGroups()
+    {
+      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());
 
-                    c.calculate();
-                    c.verdict(false, av.ConsPercGaps);
-                    ccs = new ConservationColourScheme(c, ccs.cs);
+        Vector l = tree.findLeaves( (SequenceNode) tree.getGroups()
+                                   .elementAt(i),
+                                   new Vector());
 
-                    sg.cs = ccs;
+        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;
 
-                  av.alignment.addGroup(sg);
+        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());
+        }
 
-      PaintRefresher.Refresh(this);
-      repaint();
+        SequenceGroup sg = new SequenceGroup(sequences,
+                                             "TreeGroup", cs, true, true, false,
+                                             0,
+                                             av.alignment.getWidth() - 1);
+
+        if (av.getGlobalColourScheme() != null
+            && av.getGlobalColourScheme().conservationApplied())
+        {
+          Conservation c = new Conservation("Group",
+                                            ResidueProperties.propHash, 3,
+                                            sg.getSequences(false),
+                                            sg.getStartRes(), sg.getEndRes());
+
+          c.calculate();
+          c.verdict(false, av.ConsPercGaps);
+          sg.cs.setConservation(c);
+        }
 
-  }
+        av.alignment.addGroup(sg);
+      }
 
-    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();
+    }
 }
-