Sequence colour in viewport
[jalview.git] / src / jalview / gui / TreeCanvas.java
index 944aa73..57e652c 100755 (executable)
@@ -1,6 +1,6 @@
 /*\r
  * Jalview - A Sequence Alignment Editor and Viewer\r
- * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
+ * Copyright (C) 2006 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
@@ -42,13 +42,14 @@ import javax.swing.*;
  * @version $Revision$\r
  */\r
 public class TreeCanvas extends JPanel implements MouseListener, Runnable,\r
-    Printable\r
+    Printable, MouseMotionListener\r
 {\r
     /** DOCUMENT ME!! */\r
     public static final String PLACEHOLDER = " * ";\r
     NJTree tree;\r
     JScrollPane scrollPane;\r
     AlignViewport av;\r
+    AlignmentPanel ap;\r
     Font font;\r
     FontMetrics fm;\r
     boolean fitToWindow = true;\r
@@ -61,10 +62,9 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
     String longestName;\r
     int labelLength = -1;\r
 \r
-    //RubberbandRectangle rubberband;\r
-    Vector listeners;\r
     Hashtable nameHash = new Hashtable();\r
     Hashtable nodeHash = new Hashtable();\r
+    SequenceNode highlightNode;\r
 \r
     /**\r
      * Creates a new TreeCanvas object.\r
@@ -74,13 +74,16 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
      * @param scroller DOCUMENT ME!\r
      * @param label DOCUMENT ME!\r
      */\r
-    public TreeCanvas(AlignViewport av, JScrollPane scroller)\r
+    public TreeCanvas(AlignmentPanel ap, JScrollPane scroller)\r
     {\r
-        this.av = av;\r
+        this.av = ap.av;\r
+        this.ap = ap;\r
         font = av.getFont();\r
         scrollPane = scroller;\r
         addMouseListener(this);\r
-        PaintRefresher.Register(this, av.alignment);\r
+        addMouseMotionListener(this);\r
+        PaintRefresher.Register(this, ap.av.getSequenceSetId());\r
+        ToolTipManager.sharedInstance().registerComponent(this);\r
     }\r
 \r
 \r
@@ -89,7 +92,7 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
      *\r
      * @param sequence DOCUMENT ME!\r
      */\r
-    public void TreeSelectionChanged(Sequence sequence)\r
+    public void treeSelectionChanged(SequenceI sequence)\r
     {\r
         SequenceGroup selected = av.getSelectionGroup();\r
 \r
@@ -101,9 +104,6 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
 \r
         selected.setEndRes(av.alignment.getWidth()-1);\r
         selected.addOrRemove(sequence, true);\r
-\r
-        PaintRefresher.Refresh(this, av.alignment);\r
-        repaint();\r
     }\r
 \r
     /**\r
@@ -173,14 +173,15 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
 \r
             if (node.element() instanceof SequenceI)\r
             {\r
-                if (((SequenceI) ((SequenceNode) node).element()).getColor() == Color.white)\r
+              SequenceI seq = (SequenceI)((SequenceNode) node).element();\r
+\r
+                if (av.getSequenceColour(seq) == Color.white)\r
                 {\r
                     g.setColor(Color.black);\r
                 }\r
                 else\r
                 {\r
-                    g.setColor(((SequenceI) ((SequenceNode) node).element()).getColor()\r
-                                .darker());\r
+                    g.setColor(av.getSequenceColour(seq).darker());\r
                 }\r
             }\r
             else\r
@@ -228,7 +229,7 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
             SequenceGroup selected = av.getSelectionGroup();\r
 \r
             if ((selected != null) &&\r
-                    selected.sequences.contains((SequenceI) node.element()))\r
+                    selected.getSequences(false).contains((SequenceI) node.element()))\r
             {\r
                 g.setColor(Color.gray);\r
 \r
@@ -258,7 +259,10 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
 \r
             // Draw horizontal line\r
             g.drawLine(xstart, ypos, xend, ypos);\r
-            g.fillRect(xend - 2, ypos - 2, 4, 4);\r
+            if (node == highlightNode)\r
+              g.fillRect(xend - 3, ypos - 3, 6, 6);\r
+            else\r
+              g.fillRect(xend - 2, ypos - 2, 4, 4);\r
 \r
             int ystart = (int) (((SequenceNode) node.left()).ycount * chunk) +\r
                 offy;\r
@@ -416,7 +420,7 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
 \r
             if (node.element() instanceof SequenceI)\r
             {\r
-                ((SequenceI) node.element()).setColor(c);\r
+              av.setSequenceColour((SequenceI) node.element(), c);\r
             }\r
         }\r
         else\r
@@ -657,10 +661,74 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
      *\r
      * @param e DOCUMENT ME!\r
      */\r
-    public void mouseClicked(MouseEvent e)\r
+    public void mouseClicked(MouseEvent evt)\r
     {\r
+      if(highlightNode!=null)\r
+      {\r
+        if (SwingUtilities.isRightMouseButton(evt))\r
+        {\r
+          Color col = JColorChooser.showDialog(this, "Select Background Colour",\r
+                                               highlightNode.color);\r
+\r
+          setColor(highlightNode, col);\r
+        }\r
+        else\r
+        if(evt.getClickCount()>1)\r
+        {\r
+          tree.swapNodes(highlightNode);\r
+          tree.reCount(tree.getTopNode());\r
+          tree.findHeight(tree.getTopNode());\r
+        }\r
+        else\r
+        {\r
+          Vector leaves = new Vector();\r
+          tree.findLeaves(highlightNode, leaves);\r
+\r
+          for (int i = 0; i < leaves.size(); i++)\r
+          {\r
+            SequenceI seq =\r
+                (SequenceI) ( (SequenceNode) leaves.elementAt(i)).element();\r
+            treeSelectionChanged(seq);\r
+          }\r
+        }\r
+\r
+        PaintRefresher.Refresh(this, av.getSequenceSetId());\r
+        repaint();\r
+      }\r
     }\r
 \r
+\r
+\r
+    public void mouseMoved(MouseEvent evt)\r
+    {\r
+      av.setCurrentTree(tree);\r
+\r
+      Object ob = findElement(evt.getX(), evt.getY());\r
+\r
+      if (ob instanceof SequenceNode)\r
+      {\r
+        highlightNode = (SequenceNode) ob;\r
+        this.setToolTipText(\r
+            "<html>Left click to select leaves"\r
+            + "<br>Double-click to invert leaves"\r
+            + "<br>Right click to change colour");\r
+        repaint();\r
+\r
+      }\r
+      else\r
+      {\r
+        if (highlightNode != null)\r
+        {\r
+          highlightNode = null;\r
+          setToolTipText(null);\r
+          repaint();\r
+        }\r
+      }\r
+  }\r
+\r
+    public void mouseDragged(MouseEvent ect)\r
+    {}\r
+\r
     /**\r
      * DOCUMENT ME!\r
      *\r
@@ -677,19 +745,12 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
 \r
         if (ob instanceof SequenceI)\r
         {\r
-            TreeSelectionChanged((Sequence) ob);\r
-            repaint();\r
-\r
-            return;\r
-        }\r
-        else if (ob instanceof SequenceNode)\r
-        {\r
-            SequenceNode tmpnode = (SequenceNode) ob;\r
-            tree.swapNodes(tmpnode);\r
-            tree.reCount(tree.getTopNode());\r
-            tree.findHeight(tree.getTopNode());\r
+          treeSelectionChanged( (Sequence) ob);\r
+          PaintRefresher.Refresh(this, ap.av.getSequenceSetId());\r
+          repaint();\r
+          return;\r
         }\r
-        else\r
+        else if( !(ob instanceof SequenceNode) )\r
         {\r
             // Find threshold\r
             if (tree.getMaxHeight() != 0)\r
@@ -703,76 +764,86 @@ public class TreeCanvas extends JPanel implements MouseListener, Runnable,
 \r
                 av.setSelectionGroup(null);\r
                 av.alignment.deleteAllGroups();\r
+                av.sequenceColours.clear();\r
 \r
-                for (int i = 0; i < tree.getGroups().size(); i++)\r
-                {\r
-                    Color col = new Color((int) (Math.random() * 255),\r
-                            (int) (Math.random() * 255),\r
-                            (int) (Math.random() * 255));\r
-                    setColor((SequenceNode) tree.getGroups().elementAt(i),\r
-                        col.brighter());\r
+                colourGroups();\r
+              }\r
 \r
-                    Vector l = tree.findLeaves((SequenceNode) tree.getGroups()\r
-                                                                  .elementAt(i),\r
-                            new Vector());\r
+              PaintRefresher.Refresh(this, ap.av.getSequenceSetId());\r
+              repaint();\r
+        }\r
 \r
-                    Vector sequences = new Vector();\r
 \r
-                    for (int j = 0; j < l.size(); j++)\r
-                    {\r
-                        SequenceI s1 = (SequenceI) ((SequenceNode) l.elementAt(j)).element();\r
+    }\r
 \r
-                        if (!sequences.contains(s1))\r
-                        {\r
-                            sequences.addElement(s1);\r
-                        }\r
-                    }\r
+    void colourGroups()\r
+    {\r
+      for (int i = 0; i < tree.getGroups().size(); i++)\r
+      {\r
+        Color col = new Color( (int) (Math.random() * 255),\r
+                              (int) (Math.random() * 255),\r
+                              (int) (Math.random() * 255));\r
+        setColor( (SequenceNode) tree.getGroups().elementAt(i),\r
+                 col.brighter());\r
 \r
-                    ColourSchemeI cs = null;\r
+        Vector l = tree.findLeaves( (SequenceNode) tree.getGroups()\r
+                                   .elementAt(i),\r
+                                   new Vector());\r
 \r
-                    if (av.getGlobalColourScheme() != null)\r
-                    {\r
-                      if (av.getGlobalColourScheme() instanceof UserColourScheme)\r
-                      {\r
-                        cs = new UserColourScheme(\r
-                            ( (UserColourScheme) av.getGlobalColourScheme()).getColours());\r
-\r
-                      }\r
-                      else\r
-                        cs = ColourSchemeProperty.getColour(sequences,\r
-                                                            av.alignment.getWidth(),\r
-                                                            ColourSchemeProperty.getColourName(\r
-                                                                av.getGlobalColourScheme()));\r
-\r
-                        cs.setThreshold(av.getGlobalColourScheme().getThreshold(),\r
-                                                             av.getIgnoreGapsConsensus());\r
-                    }\r
+        Vector sequences = new Vector();\r
+\r
+        for (int j = 0; j < l.size(); j++)\r
+        {\r
+          SequenceI s1 = (SequenceI) ( (SequenceNode) l.elementAt(j)).element();\r
 \r
+          if (!sequences.contains(s1))\r
+          {\r
+            sequences.addElement(s1);\r
+          }\r
+        }\r
 \r
-                    SequenceGroup sg = new SequenceGroup(sequences,\r
-                            "TreeGroup", cs, true, true, false, 0,\r
-                            av.alignment.getWidth()-1);\r
+        ColourSchemeI cs = null;\r
 \r
+        if (av.getGlobalColourScheme() != null)\r
+        {\r
+          if (av.getGlobalColourScheme() instanceof UserColourScheme)\r
+          {\r
+            cs = new UserColourScheme(\r
+                ( (UserColourScheme) av.getGlobalColourScheme()).getColours());\r
 \r
-                    if (  av.getGlobalColourScheme()!=null\r
-                       && av.getGlobalColourScheme().conservationApplied())\r
-                    {\r
-                        Conservation c = new Conservation("Group",\r
-                                ResidueProperties.propHash, 3, sg.sequences,\r
-                                sg.getStartRes(), sg.getEndRes());\r
+          }\r
+          else\r
+            cs = ColourSchemeProperty.getColour(sequences,\r
+                                                av.alignment.getWidth(),\r
+                                                ColourSchemeProperty.\r
+                                                getColourName(\r
+                                                    av.getGlobalColourScheme()));\r
+\r
+          cs.setThreshold(av.getGlobalColourScheme().getThreshold(),\r
+                          av.getIgnoreGapsConsensus());\r
+        }\r
 \r
-                        c.calculate();\r
-                        c.verdict(false, av.ConsPercGaps);\r
-                        sg.cs.setConservation(c);\r
-                    }\r
+        SequenceGroup sg = new SequenceGroup(sequences,\r
+                                             "TreeGroup", cs, true, true, false,\r
+                                             0,\r
+                                             av.alignment.getWidth() - 1);\r
 \r
-                    av.alignment.addGroup(sg);\r
-                }\r
-            }\r
+        if (av.getGlobalColourScheme() != null\r
+            && av.getGlobalColourScheme().conservationApplied())\r
+        {\r
+          Conservation c = new Conservation("Group",\r
+                                            ResidueProperties.propHash, 3,\r
+                                            sg.getSequences(false),\r
+                                            sg.getStartRes(), sg.getEndRes());\r
+\r
+          c.calculate();\r
+          c.verdict(false, av.ConsPercGaps);\r
+          sg.cs.setConservation(c);\r
         }\r
 \r
-        PaintRefresher.Refresh(this, av.alignment);\r
-        repaint();\r
+        av.alignment.addGroup(sg);\r
+      }\r
+\r
     }\r
 \r
     /**\r