Zxc200611/pocketpy
1
0
Fork 0
mirror of https://github.com/pocketpy/pocketpy synced 2025-10-20 11:30:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

...

Browse Source
This commit is contained in:
blueloveTH 2023-06-15 16:40:45 +08:00
parent 5ee3779a6b
commit 6b9f2d0fba
2 changed files with 29 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
python/_long.py
View File

@ -165,8 +165,12 @@ class long:
self.digits, self.sign = x self.digits, self.sign = x
elif type(x) is int: elif type(x) is int:
self.digits, self.sign = ulong_fromint(x) self.digits, self.sign = ulong_fromint(x)
elif type(x) is float:
self.digits, self.sign = ulong_fromint(int(x))
elif type(x) is str: elif type(x) is str:
self.digits, self.sign = ulong_fromstr(x) self.digits, self.sign = ulong_fromstr(x)
elif type(x) is long:
self.digits, self.sign = x.digits.copy(), x.sign
else: else:
raise TypeError('expected int or str') raise TypeError('expected int or str')
@ -228,11 +232,13 @@ class long:
return self.__mul__(other) return self.__mul__(other)
####################################################### #######################################################
def __divmod__(self, other: int): def __divmod__(self, other):
if type(other) is int:
assert type(other) is int and other > 0 assert type(other) is int and other > 0
assert self.sign == 1 assert self.sign == 1
q, r = ulong_divmodi(self.digits, other) q, r = ulong_divmodi(self.digits, other)
return long((q, 1)), r return long((q, 1)), r
raise NotImplementedError
def __floordiv__(self, other: int): def __floordiv__(self, other: int):
return self.__divmod__(other)[0] return self.__divmod__(other)[0]
@ -249,30 +255,22 @@ class long:
return long((ulong_powi(self.digits, other), sign)) return long((ulong_powi(self.digits, other), sign))
def __lshift__(self, other: int): def __lshift__(self, other: int):
# TODO: optimize
assert type(other) is int and other >= 0 assert type(other) is int and other >= 0
x = self.digits.copy() x = self.digits.copy()
for _ in range(other): q, r = divmod(other, PyLong_SHIFT)
x = ulong_muli(x, 2) x = [0]*q + x
for _ in range(r): x = ulong_muli(x, 2)
return long((x, self.sign)) return long((x, self.sign))
def __rshift__(self, other: int): def __rshift__(self, other: int):
# TODO: optimize
assert type(other) is int and other >= 0 assert type(other) is int and other >= 0
x = self.digits.copy() x = self.digits.copy()
for _ in range(other): q, r = divmod(other, PyLong_SHIFT)
x = ulong_floordivi(x, 2) x = x[q:]
if not x: return long(0)
for _ in range(r): x = ulong_floordivi(x, 2)
return long((x, self.sign)) return long((x, self.sign))
def __and__(self, other):
raise NotImplementedError
def __or__(self, other):
raise NotImplementedError
def __xor__(self, other):
raise NotImplementedError
def __neg__(self): def __neg__(self):
return long((self.digits, -self.sign)) return long((self.digits, -self.sign))

10
tests/09_long.py
View File

@ -14,3 +14,13 @@ assert -a == -2
assert 1 + a == 3L assert 1 + a == 3L
assert 1 - a == -1L assert 1 - a == -1L
assert 2 * a == 4L assert 2 * a == 4L
# __lshift__ and __rshift__
for i in range(29):
assert 1L << i == 2 ** i
for i in range(29):
assert 2L ** i >> i == 1L
assert 12L >> 100 == 0