X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FTreeCanvas.java;h=eea181f476701e4d2a58abcf7b5068100d337692;hb=eee02848424734a9466de35ff19636377652e59b;hp=0f85aed4b91f5eebac321418cad0691e409a4b7b;hpb=e4f54241f5c5caf8415113b39fdce78be86ae7fa;p=jalview.git

diff --git a/src/jalview/gui/TreeCanvas.java b/src/jalview/gui/TreeCanvas.java
index 0f85aed..eea181f 100755
--- a/src/jalview/gui/TreeCanvas.java
+++ b/src/jalview/gui/TreeCanvas.java
@@ -1,381 +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.schemes.*;
+
 import jalview.util.*;
-import javax.swing.*;
+
 import java.awt.*;
 import java.awt.event.*;
-import java.util.*;
-import java.io.*;
+import java.awt.print.*;
 
-public class TreeCanvas extends JPanel implements MouseListener
-{
-  NJTree tree;
-  JScrollPane scrollPane;
-  AlignViewport av;
-
-  Font font;
-  int  fontSize = 12;
-
-  boolean showDistances = false;
-  boolean showBootstrap = false;
+import java.util.*;
 
-  int offx = 20;
-  int offy = 20;
+import javax.swing.*;
 
-  int threshold;
 
-  String longestName;
-  int labelLength=-1;
+/**
+ * DOCUMENT ME!
+ *
+ * @author $author$
+ * @version $Revision$
+ */
+public class TreeCanvas extends JPanel implements MouseListener, Runnable,
+    Printable
+{
+    /** 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)
+    {
+        this.av = av;
+        font = av.getFont();
+        scrollPane = scroller;
+        addMouseListener(this);
+        PaintRefresher.Register(this, av.alignment);
+    }
 
-  //RubberbandRectangle rubberband;
 
-  Selection selected;
-  Vector    listeners;
+    /**
+     * DOCUMENT ME!
+     *
+     * @param sequence DOCUMENT ME!
+     */
+    public void TreeSelectionChanged(Sequence sequence)
+    {
+        SequenceGroup selected = av.getSelectionGroup();
 
-  Hashtable nameHash = new Hashtable();
-  Hashtable nodeHash = new Hashtable();
+        if (selected == null)
+        {
+            selected = new SequenceGroup();
+            av.setSelectionGroup(selected);
+        }
 
-  public TreeCanvas(AlignViewport av, NJTree tree, JScrollPane scroller, String label)
-  {
-    this.av = av;
-    this.tree     = tree;
-    selected = av.getSelection();
-    scrollPane = scroller;
-    addMouseListener(this);
-    tree.findHeight(tree.getTopNode());
-    longestName = label;
+        selected.setEndRes(av.alignment.getWidth()-1);
+        selected.addOrRemove(sequence, true);
 
-    PaintRefresher.Register(this);
-  }
-  public void TreeSelectionChanged(Sequence sequence)
- {
-    selected = av.getSelection();
+        PaintRefresher.Refresh(this, av.alignment);
+        repaint();
+    }
 
-    if (selected.contains(sequence))
-        selected.removeElement(sequence);
-    else
-        selected.addElement(sequence);
+    /**
+     * DOCUMENT ME!
+     *
+     * @param tree DOCUMENT ME!
+     */
+    public void setTree(NJTree tree)
+    {
+        this.tree = tree;
+        tree.findHeight(tree.getTopNode());
 
-    setSelected(selected);
-    PaintRefresher.Refresh(this);
-    repaint();
- }
+        // 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++)
+        {
+          SequenceNode lf = (SequenceNode) leaves.elementAt(i);
 
-  public void setSelected(Selection selected)
-  {
-        this.selected = selected;
-  }
+          if (lf.isPlaceholder())
+          {
+            has_placeholders = true;
+          }
 
-  public void setTree(NJTree tree) {
-    this.tree = tree;
-    tree.findHeight(tree.getTopNode());
-  }
+          if (longestName.length() < ( (Sequence) lf.element()).getName()
+              .length())
+          {
+            longestName = TreeCanvas.PLACEHOLDER +
+                ( (Sequence) lf.element()).getName();
+          }
+        }
 
-    public void drawNode(Graphics g,SequenceNode node, float chunk, float scale, int width,int offx, int offy) {
-    if (node == null) {
-      return;
+        setMarkPlaceholders(has_placeholders);
     }
 
-    if (node.left() == null && node.right() == null) {
-      // Drawing leaf node
+    /**
+     * 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;
+        }
 
-      float height = node.height;
-      float dist   = node.dist;
+        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);
+            }
 
-      int xstart = (int)((height-dist)*scale) + offx;
-      int xend =   (int)(height*scale)        + offx;
+            // Draw horizontal line
+            g.drawLine(xstart, ypos, xend, ypos);
 
-      int ypos = (int)(node.ycount * chunk) + offy;
+            String nodeLabel = "";
 
-      if (node.element() instanceof SequenceI)
-          g.setColor(((SequenceI)((SequenceNode)node).element()).getColor().darker());
-      else
-          g.setColor(Color.black);
+            if (showDistances && (node.dist > 0))
+            {
+                nodeLabel = new Format("%5.2f").form(node.dist);
+            }
 
+            if (showBootstrap)
+            {
+                if (showDistances)
+                {
+                    nodeLabel = nodeLabel + " : ";
+                }
 
-      // Draw horizontal line
-      g.drawLine(xstart,ypos,xend,ypos);
+                nodeLabel = nodeLabel + String.valueOf(node.getBootstrap());
+            }
 
-      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 (!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();
 
-      // Colour selected leaves differently
-      String name    = node.getName();
-      FontMetrics fm = g.getFontMetrics(font);
-      int charWidth  = fm.stringWidth(node.getName()) + 3;
-      int charHeight = fm.getHeight();
+            Rectangle rect = new Rectangle(xend + 20, ypos - charHeight,
+                    charWidth, charHeight);
 
-      Rectangle rect = new Rectangle(xend+20,ypos-charHeight,
-                                     charWidth,charHeight);
+            nameHash.put((SequenceI) node.element(), rect);
 
-      nameHash.put((SequenceI)node.element(),rect);
+            // Colour selected leaves differently
+            SequenceGroup selected = av.getSelectionGroup();
 
-      if (selected.contains((SequenceI)node.element())) {
-        g.setColor(Color.gray);
+            if ((selected != null) &&
+                    selected.sequences.contains((SequenceI) node.element()))
+            {
+                g.setColor(Color.gray);
 
-        g.fillRect(xend + 10, ypos - charHeight + 3,charWidth,charHeight);
-        g.setColor(Color.white);
-      }
-      g.drawString(node.getName(),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);
+                g.fillRect(xend + 10, ypos - charHeight + 3, charWidth,
+                    charHeight);
+                g.setColor(Color.white);
+            }
 
-      float height = node.height;
-      float dist   = node.dist;
+            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);
 
-      int xstart = (int)((height-dist)*scale) + offx;
-      int xend   = (int)(height       *scale) + offx;
-      int ypos   = (int)(node.ycount  *chunk) + offy;
+            float height = node.height;
+            float dist = node.dist;
 
-      g.setColor(((SequenceNode)node).color.darker());
+            int xstart = (int) ((height - dist) * scale) + offx;
+            int xend = (int) (height * scale) + offx;
+            int ypos = (int) (node.ycount * chunk) + offy;
 
-      // Draw horizontal line
-      g.drawLine(xstart,ypos,xend,ypos);
-      g.fillRect(xend-2, ypos-2, 4,4);
+            g.setColor(((SequenceNode) node).color.darker());
 
-      int ystart = (int)(((SequenceNode)node.left()) .ycount * chunk) + offy;
-      int yend   = (int)(((SequenceNode)node.right()).ycount * chunk) + offy;
+            // Draw horizontal line
+            g.drawLine(xstart, ypos, xend, ypos);
+            g.fillRect(xend - 2, ypos - 2, 4, 4);
 
-      Rectangle pos = new Rectangle(xend-2,ypos-2,5,5);
-      nodeHash.put(node,pos);
+            int ystart = (int) (((SequenceNode) node.left()).ycount * chunk) +
+                offy;
+            int yend = (int) (((SequenceNode) node.right()).ycount * chunk) +
+                offy;
 
-      g.drawLine((int)(height*scale) + offx, ystart,
-                 (int)(height*scale) + offx, yend);
+            Rectangle pos = new Rectangle(xend - 2, ypos - 2, 5, 5);
+            nodeHash.put(node, pos);
 
-      if (showDistances && node.dist > 0) {
-        g.drawString(new Format("%5.2f").form(node.dist),xstart,ypos - 5);
-      }
+            g.drawLine((int) (height * scale) + offx, ystart,
+                (int) (height * scale) + offx, yend);
 
+            if (showDistances && (node.dist > 0))
+            {
+                g.drawString(new Format("%5.2f").form(node.dist), xstart,
+                    ypos - 5);
+            }
+        }
     }
-  }
-  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, Selection sel) {
-    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,sel,top,chunk,wscale,width,offx,offy);
-  }
+        float chunk = (float) (height - (offy * 2)) / top.count;
 
-  public void pickNode(Rectangle pickBox, Selection sel, 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;
+        }
 
-      int xstart = (int)((height-dist)*scale) + offx;
-      int xend   = (int)(height*scale) + offx;
+        if ((node.left() == null) && (node.right() == null))
+        {
+            float height = node.height;
+            float dist = node.dist;
 
-      int ypos = (int)(node.ycount * chunk) + offy;
+            int xstart = (int) ((height - dist) * scale) + offx;
+            int xend = (int) (height * scale) + offx;
 
-      if (pickBox.contains(new Point(xend,ypos))) {
-        if (node.element() instanceof SequenceI) {
-          SequenceI seq = (SequenceI)node.element();
-          if (sel.contains(seq)) {
-            sel.removeElement(seq);
-          } else {
-            sel.addElement(seq);
-          }
+            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);
+        }
+    }
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param node DOCUMENT ME!
+     * @param c DOCUMENT ME!
+     */
+    public void setColor(SequenceNode 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);
         }
-      }
-    } else {
-      pickNode(pickBox,sel,(SequenceNode)node.left(), chunk,scale,width,offx,offy);
-      pickNode(pickBox,sel,(SequenceNode)node.right(),chunk,scale,width,offx,offy);
     }
-  }
 
-  public void setColor(SequenceNode node, Color c) {
-    if (node == null) {
-      return;
+    /**
+     * DOCUMENT ME!
+     */
+    void startPrinting()
+    {
+        Thread thread = new Thread(this);
+        thread.start();
     }
 
-    if (node.left() == null && node.right() == null) {
-      node.color = c;
+    // 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 (node.element() instanceof SequenceI) {
-          ((SequenceI)node.element()).setColor(c);
-      }
-    } else {
-      node.color = c;
-      setColor((SequenceNode)node.left(),c);
-      setColor((SequenceNode)node.right(),c);
+        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 void paintComponent(Graphics g) {
+        int pwidth = (int) pf.getImageableWidth();
+        int pheight = (int) pf.getImageableHeight();
 
+        int noPages = getHeight() / pheight;
 
-    font = new Font("Verdana",Font.PLAIN,fontSize);
-    g.setFont(font);
+        if (pi > noPages)
+        {
+            return Printable.NO_SUCH_PAGE;
+        }
 
-    FontMetrics fm = g.getFontMetrics(font);
+        if (pwidth > getWidth())
+        {
+            pwidth = getWidth();
+        }
 
-    if(nameHash.size()==0)
-      repaint();
+        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;
+        }
 
-    if( scrollPane.getHeight() > fm.getHeight() * nameHash.size()+offy)
-     {
-         draw(g,scrollPane.getWidth(),scrollPane.getHeight());
-         setPreferredSize(new Dimension(scrollPane.getWidth(), scrollPane.getHeight()));
-     }
-    else
-     {
-         setPreferredSize(new Dimension(getWidth(), fm.getHeight() * nameHash.size()));
-         draw( g,getWidth(), fm.getHeight() * nameHash.size());
-     }
+        draw(pg, pwidth, pheight);
 
+        return Printable.PAGE_EXISTS;
+    }
 
-    if (threshold != 0)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param g DOCUMENT ME!
+     */
+    public void paintComponent(Graphics g)
     {
-        g.setColor(Color.red);
-        g.drawLine(threshold,0,threshold,getHeight());
+        super.paintComponent(g);
+        g.setFont(font);
+
+        if(tree==null)
+        {
+          g.drawString("Calculating tree....", 20, getHeight()/2);
+        }
+        else
+        {
+          FontMetrics 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());
+          }
+
+          scrollPane.revalidate();
+        }
     }
 
-    scrollPane.revalidate();
-  }
-    public int getFontSize() {
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public int getFontSize()
+    {
         return fontSize;
     }
-    public void setFontSize(int fontSize) {
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param fontSize DOCUMENT ME!
+     */
+    public void setFontSize(int fontSize)
+    {
         this.fontSize = fontSize;
+        font = new Font(font.getName(), font.getStyle(), fontSize);
         repaint();
     }
-  public void draw(Graphics g, int width, int height) {
-      g.setColor(Color.white);
-      g.fillRect(0,0,width,height);
 
+    /**
+     * DOCUMENT ME!
+     *
+     * @param g1 DOCUMENT ME!
+     * @param width DOCUMENT ME!
+     * @param height DOCUMENT ME!
+     */
+    public void draw(Graphics g1, int width, int height)
+    {
+        Graphics2D g2 = (Graphics2D) g1;
+        g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
+            RenderingHints.VALUE_ANTIALIAS_ON);
+        g2.setColor(Color.white);
+        g2.fillRect(0, 0, width, height);
 
-      labelLength = g.getFontMetrics(font).stringWidth(longestName)+ 20;//20 allows for scrollbar
+        g2.setFont(font);
 
-      float wscale =(float)(width - labelLength -offx*2)/tree.getMaxHeight();
+        labelLength = g2.getFontMetrics(font).stringWidth(longestName) + 20; //20 allows for scrollbar
 
-      SequenceNode top = tree.getTopNode();
+        float wscale = (float) (width - labelLength - (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 ;
+        SequenceNode top = tree.getTopNode();
 
-      drawNode(g,tree.getTopNode(),chunk,wscale,width,offx,offy);
-  }
+        if (top.count == 0)
+        {
+            top.count = ((SequenceNode) top.left()).count +
+                ((SequenceNode) top.right()).count;
+        }
 
-  public void mouseReleased(MouseEvent e) { }
-  public void mouseEntered(MouseEvent e) { }
-  public void mouseExited(MouseEvent e) { }
-  public void mouseClicked(MouseEvent e) {
-  }
+        float chunk = (float) (height - (offy * 2)) / top.count;
 
-  public void mousePressed(MouseEvent e) {
-      int x = e.getX();
-      int y = e.getY();
+        drawNode(g2, tree.getTopNode(), chunk, wscale, width, offx, offy);
 
-      Object ob = findElement(x,y);
+        if (threshold != 0)
+        {
+            if (av.getCurrentTree() == tree)
+            {
+                g2.setColor(Color.red);
+            }
+            else
+            {
+                g2.setColor(Color.gray);
+            }
 
-      if (ob instanceof SequenceI)
-      {
-          TreeSelectionChanged((Sequence)ob);
-          repaint();
-          return;
+            int x = (int) ((threshold * (float) (getWidth() - labelLength -
+                (2 * offx))) + offx);
 
-      } else if (ob instanceof SequenceNode) {
-          SequenceNode tmpnode = (SequenceNode)ob;
-          tree.swapNodes(tmpnode);
-          tree.reCount(tree.getTopNode());
-          tree.findHeight(tree.getTopNode());
-      } else {
-          // Find threshold
+            g2.drawLine(x, 0, x, getHeight());
+        }
+    }
 
-          if (tree.getMaxHeight() != 0) {
-              float fthreshold = (float)(x - offx)/(float)(getWidth()-labelLength - 2*offx);
-              this.threshold = x;
-              tree.getGroups().removeAllElements();
-              tree.groupNodes(tree.getTopNode(),fthreshold);
-              setColor(tree.getTopNode(),Color.black);
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseReleased(MouseEvent e)
+    {
+    }
 
-              for (int i=0; i < tree.getGroups().size(); i++) {
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseEntered(MouseEvent e)
+    {
+    }
 
-                  Color col = new Color((int)(Math.random()*255),
-                                        (int)(Math.random()*255),
-                                        (int)(Math.random()*255));
-                  setColor((SequenceNode)tree.getGroups().elementAt(i),col.brighter());
+    /**
+     * DOCUMENT ME!
+     *
+     * @param e DOCUMENT ME!
+     */
+    public void mouseExited(MouseEvent e)
+    {
+    }
 
-                  // l is vector of Objects
-                 // Vector l = tree.findLeaves((SequenceNode)tree.getGroups().elementAt(i),new Vector());
+    /**
+     * 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();
 
-      PaintRefresher.Refresh(this);
-      repaint();
+        Object ob = findElement(x, y);
 
-  }
+        if (ob instanceof SequenceI)
+        {
+            TreeSelectionChanged((Sequence) ob);
+            repaint();
 
-    public void setShowDistances(boolean state) {
+            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();
+    }
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param state DOCUMENT ME!
+     */
+    public void setShowDistances(boolean state)
+    {
         this.showDistances = state;
         repaint();
     }
 
-    public void setShowBootstrap(boolean state) {
-      this.showBootstrap = 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();
+    }
 }
-