X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Flanguage%2Fdata-io%2Fprint-space.c;h=f077e9c39573e30760303704d0c356e95895bbf7;hb=510cc9dd9baf3108ba55cfb7893384517c9930b4;hp=39773ec0843d98588dd8c11401201e21a2f3983c;hpb=43b1296aafe7582e7dbe6c2b6a8b478d7d9b0fcf;p=pspp diff --git a/src/language/data-io/print-space.c b/src/language/data-io/print-space.c index 39773ec084..f077e9c395 100644 --- a/src/language/data-io/print-space.c +++ b/src/language/data-io/print-space.c @@ -1,5 +1,5 @@ /* PSPP - a program for statistical analysis. - Copyright (C) 2006 Free Software Foundation, Inc. + Copyright (C) 2006, 2009, 2010, 2011, 2012 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 @@ -19,17 +19,17 @@ #include #include -#include -#include -#include -#include -#include -#include -#include -#include -#include +#include "data/dataset.h" +#include "data/value.h" +#include "language/command.h" +#include "language/data-io/data-writer.h" +#include "language/data-io/file-handle.h" +#include "language/expressions/public.h" +#include "language/lexer/lexer.h" +#include "libpspp/message.h" +#include "output/driver.h" -#include "xalloc.h" +#include "gl/xalloc.h" #include "gettext.h" #define _(msgid) gettext (msgid) @@ -48,30 +48,40 @@ int cmd_print_space (struct lexer *lexer, struct dataset *ds) { struct print_space_trns *trns; - struct file_handle *handle; - struct expression *expr; + struct file_handle *handle = NULL; + struct expression *expr = NULL; struct dfm_writer *writer; + char *encoding = NULL; if (lex_match_id (lexer, "OUTFILE")) { - lex_match (lexer, '='); + lex_match (lexer, T_EQUALS); - handle = fh_parse (lexer, FH_REF_FILE); + handle = fh_parse (lexer, FH_REF_FILE, NULL); if (handle == NULL) return CMD_FAILURE; - lex_get (lexer); + + if (lex_match_id (lexer, "ENCODING")) + { + lex_match (lexer, T_EQUALS); + if (!lex_force_string (lexer)) + goto error; + + encoding = ss_xstrdup (lex_tokss (lexer)); + + lex_get (lexer); + } } else handle = NULL; - if (lex_token (lexer) != '.') + if (lex_token (lexer) != T_ENDCMD) { - expr = expr_parse (lexer, ds, EXPR_NUMBER); - if (lex_token (lexer) != '.') + expr = expr_parse (lexer, NULL, ds, VAL_NUMERIC); + if (lex_token (lexer) != T_ENDCMD) { - expr_free (expr); - lex_error (lexer, _("expecting end of command")); - return CMD_FAILURE; + lex_error (lexer, _("expecting end of command")); + goto error; } } else @@ -79,12 +89,9 @@ cmd_print_space (struct lexer *lexer, struct dataset *ds) if (handle != NULL) { - writer = dfm_open_writer (handle); + writer = dfm_open_writer (handle, encoding); if (writer == NULL) - { - expr_free (expr); - return CMD_FAILURE; - } + goto error; } else writer = NULL; @@ -95,12 +102,18 @@ cmd_print_space (struct lexer *lexer, struct dataset *ds) add_transformation (ds, print_space_trns_proc, print_space_trns_free, trns); + fh_unref (handle); return CMD_SUCCESS; + +error: + fh_unref (handle); + expr_free (expr); + return CMD_FAILURE; } /* Executes a PRINT SPACE transformation. */ static int -print_space_trns_proc (void *t_, struct ccase *c, +print_space_trns_proc (void *t_, struct ccase **c, casenumber case_num UNUSED) { struct print_space_trns *trns = t_; @@ -109,21 +122,21 @@ print_space_trns_proc (void *t_, struct ccase *c, n = 1; if (trns->expr) { - double f = expr_evaluate_num (trns->expr, c, case_num); + double f = expr_evaluate_num (trns->expr, *c, case_num); if (f == SYSMIS) - msg (SW, _("The expression on PRINT SPACE evaluated to the " - "system-missing value.")); + msg (SW, _("The expression on %s evaluated to the " + "system-missing value."), "PRINT SPACE"); else if (f < 0 || f > INT_MAX) - msg (SW, _("The expression on PRINT SPACE evaluated to %g."), f); + msg (SW, _("The expression on %s evaluated to %g."), "PRINT SPACE", f); else n = f; } while (n--) if (trns->writer == NULL) - som_blank_line (); + output_log ("%s", ""); else - dfm_put_record (trns->writer, " ", 1); + dfm_put_record (trns->writer, " ", 1); /* XXX */ if (trns->writer != NULL && dfm_write_error (trns->writer)) return TRNS_ERROR;