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 static __isl_give isl_union_pw_qpolynomial_fold
*union_pw_qpolynomial_add_union_pw_qpolynomial_fold(
245 __isl_take isl_union_pw_qpolynomial
*upwqp
,
246 __isl_take isl_union_pw_qpolynomial_fold
*upwf
)
248 return isl_union_pw_qpolynomial_fold_add_union_pw_qpolynomial(upwf
,
252 static __isl_give
struct isl_list
*union_map_apply_union_pw_qpolynomial_fold(
253 __isl_take isl_union_map
*umap
,
254 __isl_take isl_union_pw_qpolynomial_fold
*upwf
)
257 struct isl_list
*list
;
260 ctx
= isl_union_map_get_ctx(umap
);
261 list
= isl_list_alloc(ctx
, 2);
265 list
->obj
[0].type
= isl_obj_union_pw_qpolynomial_fold
;
266 list
->obj
[0].v
= isl_union_map_apply_union_pw_qpolynomial_fold(umap
,
268 list
->obj
[1].type
= isl_obj_bool
;
269 list
->obj
[1].v
= tight
? &isl_bool_true
: &isl_bool_false
;
270 if (tight
< 0 || !list
->obj
[0].v
)
275 isl_union_map_free(umap
);
276 isl_union_pw_qpolynomial_fold_free(upwf
);
282 struct isc_bin_op bin_ops
[] = {
283 { '+', isl_obj_union_set
, isl_obj_union_set
,
285 (isc_bin_op_fn
) &isl_union_set_union
},
286 { '+', isl_obj_union_map
, isl_obj_union_map
,
288 (isc_bin_op_fn
) &isl_union_map_union
},
289 { '-', isl_obj_union_set
, isl_obj_union_set
,
291 (isc_bin_op_fn
) &isl_union_set_subtract
},
292 { '-', isl_obj_union_map
, isl_obj_union_map
,
294 (isc_bin_op_fn
) &isl_union_map_subtract
},
295 { '*', isl_obj_union_set
, isl_obj_union_set
,
297 (isc_bin_op_fn
) &isl_union_set_intersect
},
298 { '*', isl_obj_union_map
, isl_obj_union_map
,
300 (isc_bin_op_fn
) &isl_union_map_intersect
},
301 { '*', isl_obj_union_map
, isl_obj_union_set
,
303 (isc_bin_op_fn
) &isl_union_map_intersect_domain
},
304 { '.', isl_obj_union_map
, isl_obj_union_map
,
306 (isc_bin_op_fn
) &isl_union_map_apply_range
},
307 { '.', isl_obj_union_map
, isl_obj_union_pw_qpolynomial
,
308 isl_obj_union_pw_qpolynomial
,
309 (isc_bin_op_fn
) &isl_union_map_apply_union_pw_qpolynomial
},
310 { '.', isl_obj_union_map
, isl_obj_union_pw_qpolynomial_fold
,
312 (isc_bin_op_fn
) &union_map_apply_union_pw_qpolynomial_fold
},
313 { ISL_TOKEN_TO
, isl_obj_union_set
, isl_obj_union_set
,
315 (isc_bin_op_fn
) &isl_union_map_from_domain_and_range
},
316 { '=', isl_obj_union_set
, isl_obj_union_set
, isl_obj_bool
,
317 (isc_bin_op_fn
) &union_set_is_equal
},
318 { '=', isl_obj_union_map
, isl_obj_union_map
, isl_obj_bool
,
319 (isc_bin_op_fn
) &union_map_is_equal
},
320 { ISL_TOKEN_LE
, isl_obj_union_set
, isl_obj_union_set
,
321 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_subset
},
322 { ISL_TOKEN_LE
, isl_obj_union_map
, isl_obj_union_map
,
323 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_subset
},
324 { ISL_TOKEN_LT
, isl_obj_union_set
, isl_obj_union_set
,
325 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_strict_subset
},
326 { ISL_TOKEN_LT
, isl_obj_union_map
, isl_obj_union_map
,
327 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_strict_subset
},
328 { ISL_TOKEN_GE
, isl_obj_union_set
, isl_obj_union_set
,
329 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_superset
},
330 { ISL_TOKEN_GE
, isl_obj_union_map
, isl_obj_union_map
,
331 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_superset
},
332 { ISL_TOKEN_GT
, isl_obj_union_set
, isl_obj_union_set
,
333 isl_obj_bool
, (isc_bin_op_fn
) &union_set_is_strict_superset
},
334 { ISL_TOKEN_GT
, isl_obj_union_map
, isl_obj_union_map
,
335 isl_obj_bool
, (isc_bin_op_fn
) &union_map_is_strict_superset
},
336 { '.', isl_obj_union_pw_qpolynomial_fold
,
337 isl_obj_union_pw_qpolynomial_fold
,
338 isl_obj_union_pw_qpolynomial_fold
,
339 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_fold
},
340 { '+', isl_obj_union_pw_qpolynomial
, isl_obj_union_pw_qpolynomial
,
341 isl_obj_union_pw_qpolynomial
,
342 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_add
},
343 { '+', isl_obj_union_pw_qpolynomial
,
344 isl_obj_union_pw_qpolynomial_fold
,
345 isl_obj_union_pw_qpolynomial_fold
,
346 (isc_bin_op_fn
) &union_pw_qpolynomial_add_union_pw_qpolynomial_fold
},
347 { '+', isl_obj_union_pw_qpolynomial_fold
,
348 isl_obj_union_pw_qpolynomial
,
349 isl_obj_union_pw_qpolynomial_fold
,
350 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_add_union_pw_qpolynomial
},
351 { '-', isl_obj_union_pw_qpolynomial
, isl_obj_union_pw_qpolynomial
,
352 isl_obj_union_pw_qpolynomial
,
353 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_sub
},
354 { '*', isl_obj_union_pw_qpolynomial
, isl_obj_union_pw_qpolynomial
,
355 isl_obj_union_pw_qpolynomial
,
356 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_mul
},
357 { '*', isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
358 isl_obj_union_pw_qpolynomial
,
359 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_intersect_domain
},
360 { '*', isl_obj_union_pw_qpolynomial_fold
, isl_obj_union_set
,
361 isl_obj_union_pw_qpolynomial_fold
,
362 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_intersect_domain
},
363 { '@', isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
364 isl_obj_union_pw_qpolynomial
,
365 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_at
},
366 { '@', isl_obj_union_pw_qpolynomial_fold
, isl_obj_union_set
,
367 isl_obj_union_pw_qpolynomial
,
368 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_at
},
369 { '%', isl_obj_union_set
, isl_obj_union_set
,
371 (isc_bin_op_fn
) &isl_union_set_gist
},
372 { '%', isl_obj_union_map
, isl_obj_union_map
,
374 (isc_bin_op_fn
) &isl_union_map_gist
},
375 { '%', isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
376 isl_obj_union_pw_qpolynomial
,
377 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_gist
},
378 { '%', isl_obj_union_pw_qpolynomial_fold
, isl_obj_union_set
,
379 isl_obj_union_pw_qpolynomial_fold
,
380 (isc_bin_op_fn
) &isl_union_pw_qpolynomial_fold_gist
},
383 struct isc_named_bin_op named_bin_ops
[] = {
384 { "cross", { -1, isl_obj_union_set
, isl_obj_union_set
,
386 (isc_bin_op_fn
) &isl_union_set_product
} },
387 { "cross", { -1, isl_obj_union_map
, isl_obj_union_map
,
389 (isc_bin_op_fn
) &isl_union_map_product
} },
393 __isl_give isl_set
*union_set_sample(__isl_take isl_union_set
*uset
)
395 return isl_set_from_basic_set(isl_union_set_sample(uset
));
398 __isl_give isl_map
*union_map_sample(__isl_take isl_union_map
*umap
)
400 return isl_map_from_basic_map(isl_union_map_sample(umap
));
403 static __isl_give
struct isl_list
*union_pw_qpolynomial_upper_bound(
404 __isl_take isl_union_pw_qpolynomial
*upwqp
)
407 struct isl_list
*list
;
410 ctx
= isl_union_pw_qpolynomial_get_ctx(upwqp
);
411 list
= isl_list_alloc(ctx
, 2);
415 list
->obj
[0].type
= isl_obj_union_pw_qpolynomial_fold
;
416 list
->obj
[0].v
= isl_union_pw_qpolynomial_bound(upwqp
,
417 isl_fold_max
, &tight
);
418 list
->obj
[1].type
= isl_obj_bool
;
419 list
->obj
[1].v
= tight
? &isl_bool_true
: &isl_bool_false
;
420 if (tight
< 0 || !list
->obj
[0].v
)
425 isl_union_pw_qpolynomial_free(upwqp
);
432 void *map_codegen(void *arg
)
435 isl_union_map
*umap
= (isl_union_map
*)arg
;
436 isl_ctx
*ctx
= isl_union_map_get_ctx(umap
);
438 CloogOptions
*options
;
439 CloogDomain
*context
;
440 CloogUnionDomain
*ud
;
442 struct clast_stmt
*stmt
;
444 state
= cloog_isl_state_malloc(ctx
);
445 options
= cloog_options_malloc(state
);
446 options
->language
= LANGUAGE_C
;
447 options
->strides
= 1;
449 ud
= cloog_union_domain_from_isl_union_map(isl_union_map_copy(umap
));
451 dim
= isl_union_map_get_dim(umap
);
452 context
= cloog_domain_from_isl_set(isl_set_universe(dim
));
454 input
= cloog_input_alloc(context
, ud
);
456 stmt
= cloog_clast_create_from_input(input
, options
);
457 clast_pprint(stdout
, stmt
, 0, options
);
458 cloog_clast_free(stmt
);
461 cloog_options_free(options
);
462 cloog_state_free(state
);
463 isl_union_map_free(umap
);
467 void *set_codegen(void *arg
)
470 isl_union_set
*uset
= (isl_union_set
*)arg
;
471 isl_ctx
*ctx
= isl_union_set_get_ctx(uset
);
473 CloogOptions
*options
;
474 CloogDomain
*context
;
475 CloogUnionDomain
*ud
;
477 struct clast_stmt
*stmt
;
479 state
= cloog_isl_state_malloc(ctx
);
480 options
= cloog_options_malloc(state
);
481 options
->language
= LANGUAGE_C
;
482 options
->strides
= 1;
484 ud
= cloog_union_domain_from_isl_union_set(isl_union_set_copy(uset
));
486 dim
= isl_union_set_get_dim(uset
);
487 context
= cloog_domain_from_isl_set(isl_set_universe(dim
));
489 input
= cloog_input_alloc(context
, ud
);
491 stmt
= cloog_clast_create_from_input(input
, options
);
492 clast_pprint(stdout
, stmt
, 0, options
);
493 cloog_clast_free(stmt
);
496 cloog_options_free(options
);
497 cloog_state_free(state
);
498 isl_union_set_free(uset
);
503 typedef void *(*isc_un_op_fn
)(void *arg
);
505 enum isl_token_type op
;
510 struct isc_named_un_op
{
514 struct isc_named_un_op named_un_ops
[] = {
515 {"aff", { -1, isl_obj_union_map
, isl_obj_union_map
,
516 (isc_un_op_fn
) &isl_union_map_affine_hull
} },
517 {"aff", { -1, isl_obj_union_set
, isl_obj_union_set
,
518 (isc_un_op_fn
) &isl_union_set_affine_hull
} },
519 {"card", { -1, isl_obj_union_set
,
520 isl_obj_union_pw_qpolynomial
,
521 (isc_un_op_fn
) &isl_union_set_card
} },
522 {"card", { -1, isl_obj_union_map
,
523 isl_obj_union_pw_qpolynomial
,
524 (isc_un_op_fn
) &isl_union_map_card
} },
525 {"coalesce", { -1, isl_obj_union_set
, isl_obj_union_set
,
526 (isc_un_op_fn
) &isl_union_set_coalesce
} },
527 {"coalesce", { -1, isl_obj_union_map
, isl_obj_union_map
,
528 (isc_un_op_fn
) &isl_union_map_coalesce
} },
529 {"coalesce", { -1, isl_obj_union_pw_qpolynomial
,
530 isl_obj_union_pw_qpolynomial
,
531 (isc_un_op_fn
) &isl_union_pw_qpolynomial_coalesce
} },
532 {"coalesce", { -1, isl_obj_union_pw_qpolynomial_fold
,
533 isl_obj_union_pw_qpolynomial_fold
,
534 (isc_un_op_fn
) &isl_union_pw_qpolynomial_fold_coalesce
} },
536 {"codegen", { -1, isl_obj_union_set
, isl_obj_none
,
538 {"codegen", { -1, isl_obj_union_map
, isl_obj_none
,
541 {"deltas", { -1, isl_obj_union_map
, isl_obj_union_set
,
542 (isc_un_op_fn
) &isl_union_map_deltas
} },
543 {"dom", { -1, isl_obj_union_map
, isl_obj_union_set
,
544 (isc_un_op_fn
) &isl_union_map_domain
} },
545 {"dom", { -1, isl_obj_union_pw_qpolynomial
, isl_obj_union_set
,
546 (isc_un_op_fn
) &isl_union_pw_qpolynomial_domain
} },
547 {"dom", { -1, isl_obj_union_pw_qpolynomial_fold
,
549 (isc_un_op_fn
) &isl_union_pw_qpolynomial_fold_domain
} },
550 {"ran", { -1, isl_obj_union_map
, isl_obj_union_set
,
551 (isc_un_op_fn
) &isl_union_map_range
} },
552 {"lexmin", { -1, isl_obj_union_map
, isl_obj_union_map
,
553 (isc_un_op_fn
) &isl_union_map_lexmin
} },
554 {"lexmax", { -1, isl_obj_union_map
, isl_obj_union_map
,
555 (isc_un_op_fn
) &isl_union_map_lexmax
} },
556 {"lexmin", { -1, isl_obj_union_set
, isl_obj_union_set
,
557 (isc_un_op_fn
) &isl_union_set_lexmin
} },
558 {"lexmax", { -1, isl_obj_union_set
, isl_obj_union_set
,
559 (isc_un_op_fn
) &isl_union_set_lexmax
} },
560 {"sample", { -1, isl_obj_union_set
, isl_obj_set
,
561 (isc_un_op_fn
) &union_set_sample
} },
562 {"sample", { -1, isl_obj_union_map
, isl_obj_map
,
563 (isc_un_op_fn
) &union_map_sample
} },
564 {"sum", { -1, isl_obj_union_pw_qpolynomial
,
565 isl_obj_union_pw_qpolynomial
,
566 (isc_un_op_fn
) &isl_union_pw_qpolynomial_sum
} },
567 {"ub", { -1, isl_obj_union_pw_qpolynomial
, isl_obj_list
,
568 (isc_un_op_fn
) &union_pw_qpolynomial_upper_bound
} },
569 {"unwrap", { -1, isl_obj_union_set
, isl_obj_union_map
,
570 (isc_un_op_fn
) &isl_union_set_unwrap
} },
571 {"wrap", { -1, isl_obj_union_map
, isl_obj_union_set
,
572 (isc_un_op_fn
) &isl_union_map_wrap
} },
576 struct isl_named_obj
{
581 static void free_obj(struct isl_obj obj
)
583 obj
.type
->free(obj
.v
);
586 static int same_name(const void *entry
, const void *val
)
588 const struct isl_named_obj
*named
= (const struct isl_named_obj
*)entry
;
590 return !strcmp(named
->name
, val
);
593 static int do_assign(struct isl_ctx
*ctx
, struct isl_hash_table
*table
,
594 char *name
, struct isl_obj obj
)
596 struct isl_hash_table_entry
*entry
;
598 struct isl_named_obj
*named
;
600 name_hash
= isl_hash_string(isl_hash_init(), name
);
601 entry
= isl_hash_table_find(ctx
, table
, name_hash
, same_name
, name
, 1);
606 free_obj(named
->obj
);
609 named
= isl_alloc_type(ctx
, struct isl_named_obj
);
624 static struct isl_obj
stored_obj(struct isl_ctx
*ctx
,
625 struct isl_hash_table
*table
, char *name
)
627 struct isl_obj obj
= { isl_obj_none
, NULL
};
628 struct isl_hash_table_entry
*entry
;
631 name_hash
= isl_hash_string(isl_hash_init(), name
);
632 entry
= isl_hash_table_find(ctx
, table
, name_hash
, same_name
, name
, 0);
634 struct isl_named_obj
*named
;
640 obj
.v
= obj
.type
->copy(obj
.v
);
644 static int is_subtype(struct isl_obj obj
, isl_obj_type super
)
646 if (obj
.type
== super
)
648 if (obj
.type
== isl_obj_map
&& super
== isl_obj_union_map
)
650 if (obj
.type
== isl_obj_set
&& super
== isl_obj_union_set
)
652 if (obj
.type
== isl_obj_pw_qpolynomial
&&
653 super
== isl_obj_union_pw_qpolynomial
)
655 if (obj
.type
== isl_obj_pw_qpolynomial_fold
&&
656 super
== isl_obj_union_pw_qpolynomial_fold
)
658 if (obj
.type
== isl_obj_union_set
&& isl_union_set_is_empty(obj
.v
))
663 static struct isl_obj
convert(struct isl_obj obj
, isl_obj_type type
)
665 if (obj
.type
== type
)
667 if (obj
.type
== isl_obj_map
&& type
== isl_obj_union_map
) {
668 obj
.type
= isl_obj_union_map
;
669 obj
.v
= isl_union_map_from_map(obj
.v
);
672 if (obj
.type
== isl_obj_set
&& type
== isl_obj_union_set
) {
673 obj
.type
= isl_obj_union_set
;
674 obj
.v
= isl_union_set_from_set(obj
.v
);
677 if (obj
.type
== isl_obj_pw_qpolynomial
&&
678 type
== isl_obj_union_pw_qpolynomial
) {
679 obj
.type
= isl_obj_union_pw_qpolynomial
;
680 obj
.v
= isl_union_pw_qpolynomial_from_pw_qpolynomial(obj
.v
);
683 if (obj
.type
== isl_obj_pw_qpolynomial_fold
&&
684 type
== isl_obj_union_pw_qpolynomial_fold
) {
685 obj
.type
= isl_obj_union_pw_qpolynomial_fold
;
686 obj
.v
= isl_union_pw_qpolynomial_fold_from_pw_qpolynomial_fold(obj
.v
);
689 if (obj
.type
== isl_obj_union_set
&& isl_union_set_is_empty(obj
.v
)) {
690 if (type
== isl_obj_union_map
) {
691 obj
.type
= isl_obj_union_map
;
694 if (type
== isl_obj_union_pw_qpolynomial
) {
695 isl_dim
*dim
= isl_union_set_get_dim(obj
.v
);
696 isl_union_set_free(obj
.v
);
697 obj
.v
= isl_union_pw_qpolynomial_zero(dim
);
698 obj
.type
= isl_obj_union_pw_qpolynomial
;
701 if (type
== isl_obj_union_pw_qpolynomial_fold
) {
702 isl_dim
*dim
= isl_union_set_get_dim(obj
.v
);
703 isl_union_set_free(obj
.v
);
704 obj
.v
= isl_union_pw_qpolynomial_fold_zero(dim
,
706 obj
.type
= isl_obj_union_pw_qpolynomial_fold
;
711 obj
.type
= isl_obj_none
;
716 static struct isc_bin_op
*read_bin_op_if_available(struct isl_stream
*s
,
720 struct isl_token
*tok
;
722 tok
= isl_stream_next_token(s
);
729 if (bin_ops
[i
].op
!= tok
->type
)
731 if (!is_subtype(lhs
, bin_ops
[i
].lhs
))
739 if (!named_bin_ops
[i
].name
)
741 if (named_bin_ops
[i
].op
.op
!= tok
->type
)
743 if (!is_subtype(lhs
, named_bin_ops
[i
].op
.lhs
))
747 return &named_bin_ops
[i
].op
;
750 isl_stream_push_token(s
, tok
);
755 static struct isc_un_op
*read_prefix_un_op_if_available(struct isl_stream
*s
)
758 struct isl_token
*tok
;
760 tok
= isl_stream_next_token(s
);
765 if (!named_un_ops
[i
].name
)
767 if (named_un_ops
[i
].op
.op
!= tok
->type
)
771 return &named_un_ops
[i
].op
;
774 isl_stream_push_token(s
, tok
);
779 static struct isc_un_op
*find_matching_un_op(struct isc_un_op
*like
,
785 if (!named_un_ops
[i
].name
)
787 if (named_un_ops
[i
].op
.op
!= like
->op
)
789 if (!is_subtype(arg
, named_un_ops
[i
].op
.arg
))
792 return &named_un_ops
[i
].op
;
798 static int is_assign(struct isl_stream
*s
)
800 struct isl_token
*tok
;
801 struct isl_token
*tok2
;
804 tok
= isl_stream_next_token(s
);
807 if (tok
->type
!= ISL_TOKEN_IDENT
) {
808 isl_stream_push_token(s
, tok
);
812 tok2
= isl_stream_next_token(s
);
814 isl_stream_push_token(s
, tok
);
817 assign
= tok2
->type
== ISL_TOKEN_DEF
;
818 isl_stream_push_token(s
, tok2
);
819 isl_stream_push_token(s
, tok
);
824 static struct isl_obj
read_obj(struct isl_stream
*s
,
825 struct isl_hash_table
*table
);
826 static struct isl_obj
read_expr(struct isl_stream
*s
,
827 struct isl_hash_table
*table
);
829 static struct isl_obj
read_un_op_expr(struct isl_stream
*s
,
830 struct isl_hash_table
*table
, struct isc_un_op
*op
)
832 struct isl_obj obj
= { isl_obj_none
, NULL
};
834 obj
= read_obj(s
, table
);
838 op
= find_matching_un_op(op
, obj
);
840 isl_assert(s
->ctx
, op
, goto error
);
841 obj
= convert(obj
, op
->arg
);
842 obj
.v
= op
->fn(obj
.v
);
848 obj
.type
= isl_obj_none
;
853 static struct isl_obj
transitive_closure(struct isl_ctx
*ctx
, struct isl_obj obj
)
855 struct isl_list
*list
;
858 if (obj
.type
== isl_obj_map
)
859 obj
= convert(obj
, isl_obj_union_map
);
860 isl_assert(ctx
, obj
.type
== isl_obj_union_map
, goto error
);
861 list
= isl_list_alloc(ctx
, 2);
865 list
->obj
[0].type
= isl_obj_union_map
;
866 list
->obj
[0].v
= isl_union_map_transitive_closure(obj
.v
, &exact
);
867 list
->obj
[1].type
= isl_obj_bool
;
868 list
->obj
[1].v
= exact
? &isl_bool_true
: &isl_bool_false
;
870 obj
.type
= isl_obj_list
;
871 if (exact
< 0 || !list
->obj
[0].v
)
877 obj
.type
= isl_obj_none
;
882 static struct isl_obj
obj_at_index(struct isl_stream
*s
, struct isl_obj obj
)
884 struct isl_list
*list
= obj
.v
;
885 struct isl_token
*tok
;
888 tok
= isl_stream_next_token(s
);
889 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
) {
890 isl_stream_error(s
, tok
, "expecting index");
892 isl_stream_push_token(s
, tok
);
895 i
= isl_int_get_si(tok
->u
.v
);
897 isl_assert(s
->ctx
, i
< list
->n
, goto error
);
898 if (isl_stream_eat(s
, ']'))
902 obj
.v
= obj
.type
->copy(obj
.v
);
909 obj
.type
= isl_obj_none
;
914 static struct isl_obj
apply(struct isl_stream
*s
, __isl_take isl_union_map
*umap
,
915 struct isl_hash_table
*table
)
919 obj
= read_expr(s
, table
);
920 isl_assert(s
->ctx
, is_subtype(obj
, isl_obj_union_set
) ||
921 is_subtype(obj
, isl_obj_union_map
), goto error
);
923 if (obj
.type
== isl_obj_set
)
924 obj
= convert(obj
, isl_obj_union_set
);
925 else if (obj
.type
== isl_obj_map
)
926 obj
= convert(obj
, isl_obj_union_map
);
927 if (obj
.type
== isl_obj_union_set
) {
928 obj
.v
= isl_union_set_apply(obj
.v
, umap
);
930 obj
.v
= isl_union_map_apply_range(obj
.v
, umap
);
934 if (isl_stream_eat(s
, ')'))
939 isl_union_map_free(umap
);
942 obj
.type
= isl_obj_none
;
947 struct add_vertex_data
{
948 struct isl_list
*list
;
952 static int add_vertex(__isl_take isl_vertex
*vertex
, void *user
)
954 struct add_vertex_data
*data
= (struct add_vertex_data
*)user
;
957 expr
= isl_vertex_get_expr(vertex
);
959 data
->list
->obj
[data
->i
].type
= isl_obj_set
;
960 data
->list
->obj
[data
->i
].v
= isl_set_from_basic_set(expr
);
963 isl_vertex_free(vertex
);
968 static struct isl_obj
vertices(struct isl_stream
*s
,
969 struct isl_hash_table
*table
)
975 isl_vertices
*vertices
= NULL
;
976 struct isl_list
*list
= NULL
;
977 struct add_vertex_data data
;
979 obj
= read_expr(s
, table
);
980 isl_assert(s
->ctx
, obj
.type
== isl_obj_set
, goto error
);
983 set
= isl_set_remove_divs(set
);
984 hull
= isl_set_convex_hull(set
);
985 vertices
= isl_basic_set_compute_vertices(hull
);
986 isl_basic_set_free(hull
);
988 ctx
= isl_vertices_get_ctx(vertices
);
989 list
= isl_list_alloc(ctx
, isl_vertices_get_n_vertices(vertices
));
996 if (isl_vertices_foreach_vertex(vertices
, &add_vertex
, &data
) < 0)
999 isl_vertices_free(vertices
);
1001 obj
.type
= isl_obj_list
;
1006 isl_vertices_free(vertices
);
1007 isl_list_free(list
);
1009 obj
.type
= isl_obj_none
;
1014 static struct isl_obj
power(struct isl_stream
*s
, struct isl_obj obj
)
1016 struct isl_token
*tok
;
1018 if (isl_stream_eat_if_available(s
, '+'))
1019 return transitive_closure(s
->ctx
, obj
);
1021 tok
= isl_stream_next_token(s
);
1022 if (!tok
|| tok
->type
!= ISL_TOKEN_VALUE
|| isl_int_cmp_si(tok
->u
.v
, -1)) {
1023 isl_stream_error(s
, tok
, "expecting -1");
1025 isl_stream_push_token(s
, tok
);
1028 isl_token_free(tok
);
1029 isl_assert(s
->ctx
, is_subtype(obj
, isl_obj_union_map
), goto error
);
1030 if (obj
.type
!= isl_obj_union_map
)
1031 obj
= convert(obj
, isl_obj_union_map
);
1033 obj
.v
= isl_union_map_reverse(obj
.v
);
1040 obj
.type
= isl_obj_none
;
1045 static struct isl_obj
read_from_file(struct isl_stream
*s
)
1048 struct isl_token
*tok
;
1049 struct isl_stream
*s_file
;
1052 tok
= isl_stream_next_token(s
);
1053 if (!tok
|| tok
->type
!= ISL_TOKEN_STRING
) {
1054 isl_stream_error(s
, tok
, "expecting filename");
1055 isl_token_free(tok
);
1059 file
= fopen(tok
->u
.s
, "r");
1060 isl_token_free(tok
);
1061 isl_assert(s
->ctx
, file
, goto error
);
1063 s_file
= isl_stream_new_file(s
->ctx
, file
);
1069 obj
= isl_stream_read_obj(s_file
);
1071 isl_stream_free(s_file
);
1076 obj
.type
= isl_obj_none
;
1081 static struct isl_obj
read_obj(struct isl_stream
*s
,
1082 struct isl_hash_table
*table
)
1084 struct isl_obj obj
= { isl_obj_none
, NULL
};
1086 struct isc_un_op
*op
= NULL
;
1088 if (isl_stream_eat_if_available(s
, '(')) {
1089 obj
= read_expr(s
, table
);
1090 if (!obj
.v
|| isl_stream_eat(s
, ')'))
1093 op
= read_prefix_un_op_if_available(s
);
1095 return read_un_op_expr(s
, table
, op
);
1097 if (isl_stream_eat_if_available(s
, read_op
))
1098 return read_from_file(s
);
1099 if (isl_stream_eat_if_available(s
, vertices_op
))
1100 return vertices(s
, table
);
1102 name
= isl_stream_read_ident_if_available(s
);
1104 obj
= stored_obj(s
->ctx
, table
, name
);
1106 obj
= isl_stream_read_obj(s
);
1111 if (isl_stream_eat_if_available(s
, '^'))
1112 obj
= power(s
, obj
);
1113 else if (obj
.type
== isl_obj_list
&& isl_stream_eat_if_available(s
, '['))
1114 obj
= obj_at_index(s
, obj
);
1115 else if (is_subtype(obj
, isl_obj_union_map
) &&
1116 isl_stream_eat_if_available(s
, '(')) {
1117 obj
= convert(obj
, isl_obj_union_map
);
1118 obj
= apply(s
, obj
.v
, table
);
1124 obj
.type
= isl_obj_none
;
1129 static struct isc_bin_op
*find_matching_bin_op(struct isc_bin_op
*like
,
1130 struct isl_obj lhs
, struct isl_obj rhs
)
1134 for (i
= 0; ; ++i
) {
1137 if (bin_ops
[i
].op
!= like
->op
)
1139 if (!is_subtype(lhs
, bin_ops
[i
].lhs
))
1141 if (!is_subtype(rhs
, bin_ops
[i
].rhs
))
1147 for (i
= 0; ; ++i
) {
1148 if (!named_bin_ops
[i
].name
)
1150 if (named_bin_ops
[i
].op
.op
!= like
->op
)
1152 if (!is_subtype(lhs
, named_bin_ops
[i
].op
.lhs
))
1154 if (!is_subtype(rhs
, named_bin_ops
[i
].op
.rhs
))
1157 return &named_bin_ops
[i
].op
;
1163 static struct isl_obj
read_expr(struct isl_stream
*s
,
1164 struct isl_hash_table
*table
)
1166 struct isl_obj obj
= { isl_obj_none
, NULL
};
1167 struct isl_obj right_obj
= { isl_obj_none
, NULL
};
1169 obj
= read_obj(s
, table
);
1171 struct isc_bin_op
*op
= NULL
;
1173 op
= read_bin_op_if_available(s
, obj
);
1177 right_obj
= read_obj(s
, table
);
1179 op
= find_matching_bin_op(op
, obj
, right_obj
);
1181 isl_assert(s
->ctx
, op
, goto error
);
1182 obj
= convert(obj
, op
->lhs
);
1183 right_obj
= convert(right_obj
, op
->rhs
);
1184 obj
.v
= op
->fn(obj
.v
, right_obj
.v
);
1190 free_obj(right_obj
);
1192 obj
.type
= isl_obj_none
;
1197 static __isl_give isl_printer
*read_line(struct isl_stream
*s
,
1198 struct isl_hash_table
*table
, __isl_take isl_printer
*p
)
1200 struct isl_obj obj
= { isl_obj_none
, NULL
};
1203 struct isc_bin_op
*op
= NULL
;
1207 if (isl_stream_is_empty(s
))
1210 assign
= is_assign(s
);
1212 lhs
= isl_stream_read_ident_if_available(s
);
1213 if (isl_stream_eat(s
, ISL_TOKEN_DEF
))
1217 obj
= read_expr(s
, table
);
1218 if (obj
.type
== isl_obj_none
|| obj
.v
== NULL
)
1220 if (isl_stream_eat(s
, ';'))
1224 if (do_assign(s
->ctx
, table
, lhs
, obj
))
1227 p
= obj
.type
->print(p
, obj
.v
);
1228 p
= isl_printer_end_line(p
);
1234 isl_stream_flush_tokens(s
);
1235 isl_stream_skip_line(s
);
1241 int free_cb(void **entry
, void *user
)
1243 struct isl_named_obj
*named
= *entry
;
1245 free_obj(named
->obj
);
1252 static void register_named_ops(struct isl_stream
*s
)
1256 read_op
= isl_stream_register_keyword(s
, "read");
1257 assert(read_op
!= ISL_TOKEN_ERROR
);
1258 vertices_op
= isl_stream_register_keyword(s
, "vertices");
1259 assert(vertices_op
!= ISL_TOKEN_ERROR
);
1261 for (i
= 0; ; ++i
) {
1262 if (!named_un_ops
[i
].name
)
1264 named_un_ops
[i
].op
.op
= isl_stream_register_keyword(s
,
1265 named_un_ops
[i
].name
);
1266 assert(named_un_ops
[i
].op
.op
!= ISL_TOKEN_ERROR
);
1269 for (i
= 0; ; ++i
) {
1270 if (!named_bin_ops
[i
].name
)
1272 named_bin_ops
[i
].op
.op
= isl_stream_register_keyword(s
,
1273 named_bin_ops
[i
].name
);
1274 assert(named_bin_ops
[i
].op
.op
!= ISL_TOKEN_ERROR
);
1278 int main(int argc
, char **argv
)
1280 struct isl_ctx
*ctx
;
1281 struct isl_stream
*s
;
1282 struct isl_hash_table
*table
;
1283 struct iscc_options
*options
;
1286 options
= iscc_options_new_with_defaults();
1288 argc
= iscc_options_parse(options
, argc
, argv
, ISL_ARG_ALL
);
1290 ctx
= isl_ctx_alloc_with_options(iscc_options_arg
, options
);
1291 s
= isl_stream_new_file(ctx
, stdin
);
1293 table
= isl_hash_table_alloc(ctx
, 10);
1295 p
= isl_printer_to_file(ctx
, stdout
);
1296 p
= isl_printer_set_output_format(p
, options
->format
);
1299 register_named_ops(s
);
1302 p
= read_line(s
, table
, p
);
1305 isl_printer_free(p
);
1306 isl_hash_table_foreach(ctx
, table
, free_cb
, NULL
);
1307 isl_hash_table_free(ctx
, table
);