* along with this program; if not, write to the Free Software\r
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA\r
*/\r
-\r
package jalview.schemes;\r
\r
-import java.util.*;\r
+import jalview.analysis.*;\r
+\r
import java.awt.*;\r
\r
-public class ResidueColourScheme implements ColourSchemeI{\r
- Color [] colors;\r
- int threshold = 0;\r
- public Vector consensus;\r
+import java.util.*;\r
+\r
+\r
+/**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @author $author$\r
+ * @version $Revision$\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
+\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
+ /**\r
+ * The colour to be calculated, manipulated and returned\r
+ */\r
+ Color currentColour = null;\r
\r
- public ResidueColourScheme(Color[] colors, int threshold) {\r
- this.colors = colors;\r
- this.threshold = threshold;\r
- }\r
\r
- public ResidueColourScheme()\r
- { }\r
\r
- public void setConsensus(Vector consensus)\r
+ /**\r
+ * Creates a new ResidueColourScheme object.\r
+ *\r
+ * @param colors DOCUMENT ME!\r
+ * @param threshold DOCUMENT ME!\r
+ */\r
+ public ResidueColourScheme(Color[] colours, int threshold)\r
{\r
- this.consensus = consensus;\r
+ this.colors = colours;\r
+ this.threshold = threshold;\r
}\r
\r
- public Color findColour(String aa)\r
+ /**\r
+ * Creates a new ResidueColourScheme object.\r
+ */\r
+ public ResidueColourScheme()\r
{\r
- return colors[((Integer)(ResidueProperties.aaHash.get(aa))).intValue()];\r
}\r
\r
- public Color findColour(String s, int j) {\r
+ /**\r
+ * Find a colour without an index in a sequence\r
+ */\r
+ public Color findColour(String aa)\r
+ {\r
+ return colors[((Integer) (ResidueProperties.aaHash.get(aa))).intValue()];\r
+ }\r
+\r
+\r
+\r
+ public Color findColour(String s, int j)\r
+ {\r
\r
- if( threshold==0 || aboveThreshold(s,j))\r
- return colors[((Integer)(ResidueProperties.aaHash.get(s))).intValue()];\r
+ int index = ((Integer) (ResidueProperties.aaHash.get(s))).intValue();\r
+\r
+ if ((threshold == 0) || aboveThreshold(ResidueProperties.aa[index], j))\r
+ {\r
+ currentColour = colors[index];\r
+ }\r
else\r
- return Color.white;\r
+ {\r
+ currentColour = Color.white;\r
+ }\r
\r
- }\r
+ if(conservationColouring)\r
+ applyConservation(j);\r
+\r
+\r
+ return currentColour;\r
+ }\r
\r
- public int getThreshold() {\r
- return threshold;\r
+\r
+ /**\r
+ * Get the percentage threshold for this colour scheme\r
+ *\r
+ * @return Returns the percentage threshold\r
+ */\r
+ public int getThreshold()\r
+ {\r
+ return threshold;\r
}\r
\r
- public void setThreshold(int ct) {\r
- threshold = ct;\r
+ /**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @param ct DOCUMENT ME!\r
+ */\r
+ public void setThreshold(int ct, boolean ignoreGaps)\r
+ {\r
+ threshold = ct;\r
+ if(ignoreGaps)\r
+ this.ignoreGaps = "pid_nogaps";\r
+ else\r
+ this.ignoreGaps = "pid_gaps";\r
}\r
\r
+ /**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @param s DOCUMENT ME!\r
+ * @param j DOCUMENT ME!\r
+ *\r
+ * @return DOCUMENT ME!\r
+ */\r
public boolean aboveThreshold(String s, int j)\r
{\r
- Hashtable hash = (Hashtable)consensus.elementAt(j);\r
-\r
- if ( ( (Integer) hash.get("maxCount")).intValue() != -1 && hash.contains(s))\r
+ if ((((Integer) consensus[j].get("maxCount")).intValue() != -1) &&\r
+ consensus[j].contains(s))\r
{\r
- //// resCount//////////////////// ///////////////seq count////////////\r
- double sc = ( (Integer) hash.get(s)).intValue() * 100.0 / ( (Integer) hash.get("size")).intValue();\r
+ if (((Float)consensus[j].get(ignoreGaps)).floatValue() >= threshold)\r
+ {\r
+ return true;\r
+ }\r
+ }\r
\r
- if (sc >= threshold)\r
- return true;\r
+ return false;\r
+ }\r
\r
- }\r
\r
- return false;\r
+ public boolean conservationApplied()\r
+ {\r
+ return conservationColouring;\r
+ }\r
+\r
+ public void setConservationInc(int i)\r
+ {\r
+ inc = i;\r
+ }\r
+\r
+ public int getConservationInc()\r
+ {\r
+ return inc;\r
+ }\r
+\r
+ /**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @param consensus DOCUMENT ME!\r
+ */\r
+ public void setConsensus(Vector vconsensus)\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
+ }\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
}\r
+\r
+\r
+ /**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @param s DOCUMENT ME!\r
+ * @param i DOCUMENT ME!\r
+ *\r
+ * @return DOCUMENT ME!\r
+ */\r
+ void applyConservation(int i)\r
+ {\r
+ if ((conservation[i] != '*') && (conservation[i] != '+'))\r
+ {\r
+ int tmp = 10;\r
+ int t = 0;\r
+\r
+ if (!jalview.util.Comparison.isGap(conservation[i]))\r
+ {\r
+ t = conservation[i]-'0';\r
+ }\r
+\r
+ while (tmp >= t)\r
+ {\r
+ lighter(inc);\r
+ tmp--;\r
+ }\r
+ }\r
+\r
+ }\r
+\r
+ /**\r
+ * DOCUMENT ME!\r
+ *\r
+ * @param c DOCUMENT ME!\r
+ * @param inc DOCUMENT ME!\r
+ *\r
+ * @return DOCUMENT ME!\r
+ */\r
+ void lighter(int inc)\r
+ {\r
+ int red = currentColour.getRed();\r
+ int blue = currentColour.getBlue();\r
+ int green = currentColour.getGreen();\r
+\r
+ if (red < (255 - inc))\r
+ {\r
+ red = red + inc;\r
+ }\r
+ else\r
+ {\r
+ red = 255;\r
+ }\r
+\r
+ if (blue < (255 - inc))\r
+ {\r
+ blue = blue + inc;\r
+ }\r
+ else\r
+ {\r
+ blue = 255;\r
+ }\r
+\r
+ if (green < (255 - inc))\r
+ {\r
+ green = green + inc;\r
+ }\r
+ else\r
+ {\r
+ green = 255;\r
+ }\r
+\r
+ currentColour = new Color(red, green, blue);\r
+ }\r
+\r
+\r
}\r
git://source.jalview.org / jalview.git / blobdiff