polly/lib/External/isl/isl_val_gmp.c

   1 #include <string.h>
   2 #include <isl/val_gmp.h>
   3 #include <isl_val_private.h>
   4
   5 /* Return a reference to an isl_val representing the integer "z".
   6  */
   7 __isl_give isl_val *isl_val_int_from_gmp(isl_ctx *ctx, mpz_t z)
   8 {
   9         isl_val *v;
  10
  11         v = isl_val_alloc(ctx);
  12         if (!v)
  13                 return NULL;
  14
  15         isl_int_set(v->n, z);
  16         isl_int_set_si(v->d, 1);
  17
  18         return v;
  19 }
  20
  21 /* Return a reference to an isl_val representing the rational value "n"/"d".
  22  */
  23 __isl_give isl_val *isl_val_from_gmp(isl_ctx *ctx, const mpz_t n, const mpz_t d)
  24 {
  25         isl_val *v;
  26
  27         v = isl_val_alloc(ctx);
  28         if (!v)
  29                 return NULL;
  30
  31         isl_int_set(v->n, n);
  32         isl_int_set(v->d, d);
  33
  34         return isl_val_normalize(v);
  35 }
  36
  37 /* Extract the numerator of a rational value "v" in "z".
  38  *
  39  * If "v" is not a rational value, then the result is undefined.
  40  */
  41 int isl_val_get_num_gmp(__isl_keep isl_val *v, mpz_t z)
  42 {
  43         if (!v)
  44                 return -1;
  45         if (!isl_val_is_rat(v))
  46                 isl_die(isl_val_get_ctx(v), isl_error_invalid,
  47                         "expecting rational value", return -1);
  48         mpz_set(z, v->n);
  49         return 0;
  50 }
  51
  52 /* Extract the denominator of a rational value "v" in "z".
  53  *
  54  * If "v" is not a rational value, then the result is undefined.
  55  */
  56 int isl_val_get_den_gmp(__isl_keep isl_val *v, mpz_t z)
  57 {
  58         if (!v)
  59                 return -1;
  60         if (!isl_val_is_rat(v))
  61                 isl_die(isl_val_get_ctx(v), isl_error_invalid,
  62                         "expecting rational value", return -1);
  63         mpz_set(z, v->d);
  64         return 0;
  65 }
  66
  67 /* Return a reference to an isl_val representing the unsigned
  68  * integer value stored in the "n" chunks of size "size" at "chunks".
  69  * The least significant chunk is assumed to be stored first.
  70  */
  71 __isl_give isl_val *isl_val_int_from_chunks(isl_ctx *ctx, size_t n,
  72         size_t size, const void *chunks)
  73 {
  74         isl_val *v;
  75
  76         v = isl_val_alloc(ctx);
  77         if (!v)
  78                 return NULL;
  79
  80         mpz_import(v->n, n, -1, size, 0, 0, chunks);
  81         isl_int_set_si(v->d, 1);
  82
  83         return v;
  84 }
  85
  86 /* Return the number of chunks of size "size" required to
  87  * store the absolute value of the numerator of "v".
  88  */
  89 isl_size isl_val_n_abs_num_chunks(__isl_keep isl_val *v, size_t size)
  90 {
  91         if (!v)
  92                 return isl_size_error;
  93
  94         if (!isl_val_is_rat(v))
  95                 isl_die(isl_val_get_ctx(v), isl_error_invalid,
  96                         "expecting rational value", return isl_size_error);
  97
  98         size *= 8;
  99         return (mpz_sizeinbase(v->n, 2) + size - 1) / size;
 100 }
 101
 102 /* Store a representation of the absolute value of the numerator of "v"
 103  * in terms of chunks of size "size" at "chunks".
 104  * The least significant chunk is stored first.
 105  * The number of chunks in the result can be obtained by calling
 106  * isl_val_n_abs_num_chunks.  The user is responsible for allocating
 107  * enough memory to store the results.
 108  *
 109  * In the special case of a zero value, isl_val_n_abs_num_chunks will
 110  * return one, while mpz_export will not fill in any chunks.  We therefore
 111  * do it ourselves.
 112  */
 113 isl_stat isl_val_get_abs_num_chunks(__isl_keep isl_val *v, size_t size,
 114         void *chunks)
 115 {
 116         if (!v || !chunks)
 117                 return isl_stat_error;
 118
 119         if (!isl_val_is_rat(v))
 120                 isl_die(isl_val_get_ctx(v), isl_error_invalid,
 121                         "expecting rational value", return isl_stat_error);
 122
 123         mpz_export(chunks, NULL, -1, size, 0, 0, v->n);
 124         if (isl_val_is_zero(v))
 125                 memset(chunks, 0, size);
 126
 127         return isl_stat_ok;
 128 }