Merge branch 'maint'
[isl.git] / isl_printer.c
blobe0821f2a2b4ef5d57ee6927d9e436b9f6b587a72
1 #include <string.h>
2 #include <isl_printer_private.h>
4 static __isl_give isl_printer *file_start_line(__isl_take isl_printer *p)
6 fprintf(p->file, "%*s%s", p->indent, "", p->prefix ? p->prefix : "");
7 return p;
10 static __isl_give isl_printer *file_end_line(__isl_take isl_printer *p)
12 fprintf(p->file, "%s\n", p->suffix ? p->suffix : "");
13 return p;
16 static __isl_give isl_printer *file_flush(__isl_take isl_printer *p)
18 fflush(p->file);
19 return p;
22 static __isl_give isl_printer *file_print_str(__isl_take isl_printer *p,
23 const char *s)
25 fprintf(p->file, "%s", s);
26 return p;
29 static __isl_give isl_printer *file_print_double(__isl_take isl_printer *p,
30 double d)
32 fprintf(p->file, "%g", d);
33 return p;
36 static __isl_give isl_printer *file_print_int(__isl_take isl_printer *p, int i)
38 fprintf(p->file, "%d", i);
39 return p;
42 static __isl_give isl_printer *file_print_isl_int(__isl_take isl_printer *p, isl_int i)
44 isl_int_print(p->file, i, p->width);
45 return p;
48 static int grow_buf(__isl_keep isl_printer *p, int extra)
50 int new_size;
51 char *new_buf;
53 if (p->buf_size == 0)
54 return -1;
56 new_size = ((p->buf_n + extra + 1) * 3) / 2;
57 new_buf = isl_realloc_array(p->ctx, p->buf, char, new_size);
58 if (!new_buf) {
59 p->buf_size = 0;
60 return -1;
62 p->buf = new_buf;
63 p->buf_size = new_size;
65 return 0;
68 static __isl_give isl_printer *str_print(__isl_take isl_printer *p,
69 const char *s, int len)
71 if (p->buf_n + len + 1 >= p->buf_size && grow_buf(p, len))
72 goto error;
73 memcpy(p->buf + p->buf_n, s, len);
74 p->buf_n += len;
76 p->buf[p->buf_n] = '\0';
77 return p;
78 error:
79 isl_printer_free(p);
80 return NULL;
83 static __isl_give isl_printer *str_print_indent(__isl_take isl_printer *p,
84 int indent)
86 int i;
88 if (p->buf_n + indent + 1 >= p->buf_size && grow_buf(p, indent))
89 goto error;
90 for (i = 0; i < indent; ++i)
91 p->buf[p->buf_n++] = ' ';
92 return p;
93 error:
94 isl_printer_free(p);
95 return NULL;
98 static __isl_give isl_printer *str_start_line(__isl_take isl_printer *p)
100 p = str_print_indent(p, p->indent);
101 if (p->prefix)
102 p = str_print(p, p->prefix, strlen(p->prefix));
103 return p;
106 static __isl_give isl_printer *str_end_line(__isl_take isl_printer *p)
108 if (p->suffix)
109 p = str_print(p, p->suffix, strlen(p->suffix));
110 p = str_print(p, "\n", strlen("\n"));
111 return p;
114 static __isl_give isl_printer *str_flush(__isl_take isl_printer *p)
116 p->buf_n = 0;
117 return p;
120 static __isl_give isl_printer *str_print_str(__isl_take isl_printer *p,
121 const char *s)
123 return str_print(p, s, strlen(s));
126 static __isl_give isl_printer *str_print_double(__isl_take isl_printer *p,
127 double d)
129 int left = p->buf_size - p->buf_n;
130 int need = snprintf(p->buf + p->buf_n, left, "%g", d);
131 if (need >= left) {
132 if (grow_buf(p, need))
133 goto error;
134 left = p->buf_size - p->buf_n;
135 need = snprintf(p->buf + p->buf_n, left, "%g", d);
137 p->buf_n += need;
138 return p;
139 error:
140 isl_printer_free(p);
141 return NULL;
144 static __isl_give isl_printer *str_print_int(__isl_take isl_printer *p, int i)
146 int left = p->buf_size - p->buf_n;
147 int need = snprintf(p->buf + p->buf_n, left, "%d", i);
148 if (need >= left) {
149 if (grow_buf(p, need))
150 goto error;
151 left = p->buf_size - p->buf_n;
152 need = snprintf(p->buf + p->buf_n, left, "%d", i);
154 p->buf_n += need;
155 return p;
156 error:
157 isl_printer_free(p);
158 return NULL;
161 static __isl_give isl_printer *str_print_isl_int(__isl_take isl_printer *p,
162 isl_int i)
164 char *s;
165 int len;
167 s = isl_int_get_str(i);
168 len = strlen(s);
169 if (len < p->width)
170 p = str_print_indent(p, p->width - len);
171 p = str_print(p, s, len);
172 isl_int_free_str(s);
173 return p;
176 struct isl_printer_ops {
177 __isl_give isl_printer *(*start_line)(__isl_take isl_printer *p);
178 __isl_give isl_printer *(*end_line)(__isl_take isl_printer *p);
179 __isl_give isl_printer *(*print_double)(__isl_take isl_printer *p,
180 double d);
181 __isl_give isl_printer *(*print_int)(__isl_take isl_printer *p, int i);
182 __isl_give isl_printer *(*print_isl_int)(__isl_take isl_printer *p,
183 isl_int i);
184 __isl_give isl_printer *(*print_str)(__isl_take isl_printer *p,
185 const char *s);
186 __isl_give isl_printer *(*flush)(__isl_take isl_printer *p);
189 static struct isl_printer_ops file_ops = {
190 file_start_line,
191 file_end_line,
192 file_print_double,
193 file_print_int,
194 file_print_isl_int,
195 file_print_str,
196 file_flush
199 static struct isl_printer_ops str_ops = {
200 str_start_line,
201 str_end_line,
202 str_print_double,
203 str_print_int,
204 str_print_isl_int,
205 str_print_str,
206 str_flush
209 __isl_give isl_printer *isl_printer_to_file(isl_ctx *ctx, FILE *file)
211 struct isl_printer *p = isl_alloc_type(ctx, struct isl_printer);
212 if (!p)
213 return NULL;
214 p->ctx = ctx;
215 isl_ctx_ref(p->ctx);
216 p->ops = &file_ops;
217 p->file = file;
218 p->buf = NULL;
219 p->buf_n = 0;
220 p->buf_size = 0;
221 p->indent = 0;
222 p->output_format = ISL_FORMAT_ISL;
223 p->prefix = NULL;
224 p->suffix = NULL;
225 p->width = 0;
227 return p;
230 __isl_give isl_printer *isl_printer_to_str(isl_ctx *ctx)
232 struct isl_printer *p = isl_alloc_type(ctx, struct isl_printer);
233 if (!p)
234 return NULL;
235 p->ctx = ctx;
236 isl_ctx_ref(p->ctx);
237 p->ops = &str_ops;
238 p->file = NULL;
239 p->buf = isl_alloc_array(ctx, char, 256);
240 if (!p->buf)
241 goto error;
242 p->buf_n = 0;
243 p->buf[0] = '\0';
244 p->buf_size = 256;
245 p->indent = 0;
246 p->output_format = ISL_FORMAT_ISL;
247 p->prefix = NULL;
248 p->suffix = NULL;
249 p->width = 0;
251 return p;
252 error:
253 isl_printer_free(p);
254 return NULL;
257 void *isl_printer_free(__isl_take isl_printer *p)
259 if (!p)
260 return NULL;
261 free(p->buf);
262 isl_ctx_deref(p->ctx);
263 free(p);
265 return NULL;
268 isl_ctx *isl_printer_get_ctx(__isl_keep isl_printer *printer)
270 return printer ? printer->ctx : NULL;
273 FILE *isl_printer_get_file(__isl_keep isl_printer *printer)
275 if (!printer)
276 return NULL;
277 if (!printer->file)
278 isl_die(isl_printer_get_ctx(printer), isl_error_invalid,
279 "not a file printer", return NULL);
280 return printer->file;
283 __isl_give isl_printer *isl_printer_set_isl_int_width(__isl_take isl_printer *p,
284 int width)
286 if (!p)
287 return NULL;
289 p->width = width;
291 return p;
294 __isl_give isl_printer *isl_printer_set_indent(__isl_take isl_printer *p,
295 int indent)
297 if (!p)
298 return NULL;
300 p->indent = indent;
302 return p;
305 __isl_give isl_printer *isl_printer_indent(__isl_take isl_printer *p,
306 int indent)
308 if (!p)
309 return NULL;
311 p->indent += indent;
312 if (p->indent < 0)
313 p->indent = 0;
315 return p;
318 __isl_give isl_printer *isl_printer_set_prefix(__isl_take isl_printer *p,
319 const char *prefix)
321 if (!p)
322 return NULL;
324 p->prefix = prefix;
326 return p;
329 __isl_give isl_printer *isl_printer_set_suffix(__isl_take isl_printer *p,
330 const char *suffix)
332 if (!p)
333 return NULL;
335 p->suffix = suffix;
337 return p;
340 __isl_give isl_printer *isl_printer_set_output_format(__isl_take isl_printer *p,
341 int output_format)
343 if (!p)
344 return NULL;
346 p->output_format = output_format;
348 return p;
351 int isl_printer_get_output_format(__isl_keep isl_printer *p)
353 if (!p)
354 return -1;
355 return p->output_format;
358 __isl_give isl_printer *isl_printer_print_str(__isl_take isl_printer *p,
359 const char *s)
361 if (!p)
362 return NULL;
363 if (!s)
364 return isl_printer_free(p);
366 return p->ops->print_str(p, s);
369 __isl_give isl_printer *isl_printer_print_double(__isl_take isl_printer *p,
370 double d)
372 if (!p)
373 return NULL;
375 return p->ops->print_double(p, d);
378 __isl_give isl_printer *isl_printer_print_int(__isl_take isl_printer *p, int i)
380 if (!p)
381 return NULL;
383 return p->ops->print_int(p, i);
386 __isl_give isl_printer *isl_printer_print_isl_int(__isl_take isl_printer *p,
387 isl_int i)
389 if (!p)
390 return NULL;
392 return p->ops->print_isl_int(p, i);
395 __isl_give isl_printer *isl_printer_start_line(__isl_take isl_printer *p)
397 if (!p)
398 return NULL;
400 return p->ops->start_line(p);
403 __isl_give isl_printer *isl_printer_end_line(__isl_take isl_printer *p)
405 if (!p)
406 return NULL;
408 return p->ops->end_line(p);
411 char *isl_printer_get_str(__isl_keep isl_printer *printer)
413 if (!printer || !printer->buf)
414 return NULL;
415 return strdup(printer->buf);
418 __isl_give isl_printer *isl_printer_flush(__isl_take isl_printer *p)
420 if (!p)
421 return NULL;
423 return p->ops->flush(p);