isl_ast_private.h

   1 #ifndef ISL_AST_PRIVATE_H
   2 #define ISL_AST_PRIVATE_H
   3
   4 #include <isl/aff.h>
   5 #include <isl/ast.h>
   6 #include <isl/set.h>
   7 #include <isl/map.h>
   8 #include <isl/vec.h>
   9 #include <isl/list.h>
  10
  11 /* An expression is either an integer, an identifier or an operation
  12  * with zero or more arguments.
  13  */
  14 struct isl_ast_expr {
  15         int ref;
  16
  17         isl_ctx *ctx;
  18
  19         enum isl_ast_expr_type type;
  20
  21         union {
  22                 isl_val *v;
  23                 isl_id *id;
  24                 struct {
  25                         enum isl_ast_op_type op;
  26                         unsigned n_arg;
  27                         isl_ast_expr **args;
  28                 } op;
  29         } u;
  30 };
  31
  32 #undef EL
  33 #define EL isl_ast_expr
  34
  35 #include <isl_list_templ.h>
  36
  37 __isl_give isl_ast_expr *isl_ast_expr_alloc_int_si(isl_ctx *ctx, int i);
  38 __isl_give isl_ast_expr *isl_ast_expr_alloc_op(isl_ctx *ctx,
  39         enum isl_ast_op_type op, int n_arg);
  40 __isl_give isl_ast_expr *isl_ast_expr_alloc_binary(enum isl_ast_op_type type,
  41         __isl_take isl_ast_expr *expr1, __isl_take isl_ast_expr *expr2);
  42
  43 #undef EL
  44 #define EL isl_ast_node
  45
  46 #include <isl_list_templ.h>
  47
  48 /* A node is either a block, an if, a for, a user node or a mark node.
  49  * "else_node" is NULL if the if node does not have an else branch.
  50  * "cond" and "inc" are NULL for degenerate for nodes.
  51  * In case of a mark node, "mark" is the mark and "node" is the marked node.
  52  */
  53 struct isl_ast_node {
  54         int ref;
  55
  56         isl_ctx *ctx;
  57         enum isl_ast_node_type type;
  58
  59         union {
  60                 struct {
  61                         isl_ast_node_list *children;
  62                 } b;
  63                 struct {
  64                         isl_ast_expr *guard;
  65                         isl_ast_node *then;
  66                         isl_ast_node *else_node;
  67                 } i;
  68                 struct {
  69                         unsigned degenerate : 1;
  70                         isl_ast_expr *iterator;
  71                         isl_ast_expr *init;
  72                         isl_ast_expr *cond;
  73                         isl_ast_expr *inc;
  74                         isl_ast_node *body;
  75                 } f;
  76                 struct {
  77                         isl_ast_expr *expr;
  78                 } e;
  79                 struct {
  80                         isl_id *mark;
  81                         isl_ast_node *node;
  82                 } m;
  83         } u;
  84
  85         isl_id *annotation;
  86 };
  87
  88 __isl_give isl_ast_node *isl_ast_node_alloc_for(__isl_take isl_id *id);
  89 __isl_give isl_ast_node *isl_ast_node_for_mark_degenerate(
  90         __isl_take isl_ast_node *node);
  91 __isl_give isl_ast_node *isl_ast_node_alloc_if(__isl_take isl_ast_expr *guard);
  92 __isl_give isl_ast_node *isl_ast_node_alloc_block(
  93         __isl_take isl_ast_node_list *list);
  94 __isl_give isl_ast_node *isl_ast_node_alloc_mark(__isl_take isl_id *id,
  95         __isl_take isl_ast_node *node);
  96 __isl_give isl_ast_node *isl_ast_node_from_ast_node_list(
  97         __isl_take isl_ast_node_list *list);
  98 __isl_give isl_ast_node *isl_ast_node_for_set_body(
  99         __isl_take isl_ast_node *node, __isl_take isl_ast_node *body);
 100 __isl_give isl_ast_node *isl_ast_node_if_set_then(
 101         __isl_take isl_ast_node *node, __isl_take isl_ast_node *child);
 102
 103 struct isl_ast_print_options {
 104         int ref;
 105         isl_ctx *ctx;
 106
 107         __isl_give isl_printer *(*print_for)(__isl_take isl_printer *p,
 108                 __isl_take isl_ast_print_options *options,
 109                 __isl_keep isl_ast_node *node, void *user);
 110         void *print_for_user;
 111         __isl_give isl_printer *(*print_user)(__isl_take isl_printer *p,
 112                 __isl_take isl_ast_print_options *options,
 113                 __isl_keep isl_ast_node *node, void *user);
 114         void *print_user_user;
 115 };
 116
 117 __isl_give isl_printer *isl_ast_node_list_print(
 118         __isl_keep isl_ast_node_list *list, __isl_take isl_printer *p,
 119         __isl_keep isl_ast_print_options *options);
 120
 121 #endif