2 * Copyright 2011 INRIA Saclay
4 * Use of this software is governed by the GNU LGPLv2.1 license
6 * Written by Sven Verdoolaege, INRIA Saclay - Ile-de-France,
7 * Parc Club Orsay Universite, ZAC des vignes, 4 rue Jacques Monod,
11 #include <isl_aff_private.h>
12 #include <isl_local_space_private.h>
13 #include <isl_mat_private.h>
16 __isl_give isl_aff *isl_aff_alloc_vec(__isl_take isl_local_space *ls,
17 __isl_take isl_vec *v)
24 aff = isl_calloc_type(v->ctx, struct isl_aff);
34 isl_local_space_free(ls);
39 __isl_give isl_aff *isl_aff_alloc(__isl_take isl_local_space *ls)
48 ctx = isl_local_space_get_ctx(ls);
49 if (!isl_local_space_divs_known(ls))
50 isl_die(ctx, isl_error_invalid, "local space has unknown divs",
53 total = isl_local_space_dim(ls, isl_dim_all);
54 v = isl_vec_alloc(ctx, 1 + 1 + total);
55 return isl_aff_alloc_vec(ls, v);
57 isl_local_space_free(ls);
61 __isl_give isl_aff *isl_aff_zero(__isl_take isl_local_space *ls)
65 aff = isl_aff_alloc(ls);
69 isl_int_set_si(aff->v->el[0], 1);
70 isl_seq_clr(aff->v->el + 1, aff->v->size - 1);
75 __isl_give isl_aff *isl_aff_copy(__isl_keep isl_aff *aff)
84 __isl_give isl_aff *isl_aff_dup(__isl_keep isl_aff *aff)
89 return isl_aff_alloc_vec(isl_local_space_copy(aff->ls),
90 isl_vec_copy(aff->v));
93 __isl_give isl_aff *isl_aff_cow(__isl_take isl_aff *aff)
101 return isl_aff_dup(aff);
104 void *isl_aff_free(__isl_take isl_aff *aff)
112 isl_local_space_free(aff->ls);
113 isl_vec_free(aff->v);
120 isl_ctx *isl_aff_get_ctx(__isl_keep isl_aff *aff)
122 return aff ? isl_local_space_get_ctx(aff->ls) : NULL;
125 int isl_aff_dim(__isl_keep isl_aff *aff, enum isl_dim_type type)
127 return aff ? isl_local_space_dim(aff->ls, type) : 0;
130 __isl_give isl_dim *isl_aff_get_dim(__isl_keep isl_aff *aff)
132 return aff ? isl_local_space_get_dim(aff->ls) : NULL;
135 __isl_give isl_local_space *isl_aff_get_local_space(__isl_keep isl_aff *aff)
137 return aff ? isl_local_space_copy(aff->ls) : NULL;
140 const char *isl_aff_get_dim_name(__isl_keep isl_aff *aff,
141 enum isl_dim_type type, unsigned pos)
143 return aff ? isl_local_space_get_dim_name(aff->ls, type, pos) : 0;
146 int isl_aff_get_denominator(__isl_keep isl_aff *aff, isl_int *v)
150 isl_int_set(*v, aff->v->el[0]);
154 int isl_aff_get_constant(__isl_keep isl_aff *aff, isl_int *v)
158 isl_int_set(*v, aff->v->el[1]);
162 int isl_aff_get_coefficient(__isl_keep isl_aff *aff,
163 enum isl_dim_type type, int pos, isl_int *v)
168 if (pos >= isl_local_space_dim(aff->ls, type))
169 isl_die(aff->v->ctx, isl_error_invalid,
170 "position out of bounds", return -1);
172 pos += isl_local_space_offset(aff->ls, type);
173 isl_int_set(*v, aff->v->el[1 + pos]);
178 __isl_give isl_aff *isl_aff_set_denominator(__isl_take isl_aff *aff, isl_int v)
180 aff = isl_aff_cow(aff);
184 aff->v = isl_vec_cow(aff->v);
186 return isl_aff_free(aff);
188 isl_int_set(aff->v->el[0], v);
193 __isl_give isl_aff *isl_aff_set_constant(__isl_take isl_aff *aff, isl_int v)
195 aff = isl_aff_cow(aff);
199 aff->v = isl_vec_cow(aff->v);
201 return isl_aff_free(aff);
203 isl_int_set(aff->v->el[1], v);
208 __isl_give isl_aff *isl_aff_add_constant(__isl_take isl_aff *aff, isl_int v)
210 if (isl_int_is_zero(v))
213 aff = isl_aff_cow(aff);
217 aff->v = isl_vec_cow(aff->v);
219 return isl_aff_free(aff);
221 isl_int_addmul(aff->v->el[1], aff->v->el[0], v);
226 __isl_give isl_aff *isl_aff_add_constant_si(__isl_take isl_aff *aff, int v)
231 isl_int_set_si(t, v);
232 aff = isl_aff_add_constant(aff, t);
238 __isl_give isl_aff *isl_aff_set_constant_si(__isl_take isl_aff *aff, int v)
240 aff = isl_aff_cow(aff);
244 aff->v = isl_vec_cow(aff->v);
246 return isl_aff_free(aff);
248 isl_int_set_si(aff->v->el[1], v);
253 __isl_give isl_aff *isl_aff_set_coefficient(__isl_take isl_aff *aff,
254 enum isl_dim_type type, int pos, isl_int v)
259 if (pos >= isl_local_space_dim(aff->ls, type))
260 isl_die(aff->v->ctx, isl_error_invalid,
261 "position out of bounds", return isl_aff_free(aff));
263 aff = isl_aff_cow(aff);
267 aff->v = isl_vec_cow(aff->v);
269 return isl_aff_free(aff);
271 pos += isl_local_space_offset(aff->ls, type);
272 isl_int_set(aff->v->el[1 + pos], v);
277 __isl_give isl_aff *isl_aff_set_coefficient_si(__isl_take isl_aff *aff,
278 enum isl_dim_type type, int pos, int v)
283 if (pos >= isl_local_space_dim(aff->ls, type))
284 isl_die(aff->v->ctx, isl_error_invalid,
285 "position out of bounds", return isl_aff_free(aff));
287 aff = isl_aff_cow(aff);
291 aff->v = isl_vec_cow(aff->v);
293 return isl_aff_free(aff);
295 pos += isl_local_space_offset(aff->ls, type);
296 isl_int_set_si(aff->v->el[1 + pos], v);
301 __isl_give isl_aff *isl_aff_add_coefficient(__isl_take isl_aff *aff,
302 enum isl_dim_type type, int pos, isl_int v)
307 if (pos >= isl_local_space_dim(aff->ls, type))
308 isl_die(aff->v->ctx, isl_error_invalid,
309 "position out of bounds", return isl_aff_free(aff));
311 aff = isl_aff_cow(aff);
315 aff->v = isl_vec_cow(aff->v);
317 return isl_aff_free(aff);
319 pos += isl_local_space_offset(aff->ls, type);
320 isl_int_addmul(aff->v->el[1 + pos], aff->v->el[0], v);
325 __isl_give isl_aff *isl_aff_add_coefficient_si(__isl_take isl_aff *aff,
326 enum isl_dim_type type, int pos, int v)
331 isl_int_set_si(t, v);
332 aff = isl_aff_add_coefficient(aff, type, pos, t);
338 __isl_give isl_div *isl_aff_get_div(__isl_keep isl_aff *aff, int pos)
343 return isl_local_space_get_div(aff->ls, pos);
346 __isl_give isl_aff *isl_aff_neg(__isl_take isl_aff *aff)
348 aff = isl_aff_cow(aff);
351 aff->v = isl_vec_cow(aff->v);
353 return isl_aff_free(aff);
355 isl_seq_neg(aff->v->el + 1, aff->v->el + 1, aff->v->size - 1);
360 /* Given f, return floor(f).
361 * If f is an integer expression, then just return f.
362 * Otherwise, create a new div d = [f] and return the expression d.
364 __isl_give isl_aff *isl_aff_floor(__isl_take isl_aff *aff)
372 if (isl_int_is_one(aff->v->el[0]))
375 aff = isl_aff_cow(aff);
379 aff->ls = isl_local_space_add_div(aff->ls, isl_vec_copy(aff->v));
381 return isl_aff_free(aff);
383 ctx = isl_aff_get_ctx(aff);
385 isl_vec_free(aff->v);
386 aff->v = isl_vec_alloc(ctx, size + 1);
387 aff->v = isl_vec_clr(aff->v);
389 return isl_aff_free(aff);
390 isl_int_set_si(aff->v->el[0], 1);
391 isl_int_set_si(aff->v->el[size], 1);
396 /* Given f, return ceil(f).
397 * If f is an integer expression, then just return f.
398 * Otherwise, create a new div d = [-f] and return the expression -d.
400 __isl_give isl_aff *isl_aff_ceil(__isl_take isl_aff *aff)
405 if (isl_int_is_one(aff->v->el[0]))
408 aff = isl_aff_neg(aff);
409 aff = isl_aff_floor(aff);
410 aff = isl_aff_neg(aff);
415 /* Apply the expansion computed by isl_merge_divs.
416 * The expansion itself is given by "exp" while the resulting
417 * list of divs is given by "div".
419 __isl_give isl_aff *isl_aff_expand_divs( __isl_take isl_aff *aff,
420 __isl_take isl_mat *div, int *exp)
427 aff = isl_aff_cow(aff);
431 old_n_div = isl_local_space_dim(aff->ls, isl_dim_div);
432 new_n_div = isl_mat_rows(div);
433 if (new_n_div < old_n_div)
434 isl_die(isl_mat_get_ctx(div), isl_error_invalid,
435 "not an expansion", goto error);
437 aff->v = isl_vec_extend(aff->v, aff->v->size + new_n_div - old_n_div);
441 offset = 1 + isl_local_space_offset(aff->ls, isl_dim_div);
443 for (i = new_n_div - 1; i >= 0; --i) {
444 if (j >= 0 && exp[j] == i) {
446 isl_int_swap(aff->v->el[offset + i],
447 aff->v->el[offset + j]);
450 isl_int_set_si(aff->v->el[offset + i], 0);
453 aff->ls = isl_local_space_replace_divs(aff->ls, isl_mat_copy(div));
464 /* Add two affine expressions that live in the same local space.
466 static __isl_give isl_aff *add_expanded(__isl_take isl_aff *aff1,
467 __isl_take isl_aff *aff2)
471 aff1 = isl_aff_cow(aff1);
475 aff1->v = isl_vec_cow(aff1->v);
481 isl_int_gcd(gcd, aff1->v->el[0], aff2->v->el[0]);
482 isl_int_divexact(f, aff2->v->el[0], gcd);
483 isl_seq_scale(aff1->v->el + 1, aff1->v->el + 1, f, aff1->v->size - 1);
484 isl_int_divexact(f, aff1->v->el[0], gcd);
485 isl_seq_addmul(aff1->v->el + 1, f, aff2->v->el + 1, aff1->v->size - 1);
486 isl_int_divexact(f, aff2->v->el[0], gcd);
487 isl_int_mul(aff1->v->el[0], aff1->v->el[0], f);
499 __isl_give isl_aff *isl_aff_add(__isl_take isl_aff *aff1,
500 __isl_take isl_aff *aff2)
510 ctx = isl_aff_get_ctx(aff1);
511 if (!isl_dim_equal(aff1->ls->dim, aff2->ls->dim))
512 isl_die(ctx, isl_error_invalid,
513 "spaces don't match", goto error);
515 if (aff1->ls->div->n_row == 0 && aff2->ls->div->n_row == 0)
516 return add_expanded(aff1, aff2);
518 exp1 = isl_alloc_array(ctx, int, aff1->ls->div->n_row);
519 exp2 = isl_alloc_array(ctx, int, aff2->ls->div->n_row);
523 div = isl_merge_divs(aff1->ls->div, aff2->ls->div, exp1, exp2);
524 aff1 = isl_aff_expand_divs(aff1, isl_mat_copy(div), exp1);
525 aff2 = isl_aff_expand_divs(aff2, div, exp2);
529 return add_expanded(aff1, aff2);
538 __isl_give isl_aff *isl_aff_sub(__isl_take isl_aff *aff1,
539 __isl_take isl_aff *aff2)
541 return isl_aff_add(aff1, isl_aff_neg(aff2));
544 __isl_give isl_aff *isl_aff_scale(__isl_take isl_aff *aff, isl_int f)
548 if (isl_int_is_one(f))
551 aff = isl_aff_cow(aff);
554 aff->v = isl_vec_cow(aff->v);
556 return isl_aff_free(aff);
559 isl_int_gcd(gcd, aff->v->el[0], f);
560 isl_int_divexact(aff->v->el[0], aff->v->el[0], gcd);
561 isl_int_divexact(gcd, f, gcd);
562 isl_seq_scale(aff->v->el + 1, aff->v->el + 1, gcd, aff->v->size - 1);
568 __isl_give isl_aff *isl_aff_scale_down(__isl_take isl_aff *aff, isl_int f)
572 if (isl_int_is_one(f))
575 aff = isl_aff_cow(aff);
578 aff->v = isl_vec_cow(aff->v);
580 return isl_aff_free(aff);
583 isl_seq_gcd(aff->v->el + 1, aff->v->size - 1, &gcd);
584 isl_int_gcd(gcd, gcd, f);
585 isl_seq_scale_down(aff->v->el + 1, aff->v->el + 1, gcd, aff->v->size - 1);
586 isl_int_divexact(gcd, f, gcd);
587 isl_int_mul(aff->v->el[0], aff->v->el[0], gcd);
593 __isl_give isl_aff *isl_aff_scale_down_ui(__isl_take isl_aff *aff, unsigned f)
601 isl_int_set_ui(v, f);
602 aff = isl_aff_scale_down(aff, v);