Download as
Browse Code »

Commit a3665303aa23d4b54484c496d5cbb8c5dae2f857

Authored by tfmoraes
1 parent d3779482
Exists in master and in 6 other branches beta3, beta4, inv3.0.1, invesalius-libift, usp-navegador, wxgtk3_b5

ENH: A box in each slice_data

Showing 2 changed files with 74 additions and 2 deletions   Show diff stats
invesalius/data/slice_data.py
  Diff comments   View file @a366530
... ... @@ -29,6 +29,7 @@ class SliceData(object):
29 29 self.orientation = 'AXIAL'
30 30 self.renderer = None
31 31 self.__create_text()
  32 + self.__create_box()
32 33  
33 34 def __create_text(self):
34 35 colour = const.ORIENTATION_COLOUR[self.orientation]
... ... @@ -41,6 +42,42 @@ class SliceData(object):
41 42 text.SetValue(self.number)
42 43 self.text = text
43 44  
  45 + def __create_box(self):
  46 + xi = yi = 0.1
  47 + xf = yf = 200
  48 + line_i = vtk.vtkLineSource()
  49 + line_i.SetPoint1((xi, yi, 0))
  50 + line_i.SetPoint2((xf, yi, 0))
  51 + self.line_i = line_i
  52 +
  53 + line_s = vtk.vtkLineSource()
  54 + line_s.SetPoint1((xi, yf, 0))
  55 + line_s.SetPoint2((xf, yf, 0))
  56 + self.line_s = line_s
  57 +
  58 + line_l = vtk.vtkLineSource()
  59 + line_l.SetPoint1((xi, yi, 0))
  60 + line_l.SetPoint2((xi, yf, 0))
  61 + self.line_l = line_l
  62 +
  63 + line_r = vtk.vtkLineSource()
  64 + line_r.SetPoint1((xf, yi, 0))
  65 + line_r.SetPoint2((xf, yf, 0))
  66 + self.line_r = line_r
  67 +
  68 + box = vtk.vtkAppendPolyData()
  69 + box.AddInput(line_i.GetOutput())
  70 + box.AddInput(line_s.GetOutput())
  71 + box.AddInput(line_l.GetOutput())
  72 + box.AddInput(line_r.GetOutput())
  73 +
  74 + box_mapper = vtk.vtkPolyDataMapper2D()
  75 + box_mapper.SetInput(box.GetOutput())
  76 +
  77 + box_actor = vtk.vtkActor2D()
  78 + box_actor.SetMapper(box_mapper)
  79 + self.box_actor = box_actor
  80 +
44 81 def SetCursor(self, cursor):
45 82 if self.cursor:
46 83 self.renderer.RemoveActor(self.cursor.actor)
... ... @@ -55,11 +92,32 @@ class SliceData(object):
55 92 self.orientation = orientation
56 93 colour = const.ORIENTATION_COLOUR[self.orientation]
57 94 self.text.SetColour(colour)
  95 + self.box_actor.GetProperty().SetColor(colour)
  96 +
  97 + def SetSize(self, size):
  98 + w, h = size
  99 + xi = yi = 0.1
  100 + xf = w - 0.1
  101 + yf = h - 0.1
  102 +
  103 + self.line_i.SetPoint1((xi, yi, 0))
  104 + self.line_i.SetPoint2((xf, yi, 0))
  105 +
  106 + self.line_s.SetPoint1((xi, yf, 0))
  107 + self.line_s.SetPoint2((xf, yf, 0))
  108 +
  109 + self.line_l.SetPoint1((xi, yi, 0))
  110 + self.line_l.SetPoint2((xi, yf, 0))
  111 +
  112 + self.line_r.SetPoint1((xf, yi, 0))
  113 + self.line_r.SetPoint2((xf, yf, 0))
58 114  
59 115 def Hide(self):
60 116 self.renderer.RemoveActor(self.actor)
61 117 self.renderer.RemoveActor(self.text.actor)
  118 + self.renderer.RemoveActor(self.box_actor)
62 119  
63 120 def Show(self):
64 121 self.renderer.AddActor(self.actor)
65 122 self.renderer.AddActor(self.text.actor)
  123 + self.renderer.AddActor(self.box_actor)
... ...
invesalius/data/viewer_slice.py
  Diff comments   View file @a366530
... ... @@ -45,8 +45,8 @@ class Viewer(wx.Panel):
45 45 def __init__(self, prnt, orientation='AXIAL'):
46 46 wx.Panel.__init__(self, prnt, size=wx.Size(320, 300))
47 47  
48   - colour = [255*c for c in const.ORIENTATION_COLOUR[orientation]]
49   - self.SetBackgroundColour(colour)
  48 + #colour = [255*c for c in const.ORIENTATION_COLOUR[orientation]]
  49 + #self.SetBackgroundColour(colour)
50 50  
51 51 # Interactor additional style
52 52 self.modes = []#['DEFAULT']
... ... @@ -830,6 +830,7 @@ class Viewer(wx.Panel):
830 830 #self.scroll.Bind(wx.EVT_SCROLL_ENDSCROLL, self.OnScrollBarRelease)
831 831 self.interactor.Bind(wx.EVT_KEY_DOWN, self.OnKeyDown)
832 832 self.interactor.Bind(wx.EVT_CONTEXT_MENU, self.OnContextMenu)
  833 + self.Bind(wx.EVT_SIZE, self.OnSize)
833 834  
834 835 def LoadImagedata(self, pubsub_evt):
835 836 imagedata, mask_dict = pubsub_evt.data
... ... @@ -853,6 +854,9 @@ class Viewer(wx.Panel):
853 854 proportion_y = 1.0 / self.layout[1]
854 855 # The (0,0) in VTK is in bottom left. So the creation from renderers
855 856 # must be # in inverted order, from the top left to bottom right
  857 + w, h = self.interactor.GetRenderWindow().GetSize()
  858 + w *= proportion_x
  859 + h *= proportion_y
856 860 n = 0
857 861 for j in xrange(self.layout[1]-1, -1, -1):
858 862 for i in xrange(self.layout[0]):
... ... @@ -868,6 +872,7 @@ class Viewer(wx.Panel):
868 872 x, y = const.TEXT_POS_LEFT_DOWN
869 873 slice_data.text.SetPosition((x+slice_xi,y+slice_yi))
870 874 slice_data.SetCursor(self.__create_cursor())
  875 + slice_data.SetSize((w, h))
871 876 self.__update_camera(slice_data)
872 877 n += 1
873 878  
... ... @@ -1097,6 +1102,7 @@ class Viewer(wx.Panel):
1097 1102 slice_data.actor = actor
1098 1103 renderer.AddActor(actor)
1099 1104 renderer.AddActor(slice_data.text.actor)
  1105 + renderer.AddActor(slice_data.box_actor)
1100 1106 return slice_data
1101 1107  
1102 1108 def __update_camera(self, slice_data):
... ... @@ -1205,6 +1211,14 @@ class Viewer(wx.Panel):
1205 1211 if evt:
1206 1212 evt.Skip()
1207 1213  
  1214 + def OnSize(self, evt):
  1215 + w, h = self.interactor.GetRenderWindow().GetSize()
  1216 + w = float(w) / self.layout[0]
  1217 + h = float(h) / self.layout[1]
  1218 + for slice_data in self.slice_data_list:
  1219 + slice_data.SetSize((w, h))
  1220 + evt.Skip()
  1221 +
1208 1222 def set_slice_number(self, index):
1209 1223 self.slice_number = index
1210 1224 for n, slice_data in enumerate(self.slice_data_list):
... ...