search function added
[jalview.git] / src / jalview / gui / AlignmentPanel.java
index df0459f..4f56474 100755 (executable)
@@ -2,21 +2,21 @@ package jalview.gui;
 \r
 import jalview.jbgui.GAlignmentPanel;\r
 import java.awt.*;\r
+import java.awt.image.*;\r
 import java.awt.event.*;\r
-import java.io.*;\r
+import java.awt.print.*;\r
+import javax.swing.*;\r
 \r
-public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListener\r
+public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListener, Printable\r
 {\r
 \r
   AlignViewport     av;\r
   OverviewPanel overviewPanel;\r
-  public SeqPanel   seqPanel;\r
-  public IdPanel    idPanel;\r
+  SeqPanel   seqPanel;\r
+  IdPanel    idPanel;\r
   public AlignFrame alignFrame;\r
-  public ScalePanel scalePanel;\r
-  public ScorePanel scorePanel;\r
-\r
-  public boolean    groupEdit = false;\r
+  ScalePanel scalePanel;\r
+  ScorePanel scorePanel;\r
 \r
   public AlignmentPanel(AlignFrame af, AlignViewport av)\r
   {\r
@@ -27,7 +27,6 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     scalePanel = new ScalePanel(av);\r
     scorePanel = new ScorePanel(av);\r
 \r
-\r
     idPanelHolder.add(idPanel, BorderLayout.CENTER);\r
     idPanel.addNotify();\r
     scalePanelHolder.add(scalePanel, BorderLayout.CENTER);\r
@@ -37,37 +36,27 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 \r
     hscroll.addAdjustmentListener(this);\r
     vscroll.addAdjustmentListener(this);\r
-    selectAll(false);\r
 \r
     addComponentListener(new ComponentAdapter()\r
    {\r
        public void componentResized(ComponentEvent evt)\r
      {\r
-     //  if(!initialising)\r
-     RefreshPanels();\r
+          RefreshPanels();\r
      }\r
    });\r
-\r
   }\r
 \r
-  public int countSelected()\r
-  {\r
-    return av.getSelection().size();\r
-  }\r
+ public void highlightSearchResults(int [] results)\r
+ {\r
+   seqPanel.seqCanvas.highlightSearchResults( results );\r
+\r
+   // do we need to scroll the panel?\r
+   if(av.getStartSeq()>results[0] || av.getEndSeq()<results[0]  || av.getStartRes()>results[1] || av.getEndRes()<results[2])\r
+       setScrollValues(results[1], results[0]);\r
+\r
+\r
+ }\r
 \r
-  public void selectAll(boolean flag) {\r
-    int i = 0;\r
-    if (flag) {\r
-      while (i < seqPanel.align.getHeight()) {\r
-        if (! av.getSelection().contains(av.getAlignment().getSequenceAt(i))) {\r
-          av.getSelection().addElement(av.getAlignment().getSequenceAt(i));\r
-        }\r
-        i++;\r
-      }\r
-    } else {\r
-      av.getSelection().clear();\r
-    }\r
-  }\r
 \r
  public OverviewPanel getOverviewPanel()\r
  {\r
@@ -87,9 +76,20 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     RefreshPanels();\r
   }\r
 \r
-  public void setColourScheme(jalview.schemes.ColourSchemeI cs)\r
+  public void setWrapAlignment(boolean wrap)\r
+  {\r
+      scalePanelHolder.setVisible(!wrap);\r
+      scorePanelHolder.setVisible(!wrap);\r
+\r
+      hscroll.setVisible(!wrap);\r
+      idSpaceFillerPanel.setVisible(!wrap);\r
+      idSpaceFillerPanel1.setVisible(!wrap);\r
+  }\r
+\r
+\r
+  public void setColourScheme(jalview.schemes.ColourSchemeI cs, boolean conservation)\r
   {\r
-    seqPanel.setColourScheme(cs);\r
+    seqPanel.setColourScheme(cs, conservation);\r
     if(overviewPanel!=null)\r
       overviewPanel.updateOverviewImage();\r
   }\r
@@ -97,17 +97,11 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 \r
   public void RefreshPanels()\r
   {\r
-\r
-  /*  try{\r
-      ObjectOutputStream out = new ObjectOutputStream(new FileOutputStream("dump.dat"));\r
-      out.writeObject( (jalview.datamodel.Alignment)av.getAlignment() );\r
-    }catch(IOException ex){ex.printStackTrace();}*/\r
-\r
     invalidate();\r
     validate();\r
+    seqPanel.seqCanvas.paintFlag = true;\r
     idPanel.idCanvas.paintFlag = true;\r
     idPanelHolder.setPreferredSize( idPanel.idCanvas.getPreferredSize() );\r
-    seqPanel.seqCanvas.paintFlag = true;\r
     setScrollValues(av.getStartRes(), av.getStartSeq());\r
     av.getConsensus(true);\r
     repaint();\r
@@ -139,7 +133,6 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 \r
     hscroll.setValues(x,hextent,0,av.getAlignment().getWidth());\r
     vscroll.setValues(y,vextent,0,av.getAlignment().getHeight() );\r
-    seqPanel.seqCanvas.paintFlag = true;\r
     repaint();\r
 \r
   }\r
@@ -166,10 +159,90 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     if(overviewPanel!=null)\r
       overviewPanel.setBoxPosition();\r
 \r
-\r
+    seqPanel.seqCanvas.paintFlag=true;\r
     repaint();\r
   }\r
 \r
 \r
+  public int print(Graphics pg, PageFormat pf, int pi) throws PrinterException\r
+  {\r
+\r
+    pg.translate((int)pf.getImageableX(), (int)pf.getImageableY());\r
+\r
+    int pwidth = (int)pf.getImageableWidth();\r
+    int pheight = (int)pf.getImageableHeight();\r
+    int idWidth = (int)idPanel.idCanvas.getLabelWidth().getWidth();\r
+\r
+    //BufferedImage printImage = new BufferedImage(pwidth,pheight,BufferedImage.TYPE_INT_RGB);\r
+   // Graphics pg = printImage.getGraphics();\r
+    pg.setColor(Color.white);\r
+    pg.fillRect(0,0,pwidth, pheight);\r
+    pg.setFont( av.getFont() );\r
+\r
+    ////////////////////////////////////\r
+    /// How many sequences and residues can we fit on a printable page?\r
+    int totalRes = (int)((pwidth - idWidth)/av.getCharWidth());\r
+    int totalSeq = (int)((pheight - 30)/av.getCharHeight())-1;\r
+    int pagesWide = av.getAlignment().getWidth() / totalRes +1;\r
+    int pagesHigh = av.getAlignment().getHeight() / totalSeq +1;\r
+\r
+    if (pi >= pagesWide*pagesHigh)\r
+     return Printable.NO_SUCH_PAGE;\r
+\r
+    /////////////////////////////\r
+    /// Only print these sequences and residues on this page\r
+    int startRes, endRes, startSeq, endSeq;\r
+    startRes = (pi % pagesWide) * totalRes;\r
+    endRes = startRes + totalRes-1;\r
+    if(endRes>av.getAlignment().getWidth())\r
+      endRes = av.getAlignment().getWidth();\r
+\r
+     startSeq = (pi / pagesWide) * totalSeq;\r
+     endSeq = startSeq + totalSeq;\r
+     if(endSeq > av.getAlignment().getHeight())\r
+       endSeq = av.getAlignment().getHeight();\r
+\r
+\r
+    ////////////////\r
+    //draw Scale\r
+    pg.translate(idWidth,0);\r
+    scalePanel.scaleCanvas.drawScale(pg, startRes, endRes, pwidth-idWidth);\r
+\r
+    pg.translate(-idWidth, 30);\r
+    ////////////////\r
+    // Draw the ids\r
+    Color currentColor=null;\r
+    Color currentTextColor=null;\r
+    for(int i=startSeq; i<endSeq; i++)\r
+    {\r
+      if (av.getSelection().contains(av.getAlignment().getSequenceAt(i)))\r
+      {\r
+        currentColor = Color.gray;\r
+        currentTextColor = Color.black;\r
+      }\r
+      else\r
+      {\r
+        currentColor = av.getAlignment().getSequenceAt(i).getColor();\r
+        currentTextColor = Color.black;\r
+      }\r
+\r
+      pg.setColor(currentColor);\r
+      pg.fillRect(0,  jalview.analysis.AlignmentUtil.getPixelHeight(startSeq, i, av.getCharHeight()),\r
+                              getWidth(),\r
+                              av.getCharHeight());\r
 \r
+      pg.setColor(currentTextColor);\r
+\r
+      String string = av.getAlignment().getSequenceAt(i).getDisplayId();\r
+      pg.drawString(string, 0,  jalview.analysis.AlignmentUtil.getPixelHeight\r
+                    (startSeq, i, av.getCharHeight()) + av.getCharHeight() - (av.getCharHeight() / 5));\r
+    }\r
+\r
+    // draw main sequence panel\r
+    pg.translate(idWidth,0);\r
+    seqPanel.seqCanvas.drawPanel(pg,startRes,endRes,startSeq,endSeq,startRes,startSeq,0);\r
+\r
+    return Printable.PAGE_EXISTS;\r
+  }\r
 }\r
+\r