import jalview.analysis.*;\r
import jalview.datamodel.*;\r
import jalview.util.*;\r
+import jalview.schemes.*;\r
import javax.swing.*;\r
import java.awt.*;\r
import java.awt.event.*;\r
import java.util.*;\r
-import java.io.*;\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
Font font;\r
int fontSize = 12;\r
\r
+ boolean fitToWindow = true;\r
boolean showDistances = false;\r
boolean showBootstrap = false;\r
\r
\r
//RubberbandRectangle rubberband;\r
\r
- Selection selected;\r
Vector listeners;\r
\r
Hashtable nameHash = new Hashtable();\r
{\r
this.av = av;\r
this.tree = tree;\r
- selected = av.getSelection();\r
scrollPane = scroller;\r
addMouseListener(this);\r
tree.findHeight(tree.getTopNode());\r
}\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
\r
nameHash.put((SequenceI)node.element(),rect);\r
\r
- if (selected.contains((SequenceI)node.element())) {\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
\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
}\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
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
- public void setColor(SequenceNode node, Color c) {\r
- if (node == null) {\r
+ public void setColor(SequenceNode node, Color c)\r
+ {\r
+ if (node == null)\r
return;\r
- }\r
\r
- if (node.left() == null && node.right() == null) {\r
+ if (node.left() == null && node.right() == null)\r
+ {\r
node.color = c;\r
\r
- if (node.element() instanceof SequenceI) {\r
+ if (node.element() instanceof SequenceI)\r
((SequenceI)node.element()).setColor(c);\r
- }\r
- } else {\r
+ } else\r
+ {\r
node.color = c;\r
setColor((SequenceNode)node.left(),c);\r
setColor((SequenceNode)node.right(),c);\r
}\r
}\r
\r
+ void startPrinting()\r
+ {\r
+ Thread thread = new Thread(this);\r
+ thread.start();\r
+ }\r
+\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
+ 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
+ pg.setFont(font);\r
+ FontMetrics fm = pg.getFontMetrics(font);\r
+\r
+ pg.translate((int)pf.getImageableX(), (int)pf.getImageableY());\r
+\r
+ int pwidth = (int) pf.getImageableWidth();\r
+ int pheight = (int) pf.getImageableHeight();\r
+\r
+ // adjust pheight to row height\r
+ pheight -= (pheight%fm.getHeight());\r
+ pg.setClip(0,0, pwidth, pheight);\r
+\r
+ int noPages = (fm.getHeight() * nameHash.size()) / pheight;\r
+\r
+ if(pi>noPages)\r
+ return Printable.NO_SUCH_PAGE;\r
+\r
+ pg.translate(0, -pheight*pi );\r
+ draw(pg, pwidth, fm.getHeight() * nameHash.size());\r
\r
- public void paintComponent(Graphics g) {\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
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
tree.groupNodes(tree.getTopNode(),fthreshold);\r
setColor(tree.getTopNode(),Color.black);\r
\r
- for (int i=0; i < tree.getGroups().size(); i++) {\r
+ av.setSelectionGroup(null);\r
+ av.alignment.deleteAllGroups();\r
+\r
+ for (int i=0; i < tree.getGroups().size(); i++)\r
+ {\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),col.brighter());\r
\r
- // l is vector of Objects\r
- // Vector l = tree.findLeaves((SequenceNode)tree.getGroups().elementAt(i),new Vector());\r
+ Vector l = tree.findLeaves((SequenceNode)tree.getGroups().elementAt(i),new Vector());\r
+ SequenceGroup sg = null;\r
+ for (int j = 0; j < l.size(); j++)\r
+ {\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
+\r
+ sg.addSequence( (Sequence) sn.element());\r
+ }\r
+\r
+ if (av.getGlobalColourScheme() instanceof ConservationColourScheme)\r
+ {\r
+ ConservationColourScheme ccs = (ConservationColourScheme) av.getGlobalColourScheme();\r
+ Conservation c = new Conservation("Group",\r
+ ResidueProperties.propHash, 3,\r
+ sg.sequences, sg.getStartRes(),\r
+ sg.getEndRes());\r
+\r
+ c.calculate();\r
+ c.verdict(false, 100);\r
+ ccs = new ConservationColourScheme(c, ccs.cs);\r
+\r
+ sg.cs = ccs;\r
+\r
+ }\r
+\r
+\r
+\r
+ av.alignment.addGroup(sg);\r
\r
}\r
}\r