cpython/Objects/longobject.c

1359 lines
27 KiB
C
Raw Normal View History

1991-05-05 17:09:44 -03:00
/***********************************************************
Copyright 1991 by Stichting Mathematisch Centrum, Amsterdam, The
Netherlands.
All Rights Reserved
Permission to use, copy, modify, and distribute this software and its
documentation for any purpose and without fee is hereby granted,
provided that the above copyright notice appear in all copies and that
both that copyright notice and this permission notice appear in
supporting documentation, and that the names of Stichting Mathematisch
Centrum or CWI not be used in advertising or publicity pertaining to
distribution of the software without specific, written prior permission.
STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
******************************************************************/
/* Long (arbitrary precision) integer object implementation */
1991-05-14 09:06:49 -03:00
/* XXX The functional organization of this file is terrible */
1991-05-05 17:09:44 -03:00
#include "allobjects.h"
#include "longintrepr.h"
#include <math.h>
1991-05-05 17:09:44 -03:00
#include <assert.h>
1991-05-14 09:06:49 -03:00
static int ticker; /* XXX Could be shared with ceval? */
#define ZABS(x) ((x) < 0 ? ~(x) : (x))
1991-05-14 09:06:49 -03:00
#define INTRCHECK(block) \
if (--ticker < 0) { \
ticker = 100; \
if (intrcheck()) { block; } \
}
1991-05-05 17:09:44 -03:00
/* Normalize (remove leading zeros from) a long int object.
Doesn't attempt to free the storage--in most cases, due to the nature
of the algorithms used, this could save at most be one word anyway. */
longobject *
long_normalize(v)
register longobject *v;
{
int j = ZABS(v->ob_size);
register int i = j;
while (i > 0 && v->ob_digit[i-1] == 0)
--i;
if (i != j)
v->ob_size = (v->ob_size < 0) ? ~i : i;
if (v->ob_size == ~0)
v->ob_size = 0;
return v;
}
/* Normalize except leave ~0 unchanged */
longobject *
long_znormalize(v)
register longobject *v;
{
int j = ZABS(v->ob_size);
1991-05-05 17:09:44 -03:00
register int i = j;
while (i > 0 && v->ob_digit[i-1] == 0)
--i;
if (i != j)
v->ob_size = (v->ob_size < 0) ? ~i : i;
1991-05-05 17:09:44 -03:00
return v;
}
/* Allocate a new long int object with size digits.
Return NULL and set exception if we run out of memory. */
longobject *
alloclongobject(size)
int size;
{
return NEWVAROBJ(longobject, &Longtype, size);
}
/* Create a new long int object from a C long int */
object *
newlongobject(ival)
long ival;
{
/* Assume a C long fits in at most 3 'digits' */
longobject *v = alloclongobject(3);
if (v != NULL) {
if (ival < 0) {
ival = -ival;
v->ob_size = ~v->ob_size;
1991-05-05 17:09:44 -03:00
}
v->ob_digit[0] = ival & MASK;
v->ob_digit[1] = (ival >> SHIFT) & MASK;
v->ob_digit[2] = (ival >> (2*SHIFT)) & MASK;
v = long_normalize(v);
}
return (object *)v;
}
/* Create a new long int object from a C double */
object *
dnewlongobject(dval)
double dval;
{
longobject *v;
double frac;
int i, ndig, expo, neg;
neg = 0;
if (dval < 0.0) {
neg = 1;
dval = -dval;
}
frac = frexp(dval, &expo); /* dval = frac*2**expo; 0.0 <= frac < 1.0 */
if (expo <= 0)
return newlongobject(0L);
ndig = (expo-1) / SHIFT + 1; /* Number of 'digits' in result */
v = alloclongobject(ndig);
if (v == NULL)
return NULL;
frac = ldexp(frac, (expo-1) % SHIFT + 1);
for (i = ndig; --i >= 0; ) {
long bits = (long)frac;
v->ob_digit[i] = bits;
frac = frac - (double)bits;
frac = ldexp(frac, SHIFT);
}
if (neg)
v->ob_size = ~v->ob_size;
return (object *)v;
}
1991-05-05 17:09:44 -03:00
/* Get a C long int from a long int object.
Returns -1 and sets an error condition if overflow occurs. */
long
getlongvalue(vv)
object *vv;
{
register longobject *v;
long x, prev;
int i, sign;
if (vv == NULL || !is_longobject(vv)) {
err_badcall();
return -1;
}
v = (longobject *)vv;
i = v->ob_size;
sign = 1;
x = 0;
if (i < 0) {
sign = -1;
i = ~i;
1991-05-05 17:09:44 -03:00
}
while (--i >= 0) {
prev = x;
x = (x << SHIFT) + v->ob_digit[i];
if ((x >> SHIFT) != prev) {
1991-12-10 10:00:03 -04:00
err_setstr(ValueError,
1991-05-05 17:09:44 -03:00
"long int too long to convert");
return -1;
}
}
return x * sign;
}
/* Get a C double from a long int object. No overflow check. */
double
dgetlongvalue(vv)
object *vv;
{
register longobject *v;
double x;
double multiplier = (double) (1L << SHIFT);
int i, sign;
if (vv == NULL || !is_longobject(vv)) {
err_badcall();
return -1;
}
v = (longobject *)vv;
i = v->ob_size;
sign = 1;
x = 0.0;
if (i < 0) {
sign = -1;
i = ~i;
1991-05-05 17:09:44 -03:00
}
while (--i >= 0) {
x = x*multiplier + v->ob_digit[i];
}
return x * sign;
}
/* Multiply by a single digit, ignoring the sign. */
longobject *
mul1(a, n)
longobject *a;
1991-05-14 09:06:49 -03:00
wdigit n;
1991-05-05 17:09:44 -03:00
{
return muladd1(a, n, (digit)0);
}
/* Multiply by a single digit and add a single digit, ignoring the sign. */
longobject *
muladd1(a, n, extra)
longobject *a;
1991-05-14 09:06:49 -03:00
wdigit n;
wdigit extra;
1991-05-05 17:09:44 -03:00
{
int size_a = ZABS(a->ob_size);
1991-05-05 17:09:44 -03:00
longobject *z = alloclongobject(size_a+1);
twodigits carry = extra;
int i;
if (z == NULL)
return NULL;
for (i = 0; i < size_a; ++i) {
carry += (twodigits)a->ob_digit[i] * n;
z->ob_digit[i] = carry & MASK;
carry >>= SHIFT;
}
z->ob_digit[i] = carry;
return long_normalize(z);
}
/* Divide a long integer by a digit, returning both the quotient
(as function result) and the remainder (through *prem).
The sign of a is ignored; n should not be zero. */
longobject *
divrem1(a, n, prem)
longobject *a;
1991-05-14 09:06:49 -03:00
wdigit n;
1991-05-05 17:09:44 -03:00
digit *prem;
{
int size = ZABS(a->ob_size);
1991-05-05 17:09:44 -03:00
longobject *z;
int i;
twodigits rem = 0;
assert(n > 0 && n <= MASK);
z = alloclongobject(size);
if (z == NULL)
return NULL;
for (i = size; --i >= 0; ) {
rem = (rem << SHIFT) + a->ob_digit[i];
z->ob_digit[i] = rem/n;
rem %= n;
}
*prem = rem;
return long_normalize(z);
}
/* Convert a long int object to a string, using a given conversion base.
Return a string object.
If base is 8 or 16, add the proper prefix '0' or '0x'. */
1991-05-05 17:09:44 -03:00
stringobject *
long_format(a, base)
longobject *a;
int base;
{
stringobject *str;
int i;
int size_a = ZABS(a->ob_size);
1991-05-05 17:09:44 -03:00
char *p;
int bits;
char sign = '\0';
assert(base >= 2 && base <= 36);
/* Compute a rough upper bound for the length of the string */
i = base;
bits = 0;
while (i > 1) {
++bits;
i >>= 1;
}
i = 6 + (size_a*SHIFT + bits-1) / bits;
1991-05-05 17:09:44 -03:00
str = (stringobject *) newsizedstringobject((char *)0, i);
if (str == NULL)
return NULL;
p = GETSTRINGVALUE(str) + i;
*p = '\0';
*--p = 'L';
if (a->ob_size < 0) {
if (a->ob_size < ~0)
sign = '-';
else
sign = '~';
}
1991-05-05 17:09:44 -03:00
INCREF(a);
do {
digit rem;
longobject *temp = divrem1(a, (digit)base, &rem);
if (temp == NULL) {
DECREF(a);
DECREF(str);
return NULL;
}
if (rem < 10)
rem += '0';
else
rem += 'A'-10;
assert(p > GETSTRINGVALUE(str));
*--p = rem;
DECREF(a);
a = temp;
1991-05-14 09:06:49 -03:00
INTRCHECK({
1991-05-05 17:09:44 -03:00
DECREF(a);
DECREF(str);
err_set(KeyboardInterrupt);
return NULL;
1991-05-14 09:06:49 -03:00
})
} while (ZABS(a->ob_size) != 0);
1991-05-05 17:09:44 -03:00
DECREF(a);
if (base == 8)
*--p = '0';
else if (base == 16) {
*--p = 'x';
*--p = '0';
}
else if (base != 10) {
*--p = '#';
*--p = '0' + base%10;
if (base > 10)
*--p = '0' + base/10;
}
1991-05-05 17:09:44 -03:00
if (sign)
*--p = sign;
if (p != GETSTRINGVALUE(str)) {
char *q = GETSTRINGVALUE(str);
assert(p > q);
do {
} while ((*q++ = *p++) != '\0');
q--;
1991-05-05 17:09:44 -03:00
resizestring((object **)&str, (int) (q - GETSTRINGVALUE(str)));
}
return str;
}
/* Convert a string to a long int object, in a given base.
Base zero implies a default depending on the number. */
object *
long_scan(str, base)
char *str;
int base;
{
int sign = 1;
longobject *z;
assert(base == 0 || base >= 2 && base <= 36);
if (*str == '+')
++str;
else if (*str == '-') {
++str;
sign = -1;
}
if (base == 0) {
if (str[0] != '0')
base = 10;
else if (str[1] == 'x' || str[1] == 'X')
base = 16;
else
base = 8;
}
if (base == 16 && str[0] == '0' && (str[1] == 'x' || str[1] == 'X'))
str += 2;
z = alloclongobject(0);
for ( ; z != NULL; ++str) {
int k = -1;
longobject *temp;
if (*str <= '9')
k = *str - '0';
else if (*str >= 'a')
k = *str - 'a' + 10;
else if (*str >= 'A')
k = *str - 'A' + 10;
if (k < 0 || k >= base)
break;
temp = muladd1(z, (digit)base, (digit)k);
DECREF(z);
z = temp;
}
if (sign < 0 && z != NULL && z->ob_size != 0)
z->ob_size = ~z->ob_size;
1991-05-05 17:09:44 -03:00
return (object *) z;
}
static longobject *x_divrem PROTO((longobject *, longobject *, longobject **));
static object *long_pos PROTO((longobject *));
1991-05-05 17:09:44 -03:00
/* Long division with remainder, top-level routine */
longobject *
long_divrem(a, b, prem)
longobject *a, *b;
longobject **prem;
{
int size_a = ZABS(a->ob_size), size_b = ZABS(b->ob_size);
1991-05-05 17:09:44 -03:00
longobject *z;
if (size_b == 0) {
if (prem != NULL)
*prem = NULL;
1991-12-10 10:00:03 -04:00
err_setstr(ZeroDivisionError, "long division or remainder");
1991-05-05 17:09:44 -03:00
return NULL;
}
if (size_a < size_b ||
1991-05-14 09:06:49 -03:00
size_a == size_b &&
a->ob_digit[size_a-1] < b->ob_digit[size_b-1]) {
1991-05-05 17:09:44 -03:00
/* |a| < |b|. */
if (prem != NULL) {
object *long_pos();
*prem = (longobject *) long_pos(a);
1991-05-05 17:09:44 -03:00
}
return alloclongobject(0);
}
if (size_b == 1) {
digit rem = 0;
z = divrem1(a, b->ob_digit[0], &rem);
if (prem != NULL) {
if (z == NULL)
*prem = NULL;
else
*prem = (longobject *)
newlongobject((long)rem);
}
}
else
z = x_divrem(a, b, prem);
/* Set the signs.
The quotient z has the sign of a*b;
the remainder r has the sign of a,
so a = b*z + r. */
if (z != NULL) {
if ((a->ob_size < 0) != (b->ob_size < 0))
z->ob_size = ~ z->ob_size;
if (prem != NULL && *prem != NULL && a->ob_size < 0 &&
(*prem)->ob_size != 0)
(*prem)->ob_size = ~ (*prem)->ob_size;
1991-05-05 17:09:44 -03:00
}
return z;
}
1991-05-14 09:06:49 -03:00
/* Unsigned long division with remainder -- the algorithm */
1991-05-05 17:09:44 -03:00
static longobject *
x_divrem(v1, w1, prem)
longobject *v1, *w1;
longobject **prem;
{
int size_v = ZABS(v1->ob_size), size_w = ZABS(w1->ob_size);
1991-05-05 17:09:44 -03:00
digit d = (twodigits)BASE / (w1->ob_digit[size_w-1] + 1);
longobject *v = mul1(v1, d);
longobject *w = mul1(w1, d);
longobject *a;
int j, k;
if (v == NULL || w == NULL) {
XDECREF(v);
XDECREF(w);
if (prem != NULL)
*prem = NULL;
return NULL;
}
assert(size_v >= size_w && size_w > 1); /* Assert checks by div() */
1991-05-14 09:06:49 -03:00
assert(v->ob_refcnt == 1); /* Since v will be used as accumulator! */
assert(size_w == ZABS(w->ob_size)); /* That's how d was calculated */
1991-05-05 17:09:44 -03:00
size_v = ZABS(v->ob_size);
1991-05-05 17:09:44 -03:00
a = alloclongobject(size_v - size_w + 1);
for (j = size_v, k = a->ob_size-1; a != NULL && k >= 0; --j, --k) {
digit vj = (j >= size_v) ? 0 : v->ob_digit[j];
twodigits q;
1991-05-14 09:06:49 -03:00
stwodigits carry = 0;
1991-05-05 17:09:44 -03:00
int i;
1991-05-14 09:06:49 -03:00
INTRCHECK({
1991-05-05 17:09:44 -03:00
DECREF(a);
a = NULL;
err_set(KeyboardInterrupt);
break;
1991-05-14 09:06:49 -03:00
})
1991-05-05 17:09:44 -03:00
if (vj == w->ob_digit[size_w-1])
q = MASK;
else
q = (((twodigits)vj << SHIFT) + v->ob_digit[j-1]) /
w->ob_digit[size_w-1];
while (w->ob_digit[size_w-2]*q >
((
((twodigits)vj << SHIFT)
+ v->ob_digit[j-1]
- q*w->ob_digit[size_w-1]
) << SHIFT)
+ v->ob_digit[j-2])
--q;
for (i = 0; i < size_w && i+k < size_v; ++i) {
twodigits z = w->ob_digit[i] * q;
digit zz = z >> SHIFT;
carry += v->ob_digit[i+k] - z + ((twodigits)zz << SHIFT);
v->ob_digit[i+k] = carry & MASK;
carry = (carry >> SHIFT) - zz;
}
if (i+k < size_v) {
carry += v->ob_digit[i+k];
v->ob_digit[i+k] = 0;
}
if (carry == 0)
a->ob_digit[k] = q;
else {
assert(carry == -1);
a->ob_digit[k] = q-1;
carry = 0;
for (i = 0; i < size_w && i+k < size_v; ++i) {
carry += v->ob_digit[i+k] + w->ob_digit[i];
v->ob_digit[i+k] = carry & MASK;
carry >>= SHIFT;
}
}
} /* for j, k */
if (a == NULL) {
if (prem != NULL)
1991-05-05 17:09:44 -03:00
*prem = NULL;
}
else {
a = long_normalize(a);
if (prem != NULL) {
1991-05-05 17:09:44 -03:00
*prem = divrem1(v, d, &d);
/* d receives the (unused) remainder */
if (*prem == NULL) {
DECREF(a);
a = NULL;
}
}
1991-05-05 17:09:44 -03:00
}
DECREF(v);
DECREF(w);
return a;
}
/* Methods */
static void
long_dealloc(v)
longobject *v;
{
DEL(v);
}
1991-06-07 13:10:43 -03:00
static int
1991-05-05 17:09:44 -03:00
long_print(v, fp, flags)
longobject *v;
FILE *fp;
int flags;
{
stringobject *str = long_format(v, 10);
1991-06-07 13:10:43 -03:00
if (str == NULL)
return -1;
fprintf(fp, "%s", GETSTRINGVALUE(str));
1991-06-07 13:10:43 -03:00
DECREF(str);
return 0;
1991-05-05 17:09:44 -03:00
}
static object *
long_repr(v)
longobject *v;
{
return (object *) long_format(v, 10);
1991-05-05 17:09:44 -03:00
}
static int
long_compare(a, b)
longobject *a, *b;
{
int sign;
if (a->ob_size != b->ob_size) {
if (ZABS(a->ob_size) == 0 && ZABS(b->ob_size) == 0)
sign = 0;
else
sign = a->ob_size - b->ob_size;
}
1991-05-05 17:09:44 -03:00
else {
int i = ZABS(a->ob_size);
1991-05-05 17:09:44 -03:00
while (--i >= 0 && a->ob_digit[i] == b->ob_digit[i])
;
if (i < 0)
sign = 0;
else
sign = (int)a->ob_digit[i] - (int)b->ob_digit[i];
}
return sign < 0 ? -1 : sign > 0 ? 1 : 0;
1991-05-05 17:09:44 -03:00
}
/* Add the absolute values of two long integers. */
static longobject *x_add PROTO((longobject *, longobject *));
static longobject *
x_add(a, b)
longobject *a, *b;
{
int size_a = ZABS(a->ob_size), size_b = ZABS(b->ob_size);
1991-05-05 17:09:44 -03:00
longobject *z;
int i;
digit carry = 0;
/* Ensure a is the larger of the two: */
if (size_a < size_b) {
{ longobject *temp = a; a = b; b = temp; }
{ int size_temp = size_a; size_a = size_b; size_b = size_temp; }
}
z = alloclongobject(size_a+1);
if (z == NULL)
return NULL;
for (i = 0; i < size_b; ++i) {
carry += a->ob_digit[i] + b->ob_digit[i];
z->ob_digit[i] = carry & MASK;
/* The following assumes unsigned shifts don't
propagate the sign bit. */
carry >>= SHIFT;
}
for (; i < size_a; ++i) {
carry += a->ob_digit[i];
z->ob_digit[i] = carry & MASK;
carry >>= SHIFT;
}
z->ob_digit[i] = carry;
return long_normalize(z);
}
/* Subtract the absolute values of two integers. */
static longobject *x_sub PROTO((longobject *, longobject *));
static longobject *
x_sub(a, b)
longobject *a, *b;
{
int size_a = ZABS(a->ob_size), size_b = ZABS(b->ob_size);
1991-05-05 17:09:44 -03:00
longobject *z;
int i;
int sign = 1;
digit borrow = 0;
/* Ensure a is the larger of the two: */
if (size_a < size_b) {
sign = -1;
{ longobject *temp = a; a = b; b = temp; }
{ int size_temp = size_a; size_a = size_b; size_b = size_temp; }
}
else if (size_a == size_b) {
/* Find highest digit where a and b differ: */
i = size_a;
while (--i >= 0 && a->ob_digit[i] == b->ob_digit[i])
;
if (i < 0)
return alloclongobject(0);
if (a->ob_digit[i] < b->ob_digit[i]) {
sign = -1;
{ longobject *temp = a; a = b; b = temp; }
}
size_a = size_b = i+1;
}
z = alloclongobject(size_a);
if (z == NULL)
return NULL;
for (i = 0; i < size_b; ++i) {
/* The following assumes unsigned arithmetic
works module 2**N for some N>SHIFT. */
borrow = a->ob_digit[i] - b->ob_digit[i] - borrow;
z->ob_digit[i] = borrow & MASK;
borrow >>= SHIFT;
borrow &= 1; /* Keep only one sign bit */
}
for (; i < size_a; ++i) {
borrow = a->ob_digit[i] - borrow;
z->ob_digit[i] = borrow & MASK;
borrow >>= SHIFT;
}
assert(borrow == 0);
if (sign < 0)
z->ob_size = ~z->ob_size;
1991-05-05 17:09:44 -03:00
return long_normalize(z);
}
static object *
long_add(a, w)
longobject *a;
object *w;
{
longobject *b;
longobject *z;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
if (a->ob_size < 0) {
if (b->ob_size < 0) {
z = x_add(a, b);
if (z != NULL && z->ob_size != 0)
z->ob_size = ~z->ob_size;
1991-05-05 17:09:44 -03:00
}
else
z = x_sub(b, a);
}
else {
if (b->ob_size < 0)
z = x_sub(a, b);
else
z = x_add(a, b);
}
return (object *)z;
}
static object *
long_sub(a, w)
longobject *a;
object *w;
{
longobject *b;
longobject *z;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
if (a->ob_size < 0) {
if (b->ob_size < 0)
z = x_sub(a, b);
else
z = x_add(a, b);
if (z != NULL && z->ob_size != 0)
z->ob_size = ~z->ob_size;
1991-05-05 17:09:44 -03:00
}
else {
if (b->ob_size < 0)
z = x_add(a, b);
else
z = x_sub(a, b);
}
return (object *)z;
}
static object *
long_mul(a, w)
longobject *a;
object *w;
{
longobject *b;
int size_a;
int size_b;
longobject *z;
int i;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
size_a = ZABS(a->ob_size);
size_b = ZABS(b->ob_size);
1991-05-05 17:09:44 -03:00
z = alloclongobject(size_a + size_b);
if (z == NULL)
return NULL;
for (i = 0; i < z->ob_size; ++i)
z->ob_digit[i] = 0;
for (i = 0; i < size_a; ++i) {
twodigits carry = 0;
twodigits f = a->ob_digit[i];
int j;
1991-05-14 09:06:49 -03:00
INTRCHECK({
1991-05-05 17:09:44 -03:00
DECREF(z);
err_set(KeyboardInterrupt);
return NULL;
1991-05-14 09:06:49 -03:00
})
1991-05-05 17:09:44 -03:00
for (j = 0; j < size_b; ++j) {
carry += z->ob_digit[i+j] + b->ob_digit[j] * f;
z->ob_digit[i+j] = carry & MASK;
carry >>= SHIFT;
}
for (; carry != 0; ++j) {
assert(i+j < z->ob_size);
carry += z->ob_digit[i+j];
z->ob_digit[i+j] = carry & MASK;
carry >>= SHIFT;
}
}
if (a->ob_size < 0)
z->ob_size = ~z->ob_size;
1991-05-05 17:09:44 -03:00
if (b->ob_size < 0)
z->ob_size = ~z->ob_size;
1991-05-05 17:09:44 -03:00
return (object *) long_normalize(z);
}
static object *
long_div(v, w)
longobject *v;
register object *w;
{
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
return (object *) long_divrem(v, (longobject *)w, (longobject **)0);
}
static object *
long_rem(v, w)
longobject *v;
register object *w;
{
longobject *div, *rem = NULL;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
div = long_divrem(v, (longobject *)w, &rem);
if (div == NULL) {
XDECREF(rem);
rem = NULL;
}
else {
DECREF(div);
}
return (object *) rem;
}
/* The expression a mod b has the value a - b*floor(a/b).
The divrem function gives the remainder after division of
|a| by |b|, with the sign of a. This is also expressed
as a - b*trunc(a/b), if trunc truncates towards zero.
Some examples:
a b a rem b a mod b
13 10 3 3
-13 10 -3 7
13 -10 3 -7
-13 -10 -3 -3
So, to get from rem to mod, we have to add b if a and b
1991-05-14 09:06:49 -03:00
have different signs. We then subtract one from the 'div'
part of the outcome to keep the invariant intact. */
1991-05-05 17:09:44 -03:00
static object *
long_divmod(v, w)
longobject *v;
register object *w;
{
object *z;
longobject *div, *rem;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
div = long_divrem(v, (longobject *)w, &rem);
if (div == NULL) {
XDECREF(rem);
return NULL;
}
if ((v->ob_size < 0) != (((longobject *)w)->ob_size < 0)) {
1991-05-14 09:06:49 -03:00
longobject *temp;
longobject *one;
temp = (longobject *) long_add(rem, w);
1991-05-05 17:09:44 -03:00
DECREF(rem);
1991-05-14 09:06:49 -03:00
rem = temp;
1991-05-05 17:09:44 -03:00
if (rem == NULL) {
DECREF(div);
return NULL;
}
1991-05-14 09:06:49 -03:00
one = (longobject *) newlongobject(1L);
if (one == NULL ||
(temp = (longobject *) long_sub(div, one)) == NULL) {
DECREF(rem);
DECREF(div);
return NULL;
}
DECREF(div);
div = temp;
1991-05-05 17:09:44 -03:00
}
z = newtupleobject(2);
if (z != NULL) {
settupleitem(z, 0, (object *) div);
settupleitem(z, 1, (object *) rem);
}
else {
DECREF(div);
DECREF(rem);
}
return z;
}
static object *
long_pow(a, w)
longobject *a;
object *w;
1991-05-05 17:09:44 -03:00
{
register longobject *b;
longobject *z;
int size_b, i;
1991-05-05 17:09:44 -03:00
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
size_b = b->ob_size;
if (size_b == ~0)
size_b = 0;
else if (size_b < 0) {
1991-12-10 10:00:03 -04:00
err_setstr(ValueError, "long integer to the negative power");
return NULL;
}
z = (longobject *)newlongobject(1L);
INCREF(a);
for (i = 0; i < size_b; ++i) {
digit bi = b->ob_digit[i];
int j;
for (j = 0; j < SHIFT; ++j) {
longobject *temp;
if (bi & 1) {
temp = (longobject *)long_mul(z, (object *)a);
DECREF(z);
z = temp;
if (z == NULL)
break;
}
bi >>= 1;
if (bi == 0 && i+1 == size_b)
break;
temp = (longobject *)long_mul(a, (object *)a);
DECREF(a);
a = temp;
if (a == NULL) {
DECREF(z);
z = NULL;
break;
}
}
if (a == NULL)
break;
}
XDECREF(a);
return (object *)z;
1991-05-05 17:09:44 -03:00
}
static object *
long_invert(v)
longobject *v;
{
longobject *z;
int i = ZABS(v->ob_size);
z = alloclongobject(i);
if (z != NULL) {
z->ob_size = ~ v->ob_size;
while (--i >= 0)
z->ob_digit[i] = v->ob_digit[i];
}
return (object *)z;
}
1991-05-05 17:09:44 -03:00
static object *
long_pos(v)
longobject *v;
{
if (v->ob_size == ~0)
return long_invert(v);
else {
INCREF(v);
return (object *)v;
}
1991-05-05 17:09:44 -03:00
}
static object *
long_neg(v)
longobject *v;
{
if (v->ob_size != 0)
return long_invert(v);
else {
INCREF(v);
return (object *)v;
1991-05-05 17:09:44 -03:00
}
}
static object *
long_abs(v)
longobject *v;
{
if (v->ob_size < 0)
return long_invert(v);
else {
INCREF(v);
return (object *)v;
}
1991-05-05 17:09:44 -03:00
}
1991-05-14 09:06:49 -03:00
static int
long_nonzero(v)
longobject *v;
{
return ZABS(v->ob_size) != 0;
}
static object *
long_rshift(a, b)
longobject *a;
object *b;
{
longobject *z;
long shiftby;
int newsize, wordshift, loshift, hishift, i, j;
digit lomask, himask;
if (!is_longobject(b)) {
err_badarg();
return NULL;
}
shiftby = getlongvalue(b);
if (shiftby == -1L && err_occurred())
return NULL;
if (shiftby < 0) {
1991-12-10 10:00:03 -04:00
err_setstr(ValueError, "negative shift count");
return NULL;
}
if (shiftby > MASK) {
1991-12-10 10:00:03 -04:00
err_setstr(ValueError, "outrageous shift count");
return NULL;
}
wordshift = shiftby / SHIFT;
newsize = ZABS(a->ob_size) - wordshift;
if (newsize <= 0) {
z = alloclongobject(0);
if (a->ob_size < 0 && z != NULL)
z->ob_size = ~0;
return (object *)z;
}
loshift = shiftby % SHIFT;
hishift = SHIFT - loshift;
lomask = ((digit)1 << hishift) - 1;
himask = MASK ^ lomask;
z = alloclongobject(newsize);
if (z == NULL)
return NULL;
if (a->ob_size < 0)
z->ob_size = ~z->ob_size;
for (i = 0, j = wordshift; i < newsize; i++, j++) {
z->ob_digit[i] = (a->ob_digit[j] >> loshift) & lomask;
if (i+1 < newsize)
z->ob_digit[i] |=
(a->ob_digit[j+1] << hishift) & himask;
}
return (object *) long_znormalize(z);
}
static object *
long_lshift(a, b)
longobject *a;
object *b;
{
longobject *z;
long shiftby;
int newsize, wordshift, loshift, hishift, i, j;
digit lomask, himask;
if (!is_longobject(b)) {
err_badarg();
return NULL;
}
shiftby = getlongvalue(b);
if (shiftby == -1L && err_occurred())
return NULL;
if (shiftby < 0) {
1991-12-10 10:00:03 -04:00
err_setstr(ValueError, "negative shift count");
return NULL;
}
if (shiftby > MASK) {
1991-12-10 10:00:03 -04:00
err_setstr(ValueError, "outrageous shift count");
return NULL;
}
if (shiftby % SHIFT == 0) {
wordshift = shiftby / SHIFT;
loshift = 0;
hishift = SHIFT;
newsize = ZABS(a->ob_size) + wordshift;
lomask = MASK;
himask = 0;
}
else {
wordshift = shiftby / SHIFT + 1;
loshift = SHIFT - shiftby%SHIFT;
hishift = shiftby % SHIFT;
newsize = ZABS(a->ob_size) + wordshift;
lomask = ((digit)1 << hishift) - 1;
himask = MASK ^ lomask;
}
z = alloclongobject(newsize);
if (z == NULL)
return NULL;
if (a->ob_size < 0)
z->ob_size = ~z->ob_size;
for (i = 0; i < wordshift; i++)
z->ob_digit[i] = 0;
for (i = wordshift, j = 0; i < newsize; i++, j++) {
if (i > 0)
z->ob_digit[i-1] |=
(a->ob_digit[j] << hishift) & himask;
z->ob_digit[i] =
(a->ob_digit[j] >> loshift) & lomask;
}
return (object *) long_znormalize(z);
}
1991-12-31 09:14:13 -04:00
#define MAX(x, y) ((x) < (y) ? (y) : (x))
#define MIN(x, y) ((x) > (y) ? (y) : (x))
/* Logical or the absolute values of two long integers.
The second value is first xor'ed with 'mask'. */
static longobject *x_or PROTO((longobject *, longobject *, int));
static longobject *
x_or(a, b, mask)
longobject *a, *b;
int mask;
{
int size_a = ZABS(a->ob_size), size_b = ZABS(b->ob_size);
1991-12-31 09:14:13 -04:00
int size_max = MAX(size_a, size_b);
int size_min = MIN(size_a, size_b);
longobject *z;
int i;
1991-12-31 09:14:13 -04:00
z = alloclongobject(size_max);
if (z == NULL)
return NULL;
1991-12-31 09:14:13 -04:00
for (i = 0; i < size_min; ++i) {
z->ob_digit[i] = a->ob_digit[i] | (b->ob_digit[i] ^ mask);
}
1991-12-31 09:14:13 -04:00
/* At most one of the following two loops executes */
for (; i < size_a; ++i) {
1991-12-31 09:14:13 -04:00
z->ob_digit[i] = a->ob_digit[i] | (0 ^ mask);
}
for (; i < size_b; ++i) {
z->ob_digit[i] = 0 | (b->ob_digit[i] ^ mask);
}
return long_znormalize(z);
}
/* Logical and the absolute values of two long integers.
The second value is first xor'ed with 'mask'. */
static longobject *x_and PROTO((longobject *, longobject *, int));
static longobject *
x_and(a, b, mask)
longobject *a, *b;
int mask;
{
int size_a = ZABS(a->ob_size), size_b = ZABS(b->ob_size);
1991-12-31 09:14:13 -04:00
int size_max = MAX(size_a, size_b);
int size_min = MIN(size_a, size_b);
longobject *z;
int i;
1991-12-31 09:14:13 -04:00
z = alloclongobject(size_max);
if (z == NULL)
return NULL;
1991-12-31 09:14:13 -04:00
for (i = 0; i < size_min; ++i) {
z->ob_digit[i] = a->ob_digit[i] & (b->ob_digit[i] ^ mask);
}
1991-12-31 09:14:13 -04:00
/* At most one of the following two loops executes */
for (; i < size_a; ++i) {
1991-12-31 09:14:13 -04:00
z->ob_digit[i] = a->ob_digit[i] & (0 ^ mask);
}
for (; i < size_b; ++i) {
z->ob_digit[i] = 0 & (b->ob_digit[i] ^ mask);
}
return long_znormalize(z);
}
/* Logical xor the absolute values of two long integers.
The second value is first xor'ed with 'mask'. */
1991-12-31 09:14:13 -04:00
static longobject *x_xor PROTO((longobject *, longobject *, int));
static longobject *
x_xor(a, b, mask)
longobject *a, *b;
int mask;
{
int size_a = ZABS(a->ob_size), size_b = ZABS(b->ob_size);
1991-12-31 09:14:13 -04:00
int size_max = MAX(size_a, size_b);
int size_min = MIN(size_a, size_b);
longobject *z;
int i;
1991-12-31 09:14:13 -04:00
z = alloclongobject(size_max);
if (z == NULL)
return NULL;
1991-12-31 09:14:13 -04:00
for (i = 0; i < size_min; ++i) {
z->ob_digit[i] = a->ob_digit[i] ^ (b->ob_digit[i] ^ mask);
}
1991-12-31 09:14:13 -04:00
/* At most one of the following two loops executes */
for (; i < size_a; ++i) {
1991-12-31 09:14:13 -04:00
z->ob_digit[i] = a->ob_digit[i] ^ (0 ^ mask);
}
for (; i < size_b; ++i) {
z->ob_digit[i] = 0 ^ (b->ob_digit[i] ^ mask);
}
return long_znormalize(z);
}
static object *
long_and(a, w)
longobject *a;
object *w;
{
longobject *b, *z;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
if (a->ob_size >= 0 && b->ob_size >= 0)
z = x_and(a, b, 0);
else if (a->ob_size >= 0 && b->ob_size < 0)
z = x_and(a, b, MASK);
else if (a->ob_size < 0 && b->ob_size >= 0)
z = x_and(b, a, MASK);
else {
z = x_or(a, b, 0);
z->ob_size = ~z->ob_size;
}
return (object *)z;
}
static object *
long_xor(a, w)
longobject *a;
object *w;
{
longobject *b, *z;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
if (a->ob_size >= 0 && b->ob_size >= 0)
z = x_xor(a, b, 0);
else if (a->ob_size >= 0 && b->ob_size < 0)
z = x_xor(a, b, MASK);
else if (a->ob_size < 0 && b->ob_size >= 0)
z = x_xor(b, a, MASK);
else {
z = x_xor(a, b, 0);
z->ob_size = ~z->ob_size;
}
return (object *)z;
}
static object *
long_or(a, w)
longobject *a;
object *w;
{
longobject *b, *z;
if (!is_longobject(w)) {
err_badarg();
return NULL;
}
b = (longobject *)w;
if (a->ob_size >= 0 && b->ob_size >= 0)
z = x_or(a, b, 0);
else {
if (a->ob_size < 0 && b->ob_size >= 0)
z = x_and(a, b, MASK);
else if (a->ob_size >= 0 && b->ob_size < 0)
z = x_and(b, a, MASK);
else
z = x_and(a, b, 0);
z->ob_size = ~z->ob_size;
}
return (object *)z;
1991-05-14 09:06:49 -03:00
}
1991-05-05 17:09:44 -03:00
static number_methods long_as_number = {
long_add, /*nb_add*/
long_sub, /*nb_subtract*/
long_mul, /*nb_multiply*/
long_div, /*nb_divide*/
long_rem, /*nb_remainder*/
long_divmod, /*nb_divmod*/
long_pow, /*nb_power*/
long_neg, /*nb_negative*/
long_pos, /*tp_positive*/
long_abs, /*tp_absolute*/
1991-05-14 09:06:49 -03:00
long_nonzero, /*tp_nonzero*/
long_invert, /*nb_invert*/
long_lshift, /*nb_lshift*/
long_rshift, /*nb_rshift*/
long_and, /*nb_and*/
long_xor, /*nb_xor*/
long_or, /*nb_or*/
1991-05-05 17:09:44 -03:00
};
typeobject Longtype = {
OB_HEAD_INIT(&Typetype)
0,
"long int",
sizeof(longobject) - sizeof(digit),
sizeof(digit),
long_dealloc, /*tp_dealloc*/
long_print, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
long_compare, /*tp_compare*/
long_repr, /*tp_repr*/
&long_as_number,/*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
};