Add progress bar to saving project

invesalius/control.py

@@ -379,6 +379,7 @@ def SaveProject(self, path=None, compress=False):
 
  proj = prj.Project()
  try:
+ dlg=dialogs.SaveProjectProgressWindow()
  prj.Project().SavePlistProject(dirpath, filename, compress)
  except PermissionError as err:
  if wx.GetApp() is None:

invesalius/gui/dialogs.py

@@ -17,7 +17,6 @@
 # PARTICULAR. Consulte a Licenca Publica Geral GNU para obter mais
 # detalhes.
 #--------------------------------------------------------------------------
-
 import itertools
 import os
 import random
@@ -5391,6 +5390,29 @@ def Update(self, msg=None, value=None):
  def Close(self):
  self.dlg.Destroy()
 
+class SaveProjectProgressWindow(object):
+ def __init__(self):
+ self.title = "InVesalius 3"
+ self.msg = _("Saving project ...")
+ self.style = wx.PD_APP_MODAL | wx.PD_APP_MODAL | wx.PD_CAN_ABORT | wx.PD_ELAPSED_TIME
+ self.dlg = wx.ProgressDialog(self.title,
+ self.msg,
+ parent=None,
+ style=self.style,
+ maximum=6)
+ self.running = True
+ self.error = None
+ self.dlg.Show()
+
+ Publisher.subscribe(self.update, 'update_progress')
+ Publisher.subscribe(self.destroy, 'destroy_progress')
+
+
+ def update(self, value):
+ self.dlg.Update(value)
+
+ def destroy(self):
+ wx.CallAfter(self.dlg.Destroy)
 
 class GoToDialog(wx.Dialog):
  def __init__(self, title=_("Go to slice ..."), init_orientation=const.AXIAL_STR):

invesalius/project.py

@@ -241,20 +241,23 @@ def SavePlistProject(self, dir_, filename, compress=False):
  project['matrix'] = matrix
  filelist[self.matrix_filename] = 'matrix.dat'
  #shutil.copyfile(self.matrix_filename, filename_tmp)
+ Publisher.sendMessage('update_progress', value=1)
 
  # Saving the masks
  masks = {}
  for index in self.mask_dict:
  masks[str(index)] = self.mask_dict[index].SavePlist(dir_temp,
  filelist)
  project['masks'] = masks
+ Publisher.sendMessage('update_progress',value=2)
 
  # Saving the surfaces
  surfaces = {}
  for index in self.surface_dict:
  surfaces[str(index)] = self.surface_dict[index].SavePlist(dir_temp,
  filelist)
  project['surfaces'] = surfaces
+ Publisher.sendMessage('update_progress', value=3)
 
  # Saving the measurements
  measurements = self.GetMeasuresDict()
@@ -264,10 +267,12 @@ def SavePlistProject(self, dir_, filename, compress=False):
  plistlib.dump(measurements, f)
  filelist[temp_mplist] = measurements_filename
  project['measurements'] = measurements_filename
+ Publisher.sendMessage('update_progress', value=4)
  os.close(fd_mplist)
 
  # Saving the annotations (empty in this version)
  project['annotations'] = {}
+ Publisher.sendMessage('update_progress', value=5)
 
  # Saving the main plist
  temp_fd, temp_plist = tempfile.mkstemp()
@@ -286,6 +291,10 @@ def SavePlistProject(self, dir_, filename, compress=False):
  for f in filelist:
  if filelist[f].endswith('.plist'):
  os.remove(f)
+ Publisher.sendMessage('update_progress', value=6)
+
+ Publisher.sendMessage('destroy_progress')
+ wx.Yield()
 
  def OpenPlistProject(self, filename):
  if not const.VTK_WARNING: