e5a1ca5b19dda4c5964d53e4f0e346c55a1ac10e
[pspp] / src / language / stats / ctables.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 2021 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 <math.h>
20 #include <errno.h>
21
22 #include "data/casegrouper.h"
23 #include "data/casereader.h"
24 #include "data/casewriter.h"
25 #include "data/data-in.h"
26 #include "data/data-out.h"
27 #include "data/dataset.h"
28 #include "data/dictionary.h"
29 #include "data/mrset.h"
30 #include "data/subcase.h"
31 #include "data/value-labels.h"
32 #include "language/command.h"
33 #include "language/dictionary/split-file.h"
34 #include "language/lexer/format-parser.h"
35 #include "language/lexer/lexer.h"
36 #include "language/lexer/token.h"
37 #include "language/lexer/variable-parser.h"
38 #include "libpspp/array.h"
39 #include "libpspp/assertion.h"
40 #include "libpspp/hash-functions.h"
41 #include "libpspp/hmap.h"
42 #include "libpspp/i18n.h"
43 #include "libpspp/message.h"
44 #include "libpspp/string-array.h"
45 #include "math/mode.h"
46 #include "math/moments.h"
47 #include "math/percentiles.h"
48 #include "math/sort.h"
49 #include "output/pivot-table.h"
50
51 #include "gl/minmax.h"
52 #include "gl/xalloc.h"
53
54 #include "gettext.h"
55 #define _(msgid) gettext (msgid)
56 #define N_(msgid) (msgid)
57
58 enum ctables_vlabel
59   {
60     CTVL_NONE = SETTINGS_VALUE_SHOW_DEFAULT,
61     CTVL_NAME = SETTINGS_VALUE_SHOW_VALUE,
62     CTVL_LABEL = SETTINGS_VALUE_SHOW_LABEL,
63     CTVL_BOTH = SETTINGS_VALUE_SHOW_BOTH,
64   };
65
66 enum ctables_function_type
67   {
68     /* A function that operates on data in a single cell.  The function does
69        not have an unweighted version. */
70     CTFT_CELL,
71
72     /* A function that operates on data in a single cell.  The function has an
73        unweighted version. */
74     CTFT_UCELL,
75
76     /* A function that operates on an area of cells.  The function has an
77        unweighted version. */
78     CTFT_AREA,
79   };
80
81 enum ctables_format
82   {
83     CTF_COUNT,
84     CTF_PERCENT,
85     CTF_GENERAL
86   };
87
88 enum ctables_function_availability
89   {
90     CTFA_ALL,                /* Any variables. */
91     CTFA_SCALE,              /* Only scale variables, totals, and subtotals. */
92     //CTFA_MRSETS,             /* Only multiple-response sets */
93   };
94
95 enum ctables_summary_function
96   {
97 #define S(ENUM, NAME, TYPE, FORMAT, AVAILABILITY) ENUM,
98 #include "ctables.inc"
99 #undef S
100   };
101
102 enum {
103 #define S(ENUM, NAME, TYPE, FORMAT, AVAILABILITY) +1
104   N_CTSF_FUNCTIONS =
105 #include "ctables.inc"
106 #undef S
107 };
108
109 struct ctables_function_info
110   {
111     struct substring basename;
112     enum ctables_function_type type;
113     enum ctables_format format;
114     enum ctables_function_availability availability;
115
116     bool may_be_unweighted;
117     bool is_area;
118   };
119 static const struct ctables_function_info ctables_function_info[N_CTSF_FUNCTIONS] = {
120 #define S(ENUM, NAME, TYPE, FORMAT, AVAILABILITY)                       \
121   [ENUM] = {                                                            \
122     .basename = SS_LITERAL_INITIALIZER (NAME),                          \
123     .type = TYPE,                                                       \
124     .format = FORMAT,                                                   \
125     .availability = AVAILABILITY,                                       \
126     .may_be_unweighted = (TYPE) == CTFT_UCELL || (TYPE) == CTFT_AREA,   \
127     .is_area = (TYPE) == CTFT_AREA                                      \
128   },
129 #include "ctables.inc"
130 #undef S
131 };
132
133 static bool ctables_summary_function_is_count (enum ctables_summary_function);
134
135 enum ctables_area_type
136   {
137     /* Within a section, where stacked variables divide one section from
138        another.
139
140        Keep CTAT_LAYER after CTAT_LAYERROW and CTAT_LAYERCOL so that
141        parse_ctables_summary_function() parses correctly. */
142     CTAT_TABLE,                  /* All layers of a whole section. */
143     CTAT_LAYERROW,               /* Row in one layer within a section. */
144     CTAT_LAYERCOL,               /* Column in one layer within a section. */
145     CTAT_LAYER,                  /* One layer within a section. */
146
147     /* Within a subtable, where a subtable pairs an innermost row variable with
148        an innermost column variable within a single layer.  */
149     CTAT_SUBTABLE,               /* Whole subtable. */
150     CTAT_ROW,                    /* Row within a subtable. */
151     CTAT_COL,                    /* Column within a subtable. */
152 #define N_CTATS 7
153   };
154
155 static const char *ctables_area_type_name[N_CTATS] = {
156   [CTAT_TABLE] = "TABLE",
157   [CTAT_LAYER] = "LAYER",
158   [CTAT_LAYERROW] = "LAYERROW",
159   [CTAT_LAYERCOL] = "LAYERCOL",
160   [CTAT_SUBTABLE] = "SUBTABLE",
161   [CTAT_ROW] = "ROW",
162   [CTAT_COL] = "COL",
163 };
164
165 struct ctables_area
166   {
167     struct hmap_node node;
168
169     const struct ctables_cell *example;
170
171     size_t sequence;
172     double d_valid;             /* Dictionary weight. */
173     double d_count;
174     double d_total;
175     double e_valid;             /* Effective weight */
176     double e_count;
177     double e_total;
178     double u_valid;             /* Unweighted. */
179     double u_count;
180     double u_total;
181     struct ctables_sum *sums;
182   };
183
184 struct ctables_sum
185   {
186     double e_sum;
187     double u_sum;
188   };
189
190 enum ctables_summary_variant
191   {
192     CSV_CELL,
193     CSV_TOTAL
194 #define N_CSVS 2
195   };
196
197 struct ctables_cell
198   {
199     /* In struct ctables_section's 'cells' hmap.  Indexed by all the values in
200        all the axes (except the scalar variable, if any). */
201     struct hmap_node node;
202
203     /* The areas that contain this cell. */
204     uint32_t omit_areas;
205     struct ctables_area *areas[N_CTATS];
206
207     bool hide;
208
209     bool postcompute;
210     enum ctables_summary_variant sv;
211
212     struct ctables_cell_axis
213       {
214         struct ctables_cell_value
215           {
216             const struct ctables_category *category;
217             union value value;
218           }
219         *cvs;
220         int leaf;
221       }
222     axes[PIVOT_N_AXES];
223
224     union ctables_summary *summaries;
225
226     //char *name;
227   };
228
229 struct ctables
230   {
231     const struct dictionary *dict;
232     struct pivot_table_look *look;
233
234     /* CTABLES has a number of extra formats that we implement via custom
235        currency specifications on an alternate fmt_settings. */
236 #define CTEF_NEGPAREN FMT_CCA
237 #define CTEF_NEQUAL   FMT_CCB
238 #define CTEF_PAREN    FMT_CCC
239 #define CTEF_PCTPAREN FMT_CCD
240     struct fmt_settings ctables_formats;
241
242     /* If this is NULL, zeros are displayed using the normal print format.
243        Otherwise, this string is displayed. */
244     char *zero;
245
246     /* If this is NULL, missing values are displayed using the normal print
247        format.  Otherwise, this string is displayed. */
248     char *missing;
249
250     /* Indexed by variable dictionary index. */
251     enum ctables_vlabel *vlabels;
252
253     struct hmap postcomputes;   /* Contains "struct ctables_postcompute"s. */
254
255     bool mrsets_count_duplicates; /* MRSETS. */
256     bool smissing_listwise;       /* SMISSING. */
257     struct variable *e_weight;    /* WEIGHT. */
258     int hide_threshold;           /* HIDESMALLCOUNTS. */
259
260     struct ctables_table **tables;
261     size_t n_tables;
262   };
263
264 static struct ctables_postcompute *ctables_find_postcompute (struct ctables *,
265                                                              const char *name);
266
267 struct ctables_postcompute
268   {
269     struct hmap_node hmap_node; /* In struct ctables's 'pcompute' hmap. */
270     char *name;                 /* Name, without leading &. */
271
272     struct msg_location *location; /* Location of definition. */
273     struct ctables_pcexpr *expr;
274     char *label;
275     struct ctables_summary_spec_set *specs;
276     bool hide_source_cats;
277   };
278
279 struct ctables_pcexpr
280   {
281     /* Precedence table:
282
283        ()
284        **
285        -
286        * /
287        - +
288     */
289     enum ctables_postcompute_op
290       {
291         /* Terminals. */
292         CTPO_CONSTANT,          /* 5 */
293         CTPO_CAT_NUMBER,        /* [5] */
294         CTPO_CAT_STRING,        /* ["STRING"] */
295         CTPO_CAT_NRANGE,        /* [LO THRU 5] */
296         CTPO_CAT_SRANGE,        /* ["A" THRU "B"] */
297         CTPO_CAT_MISSING,       /* MISSING */
298         CTPO_CAT_OTHERNM,       /* OTHERNM */
299         CTPO_CAT_SUBTOTAL,      /* SUBTOTAL */
300         CTPO_CAT_TOTAL,         /* TOTAL */
301
302         /* Nonterminals. */
303         CTPO_ADD,
304         CTPO_SUB,
305         CTPO_MUL,
306         CTPO_DIV,
307         CTPO_POW,
308         CTPO_NEG,
309       }
310     op;
311
312     union
313       {
314         /* CTPO_CAT_NUMBER. */
315         double number;
316
317         /* CTPO_CAT_STRING, in dictionary encoding. */
318         struct substring string;
319
320         /* CTPO_CAT_NRANGE. */
321         double nrange[2];
322
323         /* CTPO_CAT_SRANGE. */
324         struct substring srange[2];
325
326         /* CTPO_CAT_SUBTOTAL. */
327         size_t subtotal_index;
328
329         /* Two elements: CTPO_ADD, CTPO_SUB, CTPO_MUL, CTPO_DIV, CTPO_POW.
330            One element: CTPO_NEG. */
331         struct ctables_pcexpr *subs[2];
332       };
333
334     /* Source location. */
335     struct msg_location *location;
336   };
337
338 static void ctables_pcexpr_destroy (struct ctables_pcexpr *);
339 static struct ctables_pcexpr *ctables_pcexpr_allocate_binary (
340   enum ctables_postcompute_op, struct ctables_pcexpr *sub0,
341   struct ctables_pcexpr *sub1);
342
343 struct ctables_summary_spec_set
344   {
345     struct ctables_summary_spec *specs;
346     size_t n;
347     size_t allocated;
348
349     /* The variable to which the summary specs are applied. */
350     struct variable *var;
351
352     /* Whether the variable to which the summary specs are applied is a scale
353        variable for the purpose of summarization.
354
355        (VALIDN and TOTALN act differently for summarizing scale and categorical
356        variables.) */
357     bool is_scale;
358
359     /* If any of these optional additional scale variables are missing, then
360        treat 'var' as if it's missing too.  This is for implementing
361        SMISSING=LISTWISE. */
362     struct variable **listwise_vars;
363     size_t n_listwise_vars;
364   };
365
366 static void ctables_summary_spec_set_clone (struct ctables_summary_spec_set *,
367                                             const struct ctables_summary_spec_set *);
368 static void ctables_summary_spec_set_uninit (struct ctables_summary_spec_set *);
369
370 /* A nested sequence of variables, e.g. a > b > c. */
371 struct ctables_nest
372   {
373     struct variable **vars;
374     size_t n;
375     size_t scale_idx;
376     size_t *areas[N_CTATS];
377     size_t n_areas[N_CTATS];
378     size_t group_head;
379
380     struct ctables_summary_spec_set specs[N_CSVS];
381   };
382
383 /* A stack of nestings, e.g. nest1 + nest2 + ... + nestN. */
384 struct ctables_stack
385   {
386     struct ctables_nest *nests;
387     size_t n;
388   };
389
390 static void ctables_stack_uninit (struct ctables_stack *);
391
392 struct ctables_value
393   {
394     struct hmap_node node;
395     union value value;
396     int leaf;
397   };
398
399 struct ctables_occurrence
400   {
401     struct hmap_node node;
402     union value value;
403   };
404
405 struct ctables_section
406   {
407     /* Settings. */
408     struct ctables_table *table;
409     struct ctables_nest *nests[PIVOT_N_AXES];
410
411     /* Data. */
412     struct hmap *occurrences[PIVOT_N_AXES]; /* "struct ctables_occurrence"s. */
413     struct hmap cells;            /* Contains "struct ctables_cell"s. */
414     struct hmap areas[N_CTATS];   /* Contains "struct ctables_area"s. */
415   };
416
417 static void ctables_section_uninit (struct ctables_section *);
418
419 struct ctables_table
420   {
421     struct ctables *ctables;
422     struct ctables_axis *axes[PIVOT_N_AXES];
423     struct ctables_stack stacks[PIVOT_N_AXES];
424     struct ctables_section *sections;
425     size_t n_sections;
426     enum pivot_axis_type summary_axis;
427     struct ctables_summary_spec_set summary_specs;
428     struct variable **sum_vars;
429     size_t n_sum_vars;
430
431     enum pivot_axis_type slabels_axis;
432     bool slabels_visible;
433
434     /* The innermost category labels for axis 'a' appear on axis label_axis[a].
435
436        Most commonly, label_axis[a] == a, and in particular we always have
437        label_axis{PIVOT_AXIS_LAYER] == PIVOT_AXIS_LAYER.
438
439        If ROWLABELS or COLLABELS is specified, then one of
440        label_axis[PIVOT_AXIS_ROW] or label_axis[PIVOT_AXIS_COLUMN] can be the
441        opposite axis or PIVOT_AXIS_LAYER.  Only one of them will differ.
442
443        If any category labels are moved, then 'clabels_example' is one of the
444        variables being moved (and it is otherwise NULL).  All of the variables
445        being moved have the same width, value labels, and categories, so this
446        example variable can be used to find those out.
447
448        The remaining members in this group are relevant only if category labels
449        are moved.
450
451        'clabels_values_map' holds a "struct ctables_value" for all the values
452        that appear in all of the variables in the moved categories.  It is
453        accumulated as the data is read.  Once the data is fully read, its
454        sorted values are put into 'clabels_values' and 'n_clabels_values'.
455     */
456     enum pivot_axis_type label_axis[PIVOT_N_AXES];
457     enum pivot_axis_type clabels_from_axis;
458     const struct variable *clabels_example;
459     struct hmap clabels_values_map;
460     struct ctables_value **clabels_values;
461     size_t n_clabels_values;
462
463     /* Indexed by variable dictionary index. */
464     struct ctables_categories **categories;
465     size_t n_categories;
466
467     double cilevel;
468
469     char *caption;
470     char *corner;
471     char *title;
472
473     struct ctables_chisq *chisq;
474     struct ctables_pairwise *pairwise;
475   };
476
477 struct ctables_categories
478   {
479     size_t n_refs;
480     struct ctables_category *cats;
481     size_t n_cats;
482     bool show_empty;
483   };
484
485 struct ctables_category
486   {
487     enum ctables_category_type
488       {
489         /* Explicit category lists. */
490         CCT_NUMBER,
491         CCT_STRING,
492         CCT_NRANGE,             /* Numerical range. */
493         CCT_SRANGE,             /* String range. */
494         CCT_MISSING,
495         CCT_OTHERNM,
496         CCT_POSTCOMPUTE,
497
498         /* Totals and subtotals. */
499         CCT_SUBTOTAL,
500         CCT_TOTAL,
501
502         /* Implicit category lists. */
503         CCT_VALUE,
504         CCT_LABEL,
505         CCT_FUNCTION,
506
507         /* For contributing to TOTALN. */
508         CCT_EXCLUDED_MISSING,
509       }
510     type;
511
512     struct ctables_category *subtotal;
513
514     bool hide;
515
516     union
517       {
518         double number;           /* CCT_NUMBER. */
519         struct substring string; /* CCT_STRING, in dictionary encoding. */
520         double nrange[2];        /* CCT_NRANGE. */
521         struct substring srange[2]; /* CCT_SRANGE. */
522
523         struct
524           {
525             char *total_label;      /* CCT_SUBTOTAL, CCT_TOTAL. */
526             bool hide_subcategories; /* CCT_SUBTOTAL. */
527           };
528
529         /* CCT_POSTCOMPUTE. */
530         struct
531           {
532             const struct ctables_postcompute *pc;
533             enum fmt_type parse_format;
534           };
535
536         /* CCT_VALUE, CCT_LABEL, CCT_FUNCTION. */
537         struct
538           {
539             bool include_missing;
540             bool sort_ascending;
541
542             /* CCT_FUNCTION. */
543             enum ctables_summary_function sort_function;
544             bool weighted;
545             enum ctables_area_type area;
546             struct variable *sort_var;
547             double percentile;
548           };
549       };
550
551     /* Source location.  This is null for CCT_TOTAL, CCT_VALUE, CCT_LABEL,
552        CCT_FUNCTION, CCT_EXCLUDED_MISSING. */
553     struct msg_location *location;
554   };
555
556 static void
557 ctables_category_uninit (struct ctables_category *cat)
558 {
559   if (!cat)
560     return;
561
562   msg_location_destroy (cat->location);
563   switch (cat->type)
564     {
565     case CCT_NUMBER:
566     case CCT_NRANGE:
567     case CCT_MISSING:
568     case CCT_OTHERNM:
569     case CCT_POSTCOMPUTE:
570       break;
571
572     case CCT_STRING:
573       ss_dealloc (&cat->string);
574       break;
575
576     case CCT_SRANGE:
577       ss_dealloc (&cat->srange[0]);
578       ss_dealloc (&cat->srange[1]);
579       break;
580
581     case CCT_SUBTOTAL:
582     case CCT_TOTAL:
583       free (cat->total_label);
584       break;
585
586     case CCT_VALUE:
587     case CCT_LABEL:
588     case CCT_FUNCTION:
589       break;
590
591     case CCT_EXCLUDED_MISSING:
592       break;
593     }
594 }
595
596 static bool
597 nullable_substring_equal (const struct substring *a,
598                           const struct substring *b)
599 {
600   return !a->string ? !b->string : b->string && ss_equals (*a, *b);
601 }
602
603 static bool
604 ctables_category_equal (const struct ctables_category *a,
605                         const struct ctables_category *b)
606 {
607   if (a->type != b->type)
608     return false;
609
610   switch (a->type)
611     {
612     case CCT_NUMBER:
613       return a->number == b->number;
614
615     case CCT_STRING:
616       return ss_equals (a->string, b->string);
617
618     case CCT_NRANGE:
619       return a->nrange[0] == b->nrange[0] && a->nrange[1] == b->nrange[1];
620
621     case CCT_SRANGE:
622       return (nullable_substring_equal (&a->srange[0], &b->srange[0])
623               && nullable_substring_equal (&a->srange[1], &b->srange[1]));
624
625     case CCT_MISSING:
626     case CCT_OTHERNM:
627       return true;
628
629     case CCT_POSTCOMPUTE:
630       return a->pc == b->pc;
631
632     case CCT_SUBTOTAL:
633     case CCT_TOTAL:
634       return !strcmp (a->total_label, b->total_label);
635
636     case CCT_VALUE:
637     case CCT_LABEL:
638     case CCT_FUNCTION:
639       return (a->include_missing == b->include_missing
640               && a->sort_ascending == b->sort_ascending
641               && a->sort_function == b->sort_function
642               && a->sort_var == b->sort_var
643               && a->percentile == b->percentile);
644
645     case CCT_EXCLUDED_MISSING:
646       return true;
647     }
648
649   NOT_REACHED ();
650 }
651
652 static void
653 ctables_categories_unref (struct ctables_categories *c)
654 {
655   if (!c)
656     return;
657
658   assert (c->n_refs > 0);
659   if (--c->n_refs)
660     return;
661
662   for (size_t i = 0; i < c->n_cats; i++)
663     ctables_category_uninit (&c->cats[i]);
664   free (c->cats);
665   free (c);
666 }
667
668 static bool
669 ctables_categories_equal (const struct ctables_categories *a,
670                           const struct ctables_categories *b)
671 {
672   if (a->n_cats != b->n_cats || a->show_empty != b->show_empty)
673     return false;
674
675   for (size_t i = 0; i < a->n_cats; i++)
676     if (!ctables_category_equal (&a->cats[i], &b->cats[i]))
677       return false;
678
679   return true;
680 }
681
682 /* Chi-square test (SIGTEST). */
683 struct ctables_chisq
684   {
685     double alpha;
686     bool include_mrsets;
687     bool all_visible;
688   };
689
690 /* Pairwise comparison test (COMPARETEST). */
691 struct ctables_pairwise
692   {
693     enum { PROP, MEAN } type;
694     double alpha[2];
695     bool include_mrsets;
696     bool meansvariance_allcats;
697     bool all_visible;
698     enum { BONFERRONI = 1, BH } adjust;
699     bool merge;
700     bool apa_style;
701     bool show_sig;
702   };
703
704 struct ctables_axis
705   {
706     enum ctables_axis_op
707       {
708         /* Terminals. */
709         CTAO_VAR,
710
711         /* Nonterminals. */
712         CTAO_STACK,             /* + */
713         CTAO_NEST,              /* > */
714       }
715     op;
716
717     union
718       {
719         /* Terminals. */
720         struct
721           {
722             struct variable *var;
723             bool scale;
724             struct ctables_summary_spec_set specs[N_CSVS];
725           };
726
727         /* Nonterminals. */
728         struct ctables_axis *subs[2];
729       };
730
731     struct msg_location *loc;
732   };
733
734 static void ctables_axis_destroy (struct ctables_axis *);
735
736 struct ctables_summary_spec
737   {
738     /* The calculation to be performed.
739
740        'function' is the function to calculate.  'weighted' specifies whether
741        to use weighted or unweighted data (for functions that do not support a
742        choice, it must be true).  'calc_area' is the area over which the
743        calculation takes place (for functions that target only an individual
744        cell, it must be 0).  For CTSF_PTILE only, 'percentile' is the
745        percentile between 0 and 100 (for other functions it must be 0). */
746     enum ctables_summary_function function;
747     bool weighted;
748     enum ctables_area_type calc_area;
749     double percentile;          /* CTSF_PTILE only. */
750
751     /* How to display the result of the calculation.
752
753        'label' is a user-specified label, NULL if the user didn't specify
754        one.
755
756        'user_area' is usually the same as 'calc_area', but when category labels
757        are rotated from one axis to another it swaps rows and columns.
758
759        'format' is the format for displaying the output.  If
760        'is_ctables_format' is true, then 'format.type' is one of the special
761        CTEF_* formats instead of the standard ones. */
762     char *label;
763     enum ctables_area_type user_area;
764     struct fmt_spec format;
765     bool is_ctables_format;       /* Is 'format' one of CTEF_*? */
766
767     size_t axis_idx;
768     size_t sum_var_idx;
769   };
770
771 static void
772 ctables_summary_spec_clone (struct ctables_summary_spec *dst,
773                             const struct ctables_summary_spec *src)
774 {
775   *dst = *src;
776   dst->label = xstrdup_if_nonnull (src->label);
777 }
778
779 static void
780 ctables_summary_spec_uninit (struct ctables_summary_spec *s)
781 {
782   if (s)
783     free (s->label);
784 }
785
786 static void
787 ctables_summary_spec_set_clone (struct ctables_summary_spec_set *dst,
788                                 const struct ctables_summary_spec_set *src)
789 {
790   struct ctables_summary_spec *specs
791     = (src->n ? xnmalloc (src->n, sizeof *specs) : NULL);
792   for (size_t i = 0; i < src->n; i++)
793     ctables_summary_spec_clone (&specs[i], &src->specs[i]);
794
795   *dst = (struct ctables_summary_spec_set) {
796     .specs = specs,
797     .n = src->n,
798     .allocated = src->n,
799     .var = src->var,
800     .is_scale = src->is_scale,
801   };
802 }
803
804 static void
805 ctables_summary_spec_set_uninit (struct ctables_summary_spec_set *set)
806 {
807   for (size_t i = 0; i < set->n; i++)
808     ctables_summary_spec_uninit (&set->specs[i]);
809   free (set->listwise_vars);
810   free (set->specs);
811 }
812
813 static bool
814 parse_col_width (struct lexer *lexer, const char *name, double *width)
815 {
816   lex_match (lexer, T_EQUALS);
817   if (lex_match_id (lexer, "DEFAULT"))
818     *width = SYSMIS;
819   else if (lex_force_num_range_closed (lexer, name, 0, DBL_MAX))
820     {
821       *width = lex_number (lexer);
822       lex_get (lexer);
823     }
824   else
825     return false;
826
827   return true;
828 }
829
830 static bool
831 parse_bool (struct lexer *lexer, bool *b)
832 {
833   if (lex_match_id (lexer, "NO"))
834     *b = false;
835   else if (lex_match_id (lexer, "YES"))
836     *b = true;
837   else
838     {
839       lex_error_expecting (lexer, "YES", "NO");
840       return false;
841     }
842   return true;
843 }
844
845 static enum ctables_function_availability
846 ctables_function_availability (enum ctables_summary_function f)
847 {
848   static enum ctables_function_availability availability[] = {
849 #define S(ENUM, NAME, TYPE, FORMAT, AVAILABILITY) [ENUM] = AVAILABILITY,
850 #include "ctables.inc"
851 #undef S
852   };
853
854   return availability[f];
855 }
856
857 static bool
858 ctables_summary_function_is_count (enum ctables_summary_function f)
859 {
860   return f == CTSF_COUNT || f == CTSF_ECOUNT;
861 }
862
863 static bool
864 parse_ctables_summary_function (struct lexer *lexer,
865                                 enum ctables_summary_function *function,
866                                 bool *weighted,
867                                 enum ctables_area_type *area)
868 {
869   if (!lex_force_id (lexer))
870     return false;
871
872   struct substring name = lex_tokss (lexer);
873   *weighted = !(ss_match_byte (&name, 'U') || ss_match_byte (&name, 'u'));
874
875   bool has_area = false;
876   *area = 0;
877   for (enum ctables_area_type at = 0; at < N_CTATS; at++)
878     if (ss_match_string_case (&name, ss_cstr (ctables_area_type_name[at])))
879       {
880         has_area = true;
881         *area = at;
882
883         if (ss_equals_case (name, ss_cstr ("PCT")))
884           {
885             /* Special case where .COUNT suffix is omitted. */
886             *function = CTSF_areaPCT_COUNT;
887             lex_get (lexer);
888             return true;
889           }
890         break;
891       }
892
893   for (int f = 0; f < N_CTSF_FUNCTIONS; f++)
894     {
895       const struct ctables_function_info *cfi = &ctables_function_info[f];
896       if (ss_equals_case (cfi->basename, name))
897         {
898           *function = f;
899           if (!*weighted && !cfi->may_be_unweighted)
900             break;
901           if (has_area != cfi->is_area)
902             break;
903
904           lex_get (lexer);
905           return true;
906         }
907     }
908
909   lex_error (lexer, _("Expecting summary function name."));
910   return false;
911 }
912
913 static void
914 ctables_axis_destroy (struct ctables_axis *axis)
915 {
916   if (!axis)
917     return;
918
919   switch (axis->op)
920     {
921     case CTAO_VAR:
922       for (size_t i = 0; i < N_CSVS; i++)
923         ctables_summary_spec_set_uninit (&axis->specs[i]);
924       break;
925
926     case CTAO_STACK:
927     case CTAO_NEST:
928       ctables_axis_destroy (axis->subs[0]);
929       ctables_axis_destroy (axis->subs[1]);
930       break;
931     }
932   msg_location_destroy (axis->loc);
933   free (axis);
934 }
935
936 static struct ctables_axis *
937 ctables_axis_new_nonterminal (enum ctables_axis_op op,
938                               struct ctables_axis *sub0,
939                               struct ctables_axis *sub1,
940                               struct lexer *lexer, int start_ofs)
941 {
942   struct ctables_axis *axis = xmalloc (sizeof *axis);
943   *axis = (struct ctables_axis) {
944     .op = op,
945     .subs = { sub0, sub1 },
946     .loc = lex_ofs_location (lexer, start_ofs, lex_ofs (lexer) - 1),
947   };
948   return axis;
949 }
950
951 struct ctables_axis_parse_ctx
952   {
953     struct lexer *lexer;
954     struct dictionary *dict;
955     struct ctables *ct;
956     struct ctables_table *t;
957   };
958
959 static struct fmt_spec
960 ctables_summary_default_format (enum ctables_summary_function function,
961                                 const struct variable *var)
962 {
963   static const enum ctables_format default_formats[] = {
964 #define S(ENUM, NAME, TYPE, FORMAT, AVAILABILITY) [ENUM] = FORMAT,
965 #include "ctables.inc"
966 #undef S
967   };
968   switch (default_formats[function])
969     {
970     case CTF_COUNT:
971       return (struct fmt_spec) { .type = FMT_F, .w = 40 };
972
973     case CTF_PERCENT:
974       return (struct fmt_spec) { .type = FMT_PCT, .w = 40, .d = 1 };
975
976     case CTF_GENERAL:
977       return *var_get_print_format (var);
978
979     default:
980       NOT_REACHED ();
981     }
982 }
983
984 static const char *
985 ctables_summary_label__ (const struct ctables_summary_spec *spec)
986 {
987   bool w = spec->weighted;
988   enum ctables_area_type a = spec->user_area;
989   switch (spec->function)
990     {
991     case CTSF_COUNT:
992       return w ? N_("Count") : N_("Unweighted Count");
993
994     case CTSF_ECOUNT:
995       return N_("Adjusted Count");
996
997     case CTSF_areaPCT_COUNT:
998       switch (a)
999         {
1000         case CTAT_TABLE: return w ? N_("Table %") : N_("Unweighted Table %");
1001         case CTAT_LAYER: return w ? N_("Layer %") : N_("Unweighted Layer %");
1002         case CTAT_LAYERROW: return w ? N_("Layer Row %") : N_("Unweighted Layer Row %");
1003         case CTAT_LAYERCOL: return w ? N_("Layer Column %") : N_("Unweighted Layer Column %");
1004         case CTAT_SUBTABLE: return w ? N_("Subtable %") : N_("Unweighted Subtable %");
1005         case CTAT_ROW: return w ? N_("Row %") : N_("Unweighted Row %");
1006         case CTAT_COL: return w ? N_("Column %") : N_("Unweighted Column %");
1007         }
1008       NOT_REACHED ();
1009
1010     case CTSF_areaPCT_VALIDN:
1011       switch (a)
1012         {
1013         case CTAT_TABLE: return w ? N_("Table Valid N %") : N_("Unweighted Table Valid N %");
1014         case CTAT_LAYER: return w ? N_("Layer Valid N %") : N_("Unweighted Layer Valid N %");
1015         case CTAT_LAYERROW: return w ? N_("Layer Row Valid N %") : N_("Unweighted Layer Row Valid N %");
1016         case CTAT_LAYERCOL: return w ? N_("Layer Column Valid N %") : N_("Unweighted Layer Column Valid N %");
1017         case CTAT_SUBTABLE: return w ? N_("Subtable Valid N %") : N_("Unweighted Subtable Valid N %");
1018         case CTAT_ROW: return w ? N_("Row Valid N %") : N_("Unweighted Row Valid N %");
1019         case CTAT_COL: return w ? N_("Column Valid N %") : N_("Unweighted Column Valid N %");
1020         }
1021       NOT_REACHED ();
1022
1023     case CTSF_areaPCT_TOTALN:
1024       switch (a)
1025         {
1026         case CTAT_TABLE: return w ? N_("Table Total N %") : N_("Unweighted Table Total N %");
1027         case CTAT_LAYER: return w ? N_("Layer Total N %") : N_("Unweighted Layer Total N %");
1028         case CTAT_LAYERROW: return w ? N_("Layer Row Total N %") : N_("Unweighted Layer Row Total N %");
1029         case CTAT_LAYERCOL: return w ? N_("Layer Column Total N %") : N_("Unweighted Layer Column Total N %");
1030         case CTAT_SUBTABLE: return w ? N_("Subtable Total N %") : N_("Unweighted Subtable Total N %");
1031         case CTAT_ROW: return w ? N_("Row Total N %") : N_("Unweighted Row Total N %");
1032         case CTAT_COL: return w ? N_("Column Total N %") : N_("Unweighted Column Total N %");
1033         }
1034       NOT_REACHED ();
1035
1036     case CTSF_MAXIMUM: return N_("Maximum");
1037     case CTSF_MEAN: return w ? N_("Mean") : N_("Unweighted Mean");
1038     case CTSF_MEDIAN: return N_("Median");
1039     case CTSF_MINIMUM: return N_("Minimum");
1040     case CTSF_MISSING: return N_("Missing");
1041     case CTSF_MODE: return N_("Mode");
1042     case CTSF_PTILE: NOT_REACHED ();
1043     case CTSF_RANGE: return N_("Range");
1044     case CTSF_SEMEAN: return N_("Std Error of Mean");
1045     case CTSF_STDDEV: return N_("Std Deviation");
1046     case CTSF_SUM: return N_("Sum");
1047     case CTSF_TOTALN: return N_("Total N");
1048     case CTSF_ETOTALN: return N_("Adjusted Total N");
1049     case CTSF_VALIDN: return N_("Valid N");
1050     case CTSF_EVALIDN: return N_("Adjusted Valid N");
1051     case CTSF_VARIANCE: return N_("Variance");
1052     case CTSF_areaPCT_SUM:
1053       switch (a)
1054         {
1055         case CTAT_TABLE: return w ? N_("Table Sum %") : N_("Unweighted Table Sum %");
1056         case CTAT_LAYER: return w ? N_("Layer Sum %") : N_("Unweighted Layer Sum %");
1057         case CTAT_LAYERROW: return w ? N_("Layer Row Sum %") : N_("Unweighted Layer Row Sum %");
1058         case CTAT_LAYERCOL: return w ? N_("Layer Column Sum %") : N_("Unweighted Layer Column Sum %");
1059         case CTAT_SUBTABLE: return w ? N_("Subtable Sum %") : N_("Unweighted Subtable Sum %");
1060         case CTAT_ROW: return w ? N_("Row Sum %") : N_("Unweighted Row Sum %");
1061         case CTAT_COL: return w ? N_("Column Sum %") : N_("Unweighted Column Sum %");
1062         }
1063       NOT_REACHED ();
1064
1065     case CTSF_areaID:
1066       switch (a)
1067         {
1068         /* Don't bother translating these: they are for developers only. */
1069         case CTAT_TABLE: return "Table ID";
1070         case CTAT_LAYER: return "Layer ID";
1071         case CTAT_LAYERROW: return "Layer Row ID";
1072         case CTAT_LAYERCOL: return "Layer Column ID";
1073         case CTAT_SUBTABLE: return "Subtable ID";
1074         case CTAT_ROW: return "Row ID";
1075         case CTAT_COL: return "Column ID";
1076         }
1077       NOT_REACHED ();
1078     }
1079
1080   NOT_REACHED ();
1081 }
1082
1083 static struct pivot_value *
1084 ctables_summary_label (const struct ctables_summary_spec *spec, double cilevel)
1085 {
1086   if (!spec->label)
1087     {
1088       if (spec->function == CTSF_PTILE)
1089         {
1090           double p = spec->percentile;
1091           char *s = (spec->weighted
1092                      ? xasprintf (_("Percentile %.2f"), p)
1093                      : xasprintf (_("Unweighted Percentile %.2f"), p));
1094           return pivot_value_new_user_text_nocopy (s);
1095         }
1096       else
1097         return pivot_value_new_text (ctables_summary_label__ (spec));
1098     }
1099   else
1100     {
1101       struct substring in = ss_cstr (spec->label);
1102       struct substring target = ss_cstr (")CILEVEL");
1103
1104       struct string out = DS_EMPTY_INITIALIZER;
1105       for (;;)
1106         {
1107           size_t chunk = ss_find_substring (in, target);
1108           ds_put_substring (&out, ss_head (in, chunk));
1109           ss_advance (&in, chunk);
1110           if (!in.length)
1111             return pivot_value_new_user_text_nocopy (ds_steal_cstr (&out));
1112
1113           ss_advance (&in, target.length);
1114           ds_put_format (&out, "%g", cilevel);
1115         }
1116     }
1117 }
1118
1119 static const char *
1120 ctables_summary_function_name (enum ctables_summary_function function,
1121                                bool weighted,
1122                                enum ctables_area_type area,
1123                                char *buffer, size_t bufsize)
1124 {
1125   const struct ctables_function_info *cfi = &ctables_function_info[function];
1126   snprintf (buffer, bufsize, "%s%s%s",
1127             weighted ? "" : "U",
1128             cfi->is_area ? ctables_area_type_name[area] : "",
1129             cfi->basename.string);
1130   return buffer;
1131 }
1132
1133 static bool
1134 add_summary_spec (struct ctables_axis *axis,
1135                   enum ctables_summary_function function, bool weighted,
1136                   enum ctables_area_type area, double percentile,
1137                   const char *label, const struct fmt_spec *format,
1138                   bool is_ctables_format, const struct msg_location *loc,
1139                   enum ctables_summary_variant sv)
1140 {
1141   if (axis->op == CTAO_VAR)
1142     {
1143       char function_name[128];
1144       ctables_summary_function_name (function, weighted, area,
1145                                      function_name, sizeof function_name);
1146       const char *var_name = var_get_name (axis->var);
1147       switch (ctables_function_availability (function))
1148         {
1149 #if 0
1150         case CTFA_MRSETS:
1151           msg_at (SE, loc, _("Summary function %s applies only to multiple "
1152                              "response sets."), function_name);
1153           msg_at (SN, axis->loc, _("'%s' is not a multiple response set."),
1154                   var_name);
1155           return false;
1156 #endif
1157
1158         case CTFA_SCALE:
1159           if (!axis->scale && sv != CSV_TOTAL)
1160             {
1161               msg_at (SE, loc,
1162                       _("Summary function %s applies only to scale variables."),
1163                       function_name);
1164               msg_at (SN, axis->loc, _("'%s' is not a scale variable."),
1165                       var_name);
1166               return false;
1167             }
1168           break;
1169
1170         case CTFA_ALL:
1171           break;
1172         }
1173
1174       struct ctables_summary_spec_set *set = &axis->specs[sv];
1175       if (set->n >= set->allocated)
1176         set->specs = x2nrealloc (set->specs, &set->allocated,
1177                                  sizeof *set->specs);
1178
1179       struct ctables_summary_spec *dst = &set->specs[set->n++];
1180       *dst = (struct ctables_summary_spec) {
1181         .function = function,
1182         .weighted = weighted,
1183         .calc_area = area,
1184         .user_area = area,
1185         .percentile = percentile,
1186         .label = xstrdup_if_nonnull (label),
1187         .format = (format ? *format
1188                    : ctables_summary_default_format (function, axis->var)),
1189         .is_ctables_format = is_ctables_format,
1190       };
1191       return true;
1192     }
1193   else
1194     {
1195       for (size_t i = 0; i < 2; i++)
1196         if (!add_summary_spec (axis->subs[i], function, weighted, area,
1197                                percentile, label, format, is_ctables_format,
1198                                loc, sv))
1199           return false;
1200       return true;
1201     }
1202 }
1203
1204 static struct ctables_axis *ctables_axis_parse_stack (
1205   struct ctables_axis_parse_ctx *);
1206
1207
1208 static struct ctables_axis *
1209 ctables_axis_parse_primary (struct ctables_axis_parse_ctx *ctx)
1210 {
1211   if (lex_match (ctx->lexer, T_LPAREN))
1212     {
1213       struct ctables_axis *sub = ctables_axis_parse_stack (ctx);
1214       if (!sub || !lex_force_match (ctx->lexer, T_RPAREN))
1215         {
1216           ctables_axis_destroy (sub);
1217           return NULL;
1218         }
1219       return sub;
1220     }
1221
1222   if (!lex_force_id (ctx->lexer))
1223     return NULL;
1224
1225   int start_ofs = lex_ofs (ctx->lexer);
1226   struct variable *var = parse_variable (ctx->lexer, ctx->dict);
1227   if (!var)
1228     return NULL;
1229
1230   struct ctables_axis *axis = xmalloc (sizeof *axis);
1231   *axis = (struct ctables_axis) { .op = CTAO_VAR, .var = var };
1232
1233   axis->scale = (lex_match_phrase (ctx->lexer, "[S]") ? true
1234                  : lex_match_phrase (ctx->lexer, "[C]") ? false
1235                  : var_get_measure (var) == MEASURE_SCALE);
1236   axis->loc = lex_ofs_location (ctx->lexer, start_ofs,
1237                                 lex_ofs (ctx->lexer) - 1);
1238   if (axis->scale && var_is_alpha (var))
1239     {
1240       msg_at (SE, axis->loc, _("Cannot use string variable %s as a scale "
1241                                "variable."),
1242               var_get_name (var));
1243       ctables_axis_destroy (axis);
1244       return NULL;
1245     }
1246
1247   return axis;
1248 }
1249
1250 static bool
1251 has_digit (const char *s)
1252 {
1253   return s[strcspn (s, "0123456789")] != '\0';
1254 }
1255
1256 static bool
1257 parse_ctables_format_specifier (struct lexer *lexer, struct fmt_spec *format,
1258                                 bool *is_ctables_format)
1259 {
1260   char type[FMT_TYPE_LEN_MAX + 1];
1261   if (!parse_abstract_format_specifier__ (lexer, type, &format->w, &format->d))
1262     return false;
1263
1264   if (!strcasecmp (type, "NEGPAREN"))
1265     format->type = CTEF_NEGPAREN;
1266   else if (!strcasecmp (type, "NEQUAL"))
1267     format->type = CTEF_NEQUAL;
1268   else if (!strcasecmp (type, "PAREN"))
1269     format->type = CTEF_PAREN;
1270   else if (!strcasecmp (type, "PCTPAREN"))
1271     format->type = CTEF_PCTPAREN;
1272   else
1273     {
1274       *is_ctables_format = false;
1275       return (parse_format_specifier (lexer, format)
1276               && fmt_check_output (format)
1277               && fmt_check_type_compat (format, VAL_NUMERIC));
1278     }
1279
1280   lex_get (lexer);
1281   if (format->w < 2)
1282     {
1283       lex_next_error (lexer, -1, -1,
1284                       _("Output format %s requires width 2 or greater."), type);
1285       return false;
1286     }
1287   else if (format->d > format->w - 1)
1288     {
1289       lex_next_error (lexer, -1, -1, _("Output format %s requires width "
1290                                        "greater than decimals."), type);
1291       return false;
1292     }
1293   else
1294     {
1295       *is_ctables_format = true;
1296       return true;
1297     }
1298 }
1299
1300 static struct ctables_axis *
1301 ctables_axis_parse_postfix (struct ctables_axis_parse_ctx *ctx)
1302 {
1303   struct ctables_axis *sub = ctables_axis_parse_primary (ctx);
1304   if (!sub || !lex_match (ctx->lexer, T_LBRACK))
1305     return sub;
1306
1307   enum ctables_summary_variant sv = CSV_CELL;
1308   for (;;)
1309     {
1310       int start_ofs = lex_ofs (ctx->lexer);
1311
1312       /* Parse function. */
1313       enum ctables_summary_function function;
1314       bool weighted;
1315       enum ctables_area_type area;
1316       if (!parse_ctables_summary_function (ctx->lexer, &function, &weighted,
1317                                            &area))
1318         goto error;
1319
1320       /* Parse percentile. */
1321       double percentile = 0;
1322       if (function == CTSF_PTILE)
1323         {
1324           if (!lex_force_num_range_closed (ctx->lexer, "PTILE", 0, 100))
1325             goto error;
1326           percentile = lex_number (ctx->lexer);
1327           lex_get (ctx->lexer);
1328         }
1329
1330       /* Parse label. */
1331       char *label = NULL;
1332       if (lex_is_string (ctx->lexer))
1333         {
1334           label = ss_xstrdup (lex_tokss (ctx->lexer));
1335           lex_get (ctx->lexer);
1336         }
1337
1338       /* Parse format. */
1339       struct fmt_spec format;
1340       const struct fmt_spec *formatp;
1341       bool is_ctables_format = false;
1342       if (lex_token (ctx->lexer) == T_ID
1343           && has_digit (lex_tokcstr (ctx->lexer)))
1344         {
1345           if (!parse_ctables_format_specifier (ctx->lexer, &format,
1346                                                &is_ctables_format))
1347             {
1348               free (label);
1349               goto error;
1350             }
1351           formatp = &format;
1352         }
1353       else
1354         formatp = NULL;
1355
1356       struct msg_location *loc = lex_ofs_location (ctx->lexer, start_ofs,
1357                                                    lex_ofs (ctx->lexer) - 1);
1358       add_summary_spec (sub, function, weighted, area, percentile, label,
1359                         formatp, is_ctables_format, loc, sv);
1360       free (label);
1361       msg_location_destroy (loc);
1362
1363       lex_match (ctx->lexer, T_COMMA);
1364       if (sv == CSV_CELL && lex_match_id (ctx->lexer, "TOTALS"))
1365         {
1366           if (!lex_force_match (ctx->lexer, T_LBRACK))
1367             goto error;
1368           sv = CSV_TOTAL;
1369         }
1370       else if (lex_match (ctx->lexer, T_RBRACK))
1371         {
1372           if (sv == CSV_TOTAL && !lex_force_match (ctx->lexer, T_RBRACK))
1373             goto error;
1374           return sub;
1375         }
1376     }
1377
1378 error:
1379   ctables_axis_destroy (sub);
1380   return NULL;
1381 }
1382
1383 static const struct ctables_axis *
1384 find_scale (const struct ctables_axis *axis)
1385 {
1386   if (!axis)
1387     return NULL;
1388   else if (axis->op == CTAO_VAR)
1389     return axis->scale ? axis : NULL;
1390   else
1391     {
1392       for (size_t i = 0; i < 2; i++)
1393         {
1394           const struct ctables_axis *scale = find_scale (axis->subs[i]);
1395           if (scale)
1396             return scale;
1397         }
1398       return NULL;
1399     }
1400 }
1401
1402 static const struct ctables_axis *
1403 find_categorical_summary_spec (const struct ctables_axis *axis)
1404 {
1405   if (!axis)
1406     return NULL;
1407   else if (axis->op == CTAO_VAR)
1408     return !axis->scale && axis->specs[CSV_CELL].n ? axis : NULL;
1409   else
1410     {
1411       for (size_t i = 0; i < 2; i++)
1412         {
1413           const struct ctables_axis *sum
1414             = find_categorical_summary_spec (axis->subs[i]);
1415           if (sum)
1416             return sum;
1417         }
1418       return NULL;
1419     }
1420 }
1421
1422 static struct ctables_axis *
1423 ctables_axis_parse_nest (struct ctables_axis_parse_ctx *ctx)
1424 {
1425   int start_ofs = lex_ofs (ctx->lexer);
1426   struct ctables_axis *lhs = ctables_axis_parse_postfix (ctx);
1427   if (!lhs)
1428     return NULL;
1429
1430   while (lex_match (ctx->lexer, T_GT))
1431     {
1432       struct ctables_axis *rhs = ctables_axis_parse_postfix (ctx);
1433       if (!rhs)
1434         {
1435           ctables_axis_destroy (lhs);
1436           return NULL;
1437         }
1438
1439       struct ctables_axis *nest = ctables_axis_new_nonterminal (
1440         CTAO_NEST, lhs, rhs, ctx->lexer, start_ofs);
1441
1442       const struct ctables_axis *outer_scale = find_scale (lhs);
1443       const struct ctables_axis *inner_scale = find_scale (rhs);
1444       if (outer_scale && inner_scale)
1445         {
1446           msg_at (SE, nest->loc, _("Cannot nest scale variables."));
1447           msg_at (SN, outer_scale->loc, _("This is an outer scale variable."));
1448           msg_at (SN, inner_scale->loc, _("This is an inner scale variable."));
1449           ctables_axis_destroy (nest);
1450           return NULL;
1451         }
1452
1453       const struct ctables_axis *outer_sum = find_categorical_summary_spec (lhs);
1454       if (outer_sum)
1455         {
1456           msg_at (SE, nest->loc,
1457                   _("Summaries may only be requested for categorical variables "
1458                     "at the innermost nesting level."));
1459           msg_at (SN, outer_sum->loc,
1460                   _("This outer categorical variable has a summary."));
1461           ctables_axis_destroy (nest);
1462           return NULL;
1463         }
1464
1465       lhs = nest;
1466     }
1467
1468   return lhs;
1469 }
1470
1471 static struct ctables_axis *
1472 ctables_axis_parse_stack (struct ctables_axis_parse_ctx *ctx)
1473 {
1474   int start_ofs = lex_ofs (ctx->lexer);
1475   struct ctables_axis *lhs = ctables_axis_parse_nest (ctx);
1476   if (!lhs)
1477     return NULL;
1478
1479   while (lex_match (ctx->lexer, T_PLUS))
1480     {
1481       struct ctables_axis *rhs = ctables_axis_parse_nest (ctx);
1482       if (!rhs)
1483         {
1484           ctables_axis_destroy (lhs);
1485           return NULL;
1486         }
1487
1488       lhs = ctables_axis_new_nonterminal (CTAO_STACK, lhs, rhs,
1489                                           ctx->lexer, start_ofs);
1490     }
1491
1492   return lhs;
1493 }
1494
1495 static bool
1496 ctables_axis_parse (struct lexer *lexer, struct dictionary *dict,
1497                     struct ctables *ct, struct ctables_table *t,
1498                     enum pivot_axis_type a)
1499 {
1500   if (lex_token (lexer) == T_BY
1501       || lex_token (lexer) == T_SLASH
1502       || lex_token (lexer) == T_ENDCMD)
1503     return true;
1504
1505   struct ctables_axis_parse_ctx ctx = {
1506     .lexer = lexer,
1507     .dict = dict,
1508     .ct = ct,
1509     .t = t
1510   };
1511   t->axes[a] = ctables_axis_parse_stack (&ctx);
1512   return t->axes[a] != NULL;
1513 }
1514
1515 static void
1516 ctables_chisq_destroy (struct ctables_chisq *chisq)
1517 {
1518   free (chisq);
1519 }
1520
1521 static void
1522 ctables_pairwise_destroy (struct ctables_pairwise *pairwise)
1523 {
1524   free (pairwise);
1525 }
1526
1527 static void
1528 ctables_table_destroy (struct ctables_table *t)
1529 {
1530   if (!t)
1531     return;
1532
1533   for (size_t i = 0; i < t->n_sections; i++)
1534     ctables_section_uninit (&t->sections[i]);
1535   free (t->sections);
1536
1537   for (size_t i = 0; i < t->n_categories; i++)
1538     ctables_categories_unref (t->categories[i]);
1539   free (t->categories);
1540
1541   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
1542     {
1543       ctables_axis_destroy (t->axes[a]);
1544       ctables_stack_uninit (&t->stacks[a]);
1545     }
1546   free (t->summary_specs.specs);
1547
1548   struct ctables_value *ctv, *next_ctv;
1549   HMAP_FOR_EACH_SAFE (ctv, next_ctv, struct ctables_value, node,
1550                       &t->clabels_values_map)
1551     {
1552       value_destroy (&ctv->value, var_get_width (t->clabels_example));
1553       hmap_delete (&t->clabels_values_map, &ctv->node);
1554       free (ctv);
1555     }
1556   hmap_destroy (&t->clabels_values_map);
1557   free (t->clabels_values);
1558
1559   free (t->sum_vars);
1560   free (t->caption);
1561   free (t->corner);
1562   free (t->title);
1563   ctables_chisq_destroy (t->chisq);
1564   ctables_pairwise_destroy (t->pairwise);
1565   free (t);
1566 }
1567
1568 static void
1569 ctables_destroy (struct ctables *ct)
1570 {
1571   if (!ct)
1572     return;
1573
1574   struct ctables_postcompute *pc, *next_pc;
1575   HMAP_FOR_EACH_SAFE (pc, next_pc, struct ctables_postcompute, hmap_node,
1576                       &ct->postcomputes)
1577     {
1578       free (pc->name);
1579       msg_location_destroy (pc->location);
1580       ctables_pcexpr_destroy (pc->expr);
1581       free (pc->label);
1582       if (pc->specs)
1583         {
1584           ctables_summary_spec_set_uninit (pc->specs);
1585           free (pc->specs);
1586         }
1587       hmap_delete (&ct->postcomputes, &pc->hmap_node);
1588       free (pc);
1589     }
1590   hmap_destroy (&ct->postcomputes);
1591
1592   fmt_settings_uninit (&ct->ctables_formats);
1593   pivot_table_look_unref (ct->look);
1594   free (ct->zero);
1595   free (ct->missing);
1596   free (ct->vlabels);
1597   for (size_t i = 0; i < ct->n_tables; i++)
1598     ctables_table_destroy (ct->tables[i]);
1599   free (ct->tables);
1600   free (ct);
1601 }
1602
1603 static struct ctables_category
1604 cct_nrange (double low, double high)
1605 {
1606   return (struct ctables_category) {
1607     .type = CCT_NRANGE,
1608     .nrange = { low, high }
1609   };
1610 }
1611
1612 static struct ctables_category
1613 cct_srange (struct substring low, struct substring high)
1614 {
1615   return (struct ctables_category) {
1616     .type = CCT_SRANGE,
1617     .srange = { low, high }
1618   };
1619 }
1620
1621 static bool
1622 ctables_table_parse_subtotal (struct lexer *lexer, bool hide_subcategories,
1623                               struct ctables_category *cat)
1624 {
1625   char *total_label;
1626   if (lex_match (lexer, T_EQUALS))
1627     {
1628       if (!lex_force_string (lexer))
1629         return false;
1630
1631       total_label = ss_xstrdup (lex_tokss (lexer));
1632       lex_get (lexer);
1633     }
1634   else
1635     total_label = xstrdup (_("Subtotal"));
1636
1637   *cat = (struct ctables_category) {
1638     .type = CCT_SUBTOTAL,
1639     .hide_subcategories = hide_subcategories,
1640     .total_label = total_label
1641   };
1642   return true;
1643 }
1644
1645 static struct substring
1646 parse_substring (struct lexer *lexer, struct dictionary *dict)
1647 {
1648   struct substring s = recode_substring_pool (
1649     dict_get_encoding (dict), "UTF-8", lex_tokss (lexer), NULL);
1650   ss_rtrim (&s, ss_cstr (" "));
1651   lex_get (lexer);
1652   return s;
1653 }
1654
1655 static bool
1656 ctables_table_parse_explicit_category (struct lexer *lexer,
1657                                        struct dictionary *dict,
1658                                        struct ctables *ct,
1659                                        struct ctables_category *cat)
1660 {
1661   if (lex_match_id (lexer, "OTHERNM"))
1662     *cat = (struct ctables_category) { .type = CCT_OTHERNM };
1663   else if (lex_match_id (lexer, "MISSING"))
1664     *cat = (struct ctables_category) { .type = CCT_MISSING };
1665   else if (lex_match_id (lexer, "SUBTOTAL"))
1666     return ctables_table_parse_subtotal (lexer, false, cat);
1667   else if (lex_match_id (lexer, "HSUBTOTAL"))
1668     return ctables_table_parse_subtotal (lexer, true, cat);
1669   else if (lex_match_id (lexer, "LO"))
1670     {
1671       if (!lex_force_match_id (lexer, "THRU"))
1672         return false;
1673       if (lex_is_string (lexer))
1674         {
1675           struct substring sr0 = { .string = NULL };
1676           struct substring sr1 = parse_substring (lexer, dict);
1677           *cat = cct_srange (sr0, sr1);
1678         }
1679       else if (lex_force_num (lexer))
1680         {
1681           *cat = cct_nrange (-DBL_MAX, lex_number (lexer));
1682           lex_get (lexer);
1683         }
1684       else
1685         return false;
1686     }
1687   else if (lex_is_number (lexer))
1688     {
1689       double number = lex_number (lexer);
1690       lex_get (lexer);
1691       if (lex_match_id (lexer, "THRU"))
1692         {
1693           if (lex_match_id (lexer, "HI"))
1694             *cat = cct_nrange (number, DBL_MAX);
1695           else
1696             {
1697               if (!lex_force_num (lexer))
1698                 return false;
1699               *cat = cct_nrange (number, lex_number (lexer));
1700               lex_get (lexer);
1701             }
1702         }
1703       else
1704         *cat = (struct ctables_category) {
1705           .type = CCT_NUMBER,
1706           .number = number
1707         };
1708     }
1709   else if (lex_is_string (lexer))
1710     {
1711       struct substring s = parse_substring (lexer, dict);
1712       if (lex_match_id (lexer, "THRU"))
1713         {
1714           if (lex_match_id (lexer, "HI"))
1715             {
1716               struct substring sr1 = { .string = NULL };
1717               *cat = cct_srange (s, sr1);
1718             }
1719           else
1720             {
1721               if (!lex_force_string (lexer))
1722                 {
1723                   ss_dealloc (&s);
1724                   return false;
1725                 }
1726               struct substring sr1 = parse_substring (lexer, dict);
1727               *cat = cct_srange (s, sr1);
1728             }
1729         }
1730       else
1731         *cat = (struct ctables_category) { .type = CCT_STRING, .string = s };
1732     }
1733   else if (lex_match (lexer, T_AND))
1734     {
1735       if (!lex_force_id (lexer))
1736         return false;
1737       struct ctables_postcompute *pc = ctables_find_postcompute (
1738         ct, lex_tokcstr (lexer));
1739       if (!pc)
1740         {
1741           struct msg_location *loc = lex_get_location (lexer, -1, 0);
1742           msg_at (SE, loc, _("Unknown postcompute &%s."),
1743                   lex_tokcstr (lexer));
1744           msg_location_destroy (loc);
1745           return false;
1746         }
1747       lex_get (lexer);
1748
1749       *cat = (struct ctables_category) { .type = CCT_POSTCOMPUTE, .pc = pc };
1750     }
1751   else
1752     {
1753       lex_error (lexer, NULL);
1754       return false;
1755     }
1756
1757   return true;
1758 }
1759
1760 static bool
1761 parse_category_string (struct msg_location *location,
1762                        struct substring s, const struct dictionary *dict,
1763                        enum fmt_type format, double *n)
1764 {
1765   union value v;
1766   char *error = data_in (s, dict_get_encoding (dict), format,
1767                          settings_get_fmt_settings (), &v, 0, NULL);
1768   if (error)
1769     {
1770       msg_at (SE, location,
1771               _("Failed to parse category specification as format %s: %s."),
1772               fmt_name (format), error);
1773       free (error);
1774       return false;
1775     }
1776
1777   *n = v.f;
1778   return true;
1779 }
1780
1781 static struct ctables_category *
1782 ctables_find_category_for_postcompute__ (const struct ctables_categories *cats,
1783                                          const struct ctables_pcexpr *e)
1784 {
1785   struct ctables_category *best = NULL;
1786   size_t n_subtotals = 0;
1787   for (size_t i = 0; i < cats->n_cats; i++)
1788     {
1789       struct ctables_category *cat = &cats->cats[i];
1790       switch (e->op)
1791         {
1792         case CTPO_CAT_NUMBER:
1793           if (cat->type == CCT_NUMBER && cat->number == e->number)
1794             best = cat;
1795           break;
1796
1797         case CTPO_CAT_STRING:
1798           if (cat->type == CCT_STRING && ss_equals (cat->string, e->string))
1799             best = cat;
1800           break;
1801
1802         case CTPO_CAT_NRANGE:
1803           if (cat->type == CCT_NRANGE
1804               && cat->nrange[0] == e->nrange[0]
1805               && cat->nrange[1] == e->nrange[1])
1806             best = cat;
1807           break;
1808
1809         case CTPO_CAT_SRANGE:
1810           if (cat->type == CCT_SRANGE
1811               && nullable_substring_equal (&cat->srange[0], &e->srange[0])
1812               && nullable_substring_equal (&cat->srange[1], &e->srange[1]))
1813             best = cat;
1814           break;
1815
1816         case CTPO_CAT_MISSING:
1817           if (cat->type == CCT_MISSING)
1818             best = cat;
1819           break;
1820
1821         case CTPO_CAT_OTHERNM:
1822           if (cat->type == CCT_OTHERNM)
1823             best = cat;
1824           break;
1825
1826         case CTPO_CAT_SUBTOTAL:
1827           if (cat->type == CCT_SUBTOTAL)
1828             {
1829               n_subtotals++;
1830               if (e->subtotal_index == n_subtotals)
1831                 return cat;
1832               else if (e->subtotal_index == 0)
1833                 best = cat;
1834             }
1835           break;
1836
1837         case CTPO_CAT_TOTAL:
1838           if (cat->type == CCT_TOTAL)
1839             return cat;
1840           break;
1841
1842         case CTPO_CONSTANT:
1843         case CTPO_ADD:
1844         case CTPO_SUB:
1845         case CTPO_MUL:
1846         case CTPO_DIV:
1847         case CTPO_POW:
1848         case CTPO_NEG:
1849           NOT_REACHED ();
1850         }
1851     }
1852   if (e->op == CTPO_CAT_SUBTOTAL && e->subtotal_index == 0 && n_subtotals > 1)
1853     return NULL;
1854   return best;
1855 }
1856
1857 static struct ctables_category *
1858 ctables_find_category_for_postcompute (const struct dictionary *dict,
1859                                        const struct ctables_categories *cats,
1860                                        enum fmt_type parse_format,
1861                                        const struct ctables_pcexpr *e)
1862 {
1863   if (parse_format != FMT_F)
1864     {
1865       if (e->op == CTPO_CAT_STRING)
1866         {
1867           double number;
1868           if (!parse_category_string (e->location, e->string, dict,
1869                                       parse_format, &number))
1870             return NULL;
1871
1872           struct ctables_pcexpr e2 = {
1873             .op = CTPO_CAT_NUMBER,
1874             .number = number,
1875             .location = e->location,
1876           };
1877           return ctables_find_category_for_postcompute__ (cats, &e2);
1878         }
1879       else if (e->op == CTPO_CAT_SRANGE)
1880         {
1881           double nrange[2];
1882           if (!e->srange[0].string)
1883             nrange[0] = -DBL_MAX;
1884           else if (!parse_category_string (e->location, e->srange[0], dict,
1885                                            parse_format, &nrange[0]))
1886             return NULL;
1887
1888           if (!e->srange[1].string)
1889             nrange[1] = DBL_MAX;
1890           else if (!parse_category_string (e->location, e->srange[1], dict,
1891                                            parse_format, &nrange[1]))
1892             return NULL;
1893
1894           struct ctables_pcexpr e2 = {
1895             .op = CTPO_CAT_NRANGE,
1896             .nrange = { nrange[0], nrange[1] },
1897             .location = e->location,
1898           };
1899           return ctables_find_category_for_postcompute__ (cats, &e2);
1900         }
1901     }
1902   return ctables_find_category_for_postcompute__ (cats, e);
1903 }
1904
1905 static bool
1906 ctables_recursive_check_postcompute (struct dictionary *dict,
1907                                      const struct ctables_pcexpr *e,
1908                                      struct ctables_category *pc_cat,
1909                                      const struct ctables_categories *cats,
1910                                      const struct msg_location *cats_location)
1911 {
1912   switch (e->op)
1913     {
1914     case CTPO_CAT_NUMBER:
1915     case CTPO_CAT_STRING:
1916     case CTPO_CAT_NRANGE:
1917     case CTPO_CAT_SRANGE:
1918     case CTPO_CAT_MISSING:
1919     case CTPO_CAT_OTHERNM:
1920     case CTPO_CAT_SUBTOTAL:
1921     case CTPO_CAT_TOTAL:
1922       {
1923         struct ctables_category *cat = ctables_find_category_for_postcompute (
1924           dict, cats, pc_cat->parse_format, e);
1925         if (!cat)
1926           {
1927             if (e->op == CTPO_CAT_SUBTOTAL && e->subtotal_index == 0)
1928               {
1929                 size_t n_subtotals = 0;
1930                 for (size_t i = 0; i < cats->n_cats; i++)
1931                   n_subtotals += cats->cats[i].type == CCT_SUBTOTAL;
1932                 if (n_subtotals > 1)
1933                   {
1934                     msg_at (SE, cats_location,
1935                             ngettext ("These categories include %zu instance "
1936                                       "of SUBTOTAL or HSUBTOTAL, so references "
1937                                       "from computed categories must refer to "
1938                                       "subtotals by position, "
1939                                       "e.g. SUBTOTAL[1].",
1940                                       "These categories include %zu instances "
1941                                       "of SUBTOTAL or HSUBTOTAL, so references "
1942                                       "from computed categories must refer to "
1943                                       "subtotals by position, "
1944                                       "e.g. SUBTOTAL[1].",
1945                                       n_subtotals),
1946                             n_subtotals);
1947                     msg_at (SN, e->location,
1948                             _("This is the reference that lacks a position."));
1949                     return NULL;
1950                   }
1951               }
1952
1953             msg_at (SE, pc_cat->location,
1954                     _("Computed category &%s references a category not included "
1955                       "in the category list."),
1956                     pc_cat->pc->name);
1957             msg_at (SN, e->location, _("This is the missing category."));
1958             if (e->op == CTPO_CAT_SUBTOTAL)
1959               msg_at (SN, cats_location,
1960                       _("To fix the problem, add subtotals to the "
1961                         "list of categories here."));
1962             else if (e->op == CTPO_CAT_TOTAL)
1963               msg (SN, _("To fix the problem, add TOTAL=YES to the variable's "
1964                          "CATEGORIES specification."));
1965             else
1966               msg_at (SN, cats_location,
1967                       _("To fix the problem, add the missing category to the "
1968                         "list of categories here."));
1969             return false;
1970           }
1971         if (pc_cat->pc->hide_source_cats)
1972           cat->hide = true;
1973         return true;
1974       }
1975
1976     case CTPO_CONSTANT:
1977       return true;
1978
1979     case CTPO_ADD:
1980     case CTPO_SUB:
1981     case CTPO_MUL:
1982     case CTPO_DIV:
1983     case CTPO_POW:
1984     case CTPO_NEG:
1985       for (size_t i = 0; i < 2; i++)
1986         if (e->subs[i] && !ctables_recursive_check_postcompute (
1987               dict, e->subs[i], pc_cat, cats, cats_location))
1988           return false;
1989       return true;
1990     }
1991
1992   NOT_REACHED ();
1993 }
1994
1995 static bool
1996 all_strings (struct variable **vars, size_t n_vars,
1997              const struct ctables_category *cat)
1998 {
1999   for (size_t j = 0; j < n_vars; j++)
2000     if (var_is_numeric (vars[j]))
2001       {
2002         msg_at (SE, cat->location,
2003                 _("This category specification may be applied only to string "
2004                   "variables, but this subcommand tries to apply it to "
2005                   "numeric variable %s."),
2006                 var_get_name (vars[j]));
2007         return false;
2008       }
2009   return true;
2010 }
2011
2012 static bool
2013 ctables_table_parse_categories (struct lexer *lexer, struct dictionary *dict,
2014                                 struct ctables *ct, struct ctables_table *t)
2015 {
2016   if (!lex_match_id (lexer, "VARIABLES"))
2017     return false;
2018   lex_match (lexer, T_EQUALS);
2019
2020   struct variable **vars;
2021   size_t n_vars;
2022   if (!parse_variables (lexer, dict, &vars, &n_vars, PV_NO_SCRATCH))
2023     return false;
2024
2025   const struct fmt_spec *common_format = var_get_print_format (vars[0]);
2026   for (size_t i = 1; i < n_vars; i++)
2027     {
2028       const struct fmt_spec *f = var_get_print_format (vars[i]);
2029       if (f->type != common_format->type)
2030         {
2031           common_format = NULL;
2032           break;
2033         }
2034     }
2035   bool parse_strings
2036     = (common_format
2037        && (fmt_get_category (common_format->type)
2038            & (FMT_CAT_DATE | FMT_CAT_TIME | FMT_CAT_DATE_COMPONENT)));
2039
2040   struct ctables_categories *c = xmalloc (sizeof *c);
2041   *c = (struct ctables_categories) { .n_refs = n_vars, .show_empty = true };
2042   for (size_t i = 0; i < n_vars; i++)
2043     {
2044       struct ctables_categories **cp
2045         = &t->categories[var_get_dict_index (vars[i])];
2046       ctables_categories_unref (*cp);
2047       *cp = c;
2048     }
2049
2050   size_t allocated_cats = 0;
2051   int cats_start_ofs = -1;
2052   int cats_end_ofs = -1;
2053   if (lex_match (lexer, T_LBRACK))
2054     {
2055       cats_start_ofs = lex_ofs (lexer);
2056       do
2057         {
2058           if (c->n_cats >= allocated_cats)
2059             c->cats = x2nrealloc (c->cats, &allocated_cats, sizeof *c->cats);
2060
2061           int start_ofs = lex_ofs (lexer);
2062           struct ctables_category *cat = &c->cats[c->n_cats];
2063           if (!ctables_table_parse_explicit_category (lexer, dict, ct, cat))
2064             goto error;
2065           cat->location = lex_ofs_location (lexer, start_ofs, lex_ofs (lexer) - 1);
2066           c->n_cats++;
2067
2068           lex_match (lexer, T_COMMA);
2069         }
2070       while (!lex_match (lexer, T_RBRACK));
2071       cats_end_ofs = lex_ofs (lexer) - 1;
2072     }
2073
2074   struct ctables_category cat = {
2075     .type = CCT_VALUE,
2076     .include_missing = false,
2077     .sort_ascending = true,
2078   };
2079   bool show_totals = false;
2080   char *total_label = NULL;
2081   bool totals_before = false;
2082   while (lex_token (lexer) != T_SLASH && lex_token (lexer) != T_ENDCMD)
2083     {
2084       if (!c->n_cats && lex_match_id (lexer, "ORDER"))
2085         {
2086           lex_match (lexer, T_EQUALS);
2087           if (lex_match_id (lexer, "A"))
2088             cat.sort_ascending = true;
2089           else if (lex_match_id (lexer, "D"))
2090             cat.sort_ascending = false;
2091           else
2092             {
2093               lex_error_expecting (lexer, "A", "D");
2094               goto error;
2095             }
2096         }
2097       else if (!c->n_cats && lex_match_id (lexer, "KEY"))
2098         {
2099           lex_match (lexer, T_EQUALS);
2100           if (lex_match_id (lexer, "VALUE"))
2101             cat.type = CCT_VALUE;
2102           else if (lex_match_id (lexer, "LABEL"))
2103             cat.type = CCT_LABEL;
2104           else
2105             {
2106               cat.type = CCT_FUNCTION;
2107               if (!parse_ctables_summary_function (lexer, &cat.sort_function,
2108                                                    &cat.weighted, &cat.area))
2109                 goto error;
2110
2111               if (lex_match (lexer, T_LPAREN))
2112                 {
2113                   cat.sort_var = parse_variable (lexer, dict);
2114                   if (!cat.sort_var)
2115                     goto error;
2116
2117                   if (cat.sort_function == CTSF_PTILE)
2118                     {
2119                       lex_match (lexer, T_COMMA);
2120                       if (!lex_force_num_range_closed (lexer, "PTILE", 0, 100))
2121                         goto error;
2122                       cat.percentile = lex_number (lexer);
2123                       lex_get (lexer);
2124                     }
2125
2126                   if (!lex_force_match (lexer, T_RPAREN))
2127                     goto error;
2128                 }
2129               else if (ctables_function_availability (cat.sort_function)
2130                        == CTFA_SCALE)
2131                 {
2132                   bool UNUSED b = lex_force_match (lexer, T_LPAREN);
2133                   goto error;
2134                 }
2135             }
2136         }
2137       else if (!c->n_cats && lex_match_id (lexer, "MISSING"))
2138         {
2139           lex_match (lexer, T_EQUALS);
2140           if (lex_match_id (lexer, "INCLUDE"))
2141             cat.include_missing = true;
2142           else if (lex_match_id (lexer, "EXCLUDE"))
2143             cat.include_missing = false;
2144           else
2145             {
2146               lex_error_expecting (lexer, "INCLUDE", "EXCLUDE");
2147               goto error;
2148             }
2149         }
2150       else if (lex_match_id (lexer, "TOTAL"))
2151         {
2152           lex_match (lexer, T_EQUALS);
2153           if (!parse_bool (lexer, &show_totals))
2154             goto error;
2155         }
2156       else if (lex_match_id (lexer, "LABEL"))
2157         {
2158           lex_match (lexer, T_EQUALS);
2159           if (!lex_force_string (lexer))
2160             goto error;
2161           free (total_label);
2162           total_label = ss_xstrdup (lex_tokss (lexer));
2163           lex_get (lexer);
2164         }
2165       else if (lex_match_id (lexer, "POSITION"))
2166         {
2167           lex_match (lexer, T_EQUALS);
2168           if (lex_match_id (lexer, "BEFORE"))
2169             totals_before = true;
2170           else if (lex_match_id (lexer, "AFTER"))
2171             totals_before = false;
2172           else
2173             {
2174               lex_error_expecting (lexer, "BEFORE", "AFTER");
2175               goto error;
2176             }
2177         }
2178       else if (lex_match_id (lexer, "EMPTY"))
2179         {
2180           lex_match (lexer, T_EQUALS);
2181           if (lex_match_id (lexer, "INCLUDE"))
2182             c->show_empty = true;
2183           else if (lex_match_id (lexer, "EXCLUDE"))
2184             c->show_empty = false;
2185           else
2186             {
2187               lex_error_expecting (lexer, "INCLUDE", "EXCLUDE");
2188               goto error;
2189             }
2190         }
2191       else
2192         {
2193           if (!c->n_cats)
2194             lex_error_expecting (lexer, "ORDER", "KEY", "MISSING",
2195                                  "TOTAL", "LABEL", "POSITION", "EMPTY");
2196           else
2197             lex_error_expecting (lexer, "TOTAL", "LABEL", "POSITION", "EMPTY");
2198           goto error;
2199         }
2200     }
2201
2202   if (!c->n_cats)
2203     {
2204       if (c->n_cats >= allocated_cats)
2205         c->cats = x2nrealloc (c->cats, &allocated_cats, sizeof *c->cats);
2206       c->cats[c->n_cats++] = cat;
2207     }
2208
2209   if (show_totals)
2210     {
2211       if (c->n_cats >= allocated_cats)
2212         c->cats = x2nrealloc (c->cats, &allocated_cats, sizeof *c->cats);
2213
2214       struct ctables_category *totals;
2215       if (totals_before)
2216         {
2217           insert_element (c->cats, c->n_cats, sizeof *c->cats, 0);
2218           totals = &c->cats[0];
2219         }
2220       else
2221         totals = &c->cats[c->n_cats];
2222       c->n_cats++;
2223
2224       *totals = (struct ctables_category) {
2225         .type = CCT_TOTAL,
2226         .total_label = total_label ? total_label : xstrdup (_("Total")),
2227       };
2228     }
2229
2230   struct ctables_category *subtotal = NULL;
2231   for (size_t i = totals_before ? 0 : c->n_cats;
2232        totals_before ? i < c->n_cats : i-- > 0;
2233        totals_before ? i++ : 0)
2234     {
2235       struct ctables_category *cat = &c->cats[i];
2236       switch (cat->type)
2237         {
2238         case CCT_NUMBER:
2239         case CCT_STRING:
2240         case CCT_NRANGE:
2241         case CCT_SRANGE:
2242         case CCT_MISSING:
2243         case CCT_OTHERNM:
2244           cat->subtotal = subtotal;
2245           break;
2246
2247         case CCT_POSTCOMPUTE:
2248           break;
2249
2250         case CCT_SUBTOTAL:
2251           subtotal = cat;
2252           break;
2253
2254         case CCT_TOTAL:
2255         case CCT_VALUE:
2256         case CCT_LABEL:
2257         case CCT_FUNCTION:
2258         case CCT_EXCLUDED_MISSING:
2259           break;
2260         }
2261     }
2262
2263   if (cats_start_ofs != -1)
2264     {
2265       for (size_t i = 0; i < c->n_cats; i++)
2266         {
2267           struct ctables_category *cat = &c->cats[i];
2268           switch (cat->type)
2269             {
2270             case CCT_POSTCOMPUTE:
2271               cat->parse_format = parse_strings ? common_format->type : FMT_F;
2272               struct msg_location *cats_location
2273                 = lex_ofs_location (lexer, cats_start_ofs, cats_end_ofs);
2274               bool ok = ctables_recursive_check_postcompute (
2275                 dict, cat->pc->expr, cat, c, cats_location);
2276               msg_location_destroy (cats_location);
2277               if (!ok)
2278                 goto error;
2279               break;
2280
2281             case CCT_NUMBER:
2282             case CCT_NRANGE:
2283               for (size_t j = 0; j < n_vars; j++)
2284                 if (var_is_alpha (vars[j]))
2285                   {
2286                     msg_at (SE, cat->location,
2287                             _("This category specification may be applied "
2288                               "only to numeric variables, but this "
2289                               "subcommand tries to apply it to string "
2290                               "variable %s."),
2291                             var_get_name (vars[j]));
2292                     goto error;
2293                   }
2294               break;
2295
2296             case CCT_STRING:
2297               if (parse_strings)
2298                 {
2299                   double n;
2300                   if (!parse_category_string (cat->location, cat->string, dict,
2301                                               common_format->type, &n))
2302                     goto error;
2303
2304                   ss_dealloc (&cat->string);
2305
2306                   cat->type = CCT_NUMBER;
2307                   cat->number = n;
2308                 }
2309               else if (!all_strings (vars, n_vars, cat))
2310                 goto error;
2311               break;
2312
2313             case CCT_SRANGE:
2314               if (parse_strings)
2315                 {
2316                   double n[2];
2317
2318                   if (!cat->srange[0].string)
2319                     n[0] = -DBL_MAX;
2320                   else if (!parse_category_string (cat->location,
2321                                                    cat->srange[0], dict,
2322                                                    common_format->type, &n[0]))
2323                     goto error;
2324
2325                   if (!cat->srange[1].string)
2326                     n[1] = DBL_MAX;
2327                   else if (!parse_category_string (cat->location,
2328                                                    cat->srange[1], dict,
2329                                                    common_format->type, &n[1]))
2330                     goto error;
2331
2332                   ss_dealloc (&cat->srange[0]);
2333                   ss_dealloc (&cat->srange[1]);
2334
2335                   cat->type = CCT_NRANGE;
2336                   cat->nrange[0] = n[0];
2337                   cat->nrange[1] = n[1];
2338                 }
2339               else if (!all_strings (vars, n_vars, cat))
2340                 goto error;
2341               break;
2342
2343             case CCT_MISSING:
2344             case CCT_OTHERNM:
2345             case CCT_SUBTOTAL:
2346             case CCT_TOTAL:
2347             case CCT_VALUE:
2348             case CCT_LABEL:
2349             case CCT_FUNCTION:
2350             case CCT_EXCLUDED_MISSING:
2351               break;
2352             }
2353         }
2354     }
2355
2356   free (vars);
2357   return true;
2358
2359 error:
2360   free (vars);
2361   return false;
2362 }
2363
2364 static void
2365 ctables_nest_uninit (struct ctables_nest *nest)
2366 {
2367   free (nest->vars);
2368   for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
2369     ctables_summary_spec_set_uninit (&nest->specs[sv]);
2370   for (enum ctables_area_type at = 0; at < N_CTATS; at++)
2371     free (nest->areas[at]);
2372 }
2373
2374 static void
2375 ctables_stack_uninit (struct ctables_stack *stack)
2376 {
2377   if (stack)
2378     {
2379       for (size_t i = 0; i < stack->n; i++)
2380         ctables_nest_uninit (&stack->nests[i]);
2381       free (stack->nests);
2382     }
2383 }
2384
2385 static struct ctables_stack
2386 nest_fts (struct ctables_stack s0, struct ctables_stack s1)
2387 {
2388   if (!s0.n)
2389     return s1;
2390   else if (!s1.n)
2391     return s0;
2392
2393   struct ctables_stack stack = { .nests = xnmalloc (s0.n, s1.n * sizeof *stack.nests) };
2394   for (size_t i = 0; i < s0.n; i++)
2395     for (size_t j = 0; j < s1.n; j++)
2396       {
2397         const struct ctables_nest *a = &s0.nests[i];
2398         const struct ctables_nest *b = &s1.nests[j];
2399
2400         size_t allocate = a->n + b->n;
2401         struct variable **vars = xnmalloc (allocate, sizeof *vars);
2402         size_t n = 0;
2403         for (size_t k = 0; k < a->n; k++)
2404           vars[n++] = a->vars[k];
2405         for (size_t k = 0; k < b->n; k++)
2406           vars[n++] = b->vars[k];
2407         assert (n == allocate);
2408
2409         const struct ctables_nest *summary_src;
2410         if (!a->specs[CSV_CELL].var)
2411           summary_src = b;
2412         else if (!b->specs[CSV_CELL].var)
2413           summary_src = a;
2414         else
2415           NOT_REACHED ();
2416
2417         struct ctables_nest *new = &stack.nests[stack.n++];
2418         *new = (struct ctables_nest) {
2419           .vars = vars,
2420           .scale_idx = (a->scale_idx != SIZE_MAX ? a->scale_idx
2421                         : b->scale_idx != SIZE_MAX ? a->n + b->scale_idx
2422                         : SIZE_MAX),
2423           .n = n,
2424         };
2425         for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
2426           ctables_summary_spec_set_clone (&new->specs[sv], &summary_src->specs[sv]);
2427       }
2428   ctables_stack_uninit (&s0);
2429   ctables_stack_uninit (&s1);
2430   return stack;
2431 }
2432
2433 static struct ctables_stack
2434 stack_fts (struct ctables_stack s0, struct ctables_stack s1)
2435 {
2436   struct ctables_stack stack = { .nests = xnmalloc (s0.n + s1.n, sizeof *stack.nests) };
2437   for (size_t i = 0; i < s0.n; i++)
2438     stack.nests[stack.n++] = s0.nests[i];
2439   for (size_t i = 0; i < s1.n; i++)
2440     {
2441       stack.nests[stack.n] = s1.nests[i];
2442       stack.nests[stack.n].group_head += s0.n;
2443       stack.n++;
2444     }
2445   assert (stack.n == s0.n + s1.n);
2446   free (s0.nests);
2447   free (s1.nests);
2448   return stack;
2449 }
2450
2451 static struct ctables_stack
2452 var_fts (const struct ctables_axis *a)
2453 {
2454   struct variable **vars = xmalloc (sizeof *vars);
2455   *vars = a->var;
2456
2457   struct ctables_nest *nest = xmalloc (sizeof *nest);
2458   *nest = (struct ctables_nest) {
2459     .vars = vars,
2460     .n = 1,
2461     .scale_idx = a->scale ? 0 : SIZE_MAX,
2462   };
2463   if (a->specs[CSV_CELL].n || a->scale)
2464     for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
2465       {
2466         ctables_summary_spec_set_clone (&nest->specs[sv], &a->specs[sv]);
2467         nest->specs[sv].var = a->var;
2468         nest->specs[sv].is_scale = a->scale;
2469       }
2470   return (struct ctables_stack) { .nests = nest, .n = 1 };
2471 }
2472
2473 static struct ctables_stack
2474 enumerate_fts (enum pivot_axis_type axis_type, const struct ctables_axis *a)
2475 {
2476   if (!a)
2477     return (struct ctables_stack) { .n = 0 };
2478
2479   switch (a->op)
2480     {
2481     case CTAO_VAR:
2482       return var_fts (a);
2483
2484     case CTAO_STACK:
2485       return stack_fts (enumerate_fts (axis_type, a->subs[0]),
2486                         enumerate_fts (axis_type, a->subs[1]));
2487
2488     case CTAO_NEST:
2489       /* This should consider any of the scale variables found in the result to
2490          be linked to each other listwise for SMISSING=LISTWISE. */
2491       return nest_fts (enumerate_fts (axis_type, a->subs[0]),
2492                        enumerate_fts (axis_type, a->subs[1]));
2493     }
2494
2495   NOT_REACHED ();
2496 }
2497
2498 union ctables_summary
2499   {
2500     /* COUNT, VALIDN, TOTALN. */
2501     double count;
2502
2503     /* MINIMUM, MAXIMUM, RANGE. */
2504     struct
2505       {
2506         double min;
2507         double max;
2508       };
2509
2510     /* MEAN, SEMEAN, STDDEV, SUM, VARIANCE, *.SUM. */
2511     struct moments1 *moments;
2512
2513     /* MEDIAN, MODE, PTILE. */
2514     struct
2515       {
2516         struct casewriter *writer;
2517         double ovalid;
2518         double ovalue;
2519       };
2520
2521     /* XXX multiple response */
2522   };
2523
2524 static void
2525 ctables_summary_init (union ctables_summary *s,
2526                       const struct ctables_summary_spec *ss)
2527 {
2528   switch (ss->function)
2529     {
2530     case CTSF_COUNT:
2531     case CTSF_ECOUNT:
2532     case CTSF_areaPCT_COUNT:
2533     case CTSF_areaPCT_VALIDN:
2534     case CTSF_areaPCT_TOTALN:
2535     case CTSF_MISSING:
2536     case CTSF_TOTALN:
2537     case CTSF_ETOTALN:
2538     case CTSF_VALIDN:
2539     case CTSF_EVALIDN:
2540       s->count = 0;
2541       break;
2542
2543     case CTSF_areaID:
2544       break;
2545
2546     case CTSF_MAXIMUM:
2547     case CTSF_MINIMUM:
2548     case CTSF_RANGE:
2549       s->min = s->max = SYSMIS;
2550       break;
2551
2552     case CTSF_MEAN:
2553     case CTSF_SEMEAN:
2554     case CTSF_STDDEV:
2555     case CTSF_SUM:
2556     case CTSF_VARIANCE:
2557     case CTSF_areaPCT_SUM:
2558       s->moments = moments1_create (MOMENT_VARIANCE);
2559       break;
2560
2561     case CTSF_MEDIAN:
2562     case CTSF_MODE:
2563     case CTSF_PTILE:
2564       {
2565         struct caseproto *proto = caseproto_create ();
2566         proto = caseproto_add_width (proto, 0);
2567         proto = caseproto_add_width (proto, 0);
2568
2569         struct subcase ordering;
2570         subcase_init (&ordering, 0, 0, SC_ASCEND);
2571         s->writer = sort_create_writer (&ordering, proto);
2572         subcase_uninit (&ordering);
2573         caseproto_unref (proto);
2574
2575         s->ovalid = 0;
2576         s->ovalue = SYSMIS;
2577       }
2578       break;
2579     }
2580 }
2581
2582 static void
2583 ctables_summary_uninit (union ctables_summary *s,
2584                         const struct ctables_summary_spec *ss)
2585 {
2586   switch (ss->function)
2587     {
2588     case CTSF_COUNT:
2589     case CTSF_ECOUNT:
2590     case CTSF_areaPCT_COUNT:
2591     case CTSF_areaPCT_VALIDN:
2592     case CTSF_areaPCT_TOTALN:
2593     case CTSF_MISSING:
2594     case CTSF_TOTALN:
2595     case CTSF_ETOTALN:
2596     case CTSF_VALIDN:
2597     case CTSF_EVALIDN:
2598       break;
2599
2600     case CTSF_areaID:
2601       break;
2602
2603     case CTSF_MAXIMUM:
2604     case CTSF_MINIMUM:
2605     case CTSF_RANGE:
2606       break;
2607
2608     case CTSF_MEAN:
2609     case CTSF_SEMEAN:
2610     case CTSF_STDDEV:
2611     case CTSF_SUM:
2612     case CTSF_VARIANCE:
2613     case CTSF_areaPCT_SUM:
2614       moments1_destroy (s->moments);
2615       break;
2616
2617     case CTSF_MEDIAN:
2618     case CTSF_MODE:
2619     case CTSF_PTILE:
2620       casewriter_destroy (s->writer);
2621       break;
2622     }
2623 }
2624
2625 static void
2626 ctables_summary_add (union ctables_summary *s,
2627                      const struct ctables_summary_spec *ss,
2628                      const struct variable *var, const union value *value,
2629                      bool is_scale, bool is_scale_missing,
2630                      bool is_missing, bool excluded_missing,
2631                      double d_weight, double e_weight)
2632 {
2633   /* To determine whether a case is included in a given table for a particular
2634      kind of summary, consider the following charts for each variable in the
2635      table.  Only if "yes" appears for every variable for the summary is the
2636      case counted.
2637
2638      Categorical variables:                    VALIDN   COUNT   TOTALN
2639        Valid values in included categories       yes     yes      yes
2640        Missing values in included categories     ---     yes      yes
2641        Missing values in excluded categories     ---     ---      yes
2642        Valid values in excluded categories       ---     ---      ---
2643
2644      Scale variables:                          VALIDN   COUNT   TOTALN
2645        Valid value                               yes     yes      yes
2646        Missing value                             ---     yes      yes
2647
2648      Missing values include both user- and system-missing.  (The system-missing
2649      value is always in an excluded category.)
2650   */
2651   switch (ss->function)
2652     {
2653     case CTSF_TOTALN:
2654     case CTSF_areaPCT_TOTALN:
2655       s->count += ss->weighted ? d_weight : 1.0;
2656       break;
2657
2658     case CTSF_COUNT:
2659     case CTSF_areaPCT_COUNT:
2660       if (is_scale || !excluded_missing)
2661         s->count += ss->weighted ? d_weight : 1.0;
2662       break;
2663
2664     case CTSF_VALIDN:
2665     case CTSF_areaPCT_VALIDN:
2666       if (is_scale
2667           ? !is_scale_missing
2668           : !is_missing)
2669         s->count += ss->weighted ? d_weight : 1.0;
2670       break;
2671
2672     case CTSF_areaID:
2673       break;
2674
2675     case CTSF_MISSING:
2676       if (is_scale
2677           ? is_scale_missing
2678           : is_missing)
2679         s->count += ss->weighted ? d_weight : 1.0;
2680       break;
2681
2682     case CTSF_ECOUNT:
2683       if (is_scale || !excluded_missing)
2684         s->count += e_weight;
2685       break;
2686
2687     case CTSF_EVALIDN:
2688       if (is_scale
2689           ? !is_scale_missing
2690           : !is_missing)
2691         s->count += e_weight;
2692       break;
2693
2694     case CTSF_ETOTALN:
2695       s->count += e_weight;
2696       break;
2697
2698     case CTSF_MAXIMUM:
2699     case CTSF_MINIMUM:
2700     case CTSF_RANGE:
2701       if (!is_scale_missing)
2702         {
2703           assert (!var_is_alpha (var)); /* XXX? */
2704           if (s->min == SYSMIS || value->f < s->min)
2705             s->min = value->f;
2706           if (s->max == SYSMIS || value->f > s->max)
2707             s->max = value->f;
2708         }
2709       break;
2710
2711     case CTSF_MEAN:
2712     case CTSF_SEMEAN:
2713     case CTSF_STDDEV:
2714     case CTSF_SUM:
2715     case CTSF_VARIANCE:
2716     case CTSF_areaPCT_SUM:
2717       if (!is_scale_missing)
2718         moments1_add (s->moments, value->f, ss->weighted ? e_weight : 1.0);
2719       break;
2720
2721     case CTSF_MEDIAN:
2722     case CTSF_MODE:
2723     case CTSF_PTILE:
2724       if (!is_scale_missing)
2725         {
2726           double w = ss->weighted ? e_weight : 1.0;
2727           s->ovalid += w;
2728
2729           struct ccase *c = case_create (casewriter_get_proto (s->writer));
2730           *case_num_rw_idx (c, 0) = value->f;
2731           *case_num_rw_idx (c, 1) = w;
2732           casewriter_write (s->writer, c);
2733         }
2734       break;
2735     }
2736 }
2737
2738 static double
2739 ctables_summary_value (const struct ctables_cell *cell,
2740                        union ctables_summary *s,
2741                        const struct ctables_summary_spec *ss)
2742 {
2743   switch (ss->function)
2744     {
2745     case CTSF_COUNT:
2746     case CTSF_ECOUNT:
2747       return s->count;
2748
2749     case CTSF_areaID:
2750       return cell->areas[ss->calc_area]->sequence;
2751
2752     case CTSF_areaPCT_COUNT:
2753       {
2754         const struct ctables_area *a = cell->areas[ss->calc_area];
2755         double a_count = ss->weighted ? a->e_count : a->u_count;
2756         return a_count ? s->count / a_count * 100 : SYSMIS;
2757       }
2758
2759     case CTSF_areaPCT_VALIDN:
2760       {
2761         const struct ctables_area *a = cell->areas[ss->calc_area];
2762         double a_valid = ss->weighted ? a->e_valid : a->u_valid;
2763         return a_valid ? s->count / a_valid * 100 : SYSMIS;
2764       }
2765
2766     case CTSF_areaPCT_TOTALN:
2767       {
2768         const struct ctables_area *a = cell->areas[ss->calc_area];
2769         double a_total = ss->weighted ? a->e_total : a->u_total;
2770         return a_total ? s->count / a_total * 100 : SYSMIS;
2771       }
2772
2773     case CTSF_MISSING:
2774     case CTSF_TOTALN:
2775     case CTSF_ETOTALN:
2776     case CTSF_VALIDN:
2777     case CTSF_EVALIDN:
2778       return s->count;
2779
2780     case CTSF_MAXIMUM:
2781       return s->max;
2782
2783     case CTSF_MINIMUM:
2784       return s->min;
2785
2786     case CTSF_RANGE:
2787       return s->max != SYSMIS && s->min != SYSMIS ? s->max - s->min : SYSMIS;
2788
2789     case CTSF_MEAN:
2790       {
2791         double mean;
2792         moments1_calculate (s->moments, NULL, &mean, NULL, NULL, NULL);
2793         return mean;
2794       }
2795
2796     case CTSF_SEMEAN:
2797       {
2798         double weight, variance;
2799         moments1_calculate (s->moments, &weight, NULL, &variance, NULL, NULL);
2800         return calc_semean (variance, weight);
2801       }
2802
2803     case CTSF_STDDEV:
2804       {
2805         double variance;
2806         moments1_calculate (s->moments, NULL, NULL, &variance, NULL, NULL);
2807         return variance != SYSMIS ? sqrt (variance) : SYSMIS;
2808       }
2809
2810     case CTSF_SUM:
2811       {
2812         double weight, mean;
2813         moments1_calculate (s->moments, &weight, &mean, NULL, NULL, NULL);
2814         return weight != SYSMIS && mean != SYSMIS ? weight * mean : SYSMIS;
2815       }
2816
2817     case CTSF_VARIANCE:
2818       {
2819         double variance;
2820         moments1_calculate (s->moments, NULL, NULL, &variance, NULL, NULL);
2821         return variance;
2822       }
2823
2824     case CTSF_areaPCT_SUM:
2825       {
2826         double weight, mean;
2827         moments1_calculate (s->moments, &weight, &mean, NULL, NULL, NULL);
2828         if (weight == SYSMIS || mean == SYSMIS)
2829           return SYSMIS;
2830
2831         const struct ctables_area *a = cell->areas[ss->calc_area];
2832         const struct ctables_sum *sum = &a->sums[ss->sum_var_idx];
2833         double denom = ss->weighted ? sum->e_sum : sum->u_sum;
2834         return denom != 0 ? weight * mean / denom * 100 : SYSMIS;
2835       }
2836
2837     case CTSF_MEDIAN:
2838     case CTSF_PTILE:
2839       if (s->writer)
2840         {
2841           struct casereader *reader = casewriter_make_reader (s->writer);
2842           s->writer = NULL;
2843
2844           struct percentile *ptile = percentile_create (
2845             ss->function == CTSF_PTILE ? ss->percentile : 0.5, s->ovalid);
2846           struct order_stats *os = &ptile->parent;
2847           order_stats_accumulate_idx (&os, 1, reader, 1, 0);
2848           s->ovalue = percentile_calculate (ptile, PC_HAVERAGE);
2849           statistic_destroy (&ptile->parent.parent);
2850         }
2851       return s->ovalue;
2852
2853     case CTSF_MODE:
2854       if (s->writer)
2855         {
2856           struct casereader *reader = casewriter_make_reader (s->writer);
2857           s->writer = NULL;
2858
2859           struct mode *mode = mode_create ();
2860           struct order_stats *os = &mode->parent;
2861           order_stats_accumulate_idx (&os, 1, reader, 1, 0);
2862           s->ovalue = mode->mode;
2863           statistic_destroy (&mode->parent.parent);
2864         }
2865       return s->ovalue;
2866     }
2867
2868   NOT_REACHED ();
2869 }
2870
2871 struct ctables_cell_sort_aux
2872   {
2873     const struct ctables_nest *nest;
2874     enum pivot_axis_type a;
2875   };
2876
2877 static int
2878 ctables_cell_compare_3way (const void *a_, const void *b_, const void *aux_)
2879 {
2880   const struct ctables_cell_sort_aux *aux = aux_;
2881   struct ctables_cell *const *ap = a_;
2882   struct ctables_cell *const *bp = b_;
2883   const struct ctables_cell *a = *ap;
2884   const struct ctables_cell *b = *bp;
2885
2886   const struct ctables_nest *nest = aux->nest;
2887   for (size_t i = 0; i < nest->n; i++)
2888     if (i != nest->scale_idx)
2889       {
2890         const struct variable *var = nest->vars[i];
2891         const struct ctables_cell_value *a_cv = &a->axes[aux->a].cvs[i];
2892         const struct ctables_cell_value *b_cv = &b->axes[aux->a].cvs[i];
2893         if (a_cv->category != b_cv->category)
2894           return a_cv->category > b_cv->category ? 1 : -1;
2895
2896         const union value *a_val = &a_cv->value;
2897         const union value *b_val = &b_cv->value;
2898         switch (a_cv->category->type)
2899           {
2900           case CCT_NUMBER:
2901           case CCT_STRING:
2902           case CCT_SUBTOTAL:
2903           case CCT_TOTAL:
2904           case CCT_POSTCOMPUTE:
2905           case CCT_EXCLUDED_MISSING:
2906             /* Must be equal. */
2907             continue;
2908
2909           case CCT_NRANGE:
2910           case CCT_SRANGE:
2911           case CCT_MISSING:
2912           case CCT_OTHERNM:
2913             {
2914               int cmp = value_compare_3way (a_val, b_val, var_get_width (var));
2915               if (cmp)
2916                 return cmp;
2917             }
2918             break;
2919
2920           case CCT_VALUE:
2921             {
2922               int cmp = value_compare_3way (a_val, b_val, var_get_width (var));
2923               if (cmp)
2924                 return a_cv->category->sort_ascending ? cmp : -cmp;
2925             }
2926             break;
2927
2928           case CCT_LABEL:
2929             {
2930               const char *a_label = var_lookup_value_label (var, a_val);
2931               const char *b_label = var_lookup_value_label (var, b_val);
2932               int cmp;
2933               if (a_label)
2934                 {
2935                   if (!b_label)
2936                     return -1;
2937                   cmp = strcmp (a_label, b_label);
2938                 }
2939               else
2940                 {
2941                   if (b_label)
2942                     return 1;
2943                   cmp = value_compare_3way (a_val, b_val, var_get_width (var));
2944                 }
2945               if (cmp)
2946                 return a_cv->category->sort_ascending ? cmp : -cmp;
2947             }
2948             break;
2949
2950           case CCT_FUNCTION:
2951             NOT_REACHED ();
2952           }
2953       }
2954   return 0;
2955 }
2956
2957 static int
2958 ctables_cell_compare_leaf_3way (const void *a_, const void *b_,
2959                                 const void *aux UNUSED)
2960 {
2961   struct ctables_cell *const *ap = a_;
2962   struct ctables_cell *const *bp = b_;
2963   const struct ctables_cell *a = *ap;
2964   const struct ctables_cell *b = *bp;
2965
2966   for (enum pivot_axis_type axis = 0; axis < PIVOT_N_AXES; axis++)
2967     {
2968       int al = a->axes[axis].leaf;
2969       int bl = b->axes[axis].leaf;
2970       if (al != bl)
2971         return al > bl ? 1 : -1;
2972     }
2973   return 0;
2974 }
2975
2976 /* Algorithm:
2977
2978    For each row:
2979        For each ctables_table:
2980            For each combination of row vars:
2981                For each combination of column vars:
2982                    For each combination of layer vars:
2983                        Add entry
2984    Make a table of row values:
2985        Sort entries by row values
2986        Assign a 0-based index to each actual value
2987        Construct a dimension
2988    Make a table of column values
2989    Make a table of layer values
2990    For each entry:
2991        Fill the table entry using the indexes from before.
2992  */
2993
2994 static struct ctables_area *
2995 ctables_area_insert (struct ctables_section *s, struct ctables_cell *cell,
2996                        enum ctables_area_type area)
2997 {
2998   size_t hash = 0;
2999   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3000     {
3001       const struct ctables_nest *nest = s->nests[a];
3002       for (size_t i = 0; i < nest->n_areas[area]; i++)
3003         {
3004           size_t v_idx = nest->areas[area][i];
3005           struct ctables_cell_value *cv = &cell->axes[a].cvs[v_idx];
3006           hash = hash_pointer (cv->category, hash);
3007           if (cv->category->type != CCT_TOTAL
3008               && cv->category->type != CCT_SUBTOTAL
3009               && cv->category->type != CCT_POSTCOMPUTE)
3010             hash = value_hash (&cv->value,
3011                                var_get_width (nest->vars[v_idx]), hash);
3012         }
3013     }
3014
3015   struct ctables_area *a;
3016   HMAP_FOR_EACH_WITH_HASH (a, struct ctables_area, node, hash, &s->areas[area])
3017     {
3018       const struct ctables_cell *df = a->example;
3019       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3020         {
3021           const struct ctables_nest *nest = s->nests[a];
3022           for (size_t i = 0; i < nest->n_areas[area]; i++)
3023             {
3024               size_t v_idx = nest->areas[area][i];
3025               struct ctables_cell_value *cv1 = &df->axes[a].cvs[v_idx];
3026               struct ctables_cell_value *cv2 = &cell->axes[a].cvs[v_idx];
3027               if (cv1->category != cv2->category
3028                   || (cv1->category->type != CCT_TOTAL
3029                       && cv1->category->type != CCT_SUBTOTAL
3030                       && cv1->category->type != CCT_POSTCOMPUTE
3031                       && !value_equal (&cv1->value, &cv2->value,
3032                                        var_get_width (nest->vars[v_idx]))))
3033                 goto not_equal;
3034             }
3035         }
3036       return a;
3037
3038     not_equal: ;
3039     }
3040
3041   struct ctables_sum *sums = (s->table->n_sum_vars
3042                               ? xzalloc (s->table->n_sum_vars * sizeof *sums)
3043                               : NULL);
3044
3045   a = xmalloc (sizeof *a);
3046   *a = (struct ctables_area) { .example = cell, .sums = sums };
3047   hmap_insert (&s->areas[area], &a->node, hash);
3048   return a;
3049 }
3050
3051 static struct substring
3052 rtrim_value (const union value *v, const struct variable *var)
3053 {
3054   struct substring s = ss_buffer (CHAR_CAST (char *, v->s),
3055                                   var_get_width (var));
3056   ss_rtrim (&s, ss_cstr (" "));
3057   return s;
3058 }
3059
3060 static bool
3061 in_string_range (const union value *v, const struct variable *var,
3062                  const struct substring *srange)
3063 {
3064   struct substring s = rtrim_value (v, var);
3065   return ((!srange[0].string || ss_compare (s, srange[0]) >= 0)
3066           && (!srange[1].string || ss_compare (s, srange[1]) <= 0));
3067 }
3068
3069 static const struct ctables_category *
3070 ctables_categories_match (const struct ctables_categories *c,
3071                           const union value *v, const struct variable *var)
3072 {
3073   if (var_is_numeric (var) && v->f == SYSMIS)
3074     return NULL;
3075
3076   const struct ctables_category *othernm = NULL;
3077   for (size_t i = c->n_cats; i-- > 0; )
3078     {
3079       const struct ctables_category *cat = &c->cats[i];
3080       switch (cat->type)
3081         {
3082         case CCT_NUMBER:
3083           if (cat->number == v->f)
3084             return cat;
3085           break;
3086
3087         case CCT_STRING:
3088           if (ss_equals (cat->string, rtrim_value (v, var)))
3089             return cat;
3090           break;
3091
3092         case CCT_NRANGE:
3093           if ((cat->nrange[0] == -DBL_MAX || v->f >= cat->nrange[0])
3094               && (cat->nrange[1] == DBL_MAX || v->f <= cat->nrange[1]))
3095             return cat;
3096           break;
3097
3098         case CCT_SRANGE:
3099           if (in_string_range (v, var, cat->srange))
3100             return cat;
3101           break;
3102
3103         case CCT_MISSING:
3104           if (var_is_value_missing (var, v))
3105             return cat;
3106           break;
3107
3108         case CCT_POSTCOMPUTE:
3109           break;
3110
3111         case CCT_OTHERNM:
3112           if (!othernm)
3113             othernm = cat;
3114           break;
3115
3116         case CCT_SUBTOTAL:
3117         case CCT_TOTAL:
3118           break;
3119
3120         case CCT_VALUE:
3121         case CCT_LABEL:
3122         case CCT_FUNCTION:
3123           return (cat->include_missing || !var_is_value_missing (var, v) ? cat
3124                   : NULL);
3125
3126         case CCT_EXCLUDED_MISSING:
3127           break;
3128         }
3129     }
3130
3131   return var_is_value_missing (var, v) ? NULL : othernm;
3132 }
3133
3134 static const struct ctables_category *
3135 ctables_categories_total (const struct ctables_categories *c)
3136 {
3137   const struct ctables_category *first = &c->cats[0];
3138   const struct ctables_category *last = &c->cats[c->n_cats - 1];
3139   return (first->type == CCT_TOTAL ? first
3140           : last->type == CCT_TOTAL ? last
3141           : NULL);
3142 }
3143
3144 static struct ctables_cell *
3145 ctables_cell_insert__ (struct ctables_section *s, const struct ccase *c,
3146                        const struct ctables_category *cats[PIVOT_N_AXES][10])
3147 {
3148   size_t hash = 0;
3149   enum ctables_summary_variant sv = CSV_CELL;
3150   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3151     {
3152       const struct ctables_nest *nest = s->nests[a];
3153       for (size_t i = 0; i < nest->n; i++)
3154         if (i != nest->scale_idx)
3155           {
3156             hash = hash_pointer (cats[a][i], hash);
3157             if (cats[a][i]->type != CCT_TOTAL
3158                 && cats[a][i]->type != CCT_SUBTOTAL
3159                 && cats[a][i]->type != CCT_POSTCOMPUTE)
3160               hash = value_hash (case_data (c, nest->vars[i]),
3161                                  var_get_width (nest->vars[i]), hash);
3162             else
3163               sv = CSV_TOTAL;
3164           }
3165     }
3166
3167   struct ctables_cell *cell;
3168   HMAP_FOR_EACH_WITH_HASH (cell, struct ctables_cell, node, hash, &s->cells)
3169     {
3170       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3171         {
3172           const struct ctables_nest *nest = s->nests[a];
3173           for (size_t i = 0; i < nest->n; i++)
3174             if (i != nest->scale_idx
3175                 && (cats[a][i] != cell->axes[a].cvs[i].category
3176                     || (cats[a][i]->type != CCT_TOTAL
3177                         && cats[a][i]->type != CCT_SUBTOTAL
3178                         && cats[a][i]->type != CCT_POSTCOMPUTE
3179                         && !value_equal (case_data (c, nest->vars[i]),
3180                                          &cell->axes[a].cvs[i].value,
3181                                          var_get_width (nest->vars[i])))))
3182                 goto not_equal;
3183         }
3184
3185       return cell;
3186
3187     not_equal: ;
3188     }
3189
3190   cell = xmalloc (sizeof *cell);
3191   cell->hide = false;
3192   cell->sv = sv;
3193   cell->omit_areas = 0;
3194   cell->postcompute = false;
3195   //struct string name = DS_EMPTY_INITIALIZER;
3196   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3197     {
3198       const struct ctables_nest *nest = s->nests[a];
3199       cell->axes[a].cvs = (nest->n
3200                            ? xnmalloc (nest->n, sizeof *cell->axes[a].cvs)
3201                            : NULL);
3202       for (size_t i = 0; i < nest->n; i++)
3203         {
3204           const struct ctables_category *cat = cats[a][i];
3205           const struct variable *var = nest->vars[i];
3206           const union value *value = case_data (c, var);
3207           if (i != nest->scale_idx)
3208             {
3209               const struct ctables_category *subtotal = cat->subtotal;
3210               if (cat->hide || (subtotal && subtotal->hide_subcategories))
3211                 cell->hide = true;
3212
3213               if (cat->type == CCT_TOTAL
3214                   || cat->type == CCT_SUBTOTAL
3215                   || cat->type == CCT_POSTCOMPUTE)
3216                 {
3217                   /* XXX these should be more encompassing I think.*/
3218
3219                   switch (a)
3220                     {
3221                     case PIVOT_AXIS_COLUMN:
3222                       cell->omit_areas |= ((1u << CTAT_TABLE) |
3223                                            (1u << CTAT_LAYER) |
3224                                            (1u << CTAT_LAYERCOL) |
3225                                            (1u << CTAT_SUBTABLE) |
3226                                            (1u << CTAT_COL));
3227                       break;
3228                     case PIVOT_AXIS_ROW:
3229                       cell->omit_areas |= ((1u << CTAT_TABLE) |
3230                                            (1u << CTAT_LAYER) |
3231                                            (1u << CTAT_LAYERROW) |
3232                                            (1u << CTAT_SUBTABLE) |
3233                                            (1u << CTAT_ROW));
3234                       break;
3235                     case PIVOT_AXIS_LAYER:
3236                       cell->omit_areas |= ((1u << CTAT_TABLE) |
3237                                            (1u << CTAT_LAYER));
3238                       break;
3239                     }
3240                 }
3241               if (cat->type == CCT_POSTCOMPUTE)
3242                 cell->postcompute = true;
3243             }
3244
3245           cell->axes[a].cvs[i].category = cat;
3246           value_clone (&cell->axes[a].cvs[i].value, value, var_get_width (var));
3247
3248 #if 0
3249           if (i != nest->scale_idx)
3250             {
3251               if (!ds_is_empty (&name))
3252                 ds_put_cstr (&name, ", ");
3253               char *value_s = data_out (value, var_get_encoding (var),
3254                                         var_get_print_format (var),
3255                                         settings_get_fmt_settings ());
3256               if (cat->type == CCT_TOTAL
3257                   || cat->type == CCT_SUBTOTAL
3258                   || cat->type == CCT_POSTCOMPUTE)
3259                 ds_put_format (&name, "%s=total", var_get_name (var));
3260               else
3261                 ds_put_format (&name, "%s=%s", var_get_name (var),
3262                                value_s + strspn (value_s, " "));
3263               free (value_s);
3264             }
3265 #endif
3266         }
3267     }
3268   //cell->name = ds_steal_cstr (&name);
3269
3270   const struct ctables_nest *ss = s->nests[s->table->summary_axis];
3271   const struct ctables_summary_spec_set *specs = &ss->specs[cell->sv];
3272   cell->summaries = xmalloc (specs->n * sizeof *cell->summaries);
3273   for (size_t i = 0; i < specs->n; i++)
3274     ctables_summary_init (&cell->summaries[i], &specs->specs[i]);
3275   for (enum ctables_area_type at = 0; at < N_CTATS; at++)
3276     cell->areas[at] = ctables_area_insert (s, cell, at);
3277   hmap_insert (&s->cells, &cell->node, hash);
3278   return cell;
3279 }
3280
3281 static bool
3282 is_scale_missing (const struct ctables_summary_spec_set *specs,
3283                   const struct ccase *c)
3284 {
3285   if (!specs->is_scale)
3286     return false;
3287
3288   if (var_is_num_missing (specs->var, case_num (c, specs->var)))
3289     return true;
3290
3291   for (size_t i = 0; i < specs->n_listwise_vars; i++)
3292     {
3293       const struct variable *var = specs->listwise_vars[i];
3294       if (var_is_num_missing (var, case_num (c, var)))
3295         return true;
3296     }
3297
3298   return false;
3299 }
3300
3301 static void
3302 ctables_cell_add__ (struct ctables_section *s, const struct ccase *c,
3303                     const struct ctables_category *cats[PIVOT_N_AXES][10],
3304                     bool is_missing, bool excluded_missing,
3305                     double d_weight, double e_weight)
3306 {
3307   struct ctables_cell *cell = ctables_cell_insert__ (s, c, cats);
3308   const struct ctables_nest *ss = s->nests[s->table->summary_axis];
3309
3310   const struct ctables_summary_spec_set *specs = &ss->specs[cell->sv];
3311
3312   bool scale_missing = is_scale_missing (specs, c);
3313   for (size_t i = 0; i < specs->n; i++)
3314     ctables_summary_add (&cell->summaries[i], &specs->specs[i],
3315                          specs->var, case_data (c, specs->var), specs->is_scale,
3316                          scale_missing, is_missing, excluded_missing,
3317                          d_weight, e_weight);
3318   for (enum ctables_area_type at = 0; at < N_CTATS; at++)
3319     if (!(cell->omit_areas && (1u << at)))
3320       {
3321         struct ctables_area *a = cell->areas[at];
3322         a->d_total += d_weight;
3323         a->e_total += e_weight;
3324         a->u_total += 1.0;
3325         if (!excluded_missing)
3326           {
3327             a->d_count += d_weight;
3328             a->e_count += e_weight;
3329             a->u_count += 1.0;
3330           }
3331         if (!is_missing)
3332           {
3333             a->d_valid += d_weight;
3334             a->e_valid += e_weight;
3335             a->u_count += 1.0;
3336
3337             for (size_t i = 0; i < s->table->n_sum_vars; i++)
3338               {
3339                 /* XXX listwise_missing??? */
3340                 const struct variable *var = s->table->sum_vars[i];
3341                 double addend = case_num (c, var);
3342                 if (!var_is_num_missing (var, addend))
3343                   {
3344                     struct ctables_sum *sum = &a->sums[i];
3345                     sum->e_sum += addend * e_weight;
3346                     sum->u_sum += addend;
3347                   }
3348               }
3349           }
3350       }
3351 }
3352
3353 static void
3354 recurse_totals (struct ctables_section *s, const struct ccase *c,
3355                 const struct ctables_category *cats[PIVOT_N_AXES][10],
3356                 bool is_missing, bool excluded_missing,
3357                 double d_weight, double e_weight,
3358                 enum pivot_axis_type start_axis, size_t start_nest)
3359 {
3360   for (enum pivot_axis_type a = start_axis; a < PIVOT_N_AXES; a++)
3361     {
3362       const struct ctables_nest *nest = s->nests[a];
3363       for (size_t i = start_nest; i < nest->n; i++)
3364         {
3365           if (i == nest->scale_idx)
3366             continue;
3367
3368           const struct variable *var = nest->vars[i];
3369
3370           const struct ctables_category *total = ctables_categories_total (
3371             s->table->categories[var_get_dict_index (var)]);
3372           if (total)
3373             {
3374               const struct ctables_category *save = cats[a][i];
3375               cats[a][i] = total;
3376               ctables_cell_add__ (s, c, cats, is_missing, excluded_missing,
3377                                   d_weight, e_weight);
3378               recurse_totals (s, c, cats, is_missing, excluded_missing,
3379                               d_weight, e_weight, a, i + 1);
3380               cats[a][i] = save;
3381             }
3382         }
3383       start_nest = 0;
3384     }
3385 }
3386
3387 static void
3388 recurse_subtotals (struct ctables_section *s, const struct ccase *c,
3389                    const struct ctables_category *cats[PIVOT_N_AXES][10],
3390                    bool is_missing, bool excluded_missing,
3391                    double d_weight, double e_weight,
3392                    enum pivot_axis_type start_axis, size_t start_nest)
3393 {
3394   for (enum pivot_axis_type a = start_axis; a < PIVOT_N_AXES; a++)
3395     {
3396       const struct ctables_nest *nest = s->nests[a];
3397       for (size_t i = start_nest; i < nest->n; i++)
3398         {
3399           if (i == nest->scale_idx)
3400             continue;
3401
3402           const struct ctables_category *save = cats[a][i];
3403           if (save->subtotal)
3404             {
3405               cats[a][i] = save->subtotal;
3406               ctables_cell_add__ (s, c, cats, is_missing, excluded_missing,
3407                                   d_weight, e_weight);
3408               recurse_subtotals (s, c, cats, is_missing, excluded_missing,
3409                                  d_weight, e_weight, a, i + 1);
3410               cats[a][i] = save;
3411             }
3412         }
3413       start_nest = 0;
3414     }
3415 }
3416
3417 static void
3418 ctables_add_occurrence (const struct variable *var,
3419                         const union value *value,
3420                         struct hmap *occurrences)
3421 {
3422   int width = var_get_width (var);
3423   unsigned int hash = value_hash (value, width, 0);
3424
3425   struct ctables_occurrence *o;
3426   HMAP_FOR_EACH_WITH_HASH (o, struct ctables_occurrence, node, hash,
3427                            occurrences)
3428     if (value_equal (value, &o->value, width))
3429       return;
3430
3431   o = xmalloc (sizeof *o);
3432   value_clone (&o->value, value, width);
3433   hmap_insert (occurrences, &o->node, hash);
3434 }
3435
3436 static void
3437 ctables_cell_insert (struct ctables_section *s,
3438                      const struct ccase *c,
3439                      double d_weight, double e_weight)
3440 {
3441   const struct ctables_category *cats[PIVOT_N_AXES][10]; /* XXX */
3442
3443   /* Does at least one categorical variable have a missing value in an included
3444      or excluded category? */
3445   bool is_missing = false;
3446
3447   /* Does at least one categorical variable have a missing value in an excluded
3448      category? */
3449   bool excluded_missing = false;
3450
3451   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3452     {
3453       const struct ctables_nest *nest = s->nests[a];
3454       for (size_t i = 0; i < nest->n; i++)
3455         {
3456           if (i == nest->scale_idx)
3457             continue;
3458
3459           const struct variable *var = nest->vars[i];
3460           const union value *value = case_data (c, var);
3461
3462           bool var_missing = var_is_value_missing (var, value) != 0;
3463           if (var_missing)
3464             is_missing = true;
3465
3466           cats[a][i] = ctables_categories_match (
3467             s->table->categories[var_get_dict_index (var)], value, var);
3468           if (!cats[a][i])
3469             {
3470               if (!var_missing)
3471                 return;
3472
3473               static const struct ctables_category cct_excluded_missing = {
3474                 .type = CCT_EXCLUDED_MISSING,
3475                 .hide = true,
3476               };
3477               cats[a][i] = &cct_excluded_missing;
3478               excluded_missing = true;
3479             }
3480         }
3481     }
3482
3483   if (!excluded_missing)
3484     for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3485       {
3486         const struct ctables_nest *nest = s->nests[a];
3487         for (size_t i = 0; i < nest->n; i++)
3488           if (i != nest->scale_idx)
3489             {
3490               const struct variable *var = nest->vars[i];
3491               const union value *value = case_data (c, var);
3492               ctables_add_occurrence (var, value, &s->occurrences[a][i]);
3493             }
3494       }
3495
3496   ctables_cell_add__ (s, c, cats, is_missing, excluded_missing,
3497                       d_weight, e_weight);
3498
3499   //if (!excluded_missing)
3500     {
3501       recurse_totals (s, c, cats, is_missing, excluded_missing,
3502                       d_weight, e_weight, 0, 0);
3503       recurse_subtotals (s, c, cats, is_missing, excluded_missing,
3504                          d_weight, e_weight, 0, 0);
3505     }
3506 }
3507
3508 struct merge_item
3509   {
3510     const struct ctables_summary_spec_set *set;
3511     size_t ofs;
3512   };
3513
3514 static int
3515 merge_item_compare_3way (const struct merge_item *a, const struct merge_item *b)
3516 {
3517   const struct ctables_summary_spec *as = &a->set->specs[a->ofs];
3518   const struct ctables_summary_spec *bs = &b->set->specs[b->ofs];
3519   if (as->function != bs->function)
3520     return as->function > bs->function ? 1 : -1;
3521   else if (as->weighted != bs->weighted)
3522     return as->weighted > bs->weighted ? 1 : -1;
3523   else if (as->calc_area != bs->calc_area)
3524     return as->calc_area > bs->calc_area ? 1 : -1;
3525   else if (as->percentile != bs->percentile)
3526     return as->percentile < bs->percentile ? 1 : -1;
3527
3528   const char *as_label = as->label ? as->label : "";
3529   const char *bs_label = bs->label ? bs->label : "";
3530   return strcmp (as_label, bs_label);
3531 }
3532
3533 static void
3534 ctables_category_format_number (double number, const struct variable *var,
3535                                 struct string *s)
3536 {
3537   struct pivot_value *pv = pivot_value_new_var_value (
3538     var, &(union value) { .f = number });
3539   pivot_value_format (pv, NULL, s);
3540   pivot_value_destroy (pv);
3541 }
3542
3543 static void
3544 ctables_category_format_string (struct substring string,
3545                                 const struct variable *var, struct string *out)
3546 {
3547   int width = var_get_width (var);
3548   char *s = xmalloc (width);
3549   buf_copy_rpad (s, width, string.string, string.length, ' ');
3550   struct pivot_value *pv = pivot_value_new_var_value (
3551     var, &(union value) { .s = CHAR_CAST (uint8_t *, s) });
3552   pivot_value_format (pv, NULL, out);
3553   pivot_value_destroy (pv);
3554   free (s);
3555 }
3556
3557 static bool
3558 ctables_category_format_label (const struct ctables_category *cat,
3559                                const struct variable *var,
3560                                struct string *s)
3561 {
3562   switch (cat->type)
3563     {
3564     case CCT_NUMBER:
3565       ctables_category_format_number (cat->number, var, s);
3566       return true;
3567
3568     case CCT_STRING:
3569       ctables_category_format_string (cat->string, var, s);
3570       return true;
3571
3572     case CCT_NRANGE:
3573       ctables_category_format_number (cat->nrange[0], var, s);
3574       ds_put_format (s, " THRU ");
3575       ctables_category_format_number (cat->nrange[1], var, s);
3576       return true;
3577
3578     case CCT_SRANGE:
3579       ctables_category_format_string (cat->srange[0], var, s);
3580       ds_put_format (s, " THRU ");
3581       ctables_category_format_string (cat->srange[1], var, s);
3582       return true;
3583
3584     case CCT_MISSING:
3585       ds_put_cstr (s, "MISSING");
3586       return true;
3587
3588     case CCT_OTHERNM:
3589       ds_put_cstr (s, "OTHERNM");
3590       return true;
3591
3592     case CCT_POSTCOMPUTE:
3593       ds_put_format (s, "&%s", cat->pc->name);
3594       return true;
3595
3596     case CCT_TOTAL:
3597     case CCT_SUBTOTAL:
3598       ds_put_cstr (s, cat->total_label);
3599       return true;
3600
3601     case CCT_VALUE:
3602     case CCT_LABEL:
3603     case CCT_FUNCTION:
3604     case CCT_EXCLUDED_MISSING:
3605       return false;
3606     }
3607
3608   return false;
3609 }
3610
3611 static struct pivot_value *
3612 ctables_postcompute_label (const struct ctables_categories *cats,
3613                            const struct ctables_category *cat,
3614                            const struct variable *var)
3615 {
3616   struct substring in = ss_cstr (cat->pc->label);
3617   struct substring target = ss_cstr (")LABEL[");
3618
3619   struct string out = DS_EMPTY_INITIALIZER;
3620   for (;;)
3621     {
3622       size_t chunk = ss_find_substring (in, target);
3623       if (chunk == SIZE_MAX)
3624         {
3625           if (ds_is_empty (&out))
3626             return pivot_value_new_user_text (in.string, in.length);
3627           else
3628             {
3629               ds_put_substring (&out, in);
3630               return pivot_value_new_user_text_nocopy (ds_steal_cstr (&out));
3631             }
3632         }
3633
3634       ds_put_substring (&out, ss_head (in, chunk));
3635       ss_advance (&in, chunk + target.length);
3636
3637       struct substring idx_s;
3638       if (!ss_get_until (&in, ']', &idx_s))
3639         goto error;
3640       char *tail;
3641       long int idx = strtol (idx_s.string, &tail, 10);
3642       if (idx < 1 || idx > cats->n_cats || tail != ss_end (idx_s))
3643         goto error;
3644
3645       struct ctables_category *cat2 = &cats->cats[idx - 1];
3646       if (!ctables_category_format_label (cat2, var, &out))
3647         goto error;
3648     }
3649
3650 error:
3651   ds_destroy (&out);
3652   return pivot_value_new_user_text (cat->pc->label, SIZE_MAX);
3653 }
3654
3655 static struct pivot_value *
3656 ctables_category_create_value_label (const struct ctables_categories *cats,
3657                                      const struct ctables_category *cat,
3658                                      const struct variable *var,
3659                                      const union value *value)
3660 {
3661   return (cat->type == CCT_POSTCOMPUTE && cat->pc->label
3662           ? ctables_postcompute_label (cats, cat, var)
3663           : cat->type == CCT_TOTAL || cat->type == CCT_SUBTOTAL
3664           ? pivot_value_new_user_text (cat->total_label, SIZE_MAX)
3665           : pivot_value_new_var_value (var, value));
3666 }
3667
3668 static struct ctables_value *
3669 ctables_value_find__ (struct ctables_table *t, const union value *value,
3670                       int width, unsigned int hash)
3671 {
3672   struct ctables_value *clv;
3673   HMAP_FOR_EACH_WITH_HASH (clv, struct ctables_value, node,
3674                            hash, &t->clabels_values_map)
3675     if (value_equal (value, &clv->value, width))
3676       return clv;
3677   return NULL;
3678 }
3679
3680 static void
3681 ctables_value_insert (struct ctables_table *t, const union value *value,
3682                       int width)
3683 {
3684   unsigned int hash = value_hash (value, width, 0);
3685   struct ctables_value *clv = ctables_value_find__ (t, value, width, hash);
3686   if (!clv)
3687     {
3688       clv = xmalloc (sizeof *clv);
3689       value_clone (&clv->value, value, width);
3690       hmap_insert (&t->clabels_values_map, &clv->node, hash);
3691     }
3692 }
3693
3694 static struct ctables_value *
3695 ctables_value_find (struct ctables_table *t,
3696                     const union value *value, int width)
3697 {
3698   return ctables_value_find__ (t, value, width,
3699                                value_hash (value, width, 0));
3700 }
3701
3702 static void
3703 ctables_table_add_section (struct ctables_table *t, enum pivot_axis_type a,
3704                            size_t ix[PIVOT_N_AXES])
3705 {
3706   if (a < PIVOT_N_AXES)
3707     {
3708       size_t limit = MAX (t->stacks[a].n, 1);
3709       for (ix[a] = 0; ix[a] < limit; ix[a]++)
3710         ctables_table_add_section (t, a + 1, ix);
3711     }
3712   else
3713     {
3714       struct ctables_section *s = &t->sections[t->n_sections++];
3715       *s = (struct ctables_section) {
3716         .table = t,
3717         .cells = HMAP_INITIALIZER (s->cells),
3718       };
3719       for (a = 0; a < PIVOT_N_AXES; a++)
3720         if (t->stacks[a].n)
3721           {
3722             struct ctables_nest *nest = &t->stacks[a].nests[ix[a]];
3723             s->nests[a] = nest;
3724             s->occurrences[a] = xnmalloc (nest->n, sizeof *s->occurrences[a]);
3725             for (size_t i = 0; i < nest->n; i++)
3726               hmap_init (&s->occurrences[a][i]);
3727         }
3728       for (size_t i = 0; i < N_CTATS; i++)
3729         hmap_init (&s->areas[i]);
3730     }
3731 }
3732
3733 static double
3734 ctpo_add (double a, double b)
3735 {
3736   return a + b;
3737 }
3738
3739 static double
3740 ctpo_sub (double a, double b)
3741 {
3742   return a - b;
3743 }
3744
3745 static double
3746 ctpo_mul (double a, double b)
3747 {
3748   return a * b;
3749 }
3750
3751 static double
3752 ctpo_div (double a, double b)
3753 {
3754   return b ? a / b : SYSMIS;
3755 }
3756
3757 static double
3758 ctpo_pow (double a, double b)
3759 {
3760   int save_errno = errno;
3761   errno = 0;
3762   double result = pow (a, b);
3763   if (errno)
3764     result = SYSMIS;
3765   errno = save_errno;
3766   return result;
3767 }
3768
3769 static double
3770 ctpo_neg (double a, double b UNUSED)
3771 {
3772   return -a;
3773 }
3774
3775 struct ctables_pcexpr_evaluate_ctx
3776   {
3777     const struct ctables_cell *cell;
3778     const struct ctables_section *section;
3779     const struct ctables_categories *cats;
3780     enum pivot_axis_type pc_a;
3781     size_t pc_a_idx;
3782     size_t summary_idx;
3783     enum fmt_type parse_format;
3784   };
3785
3786 static double ctables_pcexpr_evaluate (
3787   const struct ctables_pcexpr_evaluate_ctx *, const struct ctables_pcexpr *);
3788
3789 static double
3790 ctables_pcexpr_evaluate_nonterminal (
3791   const struct ctables_pcexpr_evaluate_ctx *ctx,
3792   const struct ctables_pcexpr *e, size_t n_args,
3793   double evaluate (double, double))
3794 {
3795   double args[2] = { 0, 0 };
3796   for (size_t i = 0; i < n_args; i++)
3797     {
3798       args[i] = ctables_pcexpr_evaluate (ctx, e->subs[i]);
3799       if (!isfinite (args[i]) || args[i] == SYSMIS)
3800         return SYSMIS;
3801     }
3802   return evaluate (args[0], args[1]);
3803 }
3804
3805 static double
3806 ctables_pcexpr_evaluate_category (const struct ctables_pcexpr_evaluate_ctx *ctx,
3807                                   const struct ctables_cell_value *pc_cv)
3808 {
3809   const struct ctables_section *s = ctx->section;
3810
3811   size_t hash = 0;
3812   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3813     {
3814       const struct ctables_nest *nest = s->nests[a];
3815       for (size_t i = 0; i < nest->n; i++)
3816         if (i != nest->scale_idx)
3817           {
3818             const struct ctables_cell_value *cv
3819               = (a == ctx->pc_a && i == ctx->pc_a_idx ? pc_cv
3820                  : &ctx->cell->axes[a].cvs[i]);
3821             hash = hash_pointer (cv->category, hash);
3822             if (cv->category->type != CCT_TOTAL
3823                 && cv->category->type != CCT_SUBTOTAL
3824                 && cv->category->type != CCT_POSTCOMPUTE)
3825               hash = value_hash (&cv->value,
3826                                  var_get_width (nest->vars[i]), hash);
3827           }
3828     }
3829
3830   struct ctables_cell *tc;
3831   HMAP_FOR_EACH_WITH_HASH (tc, struct ctables_cell, node, hash, &s->cells)
3832     {
3833       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
3834         {
3835           const struct ctables_nest *nest = s->nests[a];
3836           for (size_t i = 0; i < nest->n; i++)
3837             if (i != nest->scale_idx)
3838               {
3839                 const struct ctables_cell_value *p_cv
3840                   = (a == ctx->pc_a && i == ctx->pc_a_idx ? pc_cv
3841                      : &ctx->cell->axes[a].cvs[i]);
3842                 const struct ctables_cell_value *t_cv = &tc->axes[a].cvs[i];
3843                 if (p_cv->category != t_cv->category
3844                     || (p_cv->category->type != CCT_TOTAL
3845                         && p_cv->category->type != CCT_SUBTOTAL
3846                         && p_cv->category->type != CCT_POSTCOMPUTE
3847                         && !value_equal (&p_cv->value,
3848                                          &t_cv->value,
3849                                          var_get_width (nest->vars[i]))))
3850                   goto not_equal;
3851               }
3852         }
3853
3854       goto found;
3855
3856     not_equal: ;
3857     }
3858   return 0;
3859
3860 found: ;
3861   const struct ctables_table *t = s->table;
3862   const struct ctables_nest *specs_nest = s->nests[t->summary_axis];
3863   const struct ctables_summary_spec_set *specs = &specs_nest->specs[tc->sv];
3864   return ctables_summary_value (tc, &tc->summaries[ctx->summary_idx],
3865                                 &specs->specs[ctx->summary_idx]);
3866 }
3867
3868 static double
3869 ctables_pcexpr_evaluate (const struct ctables_pcexpr_evaluate_ctx *ctx,
3870                          const struct ctables_pcexpr *e)
3871 {
3872   switch (e->op)
3873     {
3874     case CTPO_CONSTANT:
3875       return e->number;
3876
3877     case CTPO_CAT_NRANGE:
3878     case CTPO_CAT_SRANGE:
3879     case CTPO_CAT_MISSING:
3880     case CTPO_CAT_OTHERNM:
3881       {
3882         struct ctables_cell_value cv = {
3883           .category = ctables_find_category_for_postcompute (ctx->section->table->ctables->dict, ctx->cats, ctx->parse_format, e)
3884         };
3885         assert (cv.category != NULL);
3886
3887         struct hmap *occurrences = &ctx->section->occurrences[ctx->pc_a][ctx->pc_a_idx];
3888         const struct ctables_occurrence *o;
3889
3890         double sum = 0.0;
3891         const struct variable *var = ctx->section->nests[ctx->pc_a]->vars[ctx->pc_a_idx];
3892         HMAP_FOR_EACH (o, struct ctables_occurrence, node, occurrences)
3893           if (ctables_categories_match (ctx->cats, &o->value, var) == cv.category)
3894             {
3895               cv.value = o->value;
3896               sum += ctables_pcexpr_evaluate_category (ctx, &cv);
3897             }
3898         return sum;
3899       }
3900
3901     case CTPO_CAT_NUMBER:
3902     case CTPO_CAT_SUBTOTAL:
3903     case CTPO_CAT_TOTAL:
3904       {
3905         struct ctables_cell_value cv = {
3906           .category = ctables_find_category_for_postcompute (ctx->section->table->ctables->dict, ctx->cats, ctx->parse_format, e),
3907           .value = { .f = e->number },
3908         };
3909         assert (cv.category != NULL);
3910         return ctables_pcexpr_evaluate_category (ctx, &cv);
3911       }
3912
3913     case CTPO_CAT_STRING:
3914       {
3915         int width = var_get_width (ctx->section->nests[ctx->pc_a]->vars[ctx->pc_a_idx]);
3916         char *s = NULL;
3917         if (width > e->string.length)
3918           {
3919             s = xmalloc (width);
3920             buf_copy_rpad (s, width, e->string.string, e->string.length, ' ');
3921           }
3922
3923         const struct ctables_category *category
3924           = ctables_find_category_for_postcompute (
3925             ctx->section->table->ctables->dict,
3926             ctx->cats, ctx->parse_format, e);
3927         assert (category != NULL);
3928
3929         struct ctables_cell_value cv = { .category = category };
3930         if (category->type == CCT_NUMBER)
3931           cv.value.f = category->number;
3932         else if (category->type == CCT_STRING)
3933           cv.value.s = CHAR_CAST (uint8_t *, s ? s : e->string.string);
3934         else
3935           NOT_REACHED ();
3936
3937         double retval = ctables_pcexpr_evaluate_category (ctx, &cv);
3938         free (s);
3939         return retval;
3940       }
3941
3942     case CTPO_ADD:
3943       return ctables_pcexpr_evaluate_nonterminal (ctx, e, 2, ctpo_add);
3944
3945     case CTPO_SUB:
3946       return ctables_pcexpr_evaluate_nonterminal (ctx, e, 2, ctpo_sub);
3947
3948     case CTPO_MUL:
3949       return ctables_pcexpr_evaluate_nonterminal (ctx, e, 2, ctpo_mul);
3950
3951     case CTPO_DIV:
3952       return ctables_pcexpr_evaluate_nonterminal (ctx, e, 2, ctpo_div);
3953
3954     case CTPO_POW:
3955       return ctables_pcexpr_evaluate_nonterminal (ctx, e, 2, ctpo_pow);
3956
3957     case CTPO_NEG:
3958       return ctables_pcexpr_evaluate_nonterminal (ctx, e, 1, ctpo_neg);
3959     }
3960
3961   NOT_REACHED ();
3962 }
3963
3964 static const struct ctables_category *
3965 ctables_cell_postcompute (const struct ctables_section *s,
3966                           const struct ctables_cell *cell,
3967                           enum pivot_axis_type *pc_a_p,
3968                           size_t *pc_a_idx_p)
3969 {
3970   assert (cell->postcompute);
3971   const struct ctables_category *pc_cat = NULL;
3972   for (enum pivot_axis_type pc_a = 0; pc_a < PIVOT_N_AXES; pc_a++)
3973     for (size_t pc_a_idx = 0; pc_a_idx < s->nests[pc_a]->n; pc_a_idx++)
3974       {
3975         const struct ctables_cell_value *cv = &cell->axes[pc_a].cvs[pc_a_idx];
3976         if (cv->category->type == CCT_POSTCOMPUTE)
3977           {
3978             if (pc_cat)
3979               {
3980                 /* Multiple postcomputes cross each other.  The value is
3981                    undefined. */
3982                 return NULL;
3983               }
3984
3985             pc_cat = cv->category;
3986             if (pc_a_p)
3987               *pc_a_p = pc_a;
3988             if (pc_a_idx_p)
3989               *pc_a_idx_p = pc_a_idx;
3990           }
3991       }
3992
3993   assert (pc_cat != NULL);
3994   return pc_cat;
3995 }
3996
3997 static double
3998 ctables_cell_calculate_postcompute (const struct ctables_section *s,
3999                                     const struct ctables_cell *cell,
4000                                     const struct ctables_summary_spec *ss,
4001                                     struct fmt_spec *format,
4002                                     bool *is_ctables_format,
4003                                     size_t summary_idx)
4004 {
4005   enum pivot_axis_type pc_a = 0;
4006   size_t pc_a_idx = 0;
4007   const struct ctables_category *pc_cat = ctables_cell_postcompute (
4008     s, cell, &pc_a, &pc_a_idx);
4009   if (!pc_cat)
4010     return SYSMIS;
4011
4012   const struct ctables_postcompute *pc = pc_cat->pc;
4013   if (pc->specs)
4014     {
4015       for (size_t i = 0; i < pc->specs->n; i++)
4016         {
4017           const struct ctables_summary_spec *ss2 = &pc->specs->specs[i];
4018           if (ss->function == ss2->function
4019               && ss->weighted == ss2->weighted
4020               && ss->calc_area == ss2->calc_area
4021               && ss->percentile == ss2->percentile)
4022             {
4023               *format = ss2->format;
4024               *is_ctables_format = ss2->is_ctables_format;
4025               break;
4026             }
4027         }
4028     }
4029
4030   const struct variable *var = s->nests[pc_a]->vars[pc_a_idx];
4031   const struct ctables_categories *cats = s->table->categories[
4032     var_get_dict_index (var)];
4033   struct ctables_pcexpr_evaluate_ctx ctx = {
4034     .cell = cell,
4035     .section = s,
4036     .cats = cats,
4037     .pc_a = pc_a,
4038     .pc_a_idx = pc_a_idx,
4039     .summary_idx = summary_idx,
4040     .parse_format = pc_cat->parse_format,
4041   };
4042   return ctables_pcexpr_evaluate (&ctx, pc->expr);
4043 }
4044
4045 static char *
4046 ctables_format (double d, const struct fmt_spec *format,
4047                 const struct fmt_settings *settings)
4048 {
4049   const union value v = { .f = d };
4050   char *s = data_out_stretchy (&v, "UTF-8", format, settings, NULL);
4051
4052   /* The custom-currency specifications for NEQUAL, PAREN, and PCTPAREN don't
4053      produce the results we want for negative numbers, putting the negative
4054      sign in the wrong spot, before the prefix instead of after it.  We can't,
4055      in fact, produce the desired results using a custom-currency
4056      specification.  Instead, we postprocess the output, moving the negative
4057      sign into place:
4058
4059          NEQUAL:   "-N=3"  => "N=-3"
4060          PAREN:    "-(3)"  => "(-3)"
4061          PCTPAREN: "-(3%)" => "(-3%)"
4062
4063      This transformation doesn't affect NEGPAREN. */
4064   char *minus_src = strchr (s, '-');
4065   if (minus_src && (minus_src == s || minus_src[-1] != 'E'))
4066     {
4067       char *n_equals = strstr (s, "N=");
4068       char *lparen = strchr (s, '(');
4069       char *minus_dst = n_equals ? n_equals + 1 : lparen;
4070       if (minus_dst)
4071         move_element (s, minus_dst - s + 1, 1, minus_src - s, minus_dst - s);
4072     }
4073   return s;
4074 }
4075
4076 static bool
4077 all_hidden_vlabels (const struct ctables_table *t, enum pivot_axis_type a)
4078 {
4079   for (size_t i = 0; i < t->stacks[a].n; i++)
4080     {
4081       struct ctables_nest *nest = &t->stacks[a].nests[i];
4082       if (nest->n != 1 || nest->scale_idx != 0)
4083         return false;
4084
4085       enum ctables_vlabel vlabel
4086         = t->ctables->vlabels[var_get_dict_index (nest->vars[0])];
4087       if (vlabel != CTVL_NONE)
4088         return false;
4089     }
4090   return true;
4091 }
4092
4093 static void
4094 ctables_table_output (struct ctables *ct, struct ctables_table *t)
4095 {
4096   struct pivot_table *pt = pivot_table_create__ (
4097     (t->title
4098      ? pivot_value_new_user_text (t->title, SIZE_MAX)
4099      : pivot_value_new_text (N_("Custom Tables"))),
4100     "Custom Tables");
4101   if (t->caption)
4102     pivot_table_set_caption (
4103       pt, pivot_value_new_user_text (t->caption, SIZE_MAX));
4104   if (t->corner)
4105     pivot_table_set_corner_text (
4106       pt, pivot_value_new_user_text (t->corner, SIZE_MAX));
4107
4108   bool summary_dimension = (t->summary_axis != t->slabels_axis
4109                             || (!t->slabels_visible
4110                                 && t->summary_specs.n > 1));
4111   if (summary_dimension)
4112     {
4113       struct pivot_dimension *d = pivot_dimension_create (
4114         pt, t->slabels_axis, N_("Statistics"));
4115       const struct ctables_summary_spec_set *specs = &t->summary_specs;
4116       if (!t->slabels_visible)
4117         d->hide_all_labels = true;
4118       for (size_t i = 0; i < specs->n; i++)
4119         pivot_category_create_leaf (
4120           d->root, ctables_summary_label (&specs->specs[i], t->cilevel));
4121     }
4122
4123   bool categories_dimension = t->clabels_example != NULL;
4124   if (categories_dimension)
4125     {
4126       struct pivot_dimension *d = pivot_dimension_create (
4127         pt, t->label_axis[t->clabels_from_axis],
4128         t->clabels_from_axis == PIVOT_AXIS_ROW
4129         ? N_("Row Categories")
4130         : N_("Column Categories"));
4131       const struct variable *var = t->clabels_example;
4132       const struct ctables_categories *c = t->categories[var_get_dict_index (var)];
4133       for (size_t i = 0; i < t->n_clabels_values; i++)
4134         {
4135           const struct ctables_value *value = t->clabels_values[i];
4136           const struct ctables_category *cat = ctables_categories_match (c, &value->value, var);
4137           assert (cat != NULL);
4138           pivot_category_create_leaf (
4139             d->root, ctables_category_create_value_label (c, cat,
4140                                                           t->clabels_example,
4141                                                           &value->value));
4142         }
4143     }
4144
4145   pivot_table_set_look (pt, ct->look);
4146   struct pivot_dimension *d[PIVOT_N_AXES];
4147   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
4148     {
4149       static const char *names[] = {
4150         [PIVOT_AXIS_ROW] = N_("Rows"),
4151         [PIVOT_AXIS_COLUMN] = N_("Columns"),
4152         [PIVOT_AXIS_LAYER] = N_("Layers"),
4153       };
4154       d[a] = (t->axes[a] || a == t->summary_axis
4155               ? pivot_dimension_create (pt, a, names[a])
4156               : NULL);
4157       if (!d[a])
4158         continue;
4159
4160       assert (t->axes[a]);
4161
4162       for (size_t i = 0; i < t->stacks[a].n; i++)
4163         {
4164           struct ctables_nest *nest = &t->stacks[a].nests[i];
4165           struct ctables_section **sections = xnmalloc (t->n_sections,
4166                                                         sizeof *sections);
4167           size_t n_sections = 0;
4168
4169           size_t n_total_cells = 0;
4170           size_t max_depth = 0;
4171           for (size_t j = 0; j < t->n_sections; j++)
4172             if (t->sections[j].nests[a] == nest)
4173               {
4174                 struct ctables_section *s = &t->sections[j];
4175                 sections[n_sections++] = s;
4176                 n_total_cells += hmap_count (&s->cells);
4177
4178                 size_t depth = s->nests[a]->n;
4179                 max_depth = MAX (depth, max_depth);
4180               }
4181
4182           struct ctables_cell **sorted = xnmalloc (n_total_cells,
4183                                                    sizeof *sorted);
4184           size_t n_sorted = 0;
4185
4186           for (size_t j = 0; j < n_sections; j++)
4187             {
4188               struct ctables_section *s = sections[j];
4189
4190               struct ctables_cell *cell;
4191               HMAP_FOR_EACH (cell, struct ctables_cell, node, &s->cells)
4192                 if (!cell->hide)
4193                   sorted[n_sorted++] = cell;
4194               assert (n_sorted <= n_total_cells);
4195             }
4196
4197           struct ctables_cell_sort_aux aux = { .nest = nest, .a = a };
4198           sort (sorted, n_sorted, sizeof *sorted, ctables_cell_compare_3way, &aux);
4199
4200 #if 0
4201           if (a == PIVOT_AXIS_ROW)
4202             {
4203               size_t ids[N_CTATS];
4204               memset (ids, 0, sizeof ids);
4205               for (size_t j = 0; j < n_sorted; j++)
4206                 {
4207                   struct ctables_cell *cell = sorted[j];
4208                   for (enum ctables_area_type at = 0; at < N_CTATS; at++)
4209                     {
4210                       struct ctables_area *area = cell->areas[at];
4211                       if (!area->sequence)
4212                         area->sequence = ++ids[at];
4213                     }
4214                 }
4215             }
4216 #endif
4217
4218 #if 0
4219           for (size_t j = 0; j < n_sorted; j++)
4220             {
4221               printf ("%s (%s): %f/%f = %.1f%%\n", sorted[j]->name, sorted[j]->contributes_to_areas ? "y" : "n", sorted[j]->summaries[0].count, sorted[j]->areas[CTAT_COL]->e_count, sorted[j]->summaries[0].count / sorted[j]->areas[CTAT_COL]->e_count * 100.0);
4222             }
4223           printf ("\n");
4224 #endif
4225           
4226           struct ctables_level
4227             {
4228               enum ctables_level_type
4229                 {
4230                   CTL_VAR,          /* Variable label for nest->vars[var_idx]. */
4231                   CTL_CATEGORY,     /* Category for nest->vars[var_idx]. */
4232                   CTL_SUMMARY,      /* Summary functions. */
4233                 }
4234                 type;
4235
4236               enum settings_value_show vlabel; /* CTL_VAR only. */
4237               size_t var_idx;
4238             };
4239           struct ctables_level *levels = xnmalloc (1 + 2 * max_depth, sizeof *levels);
4240           size_t n_levels = 0;
4241           for (size_t k = 0; k < nest->n; k++)
4242             {
4243               enum ctables_vlabel vlabel = ct->vlabels[var_get_dict_index (nest->vars[k])];
4244               if (vlabel == CTVL_NONE && nest->scale_idx == k)
4245                 vlabel = CTVL_NAME;
4246               if (vlabel != CTVL_NONE)
4247                 {
4248                   levels[n_levels++] = (struct ctables_level) {
4249                     .type = CTL_VAR,
4250                     .vlabel = (enum settings_value_show) vlabel,
4251                     .var_idx = k,
4252                   };
4253                 }
4254
4255               if (nest->scale_idx != k
4256                   && (k != nest->n - 1 || t->label_axis[a] == a))
4257                 {
4258                   levels[n_levels++] = (struct ctables_level) {
4259                     .type = CTL_CATEGORY,
4260                     .var_idx = k,
4261                   };
4262                 }
4263             }
4264
4265           if (!summary_dimension && a == t->slabels_axis)
4266             {
4267               levels[n_levels++] = (struct ctables_level) {
4268                 .type = CTL_SUMMARY,
4269                 .var_idx = SIZE_MAX,
4270               };
4271             }
4272
4273           /* Pivot categories:
4274
4275              - variable label for nest->vars[0], if vlabel != CTVL_NONE
4276              - category for nest->vars[0], if nest->scale_idx != 0
4277              - variable label for nest->vars[1], if vlabel != CTVL_NONE
4278              - category for nest->vars[1], if nest->scale_idx != 1
4279              ...
4280              - variable label for nest->vars[n - 1], if vlabel != CTVL_NONE
4281              - category for nest->vars[n - 1], if t->label_axis[a] == a && nest->scale_idx != n - 1.
4282              - summary function, if 'a == t->slabels_axis && a ==
4283              t->summary_axis'.
4284
4285              Additional dimensions:
4286
4287              - If 'a == t->slabels_axis && a != t->summary_axis', add a summary
4288              dimension.
4289              - If 't->label_axis[b] == a' for some 'b != a', add a category
4290              dimension to 'a'.
4291           */
4292
4293
4294           struct pivot_category **groups = xnmalloc (1 + 2 * max_depth, sizeof *groups);
4295           int prev_leaf = 0;
4296           for (size_t j = 0; j < n_sorted; j++)
4297             {
4298               struct ctables_cell *cell = sorted[j];
4299               struct ctables_cell *prev = j > 0 ? sorted[j - 1] : NULL;
4300
4301               size_t n_common = 0;
4302               if (j > 0)
4303                 {
4304                   for (; n_common < n_levels; n_common++)
4305                     {
4306                       const struct ctables_level *level = &levels[n_common];
4307                       if (level->type == CTL_CATEGORY)
4308                         {
4309                           size_t var_idx = level->var_idx;
4310                           const struct ctables_category *c = cell->axes[a].cvs[var_idx].category;
4311                           if (prev->axes[a].cvs[var_idx].category != c)
4312                             break;
4313                           else if (c->type != CCT_SUBTOTAL
4314                                    && c->type != CCT_TOTAL
4315                                    && c->type != CCT_POSTCOMPUTE
4316                                    && !value_equal (&prev->axes[a].cvs[var_idx].value,
4317                                                     &cell->axes[a].cvs[var_idx].value,
4318                                                     var_get_type (nest->vars[var_idx])))
4319                             break;
4320                         }
4321                     }
4322                 }
4323
4324               for (size_t k = n_common; k < n_levels; k++)
4325                 {
4326                   const struct ctables_level *level = &levels[k];
4327                   struct pivot_category *parent = k ? groups[k - 1] : d[a]->root;
4328                   if (level->type == CTL_SUMMARY)
4329                     {
4330                       assert (k == n_levels - 1);
4331
4332                       const struct ctables_summary_spec_set *specs = &t->summary_specs;
4333                       for (size_t m = 0; m < specs->n; m++)
4334                         {
4335                           int leaf = pivot_category_create_leaf (
4336                             parent, ctables_summary_label (&specs->specs[m],
4337                                                            t->cilevel));
4338                           if (!m)
4339                             prev_leaf = leaf;
4340                         }
4341                     }
4342                   else
4343                     {
4344                       const struct variable *var = nest->vars[level->var_idx];
4345                       struct pivot_value *label;
4346                       if (level->type == CTL_VAR)
4347                         {
4348                           label = pivot_value_new_variable (var);
4349                           label->variable.show = level->vlabel;
4350                         }
4351                       else if (level->type == CTL_CATEGORY)
4352                         {
4353                           const struct ctables_cell_value *cv = &cell->axes[a].cvs[level->var_idx];
4354                           label = ctables_category_create_value_label (
4355                             t->categories[var_get_dict_index (var)],
4356                             cv->category, var, &cv->value);
4357                         }
4358                       else
4359                         NOT_REACHED ();
4360
4361                       if (k == n_levels - 1)
4362                         prev_leaf = pivot_category_create_leaf (parent, label);
4363                       else
4364                         groups[k] = pivot_category_create_group__ (parent, label);
4365                     }
4366                 }
4367
4368               cell->axes[a].leaf = prev_leaf;
4369             }
4370           free (sorted);
4371           free (groups);
4372           free (levels);
4373           free (sections);
4374
4375         }
4376
4377       d[a]->hide_all_labels = all_hidden_vlabels (t, a);
4378     }
4379
4380   {
4381     size_t n_total_cells = 0;
4382     for (size_t j = 0; j < t->n_sections; j++)
4383       n_total_cells += hmap_count (&t->sections[j].cells);
4384
4385     struct ctables_cell **sorted = xnmalloc (n_total_cells, sizeof *sorted);
4386     size_t n_sorted = 0;
4387     for (size_t j = 0; j < t->n_sections; j++)
4388       {
4389         const struct ctables_section *s = &t->sections[j];
4390         struct ctables_cell *cell;
4391         HMAP_FOR_EACH (cell, struct ctables_cell, node, &s->cells)
4392           if (!cell->hide)
4393             sorted[n_sorted++] = cell;
4394       }
4395     assert (n_sorted <= n_total_cells);
4396     sort (sorted, n_sorted, sizeof *sorted, ctables_cell_compare_leaf_3way,
4397           NULL);
4398     size_t ids[N_CTATS];
4399     memset (ids, 0, sizeof ids);
4400     for (size_t j = 0; j < n_sorted; j++)
4401       {
4402         struct ctables_cell *cell = sorted[j];
4403         for (enum ctables_area_type at = 0; at < N_CTATS; at++)
4404           {
4405             struct ctables_area *area = cell->areas[at];
4406             if (!area->sequence)
4407               area->sequence = ++ids[at];
4408           }
4409       }
4410
4411     free (sorted);
4412   }
4413
4414   for (size_t i = 0; i < t->n_sections; i++)
4415     {
4416       struct ctables_section *s = &t->sections[i];
4417
4418       struct ctables_cell *cell;
4419       HMAP_FOR_EACH (cell, struct ctables_cell, node, &s->cells)
4420         {
4421           if (cell->hide)
4422             continue;
4423
4424           const struct ctables_nest *specs_nest = s->nests[t->summary_axis];
4425           const struct ctables_summary_spec_set *specs = &specs_nest->specs[cell->sv];
4426           for (size_t j = 0; j < specs->n; j++)
4427             {
4428               size_t dindexes[5];
4429               size_t n_dindexes = 0;
4430
4431               if (summary_dimension)
4432                 dindexes[n_dindexes++] = specs->specs[j].axis_idx;
4433
4434               if (categories_dimension)
4435                 {
4436                   const struct ctables_nest *clabels_nest = s->nests[t->clabels_from_axis];
4437                   const struct variable *var = clabels_nest->vars[clabels_nest->n - 1];
4438                   const union value *value = &cell->axes[t->clabels_from_axis].cvs[clabels_nest->n - 1].value;
4439                   const struct ctables_value *ctv = ctables_value_find (t, value, var_get_width (var));
4440                   if (!ctv)
4441                     continue;
4442                   dindexes[n_dindexes++] = ctv->leaf;
4443                 }
4444
4445               for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
4446                 if (d[a])
4447                   {
4448                     int leaf = cell->axes[a].leaf;
4449                     if (a == t->summary_axis && !summary_dimension)
4450                       leaf += j;
4451                     dindexes[n_dindexes++] = leaf;
4452                   }
4453
4454               const struct ctables_summary_spec *ss = &specs->specs[j];
4455
4456               struct fmt_spec format = specs->specs[j].format;
4457               bool is_ctables_format = ss->is_ctables_format;
4458               double d = (cell->postcompute
4459                           ? ctables_cell_calculate_postcompute (
4460                             s, cell, ss, &format, &is_ctables_format, j)
4461                           : ctables_summary_value (cell, &cell->summaries[j],
4462                                                    ss));
4463
4464               struct pivot_value *value;
4465               if (ct->hide_threshold != 0
4466                   && d < ct->hide_threshold
4467                   && ctables_summary_function_is_count (ss->function))
4468                 {
4469                   value = pivot_value_new_user_text_nocopy (
4470                     xasprintf ("<%d", ct->hide_threshold));
4471                 }
4472               else if (d == 0 && ct->zero)
4473                 value = pivot_value_new_user_text (ct->zero, SIZE_MAX);
4474               else if (d == SYSMIS && ct->missing)
4475                 value = pivot_value_new_user_text (ct->missing, SIZE_MAX);
4476               else if (is_ctables_format)
4477                 value = pivot_value_new_user_text_nocopy (
4478                   ctables_format (d, &format, &ct->ctables_formats));
4479               else
4480                 {
4481                   value = pivot_value_new_number (d);
4482                   value->numeric.format = format;
4483                 }
4484               /* XXX should text values be right-justified? */
4485               pivot_table_put (pt, dindexes, n_dindexes, value);
4486             }
4487         }
4488     }
4489
4490   pivot_table_submit (pt);
4491 }
4492
4493 static bool
4494 ctables_check_label_position (struct ctables_table *t, enum pivot_axis_type a)
4495 {
4496   enum pivot_axis_type label_pos = t->label_axis[a];
4497   if (label_pos == a)
4498     return true;
4499
4500   t->clabels_from_axis = a;
4501
4502   const char *subcommand_name = a == PIVOT_AXIS_ROW ? "ROWLABELS" : "COLLABELS";
4503   const char *pos_name = label_pos == PIVOT_AXIS_LAYER ? "LAYER" : "OPPOSITE";
4504
4505   const struct ctables_stack *stack = &t->stacks[a];
4506   if (!stack->n)
4507     return true;
4508
4509   const struct ctables_nest *n0 = &stack->nests[0];
4510   if (n0->n == 0)
4511     {
4512       assert (stack->n == 1);
4513       return true;
4514     }
4515
4516   const struct variable *v0 = n0->vars[n0->n - 1];
4517   struct ctables_categories *c0 = t->categories[var_get_dict_index (v0)];
4518   t->clabels_example = v0;
4519
4520   for (size_t i = 0; i < c0->n_cats; i++)
4521     if (c0->cats[i].type == CCT_FUNCTION)
4522       {
4523         msg (SE, _("%s=%s is not allowed with sorting based "
4524                    "on a summary function."),
4525              subcommand_name, pos_name);
4526         return false;
4527       }
4528   if (n0->n - 1 == n0->scale_idx)
4529     {
4530       msg (SE, _("%s=%s requires the variables to be moved to be categorical, "
4531                  "but %s is a scale variable."),
4532            subcommand_name, pos_name, var_get_name (v0));
4533       return false;
4534     }
4535
4536   for (size_t i = 1; i < stack->n; i++)
4537     {
4538       const struct ctables_nest *ni = &stack->nests[i];
4539       assert (ni->n > 0);
4540       const struct variable *vi = ni->vars[ni->n - 1];
4541       struct ctables_categories *ci = t->categories[var_get_dict_index (vi)];
4542
4543       if (ni->n - 1 == ni->scale_idx)
4544         {
4545           msg (SE, _("%s=%s requires the variables to be moved to be "
4546                      "categorical, but %s is a scale variable."),
4547                subcommand_name, pos_name, var_get_name (vi));
4548           return false;
4549         }
4550       if (var_get_width (v0) != var_get_width (vi))
4551         {
4552           msg (SE, _("%s=%s requires the variables to be "
4553                      "moved to have the same width, but %s has "
4554                      "width %d and %s has width %d."),
4555                subcommand_name, pos_name,
4556                var_get_name (v0), var_get_width (v0),
4557                var_get_name (vi), var_get_width (vi));
4558           return false;
4559         }
4560       if (!val_labs_equal (var_get_value_labels (v0),
4561                            var_get_value_labels (vi)))
4562         {
4563           msg (SE, _("%s=%s requires the variables to be "
4564                      "moved to have the same value labels, but %s "
4565                      "and %s have different value labels."),
4566                subcommand_name, pos_name,
4567                var_get_name (v0), var_get_name (vi));
4568           return false;
4569         }
4570       if (!ctables_categories_equal (c0, ci))
4571         {
4572           msg (SE, _("%s=%s requires the variables to be "
4573                      "moved to have the same category "
4574                      "specifications, but %s and %s have different "
4575                      "category specifications."),
4576                subcommand_name, pos_name,
4577                var_get_name (v0), var_get_name (vi));
4578           return false;
4579         }
4580     }
4581
4582   return true;
4583 }
4584
4585 static size_t
4586 add_sum_var (struct variable *var,
4587              struct variable ***sum_vars, size_t *n, size_t *allocated)
4588 {
4589   for (size_t i = 0; i < *n; i++)
4590     if (var == (*sum_vars)[i])
4591       return i;
4592
4593   if (*n >= *allocated)
4594     *sum_vars = x2nrealloc (*sum_vars, allocated, sizeof **sum_vars);
4595   (*sum_vars)[*n] = var;
4596   return (*n)++;
4597 }
4598
4599 static enum ctables_area_type
4600 rotate_area (enum ctables_area_type area)
4601 {
4602   return area;
4603   switch (area)
4604     {
4605     case CTAT_TABLE:
4606     case CTAT_LAYER:
4607     case CTAT_SUBTABLE:
4608       return area;
4609
4610     case CTAT_LAYERROW:
4611       return CTAT_LAYERCOL;
4612
4613     case CTAT_LAYERCOL:
4614       return CTAT_LAYERROW;
4615
4616     case CTAT_ROW:
4617       return CTAT_COL;
4618
4619     case CTAT_COL:
4620       return CTAT_ROW;
4621     }
4622
4623   NOT_REACHED ();
4624 }
4625
4626 static void
4627 enumerate_sum_vars (const struct ctables_axis *a,
4628                     struct variable ***sum_vars, size_t *n, size_t *allocated)
4629 {
4630   if (!a)
4631     return;
4632
4633   switch (a->op)
4634     {
4635     case CTAO_VAR:
4636       for (size_t i = 0; i < N_CSVS; i++)
4637         for (size_t j = 0; j < a->specs[i].n; j++)
4638           {
4639             struct ctables_summary_spec *spec = &a->specs[i].specs[j];
4640             if (spec->function == CTSF_areaPCT_SUM)
4641               spec->sum_var_idx = add_sum_var (a->var, sum_vars, n, allocated);
4642           }
4643       break;
4644
4645     case CTAO_STACK:
4646     case CTAO_NEST:
4647       for (size_t i = 0; i < 2; i++)
4648         enumerate_sum_vars (a->subs[i], sum_vars, n, allocated);
4649       break;
4650     }
4651 }
4652
4653 static bool
4654 ctables_prepare_table (struct ctables_table *t)
4655 {
4656   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
4657     if (t->axes[a])
4658       {
4659         t->stacks[a] = enumerate_fts (a, t->axes[a]);
4660
4661         for (size_t j = 0; j < t->stacks[a].n; j++)
4662           {
4663             struct ctables_nest *nest = &t->stacks[a].nests[j];
4664             for (enum ctables_area_type at = 0; at < N_CTATS; at++)
4665               {
4666                 nest->areas[at] = xmalloc (nest->n * sizeof *nest->areas[at]);
4667                 nest->n_areas[at] = 0;
4668
4669                 bool add_vars = (at == CTAT_LAYER ? a == PIVOT_AXIS_LAYER
4670                                  : at == CTAT_LAYERROW ? a != PIVOT_AXIS_COLUMN
4671                                  : at == CTAT_LAYERCOL ? a != PIVOT_AXIS_ROW
4672                                  : at == CTAT_TABLE ? false
4673                                  : true);
4674                 if (add_vars)
4675                   for (size_t k = 0; k < nest->n; k++)
4676                     {
4677                       if (k == nest->scale_idx)
4678                         continue;
4679                       nest->areas[at][nest->n_areas[at]++] = k;
4680                     }
4681                 else if (at == CTAT_LAYERCOL && a == PIVOT_AXIS_ROW && t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_COLUMN)
4682                   {
4683                     for (size_t k = nest->n - 1; k < nest->n; k--)
4684                       {
4685                         if (k == nest->scale_idx)
4686                           continue;
4687                         nest->areas[at][nest->n_areas[at]++] = k;
4688                         break;
4689                       }
4690                   }
4691                 else if (at == CTAT_LAYERROW && a == PIVOT_AXIS_COLUMN && t->label_axis[PIVOT_AXIS_COLUMN] == PIVOT_AXIS_ROW)
4692                   {
4693                     for (size_t k = nest->n - 1; k < nest->n; k--)
4694                       {
4695                         if (k == nest->scale_idx)
4696                           continue;
4697                         nest->areas[at][nest->n_areas[at]++] = k;
4698                         break;
4699                       }
4700                   }
4701                 else if (at == CTAT_LAYER && t->label_axis[a] == PIVOT_AXIS_LAYER)
4702                   {
4703                     for (size_t k = nest->n - 1; k < nest->n; k--)
4704                       {
4705                         if (k == nest->scale_idx)
4706                           continue;
4707                         nest->areas[at][nest->n_areas[at]++] = k;
4708                         break;
4709                       }
4710                     continue;
4711                   }
4712
4713                 if (at == CTAT_SUBTABLE && t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_COLUMN)
4714                   {
4715                     size_t n_drop = (a == PIVOT_AXIS_LAYER ? 0
4716                                      : a == PIVOT_AXIS_ROW ? 2
4717                                      : 0);
4718                     for (size_t i = 0; i < n_drop; i++)
4719                       if (nest->n_areas[at] > 0)
4720                         nest->n_areas[at]--;
4721                     continue;
4722                   }
4723                 else if (at == CTAT_SUBTABLE && t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_LAYER)
4724                   {
4725                     if (a == PIVOT_AXIS_ROW)
4726                       {
4727                         size_t n = nest->n_areas[at];
4728                         if (n > 1)
4729                           {
4730                             nest->areas[at][n - 2] = nest->areas[at][n - 1];
4731                             nest->n_areas[at]--;
4732                           }
4733                       }
4734                     else if (a == PIVOT_AXIS_COLUMN)
4735                       {
4736                         if (nest->n_areas[at] > 0)
4737                           nest->n_areas[at]--;
4738                       }
4739                     continue;
4740                   }
4741                 else if (at == CTAT_SUBTABLE && t->label_axis[PIVOT_AXIS_COLUMN] == PIVOT_AXIS_ROW)
4742                   {
4743                     size_t n_drop = (a == PIVOT_AXIS_LAYER ? 0
4744                                      : a == PIVOT_AXIS_COLUMN ? 2
4745                                      : 0);
4746                     for (size_t i = 0; i < n_drop; i++)
4747                       if (nest->n_areas[at] > 0)
4748                         nest->n_areas[at]--;
4749                     continue;
4750                   }
4751                 else if (at == CTAT_ROW && a == PIVOT_AXIS_COLUMN && t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_COLUMN)
4752                   continue;
4753                 else if (at == CTAT_COL && a == PIVOT_AXIS_ROW && t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_COLUMN)
4754                   {
4755                     size_t n = nest->n_areas[at];
4756                     if (n > 1)
4757                       {
4758                         nest->areas[at][n - 2] = nest->areas[at][n - 1];
4759                         nest->n_areas[at]--;
4760                       }
4761                     continue;
4762                   }
4763                 else if (at == CTAT_COL && a == PIVOT_AXIS_ROW && t->label_axis[PIVOT_AXIS_COLUMN] == PIVOT_AXIS_ROW)
4764                   continue;
4765                 else if (at == CTAT_ROW && a == PIVOT_AXIS_COLUMN && t->label_axis[PIVOT_AXIS_COLUMN] == PIVOT_AXIS_ROW)
4766                   {
4767                     size_t n = nest->n_areas[at];
4768                     if (n > 1)
4769                       {
4770                         nest->areas[at][n - 2] = nest->areas[at][n - 1];
4771                         nest->n_areas[at]--;
4772                       }
4773                     continue;
4774                   }
4775
4776                 bool drop_last = (at == CTAT_SUBTABLE ? a != PIVOT_AXIS_LAYER
4777                                   : at == CTAT_ROW ? a == PIVOT_AXIS_COLUMN
4778                                   : at == CTAT_COL ? a == PIVOT_AXIS_ROW
4779                                   : false);
4780                 if (drop_last && nest->n_areas[at] > 0)
4781                   nest->n_areas[at]--;
4782
4783                 bool drop_additional
4784                   = ((t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_COLUMN && (at == CTAT_ROW || at == CTAT_LAYERROW) && a == PIVOT_AXIS_ROW)
4785                      || (t->label_axis[PIVOT_AXIS_COLUMN] == PIVOT_AXIS_ROW && (at == CTAT_COL || at == CTAT_LAYERCOL) && a == PIVOT_AXIS_COLUMN));
4786                 if (drop_additional && nest->n_areas[at] > 0)
4787                   nest->n_areas[at]--;
4788               }
4789           }
4790       }
4791     else
4792       {
4793         struct ctables_nest *nest = xmalloc (sizeof *nest);
4794         *nest = (struct ctables_nest) { .n = 0 };
4795         t->stacks[a] = (struct ctables_stack) { .nests = nest, .n = 1 };
4796
4797         /* There's no point in moving labels away from an axis that has no
4798            labels, so avoid dealing with the special cases around that. */
4799         t->label_axis[a] = a;
4800       }
4801
4802   struct ctables_stack *stack = &t->stacks[t->summary_axis];
4803   for (size_t i = 0; i < stack->n; i++)
4804     {
4805       struct ctables_nest *nest = &stack->nests[i];
4806       if (!nest->specs[CSV_CELL].n)
4807         {
4808           struct ctables_summary_spec_set *specs = &nest->specs[CSV_CELL];
4809           specs->specs = xmalloc (sizeof *specs->specs);
4810           specs->n = 1;
4811
4812           enum ctables_summary_function function
4813             = specs->is_scale ? CTSF_MEAN : CTSF_COUNT;
4814
4815           *specs->specs = (struct ctables_summary_spec) {
4816             .function = function,
4817             .weighted = true,
4818             .format = ctables_summary_default_format (function, specs->var),
4819           };
4820           if (!specs->var)
4821             specs->var = nest->vars[0];
4822
4823           ctables_summary_spec_set_clone (&nest->specs[CSV_TOTAL],
4824                                           &nest->specs[CSV_CELL]);
4825         }
4826       else if (!nest->specs[CSV_TOTAL].n)
4827         ctables_summary_spec_set_clone (&nest->specs[CSV_TOTAL],
4828                                         &nest->specs[CSV_CELL]);
4829
4830       if (t->label_axis[PIVOT_AXIS_ROW] == PIVOT_AXIS_COLUMN
4831           || t->label_axis[PIVOT_AXIS_COLUMN] == PIVOT_AXIS_ROW)
4832         {
4833           for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
4834             for (size_t i = 0; i < nest->specs[sv].n; i++)
4835               {
4836                 struct ctables_summary_spec *ss = &nest->specs[sv].specs[i];
4837                 const struct ctables_function_info *cfi =
4838                   &ctables_function_info[ss->function];
4839                 if (cfi->is_area)
4840                   ss->calc_area = rotate_area (ss->calc_area);
4841               }
4842         }
4843
4844       if (t->ctables->smissing_listwise)
4845         {
4846           struct variable **listwise_vars = NULL;
4847           size_t n = 0;
4848           size_t allocated = 0;
4849
4850           for (size_t j = nest->group_head; j < stack->n; j++)
4851             {
4852               const struct ctables_nest *other_nest = &stack->nests[j];
4853               if (other_nest->group_head != nest->group_head)
4854                 break;
4855
4856               if (nest != other_nest && other_nest->scale_idx < other_nest->n)
4857                 {
4858                   if (n >= allocated)
4859                     listwise_vars = x2nrealloc (listwise_vars, &allocated,
4860                                                 sizeof *listwise_vars);
4861                   listwise_vars[n++] = other_nest->vars[other_nest->scale_idx];
4862                 }
4863             }
4864           for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
4865             {
4866               if (sv > 0)
4867                 listwise_vars = xmemdup (listwise_vars,
4868                                          n * sizeof *listwise_vars);
4869               nest->specs[sv].listwise_vars = listwise_vars;
4870               nest->specs[sv].n_listwise_vars = n;
4871             }
4872         }
4873     }
4874
4875   struct ctables_summary_spec_set *merged = &t->summary_specs;
4876   struct merge_item *items = xnmalloc (N_CSVS * stack->n, sizeof *items);
4877   size_t n_left = 0;
4878   for (size_t j = 0; j < stack->n; j++)
4879     {
4880       const struct ctables_nest *nest = &stack->nests[j];
4881       if (nest->n)
4882         for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
4883           items[n_left++] = (struct merge_item) { .set = &nest->specs[sv] };
4884     }
4885
4886   while (n_left > 0)
4887     {
4888       struct merge_item min = items[0];
4889       for (size_t j = 1; j < n_left; j++)
4890         if (merge_item_compare_3way (&items[j], &min) < 0)
4891           min = items[j];
4892
4893       if (merged->n >= merged->allocated)
4894         merged->specs = x2nrealloc (merged->specs, &merged->allocated,
4895                                     sizeof *merged->specs);
4896       merged->specs[merged->n++] = min.set->specs[min.ofs];
4897
4898       for (size_t j = 0; j < n_left; )
4899         {
4900           if (merge_item_compare_3way (&items[j], &min) == 0)
4901             {
4902               struct merge_item *item = &items[j];
4903               item->set->specs[item->ofs].axis_idx = merged->n - 1;
4904               if (++item->ofs >= item->set->n)
4905                 {
4906                   items[j] = items[--n_left];
4907                   continue;
4908                 }
4909             }
4910           j++;
4911         }
4912     }
4913   free (items);
4914
4915 #if 0
4916   for (size_t j = 0; j < merged->n; j++)
4917     printf ("%s\n", ctables_summary_function_name (merged->specs[j].function));
4918
4919   for (size_t j = 0; j < stack->n; j++)
4920     {
4921       const struct ctables_nest *nest = &stack->nests[j];
4922       for (enum ctables_summary_variant sv = 0; sv < N_CSVS; sv++)
4923         {
4924           const struct ctables_summary_spec_set *specs = &nest->specs[sv];
4925           for (size_t k = 0; k < specs->n; k++)
4926             printf ("(%s, %zu) ", ctables_summary_function_name (specs->specs[k].function),
4927                     specs->specs[k].axis_idx);
4928           printf ("\n");
4929         }
4930     }
4931 #endif
4932
4933   size_t allocated_sum_vars = 0;
4934   enumerate_sum_vars (t->axes[t->summary_axis],
4935                       &t->sum_vars, &t->n_sum_vars, &allocated_sum_vars);
4936
4937   return (ctables_check_label_position (t, PIVOT_AXIS_ROW)
4938           && ctables_check_label_position (t, PIVOT_AXIS_COLUMN));
4939 }
4940
4941 static void
4942 ctables_insert_clabels_values (struct ctables_table *t, const struct ccase *c,
4943                                enum pivot_axis_type a)
4944 {
4945   struct ctables_stack *stack = &t->stacks[a];
4946   for (size_t i = 0; i < stack->n; i++)
4947     {
4948       const struct ctables_nest *nest = &stack->nests[i];
4949       const struct variable *var = nest->vars[nest->n - 1];
4950       const union value *value = case_data (c, var);
4951
4952       if (var_is_numeric (var) && value->f == SYSMIS)
4953         continue;
4954
4955       if (ctables_categories_match (t->categories [var_get_dict_index (var)],
4956                                     value, var))
4957         ctables_value_insert (t, value, var_get_width (var));
4958     }
4959 }
4960
4961 static int
4962 compare_clabels_values_3way (const void *a_, const void *b_, const void *width_)
4963 {
4964   const struct ctables_value *const *ap = a_;
4965   const struct ctables_value *const *bp = b_;
4966   const struct ctables_value *a = *ap;
4967   const struct ctables_value *b = *bp;
4968   const int *width = width_;
4969   return value_compare_3way (&a->value, &b->value, *width);
4970 }
4971
4972 static void
4973 ctables_sort_clabels_values (struct ctables_table *t)
4974 {
4975   const struct variable *v0 = t->clabels_example;
4976   int width = var_get_width (v0);
4977
4978   struct ctables_categories *c0 = t->categories[var_get_dict_index (v0)];
4979   if (c0->show_empty)
4980     {
4981       const struct val_labs *val_labs = var_get_value_labels (v0);
4982       for (const struct val_lab *vl = val_labs_first (val_labs); vl;
4983            vl = val_labs_next (val_labs, vl))
4984         if (ctables_categories_match (c0, &vl->value, v0))
4985           ctables_value_insert (t, &vl->value, width);
4986     }
4987
4988   size_t n = hmap_count (&t->clabels_values_map);
4989   t->clabels_values = xnmalloc (n, sizeof *t->clabels_values);
4990
4991   struct ctables_value *clv;
4992   size_t i = 0;
4993   HMAP_FOR_EACH (clv, struct ctables_value, node, &t->clabels_values_map)
4994     t->clabels_values[i++] = clv;
4995   t->n_clabels_values = n;
4996   assert (i == n);
4997
4998   sort (t->clabels_values, n, sizeof *t->clabels_values,
4999         compare_clabels_values_3way, &width);
5000
5001   for (size_t i = 0; i < n; i++)
5002     t->clabels_values[i]->leaf = i;
5003 }
5004
5005 static void
5006 ctables_add_category_occurrences (const struct variable *var,
5007                                   struct hmap *occurrences,
5008                                   const struct ctables_categories *cats)
5009 {
5010   const struct val_labs *val_labs = var_get_value_labels (var);
5011
5012   for (size_t i = 0; i < cats->n_cats; i++)
5013     {
5014       const struct ctables_category *c = &cats->cats[i];
5015       switch (c->type)
5016         {
5017         case CCT_NUMBER:
5018           ctables_add_occurrence (var, &(const union value) { .f = c->number },
5019                                   occurrences);
5020           break;
5021
5022         case CCT_STRING:
5023           {
5024             int width = var_get_width (var);
5025             union value value;
5026             value_init (&value, width);
5027             value_copy_buf_rpad (&value, width,
5028                                  CHAR_CAST (uint8_t *, c->string.string),
5029                                  c->string.length, ' ');
5030             ctables_add_occurrence (var, &value, occurrences);
5031             value_destroy (&value, width);
5032           }
5033           break;
5034
5035         case CCT_NRANGE:
5036           assert (var_is_numeric (var));
5037           for (const struct val_lab *vl = val_labs_first (val_labs); vl;
5038                vl = val_labs_next (val_labs, vl))
5039             if (vl->value.f >= c->nrange[0] && vl->value.f <= c->nrange[1])
5040               ctables_add_occurrence (var, &vl->value, occurrences);
5041           break;
5042
5043         case CCT_SRANGE:
5044           assert (var_is_alpha (var));
5045           for (const struct val_lab *vl = val_labs_first (val_labs); vl;
5046                vl = val_labs_next (val_labs, vl))
5047             if (in_string_range (&vl->value, var, c->srange))
5048               ctables_add_occurrence (var, &vl->value, occurrences);
5049           break;
5050
5051         case CCT_MISSING:
5052           for (const struct val_lab *vl = val_labs_first (val_labs); vl;
5053                vl = val_labs_next (val_labs, vl))
5054             if (var_is_value_missing (var, &vl->value))
5055               ctables_add_occurrence (var, &vl->value, occurrences);
5056           break;
5057
5058         case CCT_OTHERNM:
5059           for (const struct val_lab *vl = val_labs_first (val_labs); vl;
5060                vl = val_labs_next (val_labs, vl))
5061             ctables_add_occurrence (var, &vl->value, occurrences);
5062           break;
5063
5064         case CCT_POSTCOMPUTE:
5065           break;
5066
5067         case CCT_SUBTOTAL:
5068         case CCT_TOTAL:
5069           break;
5070
5071         case CCT_VALUE:
5072         case CCT_LABEL:
5073         case CCT_FUNCTION:
5074           for (const struct val_lab *vl = val_labs_first (val_labs); vl;
5075                vl = val_labs_next (val_labs, vl))
5076             if (c->include_missing || !var_is_value_missing (var, &vl->value))
5077               ctables_add_occurrence (var, &vl->value, occurrences);
5078           break;
5079
5080         case CCT_EXCLUDED_MISSING:
5081           break;
5082         }
5083     }
5084 }
5085
5086 static void
5087 ctables_section_recurse_add_empty_categories (
5088   struct ctables_section *s,
5089   const struct ctables_category *cats[PIVOT_N_AXES][10], struct ccase *c,
5090   enum pivot_axis_type a, size_t a_idx)
5091 {
5092   if (a >= PIVOT_N_AXES)
5093     ctables_cell_insert__ (s, c, cats);
5094   else if (!s->nests[a] || a_idx >= s->nests[a]->n)
5095     ctables_section_recurse_add_empty_categories (s, cats, c, a + 1, 0);
5096   else
5097     {
5098       const struct variable *var = s->nests[a]->vars[a_idx];
5099       const struct ctables_categories *categories = s->table->categories[
5100         var_get_dict_index (var)];
5101       int width = var_get_width (var);
5102       const struct hmap *occurrences = &s->occurrences[a][a_idx];
5103       const struct ctables_occurrence *o;
5104       HMAP_FOR_EACH (o, struct ctables_occurrence, node, occurrences)
5105         {
5106           union value *value = case_data_rw (c, var);
5107           value_destroy (value, width);
5108           value_clone (value, &o->value, width);
5109           cats[a][a_idx] = ctables_categories_match (categories, value, var);
5110           assert (cats[a][a_idx] != NULL);
5111           ctables_section_recurse_add_empty_categories (s, cats, c, a, a_idx + 1);
5112         }
5113
5114       for (size_t i = 0; i < categories->n_cats; i++)
5115         {
5116           const struct ctables_category *cat = &categories->cats[i];
5117           if (cat->type == CCT_POSTCOMPUTE)
5118             {
5119               cats[a][a_idx] = cat;
5120               ctables_section_recurse_add_empty_categories (s, cats, c, a, a_idx + 1);
5121             }
5122         }
5123     }
5124 }
5125
5126 static void
5127 ctables_section_add_empty_categories (struct ctables_section *s)
5128 {
5129   bool show_empty = false;
5130   for (size_t a = 0; a < PIVOT_N_AXES; a++)
5131     if (s->nests[a])
5132       for (size_t k = 0; k < s->nests[a]->n; k++)
5133         if (k != s->nests[a]->scale_idx)
5134           {
5135             const struct variable *var = s->nests[a]->vars[k];
5136             const struct ctables_categories *cats = s->table->categories[
5137               var_get_dict_index (var)];
5138             if (cats->show_empty)
5139               {
5140                 show_empty = true;
5141                 ctables_add_category_occurrences (var, &s->occurrences[a][k], cats);
5142               }
5143           }
5144   if (!show_empty)
5145     return;
5146
5147   const struct ctables_category *cats[PIVOT_N_AXES][10]; /* XXX */
5148   struct ccase *c = case_create (dict_get_proto (s->table->ctables->dict));
5149   ctables_section_recurse_add_empty_categories (s, cats, c, 0, 0);
5150   case_unref (c);
5151 }
5152
5153 static void
5154 ctables_section_clear (struct ctables_section *s)
5155 {
5156   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
5157     {
5158       const struct ctables_nest *nest = s->nests[a];
5159       for (size_t i = 0; i < nest->n; i++)
5160         if (i != nest->scale_idx)
5161           {
5162             const struct variable *var = nest->vars[i];
5163             int width = var_get_width (var);
5164             struct ctables_occurrence *o, *next;
5165             struct hmap *map = &s->occurrences[a][i];
5166             HMAP_FOR_EACH_SAFE (o, next, struct ctables_occurrence, node, map)
5167               {
5168                 value_destroy (&o->value, width);
5169                 hmap_delete (map, &o->node);
5170                 free (o);
5171               }
5172             hmap_shrink (map);
5173           }
5174     }
5175
5176   struct ctables_cell *cell, *next_cell;
5177   HMAP_FOR_EACH_SAFE (cell, next_cell, struct ctables_cell, node, &s->cells)
5178     {
5179       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
5180         {
5181           const struct ctables_nest *nest = s->nests[a];
5182           for (size_t i = 0; i < nest->n; i++)
5183             if (i != nest->scale_idx)
5184               value_destroy (&cell->axes[a].cvs[i].value,
5185                              var_get_width (nest->vars[i]));
5186           free (cell->axes[a].cvs);
5187         }
5188
5189       const struct ctables_nest *ss = s->nests[s->table->summary_axis];
5190       const struct ctables_summary_spec_set *specs = &ss->specs[cell->sv];
5191       for (size_t i = 0; i < specs->n; i++)
5192         ctables_summary_uninit (&cell->summaries[i], &specs->specs[i]);
5193       free (cell->summaries);
5194
5195       hmap_delete (&s->cells, &cell->node);
5196       free (cell);
5197     }
5198   hmap_shrink (&s->cells);
5199
5200   for (enum ctables_area_type at = 0; at < N_CTATS; at++)
5201     {
5202       struct ctables_area *area, *next_area;
5203       HMAP_FOR_EACH_SAFE (area, next_area, struct ctables_area, node,
5204                           &s->areas[at])
5205         {
5206           free (area->sums);
5207           hmap_delete (&s->areas[at], &area->node);
5208           free (area);
5209         }
5210       hmap_shrink (&s->areas[at]);
5211     }
5212 }
5213
5214 static void
5215 ctables_section_uninit (struct ctables_section *s)
5216 {
5217   ctables_section_clear (s);
5218
5219   for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
5220     {
5221       struct ctables_nest *nest = s->nests[a];
5222       for (size_t i = 0; i < nest->n; i++)
5223         hmap_destroy (&s->occurrences[a][i]);
5224       free (s->occurrences[a]);
5225     }
5226
5227   hmap_destroy (&s->cells);
5228   for (size_t i = 0; i < N_CTATS; i++)
5229     hmap_destroy (&s->areas[i]);
5230 }
5231
5232 static void
5233 ctables_table_clear (struct ctables_table *t)
5234 {
5235   for (size_t i = 0; i < t->n_sections; i++)
5236     ctables_section_clear (&t->sections[i]);
5237
5238   if (t->clabels_example)
5239     {
5240       int width = var_get_width (t->clabels_example);
5241       struct ctables_value *value, *next_value;
5242       HMAP_FOR_EACH_SAFE (value, next_value, struct ctables_value, node,
5243                           &t->clabels_values_map)
5244         {
5245           value_destroy (&value->value, width);
5246           hmap_delete (&t->clabels_values_map, &value->node);
5247           free (value);
5248         }
5249       hmap_shrink (&t->clabels_values_map);
5250
5251       free (t->clabels_values);
5252       t->clabels_values = NULL;
5253       t->n_clabels_values = 0;
5254     }
5255 }
5256
5257 static bool
5258 ctables_execute (struct dataset *ds, struct casereader *input,
5259                  struct ctables *ct)
5260 {
5261   for (size_t i = 0; i < ct->n_tables; i++)
5262     {
5263       struct ctables_table *t = ct->tables[i];
5264       t->sections = xnmalloc (MAX (1, t->stacks[PIVOT_AXIS_ROW].n) *
5265                               MAX (1, t->stacks[PIVOT_AXIS_COLUMN].n) *
5266                               MAX (1, t->stacks[PIVOT_AXIS_LAYER].n),
5267                               sizeof *t->sections);
5268       size_t ix[PIVOT_N_AXES];
5269       ctables_table_add_section (t, 0, ix);
5270     }
5271
5272   struct dictionary *dict = dataset_dict (ds);
5273
5274   bool splitting = dict_get_split_type (dict) == SPLIT_SEPARATE;
5275   struct casegrouper *grouper
5276     = (splitting
5277        ? casegrouper_create_splits (input, dict)
5278        : casegrouper_create_vars (input, NULL, 0));
5279   struct casereader *group;
5280   while (casegrouper_get_next_group (grouper, &group))
5281     {
5282       if (splitting)
5283         {
5284           struct ccase *c = casereader_peek (group, 0);
5285           if (c != NULL)
5286             {
5287               output_split_file_values (ds, c);
5288               case_unref (c);
5289             }
5290         }
5291
5292       bool warn_on_invalid = true;
5293       for (struct ccase *c = casereader_read (group); c;
5294            case_unref (c), c = casereader_read (group))
5295         {
5296           double d_weight = dict_get_case_weight (dict, c, &warn_on_invalid);
5297           double e_weight = (ct->e_weight
5298                              ? var_force_valid_weight (ct->e_weight,
5299                                                        case_num (c, ct->e_weight),
5300                                                        &warn_on_invalid)
5301                              : d_weight);
5302
5303           for (size_t i = 0; i < ct->n_tables; i++)
5304             {
5305               struct ctables_table *t = ct->tables[i];
5306
5307               for (size_t j = 0; j < t->n_sections; j++)
5308                 ctables_cell_insert (&t->sections[j], c, d_weight, e_weight);
5309
5310               for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
5311                 if (t->label_axis[a] != a)
5312                   ctables_insert_clabels_values (t, c, a);
5313             }
5314         }
5315       casereader_destroy (group);
5316
5317       for (size_t i = 0; i < ct->n_tables; i++)
5318         {
5319           struct ctables_table *t = ct->tables[i];
5320
5321           if (t->clabels_example)
5322             ctables_sort_clabels_values (t);
5323
5324           for (size_t j = 0; j < t->n_sections; j++)
5325             ctables_section_add_empty_categories (&t->sections[j]);
5326
5327           ctables_table_output (ct, t);
5328           ctables_table_clear (t);
5329         }
5330     }
5331   return casegrouper_destroy (grouper);
5332 }
5333 \f
5334 /* Postcomputes. */
5335
5336 typedef struct ctables_pcexpr *parse_recursively_func (struct lexer *,
5337                                                        struct dictionary *);
5338
5339 static void
5340 ctables_pcexpr_destroy (struct ctables_pcexpr *e)
5341 {
5342   if (e)
5343     {
5344       switch (e->op)
5345         {
5346         case CTPO_CAT_STRING:
5347           ss_dealloc (&e->string);
5348           break;
5349
5350         case CTPO_CAT_SRANGE:
5351           for (size_t i = 0; i < 2; i++)
5352             ss_dealloc (&e->srange[i]);
5353           break;
5354
5355         case CTPO_ADD:
5356         case CTPO_SUB:
5357         case CTPO_MUL:
5358         case CTPO_DIV:
5359         case CTPO_POW:
5360         case CTPO_NEG:
5361           for (size_t i = 0; i < 2; i++)
5362             ctables_pcexpr_destroy (e->subs[i]);
5363           break;
5364
5365         case CTPO_CONSTANT:
5366         case CTPO_CAT_NUMBER:
5367         case CTPO_CAT_NRANGE:
5368         case CTPO_CAT_MISSING:
5369         case CTPO_CAT_OTHERNM:
5370         case CTPO_CAT_SUBTOTAL:
5371         case CTPO_CAT_TOTAL:
5372           break;
5373         }
5374
5375       msg_location_destroy (e->location);
5376       free (e);
5377     }
5378 }
5379
5380 static struct ctables_pcexpr *
5381 ctables_pcexpr_allocate_binary (enum ctables_postcompute_op op,
5382                                 struct ctables_pcexpr *sub0,
5383                                 struct ctables_pcexpr *sub1)
5384 {
5385   struct ctables_pcexpr *e = xmalloc (sizeof *e);
5386   *e = (struct ctables_pcexpr) {
5387     .op = op,
5388     .subs = { sub0, sub1 },
5389     .location = msg_location_merged (sub0->location, sub1->location),
5390   };
5391   return e;
5392 }
5393
5394 /* How to parse an operator. */
5395 struct operator
5396   {
5397     enum token_type token;
5398     enum ctables_postcompute_op op;
5399   };
5400
5401 static const struct operator *
5402 ctables_pcexpr_match_operator (struct lexer *lexer,
5403                               const struct operator ops[], size_t n_ops)
5404 {
5405   for (const struct operator *op = ops; op < ops + n_ops; op++)
5406     if (lex_token (lexer) == op->token)
5407       {
5408         if (op->token != T_NEG_NUM)
5409           lex_get (lexer);
5410
5411         return op;
5412       }
5413
5414   return NULL;
5415 }
5416
5417 static struct ctables_pcexpr *
5418 ctables_pcexpr_parse_binary_operators__ (
5419   struct lexer *lexer, struct dictionary *dict,
5420   const struct operator ops[], size_t n_ops,
5421   parse_recursively_func *parse_next_level,
5422   const char *chain_warning, struct ctables_pcexpr *lhs)
5423 {
5424   for (int op_count = 0; ; op_count++)
5425     {
5426       const struct operator *op
5427         = ctables_pcexpr_match_operator (lexer, ops, n_ops);
5428       if (!op)
5429         {
5430           if (op_count > 1 && chain_warning)
5431             msg_at (SW, lhs->location, "%s", chain_warning);
5432
5433           return lhs;
5434         }
5435
5436       struct ctables_pcexpr *rhs = parse_next_level (lexer, dict);
5437       if (!rhs)
5438         {
5439           ctables_pcexpr_destroy (lhs);
5440           return NULL;
5441         }
5442
5443       lhs = ctables_pcexpr_allocate_binary (op->op, lhs, rhs);
5444     }
5445 }
5446
5447 static struct ctables_pcexpr *
5448 ctables_pcexpr_parse_binary_operators (
5449   struct lexer *lexer, struct dictionary *dict,
5450   const struct operator ops[], size_t n_ops,
5451   parse_recursively_func *parse_next_level, const char *chain_warning)
5452 {
5453   struct ctables_pcexpr *lhs = parse_next_level (lexer, dict);
5454   if (!lhs)
5455     return NULL;
5456
5457   return ctables_pcexpr_parse_binary_operators__ (lexer, dict, ops, n_ops,
5458                                                  parse_next_level,
5459                                                  chain_warning, lhs);
5460 }
5461
5462 static struct ctables_pcexpr *ctables_pcexpr_parse_add (struct lexer *,
5463                                                         struct dictionary *);
5464
5465 static struct ctables_pcexpr
5466 ctpo_cat_nrange (double low, double high)
5467 {
5468   return (struct ctables_pcexpr) {
5469     .op = CTPO_CAT_NRANGE,
5470     .nrange = { low, high },
5471   };
5472 }
5473
5474 static struct ctables_pcexpr
5475 ctpo_cat_srange (struct substring low, struct substring high)
5476 {
5477   return (struct ctables_pcexpr) {
5478     .op = CTPO_CAT_SRANGE,
5479     .srange = { low, high },
5480   };
5481 }
5482
5483 static struct ctables_pcexpr *
5484 ctables_pcexpr_parse_primary (struct lexer *lexer, struct dictionary *dict)
5485 {
5486   int start_ofs = lex_ofs (lexer);
5487   struct ctables_pcexpr e;
5488   if (lex_is_number (lexer))
5489     {
5490       e = (struct ctables_pcexpr) { .op = CTPO_CONSTANT,
5491                                     .number = lex_number (lexer) };
5492       lex_get (lexer);
5493     }
5494   else if (lex_match_id (lexer, "MISSING"))
5495     e = (struct ctables_pcexpr) { .op = CTPO_CAT_MISSING };
5496   else if (lex_match_id (lexer, "OTHERNM"))
5497     e = (struct ctables_pcexpr) { .op = CTPO_CAT_OTHERNM };
5498   else if (lex_match_id (lexer, "TOTAL"))
5499     e = (struct ctables_pcexpr) { .op = CTPO_CAT_TOTAL };
5500   else if (lex_match_id (lexer, "SUBTOTAL"))
5501     {
5502       size_t subtotal_index = 0;
5503       if (lex_match (lexer, T_LBRACK))
5504         {
5505           if (!lex_force_int_range (lexer, "SUBTOTAL", 1, LONG_MAX))
5506             return NULL;
5507           subtotal_index = lex_integer (lexer);
5508           lex_get (lexer);
5509           if (!lex_force_match (lexer, T_RBRACK))
5510             return NULL;
5511         }
5512       e = (struct ctables_pcexpr) { .op = CTPO_CAT_SUBTOTAL,
5513                                     .subtotal_index = subtotal_index };
5514     }
5515   else if (lex_match (lexer, T_LBRACK))
5516     {
5517       if (lex_match_id (lexer, "LO"))
5518         {
5519           if (!lex_force_match_id (lexer, "THRU"))
5520             return false;
5521
5522           if (lex_is_string (lexer))
5523             {
5524               struct substring low = { .string = NULL };
5525               struct substring high = parse_substring (lexer, dict);
5526               e = ctpo_cat_srange (low, high);
5527             }
5528           else
5529             {
5530               if (!lex_force_num (lexer))
5531                 return false;
5532               e = ctpo_cat_nrange (-DBL_MAX, lex_number (lexer));
5533               lex_get (lexer);
5534             }
5535         }
5536       else if (lex_is_number (lexer))
5537         {
5538           double number = lex_number (lexer);
5539           lex_get (lexer);
5540           if (lex_match_id (lexer, "THRU"))
5541             {
5542               if (lex_match_id (lexer, "HI"))
5543                 e = ctpo_cat_nrange (number, DBL_MAX);
5544               else
5545                 {
5546                   if (!lex_force_num (lexer))
5547                     return false;
5548                   e = ctpo_cat_nrange (number, lex_number (lexer));
5549                   lex_get (lexer);
5550                 }
5551             }
5552           else
5553             e = (struct ctables_pcexpr) { .op = CTPO_CAT_NUMBER,
5554                                           .number = number };
5555         }
5556       else if (lex_is_string (lexer))
5557         {
5558           struct substring s = parse_substring (lexer, dict);
5559
5560           if (lex_match_id (lexer, "THRU"))
5561             {
5562               struct substring high;
5563
5564               if (lex_match_id (lexer, "HI"))
5565                 high = (struct substring) { .string = NULL };
5566               else
5567                 {
5568                   if (!lex_force_string (lexer))
5569                     {
5570                       ss_dealloc (&s);
5571                       return false;
5572                     }
5573                   high = parse_substring (lexer, dict);
5574                 }
5575
5576               e = ctpo_cat_srange (s, high);
5577             }
5578           else
5579             e = (struct ctables_pcexpr) { .op = CTPO_CAT_STRING, .string = s };
5580         }
5581       else
5582         {
5583           lex_error (lexer, NULL);
5584           return NULL;
5585         }
5586
5587       if (!lex_force_match (lexer, T_RBRACK))
5588         {
5589           if (e.op == CTPO_CAT_STRING)
5590             ss_dealloc (&e.string);
5591           else if (e.op == CTPO_CAT_SRANGE)
5592             {
5593               ss_dealloc (&e.srange[0]);
5594               ss_dealloc (&e.srange[1]);
5595             }
5596           return NULL;
5597         }
5598     }
5599   else if (lex_match (lexer, T_LPAREN))
5600     {
5601       struct ctables_pcexpr *ep = ctables_pcexpr_parse_add (lexer, dict);
5602       if (!ep)
5603         return NULL;
5604       if (!lex_force_match (lexer, T_RPAREN))
5605         {
5606           ctables_pcexpr_destroy (ep);
5607           return NULL;
5608         }
5609       return ep;
5610     }
5611   else
5612     {
5613       lex_error (lexer, NULL);
5614       return NULL;
5615     }
5616
5617   e.location = lex_ofs_location (lexer, start_ofs, lex_ofs (lexer) - 1);
5618   return xmemdup (&e, sizeof e);
5619 }
5620
5621 static struct ctables_pcexpr *
5622 ctables_pcexpr_allocate_neg (struct ctables_pcexpr *sub,
5623                              struct lexer *lexer, int start_ofs)
5624 {
5625   struct ctables_pcexpr *e = xmalloc (sizeof *e);
5626   *e = (struct ctables_pcexpr) {
5627     .op = CTPO_NEG,
5628     .subs = { sub },
5629     .location = lex_ofs_location (lexer, start_ofs, lex_ofs (lexer) - 1),
5630   };
5631   return e;
5632 }
5633
5634 static struct ctables_pcexpr *
5635 ctables_pcexpr_parse_exp (struct lexer *lexer, struct dictionary *dict)
5636 {
5637   static const struct operator op = { T_EXP, CTPO_POW };
5638
5639   const char *chain_warning =
5640     _("The exponentiation operator (`**') is left-associative: "
5641       "`a**b**c' equals `(a**b)**c', not `a**(b**c)'.  "
5642       "To disable this warning, insert parentheses.");
5643
5644   if (lex_token (lexer) != T_NEG_NUM || lex_next_token (lexer, 1) != T_EXP)
5645     return ctables_pcexpr_parse_binary_operators (lexer, dict, &op, 1,
5646                                                   ctables_pcexpr_parse_primary,
5647                                                   chain_warning);
5648
5649   /* Special case for situations like "-5**6", which must be parsed as
5650      -(5**6). */
5651
5652   int start_ofs = lex_ofs (lexer);
5653   struct ctables_pcexpr *lhs = xmalloc (sizeof *lhs);
5654   *lhs = (struct ctables_pcexpr) {
5655     .op = CTPO_CONSTANT,
5656     .number = -lex_tokval (lexer),
5657     .location = lex_ofs_location (lexer, start_ofs, lex_ofs (lexer)),
5658   };
5659   lex_get (lexer);
5660
5661   struct ctables_pcexpr *node = ctables_pcexpr_parse_binary_operators__ (
5662     lexer, dict, &op, 1,
5663     ctables_pcexpr_parse_primary, chain_warning, lhs);
5664   if (!node)
5665     return NULL;
5666
5667   return ctables_pcexpr_allocate_neg (node, lexer, start_ofs);
5668 }
5669
5670 /* Parses the unary minus level. */
5671 static struct ctables_pcexpr *
5672 ctables_pcexpr_parse_neg (struct lexer *lexer, struct dictionary *dict)
5673 {
5674   int start_ofs = lex_ofs (lexer);
5675   if (!lex_match (lexer, T_DASH))
5676     return ctables_pcexpr_parse_exp (lexer, dict);
5677
5678   struct ctables_pcexpr *inner = ctables_pcexpr_parse_neg (lexer, dict);
5679   if (!inner)
5680     return NULL;
5681
5682   return ctables_pcexpr_allocate_neg (inner, lexer, start_ofs);
5683 }
5684
5685 /* Parses the multiplication and division level. */
5686 static struct ctables_pcexpr *
5687 ctables_pcexpr_parse_mul (struct lexer *lexer, struct dictionary *dict)
5688 {
5689   static const struct operator ops[] =
5690     {
5691       { T_ASTERISK, CTPO_MUL },
5692       { T_SLASH, CTPO_DIV },
5693     };
5694
5695   return ctables_pcexpr_parse_binary_operators (lexer, dict, ops,
5696                                                sizeof ops / sizeof *ops,
5697                                                ctables_pcexpr_parse_neg, NULL);
5698 }
5699
5700 /* Parses the addition and subtraction level. */
5701 static struct ctables_pcexpr *
5702 ctables_pcexpr_parse_add (struct lexer *lexer, struct dictionary *dict)
5703 {
5704   static const struct operator ops[] =
5705     {
5706       { T_PLUS, CTPO_ADD },
5707       { T_DASH, CTPO_SUB },
5708       { T_NEG_NUM, CTPO_ADD },
5709     };
5710
5711   return ctables_pcexpr_parse_binary_operators (lexer, dict,
5712                                                ops, sizeof ops / sizeof *ops,
5713                                                ctables_pcexpr_parse_mul, NULL);
5714 }
5715
5716 static struct ctables_postcompute *
5717 ctables_find_postcompute (struct ctables *ct, const char *name)
5718 {
5719   struct ctables_postcompute *pc;
5720   HMAP_FOR_EACH_WITH_HASH (pc, struct ctables_postcompute, hmap_node,
5721                            utf8_hash_case_string (name, 0), &ct->postcomputes)
5722     if (!utf8_strcasecmp (pc->name, name))
5723       return pc;
5724   return NULL;
5725 }
5726
5727 static bool
5728 ctables_parse_pcompute (struct lexer *lexer, struct dictionary *dict,
5729                         struct ctables *ct)
5730 {
5731   int pcompute_start = lex_ofs (lexer) - 1;
5732
5733   if (!lex_match (lexer, T_AND))
5734     {
5735       lex_error_expecting (lexer, "&");
5736       return false;
5737     }
5738   if (!lex_force_id (lexer))
5739     return false;
5740
5741   char *name = ss_xstrdup (lex_tokss (lexer));
5742
5743   lex_get (lexer);
5744   if (!lex_force_match (lexer, T_EQUALS)
5745       || !lex_force_match_id (lexer, "EXPR")
5746       || !lex_force_match (lexer, T_LPAREN))
5747     {
5748       free (name);
5749       return false;
5750     }
5751
5752   int expr_start = lex_ofs (lexer);
5753   struct ctables_pcexpr *expr = ctables_pcexpr_parse_add (lexer, dict);
5754   int expr_end = lex_ofs (lexer) - 1;
5755   if (!expr || !lex_force_match (lexer, T_RPAREN))
5756     {
5757       ctables_pcexpr_destroy (expr);
5758       free (name);
5759       return false;
5760     }
5761   int pcompute_end = lex_ofs (lexer) - 1;
5762
5763   struct msg_location *location = lex_ofs_location (lexer, pcompute_start,
5764                                                     pcompute_end);
5765
5766   struct ctables_postcompute *pc = ctables_find_postcompute (ct, name);
5767   if (pc)
5768     {
5769       msg_at (SW, location, _("New definition of &%s will override the "
5770                               "previous definition."),
5771               pc->name);
5772       msg_at (SN, pc->location, _("This is the previous definition."));
5773
5774       ctables_pcexpr_destroy (pc->expr);
5775       msg_location_destroy (pc->location);
5776       free (name);
5777     }
5778   else
5779     {
5780       pc = xmalloc (sizeof *pc);
5781       *pc = (struct ctables_postcompute) { .name = name };
5782       hmap_insert (&ct->postcomputes, &pc->hmap_node,
5783                    utf8_hash_case_string (pc->name, 0));
5784     }
5785   pc->expr = expr;
5786   pc->location = location;
5787   if (!pc->label)
5788     pc->label = lex_ofs_representation (lexer, expr_start, expr_end);
5789   return true;
5790 }
5791
5792 static bool
5793 ctables_parse_pproperties_format (struct lexer *lexer,
5794                                   struct ctables_summary_spec_set *sss)
5795 {
5796   *sss = (struct ctables_summary_spec_set) { .n = 0 };
5797
5798   while (lex_token (lexer) != T_ENDCMD && lex_token (lexer) != T_SLASH
5799          && !(lex_token (lexer) == T_ID
5800               && (lex_id_match (ss_cstr ("LABEL"), lex_tokss (lexer))
5801                   || lex_id_match (ss_cstr ("HIDESOURCECATS"),
5802                                    lex_tokss (lexer)))))
5803     {
5804       /* Parse function. */
5805       enum ctables_summary_function function;
5806       bool weighted;
5807       enum ctables_area_type area;
5808       if (!parse_ctables_summary_function (lexer, &function, &weighted, &area))
5809         goto error;
5810
5811       /* Parse percentile. */
5812       double percentile = 0;
5813       if (function == CTSF_PTILE)
5814         {
5815           if (!lex_force_num_range_closed (lexer, "PTILE", 0, 100))
5816             goto error;
5817           percentile = lex_number (lexer);
5818           lex_get (lexer);
5819         }
5820
5821       /* Parse format. */
5822       struct fmt_spec format;
5823       bool is_ctables_format;
5824       if (!parse_ctables_format_specifier (lexer, &format, &is_ctables_format))
5825         goto error;
5826
5827       if (sss->n >= sss->allocated)
5828         sss->specs = x2nrealloc (sss->specs, &sss->allocated,
5829                                  sizeof *sss->specs);
5830       sss->specs[sss->n++] = (struct ctables_summary_spec) {
5831         .function = function,
5832         .weighted = weighted,
5833         .calc_area = area,
5834         .user_area = area,
5835         .percentile = percentile,
5836         .format = format,
5837         .is_ctables_format = is_ctables_format,
5838       };
5839     }
5840   return true;
5841
5842 error:
5843   ctables_summary_spec_set_uninit (sss);
5844   return false;
5845 }
5846
5847 static bool
5848 ctables_parse_pproperties (struct lexer *lexer, struct ctables *ct)
5849 {
5850   struct ctables_postcompute **pcs = NULL;
5851   size_t n_pcs = 0;
5852   size_t allocated_pcs = 0;
5853
5854   while (lex_match (lexer, T_AND))
5855     {
5856       if (!lex_force_id (lexer))
5857         goto error;
5858       struct ctables_postcompute *pc
5859         = ctables_find_postcompute (ct, lex_tokcstr (lexer));
5860       if (!pc)
5861         {
5862           msg (SE, _("Unknown computed category &%s."), lex_tokcstr (lexer));
5863           goto error;
5864         }
5865       lex_get (lexer);
5866
5867       if (n_pcs >= allocated_pcs)
5868         pcs = x2nrealloc (pcs, &allocated_pcs, sizeof *pcs);
5869       pcs[n_pcs++] = pc;
5870     }
5871
5872   while (lex_token (lexer) != T_SLASH && lex_token (lexer) != T_ENDCMD)
5873     {
5874       if (lex_match_id (lexer, "LABEL"))
5875         {
5876           lex_match (lexer, T_EQUALS);
5877           if (!lex_force_string (lexer))
5878             goto error;
5879
5880           for (size_t i = 0; i < n_pcs; i++)
5881             {
5882               free (pcs[i]->label);
5883               pcs[i]->label = ss_xstrdup (lex_tokss (lexer));
5884             }
5885
5886           lex_get (lexer);
5887         }
5888       else if (lex_match_id (lexer, "FORMAT"))
5889         {
5890           lex_match (lexer, T_EQUALS);
5891
5892           struct ctables_summary_spec_set sss;
5893           if (!ctables_parse_pproperties_format (lexer, &sss))
5894             goto error;
5895
5896           for (size_t i = 0; i < n_pcs; i++)
5897             {
5898               if (pcs[i]->specs)
5899                 ctables_summary_spec_set_uninit (pcs[i]->specs);
5900               else
5901                 pcs[i]->specs = xmalloc (sizeof *pcs[i]->specs);
5902               ctables_summary_spec_set_clone (pcs[i]->specs, &sss);
5903             }
5904           ctables_summary_spec_set_uninit (&sss);
5905         }
5906       else if (lex_match_id (lexer, "HIDESOURCECATS"))
5907         {
5908           lex_match (lexer, T_EQUALS);
5909           bool hide_source_cats;
5910           if (!parse_bool (lexer, &hide_source_cats))
5911             goto error;
5912           for (size_t i = 0; i < n_pcs; i++)
5913             pcs[i]->hide_source_cats = hide_source_cats;
5914         }
5915       else
5916         {
5917           lex_error_expecting (lexer, "LABEL", "FORMAT", "HIDESOURCECATS");
5918           goto error;
5919         }
5920     }
5921   free (pcs);
5922   return true;
5923
5924 error:
5925   free (pcs);
5926   return false;
5927 }
5928
5929 static void
5930 put_strftime (struct string *out, time_t now, const char *format)
5931 {
5932   const struct tm *tm = localtime (&now);
5933   char value[128];
5934   strftime (value, sizeof value, format, tm);
5935   ds_put_cstr (out, value);
5936 }
5937
5938 static bool
5939 skip_prefix (struct substring *s, struct substring prefix)
5940 {
5941   if (ss_starts_with (*s, prefix))
5942     {
5943       ss_advance (s, prefix.length);
5944       return true;
5945     }
5946   else
5947     return false;
5948 }
5949
5950 static void
5951 put_table_expression (struct string *out, struct lexer *lexer,
5952                       struct dictionary *dict, int expr_start, int expr_end)
5953 {
5954   size_t nest = 0;
5955   for (int ofs = expr_start; ofs < expr_end; ofs++)
5956     {
5957       const struct token *t = lex_ofs_token (lexer, ofs);
5958       if (t->type == T_LBRACK)
5959         nest++;
5960       else if (t->type == T_RBRACK && nest > 0)
5961         nest--;
5962       else if (nest > 0)
5963         {
5964           /* Nothing. */
5965         }
5966       else if (t->type == T_ID)
5967         {
5968           const struct variable *var
5969             = dict_lookup_var (dict, t->string.string);
5970           const char *label = var ? var_get_label (var) : NULL;
5971           ds_put_cstr (out, label ? label : t->string.string);
5972         }
5973       else
5974         {
5975           if (ofs != expr_start && t->type != T_RPAREN && ds_last (out) != ' ')
5976             ds_put_byte (out, ' ');
5977
5978           char *repr = lex_ofs_representation (lexer, ofs, ofs);
5979           ds_put_cstr (out, repr);
5980           free (repr);
5981
5982           if (ofs + 1 != expr_end && t->type != T_LPAREN)
5983             ds_put_byte (out, ' ');
5984         }
5985     }
5986 }
5987
5988 static void
5989 put_title_text (struct string *out, struct substring in, time_t now,
5990                 struct lexer *lexer, struct dictionary *dict,
5991                 int expr_start, int expr_end)
5992 {
5993   for (;;)
5994     {
5995       size_t chunk = ss_find_byte (in, ')');
5996       ds_put_substring (out, ss_head (in, chunk));
5997       ss_advance (&in, chunk);
5998       if (ss_is_empty (in))
5999         return;
6000
6001       if (skip_prefix (&in, ss_cstr (")DATE")))
6002         put_strftime (out, now, "%x");
6003       else if (skip_prefix (&in, ss_cstr (")TIME")))
6004         put_strftime (out, now, "%X");
6005       else if (skip_prefix (&in, ss_cstr (")TABLE")))
6006         put_table_expression (out, lexer, dict, expr_start, expr_end);
6007       else
6008         {
6009           ds_put_byte (out, ')');
6010           ss_advance (&in, 1);
6011         }
6012     }
6013 }
6014
6015 int
6016 cmd_ctables (struct lexer *lexer, struct dataset *ds)
6017 {
6018   struct casereader *input = NULL;
6019
6020   struct measure_guesser *mg = measure_guesser_create (ds);
6021   if (mg)
6022     {
6023       input = proc_open (ds);
6024       measure_guesser_run (mg, input);
6025       measure_guesser_destroy (mg);
6026     }
6027
6028   size_t n_vars = dict_get_n_vars (dataset_dict (ds));
6029   enum ctables_vlabel *vlabels = xnmalloc (n_vars, sizeof *vlabels);
6030   enum settings_value_show tvars = settings_get_show_variables ();
6031   for (size_t i = 0; i < n_vars; i++)
6032     vlabels[i] = (enum ctables_vlabel) tvars;
6033
6034   struct pivot_table_look *look = pivot_table_look_unshare (
6035     pivot_table_look_ref (pivot_table_look_get_default ()));
6036   look->omit_empty = false;
6037
6038   struct ctables *ct = xmalloc (sizeof *ct);
6039   *ct = (struct ctables) {
6040     .dict = dataset_dict (ds),
6041     .look = look,
6042     .ctables_formats = FMT_SETTINGS_INIT,
6043     .vlabels = vlabels,
6044     .postcomputes = HMAP_INITIALIZER (ct->postcomputes),
6045   };
6046
6047   time_t now = time (NULL);
6048
6049   struct ctf
6050     {
6051       enum fmt_type type;
6052       const char *dot_string;
6053       const char *comma_string;
6054     };
6055   static const struct ctf ctfs[4] = {
6056     { CTEF_NEGPAREN, "(,,,)",   "(...)" },
6057     { CTEF_NEQUAL,   "-,N=,,",  "-.N=.." },
6058     { CTEF_PAREN,    "-,(,),",  "-.(.)." },
6059     { CTEF_PCTPAREN, "-,(,%),", "-.(.%)." },
6060   };
6061   bool is_dot = settings_get_fmt_settings ()->decimal == '.';
6062   for (size_t i = 0; i < 4; i++)
6063     {
6064       const char *s = is_dot ? ctfs[i].dot_string : ctfs[i].comma_string;
6065       fmt_settings_set_cc (&ct->ctables_formats, ctfs[i].type,
6066                            fmt_number_style_from_string (s));
6067     }
6068
6069   if (!lex_force_match (lexer, T_SLASH))
6070     goto error;
6071
6072   while (!lex_match_id (lexer, "TABLE"))
6073     {
6074       if (lex_match_id (lexer, "FORMAT"))
6075         {
6076           double widths[2] = { SYSMIS, SYSMIS };
6077           double units_per_inch = 72.0;
6078
6079           while (lex_token (lexer) != T_SLASH)
6080             {
6081               if (lex_match_id (lexer, "MINCOLWIDTH"))
6082                 {
6083                   if (!parse_col_width (lexer, "MINCOLWIDTH", &widths[0]))
6084                     goto error;
6085                 }
6086               else if (lex_match_id (lexer, "MAXCOLWIDTH"))
6087                 {
6088                   if (!parse_col_width (lexer, "MAXCOLWIDTH", &widths[1]))
6089                     goto error;
6090                 }
6091               else if (lex_match_id (lexer, "UNITS"))
6092                 {
6093                   lex_match (lexer, T_EQUALS);
6094                   if (lex_match_id (lexer, "POINTS"))
6095                     units_per_inch = 72.0;
6096                   else if (lex_match_id (lexer, "INCHES"))
6097                     units_per_inch = 1.0;
6098                   else if (lex_match_id (lexer, "CM"))
6099                     units_per_inch = 2.54;
6100                   else
6101                     {
6102                       lex_error_expecting (lexer, "POINTS", "INCHES", "CM");
6103                       goto error;
6104                     }
6105                 }
6106               else if (lex_match_id (lexer, "EMPTY"))
6107                 {
6108                   free (ct->zero);
6109                   ct->zero = NULL;
6110
6111                   lex_match (lexer, T_EQUALS);
6112                   if (lex_match_id (lexer, "ZERO"))
6113                     {
6114                       /* Nothing to do. */
6115                     }
6116                   else if (lex_match_id (lexer, "BLANK"))
6117                     ct->zero = xstrdup ("");
6118                   else if (lex_force_string (lexer))
6119                     {
6120                       ct->zero = ss_xstrdup (lex_tokss (lexer));
6121                       lex_get (lexer);
6122                     }
6123                   else
6124                     goto error;
6125                 }
6126               else if (lex_match_id (lexer, "MISSING"))
6127                 {
6128                   lex_match (lexer, T_EQUALS);
6129                   if (!lex_force_string (lexer))
6130                     goto error;
6131
6132                   free (ct->missing);
6133                   ct->missing = (strcmp (lex_tokcstr (lexer), ".")
6134                                  ? ss_xstrdup (lex_tokss (lexer))
6135                                  : NULL);
6136                   lex_get (lexer);
6137                 }
6138               else
6139                 {
6140                   lex_error_expecting (lexer, "MINCOLWIDTH", "MAXCOLWIDTH",
6141                                        "UNITS", "EMPTY", "MISSING");
6142                   goto error;
6143                 }
6144             }
6145
6146           if (widths[0] != SYSMIS && widths[1] != SYSMIS
6147               && widths[0] > widths[1])
6148             {
6149               msg (SE, _("MINCOLWIDTH must not be greater than MAXCOLWIDTH."));
6150               goto error;
6151             }
6152
6153           for (size_t i = 0; i < 2; i++)
6154             if (widths[i] != SYSMIS)
6155               {
6156                 int *wr = ct->look->width_ranges[TABLE_HORZ];
6157                 wr[i] = widths[i] / units_per_inch * 96.0;
6158                 if (wr[0] > wr[1])
6159                   wr[!i] = wr[i];
6160               }
6161         }
6162       else if (lex_match_id (lexer, "VLABELS"))
6163         {
6164           if (!lex_force_match_id (lexer, "VARIABLES"))
6165             goto error;
6166           lex_match (lexer, T_EQUALS);
6167
6168           struct variable **vars;
6169           size_t n_vars;
6170           if (!parse_variables (lexer, dataset_dict (ds), &vars, &n_vars,
6171                                 PV_NO_SCRATCH))
6172             goto error;
6173
6174           if (!lex_force_match_id (lexer, "DISPLAY"))
6175             {
6176               free (vars);
6177               goto error;
6178             }
6179           lex_match (lexer, T_EQUALS);
6180
6181           enum ctables_vlabel vlabel;
6182           if (lex_match_id (lexer, "DEFAULT"))
6183             vlabel = (enum ctables_vlabel) settings_get_show_variables ();
6184           else if (lex_match_id (lexer, "NAME"))
6185             vlabel = CTVL_NAME;
6186           else if (lex_match_id (lexer, "LABEL"))
6187             vlabel = CTVL_LABEL;
6188           else if (lex_match_id (lexer, "BOTH"))
6189             vlabel = CTVL_BOTH;
6190           else if (lex_match_id (lexer, "NONE"))
6191             vlabel = CTVL_NONE;
6192           else
6193             {
6194               lex_error_expecting (lexer, "DEFAULT", "NAME", "LABEL",
6195                                    "BOTH", "NONE");
6196               free (vars);
6197               goto error;
6198             }
6199
6200           for (size_t i = 0; i < n_vars; i++)
6201             ct->vlabels[var_get_dict_index (vars[i])] = vlabel;
6202           free (vars);
6203         }
6204       else if (lex_match_id (lexer, "MRSETS"))
6205         {
6206           if (!lex_force_match_id (lexer, "COUNTDUPLICATES"))
6207             goto error;
6208           lex_match (lexer, T_EQUALS);
6209           if (!parse_bool (lexer, &ct->mrsets_count_duplicates))
6210             goto error;
6211         }
6212       else if (lex_match_id (lexer, "SMISSING"))
6213         {
6214           if (lex_match_id (lexer, "VARIABLE"))
6215             ct->smissing_listwise = false;
6216           else if (lex_match_id (lexer, "LISTWISE"))
6217             ct->smissing_listwise = true;
6218           else
6219             {
6220               lex_error_expecting (lexer, "VARIABLE", "LISTWISE");
6221               goto error;
6222             }
6223         }
6224       else if (lex_match_id (lexer, "PCOMPUTE"))
6225         {
6226           if (!ctables_parse_pcompute (lexer, dataset_dict (ds), ct))
6227             goto error;
6228         }
6229       else if (lex_match_id (lexer, "PPROPERTIES"))
6230         {
6231           if (!ctables_parse_pproperties (lexer, ct))
6232             goto error;
6233         }
6234       else if (lex_match_id (lexer, "WEIGHT"))
6235         {
6236           if (!lex_force_match_id (lexer, "VARIABLE"))
6237             goto error;
6238           lex_match (lexer, T_EQUALS);
6239           ct->e_weight = parse_variable (lexer, dataset_dict (ds));
6240           if (!ct->e_weight)
6241             goto error;
6242         }
6243       else if (lex_match_id (lexer, "HIDESMALLCOUNTS"))
6244         {
6245           if (lex_match_id (lexer, "COUNT"))
6246             {
6247               lex_match (lexer, T_EQUALS);
6248               if (!lex_force_int_range (lexer, "HIDESMALLCOUNTS COUNT",
6249                                         2, INT_MAX))
6250                 goto error;
6251               ct->hide_threshold = lex_integer (lexer);
6252               lex_get (lexer);
6253             }
6254           else if (ct->hide_threshold == 0)
6255             ct->hide_threshold = 5;
6256         }
6257       else
6258         {
6259           lex_error_expecting (lexer, "FORMAT", "VLABELS", "MRSETS",
6260                                "SMISSING", "PCOMPUTE", "PPROPERTIES",
6261                                "WEIGHT", "HIDESMALLCOUNTS", "TABLE");
6262           goto error;
6263         }
6264
6265       if (!lex_force_match (lexer, T_SLASH))
6266         goto error;
6267     }
6268
6269   size_t allocated_tables = 0;
6270   do
6271     {
6272       if (ct->n_tables >= allocated_tables)
6273         ct->tables = x2nrealloc (ct->tables, &allocated_tables,
6274                                  sizeof *ct->tables);
6275
6276       struct ctables_category *cat = xmalloc (sizeof *cat);
6277       *cat = (struct ctables_category) {
6278         .type = CCT_VALUE,
6279         .include_missing = false,
6280         .sort_ascending = true,
6281       };
6282
6283       struct ctables_categories *c = xmalloc (sizeof *c);
6284       size_t n_vars = dict_get_n_vars (dataset_dict (ds));
6285       *c = (struct ctables_categories) {
6286         .n_refs = n_vars,
6287         .cats = cat,
6288         .n_cats = 1,
6289         .show_empty = true,
6290       };
6291
6292       struct ctables_categories **categories = xnmalloc (n_vars,
6293                                                          sizeof *categories);
6294       for (size_t i = 0; i < n_vars; i++)
6295         categories[i] = c;
6296
6297       struct ctables_table *t = xmalloc (sizeof *t);
6298       *t = (struct ctables_table) {
6299         .ctables = ct,
6300         .slabels_axis = PIVOT_AXIS_COLUMN,
6301         .slabels_visible = true,
6302         .clabels_values_map = HMAP_INITIALIZER (t->clabels_values_map),
6303         .label_axis = {
6304           [PIVOT_AXIS_ROW] = PIVOT_AXIS_ROW,
6305           [PIVOT_AXIS_COLUMN] = PIVOT_AXIS_COLUMN,
6306           [PIVOT_AXIS_LAYER] = PIVOT_AXIS_LAYER,
6307         },
6308         .clabels_from_axis = PIVOT_AXIS_LAYER,
6309         .categories = categories,
6310         .n_categories = n_vars,
6311         .cilevel = 95,
6312       };
6313       ct->tables[ct->n_tables++] = t;
6314
6315       lex_match (lexer, T_EQUALS);
6316       int expr_start = lex_ofs (lexer);
6317       if (!ctables_axis_parse (lexer, dataset_dict (ds), ct, t, PIVOT_AXIS_ROW))
6318         goto error;
6319       if (lex_match (lexer, T_BY))
6320         {
6321           if (!ctables_axis_parse (lexer, dataset_dict (ds),
6322                                    ct, t, PIVOT_AXIS_COLUMN))
6323             goto error;
6324
6325           if (lex_match (lexer, T_BY))
6326             {
6327               if (!ctables_axis_parse (lexer, dataset_dict (ds),
6328                                        ct, t, PIVOT_AXIS_LAYER))
6329                 goto error;
6330             }
6331         }
6332       int expr_end = lex_ofs (lexer);
6333
6334       if (!t->axes[PIVOT_AXIS_ROW] && !t->axes[PIVOT_AXIS_COLUMN]
6335           && !t->axes[PIVOT_AXIS_LAYER])
6336         {
6337           lex_error (lexer, _("At least one variable must be specified."));
6338           goto error;
6339         }
6340
6341       const struct ctables_axis *scales[PIVOT_N_AXES];
6342       size_t n_scales = 0;
6343       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
6344         {
6345           scales[a] = find_scale (t->axes[a]);
6346           if (scales[a])
6347             n_scales++;
6348         }
6349       if (n_scales > 1)
6350         {
6351           msg (SE, _("Scale variables may appear only on one axis."));
6352           if (scales[PIVOT_AXIS_ROW])
6353             msg_at (SN, scales[PIVOT_AXIS_ROW]->loc,
6354                     _("This scale variable appears on the rows axis."));
6355           if (scales[PIVOT_AXIS_COLUMN])
6356             msg_at (SN, scales[PIVOT_AXIS_COLUMN]->loc,
6357                     _("This scale variable appears on the columns axis."));
6358           if (scales[PIVOT_AXIS_LAYER])
6359             msg_at (SN, scales[PIVOT_AXIS_LAYER]->loc,
6360                     _("This scale variable appears on the layer axis."));
6361           goto error;
6362         }
6363
6364       const struct ctables_axis *summaries[PIVOT_N_AXES];
6365       size_t n_summaries = 0;
6366       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
6367         {
6368           summaries[a] = (scales[a]
6369                           ? scales[a]
6370                           : find_categorical_summary_spec (t->axes[a]));
6371           if (summaries[a])
6372             n_summaries++;
6373         }
6374       if (n_summaries > 1)
6375         {
6376           msg (SE, _("Summaries may appear only on one axis."));
6377           for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
6378             if (summaries[a])
6379               {
6380                 msg_at (SN, summaries[a]->loc,
6381                         a == PIVOT_AXIS_ROW
6382                         ? _("This variable on the rows axis has a summary.")
6383                         : a == PIVOT_AXIS_COLUMN
6384                         ? _("This variable on the columns axis has a summary.")
6385                         : _("This variable on the layers axis has a summary."));
6386                 if (scales[a])
6387                   msg_at (SN, summaries[a]->loc,
6388                           _("This is a scale variable, so it always has a "
6389                             "summary even if the syntax does not explicitly "
6390                             "specify one."));
6391               }
6392           goto error;
6393         }
6394       for (enum pivot_axis_type a = 0; a < PIVOT_N_AXES; a++)
6395         if (n_summaries ? summaries[a] : t->axes[a])
6396           {
6397             t->summary_axis = a;
6398             break;
6399           }
6400
6401       if (lex_token (lexer) == T_ENDCMD)
6402         {
6403           if (!ctables_prepare_table (t))
6404             goto error;
6405           break;
6406         }
6407       if (!lex_force_match (lexer, T_SLASH))
6408         goto error;
6409
6410       while (!lex_match_id (lexer, "TABLE") && lex_token (lexer) != T_ENDCMD)
6411         {
6412           if (lex_match_id (lexer, "SLABELS"))
6413             {
6414               while (lex_token (lexer) != T_SLASH && lex_token (lexer) != T_ENDCMD)
6415                 {
6416                   if (lex_match_id (lexer, "POSITION"))
6417                     {
6418                       lex_match (lexer, T_EQUALS);
6419                       if (lex_match_id (lexer, "COLUMN"))
6420                         t->slabels_axis = PIVOT_AXIS_COLUMN;
6421                       else if (lex_match_id (lexer, "ROW"))
6422                         t->slabels_axis = PIVOT_AXIS_ROW;
6423                       else if (lex_match_id (lexer, "LAYER"))
6424                         t->slabels_axis = PIVOT_AXIS_LAYER;
6425                       else
6426                         {
6427                           lex_error_expecting (lexer, "COLUMN", "ROW", "LAYER");
6428                           goto error;
6429                         }
6430                     }
6431                   else if (lex_match_id (lexer, "VISIBLE"))
6432                     {
6433                       lex_match (lexer, T_EQUALS);
6434                       if (!parse_bool (lexer, &t->slabels_visible))
6435                         goto error;
6436                     }
6437                   else
6438                     {
6439                       lex_error_expecting (lexer, "POSITION", "VISIBLE");
6440                       goto error;
6441                     }
6442                 }
6443             }
6444           else if (lex_match_id (lexer, "CLABELS"))
6445             {
6446               if (lex_match_id (lexer, "AUTO"))
6447                 {
6448                   t->label_axis[PIVOT_AXIS_ROW] = PIVOT_AXIS_ROW;
6449                   t->label_axis[PIVOT_AXIS_COLUMN] = PIVOT_AXIS_COLUMN;
6450                 }
6451               else if (lex_match_id (lexer, "ROWLABELS"))
6452                 {
6453                   lex_match (lexer, T_EQUALS);
6454                   if (lex_match_id (lexer, "OPPOSITE"))
6455                     t->label_axis[PIVOT_AXIS_ROW] = PIVOT_AXIS_COLUMN;
6456                   else if (lex_match_id (lexer, "LAYER"))
6457                     t->label_axis[PIVOT_AXIS_ROW] = PIVOT_AXIS_LAYER;
6458                   else
6459                     {
6460                       lex_error_expecting (lexer, "OPPOSITE", "LAYER");
6461                       goto error;
6462                     }
6463                 }
6464               else if (lex_match_id (lexer, "COLLABELS"))
6465                 {
6466                   lex_match (lexer, T_EQUALS);
6467                   if (lex_match_id (lexer, "OPPOSITE"))
6468                     t->label_axis[PIVOT_AXIS_COLUMN] = PIVOT_AXIS_ROW;
6469                   else if (lex_match_id (lexer, "LAYER"))
6470                     t->label_axis[PIVOT_AXIS_COLUMN] = PIVOT_AXIS_LAYER;
6471                   else
6472                     {
6473                       lex_error_expecting (lexer, "OPPOSITE", "LAYER");
6474                       goto error;
6475                     }
6476                 }
6477               else
6478                 {
6479                   lex_error_expecting (lexer, "AUTO", "ROWLABELS",
6480                                        "COLLABELS");
6481                   goto error;
6482                 }
6483             }
6484           else if (lex_match_id (lexer, "CRITERIA"))
6485             {
6486               if (!lex_force_match_id (lexer, "CILEVEL"))
6487                 goto error;
6488               lex_match (lexer, T_EQUALS);
6489
6490               if (!lex_force_num_range_halfopen (lexer, "CILEVEL", 0, 100))
6491                 goto error;
6492               t->cilevel = lex_number (lexer);
6493               lex_get (lexer);
6494             }
6495           else if (lex_match_id (lexer, "CATEGORIES"))
6496             {
6497               if (!ctables_table_parse_categories (lexer, dataset_dict (ds),
6498                                                    ct, t))
6499                 goto error;
6500             }
6501           else if (lex_match_id (lexer, "TITLES"))
6502             {
6503               do
6504                 {
6505                   char **textp;
6506                   if (lex_match_id (lexer, "CAPTION"))
6507                     textp = &t->caption;
6508                   else if (lex_match_id (lexer, "CORNER"))
6509                     textp = &t->corner;
6510                   else if (lex_match_id (lexer, "TITLE"))
6511                     textp = &t->title;
6512                   else
6513                     {
6514                       lex_error_expecting (lexer, "CAPTION", "CORNER", "TITLE");
6515                       goto error;
6516                     }
6517                   lex_match (lexer, T_EQUALS);
6518
6519                   struct string s = DS_EMPTY_INITIALIZER;
6520                   while (lex_is_string (lexer))
6521                     {
6522                       if (!ds_is_empty (&s))
6523                         ds_put_byte (&s, ' ');
6524                       put_title_text (&s, lex_tokss (lexer), now,
6525                                       lexer, dataset_dict (ds),
6526                                       expr_start, expr_end);
6527                       lex_get (lexer);
6528                     }
6529                   free (*textp);
6530                   *textp = ds_steal_cstr (&s);
6531                 }
6532               while (lex_token (lexer) != T_SLASH
6533                      && lex_token (lexer) != T_ENDCMD);
6534             }
6535           else if (lex_match_id (lexer, "SIGTEST"))
6536             {
6537               if (!t->chisq)
6538                 {
6539                   t->chisq = xmalloc (sizeof *t->chisq);
6540                   *t->chisq = (struct ctables_chisq) {
6541                     .alpha = .05,
6542                     .include_mrsets = true,
6543                     .all_visible = true,
6544                   };
6545                 }
6546
6547               do
6548                 {
6549                   if (lex_match_id (lexer, "TYPE"))
6550                     {
6551                       lex_match (lexer, T_EQUALS);
6552                       if (!lex_force_match_id (lexer, "CHISQUARE"))
6553                         goto error;
6554                     }
6555                   else if (lex_match_id (lexer, "ALPHA"))
6556                     {
6557                       lex_match (lexer, T_EQUALS);
6558                       if (!lex_force_num_range_halfopen (lexer, "ALPHA", 0, 1))
6559                         goto error;
6560                       t->chisq->alpha = lex_number (lexer);
6561                       lex_get (lexer);
6562                     }
6563                   else if (lex_match_id (lexer, "INCLUDEMRSETS"))
6564                     {
6565                       lex_match (lexer, T_EQUALS);
6566                       if (!parse_bool (lexer, &t->chisq->include_mrsets))
6567                         goto error;
6568                     }
6569                   else if (lex_match_id (lexer, "CATEGORIES"))
6570                     {
6571                       lex_match (lexer, T_EQUALS);
6572                       if (lex_match_id (lexer, "ALLVISIBLE"))
6573                         t->chisq->all_visible = true;
6574                       else if (lex_match_id (lexer, "SUBTOTALS"))
6575                         t->chisq->all_visible = false;
6576                       else
6577                         {
6578                           lex_error_expecting (lexer,
6579                                                "ALLVISIBLE", "SUBTOTALS");
6580                           goto error;
6581                         }
6582                     }
6583                   else
6584                     {
6585                       lex_error_expecting (lexer, "TYPE", "ALPHA",
6586                                            "INCLUDEMRSETS", "CATEGORIES");
6587                       goto error;
6588                     }
6589                 }
6590               while (lex_token (lexer) != T_SLASH
6591                      && lex_token (lexer) != T_ENDCMD);
6592             }
6593           else if (lex_match_id (lexer, "COMPARETEST"))
6594             {
6595               if (!t->pairwise)
6596                 {
6597                   t->pairwise = xmalloc (sizeof *t->pairwise);
6598                   *t->pairwise = (struct ctables_pairwise) {
6599                     .type = PROP,
6600                     .alpha = { .05, .05 },
6601                     .adjust = BONFERRONI,
6602                     .include_mrsets = true,
6603                     .meansvariance_allcats = true,
6604                     .all_visible = true,
6605                     .merge = false,
6606                     .apa_style = true,
6607                     .show_sig = false,
6608                   };
6609                 }
6610
6611               do
6612                 {
6613                   if (lex_match_id (lexer, "TYPE"))
6614                     {
6615                       lex_match (lexer, T_EQUALS);
6616                       if (lex_match_id (lexer, "PROP"))
6617                         t->pairwise->type = PROP;
6618                       else if (lex_match_id (lexer, "MEAN"))
6619                         t->pairwise->type = MEAN;
6620                       else
6621                         {
6622                           lex_error_expecting (lexer, "PROP", "MEAN");
6623                           goto error;
6624                         }
6625                     }
6626                   else if (lex_match_id (lexer, "ALPHA"))
6627                     {
6628                       lex_match (lexer, T_EQUALS);
6629
6630                       if (!lex_force_num_range_open (lexer, "ALPHA", 0, 1))
6631                         goto error;
6632                       double a0 = lex_number (lexer);
6633                       lex_get (lexer);
6634
6635                       lex_match (lexer, T_COMMA);
6636                       if (lex_is_number (lexer))
6637                         {
6638                           if (!lex_force_num_range_open (lexer, "ALPHA", 0, 1))
6639                             goto error;
6640                           double a1 = lex_number (lexer);
6641                           lex_get (lexer);
6642
6643                           t->pairwise->alpha[0] = MIN (a0, a1);
6644                           t->pairwise->alpha[1] = MAX (a0, a1);
6645                         }
6646                       else
6647                         t->pairwise->alpha[0] = t->pairwise->alpha[1] = a0;
6648                     }
6649                   else if (lex_match_id (lexer, "ADJUST"))
6650                     {
6651                       lex_match (lexer, T_EQUALS);
6652                       if (lex_match_id (lexer, "BONFERRONI"))
6653                         t->pairwise->adjust = BONFERRONI;
6654                       else if (lex_match_id (lexer, "BH"))
6655                         t->pairwise->adjust = BH;
6656                       else if (lex_match_id (lexer, "NONE"))
6657                         t->pairwise->adjust = 0;
6658                       else
6659                         {
6660                           lex_error_expecting (lexer, "BONFERRONI", "BH",
6661                                                "NONE");
6662                           goto error;
6663                         }
6664                     }
6665                   else if (lex_match_id (lexer, "INCLUDEMRSETS"))
6666                     {
6667                       lex_match (lexer, T_EQUALS);
6668                       if (!parse_bool (lexer, &t->pairwise->include_mrsets))
6669                         goto error;
6670                     }
6671                   else if (lex_match_id (lexer, "MEANSVARIANCE"))
6672                     {
6673                       lex_match (lexer, T_EQUALS);
6674                       if (lex_match_id (lexer, "ALLCATS"))
6675                         t->pairwise->meansvariance_allcats = true;
6676                       else if (lex_match_id (lexer, "TESTEDCATS"))
6677                         t->pairwise->meansvariance_allcats = false;
6678                       else
6679                         {
6680                           lex_error_expecting (lexer, "ALLCATS", "TESTEDCATS");
6681                           goto error;
6682                         }
6683                     }
6684                   else if (lex_match_id (lexer, "CATEGORIES"))
6685                     {
6686                       lex_match (lexer, T_EQUALS);
6687                       if (lex_match_id (lexer, "ALLVISIBLE"))
6688                         t->pairwise->all_visible = true;
6689                       else if (lex_match_id (lexer, "SUBTOTALS"))
6690                         t->pairwise->all_visible = false;
6691                       else
6692                         {
6693                           lex_error_expecting (lexer, "ALLVISIBLE",
6694                                                "SUBTOTALS");
6695                           goto error;
6696                         }
6697                     }
6698                   else if (lex_match_id (lexer, "MERGE"))
6699                     {
6700                       lex_match (lexer, T_EQUALS);
6701                       if (!parse_bool (lexer, &t->pairwise->merge))
6702                         goto error;
6703                     }
6704                   else if (lex_match_id (lexer, "STYLE"))
6705                     {
6706                       lex_match (lexer, T_EQUALS);
6707                       if (lex_match_id (lexer, "APA"))
6708                         t->pairwise->apa_style = true;
6709                       else if (lex_match_id (lexer, "SIMPLE"))
6710                         t->pairwise->apa_style = false;
6711                       else
6712                         {
6713                           lex_error_expecting (lexer, "APA", "SIMPLE");
6714                           goto error;
6715                         }
6716                     }
6717                   else if (lex_match_id (lexer, "SHOWSIG"))
6718                     {
6719                       lex_match (lexer, T_EQUALS);
6720                       if (!parse_bool (lexer, &t->pairwise->show_sig))
6721                         goto error;
6722                     }
6723                   else
6724                     {
6725                       lex_error_expecting (lexer, "TYPE", "ALPHA", "ADJUST",
6726                                            "INCLUDEMRSETS", "MEANSVARIANCE",
6727                                            "CATEGORIES", "MERGE", "STYLE",
6728                                            "SHOWSIG");
6729                       goto error;
6730                     }
6731                 }
6732               while (lex_token (lexer) != T_SLASH
6733                      && lex_token (lexer) != T_ENDCMD);
6734             }
6735           else
6736             {
6737               lex_error_expecting (lexer, "TABLE", "SLABELS", "CLABELS",
6738                                    "CRITERIA", "CATEGORIES", "TITLES",
6739                                    "SIGTEST", "COMPARETEST");
6740               goto error;
6741             }
6742
6743           if (!lex_match (lexer, T_SLASH))
6744             break;
6745         }
6746
6747       if (t->label_axis[PIVOT_AXIS_ROW] != PIVOT_AXIS_ROW
6748           && t->label_axis[PIVOT_AXIS_COLUMN] != PIVOT_AXIS_COLUMN)
6749         {
6750           msg (SE, _("ROWLABELS and COLLABELS may not both be specified."));
6751           goto error;
6752         }
6753
6754       if (!ctables_prepare_table (t))
6755         goto error;
6756     }
6757   while (lex_token (lexer) != T_ENDCMD);
6758
6759   if (!input)
6760     input = proc_open (ds);
6761   bool ok = ctables_execute (ds, input, ct);
6762   ok = proc_commit (ds) && ok;
6763
6764   ctables_destroy (ct);
6765   return ok ? CMD_SUCCESS : CMD_FAILURE;
6766
6767 error:
6768   if (input)
6769     proc_commit (ds);
6770   ctables_destroy (ct);
6771   return CMD_FAILURE;
6772 }
6773