48645fc29d68c2b8a803053109df8b77c0885f9e
[pspp-builds.git] / src / language / data-io / save.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-9, 2000, 2006, 2007, 2008, 2009, 2010, 2011 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 <stdlib.h>
20
21 #include "data/any-writer.h"
22 #include "data/case-map.h"
23 #include "data/case.h"
24 #include "data/casereader.h"
25 #include "data/casewriter.h"
26 #include "data/dataset.h"
27 #include "data/dictionary.h"
28 #include "data/por-file-writer.h"
29 #include "data/sys-file-writer.h"
30 #include "data/transformations.h"
31 #include "data/variable.h"
32 #include "language/command.h"
33 #include "language/data-io/file-handle.h"
34 #include "language/data-io/trim.h"
35 #include "language/lexer/lexer.h"
36 #include "libpspp/assertion.h"
37 #include "libpspp/compiler.h"
38
39 #include "gl/xalloc.h"
40
41 #include "gettext.h"
42 #define _(msgid) gettext (msgid)
43
44 /* Writing system and portable files. */
45
46 /* Type of output file. */
47 enum writer_type
48   {
49     SYSFILE_WRITER,     /* System file. */
50     PORFILE_WRITER      /* Portable file. */
51   };
52
53 /* Type of a command. */
54 enum command_type
55   {
56     XFORM_CMD,          /* Transformation. */
57     PROC_CMD            /* Procedure. */
58   };
59
60 static int parse_output_proc (struct lexer *, struct dataset *,
61                               enum writer_type);
62 static int parse_output_trns (struct lexer *, struct dataset *,
63                               enum writer_type);
64
65 int
66 cmd_save (struct lexer *lexer, struct dataset *ds)
67 {
68   return parse_output_proc (lexer, ds, SYSFILE_WRITER);
69 }
70
71 int
72 cmd_export (struct lexer *lexer, struct dataset *ds)
73 {
74   return parse_output_proc (lexer, ds, PORFILE_WRITER);
75 }
76
77 int
78 cmd_xsave (struct lexer *lexer, struct dataset *ds)
79 {
80   return parse_output_trns (lexer, ds, SYSFILE_WRITER);
81 }
82
83 int
84 cmd_xexport (struct lexer *lexer, struct dataset *ds)
85 {
86   return parse_output_trns (lexer, ds, PORFILE_WRITER);
87 }
88 \f
89 struct output_trns
90   {
91     struct casewriter *writer;          /* Writer. */
92   };
93
94 static trns_proc_func output_trns_proc;
95 static trns_free_func output_trns_free;
96 static struct casewriter *parse_write_command (struct lexer *,
97                                                struct dataset *,
98                                                enum writer_type,
99                                                enum command_type,
100                                                bool *retain_unselected);
101
102 /* Parses and performs the SAVE or EXPORT procedure. */
103 static int
104 parse_output_proc (struct lexer *lexer, struct dataset *ds,
105                    enum writer_type writer_type)
106 {
107   bool retain_unselected;
108   struct casewriter *output;
109   bool ok;
110
111   output = parse_write_command (lexer, ds, writer_type, PROC_CMD,
112                                 &retain_unselected);
113   if (output == NULL)
114     return CMD_CASCADING_FAILURE;
115
116   casereader_transfer (proc_open_filtering (ds, !retain_unselected), output);
117   ok = casewriter_destroy (output);
118   ok = proc_commit (ds) && ok;
119
120   return ok ? CMD_SUCCESS : CMD_CASCADING_FAILURE;
121 }
122
123 /* Parses the XSAVE or XEXPORT transformation command. */
124 static int
125 parse_output_trns (struct lexer *lexer, struct dataset *ds, enum writer_type writer_type)
126 {
127   struct output_trns *t = xmalloc (sizeof *t);
128   t->writer = parse_write_command (lexer, ds, writer_type, XFORM_CMD, NULL);
129   if (t->writer == NULL)
130     {
131       free (t);
132       return CMD_CASCADING_FAILURE;
133     }
134
135   add_transformation (ds, output_trns_proc, output_trns_free, t);
136   return CMD_SUCCESS;
137 }
138
139 /* Parses SAVE or XSAVE or EXPORT or XEXPORT command.
140    WRITER_TYPE identifies the type of file to write,
141    and COMMAND_TYPE identifies the type of command.
142
143    On success, returns a writer.
144    For procedures only, sets *RETAIN_UNSELECTED to true if cases
145    that would otherwise be excluded by FILTER or USE should be
146    included.
147
148    On failure, returns a null pointer. */
149 static struct casewriter *
150 parse_write_command (struct lexer *lexer, struct dataset *ds,
151                      enum writer_type writer_type,
152                      enum command_type command_type,
153                      bool *retain_unselected)
154 {
155   /* Common data. */
156   struct file_handle *handle; /* Output file. */
157   struct dictionary *dict;    /* Dictionary for output file. */
158   struct casewriter *writer;  /* Writer. */
159   struct case_map *map;       /* Map from input data to data for writer. */
160
161   /* Common options. */
162   bool print_map;             /* Print map?  TODO. */
163   bool print_short_names;     /* Print long-to-short name map.  TODO. */
164   struct sfm_write_options sysfile_opts;
165   struct pfm_write_options porfile_opts;
166
167   assert (writer_type == SYSFILE_WRITER || writer_type == PORFILE_WRITER);
168   assert (command_type == XFORM_CMD || command_type == PROC_CMD);
169   assert ((retain_unselected != NULL) == (command_type == PROC_CMD));
170
171   if (command_type == PROC_CMD)
172     *retain_unselected = true;
173
174   handle = NULL;
175   dict = dict_clone (dataset_dict (ds));
176   writer = NULL;
177   map = NULL;
178   print_map = false;
179   print_short_names = false;
180   sysfile_opts = sfm_writer_default_options ();
181   porfile_opts = pfm_writer_default_options ();
182
183   case_map_prepare_dict (dict);
184   dict_delete_scratch_vars (dict);
185
186   lex_match (lexer, T_SLASH);
187   for (;;)
188     {
189       if (lex_match_id (lexer, "OUTFILE"))
190         {
191           if (handle != NULL)
192             {
193               lex_sbc_only_once ("OUTFILE");
194               goto error;
195             }
196
197           lex_match (lexer, T_EQUALS);
198
199           handle = fh_parse (lexer, FH_REF_FILE | FH_REF_SCRATCH);
200           if (handle == NULL)
201             goto error;
202         }
203       else if (lex_match_id (lexer, "NAMES"))
204         print_short_names = true;
205       else if (lex_match_id (lexer, "PERMISSIONS"))
206         {
207           bool cw;
208
209           lex_match (lexer, T_EQUALS);
210           if (lex_match_id (lexer, "READONLY"))
211             cw = false;
212           else if (lex_match_id (lexer, "WRITEABLE"))
213             cw = true;
214           else
215             {
216               lex_error (lexer, _("expecting %s or %s"),
217                          "READONLY", "WRITEABLE");
218               goto error;
219             }
220           sysfile_opts.create_writeable = porfile_opts.create_writeable = cw;
221         }
222       else if (command_type == PROC_CMD && lex_match_id (lexer, "UNSELECTED"))
223         {
224           lex_match (lexer, T_EQUALS);
225           if (lex_match_id (lexer, "RETAIN"))
226             *retain_unselected = true;
227           else if (lex_match_id (lexer, "DELETE"))
228             *retain_unselected = false;
229           else
230             {
231               lex_error (lexer, _("expecting %s or %s"), "RETAIN", "DELETE");
232               goto error;
233             }
234         }
235       else if (writer_type == SYSFILE_WRITER
236                && lex_match_id (lexer, "COMPRESSED"))
237         sysfile_opts.compress = true;
238       else if (writer_type == SYSFILE_WRITER
239                && lex_match_id (lexer, "UNCOMPRESSED"))
240         sysfile_opts.compress = false;
241       else if (writer_type == SYSFILE_WRITER
242                && lex_match_id (lexer, "VERSION"))
243         {
244           lex_match (lexer, T_EQUALS);
245           if (!lex_force_int (lexer))
246             goto error;
247           sysfile_opts.version = lex_integer (lexer);
248           lex_get (lexer);
249         }
250       else if (writer_type == PORFILE_WRITER && lex_match_id (lexer, "TYPE"))
251         {
252           lex_match (lexer, T_EQUALS);
253           if (lex_match_id (lexer, "COMMUNICATIONS"))
254             porfile_opts.type = PFM_COMM;
255           else if (lex_match_id (lexer, "TAPE"))
256             porfile_opts.type = PFM_TAPE;
257           else
258             {
259               lex_error (lexer, _("expecting %s or %s"), "COMM", "TAPE");
260               goto error;
261             }
262         }
263       else if (writer_type == PORFILE_WRITER && lex_match_id (lexer, "DIGITS"))
264         {
265           lex_match (lexer, T_EQUALS);
266           if (!lex_force_int (lexer))
267             goto error;
268           porfile_opts.digits = lex_integer (lexer);
269           lex_get (lexer);
270         }
271       else if (!parse_dict_trim (lexer, dict))
272         goto error;
273
274       if (!lex_match (lexer, T_SLASH))
275         break;
276     }
277   if (lex_end_of_command (lexer) != CMD_SUCCESS)
278     goto error;
279
280   if (handle == NULL)
281     {
282       lex_sbc_missing (lexer, "OUTFILE");
283       goto error;
284     }
285
286   dict_delete_scratch_vars (dict);
287   dict_compact_values (dict);
288
289   if (fh_get_referent (handle) == FH_REF_FILE)
290     {
291       switch (writer_type)
292         {
293         case SYSFILE_WRITER:
294           writer = sfm_open_writer (handle, dict, sysfile_opts);
295           break;
296         case PORFILE_WRITER:
297           writer = pfm_open_writer (handle, dict, porfile_opts);
298           break;
299         }
300     }
301   else
302     writer = any_writer_open (handle, dict);
303   if (writer == NULL)
304     goto error;
305
306   map = case_map_from_dict (dict);
307   if (map != NULL)
308     writer = case_map_create_output_translator (map, writer);
309   dict_destroy (dict);
310
311   fh_unref (handle);
312   return writer;
313
314  error:
315   fh_unref (handle);
316   casewriter_destroy (writer);
317   dict_destroy (dict);
318   case_map_destroy (map);
319   return NULL;
320 }
321
322 /* Writes case *C to the system file specified on XSAVE or XEXPORT. */
323 static int
324 output_trns_proc (void *trns_, struct ccase **c, casenumber case_num UNUSED)
325 {
326   struct output_trns *t = trns_;
327   casewriter_write (t->writer, case_ref (*c));
328   return TRNS_CONTINUE;
329 }
330
331 /* Frees an XSAVE or XEXPORT transformation.
332    Returns true if successful, false if an I/O error occurred. */
333 static bool
334 output_trns_free (void *trns_)
335 {
336   struct output_trns *t = trns_;
337   bool ok = casewriter_destroy (t->writer);
338   free (t);
339   return ok;
340 }