1*5971e316Smrg #include <isl_pw_macro.h>
2*5971e316Smrg
3*5971e316Smrg /* Fix the value of the given parameter or domain dimension of "pw"
4*5971e316Smrg * to be equal to "value".
5*5971e316Smrg */
FN(PW,fix_si)6*5971e316Smrg __isl_give PW *FN(PW,fix_si)(__isl_take PW *pw, enum isl_dim_type type,
7*5971e316Smrg unsigned pos, int value)
8*5971e316Smrg {
9*5971e316Smrg int i;
10*5971e316Smrg isl_size n;
11*5971e316Smrg
12*5971e316Smrg n = FN(PW,n_piece)(pw);
13*5971e316Smrg if (n < 0)
14*5971e316Smrg return FN(PW,free)(pw);
15*5971e316Smrg
16*5971e316Smrg if (type == isl_dim_out)
17*5971e316Smrg isl_die(FN(PW,get_ctx)(pw), isl_error_invalid,
18*5971e316Smrg "cannot fix output dimension", return FN(PW,free)(pw));
19*5971e316Smrg
20*5971e316Smrg if (type == isl_dim_in)
21*5971e316Smrg type = isl_dim_set;
22*5971e316Smrg
23*5971e316Smrg for (i = n - 1; i >= 0; --i) {
24*5971e316Smrg isl_set *domain;
25*5971e316Smrg
26*5971e316Smrg domain = FN(PW,take_domain_at)(pw, i);
27*5971e316Smrg domain = isl_set_fix_si(domain, type, pos, value);
28*5971e316Smrg pw = FN(PW,restore_domain_at)(pw, i, domain);
29*5971e316Smrg pw = FN(PW,exploit_equalities_and_remove_if_empty)(pw, i);
30*5971e316Smrg }
31*5971e316Smrg
32*5971e316Smrg return pw;
33*5971e316Smrg }
34*5971e316Smrg
35*5971e316Smrg /* Fix the value of the variable at position "pos" of type "type" of "pw"
36*5971e316Smrg * to be equal to "v".
37*5971e316Smrg */
FN(PW,fix_val)38*5971e316Smrg __isl_give PW *FN(PW,fix_val)(__isl_take PW *pw,
39*5971e316Smrg enum isl_dim_type type, unsigned pos, __isl_take isl_val *v)
40*5971e316Smrg {
41*5971e316Smrg int i;
42*5971e316Smrg isl_size n;
43*5971e316Smrg
44*5971e316Smrg if (!v)
45*5971e316Smrg return FN(PW,free)(pw);
46*5971e316Smrg if (!isl_val_is_int(v))
47*5971e316Smrg isl_die(FN(PW,get_ctx)(pw), isl_error_invalid,
48*5971e316Smrg "expecting integer value", goto error);
49*5971e316Smrg
50*5971e316Smrg n = FN(PW,n_piece)(pw);
51*5971e316Smrg if (n < 0)
52*5971e316Smrg goto error;
53*5971e316Smrg
54*5971e316Smrg if (type == isl_dim_in)
55*5971e316Smrg type = isl_dim_set;
56*5971e316Smrg
57*5971e316Smrg for (i = 0; i < n; ++i) {
58*5971e316Smrg isl_set *domain;
59*5971e316Smrg
60*5971e316Smrg domain = FN(PW,take_domain_at)(pw, i);
61*5971e316Smrg domain = isl_set_fix(domain, type, pos, v->n);
62*5971e316Smrg pw = FN(PW,restore_domain_at)(pw, i, domain);
63*5971e316Smrg pw = FN(PW,exploit_equalities_and_remove_if_empty)(pw, i);
64*5971e316Smrg }
65*5971e316Smrg
66*5971e316Smrg isl_val_free(v);
67*5971e316Smrg return pw;
68*5971e316Smrg error:
69*5971e316Smrg isl_val_free(v);
70*5971e316Smrg return FN(PW,free)(pw);
71*5971e316Smrg }
72