isl_printer.c

   1 #include <string.h>
   2 #include <isl_int.h>
   3 #include <isl_printer_private.h>
   4
   5 static __isl_give isl_printer *file_start_line(__isl_take isl_printer *p)
   6 {
   7         fprintf(p->file, "%s%*s%s", p->indent_prefix ? p->indent_prefix : "",
   8                                     p->indent, "", p->prefix ? p->prefix : "");
   9         return p;
  10 }
  11
  12 static __isl_give isl_printer *file_end_line(__isl_take isl_printer *p)
  13 {
  14         fprintf(p->file, "%s\n", p->suffix ? p->suffix : "");
  15         return p;
  16 }
  17
  18 static __isl_give isl_printer *file_flush(__isl_take isl_printer *p)
  19 {
  20         fflush(p->file);
  21         return p;
  22 }
  23
  24 static __isl_give isl_printer *file_print_str(__isl_take isl_printer *p,
  25         const char *s)
  26 {
  27         fprintf(p->file, "%s", s);
  28         return p;
  29 }
  30
  31 static __isl_give isl_printer *file_print_double(__isl_take isl_printer *p,
  32         double d)
  33 {
  34         fprintf(p->file, "%g", d);
  35         return p;
  36 }
  37
  38 static __isl_give isl_printer *file_print_int(__isl_take isl_printer *p, int i)
  39 {
  40         fprintf(p->file, "%d", i);
  41         return p;
  42 }
  43
  44 static __isl_give isl_printer *file_print_isl_int(__isl_take isl_printer *p, isl_int i)
  45 {
  46         isl_int_print(p->file, i, p->width);
  47         return p;
  48 }
  49
  50 static int grow_buf(__isl_keep isl_printer *p, int extra)
  51 {
  52         int new_size;
  53         char *new_buf;
  54
  55         if (p->buf_size == 0)
  56                 return -1;
  57
  58         new_size = ((p->buf_n + extra + 1) * 3) / 2;
  59         new_buf = isl_realloc_array(p->ctx, p->buf, char, new_size);
  60         if (!new_buf) {
  61                 p->buf_size = 0;
  62                 return -1;
  63         }
  64         p->buf = new_buf;
  65         p->buf_size = new_size;
  66
  67         return 0;
  68 }
  69
  70 static __isl_give isl_printer *str_print(__isl_take isl_printer *p,
  71         const char *s, int len)
  72 {
  73         if (p->buf_n + len + 1 >= p->buf_size && grow_buf(p, len))
  74                 goto error;
  75         memcpy(p->buf + p->buf_n, s, len);
  76         p->buf_n += len;
  77
  78         p->buf[p->buf_n] = '\0';
  79         return p;
  80 error:
  81         isl_printer_free(p);
  82         return NULL;
  83 }
  84
  85 static __isl_give isl_printer *str_print_indent(__isl_take isl_printer *p,
  86         int indent)
  87 {
  88         int i;
  89
  90         if (p->buf_n + indent + 1 >= p->buf_size && grow_buf(p, indent))
  91                 goto error;
  92         for (i = 0; i < indent; ++i)
  93                 p->buf[p->buf_n++] = ' ';
  94         return p;
  95 error:
  96         isl_printer_free(p);
  97         return NULL;
  98 }
  99
 100 static __isl_give isl_printer *str_start_line(__isl_take isl_printer *p)
 101 {
 102         if (p->indent_prefix)
 103                 p = str_print(p, p->indent_prefix, strlen(p->indent_prefix));
 104         p = str_print_indent(p, p->indent);
 105         if (p->prefix)
 106                 p = str_print(p, p->prefix, strlen(p->prefix));
 107         return p;
 108 }
 109
 110 static __isl_give isl_printer *str_end_line(__isl_take isl_printer *p)
 111 {
 112         if (p->suffix)
 113                 p = str_print(p, p->suffix, strlen(p->suffix));
 114         p = str_print(p, "\n", strlen("\n"));
 115         return p;
 116 }
 117
 118 static __isl_give isl_printer *str_flush(__isl_take isl_printer *p)
 119 {
 120         p->buf_n = 0;
 121         p->buf[p->buf_n] = '\0';
 122         return p;
 123 }
 124
 125 static __isl_give isl_printer *str_print_str(__isl_take isl_printer *p,
 126         const char *s)
 127 {
 128         return str_print(p, s, strlen(s));
 129 }
 130
 131 static __isl_give isl_printer *str_print_double(__isl_take isl_printer *p,
 132         double d)
 133 {
 134         int left = p->buf_size - p->buf_n;
 135         int need = snprintf(p->buf + p->buf_n, left, "%g", d);
 136         if (need >= left) {
 137                 if (grow_buf(p, need))
 138                         goto error;
 139                 left = p->buf_size - p->buf_n;
 140                 need = snprintf(p->buf + p->buf_n, left, "%g", d);
 141         }
 142         p->buf_n += need;
 143         return p;
 144 error:
 145         isl_printer_free(p);
 146         return NULL;
 147 }
 148
 149 static __isl_give isl_printer *str_print_int(__isl_take isl_printer *p, int i)
 150 {
 151         int left = p->buf_size - p->buf_n;
 152         int need = snprintf(p->buf + p->buf_n, left, "%d", i);
 153         if (need >= left) {
 154                 if (grow_buf(p, need))
 155                         goto error;
 156                 left = p->buf_size - p->buf_n;
 157                 need = snprintf(p->buf + p->buf_n, left, "%d", i);
 158         }
 159         p->buf_n += need;
 160         return p;
 161 error:
 162         isl_printer_free(p);
 163         return NULL;
 164 }
 165
 166 static __isl_give isl_printer *str_print_isl_int(__isl_take isl_printer *p,
 167         isl_int i)
 168 {
 169         char *s;
 170         int len;
 171
 172         s = isl_int_get_str(i);
 173         len = strlen(s);
 174         if (len < p->width)
 175                 p = str_print_indent(p, p->width - len);
 176         p = str_print(p, s, len);
 177         isl_int_free_str(s);
 178         return p;
 179 }
 180
 181 struct isl_printer_ops {
 182         __isl_give isl_printer *(*start_line)(__isl_take isl_printer *p);
 183         __isl_give isl_printer *(*end_line)(__isl_take isl_printer *p);
 184         __isl_give isl_printer *(*print_double)(__isl_take isl_printer *p,
 185                 double d);
 186         __isl_give isl_printer *(*print_int)(__isl_take isl_printer *p, int i);
 187         __isl_give isl_printer *(*print_isl_int)(__isl_take isl_printer *p,
 188                                                 isl_int i);
 189         __isl_give isl_printer *(*print_str)(__isl_take isl_printer *p,
 190                                                 const char *s);
 191         __isl_give isl_printer *(*flush)(__isl_take isl_printer *p);
 192 };
 193
 194 static struct isl_printer_ops file_ops = {
 195         file_start_line,
 196         file_end_line,
 197         file_print_double,
 198         file_print_int,
 199         file_print_isl_int,
 200         file_print_str,
 201         file_flush
 202 };
 203
 204 static struct isl_printer_ops str_ops = {
 205         str_start_line,
 206         str_end_line,
 207         str_print_double,
 208         str_print_int,
 209         str_print_isl_int,
 210         str_print_str,
 211         str_flush
 212 };
 213
 214 __isl_give isl_printer *isl_printer_to_file(isl_ctx *ctx, FILE *file)
 215 {
 216         struct isl_printer *p = isl_calloc_type(ctx, struct isl_printer);
 217         if (!p)
 218                 return NULL;
 219         p->ctx = ctx;
 220         isl_ctx_ref(p->ctx);
 221         p->ops = &file_ops;
 222         p->file = file;
 223         p->buf = NULL;
 224         p->buf_n = 0;
 225         p->buf_size = 0;
 226         p->indent = 0;
 227         p->output_format = ISL_FORMAT_ISL;
 228         p->indent_prefix = NULL;
 229         p->prefix = NULL;
 230         p->suffix = NULL;
 231         p->width = 0;
 232         p->yaml_style = ISL_YAML_STYLE_FLOW;
 233
 234         return p;
 235 }
 236
 237 __isl_give isl_printer *isl_printer_to_str(isl_ctx *ctx)
 238 {
 239         struct isl_printer *p = isl_calloc_type(ctx, struct isl_printer);
 240         if (!p)
 241                 return NULL;
 242         p->ctx = ctx;
 243         isl_ctx_ref(p->ctx);
 244         p->ops = &str_ops;
 245         p->file = NULL;
 246         p->buf = isl_alloc_array(ctx, char, 256);
 247         if (!p->buf)
 248                 goto error;
 249         p->buf_n = 0;
 250         p->buf[0] = '\0';
 251         p->buf_size = 256;
 252         p->indent = 0;
 253         p->output_format = ISL_FORMAT_ISL;
 254         p->indent_prefix = NULL;
 255         p->prefix = NULL;
 256         p->suffix = NULL;
 257         p->width = 0;
 258         p->yaml_style = ISL_YAML_STYLE_FLOW;
 259
 260         return p;
 261 error:
 262         isl_printer_free(p);
 263         return NULL;
 264 }
 265
 266 __isl_null isl_printer *isl_printer_free(__isl_take isl_printer *p)
 267 {
 268         if (!p)
 269                 return NULL;
 270         free(p->buf);
 271         free(p->indent_prefix);
 272         free(p->prefix);
 273         free(p->suffix);
 274         free(p->yaml_state);
 275         isl_id_to_id_free(p->notes);
 276         isl_ctx_deref(p->ctx);
 277         free(p);
 278
 279         return NULL;
 280 }
 281
 282 isl_ctx *isl_printer_get_ctx(__isl_keep isl_printer *printer)
 283 {
 284         return printer ? printer->ctx : NULL;
 285 }
 286
 287 FILE *isl_printer_get_file(__isl_keep isl_printer *printer)
 288 {
 289         if (!printer)
 290                 return NULL;
 291         if (!printer->file)
 292                 isl_die(isl_printer_get_ctx(printer), isl_error_invalid,
 293                         "not a file printer", return NULL);
 294         return printer->file;
 295 }
 296
 297 __isl_give isl_printer *isl_printer_set_isl_int_width(__isl_take isl_printer *p,
 298         int width)
 299 {
 300         if (!p)
 301                 return NULL;
 302
 303         p->width = width;
 304
 305         return p;
 306 }
 307
 308 __isl_give isl_printer *isl_printer_set_indent(__isl_take isl_printer *p,
 309         int indent)
 310 {
 311         if (!p)
 312                 return NULL;
 313
 314         p->indent = indent;
 315
 316         return p;
 317 }
 318
 319 __isl_give isl_printer *isl_printer_indent(__isl_take isl_printer *p,
 320         int indent)
 321 {
 322         if (!p)
 323                 return NULL;
 324
 325         p->indent += indent;
 326         if (p->indent < 0)
 327                 p->indent = 0;
 328
 329         return p;
 330 }
 331
 332 /* Replace the indent prefix of "p" by "prefix".
 333  */
 334 __isl_give isl_printer *isl_printer_set_indent_prefix(__isl_take isl_printer *p,
 335         const char *prefix)
 336 {
 337         if (!p)
 338                 return NULL;
 339
 340         free(p->indent_prefix);
 341         p->indent_prefix = prefix ? strdup(prefix) : NULL;
 342
 343         return p;
 344 }
 345
 346 __isl_give isl_printer *isl_printer_set_prefix(__isl_take isl_printer *p,
 347         const char *prefix)
 348 {
 349         if (!p)
 350                 return NULL;
 351
 352         free(p->prefix);
 353         p->prefix = prefix ? strdup(prefix) : NULL;
 354
 355         return p;
 356 }
 357
 358 __isl_give isl_printer *isl_printer_set_suffix(__isl_take isl_printer *p,
 359         const char *suffix)
 360 {
 361         if (!p)
 362                 return NULL;
 363
 364         free(p->suffix);
 365         p->suffix = suffix ? strdup(suffix) : NULL;
 366
 367         return p;
 368 }
 369
 370 __isl_give isl_printer *isl_printer_set_output_format(__isl_take isl_printer *p,
 371         int output_format)
 372 {
 373         if (!p)
 374                 return NULL;
 375
 376         p->output_format = output_format;
 377
 378         return p;
 379 }
 380
 381 int isl_printer_get_output_format(__isl_keep isl_printer *p)
 382 {
 383         if (!p)
 384                 return -1;
 385         return p->output_format;
 386 }
 387
 388 /* Does "p" have a note with identifier "id"?
 389  */
 390 isl_bool isl_printer_has_note(__isl_keep isl_printer *p,
 391         __isl_keep isl_id *id)
 392 {
 393         if (!p || !id)
 394                 return isl_bool_error;
 395         if (!p->notes)
 396                 return isl_bool_false;
 397         return isl_id_to_id_has(p->notes, id);
 398 }
 399
 400 /* Retrieve the note identified by "id" from "p".
 401  * The note is assumed to exist.
 402  */
 403 __isl_give isl_id *isl_printer_get_note(__isl_keep isl_printer *p,
 404         __isl_take isl_id *id)
 405 {
 406         isl_bool has_note;
 407
 408         has_note = isl_printer_has_note(p, id);
 409         if (has_note < 0)
 410                 goto error;
 411         if (!has_note)
 412                 isl_die(isl_printer_get_ctx(p), isl_error_invalid,
 413                         "no such note", goto error);
 414
 415         return isl_id_to_id_get(p->notes, id);
 416 error:
 417         isl_id_free(id);
 418         return NULL;
 419 }
 420
 421 /* Associate "note" to the identifier "id" in "p",
 422  * replacing the previous note associated to the identifier, if any.
 423  */
 424 __isl_give isl_printer *isl_printer_set_note(__isl_take isl_printer *p,
 425         __isl_take isl_id *id, __isl_take isl_id *note)
 426 {
 427         if (!p || !id || !note)
 428                 goto error;
 429         if (!p->notes) {
 430                 p->notes = isl_id_to_id_alloc(isl_printer_get_ctx(p), 1);
 431                 if (!p->notes)
 432                         goto error;
 433         }
 434         p->notes = isl_id_to_id_set(p->notes, id, note);
 435         if (!p->notes)
 436                 return isl_printer_free(p);
 437         return p;
 438 error:
 439         isl_printer_free(p);
 440         isl_id_free(id);
 441         isl_id_free(note);
 442         return NULL;
 443 }
 444
 445 /* Keep track of whether the printing to "p" is being performed from
 446  * an isl_*_dump function as specified by "dump".
 447  */
 448 __isl_give isl_printer *isl_printer_set_dump(__isl_take isl_printer *p,
 449         int dump)
 450 {
 451         if (!p)
 452                 return NULL;
 453
 454         p->dump = dump;
 455
 456         return p;
 457 }
 458
 459 /* Set the YAML style of "p" to "yaml_style" and return the updated printer.
 460  */
 461 __isl_give isl_printer *isl_printer_set_yaml_style(__isl_take isl_printer *p,
 462         int yaml_style)
 463 {
 464         if (!p)
 465                 return NULL;
 466
 467         p->yaml_style = yaml_style;
 468
 469         return p;
 470 }
 471
 472 /* Return the YAML style of "p" or -1 on error.
 473  */
 474 int isl_printer_get_yaml_style(__isl_keep isl_printer *p)
 475 {
 476         if (!p)
 477                 return -1;
 478         return p->yaml_style;
 479 }
 480
 481 /* Push "state" onto the stack of currently active YAML elements and
 482  * return the updated printer.
 483  */
 484 static __isl_give isl_printer *push_state(__isl_take isl_printer *p,
 485         enum isl_yaml_state state)
 486 {
 487         if (!p)
 488                 return NULL;
 489
 490         if (p->yaml_size < p->yaml_depth + 1) {
 491                 enum isl_yaml_state *state;
 492                 state = isl_realloc_array(p->ctx, p->yaml_state,
 493                                         enum isl_yaml_state, p->yaml_depth + 1);
 494                 if (!state)
 495                         return isl_printer_free(p);
 496                 p->yaml_state = state;
 497                 p->yaml_size = p->yaml_depth + 1;
 498         }
 499
 500         p->yaml_state[p->yaml_depth] = state;
 501         p->yaml_depth++;
 502
 503         return p;
 504 }
 505
 506 /* Remove the innermost active YAML element from the stack and
 507  * return the updated printer.
 508  */
 509 static __isl_give isl_printer *pop_state(__isl_take isl_printer *p)
 510 {
 511         if (!p)
 512                 return NULL;
 513         p->yaml_depth--;
 514         return p;
 515 }
 516
 517 /* Set the state of the innermost active YAML element to "state" and
 518  * return the updated printer.
 519  */
 520 static __isl_give isl_printer *update_state(__isl_take isl_printer *p,
 521         enum isl_yaml_state state)
 522 {
 523         if (!p)
 524                 return NULL;
 525         if (p->yaml_depth < 1)
 526                 isl_die(isl_printer_get_ctx(p), isl_error_invalid,
 527                         "not in YAML construct", return isl_printer_free(p));
 528
 529         p->yaml_state[p->yaml_depth - 1] = state;
 530
 531         return p;
 532 }
 533
 534 /* Return the state of the innermost active YAML element.
 535  * Return isl_yaml_none if we are not inside any YAML element.
 536  */
 537 static enum isl_yaml_state current_state(__isl_keep isl_printer *p)
 538 {
 539         if (!p)
 540                 return isl_yaml_none;
 541         if (p->yaml_depth < 1)
 542                 return isl_yaml_none;
 543         return p->yaml_state[p->yaml_depth - 1];
 544 }
 545
 546 /* If we are printing a YAML document and we are at the start of an element,
 547  * print whatever is needed before we can print the actual element and
 548  * keep track of the fact that we are now printing the element.
 549  * If "eol" is set, then whatever we print is going to be the last
 550  * thing that gets printed on this line.
 551  *
 552  * If we are about the print the first key of a mapping, then nothing
 553  * extra needs to be printed.  For any other key, however, we need
 554  * to either move to the next line (in block format) or print a comma
 555  * (in flow format).
 556  * Before printing a value in a mapping, we need to print a colon.
 557  *
 558  * For sequences, in flow format, we only need to print a comma
 559  * for each element except the first.
 560  * In block format, before the first element in the sequence,
 561  * we move to a new line, print a dash and increase the indentation.
 562  * Before any other element, we print a dash on a new line,
 563  * temporarily moving the indentation back.
 564  */
 565 static __isl_give isl_printer *enter_state(__isl_take isl_printer *p,
 566         int eol)
 567 {
 568         enum isl_yaml_state state;
 569
 570         if (!p)
 571                 return NULL;
 572
 573         state = current_state(p);
 574         if (state == isl_yaml_mapping_val_start) {
 575                 if (eol)
 576                         p = p->ops->print_str(p, ":");
 577                 else
 578                         p = p->ops->print_str(p, ": ");
 579                 p = update_state(p, isl_yaml_mapping_val);
 580         } else if (state == isl_yaml_mapping_first_key_start) {
 581                 p = update_state(p, isl_yaml_mapping_key);
 582         } else if (state == isl_yaml_mapping_key_start) {
 583                 if (p->yaml_style == ISL_YAML_STYLE_FLOW)
 584                         p = p->ops->print_str(p, ", ");
 585                 else {
 586                         p = p->ops->end_line(p);
 587                         p = p->ops->start_line(p);
 588                 }
 589                 p = update_state(p, isl_yaml_mapping_key);
 590         } else if (state == isl_yaml_sequence_first_start) {
 591                 if (p->yaml_style != ISL_YAML_STYLE_FLOW) {
 592                         p = p->ops->end_line(p);
 593                         p = p->ops->start_line(p);
 594                         p = p->ops->print_str(p, "- ");
 595                         p = isl_printer_indent(p, 2);
 596                 }
 597                 p = update_state(p, isl_yaml_sequence);
 598         } else if (state == isl_yaml_sequence_start) {
 599                 if (p->yaml_style == ISL_YAML_STYLE_FLOW)
 600                         p = p->ops->print_str(p, ", ");
 601                 else {
 602                         p = p->ops->end_line(p);
 603                         p = isl_printer_indent(p, -2);
 604                         p = p->ops->start_line(p);
 605                         p = p->ops->print_str(p, "- ");
 606                         p = isl_printer_indent(p, 2);
 607                 }
 608                 p = update_state(p, isl_yaml_sequence);
 609         }
 610
 611         return p;
 612 }
 613
 614 __isl_give isl_printer *isl_printer_print_str(__isl_take isl_printer *p,
 615         const char *s)
 616 {
 617         if (!p)
 618                 return NULL;
 619         if (!s)
 620                 return isl_printer_free(p);
 621         p = enter_state(p, 0);
 622         if (!p)
 623                 return NULL;
 624         return p->ops->print_str(p, s);
 625 }
 626
 627 __isl_give isl_printer *isl_printer_print_double(__isl_take isl_printer *p,
 628         double d)
 629 {
 630         p = enter_state(p, 0);
 631         if (!p)
 632                 return NULL;
 633
 634         return p->ops->print_double(p, d);
 635 }
 636
 637 __isl_give isl_printer *isl_printer_print_int(__isl_take isl_printer *p, int i)
 638 {
 639         p = enter_state(p, 0);
 640         if (!p)
 641                 return NULL;
 642
 643         return p->ops->print_int(p, i);
 644 }
 645
 646 __isl_give isl_printer *isl_printer_print_isl_int(__isl_take isl_printer *p,
 647         isl_int i)
 648 {
 649         p = enter_state(p, 0);
 650         if (!p)
 651                 return NULL;
 652
 653         return p->ops->print_isl_int(p, i);
 654 }
 655
 656 __isl_give isl_printer *isl_printer_start_line(__isl_take isl_printer *p)
 657 {
 658         if (!p)
 659                 return NULL;
 660
 661         return p->ops->start_line(p);
 662 }
 663
 664 __isl_give isl_printer *isl_printer_end_line(__isl_take isl_printer *p)
 665 {
 666         if (!p)
 667                 return NULL;
 668
 669         return p->ops->end_line(p);
 670 }
 671
 672 char *isl_printer_get_str(__isl_keep isl_printer *printer)
 673 {
 674         if (!printer || !printer->buf)
 675                 return NULL;
 676         return strdup(printer->buf);
 677 }
 678
 679 __isl_give isl_printer *isl_printer_flush(__isl_take isl_printer *p)
 680 {
 681         if (!p)
 682                 return NULL;
 683
 684         return p->ops->flush(p);
 685 }
 686
 687 /* Start a YAML mapping and push a new state to reflect that we
 688  * are about to print the first key in a mapping.
 689  *
 690  * In flow style, print the opening brace.
 691  * In block style, move to the next line with an increased indentation,
 692  * except if this is the outer mapping or if we are inside a sequence
 693  * (in which case we have already increased the indentation and we want
 694  * to print the first key on the same line as the dash).
 695  */
 696 __isl_give isl_printer *isl_printer_yaml_start_mapping(
 697         __isl_take isl_printer *p)
 698 {
 699         enum isl_yaml_state state;
 700
 701         if (!p)
 702                 return NULL;
 703         p = enter_state(p, p->yaml_style == ISL_YAML_STYLE_BLOCK);
 704         if (!p)
 705                 return NULL;
 706         state = current_state(p);
 707         if (p->yaml_style == ISL_YAML_STYLE_FLOW)
 708                 p = p->ops->print_str(p, "{ ");
 709         else if (state != isl_yaml_none && state != isl_yaml_sequence) {
 710                 p = p->ops->end_line(p);
 711                 p = isl_printer_indent(p, 2);
 712                 p = p->ops->start_line(p);
 713         }
 714         p = push_state(p, isl_yaml_mapping_first_key_start);
 715         return p;
 716 }
 717
 718 /* Finish a YAML mapping and pop it from the state stack.
 719  *
 720  * In flow style, print the closing brace.
 721  *
 722  * In block style, first check if we are still in the
 723  * isl_yaml_mapping_first_key_start state.  If so, we have not printed
 724  * anything yet, so print "{}" to indicate an empty mapping.
 725  * If we increased the indentation in isl_printer_yaml_start_mapping,
 726  * then decrease it again.
 727  * If this is the outer mapping then print a newline.
 728  */
 729 __isl_give isl_printer *isl_printer_yaml_end_mapping(
 730         __isl_take isl_printer *p)
 731 {
 732         enum isl_yaml_state state;
 733
 734         state = current_state(p);
 735         p = pop_state(p);
 736         if (!p)
 737                 return NULL;
 738         if (p->yaml_style == ISL_YAML_STYLE_FLOW)
 739                 return p->ops->print_str(p, " }");
 740         if (state == isl_yaml_mapping_first_key_start)
 741                 p = p->ops->print_str(p, "{}");
 742         if (!p)
 743                 return NULL;
 744         state = current_state(p);
 745         if (state != isl_yaml_none && state != isl_yaml_sequence)
 746                 p = isl_printer_indent(p, -2);
 747         if (state == isl_yaml_none)
 748                 p = p->ops->end_line(p);
 749         return p;
 750 }
 751
 752 /* Start a YAML sequence and push a new state to reflect that we
 753  * are about to print the first element in a sequence.
 754  *
 755  * In flow style, print the opening bracket.
 756  */
 757 __isl_give isl_printer *isl_printer_yaml_start_sequence(
 758         __isl_take isl_printer *p)
 759 {
 760         if (!p)
 761                 return NULL;
 762         p = enter_state(p, p->yaml_style == ISL_YAML_STYLE_BLOCK);
 763         p = push_state(p, isl_yaml_sequence_first_start);
 764         if (!p)
 765                 return NULL;
 766         if (p->yaml_style == ISL_YAML_STYLE_FLOW)
 767                 p = p->ops->print_str(p, "[ ");
 768         return p;
 769 }
 770
 771 /* Finish a YAML sequence and pop it from the state stack.
 772  *
 773  * In flow style, print the closing bracket.
 774  *
 775  * In block style, check if we are still in the
 776  * isl_yaml_sequence_first_start state.  If so, we have not printed
 777  * anything yet, so print "[]" or " []" to indicate an empty sequence.
 778  * We print the extra space when we instructed enter_state not
 779  * to print a space at the end of the line.
 780  * Otherwise, undo the increase in indentation performed by
 781  * enter_state when moving away from the isl_yaml_sequence_first_start
 782  * state.
 783  * If this is the outer sequence then print a newline.
 784  */
 785 __isl_give isl_printer *isl_printer_yaml_end_sequence(
 786         __isl_take isl_printer *p)
 787 {
 788         enum isl_yaml_state state, up;
 789
 790         state = current_state(p);
 791         p = pop_state(p);
 792         if (!p)
 793                 return NULL;
 794         if (p->yaml_style == ISL_YAML_STYLE_FLOW)
 795                 return p->ops->print_str(p, " ]");
 796         up = current_state(p);
 797         if (state == isl_yaml_sequence_first_start) {
 798                 if (up == isl_yaml_mapping_val)
 799                         p = p->ops->print_str(p, " []");
 800                 else
 801                         p = p->ops->print_str(p, "[]");
 802         } else {
 803                 p = isl_printer_indent(p, -2);
 804         }
 805         if (!p)
 806                 return NULL;
 807         state = current_state(p);
 808         if (state == isl_yaml_none)
 809                 p = p->ops->end_line(p);
 810         return p;
 811 }
 812
 813 /* Mark the fact that the current element is finished and that
 814  * the next output belongs to the next element.
 815  * In particular, if we are printing a key, then prepare for
 816  * printing the subsequent value.  If we are printing a value,
 817  * prepare for printing the next key.  If we are printing an
 818  * element in a sequence, prepare for printing the next element.
 819  */
 820 __isl_give isl_printer *isl_printer_yaml_next(__isl_take isl_printer *p)
 821 {
 822         enum isl_yaml_state state;
 823
 824         if (!p)
 825                 return NULL;
 826         if (p->yaml_depth < 1)
 827                 isl_die(isl_printer_get_ctx(p), isl_error_invalid,
 828                         "not in YAML construct", return isl_printer_free(p));
 829
 830         state = current_state(p);
 831         if (state == isl_yaml_mapping_key)
 832                 state = isl_yaml_mapping_val_start;
 833         else if (state == isl_yaml_mapping_val)
 834                 state = isl_yaml_mapping_key_start;
 835         else if (state == isl_yaml_sequence)
 836                 state = isl_yaml_sequence_start;
 837         p = update_state(p, state);
 838
 839         return p;
 840 }