2 PSPP - a program for statistical analysis.
3 Copyright (C) 1997-9, 2000, 2007, 2009, 2010, 2011, 2014 Free Software Foundation, Inc.
5 This program is free software: you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation, either version 3 of the License, or
8 (at your option) any later version.
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include <gsl/gsl_histogram.h>
24 #include "data/case.h"
25 #include "data/casegrouper.h"
26 #include "data/casereader.h"
27 #include "data/dataset.h"
28 #include "data/dictionary.h"
29 #include "data/format.h"
30 #include "data/missing-values.h"
31 #include "data/settings.h"
32 #include "data/value-labels.h"
33 #include "data/variable.h"
35 #include "language/dictionary/split-file.h"
37 #include "language/command.h"
38 #include "language/lexer/lexer.h"
39 #include "language/lexer/variable-parser.h"
40 #include "language/stats/freq.h"
42 #include "libpspp/array.h"
43 #include "libpspp/bit-vector.h"
44 #include "libpspp/compiler.h"
45 #include "libpspp/hmap.h"
46 #include "libpspp/message.h"
47 #include "libpspp/misc.h"
48 #include "libpspp/pool.h"
50 #include "math/histogram.h"
51 #include "math/moments.h"
52 #include "math/chart-geometry.h"
55 #include "output/chart-item.h"
56 #include "output/charts/piechart.h"
57 #include "output/charts/plot-hist.h"
58 #include "output/tab.h"
60 #include "gl/minmax.h"
61 #include "gl/xalloc.h"
64 #define _(msgid) gettext (msgid)
65 #define N_(msgid) msgid
67 /* Percentiles to calculate. */
71 double p; /* the %ile to be calculated */
72 double value; /* the %ile's value */
73 bool show; /* True to show this percentile in the statistics box. */
77 ptile_3way (const void *_p1, const void *_p2)
79 const struct percentile *p1 = _p1;
80 const struct percentile *p2 = _p2;
85 return (p1->p > p2->p);
109 /* Array indices for STATISTICS subcommand. */
129 /* Description of statistics. */
130 static const char *st_name[FRQ_ST_count] =
150 struct hmap data; /* Hash table for accumulating counts. */
151 struct freq *valid; /* Valid freqs. */
152 int n_valid; /* Number of total freqs. */
153 const struct dictionary *dict; /* Source of entries in the table. */
155 struct freq *missing; /* Missing freqs. */
156 int n_missing; /* Number of missing freqs. */
159 double total_cases; /* Sum of weights of all cases. */
160 double valid_cases; /* Sum of weights of valid cases. */
165 double x_min; /* X axis minimum value. */
166 double x_max; /* X axis maximum value. */
167 int y_scale; /* Y axis scale: FRQ_FREQ or FRQ_PERCENT. */
169 /* Histograms only. */
170 double y_max; /* Y axis maximum value. */
171 bool draw_normal; /* Whether to draw normal curve. */
173 /* Pie charts only. */
174 bool include_missing; /* Whether to include missing values. */
177 /* Per-variable frequency data. */
180 const struct variable *var;
182 /* Freqency table. */
183 struct freq_tab tab; /* Frequencies table to use. */
186 int n_groups; /* Number of groups. */
187 double *groups; /* Groups. */
190 double stat[FRQ_ST_count];
192 /* Variable attributes. */
200 struct var_freqs *vars;
203 /* Percentiles to calculate and possibly display. */
204 struct percentile *percentiles;
205 int n_percentiles, n_show_percentiles;
207 /* Frequency table display. */
208 int max_categories; /* Maximum categories to show. */
209 int sort; /* FRQ_AVALUE or FRQ_DVALUE
210 or FRQ_AFREQ or FRQ_DFREQ. */
212 /* Statistics; number of statistics. */
216 /* Histogram and pie chart settings. */
217 struct frq_chart *hist, *pie;
221 struct freq_compare_aux
227 bool ascending_value;
230 static void calc_stats (const struct var_freqs *vf, double d[FRQ_ST_count]);
232 static void do_piechart(const struct frq_chart *pie,
233 const struct variable *var,
234 const struct freq_tab *frq_tab);
236 static void dump_statistics (const struct frq_proc *frq,
237 const struct var_freqs *vf,
238 const struct variable *wv);
241 compare_freq (const void *a_, const void *b_, const void *aux_)
243 const struct freq_compare_aux *aux = aux_;
244 const struct freq *a = a_;
245 const struct freq *b = b_;
247 if (aux->by_freq && a->count != b->count)
249 int cmp = a->count > b->count ? 1 : -1;
250 return aux->ascending_freq ? cmp : -cmp;
254 int cmp = value_compare_3way (&a->value, &b->value, aux->width);
255 return aux->ascending_value ? cmp : -cmp;
259 /* Create a gsl_histogram from a freq_tab */
260 static struct histogram *
261 freq_tab_to_hist (const struct frq_proc *frq, const struct freq_tab *ft,
262 const struct variable *var);
265 /* Displays a full frequency table for variable V. */
267 dump_freq_table (const struct var_freqs *vf, const struct variable *wv)
269 const struct fmt_spec *wfmt = wv ? var_get_print_format (wv) : &F_8_0;
270 const struct freq_tab *ft = &vf->tab;
275 double cum_total = 0.0;
276 double cum_freq = 0.0;
278 static const char *headings[] = {
287 n_categories = ft->n_valid + ft->n_missing;
288 t = tab_create (6, n_categories + 2);
289 tab_set_format (t, RC_WEIGHT, wfmt);
290 tab_headers (t, 0, 0, 1, 0);
292 for (x = 0; x < 6; x++)
293 tab_text (t, x, 0, TAB_CENTER | TAT_TITLE, gettext (headings[x]));
296 for (f = ft->valid; f < ft->missing; f++)
299 double percent, valid_percent;
301 cum_freq += f->count;
303 percent = f->count / ft->total_cases * 100.0;
304 valid_percent = f->count / ft->valid_cases * 100.0;
305 cum_total += valid_percent;
307 label = var_lookup_value_label (vf->var, &f->value);
309 tab_text (t, 0, r, TAB_LEFT, label);
311 tab_value (t, 1, r, TAB_NONE, &f->value, vf->var, NULL);
312 tab_double (t, 2, r, TAB_NONE, f->count, NULL, RC_WEIGHT);
313 tab_double (t, 3, r, TAB_NONE, percent, NULL, RC_OTHER);
314 tab_double (t, 4, r, TAB_NONE, valid_percent, NULL, RC_OTHER);
315 tab_double (t, 5, r, TAB_NONE, cum_total, NULL, RC_OTHER);
318 for (; f < &ft->valid[n_categories]; f++)
322 cum_freq += f->count;
324 label = var_lookup_value_label (vf->var, &f->value);
326 tab_text (t, 0, r, TAB_LEFT, label);
328 tab_value (t, 1, r, TAB_NONE, &f->value, vf->var, NULL);
329 tab_double (t, 2, r, TAB_NONE, f->count, NULL, RC_WEIGHT);
330 tab_double (t, 3, r, TAB_NONE,
331 f->count / ft->total_cases * 100.0, NULL, RC_OTHER);
332 tab_text (t, 4, r, TAB_NONE, _("Missing"));
336 tab_box (t, TAL_1, TAL_1, -1, TAL_1, 0, 0, 5, r);
337 tab_hline (t, TAL_2, 0, 5, 1);
338 tab_hline (t, TAL_2, 0, 5, r);
339 tab_joint_text (t, 0, r, 1, r, TAB_RIGHT | TAT_TITLE, _("Total"));
340 tab_vline (t, TAL_0, 1, r, r);
341 tab_double (t, 2, r, TAB_NONE, cum_freq, NULL, RC_WEIGHT);
342 tab_double (t, 3, r, TAB_NONE, 100.0, &F_5_1, RC_OTHER);
343 tab_double (t, 4, r, TAB_NONE, 100.0, &F_5_1, RC_OTHER);
345 tab_title (t, "%s", var_to_string (vf->var));
349 /* Statistical display. */
352 calc_percentile (double p, double valid_cases, double x1, double x2)
356 s = (settings_get_algorithm () != COMPATIBLE
357 ? modf ((valid_cases - 1) * p, &dummy)
358 : modf ((valid_cases + 1) * p - 1, &dummy));
360 return x1 + (x2 - x1) * s;
363 /* Calculates all of the percentiles for VF within FRQ. */
365 calc_percentiles (const struct frq_proc *frq, const struct var_freqs *vf)
367 const struct freq_tab *ft = &vf->tab;
368 double W = ft->valid_cases;
369 const struct freq *f;
373 assert (ft->n_valid > 0);
377 for (f = ft->valid; f < ft->missing; f++)
380 for (; percentile_idx < frq->n_percentiles; percentile_idx++)
382 struct percentile *pc = &frq->percentiles[percentile_idx];
385 tp = (settings_get_algorithm () == ENHANCED
387 : (W + 1) * pc->p - 1);
392 if (tp + 1 < rank || f + 1 >= ft->missing)
393 pc->value = f->value.f;
395 pc->value = calc_percentile (pc->p, W, f->value.f, f[1].value.f);
398 for (; percentile_idx < frq->n_percentiles; percentile_idx++)
400 struct percentile *pc = &frq->percentiles[percentile_idx];
401 pc->value = ft->valid[ft->n_valid - 1].value.f;
405 /* Returns true iff the value in struct freq F is non-missing
408 not_missing (const void *f_, const void *v_)
410 const struct freq *f = f_;
411 const struct variable *v = v_;
413 return !var_is_value_missing (v, &f->value, MV_ANY);
417 /* Summarizes the frequency table data for variable V. */
419 postprocess_freq_tab (const struct frq_proc *frq, struct var_freqs *vf)
421 struct freq_tab *ft = &vf->tab;
422 struct freq_compare_aux aux;
424 struct freq *freqs, *f;
427 /* Extract data from hash table. */
428 count = hmap_count (&ft->data);
429 freqs = freq_hmap_extract (&ft->data);
431 /* Put data into ft. */
433 ft->n_valid = partition (freqs, count, sizeof *freqs, not_missing, vf->var);
434 ft->missing = freqs + ft->n_valid;
435 ft->n_missing = count - ft->n_valid;
438 aux.by_freq = frq->sort == FRQ_AFREQ || frq->sort == FRQ_DFREQ;
439 aux.ascending_freq = frq->sort != FRQ_DFREQ;
440 aux.width = vf->width;
441 aux.ascending_value = frq->sort != FRQ_DVALUE;
442 sort (ft->valid, ft->n_valid, sizeof *ft->valid, compare_freq, &aux);
443 sort (ft->missing, ft->n_missing, sizeof *ft->missing, compare_freq, &aux);
445 /* Summary statistics. */
446 ft->valid_cases = 0.0;
447 for(i = 0 ; i < ft->n_valid ; ++i )
450 ft->valid_cases += f->count;
454 ft->total_cases = ft->valid_cases ;
455 for(i = 0 ; i < ft->n_missing ; ++i )
458 ft->total_cases += f->count;
463 /* Frees the frequency table for variable V. */
465 cleanup_freq_tab (struct var_freqs *vf)
467 free (vf->tab.valid);
468 freq_hmap_destroy (&vf->tab.data, vf->width);
471 /* Add data from case C to the frequency table. */
473 calc (struct frq_proc *frq, const struct ccase *c, const struct dataset *ds)
475 double weight = dict_get_case_weight (dataset_dict (ds), c, NULL);
478 for (i = 0; i < frq->n_vars; i++)
480 struct var_freqs *vf = &frq->vars[i];
481 const union value *value = case_data (c, vf->var);
482 size_t hash = value_hash (value, vf->width, 0);
485 f = freq_hmap_search (&vf->tab.data, value, vf->width, hash);
487 f = freq_hmap_insert (&vf->tab.data, value, vf->width, hash);
493 /* Prepares each variable that is the target of FREQUENCIES by setting
494 up its hash table. */
496 precalc (struct frq_proc *frq, struct casereader *input, struct dataset *ds)
501 c = casereader_peek (input, 0);
504 output_split_file_values (ds, c);
508 for (i = 0; i < frq->n_vars; i++)
509 hmap_init (&frq->vars[i].tab.data);
512 /* Finishes up with the variables after frequencies have been
513 calculated. Displays statistics, percentiles, ... */
515 postcalc (struct frq_proc *frq, const struct dataset *ds)
517 const struct dictionary *dict = dataset_dict (ds);
518 const struct variable *wv = dict_get_weight (dict);
521 for (i = 0; i < frq->n_vars; i++)
523 struct var_freqs *vf = &frq->vars[i];
525 postprocess_freq_tab (frq, vf);
527 /* Frequencies tables. */
528 if (vf->tab.n_valid + vf->tab.n_missing <= frq->max_categories)
529 dump_freq_table (vf, wv);
531 calc_percentiles (frq, vf);
535 dump_statistics (frq, vf, wv);
537 if (frq->hist && var_is_numeric (vf->var) && vf->tab.n_valid > 0)
539 double d[FRQ_ST_count];
540 struct histogram *histogram;
544 histogram = freq_tab_to_hist (frq, &vf->tab, vf->var);
548 chart_item_submit (histogram_chart_create (
549 histogram->gsl_hist, var_to_string(vf->var),
553 frq->hist->draw_normal));
555 statistic_destroy (&histogram->parent);
560 do_piechart(frq->pie, vf->var, &vf->tab);
562 cleanup_freq_tab (vf);
567 cmd_frequencies (struct lexer *lexer, struct dataset *ds)
571 const struct variable **vars;
573 bool sbc_barchart = false;
574 bool sbc_piechart = false;
575 bool sbc_histogram = false;
577 double pie_min = -DBL_MAX;
578 double pie_max = DBL_MAX;
579 bool pie_missing = false;
581 double hi_min = -DBL_MAX;
582 double hi_max = DBL_MAX;
583 int hi_scale = FRQ_FREQ;
584 int hi_freq = INT_MIN;
585 int hi_pcnt = INT_MIN;
586 int hi_norm = FRQ_NONORMAL;
588 frq.pool = pool_create ();
589 frq.sort = FRQ_AVALUE;
594 frq.stats = BIT_INDEX (FRQ_ST_MEAN)
595 | BIT_INDEX (FRQ_ST_STDDEV)
596 | BIT_INDEX (FRQ_ST_MINIMUM)
597 | BIT_INDEX (FRQ_ST_MAXIMUM);
601 frq.max_categories = INT_MAX;
603 frq.percentiles = NULL;
604 frq.n_percentiles = 0;
605 frq.n_show_percentiles = 0;
611 /* Accept an optional, completely pointless "/VARIABLES=" */
612 lex_match (lexer, T_SLASH);
613 if (lex_match_id (lexer, "VARIABLES"))
615 if (! lex_force_match (lexer, T_EQUALS) )
619 if (!parse_variables_const (lexer, dataset_dict (ds),
625 frq.vars = xzalloc (frq.n_vars * sizeof (*frq.vars));
626 for (i = 0; i < frq.n_vars; ++i)
628 frq.vars[i].var = vars[i];
629 frq.vars[i].width = var_get_width (vars[i]);
632 while (lex_token (lexer) != T_ENDCMD)
634 lex_match (lexer, T_SLASH);
636 if (lex_match_id (lexer, "STATISTICS"))
638 frq.stats = BIT_INDEX (FRQ_ST_MEAN)
639 | BIT_INDEX (FRQ_ST_STDDEV)
640 | BIT_INDEX (FRQ_ST_MINIMUM)
641 | BIT_INDEX (FRQ_ST_MAXIMUM);
645 if (lex_match (lexer, T_EQUALS))
651 while (lex_token (lexer) != T_ENDCMD
652 && lex_token (lexer) != T_SLASH)
654 if (lex_match_id (lexer, "DEFAULT"))
656 frq.stats = BIT_INDEX (FRQ_ST_MEAN)
657 | BIT_INDEX (FRQ_ST_STDDEV)
658 | BIT_INDEX (FRQ_ST_MINIMUM)
659 | BIT_INDEX (FRQ_ST_MAXIMUM);
663 else if (lex_match_id (lexer, "MEAN"))
665 frq.stats |= BIT_INDEX (FRQ_ST_MEAN);
668 else if (lex_match_id (lexer, "SEMEAN"))
670 frq.stats |= BIT_INDEX (FRQ_ST_SEMEAN);
673 else if (lex_match_id (lexer, "MEDIAN"))
675 frq.stats |= BIT_INDEX (FRQ_ST_MEDIAN);
678 else if (lex_match_id (lexer, "MODE"))
680 frq.stats |= BIT_INDEX (FRQ_ST_MODE);
683 else if (lex_match_id (lexer, "STDDEV"))
685 frq.stats |= BIT_INDEX (FRQ_ST_STDDEV);
688 else if (lex_match_id (lexer, "VARIANCE"))
690 frq.stats |= BIT_INDEX (FRQ_ST_MEAN);
693 else if (lex_match_id (lexer, "KURTOSIS"))
695 frq.stats |= BIT_INDEX (FRQ_ST_KURTOSIS);
698 else if (lex_match_id (lexer, "SKEWNESS"))
700 frq.stats |= BIT_INDEX (FRQ_ST_SKEWNESS);
703 else if (lex_match_id (lexer, "RANGE"))
705 frq.stats |= BIT_INDEX (FRQ_ST_RANGE);
708 else if (lex_match_id (lexer, "MINIMUM"))
710 frq.stats |= BIT_INDEX (FRQ_ST_MINIMUM);
713 else if (lex_match_id (lexer, "MAXIMUM"))
715 frq.stats |= BIT_INDEX (FRQ_ST_MAXIMUM);
718 else if (lex_match_id (lexer, "SUM"))
720 frq.stats |= BIT_INDEX (FRQ_ST_SUM);
723 else if (lex_match_id (lexer, "SESKEWNESS"))
725 frq.stats |= BIT_INDEX (FRQ_ST_SESKEWNESS);
728 else if (lex_match_id (lexer, "SEKURTOSIS"))
730 frq.stats |= BIT_INDEX (FRQ_ST_SEKURTOSIS);
733 else if (lex_match_id (lexer, "NONE"))
738 else if (lex_match (lexer, T_ALL))
741 frq.n_stats = FRQ_ST_count;
745 lex_error (lexer, NULL);
750 else if (lex_match_id (lexer, "PERCENTILES"))
752 lex_match (lexer, T_EQUALS);
753 while (lex_token (lexer) != T_ENDCMD
754 && lex_token (lexer) != T_SLASH)
756 if (lex_force_num (lexer))
759 xrealloc (frq.percentiles,
760 (frq.n_percentiles + 1)
761 * sizeof (*frq.percentiles));
762 frq.percentiles[frq.n_percentiles].p = lex_number (lexer) / 100.0;
763 frq.percentiles[frq.n_percentiles].show = true;
766 frq.n_show_percentiles++;
770 lex_error (lexer, NULL);
775 else if (lex_match_id (lexer, "FORMAT"))
777 lex_match (lexer, T_EQUALS);
778 while (lex_token (lexer) != T_ENDCMD
779 && lex_token (lexer) != T_SLASH)
781 if (lex_match_id (lexer, "TABLE"))
785 else if (lex_match_id (lexer, "NOTABLE"))
787 frq.max_categories = 0;
789 else if (lex_match_id (lexer, "AVALUE"))
791 frq.sort = FRQ_AVALUE;
793 else if (lex_match_id (lexer, "DVALUE"))
795 frq.sort = FRQ_DVALUE;
797 else if (lex_match_id (lexer, "AFREQ"))
799 frq.sort = FRQ_AFREQ;
801 else if (lex_match_id (lexer, "DFREQ"))
803 frq.sort = FRQ_DFREQ;
807 lex_error (lexer, NULL);
812 else if (lex_match_id (lexer, "NTILES"))
814 lex_match (lexer, T_EQUALS);
816 if (lex_force_int (lexer))
819 int n = lex_integer (lexer);
821 for (i = 0; i < n + 1; ++i)
824 xrealloc (frq.percentiles,
825 (frq.n_percentiles + 1)
826 * sizeof (*frq.percentiles));
827 frq.percentiles[frq.n_percentiles].p =
829 frq.percentiles[frq.n_percentiles].show = true;
832 frq.n_show_percentiles++;
837 lex_error (lexer, NULL);
841 else if (lex_match_id (lexer, "ALGORITHM"))
843 lex_match (lexer, T_EQUALS);
845 if (lex_match_id (lexer, "COMPATIBLE"))
847 settings_set_cmd_algorithm (COMPATIBLE);
849 else if (lex_match_id (lexer, "ENHANCED"))
851 settings_set_cmd_algorithm (ENHANCED);
855 lex_error (lexer, NULL);
859 else if (lex_match_id (lexer, "HISTOGRAM"))
861 lex_match (lexer, T_EQUALS);
862 sbc_histogram = true;
864 while (lex_token (lexer) != T_ENDCMD
865 && lex_token (lexer) != T_SLASH)
867 if (lex_match_id (lexer, "NORMAL"))
869 hi_norm = FRQ_NORMAL;
871 else if (lex_match_id (lexer, "NONORMAL"))
873 hi_norm = FRQ_NONORMAL;
875 else if (lex_match_id (lexer, "FREQ"))
878 if (lex_match (lexer, T_LPAREN))
880 if (lex_force_int (lexer))
882 hi_freq = lex_integer (lexer);
885 lex_error (lexer, _("Histogram frequency must be greater than zero."));
888 lex_force_match (lexer, T_RPAREN);
892 else if (lex_match_id (lexer, "PERCENT"))
894 hi_scale = FRQ_PERCENT;
895 if (lex_match (lexer, T_LPAREN))
897 if (lex_force_int (lexer))
899 hi_pcnt = lex_integer (lexer);
902 lex_error (lexer, _("Histogram percentage must be greater than zero."));
905 lex_force_match (lexer, T_RPAREN);
909 else if (lex_match_id (lexer, "MINIMUM"))
911 lex_force_match (lexer, T_LPAREN);
912 if (lex_force_num (lexer))
914 hi_min = lex_number (lexer);
917 lex_force_match (lexer, T_RPAREN);
919 else if (lex_match_id (lexer, "MAXIMUM"))
921 lex_force_match (lexer, T_LPAREN);
922 if (lex_force_num (lexer))
924 hi_max = lex_number (lexer);
927 lex_force_match (lexer, T_RPAREN);
931 lex_error (lexer, NULL);
936 else if (lex_match_id (lexer, "PIECHART"))
938 lex_match (lexer, T_EQUALS);
939 while (lex_token (lexer) != T_ENDCMD
940 && lex_token (lexer) != T_SLASH)
942 if (lex_match_id (lexer, "MINIMUM"))
944 lex_force_match (lexer, T_LPAREN);
945 if (lex_force_num (lexer))
947 pie_min = lex_number (lexer);
950 lex_force_match (lexer, T_RPAREN);
952 else if (lex_match_id (lexer, "MAXIMUM"))
954 lex_force_match (lexer, T_LPAREN);
955 if (lex_force_num (lexer))
957 pie_max = lex_number (lexer);
960 lex_force_match (lexer, T_RPAREN);
962 else if (lex_match_id (lexer, "MISSING"))
966 else if (lex_match_id (lexer, "NOMISSING"))
972 lex_error (lexer, NULL);
978 else if (lex_match_id (lexer, "MISSING"))
980 lex_match (lexer, T_EQUALS);
982 while (lex_token (lexer) != T_ENDCMD
983 && lex_token (lexer) != T_SLASH)
985 if (lex_match_id (lexer, "EXCLUDE"))
988 else if (lex_match_id (lexer, "INCLUDE"))
993 lex_error (lexer, NULL);
1000 lex_error (lexer, NULL);
1005 if (frq.stats & BIT_INDEX (FRQ_ST_MEDIAN))
1008 xrealloc (frq.percentiles,
1009 (frq.n_percentiles + 1)
1010 * sizeof (*frq.percentiles));
1012 frq.percentiles[frq.n_percentiles].p = 0.50;
1013 frq.percentiles[frq.n_percentiles].show = true;
1015 frq.n_percentiles++;
1019 /* Figure out which charts the user requested. */
1023 msg (SW, _("Bar charts are not implemented."));
1027 struct frq_chart *hist;
1029 hist = frq.hist = xmalloc (sizeof *frq.hist);
1030 hist->x_min = hi_min;
1031 hist->x_max = hi_max;
1032 hist->y_scale = hi_scale;
1033 hist->y_max = hi_scale == FRQ_FREQ ? hi_freq : hi_pcnt;
1034 hist->draw_normal = hi_norm != FRQ_NONORMAL;
1035 hist->include_missing = false;
1037 if (hist->x_min != SYSMIS && hist->x_max != SYSMIS
1038 && hist->x_min >= hist->x_max)
1040 msg (SE, _("%s for histogram must be greater than or equal to %s, "
1041 "but %s was specified as %.15g and %s as %.15g. "
1042 "%s and %s will be ignored."),
1047 hist->x_min = hist->x_max = SYSMIS;
1051 xrealloc (frq.percentiles,
1052 (frq.n_percentiles + 2)
1053 * sizeof (*frq.percentiles));
1055 frq.percentiles[frq.n_percentiles].p = 0.25;
1056 frq.percentiles[frq.n_percentiles].show = false;
1058 frq.percentiles[frq.n_percentiles + 1].p = 0.75;
1059 frq.percentiles[frq.n_percentiles + 1].show = false;
1061 frq.n_percentiles+=2;
1066 struct frq_chart *pie;
1068 pie = frq.pie = xmalloc (sizeof *frq.pie);
1069 pie->x_min = pie_min;
1070 pie->x_max = pie_max;
1071 pie->include_missing = pie_missing;
1073 if (pie->x_min != SYSMIS && pie->x_max != SYSMIS
1074 && pie->x_min >= pie->x_max)
1076 msg (SE, _("%s for pie chart must be greater than or equal to %s, "
1077 "but %s was specified as %.15g and %s as %.15g. "
1078 "%s and %s will be ignored."),
1083 pie->x_min = pie->x_max = SYSMIS;
1090 double previous_p = -1;
1091 qsort (frq.percentiles, frq.n_percentiles,
1092 sizeof (*frq.percentiles),
1095 frq.n_show_percentiles = 0;
1096 for (i = o = 0; i < frq.n_percentiles; ++i)
1098 frq.percentiles[o].p = frq.percentiles[i].p;
1100 if (frq.percentiles[i].show)
1101 frq.percentiles[o].show = true;
1103 if (frq.percentiles[i].p != previous_p)
1105 if (frq.percentiles[i].show)
1106 frq.n_show_percentiles++;
1111 previous_p = frq.percentiles[i].p;
1114 frq.n_percentiles = o;
1118 struct casegrouper *grouper;
1119 struct casereader *group;
1122 grouper = casegrouper_create_splits (proc_open (ds), dataset_dict (ds));
1123 while (casegrouper_get_next_group (grouper, &group))
1126 precalc (&frq, group, ds);
1127 for (; (c = casereader_read (group)) != NULL; case_unref (c))
1129 postcalc (&frq, ds);
1131 ok = casegrouper_destroy (grouper);
1132 ok = proc_commit (ds) && ok;
1144 calculate_iqr (const struct frq_proc *frq)
1150 /* This cannot work unless the 25th and 75th percentile are calculated */
1151 assert (frq->n_percentiles >= 2);
1152 for (i = 0; i < frq->n_percentiles; i++)
1154 struct percentile *pc = &frq->percentiles[i];
1156 if (fabs (0.25 - pc->p) < DBL_EPSILON)
1158 else if (fabs (0.75 - pc->p) < DBL_EPSILON)
1162 return q1 == SYSMIS || q3 == SYSMIS ? SYSMIS : q3 - q1;
1166 chart_includes_value (const struct frq_chart *chart,
1167 const struct variable *var,
1168 const union value *value)
1170 if (!chart->include_missing && var_is_value_missing (var, value, MV_ANY))
1173 if (var_is_numeric (var)
1174 && ((chart->x_min != SYSMIS && value->f < chart->x_min)
1175 || (chart->x_max != SYSMIS && value->f > chart->x_max)))
1181 /* Create a gsl_histogram from a freq_tab */
1182 static struct histogram *
1183 freq_tab_to_hist (const struct frq_proc *frq, const struct freq_tab *ft,
1184 const struct variable *var)
1186 double x_min, x_max, valid_freq;
1189 struct histogram *histogram;
1192 /* Find out the extremes of the x value, within the range to be included in
1193 the histogram, and sum the total frequency of those values. */
1197 for (i = 0; i < ft->n_valid; i++)
1199 const struct freq *f = &ft->valid[i];
1200 if (chart_includes_value (frq->hist, var, &f->value))
1202 x_min = MIN (x_min, f->value.f);
1203 x_max = MAX (x_max, f->value.f);
1204 valid_freq += f->count;
1209 iqr = calculate_iqr (frq);
1212 /* Freedman-Diaconis' choice of bin width. */
1213 bin_width = 2 * iqr / pow (valid_freq, 1.0 / 3.0);
1217 bin_width = (x_max - x_min) / (1 + log2 (valid_freq));
1219 histogram = histogram_create (bin_width, x_min, x_max);
1221 if ( histogram == NULL)
1224 for (i = 0; i < ft->n_valid; i++)
1226 const struct freq *f = &ft->valid[i];
1227 if (chart_includes_value (frq->hist, var, &f->value))
1228 histogram_add (histogram, f->value.f, f->count);
1235 add_slice (const struct frq_chart *pie, const struct freq *freq,
1236 const struct variable *var, struct slice *slice)
1238 if (chart_includes_value (pie, var, &freq->value))
1240 ds_init_empty (&slice->label);
1241 var_append_value_name (var, &freq->value, &slice->label);
1242 slice->magnitude = freq->count;
1249 /* Allocate an array of slices and fill them from the data in frq_tab
1250 n_slices will contain the number of slices allocated.
1251 The caller is responsible for freeing slices
1253 static struct slice *
1254 freq_tab_to_slice_array(const struct frq_chart *pie,
1255 const struct freq_tab *frq_tab,
1256 const struct variable *var,
1259 struct slice *slices;
1263 slices = xnmalloc (frq_tab->n_valid + frq_tab->n_missing, sizeof *slices);
1266 for (i = 0; i < frq_tab->n_valid; i++)
1267 n_slices += add_slice (pie, &frq_tab->valid[i], var, &slices[n_slices]);
1268 for (i = 0; i < frq_tab->n_missing; i++)
1269 n_slices += add_slice (pie, &frq_tab->missing[i], var, &slices[n_slices]);
1271 *n_slicesp = n_slices;
1277 do_piechart(const struct frq_chart *pie, const struct variable *var,
1278 const struct freq_tab *frq_tab)
1280 struct slice *slices;
1283 slices = freq_tab_to_slice_array (pie, frq_tab, var, &n_slices);
1286 msg (SW, _("Omitting pie chart for %s, which has only %d unique values."),
1287 var_get_name (var), n_slices);
1288 else if (n_slices > 50)
1289 msg (SW, _("Omitting pie chart for %s, which has over 50 unique values."),
1290 var_get_name (var));
1292 chart_item_submit (piechart_create (var_to_string(var), slices, n_slices));
1294 for (i = 0; i < n_slices; i++)
1295 ds_destroy (&slices[i].label);
1299 /* Calculates all the pertinent statistics for VF, putting them in array
1302 calc_stats (const struct var_freqs *vf, double d[FRQ_ST_count])
1304 const struct freq_tab *ft = &vf->tab;
1305 double W = ft->valid_cases;
1306 const struct freq *f;
1311 assert (ft->n_valid > 0);
1313 /* Calculate the mode. */
1316 for (f = ft->valid; f < ft->missing; f++)
1318 if (most_often < f->count)
1320 most_often = f->count;
1321 X_mode = f->value.f;
1323 else if (most_often == f->count)
1325 /* A duplicate mode is undefined.
1326 FIXME: keep track of *all* the modes. */
1331 /* Calculate moments. */
1332 m = moments_create (MOMENT_KURTOSIS);
1333 for (f = ft->valid; f < ft->missing; f++)
1334 moments_pass_one (m, f->value.f, f->count);
1335 for (f = ft->valid; f < ft->missing; f++)
1336 moments_pass_two (m, f->value.f, f->count);
1337 moments_calculate (m, NULL, &d[FRQ_ST_MEAN], &d[FRQ_ST_VARIANCE],
1338 &d[FRQ_ST_SKEWNESS], &d[FRQ_ST_KURTOSIS]);
1339 moments_destroy (m);
1341 /* Formulas below are taken from _SPSS Statistical Algorithms_. */
1342 d[FRQ_ST_MINIMUM] = ft->valid[0].value.f;
1343 d[FRQ_ST_MAXIMUM] = ft->valid[ft->n_valid - 1].value.f;
1344 d[FRQ_ST_MODE] = X_mode;
1345 d[FRQ_ST_RANGE] = d[FRQ_ST_MAXIMUM] - d[FRQ_ST_MINIMUM];
1346 d[FRQ_ST_SUM] = d[FRQ_ST_MEAN] * W;
1347 d[FRQ_ST_STDDEV] = sqrt (d[FRQ_ST_VARIANCE]);
1348 d[FRQ_ST_SEMEAN] = d[FRQ_ST_STDDEV] / sqrt (W);
1349 d[FRQ_ST_SESKEWNESS] = calc_seskew (W);
1350 d[FRQ_ST_SEKURTOSIS] = calc_sekurt (W);
1353 /* Displays a table of all the statistics requested for variable V. */
1355 dump_statistics (const struct frq_proc *frq, const struct var_freqs *vf,
1356 const struct variable *wv)
1358 const struct fmt_spec *wfmt = wv ? var_get_print_format (wv) : &F_8_0;
1359 const struct freq_tab *ft = &vf->tab;
1360 double stat_value[FRQ_ST_count];
1361 struct tab_table *t;
1364 if (var_is_alpha (vf->var))
1367 if (ft->n_valid == 0)
1369 msg (SW, _("No valid data for variable %s; statistics not displayed."),
1370 var_get_name (vf->var));
1373 calc_stats (vf, stat_value);
1375 t = tab_create (3, ((frq->stats & BIT_INDEX (FRQ_ST_MEDIAN)) ? frq->n_stats - 1 : frq->n_stats)
1376 + frq->n_show_percentiles + 2);
1377 tab_set_format (t, RC_WEIGHT, wfmt);
1378 tab_box (t, TAL_1, TAL_1, -1, -1 , 0 , 0 , 2, tab_nr(t) - 1) ;
1381 tab_vline (t, TAL_1 , 2, 0, tab_nr(t) - 1);
1382 tab_vline (t, TAL_GAP , 1, 0, tab_nr(t) - 1 ) ;
1384 r = 2; /* N missing and N valid are always dumped */
1386 for (i = 0; i < FRQ_ST_count; i++)
1388 if (FRQ_ST_MEDIAN == i)
1391 if (frq->stats & BIT_INDEX (i))
1393 tab_text (t, 0, r, TAB_LEFT | TAT_TITLE,
1394 gettext (st_name[i]));
1395 tab_double (t, 2, r, TAB_NONE, stat_value[i], NULL, RC_OTHER);
1400 tab_text (t, 0, 0, TAB_LEFT | TAT_TITLE, _("N"));
1401 tab_text (t, 1, 0, TAB_LEFT | TAT_TITLE, _("Valid"));
1402 tab_text (t, 1, 1, TAB_LEFT | TAT_TITLE, _("Missing"));
1404 tab_double (t, 2, 0, TAB_NONE, ft->valid_cases, NULL, RC_WEIGHT);
1405 tab_double (t, 2, 1, TAB_NONE, ft->total_cases - ft->valid_cases, NULL, RC_WEIGHT);
1407 for (i = 0; i < frq->n_percentiles; i++)
1409 const struct percentile *pc = &frq->percentiles[i];
1416 tab_text (t, 0, r, TAB_LEFT | TAT_TITLE, _("Percentiles"));
1420 tab_text (t, 1, r, TAB_LEFT, _("50 (Median)"));
1422 tab_double (t, 1, r, TAB_LEFT, pc->p * 100, NULL, RC_INTEGER);
1423 tab_double (t, 2, r, TAB_NONE, pc->value,
1424 var_get_print_format (vf->var), RC_OTHER);
1428 tab_title (t, "%s", var_to_string (vf->var));