From 398a73e2c25c9b07b36b778845e2286fdcc11887 Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Mon, 19 May 2014 21:42:23 -0700 Subject: [PATCH] return value on plot slice --- SimPEG/Mesh/View.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/SimPEG/Mesh/View.py b/SimPEG/Mesh/View.py index eed8d06a..a4753b5e 100644 --- a/SimPEG/Mesh/View.py +++ b/SimPEG/Mesh/View.py @@ -254,7 +254,7 @@ class TensorView(object): assert isinstance(ax, matplotlib.axes.Axes), "ax must be an matplotlib.axes.Axes" fig = ax.figure - tM._plotImage2D(v2d, vType=('CCv' if view == 'vec' else 'CC'), grid=grid, view=view, + out = tM._plotImage2D(v2d, vType=('CCv' if view == 'vec' else 'CC'), grid=grid, view=view, ax=ax, clim=clim, showIt=showIt, pcolorOpts=pcolorOpts, streamOpts=streamOpts, gridOpts=gridOpts) @@ -263,6 +263,7 @@ class TensorView(object): ax.set_xlabel('y' if normal == 'X' else 'x') ax.set_ylabel('y' if normal == 'Z' else 'z') ax.set_title('Slice %d' % ind) + return out def _plotImage2D(self, v, vType='CC', grid=False, view='real',