Alignment Annotation added
[jalview.git] / src / jalview / gui / AlignFrame.java
index 4209455..8c7c9be 100755 (executable)
@@ -32,8 +32,7 @@ public class AlignFrame extends GAlignFrame
   final AlignViewport viewport;\r
   public AlignFrame(AlignmentI al)\r
   {\r
-    super();\r
-    viewport = new AlignViewport(al,true,true,true,false);\r
+    viewport = new AlignViewport(al,true,true,false);\r
 \r
     String fontName = jalview.bin.Cache.getProperty("FONT_NAME");\r
     String fontStyle= jalview.bin.Cache.getProperty("FONT_STYLE");\r
@@ -42,10 +41,18 @@ public class AlignFrame extends GAlignFrame
       viewport.setFont( new Font(fontName,Integer.parseInt(fontStyle),Integer.parseInt(fontSize)) );\r
 \r
 \r
-\r
     alignPanel = new AlignmentPanel(this, viewport);\r
     getContentPane().add(alignPanel, java.awt.BorderLayout.CENTER);\r
 \r
+    // add conservation graph to alignment\r
+    viewport.updateConservation();\r
+    viewport.updateConsensus();\r
+\r
+    alignPanel.annotationPanel.adjustPanelHeight();\r
+    alignPanel.annotationSpaceFillerHolder.setPreferredSize(alignPanel.annotationPanel.getPreferredSize());\r
+    alignPanel.annotationScroller.setPreferredSize(alignPanel.annotationPanel.getPreferredSize());\r
+\r
+\r
     addInternalFrameListener(new InternalFrameAdapter()\r
    {\r
      public void internalFrameActivated(InternalFrameEvent evt)\r
@@ -63,11 +70,27 @@ public class AlignFrame extends GAlignFrame
 \r
   protected void saveAs_actionPerformed(ActionEvent e)\r
   {\r
-    JFileChooser chooser = new JFileChooser(jalview.bin.Cache.getProperty("LAST_DIRECTORY"));\r
+    String suffix [] = null;\r
+    if(e.getActionCommand().equals("FASTA"))\r
+      suffix = new String[]{"fa", "fasta"};\r
+    else if(e.getActionCommand().equals("MSF"))\r
+      suffix = new String[]{"msf"};\r
+    else if(e.getActionCommand().equals("CLUSTAL"))\r
+      suffix = new String[]{"aln"};\r
+    else if(e.getActionCommand().equals("BLC"))\r
+      suffix = new String[]{"blc"};\r
+    else if(e.getActionCommand().equals("PIR"))\r
+      suffix = new String[]{"pir"};\r
+    else if(e.getActionCommand().equals("PFAM"))\r
+      suffix = new String[]{"pfam"};\r
+\r
+    JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache.getProperty("LAST_DIRECTORY")\r
+        , suffix, e.getActionCommand()+" file");\r
+    chooser.setFileView(new JalviewFileView());\r
     chooser.setDialogTitle("Save Alignment to file - "+e.getActionCommand() +" format.");\r
     chooser.setToolTipText("Save");\r
     int value = chooser.showSaveDialog(this);\r
-    if(value == JFileChooser.APPROVE_OPTION)\r
+    if(value == JalviewFileChooser.APPROVE_OPTION)\r
     {\r
       String choice =  chooser.getSelectedFile().getPath();\r
       jalview.bin.Cache.setProperty("LAST_DIRECTORY", choice);\r
@@ -98,20 +121,14 @@ public class AlignFrame extends GAlignFrame
     htmlOutput = null;\r
   }\r
 \r
-  protected void createJPG_actionPerformed(ActionEvent e)\r
+  protected void createPNG_actionPerformed(ActionEvent e)\r
   {\r
+    alignPanel.makePNG();\r
+  }\r
 \r
-    int height = (viewport.alignment.getWidth() / viewport.getChunkWidth() +1) * viewport.chunkHeight;\r
-    int width = alignPanel.seqPanel.getWidth() + alignPanel.idPanel.getWidth();\r
-\r
-\r
-    if(!viewport.getWrapAlignment())\r
-    {\r
-      height = viewport.alignment.getHeight() * viewport.charHeight;\r
-      width = alignPanel.idPanel.getWidth() + viewport.alignment.getWidth() * viewport.charWidth;\r
-    }\r
-\r
-    alignPanel.makeJPG( width, height);\r
+  protected void epsFile_actionPerformed(ActionEvent e)\r
+  {\r
+    alignPanel.makeEPS();\r
   }\r
 \r
 \r
@@ -156,6 +173,7 @@ public class AlignFrame extends GAlignFrame
 \r
   Stack historyList = new Stack();\r
   Stack redoList = new Stack();\r
+  JMenuBar jMenuBar1 = new JMenuBar();\r
 \r
   void updateEditMenuBar()\r
   {\r
@@ -217,12 +235,14 @@ public class AlignFrame extends GAlignFrame
       seq = (SequenceI[]) history[1];\r
       viewport.setAlignment( new Alignment(seq) );\r
       updateEditMenuBar();\r
+      viewport.updateConsensus();\r
+      alignPanel.RefreshPanels();\r
       alignPanel.RefreshPanels();\r
   }\r
 \r
   public void moveSelectedSequences(boolean up)\r
   {\r
-    SequenceGroup sg = viewport.getRubberbandGroup();\r
+    SequenceGroup sg = viewport.getSelectionGroup();\r
     if (sg == null)\r
       return;\r
 \r
@@ -266,10 +286,10 @@ public class AlignFrame extends GAlignFrame
 \r
   protected void copy_actionPerformed(ActionEvent e)\r
   {\r
-   if(viewport.getRubberbandGroup()==null)\r
+   if(viewport.getSelectionGroup()==null)\r
      return;\r
 \r
-   SequenceGroup sg = viewport.getRubberbandGroup();\r
+   SequenceGroup sg = viewport.getSelectionGroup();\r
 \r
      Clipboard c = Toolkit.getDefaultToolkit().getSystemClipboard();\r
      StringBuffer buffer= new StringBuffer();\r
@@ -329,12 +349,14 @@ public class AlignFrame extends GAlignFrame
         int newHeight = newSeqs.length * af.viewport.getCharHeight() + 200;\r
         if (newHeight > 500)\r
           newHeight = 500;\r
-        Desktop.addInternalFrame(af, "Copied sequences", 700, newHeight);\r
+        Desktop.addInternalFrame(af, "Copied sequences", 700, 500);\r
       }\r
       else\r
       {\r
         viewport.setEndSeq(viewport.alignment.getHeight());\r
         viewport.alignment.getWidth();\r
+        viewport.updateConservation();\r
+        viewport.updateConsensus();\r
         alignPanel.RefreshPanels();\r
       }\r
 \r
@@ -352,10 +374,10 @@ public class AlignFrame extends GAlignFrame
   protected void delete_actionPerformed(ActionEvent e)\r
   {\r
     addHistoryItem("Delete");\r
-    if (viewport.getRubberbandGroup() == null)\r
+    if (viewport.getSelectionGroup() == null)\r
       return;\r
 \r
-     SequenceGroup sg = viewport.getRubberbandGroup();\r
+     SequenceGroup sg = viewport.getSelectionGroup();\r
      for (int i=0;i < sg.sequences.size(); i++)\r
      {\r
        SequenceI seq = sg.getSequenceAt(i);\r
@@ -368,7 +390,7 @@ public class AlignFrame extends GAlignFrame
           viewport.getAlignment().getSequences().setElementAt(seq, index);\r
      }\r
 \r
-     viewport.setRubberbandGroup(null);\r
+     viewport.setSelectionGroup(null);\r
      viewport.alignment.deleteGroup(sg);\r
      viewport.resetSeqLimits( alignPanel.seqPanel.seqCanvas.getHeight());\r
      if(viewport.getAlignment().getHeight()<1)\r
@@ -376,6 +398,8 @@ public class AlignFrame extends GAlignFrame
      {\r
        this.setClosed(true);\r
      }catch(Exception ex){}\r
+   viewport.updateConservation();\r
+   viewport.updateConsensus();\r
      alignPanel.RefreshPanels();\r
 \r
   }\r
@@ -388,6 +412,8 @@ public class AlignFrame extends GAlignFrame
      SequenceI[] seq = (SequenceI[]) history[1];\r
      viewport.setAlignment( new Alignment(seq) );\r
      updateEditMenuBar();\r
+     viewport.updateConsensus();\r
+     alignPanel.RefreshPanels();\r
      alignPanel.RefreshPanels();\r
   }\r
 \r
@@ -404,7 +430,7 @@ public class AlignFrame extends GAlignFrame
   protected void deleteGroups_actionPerformed(ActionEvent e)\r
   {\r
     viewport.alignment.deleteAllGroups();\r
-    viewport.getSelection().clear();\r
+    viewport.setSelectionGroup(null);\r
 \r
     alignPanel.RefreshPanels();\r
   }\r
@@ -413,29 +439,26 @@ public class AlignFrame extends GAlignFrame
 \r
   public void selectAllSequenceMenuItem_actionPerformed(ActionEvent e)\r
   {\r
-    Selection sel = viewport.getSelection();\r
+    SequenceGroup sg = new SequenceGroup();\r
     for (int i=0; i<viewport.getAlignment().getSequences().size(); i++)\r
-      sel.addElement( viewport.getAlignment().getSequenceAt(i));\r
+      sg.addSequence( viewport.getAlignment().getSequenceAt(i));\r
+    sg.setEndRes(viewport.alignment.getWidth());\r
+    viewport.setSelectionGroup(sg);\r
     PaintRefresher.Refresh(null);\r
   }\r
 \r
   public void deselectAllSequenceMenuItem_actionPerformed(ActionEvent e)\r
   {\r
-    viewport.setRubberbandGroup(null);\r
-    viewport.getSelection().clear();\r
+    viewport.setSelectionGroup(null);\r
     PaintRefresher.Refresh(null);\r
   }\r
 \r
   public void invertSequenceMenuItem_actionPerformed(ActionEvent e)\r
   {\r
-    Selection sel = viewport.getSelection();\r
+    SequenceGroup sg = viewport.getSelectionGroup();\r
     for (int i=0; i<viewport.getAlignment().getSequences().size(); i++)\r
-    {\r
-      if (sel.contains(viewport.getAlignment().getSequenceAt(i)))\r
-        sel.removeElement(viewport.getAlignment().getSequenceAt(i));\r
-      else\r
-        sel.addElement(viewport.getAlignment().getSequenceAt(i));\r
-    }\r
+      sg.addOrRemove (viewport.getAlignment().getSequenceAt(i));\r
+\r
     PaintRefresher.Refresh(null);\r
   }\r
 \r
@@ -455,6 +478,18 @@ public class AlignFrame extends GAlignFrame
       int min = colSel.getMin();\r
       viewport.getAlignment().trimLeft(min);\r
       colSel.compensateForEdit(0,min);\r
+\r
+      if(viewport.getSelectionGroup()!=null)\r
+        viewport.getSelectionGroup().adjustForRemoveLeft(min);\r
+\r
+      Vector groups = viewport.alignment.getGroups();\r
+      for(int i=0; i<groups.size(); i++)\r
+      {\r
+        SequenceGroup sg = (SequenceGroup) groups.get(i);\r
+        if(!sg.adjustForRemoveLeft(min))\r
+          viewport.alignment.deleteGroup(sg);\r
+      }\r
+\r
       alignPanel.RefreshPanels();\r
     }\r
   }\r
@@ -466,8 +501,19 @@ public class AlignFrame extends GAlignFrame
     if (colSel.size() > 0)\r
     {\r
       int max = colSel.getMax();\r
-      if(max>1)\r
-        viewport.getAlignment().trimRight(max);\r
+      viewport.getAlignment().trimRight(max);\r
+      if(viewport.getSelectionGroup()!=null)\r
+        viewport.getSelectionGroup().adjustForRemoveRight(max);\r
+\r
+      Vector groups = viewport.alignment.getGroups();\r
+      for(int i=0; i<groups.size(); i++)\r
+      {\r
+        SequenceGroup sg = (SequenceGroup) groups.get(i);\r
+        if(!sg.adjustForRemoveRight(max))\r
+          viewport.alignment.deleteGroup(sg);\r
+      }\r
+\r
+\r
 \r
       alignPanel.RefreshPanels();\r
     }\r
@@ -478,6 +524,8 @@ public class AlignFrame extends GAlignFrame
   {\r
     addHistoryItem("delete gapped columns");\r
     viewport.getAlignment().removeGaps();\r
+    viewport.updateConservation();\r
+    viewport.updateConsensus();\r
     alignPanel.RefreshPanels();\r
   }\r
 \r
@@ -498,7 +546,8 @@ public class AlignFrame extends GAlignFrame
           jSize--;\r
         }\r
     }\r
-\r
+    viewport.updateConservation();\r
+    viewport.updateConsensus();\r
     alignPanel.RefreshPanels();\r
   }\r
 \r
@@ -535,6 +584,14 @@ public class AlignFrame extends GAlignFrame
     Desktop.addInternalFrame(frame, "Change Font", 480,100);\r
   }\r
 \r
+  protected void fullSeqId_actionPerformed(ActionEvent e)\r
+  {\r
+    viewport.setShowFullId( fullSeqId.isSelected() );\r
+\r
+    alignPanel.idPanel.idCanvas.setPreferredSize( alignPanel.calculateIdWidth() );\r
+    alignPanel.RefreshPanels();\r
+  }\r
+\r
   protected void colourTextMenuItem_actionPerformed(ActionEvent e)\r
   {\r
       viewport.setColourText( colourTextMenuItem.isSelected() );\r
@@ -578,9 +635,9 @@ public class AlignFrame extends GAlignFrame
     alignPanel.RefreshPanels();\r
   }\r
 \r
-  public void consensusGraphMenuItem_actionPerformed(ActionEvent e)\r
+  public void annotationPanelMenuItem_actionPerformed(ActionEvent e)\r
   {\r
-    alignPanel.setGraphPanelVisible( consensusGraphMenuItem.isSelected() );\r
+    alignPanel.setAnnotationVisible( annotationPanelMenuItem.isSelected() );\r
   }\r
 \r
   public void overviewMenuItem_actionPerformed(ActionEvent e)\r
@@ -764,6 +821,10 @@ if ( viewport.getConservationSelected() )
   protected  void conservationMenuItem_actionPerformed(ActionEvent e)\r
   {\r
     viewport.setConservationSelected(conservationMenuItem.isSelected());\r
+    if(coloursChanging)\r
+      return;\r
+\r
+    coloursChanging = true;\r
 \r
     viewport.setAbovePIDThreshold(false);\r
     abovePIDThreshold.setSelected(false);\r
@@ -776,11 +837,17 @@ if ( viewport.getConservationSelected() )
      changeColour( ((ConservationColourScheme)cs).cs );\r
     else\r
       changeColour( cs );\r
-  }\r
 \r
+    coloursChanging = false;\r
+  }\r
 \r
+  boolean coloursChanging = false;\r
   public void abovePIDThreshold_actionPerformed(ActionEvent e)\r
   {\r
+    if(coloursChanging)\r
+      return;\r
+\r
+    coloursChanging = true;\r
     viewport.setAbovePIDThreshold(abovePIDThreshold.isSelected());\r
 \r
     conservationMenuItem.setSelected(false);\r
@@ -797,6 +864,7 @@ if ( viewport.getConservationSelected() )
     else\r
         changeColour( cs );\r
 \r
+    coloursChanging = false;\r
   }\r
 \r
 \r
@@ -870,7 +938,7 @@ if ( viewport.getConservationSelected() )
 \r
   public void pairwiseAlignmentMenuItem_actionPerformed(ActionEvent e)\r
   {\r
-    if(viewport.getSelection().size()<2)\r
+    if(viewport.getSelectionGroup().getSize()<2)\r
       JOptionPane.showInternalMessageDialog(this, "You must select at least 2 sequences.", "Invalid Selection", JOptionPane.WARNING_MESSAGE);\r
     else\r
     {\r
@@ -882,7 +950,8 @@ if ( viewport.getConservationSelected() )
 \r
   public void PCAMenuItem_actionPerformed(ActionEvent e)\r
   {\r
-    if( (viewport.getSelection().size()<4 && viewport.getSelection().size()>0)\r
+\r
+    if( (viewport.getSelectionGroup()!=null && viewport.getSelectionGroup().getSize()<4 && viewport.getSelectionGroup().getSize()>0)\r
        || viewport.getAlignment().getHeight()<4)\r
     {\r
       JOptionPane.showInternalMessageDialog(this, "Principal component analysis must take\n"\r
@@ -938,9 +1007,9 @@ if ( viewport.getConservationSelected() )
     }\r
 \r
     TreePanel tp=null;\r
-   if (viewport.getSelection() != null && viewport.getSelection().size() > 3)\r
+   if (viewport.getSelectionGroup() != null && viewport.getSelectionGroup().getSize() > 3)\r
    {\r
-     tp = new TreePanel(viewport, viewport.getSelection().asVector(),type, pwType,\r
+     tp = new TreePanel(viewport, viewport.getSelectionGroup().sequences,type, pwType,\r
                          0, viewport.alignment.getWidth());\r
    }\r
    else\r
@@ -1059,4 +1128,50 @@ if ( viewport.getConservationSelected() )
          g.drawString("Clustal Alignment Web Service running", 30,30);\r
        }\r
   }\r
+  protected void jpred_actionPerformed(ActionEvent e)\r
+{\r
+\r
+\r
+      JInternalFrame frame = new JInternalFrame();\r
+      JPredClient ct = new JPredClient((SequenceI)\r
+                                       viewport.getAlignment().getSequences().elementAt(0));\r
+    }\r
+\r
+    protected void LoadtreeMenuItem_actionPerformed(ActionEvent e) {\r
+    // Pick the tree file\r
+    JalviewFileChooser chooser = new JalviewFileChooser(jalview.bin.Cache.\r
+        getProperty("LAST_DIRECTORY"));\r
+    chooser.setFileView(new JalviewFileView());\r
+    chooser.setDialogTitle("Select a newick-like tree file");\r
+    chooser.setToolTipText("Load a tree file");\r
+    int value = chooser.showOpenDialog(null);\r
+    if (value == JalviewFileChooser.APPROVE_OPTION)\r
+    {\r
+      String choice = chooser.getSelectedFile().getPath();\r
+      jalview.bin.Cache.setProperty("LAST_DIRECTORY", choice);\r
+      TreePanel treepanel = null;\r
+      try\r
+      {\r
+        jalview.io.NewickFile fin = new jalview.io.NewickFile(choice, "File");\r
+        fin.parse();\r
+\r
+        if (fin.getTree() != null)\r
+        {\r
+          TreePanel tp = null;\r
+          tp = new TreePanel(viewport, viewport.getAlignment().getSequences(),\r
+                             fin, "FromFile", choice);\r
+          Desktop.addInternalFrame(tp, title, 600, 500);\r
+        }\r
+      }\r
+      catch (Exception ex)\r
+      {\r
+        JOptionPane.showMessageDialog(Desktop.desktop,\r
+                                      "Problem reading tree file",\r
+                                      ex.getMessage(),\r
+                                      JOptionPane.WARNING_MESSAGE);\r
+        ex.printStackTrace();\r
+      }\r
+    }\r
+  }\r
+\r
 }\r