Merge branch 'develop' into bug/JAL-2255_seq-fetcher-broken-on-linux
[jalview.git] / src / jalview / schemes / ResidueColourScheme.java
index bca98cf..03fc129 100755 (executable)
  */
 package jalview.schemes;
 
-import jalview.analysis.AAFrequency;
-import jalview.analysis.Conservation;
 import jalview.datamodel.AnnotatedCollectionI;
 import jalview.datamodel.SequenceCollectionI;
+import jalview.datamodel.SequenceGroup;
 import jalview.datamodel.SequenceI;
-import jalview.util.MessageManager;
+import jalview.util.Comparison;
 
 import java.awt.Color;
-import java.util.Hashtable;
 import java.util.Map;
 
 /**
- * DOCUMENT ME!
- * 
- * @author $author$
- * @version $Revision$
+ * Base class for residue-based colour schemes
  */
-public class ResidueColourScheme implements ColourSchemeI
+public abstract class ResidueColourScheme implements ColourSchemeI
 {
-  final int[] symbolIndex;
+  public static final String NONE = "None";
 
-  boolean conservationColouring = false;
+  public static final String USER_DEFINED = "User Defined";
 
-  Color[] colors = null;
+  /*
+   * lookup up by character value e.g. 'G' to the colors array index
+   * e.g. if symbolIndex['K'] = 11 then colors[11] is the colour for K
+   */
+  final int[] symbolIndex;
 
-  int threshold = 0;
+  /*
+   * colour for residue characters as indexed by symbolIndex
+   */
+  Color[] colors = null;
 
   /* Set when threshold colouring to either pid_gaps or pid_nogaps */
-  protected String ignoreGaps = AAFrequency.PID_GAPS;
-
-  /** Consenus as a hashtable array */
-  Hashtable[] consensus;
-
-  /** Conservation string as a char array */
-  char[] conservation;
-
-  int conservationLength = 0;
-
-  /** DOCUMENT ME!! */
-  int inc = 30;
+  protected boolean ignoreGaps = false;
 
   /**
    * Creates a new ResidueColourScheme object.
@@ -68,15 +59,11 @@ public class ResidueColourScheme implements ColourSchemeI
    *        ResidueProperties.aaIndex)
    * @param colors
    *          colours for symbols in sequences
-   * @param threshold
-   *          threshold for conservation shading
    */
-  public ResidueColourScheme(int[] aaOrnaIndex, Color[] colours,
-          int threshold)
+  public ResidueColourScheme(int[] aaOrnaIndex, Color[] colours)
   {
     symbolIndex = aaOrnaIndex;
     this.colors = colours;
-    this.threshold = threshold;
   }
 
   /**
@@ -102,231 +89,116 @@ public class ResidueColourScheme implements ColourSchemeI
    */
   public Color findColour(char c)
   {
-    return colors == null ? Color.white : colors[symbolIndex[c]];
-  }
-
-  @Override
-  public Color findColour(char c, int j, SequenceI seq)
-  {
-    Color currentColour;
-
-    if (colors != null && symbolIndex != null && (threshold == 0)
-            || aboveThreshold(c, j))
-    {
-      currentColour = colors[symbolIndex[c]];
-    }
-    else
-    {
-      currentColour = Color.white;
-    }
+    Color colour = Color.white;
 
-    if (conservationColouring)
+    if (!Comparison.isGap(c) && colors != null && symbolIndex != null
+            && c < symbolIndex.length
+            && symbolIndex[c] < colors.length)
     {
-      currentColour = applyConservation(currentColour, j);
+      colour = colors[symbolIndex[c]];
     }
 
-    return currentColour;
-  }
-
-  /**
-   * Get the percentage threshold for this colour scheme
-   * 
-   * @return Returns the percentage threshold
-   */
-  public int getThreshold()
-  {
-    return threshold;
+    return colour;
   }
 
   /**
-   * DOCUMENT ME!
-   * 
-   * @param ct
-   *          DOCUMENT ME!
+   * Default is to call the overloaded method that ignores consensus. A colour
+   * scheme that depends on consensus (for example, Blosum62), should override
+   * this method instead.
    */
-  public void setThreshold(int ct, boolean ignoreGaps)
+  @Override
+  public Color findColour(char c, int j, SequenceI seq,
+          String consensusResidue, float pid)
   {
-    threshold = ct;
-    if (ignoreGaps)
-    {
-      this.ignoreGaps = AAFrequency.PID_NOGAPS;
-    }
-    else
-    {
-      this.ignoreGaps = AAFrequency.PID_GAPS;
-    }
+    return findColour(c, j, seq);
   }
 
   /**
-   * DOCUMENT ME!
+   * Default implementation looks up the residue colour in a fixed scheme, or
+   * returns White if not found. Override this method for a colour scheme that
+   * depends on the column position or sequence.
    * 
-   * @param s
-   *          DOCUMENT ME!
+   * @param c
    * @param j
-   *          DOCUMENT ME!
-   * 
-   * @return DOCUMENT ME!
+   * @param seq
+   * @return
    */
-  public boolean aboveThreshold(char c, int j)
-  {
-    if ('a' <= c && c <= 'z')
-    {
-      // TO UPPERCASE !!!
-      // Faster than toUpperCase
-      c -= ('a' - 'A');
-    }
-
-    if (consensus == null || consensus.length < j || consensus[j] == null)
-    {
-      return false;
-    }
-
-    if ((((Integer) consensus[j].get(AAFrequency.MAXCOUNT)).intValue() != -1)
-            && consensus[j].contains(String.valueOf(c)))
-    {
-      if (((Float) consensus[j].get(ignoreGaps)).floatValue() >= threshold)
-      {
-        return true;
-      }
-    }
-
-    return false;
-  }
-
-  public boolean conservationApplied()
+  protected Color findColour(char c, int j, SequenceI seq)
   {
-    return conservationColouring;
+    return findColour(c);
   }
 
   @Override
-  public void setConservationApplied(boolean conservationApplied)
-  {
-    conservationColouring = conservationApplied;
-  }
-
-  public void setConservationInc(int i)
-  {
-    inc = i;
-  }
-
-  public int getConservationInc()
+  public void alignmentChanged(AnnotatedCollectionI alignment,
+          Map<SequenceI, SequenceCollectionI> hiddenReps)
   {
-    return inc;
   }
 
   /**
-   * DOCUMENT ME!
-   * 
-   * @param consensus
-   *          DOCUMENT ME!
+   * Answers false if the colour scheme is nucleotide or peptide specific, and
+   * the data does not match, else true. Override to modify or extend this test
+   * as required.
    */
-  public void setConsensus(Hashtable[] consensus)
+  @Override
+  public boolean isApplicableTo(AnnotatedCollectionI ac)
   {
-    if (consensus == null)
+    if (!isPeptideSpecific() && !isNucleotideSpecific())
     {
-      return;
+      return true;
     }
-
-    this.consensus = consensus;
-  }
-
-  public void setConservation(Conservation cons)
-  {
-    if (cons == null)
+    if (ac == null)
     {
-      conservationColouring = false;
-      conservation = null;
+      return true;
     }
-    else
+    /*
+     * pop-up menu on selection group before group created
+     * (no alignment context)
+     */
+    // TODO: add nucleotide flag to SequenceGroup?
+    if (ac instanceof SequenceGroup && ac.getContext() == null)
     {
-      conservationColouring = true;
-      int i, iSize = cons.getConsSequence().getLength();
-      conservation = new char[iSize];
-      for (i = 0; i < iSize; i++)
-      {
-        conservation[i] = cons.getConsSequence().getCharAt(i);
-      }
-      conservationLength = conservation.length;
+      return true;
     }
 
+    /*
+     * inspect the data context (alignment) for residue type
+     */
+    boolean nucleotide = ac.isNucleotide();
+
+    /*
+     * does data type match colour scheme type?
+     */
+    return (nucleotide && isNucleotideSpecific())
+            || (!nucleotide && isPeptideSpecific());
   }
 
   /**
-   * DOCUMENT ME!
+   * Answers true if the colour scheme is normally only for peptide data
    * 
-   * @param s
-   *          DOCUMENT ME!
-   * @param i
-   *          DOCUMENT ME!
-   * 
-   * @return DOCUMENT ME!
+   * @return
    */
-
-  Color applyConservation(Color currentColour, int i)
+  public boolean isPeptideSpecific()
   {
-
-    if ((conservationLength > i) && (conservation[i] != '*')
-            && (conservation[i] != '+'))
-    {
-      if (jalview.util.Comparison.isGap(conservation[i]))
-      {
-        currentColour = Color.white;
-      }
-      else
-      {
-        float t = 11 - (conservation[i] - '0');
-        if (t == 0)
-        {
-          return Color.white;
-        }
-
-        int red = currentColour.getRed();
-        int green = currentColour.getGreen();
-        int blue = currentColour.getBlue();
-
-        int dr = 255 - red;
-        int dg = 255 - green;
-        int db = 255 - blue;
-
-        dr *= t / 10f;
-        dg *= t / 10f;
-        db *= t / 10f;
-
-        red += (inc / 20f) * dr;
-        green += (inc / 20f) * dg;
-        blue += (inc / 20f) * db;
-
-        if (red > 255 || green > 255 || blue > 255)
-        {
-          currentColour = Color.white;
-        }
-        else
-        {
-          currentColour = new Color(red, green, blue);
-        }
-      }
-    }
-    return currentColour;
+    return false;
   }
 
-  @Override
-  public void alignmentChanged(AnnotatedCollectionI alignment,
-          Map<SequenceI, SequenceCollectionI> hiddenReps)
+  /**
+   * Answers true if the colour scheme is normally only for nucleotide data
+   * 
+   * @return
+   */
+  public boolean isNucleotideSpecific()
   {
+    return false;
   }
 
+  /**
+   * Default method returns true. Override this to return false in colour
+   * schemes that are not determined solely by the sequence symbol.
+   */
   @Override
-  public ColourSchemeI applyTo(AnnotatedCollectionI sg,
-          Map<SequenceI, SequenceCollectionI> hiddenRepSequences)
+  public boolean isSimple()
   {
-    try
-    {
-      return getClass().newInstance();
-    } catch (Exception q)
-    {
-      throw new Error(MessageManager.formatMessage(
-              "error.implementation_error_cannot_duplicate_colour_scheme",
-              new String[] { getClass().getName() }), q);
-    }
+    return true;
   }
 }