73399cc480a92b793ab48646017e1e90d36d4399
[jalview.git] / src / jalview / renderer / seqfeatures / FeatureRenderer.java
1 /*
2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
4  * 
5  * This file is part of Jalview.
6  * 
7  * Jalview is free software: you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License 
9  * as published by the Free Software Foundation, either version 3
10  * of the License, or (at your option) any later version.
11  *  
12  * Jalview is distributed in the hope that it will be useful, but 
13  * WITHOUT ANY WARRANTY; without even the implied warranty 
14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
15  * PURPOSE.  See the GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
19  * The Jalview Authors are detailed in the 'AUTHORS' file.
20  */
21 package jalview.renderer.seqfeatures;
22
23 import jalview.api.AlignViewportI;
24 import jalview.api.FeatureColourI;
25 import jalview.datamodel.Range;
26 import jalview.datamodel.SequenceFeature;
27 import jalview.datamodel.SequenceI;
28 import jalview.util.Comparison;
29 import jalview.viewmodel.seqfeatures.FeatureRendererModel;
30
31 import java.awt.AlphaComposite;
32 import java.awt.Color;
33 import java.awt.FontMetrics;
34 import java.awt.Graphics;
35 import java.awt.Graphics2D;
36 import java.util.List;
37
38 public class FeatureRenderer extends FeatureRendererModel
39 {
40   private static final AlphaComposite NO_TRANSPARENCY = AlphaComposite
41           .getInstance(AlphaComposite.SRC_OVER, 1.0f);
42
43   /**
44    * Constructor given a viewport
45    * 
46    * @param viewport
47    */
48   public FeatureRenderer(AlignViewportI viewport)
49   {
50     this.av = viewport;
51   }
52
53   /**
54    * Renders the sequence using the given feature colour between the given start
55    * and end columns. Returns true if at least one column is drawn, else false
56    * (the feature range does not overlap the start and end positions).
57    * 
58    * @param g
59    * @param seq
60    * @param featureStart
61    * @param featureEnd
62    * @param featureColour
63    * @param start
64    * @param end
65    * @param y1
66    * @param colourOnly
67    * @return
68    */
69   boolean renderFeature(Graphics g, SequenceI seq, int featureStart,
70           int featureEnd, Color featureColour, int start, int end, int y1,
71           boolean colourOnly)
72   {
73     int charHeight = av.getCharHeight();
74     int charWidth = av.getCharWidth();
75     boolean validCharWidth = av.isValidCharWidth();
76
77     if (featureStart > end || featureEnd < start)
78     {
79       return false;
80     }
81
82     if (featureStart < start)
83     {
84       featureStart = start;
85     }
86     if (featureEnd >= end)
87     {
88       featureEnd = end;
89     }
90     int pady = (y1 + charHeight) - charHeight / 5;
91
92     FontMetrics fm = g.getFontMetrics();
93     for (int i = featureStart; i <= featureEnd; i++)
94     {
95       char s = seq.getCharAt(i);
96
97       if (Comparison.isGap(s))
98       {
99         continue;
100       }
101
102       g.setColor(featureColour);
103
104       g.fillRect((i - start) * charWidth, y1, charWidth, charHeight);
105
106       if (colourOnly || !validCharWidth)
107       {
108         continue;
109       }
110
111       g.setColor(Color.white);
112       int charOffset = (charWidth - fm.charWidth(s)) / 2;
113       g.drawString(String.valueOf(s),
114               charOffset + (charWidth * (i - start)), pady);
115     }
116     return true;
117   }
118
119   /**
120    * Renders the sequence using the given SCORE feature colour between the given
121    * start and end columns. Returns true if at least one column is drawn, else
122    * false (the feature range does not overlap the start and end positions).
123    * 
124    * @param g
125    * @param seq
126    * @param fstart
127    * @param fend
128    * @param featureColour
129    * @param start
130    * @param end
131    * @param y1
132    * @param bs
133    * @param colourOnly
134    * @return
135    */
136   boolean renderScoreFeature(Graphics g, SequenceI seq, int fstart,
137           int fend, Color featureColour, int start, int end, int y1,
138           byte[] bs, boolean colourOnly)
139   {
140     if (fstart > end || fend < start)
141     {
142       return false;
143     }
144
145     if (fstart < start)
146     { // fix for if the feature we have starts before the sequence start,
147       fstart = start; // but the feature end is still valid!!
148     }
149
150     if (fend >= end)
151     {
152       fend = end;
153     }
154     int charHeight = av.getCharHeight();
155     int pady = (y1 + charHeight) - charHeight / 5;
156     int ystrt = 0, yend = charHeight;
157     if (bs[0] != 0)
158     {
159       // signed - zero is always middle of residue line.
160       if (bs[1] < 128)
161       {
162         yend = charHeight * (128 - bs[1]) / 512;
163         ystrt = charHeight - yend / 2;
164       }
165       else
166       {
167         ystrt = charHeight / 2;
168         yend = charHeight * (bs[1] - 128) / 512;
169       }
170     }
171     else
172     {
173       yend = charHeight * bs[1] / 255;
174       ystrt = charHeight - yend;
175
176     }
177
178     FontMetrics fm = g.getFontMetrics();
179     int charWidth = av.getCharWidth();
180
181     for (int i = fstart; i <= fend; i++)
182     {
183       char s = seq.getCharAt(i);
184
185       if (Comparison.isGap(s))
186       {
187         continue;
188       }
189
190       g.setColor(featureColour);
191       int x = (i - start) * charWidth;
192       g.drawRect(x, y1, charWidth, charHeight);
193       g.fillRect(x, y1 + ystrt, charWidth, yend);
194
195       if (colourOnly || !av.isValidCharWidth())
196       {
197         continue;
198       }
199
200       g.setColor(Color.black);
201       int charOffset = (charWidth - fm.charWidth(s)) / 2;
202       g.drawString(String.valueOf(s),
203               charOffset + (charWidth * (i - start)), pady);
204     }
205     return true;
206   }
207
208   /**
209    * {@inheritDoc}
210    */
211   @Override
212   public Color findFeatureColour(SequenceI seq, int column, Graphics g)
213   {
214     // column is 'base 1' but getCharAt is an array index (ie from 0)
215     if (Comparison.isGap(seq.getCharAt(column - 1)))
216     {
217       /*
218        * returning null allows the colour scheme to provide gap colour
219        * - normally white, but can be customised
220        */
221       return null;
222     }
223
224     Color renderedColour = null;
225     if (transparency == 1.0f)
226     {
227       /*
228        * simple case - just find the topmost rendered visible feature colour
229        */
230       renderedColour = findFeatureColour(seq, column);
231     }
232     else
233     {
234       /*
235        * transparency case - draw all visible features in render order to
236        * build up a composite colour on the graphics context
237        */
238       renderedColour = drawSequence(g, seq, column, column, 0, true);
239     }
240     return renderedColour;
241   }
242
243   /**
244    * Draws the sequence features on the graphics context, or just determines the
245    * colour that would be drawn (if flag colourOnly is true). Returns the last
246    * colour drawn (which may not be the effective colour if transparency
247    * applies), or null if no feature is drawn in the range given.
248    * 
249    * @param g
250    *          the graphics context to draw on (may be null if colourOnly==true)
251    * @param seq
252    * @param start
253    *          start column
254    * @param end
255    *          end column
256    * @param y1
257    *          vertical offset at which to draw on the graphics
258    * @param colourOnly
259    *          if true, only do enough to determine the colour for the position,
260    *          do not draw the character
261    * @return
262    */
263   public synchronized Color drawSequence(final Graphics g,
264           final SequenceI seq, int start, int end, int y1,
265           boolean colourOnly)
266   {
267     /*
268      * if columns are all gapped, or sequence has no features, nothing to do
269      */
270     Range visiblePositions = seq.findPositions(start+1, end+1);
271     if (visiblePositions == null || !seq.getFeatures().hasFeatures())
272     {
273       return null;
274     }
275
276     updateFeatures();
277
278     if (transparency != 1f && g != null)
279     {
280       Graphics2D g2 = (Graphics2D) g;
281       g2.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER,
282               transparency));
283     }
284
285     Color drawnColour = null;
286
287     /*
288      * iterate over features in ordering of their rendering (last is on top)
289      */
290     for (int renderIndex = 0; renderIndex < renderOrder.length; renderIndex++)
291     {
292       String type = renderOrder[renderIndex];
293       if (!showFeatureOfType(type))
294       {
295         continue;
296       }
297
298       FeatureColourI fc = getFeatureStyle(type);
299       List<SequenceFeature> overlaps = seq.getFeatures().findFeatures(
300               visiblePositions.getBegin(), visiblePositions.getEnd(), type);
301
302       if (fc.isSimpleColour())
303       {
304         filterFeaturesForDisplay(overlaps);
305       }
306
307       for (SequenceFeature sf : overlaps)
308       {
309         Color featureColour = getColor(sf, fc);
310         if (featureColour == null)
311         {
312           /*
313            * feature excluded by visibility settings, filters, or colour threshold
314            */
315           continue;
316         }
317
318         /*
319          * if feature starts/ends outside the visible range,
320          * restrict to visible positions (or if a contact feature,
321          * to a single position)
322          */
323         int visibleStart = sf.getBegin();
324         if (visibleStart < visiblePositions.getBegin())
325         {
326           visibleStart = sf.isContactFeature() ? sf.getEnd()
327                   : visiblePositions.getBegin();
328         }
329         int visibleEnd = sf.getEnd();
330         if (visibleEnd > visiblePositions.getEnd())
331         {
332           visibleEnd = sf.isContactFeature() ? sf.getBegin()
333                   : visiblePositions.getEnd();
334         }
335
336         int featureStartCol = seq.findIndex(visibleStart);
337         int featureEndCol = sf.begin == sf.end ? featureStartCol : seq
338                 .findIndex(visibleEnd);
339
340         // Color featureColour = getColour(sequenceFeature);
341
342         boolean isContactFeature = sf.isContactFeature();
343
344         if (isContactFeature)
345         {
346           boolean drawn = renderFeature(g, seq, featureStartCol - 1,
347                   featureStartCol - 1, featureColour, start, end, y1,
348                   colourOnly);
349           drawn |= renderFeature(g, seq, featureEndCol - 1,
350                   featureEndCol - 1, featureColour, start, end, y1,
351                   colourOnly);
352           if (drawn)
353           {
354             drawnColour = featureColour;
355           }
356         }
357         else
358         {
359           /*
360            * showing feature score by height of colour
361            * is not implemented as a selectable option 
362            *
363           if (av.isShowSequenceFeaturesHeight()
364                   && !Float.isNaN(sequenceFeature.score))
365           {
366             boolean drawn = renderScoreFeature(g, seq,
367                     seq.findIndex(sequenceFeature.begin) - 1,
368                     seq.findIndex(sequenceFeature.end) - 1, featureColour,
369                     start, end, y1, normaliseScore(sequenceFeature),
370                     colourOnly);
371             if (drawn)
372             {
373               drawnColour = featureColour;
374             }
375           }
376           else
377           {
378           */
379             boolean drawn = renderFeature(g, seq,
380                     featureStartCol - 1,
381                     featureEndCol - 1, featureColour,
382                     start, end, y1, colourOnly);
383             if (drawn)
384             {
385               drawnColour = featureColour;
386             }
387           /*}*/
388         }
389       }
390     }
391
392     if (transparency != 1.0f && g != null)
393     {
394       /*
395        * reset transparency
396        */
397       Graphics2D g2 = (Graphics2D) g;
398       g2.setComposite(NO_TRANSPARENCY);
399     }
400
401     return drawnColour;
402   }
403
404   /**
405    * Called when alignment in associated view has new/modified features to
406    * discover and display.
407    * 
408    */
409   @Override
410   public void featuresAdded()
411   {
412     findAllFeatures();
413   }
414
415   /**
416    * Returns the sequence feature colour rendered at the given column position,
417    * or null if none found. The feature of highest render order (i.e. on top) is
418    * found, subject to both feature type and feature group being visible, and
419    * its colour returned. This method is suitable when no feature transparency
420    * applied (only the topmost visible feature colour is rendered).
421    * <p>
422    * Note this method does not check for a gap in the column so would return the
423    * colour for features enclosing a gapped column. Check for gap before calling
424    * if different behaviour is wanted.
425    * 
426    * @param seq
427    * @param column
428    *          (1..)
429    * @return
430    */
431   Color findFeatureColour(SequenceI seq, int column)
432   {
433     /*
434      * check for new feature added while processing
435      */
436     updateFeatures();
437
438     /*
439      * inspect features in reverse renderOrder (the last in the array is 
440      * displayed on top) until we find one that is rendered at the position
441      */
442     for (int renderIndex = renderOrder.length
443             - 1; renderIndex >= 0; renderIndex--)
444     {
445       String type = renderOrder[renderIndex];
446       if (!showFeatureOfType(type))
447       {
448         continue;
449       }
450
451       List<SequenceFeature> overlaps = seq.findFeatures(column, column,
452               type);
453       for (SequenceFeature sequenceFeature : overlaps)
454       {
455         if (!featureGroupNotShown(sequenceFeature))
456         {
457           Color col = getColour(sequenceFeature);
458           if (col != null)
459           {
460             return col;
461           }
462         }
463       }
464     }
465
466     /*
467      * no displayed feature found at position
468      */
469     return null;
470   }
471 }