3 #include <clang/Basic/SourceManager.h>
4 #include <clang/AST/Decl.h>
5 #include <clang/AST/Stmt.h>
6 #include <clang/Lex/Preprocessor.h>
13 /* The location of the scop, as delimited by scop and endscop
14 * pragmas by the user.
24 clang::Preprocessor
&PP
;
25 clang::ASTContext
&ast_context
;
26 /* If autodetect is false, then loc contains the location
27 * of the scop to be extracted.
31 /* The sequence number of the next statement. */
33 /* The sequence number of the next virtual scalar. */
35 /* If autodetect is false, a scop delimited by pragmas is extracted,
36 * otherwise we take any scop that we can find.
39 /* Set if the pet_scop returned by an extract method only
40 * represents part of the input tree.
43 /* Set is nested accesses are allowed in general.
44 * This currently defaults to true.
47 /* Set if nested accesses are allowed in that part of the tree
48 * that is currently under investigation.
51 /* Maps identifiers to the last value that was assigned to them.
52 * If an identifier is mapped to NULL, then something may have
53 * been assigned, but we don't know what.
54 * assigned_value does not take a reference to the isl_pw_aff
55 * object, so each such isl_pw_aff needs to be stored in
56 * the set of "expressions".
58 std::map
<clang::ValueDecl
*, isl_pw_aff
*> assigned_value
;
59 /* A collection of isl_pw_affs used in assigned_value or other
60 * temporary maps. expressions holds a reference for each
61 * isl_pw_aff, which is freed in the destructor of PetScan.
63 std::set
<isl_pw_aff
*> expressions
;
65 /* A union of mappings of the form
66 * { identifier[] -> [i] : lower_bound <= i <= upper_bound }
68 isl_union_map
*value_bounds
;
70 PetScan(clang::Preprocessor
&PP
,
71 clang::ASTContext
&ast_context
, ScopLoc
&loc
, int autodetect
,
72 __isl_take isl_union_map
*value_bounds
) :
73 ctx(isl_union_map_get_ctx(value_bounds
)), PP(PP
),
74 ast_context(ast_context
), loc(loc
),
75 autodetect(autodetect
), value_bounds(value_bounds
),
76 n_stmt(0), n_test(0), partial(0), allow_nested(true),
77 nesting_enabled(false) { }
81 struct pet_scop
*scan(clang::FunctionDecl
*fd
);
83 static int extract_int(clang::IntegerLiteral
*expr
, isl_int
*v
);
85 void insert_expression(__isl_take isl_pw_aff
*expr
);
86 struct pet_scop
*scan(clang::Stmt
*stmt
);
88 struct pet_scop
*scan_arrays(struct pet_scop
*scop
);
89 struct pet_array
*extract_array(isl_ctx
*ctx
, clang::ValueDecl
*decl
);
90 struct pet_array
*set_upper_bounds(struct pet_array
*array
,
91 const clang::Type
*type
, int pos
);
93 struct pet_scop
*extract_non_affine_condition(clang::Expr
*cond
,
94 __isl_take isl_map
*access
);
96 struct pet_scop
*extract_conditional_assignment(clang::IfStmt
*stmt
);
98 struct pet_scop
*extract(clang::Stmt
*stmt
);
99 struct pet_scop
*extract(clang::StmtRange stmt_range
);
100 struct pet_scop
*extract(clang::IfStmt
*stmt
);
101 struct pet_scop
*extract(clang::WhileStmt
*stmt
);
102 struct pet_scop
*extract(clang::CompoundStmt
*stmt
);
103 struct pet_scop
*extract(clang::LabelStmt
*stmt
);
105 struct pet_scop
*extract(clang::Stmt
*stmt
, struct pet_expr
*expr
,
106 __isl_take isl_id
*label
= NULL
);
108 clang::BinaryOperator
*initialization_assignment(clang::Stmt
*init
);
109 clang::Decl
*initialization_declaration(clang::Stmt
*init
);
110 clang::ValueDecl
*extract_induction_variable(clang::BinaryOperator
*stmt
);
111 clang::VarDecl
*extract_induction_variable(clang::Stmt
*init
,
113 bool check_unary_increment(clang::UnaryOperator
*op
,
114 clang::ValueDecl
*iv
, isl_int
&inc
);
115 bool check_binary_increment(clang::BinaryOperator
*op
,
116 clang::ValueDecl
*iv
, isl_int
&inc
);
117 bool check_compound_increment(clang::CompoundAssignOperator
*op
,
118 clang::ValueDecl
*iv
, isl_int
&inc
);
119 bool check_increment(clang::ForStmt
*stmt
, clang::ValueDecl
*iv
,
121 struct pet_scop
*extract_for(clang::ForStmt
*stmt
);
122 struct pet_scop
*extract_infinite_loop(clang::Stmt
*body
);
123 struct pet_scop
*extract_infinite_for(clang::ForStmt
*stmt
);
125 void mark_write(struct pet_expr
*access
);
126 struct pet_expr
*extract_expr(clang::Expr
*expr
);
127 struct pet_expr
*extract_expr(clang::UnaryOperator
*expr
);
128 struct pet_expr
*extract_expr(clang::BinaryOperator
*expr
);
129 struct pet_expr
*extract_expr(clang::ImplicitCastExpr
*expr
);
130 struct pet_expr
*extract_expr(clang::FloatingLiteral
*expr
);
131 struct pet_expr
*extract_expr(clang::ParenExpr
*expr
);
132 struct pet_expr
*extract_expr(clang::ConditionalOperator
*expr
);
133 struct pet_expr
*extract_expr(clang::CallExpr
*expr
);
135 int extract_nested(__isl_keep isl_space
*space
,
136 int n_arg
, struct pet_expr
**args
,
137 std::map
<int,int> ¶m2pos
);
138 struct pet_expr
*extract_nested(struct pet_expr
*expr
, int n
,
139 std::map
<int,int> ¶m2pos
);
140 struct pet_stmt
*extract_nested(struct pet_stmt
*stmt
, int n
,
141 std::map
<int,int> ¶m2pos
);
142 struct pet_expr
*resolve_nested(struct pet_expr
*expr
);
143 struct pet_scop
*resolve_nested(struct pet_scop
*scop
);
144 struct pet_stmt
*resolve_nested(struct pet_stmt
*stmt
);
145 struct pet_expr
*extract_access_expr(clang::Expr
*expr
);
147 __isl_give isl_map
*extract_access(clang::ArraySubscriptExpr
*expr
);
148 __isl_give isl_map
*extract_access(clang::Expr
*expr
);
149 __isl_give isl_map
*extract_access(clang::ImplicitCastExpr
*expr
);
150 __isl_give isl_map
*extract_access(clang::DeclRefExpr
*expr
);
151 __isl_give isl_map
*extract_access(clang::IntegerLiteral
*expr
);
153 int extract_int(clang::Expr
*expr
, isl_int
*v
);
154 int extract_int(clang::ParenExpr
*expr
, isl_int
*v
);
156 __isl_give isl_pw_aff
*extract_affine_add(clang::BinaryOperator
*expr
);
157 __isl_give isl_pw_aff
*extract_affine_div(clang::BinaryOperator
*expr
);
158 __isl_give isl_pw_aff
*extract_affine_mod(clang::BinaryOperator
*expr
);
159 __isl_give isl_pw_aff
*extract_affine_mul(clang::BinaryOperator
*expr
);
161 isl_pw_aff
*nested_access(clang::Expr
*expr
);
163 __isl_give isl_pw_aff
*try_extract_affine(clang::Expr
*expr
);
164 bool is_affine(clang::Expr
*expr
);
165 bool is_affine_condition(clang::Expr
*expr
);
166 __isl_give isl_set
*try_extract_nested_condition(clang::Expr
*expr
);
167 bool is_nested_allowed(__isl_keep isl_set
*set
, pet_scop
*scop
);
169 __isl_give isl_pw_aff
*extract_affine(const llvm::APInt
&val
);
170 __isl_give isl_pw_aff
*extract_affine(clang::Expr
*expr
);
171 __isl_give isl_pw_aff
*extract_affine(clang::IntegerLiteral
*expr
);
172 __isl_give isl_pw_aff
*extract_affine(clang::ImplicitCastExpr
*expr
);
173 __isl_give isl_pw_aff
*extract_affine(clang::DeclRefExpr
*expr
);
174 __isl_give isl_pw_aff
*extract_affine(clang::BinaryOperator
*expr
);
175 __isl_give isl_pw_aff
*extract_affine(clang::UnaryOperator
*expr
);
176 __isl_give isl_pw_aff
*extract_affine(clang::ParenExpr
*expr
);
177 __isl_give isl_pw_aff
*extract_affine(clang::CallExpr
*expr
);
178 __isl_give isl_pw_aff
*extract_affine(clang::ArraySubscriptExpr
*expr
);
179 __isl_give isl_pw_aff
*extract_affine(clang::ConditionalOperator
*expr
);
181 __isl_give isl_pw_aff
*extract_implicit_affine(clang::Expr
*expr
);
182 __isl_give isl_set
*extract_implicit_condition(clang::Expr
*expr
);
184 __isl_give isl_set
*extract_condition(clang::UnaryOperator
*expr
);
185 __isl_give isl_set
*extract_condition(clang::Expr
*expr
);
186 __isl_give isl_set
*extract_comparison(clang::BinaryOperator
*expr
);
187 __isl_give isl_set
*extract_comparison(clang::BinaryOperatorKind op
,
188 clang::Expr
*LHS
, clang::Expr
*RHS
, clang::Stmt
*comp
);
189 __isl_give isl_set
*extract_boolean(clang::BinaryOperator
*expr
);
190 __isl_give isl_set
*extract_boolean(clang::UnaryOperator
*expr
);
192 void unsupported(clang::Stmt
*stmt
, const char *msg
= NULL
);