src/output/charts/piechart.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2004 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
  18 #include <config.h>
  19
  20 #include <float.h>
  21 #include <assert.h>
  22 #include <gsl/gsl_math.h>
  23 #include <math.h>
  24 #include <stdio.h>
  25
  26
  27 #include <output/charts/piechart.h>
  28 #include <output/charts/plot-chart.h>
  29
  30 #include <output/chart.h>
  31 #include <libpspp/str.h>
  32 #include <data/value-labels.h>
  33
  34 #include "minmax.h"
  35
  36 /* Draw a single slice of the pie */
  37 static void
  38 draw_segment(struct chart *ch,
  39              double centre_x, double centre_y,
  40              double radius,
  41              double start_angle, double segment_angle,
  42              const char *colour) ;
  43
  44
  45
  46 /* Draw a piechart */
  47 void
  48 piechart_plot(const char *title, const struct slice *slices, int n_slices)
  49 {
  50   int i;
  51   double total_magnitude=0;
  52
  53   struct chart *ch = chart_create();
  54
  55   const double left_label = ch->data_left +
  56     (ch->data_right - ch->data_left)/10.0;
  57
  58   const double right_label = ch->data_right -
  59     (ch->data_right - ch->data_left)/10.0;
  60
  61   const double centre_x = (ch->data_right + ch->data_left ) / 2.0 ;
  62   const double centre_y = (ch->data_top + ch->data_bottom ) / 2.0 ;
  63
  64   const double radius = MIN(
  65                             5.0 / 12.0 * (ch->data_top - ch->data_bottom),
  66                             1.0 / 4.0 * (ch->data_right - ch->data_left)
  67                             );
  68
  69
  70   chart_write_title(ch, title);
  71
  72   for (i = 0 ; i < n_slices ; ++i )
  73     total_magnitude += slices[i].magnitude;
  74
  75   for (i = 0 ; i < n_slices ; ++i )
  76     {
  77       static double angle=0.0;
  78
  79       const double segment_angle =
  80         slices[i].magnitude / total_magnitude * 2 * M_PI ;
  81
  82       const double label_x = centre_x -
  83         radius * sin(angle + segment_angle/2.0);
  84
  85       const double label_y = centre_y +
  86         radius * cos(angle + segment_angle/2.0);
  87
  88       /* Fill the segment */
  89       draw_segment(ch,
  90                    centre_x, centre_y, radius,
  91                    angle, segment_angle,
  92                    data_colour[i % N_CHART_COLOURS]);
  93
  94       /* Now add the labels */
  95       if ( label_x < centre_x )
  96         {
  97           pl_line_r(ch->lp, label_x, label_y,
  98                     left_label, label_y );
  99           pl_moverel_r(ch->lp,0,5);
 100           pl_alabel_r (ch->lp, 0, 0, ds_cstr (&slices[i].label));
 101         }
 102       else
 103         {
 104           pl_line_r(ch->lp,
 105                     label_x, label_y,
 106                     right_label, label_y
 107                     );
 108           pl_moverel_r(ch->lp,0,5);
 109           pl_alabel_r (ch->lp, 'r', 0, ds_cstr (&slices[i].label));
 110         }
 111
 112       angle += segment_angle;
 113
 114     }
 115
 116   /* Draw an outline to the pie */
 117   pl_filltype_r(ch->lp,0);
 118   pl_fcircle_r (ch->lp, centre_x, centre_y, radius);
 119
 120   chart_submit(ch);
 121 }
 122
 123 static void
 124 fill_segment(struct chart *ch,
 125              double x0, double y0,
 126              double radius,
 127              double start_angle, double segment_angle) ;
 128
 129
 130 /* Fill a segment with the current fill colour */
 131 static void
 132 fill_segment(struct chart *ch,
 133              double x0, double y0,
 134              double radius,
 135              double start_angle, double segment_angle)
 136 {
 137
 138   const double start_x  = x0 - radius * sin(start_angle);
 139   const double start_y  = y0 + radius * cos(start_angle);
 140
 141   const double stop_x   =
 142     x0 - radius * sin(start_angle + segment_angle);
 143
 144   const double stop_y   =
 145     y0 + radius * cos(start_angle + segment_angle);
 146
 147   assert(segment_angle <= 2 * M_PI);
 148   assert(segment_angle >= 0);
 149
 150   if ( segment_angle > M_PI )
 151     {
 152       /* Then we must draw it in two halves */
 153       fill_segment(ch, x0, y0, radius, start_angle, segment_angle / 2.0 );
 154       fill_segment(ch, x0, y0, radius, start_angle + segment_angle / 2.0,
 155                    segment_angle / 2.0 );
 156     }
 157   else
 158     {
 159       pl_move_r(ch->lp, x0, y0);
 160
 161       pl_cont_r(ch->lp, stop_x, stop_y);
 162       pl_cont_r(ch->lp, start_x, start_y);
 163
 164       pl_arc_r(ch->lp,
 165                x0, y0,
 166                stop_x, stop_y,
 167                start_x, start_y
 168                );
 169
 170       pl_endpath_r(ch->lp);
 171     }
 172 }
 173
 174
 175
 176 /* Draw a single slice of the pie */
 177 static void
 178 draw_segment(struct chart *ch,
 179              double x0, double y0,
 180              double radius,
 181              double start_angle, double segment_angle,
 182              const char *colour)
 183 {
 184   const double start_x  = x0 - radius * sin(start_angle);
 185   const double start_y  = y0 + radius * cos(start_angle);
 186
 187   pl_savestate_r(ch->lp);
 188
 189   pl_savestate_r(ch->lp);
 190   pl_colorname_r(ch->lp, colour);
 191
 192   pl_pentype_r(ch->lp,1);
 193   pl_filltype_r(ch->lp,1);
 194
 195   fill_segment(ch, x0, y0, radius, start_angle, segment_angle);
 196   pl_restorestate_r(ch->lp);
 197
 198   /* Draw line dividing segments */
 199   pl_pentype_r(ch->lp, 1);
 200   pl_fline_r(ch->lp, x0, y0, start_x, start_y);
 201
 202
 203   pl_restorestate_r(ch->lp);
 204 }
 205