Commit 50f27d79 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Add pystencils.data_types.address_of

parent 0690f5ad
......@@ -14,6 +14,33 @@ from pystencils.utils import all_equal
from sympy.logic.boolalg import Boolean
# noinspection PyPep8Naming
class address_of(sp.Function):
is_Atom = True
def __new__(cls, arg):
obj = sp.Function.__new__(cls, arg)
return obj
def canonical(self):
if hasattr(self.args[0], 'canonical'):
return self.args[0].canonical
raise NotImplementedError()
def is_commutative(self):
return self.args[0].is_commutative
def dtype(self):
if hasattr(self.args[0], 'dtype'):
return PointerType(self.args[0].dtype, restrict=True)
return PointerType('void', restrict=True)
# noinspection PyPep8Naming
class cast_func(sp.Function):
is_Atom = True
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment