Alignment Annotation added
[jalview.git] / src / jalview / gui / AlignViewport.java
index b21f47e..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,19 +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 = false;\r
-  boolean groupEdit    = false;\r
-  boolean groupDefiningMode = true;\r
+  boolean renderGaps = 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
@@ -39,94 +40,154 @@ public class AlignViewport
   Font            font = new Font("SansSerif",Font.PLAIN,10);\r
   AlignmentI      alignment;\r
 \r
-  Selection       sel    = new Selection();\r
   ColumnSelection colSel = new ColumnSelection();\r
 \r
-  OutputGenerator og;\r
-\r
-  String visibleConsensus;\r
 \r
   int threshold;\r
   int increment;\r
 \r
   NJTree currentTree = null;\r
 \r
-  int window   = 50;\r
-  int baseline = 30;\r
-\r
-  Vector kmers;\r
 \r
-  public AlignViewport(AlignmentI da,\r
-                       boolean showScores,\r
+  public AlignViewport(AlignmentI al,\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
+    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
+    setAlignment(al);\r
+    setFont( font );\r
+ }\r
+\r
+ public void showSequenceFeatures(boolean b)\r
+ {\r
+   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 AlignViewport(int startRes, int endRes,\r
-                       int startSeq, int endSeq,\r
-                       boolean showScores,\r
-                       boolean showText,\r
-                       boolean showBoxes,\r
-                       boolean wrapAlignment) {\r
+  public void updateConsensus()\r
+  {\r
+    Annotation [] annotations = new Annotation[alignment.getWidth()];\r
 \r
-    this.startRes = startRes;\r
-    this.endRes   = endRes;\r
-    this.startSeq = startSeq;\r
-    this.endSeq   = endSeq;\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
-    this.showScores    = showScores;\r
-    this.showText      = showText;\r
-    this.showBoxes     = showBoxes;\r
-    this.wrapAlignment = wrapAlignment;\r
+        mouseOver += value+"%";\r
+        annotations[i] = new Annotation(maxRes, mouseOver, ' ', value);\r
 \r
-   // og  = new AlignmentOutputGenerator(this);\r
+    }\r
 \r
-    setFont( font );\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
@@ -137,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
@@ -231,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
@@ -264,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
@@ -276,15 +343,7 @@ public class AlignViewport
   public boolean getShowBoxes() {\r
     return showBoxes;\r
   }\r
- // public CommandParser getCommandLog() {\r
- ///   return log;\r
- // }\r
-  public boolean getGroupEdit() {\r
-    return groupEdit;\r
-  }\r
-  public void setGroupEdit(boolean state) {\r
-    groupEdit = state;\r
-  }\r
+\r
   public char getGapCharacter() {\r
     return getAlignment().getGapCharacter();\r
   }\r
@@ -324,17 +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
-  public OutputGenerator getOutputGenerator() {\r
-    return og;\r
-  }\r
+\r
   public void resetSeqLimits(int height) {\r
-    setStartSeq(0);\r
     setEndSeq(height/getCharHeight());\r
   }\r
   public void setCurrentTree(NJTree tree) {\r
@@ -351,27 +405,20 @@ public class AlignViewport
     public RendererI getRenderer() {\r
        return renderer;\r
     }\r
-    public int getPIDWindow() {\r
-       return window;\r
-    }\r
-    public void setPIDWindow(int window) {\r
-       this.window = window;\r
-    }\r
-\r
-    public int getPIDBaseline() {\r
-       return baseline;\r
-    }\r
-    public void setPIDBaseline(int baseline) {\r
-       this.baseline = baseline;\r
-    }\r
 \r
-    public void setKmers(Vector kmers) {\r
-        this.kmers = kmers;\r
-    }\r
+  public void setColourAppliesToAllGroups(boolean b)\r
+  {   colourAppliesToAllGroups = b; }\r
 \r
-    public Vector getKmers() {\r
-        return this.kmers;\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