#!/usr/bin/env python import vtk from vtk.test import Testing from vtk.util.misc import vtkGetDataRoot VTK_DATA_ROOT = vtkGetDataRoot() # create tensor ellipsoids # Create the RenderWindow, Renderer and interactive renderer # ren1 = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.SetMultiSamples(0) renWin.AddRenderer(ren1) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) # # Create tensor ellipsoids # # generate tensors ptLoad = vtk.vtkPointLoad() ptLoad.SetLoadValue(100.0) ptLoad.SetSampleDimensions(6,6,6) ptLoad.ComputeEffectiveStressOn() ptLoad.SetModelBounds(-10,10,-10,10,-10,10) # extract plane of data plane = vtk.vtkImageDataGeometryFilter() plane.SetInputConnection(ptLoad.GetOutputPort()) plane.SetExtent(2,2,0,99,0,99) # Generate ellipsoids sphere = vtk.vtkSphereSource() sphere.SetThetaResolution(8)
def test(self): dataRoot = vtkGetDataRoot() reader = vtk.vtkXMLImageDataReader() reader.SetFileName("" + str(dataRoot) + "/Data/vase_4comp.vti") volume = vtk.vtkVolume() #mapper = vtk.vtkFixedPointVolumeRayCastMapper() mapper = vtk.vtkGPUVolumeRayCastMapper() mapper.SetSampleDistance(0.1) mapper.SetAutoAdjustSampleDistances(0) ren = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() iRen = vtk.vtkRenderWindowInteractor() # Set connections mapper.SetInputConnection(reader.GetOutputPort()); volume.SetMapper(mapper) ren.AddViewProp(volume) renWin.AddRenderer(ren) iRen.SetRenderWindow(renWin) # Define opacity transfer function and color functions opacityFunc1 = vtk.vtkPiecewiseFunction() opacityFunc1.AddPoint(0.0, 0.0); opacityFunc1.AddPoint(60.0, 0.1); opacityFunc1.AddPoint(255.0, 0.0); opacityFunc2 = vtk.vtkPiecewiseFunction() opacityFunc2.AddPoint(0.0, 0.0); opacityFunc2.AddPoint(60.0, 0.0); opacityFunc2.AddPoint(120.0, 0.1); opacityFunc1.AddPoint(255.0, 0.0); opacityFunc3 = vtk.vtkPiecewiseFunction() opacityFunc3.AddPoint(0.0, 0.0); opacityFunc3.AddPoint(120.0, 0.0); opacityFunc3.AddPoint(180.0, 0.1); opacityFunc3.AddPoint(255.0, 0.0); opacityFunc4 = vtk.vtkPiecewiseFunction() opacityFunc4.AddPoint(0.0, 0.0); opacityFunc4.AddPoint(180.0, 0.0); opacityFunc4.AddPoint(255.0, 0.1); # Color transfer functions color1 = vtk.vtkColorTransferFunction() color1.AddRGBPoint(0.0, 1.0, 0.0, 0.0); color1.AddRGBPoint(60.0, 1.0, 0.0, 0.0); color2 = vtk.vtkColorTransferFunction() color2.AddRGBPoint(60.0, 0.0, 0.0, 1.0); color2.AddRGBPoint(120.0, 0.0, 0.0, 1.0); color3 = vtk.vtkColorTransferFunction() color3.AddRGBPoint(120.0, 0.0, 1.0, 0.0); color3.AddRGBPoint(180.0, 0.0, 1.0, 0.0); color4 = vtk.vtkColorTransferFunction() color4.AddRGBPoint(180.0, 0.0, 0.0, 0.0); color4.AddRGBPoint(239.0, 0.0, 0.0, 0.0); # Now set the opacity and the color volumeProperty = volume.GetProperty() volumeProperty.SetIndependentComponents(1) volumeProperty.SetScalarOpacity(0, opacityFunc1) volumeProperty.SetScalarOpacity(1, opacityFunc2) volumeProperty.SetScalarOpacity(2, opacityFunc3) volumeProperty.SetScalarOpacity(3, opacityFunc4) volumeProperty.SetColor(0, color1) volumeProperty.SetColor(1, color2) volumeProperty.SetColor(2, color3) volumeProperty.SetColor(3, color4) iRen.Initialize(); ren.SetBackground(0.1,0.4,0.2) ren.ResetCamera() renWin.Render() img_file = "TestGPURayCastIndependentComponent.png" vtk.test.Testing.compareImage( iRen.GetRenderWindow(), vtk.test.Testing.getAbsImagePath(img_file), threshold=10) vtk.test.Testing.interact()
from vtk.util.misc import vtkGetDataRoot # create planes # Create the RenderWindow, Renderer # ren = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.AddRenderer( ren ) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) # create pipeline # pl3d = vtk.vtkMultiBlockPLOT3DReader() pl3d.SetXYZFileName( vtkGetDataRoot() + '/Data/combxyz.bin' ) pl3d.SetQFileName( vtkGetDataRoot() + '/Data/combq.bin' ) pl3d.SetScalarFunctionNumber( 100 ) pl3d.SetVectorFunctionNumber( 202 ) pl3d.Update() pl3d_output = pl3d.GetOutput().GetBlock(0) outline = vtk.vtkStructuredGridOutlineFilter() outline.SetInputData(pl3d_output) outlineMapper = vtk.vtkPolyDataMapper() outlineMapper.SetInputConnection(outline.GetOutputPort()) outlineActor = vtk.vtkActor() outlineActor.SetMapper(outlineMapper)
#!/usr/bin/env python import sys for i in range(0, len(sys.argv)): if sys.argv[i] == '-A' and i < len(sys.argv)-1: sys.path = sys.path + [sys.argv[i+1]] import vtk from vtk.util.misc import vtkGetDataRoot filename = vtkGetDataRoot() + '/Data/tetraMesh.vtk' reader = vtk.vtkUnstructuredGridReader() reader.SetFileName(filename) a = vtk.vtkMeshQuality() a.SetInputConnection(reader.GetOutputPort()) a.VolumeOn() a.RatioOn() a.Update() mesh = a.GetOutput().GetFieldData().GetScalars() for i in range(mesh.GetNumberOfTuples()): print mesh.GetTuple2(i) sys.exit(0)
See Copyright.txt or http://www.kitware.com/Copyright.htm for details. This software is distributed WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the above copyright notice for more information. ========================================================================= ''' import math import vtk from vtk.util.misc import vtkGetDataRoot fname = vtkGetDataRoot() + '/Data/uGridEx.vtk' def DumpQualityStats(iq, arrayname): an = iq.GetOutput().GetFieldData().GetArray(arrayname) cardinality = an.GetComponent(0, 4) range = list() range.append(an.GetComponent(0, 0)) range.append(an.GetComponent(0, 2)) average = an.GetComponent(0, 1) stdDev = math.sqrt(math.fabs(an.GetComponent(0, 3))) outStr = '%s%g%s%g%s%g\n%s%g%s%g' % ( ' cardinality: ', cardinality, ' , range: ', range[0], ' - ', range[1], ' average: ', average, ' , standard deviation: ', stdDev) return outStr
diffuseColor = [0.4, 0.7, 0.6] program.SetUniform3f("diffuseColorUniform", diffuseColor) renWin = vtk.vtkRenderWindow() renWin.SetSize(400, 400) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) ren = vtk.vtkRenderer() ren.SetBackground(0.0, 0.0, 0.0) ren.GradientBackgroundOn() renWin.AddRenderer(ren) actor = vtk.vtkActor() ren.AddActor(actor) reader = vtk.vtkPLYReader() reader.SetFileName("" + str(vtkGetDataRoot()) + "/Data/dragon.ply") norms = vtk.vtkTriangleMeshPointNormals() norms.SetInputConnection(reader.GetOutputPort()) mapper = vtk.vtkOpenGLPolyDataMapper() mapper.SetInputConnection(norms.GetOutputPort()) actor.SetMapper(mapper) actor.GetProperty().SetAmbientColor(0.2, 0.2, 1.0) actor.GetProperty().SetDiffuseColor(1.0, 0.65, 0.7) actor.GetProperty().SetSpecularColor(1.0, 1.0, 1.0) actor.GetProperty().SetSpecular(0.5) actor.GetProperty().SetDiffuse(0.7) actor.GetProperty().SetAmbient(0.5) actor.GetProperty().SetSpecularPower(20.0) actor.GetProperty().SetOpacity(1.0) sp = actor.GetShaderProperty()
import vtk from vtk.util.misc import vtkGetDataRoot ren1 = vtk.vtkRenderer() ren2 = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.AddRenderer(ren1) renWin.AddRenderer(ren2) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) xmlReader = vtk.vtkXMLUnstructuredGridReader() xmlReader.SetFileName( vtkGetDataRoot() + '/Data/cuttertest.vtu' ) plane = vtk.vtkPlane() plane.SetOrigin( 50,0,405 ) plane.SetNormal( 0,0,1 ) # pipeline for cutter producing triangles triCutter = vtk.vtkCutter() triCutter.SetInputConnection( xmlReader.GetOutputPort() ) triCutter.SetCutFunction( plane ) triMapper = vtk.vtkPolyDataMapper() triMapper.SetInputConnection( triCutter.GetOutputPort() ) triMapper.ScalarVisibilityOff() triActor = vtk.vtkActor() triActor.SetMapper( triMapper ) triActor.GetProperty().SetColor( 1,0,0 ) triActor.GetProperty().EdgeVisibilityOn()
def getVTKDataRoot(): VTK_DATA_ROOT = os.environ( 'VTK_DATA_ROOT', vtkGetDataRoot() ) return VTK_DATA_ROOT
#!/usr/bin/env python import os import vtk from vtk.util.misc import vtkGetDataRoot r = vtk.vtkPolyDataReader() r.AddFileName(vtkGetDataRoot() + "/Data/track1.binary.vtk") r.AddFileName(vtkGetDataRoot() + "/Data/track2.binary.vtk") r.AddFileName(vtkGetDataRoot() + "/Data/track3.binary.vtk") r.UpdateInformation() assert(r.GetOutputInformation(0).Length(vtk.vtkStreamingDemandDrivenPipeline.TIME_STEPS()) == 3) r.UpdateTimeStep(1) rng = r.GetOutput().GetPointData().GetScalars().GetRange(0) assert(abs(rng[0] - 0.4) < 0.0001) r.UpdateTimeStep(2) rng = r.GetOutput().GetPointData().GetScalars().GetRange(0) assert(abs(rng[1] + 5.8 ) < 0.0001)
#!/usr/bin/env python from vtk import * from vtk.util.misc import vtkGetDataRoot VTK_DATA_ROOT = vtkGetDataRoot() rdr = vtk.vtkGenericEnSightReader() rdr.SetCaseFileName(str(VTK_DATA_ROOT) + "/Data/EnSight/pitzDaily.case") rdr.Update() srf = vtkDataSetSurfaceFilter() srf.SetInputConnection(rdr.GetOutputPort()) srf.Update() aa = vtkAssignAttribute() aa.SetInputConnection(srf.GetOutputPort()) aa.Assign('UGrad', vtkDataSetAttributes.SCALARS, vtkAssignAttribute.CELL_DATA) aa.Update() renWin = vtkRenderWindow() ri = vtkRenderWindowInteractor() rr = vtkRenderer() rr.SetBackground(1, 1, 1) ac = vtkActor() dm = vtkCompositePolyDataMapper() ac.SetMapper(dm) dm.SetInputConnection(aa.GetOutputPort()) dm.SetScalarVisibility(True) dm.SetScalarRange([0, 100]) dm.UseLookupTableScalarRangeOff() rr.AddActor(ac)
def test(self): dataRoot = vtkGetDataRoot() reader = vtk.vtkXMLImageDataReader() reader.SetFileName("" + str(dataRoot) + "/Data/vase_4comp.vti") volume = vtk.vtkVolume() #mapper = vtk.vtkFixedPointVolumeRayCastMapper() mapper = vtk.vtkGPUVolumeRayCastMapper() mapper.SetBlendModeToMaximumIntensity() mapper.SetSampleDistance(0.1) mapper.SetAutoAdjustSampleDistances(0) ren = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() iRen = vtk.vtkRenderWindowInteractor() # Set connections mapper.SetInputConnection(reader.GetOutputPort()) volume.SetMapper(mapper) ren.AddViewProp(volume) renWin.AddRenderer(ren) iRen.SetRenderWindow(renWin) # Define opacity transfer function and color functions opacityFunc1 = vtk.vtkPiecewiseFunction() opacityFunc1.AddPoint(0.0, 0.0) opacityFunc1.AddPoint(60.0, 0.1) opacityFunc1.AddPoint(255.0, 0.0) opacityFunc2 = vtk.vtkPiecewiseFunction() opacityFunc2.AddPoint(0.0, 0.0) opacityFunc2.AddPoint(60.0, 0.0) opacityFunc2.AddPoint(120.0, 0.1) opacityFunc1.AddPoint(255.0, 0.0) opacityFunc3 = vtk.vtkPiecewiseFunction() opacityFunc3.AddPoint(0.0, 0.0) opacityFunc3.AddPoint(120.0, 0.0) opacityFunc3.AddPoint(180.0, 0.1) opacityFunc3.AddPoint(255.0, 0.0) opacityFunc4 = vtk.vtkPiecewiseFunction() opacityFunc4.AddPoint(0.0, 0.0) opacityFunc4.AddPoint(180.0, 0.0) opacityFunc4.AddPoint(255.0, 0.1) # Color transfer functions color1 = vtk.vtkColorTransferFunction() color1.AddRGBPoint(0.0, 1.0, 0.0, 0.0) color1.AddRGBPoint(60.0, 1.0, 0.0, 0.0) color2 = vtk.vtkColorTransferFunction() color2.AddRGBPoint(60.0, 0.0, 0.0, 1.0) color2.AddRGBPoint(120.0, 0.0, 0.0, 1.0) color3 = vtk.vtkColorTransferFunction() color3.AddRGBPoint(120.0, 0.0, 1.0, 0.0) color3.AddRGBPoint(180.0, 0.0, 1.0, 0.0) color4 = vtk.vtkColorTransferFunction() color4.AddRGBPoint(180.0, 0.0, 0.0, 0.0) color4.AddRGBPoint(239.0, 0.0, 0.0, 0.0) # Now set the opacity and the color volumeProperty = volume.GetProperty() volumeProperty.SetIndependentComponents(1) volumeProperty.SetScalarOpacity(0, opacityFunc1) volumeProperty.SetScalarOpacity(1, opacityFunc2) volumeProperty.SetScalarOpacity(2, opacityFunc3) volumeProperty.SetScalarOpacity(3, opacityFunc4) volumeProperty.SetColor(0, color1) volumeProperty.SetColor(1, color2) volumeProperty.SetColor(2, color3) volumeProperty.SetColor(3, color4) iRen.Initialize() ren.SetBackground(0.1, 0.4, 0.2) ren.ResetCamera() renWin.Render() img_file = "TestGPURayCastIndependentComponentMIP.png" vtk.test.Testing.compareImage( iRen.GetRenderWindow(), vtk.test.Testing.getAbsImagePath(img_file), threshold=10) vtk.test.Testing.interact()