X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Foutput%2Fcharts%2Fplot-hist.c;h=334818b4cad9428975f8c2e1623ea567ff57ac5c;hb=cb586666724d5fcbdb658ce471b85484f0a7babe;hp=fbf1925e635e165c3c096d119bace8123fa0d5bc;hpb=74ddfcbaa07193bab2d6cbe218fc6c16dd3cd052;p=pspp-builds.git diff --git a/src/output/charts/plot-hist.c b/src/output/charts/plot-hist.c index fbf1925e..334818b4 100644 --- a/src/output/charts/plot-hist.c +++ b/src/output/charts/plot-hist.c @@ -18,7 +18,6 @@ #include #include -#include #include #include #include @@ -26,8 +25,10 @@ #include #include +#include #include +#include #include #include #include @@ -36,157 +37,187 @@ #include "gettext.h" #define _(msgid) gettext (msgid) +static const struct chart_class histogram_chart_class; + /* Write the legend of the chart */ static void -histogram_write_legend (struct chart *ch, double n, double mean, double stddev) +histogram_write_legend (cairo_t *cr, const struct chart_geometry *geom, + double n, double mean, double stddev) { - char buf[100]; - - if (!ch) - return ; - - pl_savestate_r (ch->lp); + double y = geom->data_bottom; + cairo_save (cr); - sprintf (buf, "N = %.2f", n); - pl_move_r (ch->lp, ch->legend_left, ch->data_bottom); - pl_alabel_r (ch->lp, 0, 'b', buf); + if (n != SYSMIS) + { + char *buf = xasprintf ("N = %.2f", n); + cairo_move_to (cr, geom->legend_left, y); + chart_label (cr, 'l', 'b', geom->font_size, buf); + y += geom->font_size * 1.5; + free (buf); + } - sprintf (buf, "Mean = %.1f", mean); - pl_fmove_r (ch->lp,ch->legend_left,ch->data_bottom + ch->font_size * 1.5); - pl_alabel_r (ch->lp, 0, 'b', buf); + if (mean != SYSMIS) + { + char *buf = xasprintf ("Mean = %.1f", mean); + cairo_move_to (cr,geom->legend_left, y); + chart_label (cr, 'l', 'b', geom->font_size, buf); + y += geom->font_size * 1.5; + free (buf); + } - sprintf (buf, "Std. Dev = %.2f", stddev); - pl_fmove_r (ch->lp, ch->legend_left, ch->data_bottom + ch->font_size * 1.5 * 2); - pl_alabel_r (ch->lp, 0, 'b', buf); + if (stddev != SYSMIS) + { + char *buf = xasprintf ("Std. Dev = %.2f", stddev); + cairo_move_to (cr, geom->legend_left, y); + chart_label (cr, 'l', 'b', geom->font_size, buf); + free (buf); + } - pl_restorestate_r (ch->lp); + cairo_restore (cr); } -static void hist_draw_bar (struct chart *ch, const struct histogram *hist, int bar); - - static void -hist_draw_bar (struct chart *ch, const struct histogram *hist, int bar) +hist_draw_bar (cairo_t *cr, const struct chart_geometry *geom, + const gsl_histogram *h, int bar) { - if (!ch) - return ; - - { - double upper; - double lower; - double height; - - const size_t bins = gsl_histogram_bins (hist->gsl_hist); - const double x_pos = (ch->data_right - ch->data_left) * bar / (double) bins ; - const double width = (ch->data_right - ch->data_left) / (double) bins ; - - assert ( 0 == gsl_histogram_get_range (hist->gsl_hist, bar, &lower, &upper)); - - assert ( upper >= lower); - - height = gsl_histogram_get (hist->gsl_hist, bar) * - (ch->data_top - ch->data_bottom) / gsl_histogram_max_val (hist->gsl_hist); - - pl_savestate_r (ch->lp); - pl_move_r (ch->lp,ch->data_left, ch->data_bottom); - pl_fillcolorname_r (ch->lp, ch->fill_colour); - pl_filltype_r (ch->lp,1); - - - pl_fboxrel_r (ch->lp, - x_pos, 0, - x_pos + width, height); - - pl_restorestate_r (ch->lp); - - draw_tick (ch, TICK_ABSCISSA, - x_pos + width / 2.0, "%g", (upper + lower) / 2.0); - } + double upper; + double lower; + double height; + + const size_t bins = gsl_histogram_bins (h); + const double x_pos = (geom->data_right - geom->data_left) * bar / (double) bins ; + const double width = (geom->data_right - geom->data_left) / (double) bins ; + + assert ( 0 == gsl_histogram_get_range (h, bar, &lower, &upper)); + + assert ( upper >= lower); + + height = gsl_histogram_get (h, bar) * + (geom->data_top - geom->data_bottom) / gsl_histogram_max_val (h); + + cairo_rectangle (cr, geom->data_left + x_pos, geom->data_bottom, + width, height); + cairo_save (cr); + cairo_set_source_rgb (cr, + geom->fill_colour.red / 255.0, + geom->fill_colour.green / 255.0, + geom->fill_colour.blue / 255.0); + cairo_fill_preserve (cr); + cairo_restore (cr); + cairo_stroke (cr); + + draw_tick (cr, geom, TICK_ABSCISSA, + x_pos + width / 2.0, "%g", (upper + lower) / 2.0); } - - -void -histogram_plot (const struct histogram *hist, - const char *label, - const struct moments1 *m) +struct histogram_chart + { + struct chart chart; + gsl_histogram *gsl_hist; + char *label; + double n; + double mean; + double stddev; + bool show_normal; + }; + +/* Plots a histogram of the data in HIST with the given LABEL. + Labels the histogram with each of N, MEAN, and STDDEV that is + not SYSMIS. If all three are not SYSMIS and SHOW_NORMAL is + true, also draws a normal curve on the histogram. */ +struct chart * +histogram_chart_create (const struct histogram *hist, const char *label, + double n, double mean, double stddev, + bool show_normal) { - double mean, var, n; - - moments1_calculate (m, &n, &mean, &var, NULL, NULL); - - histogram_plot_n (hist, label, n, mean, sqrt(var), m); + struct histogram_chart *h; + + h = xmalloc (sizeof *h); + chart_init (&h->chart, &histogram_chart_class); + h->gsl_hist = hist->gsl_hist ? gsl_histogram_clone (hist->gsl_hist) : NULL; + h->label = xstrdup (label); + h->n = n; + h->mean = mean; + h->stddev = stddev; + h->show_normal = show_normal; + return &h->chart; } - -/* This function is deprecated. Don't use it in new code */ -void -histogram_plot_n (const struct histogram *hist, - const char *label, - double n, double mean, double stddev, - bool show_normal) +static void +histogram_chart_draw (const struct chart *chart, cairo_t *cr, + struct chart_geometry *geom) { + struct histogram_chart *h = UP_CAST (chart, struct histogram_chart, chart); int i; int bins; - struct chart *ch = chart_create (); - - chart_write_title (ch, _("HISTOGRAM")); + chart_write_title (cr, geom, _("HISTOGRAM")); - chart_write_ylabel (ch, _("Frequency")); - chart_write_xlabel (ch, label); + chart_write_ylabel (cr, geom, _("Frequency")); + chart_write_xlabel (cr, geom, h->label); - if ( ! hist ) /* If this happens, probably all values are SYSMIS */ + if (h->gsl_hist == NULL) { - chart_submit (ch); + /* Probably all values are SYSMIS. */ return; } - else - { - bins = gsl_histogram_bins (hist->gsl_hist); - } - chart_write_yscale (ch, 0, gsl_histogram_max_val (hist->gsl_hist), 5); + bins = gsl_histogram_bins (h->gsl_hist); + + chart_write_yscale (cr, geom, 0, gsl_histogram_max_val (h->gsl_hist), 5); - for ( i = 0 ; i < bins ; ++i ) - hist_draw_bar (ch, hist, i); + for (i = 0; i < bins; i++) + hist_draw_bar (cr, geom, h->gsl_hist, i); - histogram_write_legend (ch, n, mean, stddev); + histogram_write_legend (cr, geom, h->n, h->mean, h->stddev); - if (show_normal) + if (h->show_normal + && h->n != SYSMIS && h->mean != SYSMIS && h->stddev != SYSMIS) { /* Draw the normal curve */ + double d; + double x_min, x_max, not_used; + double abscissa_scale; + double ordinate_scale; + double range; - double d ; - double x_min, x_max, not_used ; - double abscissa_scale ; - double ordinate_scale ; - double range ; - - gsl_histogram_get_range (hist->gsl_hist, 0, &x_min, ¬_used); + gsl_histogram_get_range (h->gsl_hist, 0, &x_min, ¬_used); range = not_used - x_min; - gsl_histogram_get_range (hist->gsl_hist, bins - 1, ¬_used, &x_max); + gsl_histogram_get_range (h->gsl_hist, bins - 1, ¬_used, &x_max); - abscissa_scale = (ch->data_right - ch->data_left) / (x_max - x_min); - ordinate_scale = (ch->data_top - ch->data_bottom) / - gsl_histogram_max_val (hist->gsl_hist) ; + abscissa_scale = (geom->data_right - geom->data_left) / (x_max - x_min); + ordinate_scale = (geom->data_top - geom->data_bottom) / + gsl_histogram_max_val (h->gsl_hist); - pl_move_r (ch->lp, ch->data_left, ch->data_bottom); - for ( d = ch->data_left; - d <= ch->data_right ; - d += (ch->data_right - ch->data_left) / 100.0) + cairo_move_to (cr, geom->data_left, geom->data_bottom); + for (d = geom->data_left; + d <= geom->data_right; + d += (geom->data_right - geom->data_left) / 100.0) { - const double x = (d - ch->data_left) / abscissa_scale + x_min ; - const double y = n * range * - gsl_ran_gaussian_pdf (x - mean, stddev); + const double x = (d - geom->data_left) / abscissa_scale + x_min; + const double y = h->n * range * + gsl_ran_gaussian_pdf (x - h->mean, h->stddev); - pl_fcont_r (ch->lp, d, ch->data_bottom + y * ordinate_scale); + cairo_line_to (cr, d, geom->data_bottom + y * ordinate_scale); } - pl_endpath_r (ch->lp); + cairo_stroke (cr); } - - chart_submit (ch); } +static void +histogram_chart_destroy (struct chart *chart) +{ + struct histogram_chart *h = UP_CAST (chart, struct histogram_chart, chart); + if (h->gsl_hist != NULL) + gsl_histogram_free (h->gsl_hist); + free (h->label); + free (h); +} + +static const struct chart_class histogram_chart_class = + { + histogram_chart_draw, + histogram_chart_destroy + };