def gen(style, event):
size = renderWindow.GetSize()
start = list(style.GetStartPosition())
end = list(style.GetEndPosition())
new_start = [start[0], end[1]]
new_end = [end[0], start[1]]
print("start,end: ", start, end)
print("new_start,new_end: ", new_start, new_end)
tmp = [new_start[0] / size[0], new_start[1] / size[1]]
print("scaled bottom_left: ", tmp)
print("scaled top_right: ", new_end[0] / size[0], new_end[1] / size[1])
representation = vtk.vtkBorderRepresentation()
borderWidget = vtk.vtkBorderWidget()
representation.SetPosition(tmp[0], tmp[1])
representation.SetPosition2(new_end[0] / size[0] - tmp[0],
new_end[1] / size[1] - tmp[1])
representation.SetMoving(0)
print("Moving status: ", representation.GetMoving())
borderWidget.SetRepresentation(representation)
borderWidget.SetInteractor(interactor)
borderWidget.ResizableOn()
# borderWidget.CreateDefaultRepresentation()
# borderWidget.SelectableO()
borderWidget.On()
renderWindow.Render()
widgets.append(borderWidget)
def setScene(self, scene):
logging.debug("In TextWidget::setScene()")
if self.scene:
return
self.scene = scene
self.rep = vtk.vtkBorderRepresentation()
self.rep.GetPositionCoordinate().SetValue( self.position[0],self.position[1] )
self.rep.GetPosition2Coordinate().SetValue( .2, .045 )
self.CreateDefaultRepresentation()
self.SetRepresentation(self.rep)
self.rep.SetShowBorderToActive()
self.SetInteractor(self.scene.interactor)
self._observers = []
self._observers.append(self.AddObserver("StartInteractionEvent", self.startInteractionAction))
self._observers.append(self.AddObserver("InteractionEvent", self.interactionAction))
self.startEvent()
ren1 = vtk.vtkRenderer() renWin = vtk.vtkRenderWindow() renWin.AddRenderer(ren1) iren = vtk.vtkRenderWindowInteractor() iren.SetRenderWindow(renWin) # Create a test pipeline ss = vtk.vtkSphereSource() mapper = vtk.vtkPolyDataMapper() mapper.SetInput(ss.GetOutput()) actor = vtk.vtkActor() actor.SetMapper(mapper) # Create the widget and its representation rep = vtk.vtkBorderRepresentation() rep.ProportionalResizeOn() rep.SetShowBorderToOn() #rep.SetShowBorderToActive() widget = vtk.vtkBorderWidget() widget.SetInteractor(iren) widget.SetRepresentation(rep) widget.SelectableOff() # Add the actors to the renderer, set the background and size ren1.AddActor(actor) ren1.SetBackground(0.1, 0.2, 0.4) renWin.SetSize(300, 300) # record events
def main():
img_reader = ImageReaderFactory().GenerateImageReader("png")
img_reader.SetFileName("./kitti/image_00/000000.png")
img_actor = ImageActor(img_reader.GetOutputPort())
sphereSource = vtk.vtkSphereSource()
sphereSource.SetRadius(4.0)
sphereSource.Update()
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(sphereSource.GetOutputPort())
actor = vtk.vtkActor()
actor.SetMapper(mapper)
renderer = vtk.vtkRenderer()
renderWindow = vtk.vtkRenderWindow()
renderWindow.AddRenderer(renderer)
interactor = vtk.vtkRenderWindowInteractor()
interactor.SetRenderWindow(renderWindow)
borderWidget = vtk.vtkBorderWidget()
borderWidget.SetInteractor(interactor)
representation = vtk.vtkBorderRepresentation()
representation.SetPosition(0, 0)
representation.SetPosition2(0.1, 0.1)
borderWidget.SetRepresentation(representation)
# borderWidget.CreateDefaultRepresentation()
borderWidget.SelectableOff()
borderWidget.On()
print(representation)
style = vtk.vtkInteractorStyleRubberBand2D()
def gen(style, event):
size = renderWindow.GetSize()
start = list(style.GetStartPosition())
end = list(style.GetEndPosition())
new_start = [start[0], end[1]]
new_end = [end[0], start[1]]
print("start,end: ", start, end)
print("new_start,new_end: ", new_start, new_end)
tmp = [new_start[0] / size[0], new_start[1] / size[1]]
print("scaled bottom_left: ", tmp)
print("scaled top_right: ", new_end[0] / size[0], new_end[1] / size[1])
representation = vtk.vtkBorderRepresentation()
borderWidget = vtk.vtkBorderWidget()
representation.SetPosition(tmp[0], tmp[1])
representation.SetPosition2(new_end[0] / size[0] - tmp[0],
new_end[1] / size[1] - tmp[1])
representation.SetMoving(0)
print("Moving status: ", representation.GetMoving())
borderWidget.SetRepresentation(representation)
borderWidget.SetInteractor(interactor)
borderWidget.ResizableOn()
# borderWidget.CreateDefaultRepresentation()
# borderWidget.SelectableO()
borderWidget.On()
renderWindow.Render()
widgets.append(borderWidget)
# def stylecallback(style, event):
# size = renderWindow.GetSize()
# start = list(style.GetStartPosition())
# end = list(style.GetEndPosition())
# new_start = [start[0], end[1]]
# new_end = [end[0], start[1]]
# print("start,end: ", start, end)
# print("new_start,new_end: ", new_start, new_end)
# tmp = [new_start[0] / size[0], new_start[1] / size[1]]
# print("scaled bottom_left: ", tmp)
# print("scaled top_right: ", new_end[0] / size[0], new_end[1] / size[1])
# r = vtk.vtkBorderRepresentation()
# # representation.MovingOff()
# borderWidget = vtk.vtkBorderWidget()
# # representation.ResizableOff()
# # representation = vtk.vtkBorderRepresentation()
# # representation
# borderWidget.SelectableOff()
# # global Flag
# # if Flag:
# # borderWidget.SelectableOn()
# # Flag = False
# # else:
# # borderWidget.SelectableOff()
# # Flag = True
# # representation.SetPosition(tmp[0], tmp[1])
# # representation.SetPosition2(
# # new_end[0] / size[0] - tmp[0], new_end[1] / size[1] - tmp[1])
# # representation.SetMoving(1)
# # print("Moving status: ", representation.GetMoving())
# # borderWidget.CreateDefaultRepresentation()
# # r = borderWidget.GetBorderRepresentation()
# r.MovingOff()
# r.SetPosition(tmp[0], tmp[1])
# r.SetPosition2(
# new_end[0] / size[0] - tmp[0], new_end[1] / size[1] - tmp[1])
# r.SetMoving(1)
# r.BuildRepresentation()
# borderWidget.SetRepresentation(r)
# borderWidget.SetInteractor(interactor)
# # borderWidget.ResizableOff()
# # borderWidget.CreateDefaultRepresentation()
# # borderWidget.SelectableOn()
# borderWidget.On()
# print(r)
# renderWindow.Render()
# widgets.append(borderWidget)
# style.AddObserver("SelectionChangedEvent", stylecallback)
interactor.SetInteractorStyle(style)
renderer.AddActor(actor)
renderer.AddActor(img_actor.actor)
interactor.Initialize()
renderWindow.Render()
# borderWidget.On()
interactor.Start()
