summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorEdoardo Pasca <edo.paskino@gmail.com>2017-10-18 10:04:32 +0100
committerEdoardo Pasca <edo.paskino@gmail.com>2017-10-18 10:10:27 +0100
commit1af73a75ccab1147a8d2387b7056f91f0642549f (patch)
tree67ed90224759a7504030b9f3258f34bac42273c6 /src
parent2014650ab9fbf5a7d1c7334fa54ac0b1c5908915 (diff)
downloadregularization-1af73a75ccab1147a8d2387b7056f91f0642549f.tar.gz
regularization-1af73a75ccab1147a8d2387b7056f91f0642549f.tar.bz2
regularization-1af73a75ccab1147a8d2387b7056f91f0642549f.tar.xz
regularization-1af73a75ccab1147a8d2387b7056f91f0642549f.zip
removed viewer package from tree
Diffstat (limited to 'src')
-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.pycbin10542 -> 0 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/CILViewer2D.cpython-35.pycbin35633 -> 0 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/QVTKWidget.cpython-35.pycbin10099 -> 0 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/QVTKWidget2.cpython-35.pycbin1316 -> 0 bytes
-rw-r--r--src/Python/ccpi/viewer/__pycache__/__init__.cpython-35.pycbin210 -> 0 bytes
-rw-r--r--src/Python/ccpi/viewer/embedvtk.py75
11 files changed, 0 insertions, 1987 deletions
diff --git a/src/Python/ccpi/viewer/CILViewer.py b/src/Python/ccpi/viewer/CILViewer.py
deleted file mode 100644
index efcf8be..0000000
--- a/src/Python/ccpi/viewer/CILViewer.py
+++ /dev/null
@@ -1,361 +0,0 @@
-# -*- 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
deleted file mode 100644
index c1629af..0000000
--- a/src/Python/ccpi/viewer/CILViewer2D.py
+++ /dev/null
@@ -1,1126 +0,0 @@
-# -*- 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
deleted file mode 100644
index 906786b..0000000
--- a/src/Python/ccpi/viewer/QVTKWidget.py
+++ /dev/null
@@ -1,340 +0,0 @@
-################################################################################
-# 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
deleted file mode 100644
index e32e1c2..0000000
--- a/src/Python/ccpi/viewer/QVTKWidget2.py
+++ /dev/null
@@ -1,84 +0,0 @@
-################################################################################
-# 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
deleted file mode 100644
index 946188b..0000000
--- a/src/Python/ccpi/viewer/__init__.py
+++ /dev/null
@@ -1 +0,0 @@
-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
deleted file mode 100644
index 711f77a..0000000
--- a/src/Python/ccpi/viewer/__pycache__/CILViewer.cpython-35.pyc
+++ /dev/null
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
deleted file mode 100644
index 77c2ca8..0000000
--- a/src/Python/ccpi/viewer/__pycache__/CILViewer2D.cpython-35.pyc
+++ /dev/null
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
deleted file mode 100644
index 3d11b87..0000000
--- a/src/Python/ccpi/viewer/__pycache__/QVTKWidget.cpython-35.pyc
+++ /dev/null
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
deleted file mode 100644
index 2fa2eaf..0000000
--- a/src/Python/ccpi/viewer/__pycache__/QVTKWidget2.cpython-35.pyc
+++ /dev/null
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
deleted file mode 100644
index fcea537..0000000
--- a/src/Python/ccpi/viewer/__pycache__/__init__.cpython-35.pyc
+++ /dev/null
Binary files differ
diff --git a/src/Python/ccpi/viewer/embedvtk.py b/src/Python/ccpi/viewer/embedvtk.py
deleted file mode 100644
index b5eb0a7..0000000
--- a/src/Python/ccpi/viewer/embedvtk.py
+++ /dev/null
@@ -1,75 +0,0 @@
-# -*- 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