1 /* PSPP - a program for statistical analysis.
2 Copyright (C) 2018 Free Software Foundation, Inc.
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.
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.
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/>. */
21 #include "output/pivot-table.h"
23 #include "data/settings.h"
24 #include "libpspp/assertion.h"
25 #include "libpspp/pool.h"
26 #include "output/tab.h"
27 #include "output/table.h"
28 #include "output/table-item.h"
29 #include "output/text-item.h"
30 #include "output/table-provider.h"
32 #include "gl/minmax.h"
33 #include "gl/xalloc.h"
38 static const struct pivot_category *
39 find_category (const struct pivot_dimension *d, int dim_index,
40 const size_t *indexes, int row_ofs)
42 size_t index = indexes[dim_index];
43 assert (index < d->n_leaves);
44 for (const struct pivot_category *c = d->presentation_leaves[index];
50 row_ofs -= 1 + c->extra_depth;
57 static struct area_style *
58 area_style_override (struct pool *pool,
59 const struct area_style *in,
60 const struct cell_style *cell_,
61 const struct font_style *font_)
63 const struct cell_style *cell = cell_ ? cell_ : &in->cell_style;
64 const struct font_style *font = font_ ? font_ : &in->font_style;
66 struct area_style *out = (pool
67 ? pool_alloc (pool, sizeof *out)
68 : xmalloc (sizeof *out));
69 *out = (struct area_style) {
70 .cell_style.halign = cell->halign,
71 .cell_style.valign = cell->valign,
72 .cell_style.decimal_offset = cell->decimal_offset,
73 .cell_style.margin[H][0] = cell->margin[H][0],
74 .cell_style.margin[H][1] = cell->margin[H][1],
75 .cell_style.margin[V][0] = cell->margin[V][0],
76 .cell_style.margin[V][1] = cell->margin[V][1],
77 .font_style.fg[0] = font->fg[0],
78 .font_style.fg[1] = font->fg[1],
79 .font_style.bg[0] = font->bg[0],
80 .font_style.bg[1] = font->bg[1],
81 .font_style.typeface = (font->typeface
82 ? pool_strdup (pool, font->typeface)
84 .font_style.size = font->size,
85 .font_style.bold = font->bold,
86 .font_style.italic = font->italic,
87 .font_style.underline = font->underline,
88 .font_style.markup = font->markup,
94 fill_cell (struct tab_table *t, int x1, int y1, int x2, int y2,
95 const struct area_style *style, int style_idx,
96 const struct pivot_value *value, struct footnote **footnotes,
97 enum settings_value_show show_values,
98 enum settings_value_show show_variables,
102 struct string s = DS_EMPTY_INITIALIZER;
103 int opts = style_idx << TAB_STYLE_SHIFT;
106 bool numeric = pivot_value_format_body (value, show_values,
110 if (value->font_style && value->font_style->markup)
115 tab_joint_text (t, x1, y1, x2, y2, opts, ds_cstr (&s));
120 if (value->cell_style || value->font_style)
121 tab_add_style (t, x1, y1,
122 area_style_override (t->container, style,
126 for (size_t i = 0; i < value->n_footnotes; i++)
127 tab_add_footnote (t, x1, y1, footnotes[value->footnotes[i]->idx]);
131 static struct table_item_text *
132 pivot_value_to_table_item_text (const struct pivot_value *value,
133 const struct area_style *area,
134 struct footnote **footnotes,
135 enum settings_value_show show_values,
136 enum settings_value_show show_variables)
141 struct string s = DS_EMPTY_INITIALIZER;
142 pivot_value_format_body (value, show_values, show_variables, &s);
144 struct table_item_text *text = xmalloc (sizeof *text);
145 *text = (struct table_item_text) {
146 .content = ds_steal_cstr (&s),
147 .footnotes = xnmalloc (value->n_footnotes, sizeof *text->footnotes),
148 .n_footnotes = value->n_footnotes,
149 .style = area_style_override (
150 NULL, area, value->cell_style, value->font_style),
153 for (size_t i = 0; i < value->n_footnotes; i++)
154 text->footnotes[i] = footnotes[value->footnotes[i]->idx];
160 get_table_rule (const struct table_border_style *styles,
161 enum pivot_border style_idx)
163 return styles[style_idx].stroke | (style_idx << TAB_RULE_STYLE_SHIFT);
167 draw_line (struct tab_table *t, const struct table_border_style *styles,
168 enum pivot_border style_idx,
169 enum table_axis axis, int a, int b0, int b1)
171 int rule = get_table_rule (styles, style_idx);
173 tab_hline (t, rule, b0, b1, a);
175 tab_vline (t, rule, a, b0, b1);
179 compose_headings (struct tab_table *t,
180 const struct pivot_axis *a_axis, enum table_axis a,
181 const struct pivot_axis *b_axis,
182 const struct table_border_style *borders,
183 enum pivot_border dim_col_horz,
184 enum pivot_border dim_col_vert,
185 enum pivot_border cat_col_horz,
186 enum pivot_border cat_col_vert,
187 const size_t *column_enumeration, size_t n_columns,
188 const struct area_style *label_style, int label_style_idx,
189 const struct area_style *corner_style,
190 struct footnote **footnotes,
191 enum settings_value_show show_values,
192 enum settings_value_show show_variables,
193 bool rotate_inner_labels, bool rotate_outer_labels)
195 enum table_axis b = !a;
196 int b_size = a_axis->label_depth;
197 int a_ofs = b_axis->label_depth;
199 if (!a_axis->n_dimensions || !n_columns || !b_size)
202 int bottom_row = b_size - 1;
203 const int stride = MAX (1, a_axis->n_dimensions);
204 for (int dim_index = 0; dim_index < a_axis->n_dimensions; dim_index++)
206 const struct pivot_dimension *d = a_axis->dimensions[dim_index];
207 if (d->hide_all_labels)
210 for (int row_ofs = 0; row_ofs < d->label_depth; row_ofs++)
212 for (size_t x1 = 0; x1 < n_columns; )
214 const struct pivot_category *c = find_category (
215 d, dim_index, column_enumeration + x1 * stride, row_ofs);
223 for (x2 = x1 + 1; x2 < n_columns; x2++)
225 const struct pivot_category *c2 = find_category (
226 d, dim_index, column_enumeration + x2 * stride, row_ofs);
231 int y1 = bottom_row - row_ofs - c->extra_depth;
232 int y2 = bottom_row - row_ofs + 1;
233 bool is_outer_row = y1 == 0;
234 bool is_inner_row = y2 == b_size;
235 if (pivot_category_is_leaf (c) || c->show_label)
237 int bb[TABLE_N_AXES][2];
238 bb[a][0] = x1 + a_ofs;
239 bb[a][1] = x2 + a_ofs - 1;
242 bool rotate = ((rotate_inner_labels && is_inner_row)
243 || (rotate_outer_labels && is_outer_row));
244 fill_cell (t, bb[H][0], bb[V][0], bb[H][1], bb[V][1],
245 label_style, label_style_idx, c->name, footnotes,
246 show_values, show_variables, rotate);
248 if (pivot_category_is_leaf (c) && x2 + 1 <= n_columns)
250 enum pivot_border style
251 = (y1 == 0 && a_axis->label_depth > d->label_depth
254 draw_line (t, borders, style, b, x2 + a_ofs, y1,
257 if (pivot_category_is_leaf (c) && x1 > 0)
259 enum pivot_border style
260 = (y1 == 0 && a_axis->label_depth > d->label_depth
263 draw_line (t, borders, style, b, x1 + a_ofs, y1,
267 if (c->parent && c->parent->show_label)
268 draw_line (t, borders, cat_col_horz, a, y1,
269 x1 + a_ofs, x2 + a_ofs - 1);
275 if (d->root->show_label_in_corner && a_ofs > 0)
277 int bb[TABLE_N_AXES][2];
278 bb[a][0] = a_ofs - 1;
279 bb[a][1] = a_ofs - 1;
280 bb[b][0] = bottom_row - d->label_depth + 1;
281 bb[b][1] = bottom_row;
282 fill_cell (t, bb[H][0], bb[V][0], bb[H][1], bb[V][1],
283 corner_style, PIVOT_AREA_CORNER, d->root->name, footnotes,
284 show_values, show_variables, false);
288 draw_line (t, borders, dim_col_horz, a, bottom_row + 1, a_ofs,
291 bottom_row -= d->label_depth;
296 pivot_table_submit_layer (const struct pivot_table *pt,
297 const size_t *layer_indexes)
299 const size_t *pindexes[PIVOT_N_AXES]
300 = { [PIVOT_AXIS_LAYER] = layer_indexes };
302 struct string layer_label = DS_EMPTY_INITIALIZER;
303 const struct pivot_axis *layer_axis = &pt->axes[PIVOT_AXIS_LAYER];
304 for (size_t i = 0; i < layer_axis->n_dimensions; i++)
306 const struct pivot_dimension *d = layer_axis->dimensions[i];
309 if (!ds_is_empty (&layer_label))
310 ds_put_byte (&layer_label, '\n');
311 pivot_value_format (d->root->name, pt->show_values,
312 pt->show_variables, &layer_label);
313 ds_put_cstr (&layer_label, ": ");
314 pivot_value_format (d->data_leaves[layer_indexes[i]]->name,
315 pt->show_values, pt->show_variables,
320 size_t body[TABLE_N_AXES];
321 size_t *column_enumeration = pivot_table_enumerate_axis (
322 pt, PIVOT_AXIS_COLUMN, layer_indexes, pt->omit_empty, &body[H]);
323 size_t *row_enumeration = pivot_table_enumerate_axis (
324 pt, PIVOT_AXIS_ROW, layer_indexes, pt->omit_empty, &body[V]);
326 int stub[TABLE_N_AXES] = {
327 [H] = pt->axes[PIVOT_AXIS_ROW].label_depth,
328 [V] = pt->axes[PIVOT_AXIS_COLUMN].label_depth,
330 struct tab_table *table = tab_create (body[H] + stub[H],
332 tab_headers (table, stub[H], 0, stub[V], 0);
334 for (size_t i = 0; i < PIVOT_N_AREAS; i++)
335 table->styles[i] = area_style_override (table->container, &pt->areas[i],
338 for (size_t i = 0; i < PIVOT_N_BORDERS; i++)
340 const struct table_border_style *in = &pt->borders[i];
341 table->rule_colors[i] = pool_alloc (table->container,
342 sizeof *table->rule_colors[i]);
343 struct cell_color *out = table->rule_colors[i];
344 out->alpha = in->color.alpha;
345 out->r = in->color.r;
346 out->g = in->color.g;
347 out->b = in->color.b;
350 struct footnote **footnotes = xcalloc (pt->n_footnotes, sizeof *footnotes);
351 for (size_t i = 0; i < pt->n_footnotes; i++)
353 char *content = pivot_value_to_string (
354 pt->footnotes[i]->content, pt->show_values, pt->show_variables);
355 char *marker = pivot_value_to_string (
356 pt->footnotes[i]->marker, pt->show_values, pt->show_variables);
357 footnotes[i] = tab_create_footnote (
358 table, i, content, marker,
359 area_style_override (table->container, &pt->areas[PIVOT_AREA_FOOTER],
360 pt->footnotes[i]->content->cell_style,
361 pt->footnotes[i]->content->font_style));
366 compose_headings (table,
367 &pt->axes[PIVOT_AXIS_COLUMN], H, &pt->axes[PIVOT_AXIS_ROW],
369 PIVOT_BORDER_DIM_COL_HORZ,
370 PIVOT_BORDER_DIM_COL_VERT,
371 PIVOT_BORDER_CAT_COL_HORZ,
372 PIVOT_BORDER_CAT_COL_VERT,
373 column_enumeration, body[H],
374 &pt->areas[PIVOT_AREA_COLUMN_LABELS],
375 PIVOT_AREA_COLUMN_LABELS,
376 &pt->areas[PIVOT_AREA_CORNER], footnotes,
377 pt->show_values, pt->show_variables,
378 pt->rotate_inner_column_labels, false);
380 compose_headings (table,
381 &pt->axes[PIVOT_AXIS_ROW], V, &pt->axes[PIVOT_AXIS_COLUMN],
383 PIVOT_BORDER_DIM_ROW_VERT,
384 PIVOT_BORDER_DIM_ROW_HORZ,
385 PIVOT_BORDER_CAT_ROW_VERT,
386 PIVOT_BORDER_CAT_ROW_HORZ,
387 row_enumeration, body[V],
388 &pt->areas[PIVOT_AREA_ROW_LABELS],
389 PIVOT_AREA_ROW_LABELS,
390 &pt->areas[PIVOT_AREA_CORNER], footnotes,
391 pt->show_values, pt->show_variables,
392 false, pt->rotate_outer_row_labels);
394 size_t *dindexes = xcalloc (pt->n_dimensions, sizeof *dindexes);
396 PIVOT_ENUMERATION_FOR_EACH (pindexes[PIVOT_AXIS_ROW], row_enumeration,
397 &pt->axes[PIVOT_AXIS_ROW])
400 PIVOT_ENUMERATION_FOR_EACH (pindexes[PIVOT_AXIS_COLUMN],
402 &pt->axes[PIVOT_AXIS_COLUMN])
404 pivot_table_convert_indexes_ptod (pt, pindexes, dindexes);
405 const struct pivot_value *value = pivot_table_get (pt, dindexes);
407 x + stub[H], y + stub[V],
408 x + stub[H], y + stub[V],
409 &pt->areas[PIVOT_AREA_DATA], PIVOT_AREA_DATA,
411 pt->show_values, pt->show_variables, false);
420 if (pt->corner_text && stub[H] && stub[V])
421 fill_cell (table, 0, 0, stub[H] - 1, stub[V] - 1,
422 &pt->areas[PIVOT_AREA_CORNER], PIVOT_AREA_CORNER,
423 pt->corner_text, footnotes,
424 pt->show_values, pt->show_variables, false);
426 if (tab_nc (table) && tab_nr (table))
429 table, get_table_rule (pt->borders, PIVOT_BORDER_INNER_TOP),
430 0, tab_nc (table) - 1, 0);
432 table, get_table_rule (pt->borders, PIVOT_BORDER_INNER_BOTTOM),
433 0, tab_nc (table) - 1, tab_nr (table));
435 table, get_table_rule (pt->borders, PIVOT_BORDER_INNER_LEFT),
436 0, 0, tab_nr (table) - 1);
438 table, get_table_rule (pt->borders, PIVOT_BORDER_INNER_RIGHT),
439 tab_nc (table), 0, tab_nr (table) - 1);
443 table, get_table_rule (pt->borders, PIVOT_BORDER_DATA_TOP),
444 0, tab_nc (table) - 1, stub[V]);
447 table, get_table_rule (pt->borders, PIVOT_BORDER_DATA_LEFT),
448 stub[H], 0, tab_nr (table) - 1);
451 free (column_enumeration);
452 free (row_enumeration);
454 struct table_item *ti = table_item_create (&table->table, NULL, NULL);
458 struct table_item_text *title = pivot_value_to_table_item_text (
459 pt->title, &pt->areas[PIVOT_AREA_TITLE], footnotes,
460 pt->show_values, pt->show_variables);
461 table_item_set_title (ti, title);
462 table_item_text_destroy (title);
465 if (!ds_is_empty (&layer_label))
467 struct table_item_text *layers = table_item_text_create (
468 ds_cstr (&layer_label));
469 layers->style = area_style_override (NULL, &pt->areas[PIVOT_AREA_LAYERS],
471 table_item_set_layers (ti, layers);
472 table_item_text_destroy (layers);
474 ds_destroy (&layer_label);
479 struct table_item_text *caption = pivot_value_to_table_item_text (
480 pt->caption, &pt->areas[PIVOT_AREA_CAPTION], footnotes,
481 pt->show_values, pt->show_variables);
482 table_item_set_caption (ti, caption);
483 table_item_text_destroy (caption);
487 table_item_submit (ti);
491 pivot_table_submit (struct pivot_table *pt)
493 pivot_table_assign_label_depth (CONST_CAST (struct pivot_table *, pt));
495 int old_decimal = settings_get_decimal_char (FMT_COMMA);
496 if (pt->decimal == '.' || pt->decimal == ',')
497 settings_set_decimal_char (pt->decimal);
499 if (pt->print_all_layers)
501 size_t *layer_indexes;
503 PIVOT_AXIS_FOR_EACH (layer_indexes, &pt->axes[PIVOT_AXIS_LAYER])
505 if (pt->paginate_layers)
506 text_item_submit (text_item_create (TEXT_ITEM_EJECT_PAGE, ""));
507 pivot_table_submit_layer (pt, layer_indexes);
511 pivot_table_submit_layer (pt, pt->current_layer);
513 settings_set_decimal_char (old_decimal);
515 pivot_table_destroy (pt);