def __init__(self, new_engine=False, param=None):
'''
Input:
param: object of GraphPlotParam class
'''
if param is None:
param = GraphPlotParam()
self.TubeColor = param.TubeColor
self.TubeRadius = param.TubeRadius
self.GylphColor = param.GylphColor
self.GylphSize = param.GylphSize
self.TubeMode = {
'radius_by_scale': False,
'radius_by_color': False,
'type_by_color': False,
'branching_by_color': False,
'flow_by_color': False,
'pressure_by_color': False
}
# start engine
if new_engine:
from mayavi.core.api import Engine
e = Engine()
e.start()
# source
self.DataSource = scalar_scatter([], [], [])
self.Data = self.DataSource.data
self.Nodes = self.Data.points
self.Edges = self.Data.lines
#modules
self.Glyph = glyph(self.DataSource)
self.Glyph.glyph.glyph.clamping = True
self.Glyph.glyph.glyph.scale_factor = 2.0
self.Glyph.glyph.glyph.range = np.array([0., 1.])
self.Glyph.glyph.color_mode = 'no_coloring'
self.Glyph.glyph.scale_mode = 'data_scaling_off'
self.Glyph.actor.property.representation = 'wireframe'
self.Glyph.actor.property.color = (0.75, 0.25, 0.25)
self.Tube = tube(self.Glyph, tube_radius=self.TubeRadius)
self.TubeSurface = surface(self.Tube, color=self.TubeColor)
self.Tube.filter.vary_radius = 'vary_radius_off'
self.TubeSurface.actor.mapper.scalar_visibility = False
def __init__(self, s=None, param=None, new_engine=False):
'''
input:
param: object from StackPlotParam
'''
# if s is None:
# print('Noe input surface model')
# return
if param is None:
param = StackPlotParam()
self.SurfaceColor = param.SurfaceColor
self.SurfaceOpacity = param.SurfaceOpacity
# start engine
if new_engine:
from mayavi.core.api import Engine
e = Engine()
e.start()
# source
self.DataSource = scalar_field(
(np.random.rand(5, 5, 5) > 0.5).astype(int))
self.DataSource.origin = [0, 0, 0]
self.Data = self.DataSource.scalar_data
self.DataSource.origin = np.array([0, 0, 0])
#modules
self.Contour = contour(self.DataSource)
self.Contour.filter.contours = []
self.Contour.filter.contours = [0.5]
self.Contour.filter.auto_update_range = False
self.Surface = None
self.__UpdateSurface()
def main():
# Create the MayaVi engine and start it.
e = Engine()
# Starting the engine registers the engine with the registry and
# notifies others that the engine is ready.
e.start()
# Do this if you need to see the MayaVi tree view UI.
ev = EngineView(engine=e)
ui = ev.edit_traits()
# Create a new scene.
#scene = e.new_scene()
#renderLearningData('pattern.json')
#scene = e.new_scene()
#renderLearningData('pattern2.json')
#renderLearningData('pattern3.json')
scene = e.new_scene()
ExampleRenderer('cluster.json', scene)
scene = e.new_scene()
ExampleRenderer('nocluster.json', scene)
show()
return e, ui
def main():
# Create the MayaVi engine and start it.
e = Engine()
# Starting the engine registers the engine with the registry and
# notifies others that the engine is ready.
e.start()
# Do this if you need to see the MayaVi tree view UI.
ev = EngineView(engine=e)
ui = ev.edit_traits()
# Create a new scene.
scene = e.new_scene()
# Now create a new scene just for kicks.
scene1 = e.new_scene()
# Now setup a normal MayaVi pipeline.
src = VTKXMLFileReader()
src.initialize(join(get_data_dir(abspath(__file__)),
'fire_ug.vtu'))
e.add_source(src)
e.add_module(Outline())
e.add_module(ScalarCutPlane())
e.add_module(Streamline())
return e, ui
def main():
# Create the MayaVi engine and start it.
e = Engine()
# Starting the engine registers the engine with the registry and
# notifies others that the engine is ready.
e.start()
# Do this if you need to see the MayaVi tree view UI.
ev = EngineView(engine=e)
ui = ev.edit_traits()
# Create a new scene.
scene = e.new_scene()
# Now create a new scene just for kicks.
scene1 = e.new_scene()
# Now setup a normal MayaVi pipeline.
src = VTKXMLFileReader()
src.initialize(join(get_data_dir(abspath(__file__)),
'fire_ug.vtu'))
e.add_source(src)
e.add_module(Outline())
e.add_module(ScalarCutPlane())
e.add_module(Streamline())
return e, ui
Created on Fri Dec 21 12:59:31 2018 @author: Schmulius """ from mayavi.modules.scalar_cut_plane import ScalarCutPlane from pyface.timer.api import Timer from mayavi.core.api import Engine from mayavi.core.base import Base from mayavi.core.module import Module from mayavi.core.lut_manager import LUTManager from mayavi.core.common import handle_children_state, exception from mayavi.core.pipeline_info import PipelineInfo scp = Engine() scp.start() scene = scp.new_scene() scp = ScalarCutPlane() # set scp as ScalarCutPlane() module # add module to the scene scp.implicit_plane.normal = (1, 0, 0) # set normal to Ox axis # set origin to (i=10, j=25, k=25) i.e. integers for a structured grid scp.implicit_plane.origin = (10, 25, 25) # set origin to (x=1.0, y=2.5, z=2.5) i.e. reals for unstructured grids # scp.implicit_plane.origin = (1.0, 2.5, 2.5) scp.implicit_plane.widget.enabled = True scp.actor.property.diffuse = 0.0 # set some color properties scp.actor.property.ambient = 1.0 # scp.actor.property.opacity = 1.0 # #scp.module_manager.scalar_lut_manager.data_range = [0, 1]
# -*- coding: utf-8 -*- """ Spyder Editor """ import os from mayavi.core.api import Engine from mayavi.sources.vtk_file_reader import VTKFileReader from mayavi.modules.surface import Surface vtkFile_l = '/Users/richad/bin/lh.vtk' vtkFile_r = '/Users/richad/bin/rh.vtk' # Create the MayaVi engine and start it. engine = Engine() engine.start() #scene = engine.new_scene() # Read in VTK file and add as source surface1 = Surface() reader1 = VTKFileReader() reader1.initialize(vtkFile_l) engine.add_source(reader1) engine.add_filter(surface1, reader1) surface2 = Surface() reader2 = VTKFileReader() reader2.initialize(vtkFile_r) engine.add_source(reader2) engine.add_filter(surface2, reader2) #import networkx as nx
import os
from mayavi.core.api import Engine
from mayavi.sources.vtk_file_reader import VTKFileReader
from mayavi.modules.surface import Surface
vtkFile = 'flow.vtk'
# Create the MayaVi engine and start it.
engine = Engine()
engine.start()
scene = engine.new_scene()
# Read in VTK file and add as source
reader = VTKFileReader()
reader.initialize(vtkFile)
engine.add_source(reader)
# Add Surface Module
surface = Surface()
engine.add_module(surface)
# Move the camera
scene.scene.camera.elevation(-70)
# Save scene to image file
scene.scene.save_png('image.png')
# Create a GUI instance and start the event loop.
# This stops the window from closing
from enthought.pyface.api import GUI
gui = GUI()
from mayavi.mlab import * from os.path import join, dirname import numpy import sys import vtk from mayavi.sources.vtk_file_reader import VTKFileReader from mayavi.sources.api import VTKXMLFileReader import os from pyface.timer.api import Timer from mayavi.core.api import Engine from mayavi.sources.vtk_file_reader import VTKFileReader from mayavi.modules.surface import Surface from mayavi.modules.scalar_cut_plane import ScalarCutPlane engine = Engine() engine.start() scene = engine.new_scene() #vtkFile = 'simple.vtk' vtkFile = '/volumes/esd01/docs/jloos/data_small/runs_elmerice_refined/Mesh60_200500_0/Bump_0001.pvtu' #reader = VTKFileReader() #reader = VTKXMLFileReader() #reader.initialize(vtkFile) #engine.add_source(reader) # Add Surface Module #surface = Surface() #engine.add_module(surface),
""" Created on Mon Jan 4 18:14:18 2016 @author: ethansolomon """ import os from mayavi.core.api import Engine from mayavi.sources.vtk_file_reader import VTKFileReader from mayavi.modules.surface import Surface vtkFile_l = 'lh.vtk' vtkFile_r = 'rh.vtk' # Create the MayaVi engine and start it. engine = Engine() engine.start() scene = engine.new_scene() # Read in VTK file and add as source reader1 = VTKFileReader() reader1.initialize(vtkFile_l) engine.add_source(reader1) # Add Surface Module surface = Surface() engine.add_module(surface) # Move the camera scene.scene.camera.elevation(90) scene.scene.camera.azimuth(-180)
# mlab surface defined with the points and element
surface = mlab.pipeline.triangular_mesh_source(points[:,0], points[:,1], points[:,2], element)
# For readability, define a variable as the _vtk_obj from the surface.
# This will be the surface data in the .stl file.
surface_vtk = surface.outputs[0]._vtk_obj
stlWriter = vtk.vtkSTLWriter()
# Set the file name
stlWriter.SetFileName('test_surface.stl')
# Set the input for the stl writer. surface.output[0]._vtk_obj is a polydata object
stlWriter.SetInput(surface_vtk)
# Write the stl file
stlWriter.Write()
# View the .stl surface that was just written-----------------------------------------------
from mayavi.core.api import Engine
engine = Engine()
# Create a new figure and add that figure to the engine
fig = mlab.figure(engine = engine)
# Open the stl file
surface_data = engine.open('test_surface.stl')
# Add the opened surface to the pipeline
opened_surface = mlab.pipeline.surface(surface_data)
# Add a module to show the opened surface
mlab.pipeline.surface(opened_surface, figure = fig)
# Show the scene
mlab.show()
from mayavi.core.module import Module
from mayavi.core.lut_manager import LUTManager
from mayavi.core.common import handle_children_state, exception
from mayavi.core.pipeline_info import PipelineInfo
from mayavi import mlab
from mayavi import tools
from numpy import array
#---------------------------------------------------------------------------------------
#__________________________________ start mayavi engine _______________________________#
#---------------------------------------------------------------------------------------
try:
engine = mayavi.engine
except NameError:
from mayavi.api import Engine
engine = Engine()
engine.start()
if len(engine.scenes) == 0:
engine.new_scene()
#---------------------------------------------------------------------------------------
#__________________________________ load data ____________________________________#
#---------------------------------------------------------------------------------------
vtkxml_file_reader = engine.open(
'/Volumes/esd01/docs/jloos/data_small/runs_elmerice_refined/Mesh20_5001000_0_0.1/Mesh/ForwardRemesh100M20L0011.pvtu'
)
# Set time-step for visualisation; depending on simulation setting time-step cann range between
# 0.1 and 0.5 = 2 (two time sizes 1a)
from mayavi.modules.surface import Surface
from mayavi import mlab
# View the .stl surface that was just written-----------------------------------------------
from mayavi.core.api import Engine
# Create an engine
engine = Engine()
# Create a new figure and add that figure to the engine
#fig = mlab.figure(engine = engine)
# Open the stl file
surface_data = engine.open('Crystal I - polyhedronp.stl')
# Add the opened surface to the pipeline
opened_surface = mlab.pipeline.surface(surface_data,representation='surface')
opened_surface.actor.property.specular = .99
opened_surface.actor.property.specular_power = 100 #15.0
opened_surface.actor.property.specular_color = (1, 1, 1)
opened_surface.actor.property.diffuse = .8
opened_surface.actor.property.ambient = 0
opened_surface.actor.property.interpolation = 'phong'
# Lighting
lm = engine.current_scene.scene.light_manager
lm.number_of_lights = 4
ABCDangle = 30
