1 //===- GmpConv.cpp - Recreate LLVM IR from the Scop.  ---------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // Functions for converting between gmp objects and apint.
11 //
12 //===----------------------------------------------------------------------===//
13 #include "polly/Support/GICHelper.h"
14 #include "isl/aff.h"
15 #include "isl/map.h"
16 #include "isl/schedule.h"
17 #include "isl/set.h"
18 #include "isl/union_map.h"
19 #include "isl/union_set.h"
20 #include "isl/val.h"
21 
22 using namespace llvm;
23 
24 __isl_give isl_val *polly::isl_valFromAPInt(isl_ctx *Ctx, const APInt Int,
25                                             bool IsSigned) {
26   APInt Abs;
27   isl_val *v;
28 
29   if (IsSigned)
30     Abs = Int.abs();
31   else
32     Abs = Int;
33 
34   const uint64_t *Data = Abs.getRawData();
35   unsigned Words = Abs.getNumWords();
36 
37   v = isl_val_int_from_chunks(Ctx, Words, sizeof(uint64_t), Data);
38 
39   if (IsSigned && Int.isNegative())
40     v = isl_val_neg(v);
41 
42   return v;
43 }
44 
45 APInt polly::APIntFromVal(__isl_take isl_val *Val) {
46   uint64_t *Data;
47   int NumChunks;
48 
49   NumChunks = isl_val_n_abs_num_chunks(Val, sizeof(uint64_t));
50 
51   Data = (uint64_t *)malloc(NumChunks * sizeof(uint64_t));
52   isl_val_get_abs_num_chunks(Val, sizeof(uint64_t), Data);
53   APInt A(8 * sizeof(uint64_t) * NumChunks, NumChunks, Data);
54 
55   if (isl_val_is_neg(Val)) {
56     A = A.zext(A.getBitWidth() + 1);
57     A = -A;
58   }
59 
60   if (A.getMinSignedBits() < A.getBitWidth())
61     A = A.trunc(A.getMinSignedBits());
62 
63   free(Data);
64   isl_val_free(Val);
65   return A;
66 }
67 
68 template <typename ISLTy, typename ISL_CTX_GETTER, typename ISL_PRINTER>
69 static inline std::string stringFromIslObjInternal(__isl_keep ISLTy *isl_obj,
70                                                    ISL_CTX_GETTER ctx_getter_fn,
71                                                    ISL_PRINTER printer_fn) {
72   isl_ctx *ctx = ctx_getter_fn(isl_obj);
73   isl_printer *p = isl_printer_to_str(ctx);
74   printer_fn(p, isl_obj);
75   char *char_str = isl_printer_get_str(p);
76   std::string string(char_str);
77   free(char_str);
78   isl_printer_free(p);
79   return string;
80 }
81 
82 static inline isl_ctx *schedule_get_ctx(__isl_keep isl_schedule *schedule) {
83   return isl_union_map_get_ctx(isl_schedule_get_map(schedule));
84 }
85 
86 std::string polly::stringFromIslObj(__isl_keep isl_map *map) {
87   return stringFromIslObjInternal(map, isl_map_get_ctx, isl_printer_print_map);
88 }
89 
90 std::string polly::stringFromIslObj(__isl_keep isl_set *set) {
91   return stringFromIslObjInternal(set, isl_set_get_ctx, isl_printer_print_set);
92 }
93 
94 std::string polly::stringFromIslObj(__isl_keep isl_union_map *umap) {
95   return stringFromIslObjInternal(umap, isl_union_map_get_ctx,
96                                   isl_printer_print_union_map);
97 }
98 
99 std::string polly::stringFromIslObj(__isl_keep isl_union_set *uset) {
100   return stringFromIslObjInternal(uset, isl_union_set_get_ctx,
101                                   isl_printer_print_union_set);
102 }
103 
104 std::string polly::stringFromIslObj(__isl_keep isl_schedule *schedule) {
105   return stringFromIslObjInternal(schedule, schedule_get_ctx,
106                                   isl_printer_print_schedule);
107 }
108 
109 std::string polly::stringFromIslObj(__isl_keep isl_multi_aff *maff) {
110   return stringFromIslObjInternal(maff, isl_multi_aff_get_ctx,
111                                   isl_printer_print_multi_aff);
112 }
113 
114 std::string polly::stringFromIslObj(__isl_keep isl_pw_multi_aff *pma) {
115   return stringFromIslObjInternal(pma, isl_pw_multi_aff_get_ctx,
116                                   isl_printer_print_pw_multi_aff);
117 }
118 
119 std::string polly::stringFromIslObj(__isl_keep isl_aff *aff) {
120   return stringFromIslObjInternal(aff, isl_aff_get_ctx, isl_printer_print_aff);
121 }
122 
123 std::string polly::stringFromIslObj(__isl_keep isl_pw_aff *pwaff) {
124   return stringFromIslObjInternal(pwaff, isl_pw_aff_get_ctx,
125                                   isl_printer_print_pw_aff);
126 }
127