From f6a43abae28de938f3fad7de09a867c1279d20cb Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Tue, 28 Mar 2023 09:23:07 +0200
Subject: [PATCH] [Fix] Matplotlib arrow rendering
---
pystencils/stencil.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/pystencils/stencil.py b/pystencils/stencil.py
index 48cee7c02..10925735a 100644
--- a/pystencils/stencil.py
+++ b/pystencils/stencil.py
@@ -421,11 +421,12 @@ def plot_3d(stencil, figure=None, axes=None, data=None, textsize='8'):
FancyArrowPatch.__init__(self, (0, 0), (0, 0), *args, **kwargs)
self._verts3d = xs, ys, zs
- def draw(self, renderer):
+ def do_3d_projection(self, *_):
xs3d, ys3d, zs3d = self._verts3d
xs, ys, zs = proj3d.proj_transform(xs3d, ys3d, zs3d, self.axes.M)
self.set_positions((xs[0], ys[0]), (xs[1], ys[1]))
- FancyArrowPatch.draw(self, renderer)
+
+ return np.min(zs)
if axes is None:
if figure is None:
--
GitLab