Changeset 904830e in sasview for sansguiframe/src/sans/guiframe/data_processor.py

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:

• sansguiframe/src/sans/guiframe/data_processor.py (modified) (15 diffs)

sansguiframe/src/sans/guiframe/data_processor.py

@@ -4 +4 @@
 import wx
 import numpy
+import math
+import re
 import sys
 import copy
@@ -17 +19 @@
 from sans.guiframe.dataFitting import Data1D
 
+FUNC_DICT = {"sqrt": "math.sqrt",
+             "pow": "math.sqrt"}
+def parse_string(sentence, list):
+    """
+    Return a dictionary of column label and index or row selected
+    :param sentence: String to parse
+    :param list: list of columns label
+    """
+    toks = []
+    p2 = re.compile(r'\d+')
+    p = re.compile(r'[\+\-\*\%\/]')
+    labels = p.split(sentence)
+    col_dict = {}
+    for elt in labels:
+        rang = None
+        temp_arr = []
+        for label in  list:
+            label_pos =  elt.find(label)
+            if label_pos != -1:
+                if elt.count(',') > 0:
+                    new_temp = []
+                    temp = elt.split(label)
+                    for item in temp:
+                        range_pos = item.find(":")
+                        if range_pos != -1:
+                            rang = p2.findall(item)
+                            for i in xrange(int(rang[0]), int(rang[1])+1 ):
+                                new_temp.append(i)
+                    temp_arr += new_temp
+                else:
+                    temp = elt.split(label)
+                    for item in temp:
+                        range_pos = item.find(":")
+                        if range_pos != -1:
+                            rang = p2.findall(item)
+                            for i in xrange(int(rang[0]), int(rang[1])+1 ):
+                                temp_arr.append(i)
+                col_dict[elt] = (label, temp_arr)
+    return col_dict
 
 class GridPage(sheet.CSheet):
@@ -50 +91 @@
         """
         flag = event.CmdDown() or event.ControlDown()
-        cell = (event.GetRow(), event.GetCol())
+        row, col = event.GetRow(), event.GetCol()
+        cell = (row, col)
         if not flag:
             self.selected_cells = []
+            self.axis_value = []
+            self.axis_label = ""
         if cell in self.selected_cells:
             self.selected_cells.remove(cell)
         else:
             self.selected_cells.append(cell)
+        if row > 1:
+            if (cell) in self.selected_cells:
+                self.axis_value.append(self.GetCellValue(row, col))
+        self.axis_label = self.GetCellValue(row, col)
         event.Skip()
       
@@ -66 +114 @@
         col = event.GetCol()
         if not flag:
-            self.selected_cols  = []
+            self.selected_cols = []
+            self.axis_value = []
+            self.axis_label = ""
         if col not in self.selected_cols:
             self.selected_cols.append(col)
+        if not flag :
+            for row in range(2, self.GetNumberRows()+ 1):
+                self.axis_value.append(self.GetCellValue(row, col))
+            row = 1
+            self.axis_label = self.GetCellValue(row, col)
         event.Skip()
         
@@ -209 +264 @@
         return list_of_cells
     
+    def get_column_labels(self):
+        """
+        return dictionary of columns labels of the current page
+        """
+        pos = self.GetSelection()
+        grid = self.GetPage(pos)
+        labels = {}
+        row = 0
+        for col in range(grid.GetNumberCols()):
+            label = grid.GetCellValue(row, col)
+            if label.strip() != "" :
+                labels[label.strip()] = col
+        return labels
+        
     def create_axis_label(self, cell_list):
         """
@@ -229 +298 @@
             for element in temp:
                 temp_list.remove(element)
-            row_min, _  = temp_list[0]    
+            row_min, col  = temp_list[0]    
             row_max = row_min
+            col_name = grid.GetCellValue(0, col)
             label += str(col_name) + "[" + str(row_min) + ":"
-            print "temp_list", temp_list
             for index in xrange(len(temp_list)):
                 prev = index - 1
@@ -243 +312 @@
                         label  += "[" + str(row_min) + ":"
                 row_max = row
-               
                 if (index == len(temp_list)- 1):
-                    label +=  str(row_max) + "]"
-                    print "here"
-        print "label ", label             
-        return label 
+                    label +=  str(row_max) + "]"     
+        return label, col_name
     
     def on_close_page(self, event):
@@ -295 +361 @@
 
 class GridPanel(SPanel):
-    def __init__(self, parent,data=None, *args, **kwds):
+    def __init__(self, parent, data=None, *args, **kwds):
         SPanel.__init__(self, parent , *args, **kwds)
         self.vbox = wx.BoxSizer(wx.VERTICAL)
@@ -310 +376 @@
         self.x_axis_label = None
         self.y_axis_label = None
-        self.x_axis_value = None
-        self.y_axis_value = None
+        self.x_axis_title = None
+        self.y_axis_title = None
         self.x_axis_unit = None
         self.y_axis_unit = None
         self.plot_button = None
-        self.grid = None
+        self.notebook = None
         self.layout_grid()
         self.layout_plotting_area()
@@ -323 +389 @@
         """
         """
-        if self.grid is not None:
-            self.grid.set_data(data)
-        
-    def set_xaxis(self, label="", x=None) :
+        if self.notebook is not None:
+            self.notebook.set_data(data)
+        
+    def set_xaxis(self, label="", x=None):
+        """
+        """
         if x is None:
             x = []
         self.x = x
-        self.x_axis_value.SetValue("%s[:]" % str(label))
-        self.x_axis_label.SetValue(str(label))
-        
-    def set_yaxis(self, label="", y=None) :
+        self.x_axis_label.SetValue("%s[:]" % str(label))
+        self.x_axis_title.SetValue(str(label))
+        
+    def set_yaxis(self, label="", y=None):
+        """
+        """
         if y is None:
             y = []
         self.y = y
-        self.y_axis_value.SetValue("%s[:]" % str(label))
-        self.y_axis_label.SetValue(str(label))
-        
+        self.y_axis_label.SetValue("%s[:]" % str(label))
+        self.y_axis_title.SetValue(str(label))
+        
+    def get_plot_axis(self, col, list):
+        """
+       
+        """
+        axis = []
+        pos = self.notebook.GetSelection()
+        grid = self.notebook.GetPage(pos)
+        for row in list:
+            label = grid.GetCellValue(row - 1, col)
+            if label.strip() != "":
+                axis.append(float(label.strip()))
+        return axis
+    
     def on_plot(self, event):
         """
-        plotting
+        Evaluate the contains of textcrtl and plot result
         """ 
-        new_plot = Data1D(x=self.x, y=self.y)
+        pos = self.notebook.GetSelection()
+        grid = self.notebook.GetPage(pos)
+        column_names = {}
+        if grid is not None:
+            column_names = self.notebook.get_column_labels()
+        #evalue x
+        sentence = self.x_axis_label.GetValue()
+        if sentence.strip() == "":
+            msg = "select value for x axis"
+            raise ValueError, msg
+        dict = parse_string(sentence, column_names.keys())
+        for tok, (col_name, list) in dict.iteritems():
+            col = column_names[col_name]
+            xaxis = self.get_plot_axis(col, list)
+            sentence = sentence.replace(tok, 
+                                        "numpy.array(%s)" % str(xaxis))
+        for key, value in FUNC_DICT.iteritems():
+            sentence = sentence.replace(key.lower(), value)
+        x = eval(sentence)
+        #evaluate y
+        sentence = self.y_axis_label.GetValue()
+        if sentence.strip() == "":
+            msg = "select value for y axis"
+            raise ValueError, msg
+        dict = parse_string(sentence, column_names.keys())
+        for tok, (col_name, list) in dict.iteritems():
+            col = column_names[col_name]
+            yaxis = self.get_plot_axis(col, list)
+            sentence = sentence.replace(tok, 
+                                        "numpy.array(%s)" % str(yaxis))
+        for key, value in FUNC_DICT.iteritems():
+            sentence = sentence.replace(key, value)
+        y = eval(sentence)
+        #plotting
+        new_plot = Data1D(x=x, y=y)
         new_plot.id =  wx.NewId()
         new_plot.group_id = wx.NewId()
-        title = "%s vs %s" % (self.y_axis_label.GetValue(), self.x_axis_label.GetValue())
-        new_plot.xaxis(self.x_axis_label.GetValue(), self.x_axis_unit.GetValue())
-        new_plot.yaxis(self.y_axis_label.GetValue(), self.y_axis_unit.GetValue())
-        wx.PostEvent(self.parent.parent, 
+        title = "%s vs %s" % (self.y_axis_title.GetValue(), 
+                              self.x_axis_title.GetValue())
+        new_plot.xaxis(self.x_axis_title.GetValue(), self.x_axis_unit.GetValue())
+        new_plot.yaxis(self.y_axis_title.GetValue(), self.y_axis_unit.GetValue())
+        try:
+            title = self.notebook.GetPageText(pos)
+            wx.PostEvent(self.parent.parent, 
                              NewPlotEvent(plot=new_plot, 
-                        group_id=str(new_plot.group_id), title ="batch"))    
+                        group_id=str(new_plot.group_id), title =title))    
+        except:
+            pass
+        
     def layout_grid(self):
         """
         Draw the area related to the grid
         """
-        self.grid = Notebook(parent=self)
-        self.grid.set_data(self._data)
-        self.grid_sizer.Add(self.grid, 1, wx.EXPAND, 0)
+        self.notebook = Notebook(parent=self)
+        self.notebook.set_data(self._data)
+        self.grid_sizer.Add(self.notebook, 1, wx.EXPAND, 0)
        
     def layout_plotting_area(self):
@@ -365 +488 @@
         Draw area containing options to plot
         """
-        self.x_axis_label = wx.TextCtrl(self, -1)
-        self.y_axis_label = wx.TextCtrl(self, -1)
-        self.x_axis_value = wx.TextCtrl(self, -1, size=(200, -1))
-        self.y_axis_value = wx.TextCtrl(self, -1, size=(200, -1))
+        self.x_axis_title = wx.TextCtrl(self, -1)
+        self.y_axis_title = wx.TextCtrl(self, -1)
+        self.x_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
+        self.y_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
         self.x_axis_add = wx.Button(self, -1, "Add")
         self.x_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis, 
@@ -380 +503 @@
         wx.EVT_BUTTON(self, self.plot_button.GetId(), self.on_plot)
         self.plotting_sizer.AddMany([
-                    (wx.StaticText(self, -1, "x-axis label"), 1, wx.LEFT, 10),
-                    (self.x_axis_label, wx.TOP|wx.BOTTOM|wx.LEFT, 10),
-                    (wx.StaticText(self, -1, "x-axis value"), 1, wx.LEFT, 10),
-                    (self.x_axis_value, wx.TOP|wx.BOTTOM|wx.LEFT, 10),
-                    (self.x_axis_add, 1, wx.LEFT|wx.RIGHT, 0),
-                    (wx.StaticText(self, -1 , "unit"), 1, wx.LEFT|wx.RIGHT, 0),
-                    (self.x_axis_unit, 0, wx.LEFT, 0),
-                    (wx.StaticText(self, -1, "y-axis label"), 1, wx.LEFT, 10),
-                    (self.y_axis_label,  wx.TOP|wx.BOTTOM|wx.LEFT, 10),
-                    (wx.StaticText(self, -1, "y-axis value"), 1, wx.LEFT, 10),
-                    (self.y_axis_value, wx.TOP|wx.BOTTOM|wx.LEFT, 10),
-                    (self.y_axis_add, 1, wx.LEFT|wx.RIGHT, 0),
-                    (wx.StaticText(self, -1 , "unit"), 1, wx.LEFT|wx.RIGHT, 0),
-                    (self.y_axis_unit, 0, wx.LEFT, 0),
+                    (wx.StaticText(self, -1, "x-axis label"), 1,
+                      wx.TOP|wx.BOTTOM|wx.LEFT, 10),
+                    (self.x_axis_label, 1, wx.TOP|wx.BOTTOM, 10),
+                    (self.x_axis_add, 1, wx.TOP|wx.BOTTOM|wx.RIGHT, 10),
+                    (wx.StaticText(self, -1, "x-axis title"), 1, 
+                     wx.TOP|wx.BOTTOM|wx.LEFT, 10),
+                    (self.x_axis_title, 1, wx.TOP|wx.BOTTOM, 10),
+                    (wx.StaticText(self, -1 , "unit"), 1, 
+                     wx.TOP|wx.BOTTOM, 10),
+                    (self.x_axis_unit, 1, wx.TOP|wx.BOTTOM, 10),
+                    (wx.StaticText(self, -1, "y-axis label"), 1, 
+                     wx.BOTTOM|wx.LEFT, 10),
+                    (self.y_axis_label, wx.BOTTOM, 10),
+                    (self.y_axis_add, 1, wx.BOTTOM|wx.RIGHT, 10),
+                    (wx.StaticText(self, -1, "y-axis title"), 1, 
+                     wx.BOTTOM|wx.LEFT, 10),
+                    (self.y_axis_title,  wx.BOTTOM, 10),
+                    (wx.StaticText(self, -1 , "unit"), 1, wx.BOTTOM, 10),
+                    (self.y_axis_unit, 1, wx.BOTTOM, 10),
                       (-1, -1),
                       (-1, -1),
@@ -400 +528 @@
                       (-1, -1),
                       (-1, -1),
-                      (self.plot_button, 1, wx.LEFT, 0)])
+                      (self.plot_button, 1, wx.LEFT|wx.BOTTOM, 10)])
    
     def on_edit_axis(self, event):
@@ -406 +534 @@
         Get the selected column on  the visible grid and set values for axis
         """
-        cell_list = self.grid.on_edit_axis()
-        self.create_axis_label(cell_list)
-        
+        cell_list = self.notebook.on_edit_axis()
+        label, title = self.create_axis_label(cell_list)
+        tcrtl = event.GetEventObject()
+        if tcrtl == self.x_axis_add:
+            self.edit_axis_helper(self.x_axis_label, self.x_axis_title,
+                                   label, title)
+        elif tcrtl == self.y_axis_add:
+            self.edit_axis_helper(self.y_axis_label, self.y_axis_title,
+                                   label, title)
+            
     def create_axis_label(self, cell_list):
         """
@@ -416 +551 @@
         
         """
-        if self.grid is not None:
-            return self.grid.create_axis_label(cell_list)
+        if self.notebook is not None:
+            return self.notebook.create_axis_label(cell_list)
     
-    def edit_axis_helper(self, tcrtl_label, tcrtl_value):
-        """
-        """
+    def edit_axis_helper(self, tcrtl_label, tcrtl_title, label, title):
+        """
+        get controls to modify
+        """
+        tcrtl_label.SetValue(str(label))
+        tcrtl_title.SetValue(str(title))
+        
     def add_column(self):
         """
         """
-        if self.grid is not None:
-            self.grid.add_column()
+        if self.notebook is not None:
+            self.notebook.add_column()
         
     def on_remove_column(self):
         """
         """
-        if self.grid is not None:
-            self.grid.on_remove_column()
+        if self.notebook is not None:
+            self.notebook.on_remove_column()