add isl_qpolynomial_fold_get_ctx
[isl.git] / isl_vec.c
blob00b03320dc78bdef85fc36225e9231fc6816fb95
1 /*
2 * Copyright 2008-2009 Katholieke Universiteit Leuven
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, K.U.Leuven, Departement
7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
8 */
10 #include <isl/seq.h>
11 #include <isl/vec.h>
13 struct isl_vec *isl_vec_alloc(struct isl_ctx *ctx, unsigned size)
15 struct isl_vec *vec;
17 vec = isl_alloc_type(ctx, struct isl_vec);
18 if (!vec)
19 return NULL;
21 vec->block = isl_blk_alloc(ctx, size);
22 if (isl_blk_is_error(vec->block))
23 goto error;
25 vec->ctx = ctx;
26 isl_ctx_ref(ctx);
27 vec->ref = 1;
28 vec->size = size;
29 vec->el = vec->block.data;
31 return vec;
32 error:
33 isl_blk_free(ctx, vec->block);
34 return NULL;
37 __isl_give isl_vec *isl_vec_extend(__isl_take isl_vec *vec, unsigned size)
39 if (!vec)
40 return NULL;
41 if (size <= vec->size)
42 return vec;
44 vec = isl_vec_cow(vec);
45 if (!vec)
46 return NULL;
48 vec->block = isl_blk_extend(vec->ctx, vec->block, size);
49 if (!vec->block.data)
50 goto error;
52 vec->size = size;
53 vec->el = vec->block.data;
55 return vec;
56 error:
57 isl_vec_free(vec);
58 return NULL;
61 __isl_give isl_vec *isl_vec_zero_extend(__isl_take isl_vec *vec, unsigned size)
63 int extra;
65 if (!vec)
66 return NULL;
67 if (size <= vec->size)
68 return vec;
70 vec = isl_vec_cow(vec);
71 if (!vec)
72 return NULL;
74 extra = size - vec->size;
75 vec = isl_vec_extend(vec, size);
76 if (!vec)
77 return NULL;
79 isl_seq_clr(vec->el + size - extra, extra);
81 return vec;
84 struct isl_vec *isl_vec_copy(struct isl_vec *vec)
86 if (!vec)
87 return NULL;
89 vec->ref++;
90 return vec;
93 struct isl_vec *isl_vec_dup(struct isl_vec *vec)
95 struct isl_vec *vec2;
97 if (!vec)
98 return NULL;
99 vec2 = isl_vec_alloc(vec->ctx, vec->size);
100 isl_seq_cpy(vec2->el, vec->el, vec->size);
101 return vec2;
104 struct isl_vec *isl_vec_cow(struct isl_vec *vec)
106 struct isl_vec *vec2;
107 if (!vec)
108 return NULL;
110 if (vec->ref == 1)
111 return vec;
113 vec2 = isl_vec_dup(vec);
114 isl_vec_free(vec);
115 return vec2;
118 void isl_vec_free(struct isl_vec *vec)
120 if (!vec)
121 return;
123 if (--vec->ref > 0)
124 return;
126 isl_ctx_deref(vec->ctx);
127 isl_blk_free(vec->ctx, vec->block);
128 free(vec);
131 void isl_vec_dump(struct isl_vec *vec, FILE *out, int indent)
133 int i;
135 if (!vec) {
136 fprintf(out, "%*snull vec\n", indent, "");
137 return;
140 fprintf(out, "%*s[", indent, "");
141 for (i = 0; i < vec->size; ++i) {
142 if (i)
143 fprintf(out, ",");
144 isl_int_print(out, vec->el[i], 0);
146 fprintf(out, "]\n");
149 __isl_give isl_vec *isl_vec_clr(__isl_take isl_vec *vec)
151 vec = isl_vec_cow(vec);
152 if (!vec)
153 return NULL;
154 isl_seq_clr(vec->el, vec->size);
155 return vec;
158 void isl_vec_lcm(struct isl_vec *vec, isl_int *lcm)
160 isl_seq_lcm(vec->block.data, vec->size, lcm);
163 /* Given a rational vector, with the denominator in the first element
164 * of the vector, round up all coordinates.
166 struct isl_vec *isl_vec_ceil(struct isl_vec *vec)
168 vec = isl_vec_cow(vec);
169 if (!vec)
170 return NULL;
172 isl_seq_cdiv_q(vec->el + 1, vec->el + 1, vec->el[0], vec->size - 1);
174 isl_int_set_si(vec->el[0], 1);
176 return vec;
179 struct isl_vec *isl_vec_normalize(struct isl_vec *vec)
181 if (!vec)
182 return NULL;
183 isl_seq_normalize(vec->ctx, vec->el, vec->size);
184 return vec;
187 __isl_give isl_vec *isl_vec_scale(__isl_take isl_vec *vec, isl_int m)
189 if (isl_int_is_one(m))
190 return vec;
191 vec = isl_vec_cow(vec);
192 if (!vec)
193 return NULL;
194 isl_seq_scale(vec->el, vec->el, m, vec->size);
195 return vec;
198 __isl_give isl_vec *isl_vec_add(__isl_take isl_vec *vec1,
199 __isl_take isl_vec *vec2)
201 vec1 = isl_vec_cow(vec1);
202 if (!vec1 || !vec2)
203 goto error;
205 isl_assert(vec1->ctx, vec1->size == vec2->size, goto error);
207 isl_seq_combine(vec1->el, vec1->ctx->one, vec1->el,
208 vec1->ctx->one, vec2->el, vec1->size);
210 isl_vec_free(vec2);
211 return vec1;
212 error:
213 isl_vec_free(vec1);
214 isl_vec_free(vec2);
215 return NULL;