def _make_functions(namespace):
"""Make the functions for adding modules and add them to the
namespace automatically.
"""
# Ignore these since they are already provided.
ignore = ['axes', 'text', 'orientation_axes']
for mod in registry.modules:
func_name = camel2enthought(mod.id)
class_name = mod.id
if func_name.endswith('_module'):
func_name = func_name[:-7]
class_name = class_name[:-6]
class_name = class_name + 'Factory'
# Don't create any that are already defined or ignored.
if class_name in namespace or func_name in ignore:
continue
# The class to wrap.
klass = new.classobj(class_name,
(_AutomaticModuleFactory,),
{'__doc__': mod.help,}
)
klass._metadata = mod
# The mlab helper function.
func = make_function(klass)
# Inject class/function into the namespace and __all__.
namespace[class_name] = klass
namespace[func_name] = func
__all__.append(func_name)
def _make_functions(namespace):
"""Make the functions for adding filters and add them to the
namespace automatically.
"""
for fil in registry.filters:
func_name = camel2enthought(fil.id)
class_name = fil.id
if func_name.endswith('_filter'):
func_name = func_name[:-7]
class_name = class_name[:-6]
class_name = class_name + 'Factory'
# Don't create any that are already defined.
if class_name in namespace:
continue
# The class to wrap.
klass = new.classobj(class_name, (_AutomaticFilterFactory, ), {
'__doc__': fil.help,
})
klass._metadata = fil
# The mlab helper function.
func = make_function(klass)
# Inject class/function into the namespace and __all__.
namespace[class_name] = klass
namespace[func_name] = func
__all__.append(func_name)
def _make_functions(namespace):
"""Make the functions for adding filters and add them to the
namespace automatically.
"""
for fil in registry.filters:
func_name = camel2enthought(fil.id)
class_name = fil.id
if func_name.endswith('_filter'):
func_name = func_name[:-7]
class_name = class_name[:-6]
class_name = class_name + 'Factory'
# Don't create any that are already defined.
if class_name in namespace:
continue
# The class to wrap.
klass = new.classobj(class_name,
(_AutomaticFilterFactory,),
{'__doc__': fil.help, }
)
klass._metadata = fil
# The mlab helper function.
func = make_function(klass)
# Inject class/function into the namespace and __all__.
namespace[class_name] = klass
namespace[func_name] = func
__all__.append(func_name)
def _mode_changed(self):
v = self._target
# Workaround for different version of VTK:
if hasattr(v.glyph.glyph_source, 'glyph_source'):
g = v.glyph.glyph_source
else:
g = v.glyph
if self.mode == 'point':
g.glyph_source = tvtk.PointSource(radius=0, number_of_points=1)
else:
g.glyph_source = g.glyph_list[self.mode_]
if self.mode_ == 0:
g.glyph_source.glyph_type = self.mode[2:]
vectors = make_function(VectorsFactory)
##############################################################################
class GlyphFactory(VectorsFactory):
"""Applies the Glyph mayavi module to the given VTK data
source (Mayavi source, or VTK dataset).
"""
_target = Instance(modules.Glyph, ())
scale_mode = Trait('scalar', {'none':'data_scaling_off',
'scalar':'scale_by_scalar',
'vector':'scale_by_vector'},
help="""the scaling mode for the glyphs
('vector', 'scalar', or 'none').""")
"""Applies the Tube mayavi filter to the given VTK object."""
_target = Instance(filters.Tube, ())
tube_sides = CInt(6,
adapts='filter.number_of_sides',
desc="""number of sides of the tubes used to
represent the lines.""")
tube_radius = CFloat(0.05,
adapts='filter.radius',
desc="""radius of the tubes used to represent the
lines.""")
tube = make_function(TubeFactory)
##############################################################################
class WarpScalarFactory(PipeFactory):
"""Applies the WarpScalar mayavi filter to the given VTK object."""
_target = Instance(filters.WarpScalar, ())
warp_scale = CFloat(1.0,
adapts="filter.scale_factor",
help="scale of the warp scalar")
warp_scalar = make_function(WarpScalarFactory)
##############################################################################
class TubeFactory(PipeFactory):
"""Applies the Tube mayavi filter to the given VTK object."""
_target = Instance(filters.Tube, ())
tube_sides = CInt(6, adapts='filter.number_of_sides',
desc="""number of sides of the tubes used to
represent the lines.""")
tube_radius = CFloat(0.05, adapts='filter.radius',
desc="""radius of the tubes used to represent the
lines.""")
tube = make_function(TubeFactory)
##############################################################################
class WarpScalarFactory(PipeFactory):
"""Applies the WarpScalar mayavi filter to the given VTK object."""
_target = Instance(filters.WarpScalar, ())
warp_scale = CFloat(1.0, adapts="filter.scale_factor",
help="scale of the warp scalar")
warp_scalar = make_function(WarpScalarFactory)
##############################################################################
