display init and end slice when showing mip

Thiago Franco de Moraes
1 parent 89562751
Showing 2 changed files with 13 additions and 4 deletions Show diff stats
invesalius/data/slice_data.py
invesalius/data/viewer_slice.py
@@ -136,9 +136,13 @@ class SliceData(object):
         self.overlay_renderer.AddActor(cursor.actor)
         self.cursor = cursor
-    def SetNumber(self, number):
-        self.number = number
-        self.text.SetValue("%d" % self.number)
+    def SetNumber(self, init, end=None):
+        if end is None:
+            self.number = init
+            self.text.SetValue("%d" % self.number)
+        else:
+            self.number = init
+            self.text.SetValue("%d - %d" % (init, end))
         self.text.SetPosition(const.TEXT_POS_LEFT_DOWN_ZERO)
     def SetOrientation(self, orientation):
@@ -1254,7 +1254,12 @@ class Viewer(wx.Panel):
         for actor in self.actors_by_slice_number.get(index, []):
             self.slice_data.renderer.AddActor(actor)
-        self.slice_data.SetNumber(index)
+        if self.slice_._type_projection == const.PROJECTION_NORMAL:
+            self.slice_data.SetNumber(index)
+        else:
+            max_slices = self.slice_.GetMaxSliceNumber(self.orientation)
+            end = min(max_slices, index + self.number_slices - 1)
+            self.slice_data.SetNumber(index, end)
         self.__update_display_extent(image)
         self.cross.SetModelBounds(self.slice_data.actor.GetBounds())