1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
|
from pypy.interpreter.gateway import unwrap_spec
from pypy.interpreter.error import oefmt
from rpython.rtyper.lltypesystem import lltype
from rpython.rtyper.lltypesystem.lloperation import llop
from rpython.rlib.rarithmetic import r_uint, intmask
from rpython.rlib.rarithmetic import int_c_div, int_c_mod, mulmod
from rpython.rlib import jit
@unwrap_spec(n=int, m=int)
def int_add(space, n, m):
return space.newint(llop.int_add(lltype.Signed, n, m))
@unwrap_spec(n=int, m=int)
def int_sub(space, n, m):
return space.newint(llop.int_sub(lltype.Signed, n, m))
@unwrap_spec(n=int, m=int)
def int_mul(space, n, m):
return space.newint(llop.int_mul(lltype.Signed, n, m))
@unwrap_spec(n=int, m=int)
def int_floordiv(space, n, m):
if m == 0:
raise oefmt(space.w_ZeroDivisionError, "division by zero")
return space.newint(int_c_div(n, m))
@unwrap_spec(n=int, m=int)
def int_mod(space, n, m):
if m == 0:
raise oefmt(space.w_ZeroDivisionError, "division by zero")
return space.newint(int_c_mod(n, m))
@unwrap_spec(n=int, m=int)
def int_lshift(space, n, m):
return space.newint(llop.int_lshift(lltype.Signed, n, m))
@unwrap_spec(n=int, m=int)
def int_rshift(space, n, m):
return space.newint(llop.int_rshift(lltype.Signed, n, m))
@unwrap_spec(n=int, m=int)
def uint_rshift(space, n, m):
n = r_uint(n)
x = llop.uint_rshift(lltype.Unsigned, n, m)
return space.newint(intmask(x))
@unwrap_spec(a=int, b=int, c=int)
def int_mulmod(space, a, b, c):
if c <= 0:
raise oefmt(space.w_ValueError, "'c' must be greater than 0")
return space.newint(mulmod(a, b, c))
|