5 #include <isl_stream.h>
6 #include <isl_vertices.h>
7 #include <isl_obj_list.h>
8 #include <barvinok/barvinok.h>
13 #include <cloog/isl/cloog.h>
16 static int isl_bool_false
= 0;
17 static int isl_bool_true
= 1;
18 static int isl_bool_error
= -1;
20 static enum isl_token_type read_op
;
21 static enum isl_token_type vertices_op
;
23 struct isl_arg_choice iscc_format
[] = {
24 {"isl", ISL_FORMAT_ISL
},
25 {"omega", ISL_FORMAT_OMEGA
},
26 {"polylib", ISL_FORMAT_POLYLIB
},
27 {"latex", ISL_FORMAT_LATEX
},
33 struct barvinok_options
*barvinok
;
37 struct isl_arg iscc_options_arg
[] = {
38 ISL_ARG_CHILD(struct iscc_options
, barvinok
, "barvinok", barvinok_options_arg
,
40 ISL_ARG_CHOICE(struct iscc_options
, format
, 0, "format", \
41 iscc_format
, ISL_FORMAT_ISL
, "output format")
45 ISL_ARG_DEF(iscc_options
, struct iscc_options
, iscc_options_arg
)
47 static void *isl_obj_bool_copy(void *v
)
52 static void isl_obj_bool_free(void *v
)
56 static __isl_give isl_printer
*isl_obj_bool_print(__isl_take isl_printer
*p
,
59 if (v
== &isl_bool_true
)
60 return isl_printer_print_str(p
, "True");
61 else if (v
== &isl_bool_false
)
62 return isl_printer_print_str(p
, "False");
64 return isl_printer_print_str(p
, "Error");
67 static void *isl_obj_bool_add(void *v1
, void *v2
)
72 struct isl_obj_vtable isl_obj_bool_vtable
= {
78 #define isl_obj_bool (&isl_obj_bool_vtable)
80 int *isl_bool_from_int(int res
)
82 return res
< 0 ? &isl_bool_error
: res
? &isl_bool_true
: &isl_bool_false
;
85 int *union_map_is_equal(__isl_take isl_union_map
*map1
,
86 __isl_take isl_union_map
*map2
)
88 int res
= isl_union_map_is_equal(map1
, map2
);
89 isl_union_map_free(map1
);
90 isl_union_map_free(map2
);
91 return isl_bool_from_int(res
);
93 int *union_set_is_equal(__isl_take isl_union_set
*set1
,
94 __isl_take isl_union_set
*set2
)
96 return union_map_is_equal((isl_union_map
*)set1
, (isl_union_map
*)set2
);
99 int *union_map_is_subset(__isl_take isl_union_map
*map1
,
100 __isl_take isl_union_map
*map2
)
102 int res
= isl_union_map_is_subset(map1
, map2
);
103 isl_union_map_free(map1
);
104 isl_union_map_free(map2
);
105 return isl_bool_from_int(res
);
107 int *union_set_is_subset(__isl_take isl_union_set
*set1
,
108 __isl_take isl_union_set
*set2
)
110 return union_map_is_subset((isl_union_map
*)set1
, (isl_union_map
*)set2
);
113 int *union_map_is_strict_subset(__isl_take isl_union_map
*map1
,
114 __isl_take isl_union_map
*map2
)
116 int res
= isl_union_map_is_strict_subset(map1
, map2
);
117 isl_union_map_free(map1
);
118 isl_union_map_free(map2
);
119 return isl_bool_from_int(res
);
121 int *union_set_is_strict_subset(__isl_take isl_union_set
*set1
,
122 __isl_take isl_union_set
*set2
)
124 return union_map_is_strict_subset((isl_union_map
*)set1
,
125 (isl_union_map
*)set2
);
128 int *union_map_is_superset(__isl_take isl_union_map
*map1
,
129 __isl_take isl_union_map
*map2
)
131 return union_map_is_subset(map2
, map1
);
133 int *union_set_is_superset(__isl_take isl_union_set
*set1
,
134 __isl_take isl_union_set
*set2
)
136 return union_set_is_subset(set2
, set1
);
139 int *union_map_is_strict_superset(__isl_take isl_union_map
*map1
,
140 __isl_take isl_union_map
*map2
)
142 return union_map_is_strict_subset(map2
, map1
);
144 int *union_set_is_strict_superset(__isl_take isl_union_set
*set1
,
145 __isl_take isl_union_set
*set2
)
147 return union_set_is_strict_subset(set2
, set1
);
150 extern struct isl_obj_vtable isl_obj_list_vtable
;
151 #define isl_obj_list (&isl_obj_list_vtable)
153 typedef void *(*isc_bin_op_fn
)(void *lhs
, void *rhs
);
155 enum isl_token_type op
;
161 struct isc_named_bin_op
{
163 struct isc_bin_op op
;
167 isl_union_pw_qpolynomial
*upwqp
;
168 isl_union_pw_qpolynomial
*res
;
171 static int eval_at(__isl_take isl_point
*pnt
, void *user
)
173 struct iscc_at
*at
= (struct iscc_at
*) user
;
177 set
= isl_set_from_point(isl_point_copy(pnt
));
178 qp
= isl_union_pw_qpolynomial_eval(
179 isl_union_pw_qpolynomial_copy(at
->upwqp
), pnt
);
181 at
->res
= isl_union_pw_qpolynomial_add(at
->res
,
182 isl_union_pw_qpolynomial_from_pw_qpolynomial(
183 isl_pw_qpolynomial_alloc(set
, qp
)));
188 __isl_give isl_union_pw_qpolynomial
*isl_union_pw_qpolynomial_at(
189 __isl_take isl_union_pw_qpolynomial
*upwqp
,
190 __isl_take isl_union_set
*uset
)
195 at
.res
= isl_union_pw_qpolynomial_zero(isl_union_set_get_dim(uset
));
197 isl_union_set_foreach_point(uset
, eval_at
, &at
);
199 isl_union_pw_qpolynomial_free(upwqp
);
200 isl_union_set_free(uset
);
205 struct iscc_fold_at
{
206 isl_union_pw_qpolynomial_fold
*upwf
;
207 isl_union_pw_qpolynomial
*res
;
210 static int eval_fold_at(__isl_take isl_point
*pnt
, void *user
)
212 struct iscc_fold_at
*at
= (struct iscc_fold_at
*) user
;
216 set
= isl_set_from_point(isl_point_copy(pnt
));
217 qp
= isl_union_pw_qpolynomial_fold_eval(
218 isl_union_pw_qpolynomial_fold_copy(at
->upwf
), pnt
);
220 at
->res
= isl_union_pw_qpolynomial_add(at
->res
,
221 isl_union_pw_qpolynomial_from_pw_qpolynomial(
222 isl_pw_qpolynomial_alloc(set
, qp
)));
227 __isl_give isl_union_pw_qpolynomial
*isl_union_pw_qpolynomial_fold_at(
228 __isl_take isl_union_pw_qpolynomial_fold
*upwf
,
229 __isl_take isl_union_set
*uset
)
231 struct iscc_fold_at at
;
234 at
.res
= isl_union_pw_qpolynomial_zero(isl_union_set_get_dim(uset
));
236 isl_union_set_foreach_point(uset
, eval_fold_at
, &at
);
238 isl_union_pw_qpolynomial_fold_free(upwf
);
239 isl_union_set_free(uset
);
244 struct isc_bin_op bin_ops
[] = {
245 { '+', isl_obj_union_set
, isl_obj_union_set
,
247 (isc_bin_op_fn
) &isl_union_set_union
},
248 { '+', isl_obj_union_map
, isl_obj_union_map
,
250 (isc_bin_op_fn
) &isl_union_map_union
},
251 { '-', isl_obj_union_set
, isl_obj_union_set
,
253 (isc_bin_op_fn
) &isl_union_set_subtract
},
254 { '-', isl_obj_union_map
, isl_obj_union_map
,
256 (isc_bin_op_fn
) &isl_union_map_subtract
},
257 { '*', isl_obj_union_set
, isl_obj_union_set
,
259 (isc_bin_op_fn
) &isl_union_set_intersect
},
260 { '*', isl_obj_union_map
, isl_obj_union_map
,
262 (isc_bin_op_fn
) &isl_union_map_intersect
},
263 { '*', isl_obj_union_map
, isl_obj_union_set
,
265 (isc_bin_op_fn
) &isl_union_map_intersect_domain
},
266 { '.', isl_obj_union_map
, isl_obj_union_map
,
268 (isc_bin_op_fn
) &isl_union_map_apply_range
},
269 { ISL_TOKEN_TO
, isl_obj_union_set
, isl_obj_union_set
,
271 (isc_bin_op_fn
) &isl_union_map_from_domain_and_range
},
272 { '=', isl_obj_union_set
, isl_obj_union_set
, isl_obj_bool
,
273 (isc_bin_op_fn
) &union_set_is_equal
},
274 { '=', isl_obj_union_map
, isl_obj_union_map
, isl_obj_bool
,
275 (isc_bin_op_fn
) &union_map_is_equal
},
276 { ISL_TOKEN_LE
, isl_obj_union_set
, isl_obj_union_set
,
277 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_subset
},
278 { ISL_TOKEN_LE
, isl_obj_union_map
, isl_obj_union_map
,
279 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_subset
},
280 { ISL_TOKEN_LT
, isl_obj_union_set
, isl_obj_union_set
,
281 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_strict_subset
},
282 { ISL_TOKEN_LT
, isl_obj_union_map
, isl_obj_union_map
,
283 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_strict_subset
},
284 { ISL_TOKEN_GE
, isl_obj_union_set
, isl_obj_union_set
,
285 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_superset
},
286 { ISL_TOKEN_GE
, isl_obj_union_map
, isl_obj_union_map
,
287 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_superset
},
288 { ISL_TOKEN_GT
, isl_obj_union_set
, isl_obj_union_set
,
289 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_strict_superset
},
290 { ISL_TOKEN_GT
, isl_obj_union_map
, isl_obj_union_map
,
291 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_strict_superset
},
292 { '+', isl_obj_union_pw_qpolynomial
, isl_obj_union_pw_qpolynomial
,
293 isl_obj_union_pw_qpolynomial
,
294 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_add
},
295 { '-', isl_obj_union_pw_qpolynomial
, isl_obj_union_pw_qpolynomial
,
296 isl_obj_union_pw_qpolynomial
,
297 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_sub
},
298 { '*', isl_obj_union_pw_qpolynomial
, isl_obj_union_pw_qpolynomial
,
299 isl_obj_union_pw_qpolynomial
,
300 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_mul
},
301 { '*', isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
302 isl_obj_union_pw_qpolynomial
,
303 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_intersect_domain
},
304 { '*', isl_obj_union_pw_qpolynomial_fold
, isl_obj_union_set
,
305 isl_obj_union_pw_qpolynomial_fold
,
306 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_intersect_domain
},
307 { '@', isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
308 isl_obj_union_pw_qpolynomial
,
309 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_at
},
310 { '@', isl_obj_union_pw_qpolynomial_fold
, isl_obj_union_set
,
311 isl_obj_union_pw_qpolynomial
,
312 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_at
},
313 { '%', isl_obj_union_set
, isl_obj_union_set
,
315 (isc_bin_op_fn
) &isl_union_set_gist
},
316 { '%', isl_obj_union_map
, isl_obj_union_map
,
318 (isc_bin_op_fn
) &isl_union_map_gist
},
319 { '%', isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
320 isl_obj_union_pw_qpolynomial
,
321 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_gist
},
322 { '%', isl_obj_union_pw_qpolynomial_fold
, isl_obj_union_set
,
323 isl_obj_union_pw_qpolynomial_fold
,
324 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_gist
},
327 struct isc_named_bin_op named_bin_ops
[] = {
328 { "cross", { -1, isl_obj_union_set
, isl_obj_union_set
,
330 (isc_bin_op_fn
) &isl_union_set_product
} },
331 { "cross", { -1, isl_obj_union_map
, isl_obj_union_map
,
333 (isc_bin_op_fn
) &isl_union_map_product
} },
337 __isl_give isl_set
*union_set_sample(__isl_take isl_union_set
*uset
)
339 return isl_set_from_basic_set(isl_union_set_sample(uset
));
342 __isl_give isl_map
*union_map_sample(__isl_take isl_union_map
*umap
)
344 return isl_map_from_basic_map(isl_union_map_sample(umap
));
347 static __isl_give
struct isl_list
*union_pw_qpolynomial_upper_bound(
348 __isl_take isl_union_pw_qpolynomial
*upwqp
)
351 struct isl_list
*list
;
354 ctx
= isl_union_pw_qpolynomial_get_ctx(upwqp
);
355 list
= isl_list_alloc(ctx
, 2);
359 list
->obj
[0].type
= isl_obj_union_pw_qpolynomial_fold
;
360 list
->obj
[0].v
= isl_union_pw_qpolynomial_bound(upwqp
,
361 isl_fold_max
, &tight
);
362 list
->obj
[1].type
= isl_obj_bool
;
363 list
->obj
[1].v
= tight
? &isl_bool_true
: &isl_bool_false
;
364 if (tight
< 0 || !list
->obj
[0].v
)
374 struct iscc_codegen_data
{
382 CloogBlockList
**nextBL
;
383 CloogScatteringList
*scatterings
;
384 CloogScatteringList
**nextScat
;
387 static int add_domain(__isl_take isl_map
*map
, void *user
)
389 struct iscc_codegen_data
*data
= (struct iscc_codegen_data
*)user
;
390 isl_ctx
*ctx
= isl_map_get_ctx(map
);
393 CloogStatement
*statement
;
395 CloogScatteringList
*s
;
398 n_in
= isl_map_dim(map
, isl_dim_in
);
400 isl_dim
*dim
= isl_map_get_dim(map
);
401 data
->nparam
= isl_map_dim(map
, isl_dim_param
);
402 data
->n_scat
= isl_map_dim(map
, isl_dim_out
);
403 dim
= isl_dim_drop_inputs(dim
, 0, n_in
);
404 dim
= isl_dim_drop_outputs(dim
, 0, data
->n_scat
);
405 data
->context
= isl_set_universe(dim
);
407 isl_assert(ctx
, data
->nparam
== isl_map_dim(map
, isl_dim_param
),
409 isl_assert(ctx
, data
->n_scat
== isl_map_dim(map
, isl_dim_out
),
412 if (n_in
> data
->max_iter
)
413 data
->max_iter
= n_in
;
415 l
= cloog_loop_malloc(data
->state
);
420 domain
= isl_map_domain(isl_map_copy(map
));
421 l
->domain
= cloog_domain_from_isl_set(domain
);
423 statement
= cloog_statement_alloc(data
->state
, data
->nr
);
424 statement
->usr
= NULL
;
425 dim
= isl_map_get_dim(map
);
426 statement
->name
= strdup(isl_dim_get_tuple_name(dim
, isl_dim_in
));
428 l
->block
= cloog_block_alloc(statement
, 0, NULL
, n_in
);
430 s
= isl_alloc_type(ctx
, CloogScatteringList
);
431 s
->scatt
= cloog_scattering_from_isl_map(map
);
435 *data
->nextBL
= cloog_block_list_alloc((*data
->next
)->block
);
436 data
->next
= &(*data
->next
)->next
;
437 data
->nextBL
= &(*data
->nextBL
)->next
;
439 data
->nextScat
= &(s
->next
);
448 char **generate_names(isl_ctx
*ctx
, int n
, const char *prefix
)
453 names
= isl_alloc_array(ctx
, char *, n
);
454 for (i
= 0; i
< n
; i
++) {
455 names
[i
] = isl_alloc_array(ctx
, char, 50);
456 sprintf(names
[i
], "%s%d", prefix
, i
);
462 void *codegen(void *arg
)
465 struct iscc_codegen_data data
;
467 isl_union_map
*umap
= (isl_union_map
*)arg
;
468 isl_ctx
*ctx
= isl_union_map_get_ctx(umap
);
469 CloogOptions
*options
;
471 struct clast_stmt
*stmt
;
473 data
.state
= cloog_isl_state_malloc(ctx
);
474 options
= cloog_options_malloc(data
.state
);
475 options
->language
= LANGUAGE_C
;
476 p
= cloog_program_malloc();
478 isl_assert(ctx
, p
, goto error
);
479 p
->names
= cloog_names_malloc();
480 isl_assert(ctx
, p
->names
, goto error2
);
488 data
.next
= &p
->loop
;
489 data
.nextBL
= &p
->blocklist
;
490 data
.scatterings
= NULL
;
491 data
.nextScat
= &data
.scatterings
;
493 if (isl_union_map_foreach_map(umap
, &add_domain
, &data
) < 0)
499 p
->names
->nb_parameters
= data
.nparam
;
500 p
->names
->parameters
= isl_alloc_array(ctx
, char *, data
.nparam
);
501 dim
= isl_set_get_dim(data
.context
);
502 p
->context
= cloog_domain_from_isl_set(data
.context
);
503 for (i
= 0; i
< data
.nparam
; ++i
) {
504 const char *s
= isl_dim_get_name(dim
, isl_dim_param
, i
);
505 p
->names
->parameters
[i
] = strdup(s
);
509 p
->names
->nb_iterators
= data
.max_iter
;
510 p
->names
->iterators
= generate_names(ctx
, data
.max_iter
, "i");
512 p
->names
->nb_scattering
= data
.n_scat
;
513 p
->names
->scattering
= generate_names(ctx
, data
.n_scat
, "c");
514 p
->names
->nb_scalars
= 0;
515 p
->nb_scattdims
= data
.n_scat
;
516 p
->scaldims
= (int *)malloc(p
->nb_scattdims
*(sizeof(int)));
517 for (i
= 0; i
< p
->nb_scattdims
; i
++)
520 cloog_program_scatter(p
, data
.scatterings
, options
);
521 p
= cloog_program_generate(p
, options
);
523 stmt
= cloog_clast_create(p
, options
);
524 clast_pprint(stdout
, stmt
, 0, options
);
525 cloog_clast_free(stmt
);
528 cloog_scattering_list_free(data
.scatterings
);
530 cloog_program_free(p
);
532 cloog_options_free(options
);
533 cloog_state_free(data
.state
);
534 isl_union_map_free(umap
);
539 typedef void *(*isc_un_op_fn
)(void *arg
);
541 enum isl_token_type op
;
546 struct isc_named_un_op
{
550 struct isc_named_un_op named_un_ops
[] = {
551 {"aff", { -1, isl_obj_union_map
, isl_obj_union_map
,
552 (isc_un_op_fn
) &isl_union_map_affine_hull
} },
553 {"aff", { -1, isl_obj_union_set
, isl_obj_union_set
,
554 (isc_un_op_fn
) &isl_union_set_affine_hull
} },
555 {"card", { -1, isl_obj_union_set
,
556 isl_obj_union_pw_qpolynomial
,
557 (isc_un_op_fn
) &isl_union_set_card
} },
558 {"card", { -1, isl_obj_union_map
,
559 isl_obj_union_pw_qpolynomial
,
560 (isc_un_op_fn
) &isl_union_map_card
} },
561 {"coalesce", { -1, isl_obj_union_set
, isl_obj_union_set
,
562 (isc_un_op_fn
) &isl_union_set_coalesce
} },
563 {"coalesce", { -1, isl_obj_union_map
, isl_obj_union_map
,
564 (isc_un_op_fn
) &isl_union_map_coalesce
} },
565 {"coalesce", { -1, isl_obj_union_pw_qpolynomial
,
566 isl_obj_union_pw_qpolynomial
,
567 (isc_un_op_fn
) &isl_union_pw_qpolynomial_coalesce
} },
568 {"coalesce", { -1, isl_obj_union_pw_qpolynomial_fold
,
569 isl_obj_union_pw_qpolynomial_fold
,
570 (isc_un_op_fn
) &isl_union_pw_qpolynomial_fold_coalesce
} },
572 {"codegen", { -1, isl_obj_union_map
, isl_obj_none
, &codegen
} },
574 {"deltas", { -1, isl_obj_union_map
, isl_obj_union_set
,
575 (isc_un_op_fn
) &isl_union_map_deltas
} },
576 {"dom", { -1, isl_obj_union_map
, isl_obj_union_set
,
577 (isc_un_op_fn
) &isl_union_map_domain
} },
578 {"dom", { -1, isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
579 (isc_un_op_fn
) &isl_union_pw_qpolynomial_domain
} },
580 {"dom", { -1, isl_obj_union_pw_qpolynomial_fold
,
582 (isc_un_op_fn
) &isl_union_pw_qpolynomial_fold_domain
} },
583 {"ran", { -1, isl_obj_union_map
, isl_obj_union_set
,
584 (isc_un_op_fn
) &isl_union_map_range
} },
585 {"lexmin", { -1, isl_obj_union_map
, isl_obj_union_map
,
586 (isc_un_op_fn
) &isl_union_map_lexmin
} },
587 {"lexmax", { -1, isl_obj_union_map
, isl_obj_union_map
,
588 (isc_un_op_fn
) &isl_union_map_lexmax
} },
589 {"lexmin", { -1, isl_obj_union_set
, isl_obj_union_set
,
590 (isc_un_op_fn
) &isl_union_set_lexmin
} },
591 {"lexmax", { -1, isl_obj_union_set
, isl_obj_union_set
,
592 (isc_un_op_fn
) &isl_union_set_lexmax
} },
593 {"sample", { -1, isl_obj_union_set
, isl_obj_set
,
594 (isc_un_op_fn
) &union_set_sample
} },
595 {"sample", { -1, isl_obj_union_map
, isl_obj_map
,
596 (isc_un_op_fn
) &union_map_sample
} },
597 {"sum", { -1, isl_obj_union_pw_qpolynomial
,
598 isl_obj_union_pw_qpolynomial
,
599 (isc_un_op_fn
) &isl_union_pw_qpolynomial_sum
} },
600 {"ub", { -1, isl_obj_union_pw_qpolynomial
, isl_obj_list
,
601 (isc_un_op_fn
) &union_pw_qpolynomial_upper_bound
} },
605 struct isl_named_obj
{
610 static void free_obj(struct isl_obj obj
)
612 obj
.type
->free(obj
.v
);
615 static int same_name(const void *entry
, const void *val
)
617 const struct isl_named_obj
*named
= (const struct isl_named_obj
*)entry
;
619 return !strcmp(named
->name
, val
);
622 static int do_assign(struct isl_ctx
*ctx
, struct isl_hash_table
*table
,
623 char *name
, struct isl_obj obj
)
625 struct isl_hash_table_entry
*entry
;
627 struct isl_named_obj
*named
;
629 name_hash
= isl_hash_string(isl_hash_init(), name
);
630 entry
= isl_hash_table_find(ctx
, table
, name_hash
, same_name
, name
, 1);
635 free_obj(named
->obj
);
638 named
= isl_alloc_type(ctx
, struct isl_named_obj
);
653 static struct isl_obj
stored_obj(struct isl_ctx
*ctx
,
654 struct isl_hash_table
*table
, char *name
)
656 struct isl_obj obj
= { isl_obj_none
, NULL
};
657 struct isl_hash_table_entry
*entry
;
660 name_hash
= isl_hash_string(isl_hash_init(), name
);
661 entry
= isl_hash_table_find(ctx
, table
, name_hash
, same_name
, name
, 0);
663 struct isl_named_obj
*named
;
669 obj
.v
= obj
.type
->copy(obj
.v
);
673 static int is_subtype(struct isl_obj obj
, isl_obj_type super
)
675 if (obj
.type
== super
)
677 if (obj
.type
== isl_obj_map
&& super
== isl_obj_union_map
)
679 if (obj
.type
== isl_obj_set
&& super
== isl_obj_union_set
)
681 if (obj
.type
== isl_obj_pw_qpolynomial
&&
682 super
== isl_obj_union_pw_qpolynomial
)
684 if (obj
.type
== isl_obj_pw_qpolynomial_fold
&&
685 super
== isl_obj_union_pw_qpolynomial_fold
)
687 if (obj
.type
== isl_obj_union_set
&& isl_union_set_is_empty(obj
.v
))
692 static struct isl_obj
convert(struct isl_obj obj
, isl_obj_type type
)
694 if (obj
.type
== type
)
696 if (obj
.type
== isl_obj_map
&& type
== isl_obj_union_map
) {
697 obj
.type
= isl_obj_union_map
;
698 obj
.v
= isl_union_map_from_map(obj
.v
);
701 if (obj
.type
== isl_obj_set
&& type
== isl_obj_union_set
) {
702 obj
.type
= isl_obj_union_set
;
703 obj
.v
= isl_union_set_from_set(obj
.v
);
706 if (obj
.type
== isl_obj_pw_qpolynomial
&&
707 type
== isl_obj_union_pw_qpolynomial
) {
708 obj
.type
= isl_obj_union_pw_qpolynomial
;
709 obj
.v
= isl_union_pw_qpolynomial_from_pw_qpolynomial(obj
.v
);
712 if (obj
.type
== isl_obj_pw_qpolynomial_fold
&&
713 type
== isl_obj_union_pw_qpolynomial_fold
) {
714 obj
.type
= isl_obj_union_pw_qpolynomial_fold
;
715 obj
.v
= isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj
.v
);
718 if (obj
.type
== isl_obj_union_set
&& isl_union_set_is_empty(obj
.v
)) {
719 if (type
== isl_obj_union_map
) {
720 obj
.type
= isl_obj_union_map
;
723 if (type
== isl_obj_union_pw_qpolynomial
) {
724 isl_dim
*dim
= isl_union_set_get_dim(obj
.v
);
725 isl_union_set_free(obj
.v
);
726 obj
.v
= isl_union_pw_qpolynomial_zero(dim
);
727 obj
.type
= isl_obj_union_pw_qpolynomial
;
730 if (type
== isl_obj_union_pw_qpolynomial_fold
) {
731 isl_dim
*dim
= isl_union_set_get_dim(obj
.v
);
732 isl_union_set_free(obj
.v
);
733 obj
.v
= isl_union_pw_qpolynomial_fold_zero(dim
);
734 obj
.type
= isl_obj_union_pw_qpolynomial_fold
;
739 obj
.type
= isl_obj_none
;
744 static struct isc_bin_op
*read_bin_op_if_available(struct isl_stream
*s
,
748 struct isl_token
*tok
;
750 tok
= isl_stream_next_token(s
);
757 if (bin_ops
[i
].op
!= tok
->type
)
759 if (!is_subtype(lhs
, bin_ops
[i
].lhs
))
767 if (!named_bin_ops
[i
].name
)
769 if (named_bin_ops
[i
].op
.op
!= tok
->type
)
771 if (!is_subtype(lhs
, named_bin_ops
[i
].op
.lhs
))
775 return &named_bin_ops
[i
].op
;
778 isl_stream_push_token(s
, tok
);
783 static struct isc_un_op
*read_prefix_un_op_if_available(struct isl_stream
*s
)
786 struct isl_token
*tok
;
788 tok
= isl_stream_next_token(s
);
793 if (!named_un_ops
[i
].name
)
795 if (named_un_ops
[i
].op
.op
!= tok
->type
)
799 return &named_un_ops
[i
].op
;
802 isl_stream_push_token(s
, tok
);
807 static struct isc_un_op
*find_matching_un_op(struct isc_un_op
*like
,
813 if (!named_un_ops
[i
].name
)
815 if (named_un_ops
[i
].op
.op
!= like
->op
)
817 if (!is_subtype(arg
, named_un_ops
[i
].op
.arg
))
820 return &named_un_ops
[i
].op
;
826 static int is_assign(struct isl_stream
*s
)
828 struct isl_token
*tok
;
829 struct isl_token
*tok2
;
832 tok
= isl_stream_next_token(s
);
835 if (tok
->type
!= ISL_TOKEN_IDENT
) {
836 isl_stream_push_token(s
, tok
);
840 tok2
= isl_stream_next_token(s
);
842 isl_stream_push_token(s
, tok
);
845 assign
= tok2
->type
== ISL_TOKEN_DEF
;
846 isl_stream_push_token(s
, tok2
);
847 isl_stream_push_token(s
, tok
);
852 static struct isl_obj
read_obj(struct isl_stream
*s
,
853 struct isl_hash_table
*table
);
854 static struct isl_obj
read_expr(struct isl_stream
*s
,
855 struct isl_hash_table
*table
);
857 static struct isl_obj
read_un_op_expr(struct isl_stream
*s
,
858 struct isl_hash_table
*table
, struct isc_un_op
*op
)
860 struct isl_obj obj
= { isl_obj_none
, NULL
};
862 obj
= read_obj(s
, table
);
866 op
= find_matching_un_op(op
, obj
);
868 isl_assert(s
->ctx
, op
, goto error
);
869 obj
= convert(obj
, op
->arg
);
870 obj
.v
= op
->fn(obj
.v
);
876 obj
.type
= isl_obj_none
;
881 static struct isl_obj
transitive_closure(struct isl_ctx
*ctx
, struct isl_obj obj
)
883 struct isl_list
*list
;
886 if (obj
.type
== isl_obj_map
)
887 obj
= convert(obj
, isl_obj_union_map
);
888 isl_assert(ctx
, obj
.type
== isl_obj_union_map
, goto error
);
889 list
= isl_list_alloc(ctx
, 2);
893 list
->obj
[0].type
= isl_obj_union_map
;
894 list
->obj
[0].v
= isl_union_map_transitive_closure(obj
.v
, &exact
);
895 list
->obj
[1].type
= isl_obj_bool
;
896 list
->obj
[1].v
= exact
? &isl_bool_true
: &isl_bool_false
;
898 obj
.type
= isl_obj_list
;
899 if (exact
< 0 || !list
->obj
[0].v
)
905 obj
.type
= isl_obj_none
;
910 static struct isl_obj
obj_at_index(struct isl_stream
*s
, struct isl_obj obj
)
912 struct isl_list
*list
= obj
.v
;
913 struct isl_token
*tok
;
916 tok
= isl_stream_next_token(s
);
917 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
918 isl_stream_error(s
, tok
, "expecting index");
920 isl_stream_push_token(s
, tok
);
923 i
= isl_int_get_si(tok
->u
.v
);
925 isl_assert(s
->ctx
, i
< list
->n
, goto error
);
926 if (isl_stream_eat(s
, ']'))
930 obj
.v
= obj
.type
->copy(obj
.v
);
937 obj
.type
= isl_obj_none
;
942 static struct isl_obj
apply(struct isl_stream
*s
, __isl_take isl_union_map
*umap
,
943 struct isl_hash_table
*table
)
947 obj
= read_expr(s
, table
);
948 isl_assert(s
->ctx
, is_subtype(obj
, isl_obj_union_set
) ||
949 is_subtype(obj
, isl_obj_union_map
), goto error
);
951 if (obj
.type
== isl_obj_set
)
952 obj
= convert(obj
, isl_obj_union_set
);
953 else if (obj
.type
== isl_obj_map
)
954 obj
= convert(obj
, isl_obj_union_map
);
955 if (obj
.type
== isl_obj_union_set
) {
956 obj
.v
= isl_union_set_apply(obj
.v
, umap
);
958 obj
.v
= isl_union_map_apply_range(obj
.v
, umap
);
962 if (isl_stream_eat(s
, ')'))
967 isl_union_map_free(umap
);
970 obj
.type
= isl_obj_none
;
975 struct add_vertex_data
{
976 struct isl_list
*list
;
980 static int add_vertex(__isl_take isl_vertex
*vertex
, void *user
)
982 struct add_vertex_data
*data
= (struct add_vertex_data
*)user
;
985 expr
= isl_vertex_get_expr(vertex
);
987 data
->list
->obj
[data
->i
].type
= isl_obj_set
;
988 data
->list
->obj
[data
->i
].v
= isl_set_from_basic_set(expr
);
991 isl_vertex_free(vertex
);
996 static struct isl_obj
vertices(struct isl_stream
*s
,
997 struct isl_hash_table
*table
)
1002 isl_basic_set
*hull
;
1003 isl_vertices
*vertices
= NULL
;
1004 struct isl_list
*list
= NULL
;
1005 struct add_vertex_data data
;
1007 obj
= read_expr(s
, table
);
1008 isl_assert(s
->ctx
, obj
.type
== isl_obj_set
, goto error
);
1011 set
= isl_set_remove_divs(set
);
1012 hull
= isl_set_convex_hull(set
);
1013 vertices
= isl_basic_set_compute_vertices(hull
);
1014 isl_basic_set_free(hull
);
1016 ctx
= isl_vertices_get_ctx(vertices
);
1017 list
= isl_list_alloc(ctx
, isl_vertices_get_n_vertices(vertices
));
1024 if (isl_vertices_foreach_vertex(vertices
, &add_vertex
, &data
) < 0)
1027 isl_vertices_free(vertices
);
1029 obj
.type
= isl_obj_list
;
1034 isl_vertices_free(vertices
);
1035 isl_list_free(list
);
1037 obj
.type
= isl_obj_none
;
1042 static struct isl_obj
power(struct isl_stream
*s
, struct isl_obj obj
)
1044 struct isl_token
*tok
;
1046 if (isl_stream_eat_if_available(s
, '+'))
1047 return transitive_closure(s
->ctx
, obj
);
1049 tok
= isl_stream_next_token(s
);
1050 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
|| isl_int_cmp_si(tok
->u
.v
, -1)) {
1051 isl_stream_error(s
, tok
, "expecting -1");
1053 isl_stream_push_token(s
, tok
);
1056 isl_token_free(tok
);
1057 isl_assert(s
->ctx
, obj
.type
== isl_obj_union_map
, goto error
);
1059 obj
.v
= isl_union_map_reverse(obj
.v
);
1066 obj
.type
= isl_obj_none
;
1071 static struct isl_obj
read_from_file(struct isl_stream
*s
)
1074 struct isl_token
*tok
;
1075 struct isl_stream
*s_file
;
1078 tok
= isl_stream_next_token(s
);
1079 if (!tok
|| tok
->type
!= ISL_TOKEN_STRING
) {
1080 isl_stream_error(s
, tok
, "expecting filename");
1081 isl_token_free(tok
);
1085 file
= fopen(tok
->u
.s
, "r");
1086 isl_token_free(tok
);
1087 isl_assert(s
->ctx
, file
, goto error
);
1089 s_file
= isl_stream_new_file(s
->ctx
, file
);
1095 obj
= isl_stream_read_obj(s_file
);
1097 isl_stream_free(s_file
);
1102 obj
.type
= isl_obj_none
;
1107 static struct isl_obj
read_obj(struct isl_stream
*s
,
1108 struct isl_hash_table
*table
)
1110 struct isl_obj obj
= { isl_obj_none
, NULL
};
1112 struct isc_un_op
*op
= NULL
;
1114 if (isl_stream_eat_if_available(s
, '(')) {
1115 obj
= read_expr(s
, table
);
1116 if (!obj
.v
|| isl_stream_eat(s
, ')'))
1119 op
= read_prefix_un_op_if_available(s
);
1121 return read_un_op_expr(s
, table
, op
);
1123 if (isl_stream_eat_if_available(s
, read_op
))
1124 return read_from_file(s
);
1125 if (isl_stream_eat_if_available(s
, vertices_op
))
1126 return vertices(s
, table
);
1128 name
= isl_stream_read_ident_if_available(s
);
1130 obj
= stored_obj(s
->ctx
, table
, name
);
1132 obj
= isl_stream_read_obj(s
);
1137 if (isl_stream_eat_if_available(s
, '^'))
1138 obj
= power(s
, obj
);
1139 else if (obj
.type
== isl_obj_list
&& isl_stream_eat_if_available(s
, '['))
1140 obj
= obj_at_index(s
, obj
);
1141 else if (is_subtype(obj
, isl_obj_union_map
) &&
1142 isl_stream_eat_if_available(s
, '(')) {
1143 obj
= convert(obj
, isl_obj_union_map
);
1144 obj
= apply(s
, obj
.v
, table
);
1150 obj
.type
= isl_obj_none
;
1155 static struct isc_bin_op
*find_matching_bin_op(struct isc_bin_op
*like
,
1156 struct isl_obj lhs
, struct isl_obj rhs
)
1160 for (i
= 0; ; ++i
) {
1163 if (bin_ops
[i
].op
!= like
->op
)
1165 if (!is_subtype(lhs
, bin_ops
[i
].lhs
))
1167 if (!is_subtype(rhs
, bin_ops
[i
].rhs
))
1173 for (i
= 0; ; ++i
) {
1174 if (!named_bin_ops
[i
].name
)
1176 if (named_bin_ops
[i
].op
.op
!= like
->op
)
1178 if (!is_subtype(lhs
, named_bin_ops
[i
].op
.lhs
))
1180 if (!is_subtype(rhs
, named_bin_ops
[i
].op
.rhs
))
1183 return &named_bin_ops
[i
].op
;
1189 static struct isl_obj
read_expr(struct isl_stream
*s
,
1190 struct isl_hash_table
*table
)
1192 struct isl_obj obj
= { isl_obj_none
, NULL
};
1193 struct isl_obj right_obj
= { isl_obj_none
, NULL
};
1195 obj
= read_obj(s
, table
);
1197 struct isc_bin_op
*op
= NULL
;
1199 op
= read_bin_op_if_available(s
, obj
);
1203 right_obj
= read_obj(s
, table
);
1205 op
= find_matching_bin_op(op
, obj
, right_obj
);
1207 isl_assert(s
->ctx
, op
, goto error
);
1208 obj
= convert(obj
, op
->lhs
);
1209 right_obj
= convert(right_obj
, op
->rhs
);
1210 obj
.v
= op
->fn(obj
.v
, right_obj
.v
);
1216 free_obj(right_obj
);
1218 obj
.type
= isl_obj_none
;
1223 static __isl_give isl_printer
*read_line(struct isl_stream
*s
,
1224 struct isl_hash_table
*table
, __isl_take isl_printer
*p
)
1226 struct isl_obj obj
= { isl_obj_none
, NULL
};
1229 struct isc_bin_op
*op
= NULL
;
1233 if (isl_stream_is_empty(s
))
1236 assign
= is_assign(s
);
1238 lhs
= isl_stream_read_ident_if_available(s
);
1239 if (isl_stream_eat(s
, ISL_TOKEN_DEF
))
1243 obj
= read_expr(s
, table
);
1244 if (obj
.type
== isl_obj_none
|| obj
.v
== NULL
)
1246 if (isl_stream_eat(s
, ';'))
1250 if (do_assign(s
->ctx
, table
, lhs
, obj
))
1253 p
= obj
.type
->print(p
, obj
.v
);
1254 p
= isl_printer_end_line(p
);
1260 isl_stream_flush_tokens(s
);
1261 isl_stream_skip_line(s
);
1267 int free_cb(void **entry
, void *user
)
1269 struct isl_named_obj
*named
= *entry
;
1271 free_obj(named
->obj
);
1278 static void register_named_ops(struct isl_stream
*s
)
1282 read_op
= isl_stream_register_keyword(s
, "read");
1283 assert(read_op
!= ISL_TOKEN_ERROR
);
1284 vertices_op
= isl_stream_register_keyword(s
, "vertices");
1285 assert(vertices_op
!= ISL_TOKEN_ERROR
);
1287 for (i
= 0; ; ++i
) {
1288 if (!named_un_ops
[i
].name
)
1290 named_un_ops
[i
].op
.op
= isl_stream_register_keyword(s
,
1291 named_un_ops
[i
].name
);
1292 assert(named_un_ops
[i
].op
.op
!= ISL_TOKEN_ERROR
);
1295 for (i
= 0; ; ++i
) {
1296 if (!named_bin_ops
[i
].name
)
1298 named_bin_ops
[i
].op
.op
= isl_stream_register_keyword(s
,
1299 named_bin_ops
[i
].name
);
1300 assert(named_bin_ops
[i
].op
.op
!= ISL_TOKEN_ERROR
);
1304 int main(int argc
, char **argv
)
1306 struct isl_ctx
*ctx
;
1307 struct isl_stream
*s
;
1308 struct isl_hash_table
*table
;
1309 struct iscc_options
*options
;
1312 options
= iscc_options_new_with_defaults();
1314 argc
= iscc_options_parse(options
, argc
, argv
, ISL_ARG_ALL
);
1316 ctx
= isl_ctx_alloc_with_options(iscc_options_arg
, options
);
1317 s
= isl_stream_new_file(ctx
, stdin
);
1319 table
= isl_hash_table_alloc(ctx
, 10);
1321 p
= isl_printer_to_file(ctx
, stdout
);
1322 p
= isl_printer_set_output_format(p
, options
->format
);
1325 register_named_ops(s
);
1328 p
= read_line(s
, table
, p
);
1331 isl_printer_free(p
);
1332 isl_hash_table_foreach(ctx
, table
, free_cb
, NULL
);
1333 isl_hash_table_free(ctx
, table
);