pintos-os.org Git - pspp/blob - 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           const struct missing_values *miss = var_get_missing_values (s);
 100           if (mv_is_resizable (miss, var_get_width (t)))
 101             var_set_missing_values (t, miss);
 102         }
 103
 104       if (var_is_numeric (s))
 105         {
 106           var_set_print_format (t, var_get_print_format (s));
 107           var_set_write_format (t, var_get_write_format (s));
 108         }
 109
 110       if (var_has_attributes (s))
 111         var_set_attributes (t, var_get_attributes (s));
 112     }
 113
 114   if (!n_matched)
 115     msg (SW, _("No matching variables found between the source "
 116                "and target files."));
 117
 118   /* Data file attributes. */
 119   if (dict_has_attributes (dict))
 120     dict_set_attributes (dataset_dict (ds), dict_get_attributes (dict));
 121
 122   /* Weighting. */
 123   if (dict_get_weight (dict) != NULL)
 124     {
 125       struct variable *new_weight
 126         = dict_lookup_var (dataset_dict (ds),
 127                            var_get_name (dict_get_weight (dict)));
 128
 129       if (new_weight != NULL)
 130         dict_set_weight (dataset_dict (ds), new_weight);
 131     }
 132
 133   return lex_end_of_command (lexer);
 134 }