Alignment Annotation added
[jalview.git] / src / jalview / gui / AlignViewport.java
index 3620bc0..19a206d 100755 (executable)
@@ -1,7 +1,7 @@
 package jalview.gui;\r
 \r
 import java.awt.*;\r
-import jalview.io.*;\r
+import jalview.analysis.*;\r
 import jalview.analysis.NJTree;\r
 import jalview.datamodel.*;\r
 import jalview.schemes.*;\r
@@ -15,18 +15,20 @@ public class AlignViewport
   int startSeq;\r
   int endSeq;\r
 \r
-  boolean showScores=false;\r
+  boolean showFullId = false;\r
   boolean showText=true;\r
   boolean showColourText=false;\r
   boolean showBoxes=true;\r
   boolean wrapAlignment=false;\r
   boolean renderGaps = true;\r
-  boolean showSequenceFeatures = true;\r
+  boolean showSequenceFeatures = false;\r
 \r
+  boolean colourAppliesToAllGroups = true;\r
   ColourSchemeI globalColourScheme = null;\r
   boolean conservationColourSelected = false;\r
+  boolean abovePIDThreshold = false;\r
 \r
-  SequenceGroup rubberbandGroup = null;\r
+  SequenceGroup selectionGroup = new SequenceGroup();\r
 \r
   RendererI renderer = new SequenceRenderer(this);\r
 \r
@@ -38,10 +40,8 @@ public class AlignViewport
   Font            font = new Font("SansSerif",Font.PLAIN,10);\r
   AlignmentI      alignment;\r
 \r
-  Selection       sel    = new Selection(this);\r
   ColumnSelection colSel = new ColumnSelection();\r
 \r
-  String visibleConsensus;\r
 \r
   int threshold;\r
   int increment;\r
@@ -49,38 +49,20 @@ public class AlignViewport
   NJTree currentTree = null;\r
 \r
 \r
-  public AlignViewport(AlignmentI da,\r
-                       boolean showScores,\r
-                       boolean showText,\r
-                       boolean showBoxes,\r
-                       boolean wrapAlignment) {\r
-    this(0,da.getWidth()-1,0,da.getHeight()-1,showScores,\r
-         showText,\r
-         showBoxes,\r
-         wrapAlignment);\r
-\r
-    setAlignment(da);\r
-  }\r
-\r
-  public AlignViewport(int startRes, int endRes,\r
-                       int startSeq, int endSeq,\r
-                       boolean showScores,\r
+  public AlignViewport(AlignmentI al,\r
                        boolean showText,\r
                        boolean showBoxes,\r
                        boolean wrapAlignment) {\r
 \r
-    this.startRes = startRes;\r
-    this.endRes   = endRes;\r
-    this.startSeq = startSeq;\r
-    this.endSeq   = endSeq;\r
-\r
-    this.showScores    = showScores;\r
-    this.showText      = showText;\r
-    this.showBoxes     = showBoxes;\r
+    this.startRes = 0;\r
+    this.endRes = al.getWidth()-1;\r
+    this.startSeq = 0;\r
+    this.endSeq = al.getHeight()-1;\r
+    this.showText = showText;\r
+    this.showBoxes = showBoxes;\r
     this.wrapAlignment = wrapAlignment;\r
 \r
-   // og  = new AlignmentOutputGenerator(this);\r
-\r
+    setAlignment(al);\r
     setFont( font );\r
  }\r
 \r
@@ -89,41 +71,123 @@ public class AlignViewport
    showSequenceFeatures = b;\r
  }\r
 \r
+  AlignmentAnnotation consensus;\r
+  AlignmentAnnotation conservation;\r
+  public void updateConservation()\r
+  {\r
+    Annotation [] annotations = new Annotation[alignment.getWidth()];\r
+\r
+    Conservation cons = new jalview.analysis.Conservation("All",\r
+        jalview.schemes.ResidueProperties.propHash, 3,\r
+        alignment.getSequences(), 0,\r
+        alignment.getWidth());\r
+    cons.calculate();\r
+    cons.verdict(false, 100);\r
+\r
+    String sequence = cons.getConsSequence().getSequence();\r
+    for (int i = 0; i < alignment.getWidth(); i++)\r
+    {\r
+      int value = 0;\r
+      try\r
+      {\r
+        value = Integer.parseInt(sequence.charAt(i) + "");\r
+      }\r
+      catch (Exception ex)\r
+      {\r
+        if (sequence.charAt(i) == '*') value = 10;\r
+      }\r
+\r
+      annotations[i] = new Annotation(sequence.charAt(i) + "",\r
+                                      "Conservation graph", ' ', value);\r
+    }\r
+\r
+    if(conservation==null)\r
+    {\r
+      conservation = new AlignmentAnnotation("Conservation",\r
+                                             "Conservation of total alignment",\r
+                                             annotations, 0, 10, 1);\r
+      alignment.addAnnotation(conservation);\r
+    }\r
+    else\r
+      conservation.annotations = annotations;\r
+\r
+  }\r
+\r
+  public void updateConsensus()\r
+  {\r
+    Annotation [] annotations = new Annotation[alignment.getWidth()];\r
+\r
+    Vector cons = alignment.getAAFrequency();\r
+    Hashtable hash = null;\r
+    for (int i = 0; i < alignment.getWidth(); i++)\r
+    {\r
+        hash = (Hashtable) cons.elementAt(i);\r
+        float value = Float.parseFloat(hash.get("maxCount").toString());\r
+        value /= Float.parseFloat(hash.get("size").toString());\r
+\r
+        value *= 100;\r
+        String maxRes = hash.get("maxResidue")+" ";\r
+        String mouseOver = hash.get("maxResidue")+" ";\r
+        if(maxRes.length()>2)\r
+        {\r
+          mouseOver = "["+maxRes+"] ";\r
+          maxRes = "+ ";\r
+        }\r
+\r
+        mouseOver += value+"%";\r
+        annotations[i] = new Annotation(maxRes, mouseOver, ' ', value);\r
+\r
+    }\r
+\r
+     if(consensus==null)\r
+     {\r
+       consensus = new AlignmentAnnotation("% Identity",\r
+                                           "PID", annotations, 0f, 100f, 1);\r
+       alignment.addAnnotation(consensus);\r
+     }\r
+     else\r
+       consensus.annotations = annotations;\r
+\r
+  }\r
+\r
   public String getVisibleConsensus()\r
   {\r
     return visibleConsensus;\r
   }\r
 \r
-  Vector consensus = new Vector();\r
+  String visibleConsensus;\r
+  Vector consensusV = new Vector();\r
   public Vector getConsensus(boolean recalculate)\r
   {\r
-    if(recalculate || consensus.size()<1)\r
+    if(recalculate || consensusV.size()<1)\r
     {\r
-      consensus = alignment.getAAFrequency();\r
+      consensusV = alignment.getAAFrequency();\r
       StringBuffer sb = new StringBuffer();\r
       Hashtable hash = null;\r
-      for (int i = 0; i < consensus.size(); i++)\r
+      for (int i = 0; i < consensusV.size(); i++)\r
       {\r
-        hash = (Hashtable) consensus.elementAt(i);\r
+        hash = (Hashtable) consensusV.elementAt(i);\r
         sb.append(hash.get("maxResidue").toString().charAt(0));\r
       }\r
       visibleConsensus = sb.toString();\r
     }\r
 \r
 \r
-    return consensus;\r
+    return consensusV;\r
   }\r
 \r
-  public SequenceGroup getRubberbandGroup()\r
+\r
+  public SequenceGroup getSelectionGroup()\r
   {\r
-    return rubberbandGroup;\r
+    return selectionGroup;\r
   }\r
 \r
-  public void setRubberbandGroup(SequenceGroup sg)\r
+  public void setSelectionGroup(SequenceGroup sg)\r
   {\r
-    rubberbandGroup = sg;\r
+    selectionGroup = sg;\r
   }\r
 \r
+\r
  public boolean getConservationSelected()\r
  {\r
    return conservationColourSelected;\r
@@ -134,6 +198,16 @@ public class AlignViewport
    conservationColourSelected = b;\r
  }\r
 \r
+ public boolean getAbovePIDThreshold()\r
+ {\r
+   return abovePIDThreshold;\r
+ }\r
+\r
+ public void setAbovePIDThreshold(boolean b)\r
+ {\r
+   abovePIDThreshold = b;\r
+ }\r
+\r
   public int getStartRes() {\r
     return startRes;\r
   }\r
@@ -228,9 +302,7 @@ public class AlignViewport
   public void setAlignment(AlignmentI align) {\r
     this.alignment = align;\r
   }\r
-  public void setShowScores(boolean state) {\r
-    showScores = state;\r
-  }\r
+\r
   public void setWrapAlignment(boolean state) {\r
     wrapAlignment = state;\r
   }\r
@@ -261,9 +333,7 @@ public class AlignViewport
   public void setShowBoxes(boolean state) {\r
     showBoxes = state;\r
   }\r
-  public boolean getShowScores() {\r
-    return showScores;\r
-  }\r
+\r
   public boolean getWrapAlignment() {\r
       return wrapAlignment;\r
   }\r
@@ -313,15 +383,12 @@ public class AlignViewport
     }\r
     return -1;\r
   }\r
-  public Selection getSelection() {\r
-    return sel;\r
-  }\r
+\r
   public ColumnSelection getColumnSelection() {\r
     return colSel;\r
   }\r
 \r
   public void resetSeqLimits(int height) {\r
-    setStartSeq(0);\r
     setEndSeq(height/getCharHeight());\r
   }\r
   public void setCurrentTree(NJTree tree) {\r
@@ -339,4 +406,19 @@ public class AlignViewport
        return renderer;\r
     }\r
 \r
+  public void setColourAppliesToAllGroups(boolean b)\r
+  {   colourAppliesToAllGroups = b; }\r
+\r
+  public boolean getColourAppliesToAllGroups()\r
+  {return colourAppliesToAllGroups; }\r
+\r
+  public boolean getShowFullId()\r
+  {\r
+    return showFullId;\r
+  }\r
+\r
+  public void setShowFullId(boolean b)\r
+  {\r
+    showFullId = b;\r
+  }\r
 }\r