c9198ed2d35a1cd9ac66c51400d6b5b0d595a752
[jalview.git] / src / jalview / renderer / ContactMapRenderer.java
1 /**
2  * 
3  */
4 package jalview.renderer;
5
6 import jalview.api.AlignViewportI;
7 import jalview.datamodel.AlignmentAnnotation;
8 import jalview.datamodel.Annotation;
9 import jalview.datamodel.ColumnSelection;
10 import jalview.datamodel.ContactListI;
11 import jalview.datamodel.ContactRange;
12 import jalview.datamodel.HiddenColumns;
13 import jalview.renderer.api.AnnotationRowRendererI;
14
15 import java.awt.Color;
16 import java.awt.Graphics;
17
18 /**
19  * @author jprocter
20  *
21  */
22 public class ContactMapRenderer implements AnnotationRowRendererI
23 {
24
25   @Override
26   public void renderRow(Graphics g, int charWidth, int charHeight,
27           boolean hasHiddenColumns, AlignViewportI viewport, HiddenColumns hiddenColumns,
28           ColumnSelection columnSelection, AlignmentAnnotation _aa,
29           Annotation[] aa_annotations, int sRes, int eRes, float min,
30           float max, int y)
31   {
32     if (sRes > aa_annotations.length)
33     {
34       return;
35     }
36     eRes = Math.min(eRes, aa_annotations.length);
37
38     int x = 0, y2 = y;
39
40     g.setColor(Color.pink);
41
42     g.drawLine(x, y2, (eRes - sRes) * charWidth, y2);
43
44     int column;
45     int aaMax = aa_annotations.length - 1;
46     while (x < eRes - sRes)
47     {
48       column = sRes + x;
49       if (hasHiddenColumns)
50       {
51         column = hiddenColumns.visibleToAbsoluteColumn(column);
52       }
53
54       if (column > aaMax)
55       {
56         break;
57       }
58
59       if (aa_annotations[column] == null)
60       {
61         x++;
62         continue;
63       }
64       /*
65        * {profile type, #values, total count, char1, pct1, char2, pct2...}
66        */
67       ContactListI contacts = viewport.getContactList(_aa, column);
68       if (contacts == null)
69       {
70         return;
71       }
72       int contact_height = contacts.getContactHeight();
73       // fractional pixel height to render each contact cell
74       double pixels_per_contact = ((double) _aa.graphHeight)
75                       / (double) contact_height;
76       // fractional number of contacts covering each pixel
77       double contacts_per_pixel = 1d/pixels_per_contact;
78       // number of contacts to render at a time
79       int step = (pixels_per_contact<1) ? (int) Math.floor(contacts_per_pixel) : (int) Math.ceil(pixels_per_contact);
80       
81       int pixels_step = (int) Math.ceil(step*pixels_per_contact);
82       int cstart, cend = -1;
83       for (int ht = y2, eht = y2 - _aa.graphHeight; ht >= eht; ht -= pixels_step)
84       {
85         cstart = cend + 1;
86         cend = (int) Math.min(contact_height, cstart + step);
87         // TODO show maximum colour for range - sort of done
88         // also need a 'getMaxPosForRange(start,end)'
89         g.setColor(getColorForRange(min, max, contacts, cstart, cend));
90
91         if (pixels_step>1)
92         {
93           g.fillRect(x * charWidth, ht, charWidth, 1 + (int) pixels_step);
94         }
95         else
96         {
97           g.drawLine(x * charWidth, ht, (x + 1) * charWidth, ht);
98         }
99       }
100       x++;
101     }
102
103   }
104
105   Color minColor = Color.white, maxColor = Color.magenta;
106
107
108   Color shadeFor(float min, float max, float value)
109   {
110     return jalview.util.ColorUtils.getGraduatedColour(value, 0, minColor,
111             max, maxColor);
112   }
113
114   public Color getColorForRange(float min, float max, ContactListI cl,
115           int i, int j)
116   {
117     ContactRange cr = cl.getRangeFor(i, j);
118     // average for moment - probably more interested in maxIntProj though
119     return shadeFor(min, max, (float) cr.getMean());
120   }
121
122 }