source: sasview/guiframe/local_perspectives/data_loader/data_loader.py @ 570cb96

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 570cb96 was 570cb96, checked in by Gervaise Alina <gervyh@…>, 14 years ago

update menu item according to discussion
  • Property mode set to 100644
File size: 8.6 KB
Line 
1
2"""
3plugin DataLoader responsible of loading data
4"""
5import os
6import sys
7import wx
8import logging
9
10from DataLoader.loader import Loader
11import DataLoader.data_info as DataInfo
12from sans.guiframe.plugin_base import PluginBase
13from sans.guiframe.events import StatusEvent
14from sans.guiframe.events import NewPlotEvent
15from sans.guiframe.dataFitting import Data1D
16from sans.guiframe.dataFitting import Data2D
17from sans.guiframe.utils import parse_name
18from sans.guiframe.gui_style import GUIFRAME
19try:
20    # Try to find a local config
21    import imp
22    path = os.getcwd()
23    if(os.path.isfile("%s/%s.py" % (path, 'local_config'))) or \
24        (os.path.isfile("%s/%s.pyc" % (path, 'local_config'))):
25        fObj, path, descr = imp.find_module('local_config', [path])
26        config = imp.load_module('local_config', fObj, path, descr) 
27    else:
28        # Try simply importing local_config
29        import local_config as config
30except:
31    # Didn't find local config, load the default
32    import config
33 
34extension_list = []
35if config.APPLICATION_STATE_EXTENSION is not None:
36    extension_list.append(config.APPLICATION_STATE_EXTENSION)
37EXTENSIONS = config.PLUGIN_STATE_EXTENSIONS + extension_list   
38
39class Plugin(PluginBase):
40   
41    def __init__(self, standalone=False):
42        PluginBase.__init__(self, name="DataLoader", standalone=standalone)
43        #Default location
44        self._default_save_location = None 
45        self.loader = Loader() 
46        self._data_menu = None 
47       
48    def populate_file_menu(self):
49        """
50        get a menu item and append it under file menu of the application
51        add load file menu item and load folder item
52        """
53        #menu for data files
54        menu_list = []
55        data_file_hint = "load one or more data in the application"
56        menu_list = [('&Load Data File(s)', data_file_hint, self._load_data)]
57        gui_style = self.parent.get_style()
58        style = gui_style & GUIFRAME.MULTIPLE_APPLICATIONS
59        style1 = gui_style & GUIFRAME.DATALOADER_ON
60        if style == GUIFRAME.MULTIPLE_APPLICATIONS:
61            #menu for data from folder
62            data_folder_hint = "load multiple data in the application"
63            menu_list.append(('&Load Data Folder', data_folder_hint, 
64                              self._load_folder))
65        return menu_list
66   
67    def unused_populate_menu(self, parent):
68        """
69        """
70        # Add menu data
71        self._data_menu = wx.Menu()
72        #menu for data files
73        data_file_id = wx.NewId()
74        data_file_hint = "load one or more data in the application"
75        self._data_menu.Append(data_file_id, 
76                         '&Load Data File(s)', data_file_hint)
77        wx.EVT_MENU(self.parent, data_file_id, self._load_data)
78        gui_style = self.parent.get_style()
79        style = gui_style & GUIFRAME.MULTIPLE_APPLICATIONS
80        style1 = gui_style & GUIFRAME.DATALOADER_ON
81        if style == GUIFRAME.MULTIPLE_APPLICATIONS:
82            #menu for data from folder
83            data_folder_id = wx.NewId()
84            data_folder_hint = "load multiple data in the application"
85            self._data_menu.Append(data_folder_id, 
86                             '&Load Data Folder', data_folder_hint)
87            wx.EVT_MENU(self.parent, data_folder_id, self._load_folder)
88           
89        return [(self._data_menu, 'Data')]
90
91    def _load_data(self, event):
92        """
93        Load data
94        """
95        path = None
96        if self._default_save_location == None:
97            self._default_save_location = os.getcwd()
98       
99        cards = self.loader.get_wildcards()
100        wlist =  '|'.join(cards)
101        style = wx.OPEN|wx.FD_MULTIPLE
102        dlg = wx.FileDialog(self.parent, 
103                            "Choose a file", 
104                            self._default_save_location, "",
105                             wlist,
106                             style=style)
107        if dlg.ShowModal() == wx.ID_OK:
108            file_list = dlg.GetPaths()
109            if len(file_list) >= 0 and not(file_list[0]is None):
110                self._default_save_location = os.path.dirname(file_list[0])
111                path = self._default_save_location
112        dlg.Destroy()
113       
114        if path is None or not file_list or file_list[0] is None:
115            return
116        self.get_data(file_list)
117       
118       
119    def can_load_data(self):
120        """
121        if return True, then call handler to laod data
122        """
123        return True
124 
125       
126    def _load_folder(self, event):
127        """
128        Load entire folder
129        """
130        path = None
131        if self._default_save_location == None:
132            self._default_save_location = os.getcwd()
133        dlg = wx.DirDialog(self.parent, "Choose a directory", 
134                           self._default_save_location,
135                            style=wx.DD_DEFAULT_STYLE)
136        if dlg.ShowModal() == wx.ID_OK:
137            path = dlg.GetPath()
138            self._default_save_location = path
139        dlg.Destroy()
140        if path is not None:
141            self._default_save_location = os.path.dirname(path)
142        else:
143            return   
144        file_list = self.get_file_path(path)
145        self.get_data(file_list)
146       
147    def load_error(self, error=None):
148        """
149        Pop up an error message.
150       
151        :param error: details error message to be displayed
152        """
153        message = "The data file you selected could not be loaded.\n"
154        message += "Make sure the content of your file"
155        message += " is properly formatted.\n\n"
156       
157        if error is not None:
158            message += "When contacting the DANSE team, mention the"
159            message += " following:\n%s" % str(error)
160        dial = wx.MessageDialog(self.parent, message, 'Error Loading File',
161                                wx.OK | wx.ICON_EXCLAMATION)
162        dial.ShowModal() 
163       
164    def get_file_path(self, path):
165        """
166        Receive a list containing folder then return a list of file
167        """
168        if os.path.isdir(path):
169            return [os.path.join(os.path.abspath(path),
170                                  file) for file in os.listdir(path)]
171   
172    def get_data(self, path, format=None):
173        """
174        """
175        message = ""
176        log_msg = ''
177        output = {}
178        error_message = ""
179        for p_file in path:
180            basename  = os.path.basename(p_file)
181            root, extension = os.path.splitext(basename)
182            if extension.lower() in EXTENSIONS:
183                log_msg = "Data Loader cannot "
184                log_msg += "load: %s\n" % str(p_file)
185                log_msg += "Try File opening ...."
186                logging.info(log_msg)
187                continue
188       
189            try:
190                temp =  self.loader.load(p_file, format)
191                if temp.__class__.__name__ == "list":
192                    for item in temp:
193                        data = self.parent.create_gui_data(item, p_file)
194                        output[data.id] = data
195                else:
196                    data = self.parent.create_gui_data(temp, p_file)
197                    output[data.id] = data
198                message = "Loading Data..." + str(p_file) + "\n"
199                self.load_update(output=output, message=message)
200            except:
201                error_message = "Error while loading Data: %s\n" % str(p_file)
202                error_message += str(sys.exc_value) + "\n"
203                self.load_update(output=output, message=error_message)
204               
205        message = "Loading Data Complete! "
206        message += log_msg
207        self.load_complete(output=output, error_message=error_message,
208                       message=message, path=path)
209           
210    def load_update(self, output=None, message=""):
211        """
212        print update on the status bar
213        """
214        if message != "":
215            wx.PostEvent(self.parent, StatusEvent(status=message,
216                                                  type="progress",
217                                                   info="warning"))
218    def load_complete(self, output, message="", error_message="", path=None):
219        """
220         post message to  status bar and return list of data
221        """
222        wx.PostEvent(self.parent, StatusEvent(status=message,
223                                              info="warning",
224                                              type="stop"))
225        if error_message != "":
226            self.load_error(error_message)
227        self.parent.add_data(data_list=output)
228   
229   
230       
231   
Note: See TracBrowser for help on using the repository browser.