X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Foutput%2Fcharts%2Fplot-hist.c;h=aed72e6f485fabe1bb8440a5c5b8b3bf600e01a1;hb=507ebaea36737618ef8265a60cd3e9005d9f3457;hp=88ad0899199d939dd5411cfb8fb2656368e395a4;hpb=005ac6d163e0e535a06e9bfb223f2f75f9320c9a;p=pspp diff --git a/src/output/charts/plot-hist.c b/src/output/charts/plot-hist.c index 88ad089919..aed72e6f48 100644 --- a/src/output/charts/plot-hist.c +++ b/src/output/charts/plot-hist.c @@ -1,10 +1,10 @@ /* PSPP - a program for statistical analysis. - Copyright (C) 2004, 2009 Free Software Foundation, Inc. + Copyright (C) 2004, 2009, 2011 Free Software Foundation, Inc. This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or - (at your option) any later version. + (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of @@ -18,163 +18,50 @@ #include #include -#include #include -#include #include -#include -#include -#include - -#include -#include -#include +#include "libpspp/cast.h" +#include "math/histogram.h" +#include "math/moments.h" +#include "output/chart-provider.h" +#include "output/charts/plot-hist.h" #include "gettext.h" #define _(msgid) gettext (msgid) -/* Write the legend of the chart */ -void -histogram_write_legend(struct chart *ch, const struct normal_curve *norm) +/* 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 gsl_histogram *hist, const char *label, + double n, double mean, double stddev, + bool show_normal) { - char buf[100]; - if ( !ch ) - return ; - - pl_savestate_r(ch->lp); - - sprintf(buf,"N = %.2f",norm->N); - pl_move_r(ch->lp, ch->legend_left, ch->data_bottom); - pl_alabel_r(ch->lp,0,'b',buf); - - sprintf(buf,"Mean = %.1f",norm->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); - - sprintf(buf,"Std. Dev = %.2f",norm->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); - - pl_restorestate_r(ch->lp); + struct histogram_chart *h; + + h = xmalloc (sizeof *h); + chart_init (&h->chart, &histogram_chart_class, label); + h->gsl_hist = hist != NULL ? gsl_histogram_clone (hist) : NULL; + h->n = n; + h->mean = mean; + h->stddev = stddev; + h->show_normal = show_normal; + return &h->chart; } -static void hist_draw_bar(struct chart *ch, const gsl_histogram *hist, int bar); - - static void -hist_draw_bar(struct chart *ch, const gsl_histogram *hist, int bar) +histogram_chart_destroy (struct chart *chart) { - if ( !ch ) - return ; - - - { - double upper; - double lower; - double height; - - const size_t bins = gsl_histogram_bins(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, bar, &lower, &upper)); - - assert( upper >= lower); - - height = gsl_histogram_get(hist, bar) * - (ch->data_top - ch->data_bottom) / gsl_histogram_max_val(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); - } + struct histogram_chart *h = UP_CAST (chart, struct histogram_chart, + chart); + if (h->gsl_hist != NULL) + gsl_histogram_free (h->gsl_hist); + free (h); } - - - -void -histogram_plot(const gsl_histogram *hist, - const char *factorname, - const struct normal_curve *norm, short show_normal) -{ - int i; - int bins; - - struct chart *ch; - - ch = chart_create(); - if (ch == NULL) - return; - - chart_write_title(ch, _("HISTOGRAM")); - - chart_write_ylabel(ch, _("Frequency")); - chart_write_xlabel(ch, factorname); - - if ( ! hist ) /* If this happens, probably all values are SYSMIS */ - { - chart_submit(ch); - return ; - } - else - { - bins = gsl_histogram_bins(hist); - } - - chart_write_yscale(ch, 0, gsl_histogram_max_val(hist), 5); - - for ( i = 0 ; i < bins ; ++i ) - hist_draw_bar(ch, hist, i); - - histogram_write_legend(ch, norm); - - if ( show_normal ) +const struct chart_class histogram_chart_class = { - /* Draw the normal curve */ - - double d ; - double x_min, x_max, not_used ; - double abscissa_scale ; - double ordinate_scale ; - double range ; - - gsl_histogram_get_range(hist, 0, &x_min, ¬_used); - range = not_used - x_min; - gsl_histogram_get_range(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) ; - - 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) - { - const double x = (d - ch->data_left) / abscissa_scale + x_min ; - const double y = norm->N * range * - gsl_ran_gaussian_pdf(x - norm->mean, norm->stddev); - - pl_fcont_r(ch->lp, d, ch->data_bottom + y * ordinate_scale); - - } - pl_endpath_r(ch->lp); - - } - chart_submit(ch); -} - + histogram_chart_destroy + };