2 PSPP - a program for statistical analysis.
3 Copyright (C) 1997-9, 2000, 2007, 2009, 2010, 2011, 2014, 2015 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/barchart.h"
57 #include "output/charts/piechart.h"
58 #include "output/charts/plot-hist.h"
59 #include "output/tab.h"
61 #include "gl/minmax.h"
62 #include "gl/xalloc.h"
65 #define _(msgid) gettext (msgid)
66 #define N_(msgid) msgid
68 /* Percentiles to calculate. */
72 double p; /* the %ile to be calculated */
73 double value; /* the %ile's value */
74 bool show; /* True to show this percentile in the statistics box. */
78 ptile_3way (const void *_p1, const void *_p2)
80 const struct percentile *p1 = _p1;
81 const struct percentile *p2 = _p2;
88 if (p1->show > p2->show)
91 return (p1->show < p2->show);
94 return (p1->p > p2->p);
118 /* Array indices for STATISTICS subcommand. */
138 /* Description of statistics. */
139 static const char *st_name[FRQ_ST_count] =
159 struct hmap data; /* Hash table for accumulating counts. */
160 struct freq *valid; /* Valid freqs. */
161 int n_valid; /* Number of total freqs. */
162 const struct dictionary *dict; /* Source of entries in the table. */
164 struct freq *missing; /* Missing freqs. */
165 int n_missing; /* Number of missing freqs. */
168 double total_cases; /* Sum of weights of all cases. */
169 double valid_cases; /* Sum of weights of valid cases. */
174 double x_min; /* X axis minimum value. */
175 double x_max; /* X axis maximum value. */
176 int y_scale; /* Y axis scale: FRQ_FREQ or FRQ_PERCENT. */
178 /* Histograms only. */
179 double y_max; /* Y axis maximum value. */
180 bool draw_normal; /* Whether to draw normal curve. */
182 /* Pie charts only. */
183 bool include_missing; /* Whether to include missing values. */
186 /* Per-variable frequency data. */
189 const struct variable *var;
191 /* Freqency table. */
192 struct freq_tab tab; /* Frequencies table to use. */
195 int n_groups; /* Number of groups. */
196 double *groups; /* Groups. */
199 double stat[FRQ_ST_count];
201 /* Variable attributes. */
209 struct var_freqs *vars;
212 /* Percentiles to calculate and possibly display. */
213 struct percentile *percentiles;
214 int n_percentiles, n_show_percentiles;
216 /* Frequency table display. */
217 long int max_categories; /* Maximum categories to show. */
218 int sort; /* FRQ_AVALUE or FRQ_DVALUE
219 or FRQ_AFREQ or FRQ_DFREQ. */
221 /* Statistics; number of statistics. */
225 /* Histogram and pie chart settings. */
226 struct frq_chart *hist, *pie, *bar;
230 struct freq_compare_aux
236 bool ascending_value;
239 static void calc_stats (const struct var_freqs *vf, double d[FRQ_ST_count]);
241 static void do_piechart(const struct frq_chart *pie,
242 const struct variable *var,
243 const struct freq_tab *frq_tab);
245 static void do_barchart(const struct frq_chart *bar,
246 const struct variable **var,
247 const struct freq_tab *frq_tab);
249 static void dump_statistics (const struct frq_proc *frq,
250 const struct var_freqs *vf,
251 const struct variable *wv);
254 compare_freq (const void *a_, const void *b_, const void *aux_)
256 const struct freq_compare_aux *aux = aux_;
257 const struct freq *a = a_;
258 const struct freq *b = b_;
260 if (aux->by_freq && a->count != b->count)
262 int cmp = a->count > b->count ? 1 : -1;
263 return aux->ascending_freq ? cmp : -cmp;
267 int cmp = value_compare_3way (a->values, b->values, aux->width);
268 return aux->ascending_value ? cmp : -cmp;
272 /* Create a gsl_histogram from a freq_tab */
273 static struct histogram *
274 freq_tab_to_hist (const struct frq_proc *frq, const struct freq_tab *ft,
275 const struct variable *var);
278 /* Displays a full frequency table for variable V. */
280 dump_freq_table (const struct var_freqs *vf, const struct variable *wv)
282 const struct fmt_spec *wfmt = wv ? var_get_print_format (wv) : &F_8_0;
283 const struct freq_tab *ft = &vf->tab;
288 double cum_total = 0.0;
289 double cum_freq = 0.0;
291 static const char *headings[] = {
300 n_categories = ft->n_valid + ft->n_missing;
301 t = tab_create (6, n_categories + 2);
302 tab_set_format (t, RC_WEIGHT, wfmt);
303 tab_headers (t, 0, 0, 1, 0);
305 for (x = 0; x < 6; x++)
306 tab_text (t, x, 0, TAB_CENTER | TAT_TITLE, gettext (headings[x]));
309 for (f = ft->valid; f < ft->missing; f++)
312 double percent, valid_percent;
314 cum_freq += f->count;
316 percent = f->count / ft->total_cases * 100.0;
317 valid_percent = f->count / ft->valid_cases * 100.0;
318 cum_total += valid_percent;
320 label = var_lookup_value_label (vf->var, f->values);
322 tab_text (t, 0, r, TAB_LEFT, label);
324 tab_value (t, 1, r, TAB_NONE, f->values, vf->var, NULL);
325 tab_double (t, 2, r, TAB_NONE, f->count, NULL, RC_WEIGHT);
326 tab_double (t, 3, r, TAB_NONE, percent, NULL, RC_OTHER);
327 tab_double (t, 4, r, TAB_NONE, valid_percent, NULL, RC_OTHER);
328 tab_double (t, 5, r, TAB_NONE, cum_total, NULL, RC_OTHER);
331 for (; f < &ft->valid[n_categories]; f++)
335 cum_freq += f->count;
337 label = var_lookup_value_label (vf->var, f->values);
339 tab_text (t, 0, r, TAB_LEFT, label);
341 tab_value (t, 1, r, TAB_NONE, f->values, vf->var, NULL);
342 tab_double (t, 2, r, TAB_NONE, f->count, NULL, RC_WEIGHT);
343 tab_double (t, 3, r, TAB_NONE,
344 f->count / ft->total_cases * 100.0, NULL, RC_OTHER);
345 tab_text (t, 4, r, TAB_NONE, _("Missing"));
349 tab_box (t, TAL_1, TAL_1, -1, TAL_1, 0, 0, 5, r);
350 tab_hline (t, TAL_2, 0, 5, 1);
351 tab_hline (t, TAL_2, 0, 5, r);
352 tab_joint_text (t, 0, r, 1, r, TAB_RIGHT | TAT_TITLE, _("Total"));
353 tab_vline (t, TAL_0, 1, r, r);
354 tab_double (t, 2, r, TAB_NONE, cum_freq, NULL, RC_WEIGHT);
355 tab_double (t, 3, r, TAB_NONE, 100.0, &F_5_1, RC_OTHER);
356 tab_double (t, 4, r, TAB_NONE, 100.0, &F_5_1, RC_OTHER);
358 tab_title (t, "%s", var_to_string (vf->var));
362 /* Statistical display. */
365 calc_percentile (double p, double valid_cases, double x1, double x2)
369 s = (settings_get_algorithm () != COMPATIBLE
370 ? modf ((valid_cases - 1) * p, &dummy)
371 : modf ((valid_cases + 1) * p - 1, &dummy));
373 return x1 + (x2 - x1) * s;
376 /* Calculates all of the percentiles for VF within FRQ. */
378 calc_percentiles (const struct frq_proc *frq, const struct var_freqs *vf)
380 const struct freq_tab *ft = &vf->tab;
381 double W = ft->valid_cases;
382 const struct freq *f;
383 int percentile_idx = 0;
386 for (f = ft->valid; f < ft->missing; f++)
389 for (; percentile_idx < frq->n_percentiles; percentile_idx++)
391 struct percentile *pc = &frq->percentiles[percentile_idx];
394 tp = (settings_get_algorithm () == ENHANCED
396 : (W + 1) * pc->p - 1);
401 if (tp + 1 < rank || f + 1 >= ft->missing)
402 pc->value = f->values[0].f;
404 pc->value = calc_percentile (pc->p, W, f->values[0].f, f[1].values[0].f);
407 for (; percentile_idx < frq->n_percentiles; percentile_idx++)
409 struct percentile *pc = &frq->percentiles[percentile_idx];
410 pc->value = ft->valid[ft->n_valid - 1].values[0].f;
414 /* Returns true iff the value in struct freq F is non-missing
417 not_missing (const void *f_, const void *v_)
419 const struct freq *f = f_;
420 const struct variable *v = v_;
422 return !var_is_value_missing (v, f->values, MV_ANY);
426 /* Summarizes the frequency table data for variable V. */
428 postprocess_freq_tab (const struct frq_proc *frq, struct var_freqs *vf)
430 struct freq_tab *ft = &vf->tab;
431 struct freq_compare_aux aux;
433 struct freq *freqs, *f;
436 /* Extract data from hash table. */
437 count = hmap_count (&ft->data);
438 freqs = freq_hmap_extract (&ft->data);
440 /* Put data into ft. */
442 ft->n_valid = partition (freqs, count, sizeof *freqs, not_missing, vf->var);
443 ft->missing = freqs + ft->n_valid;
444 ft->n_missing = count - ft->n_valid;
447 aux.by_freq = frq->sort == FRQ_AFREQ || frq->sort == FRQ_DFREQ;
448 aux.ascending_freq = frq->sort != FRQ_DFREQ;
449 aux.width = vf->width;
450 aux.ascending_value = frq->sort != FRQ_DVALUE;
451 sort (ft->valid, ft->n_valid, sizeof *ft->valid, compare_freq, &aux);
452 sort (ft->missing, ft->n_missing, sizeof *ft->missing, compare_freq, &aux);
454 /* Summary statistics. */
455 ft->valid_cases = 0.0;
456 for(i = 0 ; i < ft->n_valid ; ++i )
459 ft->valid_cases += f->count;
463 ft->total_cases = ft->valid_cases ;
464 for(i = 0 ; i < ft->n_missing ; ++i )
467 ft->total_cases += f->count;
472 /* Frees the frequency table for variable V. */
474 cleanup_freq_tab (struct var_freqs *vf)
476 free (vf->tab.valid);
477 freq_hmap_destroy (&vf->tab.data, vf->width);
480 /* Add data from case C to the frequency table. */
482 calc (struct frq_proc *frq, const struct ccase *c, const struct dataset *ds)
484 double weight = dict_get_case_weight (dataset_dict (ds), c, NULL);
487 for (i = 0; i < frq->n_vars; i++)
489 struct var_freqs *vf = &frq->vars[i];
490 const union value *value = case_data (c, vf->var);
491 size_t hash = value_hash (value, vf->width, 0);
494 f = freq_hmap_search (&vf->tab.data, value, vf->width, hash);
496 f = freq_hmap_insert (&vf->tab.data, value, vf->width, hash);
502 /* Prepares each variable that is the target of FREQUENCIES by setting
503 up its hash table. */
505 precalc (struct frq_proc *frq, struct casereader *input, struct dataset *ds)
510 c = casereader_peek (input, 0);
513 output_split_file_values (ds, c);
517 for (i = 0; i < frq->n_vars; i++)
518 hmap_init (&frq->vars[i].tab.data);
521 /* Finishes up with the variables after frequencies have been
522 calculated. Displays statistics, percentiles, ... */
524 postcalc (struct frq_proc *frq, const struct dataset *ds)
526 const struct dictionary *dict = dataset_dict (ds);
527 const struct variable *wv = dict_get_weight (dict);
530 for (i = 0; i < frq->n_vars; i++)
532 struct var_freqs *vf = &frq->vars[i];
534 postprocess_freq_tab (frq, vf);
536 /* Frequencies tables. */
537 if (vf->tab.n_valid + vf->tab.n_missing <= frq->max_categories)
538 dump_freq_table (vf, wv);
540 calc_percentiles (frq, vf);
544 dump_statistics (frq, vf, wv);
546 if (frq->hist && var_is_numeric (vf->var) && vf->tab.n_valid > 0)
548 double d[FRQ_ST_count];
549 struct histogram *histogram;
553 histogram = freq_tab_to_hist (frq, &vf->tab, vf->var);
557 chart_item_submit (histogram_chart_create (
558 histogram->gsl_hist, var_to_string(vf->var),
562 frq->hist->draw_normal));
564 statistic_destroy (&histogram->parent);
569 do_piechart(frq->pie, vf->var, &vf->tab);
572 do_barchart(frq->bar, &vf->var, &vf->tab);
574 cleanup_freq_tab (vf);
579 cmd_frequencies (struct lexer *lexer, struct dataset *ds)
583 const struct variable **vars = NULL;
585 bool sbc_barchart = false;
586 bool sbc_piechart = false;
587 bool sbc_histogram = false;
589 double pie_min = -DBL_MAX;
590 double pie_max = DBL_MAX;
591 bool pie_missing = true;
593 double bar_min = -DBL_MAX;
594 double bar_max = DBL_MAX;
595 bool bar_freq = true;
597 double hi_min = -DBL_MAX;
598 double hi_max = DBL_MAX;
599 int hi_scale = FRQ_FREQ;
600 int hi_freq = INT_MIN;
601 int hi_pcnt = INT_MIN;
602 int hi_norm = FRQ_NONORMAL;
604 frq.pool = pool_create ();
605 frq.sort = FRQ_AVALUE;
610 frq.stats = BIT_INDEX (FRQ_ST_MEAN)
611 | BIT_INDEX (FRQ_ST_STDDEV)
612 | BIT_INDEX (FRQ_ST_MINIMUM)
613 | BIT_INDEX (FRQ_ST_MAXIMUM);
617 frq.max_categories = LONG_MAX;
619 frq.percentiles = NULL;
620 frq.n_percentiles = 0;
621 frq.n_show_percentiles = 0;
628 /* Accept an optional, completely pointless "/VARIABLES=" */
629 lex_match (lexer, T_SLASH);
630 if (lex_match_id (lexer, "VARIABLES"))
632 if (! lex_force_match (lexer, T_EQUALS) )
636 if (!parse_variables_const (lexer, dataset_dict (ds),
642 frq.vars = xzalloc (frq.n_vars * sizeof (*frq.vars));
643 for (i = 0; i < frq.n_vars; ++i)
645 frq.vars[i].var = vars[i];
646 frq.vars[i].width = var_get_width (vars[i]);
649 while (lex_token (lexer) != T_ENDCMD)
651 lex_match (lexer, T_SLASH);
653 if (lex_match_id (lexer, "STATISTICS"))
655 frq.stats = BIT_INDEX (FRQ_ST_MEAN)
656 | BIT_INDEX (FRQ_ST_STDDEV)
657 | BIT_INDEX (FRQ_ST_MINIMUM)
658 | BIT_INDEX (FRQ_ST_MAXIMUM);
662 if (lex_match (lexer, T_EQUALS))
668 while (lex_token (lexer) != T_ENDCMD
669 && lex_token (lexer) != T_SLASH)
671 if (lex_match_id (lexer, "DEFAULT"))
673 frq.stats = BIT_INDEX (FRQ_ST_MEAN)
674 | BIT_INDEX (FRQ_ST_STDDEV)
675 | BIT_INDEX (FRQ_ST_MINIMUM)
676 | BIT_INDEX (FRQ_ST_MAXIMUM);
680 else if (lex_match_id (lexer, "MEAN"))
682 frq.stats |= BIT_INDEX (FRQ_ST_MEAN);
685 else if (lex_match_id (lexer, "SEMEAN"))
687 frq.stats |= BIT_INDEX (FRQ_ST_SEMEAN);
690 else if (lex_match_id (lexer, "MEDIAN"))
692 frq.stats |= BIT_INDEX (FRQ_ST_MEDIAN);
695 else if (lex_match_id (lexer, "MODE"))
697 frq.stats |= BIT_INDEX (FRQ_ST_MODE);
700 else if (lex_match_id (lexer, "STDDEV"))
702 frq.stats |= BIT_INDEX (FRQ_ST_STDDEV);
705 else if (lex_match_id (lexer, "VARIANCE"))
707 frq.stats |= BIT_INDEX (FRQ_ST_VARIANCE);
710 else if (lex_match_id (lexer, "KURTOSIS"))
712 frq.stats |= BIT_INDEX (FRQ_ST_KURTOSIS);
715 else if (lex_match_id (lexer, "SKEWNESS"))
717 frq.stats |= BIT_INDEX (FRQ_ST_SKEWNESS);
720 else if (lex_match_id (lexer, "RANGE"))
722 frq.stats |= BIT_INDEX (FRQ_ST_RANGE);
725 else if (lex_match_id (lexer, "MINIMUM"))
727 frq.stats |= BIT_INDEX (FRQ_ST_MINIMUM);
730 else if (lex_match_id (lexer, "MAXIMUM"))
732 frq.stats |= BIT_INDEX (FRQ_ST_MAXIMUM);
735 else if (lex_match_id (lexer, "SUM"))
737 frq.stats |= BIT_INDEX (FRQ_ST_SUM);
740 else if (lex_match_id (lexer, "SESKEWNESS"))
742 frq.stats |= BIT_INDEX (FRQ_ST_SESKEWNESS);
745 else if (lex_match_id (lexer, "SEKURTOSIS"))
747 frq.stats |= BIT_INDEX (FRQ_ST_SEKURTOSIS);
750 else if (lex_match_id (lexer, "NONE"))
755 else if (lex_match (lexer, T_ALL))
758 frq.n_stats = FRQ_ST_count;
762 lex_error (lexer, NULL);
767 else if (lex_match_id (lexer, "PERCENTILES"))
769 lex_match (lexer, T_EQUALS);
770 while (lex_token (lexer) != T_ENDCMD
771 && lex_token (lexer) != T_SLASH)
773 if (lex_force_num (lexer))
776 xrealloc (frq.percentiles,
777 (frq.n_percentiles + 1)
778 * sizeof (*frq.percentiles));
779 frq.percentiles[frq.n_percentiles].p = lex_number (lexer) / 100.0;
780 frq.percentiles[frq.n_percentiles].show = true;
783 frq.n_show_percentiles++;
787 lex_error (lexer, NULL);
790 lex_match (lexer, T_COMMA);
793 else if (lex_match_id (lexer, "FORMAT"))
795 lex_match (lexer, T_EQUALS);
796 while (lex_token (lexer) != T_ENDCMD
797 && lex_token (lexer) != T_SLASH)
799 if (lex_match_id (lexer, "TABLE"))
802 else if (lex_match_id (lexer, "NOTABLE"))
804 frq.max_categories = 0;
806 else if (lex_match_id (lexer, "LIMIT"))
808 if (!lex_force_match (lexer, T_LPAREN)
809 || !lex_force_int (lexer))
812 frq.max_categories = lex_integer (lexer);
815 if (!lex_force_match (lexer, T_RPAREN))
818 else if (lex_match_id (lexer, "AVALUE"))
820 frq.sort = FRQ_AVALUE;
822 else if (lex_match_id (lexer, "DVALUE"))
824 frq.sort = FRQ_DVALUE;
826 else if (lex_match_id (lexer, "AFREQ"))
828 frq.sort = FRQ_AFREQ;
830 else if (lex_match_id (lexer, "DFREQ"))
832 frq.sort = FRQ_DFREQ;
836 lex_error (lexer, NULL);
841 else if (lex_match_id (lexer, "NTILES"))
843 lex_match (lexer, T_EQUALS);
845 if (lex_force_int (lexer))
848 int n = lex_integer (lexer);
850 for (i = 0; i < n + 1; ++i)
853 xrealloc (frq.percentiles,
854 (frq.n_percentiles + 1)
855 * sizeof (*frq.percentiles));
856 frq.percentiles[frq.n_percentiles].p =
858 frq.percentiles[frq.n_percentiles].show = true;
861 frq.n_show_percentiles++;
866 lex_error (lexer, NULL);
870 else if (lex_match_id (lexer, "ALGORITHM"))
872 lex_match (lexer, T_EQUALS);
874 if (lex_match_id (lexer, "COMPATIBLE"))
876 settings_set_cmd_algorithm (COMPATIBLE);
878 else if (lex_match_id (lexer, "ENHANCED"))
880 settings_set_cmd_algorithm (ENHANCED);
884 lex_error (lexer, NULL);
888 else if (lex_match_id (lexer, "HISTOGRAM"))
890 lex_match (lexer, T_EQUALS);
891 sbc_histogram = true;
893 while (lex_token (lexer) != T_ENDCMD
894 && lex_token (lexer) != T_SLASH)
896 if (lex_match_id (lexer, "NORMAL"))
898 hi_norm = FRQ_NORMAL;
900 else if (lex_match_id (lexer, "NONORMAL"))
902 hi_norm = FRQ_NONORMAL;
904 else if (lex_match_id (lexer, "FREQ"))
907 if (lex_match (lexer, T_LPAREN))
909 if (lex_force_int (lexer))
911 hi_freq = lex_integer (lexer);
914 lex_error (lexer, _("Histogram frequency must be greater than zero."));
917 if (! lex_force_match (lexer, T_RPAREN))
922 else if (lex_match_id (lexer, "PERCENT"))
924 hi_scale = FRQ_PERCENT;
925 if (lex_match (lexer, T_LPAREN))
927 if (lex_force_int (lexer))
929 hi_pcnt = lex_integer (lexer);
932 lex_error (lexer, _("Histogram percentage must be greater than zero."));
935 if (! lex_force_match (lexer, T_RPAREN))
940 else if (lex_match_id (lexer, "MINIMUM"))
942 if (! lex_force_match (lexer, T_LPAREN))
944 if (lex_force_num (lexer))
946 hi_min = lex_number (lexer);
949 if (! lex_force_match (lexer, T_RPAREN))
952 else if (lex_match_id (lexer, "MAXIMUM"))
954 if (! lex_force_match (lexer, T_LPAREN))
956 if (lex_force_num (lexer))
958 hi_max = lex_number (lexer);
961 if (! lex_force_match (lexer, T_RPAREN))
966 lex_error (lexer, NULL);
971 else if (lex_match_id (lexer, "PIECHART"))
973 lex_match (lexer, T_EQUALS);
974 while (lex_token (lexer) != T_ENDCMD
975 && lex_token (lexer) != T_SLASH)
977 if (lex_match_id (lexer, "MINIMUM"))
979 if (! lex_force_match (lexer, T_LPAREN))
981 if (lex_force_num (lexer))
983 pie_min = lex_number (lexer);
986 if (! lex_force_match (lexer, T_RPAREN))
989 else if (lex_match_id (lexer, "MAXIMUM"))
991 if (! lex_force_match (lexer, T_LPAREN))
993 if (lex_force_num (lexer))
995 pie_max = lex_number (lexer);
998 if (! lex_force_match (lexer, T_RPAREN))
1001 else if (lex_match_id (lexer, "MISSING"))
1005 else if (lex_match_id (lexer, "NOMISSING"))
1007 pie_missing = false;
1011 lex_error (lexer, NULL);
1015 sbc_piechart = true;
1017 else if (lex_match_id (lexer, "BARCHART"))
1019 lex_match (lexer, T_EQUALS);
1020 while (lex_token (lexer) != T_ENDCMD
1021 && lex_token (lexer) != T_SLASH)
1023 if (lex_match_id (lexer, "MINIMUM"))
1025 if (! lex_force_match (lexer, T_LPAREN))
1027 if (lex_force_num (lexer))
1029 bar_min = lex_number (lexer);
1032 if (! lex_force_match (lexer, T_RPAREN))
1035 else if (lex_match_id (lexer, "MAXIMUM"))
1037 if (! lex_force_match (lexer, T_LPAREN))
1039 if (lex_force_num (lexer))
1041 bar_max = lex_number (lexer);
1044 if (! lex_force_match (lexer, T_RPAREN))
1047 else if (lex_match_id (lexer, "FREQ"))
1049 if ( lex_match (lexer, T_LPAREN))
1051 if (lex_force_num (lexer))
1056 if (! lex_force_match (lexer, T_RPAREN))
1061 else if (lex_match_id (lexer, "PERCENT"))
1063 if ( lex_match (lexer, T_LPAREN))
1065 if (lex_force_num (lexer))
1070 if (! lex_force_match (lexer, T_RPAREN))
1077 lex_error (lexer, NULL);
1081 sbc_barchart = true;
1083 else if (lex_match_id (lexer, "MISSING"))
1085 lex_match (lexer, T_EQUALS);
1087 while (lex_token (lexer) != T_ENDCMD
1088 && lex_token (lexer) != T_SLASH)
1090 if (lex_match_id (lexer, "EXCLUDE"))
1093 else if (lex_match_id (lexer, "INCLUDE"))
1098 lex_error (lexer, NULL);
1103 else if (lex_match_id (lexer, "ORDER"))
1105 lex_match (lexer, T_EQUALS);
1106 if (!lex_match_id (lexer, "ANALYSIS"))
1107 lex_match_id (lexer, "VARIABLE");
1111 lex_error (lexer, NULL);
1116 if (frq.stats & BIT_INDEX (FRQ_ST_MEDIAN))
1119 xrealloc (frq.percentiles,
1120 (frq.n_percentiles + 1)
1121 * sizeof (*frq.percentiles));
1123 frq.percentiles[frq.n_percentiles].p = 0.50;
1124 frq.percentiles[frq.n_percentiles].show = true;
1126 frq.n_percentiles++;
1127 frq.n_show_percentiles++;
1131 /* Figure out which charts the user requested. */
1136 struct frq_chart *hist;
1138 hist = frq.hist = xmalloc (sizeof *frq.hist);
1139 hist->x_min = hi_min;
1140 hist->x_max = hi_max;
1141 hist->y_scale = hi_scale;
1142 hist->y_max = hi_scale == FRQ_FREQ ? hi_freq : hi_pcnt;
1143 hist->draw_normal = hi_norm != FRQ_NONORMAL;
1144 hist->include_missing = false;
1146 if (hist->x_min != SYSMIS && hist->x_max != SYSMIS
1147 && hist->x_min >= hist->x_max)
1149 msg (SE, _("%s for histogram must be greater than or equal to %s, "
1150 "but %s was specified as %.15g and %s as %.15g. "
1151 "%s and %s will be ignored."),
1156 hist->x_min = hist->x_max = SYSMIS;
1160 xrealloc (frq.percentiles,
1161 (frq.n_percentiles + 2)
1162 * sizeof (*frq.percentiles));
1164 frq.percentiles[frq.n_percentiles].p = 0.25;
1165 frq.percentiles[frq.n_percentiles].show = false;
1167 frq.percentiles[frq.n_percentiles + 1].p = 0.75;
1168 frq.percentiles[frq.n_percentiles + 1].show = false;
1170 frq.n_percentiles+=2;
1175 frq.bar = xmalloc (sizeof *frq.bar);
1176 frq.bar->x_min = bar_min;
1177 frq.bar->x_max = bar_max;
1178 frq.bar->include_missing = false;
1179 frq.bar->y_scale = bar_freq ? FRQ_FREQ : FRQ_PERCENT;
1184 struct frq_chart *pie;
1186 pie = frq.pie = xmalloc (sizeof *frq.pie);
1187 pie->x_min = pie_min;
1188 pie->x_max = pie_max;
1189 pie->include_missing = pie_missing;
1191 if (pie->x_min != SYSMIS && pie->x_max != SYSMIS
1192 && pie->x_min >= pie->x_max)
1194 msg (SE, _("%s for pie chart must be greater than or equal to %s, "
1195 "but %s was specified as %.15g and %s as %.15g. "
1196 "%s and %s will be ignored."),
1201 pie->x_min = pie->x_max = SYSMIS;
1208 double previous_p = -1;
1209 qsort (frq.percentiles, frq.n_percentiles,
1210 sizeof (*frq.percentiles),
1213 frq.n_show_percentiles = 0;
1214 for (i = o = 0; i < frq.n_percentiles; ++i)
1216 if (frq.percentiles[i].p != previous_p)
1218 frq.percentiles[o].p = frq.percentiles[i].p;
1219 frq.percentiles[o].show = frq.percentiles[i].show;
1220 if (frq.percentiles[i].show)
1221 frq.n_show_percentiles++;
1224 else if (frq.percentiles[i].show &&
1225 !frq.percentiles[o].show)
1227 frq.percentiles[o].show = true;
1228 frq.n_show_percentiles++;
1230 previous_p = frq.percentiles[i].p;
1233 frq.n_percentiles = o;
1237 struct casegrouper *grouper;
1238 struct casereader *group;
1241 grouper = casegrouper_create_splits (proc_open (ds), dataset_dict (ds));
1242 while (casegrouper_get_next_group (grouper, &group))
1245 precalc (&frq, group, ds);
1247 for (; (c = casereader_read (group)) != NULL; case_unref (c))
1249 postcalc (&frq, ds);
1250 casereader_destroy (group);
1252 ok = casegrouper_destroy (grouper);
1253 ok = proc_commit (ds) && ok;
1262 free (frq.percentiles);
1263 pool_destroy (frq.pool);
1274 free (frq.percentiles);
1275 pool_destroy (frq.pool);
1281 calculate_iqr (const struct frq_proc *frq)
1287 /* This cannot work unless the 25th and 75th percentile are calculated */
1288 assert (frq->n_percentiles >= 2);
1289 for (i = 0; i < frq->n_percentiles; i++)
1291 struct percentile *pc = &frq->percentiles[i];
1293 if (fabs (0.25 - pc->p) < DBL_EPSILON)
1295 else if (fabs (0.75 - pc->p) < DBL_EPSILON)
1299 return q1 == SYSMIS || q3 == SYSMIS ? SYSMIS : q3 - q1;
1303 chart_includes_value (const struct frq_chart *chart,
1304 const struct variable *var,
1305 const union value *value)
1307 if (!chart->include_missing && var_is_value_missing (var, value, MV_ANY))
1310 if (var_is_numeric (var)
1311 && ((chart->x_min != SYSMIS && value->f < chart->x_min)
1312 || (chart->x_max != SYSMIS && value->f > chart->x_max)))
1318 /* Create a gsl_histogram from a freq_tab */
1319 static struct histogram *
1320 freq_tab_to_hist (const struct frq_proc *frq, const struct freq_tab *ft,
1321 const struct variable *var)
1323 double x_min, x_max, valid_freq;
1326 struct histogram *histogram;
1329 /* Find out the extremes of the x value, within the range to be included in
1330 the histogram, and sum the total frequency of those values. */
1334 for (i = 0; i < ft->n_valid; i++)
1336 const struct freq *f = &ft->valid[i];
1337 if (chart_includes_value (frq->hist, var, f->values))
1339 x_min = MIN (x_min, f->values[0].f);
1340 x_max = MAX (x_max, f->values[0].f);
1341 valid_freq += f->count;
1345 if (valid_freq <= 0)
1348 iqr = calculate_iqr (frq);
1351 /* Freedman-Diaconis' choice of bin width. */
1352 bin_width = 2 * iqr / pow (valid_freq, 1.0 / 3.0);
1356 bin_width = (x_max - x_min) / (1 + log2 (valid_freq));
1358 histogram = histogram_create (bin_width, x_min, x_max);
1360 if ( histogram == NULL)
1363 for (i = 0; i < ft->n_valid; i++)
1365 const struct freq *f = &ft->valid[i];
1366 if (chart_includes_value (frq->hist, var, f->values))
1367 histogram_add (histogram, f->values[0].f, f->count);
1374 /* Allocate an array of struct freqs and fill them from the data in FRQ_TAB,
1375 according to the parameters of CATCHART
1376 N_SLICES will contain the number of slices allocated.
1377 The caller is responsible for freeing slices
1379 static struct freq *
1380 pick_cat_counts (const struct frq_chart *catchart,
1381 const struct freq_tab *frq_tab,
1386 struct freq *slices = xnmalloc (frq_tab->n_valid + frq_tab->n_missing, sizeof *slices);
1388 for (i = 0; i < frq_tab->n_valid; i++)
1390 const struct freq *f = &frq_tab->valid[i];
1391 if (f->count > catchart->x_max)
1394 if (f->count < catchart->x_min)
1397 slices[n_slices] = *f;
1402 if (catchart->include_missing)
1404 for (i = 0; i < frq_tab->n_missing; i++)
1406 const struct freq *f = &frq_tab->missing[i];
1407 slices[n_slices].count += f->count;
1410 slices[n_slices].values[0] = f->values[0];
1413 if (frq_tab->n_missing > 0)
1417 *n_slicesp = n_slices;
1422 /* Allocate an array of struct freqs and fill them from the data in FRQ_TAB,
1423 according to the parameters of CATCHART
1424 N_SLICES will contain the number of slices allocated.
1425 The caller is responsible for freeing slices
1427 static struct freq **
1428 pick_cat_counts_ptr (const struct frq_chart *catchart,
1429 const struct freq_tab *frq_tab,
1434 struct freq **slices = xnmalloc (frq_tab->n_valid + frq_tab->n_missing, sizeof *slices);
1436 for (i = 0; i < frq_tab->n_valid; i++)
1438 struct freq *f = &frq_tab->valid[i];
1439 if (f->count > catchart->x_max)
1442 if (f->count < catchart->x_min)
1445 slices[n_slices] = f;
1450 if (catchart->include_missing)
1452 for (i = 0; i < frq_tab->n_missing; i++)
1454 const struct freq *f = &frq_tab->missing[i];
1457 slices[n_slices] = xmalloc (sizeof (struct freq));
1458 slices[n_slices]->values[0] = f->values[0];
1461 slices[n_slices]->count += f->count;
1466 *n_slicesp = n_slices;
1473 do_piechart(const struct frq_chart *pie, const struct variable *var,
1474 const struct freq_tab *frq_tab)
1477 struct freq *slices = pick_cat_counts (pie, frq_tab, &n_slices);
1480 msg (SW, _("Omitting pie chart for %s, which has only %d unique values."),
1481 var_get_name (var), n_slices);
1482 else if (n_slices > 50)
1483 msg (SW, _("Omitting pie chart for %s, which has over 50 unique values."),
1484 var_get_name (var));
1486 chart_item_submit (piechart_create (var, slices, n_slices));
1493 do_barchart(const struct frq_chart *bar, const struct variable **var,
1494 const struct freq_tab *frq_tab)
1497 struct freq **slices = pick_cat_counts_ptr (bar, frq_tab, &n_slices);
1499 chart_item_submit (barchart_create (var, 1,
1500 (bar->y_scale == FRQ_FREQ) ? _("Count") : _("Percent"),
1501 (bar->y_scale == FRQ_PERCENT),
1507 /* Calculates all the pertinent statistics for VF, putting them in array
1510 calc_stats (const struct var_freqs *vf, double d[FRQ_ST_count])
1512 const struct freq_tab *ft = &vf->tab;
1513 double W = ft->valid_cases;
1514 const struct freq *f;
1516 int most_often = -1;
1517 double X_mode = SYSMIS;
1519 /* Calculate the mode. */
1520 for (f = ft->valid; f < ft->missing; f++)
1522 if (most_often < f->count)
1524 most_often = f->count;
1525 X_mode = f->values[0].f;
1527 else if (most_often == f->count)
1529 /* A duplicate mode is undefined.
1530 FIXME: keep track of *all* the modes. */
1535 /* Calculate moments. */
1536 m = moments_create (MOMENT_KURTOSIS);
1537 for (f = ft->valid; f < ft->missing; f++)
1538 moments_pass_one (m, f->values[0].f, f->count);
1539 for (f = ft->valid; f < ft->missing; f++)
1540 moments_pass_two (m, f->values[0].f, f->count);
1541 moments_calculate (m, NULL, &d[FRQ_ST_MEAN], &d[FRQ_ST_VARIANCE],
1542 &d[FRQ_ST_SKEWNESS], &d[FRQ_ST_KURTOSIS]);
1543 moments_destroy (m);
1545 /* Formulae below are taken from _SPSS Statistical Algorithms_. */
1546 d[FRQ_ST_MINIMUM] = ft->valid[0].values[0].f;
1547 d[FRQ_ST_MAXIMUM] = ft->valid[ft->n_valid - 1].values[0].f;
1548 d[FRQ_ST_MODE] = X_mode;
1549 d[FRQ_ST_RANGE] = d[FRQ_ST_MAXIMUM] - d[FRQ_ST_MINIMUM];
1550 d[FRQ_ST_SUM] = d[FRQ_ST_MEAN] * W;
1551 d[FRQ_ST_STDDEV] = sqrt (d[FRQ_ST_VARIANCE]);
1552 d[FRQ_ST_SEMEAN] = d[FRQ_ST_STDDEV] / sqrt (W);
1553 d[FRQ_ST_SESKEWNESS] = calc_seskew (W);
1554 d[FRQ_ST_SEKURTOSIS] = calc_sekurt (W);
1557 /* Displays a table of all the statistics requested for variable V. */
1559 dump_statistics (const struct frq_proc *frq, const struct var_freqs *vf,
1560 const struct variable *wv)
1562 const struct fmt_spec *wfmt = wv ? var_get_print_format (wv) : &F_8_0;
1563 const struct freq_tab *ft = &vf->tab;
1564 double stat_value[FRQ_ST_count];
1565 struct tab_table *t;
1566 int i, r = 2; /* N missing and N valid are always dumped */
1568 if (var_is_alpha (vf->var))
1571 calc_stats (vf, stat_value);
1573 t = tab_create (3, ((frq->stats & BIT_INDEX (FRQ_ST_MEDIAN)) ? frq->n_stats - 1 : frq->n_stats)
1574 + frq->n_show_percentiles + 2);
1576 tab_set_format (t, RC_WEIGHT, wfmt);
1577 tab_box (t, TAL_1, TAL_1, -1, -1 , 0 , 0 , 2, tab_nr(t) - 1) ;
1579 tab_vline (t, TAL_1 , 2, 0, tab_nr(t) - 1);
1580 tab_vline (t, TAL_GAP , 1, 0, tab_nr(t) - 1 ) ;
1582 for (i = 0; i < FRQ_ST_count; i++)
1584 if (FRQ_ST_MEDIAN == i)
1587 if (frq->stats & BIT_INDEX (i))
1589 tab_text (t, 0, r, TAB_LEFT | TAT_TITLE,
1590 gettext (st_name[i]));
1592 if (vf->tab.n_valid <= 0 && r >= 2)
1593 tab_text (t, 2, r, 0, ".");
1595 tab_double (t, 2, r, TAB_NONE, stat_value[i], NULL, RC_OTHER);
1600 tab_text (t, 0, 0, TAB_LEFT | TAT_TITLE, _("N"));
1601 tab_text (t, 1, 0, TAB_LEFT | TAT_TITLE, _("Valid"));
1602 tab_text (t, 1, 1, TAB_LEFT | TAT_TITLE, _("Missing"));
1604 tab_double (t, 2, 0, TAB_NONE, ft->valid_cases, NULL, RC_WEIGHT);
1605 tab_double (t, 2, 1, TAB_NONE, ft->total_cases - ft->valid_cases, NULL, RC_WEIGHT);
1607 for (i = 0; i < frq->n_percentiles; i++)
1609 const struct percentile *pc = &frq->percentiles[i];
1616 tab_text (t, 0, r, TAB_LEFT | TAT_TITLE, _("Percentiles"));
1619 if (vf->tab.n_valid <= 0)
1621 tab_text (t, 2, r, 0, ".");
1627 tab_text (t, 1, r, TAB_LEFT, _("50 (Median)"));
1629 tab_double (t, 1, r, TAB_LEFT, pc->p * 100, NULL, RC_INTEGER);
1630 tab_double (t, 2, r, TAB_NONE, pc->value,
1631 var_get_print_format (vf->var), RC_OTHER);
1636 tab_title (t, "%s", var_to_string (vf->var));