include/isl/int.h

   1 /*
   2  * Copyright 2008-2009 Katholieke Universiteit Leuven
   3  *
   4  * Use of this software is governed by the MIT license
   5  *
   6  * Written by Sven Verdoolaege, K.U.Leuven, Departement
   7  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
   8  */
   9
  10 #ifndef ISL_INT_H
  11 #define ISL_INT_H
  12
  13 #include <isl/hash.h>
  14 #include <string.h>
  15 #include <gmp.h>
  16 #if defined(__cplusplus)
  17 #include <iostream>
  18 #endif
  19
  20 #if defined(__cplusplus)
  21 extern "C" {
  22 #endif
  23
  24 #ifndef mp_get_memory_functions
  25 void mp_get_memory_functions(
  26                 void *(**alloc_func_ptr) (size_t),
  27                 void *(**realloc_func_ptr) (void *, size_t, size_t),
  28                 void (**free_func_ptr) (void *, size_t));
  29 #endif
  30
  31 /* isl_int is the basic integer type.  It currently always corresponds
  32  * to a gmp mpz_t, but in the future, different types such as long long
  33  * or cln::cl_I will be supported.
  34  */
  35 typedef mpz_t   isl_int;
  36
  37 #define isl_int_init(i)         mpz_init(i)
  38 #define isl_int_clear(i)        mpz_clear(i)
  39
  40 #define isl_int_set(r,i)        mpz_set(r,i)
  41 #define isl_int_set_gmp(r,i)    mpz_set(r,i)
  42 #define isl_int_set_si(r,i)     mpz_set_si(r,i)
  43 #define isl_int_set_ui(r,i)     mpz_set_ui(r,i)
  44 #define isl_int_get_gmp(i,g)    mpz_set(g,i)
  45 #define isl_int_get_si(r)       mpz_get_si(r)
  46 #define isl_int_get_ui(r)       mpz_get_ui(r)
  47 #define isl_int_get_d(r)        mpz_get_d(r)
  48 #define isl_int_get_str(r)      mpz_get_str(0, 10, r)
  49 typedef void (*isl_int_print_gmp_free_t)(void *, size_t);
  50 #define isl_int_free_str(s)                                     \
  51         do {                                                            \
  52                 isl_int_print_gmp_free_t gmp_free;                      \
  53                 mp_get_memory_functions(NULL, NULL, &gmp_free);         \
  54                 (*gmp_free)(s, strlen(s) + 1);                          \
  55         } while (0)
  56 #define isl_int_abs(r,i)        mpz_abs(r,i)
  57 #define isl_int_neg(r,i)        mpz_neg(r,i)
  58 #define isl_int_swap(i,j)       mpz_swap(i,j)
  59 #define isl_int_swap_or_set(i,j)        mpz_swap(i,j)
  60 #define isl_int_add_ui(r,i,j)   mpz_add_ui(r,i,j)
  61 #define isl_int_sub_ui(r,i,j)   mpz_sub_ui(r,i,j)
  62
  63 #define isl_int_add(r,i,j)      mpz_add(r,i,j)
  64 #define isl_int_sub(r,i,j)      mpz_sub(r,i,j)
  65 #define isl_int_mul(r,i,j)      mpz_mul(r,i,j)
  66 #define isl_int_mul_2exp(r,i,j) mpz_mul_2exp(r,i,j)
  67 #define isl_int_mul_ui(r,i,j)   mpz_mul_ui(r,i,j)
  68 #define isl_int_pow_ui(r,i,j)   mpz_pow_ui(r,i,j)
  69 #define isl_int_addmul(r,i,j)   mpz_addmul(r,i,j)
  70 #define isl_int_submul(r,i,j)   mpz_submul(r,i,j)
  71
  72 #define isl_int_gcd(r,i,j)      mpz_gcd(r,i,j)
  73 #define isl_int_lcm(r,i,j)      mpz_lcm(r,i,j)
  74 #define isl_int_divexact(r,i,j) mpz_divexact(r,i,j)
  75 #define isl_int_divexact_ui(r,i,j)      mpz_divexact_ui(r,i,j)
  76 #define isl_int_tdiv_q(r,i,j)   mpz_tdiv_q(r,i,j)
  77 #define isl_int_cdiv_q(r,i,j)   mpz_cdiv_q(r,i,j)
  78 #define isl_int_fdiv_q(r,i,j)   mpz_fdiv_q(r,i,j)
  79 #define isl_int_fdiv_r(r,i,j)   mpz_fdiv_r(r,i,j)
  80 #define isl_int_fdiv_q_ui(r,i,j)        mpz_fdiv_q_ui(r,i,j)
  81
  82 #define isl_int_read(r,s)       mpz_set_str(r,s,10)
  83 #define isl_int_print(out,i,width)                                      \
  84         do {                                                            \
  85                 char *s;                                                \
  86                 s = mpz_get_str(0, 10, i);                              \
  87                 fprintf(out, "%*s", width, s);                          \
  88                 isl_int_free_str(s);                                        \
  89         } while (0)
  90
  91 #define isl_int_sgn(i)          mpz_sgn(i)
  92 #define isl_int_cmp(i,j)        mpz_cmp(i,j)
  93 #define isl_int_cmp_si(i,si)    mpz_cmp_si(i,si)
  94 #define isl_int_eq(i,j)         (mpz_cmp(i,j) == 0)
  95 #define isl_int_ne(i,j)         (mpz_cmp(i,j) != 0)
  96 #define isl_int_lt(i,j)         (mpz_cmp(i,j) < 0)
  97 #define isl_int_le(i,j)         (mpz_cmp(i,j) <= 0)
  98 #define isl_int_gt(i,j)         (mpz_cmp(i,j) > 0)
  99 #define isl_int_ge(i,j)         (mpz_cmp(i,j) >= 0)
 100 #define isl_int_abs_eq(i,j)     (mpz_cmpabs(i,j) == 0)
 101 #define isl_int_abs_ne(i,j)     (mpz_cmpabs(i,j) != 0)
 102 #define isl_int_abs_lt(i,j)     (mpz_cmpabs(i,j) < 0)
 103 #define isl_int_abs_gt(i,j)     (mpz_cmpabs(i,j) > 0)
 104 #define isl_int_abs_ge(i,j)     (mpz_cmpabs(i,j) >= 0)
 105
 106
 107 #define isl_int_is_zero(i)      (isl_int_sgn(i) == 0)
 108 #define isl_int_is_one(i)       (isl_int_cmp_si(i,1) == 0)
 109 #define isl_int_is_negone(i)    (isl_int_cmp_si(i,-1) == 0)
 110 #define isl_int_is_pos(i)       (isl_int_sgn(i) > 0)
 111 #define isl_int_is_neg(i)       (isl_int_sgn(i) < 0)
 112 #define isl_int_is_nonpos(i)    (isl_int_sgn(i) <= 0)
 113 #define isl_int_is_nonneg(i)    (isl_int_sgn(i) >= 0)
 114 #define isl_int_is_divisible_by(i,j)    mpz_divisible_p(i,j)
 115
 116 uint32_t isl_gmp_hash(mpz_t v, uint32_t hash);
 117 #define isl_int_hash(v,h)       isl_gmp_hash(v,h)
 118
 119 #if defined(__cplusplus)
 120 }
 121 #endif
 122
 123 #if defined(__cplusplus)
 124 extern "C" { typedef void (*isl_gmp_free_t)(void *, size_t); }
 125
 126 static inline std::ostream &operator<<(std::ostream &os, isl_int i)
 127 {
 128         char *s;
 129         s = mpz_get_str(0, 10, i);
 130         os << s;
 131         isl_int_free_str(s);
 132         return os;
 133 }
 134 #endif
 135
 136 #endif