Blame isl-0.14/polytope_scan.c

Packit fb9d21
/*
Packit fb9d21
 * Copyright 2008-2009 Katholieke Universiteit Leuven
Packit fb9d21
 *
Packit fb9d21
 * Use of this software is governed by the MIT license
Packit fb9d21
 *
Packit fb9d21
 * Written by Sven Verdoolaege, K.U.Leuven, Departement
Packit fb9d21
 * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
Packit fb9d21
 */
Packit fb9d21
Packit fb9d21
#include <assert.h>
Packit fb9d21
#include <isl_map_private.h>
Packit fb9d21
#include "isl_equalities.h"
Packit fb9d21
#include <isl_seq.h>
Packit fb9d21
#include "isl_scan.h"
Packit fb9d21
#include <isl_mat_private.h>
Packit fb9d21
#include <isl_vec_private.h>
Packit fb9d21
Packit fb9d21
/* The input of this program is the same as that of the "polytope_scan"
Packit fb9d21
 * program from the barvinok distribution.
Packit fb9d21
 *
Packit fb9d21
 * Constraints of set is PolyLib format.
Packit fb9d21
 *
Packit fb9d21
 * The input set is assumed to be bounded.
Packit fb9d21
 */
Packit fb9d21
Packit fb9d21
struct scan_samples {
Packit fb9d21
	struct isl_scan_callback callback;
Packit fb9d21
	struct isl_mat *samples;
Packit fb9d21
};
Packit fb9d21
Packit fb9d21
static int scan_samples_add_sample(struct isl_scan_callback *cb,
Packit fb9d21
	__isl_take isl_vec *sample)
Packit fb9d21
{
Packit fb9d21
	struct scan_samples *ss = (struct scan_samples *)cb;
Packit fb9d21
Packit fb9d21
	ss->samples = isl_mat_extend(ss->samples, ss->samples->n_row + 1,
Packit fb9d21
						  ss->samples->n_col);
Packit fb9d21
	if (!ss->samples)
Packit fb9d21
		goto error;
Packit fb9d21
Packit fb9d21
	isl_seq_cpy(ss->samples->row[ss->samples->n_row - 1],
Packit fb9d21
		    sample->el, sample->size);
Packit fb9d21
Packit fb9d21
	isl_vec_free(sample);
Packit fb9d21
	return 0;
Packit fb9d21
error:
Packit fb9d21
	isl_vec_free(sample);
Packit fb9d21
	return -1;
Packit fb9d21
}
Packit fb9d21
Packit fb9d21
static struct isl_mat *isl_basic_set_scan_samples(struct isl_basic_set *bset)
Packit fb9d21
{
Packit fb9d21
	isl_ctx *ctx;
Packit fb9d21
	unsigned dim;
Packit fb9d21
	struct scan_samples ss;
Packit fb9d21
Packit fb9d21
	ctx = isl_basic_set_get_ctx(bset);
Packit fb9d21
	dim = isl_basic_set_total_dim(bset);
Packit fb9d21
	ss.callback.add = scan_samples_add_sample;
Packit fb9d21
	ss.samples = isl_mat_alloc(ctx, 0, 1 + dim);
Packit fb9d21
	if (!ss.samples)
Packit fb9d21
		goto error;
Packit fb9d21
Packit fb9d21
	if (isl_basic_set_scan(bset, &ss.callback) < 0) {
Packit fb9d21
		isl_mat_free(ss.samples);
Packit fb9d21
		return NULL;
Packit fb9d21
	}
Packit fb9d21
Packit fb9d21
	return ss.samples;
Packit fb9d21
error:
Packit fb9d21
	isl_basic_set_free(bset);
Packit fb9d21
	return NULL;
Packit fb9d21
}
Packit fb9d21
Packit fb9d21
static struct isl_mat *isl_basic_set_samples(struct isl_basic_set *bset)
Packit fb9d21
{
Packit fb9d21
	struct isl_mat *T;
Packit fb9d21
	struct isl_mat *samples;
Packit fb9d21
Packit fb9d21
	if (!bset)
Packit fb9d21
		return NULL;
Packit fb9d21
Packit fb9d21
	if (bset->n_eq == 0)
Packit fb9d21
		return isl_basic_set_scan_samples(bset);
Packit fb9d21
Packit fb9d21
	bset = isl_basic_set_remove_equalities(bset, &T, NULL);
Packit fb9d21
	samples = isl_basic_set_scan_samples(bset);
Packit fb9d21
	return isl_mat_product(samples, isl_mat_transpose(T));
Packit fb9d21
}
Packit fb9d21
Packit fb9d21
int main(int argc, char **argv)
Packit fb9d21
{
Packit fb9d21
	struct isl_ctx *ctx = isl_ctx_alloc();
Packit fb9d21
	struct isl_basic_set *bset;
Packit fb9d21
	struct isl_mat *samples;
Packit fb9d21
Packit fb9d21
	bset = isl_basic_set_read_from_file(ctx, stdin);
Packit fb9d21
	samples = isl_basic_set_samples(bset);
Packit fb9d21
	isl_mat_print_internal(samples, stdout, 0);
Packit fb9d21
	isl_mat_free(samples);
Packit fb9d21
	isl_ctx_free(ctx);
Packit fb9d21
Packit fb9d21
	return 0;
Packit fb9d21
}