Fix assertion for proper Huffman merge pattern: 0 == 1 modulo 1.
[pspp] / src / formats.c
1 /* PSPP - computes sample statistics.
2    Copyright (C) 1997-9, 2000 Free Software Foundation, Inc.
3    Written by Ben Pfaff <blp@gnu.org>.
4
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.
9
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.
14
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., 59 Temple Place - Suite 330, Boston, MA
18    02111-1307, USA. */
19
20 #include <config.h>
21 #include <limits.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include "command.h"
25 #include "error.h"
26 #include "lexer.h"
27 #include "misc.h"
28 #include "str.h"
29 #include "var.h"
30
31 #include "debug-print.h"
32
33 enum
34   {
35     FORMATS_PRINT = 001,
36     FORMATS_WRITE = 002
37   };
38
39 static int internal_cmd_formats (int);
40
41 int
42 cmd_print_formats (void)
43 {
44   return internal_cmd_formats (FORMATS_PRINT);
45 }
46
47 int
48 cmd_write_formats (void)
49 {
50   return internal_cmd_formats (FORMATS_WRITE);
51 }
52
53 int
54 cmd_formats (void)
55 {
56   return internal_cmd_formats (FORMATS_PRINT | FORMATS_WRITE);
57 }
58
59 int
60 internal_cmd_formats (int which)
61 {
62   /* Variables. */
63   struct variable **v;
64   int cv;
65
66   /* Format to set the variables to. */
67   struct fmt_spec f;
68
69   /* Numeric or string. */
70   int type;
71
72   /* Counter. */
73   int i;
74
75   for (;;)
76     {
77       if (token == '.')
78         break;
79
80       if (!parse_variables (default_dict, &v, &cv, PV_SAME_TYPE))
81         return CMD_PART_SUCCESS_MAYBE;
82       type = v[0]->type;
83
84       if (!lex_match ('('))
85         {
86           msg (SE, _("`(' expected after variable list"));
87           goto fail;
88         }
89       if (!parse_format_specifier (&f, 0) || !check_output_specifier (&f))
90         goto fail;
91
92       /* Catch type mismatch errors. */
93       if ((type == ALPHA) ^ (0 != (formats[f.type].cat & FCAT_STRING)))
94         {
95           msg (SE, _("Format %s may not be assigned to a %s variable."),
96                fmt_to_string (&f), type == NUMERIC ? _("numeric") : _("string"));
97           goto fail;
98         }
99
100       /* This is an additional check for string variables.  We can't
101          let the user specify an A8 format for a string variable with
102          width 4. */
103       if (type == ALPHA)
104         {
105           /* Shortest string so far. */
106           int min_len = INT_MAX;
107
108           for (i = 0; i < cv; i++)
109             min_len = min (min_len, v[i]->width);
110           if (!check_string_specifier (&f, min_len))
111             goto fail;
112         }
113
114       if (!lex_match (')'))
115         {
116           msg (SE, _("`)' expected after output format."));
117           goto fail;
118         }
119
120       for (i = 0; i < cv; i++)
121         {
122           if (which & FORMATS_PRINT)
123             v[i]->print = f;
124           if (which & FORMATS_WRITE)
125             v[i]->write = f;
126         }
127       free (v);
128       v = NULL;
129     }
130   return CMD_SUCCESS;
131
132 fail:
133   free (v);
134   return CMD_PART_SUCCESS_MAYBE;
135 }