3 #include "isl_map_private.h"
6 static unsigned n(struct isl_div
*d
, enum isl_dim_type type
)
8 struct isl_dim
*dim
= d
->bmap
->dim
;
10 case isl_dim_param
: return dim
->nparam
;
11 case isl_dim_in
: return dim
->n_in
;
12 case isl_dim_out
: return dim
->n_out
;
13 case isl_dim_div
: return d
->bmap
->n_div
;
17 static unsigned offset(struct isl_div
*d
, enum isl_dim_type type
)
19 struct isl_dim
*dim
= d
->bmap
->dim
;
21 case isl_dim_param
: return 1 + 1;
22 case isl_dim_in
: return 1 + 1 + dim
->nparam
;
23 case isl_dim_out
: return 1 + 1 + dim
->nparam
+ dim
->n_in
;
24 case isl_dim_div
: return 1 + 1 + dim
->nparam
+ dim
->n_in
+ dim
->n_out
;
28 struct isl_div
*isl_basic_map_div(struct isl_basic_map
*bmap
, int pos
)
35 isl_assert(bmap
->ctx
, pos
< bmap
->n_div
, goto error
);
37 div
= isl_alloc_type(bmap
->ctx
, struct isl_div
);
42 isl_ctx_ref(div
->ctx
);
45 div
->line
= &bmap
->div
[pos
];
49 isl_basic_map_free(bmap
);
53 struct isl_div
*isl_basic_set_div(struct isl_basic_set
*bset
, int pos
)
55 return isl_basic_map_div((struct isl_basic_map
*)bset
, pos
);
58 struct isl_div
*isl_div_alloc(struct isl_dim
*dim
)
60 struct isl_basic_map
*bmap
;
65 bmap
= isl_basic_map_alloc_dim(dim
, 1, 0, 0);
69 isl_basic_map_alloc_div(bmap
);
70 isl_seq_clr(bmap
->div
[0], 1 + 1 + isl_basic_map_total_dim(bmap
));
71 return isl_basic_map_div(bmap
, 0);
74 void isl_div_free(struct isl_div
*c
)
82 isl_basic_map_free(c
->bmap
);
83 isl_ctx_deref(c
->ctx
);
87 void isl_div_get_constant(struct isl_div
*div
, isl_int
*v
)
91 isl_int_set(*v
, div
->line
[0][1]);
94 void isl_div_get_denominator(struct isl_div
*div
, isl_int
*v
)
98 isl_int_set(*v
, div
->line
[0][0]);
101 void isl_div_get_coefficient(struct isl_div
*div
,
102 enum isl_dim_type type
, int pos
, isl_int
*v
)
107 isl_assert(div
->ctx
, pos
< n(div
, type
), return);
108 isl_int_set(*v
, div
->line
[0][offset(div
, type
) + pos
]);
111 void isl_div_set_constant(struct isl_div
*div
, isl_int v
)
115 isl_int_set(div
->line
[0][1], v
);
118 void isl_div_set_denominator(struct isl_div
*div
, isl_int v
)
122 isl_int_set(div
->line
[0][0], v
);
125 void isl_div_set_coefficient(struct isl_div
*div
,
126 enum isl_dim_type type
, int pos
, isl_int v
)
131 isl_assert(div
->ctx
, pos
< n(div
, type
), return);
132 isl_int_set(div
->line
[0][offset(div
, type
) + pos
], v
);