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