Download as
Browse Code »

Commit 93ef8d8f207716eacc302f98562313ca62dab3e1

Authored by Thiago Franco de Moraes
1 parent 5027afe1
Exists in master

Navigation without a surface and closing in navigation mode was not working

Showing 1 changed file with 7 additions and 3 deletions   Show diff stats
invesalius/data/viewer_volume.py
  Diff comments   View file @93ef8d8
... ... @@ -1140,9 +1140,13 @@ class Viewer(wx.Panel):
1140 1140 return actor_arrow
1141 1141  
1142 1142 def CenterOfMass(self):
1143   - proj = prj.Project()
1144   - surface = proj.surface_dict[0].polydata
1145 1143 barycenter = [0.0, 0.0, 0.0]
  1144 + proj = prj.Project()
  1145 + try:
  1146 + surface = proj.surface_dict[0].polydata
  1147 + except KeyError:
  1148 + print("There is not any surface created")
  1149 + return barycenter
1146 1150 n = surface.GetNumberOfPoints()
1147 1151 for i in range(n):
1148 1152 point = surface.GetPoint(i)
... ... @@ -1356,9 +1360,9 @@ class Viewer(wx.Panel):
1356 1360 def OnNavigationStatus(self, nav_status, vis_status):
1357 1361 self.nav_status = nav_status
1358 1362 self.tracts_status = vis_status[1]
1359   - self.pTarget = self.CenterOfMass()
1360 1363  
1361 1364 if self.nav_status:
  1365 + self.pTarget = self.CenterOfMass()
1362 1366 if self.obj_actor:
1363 1367 self.obj_actor.SetVisibility(self.obj_state)
1364 1368 self.x_actor.SetVisibility(self.obj_state)
... ...