Changeset dc79cfc in sasview for sansguiframe/src/sans/guiframe/gui_manager.py

Timestamp:

Sep 12, 2011 11:46:50 AM (13 years ago)

Author:

Gervaise Alina <gervyh@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

8aa5788

Parents:

83eb1b52

Message:

allow reading batch result from file

File:

• sansguiframe/src/sans/guiframe/gui_manager.py (modified) (4 diffs)

sansguiframe/src/sans/guiframe/gui_manager.py

@@ -294 +294 @@
         time_str = time.strftime("%b %d %H;%M of %Y", t)
         name = "Batch"  +  time_str
-        details = "File Generated by %s : %s.\n" % (APPLICATION_NAME,
+        details = "File Generated by %s : %s" % (APPLICATION_NAME,
                                                      str(plugin_name))
         details += "on %s.\n" % time_str 
@@ -314 +314 @@
         self.batch_frame.Show(True)
         
+    def on_read_batch_tofile(self, event):
+        """
+        Open a file dialog , extract the file to read and display values
+        into a grid
+        """
+        path = None
+        if self._default_save_location == None:
+            self._default_save_location = os.getcwd()
+        
+        dlg = wx.FileDialog(self, 
+                            "Choose a file", 
+                            self._default_save_location, "",
+                             "(*.csv)|*.csv| Text Files (*.txt)|*.txt")
+        if dlg.ShowModal() == wx.ID_OK:
+            path = dlg.GetPath()
+            if path is not None:
+                self._default_save_location = os.path.dirname(path)
+        dlg.Destroy()
+        
+        self.read_batch_tofile(file_name=path)
+        
+    def read_batch_tofile(self, file_name):
+        """
+        Extract value from file name and Display them into a grid
+        """
+        if file_name is None or file_name.strip() == "":
+            return
+        data = {}
+        fd = open(file_name, 'r')
+        _, ext = os.path.splitext(file_name)
+        separator = "\t"
+        if ext.lower() == ".csv":
+            separator = ","
+        buffer = fd.read()
+        lines = buffer.split('\n')
+        column_names_line  = ""
+        index = None
+        for index in range(len(lines)):
+            line = lines[index]
+            if line.find(separator) != -1:
+                #found the first line containing the label
+                column_names_line = line
+                first_data_index = index
+                break 
+        if column_names_line.strip() == "" or index is None:
+            return 
+        col_name_toks = column_names_line.split(separator)
+        c_index = 0
+        for col_index in range(len(col_name_toks)):
+            c_name = col_name_toks[col_index]
+            if c_name.strip() != "":
+                data[c_name] = [ lines[row].split(separator)[c_index]
+                                for row in range(index + 1, len(lines)-1)]
+                c_index += 1
+        self.open_with_localapp(data=data)
+        
     def write_batch_tofile(self, data, file_name, details=""):
         """
@@ -329 +385 @@
             separator = ","
         fd.write(str(details))
-        fd.write(separator)
-        fd.write('\n')
         for col_name  in data.keys():
              fd.write(str(col_name))
@@ -1326 +1380 @@
             wx.EVT_MENU(self, id, self._on_save_application)
             self._file_menu.AppendSeparator()
-        
+        self._file_menu.AppendSeparator()
+        id = wx.NewId()
+        hint = "Display content of the file into a grid"
+        self._file_menu.Append(id, '&Open Batch Result', hint) 
+        wx.EVT_MENU(self, id, self.on_read_batch_tofile)
+        self._file_menu.AppendSeparator()
         id = wx.NewId()
         self._file_menu.Append(id, '&Quit', 'Exit')