pintos-os.org Git - pspp/blob - src/output/charts/plot-hist-cairo.c

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 2009, 2011 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/charts/plot-hist.h"
  20
  21 #include <gsl/gsl_randist.h>
  22
  23 #include "data/val-type.h"
  24 #include "output/cairo-chart.h"
  25
  26 #include "gl/xvasprintf.h"
  27
  28 #include "gettext.h"
  29 #define _(msgid) gettext (msgid)
  30
  31 /* Write the legend of the chart */
  32 static void
  33 histogram_write_legend (cairo_t *cr, const struct xrchart_geometry *geom,
  34                         double n, double mean, double stddev)
  35 {
  36   double y = geom->data_bottom;
  37   cairo_save (cr);
  38
  39   if (n != SYSMIS)
  40     {
  41       char *buf = xasprintf ("N = %.2f", n);
  42       cairo_move_to (cr, geom->legend_left, y);
  43       xrchart_label (cr, 'l', 'b', geom->font_size, buf);
  44       y += geom->font_size * 1.5;
  45       free (buf);
  46     }
  47
  48   if (mean != SYSMIS)
  49     {
  50       char *buf = xasprintf ("Mean = %.1f", mean);
  51       cairo_move_to (cr,geom->legend_left, y);
  52       xrchart_label (cr, 'l', 'b', geom->font_size, buf);
  53       y += geom->font_size * 1.5;
  54       free (buf);
  55     }
  56
  57   if (stddev != SYSMIS)
  58     {
  59       char *buf = xasprintf ("Std. Dev = %.2f", stddev);
  60       cairo_move_to (cr, geom->legend_left, y);
  61       xrchart_label (cr, 'l', 'b', geom->font_size, buf);
  62       free (buf);
  63     }
  64
  65   cairo_restore (cr);
  66 }
  67
  68 static void
  69 hist_draw_bar (cairo_t *cr, const struct xrchart_geometry *geom,
  70                const gsl_histogram *h, int bar)
  71 {
  72   double upper;
  73   double lower;
  74   double height;
  75
  76   const size_t bins = gsl_histogram_bins (h);
  77   const double x_pos = (geom->data_right - geom->data_left) * bar / (double) bins ;
  78   const double width = (geom->data_right - geom->data_left) / (double) bins ;
  79
  80   assert ( 0 == gsl_histogram_get_range (h, bar, &lower, &upper));
  81
  82   assert ( upper >= lower);
  83
  84   height = gsl_histogram_get (h, bar) /
  85     (geom->y_max - geom->y_min) *
  86     (geom->data_top - geom->data_bottom);
  87
  88   cairo_rectangle (cr, geom->data_left + x_pos, geom->data_bottom,
  89                    width, height);
  90   cairo_save (cr);
  91   cairo_set_source_rgb (cr,
  92                         geom->fill_colour.red / 255.0,
  93                         geom->fill_colour.green / 255.0,
  94                         geom->fill_colour.blue / 255.0);
  95   cairo_fill_preserve (cr);
  96   cairo_restore (cr);
  97   cairo_stroke (cr);
  98
  99   draw_tick (cr, geom, TICK_ABSCISSA,
 100              x_pos + width / 2.0, "%g", (upper + lower) / 2.0);
 101 }
 102
 103 void
 104 xrchart_draw_histogram (const struct chart_item *chart_item, cairo_t *cr,
 105                         struct xrchart_geometry *geom)
 106 {
 107   struct histogram_chart *h = to_histogram_chart (chart_item);
 108   int i;
 109   int bins;
 110
 111   xrchart_write_title (cr, geom, _("HISTOGRAM"));
 112
 113   xrchart_write_ylabel (cr, geom, _("Frequency"));
 114   xrchart_write_xlabel (cr, geom, chart_item_get_title (chart_item));
 115
 116   if (h->gsl_hist == NULL)
 117     {
 118       /* Probably all values are SYSMIS. */
 119       return;
 120     }
 121
 122   bins = gsl_histogram_bins (h->gsl_hist);
 123
 124   xrchart_write_yscale (cr, geom, 0, gsl_histogram_max_val (h->gsl_hist), 5);
 125
 126   for (i = 0; i < bins; i++)
 127     hist_draw_bar (cr, geom, h->gsl_hist, i);
 128
 129   histogram_write_legend (cr, geom, h->n, h->mean, h->stddev);
 130
 131   if (h->show_normal
 132       && h->n != SYSMIS && h->mean != SYSMIS && h->stddev != SYSMIS)
 133     {
 134       /* Draw the normal curve */
 135       double d;
 136       double x_min, x_max, not_used;
 137       double abscissa_scale;
 138       double ordinate_scale;
 139       double range;
 140
 141       gsl_histogram_get_range (h->gsl_hist, 0, &x_min, &not_used);
 142       range = not_used - x_min;
 143       gsl_histogram_get_range (h->gsl_hist, bins - 1, &not_used, &x_max);
 144
 145       abscissa_scale = (geom->data_right - geom->data_left) / (x_max - x_min);
 146       ordinate_scale = (geom->data_top - geom->data_bottom) /
 147         gsl_histogram_max_val (h->gsl_hist);
 148
 149       cairo_move_to (cr, geom->data_left, geom->data_bottom);
 150       for (d = geom->data_left;
 151            d <= geom->data_right;
 152            d += (geom->data_right - geom->data_left) / 100.0)
 153         {
 154           const double x = (d - geom->data_left) / abscissa_scale + x_min;
 155           const double y = h->n * range *
 156             gsl_ran_gaussian_pdf (x - h->mean, h->stddev);
 157
 158           cairo_line_to (cr, d, geom->data_bottom  + y * ordinate_scale);
 159
 160         }
 161       cairo_stroke (cr);
 162     }
 163 }