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