def save_visualization(self, file_or_fname):
"""Given a file or a file name, this saves the current
visualization to the file.
"""
# Save the state of VTK's global warning display.
o = vtk.vtkObject
w = o.GetGlobalWarningDisplay()
o.SetGlobalWarningDisplay(0) # Turn it off.
try:
#FIXME: This is for streamline seed point widget position which
#does not get serialized correctly
if is_old_pipeline():
state_pickler.dump(self, file_or_fname)
else:
state = state_pickler.get_state(self)
st = state.scenes[0].children[0].children[0].children[4]
l_pos = st.seed.widget.position
st.seed.widget.position = [pos.item() for pos in l_pos]
saved_state = state_pickler.dumps(state)
file_or_fname.write(saved_state)
except (IndexError, AttributeError):
state_pickler.dump(self, file_or_fname)
finally:
# Reset the warning state.
o.SetGlobalWarningDisplay(w)
def test_pickle(self):
"""Test if pickler works correctly with FilePaths."""
t = Test()
t.f.set("t.vtk")
cwd = os.getcwd()
curdir = basename(cwd)
# Create a dummy file in the parent dir.
s = BytesIO()
# Spoof its location.
s.name = abspath(join(cwd, os.pardir, "t.mv2"))
# Dump into it
state_pickler.dump(t, s)
# Rewind the stream
s.seek(0)
# "Move" the file elsewhere
s.name = join(cwd, "foo", "test", "t.mv2")
state = state_pickler.load_state(s)
self.assertEqual(state.f.abs_pth,
join(cwd, "foo", "test", curdir, "t.vtk"))
# Create a dummy file in a subdir.
s = BytesIO()
# Spoof its location.
s.name = abspath(join(cwd, "data", "t.mv2"))
# Dump into it.
state_pickler.dump(t, s)
# Rewind the stream
s.seek(0)
# "Move" the file elsewhere
s.name = join(cwd, "foo", "test", "t.mv2")
state = state_pickler.load_state(s)
self.assertEqual(state.f.abs_pth, join(cwd, "foo", "t.vtk"))
def save_visualization(self, file_or_fname):
"""Given a file or a file name, this saves the current
visualization to the file.
"""
# Save the state of VTK's global warning display.
o = vtk.vtkObject
w = o.GetGlobalWarningDisplay()
o.SetGlobalWarningDisplay(0) # Turn it off.
try:
#FIXME: This is for streamline seed point widget position which
#does not get serialized correctly
if is_old_pipeline():
state_pickler.dump(self, file_or_fname)
else:
state = state_pickler.get_state(self)
st = state.scenes[0].children[0].children[0].children[4]
l_pos = st.seed.widget.position
st.seed.widget.position = [pos.item() for pos in l_pos]
saved_state = state_pickler.dumps(state)
file_or_fname.write(saved_state)
except (IndexError, AttributeError):
state_pickler.dump(self, file_or_fname)
finally:
# Reset the warning state.
o.SetGlobalWarningDisplay(w)
def test_dump_to_file_str(self):
"""Test if dump can take a str as file"""
obj = A()
filepath = os.path.join(tempfile.gettempdir(), "tmp.file")
try:
state_pickler.dump(obj, filepath)
finally:
os.remove(filepath)
def test_dump_to_file_str(self):
"""Test if dump can take a str as file"""
obj = A()
filepath = os.path.join(tempfile.gettempdir(), "tmp.file")
try:
state_pickler.dump(obj, filepath)
finally:
os.remove(filepath)
def save_visualization(self, file_or_fname):
"""Given a file or a file name, this saves the current
visualization to the file.
"""
# Save the state of VTK's global warning display.
o = vtk.vtkObject
w = o.GetGlobalWarningDisplay()
o.SetGlobalWarningDisplay(0) # Turn it off.
try:
state_pickler.dump(self, file_or_fname)
finally:
# Reset the warning state.
o.SetGlobalWarningDisplay(w)
def save_visualization(self, file_or_fname):
"""Given a file or a file name, this saves the current
visualization to the file.
"""
# Save the state of VTK's global warning display.
o = vtk.vtkObject
w = o.GetGlobalWarningDisplay()
o.SetGlobalWarningDisplay(0) # Turn it off.
try:
state_pickler.dump(self, file_or_fname)
finally:
# Reset the warning state.
o.SetGlobalWarningDisplay(w)
def test_pickle(self):
"""Test if pickler works correctly with FilePaths.
"""
t = Test()
t.f.set('t.vtk')
cwd = os.getcwd()
curdir = basename(cwd)
# Create a dummy file in the parent dir.
s = BytesIO()
# Spoof its location.
s.name = abspath(join(cwd, os.pardir, 't.mv2'))
# Dump into it
state_pickler.dump(t, s)
# Rewind the stream
s.seek(0)
# "Move" the file elsewhere
s.name = join(cwd, 'foo', 'test', 't.mv2')
state = state_pickler.load_state(s)
self.assertEqual(state.f.abs_pth,
join(cwd, 'foo', 'test', curdir, 't.vtk'))
# Create a dummy file in a subdir.
s = BytesIO()
# Spoof its location.
s.name = abspath(join(cwd, 'data', 't.mv2'))
# Dump into it.
state_pickler.dump(t, s)
# Rewind the stream
s.seek(0)
# "Move" the file elsewhere
s.name = join(cwd, 'foo', 'test', 't.mv2')
state = state_pickler.load_state(s)
self.assertEqual(state.f.abs_pth,
join(cwd, 'foo', 't.vtk'))
# Authors: Frederic Petit <*****@*****.**>,
# Gael Varoquaux <*****@*****.**>
# Copyright (c) 2007-2020, Enthought, Inc.
# License: BSD Style.
import os
import numpy as np
from matplotlib.cm import datad, get_cmap
from matplotlib._cm_listed import cmaps
from mayavi.core import lut as destination_module
from apptools.persistence import state_pickler
target_dir = os.path.dirname(destination_module.__file__)
values = np.linspace(0., 1., 256)
lut_dic = {}
# Some of the cmaps are listed in cm.datad, and others in _cm_listed.cmaps
cmap_names = datad.keys()
cmap_names.extend(cmaps.keys())
for name in cmap_names:
if name.endswith('_r'):
continue
lut_dic[name] = get_cmap(name)(values.copy())
out_name = os.path.join(target_dir, 'pylab_luts.pkl')
state_pickler.dump(lut_dic, out_name)
# Authors: Frederic Petit <*****@*****.**>,
# Gael Varoquaux <*****@*****.**>
# Copyright (c) 2007-2009, Enthought, Inc.
# License: BSD Style.
import os
import numpy as np
from matplotlib.cm import datad, get_cmap
from matplotlib._cm_listed import cmaps
from mayavi.core import lut as destination_module
from apptools.persistence import state_pickler
target_dir = os.path.dirname(destination_module.__file__)
values = np.linspace(0., 1., 256)
lut_dic = {}
# Some of the cmaps are listed in cm.datad, and others in _cm_listed.cmaps
cmap_names = datad.keys()
cmap_names.extend(cmaps.keys())
for name in cmap_names:
if name.endswith('_r'):
continue
lut_dic[name] = get_cmap(name)(values.copy())
out_name = os.path.join(target_dir, 'pylab_luts.pkl')
state_pickler.dump(lut_dic, out_name)
