polytope_scan.c

   1 #include <assert.h>
   2 #include "isl_equalities.h"
   3 #include "isl_seq.h"
   4 #include "isl_scan.h"
   5
   6 /* The input of this program is the same as that of the "polytope_scan"
   7  * program from the barvinok distribution.
   8  *
   9  * Constraints of set is PolyLib format.
  10  *
  11  * The input set is assumed to be bounded.
  12  */
  13
  14 struct scan_samples {
  15         struct isl_scan_callback callback;
  16         struct isl_mat *samples;
  17 };
  18
  19 static int scan_samples_add_sample(struct isl_scan_callback *cb,
  20         __isl_take isl_vec *sample)
  21 {
  22         struct scan_samples *ss = (struct scan_samples *)cb;
  23
  24         ss->samples = isl_mat_extend(ss->samples, ss->samples->n_row + 1,
  25                                                   ss->samples->n_col);
  26         if (!ss->samples)
  27                 goto error;
  28
  29         isl_seq_cpy(ss->samples->row[ss->samples->n_row - 1],
  30                     sample->el, sample->size);
  31
  32         isl_vec_free(sample);
  33         return 0;
  34 error:
  35         isl_vec_free(sample);
  36         return -1;
  37 }
  38
  39 static struct isl_mat *isl_basic_set_scan_samples(struct isl_basic_set *bset)
  40 {
  41         unsigned dim;
  42         struct scan_samples ss;
  43
  44         dim = isl_basic_set_total_dim(bset);
  45         ss.callback.add = scan_samples_add_sample;
  46         ss.samples = isl_mat_alloc(bset->ctx, 0, 1 + dim);
  47         if (!ss.samples)
  48                 goto error;
  49
  50         if (isl_basic_set_scan(bset, &ss.callback) < 0) {
  51                 isl_mat_free(ss.samples);
  52                 return NULL;
  53         }
  54
  55         return ss.samples;
  56 error:
  57         isl_basic_set_free(bset);
  58         return NULL;
  59 }
  60
  61 static struct isl_mat *isl_basic_set_samples(struct isl_basic_set *bset)
  62 {
  63         struct isl_mat *T;
  64         struct isl_mat *samples;
  65
  66         if (bset->n_eq == 0)
  67                 return isl_basic_set_scan_samples(bset);
  68
  69         bset = isl_basic_set_remove_equalities(bset, &T, NULL);
  70         samples = isl_basic_set_scan_samples(bset);
  71         return isl_mat_product(samples, isl_mat_transpose(T));
  72 }
  73
  74 int main(int argc, char **argv)
  75 {
  76         struct isl_ctx *ctx = isl_ctx_alloc();
  77         struct isl_basic_set *bset;
  78         struct isl_mat *samples;
  79
  80         bset = isl_basic_set_read_from_file(ctx, stdin, 0, ISL_FORMAT_POLYLIB);
  81         samples = isl_basic_set_samples(bset);
  82         isl_mat_dump(samples, stdout, 0);
  83         isl_mat_free(samples);
  84         isl_ctx_free(ctx);
  85
  86         return 0;
  87 }