import vtk
reader = vtk.vtkDICOMImageReader()
reader.SetFileName("/media/WDC/Dropbox/code/pCT/IM28")


viewer = vtk.vtkImageViewer2()
viewer.SetInputConnection(reader.GetOutputPort())
interactor = vtk.vtkRenderWindowInteractor()

viewer.SetupInteractor(interactor)

#viewer.SetZSlice(0)
viewer.SetColorWindow(2000)
viewer.SetColorLevel(1000)
tracer = vtk.vtkImageTracerWidget()
tracer.SetViewProp(viewer.GetImageActor())
tracer.SetInteractor(interactor)
tracer.AutoCloseOn()
tracer.ProjectToPlaneOn()
tracer.PlaceWidget(0, 50, 50, 100, 0 ,0)
# viewer.GetRenderer().ResetCamera()
viewer.Render()
interactor.Initialize()
tracer.On()
interactor.Start()

outp = vtk.vtkPolyData()
tracer.GetPath(outp)


# creating input source for polydata -> stencil
p2s = vtk.vtkPolyDataToImageStencil()
p2s.SetInputData(outp)
p2s.SetOutputSpacing(reader.GetDataSpacing())
p2s.SetOutputOrigin(reader.GetDataOrigin())
p2s.Update()


stencil = vtk.vtkImageStencil()
stencil.SetStencilConnection(p2s.GetOutputPort())
stencil.SetInputData(reader.GetOutput())

stencil.SetBackgroundColor(1,1,0,1)
stencil.SetBackgroundValue(0)
stencil.Update()

stats = vtk.vtkImageHistogramStatistics()
stats.SetInputConnection(stencil.GetOutputPort())
stats.GenerateHistogramImageOn()
stats.Update()
stats.GetMean()
stats.GetMaximum()
stats.GetMinimum()

viewer2 = vtk.vtkImageViewer()
interator = vtk.vtkRenderWindowInteractor()
viewer2.SetInputConnection(stencil.GetOutputPort())
viewer2.SetupInteractor(interator)
viewer2.SetZSlice(0)
viewer2.SetColorWindow(2000)
viewer2.SetColorLevel(1000)
viewer2.Render()