/*\r
* Jalview - A Sequence Alignment Editor and Viewer\r
-* Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
+* Copyright (C) 2006 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
*\r
* This program is free software; you can redistribute it and/or\r
* modify it under the terms of the GNU General Public License\r
*/\r
package jalview.schemes;\r
\r
+import jalview.analysis.*;\r
+\r
import java.awt.*;\r
\r
import java.util.*;\r
*/\r
public class ResidueColourScheme implements ColourSchemeI\r
{\r
+\r
+ boolean conservationColouring = false;\r
+ boolean consensusColouring = false;\r
+\r
Color[] colors;\r
int threshold = 0;\r
\r
/* Set when threshold colouring to either pid_gaps or pid_nogaps*/\r
- protected String ignoreGaps = "pid_gaps";\r
+ protected String ignoreGaps = AAFrequency.PID_GAPS;\r
+\r
+ /** Consenus as a hashtable array */\r
+ Hashtable [] consensus;\r
+\r
+ /** Conservation string as a char array */\r
+ char [] conservation;\r
+\r
+ /** DOCUMENT ME!! */\r
+ int inc = 30;\r
\r
- /** DOCUMENT ME!! */\r
- public Hashtable [] consensus;\r
\r
/**\r
* Creates a new ResidueColourScheme object.\r
* @param colors DOCUMENT ME!\r
* @param threshold DOCUMENT ME!\r
*/\r
- public ResidueColourScheme(Color[] colors, int threshold)\r
+ public ResidueColourScheme(Color[] colours, int threshold)\r
{\r
- this.colors = colors;\r
+ this.colors = colours;\r
this.threshold = threshold;\r
}\r
\r
}\r
\r
/**\r
- * DOCUMENT ME!\r
+ * Find a colour without an index in a sequence\r
+ */\r
+ public Color findColour(String aa)\r
+ {\r
+ return colors[ResidueProperties.aaIndex[aa.charAt(0)]];\r
+ }\r
+\r
+\r
+\r
+ public Color findColour(String s, int j)\r
+ {\r
+ Color currentColour;\r
+ int index = ResidueProperties.aaIndex[s.charAt(0)];\r
+\r
+ if ((threshold == 0) || aboveThreshold(ResidueProperties.aa[index], j))\r
+ {\r
+ currentColour = colors[index];\r
+ }\r
+ else\r
+ {\r
+ currentColour = Color.white;\r
+ }\r
+\r
+ if(conservationColouring)\r
+ applyConservation(currentColour, j);\r
+\r
+\r
+ return currentColour;\r
+ }\r
+\r
+\r
+ /**\r
+ * Get the percentage threshold for this colour scheme\r
*\r
- * @param consensus DOCUMENT ME!\r
+ * @return Returns the percentage threshold\r
*/\r
- public void setConsensus(Vector vconsensus)\r
+ public int getThreshold()\r
{\r
- int i, iSize=vconsensus.size();\r
- consensus = new Hashtable[iSize];\r
- for(i=0; i<iSize; i++)\r
- consensus[i] = (Hashtable)vconsensus.elementAt(i);\r
+ return threshold;\r
}\r
\r
/**\r
* DOCUMENT ME!\r
*\r
- * @param aa DOCUMENT ME!\r
- *\r
- * @return DOCUMENT ME!\r
+ * @param ct DOCUMENT ME!\r
*/\r
- public Color findColour(String aa)\r
+ public void setThreshold(int ct, boolean ignoreGaps)\r
{\r
- return colors[((Integer) (ResidueProperties.aaHash.get(aa))).intValue()];\r
+ threshold = ct;\r
+ if(ignoreGaps)\r
+ this.ignoreGaps = AAFrequency.PID_NOGAPS;\r
+ else\r
+ this.ignoreGaps = AAFrequency.PID_GAPS;\r
}\r
\r
/**\r
*\r
* @return DOCUMENT ME!\r
*/\r
- public Color findColour(String s, int j)\r
+ public boolean aboveThreshold(String s, int j)\r
{\r
- if ((threshold == 0) || aboveThreshold(s, j))\r
+ char c = s.charAt(0);\r
+ if ('a' <= c && c <= 'z')\r
{\r
- return colors[((Integer) (ResidueProperties.aaHash.get(s))).intValue()];\r
+ // TO UPPERCASE !!!\r
+ //Faster than toUpperCase\r
+ c -= ('a' - 'A');\r
+ s = String.valueOf(c);\r
}\r
- else\r
+\r
+ if (consensus == null || consensus[j] == null)\r
+ return false;\r
+\r
+ if ( ( ( (Integer) consensus[j].get(AAFrequency.MAXCOUNT)).intValue() != -1) &&\r
+ consensus[j].contains(s))\r
{\r
- return Color.white;\r
+ if ( ( (Float) consensus[j].get(ignoreGaps)).floatValue() >= threshold)\r
+ {\r
+ return true;\r
+ }\r
}\r
+\r
+ return false;\r
}\r
\r
- /**\r
- * DOCUMENT ME!\r
- *\r
- * @return DOCUMENT ME!\r
- */\r
- public int getThreshold()\r
+\r
+ public boolean conservationApplied()\r
{\r
- return threshold;\r
+ return conservationColouring;\r
}\r
\r
- /**\r
- * DOCUMENT ME!\r
- *\r
- * @param ct DOCUMENT ME!\r
- */\r
- public void setThreshold(int ct, boolean ignoreGaps)\r
+ public void setConservationInc(int i)\r
{\r
- threshold = ct;\r
- if(ignoreGaps)\r
- this.ignoreGaps = "pid_nogaps";\r
- else\r
- this.ignoreGaps = "pid_gaps";\r
+ inc = i;\r
+ }\r
+\r
+ public int getConservationInc()\r
+ {\r
+ return inc;\r
}\r
\r
/**\r
* DOCUMENT ME!\r
*\r
- * @param s DOCUMENT ME!\r
- * @param j DOCUMENT ME!\r
- *\r
- * @return DOCUMENT ME!\r
+ * @param consensus DOCUMENT ME!\r
*/\r
- public boolean aboveThreshold(String s, int j)\r
+ public void setConsensus(Hashtable [] consensus)\r
{\r
if(consensus==null)\r
- System.out.println("its null");\r
- if ((((Integer) consensus[j].get("maxCount")).intValue() != -1) &&\r
- consensus[j].contains(s))\r
- {\r
- float ratio = ((Float)consensus[j].get(ignoreGaps)).floatValue();\r
+ return;\r
\r
- if (ratio >= threshold)\r
- {\r
- return true;\r
- }\r
- }\r
+ this.consensus = consensus;\r
+ }\r
+\r
+\r
+\r
+ public void setConservation(Conservation cons)\r
+ {\r
+ if(cons==null)\r
+ {\r
+ conservationColouring = false;\r
+ conservation = null;\r
+ }\r
+ else\r
+ {\r
+ conservationColouring = true;\r
+ int i, iSize = cons.getConsSequence().getLength();\r
+ conservation = new char[iSize];\r
+ for (i = 0; i < iSize; i++)\r
+ conservation[i] = cons.getConsSequence().getCharAt(i);\r
+ }\r
\r
- return false;\r
}\r
+\r
+\r
+ /**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @param s DOCUMENT ME!\r
+ * @param i DOCUMENT ME!\r
+ *\r
+ * @return DOCUMENT ME!\r
+ */\r
+\r
+ void applyConservation(Color currentColour, int i)\r
+ {\r
+\r
+ if ((conservation[i] != '*') && (conservation[i] != '+'))\r
+ {\r
+ if(jalview.util.Comparison.isGap(conservation[i]))\r
+ {\r
+ currentColour = Color.white;\r
+ }\r
+ else\r
+ {\r
+ float t = 11 - (conservation[i] - '0');\r
+ if(t==0)\r
+ {\r
+ currentColour = Color.white;\r
+ return;\r
+ }\r
+\r
+ int red = currentColour.getRed();\r
+ int green = currentColour.getGreen();\r
+ int blue = currentColour.getBlue();\r
+\r
+ int dr = 255 - red;\r
+ int dg = 255 - green;\r
+ int db = 255 - blue;\r
+\r
+ dr *= t / 10f;\r
+ dg *= t / 10f;\r
+ db *= t / 10f;\r
+\r
+ red += (inc / 20f) * dr;\r
+ green += (inc / 20f) * dg;\r
+ blue += (inc / 20f) * db;\r
+\r
+ if (red > 255 || green > 255 || blue > 255)\r
+ currentColour = Color.white;\r
+ else\r
+ currentColour = new Color(red, green, blue);\r
+ }\r
+ }\r
+ }\r
+\r
+\r
}\r