def drawControlPoints(self):
"""
生成控制点球体
"""
# 初始化列表 保存每一个控制球体 列表为i*j*k维 因为有i*j*k个球体
self.spherelist = [[[0 for zcol in range(self.zl+1)] for col in range(self.yl+1)] for row in range(self.xl+1)]
for i, j, k in ((a, b, c) for a in range(self.xl+1) for b in range(self.yl+1) for c in range(self.zl+1)):
# 定义一个球状体widget
sphereWidget = vtk.vtkSphereWidget()
# 渲染窗口交互器实例iren是一个3D的球状体widget
sphereWidget.SetInteractor(self.iren)
# 从索引值对应到真实空间中的xyz坐标
x, y, z = self.ijk2xyz(i, j, k)
# 设置球状体在真实空间中的xyz坐标
sphereWidget.SetCenter(x, y, z)
# 设置球状体的半径大小
sphereWidget.SetRadius(self.RADISU)
# 设置球面的颜色 仍然是通过GetProperty来获取属性并进行设置
# sphereWidget.GetSphereProperty().SetColor(0, 1.0, 0)
# 设置填充球状体的表面 三种基本的属性设置方式：点方式，网格方式和面方式
sphereWidget.SetRepresentationToSurface()
# 要是没有这一行 球状体就不会显示出来了
sphereWidget.On()
# 将球状体添加到球状体的列表中
self.spherelist[i][j][k]= sphereWidget
def main():
colors = vtk.vtkNamedColors()
# colors.SetColor('bkg', [0.1, 0.2, 0.4, 1.0])
# A renderer and render window
renderer = vtk.vtkRenderer()
renderer.SetBackground(colors.GetColor3d('MidnightBlue'))
renwin = vtk.vtkRenderWindow()
renwin.AddRenderer(renderer)
renwin.SetWindowName("SphereWidget")
# An interactor
interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(renwin)
# A Sphere widget
sphereWidget = vtk.vtkSphereWidget()
sphereWidget.SetInteractor(interactor)
sphereWidget.SetRepresentationToSurface()
sphereWidget.GetSphereProperty().SetColor(colors.GetColor3d("BurlyWood"))
# Connect the event to a function
sphereWidget.AddObserver("InteractionEvent", sphereCallback)
# Start
interactor.Initialize()
renwin.Render()
sphereWidget.On()
interactor.Start()
def create_sphere(self, X, Y, Z, fn, n_sphere=0, n_render=0, n_index=0, r=0.03):
"""
Method to create the sphere that represent the interfaces points
Args:
X: X coord
Y: Y coord
Z: Z corrd
fn (int): Formation number
n_sphere (int): Number of the sphere
n_render (int): Number of the render where the sphere belongs
n_index (int): index value in the PandasDataframe of InupData.interfaces
r (float): radio of the sphere
Returns:
vtk.vtkSphereWidget
"""
s = vtk.vtkSphereWidget()
s.SetInteractor(self.interactor)
s.SetRepresentationToSurface()
s.r_f = self._e_d_avrg * r
s.PlaceWidget(X - s.r_f, X + s.r_f, Y - s.r_f, Y + s.r_f, Z - s.r_f, Z + s.r_f)
s.GetSphereProperty().SetColor(self.C_LOT[fn])
s.SetCurrentRenderer(self.ren_list[n_render])
s.n_sphere = n_sphere
s.n_render = n_render
s.index = n_index
s.AddObserver("EndInteractionEvent", self.sphereCallback) # EndInteractionEvent
s.On()
return s
def Execute(self):
if self.Surface == None:
self.PrintError('Error: no Surface.')
if self.WidgetType == "box":
self.ClipFunction = vtk.vtkPlanes()
elif self.WidgetType == "sphere":
self.ClipFunction = vtk.vtkSphere()
self.Clipper = vtk.vtkClipPolyData()
self.Clipper.SetInput(self.Surface)
self.Clipper.SetClipFunction(self.ClipFunction)
self.Clipper.GenerateClippedOutputOn()
self.Clipper.InsideOutOn()
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
mapper = vtk.vtkPolyDataMapper()
mapper.SetInput(self.Surface)
mapper.ScalarVisibilityOff()
self.Actor = vtk.vtkActor()
self.Actor.SetMapper(mapper)
self.vmtkRenderer.Renderer.AddActor(self.Actor)
if self.WidgetType == "box":
self.ClipWidget = vtk.vtkBoxWidget()
self.ClipWidget.GetFaceProperty().SetColor(0.6,0.6,0.2)
self.ClipWidget.GetFaceProperty().SetOpacity(0.25)
elif self.WidgetType == "sphere":
self.ClipWidget = vtk.vtkSphereWidget()
self.ClipWidget.GetSphereProperty().SetColor(0.6,0.6,0.2)
self.ClipWidget.GetSphereProperty().SetOpacity(0.25)
self.ClipWidget.GetSelectedSphereProperty().SetColor(0.6,0.0,0.0)
self.ClipWidget.GetSelectedSphereProperty().SetOpacity(0.75)
self.ClipWidget.SetRepresentationToSurface()
self.ClipWidget.SetPhiResolution(20)
self.ClipWidget.SetThetaResolution(20)
self.ClipWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.Display()
self.Transform = vtk.vtkTransform()
self.ClipWidget.GetTransform(self.Transform)
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
if self.CleanOutput == 1:
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInput(self.Surface)
cleaner.Update()
self.Surface = cleaner.GetOutput()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
def create_sphereWidget(center=(3.98, -30.94, 9.16), radius=5, opacity=0.8):
sphereWidget = vtk.vtkSphereWidget()
sphereWidget.SetCenter(center)
sphereWidget.SetRadius(radius)
sphereWidget.GetSphereProperty().SetOpacity(opacity)
sphereWidget.SetRepresentationToSurface()
# sphereWidget.SetKeyPressActivationValue('k') # By default, the key press activation value is 'i'.
# sphereWidget.On()
return sphereWidget
def Execute(self):
if (self.Surface == None):
self.PrintError('Error: no Surface.')
if self.WidgetType == "box":
self.ClipFunction = vtk.vtkPlanes()
elif self.WidgetType == "sphere":
self.ClipFunction = vtk.vtkSphere()
self.Clipper = vtk.vtkClipPolyData()
self.Clipper.SetInput(self.Surface)
self.Clipper.SetClipFunction(self.ClipFunction)
self.Clipper.GenerateClippedOutputOn()
self.Clipper.InsideOutOn()
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
mapper = vtk.vtkPolyDataMapper()
mapper.SetInput(self.Surface)
mapper.ScalarVisibilityOff()
self.Actor = vtk.vtkActor()
self.Actor.SetMapper(mapper)
self.vmtkRenderer.Renderer.AddActor(self.Actor)
if self.WidgetType == "box":
self.ClipWidget = vtk.vtkBoxWidget()
self.ClipWidget.GetFaceProperty().SetColor(0.6, 0.6, 0.2)
self.ClipWidget.GetFaceProperty().SetOpacity(0.25)
elif self.WidgetType == "sphere":
self.ClipWidget = vtk.vtkSphereWidget()
self.ClipWidget.GetSphereProperty().SetColor(0.6, 0.6, 0.2)
self.ClipWidget.GetSphereProperty().SetOpacity(0.25)
self.ClipWidget.GetSelectedSphereProperty().SetColor(0.6, 0.0, 0.0)
self.ClipWidget.GetSelectedSphereProperty().SetOpacity(0.75)
self.ClipWidget.SetRepresentationToSurface()
self.ClipWidget.SetPhiResolution(20)
self.ClipWidget.SetThetaResolution(20)
self.ClipWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.Display()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
if self.CleanOutput == 1:
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInput(self.Surface)
cleaner.Update()
self.Surface = cleaner.GetOutput()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
def create_sphere(self, X, Y, Z, f, n_sphere=0, n_render=0, n_index=0, r=0.03):
s = vtk.vtkSphereWidget()
s.SetInteractor(self.interactor)
s.SetRepresentationToSurface()
s.r_f = self._e_d_avrg * r
s.PlaceWidget(X-s.r_f, X+s.r_f, Y-s.r_f, Y+s.r_f, Z-s.r_f, Z+s.r_f)
s.GetSphereProperty().SetColor(self.C_LOT[f])
s.SetCurrentRenderer(self.ren_list[n_render])
s.n_sphere = n_sphere
s.n_render = n_render
s.index = n_index
s.AddObserver("InteractionEvent", self.sphereCallback)
s.On()
return s
def Execute(self):
if self.Surface == None:
self.PrintError("Error: No input surface.")
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
self.DeformedSurface = self.Surface
sphereSource = vtk.vtkSphereSource()
sphereSource.SetRadius(1)
self.SourceGlyphs = vtk.vtkGlyph3D()
self.SourceGlyphs.SetInput(self.SourceSpheres)
self.SourceGlyphs.SetSource(sphereSource.GetOutput())
self.SourceGlyphs.SetScaleModeToDataScalingOff()
self.SourceGlyphs.SetScaleFactor(self.Surface.GetLength() * 0.01)
self.SourceGlyphs.GeneratePointIdsOn()
self.SourceGlyphMapper = vtk.vtkPolyDataMapper()
self.SourceGlyphMapper.SetInput(self.SourceGlyphs.GetOutput())
self.SourceGlyphMapper.ScalarVisibilityOff()
self.SourcePointActor = vtk.vtkActor()
self.SourcePointActor.SetMapper(self.SourceGlyphMapper)
self.SourcePointActor.GetProperty().SetColor(1.0, 0.0, 0.0)
self.SourcePointActor.GetProperty().SetOpacity(self.Opacity)
self.SourcePointActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SourcePointActor)
self.TargetGlyphs = vtk.vtkGlyph3D()
self.TargetGlyphs.SetInput(self.TargetSpheres)
self.TargetGlyphs.SetSource(sphereSource.GetOutput())
self.TargetGlyphs.SetScaleModeToDataScalingOff()
self.TargetGlyphs.SetScaleFactor(self.Surface.GetLength() * 0.01)
self.TargetGlyphs.GeneratePointIdsOn()
self.TargetGlyphMapper = vtk.vtkPolyDataMapper()
self.TargetGlyphMapper.SetInput(self.TargetGlyphs.GetOutput())
self.TargetGlyphMapper.ScalarVisibilityOff()
self.TargetPointActor = vtk.vtkActor()
self.TargetPointActor.SetMapper(self.TargetGlyphMapper)
self.TargetPointActor.GetProperty().SetColor(0.0, 0.0, 1.0)
self.TargetPointActor.GetProperty().SetOpacity(self.Opacity)
self.TargetPointActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.TargetPointActor)
self.CylinderSource = vtk.vtkCylinderSource()
self.CylinderSource.SetCenter(0, 0, 0)
self.CylinderSource.SetRadius(self.Surface.GetLength() * 0.0025)
self.CylinderSource.SetHeight(0)
self.CylinderTransform = vtk.vtkTransform()
self.CylinderTransform.Identity()
self.CylinderTransformFilter = vtk.vtkTransformPolyDataFilter()
self.CylinderTransformFilter.SetTransform(self.CylinderTransform)
self.CylinderTransformFilter.SetInputConnection(self.CylinderSource.GetOutputPort())
self.DisplacementGlyphs = vtk.vtkProgrammableGlyphFilter()
self.DisplacementGlyphs.SetInput(self.SourceSpheres)
self.DisplacementGlyphs.SetSource(self.CylinderTransformFilter.GetOutput())
self.DisplacementGlyphs.SetGlyphMethod(self.PlaceCylinder)
self.DisplacementGlyphsMapper = vtk.vtkPolyDataMapper()
self.DisplacementGlyphsMapper.SetInput(self.DisplacementGlyphs.GetOutput())
self.DisplacementGlyphsMapper.ScalarVisibilityOff()
self.DisplacementGlyphsActor = vtk.vtkActor()
self.DisplacementGlyphsActor.SetMapper(self.DisplacementGlyphsMapper)
self.DisplacementGlyphsActor.GetProperty().SetColor(0.0, 1.0, 0.0)
self.DisplacementGlyphsActor.GetProperty().SetOpacity(self.Opacity)
self.DisplacementGlyphsActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.DisplacementGlyphsActor)
# self.vmtkRenderer.RenderWindowInteractor.AddObserver("KeyPressEvent", self.KeyPressed)
self.vmtkRenderer.AddKeyBinding("u", "Undo.", self.UndoCallback)
self.vmtkRenderer.AddKeyBinding("space", "Add displacement.", self.PickCallback)
self.vmtkRenderer.AddKeyBinding("s", "Select source/target.", self.SelectCallback)
self.vmtkRenderer.AddKeyBinding("n", "Skip to next source/target.", self.NextCallback)
self.vmtkRenderer.AddKeyBinding("v", "Skip to previous source/target.", self.PreviousCallback)
self.vmtkRenderer.AddKeyBinding("x", "Delete source/target.", self.DeleteCallback)
self.vmtkRenderer.AddKeyBinding("r", "Reset target to source.", self.ResetCallback)
self.vmtkRenderer.AddKeyBinding("d", "Show deformed surface.", self.ShowDeformedCallback)
self.SurfaceMapper = vtk.vtkPolyDataMapper()
self.SurfaceMapper.SetInput(self.Surface)
self.SurfaceMapper.SetScalarVisibility(0)
surfaceActor = vtk.vtkActor()
surfaceActor.SetMapper(self.SurfaceMapper)
surfaceActor.GetProperty().SetOpacity(self.Opacity)
self.vmtkRenderer.Renderer.AddActor(surfaceActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.SetRadius(self.Surface.GetLength() * 0.01)
self.SphereWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.InputInfo("Please position the mouse and press space to add source and target points, 'u' to undo\n")
self.InitializeSpheres()
self.vmtkRenderer.Render()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
def Execute(self):
if self.Surface == None:
self.PrintError("Error: No input surface.")
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInput(self.Spheres)
glyphs.SetSource(glyphSource.GetOutput())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.0)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInput(glyphs.GetOutput())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0, 0.0, 0.0)
self.SpheresActor.GetProperty().SetOpacity(self.Opacity)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
examineGlyphs = vtk.vtkGlyph3D()
examineGlyphSource = vtk.vtkSphereSource()
examineGlyphSource.SetRadius(1)
examineGlyphs.SetInput(self.ExamineSpheres)
examineGlyphs.SetSource(examineGlyphSource.GetOutput())
examineGlyphs.SetScaleModeToScaleByScalar()
examineGlyphs.SetScaleFactor(1.0)
examineGlyphMapper = vtk.vtkPolyDataMapper()
examineGlyphMapper.SetInput(examineGlyphs.GetOutput())
examineGlyphMapper.ScalarVisibilityOff()
self.ExamineSpheresActor = vtk.vtkActor()
self.ExamineSpheresActor.SetMapper(examineGlyphMapper)
self.ExamineSpheresActor.GetProperty().SetColor(0.0, 1.0, 0.0)
self.ExamineSpheresActor.GetProperty().SetOpacity(self.Opacity)
self.ExamineSpheresActor.PickableOff()
self.ExamineSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ExamineSpheresActor)
self.vmtkRenderer.AddKeyBinding("u", "Undo.", self.UndoCallback)
self.vmtkRenderer.AddKeyBinding("space", "Place picks.", self.PickCallback)
self.vmtkRenderer.AddKeyBinding("+", "Increase sphere radius.", self.IncreaseSphereRadiusCallback)
self.vmtkRenderer.AddKeyBinding("-", "Decrease sphere radius.", self.DecreaseSphereRadiusCallback)
self.vmtkRenderer.AddKeyBinding("n", "Skip to next sphere.", self.NextCallback)
self.vmtkRenderer.AddKeyBinding("v", "Skip to previous sphere.", self.PreviousCallback)
self.vmtkRenderer.AddKeyBinding("d", "Show distances graph.", self.DistancesCallback)
self.vmtkRenderer.AddKeyBinding("x", "Examine mode.", self.ExamineCallback)
# self.vmtkRenderer.RenderWindowInteractor.AddObserver("KeyPressEvent", self.KeyPressed)
self.SurfaceMapper = vtk.vtkPolyDataMapper()
self.SurfaceMapper.SetInput(self.Surface)
self.SurfaceMapper.SetScalarVisibility(self.DisplayArray)
surfaceActor = vtk.vtkActor()
surfaceActor.SetMapper(self.SurfaceMapper)
surfaceActor.GetProperty().SetOpacity(self.Opacity)
self.vmtkRenderer.Renderer.AddActor(surfaceActor)
self.ScalarBarActor = vtk.vtkScalarBarActor()
self.ScalarBarActor.SetLookupTable(self.SurfaceMapper.GetLookupTable())
self.ScalarBarActor.GetLabelTextProperty().ItalicOff()
self.ScalarBarActor.GetLabelTextProperty().BoldOff()
self.ScalarBarActor.GetLabelTextProperty().ShadowOff()
self.ScalarBarActor.SetLabelFormat("%.2f")
self.ScalarBarActor.SetTitle("distances")
self.ScalarBarActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ScalarBarActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.ExamineText = vtk.vtkTextActor()
self.ExamineText.SetInput("Examine Mode")
self.ExamineText.GetPositionCoordinate().SetCoordinateSystemToNormalizedViewport()
self.ExamineText.SetPosition(0.05, 0.95)
self.ExamineText.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor2D(self.ExamineText)
self.InputInfo("Please position the mouse and press space to add spheres, 'u' to undo\n")
any = 0
while any == 0:
self.InitializeSpheres()
self.vmtkRenderer.Render()
any = self.Spheres.GetNumberOfPoints() > 1
self.InputInfo(
"Please position the mouse and press space to add spheres, 'u' to undo\nInsert at least 2 spheres."
)
self.Surface = self.ComputeArray()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
# create a rendering window and renderer
ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
# create a renderwindowinteractor
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
# draw sphere
totalsphere = (xl + 1) * (yl + 1) * (zl + 1)
spherelist = []
for i in range(totalsphere):
sphereWidget = vtk.vtkSphereWidget()
sphereWidget.SetInteractor(iren)
x, y, z = index2realworld(i)
sphereWidget.SetCenter(x, y, z)
sphereWidget.SetRadius(0.01)
sphereWidget.SetRepresentationToSurface()
sphereWidget.On()
spherelist.append(sphereWidget)
# draw lines
sourcelist = []
mapperlist = []
actorlist = []
for i in range(3*totalsphere):
sourcelist.append(vtk.vtkLineSource())
mapperlist.append(vtk.vtkPolyDataMapper())
def Execute(self):
if (self.Mesh == None):
self.PrintError('Error: no Mesh.')
if (self.Centerlines == None):
self.PrintError('Error: no Centerlines')
#Save the centerlines
previousCenterlines = self.Centerlines
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInput(self.Centerlines)
cleaner.Update()
self.Centerlines = cleaner.GetOutput()
if self.Tolerance == -1:
self.Tolerance = 0.000001*self.Mesh.GetLength()
if self.RadiusArrayName != '':
self.RadiusArray = self.Centerlines.GetPointData().GetArray(self.RadiusArrayName)
if self.RadiusArray == None:
self.PrintError('Error : could not find radius array')
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
meshMapper = vtk.vtkDataSetMapper()
meshMapper.SetInput(self.Mesh)
meshMapper.ScalarVisibilityOff()
self.MeshActor = vtk.vtkActor()
self.MeshActor.SetMapper(meshMapper)
self.MeshActor.GetProperty().SetOpacity(0.25)
self.MeshActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.MeshActor)
centerlinesMapper = vtk.vtkDataSetMapper()
centerlinesMapper.SetInput(self.Centerlines)
centerlinesMapper.ScalarVisibilityOff()
self.CenterlinesActor = vtk.vtkActor()
self.CenterlinesActor.SetMapper(centerlinesMapper)
self.vmtkRenderer.Renderer.AddActor(self.CenterlinesActor)
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInput(self.Spheres)
glyphs.SetSource(glyphSource.GetOutput())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInput(glyphs.GetOutput())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0,0.0,0.0)
self.SpheresActor.GetProperty().SetOpacity(0.25)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
self.InterpolatedGlyphs = vtk.vtkGlyph3D()
interpolatedGlyphSource = vtk.vtkSphereSource()
interpolatedGlyphSource.SetRadius(1)
self.InterpolatedGlyphs.SetInput(self.Centerlines)
self.InterpolatedGlyphs.SetSource(interpolatedGlyphSource.GetOutput())
#scaling is off for now
self.InterpolatedGlyphs.SetScaleModeToDataScalingOff()
self.InterpolatedGlyphs.SetScaleFactor(0.)
interpolatedGlyphMapper = vtk.vtkPolyDataMapper()
interpolatedGlyphMapper.SetInput(self.InterpolatedGlyphs.GetOutput())
interpolatedGlyphMapper.ScalarVisibilityOff()
self.InterpolatedSpheresActor = vtk.vtkActor()
self.InterpolatedSpheresActor.SetMapper(interpolatedGlyphMapper)
self.InterpolatedSpheresActor.GetProperty().SetColor(0.0,1.0,0.0)
self.InterpolatedSpheresActor.GetProperty().SetOpacity(0.25)
self.InterpolatedSpheresActor.PickableOff()
self.InterpolatedSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.InterpolatedSpheresActor)
polyBallMapper = vtk.vtkPolyDataMapper()
polyBallMapper.ScalarVisibilityOff()
self.PolyBallActor = vtk.vtkActor()
self.PolyBallActor.SetMapper(polyBallMapper)
self.PolyBallActor.GetProperty().SetColor(0.0,1.0,0.0)
self.PolyBallActor.GetProperty().SetOpacity(0.25)
self.PolyBallActor.PickableOff()
self.PolyBallActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.PolyBallActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.TranslationOff()
self.SphereWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.Clipper = vtk.vtkClipDataSet()
self.Clipper.SetInput(self.Mesh)
self.Clipper.SetInsideOut(self.InsideOut)
self.Clipper.GenerateClippedOutputOn()
#self.LineClipper = vtkvmtk.vtkvmtkClipDataSetLine()
#self.LineClipper.SetInput(self.Mesh)
#self.LineClipper.SetInsideOut(self.InsideOut)
#self.LineClipper.GenerateClippedOutputOn()
self.InitializeSpheres()
self.PreviewMesh = self.Mesh
self.Display()
self.PolyBallActor.VisibilityOff()
self.ClipMesh()
if self.ClippedMesh == None:
#return an empty mesh
self.ClippedMesh = vtk.vtkUnstructuredGrid()
elif self.IncludeSurfaceCells:
#Create the surface cells
self.PreviewMesh = self.CreateSurfaceCells(self.PreviewMesh)
self.ClippedMesh = self.CreateSurfaceCells(self.ClippedMesh)
self.Mesh = self.PreviewMesh
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
if self.Mesh.GetSource():
self.Mesh.GetSource().UnRegisterAllOutputs()
#Restore the centerlines
self.Centerlines = previousCenterlines
def Execute(self):
if self.Surface == None:
self.PrintError('Error: No input surface.')
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInputData(self.Spheres)
glyphs.SetSourceConnection(glyphSource.GetOutputPort())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInputConnection(glyphs.GetOutputPort())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0, 0.0, 0.0)
self.SpheresActor.GetProperty().SetOpacity(self.Opacity)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
examineGlyphs = vtk.vtkGlyph3D()
examineGlyphSource = vtk.vtkSphereSource()
examineGlyphSource.SetRadius(1)
examineGlyphs.SetInputData(self.ExamineSpheres)
examineGlyphs.SetSourceConnection(examineGlyphSource.GetOutputPort())
examineGlyphs.SetScaleModeToScaleByScalar()
examineGlyphs.SetScaleFactor(1.)
examineGlyphMapper = vtk.vtkPolyDataMapper()
examineGlyphMapper.SetInputConnection(examineGlyphs.GetOutputPort())
examineGlyphMapper.ScalarVisibilityOff()
self.ExamineSpheresActor = vtk.vtkActor()
self.ExamineSpheresActor.SetMapper(examineGlyphMapper)
self.ExamineSpheresActor.GetProperty().SetColor(0.0, 1.0, 0.0)
self.ExamineSpheresActor.GetProperty().SetOpacity(self.Opacity)
self.ExamineSpheresActor.PickableOff()
self.ExamineSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ExamineSpheresActor)
#self.vmtkRenderer.RenderWindowInteractor.AddObserver("KeyPressEvent", self.KeyPressed)
self.vmtkRenderer.AddKeyBinding('u', 'undo', self.UndoCallback)
self.vmtkRenderer.AddKeyBinding('plus', 'Increase sphere radius',
self.PlusCallback)
self.vmtkRenderer.AddKeyBinding('minus', 'Decrease sphere radius',
self.MinusCallback)
self.vmtkRenderer.AddKeyBinding('n', 'Show next sphere',
self.NextSphereCallback)
self.vmtkRenderer.AddKeyBinding('v', 'Show previous sphere',
self.PreviousSphereCallback)
self.vmtkRenderer.AddKeyBinding('d', 'Display ', self.DisplayCallback)
self.vmtkRenderer.AddKeyBinding(
'w', 'Switch beetween examien and interact mode ',
self.ExmienModeCallback)
self.vmtkRenderer.AddKeyBinding('space', 'Pick sphere',
self.PickCallback)
self.SurfaceMapper = vtk.vtkPolyDataMapper()
self.SurfaceMapper.SetInputData(self.Surface)
self.SurfaceMapper.SetScalarVisibility(self.DisplayArray)
surfaceActor = vtk.vtkActor()
surfaceActor.SetMapper(self.SurfaceMapper)
surfaceActor.GetProperty().SetOpacity(self.Opacity)
self.vmtkRenderer.Renderer.AddActor(surfaceActor)
self.ScalarBarActor = vtk.vtkScalarBarActor()
self.ScalarBarActor.SetLookupTable(self.SurfaceMapper.GetLookupTable())
self.ScalarBarActor.GetLabelTextProperty().ItalicOff()
self.ScalarBarActor.GetLabelTextProperty().BoldOff()
self.ScalarBarActor.GetLabelTextProperty().ShadowOff()
self.ScalarBarActor.SetLabelFormat('%.2f')
self.ScalarBarActor.SetTitle('distances')
self.ScalarBarActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ScalarBarActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.TranslationOff()
self.SphereWidget.SetInteractor(
self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.ExamineText = vtk.vtkTextActor()
self.ExamineText.SetInput("Examine Mode")
self.ExamineText.GetPositionCoordinate(
).SetCoordinateSystemToNormalizedViewport()
self.ExamineText.SetPosition(0.05, 0.95)
self.ExamineText.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor2D(self.ExamineText)
self.InputInfo(
'Please position the mouse and press space to add spheres, \'u\' to undo\n'
)
any = 0
while any == 0:
self.InitializeSpheres()
self.vmtkRenderer.Render()
any = (self.Spheres.GetNumberOfPoints() > 1)
self.Surface = self.ComputeArray()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
def Execute(self):
if self.Surface == None:
self.PrintError('Error: No input surface.')
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInput(self.Spheres)
glyphs.SetSource(glyphSource.GetOutput())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInput(glyphs.GetOutput())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0,0.0,0.0)
self.SpheresActor.GetProperty().SetOpacity(self.Opacity)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
examineGlyphs = vtk.vtkGlyph3D()
examineGlyphSource = vtk.vtkSphereSource()
examineGlyphSource.SetRadius(1)
examineGlyphs.SetInput(self.ExamineSpheres)
examineGlyphs.SetSource(examineGlyphSource.GetOutput())
examineGlyphs.SetScaleModeToScaleByScalar()
examineGlyphs.SetScaleFactor(1.)
examineGlyphMapper = vtk.vtkPolyDataMapper()
examineGlyphMapper.SetInput(examineGlyphs.GetOutput())
examineGlyphMapper.ScalarVisibilityOff()
self.ExamineSpheresActor = vtk.vtkActor()
self.ExamineSpheresActor.SetMapper(examineGlyphMapper)
self.ExamineSpheresActor.GetProperty().SetColor(0.0,1.0,0.0)
self.ExamineSpheresActor.GetProperty().SetOpacity(self.Opacity)
self.ExamineSpheresActor.PickableOff()
self.ExamineSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ExamineSpheresActor)
self.vmtkRenderer.AddKeyBinding('u','Undo.',self.UndoCallback)
self.vmtkRenderer.AddKeyBinding('space','Pick.',self.PickCallback)
self.vmtkRenderer.AddKeyBinding('+','Increase sphere radius.',self.IncreaseSphereRadiusCallback)
self.vmtkRenderer.AddKeyBinding('-','Decrease sphere radius.',self.DecreaseSphereRadiusCallback)
self.vmtkRenderer.AddKeyBinding('n','Skip to next sphere.',self.NextCallback)
self.vmtkRenderer.AddKeyBinding('v','Skip to previous sphere.',self.PreviousCallback)
self.vmtkRenderer.AddKeyBinding('d','Show distances graph.',self.DistancesCallback)
self.vmtkRenderer.AddKeyBinding('a','Change parameters.',self.ParametersCallback)
self.vmtkRenderer.AddKeyBinding('w','Switch modes.',self.SwitchModeCallback)
#self.vmtkRenderer.RenderWindowInteractor.AddObserver("KeyPressEvent", self.KeyPressed)
self.SurfaceMapper = vtk.vtkPolyDataMapper()
self.SurfaceMapper.SetInput(self.Surface)
self.SurfaceMapper.SetScalarVisibility(self.DisplayArray)
surfaceActor = vtk.vtkActor()
surfaceActor.SetMapper(self.SurfaceMapper)
surfaceActor.GetProperty().SetOpacity(self.Opacity)
self.vmtkRenderer.Renderer.AddActor(surfaceActor)
self.ScalarBarActor = vtk.vtkScalarBarActor()
self.ScalarBarActor.SetLookupTable(self.SurfaceMapper.GetLookupTable())
self.ScalarBarActor.GetLabelTextProperty().ItalicOff()
self.ScalarBarActor.GetLabelTextProperty().BoldOff()
self.ScalarBarActor.GetLabelTextProperty().ShadowOff()
self.ScalarBarActor.SetLabelFormat('%.2f')
self.ScalarBarActor.SetTitle('distances')
self.ScalarBarActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ScalarBarActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.ExamineText = vtk.vtkTextActor()
self.ExamineText.SetInput("Examine Mode")
self.ExamineText.GetPositionCoordinate().SetCoordinateSystemToNormalizedViewport()
self.ExamineText.SetPosition(0.05,0.95)
self.ExamineText.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor2D(self.ExamineText)
self.InputInfo('Please position the mouse and press space to add spheres, \'u\' to undo\n')
any = 0
while any == 0 or self.ParametersChanged:
self.ParametersChanged = False
if any == 0:
self.InitializeSpheres()
self.vmtkRenderer.Render()
any = self.Spheres.GetNumberOfPoints()
self.Surface = self.ComputeDistances()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
def add_sphere_widget(self,
callback,
center=(0, 0, 0),
radius=0.5,
theta_resolution=30,
phi_resolution=30,
color=None,
style="surface",
selected_color="pink",
indices=None,
pass_widget=False,
test_callback=True):
"""Add one or many sphere widgets to a scene.
Use a sphere widget to control a vertex location.
Parameters
----------
callback : callable
The function to call back when the widget is modified. It takes a
single argument: the center of the sphere as a XYZ coordinate.
center : tuple(float)
Length 3 array for the XYZ coordinate of the sphere's center
when placing it in the scene. If more than one location is passed,
then that many widgets will be added and the callback will also
be passed the integer index of that widget.
radius : float
The radius of the sphere
theta_resolution: int , optional
Set the number of points in the longitude direction (ranging from
start_theta to end theta).
phi_resolution : int, optional
Set the number of points in the latitude direction (ranging from
start_phi to end_phi).
color : str
The color of the sphere's surface
style : str
Representation style: surface or wireframe
selected_color : str
Color of the widget when selected during interaction
pass_widget : bool
If true, the widget will be passed as the last argument of the
callback
test_callback: bool
if true, run the callback function after the widget is created.
"""
if hasattr(self, 'notebook') and self.notebook:
raise AssertionError(
'Sphere widget not available in notebook plotting')
if not hasattr(self, 'iren'):
raise AttributeError(
'Widgets must be used with an intereactive renderer. No off screen plotting.'
)
if not hasattr(self, "sphere_widgets"):
self.sphere_widgets = []
if color is None:
color = rcParams['color']
center = np.array(center)
num = 1
if center.ndim > 1:
num = len(center)
if isinstance(color, (list, tuple, np.ndarray)):
colors = color
else:
colors = [color] * num
def _the_callback(widget, event_id):
point = widget.GetCenter()
index = widget.WIDGET_INDEX
if hasattr(callback, '__call__'):
if num > 1:
args = [point, index]
else:
args = [point]
if pass_widget:
args.append(widget)
try_callback(callback, *args)
return
if indices is None:
indices = [x for x in range(num)]
for i in range(num):
if center.ndim > 1:
loc = center[i]
else:
loc = center
sphere_widget = vtk.vtkSphereWidget()
sphere_widget.WIDGET_INDEX = indices[i] # Monkey patch the index
if style in "wireframe":
sphere_widget.SetRepresentationToWireframe()
else:
sphere_widget.SetRepresentationToSurface()
sphere_widget.GetSphereProperty().SetColor(parse_color(colors[i]))
sphere_widget.GetSelectedSphereProperty().SetColor(
parse_color(selected_color))
sphere_widget.SetInteractor(self.iren)
sphere_widget.SetCurrentRenderer(self.renderer)
sphere_widget.SetRadius(radius)
sphere_widget.SetCenter(loc)
sphere_widget.SetThetaResolution(theta_resolution)
sphere_widget.SetPhiResolution(phi_resolution)
sphere_widget.Modified()
sphere_widget.On()
sphere_widget.AddObserver(vtk.vtkCommand.EndInteractionEvent,
_the_callback)
self.sphere_widgets.append(sphere_widget)
if test_callback is True:
# Test call back in the last
_the_callback(sphere_widget, None)
if num > 1:
return self.sphere_widgets
return sphere_widget
def Execute(self):
if (self.Mesh == None):
self.PrintError('Error: no Mesh.')
if (self.Centerlines == None):
self.PrintError('Error: no Centerlines')
#Save the centerlines
previousCenterlines = self.Centerlines
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.Centerlines)
cleaner.Update()
self.Centerlines = cleaner.GetOutput()
if self.Tolerance == -1:
self.Tolerance = 0.000001*self.Mesh.GetLength()
if self.RadiusArrayName != '':
self.RadiusArray = self.Centerlines.GetPointData().GetArray(self.RadiusArrayName)
if self.RadiusArray == None:
self.PrintError('Error : could not find radius array')
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
meshMapper = vtk.vtkDataSetMapper()
meshMapper.SetInputData(self.Mesh)
meshMapper.ScalarVisibilityOff()
self.MeshActor = vtk.vtkActor()
self.MeshActor.SetMapper(meshMapper)
self.MeshActor.GetProperty().SetOpacity(0.25)
self.MeshActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.MeshActor)
centerlinesMapper = vtk.vtkDataSetMapper()
centerlinesMapper.SetInputData(self.Centerlines)
centerlinesMapper.ScalarVisibilityOff()
self.CenterlinesActor = vtk.vtkActor()
self.CenterlinesActor.SetMapper(centerlinesMapper)
self.vmtkRenderer.Renderer.AddActor(self.CenterlinesActor)
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInputData(self.Spheres)
glyphs.SetSourceConnection(glyphSource.GetOutputPort())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInputConnection(glyphs.GetOutput())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0,0.0,0.0)
self.SpheresActor.GetProperty().SetOpacity(0.25)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
self.InterpolatedGlyphs = vtk.vtkGlyph3D()
interpolatedGlyphSource = vtk.vtkSphereSource()
interpolatedGlyphSource.SetRadius(1)
self.InterpolatedGlyphs.SetInputData(self.Centerlines)
self.InterpolatedGlyphs.SetSourceConnection(interpolatedGlyphSource.GetOutputPort())
#scaling is off for now
self.InterpolatedGlyphs.SetScaleModeToDataScalingOff()
self.InterpolatedGlyphs.SetScaleFactor(0.)
interpolatedGlyphMapper = vtk.vtkPolyDataMapper()
interpolatedGlyphMapper.SetInputConnection(self.InterpolatedGlyphs.GetOutputPort())
interpolatedGlyphMapper.ScalarVisibilityOff()
self.InterpolatedSpheresActor = vtk.vtkActor()
self.InterpolatedSpheresActor.SetMapper(interpolatedGlyphMapper)
self.InterpolatedSpheresActor.GetProperty().SetColor(0.0,1.0,0.0)
self.InterpolatedSpheresActor.GetProperty().SetOpacity(0.25)
self.InterpolatedSpheresActor.PickableOff()
self.InterpolatedSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.InterpolatedSpheresActor)
polyBallMapper = vtk.vtkPolyDataMapper()
polyBallMapper.ScalarVisibilityOff()
self.PolyBallActor = vtk.vtkActor()
self.PolyBallActor.SetMapper(polyBallMapper)
self.PolyBallActor.GetProperty().SetColor(0.0,1.0,0.0)
self.PolyBallActor.GetProperty().SetOpacity(0.25)
self.PolyBallActor.PickableOff()
self.PolyBallActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.PolyBallActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.TranslationOff()
self.SphereWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.Clipper = vtk.vtkClipDataSet()
self.Clipper.SetInputData(self.Mesh)
self.Clipper.SetInsideOut(self.InsideOut)
self.Clipper.GenerateClippedOutputOn()
#self.LineClipper = vtkvmtk.vtkvmtkClipDataSetLine()
#self.LineClipper.SetInputData(self.Mesh)
#self.LineClipper.SetInsideOut(self.InsideOut)
#self.LineClipper.GenerateClippedOutputOn()
self.InitializeSpheres()
self.PreviewMesh = self.Mesh
self.Display()
self.PolyBallActor.VisibilityOff()
self.ClipMesh()
if self.ClippedMesh == None:
#return an empty mesh
self.ClippedMesh = vtk.vtkUnstructuredGrid()
elif self.IncludeSurfaceCells:
#Create the surface cells
self.PreviewMesh = self.CreateSurfaceCells(self.PreviewMesh)
self.ClippedMesh = self.CreateSurfaceCells(self.ClippedMesh)
self.Mesh = self.PreviewMesh
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
if self.Mesh.GetSource():
self.Mesh.GetSource().UnRegisterAllOutputs()
#Restore the centerlines
self.Centerlines = previousCenterlines
def Execute(self):
if self.Surface == None:
self.PrintError('Error: No input surface.')
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInput(self.Spheres)
glyphs.SetSource(glyphSource.GetOutput())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInput(glyphs.GetOutput())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0, 0.0, 0.0)
self.SpheresActor.GetProperty().SetOpacity(self.Opacity)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
examineGlyphs = vtk.vtkGlyph3D()
examineGlyphSource = vtk.vtkSphereSource()
examineGlyphSource.SetRadius(1)
examineGlyphs.SetInput(self.ExamineSpheres)
examineGlyphs.SetSource(examineGlyphSource.GetOutput())
examineGlyphs.SetScaleModeToScaleByScalar()
examineGlyphs.SetScaleFactor(1.)
examineGlyphMapper = vtk.vtkPolyDataMapper()
examineGlyphMapper.SetInput(examineGlyphs.GetOutput())
examineGlyphMapper.ScalarVisibilityOff()
self.ExamineSpheresActor = vtk.vtkActor()
self.ExamineSpheresActor.SetMapper(examineGlyphMapper)
self.ExamineSpheresActor.GetProperty().SetColor(0.0, 1.0, 0.0)
self.ExamineSpheresActor.GetProperty().SetOpacity(self.Opacity)
self.ExamineSpheresActor.PickableOff()
self.ExamineSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ExamineSpheresActor)
self.vmtkRenderer.RenderWindowInteractor.AddObserver(
"KeyPressEvent", self.KeyPressed)
self.SurfaceMapper = vtk.vtkPolyDataMapper()
self.SurfaceMapper.SetInput(self.Surface)
self.SurfaceMapper.SetScalarVisibility(self.DisplayArray)
surfaceActor = vtk.vtkActor()
surfaceActor.SetMapper(self.SurfaceMapper)
surfaceActor.GetProperty().SetOpacity(self.Opacity)
self.vmtkRenderer.Renderer.AddActor(surfaceActor)
self.ScalarBarActor = vtk.vtkScalarBarActor()
self.ScalarBarActor.SetLookupTable(self.SurfaceMapper.GetLookupTable())
self.ScalarBarActor.GetLabelTextProperty().ItalicOff()
self.ScalarBarActor.GetLabelTextProperty().BoldOff()
self.ScalarBarActor.GetLabelTextProperty().ShadowOff()
self.ScalarBarActor.SetLabelFormat('%.2f')
self.ScalarBarActor.SetTitle('distances')
self.ScalarBarActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ScalarBarActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.SetInteractor(
self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.ExamineText = vtk.vtkTextActor()
self.ExamineText.SetInput("Examine Mode")
self.ExamineText.GetPositionCoordinate(
).SetCoordinateSystemToNormalizedViewport()
self.ExamineText.SetPosition(0.05, 0.95)
self.ExamineText.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor2D(self.ExamineText)
self.OutputText(
'Please position the mouse and press space to add spheres, \'u\' to undo\n'
)
any = 0
while any == 0:
self.InitializeSpheres()
self.vmtkRenderer.Render()
any = (self.Spheres.GetNumberOfPoints() > 1)
self.Surface = self.ComputeArray()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
def Execute(self):
if self.Surface == None:
self.PrintError('Error: No input surface.')
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
glyphs = vtk.vtkGlyph3D()
glyphSource = vtk.vtkSphereSource()
glyphSource.SetRadius(1)
glyphs.SetInput(self.Spheres)
glyphs.SetSource(glyphSource.GetOutput())
glyphs.SetScaleModeToScaleByScalar()
glyphs.SetScaleFactor(1.)
glyphMapper = vtk.vtkPolyDataMapper()
glyphMapper.SetInput(glyphs.GetOutput())
glyphMapper.ScalarVisibilityOff()
self.SpheresActor = vtk.vtkActor()
self.SpheresActor.SetMapper(glyphMapper)
self.SpheresActor.GetProperty().SetColor(1.0,0.0,0.0)
self.SpheresActor.GetProperty().SetOpacity(self.Opacity)
self.SpheresActor.PickableOff()
self.vmtkRenderer.Renderer.AddActor(self.SpheresActor)
examineGlyphs = vtk.vtkGlyph3D()
examineGlyphSource = vtk.vtkSphereSource()
examineGlyphSource.SetRadius(1)
examineGlyphs.SetInput(self.ExamineSpheres)
examineGlyphs.SetSource(examineGlyphSource.GetOutput())
examineGlyphs.SetScaleModeToScaleByScalar()
examineGlyphs.SetScaleFactor(1.)
examineGlyphMapper = vtk.vtkPolyDataMapper()
examineGlyphMapper.SetInput(examineGlyphs.GetOutput())
examineGlyphMapper.ScalarVisibilityOff()
self.ExamineSpheresActor = vtk.vtkActor()
self.ExamineSpheresActor.SetMapper(examineGlyphMapper)
self.ExamineSpheresActor.GetProperty().SetColor(0.0,1.0,0.0)
self.ExamineSpheresActor.GetProperty().SetOpacity(self.Opacity)
self.ExamineSpheresActor.PickableOff()
self.ExamineSpheresActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ExamineSpheresActor)
self.vmtkRenderer.RenderWindowInteractor.AddObserver("KeyPressEvent", self.KeyPressed)
self.SurfaceMapper = vtk.vtkPolyDataMapper()
self.SurfaceMapper.SetInput(self.Surface)
self.SurfaceMapper.SetScalarVisibility(self.DisplayArray)
surfaceActor = vtk.vtkActor()
surfaceActor.SetMapper(self.SurfaceMapper)
surfaceActor.GetProperty().SetOpacity(self.Opacity)
self.vmtkRenderer.Renderer.AddActor(surfaceActor)
self.ScalarBarActor = vtk.vtkScalarBarActor()
self.ScalarBarActor.SetLookupTable(self.SurfaceMapper.GetLookupTable())
self.ScalarBarActor.GetLabelTextProperty().ItalicOff()
self.ScalarBarActor.GetLabelTextProperty().BoldOff()
self.ScalarBarActor.GetLabelTextProperty().ShadowOff()
self.ScalarBarActor.SetLabelFormat('%.2f')
self.ScalarBarActor.SetTitle('distances')
self.ScalarBarActor.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor(self.ScalarBarActor)
self.SphereWidget = vtk.vtkSphereWidget()
self.SphereWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.SphereWidget.AddObserver("InteractionEvent", self.SphereCallback)
self.ExamineText = vtk.vtkTextActor()
self.ExamineText.SetInput("Examine Mode")
self.ExamineText.GetPositionCoordinate().SetCoordinateSystemToNormalizedViewport()
self.ExamineText.SetPosition(0.05,0.95)
self.ExamineText.VisibilityOff()
self.vmtkRenderer.Renderer.AddActor2D(self.ExamineText)
self.OutputText('Please position the mouse and press space to add spheres, \'u\' to undo\n')
any = 0
while any == 0:
self.InitializeSpheres()
self.vmtkRenderer.Render()
any = self.Spheres.GetNumberOfPoints()
self.Surface = self.ComputeDistances()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
demMapper.ImmediateModeRenderingOn()
demActor = vtk.vtkLODActor()
demActor.SetMapper(demMapper)
# Create the RenderWindow, Renderer and both Actors
ren = vtk.vtkRenderer()
renWin = vtk.vtkRenderWindow()
renWin.AddRenderer(ren)
iren = vtk.vtkRenderWindowInteractor()
iren.SetRenderWindow(renWin)
iren.LightFollowCameraOff()
iren.SetInteractorStyle(None)
# Associate the line widget with the interactor
sphereWidget = vtk.vtkSphereWidget()
sphereWidget.SetInteractor(iren)
sphereWidget.SetProp3D(demActor)
sphereWidget.SetPlaceFactor(4)
sphereWidget.PlaceWidget()
sphereWidget.TranslationOff()
sphereWidget.ScaleOff()
sphereWidget.HandleVisibilityOn()
# Uncomment the next line if you want to see the widget active when
# the script starts
#sphereWidget.EnabledOn()
# Actually probe the data
def MoveLight(obj, event):
global light
for i in vertex_ids:
polygon.GetPointIds().SetId(id0, i)
id0 += 1
cells.InsertNextCell(polygon)
polygonPolyData = vtk.vtkPolyData()
polygonPolyData.SetPoints(points)
polygonPolyData.SetPolys(cells)
return polygonPolyData
# draw sphere
totalsphere = (xl + 1) * (yl + 1) * (zl + 1)
spherelist = []
for i in range(totalsphere):
sphereWidget = vtk.vtkSphereWidget() # 产生一个点
sphereWidget.SetInteractor(mw.SceneManager.iren) # 用iren进行互动
x, y, z = index2realworld(i)
sphereWidget.SetCenter(x, y, z) # 设置中心
sphereWidget.SetRadius(0.01) # 设置直径
sphereWidget.SetRepresentationToSurface() # 不懂
sphereWidget.On()
spherelist.append(sphereWidget) # 球球的列表
spherelist_position = [spherelist[i].GetCenter() for i in range(totalsphere)]
# draw lines
sourcelist = []
mapperlist = []
actorlist = []
for i in range(3 * totalsphere): # 首先创建一些vtk的线
sourcelist.append(vtk.vtkLineSource())
mapperlist.append(vtk.vtkPolyDataMapper())
def Execute(self):
if self.Surface == None:
self.PrintError('Error: no Surface to for Execute().')
self.Clipper = vtk.vtkClipPolyData()
self.Clipper.SetInputData(self.Surface)
self.Clipper.GenerateClippedOutputOn()
self.Clipper.SetInsideOut(self.InsideOut)
if self.Interactive:
if self.WidgetType == "box":
self.ClipFunction = vtk.vtkPlanes()
elif self.WidgetType == "sphere":
self.ClipFunction = vtk.vtkSphere()
self.Clipper.SetClipFunction(self.ClipFunction)
self.Cutter = vtk.vtkCutter()
self.Cutter.SetInputData(self.Surface)
self.Cutter.SetCutFunction(self.ClipFunction)
self.ClippedSurface = vtk.vtkPolyData()
self.CutLines = vtk.vtkPolyData()
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputData(self.Surface)
mapper.ScalarVisibilityOff()
self.Actor = vtk.vtkActor()
self.Actor.SetMapper(mapper)
self.vmtkRenderer.Renderer.AddActor(self.Actor)
if self.WidgetType == "box":
self.ClipWidget = vtk.vtkBoxWidget()
self.ClipWidget.GetFaceProperty().SetColor(0.6,0.6,0.2)
self.ClipWidget.GetFaceProperty().SetOpacity(0.25)
elif self.WidgetType == "sphere":
self.ClipWidget = vtk.vtkSphereWidget()
self.ClipWidget.GetSphereProperty().SetColor(0.6,0.6,0.2)
self.ClipWidget.GetSphereProperty().SetOpacity(0.25)
self.ClipWidget.GetSelectedSphereProperty().SetColor(0.6,0.0,0.0)
self.ClipWidget.GetSelectedSphereProperty().SetOpacity(0.75)
self.ClipWidget.SetRepresentationToSurface()
self.ClipWidget.SetPhiResolution(20)
self.ClipWidget.SetThetaResolution(20)
self.ClipWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
# Try AutoClip here
if not self.ClipsIn == None:
self.AutoClip()
self.vmtkRenderer.AddKeyBinding('space','Clip.',self.ClipCallback)
self.vmtkRenderer.AddKeyBinding('i','Interact.',self.InteractCallback)
self.Display()
self.Transform = vtk.vtkTransform()
if self.WidgetType == "box":
self.ClipWidget.GetTransform(self.Transform)
# Add items from ClipCollection to MultiBlockDataSet
self.ClipsOut = vtk.vtkMultiBlockDataSet()
ClipTotal = self.ClipCollection.GetNumberOfItems()
BlockNum = 0
self.ClipCollection.InitTraversal()
while BlockNum < ClipTotal:
blockin = vtk.vtkPolyData()
blockin.DeepCopy(self.ClipCollection.GetNextItem())
self.PrintLog('The bounds of blockin are: ' + str(blockin.GetBounds()))
self.ClipsOut.SetBlock(BlockNum, blockin)
BlockNum += 1
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
else:
self.Surface.GetPointData().SetActiveScalars(self.ClipArrayName)
self.Clipper.GenerateClipScalarsOff()
self.Clipper.SetValue(self.ClipValue)
self.Clipper.Update()
self.Cutter = vtk.vtkContourFilter()
self.Cutter.SetInputData(self.Surface)
self.Cutter.SetValue(0,self.ClipValue)
self.Cutter.Update()
self.Surface = self.Clipper.GetOutput()
self.ClippedSurface = self.Clipper.GetClippedOutput()
self.CutLines = self.Cutter.GetOutput()
if self.CleanOutput == 1:
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.Surface)
cleaner.Update()
self.Surface = cleaner.GetOutput()
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.ClippedSurface)
cleaner.Update()
self.ClippedSurface = cleaner.GetOutput()
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.CutLines)
cleaner.Update()
stripper = vtk.vtkStripper()
stripper.SetInputConnection(cleaner.GetOutputPort())
stripper.Update()
self.CutLines = stripper.GetOutput()
def Execute(self):
if self.Surface == None:
self.PrintError('Error: no Surface.')
self.Clipper = vtk.vtkClipPolyData()
self.Clipper.SetInput(self.Surface)
self.Clipper.GenerateClippedOutputOn()
self.Clipper.SetInsideOut(self.InsideOut)
if self.Interactive:
if self.WidgetType == "box":
self.ClipFunction = vtk.vtkPlanes()
elif self.WidgetType == "sphere":
self.ClipFunction = vtk.vtkSphere()
self.Clipper.SetClipFunction(self.ClipFunction)
self.Cutter = vtk.vtkCutter()
self.Cutter.SetInput(self.Surface)
self.Cutter.SetCutFunction(self.ClipFunction)
self.ClippedSurface = vtk.vtkPolyData()
self.CutLines = vtk.vtkPolyData()
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
mapper = vtk.vtkPolyDataMapper()
mapper.SetInput(self.Surface)
mapper.ScalarVisibilityOff()
self.Actor = vtk.vtkActor()
self.Actor.SetMapper(mapper)
self.vmtkRenderer.Renderer.AddActor(self.Actor)
if self.WidgetType == "box":
self.ClipWidget = vtk.vtkBoxWidget()
self.ClipWidget.GetFaceProperty().SetColor(0.6,0.6,0.2)
self.ClipWidget.GetFaceProperty().SetOpacity(0.25)
elif self.WidgetType == "sphere":
self.ClipWidget = vtk.vtkSphereWidget()
self.ClipWidget.GetSphereProperty().SetColor(0.6,0.6,0.2)
self.ClipWidget.GetSphereProperty().SetOpacity(0.25)
self.ClipWidget.GetSelectedSphereProperty().SetColor(0.6,0.0,0.0)
self.ClipWidget.GetSelectedSphereProperty().SetOpacity(0.75)
self.ClipWidget.SetRepresentationToSurface()
self.ClipWidget.SetPhiResolution(20)
self.ClipWidget.SetThetaResolution(20)
self.ClipWidget.SetInteractor(self.vmtkRenderer.RenderWindowInteractor)
self.vmtkRenderer.AddKeyBinding('space','Clip.',self.ClipCallback)
self.vmtkRenderer.AddKeyBinding('i','Interact.',self.InteractCallback)
self.Display()
self.Transform = vtk.vtkTransform()
self.ClipWidget.GetTransform(self.Transform)
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
else:
self.Surface.GetPointData().SetActiveScalars(self.ClipArrayName)
self.Clipper.GenerateClipScalarsOff()
self.Clipper.SetValue(self.ClipValue)
self.Clipper.Update()
self.Cutter = vtk.vtkContourFilter()
self.Cutter.SetInput(self.Surface)
self.Cutter.SetValue(0,self.ClipValue)
self.Cutter.Update()
self.Surface = self.Clipper.GetOutput()
self.ClippedSurface = self.Clipper.GetClippedOutput()
self.CutLines = self.Cutter.GetOutput()
if self.CleanOutput == 1:
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInput(self.Surface)
cleaner.Update()
self.Surface = cleaner.GetOutput()
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInput(self.ClippedSurface)
cleaner.Update()
self.ClippedSurface = cleaner.GetOutput()
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInput(self.CutLines)
cleaner.Update()
stripper = vtk.vtkStripper()
stripper.SetInput(cleaner.GetOutput())
stripper.Update()
self.CutLines = stripper.GetOutput()
if self.Surface.GetSource():
self.Surface.GetSource().UnRegisterAllOutputs()
def Execute(self):
if self.Surface == None:
self.PrintError('Error: no Surface.')
self.Clipper = vtk.vtkClipPolyData()
self.Clipper.SetInputData(self.Surface)
self.Clipper.GenerateClippedOutputOn()
self.Clipper.SetInsideOut(self.InsideOut)
if self.Interactive:
if self.WidgetType == "box":
self.ClipFunction = vtk.vtkPlanes()
elif self.WidgetType == "sphere":
self.ClipFunction = vtk.vtkSphere()
self.Clipper.SetClipFunction(self.ClipFunction)
self.Cutter = vtk.vtkCutter()
self.Cutter.SetInputData(self.Surface)
self.Cutter.SetCutFunction(self.ClipFunction)
self.ClippedSurface = vtk.vtkPolyData()
self.CutLines = vtk.vtkPolyData()
if not self.vmtkRenderer:
self.vmtkRenderer = vmtkrenderer.vmtkRenderer()
self.vmtkRenderer.Initialize()
self.OwnRenderer = 1
self.vmtkRenderer.RegisterScript(self)
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputData(self.Surface)
mapper.ScalarVisibilityOff()
self.Actor = vtk.vtkActor()
self.Actor.SetMapper(mapper)
self.vmtkRenderer.Renderer.AddActor(self.Actor)
if self.WidgetType == "box":
self.ClipWidget = vtk.vtkBoxWidget()
self.ClipWidget.GetFaceProperty().SetColor(0.6, 0.6, 0.2)
self.ClipWidget.GetFaceProperty().SetOpacity(0.25)
elif self.WidgetType == "sphere":
self.ClipWidget = vtk.vtkSphereWidget()
self.ClipWidget.GetSphereProperty().SetColor(0.6, 0.6, 0.2)
self.ClipWidget.GetSphereProperty().SetOpacity(0.25)
self.ClipWidget.GetSelectedSphereProperty().SetColor(
0.6, 0.0, 0.0)
self.ClipWidget.GetSelectedSphereProperty().SetOpacity(0.75)
self.ClipWidget.SetRepresentationToSurface()
self.ClipWidget.SetPhiResolution(20)
self.ClipWidget.SetThetaResolution(20)
self.ClipWidget.SetInteractor(
self.vmtkRenderer.RenderWindowInteractor)
self.vmtkRenderer.AddKeyBinding('space', 'Clip.',
self.ClipCallback)
self.vmtkRenderer.AddKeyBinding('i', 'Interact.',
self.InteractCallback)
self.Display()
self.Transform = vtk.vtkTransform()
if self.WidgetType == "box":
self.ClipWidget.GetTransform(self.Transform)
if self.OwnRenderer:
self.vmtkRenderer.Deallocate()
else:
self.Surface.GetPointData().SetActiveScalars(self.ClipArrayName)
self.Clipper.GenerateClipScalarsOff()
self.Clipper.SetValue(self.ClipValue)
self.Clipper.Update()
self.Cutter = vtk.vtkContourFilter()
self.Cutter.SetInputData(self.Surface)
self.Cutter.SetValue(0, self.ClipValue)
self.Cutter.Update()
self.Surface = self.Clipper.GetOutput()
self.ClippedSurface = self.Clipper.GetClippedOutput()
self.CutLines = self.Cutter.GetOutput()
if self.CleanOutput == 1:
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.Surface)
cleaner.Update()
self.Surface = cleaner.GetOutput()
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.ClippedSurface)
cleaner.Update()
self.ClippedSurface = cleaner.GetOutput()
cleaner = vtk.vtkCleanPolyData()
cleaner.SetInputData(self.CutLines)
cleaner.Update()
stripper = vtk.vtkStripper()
stripper.SetInputConnection(cleaner.GetOutputPort())
stripper.Update()
self.CutLines = stripper.GetOutput()
def _createImplicit(self, implicitType, implicitName, bounds, primaryInput):
if implicitType in self._implicitTypes and \
implicitName not in self._implicitsDict:
pi = primaryInput
rwi = self.slice3dVWR.threedFrame.threedRWI
implicitInfoBounds = None
if implicitType == "Plane":
implicitWidget = vtk.vtkImplicitPlaneWidget()
implicitWidget.SetPlaceFactor(1.25)
if pi != None:
implicitWidget.SetInput(pi)
implicitWidget.PlaceWidget()
b = pi.GetBounds()
implicitWidget.SetOrigin(b[0], b[2], b[4])
implicitInfoBounds = b
elif bounds != None:
implicitWidget.PlaceWidget(bounds)
implicitWidget.SetOrigin(bounds[0], bounds[2], bounds[4])
implicitInfoBounds = bounds
else:
# this can never happen
pass
implicitWidget.SetInteractor(rwi)
implicitWidget.On()
# create the implicit function
implicitFunction = vtk.vtkPlane()
# sync it to the initial widget
self._syncPlaneFunctionToWidget(implicitWidget)
# add it to the output
self.outputImplicitFunction.AddFunction(implicitFunction)
# now add an observer to the widget
def observerImplicitPlaneWidget(widget, eventName):
# sync it to the initial widget
ret = self._syncPlaneFunctionToWidget(widget)
# also select the correct grid row
if ret != None:
name, ii = ret
row = self.findGridRowByName(name)
if row >= 0:
self._grid.SelectRow(row)
oId = implicitWidget.AddObserver('EndInteractionEvent',
observerImplicitPlaneWidget)
elif implicitType == "Sphere":
implicitWidget = vtk.vtkSphereWidget()
implicitWidget.SetPlaceFactor(1.25)
implicitWidget.TranslationOn()
implicitWidget.ScaleOn()
#implicitWidget.HandleVisibilityOn()
if pi != None:
implicitWidget.SetInput(pi)
implicitWidget.PlaceWidget()
b = pi.GetBounds()
implicitInfoBounds = b
#implicitWidget.SetOrigin(b[0], b[2], b[4])
elif bounds != None:
implicitWidget.PlaceWidget(bounds)
implicitInfoBounds = bounds
#implicitWidget.SetOrigin(bounds[0], bounds[2], bounds[4])
else:
# this can never happen
pass
implicitWidget.SetInteractor(rwi)
implicitWidget.On()
# create the implicit function
implicitFunction = vtk.vtkSphere()
# sync it to the initial widget
self._syncSphereFunctionToWidget(implicitWidget)
# add it to the output
self.outputImplicitFunction.AddFunction(implicitFunction)
# now add an observer to the widget
def observerImplicitSphereWidget(widget, eventName):
# sync it to the initial widget
ret = self._syncSphereFunctionToWidget(widget)
# also select the correct grid row
if ret != None:
name, ii = ret
row = self.findGridRowByName(name)
if row >= 0:
self._grid.SelectRow(row)
oId = implicitWidget.AddObserver('EndInteractionEvent',
observerImplicitSphereWidget)
if implicitWidget:
# set the priority so it gets interaction before the
# ImagePlaneWidget. 3D widgets have default priority 0.5,
# so we assign our widgets just a tad higher. (voiwidget
# has 0.6 for example)
# NB: in a completely weird twist of events, only slices
# added AFTER this widget will act like they have lower
# priority. The initial slice still takes events from us!
implicitWidget.SetPriority(0.7)
# add to our internal thingy
ii = implicitInfo()
ii.name = implicitName
ii.type = implicitType
ii.widget = implicitWidget
ii.bounds = implicitInfoBounds
ii.oId = oId
ii.function = implicitFunction
self._implicitsDict[implicitName] = ii
# now add to the grid
nrGridRows = self._grid.GetNumberRows()
self._grid.AppendRows()
self._grid.SetCellValue(nrGridRows, self._gridNameCol,
implicitName)
self._grid.SetCellValue(nrGridRows, self._gridTypeCol,
implicitType)
# set the relevant cells up for Boolean
for col in [self._gridEnabledCol]:
self._grid.SetCellRenderer(nrGridRows, col,
wx.grid.GridCellBoolRenderer())
self._grid.SetCellAlignment(nrGridRows, col,
wx.ALIGN_CENTRE,
wx.ALIGN_CENTRE)
self._setImplicitEnabled(ii.name, True)
