summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2017-08-23 15:02:13 +0100
committerEdoardo Pasca <edo.paskino@gmail.com>2017-08-23 15:02:13 +0100
commitda355c4292003f0218a7862be94036d09f6f221d (patch)
tree047a9ac6d1e5afab7f7c129ad21f15ddd52d5fe4
parent111becf92752bf9b9cb6dfe5c3780953570b054c (diff)
parent105b57a1e98c2bb7b3bf94c43b6c669925ebb1b9 (diff)
downloadregularization-da355c4292003f0218a7862be94036d09f6f221d.tar.gz
regularization-da355c4292003f0218a7862be94036d09f6f221d.tar.bz2
regularization-da355c4292003f0218a7862be94036d09f6f221d.tar.xz
regularization-da355c4292003f0218a7862be94036d09f6f221d.zip
Merge branch 'pythonize' of https://github.com/vais-ral/CCPi-FISTA_Reconstruction into pythonize
-rw-r--r--src/Python/ccpi/viewer/CILViewer.py361
-rw-r--r--src/Python/ccpi/viewer/CILViewer2D.py1126
-rw-r--r--src/Python/ccpi/viewer/QVTKWidget.py340
-rw-r--r--src/Python/ccpi/viewer/QVTKWidget2.py84
-rw-r--r--src/Python/ccpi/viewer/__init__.py1
-rw-r--r--src/Python/ccpi/viewer/__pycache__/CILViewer.cpython-35.pycbin0 -> 10542 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/CILViewer2D.cpython-35.pycbin0 -> 35633 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/QVTKWidget.cpython-35.pycbin0 -> 10099 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/QVTKWidget2.cpython-35.pycbin0 -> 1316 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/__init__.cpython-35.pycbin0 -> 210 bytes
-rw-r--r--src/Python/ccpi/viewer/embedvtk.py75
11 files changed, 1987 insertions, 0 deletions
diff --git a/src/Python/ccpi/viewer/CILViewer.py b/src/Python/ccpi/viewer/CILViewer.py
new file mode 100644
index 0000000..efcf8be
--- /dev/null
+++ b/src/Python/ccpi/viewer/CILViewer.py
@@ -0,0 +1,361 @@
+# -*- coding: utf-8 -*-
+# Copyright 2017 Edoardo Pasca
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import vtk
+import numpy
+import math
+from vtk.util import numpy_support
+
+SLICE_ORIENTATION_XY = 2 # Z
+SLICE_ORIENTATION_XZ = 1 # Y
+SLICE_ORIENTATION_YZ = 0 # X
+
+
+
+class CILViewer():
+ '''Simple 3D Viewer based on VTK classes'''
+
+ def __init__(self, dimx=600,dimy=600):
+ '''creates the rendering pipeline'''
+
+ # create a rendering window and renderer
+ self.ren = vtk.vtkRenderer()
+ self.renWin = vtk.vtkRenderWindow()
+ self.renWin.SetSize(dimx,dimy)
+ self.renWin.AddRenderer(self.ren)
+
+ # img 3D as slice
+ self.img3D = None
+ self.sliceno = 0
+ self.sliceOrientation = SLICE_ORIENTATION_XY
+ self.sliceActor = None
+ self.voi = None
+ self.wl = None
+ self.ia = None
+ self.sliceActorNo = 0
+ # create a renderwindowinteractor
+ self.iren = vtk.vtkRenderWindowInteractor()
+ self.iren.SetRenderWindow(self.renWin)
+
+ self.style = vtk.vtkInteractorStyleTrackballCamera()
+ self.iren.SetInteractorStyle(self.style)
+
+ self.ren.SetBackground(.1, .2, .4)
+
+ self.actors = {}
+ self.iren.RemoveObservers('MouseWheelForwardEvent')
+ self.iren.RemoveObservers('MouseWheelBackwardEvent')
+
+ self.iren.AddObserver('MouseWheelForwardEvent', self.mouseInteraction, 1.0)
+ self.iren.AddObserver('MouseWheelBackwardEvent', self.mouseInteraction, 1.0)
+
+ self.iren.RemoveObservers('KeyPressEvent')
+ self.iren.AddObserver('KeyPressEvent', self.keyPress, 1.0)
+
+
+ self.iren.Initialize()
+
+
+
+ def getRenderer(self):
+ '''returns the renderer'''
+ return self.ren
+
+ def getRenderWindow(self):
+ '''returns the render window'''
+ return self.renWin
+
+ def getInteractor(self):
+ '''returns the render window interactor'''
+ return self.iren
+
+ def getCamera(self):
+ '''returns the active camera'''
+ return self.ren.GetActiveCamera()
+
+ def createPolyDataActor(self, polydata):
+ '''returns an actor for a given polydata'''
+ mapper = vtk.vtkPolyDataMapper()
+ if vtk.VTK_MAJOR_VERSION <= 5:
+ mapper.SetInput(polydata)
+ else:
+ mapper.SetInputData(polydata)
+
+ # actor
+ actor = vtk.vtkActor()
+ actor.SetMapper(mapper)
+ #actor.GetProperty().SetOpacity(0.8)
+ return actor
+
+ def setPolyDataActor(self, actor):
+ '''displays the given polydata'''
+
+ self.ren.AddActor(actor)
+
+ self.actors[len(self.actors)+1] = [actor, True]
+ self.iren.Initialize()
+ self.renWin.Render()
+
+ def displayPolyData(self, polydata):
+ self.setPolyDataActor(self.createPolyDataActor(polydata))
+
+ def hideActor(self, actorno):
+ '''Hides an actor identified by its number in the list of actors'''
+ try:
+ if self.actors[actorno][1]:
+ self.ren.RemoveActor(self.actors[actorno][0])
+ self.actors[actorno][1] = False
+ except KeyError as ke:
+ print ("Warning Actor not present")
+
+ def showActor(self, actorno, actor = None):
+ '''Shows hidden actor identified by its number in the list of actors'''
+ try:
+ if not self.actors[actorno][1]:
+ self.ren.AddActor(self.actors[actorno][0])
+ self.actors[actorno][1] = True
+ return actorno
+ except KeyError as ke:
+ # adds it to the actors if not there already
+ if actor != None:
+ self.ren.AddActor(actor)
+ self.actors[len(self.actors)+1] = [actor, True]
+ return len(self.actors)
+
+ def addActor(self, actor):
+ '''Adds an actor to the render'''
+ return self.showActor(0, actor)
+
+
+ def saveRender(self, filename, renWin=None):
+ '''Save the render window to PNG file'''
+ # screenshot code:
+ w2if = vtk.vtkWindowToImageFilter()
+ if renWin == None:
+ renWin = self.renWin
+ w2if.SetInput(renWin)
+ w2if.Update()
+
+ writer = vtk.vtkPNGWriter()
+ writer.SetFileName("%s.png" % (filename))
+ writer.SetInputConnection(w2if.GetOutputPort())
+ writer.Write()
+
+
+ def startRenderLoop(self):
+ self.iren.Start()
+
+
+ def setupObservers(self, interactor):
+ interactor.RemoveObservers('LeftButtonPressEvent')
+ interactor.AddObserver('LeftButtonPressEvent', self.mouseInteraction)
+ interactor.Initialize()
+
+
+ def mouseInteraction(self, interactor, event):
+ if event == 'MouseWheelForwardEvent':
+ maxSlice = self.img3D.GetDimensions()[self.sliceOrientation]
+ if (self.sliceno + 1 < maxSlice):
+ self.hideActor(self.sliceActorNo)
+ self.sliceno = self.sliceno + 1
+ self.displaySliceActor(self.sliceno)
+ else:
+ minSlice = 0
+ if (self.sliceno - 1 > minSlice):
+ self.hideActor(self.sliceActorNo)
+ self.sliceno = self.sliceno - 1
+ self.displaySliceActor(self.sliceno)
+
+
+ def keyPress(self, interactor, event):
+ #print ("Pressed key %s" % interactor.GetKeyCode())
+ # Slice Orientation
+ if interactor.GetKeyCode() == "x":
+ # slice on the other orientation
+ self.sliceOrientation = SLICE_ORIENTATION_YZ
+ self.sliceno = int(self.img3D.GetDimensions()[1] / 2)
+ self.hideActor(self.sliceActorNo)
+ self.displaySliceActor(self.sliceno)
+ elif interactor.GetKeyCode() == "y":
+ # slice on the other orientation
+ self.sliceOrientation = SLICE_ORIENTATION_XZ
+ self.sliceno = int(self.img3D.GetDimensions()[1] / 2)
+ self.hideActor(self.sliceActorNo)
+ self.displaySliceActor(self.sliceno)
+ elif interactor.GetKeyCode() == "z":
+ # slice on the other orientation
+ self.sliceOrientation = SLICE_ORIENTATION_XY
+ self.sliceno = int(self.img3D.GetDimensions()[2] / 2)
+ self.hideActor(self.sliceActorNo)
+ self.displaySliceActor(self.sliceno)
+ if interactor.GetKeyCode() == "X":
+ # Change the camera view point
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.ren.GetActiveCamera().GetFocalPoint())
+ camera.SetViewUp(self.ren.GetActiveCamera().GetViewUp())
+ newposition = [i for i in self.ren.GetActiveCamera().GetFocalPoint()]
+ newposition[SLICE_ORIENTATION_YZ] = math.sqrt(newposition[SLICE_ORIENTATION_XY] ** 2 + newposition[SLICE_ORIENTATION_XZ] ** 2)
+ camera.SetPosition(newposition)
+ camera.SetViewUp(0,0,-1)
+ self.ren.SetActiveCamera(camera)
+ self.ren.ResetCamera()
+ self.ren.Render()
+ interactor.SetKeyCode("x")
+ self.keyPress(interactor, event)
+ elif interactor.GetKeyCode() == "Y":
+ # Change the camera view point
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.ren.GetActiveCamera().GetFocalPoint())
+ camera.SetViewUp(self.ren.GetActiveCamera().GetViewUp())
+ newposition = [i for i in self.ren.GetActiveCamera().GetFocalPoint()]
+ newposition[SLICE_ORIENTATION_XZ] = math.sqrt(newposition[SLICE_ORIENTATION_XY] ** 2 + newposition[SLICE_ORIENTATION_YZ] ** 2)
+ camera.SetPosition(newposition)
+ camera.SetViewUp(0,0,-1)
+ self.ren.SetActiveCamera(camera)
+ self.ren.ResetCamera()
+ self.ren.Render()
+ interactor.SetKeyCode("y")
+ self.keyPress(interactor, event)
+ elif interactor.GetKeyCode() == "Z":
+ # Change the camera view point
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.ren.GetActiveCamera().GetFocalPoint())
+ camera.SetViewUp(self.ren.GetActiveCamera().GetViewUp())
+ newposition = [i for i in self.ren.GetActiveCamera().GetFocalPoint()]
+ newposition[SLICE_ORIENTATION_XY] = math.sqrt(newposition[SLICE_ORIENTATION_YZ] ** 2 + newposition[SLICE_ORIENTATION_XZ] ** 2)
+ camera.SetPosition(newposition)
+ camera.SetViewUp(0,0,-1)
+ self.ren.SetActiveCamera(camera)
+ self.ren.ResetCamera()
+ self.ren.Render()
+ interactor.SetKeyCode("z")
+ self.keyPress(interactor, event)
+ else :
+ print ("Unhandled event %s" % interactor.GetKeyCode())
+
+
+
+ def setInput3DData(self, imageData):
+ self.img3D = imageData
+
+ def setInputAsNumpy(self, numpyarray):
+ if (len(numpy.shape(numpyarray)) == 3):
+ doubleImg = vtk.vtkImageData()
+ shape = numpy.shape(numpyarray)
+ doubleImg.SetDimensions(shape[0], shape[1], shape[2])
+ doubleImg.SetOrigin(0,0,0)
+ doubleImg.SetSpacing(1,1,1)
+ doubleImg.SetExtent(0, shape[0]-1, 0, shape[1]-1, 0, shape[2]-1)
+ #self.img3D.SetScalarType(vtk.VTK_UNSIGNED_SHORT, vtk.vtkInformation())
+ doubleImg.AllocateScalars(vtk.VTK_DOUBLE,1)
+
+ for i in range(shape[0]):
+ for j in range(shape[1]):
+ for k in range(shape[2]):
+ doubleImg.SetScalarComponentFromDouble(
+ i,j,k,0, numpyarray[i][j][k])
+ #self.setInput3DData( numpy_support.numpy_to_vtk(numpyarray) )
+ # rescale to appropriate VTK_UNSIGNED_SHORT
+ stats = vtk.vtkImageAccumulate()
+ stats.SetInputData(doubleImg)
+ stats.Update()
+ iMin = stats.GetMin()[0]
+ iMax = stats.GetMax()[0]
+ scale = vtk.VTK_UNSIGNED_SHORT_MAX / (iMax - iMin)
+
+ shiftScaler = vtk.vtkImageShiftScale ()
+ shiftScaler.SetInputData(doubleImg)
+ shiftScaler.SetScale(scale)
+ shiftScaler.SetShift(iMin)
+ shiftScaler.SetOutputScalarType(vtk.VTK_UNSIGNED_SHORT)
+ shiftScaler.Update()
+ self.img3D = shiftScaler.GetOutput()
+
+ def displaySliceActor(self, sliceno = 0):
+ self.sliceno = sliceno
+ first = False
+
+ self.sliceActor , self.voi, self.wl , self.ia = \
+ self.getSliceActor(self.img3D,
+ sliceno,
+ self.sliceActor,
+ self.voi,
+ self.wl,
+ self.ia)
+ no = self.showActor(self.sliceActorNo, self.sliceActor)
+ self.sliceActorNo = no
+
+ self.iren.Initialize()
+ self.renWin.Render()
+
+ return self.sliceActorNo
+
+
+ def getSliceActor(self,
+ imageData ,
+ sliceno=0,
+ imageActor=None ,
+ voi=None,
+ windowLevel=None,
+ imageAccumulate=None):
+ '''Slices a 3D volume and then creates an actor to be rendered'''
+ if (voi==None):
+ voi = vtk.vtkExtractVOI()
+ #voi = vtk.vtkImageClip()
+ voi.SetInputData(imageData)
+ #select one slice in Z
+ extent = [ i for i in self.img3D.GetExtent()]
+ extent[self.sliceOrientation * 2] = sliceno
+ extent[self.sliceOrientation * 2 + 1] = sliceno
+ voi.SetVOI(extent[0], extent[1],
+ extent[2], extent[3],
+ extent[4], extent[5])
+
+ voi.Update()
+ # set window/level for all slices
+ if imageAccumulate == None:
+ imageAccumulate = vtk.vtkImageAccumulate()
+
+ if (windowLevel == None):
+ windowLevel = vtk.vtkImageMapToWindowLevelColors()
+ imageAccumulate.SetInputData(imageData)
+ imageAccumulate.Update()
+ cmax = imageAccumulate.GetMax()[0]
+ cmin = imageAccumulate.GetMin()[0]
+ windowLevel.SetLevel((cmax+cmin)/2)
+ windowLevel.SetWindow(cmax-cmin)
+
+ windowLevel.SetInputData(voi.GetOutput())
+ windowLevel.Update()
+
+ if imageActor == None:
+ imageActor = vtk.vtkImageActor()
+ imageActor.SetInputData(windowLevel.GetOutput())
+ imageActor.SetDisplayExtent(extent[0], extent[1],
+ extent[2], extent[3],
+ extent[4], extent[5])
+ imageActor.Update()
+ return (imageActor , voi, windowLevel, imageAccumulate)
+
+
+ # Set interpolation on
+ def setInterpolateOn(self):
+ self.sliceActor.SetInterpolate(True)
+ self.renWin.Render()
+
+ # Set interpolation off
+ def setInterpolateOff(self):
+ self.sliceActor.SetInterpolate(False)
+ self.renWin.Render() \ No newline at end of file
diff --git a/src/Python/ccpi/viewer/CILViewer2D.py b/src/Python/ccpi/viewer/CILViewer2D.py
new file mode 100644
index 0000000..c1629af
--- /dev/null
+++ b/src/Python/ccpi/viewer/CILViewer2D.py
@@ -0,0 +1,1126 @@
+# -*- coding: utf-8 -*-
+# Copyright 2017 Edoardo Pasca
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import vtk
+import numpy
+from vtk.util import numpy_support , vtkImageImportFromArray
+from enum import Enum
+
+SLICE_ORIENTATION_XY = 2 # Z
+SLICE_ORIENTATION_XZ = 1 # Y
+SLICE_ORIENTATION_YZ = 0 # X
+
+CONTROL_KEY = 8
+SHIFT_KEY = 4
+ALT_KEY = -128
+
+
+# Converter class
+class Converter():
+
+ # Utility functions to transform numpy arrays to vtkImageData and viceversa
+ @staticmethod
+ def numpy2vtkImporter(nparray, spacing=(1.,1.,1.), origin=(0,0,0)):
+ '''Creates a vtkImageImportFromArray object and returns it.
+
+ It handles the different axis order from numpy to VTK'''
+ importer = vtkImageImportFromArray.vtkImageImportFromArray()
+ importer.SetArray(numpy.transpose(nparray).copy())
+ importer.SetDataSpacing(spacing)
+ importer.SetDataOrigin(origin)
+ return importer
+
+ @staticmethod
+ def numpy2vtk(nparray, spacing=(1.,1.,1.), origin=(0,0,0)):
+ '''Converts a 3D numpy array to a vtkImageData'''
+ importer = Converter.numpy2vtkImporter(nparray, spacing, origin)
+ importer.Update()
+ return importer.GetOutput()
+
+ @staticmethod
+ def vtk2numpy(imgdata):
+ '''Converts the VTK data to 3D numpy array'''
+ img_data = numpy_support.vtk_to_numpy(
+ imgdata.GetPointData().GetScalars())
+
+ dims = imgdata.GetDimensions()
+ dims = (dims[2],dims[1],dims[0])
+ data3d = numpy.reshape(img_data, dims)
+
+ return numpy.transpose(data3d).copy()
+
+ @staticmethod
+ def tiffStack2numpy(filename, indices,
+ extent = None , sampleRate = None ,\
+ flatField = None, darkField = None):
+ '''Converts a stack of TIFF files to numpy array.
+
+ filename must contain the whole path. The filename is supposed to be named and
+ have a suffix with the ordinal file number, i.e. /path/to/projection_%03d.tif
+
+ indices are the suffix, generally an increasing number
+
+ Optionally extracts only a selection of the 2D images and (optionally)
+ normalizes.
+ '''
+
+ stack = vtk.vtkImageData()
+ reader = vtk.vtkTIFFReader()
+ voi = vtk.vtkExtractVOI()
+
+ #directory = "C:\\Users\\ofn77899\\Documents\\CCPi\\IMAT\\20170419_crabtomo\\crabtomo\\"
+
+ stack_image = numpy.asarray([])
+ nreduced = len(indices)
+
+ for num in range(len(indices)):
+ fn = filename % indices[num]
+ print ("resampling %s" % ( fn ) )
+ reader.SetFileName(fn)
+ reader.Update()
+ print (reader.GetOutput().GetScalarTypeAsString())
+ if num == 0:
+ if (extent == None):
+ sliced = reader.GetOutput().GetExtent()
+ stack.SetExtent(sliced[0],sliced[1], sliced[2],sliced[3], 0, nreduced-1)
+ else:
+ sliced = extent
+ voi.SetVOI(extent)
+
+ if sampleRate is not None:
+ voi.SetSampleRate(sampleRate)
+ ext = numpy.asarray([(sliced[2*i+1] - sliced[2*i])/sampleRate[i] for i in range(3)], dtype=int)
+ print ("ext {0}".format(ext))
+ stack.SetExtent(0, ext[0] , 0, ext[1], 0, nreduced-1)
+ else:
+ stack.SetExtent(0, sliced[1] - sliced[0] , 0, sliced[3]-sliced[2], 0, nreduced-1)
+ if (flatField != None and darkField != None):
+ stack.AllocateScalars(vtk.VTK_FLOAT, 1)
+ else:
+ stack.AllocateScalars(reader.GetOutput().GetScalarType(), 1)
+ print ("Image Size: %d" % ((sliced[1]+1)*(sliced[3]+1) ))
+ stack_image = Converter.vtk2numpy(stack)
+ print ("Stack shape %s" % str(numpy.shape(stack_image)))
+
+ if extent!=None:
+ voi.SetInputData(reader.GetOutput())
+ voi.Update()
+ img = voi.GetOutput()
+ else:
+ img = reader.GetOutput()
+
+ theSlice = Converter.vtk2numpy(img).T[0]
+ if darkField != None and flatField != None:
+ print("Try to normalize")
+ #if numpy.shape(darkField) == numpy.shape(flatField) and numpy.shape(flatField) == numpy.shape(theSlice):
+ theSlice = Converter.normalize(theSlice, darkField, flatField, 0.01)
+ print (theSlice.dtype)
+
+
+ print ("Slice shape %s" % str(numpy.shape(theSlice)))
+ stack_image.T[num] = theSlice.copy()
+
+ return stack_image
+
+ @staticmethod
+ def normalize(projection, dark, flat, def_val=0):
+ a = (projection - dark)
+ b = (flat-dark)
+ with numpy.errstate(divide='ignore', invalid='ignore'):
+ c = numpy.true_divide( a, b )
+ c[ ~ numpy.isfinite( c )] = def_val # set to not zero if 0/0
+ return c
+
+
+
+## Utility functions to transform numpy arrays to vtkImageData and viceversa
+#def numpy2vtkImporter(nparray, spacing=(1.,1.,1.), origin=(0,0,0)):
+# return Converter.numpy2vtkImporter(nparray, spacing, origin)
+#
+#def numpy2vtk(nparray, spacing=(1.,1.,1.), origin=(0,0,0)):
+# return Converter.numpy2vtk(nparray, spacing, origin)
+#
+#def vtk2numpy(imgdata):
+# return Converter.vtk2numpy(imgdata)
+#
+#def tiffStack2numpy(filename, indices):
+# return Converter.tiffStack2numpy(filename, indices)
+
+class ViewerEvent(Enum):
+ # left button
+ PICK_EVENT = 0
+ # alt + right button + move
+ WINDOW_LEVEL_EVENT = 1
+ # shift + right button
+ ZOOM_EVENT = 2
+ # control + right button
+ PAN_EVENT = 3
+ # control + left button
+ CREATE_ROI_EVENT = 4
+ # alt + left button
+ DELETE_ROI_EVENT = 5
+ # release button
+ NO_EVENT = -1
+
+
+#class CILInteractorStyle(vtk.vtkInteractorStyleTrackballCamera):
+class CILInteractorStyle(vtk.vtkInteractorStyleImage):
+
+ def __init__(self, callback):
+ vtk.vtkInteractorStyleImage.__init__(self)
+ self.callback = callback
+ self._viewer = callback
+ priority = 1.0
+
+# self.AddObserver("MouseWheelForwardEvent" , callback.OnMouseWheelForward , priority)
+# self.AddObserver("MouseWheelBackwardEvent" , callback.OnMouseWheelBackward, priority)
+# self.AddObserver('KeyPressEvent', callback.OnKeyPress, priority)
+# self.AddObserver('LeftButtonPressEvent', callback.OnLeftButtonPressEvent, priority)
+# self.AddObserver('RightButtonPressEvent', callback.OnRightButtonPressEvent, priority)
+# self.AddObserver('LeftButtonReleaseEvent', callback.OnLeftButtonReleaseEvent, priority)
+# self.AddObserver('RightButtonReleaseEvent', callback.OnRightButtonReleaseEvent, priority)
+# self.AddObserver('MouseMoveEvent', callback.OnMouseMoveEvent, priority)
+
+ self.AddObserver("MouseWheelForwardEvent" , self.OnMouseWheelForward , priority)
+ self.AddObserver("MouseWheelBackwardEvent" , self.OnMouseWheelBackward, priority)
+ self.AddObserver('KeyPressEvent', self.OnKeyPress, priority)
+ self.AddObserver('LeftButtonPressEvent', self.OnLeftButtonPressEvent, priority)
+ self.AddObserver('RightButtonPressEvent', self.OnRightButtonPressEvent, priority)
+ self.AddObserver('LeftButtonReleaseEvent', self.OnLeftButtonReleaseEvent, priority)
+ self.AddObserver('RightButtonReleaseEvent', self.OnRightButtonReleaseEvent, priority)
+ self.AddObserver('MouseMoveEvent', self.OnMouseMoveEvent, priority)
+
+ self.InitialEventPosition = (0,0)
+
+
+ def SetInitialEventPosition(self, xy):
+ self.InitialEventPosition = xy
+
+ def GetInitialEventPosition(self):
+ return self.InitialEventPosition
+
+ def GetKeyCode(self):
+ return self.GetInteractor().GetKeyCode()
+
+ def SetKeyCode(self, keycode):
+ self.GetInteractor().SetKeyCode(keycode)
+
+ def GetControlKey(self):
+ return self.GetInteractor().GetControlKey() == CONTROL_KEY
+
+ def GetShiftKey(self):
+ return self.GetInteractor().GetShiftKey() == SHIFT_KEY
+
+ def GetAltKey(self):
+ return self.GetInteractor().GetAltKey() == ALT_KEY
+
+ def GetEventPosition(self):
+ return self.GetInteractor().GetEventPosition()
+
+ def GetEventPositionInWorldCoordinates(self):
+ pass
+
+ def GetDeltaEventPosition(self):
+ x,y = self.GetInteractor().GetEventPosition()
+ return (x - self.InitialEventPosition[0] , y - self.InitialEventPosition[1])
+
+ def Dolly(self, factor):
+ self.callback.camera.Dolly(factor)
+ self.callback.ren.ResetCameraClippingRange()
+
+ def GetDimensions(self):
+ return self._viewer.img3D.GetDimensions()
+
+ def GetInputData(self):
+ return self._viewer.img3D
+
+ def GetSliceOrientation(self):
+ return self._viewer.sliceOrientation
+
+ def SetSliceOrientation(self, orientation):
+ self._viewer.sliceOrientation = orientation
+
+ def GetActiveSlice(self):
+ return self._viewer.sliceno
+
+ def SetActiveSlice(self, sliceno):
+ self._viewer.sliceno = sliceno
+
+ def UpdatePipeline(self, reset = False):
+ self._viewer.updatePipeline(reset)
+
+ def GetActiveCamera(self):
+ return self._viewer.ren.GetActiveCamera()
+
+ def SetActiveCamera(self, camera):
+ self._viewer.ren.SetActiveCamera(camera)
+
+ def ResetCamera(self):
+ self._viewer.ren.ResetCamera()
+
+ def Render(self):
+ self._viewer.renWin.Render()
+
+ def UpdateSliceActor(self):
+ self._viewer.sliceActor.Update()
+
+ def AdjustCamera(self):
+ self._viewer.AdjustCamera()
+
+ def SaveRender(self, filename):
+ self._viewer.SaveRender(filename)
+
+ def GetRenderWindow(self):
+ return self._viewer.renWin
+
+ def GetRenderer(self):
+ return self._viewer.ren
+
+ def GetROIWidget(self):
+ return self._viewer.ROIWidget
+
+ def SetViewerEvent(self, event):
+ self._viewer.event = event
+
+ def GetViewerEvent(self):
+ return self._viewer.event
+
+ def SetInitialCameraPosition(self, position):
+ self._viewer.InitialCameraPosition = position
+
+ def GetInitialCameraPosition(self):
+ return self._viewer.InitialCameraPosition
+
+ def SetInitialLevel(self, level):
+ self._viewer.InitialLevel = level
+
+ def GetInitialLevel(self):
+ return self._viewer.InitialLevel
+
+ def SetInitialWindow(self, window):
+ self._viewer.InitialWindow = window
+
+ def GetInitialWindow(self):
+ return self._viewer.InitialWindow
+
+ def GetWindowLevel(self):
+ return self._viewer.wl
+
+ def SetROI(self, roi):
+ self._viewer.ROI = roi
+
+ def GetROI(self):
+ return self._viewer.ROI
+
+ def UpdateCornerAnnotation(self, text, corner):
+ self._viewer.updateCornerAnnotation(text, corner)
+
+ def GetPicker(self):
+ return self._viewer.picker
+
+ def GetCornerAnnotation(self):
+ return self._viewer.cornerAnnotation
+
+ def UpdateROIHistogram(self):
+ self._viewer.updateROIHistogram()
+
+
+ ############### Handle events
+ def OnMouseWheelForward(self, interactor, event):
+ maxSlice = self.GetDimensions()[self.GetSliceOrientation()]
+ shift = interactor.GetShiftKey()
+ advance = 1
+ if shift:
+ advance = 10
+
+ if (self.GetActiveSlice() + advance < maxSlice):
+ self.SetActiveSlice(self.GetActiveSlice() + advance)
+
+ self.UpdatePipeline()
+ else:
+ print ("maxSlice %d request %d" % (maxSlice, self.GetActiveSlice() + 1 ))
+
+ def OnMouseWheelBackward(self, interactor, event):
+ minSlice = 0
+ shift = interactor.GetShiftKey()
+ advance = 1
+ if shift:
+ advance = 10
+ if (self.GetActiveSlice() - advance >= minSlice):
+ self.SetActiveSlice( self.GetActiveSlice() - advance)
+ self.UpdatePipeline()
+ else:
+ print ("minSlice %d request %d" % (minSlice, self.GetActiveSlice() + 1 ))
+
+ def OnKeyPress(self, interactor, event):
+ #print ("Pressed key %s" % interactor.GetKeyCode())
+ # Slice Orientation
+ if interactor.GetKeyCode() == "X":
+ # slice on the other orientation
+ self.SetSliceOrientation ( SLICE_ORIENTATION_YZ )
+ self.SetActiveSlice( int(self.GetDimensions()[1] / 2) )
+ self.UpdatePipeline(True)
+ elif interactor.GetKeyCode() == "Y":
+ # slice on the other orientation
+ self.SetSliceOrientation ( SLICE_ORIENTATION_XZ )
+ self.SetActiveSlice ( int(self.GetInputData().GetDimensions()[1] / 2) )
+ self.UpdatePipeline(True)
+ elif interactor.GetKeyCode() == "Z":
+ # slice on the other orientation
+ self.SetSliceOrientation ( SLICE_ORIENTATION_XY )
+ self.SetActiveSlice ( int(self.GetInputData().GetDimensions()[2] / 2) )
+ self.UpdatePipeline(True)
+ if interactor.GetKeyCode() == "x":
+ # Change the camera view point
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.GetActiveCamera().GetFocalPoint())
+ camera.SetViewUp(self.GetActiveCamera().GetViewUp())
+ newposition = [i for i in self.GetActiveCamera().GetFocalPoint()]
+ newposition[SLICE_ORIENTATION_YZ] = numpy.sqrt(newposition[SLICE_ORIENTATION_XY] ** 2 + newposition[SLICE_ORIENTATION_XZ] ** 2)
+ camera.SetPosition(newposition)
+ camera.SetViewUp(0,0,-1)
+ self.SetActiveCamera(camera)
+ self.Render()
+ interactor.SetKeyCode("X")
+ self.OnKeyPress(interactor, event)
+ elif interactor.GetKeyCode() == "y":
+ # Change the camera view point
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.GetActiveCamera().GetFocalPoint())
+ camera.SetViewUp(self.GetActiveCamera().GetViewUp())
+ newposition = [i for i in self.GetActiveCamera().GetFocalPoint()]
+ newposition[SLICE_ORIENTATION_XZ] = numpy.sqrt(newposition[SLICE_ORIENTATION_XY] ** 2 + newposition[SLICE_ORIENTATION_YZ] ** 2)
+ camera.SetPosition(newposition)
+ camera.SetViewUp(0,0,-1)
+ self.SetActiveCamera(camera)
+ self.Render()
+ interactor.SetKeyCode("Y")
+ self.OnKeyPress(interactor, event)
+ elif interactor.GetKeyCode() == "z":
+ # Change the camera view point
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.GetActiveCamera().GetFocalPoint())
+ camera.SetViewUp(self.GetActiveCamera().GetViewUp())
+ newposition = [i for i in self.GetActiveCamera().GetFocalPoint()]
+ newposition[SLICE_ORIENTATION_XY] = numpy.sqrt(newposition[SLICE_ORIENTATION_YZ] ** 2 + newposition[SLICE_ORIENTATION_XZ] ** 2)
+ camera.SetPosition(newposition)
+ camera.SetViewUp(0,1,0)
+ self.SetActiveCamera(camera)
+ self.ResetCamera()
+ self.Render()
+ interactor.SetKeyCode("Z")
+ self.OnKeyPress(interactor, event)
+ elif interactor.GetKeyCode() == "a":
+ # reset color/window
+ cmax = self._viewer.ia.GetMax()[0]
+ cmin = self._viewer.ia.GetMin()[0]
+
+ self.SetInitialLevel( (cmax+cmin)/2 )
+ self.SetInitialWindow( cmax-cmin )
+
+ self.GetWindowLevel().SetLevel(self.GetInitialLevel())
+ self.GetWindowLevel().SetWindow(self.GetInitialWindow())
+
+ self.GetWindowLevel().Update()
+
+ self.UpdateSliceActor()
+ self.AdjustCamera()
+ self.Render()
+
+ elif interactor.GetKeyCode() == "s":
+ filename = "current_render"
+ self.SaveRender(filename)
+ elif interactor.GetKeyCode() == "q":
+ print ("Terminating by pressing q %s" % (interactor.GetKeyCode(), ))
+ interactor.SetKeyCode("e")
+ self.OnKeyPress(interactor, event)
+ else :
+ #print ("Unhandled event %s" % (interactor.GetKeyCode(), )))
+ pass
+
+ def OnLeftButtonPressEvent(self, interactor, event):
+ alt = interactor.GetAltKey()
+ shift = interactor.GetShiftKey()
+ ctrl = interactor.GetControlKey()
+# print ("alt pressed " + (lambda x : "Yes" if x else "No")(alt))
+# print ("shift pressed " + (lambda x : "Yes" if x else "No")(shift))
+# print ("ctrl pressed " + (lambda x : "Yes" if x else "No")(ctrl))
+
+ interactor.SetInitialEventPosition(interactor.GetEventPosition())
+
+ if ctrl and not (alt and shift):
+ self.SetViewerEvent( ViewerEvent.CREATE_ROI_EVENT )
+ wsize = self.GetRenderWindow().GetSize()
+ position = interactor.GetEventPosition()
+ self.GetROIWidget().GetBorderRepresentation().SetPosition((position[0]/wsize[0] - 0.05) , (position[1]/wsize[1] - 0.05))
+ self.GetROIWidget().GetBorderRepresentation().SetPosition2( (0.1) , (0.1))
+
+ self.GetROIWidget().On()
+ self.SetDisplayHistogram(True)
+ self.Render()
+ print ("Event %s is CREATE_ROI_EVENT" % (event))
+ elif alt and not (shift and ctrl):
+ self.SetViewerEvent( ViewerEvent.DELETE_ROI_EVENT )
+ self.GetROIWidget().Off()
+ self._viewer.updateCornerAnnotation("", 1, False)
+ self.SetDisplayHistogram(False)
+ self.Render()
+ print ("Event %s is DELETE_ROI_EVENT" % (event))
+ elif not (ctrl and alt and shift):
+ self.SetViewerEvent ( ViewerEvent.PICK_EVENT )
+ self.HandlePickEvent(interactor, event)
+ print ("Event %s is PICK_EVENT" % (event))
+
+
+ def SetDisplayHistogram(self, display):
+ if display:
+ if (self._viewer.displayHistogram == 0):
+ self.GetRenderer().AddActor(self._viewer.histogramPlotActor)
+ self.firstHistogram = 1
+ self.Render()
+
+ self._viewer.histogramPlotActor.VisibilityOn()
+ self._viewer.displayHistogram = True
+ else:
+ self._viewer.histogramPlotActor.VisibilityOff()
+ self._viewer.displayHistogram = False
+
+
+ def OnLeftButtonReleaseEvent(self, interactor, event):
+ if self.GetViewerEvent() == ViewerEvent.CREATE_ROI_EVENT:
+ #bc = self.ROIWidget.GetBorderRepresentation().GetPositionCoordinate()
+ #print (bc.GetValue())
+ self.OnROIModifiedEvent(interactor, event)
+
+ elif self.GetViewerEvent() == ViewerEvent.PICK_EVENT:
+ self.HandlePickEvent(interactor, event)
+
+ self.SetViewerEvent( ViewerEvent.NO_EVENT )
+
+ def OnRightButtonPressEvent(self, interactor, event):
+ alt = interactor.GetAltKey()
+ shift = interactor.GetShiftKey()
+ ctrl = interactor.GetControlKey()
+# print ("alt pressed " + (lambda x : "Yes" if x else "No")(alt))
+# print ("shift pressed " + (lambda x : "Yes" if x else "No")(shift))
+# print ("ctrl pressed " + (lambda x : "Yes" if x else "No")(ctrl))
+
+ interactor.SetInitialEventPosition(interactor.GetEventPosition())
+
+
+ if alt and not (ctrl and shift):
+ self.SetViewerEvent( ViewerEvent.WINDOW_LEVEL_EVENT )
+ print ("Event %s is WINDOW_LEVEL_EVENT" % (event))
+ self.HandleWindowLevel(interactor, event)
+ elif shift and not (ctrl and alt):
+ self.SetViewerEvent( ViewerEvent.ZOOM_EVENT )
+ self.SetInitialCameraPosition( self.GetActiveCamera().GetPosition())
+ print ("Event %s is ZOOM_EVENT" % (event))
+ elif ctrl and not (shift and alt):
+ self.SetViewerEvent (ViewerEvent.PAN_EVENT )
+ self.SetInitialCameraPosition ( self.GetActiveCamera().GetPosition() )
+ print ("Event %s is PAN_EVENT" % (event))
+
+ def OnRightButtonReleaseEvent(self, interactor, event):
+ print (event)
+ if self.GetViewerEvent() == ViewerEvent.WINDOW_LEVEL_EVENT:
+ self.SetInitialLevel( self.GetWindowLevel().GetLevel() )
+ self.SetInitialWindow ( self.GetWindowLevel().GetWindow() )
+ elif self.GetViewerEvent() == ViewerEvent.ZOOM_EVENT or \
+ self.GetViewerEvent() == ViewerEvent.PAN_EVENT:
+ self.SetInitialCameraPosition( () )
+
+ self.SetViewerEvent( ViewerEvent.NO_EVENT )
+
+
+ def OnROIModifiedEvent(self, interactor, event):
+
+ #print ("ROI EVENT " + event)
+ p1 = self.GetROIWidget().GetBorderRepresentation().GetPositionCoordinate()
+ p2 = self.GetROIWidget().GetBorderRepresentation().GetPosition2Coordinate()
+ wsize = self.GetRenderWindow().GetSize()
+
+ #print (p1.GetValue())
+ #print (p2.GetValue())
+ pp1 = [p1.GetValue()[0] * wsize[0] , p1.GetValue()[1] * wsize[1] , 0.0]
+ pp2 = [p2.GetValue()[0] * wsize[0] + pp1[0] , p2.GetValue()[1] * wsize[1] + pp1[1] , 0.0]
+ vox1 = self.viewport2imageCoordinate(pp1)
+ vox2 = self.viewport2imageCoordinate(pp2)
+
+ self.SetROI( (vox1 , vox2) )
+ roi = self.GetROI()
+ print ("Pixel1 %d,%d,%d Value %f" % vox1 )
+ print ("Pixel2 %d,%d,%d Value %f" % vox2 )
+ if self.GetSliceOrientation() == SLICE_ORIENTATION_XY:
+ print ("slice orientation : XY")
+ x = abs(roi[1][0] - roi[0][0])
+ y = abs(roi[1][1] - roi[0][1])
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_XZ:
+ print ("slice orientation : XY")
+ x = abs(roi[1][0] - roi[0][0])
+ y = abs(roi[1][2] - roi[0][2])
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_YZ:
+ print ("slice orientation : XY")
+ x = abs(roi[1][1] - roi[0][1])
+ y = abs(roi[1][2] - roi[0][2])
+
+ text = "ROI: %d x %d, %.2f kp" % (x,y,float(x*y)/1024.)
+ print (text)
+ self.UpdateCornerAnnotation(text, 1)
+ self.UpdateROIHistogram()
+ self.SetViewerEvent( ViewerEvent.NO_EVENT )
+
+ def viewport2imageCoordinate(self, viewerposition):
+ #Determine point index
+
+ self.GetPicker().Pick(viewerposition[0], viewerposition[1], 0.0, self.GetRenderer())
+ pickPosition = list(self.GetPicker().GetPickPosition())
+ pickPosition[self.GetSliceOrientation()] = \
+ self.GetInputData().GetSpacing()[self.GetSliceOrientation()] * self.GetActiveSlice() + \
+ self.GetInputData().GetOrigin()[self.GetSliceOrientation()]
+ print ("Pick Position " + str (pickPosition))
+
+ if (pickPosition != [0,0,0]):
+ dims = self.GetInputData().GetDimensions()
+ print (dims)
+ spac = self.GetInputData().GetSpacing()
+ orig = self.GetInputData().GetOrigin()
+ imagePosition = [int(pickPosition[i] / spac[i] + orig[i]) for i in range(3) ]
+
+ pixelValue = self.GetInputData().GetScalarComponentAsDouble(imagePosition[0], imagePosition[1], imagePosition[2], 0)
+ return (imagePosition[0], imagePosition[1], imagePosition[2] , pixelValue)
+ else:
+ return (0,0,0,0)
+
+
+
+
+ def OnMouseMoveEvent(self, interactor, event):
+ if self.GetViewerEvent() == ViewerEvent.WINDOW_LEVEL_EVENT:
+ print ("Event %s is WINDOW_LEVEL_EVENT" % (event))
+ self.HandleWindowLevel(interactor, event)
+ elif self.GetViewerEvent() == ViewerEvent.PICK_EVENT:
+ self.HandlePickEvent(interactor, event)
+ elif self.GetViewerEvent() == ViewerEvent.ZOOM_EVENT:
+ self.HandleZoomEvent(interactor, event)
+ elif self.GetViewerEvent() == ViewerEvent.PAN_EVENT:
+ self.HandlePanEvent(interactor, event)
+
+
+ def HandleZoomEvent(self, interactor, event):
+ dx,dy = interactor.GetDeltaEventPosition()
+ size = self.GetRenderWindow().GetSize()
+ dy = - 4 * dy / size[1]
+
+ print ("distance: " + str(self.GetActiveCamera().GetDistance()))
+
+ print ("\ndy: %f\ncamera dolly %f\n" % (dy, 1 + dy))
+
+ camera = vtk.vtkCamera()
+ camera.SetFocalPoint(self.GetActiveCamera().GetFocalPoint())
+ #print ("current position " + str(self.InitialCameraPosition))
+ camera.SetViewUp(self.GetActiveCamera().GetViewUp())
+ camera.SetPosition(self.GetInitialCameraPosition())
+ newposition = [i for i in self.GetInitialCameraPosition()]
+ if self.GetSliceOrientation() == SLICE_ORIENTATION_XY:
+ dist = newposition[SLICE_ORIENTATION_XY] * ( 1 + dy )
+ newposition[SLICE_ORIENTATION_XY] *= ( 1 + dy )
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_XZ:
+ newposition[SLICE_ORIENTATION_XZ] *= ( 1 + dy )
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_YZ:
+ newposition[SLICE_ORIENTATION_YZ] *= ( 1 + dy )
+ #print ("new position " + str(newposition))
+ camera.SetPosition(newposition)
+ self.SetActiveCamera(camera)
+
+ self.Render()
+
+ print ("distance after: " + str(self.GetActiveCamera().GetDistance()))
+
+ def HandlePanEvent(self, interactor, event):
+ x,y = interactor.GetEventPosition()
+ x0,y0 = interactor.GetInitialEventPosition()
+
+ ic = self.viewport2imageCoordinate((x,y))
+ ic0 = self.viewport2imageCoordinate((x0,y0))
+
+ dx = 4 *( ic[0] - ic0[0])
+ dy = 4* (ic[1] - ic0[1])
+
+ camera = vtk.vtkCamera()
+ #print ("current position " + str(self.InitialCameraPosition))
+ camera.SetViewUp(self.GetActiveCamera().GetViewUp())
+ camera.SetPosition(self.GetInitialCameraPosition())
+ newposition = [i for i in self.GetInitialCameraPosition()]
+ newfocalpoint = [i for i in self.GetActiveCamera().GetFocalPoint()]
+ if self.GetSliceOrientation() == SLICE_ORIENTATION_XY:
+ newposition[0] -= dx
+ newposition[1] -= dy
+ newfocalpoint[0] = newposition[0]
+ newfocalpoint[1] = newposition[1]
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_XZ:
+ newposition[0] -= dx
+ newposition[2] -= dy
+ newfocalpoint[0] = newposition[0]
+ newfocalpoint[2] = newposition[2]
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_YZ:
+ newposition[1] -= dx
+ newposition[2] -= dy
+ newfocalpoint[2] = newposition[2]
+ newfocalpoint[1] = newposition[1]
+ #print ("new position " + str(newposition))
+ camera.SetFocalPoint(newfocalpoint)
+ camera.SetPosition(newposition)
+ self.SetActiveCamera(camera)
+
+ self.Render()
+
+ def HandleWindowLevel(self, interactor, event):
+ dx,dy = interactor.GetDeltaEventPosition()
+ print ("Event delta %d %d" % (dx,dy))
+ size = self.GetRenderWindow().GetSize()
+
+ dx = 4 * dx / size[0]
+ dy = 4 * dy / size[1]
+ window = self.GetInitialWindow()
+ level = self.GetInitialLevel()
+
+ if abs(window) > 0.01:
+ dx = dx * window
+ else:
+ dx = dx * (lambda x: -0.01 if x <0 else 0.01)(window);
+
+ if abs(level) > 0.01:
+ dy = dy * level
+ else:
+ dy = dy * (lambda x: -0.01 if x <0 else 0.01)(level)
+
+
+ # Abs so that direction does not flip
+
+ if window < 0.0:
+ dx = -1*dx
+ if level < 0.0:
+ dy = -1*dy
+
+ # Compute new window level
+
+ newWindow = dx + window
+ newLevel = level - dy
+
+ # Stay away from zero and really
+
+ if abs(newWindow) < 0.01:
+ newWindow = 0.01 * (lambda x: -1 if x <0 else 1)(newWindow)
+
+ if abs(newLevel) < 0.01:
+ newLevel = 0.01 * (lambda x: -1 if x <0 else 1)(newLevel)
+
+ self.GetWindowLevel().SetWindow(newWindow)
+ self.GetWindowLevel().SetLevel(newLevel)
+
+ self.GetWindowLevel().Update()
+ self.UpdateSliceActor()
+ self.AdjustCamera()
+
+ self.Render()
+
+ def HandlePickEvent(self, interactor, event):
+ position = interactor.GetEventPosition()
+ #print ("PICK " + str(position))
+ vox = self.viewport2imageCoordinate(position)
+ #print ("Pixel %d,%d,%d Value %f" % vox )
+ self._viewer.cornerAnnotation.VisibilityOn()
+ self.UpdateCornerAnnotation("[%d,%d,%d] : %.2f" % vox , 0)
+ self.Render()
+
+###############################################################################
+
+
+
+class CILViewer2D():
+ '''Simple Interactive Viewer based on VTK classes'''
+
+ def __init__(self, dimx=600,dimy=600, ren=None, renWin=None,iren=None):
+ '''creates the rendering pipeline'''
+ # create a rendering window and renderer
+ if ren == None:
+ self.ren = vtk.vtkRenderer()
+ else:
+ self.ren = ren
+ if renWin == None:
+ self.renWin = vtk.vtkRenderWindow()
+ else:
+ self.renWin = renWin
+ if iren == None:
+ self.iren = vtk.vtkRenderWindowInteractor()
+ else:
+ self.iren = iren
+
+ self.renWin.SetSize(dimx,dimy)
+ self.renWin.AddRenderer(self.ren)
+
+ self.style = CILInteractorStyle(self)
+
+ self.iren.SetInteractorStyle(self.style)
+ self.iren.SetRenderWindow(self.renWin)
+ self.iren.Initialize()
+ self.ren.SetBackground(.1, .2, .4)
+
+ self.camera = vtk.vtkCamera()
+ self.camera.ParallelProjectionOn()
+ self.ren.SetActiveCamera(self.camera)
+
+ # data
+ self.img3D = None
+ self.sliceno = 0
+ self.sliceOrientation = SLICE_ORIENTATION_XY
+
+ #Actors
+ self.sliceActor = vtk.vtkImageActor()
+ self.voi = vtk.vtkExtractVOI()
+ self.wl = vtk.vtkImageMapToWindowLevelColors()
+ self.ia = vtk.vtkImageAccumulate()
+ self.sliceActorNo = 0
+
+ #initial Window/Level
+ self.InitialLevel = 0
+ self.InitialWindow = 0
+
+ #ViewerEvent
+ self.event = ViewerEvent.NO_EVENT
+
+ # ROI Widget
+ self.ROIWidget = vtk.vtkBorderWidget()
+ self.ROIWidget.SetInteractor(self.iren)
+ self.ROIWidget.CreateDefaultRepresentation()
+ self.ROIWidget.GetBorderRepresentation().GetBorderProperty().SetColor(0,1,0)
+ self.ROIWidget.AddObserver(vtk.vtkWidgetEvent.Select, self.style.OnROIModifiedEvent, 1.0)
+
+ # edge points of the ROI
+ self.ROI = ()
+
+ #picker
+ self.picker = vtk.vtkPropPicker()
+ self.picker.PickFromListOn()
+ self.picker.AddPickList(self.sliceActor)
+
+ self.iren.SetPicker(self.picker)
+
+ # corner annotation
+ self.cornerAnnotation = vtk.vtkCornerAnnotation()
+ self.cornerAnnotation.SetMaximumFontSize(12);
+ self.cornerAnnotation.PickableOff();
+ self.cornerAnnotation.VisibilityOff();
+ self.cornerAnnotation.GetTextProperty().ShadowOn();
+ self.cornerAnnotation.SetLayerNumber(1);
+
+
+
+ # cursor doesn't show up
+ self.cursor = vtk.vtkCursor2D()
+ self.cursorMapper = vtk.vtkPolyDataMapper2D()
+ self.cursorActor = vtk.vtkActor2D()
+ self.cursor.SetModelBounds(-10, 10, -10, 10, 0, 0)
+ self.cursor.SetFocalPoint(0, 0, 0)
+ self.cursor.AllOff()
+ self.cursor.AxesOn()
+ self.cursorActor.PickableOff()
+ self.cursorActor.VisibilityOn()
+ self.cursorActor.GetProperty().SetColor(1, 1, 1)
+ self.cursorActor.SetLayerNumber(1)
+ self.cursorMapper.SetInputData(self.cursor.GetOutput())
+ self.cursorActor.SetMapper(self.cursorMapper)
+
+ # Zoom
+ self.InitialCameraPosition = ()
+
+ # XY Plot actor for histogram
+ self.displayHistogram = False
+ self.firstHistogram = 0
+ self.roiIA = vtk.vtkImageAccumulate()
+ self.roiVOI = vtk.vtkExtractVOI()
+ self.histogramPlotActor = vtk.vtkXYPlotActor()
+ self.histogramPlotActor.ExchangeAxesOff();
+ self.histogramPlotActor.SetXLabelFormat( "%g" )
+ self.histogramPlotActor.SetXLabelFormat( "%g" )
+ self.histogramPlotActor.SetAdjustXLabels(3)
+ self.histogramPlotActor.SetXTitle( "Level" )
+ self.histogramPlotActor.SetYTitle( "N" )
+ self.histogramPlotActor.SetXValuesToValue()
+ self.histogramPlotActor.SetPlotColor(0, (0,1,1) )
+ self.histogramPlotActor.SetPosition(0.6,0.6)
+ self.histogramPlotActor.SetPosition2(0.4,0.4)
+
+
+
+ def GetInteractor(self):
+ return self.iren
+
+ def GetRenderer(self):
+ return self.ren
+
+ def setInput3DData(self, imageData):
+ self.img3D = imageData
+ self.installPipeline()
+
+ def setInputAsNumpy(self, numpyarray, origin=(0,0,0), spacing=(1.,1.,1.),
+ rescale=True, dtype=vtk.VTK_UNSIGNED_SHORT):
+ importer = Converter.numpy2vtkImporter(numpyarray, spacing, origin)
+ importer.Update()
+
+ if rescale:
+ # rescale to appropriate VTK_UNSIGNED_SHORT
+ stats = vtk.vtkImageAccumulate()
+ stats.SetInputData(importer.GetOutput())
+ stats.Update()
+ iMin = stats.GetMin()[0]
+ iMax = stats.GetMax()[0]
+ if (iMax - iMin == 0):
+ scale = 1
+ else:
+ if dtype == vtk.VTK_UNSIGNED_SHORT:
+ scale = vtk.VTK_UNSIGNED_SHORT_MAX / (iMax - iMin)
+ elif dtype == vtk.VTK_UNSIGNED_INT:
+ scale = vtk.VTK_UNSIGNED_INT_MAX / (iMax - iMin)
+
+ shiftScaler = vtk.vtkImageShiftScale ()
+ shiftScaler.SetInputData(importer.GetOutput())
+ shiftScaler.SetScale(scale)
+ shiftScaler.SetShift(-iMin)
+ shiftScaler.SetOutputScalarType(dtype)
+ shiftScaler.Update()
+ self.img3D = shiftScaler.GetOutput()
+ else:
+ self.img3D = importer.GetOutput()
+
+ self.installPipeline()
+
+ def displaySlice(self, sliceno = 0):
+ self.sliceno = sliceno
+
+ self.updatePipeline()
+
+ self.renWin.Render()
+
+ return self.sliceActorNo
+
+ def updatePipeline(self, resetcamera = False):
+ extent = [ i for i in self.img3D.GetExtent()]
+ extent[self.sliceOrientation * 2] = self.sliceno
+ extent[self.sliceOrientation * 2 + 1] = self.sliceno
+ self.voi.SetVOI(extent[0], extent[1],
+ extent[2], extent[3],
+ extent[4], extent[5])
+
+ self.voi.Update()
+ self.ia.Update()
+ self.wl.Update()
+ self.sliceActor.SetDisplayExtent(extent[0], extent[1],
+ extent[2], extent[3],
+ extent[4], extent[5])
+ self.sliceActor.Update()
+
+ self.updateCornerAnnotation("Slice %d/%d" % (self.sliceno + 1 , self.img3D.GetDimensions()[self.sliceOrientation]))
+
+ if self.displayHistogram:
+ self.updateROIHistogram()
+
+ self.AdjustCamera(resetcamera)
+
+ self.renWin.Render()
+
+
+ def installPipeline(self):
+ '''Slices a 3D volume and then creates an actor to be rendered'''
+
+ self.ren.AddViewProp(self.cornerAnnotation)
+
+ self.voi.SetInputData(self.img3D)
+ #select one slice in Z
+ extent = [ i for i in self.img3D.GetExtent()]
+ extent[self.sliceOrientation * 2] = self.sliceno
+ extent[self.sliceOrientation * 2 + 1] = self.sliceno
+ self.voi.SetVOI(extent[0], extent[1],
+ extent[2], extent[3],
+ extent[4], extent[5])
+
+ self.voi.Update()
+ # set window/level for current slices
+
+
+ self.wl = vtk.vtkImageMapToWindowLevelColors()
+ self.ia.SetInputData(self.voi.GetOutput())
+ self.ia.Update()
+ cmax = self.ia.GetMax()[0]
+ cmin = self.ia.GetMin()[0]
+
+ self.InitialLevel = (cmax+cmin)/2
+ self.InitialWindow = cmax-cmin
+
+
+ self.wl.SetLevel(self.InitialLevel)
+ self.wl.SetWindow(self.InitialWindow)
+
+ self.wl.SetInputData(self.voi.GetOutput())
+ self.wl.Update()
+
+ self.sliceActor.SetInputData(self.wl.GetOutput())
+ self.sliceActor.SetDisplayExtent(extent[0], extent[1],
+ extent[2], extent[3],
+ extent[4], extent[5])
+ self.sliceActor.Update()
+ self.sliceActor.SetInterpolate(False)
+ self.ren.AddActor(self.sliceActor)
+ self.ren.ResetCamera()
+ self.ren.Render()
+
+ self.AdjustCamera()
+
+ self.ren.AddViewProp(self.cursorActor)
+ self.cursorActor.VisibilityOn()
+
+ self.iren.Initialize()
+ self.renWin.Render()
+ #self.iren.Start()
+
+ def AdjustCamera(self, resetcamera = False):
+ self.ren.ResetCameraClippingRange()
+ if resetcamera:
+ self.ren.ResetCamera()
+
+
+ def getROI(self):
+ return self.ROI
+
+ def getROIExtent(self):
+ p0 = self.ROI[0]
+ p1 = self.ROI[1]
+ return (p0[0], p1[0],p0[1],p1[1],p0[2],p1[2])
+
+ ############### Handle events are moved to the interactor style
+
+
+ def viewport2imageCoordinate(self, viewerposition):
+ #Determine point index
+
+ self.picker.Pick(viewerposition[0], viewerposition[1], 0.0, self.GetRenderer())
+ pickPosition = list(self.picker.GetPickPosition())
+ pickPosition[self.sliceOrientation] = \
+ self.img3D.GetSpacing()[self.sliceOrientation] * self.sliceno + \
+ self.img3D.GetOrigin()[self.sliceOrientation]
+ print ("Pick Position " + str (pickPosition))
+
+ if (pickPosition != [0,0,0]):
+ dims = self.img3D.GetDimensions()
+ print (dims)
+ spac = self.img3D.GetSpacing()
+ orig = self.img3D.GetOrigin()
+ imagePosition = [int(pickPosition[i] / spac[i] + orig[i]) for i in range(3) ]
+
+ pixelValue = self.img3D.GetScalarComponentAsDouble(imagePosition[0], imagePosition[1], imagePosition[2], 0)
+ return (imagePosition[0], imagePosition[1], imagePosition[2] , pixelValue)
+ else:
+ return (0,0,0,0)
+
+
+
+ def GetRenderWindow(self):
+ return self.renWin
+
+
+ def startRenderLoop(self):
+ self.iren.Start()
+
+ def GetSliceOrientation(self):
+ return self.sliceOrientation
+
+ def GetActiveSlice(self):
+ return self.sliceno
+
+ def updateCornerAnnotation(self, text , idx=0, visibility=True):
+ if visibility:
+ self.cornerAnnotation.VisibilityOn()
+ else:
+ self.cornerAnnotation.VisibilityOff()
+
+ self.cornerAnnotation.SetText(idx, text)
+ self.iren.Render()
+
+ def saveRender(self, filename, renWin=None):
+ '''Save the render window to PNG file'''
+ # screenshot code:
+ w2if = vtk.vtkWindowToImageFilter()
+ if renWin == None:
+ renWin = self.renWin
+ w2if.SetInput(renWin)
+ w2if.Update()
+
+ writer = vtk.vtkPNGWriter()
+ writer.SetFileName("%s.png" % (filename))
+ writer.SetInputConnection(w2if.GetOutputPort())
+ writer.Write()
+
+ def updateROIHistogram(self):
+
+ extent = [0 for i in range(6)]
+ if self.GetSliceOrientation() == SLICE_ORIENTATION_XY:
+ print ("slice orientation : XY")
+ extent[0] = self.ROI[0][0]
+ extent[1] = self.ROI[1][0]
+ extent[2] = self.ROI[0][1]
+ extent[3] = self.ROI[1][1]
+ extent[4] = self.GetActiveSlice()
+ extent[5] = self.GetActiveSlice()+1
+ #y = abs(roi[1][1] - roi[0][1])
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_XZ:
+ print ("slice orientation : XY")
+ extent[0] = self.ROI[0][0]
+ extent[1] = self.ROI[1][0]
+ #x = abs(roi[1][0] - roi[0][0])
+ extent[4] = self.ROI[0][2]
+ extent[5] = self.ROI[1][2]
+ #y = abs(roi[1][2] - roi[0][2])
+ extent[2] = self.GetActiveSlice()
+ extent[3] = self.GetActiveSlice()+1
+ elif self.GetSliceOrientation() == SLICE_ORIENTATION_YZ:
+ print ("slice orientation : XY")
+ extent[2] = self.ROI[0][1]
+ extent[3] = self.ROI[1][1]
+ #x = abs(roi[1][1] - roi[0][1])
+ extent[4] = self.ROI[0][2]
+ extent[5] = self.ROI[1][2]
+ #y = abs(roi[1][2] - roi[0][2])
+ extent[0] = self.GetActiveSlice()
+ extent[1] = self.GetActiveSlice()+1
+
+ self.roiVOI.SetVOI(extent)
+ self.roiVOI.SetInputData(self.img3D)
+ self.roiVOI.Update()
+ irange = self.roiVOI.GetOutput().GetScalarRange()
+
+ self.roiIA.SetInputData(self.roiVOI.GetOutput())
+ self.roiIA.IgnoreZeroOff()
+ self.roiIA.SetComponentExtent(0,int(irange[1]-irange[0]-1),0,0,0,0 )
+ self.roiIA.SetComponentOrigin( int(irange[0]),0,0 );
+ self.roiIA.SetComponentSpacing( 1,0,0 );
+ self.roiIA.Update()
+
+ self.histogramPlotActor.AddDataSetInputConnection(self.roiIA.GetOutputPort())
+ self.histogramPlotActor.SetXRange(irange[0],irange[1])
+
+ self.histogramPlotActor.SetYRange( self.roiIA.GetOutput().GetScalarRange() )
+
+ \ No newline at end of file
diff --git a/src/Python/ccpi/viewer/QVTKWidget.py b/src/Python/ccpi/viewer/QVTKWidget.py
new file mode 100644
index 0000000..906786b
--- /dev/null
+++ b/src/Python/ccpi/viewer/QVTKWidget.py
@@ -0,0 +1,340 @@
+################################################################################
+# File: QVTKWidget.py
+# Author: Edoardo Pasca
+# Description: PyVE Viewer Qt widget
+#
+# License:
+# This file is part of PyVE. PyVE is an open-source image
+# analysis and visualization environment focused on medical
+# imaging. More info at http://pyve.sourceforge.net
+#
+# Copyright (c) 2011-2012 Edoardo Pasca, Lukas Batteau
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or
+# without modification, are permitted provided that the following
+# conditions are met:
+#
+# Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+# Redistributions in binary form must reproduce the above
+# copyright notice, this list of conditions and the following
+# disclaimer in the documentation and/or other materials provided
+# with the distribution. Neither name of Edoardo Pasca or Lukas
+# Batteau nor the names of any contributors may be used to endorse
+# or promote products derived from this software without specific
+# prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
+# CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
+# OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
+# OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
+# TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
+# OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
+# OF SUCH DAMAGE.
+#
+# CHANGE HISTORY
+#
+# 20120118 Edoardo Pasca Initial version
+#
+###############################################################################
+
+import os
+from PyQt5 import QtCore, QtGui, QtWidgets
+#import itk
+import vtk
+#from viewer import PyveViewer
+from ccpi.viewer.CILViewer2D import CILViewer2D , Converter
+
+class QVTKWidget(QtWidgets.QWidget):
+
+ """ A QVTKWidget for Python and Qt."""
+
+ # Map between VTK and Qt cursors.
+ _CURSOR_MAP = {
+ 0: QtCore.Qt.ArrowCursor, # VTK_CURSOR_DEFAULT
+ 1: QtCore.Qt.ArrowCursor, # VTK_CURSOR_ARROW
+ 2: QtCore.Qt.SizeBDiagCursor, # VTK_CURSOR_SIZENE
+ 3: QtCore.Qt.SizeFDiagCursor, # VTK_CURSOR_SIZENWSE
+ 4: QtCore.Qt.SizeBDiagCursor, # VTK_CURSOR_SIZESW
+ 5: QtCore.Qt.SizeFDiagCursor, # VTK_CURSOR_SIZESE
+ 6: QtCore.Qt.SizeVerCursor, # VTK_CURSOR_SIZENS
+ 7: QtCore.Qt.SizeHorCursor, # VTK_CURSOR_SIZEWE
+ 8: QtCore.Qt.SizeAllCursor, # VTK_CURSOR_SIZEALL
+ 9: QtCore.Qt.PointingHandCursor, # VTK_CURSOR_HAND
+ 10: QtCore.Qt.CrossCursor, # VTK_CURSOR_CROSSHAIR
+ }
+
+ def __init__(self, parent=None, wflags=QtCore.Qt.WindowFlags(), **kw):
+ # the current button
+ self._ActiveButton = QtCore.Qt.NoButton
+
+ # private attributes
+ self.__oldFocus = None
+ self.__saveX = 0
+ self.__saveY = 0
+ self.__saveModifiers = QtCore.Qt.NoModifier
+ self.__saveButtons = QtCore.Qt.NoButton
+ self.__timeframe = 0
+
+ # create qt-level widget
+ QtWidgets.QWidget.__init__(self, parent, wflags|QtCore.Qt.MSWindowsOwnDC)
+
+ # Link to PyVE Viewer
+ self._PyveViewer = CILViewer2D()
+ #self._Viewer = self._PyveViewer._vtkPyveViewer
+
+ self._Iren = self._PyveViewer.GetInteractor()
+ #self._Iren = self._Viewer.GetRenderWindow().GetInteractor()
+ self._RenderWindow = self._PyveViewer.GetRenderWindow()
+ #self._RenderWindow = self._Viewer.GetRenderWindow()
+
+ self._Iren.Register(self._RenderWindow)
+ self._Iren.SetRenderWindow(self._RenderWindow)
+ self._RenderWindow.SetWindowInfo(str(int(self.winId())))
+
+ # do all the necessary qt setup
+ self.setAttribute(QtCore.Qt.WA_OpaquePaintEvent)
+ self.setAttribute(QtCore.Qt.WA_PaintOnScreen)
+ self.setMouseTracking(True) # get all mouse events
+ self.setFocusPolicy(QtCore.Qt.WheelFocus)
+ self.setSizePolicy(QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding, QtWidgets.QSizePolicy.Expanding))
+
+ self._Timer = QtCore.QTimer(self)
+ #self.connect(self._Timer, QtCore.pyqtSignal('timeout()'), self.TimerEvent)
+
+ self._Iren.AddObserver('CreateTimerEvent', self.CreateTimer)
+ self._Iren.AddObserver('DestroyTimerEvent', self.DestroyTimer)
+ self._Iren.GetRenderWindow().AddObserver('CursorChangedEvent',
+ self.CursorChangedEvent)
+
+ # Destructor
+ def __del__(self):
+ self._Iren.UnRegister(self._RenderWindow)
+ #QtWidgets.QWidget.__del__(self)
+
+ # Display image data
+ def SetInput(self, imageData):
+ self._PyveViewer.setInput3DData(imageData)
+
+ # GetInteractor
+ def GetInteractor(self):
+ return self._Iren
+
+ # Display image data
+ def GetPyveViewer(self):
+ return self._PyveViewer
+
+ def __getattr__(self, attr):
+ """Makes the object behave like a vtkGenericRenderWindowInteractor"""
+ print (attr)
+ if attr == '__vtk__':
+ return lambda t=self._Iren: t
+ elif hasattr(self._Iren, attr):
+ return getattr(self._Iren, attr)
+# else:
+# raise AttributeError( self.__class__.__name__ + \
+# " has no attribute named " + attr )
+
+ def CreateTimer(self, obj, evt):
+ self._Timer.start(10)
+
+ def DestroyTimer(self, obj, evt):
+ self._Timer.stop()
+ return 1
+
+ def TimerEvent(self):
+ self._Iren.InvokeEvent("TimerEvent")
+
+ def CursorChangedEvent(self, obj, evt):
+ """Called when the CursorChangedEvent fires on the render window."""
+ # This indirection is needed since when the event fires, the current
+ # cursor is not yet set so we defer this by which time the current
+ # cursor should have been set.
+ QtCore.QTimer.singleShot(0, self.ShowCursor)
+
+ def HideCursor(self):
+ """Hides the cursor."""
+ self.setCursor(QtCore.Qt.BlankCursor)
+
+ def ShowCursor(self):
+ """Shows the cursor."""
+ vtk_cursor = self._Iren.GetRenderWindow().GetCurrentCursor()
+ qt_cursor = self._CURSOR_MAP.get(vtk_cursor, QtCore.Qt.ArrowCursor)
+ self.setCursor(qt_cursor)
+
+ def sizeHint(self):
+ return QtCore.QSize(400, 400)
+
+ def paintEngine(self):
+ return None
+
+ def paintEvent(self, ev):
+ self._RenderWindow.Render()
+
+ def resizeEvent(self, ev):
+ self._RenderWindow.Render()
+ w = self.width()
+ h = self.height()
+
+ self._RenderWindow.SetSize(w, h)
+ self._Iren.SetSize(w, h)
+
+ def _GetCtrlShiftAlt(self, ev):
+ ctrl = shift = alt = False
+
+ if hasattr(ev, 'modifiers'):
+ if ev.modifiers() & QtCore.Qt.ShiftModifier:
+ shift = True
+ if ev.modifiers() & QtCore.Qt.ControlModifier:
+ ctrl = True
+ if ev.modifiers() & QtCore.Qt.AltModifier:
+ alt = True
+ else:
+ if self.__saveModifiers & QtCore.Qt.ShiftModifier:
+ shift = True
+ if self.__saveModifiers & QtCore.Qt.ControlModifier:
+ ctrl = True
+ if self.__saveModifiers & QtCore.Qt.AltModifier:
+ alt = True
+
+ return ctrl, shift, alt
+
+ def enterEvent(self, ev):
+ if not self.hasFocus():
+ self.__oldFocus = self.focusWidget()
+ self.setFocus()
+
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ self._Iren.SetEventInformationFlipY(self.__saveX, self.__saveY,
+ ctrl, shift, chr(0), 0, None)
+ self._Iren.SetAltKey(alt)
+ self._Iren.InvokeEvent("EnterEvent")
+
+ def leaveEvent(self, ev):
+ if self.__saveButtons == QtCore.Qt.NoButton and self.__oldFocus:
+ self.__oldFocus.setFocus()
+ self.__oldFocus = None
+
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ self._Iren.SetEventInformationFlipY(self.__saveX, self.__saveY,
+ ctrl, shift, chr(0), 0, None)
+ self._Iren.SetAltKey(alt)
+ self._Iren.InvokeEvent("LeaveEvent")
+
+ def mousePressEvent(self, ev):
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ repeat = 0
+ if ev.type() == QtCore.QEvent.MouseButtonDblClick:
+ repeat = 1
+ self._Iren.SetEventInformationFlipY(ev.x(), ev.y(),
+ ctrl, shift, chr(0), repeat, None)
+
+ self._Iren.SetAltKey(alt)
+ self._ActiveButton = ev.button()
+
+ if self._ActiveButton == QtCore.Qt.LeftButton:
+ self._Iren.InvokeEvent("LeftButtonPressEvent")
+ elif self._ActiveButton == QtCore.Qt.RightButton:
+ self._Iren.InvokeEvent("RightButtonPressEvent")
+ elif self._ActiveButton == QtCore.Qt.MidButton:
+ self._Iren.InvokeEvent("MiddleButtonPressEvent")
+
+ def mouseReleaseEvent(self, ev):
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ self._Iren.SetEventInformationFlipY(ev.x(), ev.y(),
+ ctrl, shift, chr(0), 0, None)
+ self._Iren.SetAltKey(alt)
+
+ if self._ActiveButton == QtCore.Qt.LeftButton:
+ self._Iren.InvokeEvent("LeftButtonReleaseEvent")
+ elif self._ActiveButton == QtCore.Qt.RightButton:
+ self._Iren.InvokeEvent("RightButtonReleaseEvent")
+ elif self._ActiveButton == QtCore.Qt.MidButton:
+ self._Iren.InvokeEvent("MiddleButtonReleaseEvent")
+
+ def mouseMoveEvent(self, ev):
+ self.__saveModifiers = ev.modifiers()
+ self.__saveButtons = ev.buttons()
+ self.__saveX = ev.x()
+ self.__saveY = ev.y()
+
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ self._Iren.SetEventInformationFlipY(ev.x(), ev.y(),
+ ctrl, shift, chr(0), 0, None)
+ self._Iren.SetAltKey(alt)
+ self._Iren.InvokeEvent("MouseMoveEvent")
+
+ def keyPressEvent(self, ev):
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ if ev.key() < 256:
+ key = str(ev.text())
+ else:
+ key = chr(0)
+
+ self._Iren.SetEventInformationFlipY(self.__saveX, self.__saveY,
+ ctrl, shift, key, 0, None)
+ self._Iren.SetAltKey(alt)
+ self._Iren.InvokeEvent("KeyPressEvent")
+ self._Iren.InvokeEvent("CharEvent")
+
+ def keyReleaseEvent(self, ev):
+ ctrl, shift, alt = self._GetCtrlShiftAlt(ev)
+ if ev.key() < 256:
+ key = chr(ev.key())
+ else:
+ key = chr(0)
+
+ self._Iren.SetEventInformationFlipY(self.__saveX, self.__saveY,
+ ctrl, shift, key, 0, None)
+ self._Iren.SetAltKey(alt)
+ self._Iren.InvokeEvent("KeyReleaseEvent")
+
+ def wheelEvent(self, ev):
+ print ("angleDeltaX %d" % ev.angleDelta().x())
+ print ("angleDeltaY %d" % ev.angleDelta().y())
+ if ev.angleDelta().y() >= 0:
+ self._Iren.InvokeEvent("MouseWheelForwardEvent")
+ else:
+ self._Iren.InvokeEvent("MouseWheelBackwardEvent")
+
+ def GetRenderWindow(self):
+ return self._RenderWindow
+
+ def Render(self):
+ self.update()
+
+
+def QVTKExample():
+ """A simple example that uses the QVTKWidget class."""
+
+ # every QT app needs an app
+ app = QtWidgets.QApplication(['PyVE QVTKWidget Example'])
+ page_VTK = QtWidgets.QWidget()
+ page_VTK.resize(500,500)
+ layout = QtWidgets.QVBoxLayout(page_VTK)
+ # create the widget
+ widget = QVTKWidget(parent=None)
+ layout.addWidget(widget)
+
+ #reader = vtk.vtkPNGReader()
+ #reader.SetFileName("F:\Diagnostics\Images\PyVE\VTKData\Data\camscene.png")
+ reader = vtk.vtkMetaImageReader()
+ reader.SetFileName("C:\\Users\\ofn77899\\Documents\\GitHub\\CCPi-Simpleflex\\data\\head.mha")
+ reader.Update()
+
+ widget.SetInput(reader.GetOutput())
+
+ # show the widget
+ page_VTK.show()
+ # start event processing
+ app.exec_()
+
+if __name__ == "__main__":
+ QVTKExample()
diff --git a/src/Python/ccpi/viewer/QVTKWidget2.py b/src/Python/ccpi/viewer/QVTKWidget2.py
new file mode 100644
index 0000000..e32e1c2
--- /dev/null
+++ b/src/Python/ccpi/viewer/QVTKWidget2.py
@@ -0,0 +1,84 @@
+################################################################################
+# File: QVTKWidget.py
+# Author: Edoardo Pasca
+# Description: PyVE Viewer Qt widget
+#
+# License:
+# This file is part of PyVE. PyVE is an open-source image
+# analysis and visualization environment focused on medical
+# imaging. More info at http://pyve.sourceforge.net
+#
+# Copyright (c) 2011-2012 Edoardo Pasca, Lukas Batteau
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or
+# without modification, are permitted provided that the following
+# conditions are met:
+#
+# Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+# Redistributions in binary form must reproduce the above
+# copyright notice, this list of conditions and the following
+# disclaimer in the documentation and/or other materials provided
+# with the distribution. Neither name of Edoardo Pasca or Lukas
+# Batteau nor the names of any contributors may be used to endorse
+# or promote products derived from this software without specific
+# prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
+# CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
+# MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+# DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
+# OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
+# OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
+# TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
+# OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY
+# OF SUCH DAMAGE.
+#
+# CHANGE HISTORY
+#
+# 20120118 Edoardo Pasca Initial version
+#
+###############################################################################
+
+import os
+from PyQt5 import QtCore, QtGui, QtWidgets
+#import itk
+import vtk
+#from viewer import PyveViewer
+from ccpi.viewer.CILViewer2D import CILViewer2D , Converter
+from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
+
+class QVTKWidget(QVTKRenderWindowInteractor):
+
+
+ def __init__(self, parent=None, wflags=QtCore.Qt.WindowFlags(), **kw):
+ kw = dict()
+ super().__init__(parent, **kw)
+
+
+ # Link to PyVE Viewer
+ self._PyveViewer = CILViewer2D(400,400)
+ #self._Viewer = self._PyveViewer._vtkPyveViewer
+
+ self._Iren = self._PyveViewer.GetInteractor()
+ kw['iren'] = self._Iren
+ #self._Iren = self._Viewer.GetRenderWindow().GetInteractor()
+ self._RenderWindow = self._PyveViewer.GetRenderWindow()
+ #self._RenderWindow = self._Viewer.GetRenderWindow()
+ kw['rw'] = self._RenderWindow
+
+
+
+
+ def GetInteractor(self):
+ return self._Iren
+
+ # Display image data
+ def SetInput(self, imageData):
+ self._PyveViewer.setInput3DData(imageData)
+ \ No newline at end of file
diff --git a/src/Python/ccpi/viewer/__init__.py b/src/Python/ccpi/viewer/__init__.py
new file mode 100644
index 0000000..946188b
--- /dev/null
+++ b/src/Python/ccpi/viewer/__init__.py
@@ -0,0 +1 @@
+from ccpi.viewer.CILViewer import CILViewer \ No newline at end of file
diff --git a/src/Python/ccpi/viewer/__pycache__/CILViewer.cpython-35.pyc b/src/Python/ccpi/viewer/__pycache__/CILViewer.cpython-35.pyc
new file mode 100644
index 0000000..711f77a
--- /dev/null
+++ b/src/Python/ccpi/viewer/__pycache__/CILViewer.cpython-35.pyc
Binary files differ
diff --git a/src/Python/ccpi/viewer/__pycache__/CILViewer2D.cpython-35.pyc b/src/Python/ccpi/viewer/__pycache__/CILViewer2D.cpython-35.pyc
new file mode 100644
index 0000000..77c2ca8
--- /dev/null
+++ b/src/Python/ccpi/viewer/__pycache__/CILViewer2D.cpython-35.pyc
Binary files differ
diff --git a/src/Python/ccpi/viewer/__pycache__/QVTKWidget.cpython-35.pyc b/src/Python/ccpi/viewer/__pycache__/QVTKWidget.cpython-35.pyc
new file mode 100644
index 0000000..3d11b87
--- /dev/null
+++ b/src/Python/ccpi/viewer/__pycache__/QVTKWidget.cpython-35.pyc
Binary files differ
diff --git a/src/Python/ccpi/viewer/__pycache__/QVTKWidget2.cpython-35.pyc b/src/Python/ccpi/viewer/__pycache__/QVTKWidget2.cpython-35.pyc
new file mode 100644
index 0000000..2fa2eaf
--- /dev/null
+++ b/src/Python/ccpi/viewer/__pycache__/QVTKWidget2.cpython-35.pyc
Binary files differ
diff --git a/src/Python/ccpi/viewer/__pycache__/__init__.cpython-35.pyc b/src/Python/ccpi/viewer/__pycache__/__init__.cpython-35.pyc
new file mode 100644
index 0000000..fcea537
--- /dev/null
+++ b/src/Python/ccpi/viewer/__pycache__/__init__.cpython-35.pyc
Binary files differ
diff --git a/src/Python/ccpi/viewer/embedvtk.py b/src/Python/ccpi/viewer/embedvtk.py
new file mode 100644
index 0000000..b5eb0a7
--- /dev/null
+++ b/src/Python/ccpi/viewer/embedvtk.py
@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Thu Jul 27 12:18:58 2017
+
+@author: ofn77899
+"""
+
+#!/usr/bin/env python
+
+import sys
+import vtk
+from PyQt5 import QtCore, QtWidgets
+from vtk.qt.QVTKRenderWindowInteractor import QVTKRenderWindowInteractor
+import QVTKWidget2
+
+class MainWindow(QtWidgets.QMainWindow):
+
+ def __init__(self, parent = None):
+ QtWidgets.QMainWindow.__init__(self, parent)
+
+ self.frame = QtWidgets.QFrame()
+
+ self.vl = QtWidgets.QVBoxLayout()
+# self.vtkWidget = QVTKRenderWindowInteractor(self.frame)
+
+ self.vtkWidget = QVTKWidget2.QVTKWidget(self.frame)
+ self.iren = self.vtkWidget.GetInteractor()
+ self.vl.addWidget(self.vtkWidget)
+
+
+
+
+ self.ren = vtk.vtkRenderer()
+ self.vtkWidget.GetRenderWindow().AddRenderer(self.ren)
+# self.iren = self.vtkWidget.GetRenderWindow().GetInteractor()
+#
+# # Create source
+# source = vtk.vtkSphereSource()
+# source.SetCenter(0, 0, 0)
+# source.SetRadius(5.0)
+#
+# # Create a mapper
+# mapper = vtk.vtkPolyDataMapper()
+# mapper.SetInputConnection(source.GetOutputPort())
+#
+# # Create an actor
+# actor = vtk.vtkActor()
+# actor.SetMapper(mapper)
+#
+# self.ren.AddActor(actor)
+#
+# self.ren.ResetCamera()
+#
+ self.frame.setLayout(self.vl)
+ self.setCentralWidget(self.frame)
+ reader = vtk.vtkMetaImageReader()
+ reader.SetFileName("C:\\Users\\ofn77899\\Documents\\GitHub\\CCPi-Simpleflex\\data\\head.mha")
+ reader.Update()
+
+ self.vtkWidget.SetInput(reader.GetOutput())
+
+ #self.vktWidget.Initialize()
+ #self.vktWidget.Start()
+
+ self.show()
+ #self.iren.Initialize()
+
+
+if __name__ == "__main__":
+
+ app = QtWidgets.QApplication(sys.argv)
+
+ window = MainWindow()
+
+ sys.exit(app.exec_()) \ No newline at end of file