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