X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Foutput%2Fcsv.c;h=b29a6750538ac87fab0ccec4db1467ed714fbdad;hb=d950831f83004457e69c3e6e44131aef3e57efd3;hp=004558f1b1dee53974680f26e60a3817d93ebd05;hpb=ddb7b52128d8f1f54d9632dc3a15c7869e0fbcce;p=pspp diff --git a/src/output/csv.c b/src/output/csv.c index 004558f1b1..b29a675053 100644 --- a/src/output/csv.c +++ b/src/output/csv.c @@ -1,5 +1,5 @@ /* PSPP - a program for statistical analysis. - Copyright (C) 2009, 2010 Free Software Foundation, Inc. + Copyright (C) 2009, 2010, 2012, 2013, 2014 Free Software Foundation, Inc. This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -20,18 +20,21 @@ #include #include "data/file-name.h" +#include "data/file-handle-def.h" #include "libpspp/assertion.h" #include "libpspp/compiler.h" #include "libpspp/message.h" +#include "libpspp/str.h" #include "libpspp/string-map.h" #include "output/text-item.h" #include "output/driver-provider.h" #include "output/options.h" #include "output/message-item.h" +#include "output/page-eject-item.h" #include "output/table-item.h" #include "output/table-provider.h" -#include "gl/error.h" +#include "gl/minmax.h" #include "gl/xalloc.h" #include "gl/xvasprintf.h" @@ -43,9 +46,13 @@ struct csv_driver { struct output_driver driver; - char *separator; /* Comma or tab. */ - char *file_name; /* Output file name. */ - char *command_name; /* Current command. */ + char *separator; /* Field separator (usually comma or tab). */ + int quote; /* Quote character (usually ' or ") or 0. */ + char *quote_set; /* Characters that force quoting. */ + bool titles; /* Print table titles? */ + bool captions; /* Print table captions? */ + + struct file_handle *handle; FILE *file; /* Output file. */ int n_items; /* Number of items output so far. */ }; @@ -67,24 +74,31 @@ opt (struct output_driver *d, struct string_map *options, const char *key, } static struct output_driver * -csv_create (const char *file_name, enum settings_output_devices device_type, +csv_create (struct file_handle *fh, enum settings_output_devices device_type, struct string_map *o) { struct output_driver *d; struct csv_driver *csv; + char *quote; csv = xzalloc (sizeof *csv); d = &csv->driver; - output_driver_init (&csv->driver, &csv_driver_class, file_name, device_type); + output_driver_init (&csv->driver, &csv_driver_class, fh_get_file_name (fh), device_type); csv->separator = parse_string (opt (d, o, "separator", ",")); - csv->file_name = parse_string (opt (d, o, "output-file", "pspp.csv")); - csv->file = fn_open (csv->file_name, "w"); + quote = parse_string (opt (d, o, "quote", "\"")); + csv->quote = quote[0]; + free (quote); + csv->quote_set = xasprintf ("\n\r\t%s%c", csv->separator, csv->quote); + csv->titles = parse_boolean (opt (d, o, "titles", "true")); + csv->captions = parse_boolean (opt (d, o, "captions", "true")); + csv->handle = fh; + csv->file = fn_open (fh, "w"); csv->n_items = 0; if (csv->file == NULL) { - error (0, errno, _("csv: opening output file \"%s\""), csv->file_name); + msg_error (errno, _("error opening output file `%s'"), fh_get_file_name (fh)); output_driver_destroy (d); return NULL; } @@ -98,10 +112,11 @@ csv_destroy (struct output_driver *driver) struct csv_driver *csv = csv_driver_cast (driver); if (csv->file != NULL) - fn_close (csv->file_name, csv->file); + fn_close (csv->handle, csv->file); free (csv->separator); - free (csv->file_name); + free (csv->quote_set); + fh_unref (csv->handle); free (csv); } @@ -114,51 +129,112 @@ csv_flush (struct output_driver *driver) } static void -csv_output_field (FILE *file, const char *field) +csv_output_field__ (struct csv_driver *csv, struct substring field) { - while (*field == ' ') - field++; + ss_ltrim (&field, ss_cstr (" ")); - if (field[strcspn (field, "\"\n\r,\t")]) + if (csv->quote && ss_cspan (field, ss_cstr (csv->quote_set)) < field.length) { - const char *p; - - putc ('"', file); - for (p = field; *p != '\0'; p++) + putc (csv->quote, csv->file); + for (size_t i = 0; i < field.length; i++) { - if (*p == '"') - putc ('"', file); - putc (*p, file); + if (field.string[i] == csv->quote) + putc (csv->quote, csv->file); + putc (field.string[i], csv->file); } - putc ('"', file); + putc (csv->quote, csv->file); } else - fputs (field, file); + fwrite (field.string, field.length, 1, csv->file); } static void -csv_output_field_format (FILE *file, const char *format, ...) - PRINTF_FORMAT (2, 3); +csv_output_field (struct csv_driver *csv, const char *field) +{ + csv_output_field__ (csv, ss_cstr (field)); +} static void -csv_output_field_format (FILE *file, const char *format, ...) +csv_put_separator (struct csv_driver *csv) { - va_list args; - char *s; + if (csv->n_items++ > 0) + putc ('\n', csv->file); +} - va_start (args, format); - s = xvasprintf (format, args); - va_end (args); +static void +csv_output_lines (struct csv_driver *csv, const char *text_) +{ + struct substring text = ss_cstr (text_); + struct substring line; + size_t save_idx = 0; + while (ss_separate (text, ss_cstr ("\n"), &save_idx, &line)) + { + csv_output_field__ (csv, line); + putc ('\n', csv->file); + } +} - csv_output_field (file, s); - free (s); +static void +csv_format_footnotes (struct footnote **f, size_t n, struct string *s) +{ + for (size_t i = 0; i < n; i++) + ds_put_format (s, "[%s]", f[i]->marker); } static void -csv_put_separator (struct csv_driver *csv) +csv_output_table_item_text (struct csv_driver *csv, + const struct table_item_text *text, + const char *leader) { - if (csv->n_items++ > 0) - putc ('\n', csv->file); + if (!text) + return; + + struct string s = DS_EMPTY_INITIALIZER; + ds_put_format (&s, "%s: %s", leader, text->content); + csv_format_footnotes (text->footnotes, text->n_footnotes, &s); + csv_output_field (csv, ds_cstr (&s)); + ds_destroy (&s); + putc ('\n', csv->file); +} + +static void +csv_output_table_cell (struct csv_driver *csv, + const struct table_cell *cell, + const char *leader) +{ + if (!cell) + return; + + if (!(cell->options & TAB_MARKUP) && !cell->n_footnotes + && !cell->n_subscripts && !leader) + csv_output_field (csv, cell->text); + else + { + struct string s = DS_EMPTY_INITIALIZER; + + if (leader) + ds_put_format (&s, "%s: ", leader); + + if (cell->options & TAB_MARKUP) + { + char *t = output_get_text_from_markup (cell->text); + ds_put_cstr (&s, t); + free (t); + } + else + ds_put_cstr (&s, cell->text); + + if (cell->n_subscripts) + for (size_t i = 0; i < cell->n_subscripts; i++) + ds_put_format (&s, "%c%s", + i ? ',' : '_', cell->subscripts[i]); + csv_format_footnotes (cell->footnotes, cell->n_footnotes, &s); + csv_output_field (csv, ds_cstr (&s)); + ds_destroy (&s); + + if (leader) + putc ('\n', csv->file); + } } static void @@ -167,26 +243,20 @@ csv_submit (struct output_driver *driver, { struct csv_driver *csv = csv_driver_cast (driver); - output_driver_track_current_command (output_item, &csv->command_name); - if (is_table_item (output_item)) { struct table_item *table_item = to_table_item (output_item); - const char *caption = table_item_get_caption (table_item); const struct table *t = table_item_get_table (table_item); int x, y; csv_put_separator (csv); - if (caption != NULL) - { - csv_output_field_format (csv->file, "Table: %s", caption); - putc ('\n', csv->file); - } + if (csv->titles) + csv_output_table_cell (csv, table_item_get_title (table_item), "Table"); - for (y = 0; y < table_nr (t); y++) + for (y = 0; y < t->n[TABLE_VERT]; y++) { - for (x = 0; x < table_nc (t); x++) + for (x = 0; x < t->n[TABLE_HORZ]; x++) { struct table_cell cell; @@ -196,14 +266,33 @@ csv_submit (struct output_driver *driver, fputs (csv->separator, csv->file); if (x != cell.d[TABLE_HORZ][0] || y != cell.d[TABLE_VERT][0]) - csv_output_field (csv->file, ""); + csv_output_field (csv, ""); else - csv_output_field (csv->file, cell.contents); - - table_cell_free (&cell); + csv_output_table_cell (csv, &cell, NULL); } putc ('\n', csv->file); } + + if (csv->captions) + csv_output_table_item_text (csv, table_item_get_caption (table_item), + "Caption"); + + struct footnote **f; + size_t n_footnotes = table_collect_footnotes (table_item, &f); + if (n_footnotes) + { + fputs ("\nFootnotes:\n", csv->file); + + for (size_t i = 0; i < n_footnotes; i++) + { + csv_output_field (csv, f[i]->marker); + fputs (csv->separator, csv->file); + csv_output_field (csv, f[i]->content); + putc ('\n', csv->file); + } + + free (f); + } } else if (is_text_item (output_item)) { @@ -211,40 +300,37 @@ csv_submit (struct output_driver *driver, enum text_item_type type = text_item_get_type (text_item); const char *text = text_item_get_text (text_item); - if (type == TEXT_ITEM_COMMAND_OPEN || type == TEXT_ITEM_COMMAND_CLOSE - || type == TEXT_ITEM_SYNTAX) + if (type == TEXT_ITEM_SYNTAX || type == TEXT_ITEM_PAGE_TITLE) return; csv_put_separator (csv); - switch (type) - { - case TEXT_ITEM_TITLE: - csv_output_field_format (csv->file, "Title: %s", text); - break; - case TEXT_ITEM_SUBTITLE: - csv_output_field_format (csv->file, "Subtitle: %s", text); - break; - - default: - csv_output_field (csv->file, text); - break; + if (text_item->markup) + { + char *plain_text = output_get_text_from_markup (text); + csv_output_lines (csv, plain_text); + free (plain_text); } - putc ('\n', csv->file); + else + csv_output_lines (csv, text); + } + else if (is_page_eject_item (output_item)) + { + csv_put_separator (csv); + csv_output_lines (csv, ""); } else if (is_message_item (output_item)) { const struct message_item *message_item = to_message_item (output_item); - const struct msg *msg = message_item_get_msg (message_item); - char *s = msg_to_string (msg, csv->command_name); + char *s = msg_to_string (message_item_get_msg (message_item)); csv_put_separator (csv); - csv_output_field (csv->file, s); + csv_output_field (csv, s); free (s); putc ('\n', csv->file); } } -struct output_driver_factory csv_driver_factory = { "csv", csv_create }; +struct output_driver_factory csv_driver_factory = { "csv", "-", csv_create }; static const struct output_driver_class csv_driver_class = {