source: sasview/sansguiframe/src/sans/guiframe/data_processor.py @ 285c3bb

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 285c3bb was c2e5898, checked in by Jae Cho <jhjcho@…>, 12 years ago

the data in the grid copied from datainfo can now be in data operation

  • Property mode set to 100644
File size: 66.0 KB
RevLine 
[24adb89]1"""
2Implement grid used to store data
3"""
4import wx
5import numpy
[904830e]6import math
[1ec979d]7import time
[904830e]8import re
[850525c]9import os
[24adb89]10import sys
[7ad194fa]11import copy
[24adb89]12from wx.lib.scrolledpanel import ScrolledPanel
13import  wx.grid as  Grid
14import wx.aui
15from wx.aui import AuiNotebook as nb
16import wx.lib.sheet as sheet
[1c86a37]17from sans.guiframe.panel_base import PanelBase
18from sans.guiframe.utils import format_number
[24adb89]19from sans.guiframe.events import NewPlotEvent
20from sans.guiframe.events import StatusEvent 
[fe98127]21from danse.common.plottools import plottables
[24adb89]22from sans.guiframe.dataFitting import Data1D
23
[904830e]24FUNC_DICT = {"sqrt": "math.sqrt",
25             "pow": "math.sqrt"}
[73197d0]26
[5425990]27class BatchCell:
28    """
29    Object describing a cell in  the grid.
30   
31    """
32    def __init__(self):
33        self.label = ""
34        self.value = None
35        self.col = -1
36        self.row = -1
[75790dc]37        self.object = []
[5425990]38       
[73197d0]39
[904830e]40def parse_string(sentence, list):
41    """
42    Return a dictionary of column label and index or row selected
43    :param sentence: String to parse
44    :param list: list of columns label
45    """
46    toks = []
47    p2 = re.compile(r'\d+')
48    p = re.compile(r'[\+\-\*\%\/]')
49    labels = p.split(sentence)
50    col_dict = {}
51    for elt in labels:
52        rang = None
53        temp_arr = []
54        for label in  list:
55            label_pos =  elt.find(label)
[08dc9e87]56            separator_pos  = label_pos + len(label)
57            if label_pos != -1 and len(elt) >= separator_pos  and\
58                elt[separator_pos]== "[":
59                # the label contain , meaning the range selected is not
60                # continuous
[904830e]61                if elt.count(',') > 0:
62                    new_temp = []
63                    temp = elt.split(label)
64                    for item in temp:
65                        range_pos = item.find(":")
66                        if range_pos != -1:
67                            rang = p2.findall(item)
68                            for i in xrange(int(rang[0]), int(rang[1])+1 ):
69                                new_temp.append(i)
70                    temp_arr += new_temp
71                else:
[08dc9e87]72                    # continuous range
[904830e]73                    temp = elt.split(label)
74                    for item in temp:
[08dc9e87]75                        if item.strip() != "":
76                            range_pos = item.find(":")
77                            if range_pos != -1:
78                                rang = p2.findall(item)
79                                for i in xrange(int(rang[0]), int(rang[1])+1 ):
80                                    temp_arr.append(i)
[904830e]81                col_dict[elt] = (label, temp_arr)
82    return col_dict
[24adb89]83
[f4b37d1]84         
85         
86class SPanel(ScrolledPanel):
87    def __init__(self, parent, *args, **kwds):
88        ScrolledPanel.__init__(self, parent , *args, **kwds)
89        self.SetupScrolling() 
90       
[24adb89]91class GridPage(sheet.CSheet):
[9c8f3ad]92    """
93    """
[24adb89]94    def __init__(self, parent, panel=None):
95        """
96        """
97        sheet.CSheet.__init__(self, parent)
[f4b37d1]98       
99        self.AdjustScrollbars()
100        #self.SetLabelBackgroundColour('#DBD4D4')
[fd51a7c]101        self.uid = wx.NewId()
[f4b37d1]102        self.parent = parent
[24adb89]103        self.panel = panel
104        self.col_names = []
[8523a1f2]105        self.data_inputs = {}
106        self.data_outputs = {}
[d03a356]107        self.data = None
[71fa9028]108        self.details = ""
109        self.file_name = None
[9c8f3ad]110        self._cols = 50
[cbba84f]111        self._rows = 3001
[63dc6e5]112        self.last_selected_row = -1
113        self.last_selected_col = -1
[1c86a37]114        self.col_width = 30
115        self.row_height = 20
[656d65d]116        self.max_row_touse = 0
[49ad00b]117        self.axis_value = []
118        self.axis_label = ""
119        self.selected_cells = []
120        self.selected_cols = []
[75790dc]121        self.selected_rows = []
122        self.plottable_cells = []
[63dc6e5]123        self.plottable_flag = False
[1c86a37]124        self.SetColMinimalAcceptableWidth(self.col_width)
125        self.SetRowMinimalAcceptableHeight(self.row_height)
[23a1747]126        self.SetNumberRows(self._rows)
127        self.SetNumberCols(self._cols)
[86a9e6c]128        color = self.parent.GetBackgroundColour()
129        for col in range(self._cols):
130            self.SetCellBackgroundColour(0, col, color)
[f4b37d1]131        self.AutoSize()
[23477c6]132        self.list_plot_panels = {}
[1c86a37]133        self.default_col_width = 75
[0899c82]134        self.EnableEditing(True)
[1c86a37]135        if self.GetNumberCols() > 0:
136            self.default_col_width =  self.GetColSize(0)
[49ad00b]137        self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_CLICK, self.on_left_click)
[24adb89]138        self.Bind(wx.grid.EVT_GRID_LABEL_RIGHT_CLICK, self.on_right_click)
[49ad00b]139        self.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.on_selected_cell)
[656d65d]140        self.Bind(wx.grid.EVT_GRID_CMD_CELL_CHANGE, self.on_edit_cell)
[0899c82]141        self.Bind(wx.grid.EVT_GRID_CELL_RIGHT_CLICK, self.onContextMenu)
[1c86a37]142       
[656d65d]143    def on_edit_cell(self, event):
144        """
145        """
146        row, col = event.GetRow(), event.GetCol()
147        if row > self.max_row_touse:
148            self.max_row_touse = row
[14e4804]149        if self.data == None:
150            self.data = {}
[656d65d]151        event.Skip()
[49ad00b]152       
153    def on_selected_cell(self, event):
154        """
155        Handler catching cell selection
156        """
157        flag = event.CmdDown() or event.ControlDown()
[4e0dfe4]158        flag_shift =  event.ShiftDown()
[904830e]159        row, col = event.GetRow(), event.GetCol()
160        cell = (row, col)
[63dc6e5]161        event.Skip()
[4e0dfe4]162        if not flag and not flag_shift:
[08dc9e87]163            self.selected_cols = []
164            self.selected_rows = []
[49ad00b]165            self.selected_cells = []
[904830e]166            self.axis_label = ""
[08dc9e87]167            self.axis_value = []
168            self.plottable_list = []
169            self.plottable_cells = []
170            self.plottable_flag = False
171        self.last_selected_col = col
172        self.last_selected_row = row
[63dc6e5]173        if col >= 0:
[3553ad2]174            if flag:
175                label_row = row
176            else:
177                label_row = 0
[08dc9e87]178            self.axis_label = self.GetCellValue(label_row, col)
179            self.selected_cols.append(col)
[4e0dfe4]180        if flag_shift:
181            if not self.selected_rows:
182                min_r = 1
183            else:
184                min_r = min(self.selected_rows)
185            for row_s in range(min_r, row+1):
186                cel = (row_s, col)
187                if cel not in self.selected_cells:
188                    if row > 0:
189                        self.selected_cells.append(cel)
190                        self.selected_rows.append(row) 
191            for row_s in self.selected_rows: 
192                cel = (row_s, col)
193                if row_s > row:
194                    try:
195                        self.selected_cells.remove(cel)
196                    except:
197                        pass
198                    try:
199                        self.selected_rows.remove(row_s)
200                    except:
201                        pass
202        elif flag:
203            if cell not in self.selected_cells:
204                if row > 0 :
205                    self.selected_cells.append(cell)
206                    self.selected_rows.append(row)
207            else:
208                try:
209                    self.selected_cells.remove(cell)
210                except:
211                    pass
212                try:
213                    self.selected_rows.remove(row)
214                except:
215                    pass
[49ad00b]216        else:
[4e0dfe4]217            self.selected_cells.append(cell)
218            self.selected_rows.append(row)
[dadf255]219        self.axis_value = []
220        for cell_row, cell_col in self.selected_cells:
221            if cell_row > 0 and cell_row < self.max_row_touse:
222                self.axis_value.append(self.GetCellValue(cell_row, cell_col))
[86a9e6c]223
[49ad00b]224    def on_left_click(self, event):
225        """
226        Catch the left click on label mouse event
227        """
[647df0d1]228        event.Skip()
[49ad00b]229        flag = event.CmdDown() or event.ControlDown()
[4e0dfe4]230       
[49ad00b]231        col = event.GetCol()
[63dc6e5]232        row = event.GetRow()
[86a9e6c]233
[4e0dfe4]234        if not (flag):
[dadf255]235            self.selected_cols = []
[75790dc]236            self.selected_rows = []
[dadf255]237            self.selected_cells = []
238            self.axis_label = ""
[08dc9e87]239            self.axis_value = []
[63dc6e5]240            self.plottable_list = []
[75790dc]241            self.plottable_cells = []
[63dc6e5]242            self.plottable_flag = False
[75790dc]243       
[63dc6e5]244        self.last_selected_col = col
245        self.last_selected_row = row
[75790dc]246        if row != -1 and row not in self.selected_rows:
247             self.selected_rows.append(row)
248             
[647df0d1]249        if col != -1:
250            for row in range(1, self.GetNumberRows()+ 1):
251                cell = (row, col)
252                if row > 0 and row < self.max_row_touse:
253                    if cell not in self.selected_cells:
254                        self.selected_cells.append(cell)
[08dc9e87]255                    else:
256                        if flag:
257                             self.selected_cells.remove(cell)
[647df0d1]258            self.selected_cols.append(col)
259            self.axis_value = []
260            for cell_row, cell_col in self.selected_cells:
[4e0dfe4]261                val = self.GetCellValue(cell_row, cell_col)
262                if not val:
263                    self.axis_value.append(self.GetCellValue(cell_row, cell_col))
[647df0d1]264            self.axis_label = self.GetCellValue(0, col)
[25b7bf9]265            if not self.axis_label:
266                self.axis_label = " "
[75790dc]267       
[24adb89]268    def on_right_click(self, event):
[9c8f3ad]269        """
270        Catch the right click mouse
271        """
[656d65d]272       
[24adb89]273        col = event.GetCol()
[c151afc]274        row = event.GetRow()
[9ccb7e1]275        # Ignore the index column
[c151afc]276        if col < 0 or row != -1:
[9ccb7e1]277            return
[656d65d]278        self.selected_cols = []
279        self.selected_cols.append(col)
[24adb89]280        # Slicer plot popup menu
281        slicerpop = wx.Menu()
[656d65d]282        col_label_menu  = wx.Menu()
[1c86a37]283        c_name = self.GetCellValue(0, col) 
284        label = "Insert column before %s " % str(c_name)
[656d65d]285        slicerpop.AppendSubMenu(col_label_menu , 
[1c86a37]286                                 '&%s' % str(label), str(label))
[71fa9028]287        col_name = [self.GetCellValue(0, c) 
288                        for c in range(self.GetNumberCols())]
289        row = 0
290        label = self.GetCellValue(row, col)
291        self.insert_col_menu(col_label_menu, label, self)
[b18cf3d]292       
293       
294        col_after_menu  = wx.Menu()
295        label = "Insert column after %s " % str(c_name)
296        slicerpop.AppendSubMenu(col_after_menu , 
297                                 '&%s' % str(label), str(label))
298        col_name = [self.GetCellValue(0, c) 
299                        for c in range(self.GetNumberCols())]
300        self.insert_after_col_menu(col_after_menu, label, self)
301       
302       
[656d65d]303        id = wx.NewId()   
304        hint = 'Remove selected column %s'
[71fa9028]305        slicerpop.Append(id, '&Remove Column', hint)
[656d65d]306        wx.EVT_MENU(self, id, self.on_remove_column)
[24adb89]307       
[656d65d]308        pos = wx.GetMousePosition()
[24adb89]309        pos = self.ScreenToClient(pos)
310        self.PopupMenu(slicerpop, pos)
[647df0d1]311        event.Skip()
[24adb89]312       
[71fa9028]313    def insert_col_menu(self, menu, label, window):
314        """
315        """
[5531a46]316        if self.data is None:
317            return
[71fa9028]318        id = wx.NewId()
319        title = "Empty"
320        hint = 'Insert empty column before %s' % str(label)
321        menu.Append(id, title, hint)
322        wx.EVT_MENU(window, id, self.on_insert_column)
[fb0de166]323        row = 0
324        col_name = [self.GetCellValue(row, col) 
[71fa9028]325                        for col in range(self.GetNumberCols())]
326        for c_name in self.data.keys():
[9696a10b]327            if c_name not in col_name and self.data[c_name]:
[71fa9028]328                id = wx.NewId()
329                hint = "Insert %s column before the " % str(c_name)
330                hint += " %s column" % str(label)
331                menu.Append(id, '&%s' % str(c_name), hint)
332                wx.EVT_MENU(window, id, self.on_insert_column)
333           
[b18cf3d]334    def insert_after_col_menu(self, menu, label, window):
335        """
336        """
[5531a46]337        if self.data is None:
338            return
[b18cf3d]339        id = wx.NewId()
340        title = "Empty"
341        hint = 'Insert empty column after %s' % str(label)
342        menu.Append(id, title, hint)
343        wx.EVT_MENU(window, id, self.on_insert_after_column)
344        row = 0
345        col_name = [self.GetCellValue(row, col) 
346                        for col in range(self.GetNumberCols())]
347        for c_name in self.data.keys():
[9696a10b]348            if c_name not in col_name and self.data[c_name]:
[b18cf3d]349                id = wx.NewId()
350                hint = "Insert %s column after the " % str(c_name)
351                hint += " %s column" % str(label)
352                menu.Append(id, '&%s' % str(c_name), hint)
353                wx.EVT_MENU(window, id, self.on_insert_after_column)
354                               
[71fa9028]355    def on_remove_column(self, event=None):
[656d65d]356        """
357        """
358        if self.selected_cols is not None or len(self.selected_cols) > 0:
359            col = self.selected_cols[0]
[71fa9028]360            self.remove_column(col=col, numCols=1)
361           
362    def remove_column(self, col, numCols=1):
363        """
364        Remove column to the current grid
365        """
366        # add data to the grid   
367        row = 0
368        col_name = self.GetCellValue(row, col)
369        self.data[col_name] = []
370        for row in range(1, self.GetNumberRows() + 1):
371            if row < self.max_row_touse:
372                value = self.GetCellValue(row, col)
373                self.data[col_name].append(value)
374                for k , value_list in self.data.iteritems():
375                    if k != col_name:
376                        length = len(value_list)
377                        if length < self.max_row_touse:
378                            diff = self.max_row_touse - length
379                            for i in range(diff):
380                                self.data[k].append("")
381        self.DeleteCols(pos=col, numCols=numCols, updateLabels=True)
[656d65d]382           
383    def on_insert_column(self, event):
384        """
385        """
386        if self.selected_cols is not None or len(self.selected_cols) > 0:
387            col = self.selected_cols[0]
388            # add data to the grid   
389            row = 0
390            id = event.GetId()
391            col_name = event.GetEventObject().GetLabelText(id)
[71fa9028]392            self.insert_column(col=col, col_name=col_name)
393            if  not issubclass(event.GetEventObject().__class__ , wx.Menu):
394                col += 1
395                self.selected_cols[0] += 1
[b18cf3d]396               
397    def on_insert_after_column(self, event):
398        """
399        Insert the given column after the highlighted column
400        """
401        if self.selected_cols is not None or len(self.selected_cols) > 0:
402            col = self.selected_cols[0] + 1
403            # add data to the grid   
404            row = 0
405            id = event.GetId()
406            col_name = event.GetEventObject().GetLabelText(id)
407            self.insert_column(col=col, col_name=col_name)
408            if  not issubclass(event.GetEventObject().__class__ , wx.Menu):
409                self.selected_cols[0] += 1
[71fa9028]410           
411    def insert_column(self, col, col_name):
412        """
413        """ 
414         
415        row = 0
416        self.InsertCols(pos=col, numCols=1, updateLabels=True)
417        if col_name.strip() != "Empty":
418            self.SetCellValue(row, col, str(col_name.strip()))
419        if col_name in self.data.keys():
420            value_list = self.data[col_name]
421            cell_row =  1
422            for value in value_list:
[6dad639]423                label = value#format_number(value, high=True)
[b18cf3d]424                self.SetCellValue(cell_row, col, str(label))
[71fa9028]425                cell_row += 1
[b18cf3d]426        self.AutoSizeColumn(col, True)
427        width = self.GetColSize(col)
428        if width < self.default_col_width:
429           self.SetColSize(col, self.default_col_width)
[86a9e6c]430        color = self.parent.GetBackgroundColour()
431        self.SetCellBackgroundColour(0, col, color)
[b18cf3d]432        self.ForceRefresh()
[656d65d]433       
[24adb89]434    def on_set_x_axis(self, event):
[9c8f3ad]435        """
436        """
[24adb89]437        self.panel.set_xaxis(x=self.axis_value, label=self.axis_label)
438   
439    def on_set_y_axis(self, event):
[9c8f3ad]440        """
441        """
[24adb89]442        self.panel.set_yaxis(y=self.axis_value, label=self.axis_label)     
443           
[71fa9028]444    def set_data(self, data_inputs, data_outputs, details, file_name):
[24adb89]445        """
[9c8f3ad]446        Add data to the grid
[dadf255]447        :param data_inputs: data to use from the context menu of the grid
448        :param data_ouputs: default columns deplayed
[24adb89]449        """
[71fa9028]450        self.file_name = file_name
451        self.details = details
452       
[8523a1f2]453        if data_outputs is None:
454            data_outputs = {}
455        self.data_outputs = data_outputs
[71fa9028]456        if data_inputs is None:
[8523a1f2]457            data_inputs = {}
458        self.data_inputs = data_inputs
[656d65d]459        self.data = {}
460        for item in (self.data_outputs, self.data_inputs):
461            self.data.update(item)
[5531a46]462           
[0899c82]463        #if len(self.data) + len(self.data_inputs) +len(self.data_outputs) == 0:
464        #    self.EnableEditing(False)
465        #else:
466        #    self.EnableEditing(True)
[8523a1f2]467        if  len(self.data_outputs) > 0:
[9c8f3ad]468            self._cols = self.GetNumberCols()
469            self._rows = self.GetNumberRows()
[8523a1f2]470            self.col_names = self.data_outputs.keys()
[24adb89]471            self.col_names.sort() 
[9c8f3ad]472            nbr_user_cols = len(self.col_names)
473            #Add more columns to the grid if necessary
474            if nbr_user_cols > self._cols:
[ed2d86e]475                new_col_nbr = nbr_user_cols -  self._cols + 1
[9c8f3ad]476                self.AppendCols(new_col_nbr, True)
477            #Add more rows to the grid if necessary 
[ed2d86e]478            nbr_user_row = len(self.data_outputs.values()[0])
[9c8f3ad]479            if nbr_user_row > self._rows + 1:
[ed2d86e]480                new_row_nbr =  nbr_user_row - self._rows + 1
[9c8f3ad]481                self.AppendRows(new_row_nbr, True)
482            # add data to the grid   
[ed2d86e]483            wx.CallAfter(self.set_grid_values)
[1c86a37]484        self.ForceRefresh()
[ed2d86e]485   
486    def set_grid_values(self):
487        """
488        Set the values in grids
489        """
490        # add data to the grid   
491        row = 0
492        col = 0
493        cell_col = 0
494        for col_name in  self.col_names:
495            # use the first row of the grid to add user defined labels
496            self.SetCellValue(row, col, str(col_name))
497            col += 1
498            cell_row =  1
499            value_list = self.data_outputs[col_name]
500           
501            for value in value_list:
502                label = value
503                if issubclass(value.__class__, BatchCell):
504                    label = value.label
505                try:
506                    float(label)
507                    label = str(label)#format_number(label, high=True)
508                except:
509                    label = str(label)
510                self.SetCellValue(cell_row, cell_col, label)
511                self.AutoSizeColumn(cell_col, True)
512                width = self.GetColSize(cell_col)
513                if width < self.default_col_width:
514                   self.SetColSize(cell_col, self.default_col_width)
515               
516                cell_row += 1
517            cell_col += 1
518            if cell_row > self.max_row_touse:
519                self.max_row_touse = cell_row
520                         
[71fa9028]521    def get_grid_view(self):
522        """
523        Return value contained in the grid
524        """
525        grid_view = {}
526        for col in xrange(self.GetNumberCols()):
527            label = self.GetCellValue(row=0, col=col) 
528            label = label.strip()
529            if label != "":
530                grid_view[label] = []
[c27a111]531                for row in range(1, self.max_row_touse):
[71fa9028]532                    value = self.GetCellValue(row=row, col=col)
533                    if value != "":
534                        grid_view[label].append(value) 
535                    else:
536                        grid_view[label].append(None) 
537        return grid_view
538   
[86a9e6c]539    def get_nofrows(self):
540        """
541        Return number of total rows
542        """
543        return self._rows
544   
[0899c82]545    def onContextMenu(self, event):
546        """
547        Default context menu
548        """
549        id = wx.NewId()
550        c_menu = wx.Menu()
[14e4804]551        copy_menu = c_menu.Append(id, '&Copy', 'Copy the selected cells')
[0899c82]552        wx.EVT_MENU(self, id, self.on_copy)
553       
554        id = wx.NewId()
[14e4804]555        paste_menu = c_menu.Append(id, '&Paste', 'Paste the selected cells')
[0899c82]556        wx.EVT_MENU(self, id, self.on_paste)
[14e4804]557       
558        id = wx.NewId()
559        clear_menu = c_menu.Append(id, '&Clear', 'Clear the selected cells')
560        wx.EVT_MENU(self, id, self.on_clear)
561       
[0899c82]562        # enable from flag
563        has_selection = False
564        selected_cel = self.selected_cells
565        if len(selected_cel) > 0:
566            _row, _col = selected_cel[0]
567            has_selection = self.IsInSelection(_row, _col)
[14e4804]568        if len(self.selected_cols) > 0:
569            has_selection = True
570        if len(self.selected_rows) > 0:
571            has_selection = True
[0899c82]572        copy_menu.Enable(has_selection)
[14e4804]573        clear_menu.Enable(has_selection)
[0899c82]574        try:
575            # mouse event pos
576            pos_evt = event.GetPosition()
577            self.PopupMenu(c_menu, pos_evt)
578        except:
579            return
580   
581    def on_copy(self, event):
582        """
583        On copy event from the contextmenu
584        """
585        self.Copy()
586
587    def on_paste(self, event):
588        """
589        On paste event from the contextmenu
590        """
[14e4804]591        if self.data == None:
592            self.data = {}
593        if self.file_name == None:
594            self.file_name = 'copied_data'
[0899c82]595        self.Paste()
596   
[14e4804]597    def on_clear(self, event):
598        """
599        Clear the cells selected
600        """
601        self.Clear()
602   
[24adb89]603class Notebook(nb, PanelBase):
604    """
605    ## Internal name for the AUI manager
606    window_name = "Fit panel"
607    ## Title to appear on top of the window
608    """
609    window_caption = "Notebook "
610   
611    def __init__(self, parent, manager=None, data=None, *args, **kwargs):
612        """
613        """
614        nb.__init__(self, parent, -1,
[f4b37d1]615                    style=wx.aui.AUI_NB_WINDOWLIST_BUTTON| 
616                    wx.aui.AUI_BUTTON_DOWN|
[9c8f3ad]617                    wx.aui.AUI_NB_DEFAULT_STYLE|
618                    wx.CLIP_CHILDREN)
[24adb89]619        PanelBase.__init__(self, parent)
[86a9e6c]620        self.gpage_num = 1
[9c8f3ad]621        self.enable_close_button()
[24adb89]622        self.parent = parent
623        self.manager = manager
624        self.data = data
[d03a356]625        #add empty page
626        self.add_empty_page()
[86a9e6c]627        self.pageClosedEvent = wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE
628        self.Bind(self.pageClosedEvent, self.on_close_page)   
[7ad194fa]629   
[d03a356]630    def add_empty_page(self):
631        """
632        """
633        grid = GridPage(self, panel=self.parent)
634        self.AddPage(grid, "", True)
635        pos = self.GetPageIndex(grid)
[0899c82]636        title = "Table" + str(self.gpage_num)
[d03a356]637        self.SetPageText(pos, title)
638        self.SetSelection(pos)
[86a9e6c]639        self.enable_close_button()
640        self.gpage_num += 1
[f4b37d1]641        return grid , pos
[d03a356]642       
[9c8f3ad]643    def enable_close_button(self):
644        """
645        display the close button on tab for more than 1 tabs else remove the
646        close button
647        """
648        if self.GetPageCount() <= 1:
649            style = self.GetWindowStyleFlag() 
650            flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
651            if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB == flag:
652                style = style & ~wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
653                self.SetWindowStyle(style)
654        else:
655            style = self.GetWindowStyleFlag()
656            flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
657            if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB != flag:
658                style |= wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
659                self.SetWindowStyle(style)
[9680906d]660             
661    def on_edit_axis(self):
662        """
[08dc9e87]663        Return the select cell of a given selected column. Check that all cells
664        are from the same column
[9680906d]665        """
666        pos = self.GetSelection()
667        grid = self.GetPage(pos)
[4e0dfe4]668        #grid.selected_cols = [grid.GetSelectedRows()]#
[08dc9e87]669        if len(grid.selected_cols) >= 1:
[49ad00b]670            col = grid.selected_cols[0]
[08dc9e87]671            for c in grid.selected_cols:
672                if c != col:
[7ad194fa]673                    msg = "Edit axis doesn't understand this selection.\n"
[08dc9e87]674                    msg += "Please select only one column"
[7ad194fa]675                    raise ValueError, msg
[08dc9e87]676            for (cell_row, cell_col) in grid.selected_cells:
677                if cell_col != col:
678                    msg = "Cannot use cells from different columns for "
679                    msg += "this operation.\n"
680                    msg += "Please select elements of the same col.\n"
681                    raise ValueError, msg
[86a9e6c]682               
683            # Finally check the highlighted cell if any cells missing
684            self.get_highlighted_row()
[08dc9e87]685        else:
686            msg = "No item selected.\n"
687            msg += "Please select only one column or one cell"
688            raise ValueError, msg
[dadf255]689        return grid.selected_cells
690       
[86a9e6c]691    def get_highlighted_row(self):
692        """
693        Add highlight rows
694        """
695        pos = self.GetSelection()
696        grid = self.GetPage(pos)
697        col = grid.selected_cols[0]
698        # Finally check the highlighted cell if any cells missing
699        for row in range(grid.get_nofrows()):
700            if grid.IsInSelection(row, col):
701                cel = (row, col)
702                if row < 1:
703                    continue
704                if not grid.GetCellValue(row, col):
705                    continue
706                if cel not in grid.selected_cells:
707                    grid.selected_cells.append(cel)
708                       
[904830e]709    def get_column_labels(self):
710        """
711        return dictionary of columns labels of the current page
712        """
713        pos = self.GetSelection()
714        grid = self.GetPage(pos)
715        labels = {}
716        row = 0
717        for col in range(grid.GetNumberCols()):
[25b7bf9]718            label = grid.GetColLabelValue(int(col))
719            #label = grid.GetCellValue(row, col)
[904830e]720            if label.strip() != "" :
721                labels[label.strip()] = col
722        return labels
723       
[7ad194fa]724    def create_axis_label(self, cell_list):
725        """
726        Receive a list of cells and  create a string presenting the selected
727        cells.
728        :param cell_list: list of tuple
[9680906d]729       
[7ad194fa]730        """
731        pos = self.GetSelection()
732        grid = self.GetPage(pos)
733        label = ""
734        col_name = ""
[dadf255]735        def create_label(col_name,  row_min=None, row_max=None):
736            """
737            """
[4e0dfe4]738            result = " "
[dadf255]739            if row_min is not  None or row_max is not None:
740                if row_min is None:
741                    result = str(row_max) + "]"
742                elif row_max is None:
743                     result = str(col_name) + "[" + str(row_min) + ":"
744                else:
745                    result = str(col_name) +  "[" + str(row_min) + ":"
[08dc9e87]746                    result += str(row_max) + "]"
747            return str(result)
[dadf255]748           
[7ad194fa]749        if len(cell_list) > 0:
[dadf255]750            if len(cell_list) == 1:
[86a9e6c]751                row_min, col  = cell_list[0]   
752                col_name =  grid.GetColLabelValue(int(col))
753                 
754                col_title = grid.GetCellValue(0, col)
755                label = create_label(col_name, row_min+1 , row_min+1)
756                return  label,  col_title
[dadf255]757            else:
758                temp_list = copy.deepcopy(cell_list)
759                temp_list.sort()
760                length = len(temp_list)
761                row_min, col  = temp_list[0]   
762                row_max, _  = temp_list[length-1]
[86a9e6c]763                col_name = grid.GetColLabelValue(int(col))
[25b7bf9]764                col_title = grid.GetCellValue(0, col)
[86a9e6c]765
[dadf255]766                index = 0
[86a9e6c]767                for row in xrange(row_min, row_max + 1):
[dadf255]768                    if index > 0 and index < len(temp_list):
769                        new_row, _ = temp_list[index]
770                        if row != new_row:
771                            temp_list.insert(index, (None, None))
[08dc9e87]772                            if index -1 >= 0:
[dadf255]773                                new_row, _ = temp_list[index-1]
[86a9e6c]774                                if not new_row == None and new_row != ' ' :
775                                    label += create_label(col_name, None, 
776                                                          int(new_row) +1)
[4e0dfe4]777                                else:
778                                    label += "]"
[dadf255]779                                label += ","
780                            if index + 1 < len(temp_list):
781                                new_row, _ = temp_list[index + 1]
[4e0dfe4]782                                if not new_row==None:
[86a9e6c]783                                    label += create_label(col_name, 
784                                                          int(new_row)+1, None)
[4e0dfe4]785                    if row_min != None and row_max != None:
786                        if index == 0:
[86a9e6c]787                            label += create_label(col_name, 
788                                                  int(row_min)+1, None)
[4e0dfe4]789                        elif index == len(temp_list)-1:
[86a9e6c]790                            label += create_label(col_name, None, 
791                                                  int(row_max)+1)
[dadf255]792                    index += 1
[4e0dfe4]793                # clean up the list
794                label_out = ''
795                for item in label.split(','):
[86a9e6c]796                    if item.split(":")[1] == "]":
[4e0dfe4]797                        continue
798                    else:
799                        label_out += item + ","
800
801                return label_out, col_title
[7ad194fa]802   
[9c8f3ad]803    def on_close_page(self, event):
804        """
805        close the page
806        """
807        if self.GetPageCount() == 1:
808            event.Veto()
[86a9e6c]809        wx.CallAfter(self.enable_close_button)
[24adb89]810       
[71fa9028]811    def set_data(self, data_inputs, data_outputs, details="", file_name=None):
[8523a1f2]812        if data_outputs is None or data_outputs == {}:
[24adb89]813            return
[98fdccd]814        inputs, outputs = self.get_odered_results(data_inputs, data_outputs)
[f4b37d1]815        for pos in range(self.GetPageCount()):
816            grid = self.GetPage(pos)
817            if grid.data is None:
818                #Found empty page
[98fdccd]819                grid.set_data(data_inputs=inputs, 
820                              data_outputs=outputs,
[71fa9028]821                              details=details,
822                              file_name=file_name) 
[f4b37d1]823                self.SetSelection(pos) 
824                return
825               
826        grid, pos = self.add_empty_page()
[98fdccd]827        grid.set_data(data_inputs=inputs, 
828                      data_outputs=outputs,
[a84ca2a]829                      file_name=file_name,
830                      details=details)
[98fdccd]831       
832    def get_odered_results(self, inputs, outputs=None):
833        """
834        Get ordered the results
835        """
836        # Let's re-order the data from the keys in 'Data' name.
837        if outputs == None:
838            return
[e25d908]839        try:
840            # For outputs from batch
841            to_be_sort = [str(item.label) for item in outputs['Data']]
842        except:
843            # When inputs are from an external file
844            return inputs, outputs
[98fdccd]845        inds = numpy.lexsort((to_be_sort, to_be_sort))
846        for key in outputs.keys():
847            key_list = outputs[key]
848            temp_key = [item for item in key_list]
849            for ind in inds:
850                temp_key[ind] = key_list[inds[ind]]
851            outputs[key] = temp_key
852        for key in inputs.keys():
853            key_list = inputs[key]
854            if len(key_list) > 0:
855                temp_key = [item for item in key_list]
856                for ind in inds:
857                    temp_key[ind] = key_list[inds[ind]]
858                inputs[key] = temp_key
859        return inputs, outputs
[f4b37d1]860   
[9c8f3ad]861    def add_column(self):
862        """
863        Append a new column to the grid
864        """
865        pos = self.GetSelection()
866        grid = self.GetPage(pos)
867        grid.AppendCols(1, True)
868       
[dbb3914]869    def on_remove_column(self):
870        """
871        Remove the selected column from the grid
872        """
873        pos = self.GetSelection()
874        grid = self.GetPage(pos)
875        grid.on_remove_column(event=None)
876       
[24adb89]877class GridPanel(SPanel):
[8523a1f2]878    def __init__(self, parent, data_inputs=None,
879                 data_outputs=None, *args, **kwds):
[24adb89]880        SPanel.__init__(self, parent , *args, **kwds)
[f4b37d1]881       
[24adb89]882        self.vbox = wx.BoxSizer(wx.VERTICAL)
883       
[9680906d]884        self.plotting_sizer = wx.FlexGridSizer(3, 7, 10, 5)
[75790dc]885        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
[24adb89]886        self.grid_sizer = wx.BoxSizer(wx.HORIZONTAL)
887        self.vbox.AddMany([(self.grid_sizer, 1, wx.EXPAND, 0),
888                           (wx.StaticLine(self, -1), 0, wx.EXPAND, 0),
[75790dc]889                           (self.plotting_sizer),
[3e1177d]890                           (self.button_sizer, 0, wx.BOTTOM, 10)])
[24adb89]891        self.parent = parent
[8523a1f2]892        self._data_inputs = data_inputs
893        self._data_outputs = data_outputs
[24adb89]894        self.x = []
895        self.= []
[9130227]896        self.dy  = []
[24adb89]897        self.x_axis_label = None
898        self.y_axis_label = None
[9130227]899        self.dy_axis_label = None
[904830e]900        self.x_axis_title = None
901        self.y_axis_title = None
[24adb89]902        self.x_axis_unit = None
903        self.y_axis_unit = None
[75790dc]904        self.view_button = None
[24adb89]905        self.plot_button = None
[904830e]906        self.notebook = None
[86a9e6c]907        self.plot_num = 1
[23477c6]908       
[24adb89]909        self.layout_grid()
910        self.layout_plotting_area()
911        self.SetSizer(self.vbox)
[98fdccd]912
[904830e]913    def set_xaxis(self, label="", x=None):
914        """
915        """
[24adb89]916        if x is None:
917            x = []
918        self.x = x
[904830e]919        self.x_axis_label.SetValue("%s[:]" % str(label))
920        self.x_axis_title.SetValue(str(label))
[24adb89]921       
[904830e]922    def set_yaxis(self, label="", y=None):
923        """
924        """
[24adb89]925        if y is None:
926            y = []
927        self.y = y
[904830e]928        self.y_axis_label.SetValue("%s[:]" % str(label))
929        self.y_axis_title.SetValue(str(label))
[24adb89]930       
[9130227]931    def set_dyaxis(self, label="", dy=None):
932        """
933        """
934        if dy is None:
935            dy = []
936        self.dy = dy
937        self.dy_axis_label.SetValue("%s[:]" % str(label))
938               
[904830e]939    def get_plot_axis(self, col, list):
940        """
941       
942        """
943        axis = []
944        pos = self.notebook.GetSelection()
945        grid = self.notebook.GetPage(pos)
946        for row in list:
[c911f34]947            label = grid.GetCellValue(0, col)
948            value = grid.GetCellValue(row - 1, col).strip()
949            if value != "":
950                if label.lower().strip() == "data":
951                    axis.append(float(row - 1))
952                else:
[dadf255]953                    try:
954                        axis.append(float(value))
955                    except:
956                        msg = "Invalid data in row %s column %s" % (str(row),
957                                                                    str(col))
958                        wx.PostEvent(self.parent.parent, 
[0899c82]959                             StatusEvent(status=msg, info="error"))
960                        return None
[c911f34]961            else:
962                axis.append(None) 
[904830e]963        return axis
964   
[75790dc]965    def on_view(self, event):
966        """
967        Get object represented buy the given cell and plot them.
968        """
969        pos = self.notebook.GetSelection()
970        grid = self.notebook.GetPage(pos)
971        title = self.notebook.GetPageText(pos)
[86a9e6c]972        self.notebook.get_highlighted_row()
[9400de6]973        if len(grid.selected_cells) == 0:
974            msg = "Highlight a Data or Chi2 column first..."
975            wx.PostEvent(self.parent.parent, 
976                             StatusEvent(status=msg, info="error")) 
977            return
[23155ba]978        elif len(grid.selected_cells) > 20:
979            msg = "Too many data (> 20) to plot..."
980            msg += "\n Please select no more than 20 data."
981            dial = wx.MessageDialog(self, msg, 'Plotting', wx.OK)
982            wx.PostEvent(self.parent.parent, 
983                             StatusEvent(status=msg, info="error")) 
984            return
985
[75790dc]986        for cell in grid.selected_cells:
987            row, col = cell
988            label_row = 0
989            label = grid.GetCellValue(label_row, col)
990            if label in grid.data:
991                values = grid.data[label]
[344c5d8]992                if row > len(values) or row < 1:
[9400de6]993                    msg = "Invalid cell was chosen." 
994                    wx.PostEvent(self.parent.parent, StatusEvent(status=msg, 
995                                                                info="error"))
996                    continue
[344c5d8]997                else:
998                     value = values[row -1]
[75790dc]999                if issubclass(value.__class__, BatchCell):
[fe98127]1000                    if value.object is None or len(value.object) == 0:
1001                        msg = "Row %s , " % str(row)
[c27a111]1002                        msg += "Column %s is NOT " % str(label)
1003                        msg += "the results of fits to view..."
[fe98127]1004                        #raise ValueError, msg
1005                        wx.PostEvent(self.parent.parent, StatusEvent(status=msg, 
1006                                                                info="error")) 
1007                        return
[75790dc]1008                    for new_plot in value.object:
[fe98127]1009                        if new_plot is None or \
[1f83477]1010                         not issubclass(new_plot.__class__, 
[fe98127]1011                                        plottables.Plottable):
[75790dc]1012                            msg = "Row %s , " % str(row)
[c27a111]1013                            msg += "Column %s is NOT " % str(label)
1014                            msg += "the results of fits to view..."
[75790dc]1015                            #raise ValueError, msg
1016                            wx.PostEvent(self.parent.parent, 
1017                                 StatusEvent(status=msg, info="error")) 
[d560a37]1018                            return
1019                            #continue
[63f4b8e]1020                        #new_plot.name =  title + ': ' + new_plot.title
[75790dc]1021                        if issubclass(new_plot.__class__, Data1D):
[23477c6]1022                            if label in grid.list_plot_panels.keys():
1023                                group_id = grid.list_plot_panels[label]
[fd51a7c]1024                            else:
[5d192cd]1025                                group_id = str(new_plot.group_id) + str(grid.uid)
[23477c6]1026                                grid.list_plot_panels[label] = group_id
[fd51a7c]1027                            if group_id not in new_plot.list_group_id:
1028                                new_plot.group_id = group_id
1029                                new_plot.list_group_id.append(group_id)
[75790dc]1030                        else:
[18a6556]1031                            if label.lower() in ["data", "chi2"]:
[75790dc]1032                                if len(grid.selected_cells) != 1:
[b18cf3d]1033                                    msg = "2D View: Please select one data set"
[4e0dfe4]1034                                    msg += " at a time for View Fit Results."
[75790dc]1035                                    wx.PostEvent(self.parent.parent, 
1036                                                 StatusEvent(status=msg,
[1ec979d]1037                                                              info="error"))
[d560a37]1038                                    #time.sleep(0.5)
1039                                    return
[86a9e6c]1040
[fd51a7c]1041                        wx.PostEvent(self.parent.parent, 
[75790dc]1042                                     NewPlotEvent(plot=new_plot, 
[1ec979d]1043                                                group_id=str(new_plot.group_id),
1044                                                title=title)) 
[4e0dfe4]1045                        msg = "Plotting the View Fit Results  completed!"
[c27a111]1046                        wx.PostEvent( self.parent.parent, 
1047                                      StatusEvent(status=msg)) 
[75790dc]1048                else:
1049                   
1050                    msg = "Row %s , " % str(row)
[c27a111]1051                    msg += "Column %s is NOT " % str(label)
1052                    msg += "the results of fits to view..."
[75790dc]1053                    #raise ValueError, msg
1054                    wx.PostEvent(self.parent.parent, 
1055                         StatusEvent(status=msg, info="error")) 
[d560a37]1056                    #time.sleep(0.5)
1057                    return
1058                    #continue
[75790dc]1059   
[24adb89]1060    def on_plot(self, event):
1061        """
[904830e]1062        Evaluate the contains of textcrtl and plot result
[24adb89]1063        """ 
[904830e]1064        pos = self.notebook.GetSelection()
1065        grid = self.notebook.GetPage(pos)
1066        column_names = {}
1067        if grid is not None:
1068            column_names = self.notebook.get_column_labels()
[08dc9e87]1069        #evaluate x
[904830e]1070        sentence = self.x_axis_label.GetValue()
[1c86a37]1071        try:
1072            if sentence.strip() == "":
[08dc9e87]1073                msg = "Select column values for x axis"
[1c86a37]1074                raise ValueError, msg
1075        except:
[86a9e6c]1076            msg = "X axis value error."
1077            wx.PostEvent(self.parent.parent, 
1078                            StatusEvent(status=msg, info="error")) 
1079            return
[904830e]1080        dict = parse_string(sentence, column_names.keys())
[86a9e6c]1081       
[9130227]1082        try:
[86a9e6c]1083            sentence = self.get_sentence(dict, sentence, column_names)
[9130227]1084            x = eval(sentence)
1085        except:
1086            msg = "Need a proper x-range."
1087            wx.PostEvent(self.parent.parent, 
1088                             StatusEvent(status=msg, info="error")) 
1089            return
[904830e]1090        #evaluate y
1091        sentence = self.y_axis_label.GetValue()
[86a9e6c]1092        try:
1093            if sentence.strip() == "":
1094                msg = "select value for y axis"
1095                raise ValueError, msg
1096        except:
1097            msg = "Y axis value error."
1098            wx.PostEvent(self.parent.parent, 
1099                            StatusEvent(status=msg, info="error")) 
1100            return
[904830e]1101        dict = parse_string(sentence, column_names.keys())
[9130227]1102        try:
[86a9e6c]1103            sentence = self.get_sentence(dict, sentence, column_names)
[9130227]1104            y = eval(sentence)
1105        except:
1106            msg = "Need a proper y-range."
1107            wx.PostEvent(self.parent.parent, 
1108                             StatusEvent(status=msg, info="error")) 
1109            return
1110        #evaluate y
1111        sentence = self.dy_axis_label.GetValue()
1112        dy = None
1113        if sentence.strip() != "":
1114            dict = parse_string(sentence, column_names.keys())
1115            sentence = self.get_sentence(dict, sentence, column_names)
1116            try:
1117                dy = eval(sentence)
1118            except:
1119                msg = "Need a proper dy-range."
1120                wx.PostEvent(self.parent.parent, 
1121                                 StatusEvent(status=msg, info="error")) 
1122                return
1123        if len(x) != len(y) or (len(x) == 0 or len(y) == 0):
[08dc9e87]1124            msg = "Need same length for X and Y axis and both greater than 0"
1125            msg += " to plot.\n"
1126            msg += "Got X length = %s, Y length = %s" % (str(len(x)),
1127                                                          str(len(y)))
1128            wx.PostEvent(self.parent.parent, 
1129                             StatusEvent(status=msg, info="error")) 
1130            return
[9130227]1131        if dy != None and (len(y) != len(dy)): 
1132            msg = "Need same length for Y and dY axis and both greater than 0"
1133            msg += " to plot.\n"
1134            msg += "Got Y length = %s, dY length = %s" % (str(len(y)),
1135                                                          str(len(dy)))
1136            wx.PostEvent(self.parent.parent, 
1137                             StatusEvent(status=msg, info="error")) 
1138            return
[c2e5898]1139        if dy == None:
1140            dy = numpy.zeros(len(y))
[904830e]1141        #plotting
[9130227]1142        new_plot = Data1D(x=x, y=y, dy=dy)
[24adb89]1143        new_plot.id =  wx.NewId()
[c2e5898]1144        new_plot.is_data = False
[24adb89]1145        new_plot.group_id = wx.NewId()
[d560a37]1146        y_title = self.y_axis_title.GetValue()
1147        x_title = self.x_axis_title.GetValue()
1148        title = "%s_vs_%s" % (y_title, 
1149                              x_title)
1150        new_plot.xaxis(x_title, 
[1ec979d]1151                       self.x_axis_unit.GetValue())
[d560a37]1152        new_plot.yaxis(y_title, 
[1ec979d]1153                       self.y_axis_unit.GetValue())
[904830e]1154        try:
[d560a37]1155            title = y_title.strip()
[86a9e6c]1156           
1157            title += "_" + self.notebook.GetPageText(pos)
1158            title += "_" + str(self.plot_num)
1159            self.plot_num += 1
[dadf255]1160            new_plot.name = title
[1c86a37]1161            new_plot.xtransform = "x"
1162            new_plot.ytransform  = "y" 
[d560a37]1163            #new_plot.is_data = False
[904830e]1164            wx.PostEvent(self.parent.parent, 
[1ec979d]1165                        NewPlotEvent(plot=new_plot, 
[c27a111]1166                        group_id=str(new_plot.group_id), title =title)) 
1167            msg = "Plotting completed!"
1168            wx.PostEvent( self.parent.parent, 
[c71d042]1169                                      StatusEvent(status=msg)) 
1170            self.parent.parent.update_theory(data_id=new_plot.id, 
1171                                              theory=new_plot) 
[904830e]1172        except:
[dadf255]1173             wx.PostEvent(self.parent.parent, 
[75790dc]1174                             StatusEvent(status=msg, info="error")) 
[9130227]1175   
1176    def get_sentence(self, dict, sentence, column_names):
1177        """
1178        Get sentence from dict
1179        """
1180        for tok, (col_name, list) in dict.iteritems():
1181            col = column_names[col_name]
1182            axis = self.get_plot_axis(col, list)
[0899c82]1183            if axis == None:
1184                return None
[9130227]1185            sentence = sentence.replace(tok, 
1186                                        "numpy.array(%s)" % str(axis))
1187        for key, value in FUNC_DICT.iteritems():
1188            sentence = sentence.replace(key.lower(), value)
1189        return sentence
1190           
[24adb89]1191    def layout_grid(self):
1192        """
1193        Draw the area related to the grid
1194        """
[904830e]1195        self.notebook = Notebook(parent=self)
[8523a1f2]1196        self.notebook.set_data(self._data_inputs, self._data_outputs)
[904830e]1197        self.grid_sizer.Add(self.notebook, 1, wx.EXPAND, 0)
[24adb89]1198       
1199    def layout_plotting_area(self):
1200        """
1201        Draw area containing options to plot
1202        """
[3e1177d]1203        view_description = wx.StaticBox(self, -1, 'Plot Fits/Residuals')
1204        note = "To plot the fits (or residuals), click the 'View Fits' button"
1205        note += "\n after highlighting the Data names (or Chi2 values)."
1206        note_text = wx.StaticText(self, -1, note)
1207        boxsizer1 = wx.StaticBoxSizer(view_description, wx.HORIZONTAL)
[904830e]1208        self.x_axis_title = wx.TextCtrl(self, -1)
1209        self.y_axis_title = wx.TextCtrl(self, -1)
1210        self.x_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
1211        self.y_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
[9130227]1212        self.dy_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
[9680906d]1213        self.x_axis_add = wx.Button(self, -1, "Add")
1214        self.x_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis, 
1215                            id=self.x_axis_add.GetId())
1216        self.y_axis_add = wx.Button(self, -1, "Add")
1217        self.y_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis, 
1218                            id=self.y_axis_add.GetId())
[9130227]1219        self.dy_axis_add = wx.Button(self, -1, "Add")
1220        self.dy_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis, 
1221                            id=self.dy_axis_add.GetId())
[24adb89]1222        self.x_axis_unit = wx.TextCtrl(self, -1)
1223        self.y_axis_unit = wx.TextCtrl(self, -1)
[4e0dfe4]1224        self.view_button = wx.Button(self, -1, "View Fits")
[7d47789]1225        view_tip = "Highlight the data set or the Chi2 column first."
[b18cf3d]1226        self.view_button.SetToolTipString(view_tip)
[75790dc]1227        wx.EVT_BUTTON(self, self.view_button.GetId(), self.on_view)
[24adb89]1228        self.plot_button = wx.Button(self, -1, "Plot")
[7d47789]1229        plot_tip = "Highlight a column for each axis and \n"
1230        plot_tip += "click the Add buttons first."
[3e1177d]1231       
[b18cf3d]1232        self.plot_button.SetToolTipString(plot_tip)
[3e1177d]1233        boxsizer1.AddMany([(note_text, 0, wx.LEFT, 10),
1234                           (self.view_button, 0, wx.LEFT|wx.RIGHT, 10)])
1235        self.button_sizer.AddMany([(boxsizer1, 0, 
1236                                    wx.LEFT|wx.RIGHT|wx.BOTTOM, 10), 
1237                                   (self.plot_button, 0, 
1238                                    wx.LEFT|wx.TOP|wx.BOTTOM|wx.EXPAND, 12)])
[75790dc]1239       
[24adb89]1240        wx.EVT_BUTTON(self, self.plot_button.GetId(), self.on_plot)
[9680906d]1241        self.plotting_sizer.AddMany([
[9bfa73f]1242                    (wx.StaticText(self, -1, 
1243                                   "X-axis Label\nSelection Range"), 1,
[904830e]1244                      wx.TOP|wx.BOTTOM|wx.LEFT, 10),
1245                    (self.x_axis_label, 1, wx.TOP|wx.BOTTOM, 10),
1246                    (self.x_axis_add, 1, wx.TOP|wx.BOTTOM|wx.RIGHT, 10),
[9bfa73f]1247                    (wx.StaticText(self, -1, "X-axis Label"), 1, 
[904830e]1248                     wx.TOP|wx.BOTTOM|wx.LEFT, 10),
1249                    (self.x_axis_title, 1, wx.TOP|wx.BOTTOM, 10),
[7124300]1250                    (wx.StaticText(self, -1 , "X-axis Unit"), 1, 
[904830e]1251                     wx.TOP|wx.BOTTOM, 10),
1252                    (self.x_axis_unit, 1, wx.TOP|wx.BOTTOM, 10),
[9bfa73f]1253                    (wx.StaticText(self, -1, 
1254                                   "Y-axis Label\nSelection Range"), 1, 
[904830e]1255                     wx.BOTTOM|wx.LEFT, 10),
1256                    (self.y_axis_label, wx.BOTTOM, 10),
1257                    (self.y_axis_add, 1, wx.BOTTOM|wx.RIGHT, 10),
[9bfa73f]1258                    (wx.StaticText(self, -1, "Y-axis Label"), 1, 
[904830e]1259                     wx.BOTTOM|wx.LEFT, 10),
1260                    (self.y_axis_title,  wx.BOTTOM, 10),
[9bfa73f]1261                    (wx.StaticText(self, -1 , "Y-axis Unit"), 1, wx.BOTTOM, 10),
[904830e]1262                    (self.y_axis_unit, 1, wx.BOTTOM, 10),
[9130227]1263                    (wx.StaticText(self, -1, 
[3553ad2]1264                                   "dY-Bar (Optional)\nSelection Range"), 
1265                                   1, wx.BOTTOM|wx.LEFT, 10),
[9130227]1266                    (self.dy_axis_label, wx.BOTTOM, 10),
1267                    (self.dy_axis_add, 1, wx.BOTTOM|wx.RIGHT, 10),
[9680906d]1268                      (-1, -1),
1269                      (-1, -1),
1270                      (-1, -1),
1271                      (-1, -1),
1272                      (-1, -1),
1273                      (-1, -1),
[75790dc]1274                      (-1, 1)])
[24adb89]1275   
[9680906d]1276    def on_edit_axis(self, event):
1277        """
1278        Get the selected column on  the visible grid and set values for axis
1279        """
[08dc9e87]1280        try:
1281            cell_list = self.notebook.on_edit_axis()
1282        except:
1283            msg = str(sys.exc_value)
1284            wx.PostEvent(self.parent.parent, 
1285                             StatusEvent(status=msg, info="error")) 
1286            return 
[904830e]1287        label, title = self.create_axis_label(cell_list)
1288        tcrtl = event.GetEventObject()
1289        if tcrtl == self.x_axis_add:
1290            self.edit_axis_helper(self.x_axis_label, self.x_axis_title,
1291                                   label, title)
1292        elif tcrtl == self.y_axis_add:
1293            self.edit_axis_helper(self.y_axis_label, self.y_axis_title,
1294                                   label, title)
[9130227]1295        elif tcrtl == self.dy_axis_add:
1296            self.edit_axis_helper(self.dy_axis_label, None,
1297                                   label, None)
[904830e]1298           
[7ad194fa]1299    def create_axis_label(self, cell_list):
1300        """
1301        Receive a list of cells and  create a string presenting the selected
1302        cells.
1303        :param cell_list: list of tuple
[24adb89]1304       
[7ad194fa]1305        """
[904830e]1306        if self.notebook is not None:
1307            return self.notebook.create_axis_label(cell_list)
[7ad194fa]1308   
[904830e]1309    def edit_axis_helper(self, tcrtl_label, tcrtl_title, label, title):
[9680906d]1310        """
[904830e]1311        get controls to modify
[9680906d]1312        """
[9130227]1313        if label != None:
1314            tcrtl_label.SetValue(str(label))
1315        if title != None:
1316            tcrtl_title.SetValue(str(title))
[904830e]1317       
[24adb89]1318    def add_column(self):
[9c8f3ad]1319        """
1320        """
[904830e]1321        if self.notebook is not None:
1322            self.notebook.add_column()
[24adb89]1323       
[9c8f3ad]1324    def on_remove_column(self):
1325        """
1326        """
[904830e]1327        if self.notebook is not None:
1328            self.notebook.on_remove_column()
[9c8f3ad]1329       
[24adb89]1330       
1331class GridFrame(wx.Frame):
[8523a1f2]1332    def __init__(self, parent=None, data_inputs=None, data_outputs=None, id=-1, 
[d560a37]1333                 title="Grid Window", size=(800, 500)):
[24adb89]1334        wx.Frame.__init__(self, parent=parent, id=id, title=title, size=size)
1335        self.parent = parent
[8523a1f2]1336        self.panel = GridPanel(self, data_inputs, data_outputs)
[24adb89]1337        menubar = wx.MenuBar()
1338        self.SetMenuBar(menubar)
[71fa9028]1339       
1340        self.curr_col = None
1341        self.curr_grid = None
1342        self.curr_col_name = ""
[14e4804]1343        self.file = wx.Menu()
1344        menubar.Append(self.file, "&File")
[71fa9028]1345       
1346        hint = "Open file containing batch results"
[14e4804]1347        open_menu = self.file.Append(wx.NewId(), 'Open ', hint)
[71fa9028]1348        wx.EVT_MENU(self, open_menu.GetId(), self.on_open)
1349       
1350        hint = "Open the the current grid into excel"
[14e4804]1351        self.open_excel_menu = self.file.Append(wx.NewId(), 'Open with Excel', hint)
1352        wx.EVT_MENU(self, self.open_excel_menu.GetId(), self.open_with_excel)
1353        self.file.AppendSeparator()
1354        self.save_menu = self.file.Append(wx.NewId(), 'Save As', 'Save into File')
1355        wx.EVT_MENU(self, self.save_menu.GetId(), self.on_save_page)
[71fa9028]1356       
1357        self.edit = wx.Menu()
[14e4804]1358       
1359        add_table_menu = self.edit.Append(-1, 'New Table', 
1360                                          'Add a New Table')
1361        self.edit.AppendSeparator()
1362        wx.EVT_MENU(self, add_table_menu.GetId(), self.add_table)
1363       
1364        self.copy_menu = self.edit.Append(-1, 'Copy', 
1365                                          'Copy the selected cells')
[0899c82]1366        wx.EVT_MENU(self, self.copy_menu.GetId(), self.on_copy)
[14e4804]1367        self.paste_menu = self.edit.Append(-1, 'Paste', 
1368                                           'Paste the selected Cells')
[0899c82]1369        wx.EVT_MENU(self, self.paste_menu.GetId(), self.on_paste)
[95b513c]1370        self.clear_menu = self.edit.Append(-1, 'Clear', 
1371                                           'Clear the selected Cells')
1372        wx.EVT_MENU(self, self.clear_menu.GetId(), self.on_clear)
1373
[14e4804]1374        self.edit.AppendSeparator()
[71fa9028]1375        hint = "Insert column before the selected column"
1376        self.insert_before_menu = wx.Menu()
[0899c82]1377        self.insertb_sub_menu = self.edit.AppendSubMenu(self.insert_before_menu, 
[71fa9028]1378                                                      'Insert Before', hint)
[0899c82]1379        hint = "Insert column after the selected column"
1380        self.insert_after_menu = wx.Menu()
1381        self.inserta_sub_menu = self.edit.AppendSubMenu(self.insert_after_menu, 
1382                                                      'Insert After', hint)
[71fa9028]1383        hint = "Remove the selected column"
1384        self.remove_menu = self.edit.Append(-1, 'Remove Column', hint)
1385        wx.EVT_MENU(self, self.remove_menu.GetId(), self.on_remove_column)
1386       
1387        self.Bind(wx.EVT_MENU_OPEN, self.on_menu_open)
1388        menubar.Append(self.edit, "&Edit")
[cb26857]1389        self.Bind(wx.EVT_CLOSE, self.on_close)
[0899c82]1390   
1391    def on_copy(self, event):
1392        """
1393        On Copy
1394        """
[14e4804]1395        if event != None:
1396            event.Skip()
[0899c82]1397        pos = self.panel.notebook.GetSelection()
1398        grid = self.panel.notebook.GetPage(pos)
1399        grid.Copy()
1400       
1401    def on_paste(self, event):
1402        """
1403        On Paste
1404        """
[14e4804]1405        if event != None:
1406            event.Skip()
[0899c82]1407        pos = self.panel.notebook.GetSelection()
1408        grid = self.panel.notebook.GetPage(pos)
[14e4804]1409        grid.on_paste(None)
[95b513c]1410
1411    def on_clear(self, event):
1412        """
1413        On Clear
1414        """
1415        pos = self.panel.notebook.GetSelection()
1416        grid = self.panel.notebook.GetPage(pos)
1417        grid.Clear()
[0899c82]1418     
[71fa9028]1419    def GetLabelText(self, id):
[656d65d]1420        """
[0899c82]1421        Get Label Text
[656d65d]1422        """
[71fa9028]1423        for item in self.insert_before_menu.GetMenuItems():
1424            m_id = item.GetId() 
1425            if m_id == id:
1426                return item.GetLabel() 
1427   
1428    def on_remove_column(self, event):
[cb26857]1429        """
[0899c82]1430        On remove column
[cb26857]1431        """
[71fa9028]1432        pos = self.panel.notebook.GetSelection()
1433        grid = self.panel.notebook.GetPage(pos)
1434        grid.on_remove_column(event=None)
[cb26857]1435       
[71fa9028]1436    def on_menu_open(self, event):
[9c8f3ad]1437        """
[0899c82]1438        On menu open
[9c8f3ad]1439        """
[14e4804]1440        if self.file == event.GetMenu():
1441            pos = self.panel.notebook.GetSelection()
1442            grid = self.panel.notebook.GetPage(pos)
1443            has_data = (grid.data != None and grid.data != {})
1444            self.open_excel_menu.Enable(has_data) 
1445            self.save_menu.Enable(has_data) 
1446           
[71fa9028]1447        if self.edit == event.GetMenu():
1448            #get the selected column
1449            pos = self.panel.notebook.GetSelection()
1450            grid = self.panel.notebook.GetPage(pos)
1451            col_list = grid.GetSelectedCols()
[0899c82]1452            has_selection = False
1453            selected_cel = grid.selected_cells
1454            if len(selected_cel) > 0:
1455                _row, _col = selected_cel[0]
1456                has_selection = grid.IsInSelection(_row, _col)
[14e4804]1457            if len(grid.selected_cols) > 0:
1458                has_selection = True
1459            if len(grid.selected_rows) > 0:
1460                has_selection = True
[0899c82]1461            self.copy_menu.Enable(has_selection)
[95b513c]1462            self.clear_menu.Enable(has_selection)
[0899c82]1463       
[71fa9028]1464            if len(col_list) > 0:
1465                self.remove_menu.Enable(True)
1466            else:
1467                self.remove_menu.Enable(False)
[a5e749f]1468            if len(col_list) == 0 or len(col_list) > 1:
[0899c82]1469                self.insertb_sub_menu.Enable(False)
1470                self.inserta_sub_menu.Enable(False)
[71fa9028]1471                label = "Insert Column Before"
[0899c82]1472                self.insertb_sub_menu.SetText(label)
1473                label = "Insert Column After"
1474                self.inserta_sub_menu.SetText(label)
[71fa9028]1475            else:
[0899c82]1476                self.insertb_sub_menu.Enable(True)
1477                self.inserta_sub_menu.Enable(True)
[71fa9028]1478               
1479                col = col_list[0]
1480                col_name = grid.GetCellValue(row=0, col=col)
1481                label = "Insert Column Before " + str(col_name)
[0899c82]1482                self.insertb_sub_menu.SetText(label)
[71fa9028]1483                for item in self.insert_before_menu.GetMenuItems():
1484                    self.insert_before_menu.DeleteItem(item)
1485                grid.insert_col_menu(menu=self.insert_before_menu, 
1486                                     label=col_name, window=self)
[0899c82]1487                label = "Insert Column After " + str(col_name)
1488                self.inserta_sub_menu.SetText(label)
1489                for item in self.insert_after_menu.GetMenuItems():
1490                    self.insert_after_menu.DeleteItem(item)
1491                grid.insert_after_col_menu(menu=self.insert_after_menu, 
1492                                     label=col_name, window=self)
[71fa9028]1493        event.Skip()
[9c8f3ad]1494       
[71fa9028]1495 
1496       
1497    def on_save_page(self, event):
1498        """
1499        """
1500        if self.parent is not None:
1501            pos = self.panel.notebook.GetSelection()
1502            grid = self.panel.notebook.GetPage(pos)
[7b48b08]1503            if grid.file_name is None or grid.file_name.strip() == "" or \
1504                grid.data is None or len(grid.data) == 0:
1505                name = self.panel.notebook.GetPageText(pos)
1506                msg = " %s has not data to save" % str(name)
1507                wx.PostEvent(self.parent, 
1508                             StatusEvent(status=msg, info="error")) 
1509           
1510                return
[71fa9028]1511            reader, ext = os.path.splitext(grid.file_name)
1512            path = None
1513            if self.parent is not None: 
1514                location = os.path.dirname(grid.file_name)
1515                dlg = wx.FileDialog(self, "Save Project file",
1516                            location, grid.file_name, ext, wx.SAVE)
1517                path = None
1518                if dlg.ShowModal() == wx.ID_OK:
1519                    path = dlg.GetPath()
1520                dlg.Destroy()
1521                if path != None:
1522                    if self.parent is not None:
1523                        data = grid.get_grid_view()
1524                        self.parent.write_batch_tofile(data=data, 
1525                                               file_name=path,
1526                                               details=grid.details)
1527   
1528    def on_open(self, event):
[656d65d]1529        """
[71fa9028]1530        Open file containg batch result
[656d65d]1531        """
[71fa9028]1532        if self.parent is not None:
[86a9e6c]1533            self.parent.on_read_batch_tofile(self)
[71fa9028]1534           
1535    def open_with_excel(self, event):
1536        """
1537        open excel and display batch result in Excel
1538        """
1539        if self.parent is not None:
1540            pos = self.panel.notebook.GetSelection()
1541            grid = self.panel.notebook.GetPage(pos)
1542            data = grid.get_grid_view()
[7b48b08]1543            if grid.file_name is None or grid.file_name.strip() == "" or \
1544                grid.data is None or len(grid.data) == 0:
1545                name = self.panel.notebook.GetPageText(pos)
1546                msg = " %s has not data to open on excel" % str(name)
1547                wx.PostEvent(self.parent, 
1548                             StatusEvent(status=msg, info="error")) 
1549           
1550                return
[71fa9028]1551            self.parent.open_with_externalapp(data=data,
[a5e749f]1552                                              file_name=grid.file_name, 
1553                                              details=grid.details)
[71fa9028]1554           
1555    def on_close(self, event):
1556        """
1557        """
1558        self.Hide()
[656d65d]1559       
1560    def on_append_column(self, event):
[cb26857]1561        """
[9c8f3ad]1562        Append a new column to the grid
[cb26857]1563        """
[24adb89]1564        self.panel.add_column()
[cb26857]1565       
[71fa9028]1566    def set_data(self, data_inputs, data_outputs, details="", file_name=None):
[cb26857]1567        """
[14e4804]1568        Set data
[cb26857]1569        """
[a84ca2a]1570        self.panel.notebook.set_data(data_inputs=data_inputs, 
[71fa9028]1571                            file_name=file_name,
1572                            details=details,
1573                            data_outputs=data_outputs)
[14e4804]1574
1575    def add_table(self, event):
1576        """
1577        Add a new table
1578        """
1579        # DO not event.Skip(): it will make 2 pages
1580        self.panel.notebook.add_empty_page()     
[24adb89]1581     
[83eb1b52]1582class BatchOutputFrame(wx.Frame):
[73197d0]1583    """
1584    Allow to select where the result of batch will be displayed or stored
1585    """
[8523a1f2]1586    def __init__(self, parent, data_inputs, data_outputs, file_name="",
[850525c]1587                 details="", *args, **kwds):
[73197d0]1588        """
1589        :param parent: Window instantiating this dialog
1590        :param result: result to display in a grid or export to an external
1591                application.
1592        """
[850525c]1593        #kwds['style'] = wx.CAPTION|wx.SYSTEM_MENU
[83eb1b52]1594        wx.Frame.__init__(self, parent, *args, **kwds)
[73197d0]1595        self.parent = parent
[83eb1b52]1596        self.panel = wx.Panel(self)
[850525c]1597        self.file_name = file_name
1598        self.details = details
[8523a1f2]1599        self.data_inputs = data_inputs
1600        self.data_outputs = data_outputs
1601        self.data = {}
1602        for item in (self.data_outputs, self.data_inputs):
1603            self.data.update(item)
[73197d0]1604        self.flag = 1
1605        self.SetSize((300, 200))
1606        self.local_app_selected = None
1607        self.external_app_selected = None
1608        self.save_to_file = None
1609        self._do_layout()
[8523a1f2]1610   
[73197d0]1611    def _do_layout(self):
1612        """
1613        Draw the content of the current dialog window
1614        """
1615        vbox = wx.BoxSizer(wx.VERTICAL)
[83eb1b52]1616        box_description = wx.StaticBox(self.panel, -1, str("Batch Outputs"))
[73197d0]1617        hint_sizer = wx.StaticBoxSizer(box_description, wx.VERTICAL)
[83eb1b52]1618        selection_sizer = wx.GridBagSizer(5, 5)
[73197d0]1619        button_sizer = wx.BoxSizer(wx.HORIZONTAL)
[caf3a08f]1620        text = "Open with %s" % self.parent.application_name
[83eb1b52]1621        self.local_app_selected = wx.RadioButton(self.panel, -1, text,
[73197d0]1622                                                style=wx.RB_GROUP)
1623        self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
1624                    id=self.local_app_selected.GetId())
1625        text = "Open with Excel"
[83eb1b52]1626        self.external_app_selected  = wx.RadioButton(self.panel, -1, text)
[73197d0]1627        self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
1628                    id=self.external_app_selected.GetId())
[caf3a08f]1629        text = "Save to File"
[83eb1b52]1630        self.save_to_file = wx.CheckBox(self.panel, -1, text)
[73197d0]1631        self.Bind(wx.EVT_CHECKBOX, self.onselect,
1632                    id=self.save_to_file.GetId())
1633        self.local_app_selected.SetValue(True)
1634        self.external_app_selected.SetValue(False)
1635        self.save_to_file.SetValue(False)
[83eb1b52]1636        button_close = wx.Button(self.panel, -1, "Close")
1637        button_close.Bind(wx.EVT_BUTTON, id=button_close.GetId(),
1638                           handler=self.on_close)
1639        button_apply = wx.Button(self.panel, -1, "Apply")
1640        button_apply.Bind(wx.EVT_BUTTON, id=button_apply.GetId(),
1641                        handler=self.on_apply)
1642        button_apply.SetFocus()
[73197d0]1643        hint = ""
[83eb1b52]1644        hint_sizer.Add(wx.StaticText(self.panel, -1, hint))
[73197d0]1645        hint_sizer.Add(selection_sizer)
1646        #draw area containing radio buttons
1647        ix = 0
1648        iy = 0
1649        selection_sizer.Add(self.local_app_selected, (iy, ix),
1650                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1651        iy += 1
1652        selection_sizer.Add(self.external_app_selected, (iy, ix),
1653                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1654        iy += 1
1655        selection_sizer.Add(self.save_to_file, (iy, ix),
1656                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1657        #contruction the sizer contaning button
1658        button_sizer.Add((20, 20), 1, wx.EXPAND|wx.ADJUST_MINSIZE, 0)
[caf3a08f]1659
1660        button_sizer.Add(button_close, 0,
1661                        wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
[83eb1b52]1662        button_sizer.Add(button_apply, 0,
[73197d0]1663                                wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10)
[a5e749f]1664        vbox.Add(hint_sizer, 0, wx.EXPAND|wx.ALL, 10)
[83eb1b52]1665        vbox.Add(wx.StaticLine(self.panel, -1),  0, wx.EXPAND, 0)
[a5e749f]1666        vbox.Add(button_sizer, 0, wx.TOP|wx.BOTTOM, 10)
[73197d0]1667        self.SetSizer(vbox)
1668       
[83eb1b52]1669    def on_apply(self, event):
1670        """
1671        Get the user selection and display output to the selected application
1672        """
1673        if self.flag == 1:
[8523a1f2]1674            self.parent.open_with_localapp(data_inputs=self.data_inputs,
[a5e749f]1675                                           data_outputs=self.data_outputs)
[83eb1b52]1676        elif self.flag == 2:
1677            self.parent.open_with_externalapp(data=self.data, 
[a5e749f]1678                                              file_name=self.file_name,
1679                                              details=self.details)
[83eb1b52]1680    def on_close(self, event):
1681        """
1682        close the Window
1683        """
1684        self.Close()
1685       
[73197d0]1686    def onselect(self, event=None):
1687        """
1688        Receive event and display data into third party application
1689        or save data to file.
1690       
1691        """
1692        if self.save_to_file.GetValue():
[850525c]1693            reader, ext = os.path.splitext(self.file_name)
1694            path = None
1695            location = os.getcwd()
1696            if self.parent is not None: 
[83eb1b52]1697                location = os.path.dirname(self.file_name)
[850525c]1698                dlg = wx.FileDialog(self, "Save Project file",
1699                            location, self.file_name, ext, wx.SAVE)
1700                path = None
1701                if dlg.ShowModal() == wx.ID_OK:
1702                    path = dlg.GetPath()
1703                dlg.Destroy()
1704                if path != None:
1705                    if self.parent is not None and  self.data is not None:
[83eb1b52]1706                        self.parent.write_batch_tofile(data=self.data, 
[a5e749f]1707                                                       file_name=path,
1708                                                       details=self.details)
[850525c]1709        if self.local_app_selected.GetValue():
[73197d0]1710            self.flag = 1
1711        else:
1712            self.flag = 2
1713        return self.flag
1714   
1715 
1716       
[24adb89]1717if __name__ == "__main__":
1718    app = wx.App()
1719   
1720    try:
1721        data = {}
1722        j = 0
1723        for i in range(4):
1724            j += 1
1725            data["index"+str(i)] = [i/j, i*j, i, i+j]
[656d65d]1726       
1727        data_input =  copy.deepcopy(data)   
[a5e749f]1728        data_input["index5"] = [10, 20, 40, 50]
[656d65d]1729        frame = GridFrame(data_outputs=data, data_inputs=data_input)
[24adb89]1730        frame.Show(True)
1731    except:
1732        print sys.exc_value
1733       
1734    app.MainLoop()
Note: See TracBrowser for help on using the repository browser.