not showing watershed markers when mask is hidden

Thiago Franco de Moraes
1 parent 14811233
Showing 1 changed file with 1 additions and 1 deletions Show diff stats
invesalius/data/slice_.py
@@ -534,7 +534,7 @@ class Slice(object):
             self.buffer_slices[orientation].vtk_image = image
             self.buffer_slices[orientation].vtk_mask = mask
-        if self.state == const.SLICE_STATE_WATERSHED:
+        if self.state == const.SLICE_STATE_WATERSHED and self.current_mask.is_shown:
             m = self.get_aux_slice('watershed', orientation, slice_number)
             tmp_vimage = converters.to_vtk(m, self.spacing, slice_number, orientation)
             cimage = self.do_custom_colour(tmp_vimage, {0: (0.0, 0.0, 0.0, 0.0),
	@@ -534,7 +534,7 @@ class Slice(object):		@@ -534,7 +534,7 @@ class Slice(object):
534	self.buffer_slices[orientation].vtk_image = image	534	self.buffer_slices[orientation].vtk_image = image
535	self.buffer_slices[orientation].vtk_mask = mask	535	self.buffer_slices[orientation].vtk_mask = mask
536		536
537	- if self.state == const.SLICE_STATE_WATERSHED:	537	+ if self.state == const.SLICE_STATE_WATERSHED and self.current_mask.is_shown:
538	m = self.get_aux_slice('watershed', orientation, slice_number)	538	m = self.get_aux_slice('watershed', orientation, slice_number)
539	tmp_vimage = converters.to_vtk(m, self.spacing, slice_number, orientation)	539	tmp_vimage = converters.to_vtk(m, self.spacing, slice_number, orientation)
540	cimage = self.do_custom_colour(tmp_vimage, {0: (0.0, 0.0, 0.0, 0.0),	540	cimage = self.do_custom_colour(tmp_vimage, {0: (0.0, 0.0, 0.0, 0.0),