polytope_scan.c

   1 #include <assert.h>
   2 #include <unistd.h>
   3 #include <stdlib.h>
   4 #include <strings.h>
   5 #include <barvinok/util.h>
   6 #include <barvinok/options.h>
   7 #include <barvinok/basis_reduction.h>
   8
   9 #define ALLOCN(type,n) (type*)malloc((n) * sizeof(type))
  10
  11 struct options {
  12     struct barvinok_options *barvinok;
  13     int direct;
  14 };
  15
  16 ISL_ARGS_START(struct options, options_args)
  17 ISL_ARG_CHILD(struct options, barvinok, NULL, &barvinok_options_args, NULL)
  18 ISL_ARG_BOOL(struct options, direct, 'd', "direct", 0,
  19         "don't apply generalized basis reduction first")
  20 ISL_ARGS_END
  21
  22 ISL_ARG_DEF(options, struct options, options_args)
  23
  24 static void scan_poly(Polyhedron *S, int pos, Value *z, Matrix *T)
  25 {
  26     if (!S) {
  27         int k;
  28         Vector *v;
  29
  30         v = Vector_Alloc(T->NbRows);
  31         Matrix_Vector_Product(T, z+1, v->p);
  32         value_print(stdout, VALUE_FMT, v->p[0]);
  33         for (k=1; k < pos; ++k) {
  34             printf(", ");
  35             value_print(stdout,VALUE_FMT, v->p[k]);
  36         }
  37         Vector_Free(v);
  38         printf("\n");
  39     } else {
  40         int ok;
  41         Value LB, UB, tmp;
  42         value_init(LB);
  43         value_init(UB);
  44         value_init(tmp);
  45         ok = !(lower_upper_bounds(1+pos, S, z, &LB, &UB));
  46         assert(ok);
  47         for (value_assign(tmp,LB); value_le(tmp,UB); value_increment(tmp,tmp)) {
  48             value_assign(z[pos+1], tmp);
  49             scan_poly(S->next, pos+1, z, T);
  50         }
  51         value_set_si(z[pos+1], 0);
  52         value_clear(LB);
  53         value_clear(UB);
  54         value_clear(tmp);
  55     }
  56 }
  57
  58 int main(int argc, char **argv)
  59 {
  60     Polyhedron *A, *P, *U, *S;
  61     Value *p;
  62     int i, j, ok;
  63     Matrix *basis, *T, *inv;
  64     struct options *options = options_new_with_defaults();
  65
  66     argc = options_parse(options, argc, argv, ISL_ARG_ALL);
  67
  68     A = Polyhedron_Read(options->barvinok->MaxRays);
  69
  70     if (options->direct) {
  71         inv = Identity(A->Dimension+1);
  72         P = A;
  73     } else {
  74         basis = Polyhedron_Reduced_Basis(A, options->barvinok);
  75
  76         T = Matrix_Alloc(A->Dimension+1, A->Dimension+1);
  77         inv = Matrix_Alloc(A->Dimension+1, A->Dimension+1);
  78         for (i = 0; i < A->Dimension; ++i)
  79             for (j = 0; j < A->Dimension; ++j)
  80                 value_assign(T->p[i][j], basis->p[i][j]);
  81         value_set_si(T->p[A->Dimension][A->Dimension], 1);
  82         Matrix_Free(basis);
  83
  84         ok = Matrix_Inverse(T, inv);
  85         assert(ok);
  86         Matrix_Free(T);
  87
  88         P = Polyhedron_Preimage(A, inv, options->barvinok->MaxRays);
  89         Polyhedron_Free(A);
  90     }
  91
  92     U = Universe_Polyhedron(0);
  93     S = Polyhedron_Scan(P, U, options->barvinok->MaxRays);
  94
  95     p = ALLOCN(Value, P->Dimension+2);
  96     for(i=0;i<=P->Dimension;i++) {
  97         value_init(p[i]);
  98         value_set_si(p[i],0);
  99     }
 100     value_init(p[i]);
 101     value_set_si(p[i],1);
 102
 103     scan_poly(S, 0, p, inv);
 104
 105     Matrix_Free(inv);
 106     for(i=0;i<=(P->Dimension+1);i++)
 107         value_clear(p[i]);
 108     free(p);
 109     Domain_Free(S);
 110     Polyhedron_Free(P);
 111     Polyhedron_Free(U);
 112     options_free(options);
 113     return 0;
 114 }