pintos-os.org Git - pspp/blob - src/output/spv-driver.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/driver-provider.h"
  20
  21 #include <stdlib.h>
  22
  23 #include "data/file-handle-def.h"
  24 #include "libpspp/cast.h"
  25 #include "output/group-item.h"
  26 #include "output/page-eject-item.h"
  27 #include "output/page-setup-item.h"
  28 #include "output/table-item.h"
  29 #include "output/text-item.h"
  30 #include "output/spv/spv-writer.h"
  31
  32 #include "gl/xalloc.h"
  33
  34 #include "gettext.h"
  35 #define _(msgid) gettext (msgid)
  36
  37 struct spv_driver
  38   {
  39     struct output_driver driver;
  40     struct spv_writer *writer;
  41     struct file_handle *handle;
  42   };
  43
  44 static const struct output_driver_class spv_driver_class;
  45
  46 static struct spv_driver *
  47 spv_driver_cast (struct output_driver *driver)
  48 {
  49   assert (driver->class == &spv_driver_class);
  50   return UP_CAST (driver, struct spv_driver, driver);
  51 }
  52
  53 static struct output_driver *
  54 spv_create (struct file_handle *fh, enum settings_output_devices device_type,
  55              struct string_map *o UNUSED)
  56 {
  57   struct output_driver *d;
  58   struct spv_driver *spv;
  59
  60   spv = xzalloc (sizeof *spv);
  61   d = &spv->driver;
  62   spv->handle = fh;
  63   output_driver_init (&spv->driver, &spv_driver_class, fh_get_file_name (fh),
  64                       device_type);
  65
  66   char *error = spv_writer_open (fh_get_file_name (fh), &spv->writer);
  67   if (spv->writer == NULL)
  68     {
  69       msg (ME, "%s", error);
  70       goto error;
  71     }
  72
  73   return d;
  74
  75  error:
  76   output_driver_destroy (d);
  77   return NULL;
  78 }
  79
  80 static void
  81 spv_destroy (struct output_driver *driver)
  82 {
  83   struct spv_driver *spv = spv_driver_cast (driver);
  84
  85   char *error = spv_writer_close (spv->writer);
  86   if (error)
  87     msg (ME, "%s", error);
  88   fh_unref (spv->handle);
  89   free (spv);
  90 }
  91
  92 static void
  93 spv_submit (struct output_driver *driver,
  94              const struct output_item *output_item)
  95 {
  96   struct spv_driver *spv = spv_driver_cast (driver);
  97
  98   if (is_group_open_item (output_item))
  99     spv_writer_open_heading (spv->writer,
 100                              to_group_open_item (output_item)->command_name,
 101                              to_group_open_item (output_item)->command_name);
 102   else if (is_group_close_item (output_item))
 103     spv_writer_close_heading (spv->writer);
 104   else if (is_table_item (output_item))
 105     {
 106       const struct table_item *table_item = to_table_item (output_item);
 107       if (table_item->pt)
 108         spv_writer_put_table (spv->writer, table_item->pt);
 109     }
 110   else if (is_text_item (output_item))
 111     {
 112       char *command_id = output_get_command_name ();
 113       spv_writer_put_text (spv->writer, to_text_item (output_item),
 114                            command_id);
 115       free (command_id);
 116     }
 117   else if (is_page_eject_item (output_item))
 118     spv_writer_eject_page (spv->writer);
 119   else if (is_page_setup_item (output_item))
 120     spv_writer_set_page_setup (spv->writer,
 121                                to_page_setup_item (output_item)->page_setup);
 122 }
 123
 124 struct output_driver_factory spv_driver_factory =
 125   { "spv", "pspp.spv", spv_create };
 126
 127 static const struct output_driver_class spv_driver_class =
 128   {
 129     "spv",
 130     spv_destroy,
 131     spv_submit,
 132     NULL,
 133   };