Commit 61ecc054e5845ac98fd98a783583b48cbb0e8f44
1 parent
75509c1d
Exists in
ff_mask
Testings
Showing
1 changed file
with
14 additions
and
2 deletions
Show diff stats
invesalius/data/styles.py
| @@ -1806,10 +1806,22 @@ class FlooFillMaskInteractorStyle(DefaultInteractorStyle): | @@ -1806,10 +1806,22 @@ class FlooFillMaskInteractorStyle(DefaultInteractorStyle): | ||
| 1806 | (0, -1, 0), | 1806 | (0, -1, 0), |
| 1807 | (0, 1, 0)) | 1807 | (0, 1, 0)) |
| 1808 | 1808 | ||
| 1809 | + neighbor_iter = [] | ||
| 1810 | + for i in xrange(-1, 2): | ||
| 1811 | + for j in xrange(-1, 2): | ||
| 1812 | + for k in xrange(-1, 2): | ||
| 1813 | + neighbor_iter.append((i, j, k)) | ||
| 1814 | + | ||
| 1809 | if iren.GetControlKey(): | 1815 | if iren.GetControlKey(): |
| 1810 | - floodfill.floodfill_threshold(cp_mask, [[x, y, z]], 254, 255, 1, neighbor_iter, mask) | 1816 | + t0 = 254 |
| 1817 | + t1 = 255 | ||
| 1818 | + fill = 1 | ||
| 1811 | else: | 1819 | else: |
| 1812 | - floodfill.floodfill_threshold(cp_mask, [[x, y, z]], 0, 1, 254, neighbor_iter, mask) | 1820 | + t0 = 0 |
| 1821 | + t1 = 1 | ||
| 1822 | + fill = 254 | ||
| 1823 | + | ||
| 1824 | + floodfill.floodfill_threshold(cp_mask, [[x, y, z]], t0, t1, fill, neighbor_iter, mask) | ||
| 1813 | 1825 | ||
| 1814 | self.viewer.slice_.buffer_slices['AXIAL'].discard_mask() | 1826 | self.viewer.slice_.buffer_slices['AXIAL'].discard_mask() |
| 1815 | self.viewer.slice_.buffer_slices['CORONAL'].discard_mask() | 1827 | self.viewer.slice_.buffer_slices['CORONAL'].discard_mask() |