Changeset 904830e in sasview for sansguiframe/src


Ignore:
Timestamp:
Sep 8, 2011 11:20:59 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:
5637362
Parents:
a0106ac
Message:

working on batch gui

File:
1 edited

Legend:

Unmodified
Added
Removed
  • sansguiframe/src/sans/guiframe/data_processor.py

    r7ad194fa r904830e  
    44import wx 
    55import numpy 
     6import math 
     7import re 
    68import sys 
    79import copy 
     
    1719from sans.guiframe.dataFitting import Data1D 
    1820 
     21FUNC_DICT = {"sqrt": "math.sqrt", 
     22             "pow": "math.sqrt"} 
     23def parse_string(sentence, list): 
     24    """ 
     25    Return a dictionary of column label and index or row selected 
     26    :param sentence: String to parse 
     27    :param list: list of columns label 
     28    """ 
     29    toks = [] 
     30    p2 = re.compile(r'\d+') 
     31    p = re.compile(r'[\+\-\*\%\/]') 
     32    labels = p.split(sentence) 
     33    col_dict = {} 
     34    for elt in labels: 
     35        rang = None 
     36        temp_arr = [] 
     37        for label in  list: 
     38            label_pos =  elt.find(label) 
     39            if label_pos != -1: 
     40                if elt.count(',') > 0: 
     41                    new_temp = [] 
     42                    temp = elt.split(label) 
     43                    for item in temp: 
     44                        range_pos = item.find(":") 
     45                        if range_pos != -1: 
     46                            rang = p2.findall(item) 
     47                            for i in xrange(int(rang[0]), int(rang[1])+1 ): 
     48                                new_temp.append(i) 
     49                    temp_arr += new_temp 
     50                else: 
     51                    temp = elt.split(label) 
     52                    for item in temp: 
     53                        range_pos = item.find(":") 
     54                        if range_pos != -1: 
     55                            rang = p2.findall(item) 
     56                            for i in xrange(int(rang[0]), int(rang[1])+1 ): 
     57                                temp_arr.append(i) 
     58                col_dict[elt] = (label, temp_arr) 
     59    return col_dict 
    1960 
    2061class GridPage(sheet.CSheet): 
     
    5091        """ 
    5192        flag = event.CmdDown() or event.ControlDown() 
    52         cell = (event.GetRow(), event.GetCol()) 
     93        row, col = event.GetRow(), event.GetCol() 
     94        cell = (row, col) 
    5395        if not flag: 
    5496            self.selected_cells = [] 
     97            self.axis_value = [] 
     98            self.axis_label = "" 
    5599        if cell in self.selected_cells: 
    56100            self.selected_cells.remove(cell) 
    57101        else: 
    58102            self.selected_cells.append(cell) 
     103        if row > 1: 
     104            if (cell) in self.selected_cells: 
     105                self.axis_value.append(self.GetCellValue(row, col)) 
     106        self.axis_label = self.GetCellValue(row, col) 
    59107        event.Skip() 
    60108       
     
    66114        col = event.GetCol() 
    67115        if not flag: 
    68             self.selected_cols  = [] 
     116            self.selected_cols = [] 
     117            self.axis_value = [] 
     118            self.axis_label = "" 
    69119        if col not in self.selected_cols: 
    70120            self.selected_cols.append(col) 
     121        if not flag : 
     122            for row in range(2, self.GetNumberRows()+ 1): 
     123                self.axis_value.append(self.GetCellValue(row, col)) 
     124            row = 1 
     125            self.axis_label = self.GetCellValue(row, col) 
    71126        event.Skip() 
    72127         
     
    209264        return list_of_cells 
    210265     
     266    def get_column_labels(self): 
     267        """ 
     268        return dictionary of columns labels of the current page 
     269        """ 
     270        pos = self.GetSelection() 
     271        grid = self.GetPage(pos) 
     272        labels = {} 
     273        row = 0 
     274        for col in range(grid.GetNumberCols()): 
     275            label = grid.GetCellValue(row, col) 
     276            if label.strip() != "" : 
     277                labels[label.strip()] = col 
     278        return labels 
     279         
    211280    def create_axis_label(self, cell_list): 
    212281        """ 
     
    229298            for element in temp: 
    230299                temp_list.remove(element) 
    231             row_min, _  = temp_list[0]     
     300            row_min, col  = temp_list[0]     
    232301            row_max = row_min 
     302            col_name = grid.GetCellValue(0, col) 
    233303            label += str(col_name) + "[" + str(row_min) + ":" 
    234             print "temp_list", temp_list 
    235304            for index in xrange(len(temp_list)): 
    236305                prev = index - 1 
     
    243312                        label  += "[" + str(row_min) + ":" 
    244313                row_max = row 
    245                 
    246314                if (index == len(temp_list)- 1): 
    247                     label +=  str(row_max) + "]" 
    248                     print "here" 
    249         print "label ", label              
    250         return label  
     315                    label +=  str(row_max) + "]"      
     316        return label, col_name 
    251317     
    252318    def on_close_page(self, event): 
     
    295361 
    296362class GridPanel(SPanel): 
    297     def __init__(self, parent,data=None, *args, **kwds): 
     363    def __init__(self, parent, data=None, *args, **kwds): 
    298364        SPanel.__init__(self, parent , *args, **kwds) 
    299365        self.vbox = wx.BoxSizer(wx.VERTICAL) 
     
    310376        self.x_axis_label = None 
    311377        self.y_axis_label = None 
    312         self.x_axis_value = None 
    313         self.y_axis_value = None 
     378        self.x_axis_title = None 
     379        self.y_axis_title = None 
    314380        self.x_axis_unit = None 
    315381        self.y_axis_unit = None 
    316382        self.plot_button = None 
    317         self.grid = None 
     383        self.notebook = None 
    318384        self.layout_grid() 
    319385        self.layout_plotting_area() 
     
    323389        """ 
    324390        """ 
    325         if self.grid is not None: 
    326             self.grid.set_data(data) 
    327          
    328     def set_xaxis(self, label="", x=None) : 
     391        if self.notebook is not None: 
     392            self.notebook.set_data(data) 
     393         
     394    def set_xaxis(self, label="", x=None): 
     395        """ 
     396        """ 
    329397        if x is None: 
    330398            x = [] 
    331399        self.x = x 
    332         self.x_axis_value.SetValue("%s[:]" % str(label)) 
    333         self.x_axis_label.SetValue(str(label)) 
    334          
    335     def set_yaxis(self, label="", y=None) : 
     400        self.x_axis_label.SetValue("%s[:]" % str(label)) 
     401        self.x_axis_title.SetValue(str(label)) 
     402         
     403    def set_yaxis(self, label="", y=None): 
     404        """ 
     405        """ 
    336406        if y is None: 
    337407            y = [] 
    338408        self.y = y 
    339         self.y_axis_value.SetValue("%s[:]" % str(label)) 
    340         self.y_axis_label.SetValue(str(label)) 
    341          
     409        self.y_axis_label.SetValue("%s[:]" % str(label)) 
     410        self.y_axis_title.SetValue(str(label)) 
     411         
     412    def get_plot_axis(self, col, list): 
     413        """ 
     414        
     415        """ 
     416        axis = [] 
     417        pos = self.notebook.GetSelection() 
     418        grid = self.notebook.GetPage(pos) 
     419        for row in list: 
     420            label = grid.GetCellValue(row - 1, col) 
     421            if label.strip() != "": 
     422                axis.append(float(label.strip())) 
     423        return axis 
     424     
    342425    def on_plot(self, event): 
    343426        """ 
    344         plotting 
     427        Evaluate the contains of textcrtl and plot result 
    345428        """  
    346         new_plot = Data1D(x=self.x, y=self.y) 
     429        pos = self.notebook.GetSelection() 
     430        grid = self.notebook.GetPage(pos) 
     431        column_names = {} 
     432        if grid is not None: 
     433            column_names = self.notebook.get_column_labels() 
     434        #evalue x 
     435        sentence = self.x_axis_label.GetValue() 
     436        if sentence.strip() == "": 
     437            msg = "select value for x axis" 
     438            raise ValueError, msg 
     439        dict = parse_string(sentence, column_names.keys()) 
     440        for tok, (col_name, list) in dict.iteritems(): 
     441            col = column_names[col_name] 
     442            xaxis = self.get_plot_axis(col, list) 
     443            sentence = sentence.replace(tok,  
     444                                        "numpy.array(%s)" % str(xaxis)) 
     445        for key, value in FUNC_DICT.iteritems(): 
     446            sentence = sentence.replace(key.lower(), value) 
     447        x = eval(sentence) 
     448        #evaluate y 
     449        sentence = self.y_axis_label.GetValue() 
     450        if sentence.strip() == "": 
     451            msg = "select value for y axis" 
     452            raise ValueError, msg 
     453        dict = parse_string(sentence, column_names.keys()) 
     454        for tok, (col_name, list) in dict.iteritems(): 
     455            col = column_names[col_name] 
     456            yaxis = self.get_plot_axis(col, list) 
     457            sentence = sentence.replace(tok,  
     458                                        "numpy.array(%s)" % str(yaxis)) 
     459        for key, value in FUNC_DICT.iteritems(): 
     460            sentence = sentence.replace(key, value) 
     461        y = eval(sentence) 
     462        #plotting 
     463        new_plot = Data1D(x=x, y=y) 
    347464        new_plot.id =  wx.NewId() 
    348465        new_plot.group_id = wx.NewId() 
    349         title = "%s vs %s" % (self.y_axis_label.GetValue(), self.x_axis_label.GetValue()) 
    350         new_plot.xaxis(self.x_axis_label.GetValue(), self.x_axis_unit.GetValue()) 
    351         new_plot.yaxis(self.y_axis_label.GetValue(), self.y_axis_unit.GetValue()) 
    352         wx.PostEvent(self.parent.parent,  
     466        title = "%s vs %s" % (self.y_axis_title.GetValue(),  
     467                              self.x_axis_title.GetValue()) 
     468        new_plot.xaxis(self.x_axis_title.GetValue(), self.x_axis_unit.GetValue()) 
     469        new_plot.yaxis(self.y_axis_title.GetValue(), self.y_axis_unit.GetValue()) 
     470        try: 
     471            title = self.notebook.GetPageText(pos) 
     472            wx.PostEvent(self.parent.parent,  
    353473                             NewPlotEvent(plot=new_plot,  
    354                         group_id=str(new_plot.group_id), title ="batch"))     
     474                        group_id=str(new_plot.group_id), title =title))     
     475        except: 
     476            pass 
     477         
    355478    def layout_grid(self): 
    356479        """ 
    357480        Draw the area related to the grid 
    358481        """ 
    359         self.grid = Notebook(parent=self) 
    360         self.grid.set_data(self._data) 
    361         self.grid_sizer.Add(self.grid, 1, wx.EXPAND, 0) 
     482        self.notebook = Notebook(parent=self) 
     483        self.notebook.set_data(self._data) 
     484        self.grid_sizer.Add(self.notebook, 1, wx.EXPAND, 0) 
    362485        
    363486    def layout_plotting_area(self): 
     
    365488        Draw area containing options to plot 
    366489        """ 
    367         self.x_axis_label = wx.TextCtrl(self, -1) 
    368         self.y_axis_label = wx.TextCtrl(self, -1) 
    369         self.x_axis_value = wx.TextCtrl(self, -1, size=(200, -1)) 
    370         self.y_axis_value = wx.TextCtrl(self, -1, size=(200, -1)) 
     490        self.x_axis_title = wx.TextCtrl(self, -1) 
     491        self.y_axis_title = wx.TextCtrl(self, -1) 
     492        self.x_axis_label = wx.TextCtrl(self, -1, size=(200, -1)) 
     493        self.y_axis_label = wx.TextCtrl(self, -1, size=(200, -1)) 
    371494        self.x_axis_add = wx.Button(self, -1, "Add") 
    372495        self.x_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis,  
     
    380503        wx.EVT_BUTTON(self, self.plot_button.GetId(), self.on_plot) 
    381504        self.plotting_sizer.AddMany([ 
    382                     (wx.StaticText(self, -1, "x-axis label"), 1, wx.LEFT, 10), 
    383                     (self.x_axis_label, wx.TOP|wx.BOTTOM|wx.LEFT, 10), 
    384                     (wx.StaticText(self, -1, "x-axis value"), 1, wx.LEFT, 10), 
    385                     (self.x_axis_value, wx.TOP|wx.BOTTOM|wx.LEFT, 10), 
    386                     (self.x_axis_add, 1, wx.LEFT|wx.RIGHT, 0), 
    387                     (wx.StaticText(self, -1 , "unit"), 1, wx.LEFT|wx.RIGHT, 0), 
    388                     (self.x_axis_unit, 0, wx.LEFT, 0), 
    389                     (wx.StaticText(self, -1, "y-axis label"), 1, wx.LEFT, 10), 
    390                     (self.y_axis_label,  wx.TOP|wx.BOTTOM|wx.LEFT, 10), 
    391                     (wx.StaticText(self, -1, "y-axis value"), 1, wx.LEFT, 10), 
    392                     (self.y_axis_value, wx.TOP|wx.BOTTOM|wx.LEFT, 10), 
    393                     (self.y_axis_add, 1, wx.LEFT|wx.RIGHT, 0), 
    394                     (wx.StaticText(self, -1 , "unit"), 1, wx.LEFT|wx.RIGHT, 0), 
    395                     (self.y_axis_unit, 0, wx.LEFT, 0), 
     505                    (wx.StaticText(self, -1, "x-axis label"), 1, 
     506                      wx.TOP|wx.BOTTOM|wx.LEFT, 10), 
     507                    (self.x_axis_label, 1, wx.TOP|wx.BOTTOM, 10), 
     508                    (self.x_axis_add, 1, wx.TOP|wx.BOTTOM|wx.RIGHT, 10), 
     509                    (wx.StaticText(self, -1, "x-axis title"), 1,  
     510                     wx.TOP|wx.BOTTOM|wx.LEFT, 10), 
     511                    (self.x_axis_title, 1, wx.TOP|wx.BOTTOM, 10), 
     512                    (wx.StaticText(self, -1 , "unit"), 1,  
     513                     wx.TOP|wx.BOTTOM, 10), 
     514                    (self.x_axis_unit, 1, wx.TOP|wx.BOTTOM, 10), 
     515                    (wx.StaticText(self, -1, "y-axis label"), 1,  
     516                     wx.BOTTOM|wx.LEFT, 10), 
     517                    (self.y_axis_label, wx.BOTTOM, 10), 
     518                    (self.y_axis_add, 1, wx.BOTTOM|wx.RIGHT, 10), 
     519                    (wx.StaticText(self, -1, "y-axis title"), 1,  
     520                     wx.BOTTOM|wx.LEFT, 10), 
     521                    (self.y_axis_title,  wx.BOTTOM, 10), 
     522                    (wx.StaticText(self, -1 , "unit"), 1, wx.BOTTOM, 10), 
     523                    (self.y_axis_unit, 1, wx.BOTTOM, 10), 
    396524                      (-1, -1), 
    397525                      (-1, -1), 
     
    400528                      (-1, -1), 
    401529                      (-1, -1), 
    402                       (self.plot_button, 1, wx.LEFT, 0)]) 
     530                      (self.plot_button, 1, wx.LEFT|wx.BOTTOM, 10)]) 
    403531    
    404532    def on_edit_axis(self, event): 
     
    406534        Get the selected column on  the visible grid and set values for axis 
    407535        """ 
    408         cell_list = self.grid.on_edit_axis() 
    409         self.create_axis_label(cell_list) 
    410          
     536        cell_list = self.notebook.on_edit_axis() 
     537        label, title = self.create_axis_label(cell_list) 
     538        tcrtl = event.GetEventObject() 
     539        if tcrtl == self.x_axis_add: 
     540            self.edit_axis_helper(self.x_axis_label, self.x_axis_title, 
     541                                   label, title) 
     542        elif tcrtl == self.y_axis_add: 
     543            self.edit_axis_helper(self.y_axis_label, self.y_axis_title, 
     544                                   label, title) 
     545             
    411546    def create_axis_label(self, cell_list): 
    412547        """ 
     
    416551         
    417552        """ 
    418         if self.grid is not None: 
    419             return self.grid.create_axis_label(cell_list) 
     553        if self.notebook is not None: 
     554            return self.notebook.create_axis_label(cell_list) 
    420555     
    421     def edit_axis_helper(self, tcrtl_label, tcrtl_value): 
    422         """ 
    423         """ 
     556    def edit_axis_helper(self, tcrtl_label, tcrtl_title, label, title): 
     557        """ 
     558        get controls to modify 
     559        """ 
     560        tcrtl_label.SetValue(str(label)) 
     561        tcrtl_title.SetValue(str(title)) 
     562         
    424563    def add_column(self): 
    425564        """ 
    426565        """ 
    427         if self.grid is not None: 
    428             self.grid.add_column() 
     566        if self.notebook is not None: 
     567            self.notebook.add_column() 
    429568         
    430569    def on_remove_column(self): 
    431570        """ 
    432571        """ 
    433         if self.grid is not None: 
    434             self.grid.on_remove_column() 
     572        if self.notebook is not None: 
     573            self.notebook.on_remove_column() 
    435574         
    436575         
Note: See TracChangeset for help on using the changeset viewer.