Skip to content
Snippets Groups Projects
Commit 19116526 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Add MeshNormalFunctor

parent 474d6f21
No related branches found
No related tags found
No related merge requests found
...@@ -328,3 +328,20 @@ class DynamicFunction(sp.Function): ...@@ -328,3 +328,20 @@ class DynamicFunction(sp.Function):
def __repr__(self): def __repr__(self):
return self.__str__() return self.__str__()
class MeshNormalFunctor(DynamicFunction):
def __new__(cls, mesh_name, base_dtype, *args):
from pystencils.data_types import TypedMatrixSymbol
A = TypedMatrixSymbol('A', 3, 1, base_dtype, 'Vector3<real_t>')
obj = DynamicFunction.__new__(cls,
TypedSymbol(str(mesh_name),
'std::function<Vector3<real_t>(int, int, int)>'),
A.dtype,
*args)
obj.mesh_name = mesh_name
return obj
def __getnewargs__(self):
return self.mesh_name, self.dtype.base_dtype, self.args[2:]
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment