work on pivot table and FREQUENCIES usage
[pspp] / src / output / charts / boxplot-cairo.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 2009 Free Software Foundation, Inc.
3
4    This program is free software: you can redistribute it and/or modify
5    it under the terms of the GNU General Public License as published by
6    the Free Software Foundation, either version 3 of the License, or
7    (at your option) any later version.
8
9    This program is distributed in the hope that it will be useful,
10    but WITHOUT ANY WARRANTY; without even the implied warranty of
11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12    GNU General Public License for more details.
13
14    You should have received a copy of the GNU General Public License
15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
16
17 #include <config.h>
18
19 #include <output/cairo-chart.h>
20
21 #include <math.h>
22
23 #include <output/charts/boxplot.h>
24 #include <math/box-whisker.h>
25 #include <math/chart-geometry.h>
26
27 /* Draw an OUTLIER on the plot CH
28  * at CENTRELINE
29  */
30 static void
31 draw_case (cairo_t *cr, const struct xrchart_geometry *geom, double centreline,
32            const struct outlier *outlier)
33 {
34   double y = geom->data_bottom + (outlier->value - geom->y_min) * geom->ordinate_scale;
35   xrchart_draw_marker (cr, centreline, y,
36                      outlier->extreme ? XRMARKER_ASTERISK : XRMARKER_CIRCLE,
37                      20);
38
39   cairo_move_to (cr, centreline + 10, y);
40   xrchart_label (cr, 'l', 'c', geom->font_size, ds_cstr (&outlier->label));
41 }
42
43 static void
44 boxplot_draw_box (cairo_t *cr, const struct xrchart_geometry *geom,
45                   double box_centre,
46                   double box_width,
47                   const struct box_whisker *bw,
48                   const char *name)
49 {
50   double whisker[2];
51   double hinge[3];
52   struct ll *ll;
53
54   const struct ll_list *outliers;
55
56   const double box_left = box_centre - box_width / 2.0;
57
58   const double box_right = box_centre + box_width / 2.0;
59
60   double box_bottom ;
61   double box_top ;
62   double bottom_whisker ;
63   double top_whisker ;
64
65   box_whisker_whiskers (bw, whisker);
66   box_whisker_hinges (bw, hinge);
67
68   box_bottom = geom->data_bottom + (hinge[0] - geom->y_min ) * geom->ordinate_scale;
69
70   box_top = geom->data_bottom + (hinge[2] - geom->y_min ) * geom->ordinate_scale;
71
72   bottom_whisker = geom->data_bottom + (whisker[0] - geom->y_min) *
73     geom->ordinate_scale;
74
75   top_whisker = geom->data_bottom + (whisker[1] - geom->y_min) * geom->ordinate_scale;
76
77   /* Draw the box */
78   cairo_rectangle (cr,
79                    box_left,
80                    box_bottom,
81                    box_right - box_left,
82                    box_top - box_bottom);
83   cairo_save (cr);
84   cairo_set_source_rgb (cr,
85                         geom->fill_colour.red / 255.0,
86                         geom->fill_colour.green / 255.0,
87                         geom->fill_colour.blue / 255.0);
88   cairo_fill (cr);
89   cairo_restore (cr);
90   cairo_stroke (cr);
91
92   /* Draw the median */
93   cairo_save (cr);
94   cairo_set_line_width (cr, cairo_get_line_width (cr) * 5);
95   cairo_move_to (cr,
96                  box_left,
97                  geom->data_bottom + (hinge[1] - geom->y_min) * geom->ordinate_scale);
98   cairo_line_to (cr,
99                  box_right,
100                  geom->data_bottom + (hinge[1] - geom->y_min) * geom->ordinate_scale);
101   cairo_stroke (cr);
102   cairo_restore (cr);
103
104   /* Draw the bottom whisker */
105   cairo_move_to (cr, box_left, bottom_whisker);
106   cairo_line_to (cr, box_right, bottom_whisker);
107   cairo_stroke (cr);
108
109   /* Draw top whisker */
110   cairo_move_to (cr, box_left, top_whisker);
111   cairo_line_to (cr, box_right, top_whisker);
112   cairo_stroke (cr);
113
114   /* Draw centre line.
115      (bottom half) */
116   cairo_move_to (cr, box_centre, bottom_whisker);
117   cairo_line_to (cr, box_centre, box_bottom);
118   cairo_stroke (cr);
119
120   /* (top half) */
121   cairo_move_to (cr, box_centre, top_whisker);
122   cairo_line_to (cr, box_centre, box_top);
123   cairo_stroke (cr);
124
125   outliers = box_whisker_outliers (bw);
126   for (ll = ll_head (outliers);
127        ll != ll_null (outliers); ll = ll_next (ll))
128     {
129       const struct outlier *outlier = ll_data (ll, struct outlier, ll);
130       draw_case (cr, geom, box_centre, outlier);
131     }
132
133   /* Draw  tick  mark on x axis */
134   draw_tick(cr, geom, TICK_ABSCISSA, box_centre - geom->data_left, "%s", name);
135 }
136
137 static void
138 boxplot_draw_yscale (cairo_t *cr, struct xrchart_geometry *geom,
139                      double y_max, double y_min)
140 {
141   double y_tick;
142   double d;
143
144   geom->y_max = y_max;
145   geom->y_min = y_min;
146
147   y_tick = chart_rounded_tick (fabs (geom->y_max - geom->y_min) / 5.0);
148
149   geom->y_min = (ceil (geom->y_min / y_tick) - 1.0) * y_tick;
150
151   geom->y_max = (floor (geom->y_max / y_tick) + 1.0) * y_tick;
152
153   geom->ordinate_scale = (fabs (geom->data_top - geom->data_bottom)
154                           / fabs (geom->y_max - geom->y_min));
155
156   for (d = geom->y_min; d <= geom->y_max; d += y_tick)
157     draw_tick (cr, geom, TICK_ORDINATE,
158                (d - geom->y_min) * geom->ordinate_scale, "%g", d);
159 }
160
161 void
162 xrchart_draw_boxplot (const struct chart_item *chart_item, cairo_t *cr,
163                       struct xrchart_geometry *geom)
164 {
165   const struct boxplot *boxplot = to_boxplot (chart_item);
166   double box_width;
167   size_t i;
168
169   boxplot_draw_yscale (cr, geom, boxplot->y_max, boxplot->y_min);
170   xrchart_write_title (cr, geom, "%s", chart_item->title);
171
172   box_width = (geom->data_right - geom->data_left) / boxplot->n_boxes / 2.0;
173   for (i = 0; i < boxplot->n_boxes; i++)
174     {
175       const struct boxplot_box *box = &boxplot->boxes[i];
176       const double box_centre = (i * 2 + 1) * box_width + geom->data_left;
177       boxplot_draw_box (cr, geom, box_centre, box_width, box->bw, box->label);
178     }
179 }