From 63bab1b2e935c53a55aec91975619b80f2459e51 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 22 Jan 2020 11:13:52 +0100
Subject: [PATCH] Add _latex representation for InterpolatorAccess
---
pystencils/interpolation_astnodes.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/pystencils/interpolation_astnodes.py b/pystencils/interpolation_astnodes.py
index 46946194a..0d27475c9 100644
--- a/pystencils/interpolation_astnodes.py
+++ b/pystencils/interpolation_astnodes.py
@@ -172,6 +172,11 @@ class InterpolatorAccess(TypedSymbol):
def __repr__(self):
return self.__str__()
+ def _latex(self, printer, *_):
+ n = self.field.latex_name if self.field.latex_name else self.field.name
+ foo = ", ".join(str(printer.doprint(o)) for o in self.offsets)
+ return f'{n}_{{interpolator}}\\left({foo}\\right)'
+
@property
def ndim(self):
return len(self.offsets)
--
GitLab