Commit b14ca6cf6669832c018d2bbe7bb5c61988d2eabf

Authored by tfmoraes
1 parent 5ba543ca

Some tests with surface creation

invesalius/data/slice_.py
@@ -583,7 +583,7 @@ class Slice(object): @@ -583,7 +583,7 @@ class Slice(object):
583 583
584 mask.matrix.flush() 584 mask.matrix.flush()
585 585
586 - ps.Publisher().sendMessage('Create surface', (mask, self.spacing)) 586 + ps.Publisher().sendMessage('Create surface', (self.matrix, self.matrix_filename, mask, self.spacing))
587 587
588 def GetOutput(self): 588 def GetOutput(self):
589 return self.blend_filter.GetOutput() 589 return self.blend_filter.GetOutput()
invesalius/data/surface.py
@@ -375,8 +375,7 @@ class SurfaceManager(): @@ -375,8 +375,7 @@ class SurfaceManager():
375 """ 375 """
376 Create surface actor, save into project and send it to viewer. 376 Create surface actor, save into project and send it to viewer.
377 """ 377 """
378 - surface_data = pubsub_evt.data  
379 - mask, spacing = pubsub_evt.data 378 + matrix, filename_img, mask, spacing = pubsub_evt.data
380 min_value, max_value = mask.threshold_range 379 min_value, max_value = mask.threshold_range
381 fill_holes = True 380 fill_holes = True
382 381
@@ -439,7 +438,6 @@ class SurfaceManager(): @@ -439,7 +438,6 @@ class SurfaceManager():
439 438
440 language = ses.Session().language 439 language = ses.Session().language
441 440
442 - filename_img = mask.temp_file  
443 overwrite = 0 441 overwrite = 0
444 442
445 if (prj.Project().original_orientation == const.CORONAL): 443 if (prj.Project().original_orientation == const.CORONAL):
@@ -453,8 +451,8 @@ class SurfaceManager(): @@ -453,8 +451,8 @@ class SurfaceManager():
453 o_piece = 1 451 o_piece = 1
454 piece_size = 40 452 piece_size = 40
455 453
456 - n_pieces = int(round(mask.matrix.shape[0] / piece_size + 0.5, 0))  
457 - print "n_pieces", n_pieces, mask.matrix.shape 454 + n_pieces = int(round(matrix.shape[0] / piece_size + 0.5, 0))
  455 + print "n_pieces", n_pieces, matrix.shape
458 456
459 q_in = multiprocessing.Queue() 457 q_in = multiprocessing.Queue()
460 q_out = multiprocessing.Queue() 458 q_out = multiprocessing.Queue()
@@ -462,7 +460,7 @@ class SurfaceManager(): @@ -462,7 +460,7 @@ class SurfaceManager():
462 p = [] 460 p = []
463 for i in xrange(n_processors): 461 for i in xrange(n_processors):
464 sp = surface_process.SurfaceProcess(pipe_in, filename_img, 462 sp = surface_process.SurfaceProcess(pipe_in, filename_img,
465 - mask.matrix.shape, mask.matrix.dtype, spacing, 463 + matrix.shape, matrix.dtype, spacing,
466 mode, min_value, max_value, 464 mode, min_value, max_value,
467 decimate_reduction, smooth_relaxation_factor, 465 decimate_reduction, smooth_relaxation_factor,
468 smooth_iterations, language, fill_holes, keep_largest, 466 smooth_iterations, language, fill_holes, keep_largest,