Commit 477785f5 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Allow kwargs in fd.Diff.interpolated_access

parent 39fed6f8
......@@ -111,15 +111,15 @@ class Diff(sp.Expr):
def __str__(self):
return "D(%s)" % self.arg
def interpolated_access(self, offset):
def interpolated_access(self, offset, **kwargs):
"""Represents an interpolated access on a spatially differentiated field
Args:
offset (Tuple[sympy.Expr]): Absolute position to determine the value of the spatial derivative
"""
from pystencils.interpolation_astnodes import DiffInterpolatorAccess
assert isinstance(self.argument, Field), "Must be field to enable interpolated accesses"
return DiffInterpolatorAccess(self.argument.interpolated_access(offset), self.target)
assert isinstance(self.arg.field, Field), "Must be field to enable interpolated accesses"
return DiffInterpolatorAccess(self.arg.field.interpolated_access(offset, **kwargs).symbol, self.target, *offset)
class DiffOperator(sp.Expr):
......
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