JAL-98 ProfilesI container for profiles for columns

[jalview.git] / src / jalview / schemes / PIDColourScheme.java

diff --git a/src/jalview/schemes/PIDColourScheme.java b/src/jalview/schemes/PIDColourScheme.java
index eadf09b..0ad5b5c 100755 (executable)
--- a/src/jalview/schemes/PIDColourScheme.java
+++ b/src/jalview/schemes/PIDColourScheme.java
@@ -1,48 +1,97 @@
-package jalview.schemes;\r
-\r
-import jalview.datamodel.*;\r
-import java.util.*;\r
-import java.awt.*;\r
-\r
-public class PIDColourScheme extends ResidueColourScheme {\r
-    public Color[] pidColours;\r
-    public float[] thresholds;\r
-\r
-    SequenceGroup group;\r
-\r
-    public PIDColourScheme() {\r
-       this.pidColours = ResidueProperties.pidColours;\r
-       this.thresholds = ResidueProperties.pidThresholds;\r
-    }\r
-\r
-\r
-\r
-  public Color findColour(String s, int j) {\r
-\r
-    if( threshold!=0 && !aboveThreshold(s,j))\r
-      return Color.white;\r
-\r
-    Color     c    = Color.white;\r
-      Hashtable hash =  (Hashtable)consensus.elementAt(j);\r
-\r
-      double sc = 0;\r
-      if ( Integer.parseInt( hash.get("maxCount").toString() ) != -1 && hash.contains(s))\r
-      {\r
-        sc = ( (Integer) hash.get(s)).intValue() * 100.0 / ((Integer)hash.get("size")).intValue() ;\r
-        // MC Should be isGap\r
-        if (!jalview.util.Comparison.isGap((s.charAt(0))))\r
-        {\r
-          for (int i = 0; i < thresholds.length; i++)\r
-          {\r
-            if (sc > thresholds[i])\r
-            {\r
-              c = pidColours[i];\r
-              break;\r
-            }\r
-          }\r
-        }\r
-\r
-      }\r
-      return c;\r
-  }\r
-}\r
+/*
+ * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
+ * Copyright (C) $$Year-Rel$$ The Jalview Authors
+ * 
+ * This file is part of Jalview.
+ * 
+ * Jalview is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License 
+ * as published by the Free Software Foundation, either version 3
+ * of the License, or (at your option) any later version.
+ *  
+ * Jalview is distributed in the hope that it will be useful, but 
+ * WITHOUT ANY WARRANTY; without even the implied warranty 
+ * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
+ * PURPOSE.  See the GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License
+ * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
+ * The Jalview Authors are detailed in the 'AUTHORS' file.
+ */
+package jalview.schemes;
+
+import jalview.datamodel.ProfileI;
+import jalview.datamodel.SequenceGroup;
+import jalview.datamodel.SequenceI;
+import jalview.util.Comparison;
+
+import java.awt.Color;
+
+public class PIDColourScheme extends ResidueColourScheme
+{
+  public Color[] pidColours;
+
+  public float[] thresholds;
+
+  SequenceGroup group;
+
+  public PIDColourScheme()
+  {
+    this.pidColours = ResidueProperties.pidColours;
+    this.thresholds = ResidueProperties.pidThresholds;
+  }
+
+  @Override
+  public Color findColour(char c, int j, SequenceI seq)
+  {
+    if ('a' <= c && c <= 'z')
+    {
+      c -= ('a' - 'A');
+    }
+
+    if (consensus == null || consensus.get(j) == null)
+    {
+      return Color.white;
+    }
+
+    if ((threshold != 0) && !aboveThreshold(c, j))
+    {
+      return Color.white;
+    }
+
+    Color currentColour = Color.white;
+
+    double sc = 0;
+
+
+    /*
+     * test whether this is the consensus (or joint consensus) residue
+     */
+    ProfileI profile = consensus.get(j);
+    boolean matchesConsensus = profile.getModalResidue().contains(
+            String.valueOf(c));
+    if (matchesConsensus)
+    {
+      sc = profile.getPercentageIdentity(ignoreGaps);
+
+      if (!Comparison.isGap(c))
+      {
+        for (int i = 0; i < thresholds.length; i++)
+        {
+          if (sc > thresholds[i])
+          {
+            currentColour = pidColours[i];
+            break;
+          }
+        }
+      }
+    }
+
+    if (conservationColouring)
+    {
+      currentColour = applyConservation(currentColour, j);
+    }
+
+    return currentColour;
+  }
+}