X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FAlignmentPanel.java;h=73cee2d7fefd691fa2c83a007479c257519409b3;hb=7ee3b5a7f099858cdde2318c719121b39d9e466e;hp=3f7a78164da803c1748044fd9ba7107422bb354e;hpb=44d9bd9b511490baec5302be3b7756be0adebde3;p=jalview.git

diff --git a/src/jalview/gui/AlignmentPanel.java b/src/jalview/gui/AlignmentPanel.java
index 3f7a781..73cee2d 100755
--- a/src/jalview/gui/AlignmentPanel.java
+++ b/src/jalview/gui/AlignmentPanel.java
@@ -7,6 +7,12 @@ import jalview.datamodel.*;
 import java.awt.*;
 import java.awt.event.*;
 import java.awt.print.*;
+import java.io.*;
+import java.awt.image.*;
+import org.jibble.epsgraphics.*;
+import javax.imageio.*;
+
+
 
 public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListener, Printable
 {
@@ -15,41 +21,159 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
   OverviewPanel overviewPanel;
   SeqPanel   seqPanel;
   IdPanel    idPanel;
-  SecondaryStructurePanel ssPanel;
+  IdwidthAdjuster idwidthAdjuster;
   public AlignFrame alignFrame;
   ScalePanel scalePanel;
-  ScorePanel scorePanel;
+  AnnotationPanel annotationPanel;
+  AnnotationLabels alabels;
+
+  // this value is set false when selection area being dragged
+  boolean fastPaint = true;
 
-  public AlignmentPanel(AlignFrame af, AlignViewport av)
+  public AlignmentPanel(AlignFrame af, final AlignViewport av)
   {
     alignFrame = af;
     this.av         = av;
     seqPanel        = new SeqPanel  (av, this);
     idPanel         = new IdPanel   (av, this);
+
     scalePanel = new ScalePanel(av, this);
-    scorePanel = new ScorePanel(av);
-    ssPanel = new SecondaryStructurePanel(av);
 
-    secondaryPanelHolder.add(ssPanel, BorderLayout.CENTER);
     idPanelHolder.add(idPanel, BorderLayout.CENTER);
-    idPanel.addNotify();
+    idwidthAdjuster = new IdwidthAdjuster(this);
+    idSpaceFillerPanel1.add(idwidthAdjuster, BorderLayout.CENTER);
+
+    annotationPanel = new AnnotationPanel(this);
+    alabels = new AnnotationLabels(this);
+
+    annotationSpaceFillerHolder.setPreferredSize(annotationPanel.getPreferredSize());
+    annotationScroller.setPreferredSize(annotationPanel.getPreferredSize());
+    annotationScroller.setViewportView(annotationPanel);
+    annotationSpaceFillerHolder.add(alabels, BorderLayout.CENTER);
+
+    Dimension d = calculateIdWidth();
+    d.setSize( d.width+4, d.height);
+    idPanel.idCanvas.setPreferredSize( d );
+    hscrollFillerPanel.setPreferredSize( d );
+
     scalePanelHolder.add(scalePanel, BorderLayout.CENTER);
-    scorePanelHolder.add(scorePanel, BorderLayout.CENTER);
     seqPanelHolder.add(seqPanel, BorderLayout.CENTER);
-    setScrollValues(0,0);
+
+
+    if(seqPanel.seqCanvas.getWidth()==0)
+    javax.swing.SwingUtilities.invokeLater(new Runnable()
+     {
+       public void run()
+       {
+         setScrollValues(0,0);
+         Dimension d = calculateIdWidth();
+         d.setSize( d.width+4, d.height);
+         idPanel.idCanvas.setPreferredSize( d );
+         RefreshPanels();
+       }
+    });
+
 
     hscroll.addAdjustmentListener(this);
     vscroll.addAdjustmentListener(this);
 
     addComponentListener(new ComponentAdapter()
    {
-       public void componentResized(ComponentEvent evt)
+     public void componentResized(ComponentEvent evt)
      {
           RefreshPanels();
      }
    });
+
+   setFocusable(true);
+   addKeyListener(new KeyAdapter()
+   {
+     public void keyPressed(KeyEvent evt)
+     {
+       switch(evt.getKeyCode())
+       {
+         case  27: // escape key
+           av.setSelectionGroup(null);
+           RefreshPanels();
+           break;
+         case KeyEvent.VK_X:
+           if(evt.isControlDown())
+           alignFrame.cut_actionPerformed(null);
+           break;
+         case KeyEvent.VK_C:
+         if(evt.isControlDown())
+           alignFrame.copy_actionPerformed(null);
+           break;
+         case KeyEvent.VK_V:
+          if(evt.isControlDown())
+           alignFrame.paste(true);
+           break;
+         case KeyEvent.VK_A:
+         if(evt.isControlDown())
+           alignFrame.selectAllSequenceMenuItem_actionPerformed(null);
+           break;
+        case KeyEvent.VK_DOWN:
+          alignFrame.moveSelectedSequences(false);
+          break;
+        case KeyEvent.VK_UP:
+          alignFrame.moveSelectedSequences(true);
+          break;
+        case KeyEvent.VK_F:
+         if(evt.isControlDown())
+          alignFrame.findMenuItem_actionPerformed(null);
+          break;
+       }
+     }
+   });
   }
 
+  Dimension calculateIdWidth()
+  {
+    Graphics g = this.getGraphics();
+    if(g==null)
+    {
+      javax.swing.JFrame f = new javax.swing.JFrame();
+      f.addNotify();
+      g = f.getGraphics();
+    }
+
+    FontMetrics fm = g.getFontMetrics(av.font);
+    AlignmentI al = av.getAlignment();
+
+       int i   = 0;
+       int idWidth = 0;
+       String id;
+       while (i < al.getHeight() && al.getSequenceAt(i) != null)
+       {
+         SequenceI s   = al.getSequenceAt(i);
+         if(av.getShowFullId())
+           id   = s.getDisplayId();
+         else
+           id = s.getName();
+
+         if (fm.stringWidth(id) > idWidth)
+           idWidth = fm.stringWidth(id);
+         i++;
+       }
+
+       // Also check annotation label widths
+       i=0;
+       if(al.getAlignmentAnnotation()!=null)
+       {
+         fm = g.getFontMetrics(alabels.getFont());
+         while (i < al.getAlignmentAnnotation().length)
+         {
+           String label = al.getAlignmentAnnotation()[i].label;
+           if (fm.stringWidth(label) > idWidth)
+             idWidth = fm.stringWidth(label);
+           i++;
+         }
+       }
+
+       return new Dimension(idWidth, 12);
+  }
+
+
  public void highlightSearchResults(int [] results)
  {
    seqPanel.seqCanvas.highlightSearchResults( results );
@@ -76,30 +200,39 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
  }
 
 
-  public void setGraphPanelVisible(boolean b)
+  public void setAnnotationVisible(boolean b)
   {
-    idSpaceFillerPanel.setVisible(b);
-    scorePanelHolder.setVisible(b);
+    annotationSpaceFillerHolder.setVisible(b);
+    annotationScroller.setVisible(b);
+    javax.swing.SwingUtilities.invokeLater(new Runnable()
+     {
+       public void run()
+       {
+         RefreshPanels();
+       }
+    });
 
-    RefreshPanels();
-    // bit annoying to call this twice, can you do better?
-    RefreshPanels();
   }
 
-  public void setSecondaryStructureVisible(boolean b)
-  {
-    secondaryPanelHolder.setVisible(b);
-    RefreshPanels();
-  }
 
   public void setWrapAlignment(boolean wrap)
   {
-      scalePanelHolder.setVisible(!wrap);
-      scorePanelHolder.setVisible(!wrap);
+    scalePanelHolder.setVisible(!wrap);
+    hscroll.setVisible(!wrap);
+    idwidthAdjuster.setVisible(!wrap);
+
+    av.setShowAnnotation(!wrap);
+    annotationScroller.setVisible(!wrap);
+    annotationSpaceFillerHolder.setVisible(!wrap);
+    idSpaceFillerPanel1.setVisible(!wrap);
+    javax.swing.SwingUtilities.invokeLater(new Runnable()
+     {
+       public void run()
+       {
+         RefreshPanels();
+       }
+    });
 
-      hscroll.setVisible(!wrap);
-      idSpaceFillerPanel.setVisible(!wrap);
-      idSpaceFillerPanel1.setVisible(!wrap);
   }
 
 
@@ -126,17 +259,27 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     RefreshPanels();
   }
 
+
   public void RefreshPanels()
   {
-
+    requestFocus();
     invalidate();
-    seqPanel.seqCanvas.paintFlag = true;
-    idPanel.idCanvas.paintFlag = true;
-    idPanelHolder.setPreferredSize( idPanel.idCanvas.getPreferredSize() );
-    setScrollValues(av.getStartRes(), av.getStartSeq());
-    av.getConsensus(true);
-    if(overviewPanel!=null)
-      overviewPanel.updateOverviewImage();
+
+    Dimension d = idPanel.idCanvas.getPreferredSize();
+    idPanelHolder.setPreferredSize(d);
+    hscrollFillerPanel.setPreferredSize(new Dimension(d.width, 12));
+
+    if (av.getWrapAlignment())
+    {
+      int max = av.alignment.getWidth() / seqPanel.seqCanvas.getWrappedCanvasWidth(seqPanel.seqCanvas.getWidth());
+      vscroll.setValues(0, 1, 0, max);
+    }
+    else
+    {
+      if (overviewPanel != null)
+        overviewPanel.updateOverviewImage();
+      setScrollValues(av.getStartRes(), av.getStartSeq());
+    }
 
     validate();
     repaint();
@@ -145,10 +288,58 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
   int hextent = 0;
   int vextent = 0;
 
+  // return value is true if the scroll is valid
+  public boolean scrollUp(boolean up)
+  {
+    if(up)
+    {
+      if(vscroll.getValue()<1)
+        return false;
+      fastPaint  = false;
+      vscroll.setValue(vscroll.getValue() - 1);
+    }
+    else
+    {
+     if(vextent+vscroll.getValue() >= av.getAlignment().getHeight())
+       return false;
+      fastPaint  = false;
+      vscroll.setValue(vscroll.getValue() + 1);
+    }
+    fastPaint = true;
+    return true;
+  }
+
+  public boolean scrollRight(boolean right)
+  {
+
+    if (right)
+    {
+      if (hscroll.getValue() < 1)
+        return false;
+      fastPaint = false;
+      hscroll.setValue(hscroll.getValue() - 1);
+    }
+    else
+    {
+      if (hextent + hscroll.getValue() >= av.getAlignment().getWidth())
+        return false;
+      fastPaint = false;
+      hscroll.setValue(hscroll.getValue() + 1);
+    }
+    fastPaint = true;
+    return true;
+  }
+
+
   public void setScrollValues(int x, int y)
   {
-    hextent = seqPanel.seqCanvas.getWidth()/av.getCharWidth();
-    vextent = seqPanel.seqCanvas.getHeight()/av.getCharHeight();
+    hextent = seqPanel.seqCanvas.getWidth()/av.charWidth;
+    vextent = seqPanel.seqCanvas.getHeight()/av.charHeight;
+
+    if(hextent > av.alignment.getWidth())
+      hextent = av.alignment.getWidth();
+    if(vextent > av.alignment.getHeight())
+      vextent = av.alignment.getHeight();
 
     if(hextent+x  >  av.getAlignment().getWidth())
       x =  av.getAlignment().getWidth()- hextent;
@@ -165,14 +356,13 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     hscroll.setValues(x,hextent,0,av.getAlignment().getWidth());
     vscroll.setValues(y,vextent,0,av.getAlignment().getHeight() );
 
-
-    repaint();
-
   }
 
 
   public void adjustmentValueChanged(AdjustmentEvent evt)
   {
+    int oldX = av.getStartRes();
+    int oldY = av.getStartSeq();
 
     if (evt.getSource() == hscroll)
     {
@@ -183,28 +373,56 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 
     if (evt.getSource() == vscroll)
     {
-        int offy = vscroll.getValue();
+      int offy = vscroll.getValue();
+      if (av.getWrapAlignment())
+      {
+        int rowSize = seqPanel.seqCanvas.getWrappedCanvasWidth(seqPanel.seqCanvas.getWidth());
+        av.setStartRes( vscroll.getValue() * rowSize );
+        av.setEndRes( (vscroll.getValue()+1) * rowSize );
+      }
+      else
+      {
         av.setStartSeq(offy);
-        av.setEndSeq(offy + seqPanel.seqCanvas.getHeight()/av.getCharHeight());
+        av.setEndSeq(offy + seqPanel.seqCanvas.getHeight() / av.getCharHeight());
+      }
     }
 
 
     if(overviewPanel!=null)
       overviewPanel.setBoxPosition();
 
-    seqPanel.seqCanvas.paintFlag=true;
-    repaint();
-  }
+    if(av.getWrapAlignment() || !fastPaint)
+      repaint();
+    else
+    {
+      seqPanel.seqCanvas.fastPaint(av.getStartRes() - oldX,
+                                   av.getStartSeq() - oldY);
+      idPanel.idCanvas.fastPaint(av.getStartSeq() - oldY);
+      scalePanel.repaint();
+      if (av.getShowAnnotation())
+        annotationPanel.fastPaint(av.getStartRes() - oldX);
+    }
 
+  }
 
   public int print(Graphics pg, PageFormat pf, int pi) throws PrinterException
   {
-
     pg.translate((int)pf.getImageableX(), (int)pf.getImageableY());
 
-    int pwidth = (int)pf.getImageableWidth();
-    int pheight = (int)pf.getImageableHeight();
-    int idWidth = (int)idPanel.idCanvas.getLabelWidth().getWidth();
+    int pwidth = (int) pf.getImageableWidth();
+    int pheight = (int) pf.getImageableHeight();
+
+    if (av.getWrapAlignment())
+      return printWrappedAlignment(pg, pwidth,pheight, pi);
+    else
+      return printUnwrapped(pg,pwidth, pheight,pi);
+  }
+
+  public int printUnwrapped(Graphics pg, int pwidth, int pheight, int pi) throws PrinterException
+  {
+
+    int idWidth = calculateIdWidth().width + 4;
+
 
     pg.setColor(Color.white);
     pg.fillRect(0,0,pwidth, pheight);
@@ -237,7 +455,7 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     ////////////////
     //draw Scale
     pg.translate(idWidth,0);
-    scalePanel.scaleCanvas.drawScale(pg, startRes, endRes, pwidth-idWidth);
+    scalePanel.drawScale(pg, startRes, endRes, pwidth-idWidth);
 
     pg.translate(-idWidth, 30);
     ////////////////
@@ -246,7 +464,7 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
     Color currentTextColor=null;
     for(int i=startSeq; i<endSeq; i++)
     {
-      if (av.getSelection().contains(av.getAlignment().getSequenceAt(i)))
+      if (av.getSelectionGroup()!=null && av.getSelectionGroup().sequences.contains(av.getAlignment().getSequenceAt(i)))
       {
         currentColor = Color.gray;
         currentTextColor = Color.black;
@@ -264,20 +482,190 @@ public class AlignmentPanel extends GAlignmentPanel implements AdjustmentListene
 
       pg.setColor(currentTextColor);
 
-      String string = av.getAlignment().getSequenceAt(i).getDisplayId();
+      String string = av.getAlignment().getSequenceAt(i).getName();
+      if(av.getShowFullId())
+        string = av.getAlignment().getSequenceAt(i).getDisplayId();
+
       pg.drawString(string, 0,  jalview.analysis.AlignmentUtil.getPixelHeight
                     (startSeq, i, av.getCharHeight()) + av.getCharHeight() - (av.getCharHeight() / 5));
     }
 
     // draw main sequence panel
     pg.translate(idWidth,0);
-    pg.setClip(0,0,pwidth-idWidth, pheight);
     seqPanel.seqCanvas.drawPanel(pg,startRes,endRes,startSeq,endSeq,startRes,startSeq,0);
 
+
+    if(av.getShowAnnotation())
+      {
+        pg.translate(-idWidth,(endSeq-startSeq)*av.charHeight);
+        alabels.drawComponent((Graphics2D)pg);
+        pg.translate(idWidth,0);
+        annotationPanel.drawComponent((Graphics2D) pg, startRes, endRes+1);
+      }
+
     return Printable.PAGE_EXISTS;
   }
 
+
+  public int printWrappedAlignment(Graphics pg, int pwidth, int pheight, int pi) throws PrinterException
+  {
+
+    int idWidth = calculateIdWidth().width+4;
+
+    if( seqPanel.seqCanvas.getWidth() < pwidth-idWidth)
+      pwidth = seqPanel.seqCanvas.getWidth() + idWidth;
+
+
+    pg.setColor(Color.white);
+    pg.fillRect(0,0,pwidth, pheight);
+    pg.setFont( av.getFont() );
+
+    ////////////////////////////////////
+    /// How many sequences and residues can we fit on a printable page?
+    AlignmentI da = av.alignment;
+    int endy   = da.getHeight();
+    int chunkHeight =  (da.getHeight() + 2)*av.charHeight;
+    int chunkWidth  =   (pwidth-idWidth)/av.charWidth;
+
+    int noChunksOnPage = pheight / chunkHeight;
+    int totalChunks = da.getWidth() / chunkWidth;
+
+    if ( pi*noChunksOnPage > totalChunks )
+     return Printable.NO_SUCH_PAGE;
+
+    ////////////////
+    // Draw the ids
+    pg.setClip(0,0,pwidth, noChunksOnPage*chunkHeight);
+
+    int row = pi*noChunksOnPage;
+    pg.setColor(Color.black);
+    for(int ypos=2*av.charHeight;
+        ypos <= pheight && row*chunkWidth<da.getWidth();
+        ypos += chunkHeight, row++ )
+    {
+      for (int i = 0; i < endy; i++)
+      {
+        SequenceI s = da.getSequenceAt(i);
+        String string = s.getName();
+        if (av.getShowFullId())
+          string = s.getDisplayId();
+
+        pg.drawString(string, 0,
+                      AlignmentUtil.getPixelHeight(0, i, av.charHeight) + ypos +
+                      av.charHeight - (av.charHeight / 5));
+      }
+    }
+
+    // draw main sequence panel
+    pg.translate(idWidth,0);
+    seqPanel.seqCanvas.drawWrappedPanel(pg, pwidth-idWidth, pheight, pi*noChunksOnPage*chunkWidth);
+
+
+    return Printable.PAGE_EXISTS;
+
+  }
+
+
+  public void makeEPS()
+  {
+    int height = (av.alignment.getWidth() / av.getChunkWidth() +1) * av.chunkHeight;
+    int width = seqPanel.getWidth() + idPanel.getWidth();
+
+    if (!av.getWrapAlignment())
+    {
+      height = (av.alignment.getHeight()+1) * av.charHeight + 30;
+      width = idPanel.getWidth() + av.alignment.getWidth() * av.charWidth;
+    }
+    if(av.getShowAnnotation())
+   {
+     height += annotationPanel.getPreferredSize().height;
+   }
+
+    try
+    {
+      jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(
+          "LAST_DIRECTORY"), new String[]{"eps"}, "Encapsulated Postscript");
+      chooser.setFileView(new jalview.io.JalviewFileView());
+      chooser.setDialogTitle("Create EPS file from alignment");
+      chooser.setToolTipText("Save");
+
+      int value = chooser.showSaveDialog(this);
+      if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)
+        return;
+
+      jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getPath());
+      FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());
+      EpsGraphics2D pg = new EpsGraphics2D("Example", out, 0, 0, width, height);
+
+        if (av.getWrapAlignment())
+          printWrappedAlignment(pg, width, height, 0);
+        else
+          printUnwrapped(pg, width, height, 0);
+
+
+        pg.flush();
+        pg.close();
+    }
+    catch (Exception ex)
+    {
+      ex.printStackTrace();
+    }
+  }
+
+  public void makePNG()
+  {
+      int height = (av.alignment.getWidth() / av.getChunkWidth() +1) * av.chunkHeight;
+      int width = seqPanel.getWidth() + idPanel.getWidth();
+
+      if (!av.getWrapAlignment())
+      {
+        height = (av.alignment.getHeight()+1) * av.charHeight + 30;
+        width = idPanel.getWidth() + av.alignment.getWidth() * av.charWidth;
+      }
+
+      if(av.getShowAnnotation())
+      {
+        height += annotationPanel.getPreferredSize().height;
+      }
+
+
+System.out.println(width +" "+height);
+
+    try
+    {
+      jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(jalview.bin.Cache.getProperty(
+          "LAST_DIRECTORY"), new String[]{"png"}, "Portable network graphics");
+      chooser.setFileView(new jalview.io.JalviewFileView());
+      chooser.setDialogTitle("Create EPS file from alignment");
+      chooser.setToolTipText("Save");
+
+      int value = chooser.showSaveDialog(this);
+      if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)
+        return;
+
+      jalview.bin.Cache.setProperty("LAST_DIRECTORY",chooser.getSelectedFile().getPath());
+      FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());
+
+      BufferedImage bi = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
+      Graphics png = bi.getGraphics();
+
+
+        if (av.getWrapAlignment())
+          printWrappedAlignment(png, width, height, 0);
+        else
+          printUnwrapped(png, width, height, 0);
+
+        ImageIO.write(bi, "png", out);
+        out.close();
+    }
+    catch (Exception ex)
+    {
+      ex.printStackTrace();
+    }
+  }
+
 }
 
 
 
+