2 * Copyright 2008-2009 Katholieke Universiteit Leuven
4 * Use of this software is governed by the MIT license
6 * Written by Sven Verdoolaege, K.U.Leuven, Departement
7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
10 #include <isl_ctx_private.h>
14 isl_ctx
*isl_vec_get_ctx(__isl_keep isl_vec
*vec
)
16 return vec
? vec
->ctx
: NULL
;
19 struct isl_vec
*isl_vec_alloc(struct isl_ctx
*ctx
, unsigned size
)
23 vec
= isl_alloc_type(ctx
, struct isl_vec
);
27 vec
->block
= isl_blk_alloc(ctx
, size
);
28 if (isl_blk_is_error(vec
->block
))
35 vec
->el
= vec
->block
.data
;
39 isl_blk_free(ctx
, vec
->block
);
43 __isl_give isl_vec
*isl_vec_extend(__isl_take isl_vec
*vec
, unsigned size
)
47 if (size
<= vec
->size
)
50 vec
= isl_vec_cow(vec
);
54 vec
->block
= isl_blk_extend(vec
->ctx
, vec
->block
, size
);
59 vec
->el
= vec
->block
.data
;
67 __isl_give isl_vec
*isl_vec_zero_extend(__isl_take isl_vec
*vec
, unsigned size
)
73 if (size
<= vec
->size
)
76 vec
= isl_vec_cow(vec
);
80 extra
= size
- vec
->size
;
81 vec
= isl_vec_extend(vec
, size
);
85 isl_seq_clr(vec
->el
+ size
- extra
, extra
);
90 /* Return a vector containing the elements of "vec1" followed by
93 __isl_give isl_vec
*isl_vec_concat(__isl_take isl_vec
*vec1
,
94 __isl_take isl_vec
*vec2
)
99 if (vec2
->size
== 0) {
104 if (vec1
->size
== 0) {
109 vec1
= isl_vec_extend(vec1
, vec1
->size
+ vec2
->size
);
113 isl_seq_cpy(vec1
->el
+ vec1
->size
- vec2
->size
, vec2
->el
, vec2
->size
);
123 struct isl_vec
*isl_vec_copy(struct isl_vec
*vec
)
132 struct isl_vec
*isl_vec_dup(struct isl_vec
*vec
)
134 struct isl_vec
*vec2
;
138 vec2
= isl_vec_alloc(vec
->ctx
, vec
->size
);
141 isl_seq_cpy(vec2
->el
, vec
->el
, vec
->size
);
145 struct isl_vec
*isl_vec_cow(struct isl_vec
*vec
)
147 struct isl_vec
*vec2
;
154 vec2
= isl_vec_dup(vec
);
159 void *isl_vec_free(__isl_take isl_vec
*vec
)
167 isl_ctx_deref(vec
->ctx
);
168 isl_blk_free(vec
->ctx
, vec
->block
);
174 int isl_vec_size(__isl_keep isl_vec
*vec
)
176 return vec
? vec
->size
: -1;
179 int isl_vec_get_element(__isl_keep isl_vec
*vec
, int pos
, isl_int
*v
)
184 if (pos
< 0 || pos
>= vec
->size
)
185 isl_die(vec
->ctx
, isl_error_invalid
, "position out of range",
187 isl_int_set(*v
, vec
->el
[pos
]);
191 __isl_give isl_vec
*isl_vec_set_element(__isl_take isl_vec
*vec
,
194 vec
= isl_vec_cow(vec
);
197 if (pos
< 0 || pos
>= vec
->size
)
198 isl_die(vec
->ctx
, isl_error_invalid
, "position out of range",
200 isl_int_set(vec
->el
[pos
], v
);
207 __isl_give isl_vec
*isl_vec_set_element_si(__isl_take isl_vec
*vec
,
210 vec
= isl_vec_cow(vec
);
213 if (pos
< 0 || pos
>= vec
->size
)
214 isl_die(vec
->ctx
, isl_error_invalid
, "position out of range",
216 isl_int_set_si(vec
->el
[pos
], v
);
223 int isl_vec_is_equal(__isl_keep isl_vec
*vec1
, __isl_keep isl_vec
*vec2
)
228 if (vec1
->size
!= vec2
->size
)
231 return isl_seq_eq(vec1
->el
, vec2
->el
, vec1
->size
);
234 __isl_give isl_printer
*isl_printer_print_vec(__isl_take isl_printer
*printer
,
235 __isl_keep isl_vec
*vec
)
239 if (!printer
|| !vec
)
242 printer
= isl_printer_print_str(printer
, "[");
243 for (i
= 0; i
< vec
->size
; ++i
) {
245 printer
= isl_printer_print_str(printer
, ",");
246 printer
= isl_printer_print_isl_int(printer
, vec
->el
[i
]);
248 printer
= isl_printer_print_str(printer
, "]");
252 isl_printer_free(printer
);
256 void isl_vec_dump(struct isl_vec
*vec
)
258 isl_printer
*printer
;
263 printer
= isl_printer_to_file(vec
->ctx
, stderr
);
264 printer
= isl_printer_print_vec(printer
, vec
);
265 printer
= isl_printer_end_line(printer
);
267 isl_printer_free(printer
);
270 __isl_give isl_vec
*isl_vec_set(__isl_take isl_vec
*vec
, isl_int v
)
272 vec
= isl_vec_cow(vec
);
275 isl_seq_set(vec
->el
, v
, vec
->size
);
279 __isl_give isl_vec
*isl_vec_set_si(__isl_take isl_vec
*vec
, int v
)
281 vec
= isl_vec_cow(vec
);
284 isl_seq_set_si(vec
->el
, v
, vec
->size
);
288 __isl_give isl_vec
*isl_vec_clr(__isl_take isl_vec
*vec
)
290 vec
= isl_vec_cow(vec
);
293 isl_seq_clr(vec
->el
, vec
->size
);
297 void isl_vec_lcm(struct isl_vec
*vec
, isl_int
*lcm
)
299 isl_seq_lcm(vec
->block
.data
, vec
->size
, lcm
);
302 /* Given a rational vector, with the denominator in the first element
303 * of the vector, round up all coordinates.
305 struct isl_vec
*isl_vec_ceil(struct isl_vec
*vec
)
307 vec
= isl_vec_cow(vec
);
311 isl_seq_cdiv_q(vec
->el
+ 1, vec
->el
+ 1, vec
->el
[0], vec
->size
- 1);
313 isl_int_set_si(vec
->el
[0], 1);
318 struct isl_vec
*isl_vec_normalize(struct isl_vec
*vec
)
322 isl_seq_normalize(vec
->ctx
, vec
->el
, vec
->size
);
326 __isl_give isl_vec
*isl_vec_neg(__isl_take isl_vec
*vec
)
328 vec
= isl_vec_cow(vec
);
331 isl_seq_neg(vec
->el
, vec
->el
, vec
->size
);
335 __isl_give isl_vec
*isl_vec_scale(__isl_take isl_vec
*vec
, isl_int m
)
337 if (isl_int_is_one(m
))
339 vec
= isl_vec_cow(vec
);
342 isl_seq_scale(vec
->el
, vec
->el
, m
, vec
->size
);
346 /* Reduce the elements of "vec" modulo "m".
348 __isl_give isl_vec
*isl_vec_fdiv_r(__isl_take isl_vec
*vec
, isl_int m
)
350 vec
= isl_vec_cow(vec
);
354 isl_seq_fdiv_r(vec
->el
, vec
->el
, m
, vec
->size
);
359 __isl_give isl_vec
*isl_vec_add(__isl_take isl_vec
*vec1
,
360 __isl_take isl_vec
*vec2
)
362 vec1
= isl_vec_cow(vec1
);
366 isl_assert(vec1
->ctx
, vec1
->size
== vec2
->size
, goto error
);
368 isl_seq_combine(vec1
->el
, vec1
->ctx
->one
, vec1
->el
,
369 vec1
->ctx
->one
, vec2
->el
, vec1
->size
);
379 static int qsort_int_cmp(const void *p1
, const void *p2
)
381 const isl_int
*i1
= (const isl_int
*) p1
;
382 const isl_int
*i2
= (const isl_int
*) p2
;
384 return isl_int_cmp(*i1
, *i2
);
387 __isl_give isl_vec
*isl_vec_sort(__isl_take isl_vec
*vec
)
392 qsort(vec
->el
, vec
->size
, sizeof(*vec
->el
), &qsort_int_cmp
);
397 __isl_give isl_vec
*isl_vec_drop_els(__isl_take isl_vec
*vec
,
398 unsigned pos
, unsigned n
)
402 vec
= isl_vec_cow(vec
);
406 if (pos
+ n
> vec
->size
)
407 isl_die(vec
->ctx
, isl_error_invalid
,
408 "range out of bounds", goto error
);
410 if (pos
+ n
!= vec
->size
)
411 isl_seq_cpy(vec
->el
+ pos
, vec
->el
+ pos
+ n
,
412 vec
->size
- pos
- n
);
422 __isl_give isl_vec
*isl_vec_insert_els(__isl_take isl_vec
*vec
,
423 unsigned pos
, unsigned n
)
433 isl_die(vec
->ctx
, isl_error_invalid
,
434 "position out of bounds", goto error
);
436 ext
= isl_vec_alloc(vec
->ctx
, vec
->size
+ n
);
440 isl_seq_cpy(ext
->el
, vec
->el
, pos
);
441 isl_seq_cpy(ext
->el
+ pos
+ n
, vec
->el
+ pos
, vec
->size
- pos
);
451 __isl_give isl_vec
*isl_vec_insert_zero_els(__isl_take isl_vec
*vec
,
452 unsigned pos
, unsigned n
)
454 vec
= isl_vec_insert_els(vec
, pos
, n
);
458 isl_seq_clr(vec
->el
+ pos
, n
);