por-file-writer: Use xasprintf() to avoid careful buffer sizing.
[pspp] / src / data / value.h
1 /* PSPP - a program for statistical analysis.
2    Copyright (C) 1997-9, 2000, 2007, 2009, 2010, 2012 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 #ifndef DATA_VALUE_H
18 #define DATA_VALUE_H 1
19
20 #include <assert.h>
21 #include <stdbool.h>
22 #include <stdlib.h>
23 #include <stdint.h>
24 #include <string.h>
25 #include "xalloc.h"
26 \f
27 /* Maximum length of a "short" string, that is represented in
28    "union value" without a separate pointer.
29
30    This is an implementation detail of the "union value" code.
31    There is little reason for client code to use it. */
32 #define MAX_SHORT_STRING 8
33
34 /* A numeric or string value.
35
36    The client is responsible for keeping track of the value's
37    width.
38
39    This structure is semi-opaque:
40
41        - If the value is a number, clients may access the 'f'
42          member directly.
43
44        - Clients should not access other members directly.
45 */
46 union value
47   {
48     double f;
49     uint8_t short_string[MAX_SHORT_STRING];
50     uint8_t *long_string;
51   };
52
53 static inline void value_init (union value *, int width);
54 static inline void value_clone (union value *, const union value *, int width);
55 static inline bool value_needs_init (int width);
56 static inline bool value_try_init (union value *, int width);
57 static inline void value_destroy (union value *, int width);
58
59 static inline double value_num (const union value *);
60 static inline const uint8_t *value_str (const union value *, int width);
61 static inline uint8_t *value_str_rw (union value *, int width);
62
63 static inline void value_copy (union value *, const union value *, int width);
64 void value_copy_rpad (union value *, int dst_width,
65                       const union value *, int src_width,
66                       char pad);
67 void value_copy_str_rpad (union value *, int dst_width, const uint8_t *,
68                           char pad);
69 void value_copy_buf_rpad (union value *dst, int dst_width,
70                           const uint8_t *src, size_t src_len, char pad);
71 void value_set_missing (union value *, int width);
72 int value_compare_3way (const union value *, const union value *, int width);
73 bool value_equal (const union value *, const union value *, int width);
74 unsigned int value_hash (const union value *, int width, unsigned int basis);
75
76 bool value_is_resizable (const union value *, int old_width, int new_width);
77 bool value_needs_resize (int old_width, int new_width);
78 void value_resize (union value *, int old_width, int new_width);
79
80 bool value_is_spaces (const union value *, int width);
81
82 static inline void value_swap (union value *, union value *);
83
84 struct pool;
85 void value_init_pool (struct pool *, union value *, int width);
86 void value_clone_pool (struct pool *, union value *, const union value *,
87                        int width);
88 void value_resize_pool (struct pool *, union value *,
89                         int old_width, int new_width);
90 \f
91 /* Initializes V as a value of the given WIDTH, where 0
92    represents a numeric value and a positive integer represents a
93    string value WIDTH bytes long.
94
95    A WIDTH of -1 is ignored.
96
97    The contents of value V are indeterminate after
98    initialization. */
99 static inline void
100 value_init (union value *v, int width)
101 {
102   if (width > MAX_SHORT_STRING)
103     v->long_string = xmalloc (width);
104 }
105
106 /* Initializes V as a value of the given WIDTH, as with value_init(), and
107    copies SRC's value into V as its initial value. */
108 static inline void
109 value_clone (union value *v, const union value *src, int width)
110 {
111   if (width <= MAX_SHORT_STRING)
112     *v = *src;
113   else
114     v->long_string = xmemdup (src->long_string, width);
115 }
116
117 /* Returns true if a value of the given WIDTH actually needs to
118    have the value_init and value_destroy functions called, false
119    if those functions are no-ops for values of the given WIDTH.
120
121    Using this function is only a valuable optimization if a large
122    number of values of the given WIDTH are to be initialized*/
123 static inline bool
124 value_needs_init (int width)
125 {
126   return width > MAX_SHORT_STRING;
127 }
128
129 /* Same as value_init, except that failure to allocate memory
130    causes it to return false instead of terminating the
131    program.  On success, returns true. */
132 static inline bool
133 value_try_init (union value *v, int width)
134 {
135   if (width > MAX_SHORT_STRING)
136     {
137       v->long_string = malloc (width);
138       return v->long_string != NULL;
139     }
140   else
141     return true;
142 }
143
144 /* Frees any memory allocated by value_init for V, which must
145    have the given WIDTH. */
146 static inline void
147 value_destroy (union value *v, int width)
148 {
149   if (width > MAX_SHORT_STRING)
150     free (v->long_string);
151 }
152
153 /* Returns the numeric value in V, which must have width 0. */
154 static inline double
155 value_num (const union value *v)
156 {
157   return v->f;
158 }
159
160 /* Returns the string value in V, which must have width WIDTH.
161
162    The returned value is not null-terminated.
163
164    It is important that WIDTH be the actual value that was passed
165    to value_init.  Passing, e.g., a smaller value because only
166    that number of bytes will be accessed will not always work. */
167 static inline const uint8_t *
168 value_str (const union value *v, int width)
169 {
170   assert (width > 0);
171   return (width > MAX_SHORT_STRING ? v->long_string : v->short_string);
172 }
173
174 /* Returns the string value in V, which must have width WIDTH.
175
176    The returned value is not null-terminated.
177
178    It is important that WIDTH be the actual value that was passed
179    to value_init.  Passing, e.g., a smaller value because only
180    that number of bytes will be accessed will not always work. */
181 static inline uint8_t *
182 value_str_rw (union value *v, int width)
183 {
184   assert (width > 0);
185   return (width > MAX_SHORT_STRING ? v->long_string : v->short_string);
186 }
187
188 /* Copies SRC to DST, given that they both contain data of the
189    given WIDTH. */
190 static inline void
191 value_copy (union value *dst, const union value *src, int width)
192 {
193   if (width <= MAX_SHORT_STRING)
194     *dst = *src;
195   else if (dst != src)
196     memcpy (dst->long_string, src->long_string, width);
197 }
198
199 /* Exchanges the contents of A and B. */
200 static inline void
201 value_swap (union value *a, union value *b)
202 {
203   union value tmp = *a;
204   *a = *b;
205   *b = tmp;
206 }
207
208 #endif /* data/value.h */