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 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 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)