isl_ast_codegen.c: create_node_scaled: inline call to isl_ast_graft_alloc_level
[isl.git] / codegen.c
blobe9cd692fe8d7ca44bb5e7184023b5791cbc9b76b
1 /*
2 * Copyright 2012 Ecole Normale Superieure
4 * Use of this software is governed by the MIT license
6 * Written by Sven Verdoolaege,
7 * Ecole Normale Superieure, 45 rue d’Ulm, 75230 Paris, France
8 */
10 /* This program prints an AST that scans the domain elements of
11 * the domain of a given schedule in the order of their image(s).
13 * The input consists of three sets/relations.
14 * - a schedule
15 * - a context
16 * - a relation describing AST generation options
19 #include <assert.h>
20 #include <isl/ast.h>
21 #include <isl/ast_build.h>
22 #include <isl/options.h>
23 #include <isl/set.h>
25 struct options {
26 struct isl_options *isl;
27 unsigned atomic;
28 unsigned separate;
31 ISL_ARGS_START(struct options, options_args)
32 ISL_ARG_CHILD(struct options, isl, "isl", &isl_options_args, "isl options")
33 ISL_ARG_BOOL(struct options, atomic, 0, "atomic", 0,
34 "globally set the atomic option")
35 ISL_ARG_BOOL(struct options, separate, 0, "separate", 0,
36 "globally set the separate option")
37 ISL_ARGS_END
39 ISL_ARG_DEF(options, struct options, options_args)
41 /* Return a universal, 1-dimensional set with the given name.
43 static __isl_give isl_union_set *universe(isl_ctx *ctx, const char *name)
45 isl_space *space;
47 space = isl_space_set_alloc(ctx, 0, 1);
48 space = isl_space_set_tuple_name(space, isl_dim_set, name);
49 return isl_union_set_from_set(isl_set_universe(space));
52 /* Set the "name" option for the entire schedule domain.
54 static __isl_give isl_union_map *set_universe(__isl_take isl_union_map *opt,
55 __isl_keep isl_union_map *schedule, const char *name)
57 isl_ctx *ctx;
58 isl_union_set *domain, *target;
59 isl_union_map *option;
61 ctx = isl_union_map_get_ctx(opt);
63 domain = isl_union_map_range(isl_union_map_copy(schedule));
64 domain = isl_union_set_universe(domain);
65 target = universe(ctx, name);
66 option = isl_union_map_from_domain_and_range(domain, target);
67 opt = isl_union_map_union(opt, option);
69 return opt;
72 /* Update the build options based on the user-specified options.
74 * If the --separate or --atomic options were specified, then
75 * we clear any separate or atomic options that may already exist in "opt".
77 static __isl_give isl_ast_build *set_options(__isl_take isl_ast_build *build,
78 __isl_take isl_union_map *opt, struct options *options,
79 __isl_keep isl_union_map *schedule)
81 if (options->separate || options->atomic) {
82 isl_ctx *ctx;
83 isl_union_set *target;
85 ctx = isl_union_map_get_ctx(schedule);
87 target = universe(ctx, "separate");
88 opt = isl_union_map_subtract_range(opt, target);
89 target = universe(ctx, "atomic");
90 opt = isl_union_map_subtract_range(opt, target);
93 if (options->separate)
94 opt = set_universe(opt, schedule, "separate");
95 if (options->atomic)
96 opt = set_universe(opt, schedule, "atomic");
98 build = isl_ast_build_set_options(build, opt);
100 return build;
103 int main(int argc, char **argv)
105 isl_ctx *ctx;
106 isl_set *context;
107 isl_union_map *schedule;
108 isl_union_map *options_map;
109 isl_ast_build *build;
110 isl_ast_node *tree;
111 struct options *options;
112 isl_printer *p;
114 options = options_new_with_defaults();
115 assert(options);
116 argc = options_parse(options, argc, argv, ISL_ARG_ALL);
118 ctx = isl_ctx_alloc_with_options(&options_args, options);
120 schedule = isl_union_map_read_from_file(ctx, stdin);
121 context = isl_set_read_from_file(ctx, stdin);
122 options_map = isl_union_map_read_from_file(ctx, stdin);
124 build = isl_ast_build_from_context(context);
125 build = set_options(build, options_map, options, schedule);
126 tree = isl_ast_build_ast_from_schedule(build, schedule);
127 isl_ast_build_free(build);
129 p = isl_printer_to_file(ctx, stdout);
130 p = isl_printer_set_output_format(p, ISL_FORMAT_C);
131 p = isl_printer_print_ast_node(p, tree);
132 isl_printer_free(p);
134 isl_ast_node_free(tree);
136 isl_ctx_free(ctx);
137 return 0;