diff --git a/pystencils/fd/derivative.py b/pystencils/fd/derivative.py index 7acd245059615ded27e2c1fe023e344b48c34a6e..e3fa11108acf4f59747fd806c45774070838ab7c 100644 --- a/pystencils/fd/derivative.py +++ b/pystencils/fd/derivative.py @@ -111,6 +111,11 @@ class Diff(sp.Expr): def __str__(self): return "D(%s)" % self.arg + def interpolated_access(self, offset): + 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) + class DiffOperator(sp.Expr): """Un-applied differential, i.e. differential operator