379 lines
6.7 KiB
C
379 lines
6.7 KiB
C
/*************************************************************************
|
|
* *
|
|
* YAP Prolog *
|
|
* *
|
|
* Yap Prolog was developed at NCCUP - Universidade do Porto *
|
|
* *
|
|
* Copyright L.Damas, V.S.Costa and Universidade do Porto 1985-1997 *
|
|
* *
|
|
**************************************************************************
|
|
* *
|
|
* File: gmp_support.c *
|
|
* Last rev: *
|
|
* mods: *
|
|
* comments: bignum code *
|
|
* *
|
|
*************************************************************************/
|
|
|
|
#include "Yap.h"
|
|
#include "Yatom.h"
|
|
#include "Heap.h"
|
|
#include "eval.h"
|
|
|
|
#if USE_GMP
|
|
|
|
static inline Term
|
|
MkBigAndClose(MP_INT *new)
|
|
{
|
|
Term t = Yap_MkBigIntTerm(new);
|
|
mpz_clear(new);
|
|
if (t == TermNil) {
|
|
return Yap_ArithError(RESOURCE_ERROR_STACK, t, ">>/2");
|
|
}
|
|
return t;
|
|
}
|
|
|
|
/* add i + j using temporary bigint new */
|
|
Term
|
|
Yap_gmp_add_ints(Int i, Int j)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new,i);
|
|
if (j > 0) {
|
|
mpz_add_ui(&new, &new, j);
|
|
} else {
|
|
if (j-1 > 0) { /* negative overflow */
|
|
mpz_sub_ui(&new, &new, -(j+1));
|
|
mpz_sub_ui(&new, &new, 1);
|
|
} else {
|
|
mpz_sub_ui(&new, &new, -j);
|
|
}
|
|
}
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_sub_ints(Int i, Int j)
|
|
{
|
|
MP_INT new;
|
|
Term t;
|
|
|
|
mpz_init_set_si(&new,i);
|
|
if (j > 0) {
|
|
mpz_sub_ui(&new, &new, j);
|
|
} else {
|
|
if (j-1 > 0) { /* negative overflow */
|
|
mpz_add_ui(&new, &new, -(j+1));
|
|
mpz_add_ui(&new, &new, 1);
|
|
} else {
|
|
mpz_add_ui(&new, &new, -j);
|
|
}
|
|
}
|
|
return MkBigAndClose(&new);
|
|
t = Yap_MkBigIntTerm(&new);
|
|
mpz_clear(&new);
|
|
return t;
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_mul_ints(Int i, Int j)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new,i);
|
|
mpz_mul_si(&new, &new, j);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_sll_ints(Int i, Int j)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new,i);
|
|
mpz_mul_2exp(&new, &new, j);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* add i + b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_add_int_big(Int i, MP_INT *b)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_add(&new, &new, b);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* sub i - b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_sub_int_big(Int i, MP_INT *b)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_sub(&new, &new, b);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* add i + b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_mul_int_big(Int i, MP_INT *b)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_mul(&new, &new, b);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* sub i - b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_sub_big_int(MP_INT *b, Int i)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_neg(&new, &new);
|
|
mpz_add(&new, &new, b);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* div i / b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_div_big_int(MP_INT *b, Int i)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b);
|
|
if (yap_flags[INTEGER_ROUNDING_FLAG] == 0) {
|
|
if (i > 0) {
|
|
mpz_tdiv_q_ui(&new, &new, i);
|
|
} else if (i == 0) {
|
|
return Yap_ArithError(EVALUATION_ERROR_ZERO_DIVISOR, MkIntTerm(0), "// /2");
|
|
} else {
|
|
/* we do not handle MIN_INT */
|
|
mpz_tdiv_q_ui(&new, &new, -i);
|
|
mpz_neg(&new, &new);
|
|
}
|
|
} else {
|
|
if (i > 0) {
|
|
mpz_fdiv_q_ui(&new, &new, i);
|
|
} else if (i == 0) {
|
|
return Yap_ArithError(EVALUATION_ERROR_ZERO_DIVISOR, MkIntTerm(0), "// /2");
|
|
} else {
|
|
/* we do not handle MIN_INT */
|
|
mpz_fdiv_q_ui(&new, &new, -i);
|
|
mpz_neg(&new, &new);
|
|
}
|
|
}
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* sub i - b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_and_int_big(Int i, MP_INT *b)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_and(&new, &new, b);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* sub i - b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_ior_int_big(Int i, MP_INT *b)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_ior(&new, &new, b);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* add i + b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_sll_big_int(MP_INT *b, Int i)
|
|
{
|
|
MP_INT new;
|
|
|
|
if (i > 0) {
|
|
mpz_init_set(&new, b);
|
|
mpz_mul_2exp(&new, &new, i);
|
|
} else if (i == 0) {
|
|
mpz_init_set(&new, b);
|
|
} else {
|
|
mpz_init_set(&new, b);
|
|
if (i == Int_MIN) {
|
|
return Yap_ArithError(RESOURCE_ERROR_HUGE_INT, MkIntegerTerm(i), "<</2");
|
|
}
|
|
mpz_tdiv_q_2exp(&new, &new, -i);
|
|
}
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_add_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b1);
|
|
mpz_add(&new, &new, b2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_sub_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b1);
|
|
mpz_sub(&new, &new, b2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_mul_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b1);
|
|
mpz_mul(&new, &new, b2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
/* div i / b using temporary bigint new */
|
|
Term
|
|
Yap_gmp_div_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b1);
|
|
if (yap_flags[INTEGER_ROUNDING_FLAG] == 0) {
|
|
mpz_tdiv_q(&new, &new, b2);
|
|
} else {
|
|
mpz_fdiv_q(&new, &new, b2);
|
|
}
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_and_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b1);
|
|
mpz_and(&new, &new, b2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_ior_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set(&new, b1);
|
|
mpz_ior(&new, &new, b2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_mod_big_big(MP_INT *b1, MP_INT *b2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init(&new);
|
|
mpz_fdiv_r(&new, b1, b2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_mod_big_int(MP_INT *b, Int i2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i2);
|
|
mpz_fdiv_r(&new, b, &new);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_mod_int_big(Int i1, MP_INT *b)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i1);
|
|
mpz_fdiv_r(&new, &new, b);
|
|
RBIG(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_add_float_big(Float d, MP_INT *b)
|
|
{
|
|
return MkFloatTerm(d+mpz_get_d(b));
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_sub_float_big(Float d, MP_INT *b)
|
|
{
|
|
return MkFloatTerm(d-mpz_get_d(b));
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_sub_big_float(MP_INT *b, Float d)
|
|
{
|
|
return MkFloatTerm(mpz_get_d(b)-d);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_mul_float_big(Float d, MP_INT *b)
|
|
{
|
|
return MkFloatTerm(d*mpz_get_d(b));
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_exp_ints(Int i1, Int i2)
|
|
{
|
|
MP_INT new;
|
|
|
|
mpz_init_set_si(&new, i1);
|
|
mpz_pow_ui (&new, &new, (unsigned long int)i2);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_exp_big_int(MP_INT *b, Int i)
|
|
{
|
|
MP_INT new;
|
|
|
|
if (b > 0) {
|
|
mpz_init(&new);
|
|
mpz_pow_ui (&new, b, (unsigned long int)i);
|
|
} else {
|
|
MP_INT new;
|
|
if (b==0) return MkIntTerm(1);
|
|
|
|
mpz_init_set_si(&new, i);
|
|
mpz_powm (&new, b, &new, b);
|
|
}
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
Term
|
|
Yap_gmp_big_from_64bits(YAP_LONG_LONG i)
|
|
{
|
|
char s[64];
|
|
MP_INT new;
|
|
|
|
sprintf(s, "%lld", (long long int)i);
|
|
mpz_init_set_str (&new, s, 10);
|
|
return MkBigAndClose(&new);
|
|
}
|
|
|
|
#endif
|
|
|
|
|