def initImages(self): global show_height, show_width, show_thickness if self.slice_assembly: collectioniteractor1 = vtk.vtkCollectionIterator() collectioniteractor1.SetCollection(self.slice_assembly) collectioniteractor1.InitTraversal() for i in xrange(self.slice_assembly.GetNumberOfItems()): collectioniteractor1.GetCurrentObject().GetViewProp( ).SetVisibility(False) collectioniteractor1.GoToNextItem() if self.ren1AssemblyList[self.line_index]: self.line_assembly_list = self.ren1AssemblyList[self.line_index] else: self.line_assembly_list = [None] * len(self.lines[self.line_index]) self.ren1AssemblyList[self.line_index] = self.line_assembly_list t0 = time.clock() self.images = [None] * len(self.lines[self.line_index]) self.roi_list = [] self.image_slice = 0 self.countRoiInLine() if self.line_assembly_list[self.image_slice]: self.slice_assembly = self.line_assembly_list[self.image_slice] collectioniteractor1 = vtk.vtkCollectionIterator() collectioniteractor1.SetCollection(self.slice_assembly) collectioniteractor1.InitTraversal() for i in xrange(self.slice_assembly.GetNumberOfItems()): collectioniteractor1.GetCurrentObject().GetViewProp( ).SetVisibility(True) collectioniteractor1.GoToNextItem() else: self.slice_assembly = vtk.vtkAssemblyPath() self.line_assembly_list[self.image_slice] = self.slice_assembly self.load_mx_num = 100 if self.load_mx_num > len(self.lines[self.line_index]): real_load_num = len(self.lines[self.line_index]) else: real_load_num = self.load_mx_num # print real_load_num, len(self.lines[self.line_index]) for i in xrange(real_load_num): x0, y0, z0 = self.lines[self.line_index][i][2:5] x_s = int((x0 * 10 / 3 - show_width / 2)) y_s = int((y0 * 10 / 3 - show_height / 2)) z_s = int(z0) tmp_image = np.zeros((show_height, show_width), dtype=np.uint16) for j in xrange((-show_thickness), show_thickness + 1): z_i = z_s + j if z_i < 0: z_i = 0 image_roi = (x_s, y_s, z_i, show_width, show_height, 1) image_j = self.loadImageFromMostd(image_roi) tmp_image = np.maximum(tmp_image, image_j) self.images[i] = tmp_image print time.clock() - t0
def sliceForward(self, obj, e): global swc, main_win, mostd_src, show_width, show_height cur_line_index = main_win.line_index cur_line = main_win.lines[cur_line_index] cur_line_length = len(cur_line) cur_slice = main_win.image_slice if cur_slice + 1 < cur_line_length: next_slice = cur_slice + 1 main_win.show_image = main_win.getImageFromLoaded(next_slice) main_win.image_str = main_win.show_image.tostring() main_win.im_importer.CopyImportVoidPointer(main_win.image_str, len(main_win.image_str)) main_win.im_importer.Modified() main_win.im_importer.Update() main_win.imageLevelFilter.SetInputConnection( main_win.im_importer.GetOutputPort()) main_win.imageLevelFilter.Modified() main_win.imageLevelFilter.Update() main_win.image_actor.SetInput( main_win.imageLevelFilter.GetOutput()) main_win.image_actor.Modified() main_win.image_slice = next_slice if (main_win.image_slice < 5) or (main_win.image_slice > len( main_win.lines[main_win.line_index]) - 5): main_win.sphere_actor_ren2.SetVisibility(False) else: main_win.sphere_actor_ren2.SetVisibility(True) if main_win.slice_assembly: collectionInteractor1 = vtk.vtkCollectionIterator() collectionInteractor1.SetCollection(main_win.slice_assembly) collectionInteractor1.InitTraversal() for i in xrange(main_win.slice_assembly.GetNumberOfItems()): collectionInteractor1.GetCurrentObject().GetViewProp( ).SetVisibility(False) collectionInteractor1.GoToNextItem() main_win.slice_assembly = main_win.line_assembly_list[ main_win.image_slice] if main_win.slice_assembly: collectionInteractor1 = vtk.vtkCollectionIterator() collectionInteractor1.SetCollection(main_win.slice_assembly) collectionInteractor1.InitTraversal() for i in xrange(main_win.slice_assembly.GetNumberOfItems()): collectionInteractor1.GetCurrentObject().GetViewProp( ).SetVisibility(True) collectionInteractor1.GoToNextItem() else: main_win.slice_assembly = vtk.vtkAssemblyPath() main_win.line_assembly_list[ main_win.image_slice] = main_win.slice_assembly main_win.vtk_widget.Render() main_win.renderActorsInSlice() else: next_slice = cur_slice
def initImages(self): global show_height, show_width, show_thickness if self.slice_assembly: collectioniteractor1 = vtk.vtkCollectionIterator() collectioniteractor1.SetCollection(self.slice_assembly) collectioniteractor1.InitTraversal() for i in xrange(self.slice_assembly.GetNumberOfItems()): collectioniteractor1.GetCurrentObject().GetViewProp().SetVisibility(False) collectioniteractor1.GoToNextItem() if self.ren1AssemblyList[self.line_index]: self.line_assembly_list = self.ren1AssemblyList[self.line_index] else: self.line_assembly_list = [None] * len(self.lines[self.line_index]) self.ren1AssemblyList[self.line_index] = self.line_assembly_list t0 = time.clock() self.images = [None] * len(self.lines[self.line_index]) self.roi_list = [] self.image_slice = 0 self.countRoiInLine() if self.line_assembly_list[self.image_slice]: self.slice_assembly = self.line_assembly_list[self.image_slice] collectioniteractor1 = vtk.vtkCollectionIterator() collectioniteractor1.SetCollection(self.slice_assembly) collectioniteractor1.InitTraversal() for i in xrange(self.slice_assembly.GetNumberOfItems()): collectioniteractor1.GetCurrentObject().GetViewProp().SetVisibility(True) collectioniteractor1.GoToNextItem() else: self.slice_assembly = vtk.vtkAssemblyPath() self.line_assembly_list[self.image_slice] = self.slice_assembly self.load_mx_num = 100 if self.load_mx_num > len(self.lines[self.line_index]): real_load_num = len(self.lines[self.line_index]) else: real_load_num = self.load_mx_num # print real_load_num, len(self.lines[self.line_index]) for i in xrange(real_load_num): x0, y0, z0 = self.lines[self.line_index][i][2 : 5] x_s = int((x0 * 10 / 3 - show_width / 2)) y_s = int((y0 * 10 / 3 - show_height / 2)) z_s = int(z0) tmp_image = np.zeros((show_height, show_width), dtype=np.uint16) for j in xrange((- show_thickness), show_thickness + 1): z_i = z_s + j if z_i < 0: z_i = 0 image_roi = (x_s, y_s, z_i, show_width, show_height, 1) image_j = self.loadImageFromMostd(image_roi) tmp_image = np.maximum(tmp_image, image_j); self.images[i] = tmp_image print time.clock() - t0
def sliceForward(self, obj, e): global swc, main_win, mostd_src, show_width, show_height cur_line_index = main_win.line_index cur_line = main_win.lines[cur_line_index] cur_line_length = len(cur_line) cur_slice = main_win.image_slice if cur_slice + 1 < cur_line_length: next_slice = cur_slice + 1 main_win.show_image = main_win.getImageFromLoaded(next_slice) main_win.image_str = main_win.show_image.tostring() main_win.im_importer.CopyImportVoidPointer(main_win.image_str, len(main_win.image_str)) main_win.im_importer.Modified() main_win.im_importer.Update() main_win.imageLevelFilter.SetInputConnection(main_win.im_importer.GetOutputPort()) main_win.imageLevelFilter.Modified() main_win.imageLevelFilter.Update() main_win.image_actor.SetInput(main_win.imageLevelFilter.GetOutput()) main_win.image_actor.Modified() main_win.image_slice = next_slice if (main_win.image_slice < 5) or (main_win.image_slice > len(main_win.lines[main_win.line_index]) - 5): main_win.sphere_actor_ren2.SetVisibility(False) else: main_win.sphere_actor_ren2.SetVisibility(True) if main_win.slice_assembly: collectionInteractor1 = vtk.vtkCollectionIterator() collectionInteractor1.SetCollection(main_win.slice_assembly) collectionInteractor1.InitTraversal() for i in xrange(main_win.slice_assembly.GetNumberOfItems()): collectionInteractor1.GetCurrentObject().GetViewProp().SetVisibility(False) collectionInteractor1.GoToNextItem() main_win.slice_assembly = main_win.line_assembly_list[main_win.image_slice] if main_win.slice_assembly: collectionInteractor1 = vtk.vtkCollectionIterator() collectionInteractor1.SetCollection(main_win.slice_assembly) collectionInteractor1.InitTraversal() for i in xrange(main_win.slice_assembly.GetNumberOfItems()): collectionInteractor1.GetCurrentObject().GetViewProp().SetVisibility(True) collectionInteractor1.GoToNextItem() else: main_win.slice_assembly = vtk.vtkAssemblyPath() main_win.line_assembly_list[main_win.image_slice] = main_win.slice_assembly main_win.vtk_widget.Render() main_win.renderActorsInSlice() else: next_slice = cur_slice
def pickActor(obj, e): global iren, ren, aspath picker1 = vtk.vtkPropPicker() x, y = iren.GetEventPosition() picker1.Pick(x, y, 0, ren) print picker1.GetActor() i1 = vtk.vtkCollectionIterator() aspath.InitTraversal(i1) for i in xrange(aspath.GetNumberOfItems()): actor_i = aspath.GetNextItemAsObject() print id(actor_i) print '____________________________' aspath.InitTraversal() for i in xrange(aspath.GetNumberOfItems()): actor_i = aspath.GetNextItemAsObject() print id(actor_i)
def pickActor(obj, e): global iren, ren, aspath picker1 = vtk.vtkPropPicker() x, y = iren.GetEventPosition() picker1.Pick(x, y, 0, ren) print picker1.GetActor() i1 = vtk.vtkCollectionIterator() aspath.InitTraversal(i1) for i in xrange(aspath.GetNumberOfItems()): actor_i = aspath.GetNextItemAsObject() print id(actor_i) print '____________________________' aspath.InitTraversal() for i in xrange(aspath.GetNumberOfItems()): actor_i = aspath.GetNextItemAsObject() print id(actor_i)
import vtk c = vtk.vtkCollection() i = vtk.vtkCollectionIterator() a1 = vtk.vtkActor() a2 = vtk.vtkActor() c.AddItem(a1) c.AddItem(a2) c.InitTraversal() ai = c.GetNextItemAsObject() matrix1 = vtk.vtkMatrix4x4() print matrix1 s = vtk.vtkAssemblyPath() s.AddNode(a1, matrix1) s.AddNode(a2, matrix1) s.InitTraversal() ai = s.GetNextNode() while ai: ai = s.GetNextNode()
import vtk c = vtk.vtkCollection() i = vtk.vtkCollectionIterator() a1 = vtk.vtkActor() a2 = vtk.vtkActor() c.AddItem(a1) c.AddItem(a2) c.InitTraversal() ai = c.GetNextItemAsObject() matrix1 = vtk.vtkMatrix4x4() print matrix1 s = vtk.vtkAssemblyPath() s.AddNode(a1, matrix1) s.AddNode(a2, matrix1) s.InitTraversal() ai = s.GetNextNode() while ai: ai = s.GetNextNode()