def get_data_names(self, source=None, detailed=False):
if source is None:
mlab.options.offscreen = self.offscreen
mlab.figure(fgcolor=self.fgcolor,
bgcolor=self.bgcolor,
size=(1, 1))
source = mlab.pipeline.open(self.filename)
point_scalar_names = sorted(source._point_scalars_list[:-1])
point_vector_names = sorted(source._point_vectors_list[:-1])
point_tensor_names = sorted(source._point_tensors_list[:-1])
cell_scalar_names = sorted(source._cell_scalars_list[:-1])
cell_vector_names = sorted(source._cell_vectors_list[:-1])
cell_tensor_names = sorted(source._cell_tensors_list[:-1])
p_names = [['point', 'scalars', name] for name in point_scalar_names]
p_names += [['point', 'vectors', name] for name in point_vector_names]
p_names += [['point', 'tensors', name] for name in point_tensor_names]
c_names = [['cell', 'scalars', name] for name in cell_scalar_names]
c_names += [['cell', 'vectors', name] for name in cell_vector_names]
c_names += [['cell', 'tensors', name] for name in cell_tensor_names]
if detailed:
return p_names, c_names
else:
return p_names + c_names
def get_data_names(self, source=None, detailed=False):
if source is None:
mlab.options.offscreen = self.offscreen
mlab.figure(fgcolor=self.fgcolor, bgcolor=self.bgcolor,
size=(1, 1))
source = mlab.pipeline.open(self.filename)
point_scalar_names = sorted(source._point_scalars_list[:-1])
point_vector_names = sorted(source._point_vectors_list[:-1])
point_tensor_names = sorted(source._point_tensors_list[:-1])
cell_scalar_names = sorted(source._cell_scalars_list[:-1])
cell_vector_names = sorted(source._cell_vectors_list[:-1])
cell_tensor_names = sorted(source._cell_tensors_list[:-1])
p_names = [['point', 'scalars', name] for name in point_scalar_names]
p_names += [['point', 'vectors', name] for name in point_vector_names]
p_names += [['point', 'tensors', name] for name in point_tensor_names]
c_names = [['cell', 'scalars', name] for name in cell_scalar_names]
c_names += [['cell', 'vectors', name] for name in cell_vector_names]
c_names += [['cell', 'tensors', name] for name in cell_tensor_names]
if detailed:
return p_names, c_names
else:
return p_names + c_names
def call_mlab(self, scene=None, show=True, is_3d=False,
view=None, roll=None,
fgcolor=(0.0, 0.0, 0.0), bgcolor=(1.0, 1.0, 1.0),
layout='rowcol', scalar_mode='iso_surface',
vector_mode='arrows_norm', rel_scaling=None, clamping=False,
ranges=None, is_scalar_bar=False, is_wireframe=False,
opacity=None, subdomains_args=None, rel_text_width=None,
fig_filename='view.png', resolution=None,
filter_names=None, only_names=None, group_names=None,
step=None, time=None,
anti_aliasing=None, domain_specific=None):
"""
By default, all data (point, cell, scalars, vectors, tensors)
are plotted in a grid layout, except data named 'node_groups',
'mat_id' which are usually not interesting.
Parameters
----------
show : bool
Call mlab.show().
is_3d : bool
If True, use scalar cut planes instead of surface for certain
datasets. Also sets 3D view mode.
view : tuple
Azimuth, elevation angles, distance and focal point as in
`mlab.view()`.
roll : float
Roll angle tuple as in mlab.roll().
fgcolor : tuple of floats (R, G, B)
The foreground color, that is the color of all text
annotation labels (axes, orientation axes, scalar bar
labels).
bgcolor : tuple of floats (R, G, B)
The background color.
layout : str
Grid layout for placing the datasets. Possible values are:
'row', 'col', 'rowcol', 'colrow'.
scalar_mode : str
Mode for plotting scalars and tensor magnitudes, one of
'cut_plane', 'iso_surface', 'both'.
vector_mode : str
Mode for plotting vectors, one of 'arrows', 'norm', 'arrows_norm',
'warp_norm'.
rel_scaling : float
Relative scaling of glyphs for vector datasets.
clamping : bool
Clamping for vector datasets.
ranges : dict
List of data ranges in the form {name : (min, max), ...}.
is_scalar_bar : bool
If True, show a scalar bar for each data.
is_wireframe : bool
If True, show a wireframe of mesh surface bar for each data.
opacity : float
Global surface and wireframe opacity setting in [0.0, 1.0],
subdomains_args : tuple
Tuple of (mat_id_name, threshold_limits, single_color), see
:func:`add_subdomains_surface`, or None.
rel_text_width : float
Relative text width.
fig_filename : str
File name for saving the resulting scene figure, if
self.auto_screenshot is True.
resolution : tuple
Scene and figure resolution. If None, it is set
automatically according to the layout.
filter_names : list of strings
Omit the listed datasets. If None, it is initialized to
['node_groups', 'mat_id']. Pass [] if you need no filtering.
only_names : list of strings
Draw only the listed datasets. If None, it is initialized all names
besides those in filter_names.
group_names : list of tuples
List of data names in the form [(name1, ..., nameN), (...)]. Plots
of data named in each group are superimposed. Repetitions of names
are possible.
step : int, optional
If not None, the time step to display. The closest higher step is
used if the desired one is not available. Has precedence over
`time`.
time : float, optional
If not None, the time of the time step to display. The closest
higher time is used if the desired one is not available.
anti_aliasing : int
Value of anti-aliasing.
domain_specific : dict
Domain-specific drawing functions and configurations.
"""
self.fgcolor = fgcolor
self.bgcolor = bgcolor
if filter_names is None:
filter_names = ['node_groups', 'mat_id']
if rel_text_width is None:
rel_text_width = 0.02
if isinstance(scalar_mode, basestr):
if scalar_mode == 'both':
scalar_mode = ('cut_plane', 'iso_surface')
elif scalar_mode in ('cut_plane', 'iso_surface'):
scalar_mode = (scalar_mode,)
else:
raise ValueError('bad value of scalar_mode parameter! (%s)'
% scalar_mode)
else:
for sm in scalar_mode:
if not sm in ('cut_plane', 'iso_surface'):
raise ValueError('bad value of scalar_mode parameter! (%s)'
% sm)
if isinstance(vector_mode, basestr):
if vector_mode == 'arrows_norm':
vector_mode = ('arrows', 'norm')
elif vector_mode == 'warp_norm':
vector_mode = ('warp', 'norm')
elif vector_mode in ('arrows', 'norm'):
vector_mode = (vector_mode,)
elif vector_mode == 'cut_plane':
if is_3d:
vector_mode = ('cut_plane',)
else:
vector_mode = ('arrows',)
else:
raise ValueError('bad value of vector_mode parameter! (%s)'
% vector_mode)
else:
for vm in vector_mode:
if not vm in ('arrows', 'norm', 'warp'):
raise ValueError('bad value of vector_mode parameter! (%s)'
% vm)
mlab.options.offscreen = self.offscreen
self.size_hint = self.get_size_hint(layout, resolution=resolution)
is_new_scene = False
if scene is not None:
if scene is not self.scene:
is_new_scene = True
self.scene = scene
gui = None
else:
if (self.scene is not None) and (not self.scene.running):
self.scene = None
if self.scene is None:
if self.offscreen:
gui = None
scene = mlab.figure(fgcolor=fgcolor, bgcolor=bgcolor,
size=self.size_hint)
else:
gui = ViewerGUI(viewer=self,
fgcolor=fgcolor, bgcolor=bgcolor)
scene = gui.scene.mayavi_scene
if scene is not self.scene:
is_new_scene = True
self.scene = scene
else:
gui = self.gui
scene = self.scene
self.engine = mlab.get_engine()
self.engine.current_scene = self.scene
self.gui = gui
self.file_source = create_file_source(self.filename, watch=self.watch,
offscreen=self.offscreen)
steps, times = self.file_source.get_ts_info()
has_several_times = len(times) > 0
has_several_steps = has_several_times or (len(steps) > 0)
if gui is not None:
gui.has_several_steps = has_several_steps
self.reload_source = reload_source = ReloadSource()
reload_source._viewer = self
reload_source._source = self.file_source
if has_several_steps:
self.set_step = set_step = SetStep()
set_step._viewer = self
set_step._source = self.file_source
if step is not None:
step = step if step >= 0 else steps[-1] + step + 1
assert_(steps[0] <= step <= steps[-1],
msg='invalid time step! (%d <= %d <= %d)'
% (steps[0], step, steps[-1]))
set_step.step = step
elif time is not None:
assert_(times[0] <= time <= times[-1],
msg='invalid time! (%e <= %e <= %e)'
% (times[0], time, times[-1]))
set_step.time = time
else:
set_step.step = steps[0]
if self.watch:
self.file_source.setup_notification(set_step, 'file_changed')
if gui is not None:
gui.set_step = set_step
else:
if self.watch:
self.file_source.setup_notification(reload_source,
'reload_source')
self.options.update(get_arguments(omit = ['self', 'file_source']))
if gui is None:
self.render_scene(scene, self.options)
self.reset_view()
if is_scalar_bar:
self.show_scalar_bars(self.scalar_bars)
else:
traits_view = View(
Item('scene', editor=SceneEditor(scene_class=MayaviScene),
show_label=False,
width=self.size_hint[0], height=self.size_hint[1],
style='custom',
),
Group(Item('set_step', defined_when='set_step is not None',
show_label=False, style='custom'),
),
HGroup(spring,
Item('button_make_snapshots_steps', show_label=False,
enabled_when='has_several_steps == True'),
Item('button_make_animation_steps', show_label=False,
enabled_when='has_several_steps == True'),
spring,
Item('button_make_snapshots_times', show_label=False,
enabled_when='has_several_steps == True'),
Item('button_make_animation_times', show_label=False,
enabled_when='has_several_steps == True'),
spring,),
HGroup(spring,
Item('button_reload', show_label=False),
Item('button_view', show_label=False),
Item('button_quit', show_label=False)),
resizable=True,
buttons=[],
handler=ClosingHandler(),
)
if is_new_scene:
if show:
gui.configure_traits(view=traits_view)
else:
gui.edit_traits(view=traits_view)
if self.auto_screenshot:
self.save_image(fig_filename)
return gui
def call_mlab(self,
scene=None,
show=True,
is_3d=False,
view=None,
roll=None,
parallel_projection=False,
fgcolor=(0.0, 0.0, 0.0),
bgcolor=(1.0, 1.0, 1.0),
colormap='blue-red',
layout='rowcol',
scalar_mode='iso_surface',
vector_mode='arrows_norm',
rel_scaling=None,
clamping=False,
ranges=None,
is_scalar_bar=False,
is_wireframe=False,
opacity=None,
subdomains_args=None,
rel_text_width=None,
fig_filename='view.png',
resolution=None,
filter_names=None,
only_names=None,
group_names=None,
step=None,
time=None,
anti_aliasing=None,
domain_specific=None):
"""
By default, all data (point, cell, scalars, vectors, tensors)
are plotted in a grid layout, except data named 'node_groups',
'mat_id' which are usually not interesting.
Parameters
----------
show : bool
Call mlab.show().
is_3d : bool
If True, use scalar cut planes instead of surface for certain
datasets. Also sets 3D view mode.
view : tuple
Azimuth, elevation angles, distance and focal point as in
`mlab.view()`.
roll : float
Roll angle tuple as in mlab.roll().
parallel_projection: bool
If True, use parallel projection.
fgcolor : tuple of floats (R, G, B)
The foreground color, that is the color of all text
annotation labels (axes, orientation axes, scalar bar
labels).
bgcolor : tuple of floats (R, G, B)
The background color.
colormap : str
The colormap name.
layout : str
Grid layout for placing the datasets. Possible values are:
'row', 'col', 'rowcol', 'colrow'.
scalar_mode : str
Mode for plotting scalars and tensor magnitudes, one of
'cut_plane', 'iso_surface', 'both'.
vector_mode : str
Mode for plotting vectors, one of 'arrows', 'norm', 'arrows_norm',
'warp_norm'.
rel_scaling : float
Relative scaling of glyphs for vector datasets.
clamping : bool
Clamping for vector datasets.
ranges : dict
List of data ranges in the form {name : (min, max), ...}.
is_scalar_bar : bool
If True, show a scalar bar for each data.
is_wireframe : bool
If True, show a wireframe of mesh surface bar for each data.
opacity : float
Global surface and wireframe opacity setting in [0.0, 1.0],
subdomains_args : tuple
Tuple of (mat_id_name, threshold_limits, single_color), see
:func:`add_subdomains_surface`, or None.
rel_text_width : float
Relative text width.
fig_filename : str
File name for saving the resulting scene figure.
resolution : tuple
Scene and figure resolution. If None, it is set
automatically according to the layout.
filter_names : list of strings
Omit the listed datasets. If None, it is initialized to
['node_groups', 'mat_id']. Pass [] if you need no filtering.
only_names : list of strings
Draw only the listed datasets. If None, it is initialized all names
besides those in filter_names.
group_names : list of tuples
List of data names in the form [(name1, ..., nameN), (...)]. Plots
of data named in each group are superimposed. Repetitions of names
are possible.
step : int, optional
If not None, the time step to display. The closest higher step is
used if the desired one is not available. Has precedence over
`time`.
time : float, optional
If not None, the time of the time step to display. The closest
higher time is used if the desired one is not available.
anti_aliasing : int
Value of anti-aliasing.
domain_specific : dict
Domain-specific drawing functions and configurations.
"""
self.fgcolor = fgcolor
self.bgcolor = bgcolor
self.colormap = colormap
if filter_names is None:
filter_names = ['node_groups', 'mat_id']
if rel_text_width is None:
rel_text_width = 0.02
if isinstance(scalar_mode, basestr):
if scalar_mode == 'both':
scalar_mode = ('cut_plane', 'iso_surface')
elif scalar_mode in ('cut_plane', 'iso_surface'):
scalar_mode = (scalar_mode, )
else:
raise ValueError('bad value of scalar_mode parameter! (%s)' %
scalar_mode)
else:
for sm in scalar_mode:
if not sm in ('cut_plane', 'iso_surface'):
raise ValueError(
'bad value of scalar_mode parameter! (%s)' % sm)
if isinstance(vector_mode, basestr):
if vector_mode == 'arrows_norm':
vector_mode = ('arrows', 'norm')
elif vector_mode == 'warp_norm':
vector_mode = ('warp', 'norm')
elif vector_mode in ('arrows', 'norm'):
vector_mode = (vector_mode, )
elif vector_mode == 'cut_plane':
if is_3d:
vector_mode = ('cut_plane', )
else:
vector_mode = ('arrows', )
else:
raise ValueError('bad value of vector_mode parameter! (%s)' %
vector_mode)
else:
for vm in vector_mode:
if not vm in ('arrows', 'norm', 'warp'):
raise ValueError(
'bad value of vector_mode parameter! (%s)' % vm)
mlab.options.offscreen = self.offscreen
self.size_hint = self.get_size_hint(layout, resolution=resolution)
is_new_scene = False
if scene is not None:
if scene is not self.scene:
is_new_scene = True
self.scene = scene
gui = None
else:
if (self.scene is not None) and (not self.scene.running):
self.scene = None
if self.scene is None:
if self.offscreen or not show:
gui = None
scene = mlab.figure(fgcolor=fgcolor,
bgcolor=bgcolor,
size=self.size_hint)
else:
gui = ViewerGUI(viewer=self,
fgcolor=fgcolor,
bgcolor=bgcolor)
scene = gui.scene.mayavi_scene
if scene is not self.scene:
is_new_scene = True
self.scene = scene
else:
gui = self.gui
scene = self.scene
self.engine = mlab.get_engine()
self.engine.current_scene = self.scene
self.gui = gui
self.file_source = create_file_source(self.filename,
watch=self.watch,
offscreen=self.offscreen)
steps, times = self.file_source.get_ts_info()
has_several_times = len(times) > 1
has_several_steps = has_several_times or (len(steps) > 1)
if gui is not None:
gui.has_several_steps = has_several_steps
self.reload_source = reload_source = ReloadSource()
reload_source._viewer = self
reload_source._source = self.file_source
if has_several_steps:
self.set_step = set_step = SetStep()
set_step._viewer = self
set_step._source = self.file_source
if step is not None:
step = step if step >= 0 else steps[-1] + step + 1
assert_(steps[0] <= step <= steps[-1],
msg='invalid time step! (%d <= %d <= %d)' %
(steps[0], step, steps[-1]))
set_step.step = step
elif time is not None:
assert_(times[0] <= time <= times[-1],
msg='invalid time! (%e <= %e <= %e)' %
(times[0], time, times[-1]))
set_step.time = time
else:
set_step.step = steps[0]
if self.watch:
self.file_source.setup_notification(set_step, 'file_changed')
if gui is not None:
gui.set_step = set_step
else:
if self.watch:
self.file_source.setup_notification(reload_source,
'reload_source')
self.options.update(get_arguments(omit=['self', 'file_source']))
if gui is None:
self.render_scene(scene, self.options)
self.reset_view()
if is_scalar_bar:
self.show_scalar_bars(self.scalar_bars)
else:
traits_view = View(
Item(
'scene',
editor=SceneEditor(scene_class=MayaviScene),
show_label=False,
width=self.size_hint[0],
height=self.size_hint[1],
style='custom',
),
Group(
Item('set_step',
defined_when='set_step is not None',
show_label=False,
style='custom'), ),
HGroup(
spring,
Item('button_make_snapshots_steps',
show_label=False,
enabled_when='has_several_steps == True'),
Item('button_make_animation_steps',
show_label=False,
enabled_when='has_several_steps == True'),
spring,
Item('button_make_snapshots_times',
show_label=False,
enabled_when='has_several_steps == True'),
Item('button_make_animation_times',
show_label=False,
enabled_when='has_several_steps == True'),
spring,
),
HGroup(spring, Item('button_reload', show_label=False),
Item('button_view', show_label=False),
Item('button_quit', show_label=False)),
resizable=True,
buttons=[],
handler=ClosingHandler(),
)
if is_new_scene:
if show:
gui.configure_traits(view=traits_view)
else:
gui.edit_traits(view=traits_view)
return gui
def call_mlab(self, scene=None, show=True, is_3d=False, view=None, roll=None,
layout='rowcol', scalar_mode='iso_surface',
vector_mode='arrows_norm', rel_scaling=None, clamping=False,
ranges=None, is_scalar_bar=False, is_wireframe=False,
rel_text_width=None,
fig_filename='view.png', resolution = None,
filter_names=None, only_names=None, group_names=None, step=0,
anti_aliasing=None):
"""By default, all data (point, cell, scalars, vectors, tensors) are
plotted in a grid layout, except data named 'node_groups', 'mat_id' which
are usually not interesting.
Parameters:
show : bool
Call mlab.show().
is_3d : bool
If True, use scalar cut planes instead of surface for certain
datasets. Also sets 3D view mode.
view : tuple
Azimuth, elevation angles as in mlab.view().
roll : float
Roll angle tuple as in mlab.roll().
layout : str
Grid layout for placing the datasets. Possible values are:
'row', 'col', 'rowcol', 'colrow'.
scalar_mode : str
Mode for plotting scalars and tensor magnitudes, one of
'cut_plane', 'iso_surface', 'both'.
vector_mode : str
Mode for plotting vectors, one of 'arrows', 'norm', 'arrows_norm',
'warp_norm'.
rel_scaling : float
Relative scaling of glyphs for vector datasets.
clamping : bool
Clamping for vector datasets.
ranges : dict
List of data ranges in the form {name : (min, max), ...}.
is_scalar_bar : bool
If True, show a scalar bar for each data.
is_wireframe : bool
If True, show a wireframe of mesh surface bar for each data.
rel_text_width : float
Relative text width.
fig_filename : str
File name for saving the resulting scene figure, if
self.auto_screenshot is True.
resolution : tuple
Scene and figure resolution. If None, it is set
automatically according to the layout.
filter_names : list of strings
Omit the listed datasets. If None, it is initialized to
['node_groups', 'mat_id']. Pass [] if you need no filtering.
only_names : list of strings
Draw only the listed datasets. If None, it is initialized all names
besides those in filter_names.
group_names : list of tuples
List of data names in the form [(name1, ..., nameN), (...)]. Plots
of data named in each group are superimposed. Repetitions of names
are possible.
step : int
The time step to display.
anti_aliasing : int
Value of anti-aliasing.
"""
if filter_names is None:
filter_names = ['node_groups', 'mat_id']
if rel_text_width is None:
rel_text_width = 0.02
if isinstance(scalar_mode, str):
if scalar_mode == 'both':
scalar_mode = ('cut_plane', 'iso_surface')
elif scalar_mode in ('cut_plane', 'iso_surface'):
scalar_mode = (scalar_mode,)
else:
raise ValueError('bad value of scalar_mode parameter! (%s)'
% scalar_mode)
else:
for sm in scalar_mode:
if not sm in ('cut_plane', 'iso_surface'):
raise ValueError('bad value of scalar_mode parameter! (%s)'
% sm)
if isinstance(vector_mode, str):
if vector_mode == 'arrows_norm':
vector_mode = ('arrows', 'norm')
elif vector_mode == 'warp_norm':
vector_mode = ('warp', 'norm')
elif vector_mode in ('arrows', 'norm'):
vector_mode = (vector_mode,)
else:
raise ValueError('bad value of vector_mode parameter! (%s)'
% vector_mode)
else:
for vm in vector_mode:
if not vm in ('arrows', 'norm', 'warp'):
raise ValueError('bad value of vector_mode parameter! (%s)'
% sm)
mlab.options.offscreen = self.offscreen
self.size_hint = self.get_size_hint(layout, resolution=resolution)
is_new_scene = False
if scene is not None:
if scene is not self.scene:
is_new_scene = True
self.scene = scene
gui = None
else:
if (self.scene is not None) and (not self.scene.running):
self.scene = None
if self.scene is None:
if self.offscreen:
gui = None
scene = mlab.figure(bgcolor=(1,1,1), fgcolor=(0, 0, 0),
size=self.size_hint)
else:
gui = ViewerGUI(viewer=self)
scene = gui.scene.mayavi_scene
if scene is not self.scene:
is_new_scene = True
self.scene = scene
else:
gui = self.gui
scene = self.scene
self.engine = mlab.get_engine()
self.engine.current_scene = self.scene
self.gui = gui
scene.scene.disable_render = True
self.file_source = create_file_source(self.filename, watch=self.watch,
offscreen=self.offscreen)
has_several_steps = (nm.diff(self.file_source.get_step_range()) > 0)[0]
if gui is not None:
gui.has_several_steps = has_several_steps
if has_several_steps:
self.set_step = set_step = SetStep()
set_step._viewer = self
set_step._source = self.file_source
set_step.step = step
self.file_source.setup_notification(set_step, 'file_changed')
if gui is not None:
gui.set_step = set_step
self.build_mlab_pipeline(is_3d=is_3d,
layout=layout,
scalar_mode=scalar_mode,
vector_mode=vector_mode,
rel_scaling=rel_scaling,
clamping=clamping,
ranges=ranges,
is_scalar_bar=is_scalar_bar,
is_wireframe=is_wireframe,
rel_text_width=rel_text_width,
filter_names=filter_names,
group_names=group_names,
only_names=only_names)
scene.scene.reset_zoom()
self.options.update(get_arguments(omit = ['self', 'file_source']))
if view is None:
if is_3d or self.is_3d_data:
self.view = (45, 45)
else:
self.view = (0, 0)
else:
self.view = view
self.roll = roll
scene.scene.disable_render = False
if anti_aliasing is not None:
scene.scene.anti_aliasing_frames = anti_aliasing
if gui is None:
self.reset_view()
if is_scalar_bar:
self.show_scalar_bars(self.scalar_bars)
if self.animate:
self.save_animation(fig_filename)
else:
self.save_image(fig_filename)
else:
traits_view = View(
Item('scene', editor=SceneEditor(scene_class=MayaviScene),
show_label=False,
width=self.size_hint[0], height=self.size_hint[1],
style='custom',
),
Group(Item('set_step', defined_when='set_step is not None',
show_label=False, style='custom'),
),
HGroup(spring,
Item('button_view', show_label=False),
Item('button_make_animation', show_label=False,
enabled_when='has_several_steps == True'),),
resizable=True,
buttons=['OK'],
)
if is_new_scene:
if show:
gui.configure_traits(view=traits_view)
else:
gui.edit_traits(view=traits_view)
if self.auto_screenshot:
self.save_image(fig_filename)
return gui