1/* 2 * Copyright 2008-2009 Katholieke Universiteit Leuven 3 * 4 * Use of this software is governed by the MIT license 5 * 6 * Written by Sven Verdoolaege, K.U.Leuven, Departement 7 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium 8 */ 9 10#include <assert.h> 11#include <isl_map_private.h> 12#include "isl_equalities.h" 13#include <isl/seq.h> 14#include "isl_scan.h" 15#include <isl_mat_private.h> 16 17/* The input of this program is the same as that of the "polytope_scan" 18 * program from the barvinok distribution. 19 * 20 * Constraints of set is PolyLib format. 21 * 22 * The input set is assumed to be bounded. 23 */ 24 25struct scan_samples { 26 struct isl_scan_callback callback; 27 struct isl_mat *samples; 28}; 29 30static int scan_samples_add_sample(struct isl_scan_callback *cb, 31 __isl_take isl_vec *sample) 32{ 33 struct scan_samples *ss = (struct scan_samples *)cb; 34 35 ss->samples = isl_mat_extend(ss->samples, ss->samples->n_row + 1, 36 ss->samples->n_col); 37 if (!ss->samples) 38 goto error; 39 40 isl_seq_cpy(ss->samples->row[ss->samples->n_row - 1], 41 sample->el, sample->size); 42 43 isl_vec_free(sample); 44 return 0; 45error: 46 isl_vec_free(sample); 47 return -1; 48} 49 50static struct isl_mat *isl_basic_set_scan_samples(struct isl_basic_set *bset) 51{ 52 isl_ctx *ctx; 53 unsigned dim; 54 struct scan_samples ss; 55 56 ctx = isl_basic_set_get_ctx(bset); 57 dim = isl_basic_set_total_dim(bset); 58 ss.callback.add = scan_samples_add_sample; 59 ss.samples = isl_mat_alloc(ctx, 0, 1 + dim); 60 if (!ss.samples) 61 goto error; 62 63 if (isl_basic_set_scan(bset, &ss.callback) < 0) { 64 isl_mat_free(ss.samples); 65 return NULL; 66 } 67 68 return ss.samples; 69error: 70 isl_basic_set_free(bset); 71 return NULL; 72} 73 74static struct isl_mat *isl_basic_set_samples(struct isl_basic_set *bset) 75{ 76 struct isl_mat *T; 77 struct isl_mat *samples; 78 79 if (!bset) 80 return NULL; 81 82 if (bset->n_eq == 0) 83 return isl_basic_set_scan_samples(bset); 84 85 bset = isl_basic_set_remove_equalities(bset, &T, NULL); 86 samples = isl_basic_set_scan_samples(bset); 87 return isl_mat_product(samples, isl_mat_transpose(T)); 88} 89 90int main(int argc, char **argv) 91{ 92 struct isl_ctx *ctx = isl_ctx_alloc(); 93 struct isl_basic_set *bset; 94 struct isl_mat *samples; 95 96 bset = isl_basic_set_read_from_file(ctx, stdin); 97 samples = isl_basic_set_samples(bset); 98 isl_mat_print_internal(samples, stdout, 0); 99 isl_mat_free(samples); 100 isl_ctx_free(ctx); 101 102 return 0; 103} 104