isl_div.c

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