expr_plus.cc

   1 /*
   2  * Copyright 2011      Leiden University. All rights reserved.
   3  * Copyright 2013-2015 Ecole Normale Superieure. All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  *    1. Redistributions of source code must retain the above copyright
  10  *       notice, this list of conditions and the following disclaimer.
  11  *
  12  *    2. Redistributions in binary form must reproduce the above
  13  *       copyright notice, this list of conditions and the following
  14  *       disclaimer in the documentation and/or other materials provided
  15  *       with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY LEIDEN UNIVERSITY ''AS IS'' AND ANY
  18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  20  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL LEIDEN UNIVERSITY OR
  21  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  22  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  23  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
  24  * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  27  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28  *
  29  * The views and conclusions contained in the software and documentation
  30  * are those of the authors and should not be interpreted as
  31  * representing official policies, either expressed or implied, of
  32  * Leiden University.
  33  */
  34
  35 #include <clang/AST/ASTContext.h>
  36 #include <clang/AST/Decl.h>
  37 #include <clang/AST/Type.h>
  38
  39 #include <isl/aff.h>
  40 #include <isl/id.h>
  41
  42 #include "clang.h"
  43 #include "expr.h"
  44 #include "expr_plus.h"
  45 #include "id.h"
  46
  47 using namespace clang;
  48
  49 /* Return the depth of the array accessed by the index expression "index".
  50  * If "index" is an affine expression, i.e., if it does not access
  51  * any array, then return 1.
  52  * If "index" represent a member access, i.e., if its range is a wrapped
  53  * relation, then return the sum of the depth of the array of structures
  54  * and that of the member inside the structure.
  55  */
  56 static int extract_depth(__isl_keep isl_multi_pw_aff *index)
  57 {
  58         isl_id *id;
  59         QualType qt;
  60
  61         if (!index)
  62                 return -1;
  63
  64         if (isl_multi_pw_aff_range_is_wrapping(index)) {
  65                 int domain_depth, range_depth;
  66                 isl_multi_pw_aff *domain, *range;
  67
  68                 domain = isl_multi_pw_aff_copy(index);
  69                 domain = isl_multi_pw_aff_range_factor_domain(domain);
  70                 domain_depth = extract_depth(domain);
  71                 isl_multi_pw_aff_free(domain);
  72                 range = isl_multi_pw_aff_copy(index);
  73                 range = isl_multi_pw_aff_range_factor_range(range);
  74                 range_depth = extract_depth(range);
  75                 isl_multi_pw_aff_free(range);
  76
  77                 return domain_depth + range_depth;
  78         }
  79
  80         if (!isl_multi_pw_aff_has_tuple_id(index, isl_dim_out))
  81                 return 1;
  82
  83         id = isl_multi_pw_aff_get_tuple_id(index, isl_dim_out);
  84         if (!id)
  85                 return -1;
  86         qt = pet_id_get_array_type(id);
  87         isl_id_free(id);
  88
  89         return pet_clang_array_depth(qt);
  90 }
  91
  92 /* Return the depth of the array accessed by the access expression "expr".
  93  */
  94 static int extract_depth(__isl_keep pet_expr *expr)
  95 {
  96         isl_multi_pw_aff *index;
  97         int depth;
  98
  99         index = pet_expr_access_get_index(expr);
 100         depth = extract_depth(index);
 101         isl_multi_pw_aff_free(index);
 102
 103         return depth;
 104 }
 105
 106 /* Convert the index expression "index" into an access pet_expr of type "qt".
 107  */
 108 __isl_give pet_expr *pet_expr_access_from_index(QualType qt,
 109         __isl_take pet_expr *index, ASTContext &ast_context)
 110 {
 111         int depth;
 112         int type_size;
 113
 114         depth = extract_depth(index);
 115         type_size = pet_clang_get_type_size(qt, ast_context);
 116
 117         index = pet_expr_set_type_size(index, type_size);
 118         index = pet_expr_access_set_depth(index, depth);
 119
 120         return index;
 121 }
 122
 123 /* Construct an access pet_expr for an access
 124  * to the variable represented by "id".
 125  */
 126 __isl_give pet_expr *pet_expr_access_from_id(__isl_take isl_id *id,
 127         ASTContext &ast_context)
 128 {
 129         QualType qt;
 130         pet_expr *expr;
 131
 132         qt = pet_id_get_array_type(id);
 133         expr = pet_id_create_index_expr(id);
 134         return pet_expr_access_from_index(qt, expr, ast_context);
 135 }