Changeset 75fbd17 in sasview for guiframe/local_perspectives


Ignore:
Timestamp:
Feb 9, 2011 12:04:24 AM (14 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:
c83a5af
Parents:
8240eab
Message:

working on save state

File:
1 edited

Legend:

Unmodified
Added
Removed

  • guiframe/local_perspectives/data_loader/data_loader.py

    rec489f5 r75fbd17  
    1616from sans.guiframe.dataFitting import Data2D 
    1717from sans.guiframe.utils import parse_name 
     18from sans.guiframe.gui_style import GUIFRAME 
    1819 
    19 STATE_FILE_EXT = ['SansView files (*.svs)|*.svs', 
    20                   'P(r) files (*.prv)|*.prv', 
    21                   'Fitting files (*.fitv)|*.fitv', 
    22                   'Invariant files (*.inv)|*.inv'] 
    2320EXTENSIONS = ['.svs', '.prv','.fitv', '.inv'] 
    2421 
    … …  
    3027        self._default_save_location = None   
    3128        self.data_name_dict = {} 
    32         self.loader = Loader()    
     29        self.loader = Loader()   
     30        self._data_menu = None  
    3331         
    34     def populate_file_menu(self): 
     32    def populate_menu(self, parent): 
    3533        """ 
    36         get a menu item and append it under file menu of the application 
    37         add load file menu item and load folder item 
    3834        """ 
    39          
    40         hint_load_file = "Read state's files and load them into the application" 
    41         return [["Open State from File", hint_load_file, self.load_file]] 
    42    
    43     def load_data(self, event): 
     35         # Add menu data  
     36        self._data_menu = wx.Menu() 
     37        #menu for data files 
     38        data_file_id = wx.NewId() 
     39        data_file_hint = "load one or more data in the application" 
     40        self._data_menu.Append(data_file_id,  
     41                         '&Load Data File(s)', data_file_hint) 
     42        wx.EVT_MENU(self.parent, data_file_id, self._load_data) 
     43        gui_style = self.parent.get_style() 
     44        style = gui_style & GUIFRAME.MULTIPLE_APPLICATIONS 
     45        style1 = gui_style & GUIFRAME.DATALOADER_ON 
     46        if style == GUIFRAME.MULTIPLE_APPLICATIONS: 
     47            #menu for data from folder 
     48            data_folder_id = wx.NewId() 
     49            data_folder_hint = "load multiple data in the application" 
     50            self._data_menu.Append(data_folder_id,  
     51                             '&Load Data Folder', data_folder_hint) 
     52            wx.EVT_MENU(self.parent, data_folder_id, self._load_folder) 
     53             
     54        return [(self._data_menu, 'Data')] 
     55 
     56    def _load_data(self, event): 
    4457        """ 
    4558        Load data 
    4659        """ 
    47         flag = True 
    48         file_list = self.choose_data_file(flag) 
     60        if self._default_save_location == None: 
     61            self._default_save_location = os.getcwd() 
     62         
     63        cards = self.loader.get_wildcards() 
     64        wlist =  '|'.join(cards) 
     65        style = wx.OPEN|wx.FD_MULTIPLE 
     66        dlg = wx.FileDialog(self.parent,  
     67                            "Choose a file",  
     68                            self._default_save_location, "", 
     69                             wlist, 
     70                             style=style) 
     71        if dlg.ShowModal() == wx.ID_OK: 
     72            file_list = dlg.GetPaths() 
     73            if len(file_list) >= 0 and not(file_list[0]is None): 
     74                self._default_save_location = os.path.dirname(file_list[0]) 
     75        dlg.Destroy() 
     76         
    4977        if not file_list or file_list[0] is None: 
    5078            return 
    51         self.get_data(file_list, flag=flag) 
     79        self.get_data(file_list) 
     80        self.parent.show_data_panel(event=None) 
    5281         
    5382    def can_load_data(self): 
    … …  
    5685        """ 
    5786        return True 
    58      
    59     def load_file(self, event): 
    60         """ 
    61         Load  sansview defined files 
    62         """ 
    63         flag = False 
    64         file_list = self.choose_data_file(flag) 
    65         if not file_list or file_list[0] is None: 
    66             return 
    67         self.get_state(file_list) 
     87  
    6888        
    69     def load_folder(self, event): 
     89    def _load_folder(self, event): 
    7090        """ 
    7191        Load entire folder 
    7292        """ 
    73         flag = True 
    74         path = self.choose_data_folder(flag) 
    75         if path is None: 
    76             return 
    77         file_list = self.get_file_path(path) 
    78         self.get_data(file_list, flag=flag) 
    79         
    80     def get_wild_card(self, flag=True): 
    81         """ 
    82         :param flag: is True load only data file, else load state file 
    83          return wild cards 
    84         """ 
    85         if flag: 
    86             cards = self.loader.get_wildcards() 
    87             for item in STATE_FILE_EXT: 
    88                 if item in cards: 
    89                     cards.remove(item) 
    90         else: 
    91             cards = STATE_FILE_EXT 
    92         return '|'.join(cards) 
    93          
    94          
    95     def choose_data_file(self, flag=True): 
    96         """ 
    97         Open the file dialog to load file(s) 
    98         """ 
    99         path = None 
    10093        if self._default_save_location == None: 
    10194            self._default_save_location = os.getcwd() 
    102          
    103         cards = self.loader.get_wildcards() 
    104         wlist = self.get_wild_card(flag) 
    105         if flag: 
    106             style = wx.OPEN|wx.FD_MULTIPLE 
    107         else: 
    108             style = wx.OPEN|wx.FD_DEFAULT_STYLE 
    109              
    110         dlg = wx.FileDialog(self.parent,  
    111                             "Choose a file",  
    112                             self._default_save_location, "", 
    113                              wlist, 
    114                              style=style) 
    115         if dlg.ShowModal() == wx.ID_OK: 
    116             path = dlg.GetPaths() 
    117             if len(path) >= 0 and not(path[0]is None): 
    118                 self._default_save_location = os.path.dirname(path[0]) 
    119         dlg.Destroy() 
    120         return path 
    121      
    122     def choose_data_folder(self, flag=True): 
    123         """ 
    124         :param flag: is True load only data file, else load state file 
    125         return a list of folder to read 
    126         """ 
    127         path = None 
    128         if self._default_save_location == None: 
    129             self._default_save_location = os.getcwd() 
    130          
    131         wlist = self.get_wild_card(flag) 
    132          
    13395        dlg = wx.DirDialog(self.parent, "Choose a directory",  
    13496                           self._default_save_location, 
    … …  
    138100            self._default_save_location = path 
    139101        dlg.Destroy() 
    140         return path 
    141      
     102        if path is not None: 
     103            self._default_save_location = os.path.dirname(path) 
     104        else: 
     105            return     
     106        file_list = self.get_file_path(path) 
     107        self.get_data(file_list) 
     108        self.parent.show_data_panel(event=None) 
     109    
     110    
    142111    def load_error(self, error=None): 
    143112        """ 
    … …  
    164133            return [os.path.join(os.path.abspath(path), 
    165134                                  file) for file in os.listdir(path)] 
    166      
    167     def get_state(self, path, format=None): 
    168         """ 
    169         read state 
    170         """ 
    171         output = [] 
    172         for p_file in path: 
    173             basename  = os.path.basename(p_file) 
    174             root, extension = os.path.splitext(basename) 
    175             for extension in EXTENSIONS: 
    176                 temp = self.loader.load(p_file, extension) 
    177                 if temp.__class__.__name__ == "list": 
    178                     for item in temp: 
    179                         data = self.create_data(item, p_file) 
    180                 else: 
    181                     data = self.create_data(temp, p_file) 
    182                 output.append(data) 
    183             self.load_complete(output=output, path=path) 
    184                     
    185     def get_data(self, path, format=None, flag=True): 
     135    
     136    def get_data(self, path, format=None): 
    186137        """ 
    187138        """ 
    … …  
    193144            basename  = os.path.basename(p_file) 
    194145            root, extension = os.path.splitext(basename) 
    195             if flag: 
    196                 if extension.lower() in EXTENSIONS: 
    197                     log_msg = "Data Loader cannot " 
    198                     log_msg += "load: %s\n" % str(p_file) 
    199                     log_msg += "Try File -> open ...." 
    200                     logging.info(log_msg) 
    201                     continue 
    202             else: 
    203                 if extension.lower() not in EXTENSIONS: 
    204                     log_msg = "File Loader cannot" 
    205                     log_msg += " load: %s\n" % str(p_file) 
    206                     log_msg += "Try Data -> Load ...." 
    207                     logging.info(log_msg) 
    208                     continue 
     146            if extension.lower() in EXTENSIONS: 
     147                log_msg = "Data Loader cannot " 
     148                log_msg += "load: %s\n" % str(p_file) 
     149                log_msg += "Try File opening ...." 
     150                logging.info(log_msg) 
     151                continue 
     152         
    209153            try: 
    210154                temp =  self.loader.load(p_file, format) 
    211155                if temp.__class__.__name__ == "list": 
    212                     print "list", temp 
    213156                    for item in temp: 
    214                         data = self.create_data(item, p_file) 
     157                        data = self.parent.create_gui_data(item, p_file) 
    215158                        output.append(data) 
    216159                else: 
    217                     print "not list", [temp] 
    218                     data = self.create_data(temp, p_file) 
     160                    data = self.parent.create_gui_data(temp, p_file) 
    219161                    output.append(data) 
    220                 message = "Loading ..." + str(p_file) + "\n" 
     162                message = "Loading Data..." + str(p_file) + "\n" 
    221163                self.load_update(output=output, message=message) 
    222164            except: 
    223                 raise 
    224                 error_message = "Error while loading: %s\n" % str(p_file) 
     165                error_message = "Error while loading Data: %s\n" % str(p_file) 
    225166                error_message += str(sys.exc_value) + "\n" 
    226167                self.load_update(output=output, message=error_message) 
    227168                 
    228         message = "Loading Complete! " 
     169        message = "Loading Data Complete! " 
    229170        message += log_msg 
    230171        self.load_complete(output=output, error_message=error_message, 
    231                        message=message, path=path, flag=flag) 
     172                       message=message, path=path) 
    232173             
    233     
    234174    def load_update(self, output=None, message=""): 
    235175        """ 
    … …  
    240180                                                  type="progress", 
    241181                                                   info="warning")) 
    242          
    243     def load_complete(self, output, message="", error_message="",  
    244                       path=None, flag=True): 
     182    def load_complete(self, output, message="", error_message="", path=None): 
    245183        """ 
    246184         post message to  status bar and return list of data 
    … …  
    251189        if error_message != "": 
    252190            self.load_error(error_message) 
    253         self.parent.add_data(output, flag=flag) 
     191        self.parent.add_data(output) 
     192    
     193     
    254194         
    255     def create_data(self, data, path): 
    256         """ 
    257         Receive data from loader and create a data to use for guiframe 
    258         """ 
    259          
    260         if issubclass(DataInfo.Data2D, data.__class__): 
    261             new_plot = Data2D(image=None, err_image=None)  
    262         else:  
    263             new_plot = Data1D(x=[], y=[], dx=None, dy=None) 
    264             
    265         new_plot.copy_from_datainfo(data)  
    266         data.clone_without_data(clone=new_plot)   
    267         #creating a name for data 
    268         name = "" 
    269         title = "" 
    270         file_name = "" 
    271         if path is not None: 
    272             file_name = os.path.basename(path) 
    273         if data.run: 
    274             name = data.run[0] 
    275         if name == "": 
    276             name = file_name 
    277         ## name of the data allow to differentiate data when plotted 
    278         name = parse_name(name=name, expression="_") 
    279          
    280         max_char = name.find("[") 
    281         if max_char < 0: 
    282             max_char = len(name) 
    283         name = name[0:max_char] 
    284          
    285         if name not in self.data_name_dict: 
    286             self.data_name_dict[name] = 0 
    287         else: 
    288             self.data_name_dict[name] += 1 
    289             name = name + " [" + str(self.data_name_dict[name]) + "]" 
    290         #find title 
    291         if data.title.strip(): 
    292             title = data.title 
    293         if title.strip() == "": 
    294             title = file_name 
    295          
    296         if new_plot.filename.strip() == "": 
    297             new_plot.filename = file_name 
    298          
    299         new_plot.name = name 
    300         new_plot.title = title 
    301         ## allow to highlight data when plotted 
    302         new_plot.interactive = True 
    303         ## when 2 data have the same id override the 1 st plotted 
    304         new_plot.id = name 
    305         ##group_id specify on which panel to plot this data 
    306         new_plot.group_id = name 
    307         new_plot.is_data = True 
    308         new_plot.path = path 
    309         ##post data to plot 
    310         # plot data 
    311         return new_plot 
    312          
    313         
     195    
Note: See TracChangeset for help on using the changeset viewer.