integer_functions.py 1.4 KB
 Martin Bauer committed May 11, 2018 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 ``````import sympy as sp from pystencils.data_types import get_type_of_expression, collate_types from pystencils.sympyextensions import is_integer_sequence bitwise_xor = sp.Function("bitwise_xor") bit_shift_right = sp.Function("bit_shift_right") bit_shift_left = sp.Function("bit_shift_left") bitwise_and = sp.Function("bitwise_and") bitwise_or = sp.Function("bitwise_or") # noinspection PyPep8Naming class modulo_floor(sp.Function): """Returns the next smaller integer divisible by given divisor. Examples: >>> modulo_floor(9, 4) 8 >>> modulo_floor(11, 4) 8 >>> modulo_floor(12, 4) 12 >>> from pystencils import TypedSymbol >>> a, b = TypedSymbol("a", "int64"), TypedSymbol("b", "int32") >>> modulo_floor(a, b).to_c(str) '(int64_t)((a) / (b)) * (b)' """ nargs = 2 def __new__(cls, integer, divisor): if is_integer_sequence((integer, divisor)): return (int(integer) // int(divisor)) * divisor else: return super().__new__(cls, integer, divisor) def to_c(self, print_func): dtype = collate_types((get_type_of_expression(self.args[0]), get_type_of_expression(self.args[1]))) assert dtype.is_int() return "({dtype})(({0}) / ({1})) * ({1})".format(print_func(self.args[0]), print_func(self.args[1]), dtype=dtype)``````