User can use the whitaker smoothing

Thiago Franco de Moraes
1 parent 8a446404
Showing 3 changed files with 26 additions and 5 deletions Show diff stats
invesalius/data/surface.py
invesalius/data/surface_process.py
invesalius/gui/dialogs.py
@@ -24,6 +24,7 @@ import random
 import tempfile
 import weakref
+import numpy as np
 import vtk
 import wx
 from wx.lib.pubsub import pub as Publisher
@@ -37,6 +38,8 @@ import surface_process
 import utils as utl
 import vtk_utils as vu
+from data import smooth_cy
+
 try:
     import ca_smoothing
 except ImportError:
@@ -465,13 +468,27 @@ class SurfaceManager():
         q_in = multiprocessing.Queue()
         q_out = multiprocessing.Queue()
+        if algorithm == 'Whitaker':
+            iteractions = 10
+            bsize = 4
+            mask_tfile = tempfile.mktemp()
+            mmask = np.memmap(mask_tfile, shape=mask.matrix.shape, dtype='float64', mode='w+')
+            smooth_cy.smooth(mask.matrix, iteractions, bsize, mmask)
+            mmask[0, :, :] = -1
+            mmask[:, 0, :] = -1
+            mmask[:, :, 0] = -1
+
+        else:
+            mask_tfile = mask.temp_file
+            mmask = mask.matrix
+
         p = []
         for i in xrange(n_processors):
             sp = surface_process.SurfaceProcess(pipe_in, filename_img,
                                                 matrix.shape, matrix.dtype,
-                                                mask.temp_file,
-                                                mask.matrix.shape,
-                                                mask.matrix.dtype,
+                                                mask_tfile,
+                                                mmask.shape,
+                                                mmask.dtype,
                                                 spacing,
                                                 mode, min_value, max_value,
                                                 decimate_reduction,
@@ -151,7 +151,10 @@ class SurfaceProcess(multiprocessing.Process):
         contour.SetInputData(image)
         #contour.SetInput(flip.GetOutput())
         if self.from_binary:
-            contour.SetValue(0, 127) # initial threshold
+            if self.algorithm == 'Whitaker':
+                contour.SetValue(0, 0.0) # initial threshold
+            else:
+                contour.SetValue(0, 127) # initial threshold
         else:
             contour.SetValue(0, self.min_value) # initial threshold
             contour.SetValue(1, self.max_value) # final threshold
@@ -1255,7 +1255,8 @@ class SurfaceMethodPanel(wx.Panel):
         self.mask_edited = mask_edited
         self.alg_types = {_(u'Default'): 'Default',
                           _(u'Context aware smoothing'): 'ca_smoothing',
-                          _(u'Binary'): 'Binary'}
+                          _(u'Binary'): 'Binary',
+                          _(u'Reducing Aliasing'): 'Whitaker'}
         self.edited_imp = [_(u'Default'), ]
         self._build_widgets()