scop_plus.cc

   1 /*
   2  * Copyright 2011 Leiden University. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  *    1. Redistributions of source code must retain the above copyright
   9  *       notice, this list of conditions and the following disclaimer.
  10  *
  11  *    2. Redistributions in binary form must reproduce the above
  12  *       copyright notice, this list of conditions and the following
  13  *       disclaimer in the documentation and/or other materials provided
  14  *       with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY LEIDEN UNIVERSITY ''AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL LEIDEN UNIVERSITY OR
  20  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  21  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  22  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
  23  * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  *
  28  * The views and conclusions contained in the software and documentation
  29  * are those of the authors and should not be interpreted as
  30  * representing official policies, either expressed or implied, of
  31  * Leiden University.
  32  */
  33
  34 #include <set>
  35
  36 #include "scop_plus.h"
  37
  38 using namespace std;
  39 using namespace clang;
  40
  41 static void access_collect_arrays(struct pet_expr *expr,
  42         set<ValueDecl *> &arrays)
  43 {
  44         isl_id *id;
  45         ValueDecl *decl;
  46
  47         if (!isl_map_has_tuple_id(expr->acc.access, isl_dim_out))
  48                 return;
  49         id = isl_map_get_tuple_id(expr->acc.access, isl_dim_out);
  50         if (!id)
  51                 return;
  52
  53         decl = (ValueDecl *)isl_id_get_user(id);
  54         isl_id_free(id);
  55
  56         arrays.insert(decl);
  57 }
  58
  59 static void expr_collect_arrays(struct pet_expr *expr, set<ValueDecl *> &arrays)
  60 {
  61         if (!expr)
  62                 return;
  63
  64         for (int i = 0; i < expr->n_arg; ++i)
  65                 expr_collect_arrays(expr->args[i], arrays);
  66
  67         if (expr->type == pet_expr_access)
  68                 access_collect_arrays(expr, arrays);
  69 }
  70
  71 static void stmt_collect_arrays(struct pet_stmt *stmt, set<ValueDecl *> &arrays)
  72 {
  73         if (!stmt)
  74                 return;
  75         expr_collect_arrays(stmt->body, arrays);
  76 }
  77
  78 /* Collect the set of all accessed arrays (or scalars) in "scop"
  79  * and put them in "arrays".
  80  */
  81 void pet_scop_collect_arrays(struct pet_scop *scop, set<ValueDecl *> &arrays)
  82 {
  83         if (!scop)
  84                 return;
  85
  86         for (int i = 0; i < scop->n_stmt; ++i)
  87                 stmt_collect_arrays(scop->stmts[i], arrays);
  88 }