def make_data(): """Make some test numpy data and create a TVTK data object from it that we will visualize. """ x = numpy.arange(-7., 7.05, 0.1) y = numpy.arange(-5., 5.05, 0.05) s = mlab.SurfRegular(x, y, f) return s.data
def make_data(): """Make some test numpy data and create a TVTK data object from it that we will visualize. """ def f(x, y): """Some test function. """ return numpy.sin(x * y) / (x * y) x = numpy.arange(-7., 7.05, 0.1) y = numpy.arange(-5., 5.05, 0.05) s = mlab.SurfRegular(x, y, f) return s.data
# it in !MayaVi2. A more detailed version of this script is given in the # examples pages [:Cookbook/MayaVi/Examples]. # # <codecell> import numpy def f(x, y): return numpy.sin(x * y) / (x * y) x = numpy.arange(-7., 7.05, 0.1) y = numpy.arange(-5., 5.05, 0.05) from enthought.tvtk.tools import mlab s = mlab.SurfRegular(x, y, f) from enthought.mayavi.sources.vtk_data_source import VTKDataSource d = VTKDataSource() d.data = s.data mayavi.add_source(d) from enthought.mayavi.filters.warp_scalar import WarpScalar w = WarpScalar() mayavi.add_filter(w) from enthought.mayavi.modules.outline import Outline from enthought.mayavi.modules.surface import Surface o = Outline() s = Surface() mayavi.add_module(o) mayavi.add_module(s) # <markdowncell>
def add_surface_data_immediate(self, xrange, yrange, f): s1 = mlab.SurfRegular(xrange, yrange, scipy.vectorize(f), scale=self.scale) self.add_source_data(s1.data)
def run(self): s1 = mlab.SurfRegular(self.xrange, self.yrange, scipy.vectorize(self.f), scale=self.p.scale) GUI.invoke_later(self.p.add_source_data, s1.data)
def add_data(tvtk_data): """Add a TVTK data object `tvtk_data` to the mayavi pipleine. """ d = VTKDataSource() d.data = tvtk_data mayavi.add_source(d) return d def surf_regular(source): """Now visualize the data as done in mlab. """ w = WarpScalar() source.add_child(w) s = Surface() w.add_child(s) # 3D visualization of f: if __name__ == "__main__": mayavi.new_scene() for r in riskfree: riskFreeQuote.setValue(r) s1 = mlab.SurfRegular(spot, vol, scipy.vectorize(f), scale=[1, 100, 1]) s1.lut.alpha_range = (0.2, 0.2) d = add_data(s1.data) surf_regular(d)