Fixed bug rendering joined columns
[pspp-builds.git] / src / output / odt.c
index 6511aa3f10022bf23e675b32564c9fe1a4fec32b..8cc3da9ae72324fc6a38be4cf4e47bedf6d4af1c 100644 (file)
 #include <config.h>
 
 
+/* A driver for creating OpenDocument Format text files from PSPP's output */
+
 #include <libpspp/assertion.h>
-#include <libpspp/start-date.h>
 #include <libpspp/version.h>
-#include <output/afm.h>
-#include <output/chart-provider.h>
+
 #include <output/manager.h>
 #include <output/output.h>
 #include <output/table.h>
 
+#include <time.h>
+#include <pwd.h>
 #include <sys/stat.h>
 #include <sys/types.h>
 
 
 #include <libxml/xmlwriter.h>
 
-#include <stdlib.h>
-
-#include "error.h"
-#include "intprops.h"
-#include "minmax.h"
 #include "xalloc.h"
 
 #define _xml(X) (const xmlChar *)(X)
@@ -67,10 +64,12 @@ create_mimetype (const char *dirname)
   fp = fopen (ds_cstr (&filename), "w");
   ds_destroy (&filename);
 
+  assert (fp);
   fprintf (fp, "application/vnd.oasis.opendocument.text");
   fclose (fp);
 }
 
+/* Create a new XML file called FILENAME in the temp directory, and return a writer for it */
 static xmlTextWriterPtr
 create_writer (const struct odt_driver_ext *driver, const char *filename)
 {
@@ -95,6 +94,162 @@ create_writer (const struct odt_driver_ext *driver, const char *filename)
   return w;
 }
 
+
+static void
+register_file (struct odt_driver_ext *x, const char *filename)
+{
+  assert (x->manifest_wtr);
+  xmlTextWriterStartElement (x->manifest_wtr, _xml("manifest:file-entry"));
+  xmlTextWriterWriteAttribute (x->manifest_wtr, _xml("manifest:media-type"),  _xml("text/xml"));
+  xmlTextWriterWriteAttribute (x->manifest_wtr, _xml("manifest:full-path"),  _xml (filename));
+  xmlTextWriterEndElement (x->manifest_wtr);
+}
+
+static void
+write_style_data (struct odt_driver_ext *x)
+{
+  xmlTextWriterPtr w = create_writer (x, "styles.xml");
+  register_file (x, "styles.xml");
+
+  xmlTextWriterStartElement (w, _xml ("office:document-styles"));
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:office"),
+                              _xml ("urn:oasis:names:tc:opendocument:xmlns:office:1.0"));
+
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:style"),
+                              _xml ("urn:oasis:names:tc:opendocument:xmlns:style:1.0"));
+
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:fo"),
+                              _xml ("urn:oasis:names:tc:opendocument:xmlns:xsl-fo-compatible:1.0") );
+
+  xmlTextWriterWriteAttribute (w, _xml ("office:version"),  _xml ("1.1"));
+                              
+
+
+  xmlTextWriterStartElement (w, _xml ("office:styles"));
+
+
+  {
+    xmlTextWriterStartElement (w, _xml ("style:style"));
+    xmlTextWriterWriteAttribute (w, _xml ("style:name"),
+                                _xml ("Standard"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:family"),
+                                _xml ("paragraph"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:class"),
+                                _xml ("text"));
+
+    xmlTextWriterEndElement (w); /* style:style */
+  }
+
+  {
+    xmlTextWriterStartElement (w, _xml ("style:style"));
+    xmlTextWriterWriteAttribute (w, _xml ("style:name"),
+                                _xml ("Table_20_Contents"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:display-name"),
+                                _xml ("Table Contents"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:family"),
+                                _xml ("paragraph"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:parent-style-name"),
+                                _xml ("Standard"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:class"),
+                                _xml ("extra"));
+
+    xmlTextWriterEndElement (w); /* style:style */
+  }
+
+  {
+    xmlTextWriterStartElement (w, _xml ("style:style"));
+    xmlTextWriterWriteAttribute (w, _xml ("style:name"),
+                                _xml ("Table_20_Heading"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:display-name"),
+                                _xml ("Table Heading"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:family"),
+                                _xml ("paragraph"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:parent-style-name"),
+                                _xml ("Table_20_Contents"));
+
+    xmlTextWriterWriteAttribute (w, _xml ("style:class"),
+                                _xml ("extra"));
+
+
+    xmlTextWriterStartElement (w, _xml ("style:text-properties"));
+    xmlTextWriterWriteAttribute (w, _xml ("fo:font-weight"), _xml ("bold"));
+    xmlTextWriterWriteAttribute (w, _xml ("style:font-weight-asian"), _xml ("bold"));
+    xmlTextWriterWriteAttribute (w, _xml ("style:font-weight-complex"), _xml ("bold"));
+    xmlTextWriterEndElement (w); /* style:text-properties */
+
+    xmlTextWriterEndElement (w); /* style:style */
+  }
+
+
+  xmlTextWriterEndElement (w); /* office:styles */
+  xmlTextWriterEndElement (w); /* office:document-styles */
+
+  xmlTextWriterEndDocument (w);
+  xmlFreeTextWriter (w);
+}
+
+static void
+write_meta_data (struct odt_driver_ext *x)
+{
+  xmlTextWriterPtr w = create_writer (x, "meta.xml");
+  register_file (x, "meta.xml");
+
+  xmlTextWriterStartElement (w, _xml ("office:document-meta"));
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:office"), _xml ("urn:oasis:names:tc:opendocument:xmlns:office:1.0"));
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:dc"),  _xml ("http://purl.org/dc/elements/1.1/"));
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:meta"), _xml ("urn:oasis:names:tc:opendocument:xmlns:meta:1.0"));
+  xmlTextWriterWriteAttribute (w, _xml ("xmlns:ooo"), _xml("http://openoffice.org/2004/office"));
+  xmlTextWriterWriteAttribute (w, _xml ("office:version"),  _xml("1.1"));
+
+  xmlTextWriterStartElement (w, _xml ("office:meta"));
+  {
+    xmlTextWriterStartElement (w, _xml ("meta:generator"));
+    xmlTextWriterWriteString (w, _xml (stat_version));
+    xmlTextWriterEndElement (w);
+  }
+
+
+  {
+    char buf[30];
+    struct passwd *pw = getpwuid (getuid ());
+    time_t t = time (NULL);
+    struct tm *tm =  localtime (&t);
+
+    strftime (buf, 30, "%Y-%m-%dT%H:%M:%S", tm);
+
+    xmlTextWriterStartElement (w, _xml ("meta:initial-creator"));
+    xmlTextWriterWriteString (w, _xml (strtok (pw->pw_gecos, ",")));
+    xmlTextWriterEndElement (w);
+
+    xmlTextWriterStartElement (w, _xml ("meta:creation-date"));
+    xmlTextWriterWriteString (w, _xml (buf));
+    xmlTextWriterEndElement (w);
+
+    xmlTextWriterStartElement (w, _xml ("dc:creator"));
+    xmlTextWriterWriteString (w, _xml (strtok (pw->pw_gecos, ",")));
+
+    xmlTextWriterEndElement (w);
+
+    xmlTextWriterStartElement (w, _xml ("dc:date"));
+    xmlTextWriterWriteString (w, _xml (buf));
+    xmlTextWriterEndElement (w);
+  }
+
+  xmlTextWriterEndElement (w);
+  xmlTextWriterEndElement (w);
+  xmlTextWriterEndDocument (w);
+  xmlFreeTextWriter (w);
+}
+
 static bool
 odt_open_driver (const char *name, int types, struct substring option_string)
 {
@@ -108,10 +263,9 @@ odt_open_driver (const char *name, int types, struct substring option_string)
   x->dirname = xstrdup ("odt-XXXXXX");
   mkdtemp (x->dirname);
 
-  printf ("Dir name is \"%s\"\n", x->dirname);
-
   create_mimetype (x->dirname);
 
+  /* Create the manifest */
   x->manifest_wtr = create_writer (x, "META-INF/manifest.xml");
 
   xmlTextWriterStartElement (x->manifest_wtr, _xml("manifest:manifest"));
@@ -119,20 +273,21 @@ odt_open_driver (const char *name, int types, struct substring option_string)
                               _xml("urn:oasis:names:tc:opendocument:xmlns:manifest:1.0"));
 
 
+  /* Add a manifest entry for the document as a whole */
   xmlTextWriterStartElement (x->manifest_wtr, _xml("manifest:file-entry"));
   xmlTextWriterWriteAttribute (x->manifest_wtr, _xml("manifest:media-type"),  _xml("application/vnd.oasis.opendocument.text"));
   xmlTextWriterWriteAttribute (x->manifest_wtr, _xml("manifest:full-path"),  _xml("/"));
   xmlTextWriterEndElement (x->manifest_wtr);
 
 
-  x->content_wtr = create_writer (x, "content.xml");
+  write_meta_data (x);
+  write_style_data (x);
 
-  xmlTextWriterStartElement (x->manifest_wtr, _xml("manifest:file-entry"));
-  xmlTextWriterWriteAttribute (x->manifest_wtr, _xml("manifest:media-type"),  _xml("text/xml"));
-  xmlTextWriterWriteAttribute (x->manifest_wtr, _xml("manifest:full-path"),  _xml("content.xml"));
-  xmlTextWriterEndElement (x->manifest_wtr);
+  x->content_wtr = create_writer (x, "content.xml");
+  register_file (x, "content.xml");
 
 
+  /* Some necessary junk at the start */
   xmlTextWriterStartElement (x->content_wtr, _xml("office:document-content"));
   xmlTextWriterWriteAttribute (x->content_wtr, _xml("xmlns:office"),
                               _xml("urn:oasis:names:tc:opendocument:xmlns:office:1.0"));
@@ -149,6 +304,8 @@ odt_open_driver (const char *name, int types, struct substring option_string)
   xmlTextWriterStartElement (x->content_wtr, _xml("office:text"));
 
 
+
+  /* Close the manifest */
   xmlTextWriterEndElement (x->manifest_wtr);
   xmlTextWriterEndDocument (x->manifest_wtr);
   xmlFreeTextWriter (x->manifest_wtr);
@@ -159,28 +316,44 @@ odt_open_driver (const char *name, int types, struct substring option_string)
 static bool
 odt_close_driver (struct outp_driver *this)
 {
- struct odt_driver_ext *x = this->ext;
+  struct string zip_cmd;
+  struct string rm_cmd;
+  struct odt_driver_ext *x = this->ext;
 
- xmlTextWriterEndElement (x->content_wtr); /* office:text */
- xmlTextWriterEndElement (x->content_wtr); /* office:body */
- xmlTextWriterEndElement (x->content_wtr); /* office:document-content */
 xmlTextWriterEndElement (x->content_wtr); /* office:text */
 xmlTextWriterEndElement (x->content_wtr); /* office:body */
 xmlTextWriterEndElement (x->content_wtr); /* office:document-content */
 
- xmlTextWriterEndDocument (x->content_wtr);
- xmlFreeTextWriter (x->content_wtr);
 xmlTextWriterEndDocument (x->content_wtr);
 xmlFreeTextWriter (x->content_wtr);
 
- return true;
+  /* Zip up the directory */
+  ds_init_empty (&zip_cmd);
+  ds_put_format (&zip_cmd, "cd %s ; rm -f ../pspp.odt; zip -q -X ../pspp.odt mimetype; zip -q -X -u -r ../pspp.odt .", x->dirname);
+  system (ds_cstr (&zip_cmd));
+  ds_destroy (&zip_cmd);
+
+
+  /* Remove the temp dir */
+  ds_init_empty (&rm_cmd);
+  ds_put_format (&rm_cmd, "rm -r %s", x->dirname);
+  system (ds_cstr (&rm_cmd));
+  ds_destroy (&rm_cmd);
+
+  free (x->dirname);
+  free (x);
+
+  return true;
 }
 
 static void
 odt_open_page (struct outp_driver *this)
 {
- printf ("%s\n", __FUNCTION__);
 }
 
 static void
 odt_close_page (struct outp_driver *this)
 {
- printf ("%s\n", __FUNCTION__);
 }
 
 static void
@@ -189,6 +362,8 @@ odt_output_chart (struct outp_driver *this, const struct chart *chart)
  printf ("%s\n", __FUNCTION__);
 }
 
+
+/* Submit a table to the ODT driver */
 static void
 odt_submit (struct outp_driver *this, struct som_entity *e)
 {
@@ -197,11 +372,14 @@ odt_submit (struct outp_driver *this, struct som_entity *e)
   struct odt_driver_ext *x = this->ext;
   struct tab_table *tab = e->ext;
 
+
+  /* Write a heading for the table */
   xmlTextWriterStartElement (x->content_wtr, _xml("text:h"));
   xmlTextWriterWriteFormatAttribute (x->content_wtr, _xml("text:level"), "%d", e->subtable_num == 1 ? 2 : 3);
   xmlTextWriterWriteString (x->content_wtr, _xml (tab->title) );
   xmlTextWriterEndElement (x->content_wtr);
 
+  /* Start table */
   xmlTextWriterStartElement (x->content_wtr, _xml("table:table"));
   xmlTextWriterWriteFormatAttribute (x->content_wtr, _xml("table:name"), 
                                     "TABLE-%d.%d", e->table_num, e->subtable_num);
@@ -217,21 +395,56 @@ odt_submit (struct outp_driver *this, struct som_entity *e)
   if ( tab->t > 0)
     xmlTextWriterStartElement (x->content_wtr, _xml("table:table-header-rows"));
     
+
+  /* Write all the rows */
   for (r = 0 ; r < tab->nr; ++r)
     {
+      int spanned_columns = 0;
       /* Start row definition */
       xmlTextWriterStartElement (x->content_wtr, _xml("table:table-row"));
 
+      /* Write all the columns */
       for (c = 0 ; c < tab->nc ; ++c)
        {
-         int opts = tab->ct[tab->nc * r + c];
-         xmlTextWriterStartElement (x->content_wtr, _xml("table:table-cell"));
-         xmlTextWriterWriteAttribute (x->content_wtr, _xml("office:value-type"), _xml("string"));
+         char *s = NULL;
+         unsigned int opts = tab->ct[tab->nc * r + c];
+         struct substring ss = tab->cc[tab->nc * r + c];
 
-         if (! (opts & TAB_EMPTY) ) 
+         if (opts & TAB_EMPTY)
            {
-             char *s = ss_xstrdup (tab->cc[tab->nc * r + c]);
+             xmlTextWriterStartElement (x->content_wtr, _xml("table:table-cell"));
+             xmlTextWriterEndElement (x->content_wtr);
+             continue;
+           }
+
+         if ( opts & TAB_JOIN)
+           {
+             if ( spanned_columns == 0)
+               {
+                 struct tab_joined_cell *j = (struct tab_joined_cell*) ss_data (ss);
+                 s = ss_xstrdup (j->contents);
+               }
+           }
+         else
+           s = ss_xstrdup (ss);
+
+         if ( spanned_columns == 0 )
+           {
+             xmlTextWriterStartElement (x->content_wtr, _xml("table:table-cell"));
+             xmlTextWriterWriteAttribute (x->content_wtr, _xml("office:value-type"), _xml("string"));
+
+             if ( opts & TAB_JOIN )
+               {
+                 struct tab_joined_cell *j = (struct tab_joined_cell*) ss_data (ss);
+                 spanned_columns = j->x2 - j->x1;
+
+                 xmlTextWriterWriteFormatAttribute (x->content_wtr,
+                                                    _xml("table:number-columns-spanned"),
+                                                    "%d", spanned_columns);
+               }
+
              xmlTextWriterStartElement (x->content_wtr, _xml("text:p"));
+
              if ( r < tab->t || c < tab->l )
                xmlTextWriterWriteAttribute (x->content_wtr, _xml("text:style-name"), _xml("Table_20_Heading"));
              else
@@ -239,16 +452,24 @@ odt_submit (struct outp_driver *this, struct som_entity *e)
 
              xmlTextWriterWriteString (x->content_wtr, _xml (s));
          
+             xmlTextWriterEndElement (x->content_wtr); /* text:p */
+             xmlTextWriterEndElement (x->content_wtr); /* table:table-cell */
+           }
+         else
+           {
+             xmlTextWriterStartElement (x->content_wtr, _xml("table:covered-table-cell"));
              xmlTextWriterEndElement (x->content_wtr);
-             free (s);
            }
-         xmlTextWriterEndElement (x->content_wtr);
+         if ( opts & TAB_JOIN )
+           spanned_columns --;
+
+         free (s);
        }
   
       xmlTextWriterEndElement (x->content_wtr); /* row */
 
       if ( tab->t > 0 && r == tab->t - 1)
-       xmlTextWriterEndElement (x->content_wtr);
+       xmlTextWriterEndElement (x->content_wtr); /* table-header-rows */
     }
 
   xmlTextWriterEndElement (x->content_wtr); /* table */
@@ -258,7 +479,7 @@ odt_submit (struct outp_driver *this, struct som_entity *e)
 /* ODT driver class. */
 const struct outp_class odt_class =
 {
-  "odt",
+  "odf",
   1,
 
   odt_open_driver,