added quality calculation from Jalview118, and scale dependent
[jalview.git] / src / jalview / gui / AlignViewport.java
index 7719f02..5535dbb 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,20 +15,21 @@ 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 = false;\r
+  boolean showAnnotation = true;\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
@@ -40,49 +41,32 @@ 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
 \r
   NJTree currentTree = null;\r
 \r
+  boolean scaleAboveWrapped = false;\r
+  boolean scaleLeftWrapped  = true;\r
+  boolean scaleRightWrapped = true;\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
@@ -91,41 +75,158 @@ public class AlignViewport
    showSequenceFeatures = b;\r
  }\r
 \r
+  AlignmentAnnotation consensus;\r
+  AlignmentAnnotation conservation;\r
+  AlignmentAnnotation quality;\r
+\r
+  public void updateConservation()\r
+  {\r
+    int percGaps = 100; // JBPNote : This should be a scalable property!\r
+    Conservation cons = new jalview.analysis.Conservation("All",\r
+        jalview.schemes.ResidueProperties.propHash, 1,\r
+        alignment.getSequences(), 0,\r
+        alignment.getWidth());\r
+    cons.calculate();\r
+    cons.verdict(false, percGaps);\r
+    cons.findQuality();\r
+    int alWidth = alignment.getWidth();\r
+    Annotation [] annotations = new Annotation[alWidth];\r
+    Annotation [] qannotations = new Annotation[alWidth];\r
+    String sequence = cons.getConsSequence().getSequence();\r
+    float minR,minG,minB, maxR,maxG,maxB;\r
+    minR = 0.3f;\r
+    minG = 0.0f;\r
+    minB = 0f;\r
+    maxR = 1.0f-minR; maxG=0.9f-minG; maxB=0f-minB; // scalable range for colouring both Conservation and Quality\r
+    float min = 0f;\r
+    float max = 11f;\r
+    float qmin = cons.qualityRange[0].floatValue();\r
+    float qmax = cons.qualityRange[1].floatValue();\r
+\r
+    for (int i = 0; i < alWidth; i++)\r
+    {\r
+      float value = 0;\r
+      try\r
+        {\r
+          value = Integer.parseInt(sequence.charAt(i) + "");\r
+        }\r
+      catch (Exception ex)\r
+        {\r
+          if (sequence.charAt(i) == '*') value = 11;\r
+          if (sequence.charAt(i) == '+') value = 10;\r
+        }\r
+      float vprop = value-min;\r
+      vprop/=max;\r
+      annotations[i] = new Annotation(sequence.charAt(i) + "",\r
+                                      "Conservation graph", ' ', value, new Color(minR+maxR*vprop, minG+maxG*vprop, minB+maxB*vprop));\r
+      // Quality calc\r
+      value = ((Double) cons.quality.get(i)).floatValue();\r
+      vprop = value - qmin;\r
+      vprop/=qmax;\r
+      qannotations[i] = new Annotation(sequence.charAt(i) + "",\r
+                                      new String().valueOf(value), ' ', value, new Color(minR+maxR*vprop, minG+maxG*vprop, minB+maxB*vprop));\r
+    }\r
+\r
+    if(conservation==null)\r
+    {\r
+      conservation = new AlignmentAnnotation("Conservation",\r
+                                             "Conservation of total alignment less than "+percGaps+"% gaps",\r
+                                             annotations,\r
+                                             0f, // cons.qualityRange[0].floatValue(),\r
+                                             11f, // cons.qualityRange[1].floatValue()\r
+                                             1);\r
+      alignment.addAnnotation(conservation);\r
+      quality = new AlignmentAnnotation("Quality",\r
+                                        "Alignment Quality based on Blosum62 scores",\r
+                                        qannotations,\r
+                                        cons.qualityRange[0].floatValue(),\r
+                                        cons.qualityRange[1].floatValue(),\r
+                                        1);\r
+      alignment.addAnnotation(quality);\r
+    }\r
+    else {\r
+      conservation.annotations = annotations;\r
+      quality.annotations = qannotations;\r
+    }\r
+\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, alWidth=alignment.getWidth(); i<alWidth; 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 += (int)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, sz=consensusV.size(); i < sz; 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
@@ -178,7 +279,7 @@ public class AlignViewport
   public void setEndRes(int res) {\r
     if (res > alignment.getWidth()-1) {\r
       System.out.println(" Corrected res from " + res + " to maximum " + (alignment.getWidth()-1));\r
-       res = alignment.getWidth() -1;\r
+       res = alignment.getWidth()-1;\r
     }\r
     if (res < 0) {\r
       res = 0;\r
@@ -240,9 +341,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
@@ -273,9 +372,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
@@ -325,15 +422,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
@@ -357,5 +451,43 @@ public class AlignViewport
   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
+  public boolean getShowAnnotation()\r
+  {\r
+    return showAnnotation;\r
+  }\r
+\r
+  public void setShowAnnotation(boolean b)\r
+  {\r
+    showAnnotation = b;\r
+  }\r
+\r
+  public boolean getScaleAboveWrapped()\r
+  { return scaleAboveWrapped;}\r
+\r
+  public boolean getScaleLeftWrapped()\r
+  { return scaleLeftWrapped; }\r
+\r
+  public boolean getScaleRightWrapped()\r
+  { return scaleRightWrapped; }\r
+\r
+  public void setScaleAboveWrapped(boolean b)\r
+  { scaleAboveWrapped = b; }\r
+\r
+  public void setScaleLeftWrapped(boolean b)\r
+  { scaleLeftWrapped = b; }\r
+\r
+  public void setScaleRightWrapped(boolean b)\r
+  { scaleRightWrapped = b; }\r
+\r
 \r
 }\r