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