WindowLevelInterface.py from matplotlib at Krugle
Show WindowLevelInterface.py syntax highlighted
# a simple user interface that manipulates window level.
import os
try:
VTK_DATA = os.environ['VTK_DATA']
except KeyError:
VTK_DATA = '../../../vtkdata/'
# places in the Tkinter top window.
from Tkinter import *
import sys, string
VTK_VOID = 0
VTK_BIT = 1
VTK_CHAR = 2
VTK_UNSIGNED_CHAR = 3
VTK_SHORT = 4
VTK_UNSIGNED_SHORT = 5
VTK_INT = 6
VTK_UNSIGNED_INT = 7
VTK_LONG = 8
VTK_UNSIGNED_LONG = 9
VTK_FLOAT = 10
VTK_DOUBLE = 11
VTK_IMAGE_X_AXIS = 0
VTK_IMAGE_Y_AXIS = 1
VTK_IMAGE_Z_AXIS = 2
VTK_CLAW_NEAREST_NETWORK = 0
VTK_CLAW_NEAREST_MINIMUM = 1
VTK_CLAW_NEAREST_GLOBAL = 2
VTK_CLAW_PIONEER_LOCAL = 3
VTK_CLAW_PIONEER_GLOBAL = 4
VTK_CLAW_WELL_NOISE = 6
VTK_CLAW_WELL_DIRECTED_NOISE= 7
VTK_CLAW_MINIMUM_WELL = 8
VTK_CLAW_INSERT = 9
VTK_CLAW_NARROW_WEL = 10
class WindowLevelInterface(Frame):
def __init__(self,viewer=None,**kw):
apply(Frame.__init__,(self,),kw)
if (viewer == None):
print "Must specify a vtkImageViewer"
self.viewer = viewer
w = viewer.GetColorWindow()
l = viewer.GetColorLevel()
sliceNumber = viewer.GetZSlice()
zMin = viewer.GetWholeZMin()
zMax = viewer.GetWholeZMax()
self.slice = Frame(self)
self.slice_label = Label(self.slice,text="Slice")
self.slice_scale = Scale(self.slice,from_=zMin,to=zMax, \
orient='horizontal', \
command=self.SetSlice, \
variable='sliceNumber')
self.wl = Frame(self)
self.wl_f1 = Frame(self.wl)
self.wl_f1_windowLabel = Label(self.wl_f1,text="Window")
self.wl_f1_window = Scale(self.wl_f1,from_=1,to=w*2, \
orient='horizontal', \
command=self.SetWindow, \
variable='window')
self.wl_f2 = Frame(self.wl)
self.wl_f2_levelLabel = Label(self.wl_f2,text="Level")
self.wl_f2_level = Scale(self.wl_f2,from_=l-w,to=l+w, \
orient='horizontal', \
command=self.SetLevel, \
variable='level')
self.wl_video = Checkbutton(self.wl,text="Inverse Video", \
command=self.SetInverseVideo,\
variable='video')
# resolutions less than 1.0
if (w < 20):
res = 0.05*w
self.wl_f1_window.configure(resolution=res,\
from_=res,to=2.0*2)
self.wl_f2_level.configure(resolution=res,\
from_=1.0*(l-w),to=1.0*(l+w))
self.slice_scale.set(sliceNumber)
self.wl_f1_window.set(w)
self.wl_f2_level.set(l)
self.ex = Frame(self)
self.ex_exit = Button(self.ex,text="Exit",\
command=sys.exit)
self.slice.pack(side='top')
self.wl.pack(side='top')
self.ex.pack(side='top')
self.slice_label.pack(side='left')
self.slice_scale.pack(side='left')
self.wl_f1.pack(side='top')
self.wl_f2.pack(side='top')
self.wl_video.pack(side='top')
self.wl_f1_windowLabel.pack(side='left')
self.wl_f1_window.pack(side='left')
self.wl_f2_levelLabel.pack(side='left')
self.wl_f2_level.pack(side='left')
self.ex_exit.pack(side='left')
self.pack()
self.mainloop()
def SetSlice(self,slice):
self.viewer.SetZSlice(string.atoi(slice))
self.viewer.Render()
def SetWindow(self,window):
if string.atoi(self.getvar('video')):
self.viewer.SetColorWindow(-string.atof(window))
else:
self.viewer.SetColorWindow(string.atof(window))
self.viewer.Render()
def SetLevel(self,level):
self.viewer.SetColorLevel(string.atof(level))
self.viewer.Render()
def SetInverseVideo(self):
if string.atoi(self.getvar('video')):
self.viewer.SetColorWindow(-string.atof(self.getvar('window')))
else:
self.viewer.SetColorWindow(string.atof(self.getvar('window')))
self.viewer.Render()
See more files for this project here