scan.h

   1 #include <map>
   2
   3 #include <clang/Basic/SourceManager.h>
   4 #include <clang/AST/Decl.h>
   5 #include <clang/AST/Stmt.h>
   6 #include <clang/Lex/Preprocessor.h>
   7
   8 #include <isl/ctx.h>
   9 #include <isl/map.h>
  10 #include <isl/val.h>
  11
  12 #include "scop.h"
  13
  14 /* The location of the scop, as delimited by scop and endscop
  15  * pragmas by the user.
  16  */
  17 struct ScopLoc {
  18         ScopLoc() : end(0) {}
  19
  20         unsigned start;
  21         unsigned end;
  22 };
  23
  24 /* Compare two RecordDecl pointers based on their names.
  25  */
  26 struct less_name {
  27         bool operator()(const clang::RecordDecl *x,
  28                         const clang::RecordDecl *y) {
  29                 return x->getNameAsString().compare(y->getNameAsString()) < 0;
  30         }
  31 };
  32
  33 /* A sorted set of RecordDecl pointers.  The actual order is not important,
  34  * only that it is consistent across platforms.
  35  */
  36 typedef std::set<clang::RecordDecl *, less_name> lex_recorddecl_set;
  37
  38 struct PetScan {
  39         clang::Preprocessor &PP;
  40         clang::ASTContext &ast_context;
  41         /* If autodetect is false, then loc contains the location
  42          * of the scop to be extracted.
  43          */
  44         ScopLoc &loc;
  45         isl_ctx *ctx;
  46         pet_options *options;
  47         /* The sequence number of the next statement. */
  48         int n_stmt;
  49         /* The sequence number of the next virtual scalar. */
  50         int n_test;
  51         /* Set if the pet_scop returned by an extract method only
  52          * represents part of the input tree.
  53          */
  54         bool partial;
  55         /* Set is nested accesses are allowed in general.
  56          * This currently defaults to true.
  57          */
  58         bool allow_nested;
  59         /* Set if nested accesses are allowed in that part of the tree
  60          * that is currently under investigation.
  61          */
  62         bool nesting_enabled;
  63         /* Maps identifiers to the last value that was assigned to them.
  64          * If an identifier is mapped to NULL, then something may have
  65          * been assigned, but we don't know what.
  66          * assigned_value does not take a reference to the isl_pw_aff
  67          * object, so each such isl_pw_aff needs to be stored in
  68          * the set of "expressions".
  69          */
  70         std::map<clang::ValueDecl *, isl_pw_aff *> assigned_value;
  71         /* A collection of isl_pw_affs used in assigned_value or other
  72          * temporary maps.  expressions holds a reference for each
  73          * isl_pw_aff, which is freed in the destructor of PetScan.
  74          */
  75         std::set<isl_pw_aff *> expressions;
  76
  77         /* A union of mappings of the form
  78          *      { identifier[] -> [i] : lower_bound <= i <= upper_bound }
  79          */
  80         isl_union_map *value_bounds;
  81
  82         PetScan(clang::Preprocessor &PP,
  83                 clang::ASTContext &ast_context, ScopLoc &loc,
  84                 pet_options *options, __isl_take isl_union_map *value_bounds) :
  85                 ctx(isl_union_map_get_ctx(value_bounds)), PP(PP),
  86                 ast_context(ast_context), loc(loc),
  87                 options(options), value_bounds(value_bounds),
  88                 n_stmt(0), n_test(0), partial(0), allow_nested(true),
  89                 nesting_enabled(false) { }
  90
  91         ~PetScan();
  92
  93         struct pet_scop *scan(clang::FunctionDecl *fd);
  94
  95         static __isl_give isl_val *extract_int(isl_ctx *ctx,
  96                 clang::IntegerLiteral *expr);
  97         static __isl_give isl_val *extract_unsigned(isl_ctx *ctx,
  98                 const llvm::APInt &val);
  99 private:
 100         void assign(__isl_keep pet_expr *lhs, clang::Expr *rhs);
 101
 102         void insert_expression(__isl_take isl_pw_aff *expr);
 103         struct pet_scop *scan(clang::Stmt *stmt);
 104
 105         struct pet_scop *scan_arrays(struct pet_scop *scop);
 106         struct pet_array *extract_array(isl_ctx *ctx, clang::ValueDecl *decl,
 107                 lex_recorddecl_set *types);
 108         struct pet_array *extract_array(isl_ctx *ctx,
 109                 std::vector<clang::ValueDecl *> decls,
 110                 lex_recorddecl_set *types);
 111         struct pet_array *set_upper_bounds(struct pet_array *array,
 112                 const clang::Type *type, int pos);
 113
 114         struct pet_scop *extract_non_affine_condition(clang::Expr *cond,
 115                 int stmt_nr, __isl_take isl_multi_pw_aff *index);
 116
 117         struct pet_scop *extract_conditional_assignment(clang::IfStmt *stmt);
 118         struct pet_scop *extract_non_affine_if(clang::Expr *cond,
 119                 struct pet_scop *scop_then, struct pet_scop *scop_else,
 120                 bool have_else, int stmt_id);
 121
 122         struct pet_scop *kill(clang::Stmt *stmt, struct pet_array *array);
 123
 124         struct pet_scop *extract(clang::Stmt *stmt,
 125                 bool skip_declarations = false);
 126         struct pet_scop *extract(clang::StmtRange stmt_range, bool block,
 127                 bool skip_declarations);
 128         struct pet_scop *extract(clang::IfStmt *stmt);
 129         struct pet_scop *extract(clang::WhileStmt *stmt);
 130         struct pet_scop *extract(clang::CompoundStmt *stmt,
 131                 bool skip_declarations = false);
 132         struct pet_scop *extract(clang::LabelStmt *stmt);
 133         struct pet_scop *extract(clang::ContinueStmt *stmt);
 134         struct pet_scop *extract(clang::BreakStmt *stmt);
 135         struct pet_scop *extract(clang::DeclStmt *expr);
 136
 137         struct pet_scop *update_scop_start_end(struct pet_scop *scop,
 138                 clang::SourceRange range, bool skip_semi);
 139         struct pet_scop *extract(__isl_take pet_expr *expr,
 140                 clang::SourceRange range, bool skip_semi,
 141                 __isl_take isl_id *label = NULL);
 142         struct pet_stmt *extract_kill(struct pet_scop *scop);
 143
 144         clang::BinaryOperator *initialization_assignment(clang::Stmt *init);
 145         clang::Decl *initialization_declaration(clang::Stmt *init);
 146         clang::ValueDecl *extract_induction_variable(clang::BinaryOperator *stmt);
 147         clang::VarDecl *extract_induction_variable(clang::Stmt *init,
 148                                 clang::Decl *stmt);
 149         __isl_give pet_expr *extract_unary_increment(clang::UnaryOperator *op,
 150                                 clang::ValueDecl *iv);
 151         __isl_give pet_expr *extract_binary_increment(
 152                                 clang::BinaryOperator *op,
 153                                 clang::ValueDecl *iv);
 154         __isl_give pet_expr *extract_compound_increment(
 155                                 clang::CompoundAssignOperator *op,
 156                                 clang::ValueDecl *iv);
 157         __isl_give pet_expr *extract_increment(clang::ForStmt *stmt,
 158                                 clang::ValueDecl *iv);
 159         struct pet_scop *extract_for(clang::ForStmt *stmt);
 160         struct pet_scop *extract_non_affine_for(clang::ForStmt *stmt,
 161                 clang::ValueDecl *iv,
 162                 __isl_take pet_expr *init, __isl_take pet_expr *inc);
 163         struct pet_scop *extract_infinite_loop(clang::Stmt *body);
 164         struct pet_scop *extract_infinite_for(clang::ForStmt *stmt);
 165         struct pet_scop *extract_affine_while(__isl_take isl_pw_aff *pa,
 166                                 clang::Stmt *body);
 167         struct pet_scop *extract_while(clang::Expr *cond, int test_nr,
 168                 int stmt_nr, struct pet_scop *scop_body,
 169                 struct pet_scop *scop_inc);
 170
 171         __isl_give pet_expr *mark_write(__isl_take pet_expr *access);
 172         __isl_give pet_expr *extract_assume(clang::Expr *expr);
 173         __isl_give pet_expr *extract_argument(clang::FunctionDecl *fd, int pos,
 174                 clang::Expr *expr);
 175         __isl_give pet_expr *extract_expr(clang::Expr *expr);
 176         __isl_give pet_expr *extract_expr(clang::UnaryOperator *expr);
 177         __isl_give pet_expr *extract_expr(clang::BinaryOperator *expr);
 178         __isl_give pet_expr *extract_expr(clang::ImplicitCastExpr *expr);
 179         __isl_give pet_expr *extract_expr(clang::IntegerLiteral *expr);
 180         __isl_give pet_expr *extract_expr(clang::FloatingLiteral *expr);
 181         __isl_give pet_expr *extract_expr(clang::ParenExpr *expr);
 182         __isl_give pet_expr *extract_expr(clang::ConditionalOperator *expr);
 183         __isl_give pet_expr *extract_expr(clang::CallExpr *expr);
 184         __isl_give pet_expr *extract_expr(clang::CStyleCastExpr *expr);
 185
 186         int extract_nested(__isl_keep isl_space *space,
 187                 int n_arg, pet_expr **args, std::map<int,int> &param2pos);
 188         __isl_give pet_expr *extract_nested(__isl_take pet_expr *expr, int n,
 189                 std::map<int,int> &param2pos);
 190         struct pet_stmt *extract_nested(struct pet_stmt *stmt, int n,
 191                 std::map<int,int> &param2pos);
 192         __isl_give pet_expr *resolve_nested(__isl_take pet_expr *expr);
 193         struct pet_scop *resolve_nested(struct pet_scop *scop);
 194         struct pet_stmt *resolve_nested(struct pet_stmt *stmt);
 195         __isl_give pet_expr *extract_access_expr(clang::QualType qt,
 196                 __isl_take isl_multi_pw_aff *index);
 197         __isl_give pet_expr *extract_access_expr(clang::Expr *expr);
 198         __isl_give pet_expr *extract_access_expr(clang::ValueDecl *decl);
 199
 200         __isl_give isl_multi_pw_aff *extract_index(
 201                 clang::ArraySubscriptExpr *expr);
 202         __isl_give isl_multi_pw_aff *extract_index(clang::Expr *expr);
 203         __isl_give isl_multi_pw_aff *extract_index(
 204                 clang::ImplicitCastExpr *expr);
 205         __isl_give isl_multi_pw_aff *extract_index(clang::DeclRefExpr *expr);
 206         __isl_give isl_multi_pw_aff *extract_index(clang::ValueDecl *decl);
 207         __isl_give isl_multi_pw_aff *extract_index(
 208                 clang::IntegerLiteral *expr);
 209         __isl_give isl_multi_pw_aff *extract_index(clang::MemberExpr *expr);
 210
 211         __isl_give isl_val *extract_int(clang::Expr *expr);
 212         __isl_give isl_val *extract_int(clang::ParenExpr *expr);
 213
 214         __isl_give isl_pw_aff *try_extract_affine(clang::Expr *expr);
 215         __isl_give isl_pw_aff *try_extract_affine_condition(clang::Expr *expr);
 216         bool is_affine_condition(clang::Expr *expr);
 217         __isl_give isl_pw_aff *try_extract_nested_condition(clang::Expr *expr);
 218         bool is_nested_allowed(__isl_keep isl_pw_aff *pa, pet_scop *scop);
 219
 220         __isl_give isl_pw_aff *extract_affine(const llvm::APInt &val);
 221         __isl_give isl_pw_aff *extract_affine(clang::Expr *expr);
 222
 223         __isl_give isl_pw_aff *extract_condition(clang::Expr *expr);
 224         __isl_give isl_pw_aff *extract_comparison(clang::BinaryOperator *expr);
 225         __isl_give isl_pw_aff *extract_comparison(clang::BinaryOperatorKind op,
 226                 clang::Expr *LHS, clang::Expr *RHS, clang::Stmt *comp);
 227
 228         void report(clang::Stmt *stmt, unsigned id);
 229         void unsupported(clang::Stmt *stmt);
 230         void report_prototype_required(clang::Stmt *stmt);
 231         void report_missing_increment(clang::Stmt *stmt);
 232 };