scop_plus.cc

   1 #include <set>
   2
   3 #include "scop_plus.h"
   4
   5 using namespace std;
   6 using namespace clang;
   7
   8 static void access_collect_arrays(struct pet_expr *expr,
   9         set<ValueDecl *> &arrays)
  10 {
  11         isl_id *id;
  12         ValueDecl *decl;
  13
  14         id = isl_map_get_tuple_id(expr->acc.access, isl_dim_out);
  15         if (!id)
  16                 return;
  17
  18         decl = (ValueDecl *)isl_id_get_user(id);
  19         isl_id_free(id);
  20
  21         arrays.insert(decl);
  22 }
  23
  24 static void expr_collect_arrays(struct pet_expr *expr, set<ValueDecl *> &arrays)
  25 {
  26         if (!expr)
  27                 return;
  28
  29         for (int i = 0; i < expr->n_arg; ++i)
  30                 expr_collect_arrays(expr->args[i], arrays);
  31
  32         if (expr->type == pet_expr_access)
  33                 access_collect_arrays(expr, arrays);
  34 }
  35
  36 static void stmt_collect_arrays(struct pet_stmt *stmt, set<ValueDecl *> &arrays)
  37 {
  38         if (!stmt)
  39                 return;
  40         expr_collect_arrays(stmt->body, arrays);
  41 }
  42
  43 /* Collect the set of all accessed arrays (or scalars) in "scop"
  44  * and put them in "arrays".
  45  */
  46 void pet_scop_collect_arrays(struct pet_scop *scop, set<ValueDecl *> &arrays)
  47 {
  48         if (!scop)
  49                 return;
  50
  51         for (int i = 0; i < scop->n_stmt; ++i)
  52                 stmt_collect_arrays(scop->stmts[i], arrays);
  53 }