ADD: Added stereo mode support

Paulo Henrique Junqueira Amorim
1 parent b7a728d9
Showing 6 changed files with 87 additions and 1 deletions Show diff stats
.gitattributes
icons/3D_glasses.png
icons/3D_glasses_original.png
invesalius/constants.py
invesalius/data/viewer_volume.py
invesalius/gui/default_viewers.py
@@ -18,6 +18,8 @@ docs/devel/sendmessages.txt -text
 docs/devel/subscribes.txt -text
 docs/user_guide_invesalius3a.odt -text
 docs/user_guide_invesalius3a.pdf -text
+icons/3D_glasses.png -text
+icons/3D_glasses_original.png -text
 icons/Floppy.png -text
 icons/annotation.png -text
 icons/brush_circle.jpg -text
@@ -38,6 +38,17 @@ DEFAULT_MEASURE_RADIUS = 1
 DEFAULT_MEASURE_TYPE = MEASURE_LINEAR
+STEREO_OFF = _("Off") 
+STEREO_RED_BLUE = _("RedBlue") 
+STEREO_CRISTAL = _("CristalEyes")
+STEREO_INTERLACED = _("Interlaced")
+STEREO_LEFT = _("Left")
+STEREO_RIGHT = _("Right")
+STEREO_DRESDEN = _("Dresden")
+STEREO_CHECKBOARD = _("Checkboard")
+STEREO_ANAGLYPH = _("Anaglyph")
+
+
 # VTK text
 TEXT_SIZE_SMALL = 11
 TEXT_SIZE = 12
@@ -171,6 +171,36 @@ class Viewer(wx.Panel):
                 'Set ball reference position')
         ps.Publisher().subscribe(self.SetBallReferencePositionBasedOnBound,
                 'Set ball reference position based on bound')
+        ps.Publisher().subscribe(self.SetStereoMode, 'Set stereo mode')
+
+    def SetStereoMode(self, pubsub_evt):
+        mode = pubsub_evt.data
+        ren_win = self.interactor.GetRenderWindow()
+        
+        if mode == const.STEREO_OFF:
+            ren_win.StereoRenderOff()
+        else:
+            ren_win.StereoRenderOn()
+
+            if mode == const.STEREO_RED_BLUE:
+                ren_win.SetStereoTypeToRedBlue()
+            elif mode == const.STEREO_CRISTAL:
+                ren_win.SetStereoTypeToCrystalEyes()
+            elif mode == const.STEREO_INTERLACED:
+                ren_win.SetStereoTypeToInterlaced()
+            elif mode == const.STEREO_LEFT:
+                ren_win.SetStereoTypeToLeft()
+            elif mode == const.STEREO_RIGHT:
+                ren_win.SetStereoTypeToRight()
+            elif mode == const.STEREO_DRESDEN:
+                ren_win.SetStereoTypeToDresden()
+            elif mode == const.STEREO_CHECKBOARD:
+                ren_win.SetStereoTypeToCheckerboard()
+            elif mode == const.STEREO_ANAGLYPH:
+                ren_win.SetStereoTypeToAnaglyph()
+
+        self.interactor.Render()
+
     def CreateBallReference(self):
         self.ball_reference = vtk.vtkSphereSource()
@@ -303,7 +303,7 @@ import wx.lib.pubsub as ps
 import constants as const
 import widgets.colourselect as csel
-[BUTTON_RAYCASTING, BUTTON_VIEW, BUTTON_SLICE_PLANE] = [wx.NewId() for num in xrange(3)]
+[BUTTON_RAYCASTING, BUTTON_VIEW, BUTTON_SLICE_PLANE, BUTTON_3D_STEREO] = [wx.NewId() for num in xrange(4)]
 RAYCASTING_TOOLS = wx.NewId()
 ID_TO_NAME = {}
@@ -311,6 +311,9 @@ ID_TO_TOOL = {}
 ID_TO_TOOL_ITEM = {}
 TOOL_STATE = {}
 ID_TO_ITEMSLICEMENU = {}
+ID_TO_ITEM_3DSTEREO = {}
+ID_TO_STEREO_NAME = {}
+
 class VolumeViewerCover(wx.Panel):
     def __init__(self, parent):
@@ -337,15 +340,24 @@ class VolumeToolPanel(wx.Panel):
                                     wx.BITMAP_TYPE_PNG)
+        BMP_3D_STEREO = wx.Bitmap("../icons/3D_glasses.png",
+                                    wx.BITMAP_TYPE_PNG)
+
+
         button_raycasting = pbtn.PlateButton(self, BUTTON_RAYCASTING,"",
                 BMP_RAYCASTING, style=pbtn.PB_STYLE_SQUARE,
                 size=(24,24))
+        button_stereo = pbtn.PlateButton(self, BUTTON_3D_STEREO,"",
+                BMP_3D_STEREO, style=pbtn.PB_STYLE_SQUARE,
+                    size=(24,24))
+
         button_slice_plane = self.button_slice_plane = pbtn.PlateButton(self, BUTTON_SLICE_PLANE,"",
         BMP_SLICE_PLANE, style=pbtn.PB_STYLE_SQUARE,
         size=(24,24))
         self.button_raycasting = button_raycasting
+        self.button_stereo = button_stereo
         # VOLUME VIEW ANGLE BUTTON
         BMP_FRONT = wx.Bitmap(ID_TO_BMP[const.VOL_FRONT][1],
@@ -373,6 +385,8 @@ class VolumeToolPanel(wx.Panel):
         sizer.Add(button_raycasting, 0, wx.TOP|wx.BOTTOM, 1)
         sizer.Add(button_view, 0, wx.TOP|wx.BOTTOM, 1)
         sizer.Add(button_slice_plane, 0, wx.TOP|wx.BOTTOM, 1)
+        sizer.Add(button_stereo, 0, wx.TOP|wx.BOTTOM, 1)
+
         sizer.Fit(self)
@@ -401,11 +415,15 @@ class VolumeToolPanel(wx.Panel):
         self.button_raycasting.Bind(wx.EVT_LEFT_DOWN, self.OnButtonRaycasting)
         self.button_view.Bind(wx.EVT_LEFT_DOWN, self.OnButtonView)
         self.button_colour.Bind(csel.EVT_COLOURSELECT, self.OnSelectColour)
+        self.button_stereo.Bind(wx.EVT_LEFT_DOWN, self.OnButtonStereo)
     def OnButtonRaycasting(self, evt):
         # MENU RELATED TO RAYCASTING TYPES
         self.button_raycasting.PopupMenu(self.menu_raycasting)
+    def OnButtonStereo(self, evt):
+        self.button_stereo.PopupMenu(self.stereo_menu)
+
     def OnButtonView(self, evt):
         self.button_view.PopupMenu(self.menu_view)
@@ -481,6 +499,25 @@ class VolumeToolPanel(wx.Panel):
         slice_plane_menu.Bind(wx.EVT_MENU, self.OnMenuPlaneSlice)
+
+        #3D Stereo Buttons
+        self.stereo_menu = stereo_menu = wx.Menu()
+        itens = [const.STEREO_OFF, const.STEREO_RED_BLUE,const.STEREO_ANAGLYPH, const.STEREO_CRISTAL, 
+                 const.STEREO_INTERLACED, const.STEREO_LEFT, const.STEREO_RIGHT, const.STEREO_DRESDEN, 
+                 const.STEREO_CHECKBOARD]
+ 
+        for value in itens:
+            new_id = wx.NewId()
+
+            item = wx.MenuItem(stereo_menu, new_id, value, 
+                                                kind = wx.ITEM_RADIO)
+
+            ID_TO_ITEM_3DSTEREO[new_id] = item
+            ID_TO_STEREO_NAME[new_id] = value 
+            stereo_menu.AppendItem(item)
+
+        stereo_menu.Bind(wx.EVT_MENU, self.OnMenuStereo)
+
         self.Fit()
         self.Update()
@@ -508,6 +545,12 @@ class VolumeToolPanel(wx.Panel):
         else:
             ps.Publisher().sendMessage('Enable plane', label)
+    def OnMenuStereo(self, evt):
+        id = evt.GetId() 
+        mode = ID_TO_STEREO_NAME[id]
+        ps.Publisher().sendMessage('Set stereo mode', mode)
+
+
     def Uncheck(self, pubsub_evt):        
         for item in self.slice_plane_menu.GetMenuItems():
             if (item.IsChecked()):
	@@ -38,6 +38,17 @@ DEFAULT_MEASURE_RADIUS = 1		@@ -38,6 +38,17 @@ DEFAULT_MEASURE_RADIUS = 1
38	DEFAULT_MEASURE_TYPE = MEASURE_LINEAR	38	DEFAULT_MEASURE_TYPE = MEASURE_LINEAR
39		39
40		40
		41	+STEREO_OFF = _("Off")
		42	+STEREO_RED_BLUE = _("RedBlue")
		43	+STEREO_CRISTAL = _("CristalEyes")
		44	+STEREO_INTERLACED = _("Interlaced")
		45	+STEREO_LEFT = _("Left")
		46	+STEREO_RIGHT = _("Right")
		47	+STEREO_DRESDEN = _("Dresden")
		48	+STEREO_CHECKBOARD = _("Checkboard")
		49	+STEREO_ANAGLYPH = _("Anaglyph")
		50	+
		51	+
41	# VTK text	52	# VTK text
42	TEXT_SIZE_SMALL = 11	53	TEXT_SIZE_SMALL = 11
43	TEXT_SIZE = 12	54	TEXT_SIZE = 12
	@@ -171,6 +171,36 @@ class Viewer(wx.Panel):		@@ -171,6 +171,36 @@ class Viewer(wx.Panel):
171	'Set ball reference position')	171	'Set ball reference position')
172	ps.Publisher().subscribe(self.SetBallReferencePositionBasedOnBound,	172	ps.Publisher().subscribe(self.SetBallReferencePositionBasedOnBound,
173	'Set ball reference position based on bound')	173	'Set ball reference position based on bound')
		174	+ ps.Publisher().subscribe(self.SetStereoMode, 'Set stereo mode')
		175	+
		176	+ def SetStereoMode(self, pubsub_evt):
		177	+ mode = pubsub_evt.data
		178	+ ren_win = self.interactor.GetRenderWindow()
		179	+
		180	+ if mode == const.STEREO_OFF:
		181	+ ren_win.StereoRenderOff()
		182	+ else:
		183	+ ren_win.StereoRenderOn()
		184	+
		185	+ if mode == const.STEREO_RED_BLUE:
		186	+ ren_win.SetStereoTypeToRedBlue()
		187	+ elif mode == const.STEREO_CRISTAL:
		188	+ ren_win.SetStereoTypeToCrystalEyes()
		189	+ elif mode == const.STEREO_INTERLACED:
		190	+ ren_win.SetStereoTypeToInterlaced()
		191	+ elif mode == const.STEREO_LEFT:
		192	+ ren_win.SetStereoTypeToLeft()
		193	+ elif mode == const.STEREO_RIGHT:
		194	+ ren_win.SetStereoTypeToRight()
		195	+ elif mode == const.STEREO_DRESDEN:
		196	+ ren_win.SetStereoTypeToDresden()
		197	+ elif mode == const.STEREO_CHECKBOARD:
		198	+ ren_win.SetStereoTypeToCheckerboard()
		199	+ elif mode == const.STEREO_ANAGLYPH:
		200	+ ren_win.SetStereoTypeToAnaglyph()
		201	+
		202	+ self.interactor.Render()
		203	+
174		204
175	def CreateBallReference(self):	205	def CreateBallReference(self):
176	self.ball_reference = vtk.vtkSphereSource()	206	self.ball_reference = vtk.vtkSphereSource()