added z

Thiago Franco de Moraes
1 parent 5a215327
Showing 1 changed file with 4 additions and 1 deletions Show diff stats
invesalius/data/styles.py
@@ -2177,7 +2177,10 @@ class FloodFillSegmentInteractorStyle(DefaultInteractorStyle):
             for j in xrange(int(y-1), int(y+2)):
                 if j < 0 or j >= bool_mask.shape[1]:
                     continue
-                bool_mask[0, j, i] = True
+                for k in xrange(int(z-1), int(z+2)):
+                    if k < 0 or k >= bool_mask.shape[0]:
+                        continue
+                    bool_mask[k, j, i] = True
  
  
         for i in xrange(3):
...	...	@@ -2177,7 +2177,10 @@ class FloodFillSegmentInteractorStyle(DefaultInteractorStyle):
2177	2177	for j in xrange(int(y-1), int(y+2)):
2178	2178	if j < 0 or j >= bool_mask.shape[1]:
2179	2179	continue
2180		- bool_mask[0, j, i] = True
	2180	+ for k in xrange(int(z-1), int(z+2)):
	2181	+ if k < 0 or k >= bool_mask.shape[0]:
	2182	+ continue
	2183	+ bool_mask[k, j, i] = True
2181	2184
2182	2185
2183	2186	for i in xrange(3):
...	...