From 7777156a971e5739852cc69ec8ac339caca4c76e Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 15 Jan 2020 11:47:05 +0100
Subject: [PATCH] Add fd.Diff.interpolated_access
---
pystencils/fd/derivative.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/pystencils/fd/derivative.py b/pystencils/fd/derivative.py
index 7acd24505..e3fa11108 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
--
GitLab