text-item: Simplify text_item further to just hold a pivot_value.
[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-break-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   char *s = text_item_get_plain_text (text);
318   write_text (w, s);
319   free (s);
320   end_elem (w); /* html */
321   end_elem (w); /* vtx:text */
322   end_elem (w); /* container */
323
324   if (!initial_depth)
325     spv_writer_close_file (w, "");
326 }
327
328 static cairo_status_t
329 write_to_zip (void *zw_, const unsigned char *data, unsigned int length)
330 {
331   struct zip_writer *zw = zw_;
332
333   zip_writer_add_write (zw, data, length);
334   return CAIRO_STATUS_SUCCESS;
335 }
336
337 static void
338 spv_writer_put_image (struct spv_writer *w, const struct output_item *item,
339                       cairo_surface_t *image)
340 {
341   bool initial_depth = w->heading_depth;
342   if (!initial_depth)
343     spv_writer_open_file (w);
344
345   char *uri = xasprintf ("%010d_Imagegeneric.png", ++w->n_tables);
346
347   start_container (w, item);
348
349   start_elem (w, "object");
350   write_attr (w, "type", "unknown");
351   write_attr (w, "uri", uri);
352   end_elem (w); /* object */
353   end_elem (w); /* container */
354
355   if (!initial_depth)
356     spv_writer_close_file (w, "");
357
358   zip_writer_add_start (w->zw, uri);
359   cairo_surface_write_to_png_stream (image, write_to_zip, w->zw);
360   zip_writer_add_finish (w->zw);
361
362   free (uri);
363 }
364 \f
365 #define H TABLE_HORZ
366 #define V TABLE_VERT
367
368 struct buf
369   {
370     uint8_t *data;
371     size_t len;
372     size_t allocated;
373   };
374
375 static uint8_t *
376 put_uninit (struct buf *b, size_t n)
377 {
378   while (b->allocated - b->len < n)
379     b->data = x2nrealloc (b->data, &b->allocated, sizeof b->data);
380   uint8_t *p = &b->data[b->len];
381   b->len += n;
382   return p;
383 }
384
385 static void
386 put_byte (struct buf *b, uint8_t byte)
387 {
388   *put_uninit (b, 1) = byte;
389 }
390
391 static void
392 put_bool (struct buf *b, bool boolean)
393 {
394   put_byte (b, boolean);
395 }
396
397 static void
398 put_bytes (struct buf *b, const char *bytes, size_t n)
399 {
400   memcpy (put_uninit (b, n), bytes, n);
401 }
402
403 static void
404 put_u16 (struct buf *b, uint16_t x)
405 {
406   put_uint16 (native_to_le16 (x), put_uninit (b, sizeof x));
407 }
408
409 static void
410 put_u32 (struct buf *b, uint32_t x)
411 {
412   put_uint32 (native_to_le32 (x), put_uninit (b, sizeof x));
413 }
414
415 static void
416 put_u64 (struct buf *b, uint64_t x)
417 {
418   put_uint64 (native_to_le64 (x), put_uninit (b, sizeof x));
419 }
420
421 static void
422 put_be32 (struct buf *b, uint32_t x)
423 {
424   put_uint32 (native_to_be32 (x), put_uninit (b, sizeof x));
425 }
426
427 static void
428 put_double (struct buf *b, double x)
429 {
430   float_convert (FLOAT_NATIVE_DOUBLE, &x,
431                  FLOAT_IEEE_DOUBLE_LE, put_uninit (b, 8));
432 }
433
434 static void
435 put_float (struct buf *b, float x)
436 {
437   float_convert (FLOAT_NATIVE_FLOAT, &x,
438                  FLOAT_IEEE_SINGLE_LE, put_uninit (b, 4));
439 }
440
441 static void
442 put_string (struct buf *b, const char *s_)
443 {
444   const char *s = s_ ? s_ : "";
445   size_t len = strlen (s);
446   put_u32 (b, len);
447   memcpy (put_uninit (b, len), s, len);
448 }
449
450 static void
451 put_bestring (struct buf *b, const char *s_)
452 {
453   const char *s = s_ ? s_ : "";
454   size_t len = strlen (s);
455   put_be32 (b, len);
456   memcpy (put_uninit (b, len), s, len);
457 }
458
459 static size_t
460 start_count (struct buf *b)
461 {
462   put_u32 (b, 0);
463   return b->len;
464 }
465
466 static void
467 end_count_u32 (struct buf *b, size_t start)
468 {
469   put_uint32 (native_to_le32 (b->len - start), &b->data[start - 4]);
470 }
471
472 static void
473 end_count_be32 (struct buf *b, size_t start)
474 {
475   put_uint32 (native_to_be32 (b->len - start), &b->data[start - 4]);
476 }
477
478 static void
479 put_color (struct buf *buf, const struct cell_color *color)
480 {
481   char *s = xasprintf ("#%02"PRIx8"%02"PRIx8"%02"PRIx8,
482                        color->r, color->g, color->b);
483   put_string (buf, s);
484   free (s);
485 }
486
487 static void
488 put_font_style (struct buf *buf, const struct font_style *font_style)
489 {
490   put_bool (buf, font_style->bold);
491   put_bool (buf, font_style->italic);
492   put_bool (buf, font_style->underline);
493   put_bool (buf, 1);
494   put_color (buf, &font_style->fg[0]);
495   put_color (buf, &font_style->bg[0]);
496   put_string (buf, font_style->typeface ? font_style->typeface : "SansSerif");
497   put_byte (buf, ceil (font_style->size * 1.33));
498 }
499
500 static void
501 put_halign (struct buf *buf, enum table_halign halign,
502             uint32_t mixed, uint32_t decimal)
503 {
504   put_u32 (buf, (halign == TABLE_HALIGN_RIGHT ? 4
505                  : halign == TABLE_HALIGN_LEFT ? 2
506                  : halign == TABLE_HALIGN_CENTER ? 0
507                  : halign == TABLE_HALIGN_MIXED ? mixed
508                  : decimal));
509 }
510
511 static void
512 put_valign (struct buf *buf, enum table_valign valign)
513 {
514   put_u32 (buf, (valign == TABLE_VALIGN_TOP ? 1
515                  : valign == TABLE_VALIGN_CENTER ? 0
516                  : 3));
517 }
518
519 static void
520 put_cell_style (struct buf *buf, const struct cell_style *cell_style)
521 {
522   put_halign (buf, cell_style->halign, 0xffffffad, 6);
523   put_valign (buf, cell_style->valign);
524   put_double (buf, cell_style->decimal_offset);
525   put_u16 (buf, cell_style->margin[H][0]);
526   put_u16 (buf, cell_style->margin[H][1]);
527   put_u16 (buf, cell_style->margin[V][0]);
528   put_u16 (buf, cell_style->margin[V][1]);
529 }
530
531 static void UNUSED
532 put_style_pair (struct buf *buf, const struct font_style *font_style,
533                 const struct cell_style *cell_style)
534 {
535   if (font_style)
536     {
537       put_byte (buf, 0x31);
538       put_font_style (buf, font_style);
539     }
540   else
541     put_byte (buf, 0x58);
542
543   if (cell_style)
544     {
545       put_byte (buf, 0x31);
546       put_cell_style (buf, cell_style);
547     }
548   else
549     put_byte (buf, 0x58);
550 }
551
552 static void
553 put_value_mod (struct buf *buf, const struct pivot_value *value,
554                const char *template)
555 {
556   if (value->n_footnotes || value->n_subscripts
557       || template || value->font_style || value->cell_style)
558     {
559       put_byte (buf, 0x31);
560
561       /* Footnotes. */
562       put_u32 (buf, value->n_footnotes);
563       for (size_t i = 0; i < value->n_footnotes; i++)
564         put_u16 (buf, value->footnote_indexes[i]);
565
566       /* Subscripts. */
567       put_u32 (buf, value->n_subscripts);
568       for (size_t i = 0; i < value->n_subscripts; i++)
569         put_string (buf, value->subscripts[i]);
570
571       /* Template and style. */
572       uint32_t v3_start = start_count (buf);
573       uint32_t template_string_start = start_count (buf);
574       if (template)
575         {
576           uint32_t inner_start = start_count (buf);
577           end_count_u32 (buf, inner_start);
578
579           put_byte (buf, 0x31);
580           put_string (buf, template);
581         }
582       end_count_u32 (buf, template_string_start);
583       put_style_pair (buf, value->font_style, value->cell_style);
584       end_count_u32 (buf, v3_start);
585     }
586   else
587     put_byte (buf, 0x58);
588 }
589
590 static void
591 put_format (struct buf *buf, const struct fmt_spec *f, bool honor_small)
592 {
593   int type = f->type == FMT_F && honor_small ? 40 : fmt_to_io (f->type);
594   put_u32 (buf, (type << 16) | (f->w << 8) | f->d);
595 }
596
597 static int
598 show_values_to_spvlb (enum settings_value_show show)
599 {
600   return (show == SETTINGS_VALUE_SHOW_DEFAULT ? 0
601           : show == SETTINGS_VALUE_SHOW_VALUE ? 1
602           : show == SETTINGS_VALUE_SHOW_LABEL ? 2
603           : 3);
604 }
605
606 static void
607 put_show_values (struct buf *buf, enum settings_value_show show)
608 {
609   put_byte (buf, show_values_to_spvlb (show));
610 }
611
612 static void
613 put_value (struct buf *buf, const struct pivot_value *value)
614 {
615   switch (value->type)
616     {
617     case PIVOT_VALUE_NUMERIC:
618       if (value->numeric.var_name || value->numeric.value_label)
619         {
620           put_byte (buf, 2);
621           put_value_mod (buf, value, NULL);
622           put_format (buf, &value->numeric.format, value->numeric.honor_small);
623           put_double (buf, value->numeric.x);
624           put_string (buf, value->numeric.var_name);
625           put_string (buf, value->numeric.value_label);
626           put_show_values (buf, value->numeric.show);
627         }
628       else
629         {
630           put_byte (buf, 1);
631           put_value_mod (buf, value, NULL);
632           put_format (buf, &value->numeric.format, value->numeric.honor_small);
633           put_double (buf, value->numeric.x);
634         }
635       break;
636
637     case PIVOT_VALUE_STRING:
638       put_byte (buf, 4);
639       put_value_mod (buf, value, NULL);
640       size_t len = strlen (value->string.s);
641       if (value->string.hex)
642         put_format (buf, &(struct fmt_spec) { FMT_AHEX, len * 2, 0 }, false);
643       else
644         put_format (buf, &(struct fmt_spec) { FMT_A, len, 0 }, false);
645       put_string (buf, value->string.value_label);
646       put_string (buf, value->string.var_name);
647       put_show_values (buf, value->string.show);
648       put_string (buf, value->string.s);
649       break;
650
651     case PIVOT_VALUE_VARIABLE:
652       put_byte (buf, 5);
653       put_value_mod (buf, value, NULL);
654       put_string (buf, value->variable.var_name);
655       put_string (buf, value->variable.var_label);
656       put_show_values (buf, value->variable.show);
657       break;
658
659     case PIVOT_VALUE_TEXT:
660       put_byte (buf, 3);
661       put_string (buf, value->text.local);
662       put_value_mod (buf, value, NULL);
663       put_string (buf, value->text.id);
664       put_string (buf, value->text.c);
665       put_byte (buf, 1);        /* XXX user-provided */
666       break;
667
668     case PIVOT_VALUE_TEMPLATE:
669       put_byte (buf, 0);
670       put_value_mod (buf, value, value->template.id);
671       put_string (buf, value->template.local);
672       put_u32 (buf, value->template.n_args);
673       for (size_t i = 0; i < value->template.n_args; i++)
674         {
675           const struct pivot_argument *arg = &value->template.args[i];
676           assert (arg->n >= 1);
677           if (arg->n > 1)
678             {
679               put_u32 (buf, arg->n);
680               put_u32 (buf, 0);
681               for (size_t j = 0; j < arg->n; j++)
682                 {
683                   if (j > 0)
684                     put_bytes (buf, "\0\0\0\0", 4);
685                   put_value (buf, arg->values[j]);
686                 }
687             }
688           else
689             {
690               put_u32 (buf, 0);
691               put_value (buf, arg->values[0]);
692             }
693         }
694       break;
695
696     default:
697       NOT_REACHED ();
698     }
699 }
700
701 static void
702 put_optional_value (struct buf *buf, const struct pivot_value *value)
703 {
704   if (value)
705     {
706       put_byte (buf, 0x31);
707       put_value (buf, value);
708     }
709   else
710     put_byte (buf, 0x58);
711 }
712
713 static void
714 put_category (struct buf *buf, const struct pivot_category *c)
715 {
716   put_value (buf, c->name);
717   if (pivot_category_is_leaf (c))
718     {
719       put_bytes (buf, "\0\0\0", 3);
720       put_u32 (buf, 2);
721       put_u32 (buf, c->data_index);
722       put_u32 (buf, 0);
723     }
724   else
725     {
726       put_bytes (buf, "\0\0\1", 3);
727       put_u32 (buf, 0);         /* x23 */
728       put_u32 (buf, -1);
729       put_u32 (buf, c->n_subs);
730       for (size_t i = 0; i < c->n_subs; i++)
731         put_category (buf, c->subs[i]);
732     }
733 }
734
735 static void
736 put_y0 (struct buf *buf, const struct pivot_table *table)
737 {
738   put_u32 (buf, table->settings.epoch);
739   put_byte (buf, table->settings.decimal);
740   put_byte (buf, ',');
741 }
742
743 static void
744 put_custom_currency (struct buf *buf, const struct pivot_table *table)
745 {
746   put_u32 (buf, 5);
747   for (int i = 0; i < 5; i++)
748     {
749       enum fmt_type types[5] = { FMT_CCA, FMT_CCB, FMT_CCC, FMT_CCD, FMT_CCE };
750       char *cc = fmt_number_style_to_string (fmt_settings_get_style (
751                                                &table->settings, types[i]));
752       put_string (buf, cc);
753       free (cc);
754     }
755 }
756
757 static void
758 put_x1 (struct buf *buf, const struct pivot_table *table)
759 {
760   put_byte (buf, 0);            /* x14 */
761   put_byte (buf, table->show_title ? 1 : 10);
762   put_byte (buf, 0);            /* x16 */
763   put_byte (buf, 0);            /* lang */
764   put_show_values (buf, table->show_variables);
765   put_show_values (buf, table->show_values);
766   put_u32 (buf, -1);            /* x18 */
767   put_u32 (buf, -1);            /* x19 */
768   for (int i = 0; i < 17; i++)
769     put_byte (buf, 0);
770   put_bool (buf, false);        /* x20 */
771   put_byte (buf, table->show_caption);
772 }
773
774 static void
775 put_x2 (struct buf *buf)
776 {
777   put_u32 (buf, 0);             /* n-row-heights */
778   put_u32 (buf, 0);             /* n-style-map */
779   put_u32 (buf, 0);             /* n-styles */
780   put_u32 (buf, 0);
781 }
782
783 static void
784 put_y1 (struct buf *buf, const struct pivot_table *table)
785 {
786   put_string (buf, table->command_c);
787   put_string (buf, table->command_local);
788   put_string (buf, table->language);
789   put_string (buf, "UTF-8");    /* XXX */
790   put_string (buf, table->locale);
791   put_bytes (buf, "\0\0\1\1", 4);
792   put_y0 (buf, table);
793 }
794
795 static void
796 put_y2 (struct buf *buf, const struct pivot_table *table)
797 {
798   put_custom_currency (buf, table);
799   put_byte (buf, '.');
800   put_bool (buf, 0);
801 }
802
803 static void
804 put_x3 (struct buf *buf, const struct pivot_table *table)
805 {
806   put_byte (buf, 1);
807   put_byte (buf, 0);
808   put_byte (buf, 4);            /* x21 */
809   put_byte (buf, 0);
810   put_byte (buf, 0);
811   put_byte (buf, 0);
812   put_y1 (buf, table);
813   put_double (buf, table->small);
814   put_byte (buf, 1);
815   put_string (buf, table->dataset);
816   put_string (buf, table->datafile);
817   put_u32 (buf, 0);
818   put_u32 (buf, table->date);
819   put_u32 (buf, 0);
820   put_y2 (buf, table);
821 }
822
823 static uint32_t
824 encode_current_layer (const struct pivot_table *table)
825 {
826   uint32_t current_layer = 0;
827
828   const struct pivot_axis *axis = &table->axes[PIVOT_AXIS_LAYER];
829   for (size_t i = axis->n_dimensions - 1; i < axis->n_dimensions; i--)
830     {
831       const struct pivot_dimension *d = axis->dimensions[i];
832       current_layer = current_layer * d->n_leaves + table->current_layer[i];
833     }
834
835   return current_layer;
836 }
837
838 static void
839 put_light_table (struct buf *buf, uint64_t table_id,
840                  const struct pivot_table *table)
841 {
842   /* Header. */
843   put_bytes (buf, "\1\0", 2);
844   put_u32 (buf, 3);
845   put_bool (buf, true);
846   put_bool (buf, false);
847   put_bool (buf, table->rotate_inner_column_labels);
848   put_bool (buf, table->rotate_outer_row_labels);
849   put_bool (buf, true);
850   put_u32 (buf, 0x15);
851   put_u32 (buf, table->look->width_ranges[H][0]);
852   put_u32 (buf, table->look->width_ranges[H][1]);
853   put_u32 (buf, table->look->width_ranges[V][0]);
854   put_u32 (buf, table->look->width_ranges[V][1]);
855   put_u64 (buf, table_id);
856
857   /* Titles. */
858   put_value (buf, table->title);
859   put_value (buf, table->subtype);
860   put_optional_value (buf, table->title);
861   put_optional_value (buf, table->corner_text);
862   put_optional_value (buf, table->caption);
863
864   /* Footnotes. */
865   put_u32 (buf, table->n_footnotes);
866   for (size_t i = 0; i < table->n_footnotes; i++)
867     {
868       const struct pivot_footnote *f = table->footnotes[i];
869       put_value (buf, f->content);
870       put_optional_value (buf, f->marker);
871       put_u32 (buf, f->show ? 1 : -1);
872     }
873
874   /* Areas. */
875   for (size_t i = 0; i < PIVOT_N_AREAS; i++)
876     {
877       const struct table_area_style *a = &table->look->areas[i];
878       put_byte (buf, i + 1);
879       put_byte (buf, 0x31);
880       put_string (buf, (a->font_style.typeface
881                         ? a->font_style.typeface
882                         : "SansSerif"));
883       put_float (buf, ceil (a->font_style.size * 1.33));
884       put_u32 (buf, ((a->font_style.bold ? 1 : 0)
885                      | (a->font_style.italic ? 2 : 0)));
886       put_bool (buf, a->font_style.underline);
887       put_halign (buf, a->cell_style.halign, 64173, 61453);
888       put_valign (buf, a->cell_style.valign);
889
890       put_color (buf, &a->font_style.fg[0]);
891       put_color (buf, &a->font_style.bg[0]);
892
893       bool alt
894         = (!cell_color_equal (&a->font_style.fg[0], &a->font_style.fg[1])
895            || !cell_color_equal (&a->font_style.bg[0], &a->font_style.bg[1]));
896       put_bool (buf, alt);
897       if (alt)
898         {
899           put_color (buf, &a->font_style.fg[1]);
900           put_color (buf, &a->font_style.bg[1]);
901         }
902       else
903         {
904           put_string (buf, "");
905           put_string (buf, "");
906         }
907
908       put_u32 (buf, a->cell_style.margin[H][0]);
909       put_u32 (buf, a->cell_style.margin[H][1]);
910       put_u32 (buf, a->cell_style.margin[V][0]);
911       put_u32 (buf, a->cell_style.margin[V][1]);
912     }
913
914   /* Borders. */
915   uint32_t borders_start = start_count (buf);
916   put_be32 (buf, 1);
917   put_be32 (buf, PIVOT_N_BORDERS);
918   for (size_t i = 0; i < PIVOT_N_BORDERS; i++)
919     {
920       const struct table_border_style *b = &table->look->borders[i];
921       put_be32 (buf, i);
922       put_be32 (buf, (b->stroke == TABLE_STROKE_NONE ? 0
923                       : b->stroke == TABLE_STROKE_SOLID ? 1
924                       : b->stroke == TABLE_STROKE_DASHED ? 2
925                       : b->stroke == TABLE_STROKE_THICK ? 3
926                       : b->stroke == TABLE_STROKE_THIN ? 4
927                       : 5));
928       put_be32 (buf, ((b->color.alpha << 24)
929                       | (b->color.r << 16)
930                       | (b->color.g << 8)
931                       | b->color.b));
932     }
933   put_bool (buf, table->show_grid_lines);
934   put_bytes (buf, "\0\0\0", 3);
935   end_count_u32 (buf, borders_start);
936
937   /* Print Settings. */
938   uint32_t ps_start = start_count (buf);
939   put_be32 (buf, 1);
940   put_bool (buf, table->look->print_all_layers);
941   put_bool (buf, table->look->paginate_layers);
942   put_bool (buf, table->look->shrink_to_fit[H]);
943   put_bool (buf, table->look->shrink_to_fit[V]);
944   put_bool (buf, table->look->top_continuation);
945   put_bool (buf, table->look->bottom_continuation);
946   put_be32 (buf, table->look->n_orphan_lines);
947   put_bestring (buf, table->look->continuation);
948   end_count_u32 (buf, ps_start);
949
950   /* Table Settings. */
951   uint32_t ts_start = start_count (buf);
952   put_be32 (buf, 1);
953   put_be32 (buf, 4);
954   put_be32 (buf, encode_current_layer (table));
955   put_bool (buf, table->look->omit_empty);
956   put_bool (buf, table->look->row_labels_in_corner);
957   put_bool (buf, !table->look->show_numeric_markers);
958   put_bool (buf, table->look->footnote_marker_superscripts);
959   put_byte (buf, 0);
960   uint32_t keep_start = start_count (buf);
961   put_be32 (buf, 0);            /* n-row-breaks */
962   put_be32 (buf, 0);            /* n-column-breaks */
963   put_be32 (buf, 0);            /* n-row-keeps */
964   put_be32 (buf, 0);            /* n-column-keeps */
965   put_be32 (buf, 0);            /* n-row-point-keeps */
966   put_be32 (buf, 0);            /* n-column-point-keeps */
967   end_count_be32 (buf, keep_start);
968   put_bestring (buf, table->notes);
969   put_bestring (buf, table->look->name);
970   for (size_t i = 0; i < 82; i++)
971     put_byte (buf, 0);
972   end_count_u32 (buf, ts_start);
973
974   /* Formats. */
975   put_u32 (buf, 0);             /* n-widths */
976   put_string (buf, "en_US.ISO_8859-1:1987"); /* XXX */
977   put_u32 (buf, 0);                /* XXX current-layer */
978   put_bool (buf, 0);
979   put_bool (buf, 0);
980   put_bool (buf, 1);
981   put_y0 (buf, table);
982   put_custom_currency (buf, table);
983   uint32_t formats_start = start_count (buf);
984   uint32_t x1_start = start_count (buf);
985   put_x1 (buf, table);
986   uint32_t x2_start = start_count (buf);
987   put_x2 (buf);
988   end_count_u32 (buf, x2_start);
989   end_count_u32 (buf, x1_start);
990   uint32_t x3_start = start_count (buf);
991   put_x3 (buf, table);
992   end_count_u32 (buf, x3_start);
993   end_count_u32 (buf, formats_start);
994
995   /* Dimensions. */
996   put_u32 (buf, table->n_dimensions);
997   int *x2 = xnmalloc (table->n_dimensions, sizeof *x2);
998   for (size_t i = 0; i < table->axes[PIVOT_AXIS_LAYER].n_dimensions; i++)
999     x2[i] = 2;
1000   for (size_t i = 0; i < table->axes[PIVOT_AXIS_ROW].n_dimensions; i++)
1001     x2[i + table->axes[PIVOT_AXIS_LAYER].n_dimensions] = 0;
1002   for (size_t i = 0; i < table->axes[PIVOT_AXIS_COLUMN].n_dimensions; i++)
1003     x2[i
1004        + table->axes[PIVOT_AXIS_LAYER].n_dimensions
1005        + table->axes[PIVOT_AXIS_ROW].n_dimensions] = 1;
1006   for (size_t i = 0; i < table->n_dimensions; i++)
1007     {
1008       const struct pivot_dimension *d = table->dimensions[i];
1009       put_value (buf, d->root->name);
1010       put_byte (buf, 0);        /* x1 */
1011       put_byte (buf, x2[i]);
1012       put_u32 (buf, 2);         /* x3 */
1013       put_bool (buf, !d->root->show_label);
1014       put_bool (buf, d->hide_all_labels);
1015       put_bool (buf, 1);
1016       put_u32 (buf, i);
1017
1018       put_u32 (buf, d->root->n_subs);
1019       for (size_t j = 0; j < d->root->n_subs; j++)
1020         put_category (buf, d->root->subs[j]);
1021     }
1022   free (x2);
1023
1024   /* Axes. */
1025   put_u32 (buf, table->axes[PIVOT_AXIS_LAYER].n_dimensions);
1026   put_u32 (buf, table->axes[PIVOT_AXIS_ROW].n_dimensions);
1027   put_u32 (buf, table->axes[PIVOT_AXIS_COLUMN].n_dimensions);
1028   for (size_t i = 0; i < table->axes[PIVOT_AXIS_LAYER].n_dimensions; i++)
1029     put_u32 (buf, table->axes[PIVOT_AXIS_LAYER].dimensions[i]->top_index);
1030   for (size_t i = 0; i < table->axes[PIVOT_AXIS_ROW].n_dimensions; i++)
1031     put_u32 (buf, table->axes[PIVOT_AXIS_ROW].dimensions[i]->top_index);
1032   for (size_t i = 0; i < table->axes[PIVOT_AXIS_COLUMN].n_dimensions; i++)
1033     put_u32 (buf, table->axes[PIVOT_AXIS_COLUMN].dimensions[i]->top_index);
1034
1035   /* Cells. */
1036   put_u32 (buf, hmap_count (&table->cells));
1037   const struct pivot_cell *cell;
1038   HMAP_FOR_EACH (cell, struct pivot_cell, hmap_node, &table->cells)
1039     {
1040       uint64_t index = 0;
1041       for (size_t j = 0; j < table->n_dimensions; j++)
1042         index = (table->dimensions[j]->n_leaves * index) + cell->idx[j];
1043       put_u64 (buf, index);
1044
1045       put_value (buf, cell->value);
1046     }
1047 }
1048
1049 static void
1050 spv_writer_put_table (struct spv_writer *w, const struct table_item *item)
1051 {
1052   int table_id = ++w->n_tables;
1053
1054   bool initial_depth = w->heading_depth;
1055   if (!initial_depth)
1056     spv_writer_open_file (w);
1057
1058   start_container (w, &item->output_item);
1059
1060   char *subtype = (item->pt->subtype
1061                    ? pivot_value_to_string (item->pt->subtype, item->pt)
1062                    : xstrdup ("unknown"));
1063
1064   start_elem (w, "vtb:table");
1065   write_attr (w, "commandName", item->pt->command_c);
1066   write_attr (w, "type", "table"); /* XXX */
1067   write_attr (w, "subType", subtype);
1068   write_attr_format (w, "tableId", "%d", table_id);
1069
1070   free (subtype);
1071
1072   start_elem (w, "vtb:tableStructure");
1073   start_elem (w, "vtb:dataPath");
1074   char *data_path = xasprintf ("%010d_lightTableData.bin", table_id);
1075   write_text (w, data_path);
1076   end_elem (w); /* vtb:dataPath */
1077   end_elem (w); /* vtb:tableStructure */
1078   end_elem (w); /* vtb:table */
1079   end_elem (w); /* container */
1080
1081   if (!initial_depth)
1082     spv_writer_close_file (w, "");
1083
1084   struct buf buf = { NULL, 0, 0 };
1085   put_light_table (&buf, table_id, item->pt);
1086   zip_writer_add_memory (w->zw, data_path, buf.data, buf.len);
1087   free (buf.data);
1088
1089   free (data_path);
1090 }
1091 \f
1092 void
1093 spv_writer_write (struct spv_writer *w, const struct output_item *item)
1094 {
1095   if (is_group_open_item (item))
1096     spv_writer_open_heading (w,
1097                              to_group_open_item (item)->command_name,
1098                              to_group_open_item (item)->command_name);
1099   else if (is_group_close_item (item))
1100     spv_writer_close_heading (w);
1101   else if (is_table_item (item))
1102     spv_writer_put_table (w, to_table_item (item));
1103   else if (is_chart_item (item))
1104     {
1105       cairo_surface_t *surface = xr_draw_image_chart (
1106         to_chart_item (item),
1107         &(struct cell_color) CELL_COLOR_BLACK,
1108         &(struct cell_color) CELL_COLOR_WHITE);
1109       if (cairo_surface_status (surface) == CAIRO_STATUS_SUCCESS)
1110         spv_writer_put_image (w, item, surface);
1111       cairo_surface_destroy (surface);
1112     }
1113   else if (is_image_item (item))
1114     spv_writer_put_image (w, item, to_image_item (item)->image);
1115   else if (is_text_item (item))
1116     {
1117       char *command_id = output_get_command_name ();
1118       spv_writer_put_text (w, to_text_item (item),
1119                            command_id);
1120       free (command_id);
1121     }
1122   else if (is_page_break_item (item))
1123     w->need_page_break = true;
1124   else if (is_page_setup_item (item))
1125     {
1126       page_setup_destroy (w->page_setup);
1127       w->page_setup = page_setup_clone (to_page_setup_item (item)->page_setup);
1128     }
1129 }