1 /* PSPP - computes sample statistics.
2 Copyright (C) 1997-2004 Free Software Foundation, Inc.
3 Written by Ben Pfaff <blp@gnu.org>.
5 This program is free software; you can redistribute it and/or
6 modify it under the terms of the GNU General Public License as
7 published by the Free Software Foundation; either version 2 of the
8 License, or (at your option) any later version.
10 This program is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
21 #include "data-writer.h"
27 #include "file-handle.h"
32 #define _(msgid) gettext (msgid)
34 /* Data file writer. */
37 struct file_handle *fh; /* File handle. */
38 struct file_ext file; /* Associated file. */
39 char *bounce; /* Bounce buffer for fixed-size fields. */
42 /* Opens a file handle for writing as a data file. */
44 dfm_open_writer (struct file_handle *fh)
49 aux = fh_open (fh, FH_REF_FILE, "data file", "ws");
55 w = *aux = xmalloc (sizeof *w);
60 w->file.filename = xstrdup (fh_get_filename (w->fh));
63 w->file.sequence_no = NULL;
65 w->file.postopen = NULL;
66 w->file.preclose = NULL;
68 if (!fn_open_ext (&w->file))
70 msg (ME, _("An error occurred while opening \"%s\" for writing "
71 "as a data file: %s."),
72 fh_get_filename (w->fh), strerror (errno));
83 /* Returns false if an I/O error occurred on WRITER, true otherwise. */
85 dfm_write_error (const struct dfm_writer *writer)
87 return ferror (writer->file.file);
90 /* Writes record REC having length LEN to the file corresponding to
91 HANDLE. REC is not null-terminated. Returns nonzero on success,
94 dfm_put_record (struct dfm_writer *w, const char *rec, size_t len)
98 if (dfm_write_error (w))
101 if (fh_get_mode (w->fh) == FH_MODE_BINARY
102 && len < fh_get_record_width (w->fh))
104 size_t rec_width = fh_get_record_width (w->fh);
105 if (w->bounce == NULL)
106 w->bounce = xmalloc (rec_width);
107 memcpy (w->bounce, rec, len);
108 memset (&w->bounce[len], 0, rec_width - len);
113 fwrite (rec, len, 1, w->file.file);
114 return !dfm_write_error (w);
117 /* Closes data file writer W. */
119 dfm_close_writer (struct dfm_writer *w)
125 if (fh_close (w->fh, "data file", "ws"))
129 if (w->file.file != NULL)
131 ok = !dfm_write_error (w);
132 if (!fn_close_ext (&w->file))
136 msg (ME, _("I/O error occurred writing data file \"%s\"."),
137 fh_get_filename (w->fh));
139 free (w->file.filename);
140 w->file.filename = NULL;