913c870ece08fadc4752f708cd939530bc3b71e7
[jalview.git] / src / jalview / renderer / OverviewRenderer.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;
22
23 import jalview.api.AlignmentColsCollectionI;
24 import jalview.api.AlignmentRowsCollectionI;
25 import jalview.datamodel.AlignmentAnnotation;
26 import jalview.datamodel.Annotation;
27 import jalview.datamodel.SequenceI;
28 import jalview.renderer.seqfeatures.FeatureColourFinder;
29 import jalview.renderer.seqfeatures.FeatureRenderer;
30 import jalview.viewmodel.OverviewDimensions;
31
32 import java.awt.Color;
33 import java.awt.Graphics;
34 import java.awt.image.BufferedImage;
35
36 public class OverviewRenderer
37 {
38   private FeatureColourFinder finder;
39
40   private jalview.api.SequenceRenderer sr;
41
42   // image to render on
43   private BufferedImage miniMe;
44
45   // raw number of pixels to allocate to each column
46   private float pixelsPerCol;
47
48   // raw number of pixels to allocate to each row
49   private float pixelsPerSeq;
50
51   // flag to indicate whether to halt drawing
52   private boolean redraw = false;
53
54   public OverviewRenderer(jalview.api.SequenceRenderer seqRenderer,
55           FeatureRenderer fr, OverviewDimensions od)
56   {
57     sr = seqRenderer;
58     finder = new FeatureColourFinder(fr);
59
60     pixelsPerCol = od.getPixelsPerCol();
61     pixelsPerSeq = od.getPixelsPerSeq();
62     miniMe = new BufferedImage(od.getWidth(), od.getHeight(),
63             BufferedImage.TYPE_INT_RGB);
64   }
65
66   /**
67    * Draw alignment rows and columns onto an image
68    * 
69    * @param rit
70    *          Iterator over rows to be drawn
71    * @param cit
72    *          Iterator over columns to be drawn
73    * @return image containing the drawing
74    */
75   public BufferedImage draw(AlignmentRowsCollectionI rows,
76           AlignmentColsCollectionI cols)
77   {
78     int rgbcolor = Color.white.getRGB();
79     int seqIndex = 0;
80     int pixelRow = 0;
81
82     for (int alignmentRow : rows)
83     {
84       if (redraw)
85       {
86         break;
87       }
88
89       // get details of this alignment row
90       boolean hidden = rows.isHidden(alignmentRow);
91       SequenceI seq = rows.getSequence(alignmentRow);
92
93       // calculate where this row extends to in pixels
94       int endRow = Math.min(Math.round((seqIndex + 1) * pixelsPerSeq) - 1,
95               miniMe.getHeight() - 1);
96
97       int colIndex = 0;
98       int pixelCol = 0;
99       for (int alignmentCol : cols)
100       {
101         if (redraw)
102         {
103           break;
104         }
105
106         // calculate where this column extends to in pixels
107         int endCol = Math.min(
108                 Math.round((colIndex + 1) * pixelsPerCol) - 1,
109                 miniMe.getWidth() - 1);
110
111         // don't do expensive colour determination if we're not going to use it
112         // NB this is important to avoid performance issues in the overview
113         // panel
114         if (pixelCol <= endCol)
115         {
116           // determine the colour based on the sequence and column position
117           rgbcolor = getColumnColourFromSequence(seq,
118                   hidden || cols.isHidden(alignmentCol), alignmentCol,
119                   finder);
120
121           // fill in the appropriate number of pixels
122           for (int row = pixelRow; row <= endRow; ++row)
123           {
124             for (int col = pixelCol; col <= endCol; ++col)
125             {
126               miniMe.setRGB(col, row, rgbcolor);
127             }
128           }
129
130           pixelCol = endCol + 1;
131         }
132         colIndex++;
133       }
134       pixelRow = endRow + 1;
135       seqIndex++;
136     }
137     return miniMe;
138   }
139
140   /*
141    * Find the colour of a sequence at a specified column position
142    */
143   private int getColumnColourFromSequence(jalview.datamodel.SequenceI seq,
144           boolean isHidden, int lastcol, FeatureColourFinder fcfinder)
145   {
146     Color color = Color.white;
147
148     if ((seq != null) && (seq.getLength() > lastcol))
149     {
150       color = sr.getResidueColour(seq, lastcol, fcfinder);
151     }
152
153     if (isHidden)
154     {
155       color = color.darker().darker();
156     }
157
158     return color.getRGB();
159   }
160
161   /**
162    * Draw the alignment annotation in the overview panel
163    * 
164    * @param g
165    *          the graphics object to draw on
166    * @param anno
167    *          alignment annotation information
168    * @param charWidth
169    *          alignment character width value
170    * @param y
171    *          y-position for the annotation graph
172    * @param cols
173    *          the collection of columns used in the overview panel
174    */
175   public void drawGraph(Graphics g, AlignmentAnnotation anno, int charWidth,
176           int y, AlignmentColsCollectionI cols)
177   {
178     Annotation[] annotations = anno.annotations;
179     g.setColor(Color.white);
180     g.fillRect(0, 0, miniMe.getWidth(), y);
181
182     int height;
183     int colIndex = 0;
184     int pixelCol = 0;
185     for (int alignmentCol : cols)
186     {
187       if (redraw)
188       {
189         break;
190       }
191       if (alignmentCol >= annotations.length)
192       {
193         break; // no more annotations to draw here
194       }
195       else
196       {
197         int endCol = Math.min(
198                 Math.round((colIndex + 1) * pixelsPerCol) - 1,
199                 miniMe.getWidth() - 1);
200
201         if (annotations[alignmentCol] != null)
202         {
203           if (annotations[alignmentCol].colour == null)
204           {
205             g.setColor(Color.black);
206           }
207           else
208           {
209             g.setColor(annotations[alignmentCol].colour);
210           }
211
212           height = (int) ((annotations[alignmentCol].value / anno.graphMax) * y);
213           if (height > y)
214           {
215             height = y;
216           }
217
218           g.fillRect(pixelCol, y - height, endCol - pixelCol + 1, height);
219         }
220         pixelCol = endCol + 1;
221         colIndex++;
222       }
223     }
224   }
225
226   public void setRedraw(boolean b)
227   {
228     synchronized (this)
229     {
230       redraw = b;
231     }
232   }
233 }