add support for generating ASTs from schedule relations
[isl.git] / include / isl / ast.h
blob4ddc9137b92d07c9108bddabd0e8feef65bfd659
1 #ifndef ISL_AST_H
2 #define ISL_AST_H
4 #include <isl/ctx.h>
5 #include <isl/id.h>
6 #include <isl/list.h>
7 #include <isl/printer.h>
9 #if defined(__cplusplus)
10 extern "C" {
11 #endif
13 struct isl_ast_expr;
14 typedef struct isl_ast_expr isl_ast_expr;
16 struct isl_ast_node;
17 typedef struct isl_ast_node isl_ast_node;
19 enum isl_ast_op_type {
20 isl_ast_op_error = -1,
21 isl_ast_op_and,
22 isl_ast_op_and_then,
23 isl_ast_op_or,
24 isl_ast_op_or_else,
25 isl_ast_op_max,
26 isl_ast_op_min,
27 isl_ast_op_minus,
28 isl_ast_op_add,
29 isl_ast_op_sub,
30 isl_ast_op_mul,
31 isl_ast_op_div,
32 isl_ast_op_fdiv_q, /* Round towards -infty */
33 isl_ast_op_pdiv_q, /* Dividend is non-negative */
34 isl_ast_op_pdiv_r, /* Dividend is non-negative */
35 isl_ast_op_cond,
36 isl_ast_op_select,
37 isl_ast_op_eq,
38 isl_ast_op_le,
39 isl_ast_op_ge,
40 isl_ast_op_call
43 enum isl_ast_expr_type {
44 isl_ast_expr_error = -1,
45 isl_ast_expr_op,
46 isl_ast_expr_id,
47 isl_ast_expr_int
50 enum isl_ast_node_type {
51 isl_ast_node_error = -1,
52 isl_ast_node_for = 1,
53 isl_ast_node_if,
54 isl_ast_node_block,
55 isl_ast_node_user
58 struct isl_ast_print_options;
59 typedef struct isl_ast_print_options isl_ast_print_options;
61 ISL_DECLARE_LIST(ast_expr)
62 ISL_DECLARE_LIST(ast_node)
64 int isl_options_set_ast_iterator_type(isl_ctx *ctx, const char *val);
65 const char *isl_options_get_ast_iterator_type(isl_ctx *ctx);
67 __isl_give isl_ast_expr *isl_ast_expr_from_id(__isl_take isl_id *id);
68 __isl_give isl_ast_expr *isl_ast_expr_neg(__isl_take isl_ast_expr *expr);
69 __isl_give isl_ast_expr *isl_ast_expr_add(__isl_take isl_ast_expr *expr1,
70 __isl_take isl_ast_expr *expr2);
71 __isl_give isl_ast_expr *isl_ast_expr_sub(__isl_take isl_ast_expr *expr1,
72 __isl_take isl_ast_expr *expr2);
73 __isl_give isl_ast_expr *isl_ast_expr_mul(__isl_take isl_ast_expr *expr1,
74 __isl_take isl_ast_expr *expr2);
75 __isl_give isl_ast_expr *isl_ast_expr_div(__isl_take isl_ast_expr *expr1,
76 __isl_take isl_ast_expr *expr2);
77 __isl_give isl_ast_expr *isl_ast_expr_and(__isl_take isl_ast_expr *expr1,
78 __isl_take isl_ast_expr *expr2);
79 __isl_give isl_ast_expr *isl_ast_expr_or(__isl_take isl_ast_expr *expr1,
80 __isl_take isl_ast_expr *expr2);
82 __isl_give isl_ast_expr *isl_ast_expr_copy(__isl_keep isl_ast_expr *expr);
83 void *isl_ast_expr_free(__isl_take isl_ast_expr *expr);
85 isl_ctx *isl_ast_expr_get_ctx(__isl_keep isl_ast_expr *expr);
86 enum isl_ast_expr_type isl_ast_expr_get_type(__isl_keep isl_ast_expr *expr);
87 int isl_ast_expr_get_int(__isl_keep isl_ast_expr *expr, isl_int *v);
88 __isl_give isl_id *isl_ast_expr_get_id(__isl_keep isl_ast_expr *expr);
90 enum isl_ast_op_type isl_ast_expr_get_op_type(__isl_keep isl_ast_expr *expr);
91 int isl_ast_expr_get_op_n_arg(__isl_keep isl_ast_expr *expr);
92 __isl_give isl_ast_expr *isl_ast_expr_get_op_arg(__isl_keep isl_ast_expr *expr,
93 int pos);
95 __isl_give isl_printer *isl_printer_print_ast_expr(__isl_take isl_printer *p,
96 __isl_keep isl_ast_expr *expr);
97 void isl_ast_expr_dump(__isl_keep isl_ast_expr *expr);
98 __isl_give char *isl_ast_expr_to_str(__isl_keep isl_ast_expr *expr);
100 __isl_give isl_ast_node *isl_ast_node_alloc_user(__isl_take isl_ast_expr *expr);
101 __isl_give isl_ast_node *isl_ast_node_copy(__isl_keep isl_ast_node *node);
102 void *isl_ast_node_free(__isl_take isl_ast_node *node);
104 isl_ctx *isl_ast_node_get_ctx(__isl_keep isl_ast_node *node);
105 enum isl_ast_node_type isl_ast_node_get_type(__isl_keep isl_ast_node *node);
107 __isl_give isl_ast_node *isl_ast_node_set_annotation(
108 __isl_take isl_ast_node *node, __isl_take isl_id *annotation);
109 __isl_give isl_id *isl_ast_node_get_annotation(__isl_keep isl_ast_node *node);
111 __isl_give isl_ast_expr *isl_ast_node_for_get_iterator(
112 __isl_keep isl_ast_node *node);
113 __isl_give isl_ast_expr *isl_ast_node_for_get_init(
114 __isl_keep isl_ast_node *node);
115 __isl_give isl_ast_expr *isl_ast_node_for_get_cond(
116 __isl_keep isl_ast_node *node);
117 __isl_give isl_ast_expr *isl_ast_node_for_get_inc(
118 __isl_keep isl_ast_node *node);
119 __isl_give isl_ast_node *isl_ast_node_for_get_body(
120 __isl_keep isl_ast_node *node);
121 int isl_ast_node_for_is_degenerate(__isl_keep isl_ast_node *node);
123 __isl_give isl_ast_expr *isl_ast_node_if_get_cond(
124 __isl_keep isl_ast_node *node);
125 __isl_give isl_ast_node *isl_ast_node_if_get_then(
126 __isl_keep isl_ast_node *node);
127 int isl_ast_node_if_has_else(__isl_keep isl_ast_node *node);
128 __isl_give isl_ast_node *isl_ast_node_if_get_else(
129 __isl_keep isl_ast_node *node);
131 __isl_give isl_ast_node_list *isl_ast_node_block_get_children(
132 __isl_keep isl_ast_node *node);
134 __isl_give isl_ast_expr *isl_ast_node_user_get_expr(
135 __isl_keep isl_ast_node *node);
137 __isl_give isl_printer *isl_printer_print_ast_node(__isl_take isl_printer *p,
138 __isl_keep isl_ast_node *node);
139 void isl_ast_node_dump(__isl_keep isl_ast_node *node);
141 __isl_give isl_ast_print_options *isl_ast_print_options_alloc(isl_ctx *ctx);
142 void *isl_ast_print_options_free(__isl_take isl_ast_print_options *options);
144 __isl_give isl_ast_print_options *isl_ast_print_options_set_print_user(
145 __isl_take isl_ast_print_options *options,
146 __isl_give isl_printer *(*print_user)(__isl_take isl_printer *p,
147 __isl_keep isl_ast_node *node, void *user),
148 void *user);
149 __isl_give isl_ast_print_options *isl_ast_print_options_set_print_for(
150 __isl_take isl_ast_print_options *options,
151 __isl_give isl_printer *(*print_for)(__isl_take isl_printer *p,
152 __isl_keep isl_ast_node *node, void *user),
153 void *user);
155 int isl_ast_node_foreach_ast_op_type(__isl_keep isl_ast_node *node,
156 int (*fn)(enum isl_ast_op_type type, void *user), void *user);
157 __isl_give isl_printer *isl_ast_op_type_print_macro(
158 enum isl_ast_op_type type, __isl_take isl_printer *p);
159 __isl_give isl_printer *isl_ast_node_print_macros(
160 __isl_keep isl_ast_node *node, __isl_take isl_printer *p);
161 __isl_give isl_printer *isl_ast_node_print(__isl_keep isl_ast_node *node,
162 __isl_take isl_printer *p,
163 __isl_keep isl_ast_print_options *options);
164 __isl_give isl_printer *isl_ast_node_for_print(__isl_keep isl_ast_node *node,
165 __isl_take isl_printer *p,
166 __isl_keep isl_ast_print_options *options);
167 __isl_give isl_printer *isl_ast_node_if_print(__isl_keep isl_ast_node *node,
168 __isl_take isl_printer *p,
169 __isl_keep isl_ast_print_options *options);
171 #if defined(__cplusplus)
173 #endif
175 #endif