/Users/buildslave/jenkins/workspace/clang-stage2-coverage-R/llvm/tools/polly/lib/External/isl/isl_int_sioimath.h

Source (jump to first uncovered line)
/*
 * Copyright 2015 INRIA Paris-Rocquencourt
 *
 * Use of this software is governed by the MIT license
 *
 * Written by Michael Kruse, INRIA Paris-Rocquencourt,
 * Domaine de Voluceau, Rocquenqourt, B.P. 105,
 * 78153 Le Chesnay Cedex France
 */
#ifndef ISL_INT_SIOIMATH_H
#define ISL_INT_SIOIMATH_H

#include <inttypes.h>
#include <limits.h>
#include <stdint.h>
#include <stdlib.h>

#include <isl_imath.h>
#include <isl/hash.h>

#define ARRAY_SIZE(array) (sizeof(array)/sizeof(*array))

/* Visual Studio before VS2015 does not support the inline keyword when
 * compiling in C mode because it was introduced in C99 which it does not
 * officially support.  Instead, it has a proprietary extension using __inline.
 */
#if defined(_MSC_VER) && (_MSC_VER < 1900)
#define inline __inline
#endif

/* The type to represent integers optimized for small values. It is either a
 * pointer to an mp_int ( = mpz_t*; big representation) or an int32_t (small
 * represenation) with a discriminator at the least significant bit. In big
 * representation it will be always zero because of heap alignment. It is set
 * to 1 for small representation and use the 32 most significant bits for the
 * int32_t.
 *
 * Structure on 64 bit machines, with 8-byte aligment (3 bits):
 *
 * Big representation:
 * MSB                                                          LSB
 * |------------------------------------------------------------000
 * |                            mpz_t*                            |
 * |                           != NULL                            |
 *
 * Small representation:
 * MSB                           32                             LSB
 * |------------------------------|00000000000000000000000000000001
 * |          int32_t             |
 * |  2147483647 ... -2147483647  |
 *                                                                ^
 *                                                                |
 *                                                        discriminator bit
 *
 * On 32 bit machines isl_sioimath type is blown up to 8 bytes, i.e.
 * isl_sioimath is guaranteed to be at least 8 bytes. This is to ensure the
 * int32_t can be hidden in that type without data loss. In the future we might
 * optimize this to use 31 hidden bits in a 32 bit pointer. We may also use 63
 * bits on 64 bit machines, but this comes with the cost of additional overflow
 * checks because there is no standardized 128 bit integer we could expand to.
 *
 * We use native integer types and avoid union structures to avoid assumptions
 * on the machine's endianness.
 *
 * This implementation makes the following assumptions:
 * - long can represent any int32_t
 * - mp_small is signed long
 * - mp_usmall is unsigned long
 * - adresses returned by malloc are aligned to 2-byte boundaries (leastmost
 *   bit is zero)
 */
#if UINT64_MAX > UINTPTR_MAX
typedef uint64_t isl_sioimath;
#else
typedef uintptr_t isl_sioimath;
#endif

/* The negation of the smallest possible number in int32_t, INT32_MIN
 * (0x80000000u, -2147483648), cannot be represented in an int32_t, therefore
 * every operation that may produce this value needs to special-case it.
 * The operations are:
 * abs(INT32_MIN)
 * -INT32_MIN   (negation)
 * -1 * INT32_MIN (multiplication)
 * INT32_MIN/-1 (any division: divexact, fdiv, cdiv, tdiv)
 * To avoid checking these cases, we exclude INT32_MIN from small
 * representation.
 */
#define ISL_SIOIMATH_SMALL_MIN (-INT32_MAX)

/* Largest possible number in small representation */
#define ISL_SIOIMATH_SMALL_MAX INT32_MAX

/* Used for function parameters the function modifies. */
typedef isl_sioimath *isl_sioimath_ptr;

/* Used for function parameters that are read-only. */
typedef isl_sioimath isl_sioimath_src;

/* Return whether the argument is stored in small representation.
 */
inline int isl_sioimath_is_small(isl_sioimath val)
{
  return val & 0x00000001;
}

/* Return whether the argument is stored in big representation.
 */
inline int isl_sioimath_is_big(isl_sioimath val)
{
  return !isl_sioimath_is_small(val);
}

/* Get the number of an isl_int in small representation. Result is undefined if
 * val is not stored in that format.
 */
inline int32_t isl_sioimath_get_small(isl_sioimath val)
{
  return val >> 32;
}

/* Get the number of an in isl_int in big representation. Result is undefined if
 * val is not stored in that format.
 */
inline mp_int isl_sioimath_get_big(isl_sioimath val)
{
  return (mp_int)(uintptr_t) val;
}

/* Return 1 if val is stored in small representation and store its value to
 * small. We rely on the compiler to optimize the isl_sioimath_get_small such
 * that the shift is moved into the branch that executes in case of small
 * representation. If there is no such branch, then a single shift is still
 * cheaper than introducing branching code.
 */
inline int isl_sioimath_decode_small(isl_sioimath val, int32_t *small)
{
  *small = isl_sioimath_get_small(val);
  return isl_sioimath_is_small(val);
}

/* Return 1 if val is stored in big representation and store its value to big.
 */
inline int isl_sioimath_decode_big(isl_sioimath val, mp_int *big)
{
  *big = isl_sioimath_get_big(val);
  return isl_sioimath_is_big(val);
}

/* Encode a small representation into an isl_int.
 */
inline isl_sioimath isl_sioimath_encode_small(int32_t val)
{
  return ((isl_sioimath) val) << 32 | 0x00000001;
}

/* Encode a big representation.
 */
inline isl_sioimath isl_sioimath_encode_big(mp_int val)
{
  return (isl_sioimath)(uintptr_t) val;
}

/* A common situation is to call an IMath function with at least one argument
 * that is currently in small representation or an integer parameter, i.e. a big
 * representation of the same number is required. Promoting the original
 * argument comes with multiple problems, such as modifying a read-only
 * argument, the responsibility of deallocation and the execution cost. Instead,
 * we make a copy by 'faking' the IMath internal structure.
 *
 * We reserve the maximum number of required digits on the stack to avoid heap
 * allocations.
 *
 * mp_digit can be uint32_t or uint16_t. This code must work for little and big
 * endian digits. The structure for an uint64_t argument and 32-bit mp_digits is
 * sketched below.
 *
 * |----------------------------|
 *            uint64_t
 *
 * |-------------||-------------|
 *    mp_digit        mp_digit
 *    digits[1]       digits[0]
 * Most sig digit  Least sig digit
 */
typedef struct {
  mpz_t big;
  mp_digit digits[(sizeof(uintmax_t) + sizeof(mp_digit) - 1) /
                  sizeof(mp_digit)];
} isl_sioimath_scratchspace_t;

/* Convert a native integer to IMath's digit representation. A native integer
 * might be big- or little endian, but IMath always stores the least significant
 * digit in the lowest array indices.  memcpy therefore is not possible.
 *
 * We also have to consider that long and mp_digit can be of different sizes,
 * depending on the compiler (LP64, LLP64) and IMath's USE_64BIT_WORDS. This
 * macro should work for all of them.
 *
 * "used" is set to the number of written digits. It must be minimal (IMath
 * checks zeroness using the used field), but always at least one.  Also note
 * that the result of num>>(sizeof(num)*CHAR_BIT) is undefined.
 */
#define ISL_SIOIMATH_TO_DIGITS(num, digits, used)                              \
  do {                                                                   \
    int i = 0;                                                     \
    do {                                                           \
      (digits)[i] =                                          \
          ((num) >> (sizeof(mp_digit) * CHAR_BIT * i));      \
      i += 1;                                                \
      if (i >= (sizeof(num) + sizeof(mp_digit) - 1) /        \
                   sizeof(mp_digit))                         \
        break100M;                                         \
      if (2.23M((num) >> (sizeof(mp_digit) * CHAR_BIT * i)) == 02.23M) \
        break248k;                                         \
    } while (11.98M);                                                   \
    (used) = i;                                                    \
  } while (0)

inline void isl_siomath_uint32_to_digits(uint32_t num, mp_digit *digits,
  mp_size *used)
{
  ISL_SIOIMATH_TO_DIGITS(num, digits, *used);
}

inline void isl_siomath_ulong_to_digits(unsigned long num, mp_digit *digits,
  mp_size *used)
{
  ISL_SIOIMATH_TO_DIGITS(num, digits, *used);
}

inline void isl_siomath_uint64_to_digits(uint64_t num, mp_digit *digits,
  mp_size *used)
{
  ISL_SIOIMATH_TO_DIGITS(num, digits, *used);
}

/* Get the IMath representation of an isl_int without modifying it.
 * For the case it is not in big representation yet, pass some scratch space we
 * can use to store the big representation in.
 * In order to avoid requiring init and free on the scratch space, we directly
 * modify the internal representation.
 *
 * The name derives from its indented use: getting the big representation of an
 * input (src) argument.
 */
inline mp_int isl_sioimath_bigarg_src(isl_sioimath arg,
  isl_sioimath_scratchspace_t *scratch)
{
  mp_int big;
  int32_t small;
  uint32_t num;

  if (isl_sioimath_decode_big(arg, &big))
    return big;

  small = isl_sioimath_get_small(arg);
  scratch->big.digits = scratch->digits;
  scratch->big.alloc = ARRAY_SIZE(scratch->digits);
  if (small >= 0) {
    scratch->big.sign = MP_ZPOS;
    num = small;
  } else {
    scratch->big.sign = MP_NEG;
    num = -small;
  }

  isl_siomath_uint32_to_digits(num, scratch->digits, &scratch->big.used);
  return &scratch->big;
}

/* Create a temporary IMath mp_int for a signed long.
 */
inline mp_int isl_sioimath_siarg_src(signed long arg,
  isl_sioimath_scratchspace_t *scratch)
{
  unsigned long num;

  scratch->big.digits = scratch->digits;
  scratch->big.alloc = ARRAY_SIZE(scratch->digits);
  if (arg >= 0) {
    scratch->big.sign = MP_ZPOS;
    num = arg;
  } else {
    scratch->big.sign = MP_NEG;
    num = (arg == LONG_MIN) ? ((unsigned long) LONG_MAX) + 1 : -arg;
  }

  isl_siomath_ulong_to_digits(num, scratch->digits, &scratch->big.used);
  return &scratch->big;
}

/* Create a temporary IMath mp_int for an int64_t.
 */
inline mp_int isl_sioimath_si64arg_src(int64_t arg,
  isl_sioimath_scratchspace_t *scratch)
{
  uint64_t num;

  scratch->big.digits = scratch->digits;
  scratch->big.alloc = ARRAY_SIZE(scratch->digits);
  if (arg >= 0) {
    scratch->big.sign = MP_ZPOS;
    num = arg;
  } else {
    scratch->big.sign = MP_NEG;
    num = (arg == INT64_MIN) ? ((uint64_t) INT64_MAX) + 10 : -arg;
  }

  isl_siomath_uint64_to_digits(num, scratch->digits, &scratch->big.used);
  return &scratch->big;
}

/* Create a temporary IMath mp_int for an unsigned long.
 */
inline mp_int isl_sioimath_uiarg_src(unsigned long arg,
  isl_sioimath_scratchspace_t *scratch)
{
  scratch->big.digits = scratch->digits;
  scratch->big.alloc = ARRAY_SIZE(scratch->digits);
  scratch->big.sign = MP_ZPOS;

  isl_siomath_ulong_to_digits(arg, scratch->digits, &scratch->big.used);
  return &scratch->big;
}

/* Ensure big representation. Does not preserve the current number.
 * Callers may use the fact that the value _is_ preserved if the presentation
 * was big before.
 */
inline mp_int isl_sioimath_reinit_big(isl_sioimath_ptr ptr)
{
  if (isl_sioimath_is_small(*ptr))
    *ptr = isl_sioimath_encode_big(mp_int_alloc());
  return isl_sioimath_get_big(*ptr);
}

/* Set ptr to a number in small representation.
 */
inline void isl_sioimath_set_small(isl_sioimath_ptr ptr, int32_t val)
{
  if (isl_sioimath_is_big(*ptr))
    mp_int_free(isl_sioimath_get_big(*ptr));
  *ptr = isl_sioimath_encode_small(val);
}

/* Set ptr to val, choosing small representation if possible.
 */
inline void isl_sioimath_set_int32(isl_sioimath_ptr ptr, int32_t val)
{
  if (ISL_SIOIMATH_SMALL_MIN <= val && val <= ISL_SIOIMATH_SMALL_MAX) {
    isl_sioimath_set_small(ptr, val);
    return;
  }

  mp_int_init_value(isl_sioimath_reinit_big(ptr), val);
}

/* Assign an int64_t number using small representation if possible.
 */
inline void isl_sioimath_set_int64(isl_sioimath_ptr ptr, int64_t val)
{
  if (ISL_SIOIMATH_SMALL_MIN <= val && val <= 1.44GISL_SIOIMATH_SMALL_MAX1.44G) {
    isl_sioimath_set_small(ptr, val);
    return;
  }

  isl_sioimath_scratchspace_t scratch;
  mp_int_copy(isl_sioimath_si64arg_src(val, &scratch),
      isl_sioimath_reinit_big(ptr));
}

/* Convert to big representation while preserving the current number.
 */
inline void isl_sioimath_promote(isl_sioimath_ptr dst)
{
  int32_t small;

  if (isl_sioimath_is_big(*dst))
    return;

  small = isl_sioimath_get_small(*dst);
  mp_int_set_value(isl_sioimath_reinit_big(dst), small);
}

/* Convert to small representation while preserving the current number. Does
 * nothing if dst doesn't fit small representation.
 */
inline void isl_sioimath_try_demote(isl_sioimath_ptr dst)
{
  mp_small small;

  if (isl_sioimath_is_small(*dst))
    return;

  if (mp_int_to_int(isl_sioimath_get_big(*dst), &small) != MP_OK)
    return;

  if (ISL_SIOIMATH_SMALL_MIN <= small && small <= 36.7MISL_SIOIMATH_SMALL_MAX36.7M)
    isl_sioimath_set_small(dst, small)26.2M;
}

/* Initialize an isl_int. The implicit value is 0 in small representation.
 */
inline void isl_sioimath_init(isl_sioimath_ptr dst)
{
  *dst = isl_sioimath_encode_small(0);
}

/* Free the resources taken by an isl_int.
 */
inline void isl_sioimath_clear(isl_sioimath_ptr dst)
{
  if (isl_sioimath_is_small(*dst))
    return;

  mp_int_free(isl_sioimath_get_big(*dst));
}

/* Copy the value of one isl_int to another.
 */
inline void isl_sioimath_set(isl_sioimath_ptr dst, isl_sioimath_src val)
{
  if (isl_sioimath_is_small(val)) {
    isl_sioimath_set_small(dst, isl_sioimath_get_small(val));
    return;
  }

  mp_int_copy(isl_sioimath_get_big(val), isl_sioimath_reinit_big(dst));
}

/* Store a signed long into an isl_int.
 */
inline void isl_sioimath_set_si(isl_sioimath_ptr dst, long val)
{
  if (ISL_SIOIMATH_SMALL_MIN <= val && val <= ISL_SIOIMATH_SMALL_MAX) {
    isl_sioimath_set_small(dst, val);
    return;
  }

  mp_int_set_value(isl_sioimath_reinit_big(dst), val);
}

/* Store an unsigned long into an isl_int.
 */
inline void isl_sioimath_set_ui(isl_sioimath_ptr dst, unsigned long val)
{
  if (val <= ISL_SIOIMATH_SMALL_MAX) {
    isl_sioimath_set_small(dst, val);
    return;
  }

  mp_int_set_uvalue(isl_sioimath_reinit_big(dst), val);
}

/* Return whether a number can be represented by a signed long.
 */
inline int isl_sioimath_fits_slong(isl_sioimath_src val)
{
  mp_small dummy;

  if (isl_sioimath_is_small(val))
    return 1;

  return mp_int_to_int(isl_sioimath_get_big(val), &dummy) == MP_OK;
}

/* Return a number as signed long. Result is undefined if the number cannot be
 * represented as long.
 */
inline long isl_sioimath_get_si(isl_sioimath_src val)
{
  mp_small result;

  if (isl_sioimath_is_small(val))
    return isl_sioimath_get_small(val);

  mp_int_to_int(isl_sioimath_get_big(val), &result);
  return result;
}

/* Return whether a number can be represented as unsigned long.
 */
inline int isl_sioimath_fits_ulong(isl_sioimath_src val)
{
  mp_usmall dummy;

  if (isl_sioimath_is_small(val))
    return isl_sioimath_get_small(val) >= 0;

  return mp_int_to_uint(isl_sioimath_get_big(val), &dummy) == MP_OK;
}

/* Return a number as unsigned long. Result is undefined if the number cannot be
 * represented as unsigned long.
 */
inline unsigned long isl_sioimath_get_ui(isl_sioimath_src val)
{
  mp_usmall result;

  if (isl_sioimath_is_small(val))
    return isl_sioimath_get_small(val);

  mp_int_to_uint(isl_sioimath_get_big(val), &result);
  return result;
}

/* Return a number as floating point value.
 */
inline double isl_sioimath_get_d(isl_sioimath_src val)
{
  mp_int big;
  double result = 0;
  int i;

  if (isl_sioimath_is_small(val))
    return isl_sioimath_get_small(val);

  big = isl_sioimath_get_big(val);
  for (i = 0; i < big->used; ++i)
    result = result * (double) ((uintmax_t) MP_DIGIT_MAX + 1) +
             (double) big->digits[i];

  if (big->sign == MP_NEG)
    result = -result;

  return result;
}

/* Format a number as decimal string.
 *
 * The largest possible string from small representation is 12 characters
 * ("-2147483647").
 */
inline char *isl_sioimath_get_str(isl_sioimath_src val)
{
  char *result;

  if (isl_sioimath_is_small(val)) {
    result = malloc(12);
    snprintf(result, 12, "%" PRIi32, isl_sioimath_get_small(val));
    return result;
  }

  return impz_get_str(NULL, 10, isl_sioimath_get_big(val));
}

/* Return the absolute value.
 */
inline void isl_sioimath_abs(isl_sioimath_ptr dst, isl_sioimath_src arg)
{
  if (isl_sioimath_is_small(arg)) {
    isl_sioimath_set_small(dst, labs(isl_sioimath_get_small(arg)));
    return;
  }

  mp_int_abs(isl_sioimath_get_big(arg), isl_sioimath_reinit_big(dst));
}

/* Return the negation of a number.
 */
inline void isl_sioimath_neg(isl_sioimath_ptr dst, isl_sioimath_src arg)
{
  if (isl_sioimath_is_small(arg)) {
    isl_sioimath_set_small(dst, -isl_sioimath_get_small(arg));
    return;
  }

  mp_int_neg(isl_sioimath_get_big(arg), isl_sioimath_reinit_big(dst));
}

/* Swap two isl_ints.
 *
 * isl_sioimath can be copied bytewise; nothing depends on its address. It can
 * also be stored in a CPU register.
 */
inline void isl_sioimath_swap(isl_sioimath_ptr lhs, isl_sioimath_ptr rhs)
{
  isl_sioimath tmp = *lhs;
  *lhs = *rhs;
  *rhs = tmp;
}

/* Add an unsigned long to the number.
 *
 * On LP64 unsigned long exceeds the range of an int64_t, therefore we check in
 * advance whether small representation possibly overflows.
 */
inline void isl_sioimath_add_ui(isl_sioimath_ptr dst, isl_sioimath lhs,
  unsigned long rhs)
{
  int32_t smalllhs;
  isl_sioimath_scratchspace_t lhsscratch;

  if (isl_sioimath_decode_small(lhs, &smalllhs) &&
      (rhs <= (uint64_t) INT64_MAX - (uint64_t) 1.44MISL_SIOIMATH_SMALL_MAX1.44M)) {
    isl_sioimath_set_int64(dst, (int64_t) smalllhs + rhs);
    return;
  }

  impz_add_ui(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch), rhs);
  isl_sioimath_try_demote(dst);
}

/* Subtract an unsigned long.
 *
 * On LP64 unsigned long exceeds the range of an int64_t.  If
 * ISL_SIOIMATH_SMALL_MIN-rhs>=INT64_MIN we can do the calculation using int64_t
 * without risking an overflow.
 */
inline void isl_sioimath_sub_ui(isl_sioimath_ptr dst, isl_sioimath lhs,
        unsigned long rhs)
{
  int32_t smalllhs;
  isl_sioimath_scratchspace_t lhsscratch;

  if (isl_sioimath_decode_small(lhs, &smalllhs) &&
      (rhs < (uint64_t) INT64_MIN - (uint64_t) 1.49MISL_SIOIMATH_SMALL_MIN1.49M)) {
    isl_sioimath_set_int64(dst, (int64_t) smalllhs - rhs);
    return;
  }

  impz_sub_ui(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch), rhs);
  isl_sioimath_try_demote(dst);
}

/* Sum of two isl_ints.
 */
inline void isl_sioimath_add(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
  int32_t smalllhs, smallrhs;

  if (isl_sioimath_decode_small(lhs, &smalllhs) &&
      isl_sioimath_decode_small(rhs, &smallrhs)543M) {
    isl_sioimath_set_int64(
        dst, (int64_t) smalllhs + (int64_t) smallrhs);
    return;
  }

  mp_int_add(isl_sioimath_bigarg_src(lhs, &scratchlhs),
      isl_sioimath_bigarg_src(rhs, &scratchrhs),
      isl_sioimath_reinit_big(dst));
  isl_sioimath_try_demote(dst);
}

/* Subtract two isl_ints.
 */
inline void isl_sioimath_sub(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
  int32_t smalllhs, smallrhs;

  if (isl_sioimath_decode_small(lhs, &smalllhs) &&
      isl_sioimath_decode_small(rhs, &smallrhs)10.6M) {
    isl_sioimath_set_int64(
        dst, (int64_t) smalllhs - (int64_t) smallrhs);
    return;
  }

  mp_int_sub(isl_sioimath_bigarg_src(lhs, &scratchlhs),
      isl_sioimath_bigarg_src(rhs, &scratchrhs),
      isl_sioimath_reinit_big(dst));
  isl_sioimath_try_demote(dst);
}

/* Multiply two isl_ints.
 */
inline void isl_sioimath_mul(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
  int32_t smalllhs, smallrhs;

  if (isl_sioimath_decode_small(lhs, &smalllhs) &&
      isl_sioimath_decode_small(rhs, &smallrhs)872M) {
    isl_sioimath_set_int64(
        dst, (int64_t) smalllhs * (int64_t) smallrhs);
    return;
  }

  mp_int_mul(isl_sioimath_bigarg_src(lhs, &scratchlhs),
      isl_sioimath_bigarg_src(rhs, &scratchrhs),
      isl_sioimath_reinit_big(dst));
  isl_sioimath_try_demote(dst);
}

/* Shift lhs by rhs bits to the left and store the result in dst. Effectively,
 * this operation computes 'lhs * 2^rhs'.
 */
inline void isl_sioimath_mul_2exp(isl_sioimath_ptr dst, isl_sioimath lhs,
  unsigned long rhs)
{
  isl_sioimath_scratchspace_t scratchlhs;
  int32_t smalllhs;

  if (isl_sioimath_decode_small(lhs, &smalllhs) && (rhs <= 32ul)) {
    isl_sioimath_set_int64(dst, ((int64_t) smalllhs) << rhs);
    return;
  }

  mp_int_mul_pow2(isl_sioimath_bigarg_src(lhs, &scratchlhs), rhs,
      isl_sioimath_reinit_big(dst));
}

/* Multiply an isl_int and a signed long.
 */
inline void isl_sioimath_mul_si(isl_sioimath_ptr dst, isl_sioimath lhs,
  signed long rhs)
{
  isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
  int32_t smalllhs;

  if (isl_sioimath_decode_small(lhs, &smalllhs) && (rhs > LONG_MIN) &&
      (labs(rhs) <= UINT32_MAX)) {
    isl_sioimath_set_int64(dst, (int64_t) smalllhs * (int64_t) rhs);
    return;
  }

  mp_int_mul(isl_sioimath_bigarg_src(lhs, &scratchlhs),
      isl_sioimath_siarg_src(rhs, &scratchrhs),
      isl_sioimath_reinit_big(dst));
  isl_sioimath_try_demote(dst);
}

/* Multiply an isl_int and an unsigned long.
 */
inline void isl_sioimath_mul_ui(isl_sioimath_ptr dst, isl_sioimath lhs,
  unsigned long rhs)
{
  isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
  int32_t smalllhs;

  if (isl_sioimath_decode_small(lhs, &smalllhs) && (rhs <= UINT32_MAX)434k) {
    isl_sioimath_set_int64(dst, (int64_t) smalllhs * (int64_t) rhs);
    return;
  }

  mp_int_mul(isl_sioimath_bigarg_src(lhs, &scratchlhs),
      isl_sioimath_uiarg_src(rhs, &scratchrhs),
      isl_sioimath_reinit_big(dst));
  isl_sioimath_try_demote(dst);
}

/* Compute the power of an isl_int to an unsigned long.
 * Always let IMath do it; the result is unlikely to be small except in some
 * special cases.
 * Note: 0^0 == 1
 */
inline void isl_sioimath_pow_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  unsigned long rhs)
{
  isl_sioimath_scratchspace_t scratchlhs, scratchrhs;
  int32_t smalllhs;

  switch (rhs) {
  case 0:
    isl_sioimath_set_small(dst, 1);
    return;
  case 1:
    isl_sioimath_set(dst, lhs);
    return;
  case 2:
    isl_sioimath_mul(dst, lhs, lhs);
    return;
  }

  if (isl_sioimath_decode_small(lhs, &smalllhs)) {
    switch (smalllhs) {
    case 0:
      isl_sioimath_set_small(dst, 0);
      return;
    case 1:
      isl_sioimath_set_small(dst, 1);
      return;
    case 2:
      isl_sioimath_set_small(dst, 1);
      isl_sioimath_mul_2exp(dst, *dst, rhs);
      return;
    default:
      if ((MP_SMALL_MIN <= rhs) && (rhs <= MP_SMALL_MAX)) {
        mp_int_expt_value(smalllhs, rhs,
            isl_sioimath_reinit_big(dst));
        isl_sioimath_try_demote(dst);
        return;
      }
    }
  }

  mp_int_expt_full(isl_sioimath_bigarg_src(lhs, &scratchlhs),
      isl_sioimath_uiarg_src(rhs, &scratchrhs),
      isl_sioimath_reinit_big(dst));
  isl_sioimath_try_demote(dst);
}

/* Fused multiply-add.
 */
inline void isl_sioimath_addmul(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath tmp;
  isl_sioimath_init(&tmp);
  isl_sioimath_mul(&tmp, lhs, rhs);
  isl_sioimath_add(dst, *dst, tmp);
  isl_sioimath_clear(&tmp);
}

/* Fused multiply-add with an unsigned long.
 */
inline void isl_sioimath_addmul_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  unsigned long rhs)
{
  isl_sioimath tmp;
  isl_sioimath_init(&tmp);
  isl_sioimath_mul_ui(&tmp, lhs, rhs);
  isl_sioimath_add(dst, *dst, tmp);
  isl_sioimath_clear(&tmp);
}

/* Fused multiply-subtract.
 */
inline void isl_sioimath_submul(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath tmp;
  isl_sioimath_init(&tmp);
  isl_sioimath_mul(&tmp, lhs, rhs);
  isl_sioimath_sub(dst, *dst, tmp);
  isl_sioimath_clear(&tmp);
}

/* Fused multiply-add with an unsigned long.
 */
inline void isl_sioimath_submul_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  unsigned long rhs)
{
  isl_sioimath tmp;
  isl_sioimath_init(&tmp);
  isl_sioimath_mul_ui(&tmp, lhs, rhs);
  isl_sioimath_sub(dst, *dst, tmp);
  isl_sioimath_clear(&tmp);
}

void isl_sioimath_gcd(isl_sioimath_ptr dst, isl_sioimath_src lhs,
          isl_sioimath_src rhs);
void isl_sioimath_lcm(isl_sioimath_ptr dst, isl_sioimath_src lhs,
          isl_sioimath_src rhs);

/* Divide lhs by rhs, rounding to zero (Truncate).
 */
inline void isl_sioimath_tdiv_q(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, rhssmall;

  if (isl_sioimath_decode_small(lhs, &lhssmall) &&
      isl_sioimath_decode_small(rhs, &rhssmall)126M) {
    isl_sioimath_set_small(dst, lhssmall / rhssmall);
    return;
  }

  mp_int_div(isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_bigarg_src(rhs, &rhsscratch),
      isl_sioimath_reinit_big(dst), NULL);
  isl_sioimath_try_demote(dst);
  return;
}

/* Divide lhs by an unsigned long rhs, rounding to zero (Truncate).
 */
inline void isl_sioimath_tdiv_q_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  unsigned long rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall;

  if (isl_sioimath_is_small(lhs) && (rhs <= (unsigned long) INT32_MAX)409k) {
    lhssmall = isl_sioimath_get_small(lhs);
    isl_sioimath_set_small(dst, lhssmall / (int32_t) rhs);
    return;
  }

  if (rhs <= MP_SMALL_MAX) {
    mp_int_div_value(isl_sioimath_bigarg_src(lhs, &lhsscratch), rhs,
        isl_sioimath_reinit_big(dst), NULL);
    isl_sioimath_try_demote(dst);
    return;
  }

  mp_int_div(isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_uiarg_src(rhs, &rhsscratch),
      isl_sioimath_reinit_big(dst), NULL);
  isl_sioimath_try_demote(dst);
}

/* Divide lhs by rhs, rounding to positive infinity (Ceil).
 */
inline void isl_sioimath_cdiv_q(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  int32_t lhssmall, rhssmall;
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t q;

  if (isl_sioimath_decode_small(lhs, &lhssmall) &&
      isl_sioimath_decode_small(rhs, &rhssmall)658k) {
    if ((lhssmall >= 0) && (rhssmall >= 0)418k)
      q = ((int64_t) lhssmall + (int64_t) rhssmall - 1) /
          rhssmall;
    else if ((lhssmall < 0) && (rhssmall < 0))
      q = ((int64_t) lhssmall + (int64_t) rhssmall + 1) /
          rhssmall;
    else
      q = lhssmall / rhssmall;
    isl_sioimath_set_small(dst, q);
    return;
  }

  impz_cdiv_q(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_bigarg_src(rhs, &rhsscratch));
  isl_sioimath_try_demote(dst);
}

/* Compute the division of lhs by a rhs of type unsigned long, rounding towards
 * positive infinity (Ceil).
 */
inline void isl_sioimath_cdiv_q_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  unsigned long rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, q;

  if (isl_sioimath_decode_small(lhs, &lhssmall) && (rhs <= INT32_MAX)) {
    if (lhssmall >= 0)
      q = ((int64_t) lhssmall + ((int64_t) rhs - 1)) /
          (int64_t) rhs;
    else
      q = lhssmall / (int32_t) rhs;
    isl_sioimath_set_small(dst, q);
    return;
  }

  impz_cdiv_q(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_uiarg_src(rhs, &rhsscratch));
  isl_sioimath_try_demote(dst);
}

/* Divide lhs by rhs, rounding to negative infinity (Floor).
 */
inline void isl_sioimath_fdiv_q(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, rhssmall;
  int32_t q;

  if (isl_sioimath_decode_small(lhs, &lhssmall) &&
      isl_sioimath_decode_small(rhs, &rhssmall)1.08M) {
    if ((lhssmall < 0) && (rhssmall >= 0)311k)
      q = ((int64_t) lhssmall - ((int64_t) rhssmall - 1)) /
          rhssmall;
    else if ((lhssmall >= 0) && (rhssmall < 0))
      q = ((int64_t) lhssmall - ((int64_t) rhssmall + 1)) /
          rhssmall;
    else
      q = lhssmall / rhssmall;
    isl_sioimath_set_small(dst, q);
    return;
  }

  impz_fdiv_q(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_bigarg_src(rhs, &rhsscratch));
  isl_sioimath_try_demote(dst);
}

/* Compute the division of lhs by a rhs of type unsigned long, rounding towards
 * negative infinity (Floor).
 */
inline void isl_sioimath_fdiv_q_ui(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  unsigned long rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, q;

  if (isl_sioimath_decode_small(lhs, &lhssmall) && (rhs <= INT32_MAX)141k) {
    if (lhssmall >= 0)
      q = (uint32_t) lhssmall / rhs;
    else
      q = ((int64_t) lhssmall - ((int64_t) rhs - 1)) /
          (int64_t) rhs;
    isl_sioimath_set_small(dst, q);
    return;
  }

  impz_fdiv_q(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_uiarg_src(rhs, &rhsscratch));
  isl_sioimath_try_demote(dst);
}

/* Get the remainder of: lhs divided by rhs rounded towards negative infinite
 * (Floor).
 */
inline void isl_sioimath_fdiv_r(isl_sioimath_ptr dst, isl_sioimath_src lhs,
  isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int64_t lhssmall, rhssmall;
  int32_t r;

  if (isl_sioimath_is_small(lhs) && isl_sioimath_is_small(rhs)213k) {
    lhssmall = isl_sioimath_get_small(lhs);
    rhssmall = isl_sioimath_get_small(rhs);
    r = (rhssmall + lhssmall % rhssmall) % rhssmall;
    isl_sioimath_set_small(dst, r);
    return;
  }

  impz_fdiv_r(isl_sioimath_reinit_big(dst),
      isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_bigarg_src(rhs, &rhsscratch));
  isl_sioimath_try_demote(dst);
}

void isl_sioimath_read(isl_sioimath_ptr dst, const char *str);

/* Return:
 *   +1 for a positive number
 *   -1 for a negative number
 *    0 if the number is zero
 */
inline int isl_sioimath_sgn(isl_sioimath_src arg)
{
  int32_t small;

  if (isl_sioimath_decode_small(arg, &small))
    return (small > 0) - (small < 0);

  return mp_int_compare_zero(isl_sioimath_get_big(arg));
}

/* Return:
 *   +1 if lhs > rhs
 *   -1 if lhs < rhs
 *    0 if lhs = rhs
 */
inline int isl_sioimath_cmp(isl_sioimath_src lhs, isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, rhssmall;

  if (isl_sioimath_decode_small(lhs, &lhssmall) &&
      isl_sioimath_decode_small(rhs, &rhssmall)257M)
    return (lhssmall > rhssmall) - (lhssmall < rhssmall);

  if (isl_sioimath_decode_small(rhs, &rhssmall))
    return mp_int_compare_value(
        isl_sioimath_bigarg_src(lhs, &lhsscratch), rhssmall);

  if (isl_sioimath_decode_small(lhs, &lhssmall))
    return -mp_int_compare_value(
               isl_sioimath_bigarg_src(rhs, &rhsscratch), lhssmall);

  return mp_int_compare(
      isl_sioimath_get_big(lhs), isl_sioimath_get_big(rhs));
}

/* As isl_sioimath_cmp, but with signed long rhs.
 */
inline int isl_sioimath_cmp_si(isl_sioimath_src lhs, signed long rhs)
{
  int32_t lhssmall;

  if (isl_sioimath_decode_small(lhs, &lhssmall))
    return (lhssmall > rhs) - (lhssmall < rhs);

  return mp_int_compare_value(isl_sioimath_get_big(lhs), rhs);
}

/* Return:
 *   +1 if |lhs| > |rhs|
 *   -1 if |lhs| < |rhs|
 *    0 if |lhs| = |rhs|
 */
inline int isl_sioimath_abs_cmp(isl_sioimath_src lhs, isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, rhssmall;

  if (isl_sioimath_decode_small(lhs, &lhssmall) &&
      isl_sioimath_decode_small(rhs, &rhssmall)68.5M) {
    lhssmall = labs(lhssmall);
    rhssmall = labs(rhssmall);
    return (lhssmall > rhssmall) - (lhssmall < rhssmall);
  }

  return mp_int_compare_unsigned(
      isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_bigarg_src(rhs, &rhsscratch));
}

/* Return whether lhs is divisible by rhs.
 * In particular, can rhs be multiplied by some integer to result in lhs?
 * If rhs is zero, then this means lhs has to be zero too.
 */
inline int isl_sioimath_is_divisible_by(isl_sioimath_src lhs,
          isl_sioimath_src rhs)
{
  isl_sioimath_scratchspace_t lhsscratch, rhsscratch;
  int32_t lhssmall, rhssmall;
  mpz_t rem;
  int cmp;

  if (isl_sioimath_sgn(rhs) == 0)
    return isl_sioimath_sgn(lhs) == 0;

  if (isl_sioimath_decode_small(lhs, &lhssmall) &&
      isl_sioimath_decode_small(rhs, &rhssmall)7.13M)
    return lhssmall % rhssmall == 0;

  if (isl_sioimath_decode_small(rhs, &rhssmall))
    return mp_int_divisible_value(
        isl_sioimath_bigarg_src(lhs, &lhsscratch), rhssmall);

  mp_int_init(&rem);
  mp_int_div(isl_sioimath_bigarg_src(lhs, &lhsscratch),
      isl_sioimath_bigarg_src(rhs, &rhsscratch), NULL, &rem);
  cmp = mp_int_compare_zero(&rem);
  mp_int_clear(&rem);
  return cmp == 0;
}

/* Return a hash code of an isl_sioimath.
 * The hash code for a number in small and big representation must be identical
 * on the same machine because small representation if not obligatory if fits.
 */
inline uint32_t isl_sioimath_hash(isl_sioimath_src arg, uint32_t hash)
{
  int32_t small;
  int i;
  uint32_t num;
  mp_digit digits[(sizeof(uint32_t) + sizeof(mp_digit) - 1) /
                  sizeof(mp_digit)];
  mp_size used;
  const unsigned char *digitdata = (const unsigned char *) &digits;

  if (isl_sioimath_decode_small(arg, &small)) {
    if (small < 0)
      isl_hash_byte26.4M(hash, 0xFF);
    num = labs(small);

    isl_siomath_uint32_to_digits(num, digits, &used);
    for (i = 0; i < used * sizeof(mp_digit); i += 1206M)
      isl_hash_byte(hash, digitdata[i]);
    return hash;
  }

  return isl_imath_hash(isl_sioimath_get_big(arg), hash);
}

/* Return the number of digits in a number of the given base or more, i.e. the
 * string length without sign and null terminator.
 *
 * Current implementation for small representation returns the maximal number
 * of binary digits in that representation, which can be much larger than the
 * smallest possible solution.
 */
inline size_t isl_sioimath_sizeinbase(isl_sioimath_src arg, int base)
{
  int32_t small;

  if (isl_sioimath_decode_small(arg, &small))
    return sizeof(int32_t) * CHAR_BIT - 1;

  return impz_sizeinbase(isl_sioimath_get_big(arg), base);
}

void isl_sioimath_print(FILE *out, isl_sioimath_src i, int width);
void isl_sioimath_dump(isl_sioimath_src arg);

typedef isl_sioimath isl_int[1];
#define isl_int_init(i)     isl_sioimath_init((i))
#define isl_int_clear(i)    isl_sioimath_clear((i))

#define isl_int_set(r, i)   isl_sioimath_set((r), *(i))
#define isl_int_set_si(r, i)    isl_sioimath_set_si((r), i)
#define isl_int_set_ui(r, i)    isl_sioimath_set_ui((r), i)
#define isl_int_fits_slong(r)   isl_sioimath_fits_slong(*(r))
#define isl_int_get_si(r)   isl_sioimath_get_si(*(r))
#define isl_int_fits_ulong(r)   isl_sioimath_fits_ulong(*(r))
#define isl_int_get_ui(r)   isl_sioimath_get_ui(*(r))
#define isl_int_get_d(r)    isl_sioimath_get_d(*(r))
#define isl_int_get_str(r)    isl_sioimath_get_str(*(r))
#define isl_int_abs(r, i)   isl_sioimath_abs((r), *(i))
#define isl_int_neg(r, i)   isl_sioimath_neg((r), *(i))
#define isl_int_swap(i, j)    isl_sioimath_swap((i), (j))
#define isl_int_swap_or_set(i, j) isl_sioimath_swap((i), (j))
#define isl_int_add_ui(r, i, j)   isl_sioimath_add_ui((r), *(i), j)
#define isl_int_sub_ui(r, i, j)   isl_sioimath_sub_ui((r), *(i), j)

#define isl_int_add(r, i, j)    isl_sioimath_add((r), *(i), *(j))
#define isl_int_sub(r, i, j)    isl_sioimath_sub((r), *(i), *(j))
#define isl_int_mul(r, i, j)    isl_sioimath_mul((r), *(i), *(j))
#define isl_int_mul_2exp(r, i, j) isl_sioimath_mul_2exp((r), *(i), j)
#define isl_int_mul_si(r, i, j)   isl_sioimath_mul_si((r), *(i), j)
#define isl_int_mul_ui(r, i, j)   isl_sioimath_mul_ui((r), *(i), j)
#define isl_int_pow_ui(r, i, j)   isl_sioimath_pow_ui((r), *(i), j)
#define isl_int_addmul(r, i, j)   isl_sioimath_addmul((r), *(i), *(j))
#define isl_int_addmul_ui(r, i, j)  isl_sioimath_addmul_ui((r), *(i), j)
#define isl_int_submul(r, i, j)   isl_sioimath_submul((r), *(i), *(j))
#define isl_int_submul_ui(r, i, j)  isl_sioimath_submul_ui((r), *(i), j)

#define isl_int_gcd(r, i, j)    isl_sioimath_gcd((r), *(i), *(j))
#define isl_int_lcm(r, i, j)    isl_sioimath_lcm((r), *(i), *(j))
#define isl_int_divexact(r, i, j) isl_sioimath_tdiv_q((r), *(i), *(j))
#define isl_int_divexact_ui(r, i, j)  isl_sioimath_tdiv_q_ui((r), *(i), j)
#define isl_int_tdiv_q(r, i, j)   isl_sioimath_tdiv_q((r), *(i), *(j))
#define isl_int_cdiv_q(r, i, j)   isl_sioimath_cdiv_q((r), *(i), *(j))
#define isl_int_cdiv_q_ui(r, i, j)  isl_sioimath_cdiv_q_ui((r), *(i), j)
#define isl_int_fdiv_q(r, i, j)   isl_sioimath_fdiv_q((r), *(i), *(j))
#define isl_int_fdiv_r(r, i, j)   isl_sioimath_fdiv_r((r), *(i), *(j))
#define isl_int_fdiv_q_ui(r, i, j)  isl_sioimath_fdiv_q_ui((r), *(i), j)

#define isl_int_read(r, s)    isl_sioimath_read((r), s)
#define isl_int_sgn(i)      isl_sioimath_sgn(*(i))
#define isl_int_cmp(i, j)   isl_sioimath_cmp(*(i), *(j))
#define isl_int_cmp_si(i, si)   isl_sioimath_cmp_si(*(i), si)
#define isl_int_eq(i, j)    (isl_sioimath_cmp(*(i), *(j)) == 0)
#define isl_int_ne(i, j)    (isl_sioimath_cmp(*(i), *(j)) != 0)
#define isl_int_lt(i, j)    (isl_sioimath_cmp(*(i), *(j)) < 0)
#define isl_int_le(i, j)    (isl_sioimath_cmp(*(i), *(j)) <= 0)
#define isl_int_gt(i, j)    (isl_sioimath_cmp(*(i), *(j)) > 0)
#define isl_int_ge(i, j)    (isl_sioimath_cmp(*(i), *(j)) >= 0)
#define isl_int_abs_cmp(i, j)   isl_sioimath_abs_cmp(*(i), *(j))
#define isl_int_abs_eq(i, j)    (isl_sioimath_abs_cmp(*(i), *(j)) == 0)
#define isl_int_abs_ne(i, j)    (isl_sioimath_abs_cmp(*(i), *(j)) != 0)
#define isl_int_abs_lt(i, j)    (isl_sioimath_abs_cmp(*(i), *(j)) < 0)
#define isl_int_abs_gt(i, j)    (isl_sioimath_abs_cmp(*(i), *(j)) > 0)
#define isl_int_abs_ge(i, j)    (isl_sioimath_abs_cmp(*(i), *(j)) >= 0)
#define isl_int_is_divisible_by(i, j) isl_sioimath_is_divisible_by(*(i), *(j))

#define isl_int_hash(v, h)    isl_sioimath_hash(*(v), h)
#define isl_int_free_str(s)   free(s)
#define isl_int_print(out, i, width)  isl_sioimath_print(out, *(i), width)

#endif /* ISL_INT_SIOIMATH_H */

Coverage Report

Created: 2019-07-24 05:18