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