GPL license added
[jalview.git] / src / jalview / gui / TreeCanvas.java
index c39e407..5ebf323 100755 (executable)
@@ -1,3 +1,22 @@
+/*\r
+* Jalview - A Sequence Alignment Editor and Viewer\r
+* Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
+*\r
+* This program is free software; you can redistribute it and/or\r
+* modify it under the terms of the GNU General Public License\r
+* as published by the Free Software Foundation; either version 2\r
+* of the License, or (at your option) any later version.\r
+*\r
+* This program is distributed in the hope that it will be useful,\r
+* but WITHOUT ANY WARRANTY; without even the implied warranty of\r
+* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
+* GNU General Public License for more details.\r
+*\r
+* You should have received a copy of the GNU General Public License\r
+* along with this program; if not, write to the Free Software\r
+* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA\r
+*/\r
+\r
 package jalview.gui;\r
 \r
 \r
@@ -9,30 +28,32 @@ import javax.swing.*;
 import java.awt.*;\r
 import java.awt.event.*;\r
 import java.util.*;\r
+import java.awt.print.*;\r
 \r
-public class TreeCanvas extends JPanel implements MouseListener\r
+public class TreeCanvas extends JPanel implements MouseListener, Runnable, Printable\r
 {\r
   NJTree tree;\r
   JScrollPane scrollPane;\r
   AlignViewport av;\r
-\r
+  public static final String PLACEHOLDER=" * ";\r
   Font font;\r
   int  fontSize = 12;\r
 \r
+  boolean fitToWindow = true;\r
   boolean showDistances = false;\r
   boolean showBootstrap = false;\r
+  boolean markPlaceholders = false;\r
 \r
   int offx = 20;\r
   int offy = 20;\r
 \r
-  int threshold;\r
+  float threshold;\r
 \r
   String longestName;\r
   int labelLength=-1;\r
 \r
   //RubberbandRectangle rubberband;\r
 \r
-  Selection selected;\r
   Vector    listeners;\r
 \r
   Hashtable nameHash = new Hashtable();\r
@@ -42,7 +63,6 @@ public class TreeCanvas extends JPanel implements MouseListener
   {\r
     this.av = av;\r
     this.tree     = tree;\r
-    selected = av.getSelection();\r
     scrollPane = scroller;\r
     addMouseListener(this);\r
     tree.findHeight(tree.getTopNode());\r
@@ -52,23 +72,22 @@ public class TreeCanvas extends JPanel implements MouseListener
   }\r
   public void TreeSelectionChanged(Sequence sequence)\r
  {\r
-    selected = av.getSelection();\r
+    SequenceGroup selected = av.getSelectionGroup();\r
+    if(selected == null)\r
+    {\r
+      selected = new SequenceGroup();\r
+      av.setSelectionGroup(selected);\r
+    }\r
+\r
+    selected.setEndRes(av.alignment.getWidth());\r
+    selected.addOrRemove(sequence);\r
 \r
-    if (selected.contains(sequence))\r
-        selected.removeElement(sequence);\r
-    else\r
-        selected.addElement(sequence);\r
 \r
-    setSelected(selected);\r
     PaintRefresher.Refresh(this);\r
     repaint();\r
  }\r
 \r
 \r
-  public void setSelected(Selection selected)\r
-  {\r
-        this.selected = selected;\r
-  }\r
 \r
   public void setTree(NJTree tree) {\r
     this.tree = tree;\r
@@ -123,10 +142,9 @@ public class TreeCanvas extends JPanel implements MouseListener
       g.drawString(nodeLabel,xstart,ypos - 10);\r
       }\r
 \r
-      // Colour selected leaves differently\r
-      String name    = node.getName();\r
+      String name    = (markPlaceholders && node.isPlaceholder()) ? (PLACEHOLDER+node.getName()) : node.getName();\r
       FontMetrics fm = g.getFontMetrics(font);\r
-      int charWidth  = fm.stringWidth(node.getName()) + 3;\r
+      int charWidth  = fm.stringWidth(name) + 3;\r
       int charHeight = fm.getHeight();\r
 \r
       Rectangle rect = new Rectangle(xend+20,ypos-charHeight,\r
@@ -134,13 +152,15 @@ public class TreeCanvas extends JPanel implements MouseListener
 \r
       nameHash.put((SequenceI)node.element(),rect);\r
 \r
-      if (selected.contains((SequenceI)node.element())) {\r
+      // Colour selected leaves differently\r
+      SequenceGroup selected = av.getSelectionGroup();\r
+      if (selected!=null && selected.sequences.contains((SequenceI)node.element())) {\r
         g.setColor(Color.gray);\r
 \r
         g.fillRect(xend + 10, ypos - charHeight + 3,charWidth,charHeight);\r
         g.setColor(Color.white);\r
       }\r
-      g.drawString(node.getName(),xend+10,ypos);\r
+      g.drawString(name,xend+10,ypos);\r
       g.setColor(Color.black);\r
     } else {\r
       drawNode(g,(SequenceNode)node.left(), chunk,scale,width,offx,offy);\r
@@ -201,7 +221,7 @@ public class TreeCanvas extends JPanel implements MouseListener
 \r
   }\r
 \r
-  public void pickNodes(Rectangle pickBox, Selection sel) {\r
+  public void pickNodes(Rectangle pickBox) {\r
     int width  = getWidth();\r
     int height = getHeight();\r
 \r
@@ -214,10 +234,10 @@ public class TreeCanvas extends JPanel implements MouseListener
     }\r
     float chunk = (float)(height-offy*2)/top.count;\r
 \r
-    pickNode(pickBox,sel,top,chunk,wscale,width,offx,offy);\r
+    pickNode(pickBox,top,chunk,wscale,width,offx,offy);\r
   }\r
 \r
-  public void pickNode(Rectangle pickBox, Selection sel, SequenceNode node, float chunk, float scale, int width,int offx, int offy) {\r
+  public void pickNode(Rectangle pickBox, SequenceNode node, float chunk, float scale, int width,int offx, int offy) {\r
     if (node == null) {\r
       return;\r
     }\r
@@ -234,16 +254,14 @@ public class TreeCanvas extends JPanel implements MouseListener
       if (pickBox.contains(new Point(xend,ypos))) {\r
         if (node.element() instanceof SequenceI) {\r
           SequenceI seq = (SequenceI)node.element();\r
-          if (sel.contains(seq)) {\r
-            sel.removeElement(seq);\r
-          } else {\r
-            sel.addElement(seq);\r
-          }\r
+          SequenceGroup sg = av.getSelectionGroup();\r
+          if(sg!=null)\r
+            sg.addOrRemove(seq);\r
         }\r
       }\r
     } else {\r
-      pickNode(pickBox,sel,(SequenceNode)node.left(), chunk,scale,width,offx,offy);\r
-      pickNode(pickBox,sel,(SequenceNode)node.right(),chunk,scale,width,offx,offy);\r
+      pickNode(pickBox,(SequenceNode)node.left(), chunk,scale,width,offx,offy);\r
+      pickNode(pickBox,(SequenceNode)node.right(),chunk,scale,width,offx,offy);\r
     }\r
   }\r
 \r
@@ -266,10 +284,78 @@ public class TreeCanvas extends JPanel implements MouseListener
     }\r
   }\r
 \r
+  void startPrinting()\r
+  {\r
+    Thread thread = new Thread(this);\r
+    thread.start();\r
+  }\r
 \r
-  public void paintComponent(Graphics g) {\r
+  // put printing in a thread to avoid painting problems\r
+  public void run()\r
+  {\r
+    PrinterJob printJob = PrinterJob.getPrinterJob();\r
+    PageFormat pf = printJob.pageDialog(printJob.defaultPage());\r
+\r
+    printJob.setPrintable(this, pf);\r
+    if (printJob.printDialog())\r
+    {\r
+      try\r
+      {\r
+        printJob.print();\r
+      }\r
+      catch (Exception PrintException)\r
+      {\r
+        PrintException.printStackTrace();\r
+      }\r
+    }\r
+  }\r
 \r
 \r
+  public int print(Graphics pg, PageFormat pf, int pi) throws PrinterException\r
+  {\r
+\r
+    pg.setFont(font);\r
+    pg.translate((int)pf.getImageableX(), (int)pf.getImageableY());\r
+    int pwidth = (int) pf.getImageableWidth();\r
+    int pheight = (int) pf.getImageableHeight();\r
+\r
+    int noPages = getHeight() / pheight;\r
+    if(pi>noPages)\r
+      return Printable.NO_SUCH_PAGE;\r
+\r
+\r
+    if (pwidth > getWidth())\r
+        pwidth = getWidth();\r
+\r
+    if(fitToWindow)\r
+    {\r
+      if (pheight > getHeight())\r
+        pheight = getHeight();\r
+\r
+      noPages = 0;\r
+    }\r
+    else\r
+    {\r
+\r
+        FontMetrics fm = pg.getFontMetrics(font);\r
+        int height = fm.getHeight() * nameHash.size();\r
+        pg.translate(0, -pi*pheight  );\r
+        pg.setClip(0,pi*pheight, pwidth,pi*pheight + pheight);\r
+       // translate number of pages,\r
+       // height is screen size as this is the\r
+       // non overlapping text size\r
+        pheight = height;\r
+    }\r
+\r
+    draw(pg, pwidth, pheight);\r
+\r
+    return Printable.PAGE_EXISTS;\r
+\r
+  }\r
+\r
+  public void paintComponent(Graphics g)\r
+  {\r
+\r
     font = new Font("Verdana",Font.PLAIN,fontSize);\r
     g.setFont(font);\r
 \r
@@ -279,24 +365,17 @@ public class TreeCanvas extends JPanel implements MouseListener
       repaint();\r
 \r
 \r
-    if( scrollPane.getHeight() > fm.getHeight() * nameHash.size()+offy)\r
+    if( fitToWindow || (!fitToWindow && scrollPane.getHeight() > fm.getHeight() * nameHash.size()+offy ) )\r
      {\r
          draw(g,scrollPane.getWidth(),scrollPane.getHeight());\r
-         setPreferredSize(new Dimension(scrollPane.getWidth(), scrollPane.getHeight()));\r
+         setPreferredSize(null);\r
      }\r
     else\r
      {\r
-         setPreferredSize(new Dimension(getWidth(), fm.getHeight() * nameHash.size()));\r
-         draw( g,getWidth(), fm.getHeight() * nameHash.size());\r
+         setPreferredSize(new Dimension(scrollPane.getWidth(), fm.getHeight() * nameHash.size()));\r
+         draw( g,scrollPane.getWidth(), fm.getHeight() * nameHash.size());\r
      }\r
 \r
-\r
-    if (threshold != 0)\r
-    {\r
-        g.setColor(Color.red);\r
-        g.drawLine(threshold,0,threshold,getHeight());\r
-    }\r
-\r
     scrollPane.revalidate();\r
   }\r
     public int getFontSize() {\r
@@ -306,12 +385,15 @@ public class TreeCanvas extends JPanel implements MouseListener
         this.fontSize = fontSize;\r
         repaint();\r
     }\r
-  public void draw(Graphics g, int width, int height) {\r
-      g.setColor(Color.white);\r
-      g.fillRect(0,0,width,height);\r
+  public void draw(Graphics g1, int width, int height) {\r
+\r
+      Graphics2D g2 = (Graphics2D)g1;\r
+      g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);\r
+      g2.setColor(Color.white);\r
+      g2.fillRect(0,0,width,height);\r
 \r
 \r
-      labelLength = g.getFontMetrics(font).stringWidth(longestName)+ 20;//20 allows for scrollbar\r
+      labelLength = g2.getFontMetrics(font).stringWidth(longestName)+ 20;//20 allows for scrollbar\r
 \r
       float wscale =(float)(width - labelLength -offx*2)/tree.getMaxHeight();\r
 \r
@@ -322,7 +404,20 @@ public class TreeCanvas extends JPanel implements MouseListener
       }\r
       float chunk = (float)(height-offy*2)/top.count ;\r
 \r
-      drawNode(g,tree.getTopNode(),chunk,wscale,width,offx,offy);\r
+      drawNode(g2,tree.getTopNode(),chunk,wscale,width,offx,offy);\r
+\r
+      if (threshold != 0)\r
+      {\r
+        if(av.getCurrentTree() == tree)\r
+          g2.setColor(Color.red);\r
+        else\r
+          g2.setColor(Color.gray);\r
+\r
+          int x = (int)(    threshold * (float)(getWidth()-labelLength - 2*offx) +offx   ) ;\r
+\r
+          g2.drawLine(x,0,x,getHeight());\r
+      }\r
+\r
   }\r
 \r
   public void mouseReleased(MouseEvent e) { }\r
@@ -332,6 +427,9 @@ public class TreeCanvas extends JPanel implements MouseListener
   }\r
 \r
   public void mousePressed(MouseEvent e) {\r
+\r
+      av.setCurrentTree(tree);\r
+\r
       int x = e.getX();\r
       int y = e.getY();\r
 \r
@@ -352,13 +450,13 @@ public class TreeCanvas extends JPanel implements MouseListener
           // Find threshold\r
 \r
           if (tree.getMaxHeight() != 0) {\r
-              float fthreshold = (float)(x - offx)/(float)(getWidth()-labelLength - 2*offx);\r
-              this.threshold = x;\r
+              threshold = (float)(x - offx)/(float)(getWidth()-labelLength - 2*offx);\r
+\r
               tree.getGroups().removeAllElements();\r
-              tree.groupNodes(tree.getTopNode(),fthreshold);\r
+              tree.groupNodes(tree.getTopNode(),threshold);\r
               setColor(tree.getTopNode(),Color.black);\r
 \r
-              av.sel.clear();\r
+              av.setSelectionGroup(null);\r
               av.alignment.deleteAllGroups();\r
 \r
               for (int i=0; i < tree.getGroups().size(); i++)\r
@@ -375,7 +473,7 @@ public class TreeCanvas extends JPanel implements MouseListener
                   {\r
                     SequenceNode sn = (SequenceNode) l.elementAt(j);\r
                     if(sg==null)\r
-                       sg  = new SequenceGroup("TreeGroup", av.getGlobalColourScheme(), true, true,false,0,av.alignment.getWidth());\r
+                       sg = new SequenceGroup("TreeGroup", av.getGlobalColourScheme(), true, true,false,0,av.alignment.getWidth());\r
 \r
                     sg.addSequence( (Sequence) sn.element());\r
                   }\r
@@ -389,7 +487,7 @@ public class TreeCanvas extends JPanel implements MouseListener
                                                       sg.getEndRes());\r
 \r
                     c.calculate();\r
-                    c.verdict(false, 100);\r
+                    c.verdict(false, av.ConsPercGaps);\r
                     ccs = new ConservationColourScheme(c, ccs.cs);\r
 \r
                     sg.cs = ccs;\r
@@ -418,6 +516,10 @@ public class TreeCanvas extends JPanel implements MouseListener
       this.showBootstrap = state;\r
       repaint();\r
     }\r
+    public void setMarkPlaceholders(boolean state) {\r
+            this.markPlaceholders = state;\r
+            repaint();\r
+    }\r
 \r
 }\r
 \r