2 #include <bernstein/bernstein.h>
3 #include <bernstein/piecewise_lst.h>
4 #include <barvinok/barvinok.h>
5 #include <barvinok/util.h>
6 #include <barvinok/bernstein.h>
7 #include <barvinok/options.h>
10 using namespace bernstein
;
19 ex
evalue2ex(evalue
*e
, const exvector
& vars
)
21 if (value_notzero_p(e
->d
))
22 return value2numeric(e
->x
.n
)/value2numeric(e
->d
);
23 if (e
->x
.p
->type
!= polynomial
)
26 for (int i
= e
->x
.p
->size
-1; i
>= 0; --i
) {
27 poly
*= vars
[e
->x
.p
->pos
-1];
28 ex t
= evalue2ex(&e
->x
.p
->arr
[i
], vars
);
29 if (is_exactly_a
<fail
>(t
))
36 static int type_offset(enode
*p
)
38 return p
->type
== fractional
? 1 :
39 p
->type
== flooring
? 1 : 0;
42 typedef pair
<bool, const evalue
*> typed_evalue
;
44 static ex
evalue2ex_add_var(evalue
*e
, exvector
& extravar
,
45 vector
<typed_evalue
>& expr
, bool is_fract
)
49 for (int i
= 0; i
< expr
.size(); ++i
) {
50 if (is_fract
== expr
[i
].first
&& eequal(e
, expr
[i
].second
)) {
51 base_var
= extravar
[i
];
59 snprintf(name
, sizeof(name
), "f%c%d", is_fract
? 'r' : 'l', expr
.size());
60 extravar
.push_back(base_var
= symbol(name
));
61 expr
.push_back(typed_evalue(is_fract
, e
));
66 /* For the argument e=(f/d) of a fractional, return (d-1)/d times
67 * a variable in [0,1] (see setup_constraints).
69 static ex
evalue2ex_get_fract(evalue
*e
, exvector
& extravar
,
70 vector
<typed_evalue
>& expr
)
78 den
= value2numeric(d
);
82 ex base_var
= evalue2ex_add_var(e
, extravar
, expr
, true);
87 static ex
evalue2ex_r(const evalue
*e
, const exvector
& vars
,
88 exvector
& extravar
, vector
<typed_evalue
>& expr
,
91 if (value_notzero_p(e
->d
))
92 return value2numeric(e
->x
.n
)/value2numeric(e
->d
);
96 switch (e
->x
.p
->type
) {
98 base_var
= vars
[e
->x
.p
->pos
-1];
101 base_var
= evalue2ex_add_var(&e
->x
.p
->arr
[0], extravar
, expr
, false);
104 base_var
= evalue2ex_get_fract(&e
->x
.p
->arr
[0], extravar
, expr
);
108 return evalue2ex_r(&e
->x
.p
->arr
[VALUE_TO_INT(coset
->p
[e
->x
.p
->pos
-1])],
109 vars
, extravar
, expr
, coset
);
114 int offset
= type_offset(e
->x
.p
);
115 for (int i
= e
->x
.p
->size
-1; i
>= offset
; --i
) {
117 ex t
= evalue2ex_r(&e
->x
.p
->arr
[i
], vars
, extravar
, expr
, coset
);
118 if (is_exactly_a
<fail
>(t
))
125 /* For each t = floor(e/d), set up two constraints
128 * -e + d t + d-1 >= 0
130 * e is assumed to be an affine expression.
132 * For each t = fract(e/d), set up two constraints
137 static Matrix
*setup_constraints(const vector
<typed_evalue
> expr
, int nvar
)
139 int extra
= expr
.size();
142 Matrix
*M
= Matrix_Alloc(2*extra
, 1+extra
+nvar
+1);
143 for (int i
= 0; i
< extra
; ++i
) {
145 value_set_si(M
->p
[2*i
][0], 1);
146 value_set_si(M
->p
[2*i
][1+i
], -1);
147 value_set_si(M
->p
[2*i
][1+extra
+nvar
], 1);
148 value_set_si(M
->p
[2*i
+1][0], 1);
149 value_set_si(M
->p
[2*i
+1][1+i
], 1);
151 Value
*d
= &M
->p
[2*i
][1+i
];
153 evalue_denom(expr
[i
].second
, d
);
155 for (e
= expr
[i
].second
; value_zero_p(e
->d
); e
= &e
->x
.p
->arr
[0]) {
156 assert(e
->x
.p
->type
== polynomial
);
157 assert(e
->x
.p
->size
== 2);
158 evalue
*c
= &e
->x
.p
->arr
[1];
159 value_multiply(M
->p
[2*i
][1+extra
+e
->x
.p
->pos
-1], *d
, c
->x
.n
);
160 value_division(M
->p
[2*i
][1+extra
+e
->x
.p
->pos
-1],
161 M
->p
[2*i
][1+extra
+e
->x
.p
->pos
-1], c
->d
);
163 value_multiply(M
->p
[2*i
][1+extra
+nvar
], *d
, e
->x
.n
);
164 value_division(M
->p
[2*i
][1+extra
+nvar
], M
->p
[2*i
][1+extra
+nvar
], e
->d
);
165 value_oppose(*d
, *d
);
166 value_set_si(M
->p
[2*i
][0], -1);
167 Vector_Scale(M
->p
[2*i
], M
->p
[2*i
+1], M
->p
[2*i
][0], 1+extra
+nvar
+1);
168 value_set_si(M
->p
[2*i
][0], 1);
169 value_subtract(M
->p
[2*i
+1][1+extra
+nvar
], M
->p
[2*i
+1][1+extra
+nvar
], *d
);
170 value_decrement(M
->p
[2*i
+1][1+extra
+nvar
], M
->p
[2*i
+1][1+extra
+nvar
]);
176 static bool evalue_is_periodic(const evalue
*e
, Vector
*periods
)
179 bool is_periodic
= false;
181 if (value_notzero_p(e
->d
))
184 assert(e
->x
.p
->type
!= partition
);
185 if (e
->x
.p
->type
== periodic
) {
188 value_set_si(size
, e
->x
.p
->size
);
189 value_lcm(periods
->p
[e
->x
.p
->pos
-1], size
, &periods
->p
[e
->x
.p
->pos
-1]);
193 offset
= type_offset(e
->x
.p
);
194 for (i
= e
->x
.p
->size
-1; i
>= offset
; --i
)
195 is_periodic
= evalue_is_periodic(&e
->x
.p
->arr
[i
], periods
) || is_periodic
;
199 static ex
evalue2lst(const evalue
*e
, const exvector
& vars
,
200 exvector
& extravar
, vector
<typed_evalue
>& expr
,
203 Vector
*coset
= Vector_Alloc(periods
->Size
);
207 list
.append(evalue2ex_r(e
, vars
, extravar
, expr
, coset
));
208 for (i
= coset
->Size
-1; i
>= 0; --i
) {
209 value_increment(coset
->p
[i
], coset
->p
[i
]);
210 if (value_lt(coset
->p
[i
], periods
->p
[i
]))
212 value_set_si(coset
->p
[i
], 0);
221 ex
evalue2ex(const evalue
*e
, const exvector
& vars
, exvector
& floorvar
,
222 Matrix
**C
, Vector
**p
)
224 vector
<typed_evalue
> expr
;
225 Vector
*periods
= Vector_Alloc(vars
.size());
228 for (int i
= 0; i
< periods
->Size
; ++i
)
229 value_set_si(periods
->p
[i
], 1);
230 if (evalue_is_periodic(e
, periods
)) {
236 Vector_Free(periods
);
238 ex poly
= evalue2ex_r(e
, vars
, floorvar
, expr
, NULL
);
239 Matrix
*M
= setup_constraints(expr
, vars
.size());
245 /* if the evalue is a relation, we use the relation to cut off the
246 * the edges of the domain
248 static void evalue_extract_poly(evalue
*e
, int i
, Polyhedron
**D
, evalue
**poly
,
251 *D
= EVALUE_DOMAIN(e
->x
.p
->arr
[2*i
]);
252 *poly
= e
= &e
->x
.p
->arr
[2*i
+1];
253 if (value_notzero_p(e
->d
))
255 if (e
->x
.p
->type
!= relation
)
257 if (e
->x
.p
->size
> 2)
259 evalue
*fr
= &e
->x
.p
->arr
[0];
260 assert(value_zero_p(fr
->d
));
261 assert(fr
->x
.p
->type
== fractional
);
262 assert(fr
->x
.p
->size
== 3);
263 Matrix
*T
= Matrix_Alloc(2, (*D
)->Dimension
+1);
264 value_set_si(T
->p
[1][(*D
)->Dimension
], 1);
266 /* convert argument of fractional to polylib */
267 /* the argument is assumed to be linear */
268 evalue
*p
= &fr
->x
.p
->arr
[0];
269 evalue_denom(p
, &T
->p
[1][(*D
)->Dimension
]);
270 for (;value_zero_p(p
->d
); p
= &p
->x
.p
->arr
[0]) {
271 assert(p
->x
.p
->type
== polynomial
);
272 assert(p
->x
.p
->size
== 2);
273 assert(value_notzero_p(p
->x
.p
->arr
[1].d
));
274 int pos
= p
->x
.p
->pos
- 1;
275 value_assign(T
->p
[0][pos
], p
->x
.p
->arr
[1].x
.n
);
276 value_multiply(T
->p
[0][pos
], T
->p
[0][pos
], T
->p
[1][(*D
)->Dimension
]);
277 value_division(T
->p
[0][pos
], T
->p
[0][pos
], p
->x
.p
->arr
[1].d
);
279 int pos
= (*D
)->Dimension
;
280 value_assign(T
->p
[0][pos
], p
->x
.n
);
281 value_multiply(T
->p
[0][pos
], T
->p
[0][pos
], T
->p
[1][(*D
)->Dimension
]);
282 value_division(T
->p
[0][pos
], T
->p
[0][pos
], p
->d
);
284 Polyhedron
*E
= NULL
;
285 for (Polyhedron
*P
= *D
; P
; P
= P
->next
) {
286 Polyhedron
*I
= Polyhedron_Image(P
, T
, MaxRays
);
287 I
= DomainConstraintSimplify(I
, MaxRays
);
288 Polyhedron
*R
= Polyhedron_Preimage(I
, T
, MaxRays
);
290 Polyhedron
*next
= P
->next
;
292 Polyhedron
*S
= DomainIntersection(P
, R
, MaxRays
);
298 E
= DomainConcat(S
, E
);
303 *poly
= &e
->x
.p
->arr
[1];
306 piecewise_lst
*evalue_bernstein_coefficients(piecewise_lst
*pl_all
, evalue
*e
,
307 Polyhedron
*ctx
, const exvector
& params
)
310 barvinok_options
*options
= barvinok_options_new_with_defaults();
311 pl
= evalue_bernstein_coefficients(pl_all
, e
, ctx
, params
, options
);
312 barvinok_options_free(options
);
316 static piecewise_lst
*bernstein_coefficients(piecewise_lst
*pl_all
,
317 Polyhedron
*D
, const ex
& poly
,
319 const exvector
& params
, const exvector
& floorvar
,
320 barvinok_options
*options
);
322 static piecewise_lst
*bernstein_coefficients_product(piecewise_lst
*pl_all
,
323 Polyhedron
*F
, Matrix
*T
, const ex
& poly
,
325 const exvector
& params
, const exvector
& vars
,
326 barvinok_options
*options
)
330 for (Polyhedron
*G
= F
; G
; G
= G
->next
)
334 unsigned nparam
= params
.size();
335 unsigned nvar
= vars
.size();
336 unsigned constraints
;
337 Polyhedron
*C
= NULL
;
339 /* More context constraints */
340 if (F
->Dimension
== ctx
->Dimension
) {
350 M
= Matrix_Alloc(F
->NbConstraints
, 1+nvar
+nparam
+1);
351 for (int i
= 0; i
< F
->NbConstraints
; ++i
) {
352 Vector_Copy(F
->Constraint
[i
], M
->p
[i
], 1+F
->Dimension
-nparam
);
353 Vector_Copy(F
->Constraint
[i
]+1+F
->Dimension
-nparam
,
354 M
->p
[i
]+1+nvar
, nparam
+1);
356 P
= Constraints2Polyhedron(M
, options
->MaxRays
);
359 constraints
= C
? C
->NbConstraints
: 0;
360 constraints
+= ctx
->NbConstraints
;
361 for (Polyhedron
*G
= F
->next
; G
; G
= G
->next
)
362 constraints
+= G
->NbConstraints
;
364 unsigned total_var
= nvar
-(F
->Dimension
-nparam
);
367 M
= Matrix_Alloc(constraints
, 1+total_var
+nparam
+1);
368 for (Polyhedron
*G
= F
->next
; G
; G
= G
->next
) {
369 unsigned this_var
= G
->Dimension
- nparam
;
370 for (int i
= 0; i
< G
->NbConstraints
; ++i
) {
371 value_assign(M
->p
[c
+i
][0], G
->Constraint
[i
][0]);
372 Vector_Copy(G
->Constraint
[i
]+1, M
->p
[c
+i
]+1+skip
, this_var
);
373 Vector_Copy(G
->Constraint
[i
]+1+this_var
, M
->p
[c
+i
]+1+total_var
,
376 c
+= G
->NbConstraints
;
379 assert(skip
== total_var
);
381 for (int i
= 0; i
< C
->NbConstraints
; ++i
) {
382 value_assign(M
->p
[c
+i
][0], C
->Constraint
[i
][0]);
383 Vector_Copy(C
->Constraint
[i
]+1, M
->p
[c
+i
]+1+total_var
,
386 c
+= C
->NbConstraints
;
388 for (int i
= 0; i
< ctx
->NbConstraints
; ++i
) {
389 value_assign(M
->p
[c
+i
][0], ctx
->Constraint
[i
][0]);
390 Vector_Copy(ctx
->Constraint
[i
]+1, M
->p
[c
+i
]+1+total_var
, nparam
+1);
392 PC
= Constraints2Polyhedron(M
, options
->MaxRays
);
395 exvector newvars
= constructVariableVector(nvar
, "t");
396 matrix
subs(1, nvar
);
397 for (int i
= 0; i
< nvar
; ++i
)
398 for (int j
= 0; j
< nvar
; ++j
)
399 subs(0,i
) += value2numeric(T
->p
[i
][j
]) * newvars
[j
];
401 ex newpoly
= replaceVariablesInPolynomial(poly
, vars
, subs
);
404 P_vars
.insert(P_vars
.end(), newvars
.begin(),
405 newvars
.begin()+(F
->Dimension
-nparam
));
407 P_params
.insert(P_params
.end(), newvars
.begin()+(F
->Dimension
-nparam
),
409 P_params
.insert(P_params
.end(), params
.begin(), params
.end());
411 pl
= bernstein_coefficients(NULL
, P
, newpoly
, PC
, P_params
, P_vars
, options
);
415 unsigned done
= F
->Dimension
-nparam
;
416 for (F
= F
->next
; F
; F
= F
->next
) {
418 pl_vars
.insert(pl_vars
.end(), newvars
.begin()+done
,
419 newvars
.begin()+done
+(F
->Dimension
-nparam
));
421 pl_params
.insert(pl_params
.end(), newvars
.begin()+done
+(F
->Dimension
-nparam
),
423 pl_params
.insert(pl_params
.end(), params
.begin(), params
.end());
424 piecewise_lst
*new_pl
= NULL
;
425 Polyhedron
*U
= Universe_Polyhedron(pl_params
.size());
427 for (int i
= 0; i
< pl
->list
.size(); ++i
) {
428 Polyhedron
*D
= pl
->list
[i
].first
;
429 lst polys
= pl
->list
[i
].second
;
430 lst::const_iterator j
;
431 for (j
= polys
.begin(); j
!= polys
.end(); ++j
) {
432 new_pl
= bernstein_coefficients(new_pl
, D
, *j
, U
, pl_params
,
441 done
+= F
->Dimension
-nparam
;
447 pl_all
->combine(*pl
);
454 static piecewise_lst
*bernstein_coefficients_polyhedron(piecewise_lst
*pl_all
,
455 Polyhedron
*P
, const ex
& poly
,
457 const exvector
& params
, const exvector
& floorvar
,
458 barvinok_options
*options
)
460 if (Polyhedron_is_unbounded(P
, ctx
->Dimension
, options
->MaxRays
)) {
461 fprintf(stderr
, "warning: unbounded domain skipped\n");
462 Polyhedron_Print(stderr
, P_VALUE_FMT
, P
);
467 Polyhedron
*F
= Polyhedron_Factor(P
, ctx
->Dimension
, &T
, options
->MaxRays
);
469 pl_all
= bernstein_coefficients_product(pl_all
, F
, T
, poly
, ctx
, params
,
476 unsigned PP_MaxRays
= options
->MaxRays
;
477 if (PP_MaxRays
& POL_NO_DUAL
)
480 Param_Polyhedron
*PP
= Polyhedron2Param_Domain(P
, ctx
, PP_MaxRays
);
482 piecewise_lst
*pl
= new piecewise_lst(params
);
483 for (Param_Domain
*Q
= PP
->D
; Q
; Q
= Q
->next
) {
484 matrix VM
= domainVertices(PP
, Q
, params
);
485 lst coeffs
= bernsteinExpansion(VM
, poly
, floorvar
, params
);
486 pl
->list
.push_back(guarded_lst(Polyhedron_Copy(Q
->Domain
), coeffs
));
488 Param_Polyhedron_Free(PP
);
492 pl_all
->combine(*pl
);
499 static piecewise_lst
*bernstein_coefficients(piecewise_lst
*pl_all
,
500 Polyhedron
*D
, const ex
& poly
,
502 const exvector
& params
, const exvector
& floorvar
,
503 barvinok_options
*options
)
505 if (!D
->next
&& emptyQ2(D
))
508 for (Polyhedron
*P
= D
; P
; P
= P
->next
) {
509 /* This shouldn't happen */
512 Polyhedron
*next
= P
->next
;
514 pl_all
= bernstein_coefficients_polyhedron(pl_all
, P
, poly
, ctx
,
515 params
, floorvar
, options
);
521 /* Compute the coefficients of the polynomial corresponding to each coset
522 * on its own domain. This allows us to cut the domain on multiples of
524 * To perform the cutting for a coset "i mod n = c" we map the domain
525 * to the quotient space trough "i = i' n + c", simplify the constraints
526 * (implicitly) and then map back to the original space.
528 static piecewise_lst
*bernstein_coefficients_periodic(piecewise_lst
*pl_all
,
529 Polyhedron
*D
, const evalue
*e
,
530 Polyhedron
*ctx
, const exvector
& vars
,
531 const exvector
& params
, Vector
*periods
,
532 barvinok_options
*options
)
534 assert(D
->Dimension
== periods
->Size
);
535 Matrix
*T
= Matrix_Alloc(D
->Dimension
+1, D
->Dimension
+1);
536 Matrix
*T2
= Matrix_Alloc(D
->Dimension
+1, D
->Dimension
+1);
537 Vector
*coset
= Vector_Alloc(periods
->Size
);
539 vector
<typed_evalue
> expr
;
540 exvector allvars
= vars
;
541 allvars
.insert(allvars
.end(), params
.begin(), params
.end());
543 value_set_si(T2
->p
[D
->Dimension
][D
->Dimension
], 1);
544 for (int i
= 0; i
< D
->Dimension
; ++i
) {
545 value_assign(T
->p
[i
][i
], periods
->p
[i
]);
546 value_lcm(T2
->p
[D
->Dimension
][D
->Dimension
], periods
->p
[i
],
547 &T2
->p
[D
->Dimension
][D
->Dimension
]);
549 value_set_si(T
->p
[D
->Dimension
][D
->Dimension
], 1);
550 for (int i
= 0; i
< D
->Dimension
; ++i
)
551 value_division(T2
->p
[i
][i
], T2
->p
[D
->Dimension
][D
->Dimension
],
555 ex poly
= evalue2ex_r(e
, allvars
, extravar
, expr
, coset
);
556 assert(extravar
.size() == 0);
557 assert(expr
.size() == 0);
558 Polyhedron
*E
= DomainPreimage(D
, T
, options
->MaxRays
);
559 Polyhedron
*F
= DomainPreimage(E
, T2
, options
->MaxRays
);
562 pl_all
= bernstein_coefficients(pl_all
, F
, poly
, ctx
, params
,
565 for (i
= D
->Dimension
-1; i
>= 0; --i
) {
566 value_increment(coset
->p
[i
], coset
->p
[i
]);
567 value_increment(T
->p
[i
][D
->Dimension
], T
->p
[i
][D
->Dimension
]);
568 value_subtract(T2
->p
[i
][D
->Dimension
], T2
->p
[i
][D
->Dimension
],
570 if (value_lt(coset
->p
[i
], periods
->p
[i
]))
572 value_set_si(coset
->p
[i
], 0);
573 value_set_si(T
->p
[i
][D
->Dimension
], 0);
574 value_set_si(T2
->p
[i
][D
->Dimension
], 0);
585 piecewise_lst
*evalue_bernstein_coefficients(piecewise_lst
*pl_all
, evalue
*e
,
586 Polyhedron
*ctx
, const exvector
& params
,
587 barvinok_options
*options
)
589 unsigned nparam
= ctx
->Dimension
;
590 if (EVALUE_IS_ZERO(*e
))
592 assert(value_zero_p(e
->d
));
593 assert(e
->x
.p
->type
== partition
);
594 assert(e
->x
.p
->size
>= 2);
595 unsigned nvars
= EVALUE_DOMAIN(e
->x
.p
->arr
[0])->Dimension
- nparam
;
597 exvector vars
= constructVariableVector(nvars
, "v");
598 exvector allvars
= vars
;
599 allvars
.insert(allvars
.end(), params
.begin(), params
.end());
601 for (int i
= 0; i
< e
->x
.p
->size
/2; ++i
) {
608 evalue_extract_poly(e
, i
, &E
, &EP
, options
->MaxRays
);
609 ex poly
= evalue2ex(EP
, allvars
, floorvar
, &M
, &periods
);
610 floorvar
.insert(floorvar
.end(), vars
.begin(), vars
.end());
612 Polyhedron
*AE
= align_context(E
, M
->NbColumns
-2, options
->MaxRays
);
613 if (E
!= EVALUE_DOMAIN(e
->x
.p
->arr
[2*i
]))
615 E
= DomainAddConstraints(AE
, M
, options
->MaxRays
);
619 if (is_exactly_a
<fail
>(poly
)) {
620 if (E
!= EVALUE_DOMAIN(e
->x
.p
->arr
[2*i
]))
626 pl_all
= bernstein_coefficients_periodic(pl_all
, E
, EP
, ctx
, vars
,
627 params
, periods
, options
);
629 pl_all
= bernstein_coefficients(pl_all
, E
, poly
, ctx
, params
,
632 Vector_Free(periods
);
633 if (E
!= EVALUE_DOMAIN(e
->x
.p
->arr
[2*i
]))