def setup(self):
m = mlab
fname = self.vis3d.file_list[0]
var = self.vis3d.var
self.d = VTKXMLFileReader()
self.d.initialize(fname)
self.src = m.pipeline.add_dataset(self.d)
self.warp_vector = m.pipeline.warp_vector(self.src)
#self.surf = m.pipeline.surface(self.warp_vector)
engine = m.get_engine()
self.etc = ExtractTensorComponents()
engine.add_filter(self.etc, self.warp_vector)
surface2 = Surface()
self.surface = surface2
engine.add_filter(surface2, self.etc)
lut = self.etc.children[0]
lut.scalar_lut_manager.set(show_scalar_bar=True,
show_legend=True,
data_name=var)
lut.scalar_lut_manager.scalar_bar.orientation = 'horizontal'
lut.scalar_lut_manager.scalar_bar_representation.trait_set(
maximum_size=np.array([100000, 100000]),
minimum_size=np.array([1, 1]),
position=np.array([0.3, 0.05]),
position2=np.array([0.65, 0.1]),
)
lut.scalar_lut_manager.label_text_property.trait_set(
font_family='times', italic=False, bold=False)
lut.scalar_lut_manager.title_text_property.trait_set(
font_family='times', italic=False, bold=False)
self.etc.filter.scalar_mode = 'component'
def rebuild_pipeline(self, pd):
src = VTKFileReader(base_file_name='%s_0.vtk' % self.position)
self.engine.add_source(src)
if self.warp:
self.engine.add_filter(WarpVector())
if self.name in src._point_tensors_list:
src.point_tensors_name = self.name
self.engine.add_filter(ExtractTensorComponents())
elif self.name in src._point_vectors_list:
src.point_vectors_name = self.name
self.engine.add_filter(ExtractVectorComponents())
elif self.name in src._point_scalars_list:
src.point_scalars_name = self.name
s = Surface()
s.actor.property.point_size = 5.
self.engine.add_module(s)
src.scene.z_plus_view()
def rebuild_pipeline(self, pd):
src = VTKDataSource(name=self.name, data=pd)
self.engine.add_source(src)
if self.warp:
self.engine.add_filter(WarpVector())
if self.name in src._point_tensors_list:
src.point_tensors_name = self.name
self.engine.add_filter(ExtractTensorComponents())
elif self.name in src._point_vectors_list:
src.point_vectors_name = self.name
self.engine.add_filter(ExtractVectorComponents())
elif self.name in src._point_scalars_list:
src.point_scalars_name = self.name
s = Surface()
s.actor.property.point_size = 5.
self.engine.add_module(s)
src.scene.z_plus_view()
def view():
from mayavi.sources.vtk_data_source import VTKDataSource
from mayavi.modules.outline import Outline
from mayavi.modules.surface import Surface
from mayavi.modules.vectors import Vectors
from mayavi.filters.api import WarpVector, ExtractTensorComponents
mayavi.new_scene()
# The single type one
src = VTKDataSource(data=ug)
mayavi.add_source(src)
warp_vector = WarpVector()
mayavi.add_filter(warp_vector, src)
surface = Surface()
mayavi.add_filter(surface, warp_vector)
etc = ExtractTensorComponents()
mayavi.add_filter(etc, warp_vector)
surface2 = Surface()
mayavi.add_filter(surface2, etc)
etc.filter.scalar_mode = 'component'
def setup(self):
m = mlab
fname = self.vis3d.file_list[0]
self.d = VTKXMLFileReader()
self.d.initialize(fname)
self.src = m.pipeline.add_dataset(self.d)
self.warp_vector = m.pipeline.warp_vector(self.src)
self.surf = m.pipeline.surface(self.warp_vector)
engine = m.get_engine()
etc = ExtractTensorComponents()
engine.add_filter(etc, self.warp_vector)
surface2 = Surface()
engine.add_filter(surface2, etc)
etc.filter.scalar_mode = 'component'
lut = etc.children[0]
lut.scalar_lut_manager.set(show_scalar_bar=True,
show_legend=True,
data_name='strain field')
lut.scalar_lut_manager.scalar_bar.set(height=0.1,
width=0.7,
position=np.array([0.1, 0.1]))
def mlab_view(dataset):
fig = mlab.figure(bgcolor=(1, 1, 1),
fgcolor=(0, 0, 0),
figure=dataset.class_name[3:])
engine = mlab.get_engine()
scene = engine.scenes[0]
scene.scene.z_plus_view()
src = mlab.pipeline.add_dataset(dataset)
warp_vector = mlab.pipeline.warp_vector(src)
surf = mlab.pipeline.surface(warp_vector)
etc = ExtractTensorComponents()
engine.add_filter(etc, warp_vector)
surface2 = Surface()
engine.add_filter(surface2, etc)
etc.filter.scalar_mode = 'component'
lut = etc.children[0]
lut.scalar_lut_manager.show_scalar_bar = True
lut.scalar_lut_manager.show_legend = True
lut.scalar_lut_manager.scalar_bar.height = 0.8
lut.scalar_lut_manager.scalar_bar.width = 0.17
lut.scalar_lut_manager.scalar_bar.position = np.array([0.82, 0.1])