ENH: Returned dialog while generating MPR

Paulo Henrique Junqueira Amorim
1 parent 8d148544
Showing 2 changed files with 8 additions and 8 deletions Show diff stats
invesalius/data/imagedata_utils.py
invesalius/reader/dicom_reader.py
@@ -461,6 +461,9 @@ def dcm2memmap(files, slice_size, orientation):
     From a list of dicom files it creates memmap file in the temp folder and
     returns it and its related filename.
     """
+    message = _("Generating multiplanar visualization...")
+    update_progress= vtk_utils.ShowProgress(len(files), dialog_type = "ProgressDialog")
+
     temp_file = tempfile.mktemp()
     if orientation == 'SAGITTAL':
@@ -472,6 +475,7 @@ def dcm2memmap(files, slice_size, orientation):
     print shape
     matrix = numpy.memmap(temp_file, mode='w+', dtype='int16', shape=shape)
     dcm_reader = vtkgdcm.vtkGDCMImageReader()
+    cont = 0
     for n, f in enumerate(files):
         dcm_reader.SetFileName(f)
         dcm_reader.Update()
@@ -486,6 +490,8 @@ def dcm2memmap(files, slice_size, orientation):
         else:
             array.shape = matrix.shape[1], matrix.shape[2]
             matrix[n] = array
+        update_progress(cont,message)
+        cont += 1
     matrix.flush()
     return matrix, temp_file
@@ -86,27 +86,21 @@ tag_labels = {}
 main_dict = {}
 dict_file = {}
-class LoadDicom:#(threading.Thread):
+class LoadDicom:
     def __init__(self, grouper, filepath):
-        #threading.Thread.__init__(self)
         self.grouper = grouper
         if sys.platform == 'win32':
             self.filepath = filepath.encode(utils.get_system_encoding())
         else:
             self.filepath = filepath
+        
         self.run()
     def run(self):
         grouper = self.grouper
-        #while 1:
-            
-        #filepath = q.get()
-        #if not filepath:
-        #    break
-        
         reader = gdcm.Reader()
         reader.SetFileName(self.filepath)