isl_param_util.c

   1 #include <isl_aff_polylib.h>
   2 #include <isl_set_polylib.h>
   3 #include <isl/vertices.h>
   4 #include "isl_param_util.h"
   5
   6 #define INT_BITS (sizeof(unsigned) * 8)
   7
   8 static int add_vertex(__isl_take isl_vertex *vertex, void *user)
   9 {
  10         Param_Vertices ***next_V = (Param_Vertices ***) user;
  11         Param_Vertices *V;
  12         Polyhedron *D;
  13         isl_basic_set *dom;
  14         isl_multi_aff *expr;
  15         isl_ctx *ctx;
  16
  17         ctx = isl_vertex_get_ctx(vertex);
  18
  19         dom = isl_vertex_get_domain(vertex);
  20         D = isl_basic_set_to_polylib(dom);
  21         isl_basic_set_free(dom);
  22
  23         expr = isl_vertex_get_expr(vertex);
  24
  25         V = isl_alloc_type(ctx, Param_Vertices);
  26         V->Vertex = isl_multi_aff_to_polylib(expr);
  27         V->Domain = Polyhedron2Constraints(D);
  28         V->Facets = NULL;
  29         V->next = NULL;
  30
  31         Polyhedron_Free(D);
  32         isl_multi_aff_free(expr);
  33
  34         **next_V = V;
  35         *next_V = &V->next;
  36
  37         isl_vertex_free(vertex);
  38
  39         return 0;
  40 }
  41
  42 struct bv_add_chamber_data {
  43         Param_Domain **next_D;
  44         int vertex_len;
  45         Param_Domain *dom;
  46 };
  47
  48 static int add_chamber_vertex(__isl_take isl_vertex *vertex, void *user)
  49 {
  50         int j;
  51         struct bv_add_chamber_data *data = (struct bv_add_chamber_data *)user;
  52         unsigned v;
  53
  54         v = isl_vertex_get_id(vertex);
  55         data->dom->F[v / INT_BITS] |= 1u << (INT_BITS - (v % INT_BITS) - 1);
  56
  57         isl_vertex_free(vertex);
  58
  59         return 0;
  60 }
  61
  62 static int add_chamber(__isl_take isl_cell *cell, void *user)
  63 {
  64         struct bv_add_chamber_data *data = (struct bv_add_chamber_data *)user;
  65         isl_ctx *ctx;
  66         isl_basic_set *domain;
  67
  68         ctx = isl_cell_get_ctx(cell);
  69
  70         domain = isl_cell_get_domain(cell);
  71
  72         data->dom = isl_alloc_type(ctx, Param_Domain);
  73         data->dom->Domain = isl_basic_set_to_polylib(domain);
  74         data->dom->F = isl_calloc_array(ctx, unsigned, data->vertex_len);
  75         data->dom->next = NULL;
  76
  77         isl_basic_set_free(domain);
  78
  79         *data->next_D = data->dom;
  80         data->next_D = &data->dom->next;
  81
  82         isl_cell_foreach_vertex(cell, &add_chamber_vertex, data);
  83
  84         isl_cell_free(cell);
  85
  86         return 0;
  87 }
  88
  89 Param_Polyhedron *ISL_P2PP(Polyhedron *P, Polyhedron *C,
  90                           struct barvinok_options *options)
  91 {
  92         int i, j;
  93         isl_ctx *ctx = isl_ctx_alloc();
  94         isl_space *dim;
  95         isl_basic_set *bset, *context;
  96         isl_vertices *vertices;
  97         unsigned nparam = C->Dimension;
  98         unsigned nvar = P->Dimension - nparam;
  99         Param_Polyhedron *PP = isl_calloc_type(ctx, Param_Polyhedron);
 100         Param_Vertices **next_V;
 101         struct bv_add_chamber_data data;
 102
 103         dim = isl_space_set_alloc(ctx, nparam, nvar);
 104         bset = isl_basic_set_new_from_polylib(P, dim);
 105         dim = isl_space_set_alloc(ctx, nparam, 0);
 106         context = isl_basic_set_new_from_polylib(C, dim);
 107
 108         bset = isl_basic_set_intersect(bset, context);
 109
 110         vertices = isl_basic_set_compute_vertices(bset);
 111         isl_basic_set_free(bset);
 112
 113         PP->Rays = NULL;
 114         PP->nbV = isl_vertices_get_n_vertices(vertices);
 115         PP->Constraints = Polyhedron2Constraints(P);
 116
 117         next_V = &PP->V;
 118         isl_vertices_foreach_vertex(vertices, &add_vertex, &next_V);
 119
 120         data.next_D = &PP->D;
 121         data.vertex_len = (PP->nbV + INT_BITS - 1)/INT_BITS;
 122         isl_vertices_foreach_cell(vertices, &add_chamber, &data);
 123
 124         isl_vertices_free(vertices);
 125
 126         isl_ctx_free(ctx);
 127
 128         return PP;
 129 }