def __init__(self, pool=None):
super().__init__()
# self.pool = pool
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.sol_elements = None
self.basepath = None
self.image = None
self.alt_image = None
self.lut = None
self.viewer_center = (0,0,0)
### default LUT (in QIA format) ###
# self.roi_lut = ipplut.new({
# 0: (0,0,0,0),
# 1: (0,1,0,self.ui.opacity.value()/100),
# "table": (0,1)
# })
self.roi = None
### enable all of these when ready
self.views = (
self.ui.axial,
self.ui.coronal,
# self.ui.sagittal,
# self.ui.three_d
)
# connecting functionalities
for i, v in enumerate(self.views):
print("connect try")
v.location_clicked = self.location_clicked
v.update_centers = self.update_centers
# v.update_window_level = self.update_window_level ### Disabled for now because it's not working
print("connected")
v.viewer_id = i
### Connect more things ###
# for v in self.views:
# for i in [q for q in self.views if q!=v]:
# v.positionChanged.connect(i.setPosition)
# v.zoomed.connect(i.setZoom)
# if v!=self.ui.three_d and i!=self.ui.three_d:
# v.cameraReset.connect(i.resetCamera)
self.show_crosshair(self.ui.show_crosshair.isChecked())
### Initializing opacity
self.opacity_changed(0) # what is the zero there for
self.ui.actionRefresh.setEnabled(False)
self.loading_solution_file = False
self.render_progress = QProgressBar(self)
self.render_progress.setFormat("Rendering %p%")
# self.ui.three_d.progress.connect(self.render_progress.setValue) #temp disabled
self.ui.statusbar.addPermanentWidget(self.render_progress)
# global window level to make it all the same
self.window_level = vtkImageMapToWindowLevelColors()
def ShowSlice(self, index = 0):
bitmap = self.bitmap_list[index]
# UPDATE GUI
## Text related to size
value = STR_SIZE %(bitmap[3], bitmap[4])
self.text_image_size.SetValue(value)
value1 = ''
value2 = ''
value = "%s\n%s" %(value1, value2)
self.text_image_location.SetValue(value)
#self.text_patient.SetValue(value)
self.text_patient.SetValue('')
#self.text_acquisition.SetValue(value)
self.text_acquisition.SetValue('')
n_array = bitmap_reader.ReadBitmap(bitmap[0])
image = converters.to_vtk(n_array, spacing=(1,1,1),\
slice_number=1, orientation="AXIAL")
# ADJUST CONTRAST
window_level = n_array.max()/2
window_width = n_array.max()
colorer = vtkImageMapToWindowLevelColors()
colorer.SetInputData(image)
colorer.SetWindow(float(window_width))
colorer.SetLevel(float(window_level))
colorer.Update()
if self.actor is None:
self.actor = vtkImageActor()
self.renderer.AddActor(self.actor)
# PLOT IMAGE INTO VIEWER
self.actor.SetInputData(colorer.GetOutput())
self.renderer.ResetCamera()
self.interactor.Render()
# Setting slider position
self.slider.SetValue(index)
def ShowBlackSlice(self, pub_sub):
n_array = numpy.zeros((100,100))
self.text_image_size.SetValue('')
image = converters.to_vtk(n_array, spacing=(1,1,1),\
slice_number=1, orientation="AXIAL")
colorer = vtkImageMapToWindowLevelColors()
colorer.SetInputData(image)
colorer.Update()
if self.actor is None:
self.actor = vtkImageActor()
self.renderer.AddActor(self.actor)
# PLOT IMAGE INTO VIEWER
self.actor.SetInputData(colorer.GetOutput())
self.renderer.ResetCamera()
self.interactor.Render()
# Setting slider position
self.slider.SetValue(0)
def ShowSlice(self, index=0):
try:
dicom = self.dicom_list[index]
except IndexError:
dicom = self.dicom_list[0]
if self.actor is None:
self.__init_vtk()
# UPDATE GUI
## Text related to size
value = STR_SIZE % (dicom.image.size[0], dicom.image.size[1])
self.text_image_size.SetValue(value)
## Text related to slice position
if not (dicom.image.spacing):
value1 = ''
else:
value1 = STR_SPC % (dicom.image.spacing[2])
if dicom.image.orientation_label == 'AXIAL':
value2 = STR_LOCAL % (dicom.image.position[2])
elif dicom.image.orientation_label == 'CORONAL':
value2 = STR_LOCAL % (dicom.image.position[1])
elif dicom.image.orientation_label == 'SAGITTAL':
value2 = STR_LOCAL % (dicom.image.position[0])
else:
value2 = ''
value = "%s\n%s" % (value1, value2)
self.text_image_location.SetValue(value)
## Text related to patient/ acquisiiton data
value = STR_PATIENT %(dicom.patient.id,\
dicom.acquisition.protocol_name)
self.text_patient.SetValue(value)
## Text related to acquisition date and time
value = STR_ACQ % (dicom.acquisition.date, dicom.acquisition.time)
self.text_acquisition.SetValue(value)
if isinstance(dicom.image.thumbnail_path, list):
reader = vtkPNGReader()
if _has_win32api:
reader.SetFileName(
win32api.GetShortPathName(
dicom.image.thumbnail_path[index]).encode(
const.FS_ENCODE))
else:
reader.SetFileName(dicom.image.thumbnail_path[index])
reader.Update()
image = reader.GetOutput()
else:
filename = dicom.image.file
if _has_win32api:
filename = win32api.GetShortPathName(filename).encode(
const.FS_ENCODE)
np_image = imagedata_utils.read_dcm_slice_as_np2(filename)
vtk_image = converters.to_vtk(np_image, dicom.image.spacing, 0,
'AXIAL')
# ADJUST CONTRAST
window_level = dicom.image.level
window_width = dicom.image.window
colorer = vtkImageMapToWindowLevelColors()
colorer.SetInputData(vtk_image)
colorer.SetWindow(float(window_width))
colorer.SetLevel(float(window_level))
colorer.Update()
image = colorer.GetOutput()
flip = vtkImageFlip()
flip.SetInputData(image)
flip.SetFilteredAxis(1)
flip.FlipAboutOriginOn()
flip.ReleaseDataFlagOn()
flip.Update()
if self.actor is None:
self.actor = vtkImageActor()
self.renderer.AddActor(self.actor)
self.canvas.modified = True
# PLOT IMAGE INTO VIEWER
self.actor.SetInputData(flip.GetOutput())
self.renderer.ResetCamera()
self.interactor.Render()
# Setting slider position
self.slider.SetValue(index)
def __init__(self, *args, **kwargs):
super(PyView2D, self).__init__(*args, **kwargs)
self.block_signal = False
self.reader = vtkDICOMImageReader()
self.reader2 = vtkDICOMImageReader()
self.window = self.GetRenderWindow()
self.window.BordersOff(
) #attempt to remove imageactor borders, can remove
self.center = [0, 0, 0]
self.current_image = vtk.vtkImageData()
self.viewer_id = "default"
self.viewer_setup() # flexible for future
self.image_loaded = False
self.viewer_initialized = False
# self.read_image(image_path=None, orientation="axial") # flexible for future
self.window_level = vtkImageMapToWindowLevelColors()
self.img = self.map_img()
self.roi = self.map_roi()
self.px_coord_text_prop = vtkTextProperty()
self.px_coord_text_mapper = vtkTextMapper()
self.px_coord_text_actor = vtkActor2D()
self.world_coord_text_prop = vtkTextProperty()
self.world_coord_text_mapper = vtkTextMapper()
self.world_coord_text_actor = vtkActor2D()
self.usage_text_prop = vtkTextProperty()
self.usage_text_mapper = vtkTextMapper()
self.usage_text_actor = vtkActor2D()
self.renderer = vtkRenderer() #
self.add_text()
self.renderer.AddActor(self.img) #
self.renderer.AddActor(self.roi) #
self.renderer.SetBackground(0.2, 0.3, 0.4) #
### Wasil added ###
#QVTKRenderWindowInteractor relays Qt events to VTK
# self.frame = Qt.QFrame() # do i need this
# self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
# miu_viewer = PyView2D()
# vtkWidget has its own RenderWindow
# self.ren = vtk.vtkRenderer()
# or
# self.ren = miu_viewer.renderer # should have the actors already
# self.vtkWidget.GetRenderWindow().AddRenderer(self.renderer)
# self.vtkWidget.GetRenderWindow().SetSize(1000, 1000)
# miu_viewer.window.Render()
# or
# self.vtkWidget.GetRenderWindow().Render()
##########################
# have vtkRenderWindow
# add VTKrenderer
# self.window = vtkRenderWindow()
self.window.AddRenderer(self.renderer) #
# self.window.SetSize(1000, 1000)
self.window.SetSize(300, 300)
self.interactor_style = vtkInteractorStyleImage()
self.interactor_style.SetInteractionModeToImageSlicing()
# self.interactor = vtkRenderWindowInteractor() #
# self.iren is the interactor (also from RenderWindow)
self.interactor = self.GetRenderWindow().GetInteractor()
self.interactor.SetInteractorStyle(self.interactor_style) #
self.window.SetInteractor(self.interactor) #
self.window_level.SetWindow(1000)
self.window_level.SetLevel(200)
self.window_level.Update()
self.window.Render()
### moved this down to after image is loaded
# self.interactor_style.AddObserver("MouseWheelForwardEvent", self.scroll_forward_callback)
# self.interactor_style.AddObserver("MouseWheelBackwardEvent", self.scroll_backward_callback)
# self.interactor_style.AddObserver("MouseMoveEvent", self.mouse_move_callback)
# self.interactor_style.AddObserver("KeyPressEvent", self.key_press_callback)
# self.interactor_style.AddObserver("LeftButtonPressEvent", self.left_press_callback)
# self.window.AddObserver("ModifiedEvent", self.window_mod_callback)
self.actions = {
"Slicing": 0,
"Cursor": 0,
"CurrentPos": -1,
"LastPos": -1,
"DoubleClick": 0
}
table.SetTableValue(1, 0.0, 1.0, 0.0, 1.0) # value, r, g, b, a
table.Build()
# Create a greyscale lookup table
table2 = vtkLookupTable()
table2.SetRange(0, 2000) # image intensity range
table2.SetValueRange(0.0, 1.0) # from black to white
table2.SetSaturationRange(0.0, 0.0) # no color saturation
table2.SetRampToSQRT()
table2.Build()
# Map the image through the lookup table
print("mapping colors...")
windowLevel = vtkImageMapToWindowLevelColors()
windowLevel.SetInputConnection(reslice2.GetOutputPort())
color = vtkImageMapToColors()
color.SetLookupTable(table)
color.SetInputConnection(reslice.GetOutputPort())
color2 = vtkImageMapToColors()
color2.SetLookupTable(table2)
color2.SetInputConnection(windowLevel.GetOutputPort())
# Display the image
actor = vtkImageActor()
actor.GetMapper().SetInputConnection(color.GetOutputPort())
original = vtkImageActor()
def __init__(self):
VTK_DATA_ROOT = vtkGetDataRoot()
self.reader = vtkDICOMImageReader()
self.folder = "/Users/nandana/Downloads/image_ex"
self.reader.SetDirectoryName(self.folder)
self.reader.SetFilePrefix(VTK_DATA_ROOT + "/Data/headsq/quarter")
self.reader.SetDataExtent(0, 63, 0, 63, 1, 93)
self.reader.SetDataSpacing(3.2, 3.2, 1.5)
self.reader.SetDataOrigin(0.0, 0.0, 0.0)
self.reader.SetDataScalarTypeToUnsignedShort()
self.reader.UpdateWholeExtent()
self.reader.Update()
self.center = self.calculate_center()
self.axial = vtkMatrix4x4()
self.axial.DeepCopy((1, 0, 0, self.center[0], 0, 1, 0, self.center[1],
0, 0, 1, self.center[2], 0, 0, 0, 1))
self.coronal = vtkMatrix4x4()
self.coronal.DeepCopy(
(1, 0, 0, self.center[0], 0, 0, 1, self.center[1], 0, -1, 0,
self.center[2], 0, 0, 0, 1))
self.sagittal = vtkMatrix4x4()
self.sagittal.DeepCopy(
(0, 0, -1, self.center[0], 1, 0, 0, self.center[1], 0, -1, 0,
self.center[2], 0, 0, 0, 1))
self.oblique = vtkMatrix4x4()
self.oblique.DeepCopy(
(1, 0, 0, self.center[0], 0, 0.866025, -0.5, self.center[1], 0,
0.5, 0.866025, self.center[2], 0, 0, 0, 1))
self.img_reslice = vtkImageReslice()
self.roi_reslice = vtkImageReslice()
self.img_reslice.SetInputConnection(0, self.reader.GetOutputPort())
self.img_reslice.SetOutputDimensionality(2)
self.img_reslice.SetInterpolationModeToLinear()
self.roi_reslice.SetInputConnection(0, self.reader.GetOutputPort())
self.roi_reslice.SetOutputDimensionality(2)
self.roi_reslice.SetInterpolationModeToLinear()
self.set_orientation(self.axial)
self.img_table = vtkLookupTable()
self.roi_table = vtkLookupTable()
self.window_level = vtkImageMapToWindowLevelColors()
self.img_color = vtkImageMapToColors()
self.roi_color = vtkImageMapToColors()
self.img = self.map_img()
self.roi = self.map_roi()
self.px_coord_text_prop = vtkTextProperty()
self.px_coord_text_mapper = vtkTextMapper()
self.px_coord_text_actor = vtkActor2D()
self.world_coord_text_prop = vtkTextProperty()
self.world_coord_text_mapper = vtkTextMapper()
self.world_coord_text_actor = vtkActor2D()
self.usage_text_prop = vtkTextProperty()
self.usage_text_mapper = vtkTextMapper()
self.usage_text_actor = vtkActor2D()
self.renderer = vtkRenderer()
self.add_text()
self.renderer.AddActor(self.img)
self.renderer.AddActor(self.roi)
self.renderer.SetBackground(0.2, 0.3, 0.4)
self.window = vtkRenderWindow()
self.window.AddRenderer(self.renderer)
self.window.SetSize(1000, 1000)
self.interactor_style = vtkInteractorStyleImage()
self.interactor_style.SetInteractionModeToImageSlicing()
self.interactor = vtkRenderWindowInteractor()
self.interactor.SetInteractorStyle(self.interactor_style)
self.window.SetInteractor(self.interactor)
self.window_level.SetWindow(1000)
self.window_level.SetLevel(200)
self.window_level.Update()
self.window.Render()
self.interactor_style.AddObserver("MouseWheelForwardEvent",
self.scroll_forward_callback)
self.interactor_style.AddObserver("MouseWheelBackwardEvent",
self.scroll_backward_callback)
self.interactor_style.AddObserver("MouseMoveEvent",
self.mouse_move_callback)
self.interactor_style.AddObserver("KeyPressEvent",
self.key_press_callback)
self.interactor_style.AddObserver("LeftButtonPressEvent",
self.left_press_callback)
self.window.AddObserver("ModifiedEvent", self.window_mod_callback)
self.actions = {
"Slicing": 0,
"Cursor": 0,
"CurrentPos": -1,
"LastPos": -1,
"DoubleClick": 0
}