From 381bd68b0c0ef9d18639aadd5371679f56e59476 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 27 Feb 2020 17:32:14 +0100
Subject: [PATCH] Trigger magic
---
.../framework_integration/astnodes.py | 15 ++++++++++++++-
1 file changed, 14 insertions(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py
index f530c97..efe5e6a 100644
--- a/src/pystencils_autodiff/framework_integration/astnodes.py
+++ b/src/pystencils_autodiff/framework_integration/astnodes.py
@@ -400,10 +400,23 @@ class CustomFunctionCall(JinjaCppFile):
def symbols_defined(self):
return set(self.ast_dict.fields_accessed)
+ @property
+ def fields_accessed(self):
+ return [f.name for f in self.ast_dict.fields_accessed]
+
@property
def function_name(self):
return self.ast_dict.function_name
@property
def undefined_symbols(self):
- return set(self.ast_dict.args)
+ return set(self.ast_dict.args)
+
+ def subs(self, subs_dict):
+ self.ast_dict.args = list(map(lambda x: x.subs(subs_dict), self.ast_dict.args))
+
+ def atoms(self, types=None):
+ if types:
+ return set(a for a in self.args if isinstance(a, types))
+ else:
+ return set(self.args)
--
GitLab