Checking if exist mask when showing or hiding mask in datanotebook

Thiago Franco de Moraes
1 parent 158a65d9
Showing 1 changed file with 4 additions and 2 deletions Show diff stats
invesalius/gui/data_notebook.py
@@ -445,10 +445,12 @@ class MasksListCtrlPanel(wx.ListCtrl, listmix.TextEditMixin):
                 self.SetItemImage(key, 0)
  
     def __hide_current_mask(self, pubsub_evt):
-        self.SetItemImage(self.current_index, 0)
+        if self.mask_list_index:
+            self.SetItemImage(self.current_index, 0)
  
     def __show_current_mask(self, pubsub_evt):
-        self.SetItemImage(self.current_index, 1)
+        if self.mask_list_index:
+            self.SetItemImage(self.current_index, 1)
  
     def __init_columns(self):
...	...	@@ -445,10 +445,12 @@ class MasksListCtrlPanel(wx.ListCtrl, listmix.TextEditMixin):
445	445	self.SetItemImage(key, 0)
446	446
447	447	def __hide_current_mask(self, pubsub_evt):
448		- self.SetItemImage(self.current_index, 0)
	448	+ if self.mask_list_index:
	449	+ self.SetItemImage(self.current_index, 0)
449	450
450	451	def __show_current_mask(self, pubsub_evt):
451		- self.SetItemImage(self.current_index, 1)
	452	+ if self.mask_list_index:
	453	+ self.SetItemImage(self.current_index, 1)
452	454
453	455	def __init_columns(self):
454	456
...	...