not allow to set slice_number greater than image shape and lower than 0

Thiago Franco de Moraes
1 parent 53c78d4f
Showing 1 changed file with 5 additions and 0 deletions Show diff stats
invesalius/data/viewer_slice.py
@@ -1431,6 +1431,11 @@ class Viewer(wx.Panel):
         self.overwrite_mask = flag
  
     def set_slice_number(self, index):
+        max_slice_number = sl.Slice().GetNumberOfSlices(self.orientation)
+        if index < 0:
+            index = 0
+        if index >= max_slice_number:
+            index = max_slice_number - 1
         inverted = self.mip_ctrls.inverted.GetValue()
         border_size = self.mip_ctrls.border_spin.GetValue()
         image = self.slice_.GetSlices(self.orientation, index,
...	...	@@ -1431,6 +1431,11 @@ class Viewer(wx.Panel):
1431	1431	self.overwrite_mask = flag
1432	1432
1433	1433	def set_slice_number(self, index):
	1434	+ max_slice_number = sl.Slice().GetNumberOfSlices(self.orientation)
	1435	+ if index < 0:
	1436	+ index = 0
	1437	+ if index >= max_slice_number:
	1438	+ index = max_slice_number - 1
1434	1439	inverted = self.mip_ctrls.inverted.GetValue()
1435	1440	border_size = self.mip_ctrls.border_spin.GetValue()
1436	1441	image = self.slice_.GetSlices(self.orientation, index,
...	...