fix problem when in watershed and user remove the showing mask

Thiago Franco de Moraes
1 parent c8c3f8af
Showing 1 changed file with 1 additions and 1 deletions Show diff stats
invesalius/data/slice_.py
@@ -627,7 +627,7 @@ class Slice(metaclass=utils.Singleton):
             self.buffer_slices[orientation].vtk_image = image
             self.buffer_slices[orientation].vtk_mask = mask
  
-        if self.to_show_aux == "watershed" and self.current_mask.is_shown:
+        if self.to_show_aux == "watershed" and self.current_mask is not None 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(
...	...	@@ -627,7 +627,7 @@ class Slice(metaclass=utils.Singleton):
627	627	self.buffer_slices[orientation].vtk_image = image
628	628	self.buffer_slices[orientation].vtk_mask = mask
629	629
630		- if self.to_show_aux == "watershed" and self.current_mask.is_shown:
	630	+ if self.to_show_aux == "watershed" and self.current_mask is not None and self.current_mask.is_shown:
631	631	m = self.get_aux_slice("watershed", orientation, slice_number)
632	632	tmp_vimage = converters.to_vtk(m, self.spacing, slice_number, orientation)
633	633	cimage = self.do_custom_colour(
...	...