def _annotateScene(self): script = self.script # Domain (axes and outline) script.add_source(VTKDataSource(data=vtk_geometry.domain())) script.engine.current_object.name = "Domain" outline = Outline() script.add_module(outline) outline.actor.property.opacity = 0.2 axes = Axes() axes.axes.x_label = "X" axes.axes.y_label = "Y" axes.axes.z_label = "Z" axes.axes.label_format = "%-0.1f" script.add_module(axes) return
def _setupScene(self, showFault=True, showMaterials=True, showFaultTaper=False): """ Plot axes, fault surface, and materials. """ from enthought.mayavi.modules.outline import Outline from enthought.mayavi.modules.axes import Axes from enthought.mayavi.modules.surface import Surface # Create rendering scene script = self.script script.new_scene() script.engine.current_scene.scene.background = (1, 1, 1) script.engine.current_scene.scene.foreground = (0, 0, 0) vtk_geometry.setCamera(script.engine.current_scene.scene.camera) # Domain (axes and outline) script.add_source(VTKDataSource(data=vtk_geometry.domain())) script.engine.current_object.name = "Domain" outline = Outline() script.add_module(outline) outline.actor.property.opacity = 0.2 axes = Axes() axes.axes.x_label = "X (km)" axes.axes.y_label = "Y (km)" axes.axes.z_label = "Z (km)" axes.axes.label_format = "%-0.1f" script.add_module(axes) # Fault surface if showFault: srcs = vtk_geometry.fault(showTaper=showFaultTaper) for src in srcs: script.add_source(VTKDataSource(data=src['object'])) script.engine.current_object.name = "Fault %s" % src['name'] surf = Surface() script.add_module(surf) surf.actor.property.color = (1, 0, 0) if src['name'] == "taper": surf.actor.property.opacity = 0.1 else: surf.actor.property.opacity = 0.3 # Materials if showMaterials: srcs = vtk_geometry.materials() for src in srcs: script.add_source(VTKDataSource(data=src['object'])) script.engine.current_object.name = "Material %s" % src['name'] surf = Surface() script.add_module(surf) surf.actor.property.opacity = 0.1 if src['name'] == "elastic": surf.actor.property.color = (1, 1, 0) elif src['name'] == "viscoelastic": surf.actor.property.color = (0, 1, 1) return
def _setupScene(self, showFault=True, showMaterials=True, showFaultTaper=False): """ Plot axes, fault surface, and materials. """ from enthought.mayavi.modules.outline import Outline from enthought.mayavi.modules.axes import Axes from enthought.mayavi.modules.surface import Surface # Create rendering scene script = self.script script.new_scene() script.engine.current_scene.scene.background = (1,1,1) script.engine.current_scene.scene.foreground = (0,0,0) vtk_geometry.setCamera(script.engine.current_scene.scene.camera) # Domain (axes and outline) script.add_source(VTKDataSource(data=vtk_geometry.domain())) script.engine.current_object.name = "Domain" outline = Outline() script.add_module(outline) outline.actor.property.opacity = 0.2 axes = Axes() axes.axes.x_label = "X (km)" axes.axes.y_label = "Y (km)" axes.axes.z_label = "Z (km)" axes.axes.label_format = "%-0.1f" script.add_module(axes) # Fault surface if showFault: srcs = vtk_geometry.fault(showTaper=showFaultTaper) for src in srcs: script.add_source(VTKDataSource(data=src['object'])) script.engine.current_object.name = "Fault %s" % src['name'] surf = Surface() script.add_module(surf) surf.actor.property.color = (1,0,0) if src['name'] == "taper": surf.actor.property.opacity = 0.1 else: surf.actor.property.opacity = 0.3 # Materials if showMaterials: srcs = vtk_geometry.materials() for src in srcs: script.add_source(VTKDataSource(data=src['object'])) script.engine.current_object.name = "Material %s" % src['name'] surf = Surface() script.add_module(surf) surf.actor.property.opacity = 0.1 if src['name'] == "elastic": surf.actor.property.color = (1,1,0) elif src['name'] == "viscoelastic": surf.actor.property.color = (0,1,1) return