correct eps png bug missing last 2 lines
[jalview.git] / src / jalview / gui / AlignmentPanel.java
index 3ce9fa1..76fc8d5 100755 (executable)
@@ -7,6 +7,12 @@ import jalview.datamodel.*;
 import java.awt.*;\r
 import java.awt.event.*;\r
 import java.awt.print.*;\r
+import java.io.*;\r
+import java.awt.image.*;\r
+import org.jibble.epsgraphics.*;\r
+import javax.imageio.*;\r
+\r
+\r
 \r
 public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListener, Printable\r
 {\r
@@ -26,13 +32,22 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     this.av         = av;\r
     seqPanel        = new SeqPanel  (av, this);\r
     idPanel         = new IdPanel   (av, this);\r
+\r
     scalePanel = new ScalePanel(av, this);\r
     scorePanel = new ScorePanel(av);\r
     ssPanel = new SecondaryStructurePanel(av);\r
 \r
     secondaryPanelHolder.add(ssPanel, BorderLayout.CENTER);\r
+\r
     idPanelHolder.add(idPanel, BorderLayout.CENTER);\r
-    idPanel.addNotify();\r
+    IdwidthAdjuster iap = new IdwidthAdjuster(this);\r
+    idSpaceFillerPanel1.add(iap, BorderLayout.CENTER);\r
+\r
+    Dimension d = calculateIdWidth();\r
+    d.setSize( d.width+4, d.height);\r
+    idPanel.idCanvas.setPreferredSize( d );\r
+    hscrollFillerPanel.setPreferredSize( d );\r
+\r
     scalePanelHolder.add(scalePanel, BorderLayout.CENTER);\r
     scorePanelHolder.add(scorePanel, BorderLayout.CENTER);\r
     seqPanelHolder.add(seqPanel, BorderLayout.CENTER);\r
@@ -50,8 +65,8 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
    });\r
 \r
 \r
-   hscroll.setFocusable(false);\r
-   vscroll.setFocusable(false);\r
+  // hscroll.setFocusable(false);\r
+  // vscroll.setFocusable(false);\r
    setFocusable(true);\r
 \r
    addKeyListener(new KeyAdapter()\r
@@ -61,7 +76,7 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
        switch(evt.getKeyCode())\r
        {\r
          case  27: // escape key\r
-           av.setRubberbandGroup(null);\r
+           av.setSelectionGroup(null);\r
            RefreshPanels();\r
            break;\r
          case KeyEvent.VK_X:\r
@@ -90,6 +105,38 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
    });\r
   }\r
 \r
+  Dimension calculateIdWidth()\r
+  {\r
+    Graphics g = this.getGraphics();\r
+    if(g==null)\r
+    {\r
+      javax.swing.JFrame f = new javax.swing.JFrame();\r
+      f.addNotify();\r
+      g = f.getGraphics();\r
+    }\r
+\r
+    FontMetrics fm = g.getFontMetrics(av.font);\r
+    AlignmentI al = av.getAlignment();\r
+\r
+       int i   = 0;\r
+       int idWidth = 0;\r
+       String id;\r
+       while (i < al.getHeight() && al.getSequenceAt(i) != null)\r
+       {\r
+         SequenceI s   = al.getSequenceAt(i);\r
+         if(av.getShowFullId())\r
+           id   = s.getDisplayId();\r
+         else\r
+           id = s.getName();\r
+\r
+         if (fm.stringWidth(id) > idWidth)\r
+           idWidth = fm.stringWidth(id);\r
+         i++;\r
+       }\r
+\r
+       return new Dimension(idWidth, 12);\r
+  }\r
+\r
 \r
  public void highlightSearchResults(int [] results)\r
  {\r
@@ -174,34 +221,75 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 \r
   public void RefreshPanels()\r
   {\r
-    requestFocus();\r
-    invalidate();\r
-    seqPanel.seqCanvas.paintFlag = true;\r
-    idPanel.idCanvas.paintFlag = true;\r
-    idPanelHolder.setPreferredSize( idPanel.idCanvas.getPreferredSize() );\r
+              requestFocus();\r
+              invalidate();\r
+              idPanelHolder.setPreferredSize(idPanel.idCanvas.getPreferredSize());\r
+              if(idPanel.idCanvas.getPreferredSize()!=null)\r
+                hscrollFillerPanel.setPreferredSize(new Dimension(idPanel.idCanvas.getPreferredSize().width, 12));\r
+              idSpaceFillerPanel1.setPreferredSize(new Dimension(500,\r
+                  av.charHeight / 2 + 12));\r
+              scalePanelHolder.setPreferredSize(new Dimension(500,\r
+                  av.charHeight / 2 + 12));\r
+              if (av.getWrapAlignment())\r
+              {\r
+                int max = av.alignment.getWidth() /\r
+                    (seqPanel.seqCanvas.getWidth() / av.charWidth) + 1;\r
+\r
+                int h = (av.alignment.getHeight() + 2) * av.charHeight;\r
+                vextent = seqPanel.seqCanvas.getHeight() / h;\r
+                vscroll.setValues(0, vextent, 0, max);\r
+              }\r
+              else\r
+\r
+                setScrollValues(av.getStartRes(), av.getStartSeq());\r
+              av.getConsensus(true);\r
+              if (overviewPanel != null)\r
+                overviewPanel.updateOverviewImage();\r
+\r
+          validate();\r
+          repaint();\r
+  }\r
+  int hextent = 0;\r
+  int vextent = 0;\r
 \r
-    if(av.getWrapAlignment())\r
-    {\r
-      int max = av.alignment.getWidth()/ (seqPanel.seqCanvas.getWidth()/av.charWidth)+1;\r
 \r
-      int h  = (av.alignment.getHeight() + 2)*av.charHeight;\r
-      vextent = seqPanel.seqCanvas.getHeight()/h;\r
-      vscroll.setValues(0,vextent,0,max);\r
+  // return value is true if the scroll is valid\r
+  public boolean scrollUp(boolean up)\r
+  {\r
+    if(up)\r
+    {\r
+      if(vscroll.getValue()<1)\r
+        return false;\r
+      vscroll.setValue(vscroll.getValue() - 1);\r
     }\r
     else\r
+    {\r
+     if(vextent+vscroll.getValue() >= av.getAlignment().getHeight())\r
+       return false;\r
+      vscroll.setValue(vscroll.getValue() + 1);\r
+    }\r
 \r
-    setScrollValues(av.getStartRes(), av.getStartSeq());\r
-    av.getConsensus(true);\r
-    if(overviewPanel!=null)\r
-      overviewPanel.updateOverviewImage();\r
+    return true;\r
+  }\r
 \r
-    validate();\r
-    repaint();\r
+  public boolean scrollRight(boolean right)\r
+  {\r
+    if(right)\r
+   {\r
+     if(hscroll.getValue()<1)\r
+       return false;\r
+     hscroll.setValue(hscroll.getValue() - 1);\r
+   }\r
+   else\r
+   {\r
+    if(hextent+hscroll.getValue() >= av.getAlignment().getWidth())\r
+      return false;\r
+     hscroll.setValue(hscroll.getValue() + 1);\r
+   }\r
 \r
-  }\r
+   return true;\r
+ }\r
 \r
-  int hextent = 0;\r
-  int vextent = 0;\r
 \r
   public void setScrollValues(int x, int y)\r
   {\r
@@ -245,14 +333,6 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
       if (av.getWrapAlignment())\r
       {\r
         av.setStartRes( vscroll.getValue() * av.getChunkWidth());\r
-        // System.out.println(vscroll.getValue()+" "+ av.getChunkWidth());\r
-      //  int resSpan = av.alignment.getWidth()/ (seqPanel.seqCanvas.getWidth()/av.charWidth)+1;\r
-     //   int h  = (av.alignment.getHeight() + 2)*av.charHeight;\r
-      //  vextent = seqPanel.seqCanvas.getHeight()/h;\r
-     //   vscroll.setValues(0,vextent,0,max);\r
-\r
-\r
-      // av.setStartSeq(  );\r
       }\r
       else\r
       {\r
@@ -265,24 +345,26 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     if(overviewPanel!=null)\r
       overviewPanel.setBoxPosition();\r
 \r
-    seqPanel.seqCanvas.paintFlag=true;\r
     repaint();\r
   }\r
 \r
-\r
-\r
   public int print(Graphics pg, PageFormat pf, int pi) throws PrinterException\r
   {\r
-    if(av.getWrapAlignment())\r
-    {\r
-     return PrintWrappedAlignment(pg, pf, pi);\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
+    int pwidth = (int) pf.getImageableWidth();\r
+    int pheight = (int) pf.getImageableHeight();\r
+\r
+    if (av.getWrapAlignment())\r
+      return printWrappedAlignment(pg, pwidth,pheight, pi);\r
+    else\r
+      return printUnwrapped(pg,pwidth, pheight,pi);\r
+  }\r
+\r
+  public int printUnwrapped(Graphics pg, int pwidth, int pheight, int pi) throws PrinterException\r
+  {\r
+\r
+    int idWidth = calculateIdWidth().width;\r
 \r
 \r
     pg.setColor(Color.white);\r
@@ -315,17 +397,17 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 \r
     ////////////////\r
     //draw Scale\r
-    pg.translate(idWidth,0);\r
-    scalePanel.scaleCanvas.drawScale(pg, startRes, endRes, pwidth-idWidth);\r
+    pg.translate(30,0);\r
+    scalePanel.drawScale(pg, startRes, endRes, pwidth-idWidth);\r
 \r
-    pg.translate(-idWidth, 30);\r
+    pg.translate(-30, 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
+      if (av.getSelectionGroup()!=null && av.getSelectionGroup().sequences.contains(av.getAlignment().getSequenceAt(i)))\r
       {\r
         currentColor = Color.gray;\r
         currentTextColor = Color.black;\r
@@ -343,46 +425,32 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 \r
       pg.setColor(currentTextColor);\r
 \r
-      String string = av.getAlignment().getSequenceAt(i).getDisplayId();\r
+      String string = av.getAlignment().getSequenceAt(i).getName();\r
+      if(av.getShowFullId())\r
+        string = av.getAlignment().getSequenceAt(i).getDisplayId();\r
+\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
+    pg.translate(idWidth+4,0);\r
     pg.setClip(0,0,pwidth-idWidth, pheight);\r
     seqPanel.seqCanvas.drawPanel(pg,startRes,endRes,startSeq,endSeq,startRes,startSeq,0);\r
 \r
     return Printable.PAGE_EXISTS;\r
   }\r
 \r
-  class VPanel extends javax.swing.JPanel\r
-  {\r
-    Image image;\r
-    public VPanel(Image i)\r
-    {\r
-      image = i;\r
-      repaint();\r
-    }\r
-    public void paintComponent(Graphics g)\r
-    {\r
-      if(image!=null)\r
-        g.drawImage(image,0,0,this);\r
-    }\r
-\r
-  }\r
 \r
-  public int PrintWrappedAlignment(Graphics pg, PageFormat pf, int pi)\r
+  public int printWrappedAlignment(Graphics pg, int pwidth, int pheight, int pi) throws PrinterException\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
+    int idWidth = calculateIdWidth().width;\r
 \r
     if( seqPanel.seqCanvas.getWidth() < pwidth-idWidth)\r
       pwidth = seqPanel.seqCanvas.getWidth() + idWidth;\r
 \r
+\r
     pg.setColor(Color.white);\r
     pg.fillRect(0,0,pwidth, pheight);\r
     pg.setFont( av.getFont() );\r
@@ -395,7 +463,7 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     int chunkWidth  =   (pwidth-idWidth)/av.charWidth;\r
 \r
     int noChunksOnPage = pheight / chunkHeight;\r
-    int totalChunks = endy / chunkHeight;\r
+    int totalChunks = da.getWidth() / chunkWidth;\r
 \r
     if ( pi*noChunksOnPage > totalChunks )\r
      return Printable.NO_SUCH_PAGE;\r
@@ -413,21 +481,116 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
       for (int i = 0; i < endy; i++)\r
       {\r
         SequenceI s = da.getSequenceAt(i);\r
-        pg.drawString(s.getDisplayId(), 0,\r
+        String string = s.getName();\r
+        if (av.getShowFullId())\r
+          string = s.getDisplayId();\r
+\r
+        pg.drawString(string, 0,\r
                       AlignmentUtil.getPixelHeight(0, i, av.charHeight) + ypos +\r
                       av.charHeight - (av.charHeight / 5));\r
       }\r
     }\r
 \r
     // draw main sequence panel\r
-    pg.translate(idWidth,0);\r
+    pg.translate(idWidth+4,0);\r
     seqPanel.seqCanvas.drawWrappedPanel(pg, pwidth-idWidth, pheight, pi*noChunksOnPage*chunkWidth);\r
 \r
 \r
-    return Printable.NO_SUCH_PAGE;//.PAGE_EXISTS;\r
+    return Printable.PAGE_EXISTS;\r
+\r
+  }\r
+\r
+\r
+  public void makeEPS()\r
+  {\r
+    int height = (av.alignment.getWidth() / av.getChunkWidth() +1) * av.chunkHeight;\r
+    int width = seqPanel.getWidth() + idPanel.getWidth();\r
+\r
+    if (!av.getWrapAlignment())\r
+    {\r
+      height = (av.alignment.getHeight()+1) * av.charHeight + 30;\r
+      width = idPanel.getWidth() + av.alignment.getWidth() * av.charWidth;\r
+    }\r
+\r
+    try\r
+    {\r
+      jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(\r
+          "LAST_DIRECTORY"), new String[]{"eps"}, "Encapsulated Postscript");\r
+      chooser.setFileView(new jalview.io.JalviewFileView());\r
+      chooser.setDialogTitle("Create EPS file from alignment");\r
+      chooser.setToolTipText("Save");\r
+\r
+      int value = chooser.showSaveDialog(this);\r
+      if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
+        return;\r
+\r
+      jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getPath());\r
+      FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
+      EpsGraphics2D pg = new EpsGraphics2D("Example", out, 0, 0, width, height);\r
+\r
+        if (av.getWrapAlignment())\r
+          printWrappedAlignment(pg, width, height, 0);\r
+        else\r
+          printUnwrapped(pg, width, height, 0);\r
+\r
 \r
+        pg.flush();\r
+        pg.close();\r
+    }\r
+    catch (Exception ex)\r
+    {\r
+      ex.printStackTrace();\r
+    }\r
+  }\r
+\r
+  public void makePNG()\r
+  {\r
+      int height = (av.alignment.getWidth() / av.getChunkWidth() +1) * av.chunkHeight;\r
+      int width = seqPanel.getWidth() + idPanel.getWidth();\r
+\r
+      if (!av.getWrapAlignment())\r
+      {\r
+        height = (av.alignment.getHeight()+1) * av.charHeight + 30;\r
+        width = idPanel.getWidth() + av.alignment.getWidth() * av.charWidth;\r
+      }\r
+\r
+\r
+\r
+    try\r
+    {\r
+      jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(\r
+          "LAST_DIRECTORY"), new String[]{"png"}, "Portable network graphics");\r
+      chooser.setFileView(new jalview.io.JalviewFileView());\r
+      chooser.setDialogTitle("Create EPS file from alignment");\r
+      chooser.setToolTipText("Save");\r
+\r
+      int value = chooser.showSaveDialog(this);\r
+      if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
+        return;\r
+\r
+      jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getPath());\r
+      FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
+\r
+      BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);\r
+      Graphics png = bi.getGraphics();\r
+\r
+\r
+        if (av.getWrapAlignment())\r
+          printWrappedAlignment(png, width, height, 0);\r
+        else\r
+          printUnwrapped(png, width, height, 0);\r
+\r
+        ImageIO.write(bi, "png", out);\r
+        out.close();\r
+    }\r
+    catch (Exception ex)\r
+    {\r
+      ex.printStackTrace();\r
+    }\r
   }\r
+\r
 }\r
 \r
 \r
 \r
+\r