Copying mask after applying threshold to all slices

Thiago Franco de Moraes
1 parent 21f2f380
Showing 2 changed files with 6 additions and 5 deletions Show diff stats
invesalius/data/floodfill.pyx
invesalius/data/styles.py
@@ -120,8 +120,8 @@ def floodfill_threshold(np.ndarray[image_t, ndim=3] data, list seeds, int t0, in
     with nogil:
         while stack.size():
-            c = stack.front()
-            stack.pop_front()
+            c = stack.back()
+            stack.pop_back()
             x = c.x
             y = c.y
@@ -1802,12 +1802,10 @@ class FloodFillMaskInteractorStyle(DefaultInteractorStyle):
             position = viewer.calculate_matrix_position(coord)
         x, y, z = self.calcultate_scroll_position(position)
-        mask = self.viewer.slice_.current_mask.matrix[1:, 1:, 1:]
-
-        cp_mask = mask.copy()
         from_3d = True
         if self.config.target == "3D":
+            print "DOING 3D, manolo!"
             neighbor_iter = ((-1, 0, 0),
                              (1,  0, 0),
                              (0,  -1, 0),
@@ -1821,6 +1819,9 @@ class FloodFillMaskInteractorStyle(DefaultInteractorStyle):
                              (0,  -1, 0),
                              (0,  1, 0))
+        mask = self.viewer.slice_.current_mask.matrix[1:, 1:, 1:]
+        cp_mask = mask.copy()
+
         #  neighbor_iter = []
         #  for i in xrange(-1, 2):
             #  for j in xrange(-1, 2):