isl_div.c

   1 #include <isl_div.h>
   2 #include <isl_map.h>
   3 #include "isl_map_private.h"
   4 #include "isl_seq.h"
   5
   6 static unsigned n(struct isl_div *d, enum isl_dim_type type)
   7 {
   8         struct isl_dim *dim = d->bmap->dim;
   9         switch (type) {
  10         case isl_dim_param:     return dim->nparam;
  11         case isl_dim_in:        return dim->n_in;
  12         case isl_dim_out:       return dim->n_out;
  13         case isl_dim_div:       return d->bmap->n_div;
  14         }
  15 }
  16
  17 static unsigned offset(struct isl_div *d, enum isl_dim_type type)
  18 {
  19         struct isl_dim *dim = d->bmap->dim;
  20         switch (type) {
  21         case isl_dim_param: return 1 + 1;
  22         case isl_dim_in:    return 1 + 1 + dim->nparam;
  23         case isl_dim_out:   return 1 + 1 + dim->nparam + dim->n_in;
  24         case isl_dim_div:   return 1 + 1 + dim->nparam + dim->n_in + dim->n_out;
  25         }
  26 }
  27
  28 struct isl_div *isl_basic_map_div(struct isl_basic_map *bmap, int pos)
  29 {
  30         struct isl_div *div;
  31
  32         if (!bmap)
  33                 goto error;
  34
  35         isl_assert(bmap->ctx, pos < bmap->n_div, goto error);
  36
  37         div = isl_alloc_type(bmap->ctx, struct isl_div);
  38         if (!div)
  39                 goto error;
  40
  41         div->ctx = bmap->ctx;
  42         isl_ctx_ref(div->ctx);
  43         div->ref = 1;
  44         div->bmap = bmap;
  45         div->line = &bmap->div[pos];
  46
  47         return div;
  48 error:
  49         isl_basic_map_free(bmap);
  50         return NULL;
  51 }
  52
  53 struct isl_div *isl_basic_set_div(struct isl_basic_set *bset, int pos)
  54 {
  55         return isl_basic_map_div((struct isl_basic_map *)bset, pos);
  56 }
  57
  58 struct isl_div *isl_div_alloc(struct isl_dim *dim)
  59 {
  60         struct isl_basic_map *bmap;
  61
  62         if (!dim)
  63                 return NULL;
  64
  65         bmap = isl_basic_map_alloc_dim(dim, 1, 0, 0);
  66         if (!bmap)
  67                 return NULL;
  68
  69         isl_basic_map_alloc_div(bmap);
  70         isl_seq_clr(bmap->div[0], 1 + 1 + isl_basic_map_total_dim(bmap));
  71         return isl_basic_map_div(bmap, 0);
  72 }
  73
  74 void isl_div_free(struct isl_div *c)
  75 {
  76         if (!c)
  77                 return;
  78
  79         if (--c->ref > 0)
  80                 return;
  81
  82         isl_basic_map_free(c->bmap);
  83         isl_ctx_deref(c->ctx);
  84         free(c);
  85 }
  86
  87 void isl_div_get_constant(struct isl_div *div, isl_int *v)
  88 {
  89         if (!div)
  90                 return;
  91         isl_int_set(*v, div->line[0][1]);
  92 }
  93
  94 void isl_div_get_denominator(struct isl_div *div, isl_int *v)
  95 {
  96         if (!div)
  97                 return;
  98         isl_int_set(*v, div->line[0][0]);
  99 }
 100
 101 void isl_div_get_coefficient(struct isl_div *div,
 102         enum isl_dim_type type, int pos, isl_int *v)
 103 {
 104         if (!div)
 105                 return;
 106
 107         isl_assert(div->ctx, pos < n(div, type), return);
 108         isl_int_set(*v, div->line[0][offset(div, type) + pos]);
 109 }
 110
 111 void isl_div_set_constant(struct isl_div *div, isl_int v)
 112 {
 113         if (!div)
 114                 return;
 115         isl_int_set(div->line[0][1], v);
 116 }
 117
 118 void isl_div_set_denominator(struct isl_div *div, isl_int v)
 119 {
 120         if (!div)
 121                 return;
 122         isl_int_set(div->line[0][0], v);
 123 }
 124
 125 void isl_div_set_coefficient(struct isl_div *div,
 126         enum isl_dim_type type, int pos, isl_int v)
 127 {
 128         if (!div)
 129                 return;
 130
 131         isl_assert(div->ctx, pos < n(div, type), return);
 132         isl_int_set(div->line[0][offset(div, type) + pos], v);
 133 }