def setUp(self):
"""Initial setting up of test fixture, automatically called by TestCase before any other test method is invoked"""
e = NullEngine()
# Uncomment to see visualization for debugging etc.
#e = Engine()
e.start()
e.new_scene()
self.e = e
r = VTKXMLFileReader()
r.initialize(get_example_data('pyramid_ug.vtu'))
e.add_source(r)
r.point_scalars_name = 'temperature'
o = Outline()
e.add_module(o)
c = Contour()
e.add_filter(c)
n = PolyDataNormals()
e.add_filter(n)
aa = SetActiveAttribute()
e.add_filter(aa)
aa.point_scalars_name = 'pressure'
s = Surface()
e.add_module(s)
self.scene = e.current_scene
return
def main():
mayavi.new_scene()
# Read the example data: fire_ug.vtu.
r = VTKXMLFileReader()
filename = join(mayavi2.get_data_dir(dirname(abspath(__file__))),
'fire_ug.vtu')
r.initialize(filename)
mayavi.add_source(r)
# Set the active point scalars to 'u'.
r.point_scalars_name = 'u'
# Simple outline for the data.
o = Outline()
mayavi.add_module(o)
# Branch the pipeline with a contour -- the outline above is
# directly attached to the source whereas the contour below is a
# filter and will branch the flow of data. An isosurface in the
# 'u' data attribute is generated and normals generated for it.
c = Contour()
mayavi.add_filter(c)
n = PolyDataNormals()
mayavi.add_filter(n)
# Now we want to show the temperature 't' on the surface of the 'u'
# iso-contour. This is easily done by using the SetActiveAttribute
# filter below.
aa = SetActiveAttribute()
mayavi.add_filter(aa)
aa.point_scalars_name = 't'
# Now view the iso-contours of 't' with a Surface filter.
s = Surface(enable_contours=True)
mayavi.add_module(s)
def setUp(self):
"""Initial setting up of test fixture, automatically called by TestCase before any other test method is invoked"""
e = NullEngine()
# Uncomment to see visualization for debugging etc.
#e = Engine()
e.start()
e.new_scene()
self.e = e
sgrid = datasets.generateStructuredGrid()
src = VTKDataSource(data=sgrid)
e.add_source(src)
c = Contour()
# `name` is used for the notebook tabs.
n = PolyDataNormals(name='Normals')
o = Optional(filter=n, label_text='Compute normals')
coll = Collection(filters=[c, o], name='IsoSurface')
e.add_filter(coll)
s = Surface()
e.add_module(s)
self.coll = coll
self.scene = e.current_scene
return
for file0 in os.listdir('.'):
if fnmatch.fnmatch(file0, string):
filename = file0
print filename
savename = filename
scene = engine.new_scene()
scene.scene.off_screen_rendering = True
vtk_file_reader = engine.open(filename, scene)
scene.scene.background = (1.0, 1.0, 1.0)
from enthought.mayavi.filters.contour import Contour
contour = Contour()
engine.add_filter(contour, obj=None)
from enthought.mayavi.filters.set_active_attribute import SetActiveAttribute
set_active_attribute = SetActiveAttribute()
engine.add_filter(set_active_attribute, obj=None)
from enthought.mayavi.modules.surface import Surface
surface = Surface()
engine.add_module(surface, obj=None)
##Orientation axes
orientation_axes = OrientationAxes()
engine.add_module(orientation_axes, obj=None)
orientation_axes.text_property.shadow_offset = array([1, -1])
orientation_axes.text_property.font_family = 'times'
orientation_axes.text_property.shadow_offset = array([1, -1])
orientation_axes.text_property.font_size = 15
def do(self):
"""Test for the SetActiveAttribute filter.
"""
from enthought.mayavi.sources.api import VTKXMLFileReader
from enthought.mayavi.filters.contour import Contour
from enthought.mayavi.filters.api import PolyDataNormals
from enthought.mayavi.filters.set_active_attribute import SetActiveAttribute
from enthought.mayavi.modules.api import Surface, Outline
mayavi = script = self.script
scene = self.new_scene()
r = VTKXMLFileReader()
r.initialize(get_example_data('fire_ug.vtu'))
mayavi.add_source(r)
r.point_scalars_name = 'u'
o = Outline()
mayavi.add_module(o)
c = Contour()
mayavi.add_filter(c)
n = PolyDataNormals()
mayavi.add_filter(n)
aa = SetActiveAttribute()
mayavi.add_filter(aa)
aa.point_scalars_name = 't'
s = Surface()
mayavi.add_module(s)
scene.scene.isometric_view()
# Check if things are OK.
self.check()
############################################################
# Test if saving a visualization and restoring it works.
# Save visualization.
f = StringIO()
f.name = abspath('test.mv2') # We simulate a file.
mayavi.save_visualization(f)
f.seek(0) # So we can read this saved data.
# Remove existing scene.
engine = mayavi.engine
engine.close_scene(s)
# Load visualization
mayavi.load_visualization(f)
s = engine.current_scene
# Now do the check.
s.scene.isometric_view()
self.check()
############################################################
# Test if the Mayavi2 visualization can be deep-copied.
# Pop the source object.
source = s.children.pop()
# Add it back to see if that works without error.
s.children.append(source)
# Now do the check.
self.check()
# Now deepcopy the source and replace the existing one with
# the copy. This basically simulates cutting/copying the
# object from the UI via the right-click menu on the tree
# view, and pasting the copy back.
source1 = copy.deepcopy(source)
s.children[0] = source1
# Now do the check.
s.scene.isometric_view()
self.check()
def do(self):
############################################################
# Imports.
script = self.script
from enthought.mayavi.sources.vtk_file_reader import VTKFileReader
from enthought.mayavi.filters.contour import Contour
from enthought.mayavi.filters.optional import Optional
from enthought.mayavi.filters.collection import Collection
from enthought.mayavi.filters.api import PolyDataNormals
from enthought.mayavi.modules.api import Surface
############################################################
# Create a new scene and set up the visualization.
s = self.new_scene()
# Read a VTK (old style) data file.
r = VTKFileReader()
r.initialize(get_example_data('heart.vtk'))
script.add_source(r)
c = Contour()
# `name` is used for the notebook tabs.
n = PolyDataNormals(name='Normals')
o = Optional(filter=n, label_text='Compute normals')
coll = Collection(filters=[c, o], name='IsoSurface')
script.add_filter(coll)
s = Surface()
script.add_module(s)
########################################
# do the testing.
def check(coll):
"""Check if test status is OK given the collection."""
c, o = coll.filters
c = c.filter
n = o.filter
assert coll.outputs[0].point_data.scalars.range == (127.5, 127.5)
# Adding a contour should create the appropriate output in
# the collection.
c.contours.append(200)
assert coll.outputs[0].point_data.scalars.range == (127.5, 200.0)
# the collection's output should be that of the normals.
assert coll.outputs[0] is n.outputs[0]
# disable the optional filter and check.
o.enabled = False
assert 'disabled' in o.name
assert coll.outputs[0] is c.outputs[0]
# Set back everything to original state.
c.contours.pop()
o.enabled = True
assert coll.outputs[0].point_data.scalars.range == (127.5, 127.5)
assert coll.outputs[0] is n.outputs[0]
assert 'disabled' not in o.name
check(coll)
############################################################
# Test if saving a visualization and restoring it works.
# Save visualization.
f = StringIO()
f.name = abspath('test.mv2') # We simulate a file.
script.save_visualization(f)
f.seek(0) # So we can read this saved data.
# Remove existing scene.
engine = script.engine
engine.close_scene(s)
# Load visualization
script.load_visualization(f)
s = engine.current_scene
# Now do the check.
coll = s.children[0].children[0]
check(coll)
############################################################
# Test if the Mayavi2 visualization can be deep-copied.
# Pop the source object.
source = s.children.pop()
# Add it back to see if that works without error.
s.children.append(source)
# Now do the check.
coll = s.children[0].children[0]
check(coll)
# Now deepcopy the source and replace the existing one with
# the copy. This basically simulates cutting/copying the
# object from the UI via the right-click menu on the tree
# view, and pasting the copy back.
source1 = copy.deepcopy(source)
s.children[0] = source1
# Now do the check.
coll = s.children[0].children[0]
check(coll)