/*\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
import jalview.util.*;\r
\r
public class TreeCanvas\r
- extends Panel implements MouseListener\r
+ extends Panel implements MouseListener, MouseMotionListener\r
{\r
NJTree tree;\r
ScrollPane scrollPane;\r
String longestName;\r
int labelLength = -1;\r
\r
- //RubberbandRectangle rubberband;\r
-\r
- Vector listeners;\r
-\r
Hashtable nameHash = new Hashtable();\r
Hashtable nodeHash = new Hashtable();\r
\r
+ SequenceNode highlightNode;\r
+\r
+\r
public TreeCanvas(AlignViewport av, ScrollPane scroller)\r
{\r
this.av = av;\r
font = av.getFont();\r
scrollPane = scroller;\r
addMouseListener(this);\r
+ addMouseMotionListener(this);\r
setLayout(null);\r
\r
- PaintRefresher.Register(this, av.alignment);\r
+ PaintRefresher.Register(this, av.getSequenceSetId());\r
}\r
\r
- public void TreeSelectionChanged(Sequence sequence)\r
+ public void treeSelectionChanged(SequenceI sequence)\r
{\r
SequenceGroup selected = av.getSelectionGroup();\r
if (selected == null)\r
\r
selected.setEndRes(av.alignment.getWidth()-1);\r
selected.addOrRemove(sequence, true);\r
-\r
- PaintRefresher.Refresh(this, av.alignment);\r
- repaint();\r
}\r
\r
public void setTree(NJTree tree)\r
\r
if (node.element() instanceof SequenceI)\r
{\r
- if ( ( (SequenceI) ( (SequenceNode) node).element()).getColor() ==\r
- 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
}\r
// Colour selected leaves differently\r
SequenceGroup selected = av.getSelectionGroup();\r
if (selected != null &&\r
- selected.sequences.contains( (SequenceI) node.element()))\r
+ selected.getSequences(null).contains( (SequenceI) node.element()))\r
{\r
g.setColor(Color.gray);\r
\r
\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) + offy;\r
int yend = (int) ( ( (SequenceNode) node.right()).ycount * chunk) + offy;\r
\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
}\r
}\r
\r
- public void paint(Graphics g)\r
+ public void update(Graphics g)\r
{\r
+ paint(g);\r
+ }\r
\r
+ public void paint(Graphics g)\r
+ {\r
if(tree==null)\r
return;\r
\r
- g.setFont(font);\r
-\r
-\r
- FontMetrics fm = g.getFontMetrics(font);\r
-\r
if (nameHash.size() == 0)\r
{\r
repaint();\r
}\r
\r
- if (fitToWindow ||\r
- (!fitToWindow &&\r
- scrollPane.getSize().height > fm.getHeight() * nameHash.size() + offy))\r
+ int width = scrollPane.getSize().width;\r
+ int height = scrollPane.getSize().height;\r
+ if(!fitToWindow)\r
{\r
- draw(g, scrollPane.getSize().width, scrollPane.getSize().height);\r
+ height = g.getFontMetrics(font).getHeight() * nameHash.size();\r
}\r
- else\r
+\r
+ if(getSize().width>width)\r
{\r
- setSize(new Dimension(scrollPane.getSize().width,\r
- fm.getHeight() * nameHash.size()));\r
- draw(g, scrollPane.getSize().width, fm.getHeight() * nameHash.size());\r
+ setSize(new Dimension(width,height));\r
+ scrollPane.validate();\r
+ return;\r
}\r
\r
- scrollPane.validate();\r
+ setSize(new Dimension(width,height));\r
+\r
+\r
+ g.setFont(font);\r
+\r
+ draw(g, width, height);\r
+\r
}\r
\r
\r
public void mouseExited(MouseEvent e)\r
{}\r
\r
- public void mouseClicked(MouseEvent e)\r
+ public void mouseClicked(MouseEvent evt)\r
{\r
+ if (highlightNode != null)\r
+ {\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
+ public void mouseDragged(MouseEvent ect)\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
+ repaint();\r
+ }\r
+ else\r
+ {\r
+ if (highlightNode != null)\r
+ {\r
+ highlightNode = null;\r
+ repaint();\r
+ }\r
+ }\r
}\r
\r
public void mousePressed(MouseEvent e)\r
\r
if (ob instanceof SequenceI)\r
{\r
- TreeSelectionChanged( (Sequence) ob);\r
+ treeSelectionChanged( (Sequence) ob);\r
+ PaintRefresher.Refresh(this, av.getSequenceSetId());\r
repaint();\r
return;\r
-\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
- }\r
- else\r
+ else if ( !(ob instanceof SequenceNode))\r
{\r
// Find threshold\r
\r
\r
av.setSelectionGroup(null);\r
av.alignment.deleteAllGroups();\r
+ av.sequenceColours=null;\r
+\r
+ colourGroups();\r
+\r
+ }\r
+ }\r
+\r
+ PaintRefresher.Refresh(this, av.getSequenceSetId());\r
+ repaint();\r
\r
- for (int i = 0; i < tree.getGroups().size(); i++)\r
+ }\r
+\r
+ void colourGroups()\r
+ {\r
+ for (int i = 0; i < tree.getGroups().size(); i++)\r
{\r
\r
Color col = new Color( (int) (Math.random() * 255),\r
av.getIgnoreGapsConsensus());\r
}\r
\r
- SequenceGroup sg = new SequenceGroup(sequences, "TreeGroup",\r
+ SequenceGroup sg = new SequenceGroup(sequences, "",\r
cs, true, true,\r
false, 0, av.alignment.getWidth()-1);\r
\r
+ sg.setName("JTreeGroup:"+sg.hashCode());\r
+\r
\r
if ( av.getGlobalColourScheme()!=null\r
&& av.getGlobalColourScheme().conservationApplied())\r
{\r
Conservation c = new Conservation("Group",\r
ResidueProperties.propHash, 3,\r
- sg.sequences, sg.getStartRes(),\r
+ sg.getSequences(null),\r
+ sg.getStartRes(),\r
sg.getEndRes());\r
\r
c.calculate();\r
av.alignment.addGroup(sg);\r
\r
}\r
- }\r
- }\r
-\r
- PaintRefresher.Refresh(this, av.alignment);\r
- repaint();\r
\r
}\r
\r
git://source.jalview.org / jalview.git / blobdiff