after merge
[jalview.git] / src / jalview / appletgui / FeatureRenderer.java
1 /*\r
2  * Jalview - A Sequence Alignment Editor and Viewer\r
3  * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
4  *\r
5  * This program is free software; you can redistribute it and/or\r
6  * modify it under the terms of the GNU General Public License\r
7  * as published by the Free Software Foundation; either version 2\r
8  * of the License, or (at your option) any later version.\r
9  *\r
10  * This program is distributed in the hope that it will be useful,\r
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
13  * GNU General Public License for more details.\r
14  *\r
15  * You should have received a copy of the GNU General Public License\r
16  * along with this program; if not, write to the Free Software\r
17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA\r
18  */\r
19 package jalview.appletgui;\r
20 \r
21 import jalview.datamodel.*;\r
22 \r
23 import java.awt.*;\r
24 \r
25 import java.util.*;\r
26 \r
27 \r
28 /**\r
29  * DOCUMENT ME!\r
30  *\r
31  * @author $author$\r
32  * @version $Revision$\r
33  */\r
34 public class FeatureRenderer\r
35 {\r
36     AlignViewport av;\r
37     SequenceGroup currentSequenceGroup = null;\r
38     SequenceGroup[] allGroups = null;\r
39     Color resBoxColour;\r
40     Graphics graphics;\r
41     float transparency = .4f;\r
42 \r
43     // The following vector holds the features which are\r
44     // to be added, in the correct order or rendering\r
45     Vector featuresDisplayed;\r
46 \r
47     /**\r
48      * Creates a new FeatureRenderer object.\r
49      *\r
50      * @param av DOCUMENT ME!\r
51      */\r
52     public FeatureRenderer(AlignViewport av)\r
53     {\r
54         this.av = av;\r
55         initColours();\r
56         String version = System.getProperty("java.version");\r
57         if (version.indexOf("1.1") == 0)\r
58           highlightTransparent = false;\r
59 \r
60     }\r
61 \r
62 boolean highlightTransparent = true;\r
63     /**\r
64      * DOCUMENT ME!\r
65      *\r
66      * @param g DOCUMENT ME!\r
67      * @param seq DOCUMENT ME!\r
68      * @param sg DOCUMENT ME!\r
69      * @param start DOCUMENT ME!\r
70      * @param end DOCUMENT ME!\r
71      * @param x1 DOCUMENT ME!\r
72      * @param y1 DOCUMENT ME!\r
73      * @param width DOCUMENT ME!\r
74      * @param height DOCUMENT ME!\r
75      */\r
76     public void drawSequence(Graphics g, SequenceI seq, SequenceGroup[] sg,\r
77         int start, int end, int x1, int y1, int width, int height)\r
78     {\r
79 \r
80         if(seq.getSequenceFeatures()==null)\r
81           return;\r
82 \r
83         Enumeration e = null, e2;\r
84         String type;\r
85         if(featuresDisplayed!=null)\r
86           e = featuresDisplayed.elements();\r
87         else\r
88          e = seq.getSequenceFeatures().elements();\r
89 \r
90        if(highlightTransparent)\r
91         {\r
92         //  MyGraphics.SetTransparency(g, transparency);\r
93         }\r
94 \r
95 \r
96         while (e.hasMoreElements())\r
97         {\r
98             SequenceFeature sf=null;\r
99             if(featuresDisplayed!=null)\r
100             {\r
101               e2 = seq.getSequenceFeatures().elements();\r
102               type = e.nextElement().toString();\r
103               while(e2.hasMoreElements())\r
104               {\r
105                 sf = (SequenceFeature) e2.nextElement();\r
106                 if(sf.getType().equals(type))\r
107                   break;\r
108                 else\r
109                   sf = null;\r
110               }\r
111             }\r
112             else\r
113             {\r
114               sf = (SequenceFeature) e.nextElement();\r
115               type = sf.getType();\r
116             }\r
117 \r
118             if(sf==null)\r
119               continue;\r
120 \r
121 \r
122             if (sf.getBegin() > seq.getEnd())\r
123             {\r
124                 continue;\r
125             }\r
126 \r
127             int fstart = seq.findIndex(sf.getBegin()) - 1;\r
128             int fend = seq.findIndex(sf.getEnd()) - 1;\r
129 \r
130             if (((fstart <= end) && (fend >= start)))\r
131             {\r
132                 if (fstart < start)\r
133                 { // fix for if the feature we have starts before the sequence start,\r
134                     fstart = start; // but the feature end is still valid!!\r
135                 }\r
136 \r
137                 if (fend >= end)\r
138                 {\r
139                     fend = end;\r
140                 }\r
141 \r
142                 if (fstart == fend)\r
143                 {\r
144                     g.setColor(getColour(type));\r
145                     g.fillRoundRect((fstart - start) * width, y1, width,\r
146                         height, 4, 4);\r
147                     g.setColor(Color.white);\r
148 \r
149                     char s = seq.getSequence().charAt(fstart);\r
150                     FontMetrics fm = g.getFontMetrics();\r
151                     int charOffset = (width - fm.charWidth(s)) / 2;\r
152                     int pady = height / 5;\r
153                     g.drawString(String.valueOf(s),\r
154                         charOffset + x1 + (width * (fstart - start)),\r
155                         (y1 + height) - pady);\r
156                 }\r
157                 else\r
158                 {\r
159                     for (int i = fstart; i <= fend; i++)\r
160                     {\r
161                         char s = seq.getSequence().charAt(i);\r
162 \r
163                         if (jalview.util.Comparison.isGap(s))\r
164                         {\r
165                             continue;\r
166                         }\r
167 \r
168                         g.setColor(getColour(type));\r
169                         g.fillRect((i - start) * width, y1, width, height);\r
170 \r
171                         g.setColor(Color.white);\r
172 \r
173                         FontMetrics fm = g.getFontMetrics();\r
174                         int charOffset = (width - fm.charWidth(s)) / 2;\r
175                         int pady = height / 5;\r
176                         g.drawString(String.valueOf(s),\r
177                             charOffset + x1 + (width * (i - start)),\r
178                             (y1 + height) - pady);\r
179                     }\r
180                 }\r
181             }\r
182           }\r
183 \r
184           if (highlightTransparent)\r
185           {\r
186         //    MyGraphics.SetTransparency(g, 1.0f);\r
187           }\r
188     }\r
189 \r
190     public Color getColour(String featureType)\r
191     {\r
192       return (Color)featureColours.get(featureType);\r
193     }\r
194 \r
195     public void setColour(String featureType, Color col)\r
196     {\r
197       featureColours.put(featureType, col);\r
198     }\r
199 \r
200     public void setTransparency(float value)\r
201     {\r
202       transparency = value;\r
203     }\r
204 \r
205     public float getTransparency()\r
206     {\r
207       return transparency;\r
208     }\r
209 \r
210     public void setFeaturePriority(Object [][] data)\r
211     {\r
212       // The feature table will display high priority\r
213       // features at the top, but theses are the ones\r
214       // we need to render last, so invert the data\r
215       featuresDisplayed = new Vector();\r
216       for(int i=data.length-1; i>-1; i--)\r
217       {\r
218        String type = data[i][0].toString();\r
219        setColour(type, (Color)data[i][1]);\r
220        if( ((Boolean)data[i][2]).booleanValue() )\r
221          featuresDisplayed.addElement(type);\r
222       }\r
223     }\r
224 \r
225     Hashtable featureColours = new Hashtable();\r
226     void initColours()\r
227     {\r
228       featureColours.put("active site", new Color(255, 75, 0));\r
229       featureColours.put("binding site", new Color(245, 85, 0));\r
230       featureColours.put("calcium-binding region", new Color(235, 95, 0));\r
231       featureColours.put("chain", new Color(225, 105, 0));\r
232       featureColours.put("coiled-coil region", new Color(215, 115, 0));\r
233       featureColours.put("compositionally biased region", new Color(205, 125, 0));\r
234       featureColours.put("cross-link", new Color(195, 135, 0));\r
235       featureColours.put("disulfide bond", new Color(185, 145, 0));\r
236       featureColours.put("DNA-binding region", new Color(175, 155, 0));\r
237       featureColours.put("domain", new Color(165, 165, 0));\r
238       featureColours.put("glycosylation site", new Color(155, 175, 0));\r
239       featureColours.put("helix", new Color(145, 185, 0));\r
240       featureColours.put("initiator methionine", new Color(135, 195, 5));\r
241       featureColours.put("lipid moiety-binding region", new Color(125, 205, 15));\r
242       featureColours.put("metal ion-binding site", new Color(115, 215, 25));\r
243       featureColours.put("modified residue", new Color(105, 225, 35));\r
244       featureColours.put("mutagenesis site", new Color(95, 235, 45));\r
245       featureColours.put("non-consecutive residues", new Color(85, 245, 55));\r
246       featureColours.put("non-terminal residue", new Color(75, 255, 65));\r
247       featureColours.put("nucleotide phosphate-binding region",\r
248                          new Color(65, 245, 75));\r
249       featureColours.put("peptide", new Color(55, 235, 85));\r
250       featureColours.put("propeptide", new Color(45, 225, 95));\r
251       featureColours.put("region of interest", new Color(35, 215, 105));\r
252       featureColours.put("repeat", new Color(25, 205, 115));\r
253       featureColours.put("selenocysteine", new Color(15, 195, 125));\r
254       featureColours.put("sequence conflict", new Color(5, 185, 135));\r
255       featureColours.put("sequence variant", new Color(0, 175, 145));\r
256       featureColours.put("short sequence motif", new Color(0, 165, 155));\r
257       featureColours.put("signal peptide", new Color(0, 155, 165));\r
258       featureColours.put("site", new Color(0, 145, 175));\r
259       featureColours.put("splice variant", new Color(0, 135, 185));\r
260       featureColours.put("strand", new Color(0, 125, 195));\r
261       featureColours.put("topological domain", new Color(0, 115, 205));\r
262       featureColours.put("transit peptide", new Color(0, 105, 215));\r
263       featureColours.put("transmembrane region", new Color(0, 95, 225));\r
264       featureColours.put("turn", new Color(0, 85, 235));\r
265       featureColours.put("unsure residue", new Color(0, 75, 245));\r
266       featureColours.put("zinc finger region", new Color(0, 65, 255));\r
267     }\r
268 \r
269 }\r
270 \r
271 \r
272 \r