c2de9318ae81b13462c940aae839a580f886c2d4
[pspp-builds.git] / src / language / dictionary / apply-dictionary.c
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-9, 2000, 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-reader.h"
22 #include "data/casereader.h"
23 #include "data/dataset.h"
24 #include "data/dictionary.h"
25 #include "data/file-handle-def.h"
26 #include "data/missing-values.h"
27 #include "data/value-labels.h"
28 #include "data/variable.h"
29 #include "language/command.h"
30 #include "language/data-io/file-handle.h"
31 #include "language/lexer/lexer.h"
32 #include "libpspp/message.h"
33 #include "libpspp/str.h"
34
35 #include "gettext.h"
36 #define _(msgid) gettext (msgid)
37
38 /* Parses and executes APPLY DICTIONARY. */
39 int
40 cmd_apply_dictionary (struct lexer *lexer, struct dataset *ds)
41 {
42   struct file_handle *handle;
43   struct casereader *reader;
44   struct dictionary *dict;
45
46   int n_matched = 0;
47
48   int i;
49
50   lex_match_id (lexer, "FROM");
51   lex_match (lexer, T_EQUALS);
52
53   handle = fh_parse (lexer, FH_REF_FILE, dataset_session (ds));
54   if (!handle)
55     return CMD_FAILURE;
56   reader = any_reader_open (handle, &dict);
57   fh_unref (handle);
58   if (dict == NULL)
59     return CMD_FAILURE;
60
61   casereader_destroy (reader);
62
63   for (i = 0; i < dict_get_var_cnt (dict); i++)
64     {
65       struct variable *s = dict_get_var (dict, i);
66       struct variable *t = dict_lookup_var (dataset_dict (ds),
67                                             var_get_name (s));
68       if (t == NULL)
69         continue;
70
71       n_matched++;
72       if (var_get_type (s) != var_get_type (t))
73         {
74           msg (SW, _("Variable %s is %s in target file, but %s in "
75                      "source file."),
76                var_get_name (s),
77                var_is_alpha (t) ? _("string") : _("numeric"),
78                var_is_alpha (s) ? _("string") : _("numeric"));
79           continue;
80         }
81
82       if (var_has_label (s))
83         var_set_label (t, var_get_label (s), false);
84
85       if (var_has_value_labels (s))
86         {
87           const struct val_labs *value_labels = var_get_value_labels (s);
88           if (val_labs_can_set_width (value_labels, var_get_width (t)))
89             var_set_value_labels (s, value_labels);
90         }
91
92       if (var_has_missing_values (s))
93         {
94           const struct missing_values *miss = var_get_missing_values (s);
95           if (mv_is_resizable (miss, var_get_width (t)))
96             var_set_missing_values (t, miss);
97         }
98
99       if (var_is_numeric (s))
100         {
101           var_set_print_format (t, var_get_print_format (s));
102           var_set_write_format (t, var_get_write_format (s));
103         }
104
105       if (var_has_attributes (s)) 
106         var_set_attributes (t, var_get_attributes (s));
107     }
108
109   if (!n_matched)
110     msg (SW, _("No matching variables found between the source "
111                "and target files."));
112
113   /* Data file attributes. */
114   if (dict_has_attributes (dict))
115     dict_set_attributes (dataset_dict (ds), dict_get_attributes (dict));
116
117   /* Weighting. */
118   if (dict_get_weight (dict) != NULL)
119     {
120       struct variable *new_weight
121         = dict_lookup_var (dataset_dict (ds),
122                            var_get_name (dict_get_weight (dict)));
123
124       if (new_weight != NULL)
125         dict_set_weight (dataset_dict (ds), new_weight);
126     }
127
128   return CMD_SUCCESS;
129 }