def treader(fname, engine):
from enthought.tvtk.api import tvtk
from enthought.mayavi.sources.vtk_data_source import VTKDataSource
from enthought.mayavi.sources.api import ArraySource
# Do your own reader stuff here, I'm just reading a VTK file with a
# different extension here.
src = ArraySource()
src.scalar_name=fname
import os.path
src.name=os.path.split(fname)[-1]
return src
def spe_reader(fname, engine):
"""Reader for .zzz files.
Parameters:
-----------
fname -- Filename to be read.
engine -- The engine the source will be associated with.
"""
from enthought.tvtk.api import tvtk
from enthought.mayavi.sources.vtk_data_source import VTKDataSource
# Do your own reader stuff here, I'm just reading a VTK file with a
# different extension here.
import SpeFile as spe
r = spe.SpeFile(fname)
from enthought.mayavi.sources.api import ArraySource
src = ArraySource()
src.scalar_data = r.GetNumArray()
src.scalar_name = fname
import os.path
src.name = os.path.split(fname)[-1]
return src
