if fig is None:

fig = mlab.gcf()

try:

# scene.set_size doesn't seem to work on linux && os x, so

# go into the backend and do it by hand

if sys.platform == "darwin" or sys.platform.startswith('linux'):

toolkit = mayavi.ETSConfig.toolkit

if toolkit == 'qt4':

sc = fig.scene

window_height = sc.control.parent().size().height()

render_height = sc.render_window.size[1]

h = window_height - render_height

sc.control.parent().resize(size[0], size[1] + h)

elif toolkit == 'wx':

w, h = size[0], size[1]

fig.scene.control.Parent.Parent.SetClientSizeWH(w, h)

else:

print("Unknown mayavi backend {0} (not qt4 or "

"wx); not resizing.".format(toolkit), file=sys.stderr)

else:

fig.scene.set_size(size)

except Exception as e:

"""Try to show the window; only does something on Qt backend"""

try:

# fig.scene.control.parent().show()

fig.scene.control.parent().showNormal()

except Exception as e: # pylint: disable=broad-except,unused-variable

if debug:

"""Try to hide the window; only does something on Qt backend"""

try:

# fig.scene.control.parent().hide()

fig.scene.control.parent().showMinimized()

except Exception as e: # pylint: disable=broad-except,unused-variable

if debug:

**kwargs):

"""Display a mayavi figure inline in an ipython notebook.

This function takes a screenshot of a figure and blits it to a matplotlib

figure using matplotlib.pyplot.imshow()

Args:

fig: A mayavi figure, if not specified, uses mlab.gcf()

size (None, tuple): if given, resize the scene in pixels (x, y)

hide (bool): if True, try to hide the render window

kwargs: passed to mayavi.mlab.screenshot()

"""

from matplotlib import pyplot as plt

if fig is None:

fig = mlab.gcf()

# try to show the window... Qt backend only, necessary if this is

# the 2nd call, and we hid the previous window

# imayavi_show_window(fig)

if size is not None:

_resize_window(size, fig=fig)

pixmap = mlab.screenshot(fig, mode='rgb', antialiased=True)

# try to hide the window... Qt backend only

if hide:

imayavi_hide_window(fig)

pltfig = plt.figure()

dpi = pltfig.get_dpi()

pltfig.set_size_inches([s / dpi for s in fig.scene.get_size()])

ax = pltfig.gca()

ax.imshow(pixmap)

ax.axis('off')

plt.show()

**kwargs):

"""Display a mayavi figure inline in an ipython notebook.

This function takes a screenshot of a figure and blits it to a matplotlib

figure using matplotlib.pyplot.imshow()

Args:

fig: A mayavi figure, if not specified, uses mlab.gcf()

size (None, tuple): if given, resize the scene in pixels (x, y)

hide (bool): if True, try to hide the render window

kwargs: passed to mayavi.mlab.screenshot()

"""

# from matplotlib import pyplot as plt

if fig is None:

fig = mlab.gcf()

# fig.bgcolor = (1, 1, 1)

# fig = mlab.figure(20, bgcolor = (1, 1, 1))

# try to show the window... Qt backend only, necessary if this is

# the 2nd call, and we hid the previous window

imayavi_show_window(fig)

if size is not None:

_resize_window(size, fig=fig)

pixmap = mlab.screenshot(fig, mode='rgb', antialiased=True)

# try to hide the window... Qt backend only

# if hide:

# imayavi_hide_window(fig)

# imayavi_clear_data()

# mlab.clf(fig)

# mlab.close(fig)

"""shortcut for `mayavi.mlab.show(stop=True)`"""

if sys.platform != "darwin":

print("Warning: Since linux uses offscreen rendering, imayavi_show()\n"

"won't work. To interact with a plot, turn off\n"

"offscreen rendering and recreate the plot. Remember to\n"

"re-enable offscreen rendering if making future inline plots.\n",

file=sys.stderr)

mlab.options.offscreen = False

if fig is None:

fig = mlab.gcf()

print("mlab.gcf()")

imayavi_show_window(fig)

if size is not None:

_resize_window(size, fig=fig)

print("_resize_window(size, fig=fig)")

"""Safely remove a specific vtk source

Args:

src (vtk_data_source): vtk data source to remove

"""

src.stop()

try:

try:

src.data.release_data()

except TraitError:

src.data.release_data_flag = 1

src.cell_scalars_name = ''

src.cell_tensors_name = ''

src.cell_vectors_name = ''

src.point_scalars_name = ''

src.point_tensors_name = ''

src.point_vectors_name = ''

except AttributeError:

pass

src.start()

src.stop()

"""Workaround for Mayavi / VTK memory leak

This is needed when Mayavi/VTK keeps a reference to source data

when you would expect it to be freed like on a call to `mlab.clf()`

or when removing sources from the pipeline.

Note:

This must be called when the pipeline still has the source, so

before a call to `mlab.clf()`, etc.

1. Set release_data_flag on all sources' data

2. Remove reference to the data

3. Remove the data source

Args:

scene (None, mayavi.core.scene.Scene, or 'all'): if None, gets

current scene; if Scene object, just that one; if 'all',

act on all scenes in the current engine. Can also be a list

of Scene objects

"""

if scenes is None:

scenes = [mlab.get_engine().current_scene]

elif scenes == "all":

scenes = mlab.get_engine().scenes

if not isinstance(scenes, (list, tuple)):

scenes = [scenes]

if all(s is None for s in scenes):

return

for s in scenes:

s.stop()

for child in list(s.children):

imayavi_remove_source(child)

s.start()