03a415146d795b0a95decfd868042641a9cc6ead
[pspp] / src / output / spv / spv-writer.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 2019 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 "output/spv/spv-writer.h"
20
21 #include <cairo.h>
22 #include <inttypes.h>
23 #include <libxml/xmlwriter.h>
24 #include <math.h>
25 #include <stdlib.h>
26 #include <time.h>
27
28 #include "libpspp/array.h"
29 #include "libpspp/assertion.h"
30 #include "libpspp/cast.h"
31 #include "libpspp/float-format.h"
32 #include "libpspp/integer-format.h"
33 #include "libpspp/temp-file.h"
34 #include "libpspp/version.h"
35 #include "libpspp/zip-writer.h"
36 #include "output/cairo-chart.h"
37 #include "output/chart-item.h"
38 #include "output/driver.h"
39 #include "output/group-item.h"
40 #include "output/image-item.h"
41 #include "output/page-eject-item.h"
42 #include "output/page-setup-item.h"
43 #include "output/pivot-table.h"
44 #include "output/table-item.h"
45 #include "output/text-item.h"
46
47 #include "gl/xalloc.h"
48 #include "gl/xvasprintf.h"
49
50 #include "gettext.h"
51 #define _(msgid) gettext (msgid)
52 #define N_(msgid) (msgid)
53
54 struct spv_writer
55   {
56     struct zip_writer *zw;
57
58     FILE *heading;
59     int heading_depth;
60     xmlTextWriter *xml;
61
62     int n_tables;
63
64     int n_headings;
65     struct page_setup *page_setup;
66     bool need_page_break;
67   };
68
69 static void spv_writer_close_heading (struct spv_writer *);
70
71 char * WARN_UNUSED_RESULT
72 spv_writer_open (const char *filename, struct spv_writer **writerp)
73 {
74   *writerp = NULL;
75
76   struct zip_writer *zw = zip_writer_create (filename);
77   if (!zw)
78     return xasprintf (_("%s: create failed"), filename);
79
80   struct spv_writer *w = xmalloc (sizeof *w);
81   *w = (struct spv_writer) { .zw = zw };
82   *writerp = w;
83   return NULL;
84 }
85
86 char * WARN_UNUSED_RESULT
87 spv_writer_close (struct spv_writer *w)
88 {
89   if (!w)
90     return NULL;
91
92   zip_writer_add_string (w->zw, "META-INF/MANIFEST.MF", "allowPivoting=true");
93
94   while (w->heading_depth)
95     spv_writer_close_heading (w);
96
97   char *error = NULL;
98   if (!zip_writer_close (w->zw))
99     error = xstrdup (_("I/O error writing SPV file"));
100
101   page_setup_destroy (w->page_setup);
102   free (w);
103   return error;
104 }
105
106 static void
107 write_attr (struct spv_writer *w, const char *name, const char *value)
108 {
109   xmlTextWriterWriteAttribute (w->xml,
110                                CHAR_CAST (xmlChar *, name),
111                                CHAR_CAST (xmlChar *, value));
112 }
113
114 static void PRINTF_FORMAT (3, 4)
115 write_attr_format (struct spv_writer *w, const char *name,
116                    const char *format, ...)
117 {
118   va_list args;
119   va_start (args, format);
120   char *value = xvasprintf (format, args);
121   va_end (args);
122
123   write_attr (w, name, value);
124   free (value);
125 }
126
127 static void
128 start_elem (struct spv_writer *w, const char *name)
129 {
130   xmlTextWriterStartElement (w->xml, CHAR_CAST (xmlChar *, name));
131 }
132
133 static void
134 end_elem (struct spv_writer *w)
135 {
136   xmlTextWriterEndElement (w->xml);
137 }
138
139 static void
140 write_text (struct spv_writer *w, const char *text)
141 {
142   xmlTextWriterWriteString (w->xml, CHAR_CAST (xmlChar *, text));
143 }
144
145 static void
146 write_page_heading (struct spv_writer *w, const struct page_heading *h,
147                     const char *name)
148 {
149   start_elem (w, name);
150   if (h->n)
151     {
152       start_elem (w, "pageParagraph");
153       for (size_t i = 0; i < h->n; i++)
154         {
155           start_elem (w, "text");
156           write_attr (w, "type", "title");
157           write_text (w, h->paragraphs[i].markup); /* XXX */
158           end_elem (w);
159         }
160       end_elem (w);
161     }
162   end_elem (w);
163 }
164
165 static void
166 write_page_setup (struct spv_writer *w, const struct page_setup *ps)
167 {
168   start_elem (w, "pageSetup");
169   write_attr_format (w, "initial-page-number", "%d", ps->initial_page_number);
170   write_attr (w, "chart-size",
171               (ps->chart_size == PAGE_CHART_AS_IS ? "as-is"
172                : ps->chart_size == PAGE_CHART_FULL_HEIGHT ? "full-height"
173                : ps->chart_size == PAGE_CHART_HALF_HEIGHT ? "half-height"
174                : "quarter-height"));
175   write_attr_format (w, "margin-left", "%.2fin", ps->margins[TABLE_HORZ][0]);
176   write_attr_format (w, "margin-right", "%.2fin", ps->margins[TABLE_HORZ][1]);
177   write_attr_format (w, "margin-top", "%.2fin", ps->margins[TABLE_VERT][0]);
178   write_attr_format (w, "margin-bottom", "%.2fin", ps->margins[TABLE_VERT][1]);
179   write_attr_format (w, "paper-height", "%.2fin", ps->paper[TABLE_VERT]);
180   write_attr_format (w, "paper-width", "%.2fin", ps->paper[TABLE_HORZ]);
181   write_attr (w, "reference-orientation",
182               ps->orientation == PAGE_PORTRAIT ? "portrait" : "landscape");
183   write_attr_format (w, "space-after", "%.1fpt", ps->object_spacing * 72.0);
184   write_page_heading (w, &ps->headings[0], "pageHeader");
185   write_page_heading (w, &ps->headings[1], "pageFooter");
186   end_elem (w);
187 }
188
189 static bool
190 spv_writer_open_file (struct spv_writer *w)
191 {
192   w->heading = create_temp_file ();
193   if (!w->heading)
194     return false;
195
196   w->xml = xmlNewTextWriter (xmlOutputBufferCreateFile (w->heading, NULL));
197   xmlTextWriterStartDocument (w->xml, NULL, "UTF-8", NULL);
198   start_elem (w, "heading");
199
200   time_t t = time (NULL);
201   struct tm *tm = gmtime (&t);
202   char *tm_s = asctime (tm);
203   write_attr (w, "creation-date-time", tm_s);
204
205   write_attr (w, "creator", version);
206
207   write_attr (w, "creator-version", "21");
208
209   write_attr (w, "xmlns", "http://xml.spss.com/spss/viewer/viewer-tree");
210   write_attr (w, "xmlns:vps", "http://xml.spss.com/spss/viewer/viewer-pagesetup");
211   write_attr (w, "xmlns:vtx", "http://xml.spss.com/spss/viewer/viewer-text");
212   write_attr (w, "xmlns:vtb", "http://xml.spss.com/spss/viewer/viewer-table");
213
214   start_elem (w, "label");
215   write_text (w, _("Output"));
216   end_elem (w);
217
218   if (w->page_setup)
219     {
220       write_page_setup (w, w->page_setup);
221
222       page_setup_destroy (w->page_setup);
223       w->page_setup = NULL;
224     }
225
226   return true;
227 }
228
229 static void
230 spv_writer_open_heading (struct spv_writer *w, const char *command_id,
231                          const char *label)
232 {
233   if (!w->heading)
234     {
235       if (!spv_writer_open_file (w))
236         return;
237     }
238
239   w->heading_depth++;
240   start_elem (w, "heading");
241   write_attr (w, "commandName", command_id);
242   /* XXX locale */
243   /* XXX olang */
244
245   start_elem (w, "label");
246   write_text (w, label);
247   end_elem (w);
248 }
249
250 static void
251 spv_writer_close_file (struct spv_writer *w, const char *infix)
252 {
253   if (!w->heading)
254     return;
255
256   end_elem (w);
257   xmlTextWriterEndDocument (w->xml);
258   xmlFreeTextWriter (w->xml);
259
260   char *member_name = xasprintf ("outputViewer%010d%s.xml",
261                                  w->n_headings++, infix);
262   zip_writer_add (w->zw, w->heading, member_name);
263   free (member_name);
264
265   w->heading = NULL;
266 }
267
268 static void
269 spv_writer_close_heading (struct spv_writer *w)
270 {
271   const char *infix = "";
272   if (w->heading_depth)
273     {
274       infix = "_heading";
275       end_elem (w);
276       w->heading_depth--;
277     }
278
279   if (!w->heading_depth)
280     spv_writer_close_file (w, infix);
281 }
282
283 static void
284 start_container (struct spv_writer *w, const struct output_item *item)
285 {
286   start_elem (w, "container");
287   write_attr (w, "visibility", "visible");
288   if (w->need_page_break)
289     {
290       write_attr (w, "page-break-before", "always");
291       w->need_page_break = false;
292     }
293
294   start_elem (w, "label");
295   write_text (w, output_item_get_label (item));
296   end_elem (w);
297 }
298
299 static void
300 spv_writer_put_text (struct spv_writer *w, const struct text_item *text,
301                      const char *command_id)
302 {
303   bool initial_depth = w->heading_depth;
304   if (!initial_depth)
305     spv_writer_open_file (w);
306
307   start_container (w, &text->output_item);
308
309   start_elem (w, "vtx:text");
310   write_attr (w, "type", (text->type == TEXT_ITEM_TITLE ? "title"
311                           : text->type == TEXT_ITEM_PAGE_TITLE ? "page-title"
312                           : "log"));
313   if (command_id)
314     write_attr (w, "commandName", command_id);
315
316   start_elem (w, "html");
317   write_text (w, text->text);   /* XXX */
318   end_elem (w); /* html */
319   end_elem (w); /* vtx:text */
320   end_elem (w); /* container */
321
322   if (!initial_depth)
323     spv_writer_close_file (w, "");
324 }
325
326 static cairo_status_t
327 write_to_zip (void *zw_, const unsigned char *data, unsigned int length)
328 {
329   struct zip_writer *zw = zw_;
330
331   zip_writer_add_write (zw, data, length);
332   return CAIRO_STATUS_SUCCESS;
333 }
334
335 static void
336 spv_writer_put_image (struct spv_writer *w, const struct output_item *item,
337                       cairo_surface_t *image)
338 {
339   bool initial_depth = w->heading_depth;
340   if (!initial_depth)
341     spv_writer_open_file (w);
342
343   char *uri = xasprintf ("%010d_Imagegeneric.png", ++w->n_tables);
344
345   start_container (w, item);
346
347   start_elem (w, "object");
348   write_attr (w, "type", "unknown");
349   write_attr (w, "uri", uri);
350   end_elem (w); /* object */
351   end_elem (w); /* container */
352
353   if (!initial_depth)
354     spv_writer_close_file (w, "");
355
356   zip_writer_add_start (w->zw, uri);
357   cairo_surface_write_to_png_stream (image, write_to_zip, w->zw);
358   zip_writer_add_finish (w->zw);
359
360   free (uri);
361 }
362 \f
363 #define H TABLE_HORZ
364 #define V TABLE_VERT
365
366 struct buf
367   {
368     uint8_t *data;
369     size_t len;
370     size_t allocated;
371   };
372
373 static uint8_t *
374 put_uninit (struct buf *b, size_t n)
375 {
376   while (b->allocated - b->len < n)
377     b->data = x2nrealloc (b->data, &b->allocated, sizeof b->data);
378   uint8_t *p = &b->data[b->len];
379   b->len += n;
380   return p;
381 }
382
383 static void
384 put_byte (struct buf *b, uint8_t byte)
385 {
386   *put_uninit (b, 1) = byte;
387 }
388
389 static void
390 put_bool (struct buf *b, bool boolean)
391 {
392   put_byte (b, boolean);
393 }
394
395 static void
396 put_bytes (struct buf *b, const char *bytes, size_t n)
397 {
398   memcpy (put_uninit (b, n), bytes, n);
399 }
400
401 static void
402 put_u16 (struct buf *b, uint16_t x)
403 {
404   put_uint16 (native_to_le16 (x), put_uninit (b, sizeof x));
405 }
406
407 static void
408 put_u32 (struct buf *b, uint32_t x)
409 {
410   put_uint32 (native_to_le32 (x), put_uninit (b, sizeof x));
411 }
412
413 static void
414 put_u64 (struct buf *b, uint64_t x)
415 {
416   put_uint64 (native_to_le64 (x), put_uninit (b, sizeof x));
417 }
418
419 static void
420 put_be32 (struct buf *b, uint32_t x)
421 {
422   put_uint32 (native_to_be32 (x), put_uninit (b, sizeof x));
423 }
424
425 static void
426 put_double (struct buf *b, double x)
427 {
428   float_convert (FLOAT_NATIVE_DOUBLE, &x,
429                  FLOAT_IEEE_DOUBLE_LE, put_uninit (b, 8));
430 }
431
432 static void
433 put_float (struct buf *b, float x)
434 {
435   float_convert (FLOAT_NATIVE_FLOAT, &x,
436                  FLOAT_IEEE_SINGLE_LE, put_uninit (b, 4));
437 }
438
439 static void
440 put_string (struct buf *b, const char *s_)
441 {
442   const char *s = s_ ? s_ : "";
443   size_t len = strlen (s);
444   put_u32 (b, len);
445   memcpy (put_uninit (b, len), s, len);
446 }
447
448 static void
449 put_bestring (struct buf *b, const char *s_)
450 {
451   const char *s = s_ ? s_ : "";
452   size_t len = strlen (s);
453   put_be32 (b, len);
454   memcpy (put_uninit (b, len), s, len);
455 }
456
457 static size_t
458 start_count (struct buf *b)
459 {
460   put_u32 (b, 0);
461   return b->len;
462 }
463
464 static void
465 end_count_u32 (struct buf *b, size_t start)
466 {
467   put_uint32 (native_to_le32 (b->len - start), &b->data[start - 4]);
468 }
469
470 static void
471 end_count_be32 (struct buf *b, size_t start)
472 {
473   put_uint32 (native_to_be32 (b->len - start), &b->data[start - 4]);
474 }
475
476 static void
477 put_color (struct buf *buf, const struct cell_color *color)
478 {
479   char *s = xasprintf ("#%02"PRIx8"%02"PRIx8"%02"PRIx8,
480                        color->r, color->g, color->b);
481   put_string (buf, s);
482   free (s);
483 }
484
485 static void
486 put_font_style (struct buf *buf, const struct font_style *font_style)
487 {
488   put_bool (buf, font_style->bold);
489   put_bool (buf, font_style->italic);
490   put_bool (buf, font_style->underline);
491   put_bool (buf, 1);
492   put_color (buf, &font_style->fg[0]);
493   put_color (buf, &font_style->bg[0]);
494   put_string (buf, font_style->typeface ? font_style->typeface : "SansSerif");
495   put_byte (buf, ceil (font_style->size * 1.33));
496 }
497
498 static void
499 put_halign (struct buf *buf, enum table_halign halign,
500             uint32_t mixed, uint32_t decimal)
501 {
502   put_u32 (buf, (halign == TABLE_HALIGN_RIGHT ? 4
503                  : halign == TABLE_HALIGN_LEFT ? 2
504                  : halign == TABLE_HALIGN_CENTER ? 0
505                  : halign == TABLE_HALIGN_MIXED ? mixed
506                  : decimal));
507 }
508
509 static void
510 put_valign (struct buf *buf, enum table_valign valign)
511 {
512   put_u32 (buf, (valign == TABLE_VALIGN_TOP ? 1
513                  : valign == TABLE_VALIGN_CENTER ? 0
514                  : 3));
515 }
516
517 static void
518 put_cell_style (struct buf *buf, const struct cell_style *cell_style)
519 {
520   put_halign (buf, cell_style->halign, 0xffffffad, 6);
521   put_valign (buf, cell_style->valign);
522   put_double (buf, cell_style->decimal_offset);
523   put_u16 (buf, cell_style->margin[H][0]);
524   put_u16 (buf, cell_style->margin[H][1]);
525   put_u16 (buf, cell_style->margin[V][0]);
526   put_u16 (buf, cell_style->margin[V][1]);
527 }
528
529 static void UNUSED
530 put_style_pair (struct buf *buf, const struct font_style *font_style,
531                 const struct cell_style *cell_style)
532 {
533   if (font_style)
534     {
535       put_byte (buf, 0x31);
536       put_font_style (buf, font_style);
537     }
538   else
539     put_byte (buf, 0x58);
540
541   if (cell_style)
542     {
543       put_byte (buf, 0x31);
544       put_cell_style (buf, cell_style);
545     }
546   else
547     put_byte (buf, 0x58);
548 }
549
550 static void
551 put_value_mod (struct buf *buf, const struct pivot_value *value,
552                const char *template)
553 {
554   if (value->n_footnotes || value->n_subscripts
555       || template || value->font_style || value->cell_style)
556     {
557       put_byte (buf, 0x31);
558
559       /* Footnotes. */
560       put_u32 (buf, value->n_footnotes);
561       for (size_t i = 0; i < value->n_footnotes; i++)
562         put_u16 (buf, value->footnote_indexes[i]);
563
564       /* Subscripts. */
565       put_u32 (buf, value->n_subscripts);
566       for (size_t i = 0; i < value->n_subscripts; i++)
567         put_string (buf, value->subscripts[i]);
568
569       /* Template and style. */
570       uint32_t v3_start = start_count (buf);
571       uint32_t template_string_start = start_count (buf);
572       if (template)
573         {
574           uint32_t inner_start = start_count (buf);
575           end_count_u32 (buf, inner_start);
576
577           put_byte (buf, 0x31);
578           put_string (buf, template);
579         }
580       end_count_u32 (buf, template_string_start);
581       put_style_pair (buf, value->font_style, value->cell_style);
582       end_count_u32 (buf, v3_start);
583     }
584   else
585     put_byte (buf, 0x58);
586 }
587
588 static void
589 put_format (struct buf *buf, const struct fmt_spec *f, bool honor_small)
590 {
591   int type = f->type == FMT_F && honor_small ? 40 : fmt_to_io (f->type);
592   put_u32 (buf, (type << 16) | (f->w << 8) | f->d);
593 }
594
595 static int
596 show_values_to_spvlb (enum settings_value_show show)
597 {
598   return (show == SETTINGS_VALUE_SHOW_DEFAULT ? 0
599           : show == SETTINGS_VALUE_SHOW_VALUE ? 1
600           : show == SETTINGS_VALUE_SHOW_LABEL ? 2
601           : 3);
602 }
603
604 static void
605 put_show_values (struct buf *buf, enum settings_value_show show)
606 {
607   put_byte (buf, show_values_to_spvlb (show));
608 }
609
610 static void
611 put_value (struct buf *buf, const struct pivot_value *value)
612 {
613   switch (value->type)
614     {
615     case PIVOT_VALUE_NUMERIC:
616       if (value->numeric.var_name || value->numeric.value_label)
617         {
618           put_byte (buf, 2);
619           put_value_mod (buf, value, NULL);
620           put_format (buf, &value->numeric.format, value->numeric.honor_small);
621           put_double (buf, value->numeric.x);
622           put_string (buf, value->numeric.var_name);
623           put_string (buf, value->numeric.value_label);
624           put_show_values (buf, value->numeric.show);
625         }
626       else
627         {
628           put_byte (buf, 1);
629           put_value_mod (buf, value, NULL);
630           put_format (buf, &value->numeric.format, value->numeric.honor_small);
631           put_double (buf, value->numeric.x);
632         }
633       break;
634
635     case PIVOT_VALUE_STRING:
636       put_byte (buf, 4);
637       put_value_mod (buf, value, NULL);
638       size_t len = strlen (value->string.s);
639       if (value->string.hex)
640         put_format (buf, &(struct fmt_spec) { FMT_AHEX, len * 2, 0 }, false);
641       else
642         put_format (buf, &(struct fmt_spec) { FMT_A, len, 0 }, false);
643       put_string (buf, value->string.value_label);
644       put_string (buf, value->string.var_name);
645       put_show_values (buf, value->string.show);
646       put_string (buf, value->string.s);
647       break;
648
649     case PIVOT_VALUE_VARIABLE:
650       put_byte (buf, 5);
651       put_value_mod (buf, value, NULL);
652       put_string (buf, value->variable.var_name);
653       put_string (buf, value->variable.var_label);
654       put_show_values (buf, value->variable.show);
655       break;
656
657     case PIVOT_VALUE_TEXT:
658       put_byte (buf, 3);
659       put_string (buf, value->text.local);
660       put_value_mod (buf, value, NULL);
661       put_string (buf, value->text.id);
662       put_string (buf, value->text.c);
663       put_byte (buf, 1);        /* XXX user-provided */
664       break;
665
666     case PIVOT_VALUE_TEMPLATE:
667       put_byte (buf, 0);
668       put_value_mod (buf, value, value->template.id);
669       put_string (buf, value->template.local);
670       put_u32 (buf, value->template.n_args);
671       for (size_t i = 0; i < value->template.n_args; i++)
672         {
673           const struct pivot_argument *arg = &value->template.args[i];
674           assert (arg->n >= 1);
675           if (arg->n > 1)
676             {
677               put_u32 (buf, arg->n);
678               put_u32 (buf, 0);
679               for (size_t j = 0; j < arg->n; j++)
680                 {
681                   if (j > 0)
682                     put_bytes (buf, "\0\0\0\0", 4);
683                   put_value (buf, arg->values[j]);
684                 }
685             }
686           else
687             {
688               put_u32 (buf, 0);
689               put_value (buf, arg->values[0]);
690             }
691         }
692       break;
693
694     default:
695       NOT_REACHED ();
696     }
697 }
698
699 static void
700 put_optional_value (struct buf *buf, const struct pivot_value *value)
701 {
702   if (value)
703     {
704       put_byte (buf, 0x31);
705       put_value (buf, value);
706     }
707   else
708     put_byte (buf, 0x58);
709 }
710
711 static void
712 put_category (struct buf *buf, const struct pivot_category *c)
713 {
714   put_value (buf, c->name);
715   if (pivot_category_is_leaf (c))
716     {
717       put_bytes (buf, "\0\0\0", 3);
718       put_u32 (buf, 2);
719       put_u32 (buf, c->data_index);
720       put_u32 (buf, 0);
721     }
722   else
723     {
724       put_bytes (buf, "\0\0\1", 3);
725       put_u32 (buf, 0);         /* x23 */
726       put_u32 (buf, -1);
727       put_u32 (buf, c->n_subs);
728       for (size_t i = 0; i < c->n_subs; i++)
729         put_category (buf, c->subs[i]);
730     }
731 }
732
733 static void
734 put_y0 (struct buf *buf, const struct pivot_table *table)
735 {
736   put_u32 (buf, table->settings.epoch);
737   put_byte (buf, table->settings.decimal);
738   put_byte (buf, ',');
739 }
740
741 static void
742 put_custom_currency (struct buf *buf, const struct pivot_table *table)
743 {
744   put_u32 (buf, 5);
745   for (int i = 0; i < 5; i++)
746     {
747       enum fmt_type types[5] = { FMT_CCA, FMT_CCB, FMT_CCC, FMT_CCD, FMT_CCE };
748       char *cc = fmt_number_style_to_string (fmt_settings_get_style (
749                                                &table->settings, types[i]));
750       put_string (buf, cc);
751       free (cc);
752     }
753 }
754
755 static void
756 put_x1 (struct buf *buf, const struct pivot_table *table)
757 {
758   put_byte (buf, 0);            /* x14 */
759   put_byte (buf, table->show_title ? 1 : 10);
760   put_byte (buf, 0);            /* x16 */
761   put_byte (buf, 0);            /* lang */
762   put_show_values (buf, table->show_variables);
763   put_show_values (buf, table->show_values);
764   put_u32 (buf, -1);            /* x18 */
765   put_u32 (buf, -1);            /* x19 */
766   for (int i = 0; i < 17; i++)
767     put_byte (buf, 0);
768   put_bool (buf, false);        /* x20 */
769   put_byte (buf, table->show_caption);
770 }
771
772 static void
773 put_x2 (struct buf *buf)
774 {
775   put_u32 (buf, 0);             /* n-row-heights */
776   put_u32 (buf, 0);             /* n-style-map */
777   put_u32 (buf, 0);             /* n-styles */
778   put_u32 (buf, 0);
779 }
780
781 static void
782 put_y1 (struct buf *buf, const struct pivot_table *table)
783 {
784   put_string (buf, table->command_c);
785   put_string (buf, table->command_local);
786   put_string (buf, table->language);
787   put_string (buf, "UTF-8");    /* XXX */
788   put_string (buf, table->locale);
789   put_bytes (buf, "\0\0\1\1", 4);
790   put_y0 (buf, table);
791 }
792
793 static void
794 put_y2 (struct buf *buf, const struct pivot_table *table)
795 {
796   put_custom_currency (buf, table);
797   put_byte (buf, '.');
798   put_bool (buf, 0);
799 }
800
801 static void
802 put_x3 (struct buf *buf, const struct pivot_table *table)
803 {
804   put_byte (buf, 1);
805   put_byte (buf, 0);
806   put_byte (buf, 4);            /* x21 */
807   put_byte (buf, 0);
808   put_byte (buf, 0);
809   put_byte (buf, 0);
810   put_y1 (buf, table);
811   put_double (buf, table->small);
812   put_byte (buf, 1);
813   put_string (buf, table->dataset);
814   put_string (buf, table->datafile);
815   put_u32 (buf, 0);
816   put_u32 (buf, table->date);
817   put_u32 (buf, 0);
818   put_y2 (buf, table);
819 }
820
821 static uint32_t
822 encode_current_layer (const struct pivot_table *table)
823 {
824   uint32_t current_layer = 0;
825
826   const struct pivot_axis *axis = &table->axes[PIVOT_AXIS_LAYER];
827   for (size_t i = axis->n_dimensions - 1; i < axis->n_dimensions; i--)
828     {
829       const struct pivot_dimension *d = axis->dimensions[i];
830       current_layer = current_layer * d->n_leaves + table->current_layer[i];
831     }
832
833   return current_layer;
834 }
835
836 static void
837 put_light_table (struct buf *buf, uint64_t table_id,
838                  const struct pivot_table *table)
839 {
840   /* Header. */
841   put_bytes (buf, "\1\0", 2);
842   put_u32 (buf, 3);
843   put_bool (buf, true);
844   put_bool (buf, false);
845   put_bool (buf, table->rotate_inner_column_labels);
846   put_bool (buf, table->rotate_outer_row_labels);
847   put_bool (buf, true);
848   put_u32 (buf, 0x15);
849   put_u32 (buf, table->look->width_ranges[H][0]);
850   put_u32 (buf, table->look->width_ranges[H][1]);
851   put_u32 (buf, table->look->width_ranges[V][0]);
852   put_u32 (buf, table->look->width_ranges[V][1]);
853   put_u64 (buf, table_id);
854
855   /* Titles. */
856   put_value (buf, table->title);
857   put_value (buf, table->subtype);
858   put_optional_value (buf, table->title);
859   put_optional_value (buf, table->corner_text);
860   put_optional_value (buf, table->caption);
861
862   /* Footnotes. */
863   put_u32 (buf, table->n_footnotes);
864   for (size_t i = 0; i < table->n_footnotes; i++)
865     {
866       const struct pivot_footnote *f = table->footnotes[i];
867       put_value (buf, f->content);
868       put_optional_value (buf, f->marker);
869       put_u32 (buf, f->show ? 1 : -1);
870     }
871
872   /* Areas. */
873   for (size_t i = 0; i < PIVOT_N_AREAS; i++)
874     {
875       const struct table_area_style *a = &table->look->areas[i];
876       put_byte (buf, i + 1);
877       put_byte (buf, 0x31);
878       put_string (buf, (a->font_style.typeface
879                         ? a->font_style.typeface
880                         : "SansSerif"));
881       put_float (buf, ceil (a->font_style.size * 1.33));
882       put_u32 (buf, ((a->font_style.bold ? 1 : 0)
883                      | (a->font_style.italic ? 2 : 0)));
884       put_bool (buf, a->font_style.underline);
885       put_halign (buf, a->cell_style.halign, 64173, 61453);
886       put_valign (buf, a->cell_style.valign);
887
888       put_color (buf, &a->font_style.fg[0]);
889       put_color (buf, &a->font_style.bg[0]);
890
891       bool alt
892         = (!cell_color_equal (&a->font_style.fg[0], &a->font_style.fg[1])
893            || !cell_color_equal (&a->font_style.bg[0], &a->font_style.bg[1]));
894       put_bool (buf, alt);
895       if (alt)
896         {
897           put_color (buf, &a->font_style.fg[1]);
898           put_color (buf, &a->font_style.bg[1]);
899         }
900       else
901         {
902           put_string (buf, "");
903           put_string (buf, "");
904         }
905
906       put_u32 (buf, a->cell_style.margin[H][0]);
907       put_u32 (buf, a->cell_style.margin[H][1]);
908       put_u32 (buf, a->cell_style.margin[V][0]);
909       put_u32 (buf, a->cell_style.margin[V][1]);
910     }
911
912   /* Borders. */
913   uint32_t borders_start = start_count (buf);
914   put_be32 (buf, 1);
915   put_be32 (buf, PIVOT_N_BORDERS);
916   for (size_t i = 0; i < PIVOT_N_BORDERS; i++)
917     {
918       const struct table_border_style *b = &table->look->borders[i];
919       put_be32 (buf, i);
920       put_be32 (buf, (b->stroke == TABLE_STROKE_NONE ? 0
921                       : b->stroke == TABLE_STROKE_SOLID ? 1
922                       : b->stroke == TABLE_STROKE_DASHED ? 2
923                       : b->stroke == TABLE_STROKE_THICK ? 3
924                       : b->stroke == TABLE_STROKE_THIN ? 4
925                       : 5));
926       put_be32 (buf, ((b->color.alpha << 24)
927                       | (b->color.r << 16)
928                       | (b->color.g << 8)
929                       | b->color.b));
930     }
931   put_bool (buf, table->show_grid_lines);
932   put_bytes (buf, "\0\0\0", 3);
933   end_count_u32 (buf, borders_start);
934
935   /* Print Settings. */
936   uint32_t ps_start = start_count (buf);
937   put_be32 (buf, 1);
938   put_bool (buf, table->look->print_all_layers);
939   put_bool (buf, table->look->paginate_layers);
940   put_bool (buf, table->look->shrink_to_fit[H]);
941   put_bool (buf, table->look->shrink_to_fit[V]);
942   put_bool (buf, table->look->top_continuation);
943   put_bool (buf, table->look->bottom_continuation);
944   put_be32 (buf, table->look->n_orphan_lines);
945   put_bestring (buf, table->look->continuation);
946   end_count_u32 (buf, ps_start);
947
948   /* Table Settings. */
949   uint32_t ts_start = start_count (buf);
950   put_be32 (buf, 1);
951   put_be32 (buf, 4);
952   put_be32 (buf, encode_current_layer (table));
953   put_bool (buf, table->look->omit_empty);
954   put_bool (buf, table->look->row_labels_in_corner);
955   put_bool (buf, !table->look->show_numeric_markers);
956   put_bool (buf, table->look->footnote_marker_superscripts);
957   put_byte (buf, 0);
958   uint32_t keep_start = start_count (buf);
959   put_be32 (buf, 0);            /* n-row-breaks */
960   put_be32 (buf, 0);            /* n-column-breaks */
961   put_be32 (buf, 0);            /* n-row-keeps */
962   put_be32 (buf, 0);            /* n-column-keeps */
963   put_be32 (buf, 0);            /* n-row-point-keeps */
964   put_be32 (buf, 0);            /* n-column-point-keeps */
965   end_count_be32 (buf, keep_start);
966   put_bestring (buf, table->notes);
967   put_bestring (buf, table->look->name);
968   for (size_t i = 0; i < 82; i++)
969     put_byte (buf, 0);
970   end_count_u32 (buf, ts_start);
971
972   /* Formats. */
973   put_u32 (buf, 0);             /* n-widths */
974   put_string (buf, "en_US.ISO_8859-1:1987"); /* XXX */
975   put_u32 (buf, 0);                /* XXX current-layer */
976   put_bool (buf, 0);
977   put_bool (buf, 0);
978   put_bool (buf, 1);
979   put_y0 (buf, table);
980   put_custom_currency (buf, table);
981   uint32_t formats_start = start_count (buf);
982   uint32_t x1_start = start_count (buf);
983   put_x1 (buf, table);
984   uint32_t x2_start = start_count (buf);
985   put_x2 (buf);
986   end_count_u32 (buf, x2_start);
987   end_count_u32 (buf, x1_start);
988   uint32_t x3_start = start_count (buf);
989   put_x3 (buf, table);
990   end_count_u32 (buf, x3_start);
991   end_count_u32 (buf, formats_start);
992
993   /* Dimensions. */
994   put_u32 (buf, table->n_dimensions);
995   int *x2 = xnmalloc (table->n_dimensions, sizeof *x2);
996   for (size_t i = 0; i < table->axes[PIVOT_AXIS_LAYER].n_dimensions; i++)
997     x2[i] = 2;
998   for (size_t i = 0; i < table->axes[PIVOT_AXIS_ROW].n_dimensions; i++)
999     x2[i + table->axes[PIVOT_AXIS_LAYER].n_dimensions] = 0;
1000   for (size_t i = 0; i < table->axes[PIVOT_AXIS_COLUMN].n_dimensions; i++)
1001     x2[i
1002        + table->axes[PIVOT_AXIS_LAYER].n_dimensions
1003        + table->axes[PIVOT_AXIS_ROW].n_dimensions] = 1;
1004   for (size_t i = 0; i < table->n_dimensions; i++)
1005     {
1006       const struct pivot_dimension *d = table->dimensions[i];
1007       put_value (buf, d->root->name);
1008       put_byte (buf, 0);        /* x1 */
1009       put_byte (buf, x2[i]);
1010       put_u32 (buf, 2);         /* x3 */
1011       put_bool (buf, !d->root->show_label);
1012       put_bool (buf, d->hide_all_labels);
1013       put_bool (buf, 1);
1014       put_u32 (buf, i);
1015
1016       put_u32 (buf, d->root->n_subs);
1017       for (size_t j = 0; j < d->root->n_subs; j++)
1018         put_category (buf, d->root->subs[j]);
1019     }
1020   free (x2);
1021
1022   /* Axes. */
1023   put_u32 (buf, table->axes[PIVOT_AXIS_LAYER].n_dimensions);
1024   put_u32 (buf, table->axes[PIVOT_AXIS_ROW].n_dimensions);
1025   put_u32 (buf, table->axes[PIVOT_AXIS_COLUMN].n_dimensions);
1026   for (size_t i = 0; i < table->axes[PIVOT_AXIS_LAYER].n_dimensions; i++)
1027     put_u32 (buf, table->axes[PIVOT_AXIS_LAYER].dimensions[i]->top_index);
1028   for (size_t i = 0; i < table->axes[PIVOT_AXIS_ROW].n_dimensions; i++)
1029     put_u32 (buf, table->axes[PIVOT_AXIS_ROW].dimensions[i]->top_index);
1030   for (size_t i = 0; i < table->axes[PIVOT_AXIS_COLUMN].n_dimensions; i++)
1031     put_u32 (buf, table->axes[PIVOT_AXIS_COLUMN].dimensions[i]->top_index);
1032
1033   /* Cells. */
1034   put_u32 (buf, hmap_count (&table->cells));
1035   const struct pivot_cell *cell;
1036   HMAP_FOR_EACH (cell, struct pivot_cell, hmap_node, &table->cells)
1037     {
1038       uint64_t index = 0;
1039       for (size_t j = 0; j < table->n_dimensions; j++)
1040         index = (table->dimensions[j]->n_leaves * index) + cell->idx[j];
1041       put_u64 (buf, index);
1042
1043       put_value (buf, cell->value);
1044     }
1045 }
1046
1047 static void
1048 spv_writer_put_table (struct spv_writer *w, const struct table_item *item)
1049 {
1050   int table_id = ++w->n_tables;
1051
1052   bool initial_depth = w->heading_depth;
1053   if (!initial_depth)
1054     spv_writer_open_file (w);
1055
1056   start_container (w, &item->output_item);
1057
1058   char *subtype = (item->pt->subtype
1059                    ? pivot_value_to_string (item->pt->subtype, item->pt)
1060                    : xstrdup ("unknown"));
1061
1062   start_elem (w, "vtb:table");
1063   write_attr (w, "commandName", item->pt->command_c);
1064   write_attr (w, "type", "table"); /* XXX */
1065   write_attr (w, "subType", subtype);
1066   write_attr_format (w, "tableId", "%d", table_id);
1067
1068   free (subtype);
1069
1070   start_elem (w, "vtb:tableStructure");
1071   start_elem (w, "vtb:dataPath");
1072   char *data_path = xasprintf ("%010d_lightTableData.bin", table_id);
1073   write_text (w, data_path);
1074   end_elem (w); /* vtb:dataPath */
1075   end_elem (w); /* vtb:tableStructure */
1076   end_elem (w); /* vtb:table */
1077   end_elem (w); /* container */
1078
1079   if (!initial_depth)
1080     spv_writer_close_file (w, "");
1081
1082   struct buf buf = { NULL, 0, 0 };
1083   put_light_table (&buf, table_id, item->pt);
1084   zip_writer_add_memory (w->zw, data_path, buf.data, buf.len);
1085   free (buf.data);
1086
1087   free (data_path);
1088 }
1089 \f
1090 void
1091 spv_writer_write (struct spv_writer *w, const struct output_item *item)
1092 {
1093   if (is_group_open_item (item))
1094     spv_writer_open_heading (w,
1095                              to_group_open_item (item)->command_name,
1096                              to_group_open_item (item)->command_name);
1097   else if (is_group_close_item (item))
1098     spv_writer_close_heading (w);
1099   else if (is_table_item (item))
1100     spv_writer_put_table (w, to_table_item (item));
1101   else if (is_chart_item (item))
1102     {
1103       cairo_surface_t *surface = xr_draw_image_chart (
1104         to_chart_item (item),
1105         &(struct cell_color) CELL_COLOR_BLACK,
1106         &(struct cell_color) CELL_COLOR_WHITE);
1107       if (cairo_surface_status (surface) == CAIRO_STATUS_SUCCESS)
1108         spv_writer_put_image (w, item, surface);
1109       cairo_surface_destroy (surface);
1110     }
1111   else if (is_image_item (item))
1112     spv_writer_put_image (w, item, to_image_item (item)->image);
1113   else if (is_text_item (item))
1114     {
1115       char *command_id = output_get_command_name ();
1116       spv_writer_put_text (w, to_text_item (item),
1117                            command_id);
1118       free (command_id);
1119     }
1120   else if (is_page_eject_item (item))
1121     w->need_page_break = true;
1122   else if (is_page_setup_item (item))
1123     {
1124       page_setup_destroy (w->page_setup);
1125       w->page_setup = page_setup_clone (to_page_setup_item (item)->page_setup);
1126     }
1127 }