diff --git a/invesalius/data/slice_data.py b/invesalius/data/slice_data.py index a2cb94b..7f61fc8 100644 --- a/invesalius/data/slice_data.py +++ b/invesalius/data/slice_data.py @@ -18,20 +18,25 @@ #-------------------------------------------------------------------------- import vtk +import constants as const + class SliceData(object): def __init__(self): - self.renderer = None self.actor = None - self.number = 0 self.cursor = None + self.number = 0 + self.orientation = 'AXIAL' + self.renderer = None self.__create_text() def __create_text(self): + colour = const.ORIENTATION_COLOUR[self.orientation] text_property = vtk.vtkTextProperty() + text_property.SetColor(colour) text_property.SetFontSize(16) text_property.SetFontFamilyToTimes() text_property.BoldOn() - #text_property.SetColor(colour) + self.text_property = text_property text_actor = vtk.vtkTextActor() text_actor.SetInput("%d" % self.number) @@ -49,6 +54,12 @@ class SliceData(object): self.renderer.AddActor(cursor.actor) self.cursor = cursor + def SetOrientation(self, orientation): + self.orientation = orientation + colour = const.ORIENTATION_COLOUR[self.orientation] + self.text_property.SetColor(colour) + self.text_actor.GetTextProperty().ShallowCopy(self.text_property) + def Hide(self): self.renderer.RemoveActor(self.actor) self.renderer.RemoveActor(self.text_actor) diff --git a/invesalius/data/viewer_slice.py b/invesalius/data/viewer_slice.py index b8e4615..39154a2 100755 --- a/invesalius/data/viewer_slice.py +++ b/invesalius/data/viewer_slice.py @@ -1021,6 +1021,7 @@ class Viewer(wx.Panel): actor = vtk.vtkImageActor() actor.SetInput(imagedata) slice_data = SliceData() + slice_data.SetOrientation(self.orientation) slice_data.renderer = renderer slice_data.actor = actor renderer.AddActor(actor) -- libgit2 0.21.2