source: sasview/sansguiframe/src/sans/guiframe/data_processor.py @ 4bab790

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 4bab790 was 5531a46, checked in by Gervaise Alina <gervyh@…>, 13 years ago

make sure the grid is not editable when there is no data

  • Property mode set to 100644
File size: 52.4 KB
Line 
1"""
2Implement grid used to store data
3"""
4import wx
5import numpy
6import math
7import time
8import re
9import os
10import sys
11import copy
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
17from sans.guiframe.panel_base import PanelBase
18from sans.guiframe.utils import format_number
19from sans.guiframe.events import NewPlotEvent
20from sans.guiframe.events import StatusEvent 
21from danse.common.plottools import plottables
22from sans.guiframe.dataFitting import Data1D
23
24FUNC_DICT = {"sqrt": "math.sqrt",
25             "pow": "math.sqrt"}
26
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
37        self.object = []
38       
39
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)
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
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:
72                    # continuous range
73                    temp = elt.split(label)
74                    for item in temp:
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)
81                col_dict[elt] = (label, temp_arr)
82    return col_dict
83
84         
85         
86class SPanel(ScrolledPanel):
87    def __init__(self, parent, *args, **kwds):
88        ScrolledPanel.__init__(self, parent , *args, **kwds)
89        self.SetupScrolling() 
90       
91class GridPage(sheet.CSheet):
92    """
93    """
94    def __init__(self, parent, panel=None):
95        """
96        """
97        sheet.CSheet.__init__(self, parent)
98       
99        self.AdjustScrollbars()
100        #self.SetLabelBackgroundColour('#DBD4D4')
101        self.uid = wx.NewId()
102        self.parent = parent
103        self.panel = panel
104        self.col_names = []
105        self.data_inputs = {}
106        self.data_outputs = {}
107        self.data = None
108        self.details = ""
109        self.file_name = None
110        self._cols = 50
111        self._rows = 51
112        self.last_selected_row = -1
113        self.last_selected_col = -1
114        self.col_width = 30
115        self.row_height = 20
116        self.max_row_touse = 0
117        self.axis_value = []
118        self.axis_label = ""
119        self.selected_cells = []
120        self.selected_cols = []
121        self.selected_rows = []
122        self.plottable_cells = []
123        self.plottable_flag = False
124        self.SetColMinimalAcceptableWidth(self.col_width)
125        self.SetRowMinimalAcceptableHeight(self.row_height)
126        self.SetNumberRows(self._cols)
127        self.SetNumberCols(self._rows)
128        self.AutoSize()
129        self.list_plot_panels = {}
130        self.default_col_width = 75
131        self.EnableEditing(False)
132        if self.GetNumberCols() > 0:
133            self.default_col_width =  self.GetColSize(0)
134        self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_CLICK, self.on_left_click)
135        self.Bind(wx.grid.EVT_GRID_LABEL_RIGHT_CLICK, self.on_right_click)
136        self.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.on_selected_cell)
137        self.Bind(wx.grid.EVT_GRID_CMD_CELL_CHANGE, self.on_edit_cell)
138       
139       
140    def on_edit_cell(self, event):
141        """
142        """
143        row, col = event.GetRow(), event.GetCol()
144        if row > self.max_row_touse:
145            self.max_row_touse = row
146        event.Skip()
147       
148    def on_selected_cell(self, event):
149        """
150        Handler catching cell selection
151        """
152        flag = event.CmdDown() or event.ControlDown()
153        row, col = event.GetRow(), event.GetCol()
154        cell = (row, col)
155        event.Skip()
156        if not flag:
157            self.selected_cols = []
158            self.selected_rows = []
159            self.selected_cells = []
160            self.axis_label = ""
161            self.axis_value = []
162            self.plottable_list = []
163            self.plottable_cells = []
164            self.plottable_flag = False
165        self.last_selected_col = col
166        self.last_selected_row = row
167        if col >= 0:
168            label_row = 0
169            self.axis_label = self.GetCellValue(label_row, col)
170            self.selected_cols.append(col)
171        if cell not in self.selected_cells:
172            if row > 0:
173                self.selected_cells.append(cell)
174                self.selected_rows.append(row)
175        else:
176            if flag:
177                self.selected_cells.remove(cell)
178        self.axis_value = []
179        for cell_row, cell_col in self.selected_cells:
180            if cell_row > 0 and cell_row < self.max_row_touse:
181                self.axis_value.append(self.GetCellValue(cell_row, cell_col))
182
183               
184    def on_left_click(self, event):
185        """
186        Catch the left click on label mouse event
187        """
188        event.Skip()
189        flag = event.CmdDown() or event.ControlDown()
190        col = event.GetCol()
191        row = event.GetRow()
192        if not flag:
193            self.selected_cols = []
194            self.selected_rows = []
195            self.selected_cells = []
196            self.axis_label = ""
197            self.axis_value = []
198            self.plottable_list = []
199            self.plottable_cells = []
200            self.plottable_flag = False
201       
202        self.last_selected_col = col
203        self.last_selected_row = row
204        if row != -1 and row not in self.selected_rows:
205             self.selected_rows.append(row)
206             
207        if col != -1:
208            for row in range(1, self.GetNumberRows()+ 1):
209                cell = (row, col)
210                if row > 0 and row < self.max_row_touse:
211                    if cell not in self.selected_cells:
212                        self.selected_cells.append(cell)
213                    else:
214                        if flag:
215                             self.selected_cells.remove(cell)
216            self.selected_cols.append(col)
217            self.axis_value = []
218            for cell_row, cell_col in self.selected_cells:
219                self.axis_value.append(self.GetCellValue(cell_row, cell_col))
220            self.axis_label = self.GetCellValue(0, col)
221       
222    def on_right_click(self, event):
223        """
224        Catch the right click mouse
225        """
226       
227        col = event.GetCol()
228        self.selected_cols = []
229        self.selected_cols.append(col)
230        # Slicer plot popup menu
231        slicerpop = wx.Menu()
232        col_label_menu  = wx.Menu()
233        c_name = self.GetCellValue(0, col) 
234        label = "Insert column before %s " % str(c_name)
235        slicerpop.AppendSubMenu(col_label_menu , 
236                                 '&%s' % str(label), str(label))
237        col_name = [self.GetCellValue(0, c) 
238                        for c in range(self.GetNumberCols())]
239        row = 0
240        label = self.GetCellValue(row, col)
241        self.insert_col_menu(col_label_menu, label, self)
242       
243       
244        col_after_menu  = wx.Menu()
245        label = "Insert column after %s " % str(c_name)
246        slicerpop.AppendSubMenu(col_after_menu , 
247                                 '&%s' % str(label), str(label))
248        col_name = [self.GetCellValue(0, c) 
249                        for c in range(self.GetNumberCols())]
250        self.insert_after_col_menu(col_after_menu, label, self)
251       
252       
253        id = wx.NewId()   
254        hint = 'Remove selected column %s'
255        slicerpop.Append(id, '&Remove Column', hint)
256        wx.EVT_MENU(self, id, self.on_remove_column)
257       
258        pos = wx.GetMousePosition()
259        pos = self.ScreenToClient(pos)
260        self.PopupMenu(slicerpop, pos)
261        event.Skip()
262       
263    def insert_col_menu(self, menu, label, window):
264        """
265        """
266        if self.data is None:
267            return
268        id = wx.NewId()
269        title = "Empty"
270        hint = 'Insert empty column before %s' % str(label)
271        menu.Append(id, title, hint)
272        wx.EVT_MENU(window, id, self.on_insert_column)
273        row = 0
274        col_name = [self.GetCellValue(row, col) 
275                        for col in range(self.GetNumberCols())]
276        for c_name in self.data.keys():
277            if c_name not in col_name:
278                id = wx.NewId()
279                hint = "Insert %s column before the " % str(c_name)
280                hint += " %s column" % str(label)
281                menu.Append(id, '&%s' % str(c_name), hint)
282                wx.EVT_MENU(window, id, self.on_insert_column)
283           
284    def insert_after_col_menu(self, menu, label, window):
285        """
286        """
287        if self.data is None:
288            return
289        id = wx.NewId()
290        title = "Empty"
291        hint = 'Insert empty column after %s' % str(label)
292        menu.Append(id, title, hint)
293        wx.EVT_MENU(window, id, self.on_insert_after_column)
294        row = 0
295        col_name = [self.GetCellValue(row, col) 
296                        for col in range(self.GetNumberCols())]
297        for c_name in self.data.keys():
298            if c_name not in col_name:
299                id = wx.NewId()
300                hint = "Insert %s column after the " % str(c_name)
301                hint += " %s column" % str(label)
302                menu.Append(id, '&%s' % str(c_name), hint)
303                wx.EVT_MENU(window, id, self.on_insert_after_column)
304                               
305    def on_remove_column(self, event=None):
306        """
307        """
308        if self.selected_cols is not None or len(self.selected_cols) > 0:
309            col = self.selected_cols[0]
310            self.remove_column(col=col, numCols=1)
311           
312    def remove_column(self, col, numCols=1):
313        """
314        Remove column to the current grid
315        """
316        # add data to the grid   
317        row = 0
318        col_name = self.GetCellValue(row, col)
319        self.data[col_name] = []
320        for row in range(1, self.GetNumberRows() + 1):
321            if row < self.max_row_touse:
322                value = self.GetCellValue(row, col)
323                self.data[col_name].append(value)
324                for k , value_list in self.data.iteritems():
325                    if k != col_name:
326                        length = len(value_list)
327                        if length < self.max_row_touse:
328                            diff = self.max_row_touse - length
329                            for i in range(diff):
330                                self.data[k].append("")
331        self.DeleteCols(pos=col, numCols=numCols, updateLabels=True)
332           
333    def on_insert_column(self, event):
334        """
335        """
336        if self.selected_cols is not None or len(self.selected_cols) > 0:
337            col = self.selected_cols[0]
338            # add data to the grid   
339            row = 0
340            id = event.GetId()
341            col_name = event.GetEventObject().GetLabelText(id)
342            self.insert_column(col=col, col_name=col_name)
343            if  not issubclass(event.GetEventObject().__class__ , wx.Menu):
344                col += 1
345                self.selected_cols[0] += 1
346               
347    def on_insert_after_column(self, event):
348        """
349        Insert the given column after the highlighted column
350        """
351        if self.selected_cols is not None or len(self.selected_cols) > 0:
352            col = self.selected_cols[0] + 1
353            # add data to the grid   
354            row = 0
355            id = event.GetId()
356            col_name = event.GetEventObject().GetLabelText(id)
357            self.insert_column(col=col, col_name=col_name)
358            if  not issubclass(event.GetEventObject().__class__ , wx.Menu):
359                self.selected_cols[0] += 1
360           
361    def insert_column(self, col, col_name):
362        """
363        """ 
364         
365        row = 0
366        self.InsertCols(pos=col, numCols=1, updateLabels=True)
367        if col_name.strip() != "Empty":
368            self.SetCellValue(row, col, str(col_name.strip()))
369        if col_name in self.data.keys():
370            value_list = self.data[col_name]
371            cell_row =  1
372            for value in value_list:
373                label = value#format_number(value, high=True)
374                self.SetCellValue(cell_row, col, str(label))
375                cell_row += 1
376        self.AutoSizeColumn(col, True)
377        width = self.GetColSize(col)
378        if width < self.default_col_width:
379           self.SetColSize(col, self.default_col_width)
380        self.ForceRefresh()
381       
382    def on_set_x_axis(self, event):
383        """
384        """
385        self.panel.set_xaxis(x=self.axis_value, label=self.axis_label)
386   
387    def on_set_y_axis(self, event):
388        """
389        """
390        self.panel.set_yaxis(y=self.axis_value, label=self.axis_label)     
391           
392    def set_data(self, data_inputs, data_outputs, details, file_name):
393        """
394        Add data to the grid
395        :param data_inputs: data to use from the context menu of the grid
396        :param data_ouputs: default columns deplayed
397        """
398        self.file_name = file_name
399        self.details = details
400       
401        if data_outputs is None:
402            data_outputs = {}
403        self.data_outputs = data_outputs
404        if data_inputs is None:
405            data_inputs = {}
406        self.data_inputs = data_inputs
407        self.data = {}
408        for item in (self.data_outputs, self.data_inputs):
409            self.data.update(item)
410           
411        if len(self.data) + len(self.data_inputs) +len(self.data_outputs) == 0:
412            self.EnableEditing(False)
413        else:
414            self.EnableEditing(True)
415        if  len(self.data_outputs) > 0:
416            self._cols = self.GetNumberCols()
417            self._rows = self.GetNumberRows()
418            self.col_names = self.data_outputs.keys()
419            self.col_names.sort() 
420            nbr_user_cols = len(self.col_names)
421            #Add more columns to the grid if necessary
422            if nbr_user_cols > self._cols:
423                new_col_nbr = nbr_user_cols -  self._cols
424                self.AppendCols(new_col_nbr, True)
425            #Add more rows to the grid if necessary 
426            nbr_user_row = len(self.data_outputs.values())
427            if nbr_user_row > self._rows + 1:
428                new_row_nbr =  nbr_user_row - self._rows
429                self.AppendRows(new_row_nbr, True)
430            # add data to the grid   
431            row = 0
432            col = 0
433            cell_col = 0
434            for col_name in  self.col_names:
435                # use the first row of the grid to add user defined labels
436                self.SetCellValue(row, col, str(col_name))
437                col += 1
438                cell_row =  1
439                value_list = self.data_outputs[col_name]
440               
441                for value in value_list:
442                    label = value
443                    if issubclass(value.__class__, BatchCell):
444                        label = value.label
445                    try:
446                        float(label)
447                        label = str(label)#format_number(label, high=True)
448                    except:
449                        label = str(label)
450                    self.SetCellValue(cell_row, cell_col, label)
451                    self.AutoSizeColumn(cell_col, True)
452                    width = self.GetColSize(cell_col)
453                    if width < self.default_col_width:
454                       self.SetColSize(cell_col, self.default_col_width)
455                   
456                    cell_row += 1
457                cell_col += 1
458                if cell_row > self.max_row_touse:
459                    self.max_row_touse = cell_row
460        self.ForceRefresh()
461       
462    def get_grid_view(self):
463        """
464        Return value contained in the grid
465        """
466        grid_view = {}
467        for col in xrange(self.GetNumberCols()):
468            label = self.GetCellValue(row=0, col=col) 
469            label = label.strip()
470            if label != "":
471                grid_view[label] = []
472                for row in range(1, self.max_row_touse):
473                    value = self.GetCellValue(row=row, col=col)
474                    if value != "":
475                        grid_view[label].append(value) 
476                    else:
477                        grid_view[label].append(None) 
478        return grid_view
479   
480class Notebook(nb, PanelBase):
481    """
482    ## Internal name for the AUI manager
483    window_name = "Fit panel"
484    ## Title to appear on top of the window
485    """
486    window_caption = "Notebook "
487   
488    def __init__(self, parent, manager=None, data=None, *args, **kwargs):
489        """
490        """
491        nb.__init__(self, parent, -1,
492                    style=wx.aui.AUI_NB_WINDOWLIST_BUTTON| 
493                    wx.aui.AUI_BUTTON_DOWN|
494                    wx.aui.AUI_NB_DEFAULT_STYLE|
495                    wx.CLIP_CHILDREN)
496        PanelBase.__init__(self, parent)
497        self.enable_close_button()
498        self.parent = parent
499        self.manager = manager
500        self.data = data
501        #add empty page
502        self.add_empty_page()
503       
504        self.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE, self.on_close_page)
505   
506    def add_empty_page(self):
507        """
508        """
509        grid = GridPage(self, panel=self.parent)
510        self.AddPage(grid, "", True)
511        pos = self.GetPageIndex(grid)
512        title = "Batch" + str(self.GetPageCount())
513        self.SetPageText(pos, title)
514        self.SetSelection(pos)
515        return grid , pos
516       
517    def enable_close_button(self):
518        """
519        display the close button on tab for more than 1 tabs else remove the
520        close button
521        """
522        if self.GetPageCount() <= 1:
523            style = self.GetWindowStyleFlag() 
524            flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
525            if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB == flag:
526                style = style & ~wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
527                self.SetWindowStyle(style)
528        else:
529            style = self.GetWindowStyleFlag()
530            flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
531            if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB != flag:
532                style |= wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
533                self.SetWindowStyle(style)
534             
535    def on_edit_axis(self):
536        """
537        Return the select cell of a given selected column. Check that all cells
538        are from the same column
539        """
540        pos = self.GetSelection()
541        grid = self.GetPage(pos)
542        if len(grid.selected_cols) >= 1:
543            col = grid.selected_cols[0]
544            for c in grid.selected_cols:
545                if c != col:
546                    msg = "Edit axis doesn't understand this selection.\n"
547                    msg += "Please select only one column"
548                    raise ValueError, msg
549            for (cell_row, cell_col) in grid.selected_cells:
550                if cell_col != col:
551                    msg = "Cannot use cells from different columns for "
552                    msg += "this operation.\n"
553                    msg += "Please select elements of the same col.\n"
554                    raise ValueError, msg
555        else:
556            msg = "No item selected.\n"
557            msg += "Please select only one column or one cell"
558            raise ValueError, msg
559       
560        return grid.selected_cells
561       
562   
563    def get_column_labels(self):
564        """
565        return dictionary of columns labels of the current page
566        """
567        pos = self.GetSelection()
568        grid = self.GetPage(pos)
569        labels = {}
570        row = 0
571        for col in range(grid.GetNumberCols()):
572            label = grid.GetCellValue(row, col)
573            if label.strip() != "" :
574                labels[label.strip()] = col
575        return labels
576       
577    def create_axis_label(self, cell_list):
578        """
579        Receive a list of cells and  create a string presenting the selected
580        cells.
581        :param cell_list: list of tuple
582       
583        """
584        pos = self.GetSelection()
585        grid = self.GetPage(pos)
586        label = ""
587        col_name = ""
588        def create_label(col_name,  row_min=None, row_max=None):
589            """
590            """
591            result = ""
592            if row_min is not  None or row_max is not None:
593                if row_min is None:
594                    result = str(row_max) + "]"
595                elif row_max is None:
596                     result = str(col_name) + "[" + str(row_min) + ":"
597                else:
598                    result = str(col_name) +  "[" + str(row_min) + ":"
599                    result += str(row_max) + "]"
600            return str(result)
601           
602        if len(cell_list) > 0:
603            if len(cell_list) == 1:
604                 row_min, col  = cell_list[0]   
605                 col_name = grid.GetCellValue(0, col)
606                 label = create_label(col_name, row_min+1 , row_min+1)
607                 return  label,  col_name
608            else:
609                temp_list = copy.deepcopy(cell_list)
610                temp_list.sort()
611                length = len(temp_list)
612                row_min, col  = temp_list[0]   
613                row_max, _  = temp_list[length-1]
614                col_name = grid.GetCellValue(0, col)
615                index = 0
616                for row in xrange(row_min, row_max +1):
617                    if index > 0 and index < len(temp_list):
618                        new_row, _ = temp_list[index]
619                        if row != new_row:
620                            temp_list.insert(index, (None, None))
621                            if index -1 >= 0:
622                                new_row, _ = temp_list[index-1]
623                                label += create_label(col_name, None, new_row +1)
624                                label += ","
625                            if index + 1 < len(temp_list):
626                                new_row, _ = temp_list[index + 1]
627                                label += create_label(col_name, new_row+1, None)
628                    if index == 0:
629                        label += create_label(col_name,  row_min+1, None)
630                    elif index == len(temp_list)-1:
631                        label += create_label(col_name, None, row_max+1)
632                    index += 1
633                return label, col_name
634   
635    def on_close_page(self, event):
636        """
637        close the page
638        """
639        if self.GetPageCount() == 1:
640            event.Veto()
641        self.enable_close_button()
642       
643    def set_data(self, data_inputs, data_outputs, details="", file_name=None):
644        if data_outputs is None or data_outputs == {}:
645            return
646       
647        for pos in range(self.GetPageCount()):
648            grid = self.GetPage(pos)
649            if grid.data is None:
650                #Found empty page
651                grid.set_data(data_inputs=data_inputs, 
652                              data_outputs=data_outputs,
653                              details=details,
654                              file_name=file_name) 
655                self.SetSelection(pos) 
656                return
657               
658        grid, pos = self.add_empty_page()
659        grid.set_data(data_inputs=data_inputs, 
660                      data_outputs=data_outputs,
661                      file_name=file_name,
662                      details=details)
663   
664    def add_column(self):
665        """
666        Append a new column to the grid
667        """
668        pos = self.GetSelection()
669        grid = self.GetPage(pos)
670        grid.AppendCols(1, True)
671       
672   
673
674
675
676class GridPanel(SPanel):
677    def __init__(self, parent, data_inputs=None,
678                 data_outputs=None, *args, **kwds):
679        SPanel.__init__(self, parent , *args, **kwds)
680       
681        self.vbox = wx.BoxSizer(wx.VERTICAL)
682       
683        self.plotting_sizer = wx.FlexGridSizer(3, 7, 10, 5)
684        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
685        self.grid_sizer = wx.BoxSizer(wx.HORIZONTAL)
686        self.vbox.AddMany([(self.grid_sizer, 1, wx.EXPAND, 0),
687                           (wx.StaticLine(self, -1), 0, wx.EXPAND, 0),
688                           (self.plotting_sizer),
689                           (self.button_sizer)])
690        self.parent = parent
691        self._data_inputs = data_inputs
692        self._data_outputs = data_outputs
693        self.x = []
694        self.= []
695        self.x_axis_label = None
696        self.y_axis_label = None
697        self.x_axis_title = None
698        self.y_axis_title = None
699        self.x_axis_unit = None
700        self.y_axis_unit = None
701        self.view_button = None
702        self.plot_button = None
703        self.notebook = None
704       
705        self.layout_grid()
706        self.layout_plotting_area()
707        self.SetSizer(self.vbox)
708 
709    def set_xaxis(self, label="", x=None):
710        """
711        """
712        if x is None:
713            x = []
714        self.x = x
715        self.x_axis_label.SetValue("%s[:]" % str(label))
716        self.x_axis_title.SetValue(str(label))
717       
718    def set_yaxis(self, label="", y=None):
719        """
720        """
721        if y is None:
722            y = []
723        self.y = y
724        self.y_axis_label.SetValue("%s[:]" % str(label))
725        self.y_axis_title.SetValue(str(label))
726       
727    def get_plot_axis(self, col, list):
728        """
729       
730        """
731        axis = []
732        pos = self.notebook.GetSelection()
733        grid = self.notebook.GetPage(pos)
734        for row in list:
735            label = grid.GetCellValue(0, col)
736            value = grid.GetCellValue(row - 1, col).strip()
737            if value != "":
738                if label.lower().strip() == "data":
739                    axis.append(float(row - 1))
740                else:
741                    try:
742                        axis.append(float(value))
743                    except:
744                        msg = "Invalid data in row %s column %s" % (str(row),
745                                                                    str(col))
746                        wx.PostEvent(self.parent.parent, 
747                             StatusEvent(status=msg, info="error")) 
748            else:
749                axis.append(None) 
750        return axis
751   
752    def on_save_column(self, parent):
753        """
754        """
755        pos = self.notebook.GetSelection()
756        grid = self.notebook.GetPage(pos)
757        if parent is not None and  self.data is not None:
758            parent.write_batch_tofile(data=grid.data, 
759                                               file_name=path,
760                                               details=self.details)
761       
762    def on_view(self, event):
763        """
764        Get object represented buy the given cell and plot them.
765        """
766        pos = self.notebook.GetSelection()
767        grid = self.notebook.GetPage(pos)
768        title = self.notebook.GetPageText(pos)
769        if len(grid.selected_cells) == 0:
770            msg = "Highlight a Data or Chi2 column first..."
771            wx.PostEvent(self.parent.parent, 
772                             StatusEvent(status=msg, info="error")) 
773            return
774        for cell in grid.selected_cells:
775            row, col = cell
776            label_row = 0
777            label = grid.GetCellValue(label_row, col)
778            if label in grid.data:
779                values = grid.data[label]
780                if row > len(values) or row < 1:
781                    msg = "Invalid cell was chosen." 
782                    wx.PostEvent(self.parent.parent, StatusEvent(status=msg, 
783                                                                info="error"))
784                    time.sleep(0.5)
785                    continue
786                else:
787                     value = values[row -1]
788                if issubclass(value.__class__, BatchCell):
789                    if value.object is None or len(value.object) == 0:
790                        msg = "Row %s , " % str(row)
791                        msg += "Column %s is NOT " % str(label)
792                        msg += "the results of fits to view..."
793                        #raise ValueError, msg
794                        wx.PostEvent(self.parent.parent, StatusEvent(status=msg, 
795                                                                info="error")) 
796                        return
797                    for new_plot in value.object:
798                        if new_plot is None or \
799                         not issubclass(new_plot.__class__, 
800                                        plottables.Plottable):
801                            msg = "Row %s , " % str(row)
802                            msg += "Column %s is NOT " % str(label)
803                            msg += "the results of fits to view..."
804                            #raise ValueError, msg
805                            wx.PostEvent(self.parent.parent, 
806                                 StatusEvent(status=msg, info="error")) 
807                            time.sleep(0.5)
808                            continue
809                        #new_plot.name =  title + ': ' + new_plot.title
810                        if issubclass(new_plot.__class__, Data1D):
811                            if label in grid.list_plot_panels.keys():
812                                group_id = grid.list_plot_panels[label]
813                            else:
814                                group_id = str(new_plot.group_id) + str(grid.uid)
815                                grid.list_plot_panels[label] = group_id
816                            if group_id not in new_plot.list_group_id:
817                                new_plot.group_id = group_id
818                                new_plot.list_group_id.append(group_id)
819                        else:
820                            if label.lower() in ["data", "chi2"]:
821                                if len(grid.selected_cells) != 1:
822                                    msg = "2D View: Please select one data set"
823                                    msg += " at a time for View Results."
824                                    wx.PostEvent(self.parent.parent, 
825                                                 StatusEvent(status=msg,
826                                                              info="error"))
827                                    time.sleep(0.5) 
828                                    continue 
829                        """
830                        wx.PostEvent(self.parent.parent,
831                                     NewPlotEvent(action="clear",
832                                                  group_id=str(group_id),
833                                                  title=title))
834                        """ 
835                        wx.PostEvent(self.parent.parent, 
836                                     NewPlotEvent(plot=new_plot, 
837                                                group_id=str(new_plot.group_id),
838                                                title=title)) 
839                        msg = "Plotting the View Results  completed!"
840                        wx.PostEvent( self.parent.parent, 
841                                      StatusEvent(status=msg)) 
842                else:
843                   
844                    msg = "Row %s , " % str(row)
845                    msg += "Column %s is NOT " % str(label)
846                    msg += "the results of fits to view..."
847                    #raise ValueError, msg
848                    wx.PostEvent(self.parent.parent, 
849                         StatusEvent(status=msg, info="error")) 
850                    time.sleep(0.5)
851                    continue
852   
853       
854     
855   
856    def on_plot(self, event):
857        """
858        Evaluate the contains of textcrtl and plot result
859        """ 
860        pos = self.notebook.GetSelection()
861        grid = self.notebook.GetPage(pos)
862        column_names = {}
863        if grid is not None:
864            column_names = self.notebook.get_column_labels()
865        #evaluate x
866        sentence = self.x_axis_label.GetValue()
867        try:
868            if sentence.strip() == "":
869                msg = "Select column values for x axis"
870                raise ValueError, msg
871        except:
872             wx.PostEvent(self.parent.parent, 
873                             StatusEvent(status=msg, info="error")) 
874             return
875
876        dict = parse_string(sentence, column_names.keys())
877        for tok, (col_name, list) in dict.iteritems():
878            col = column_names[col_name]
879            xaxis = self.get_plot_axis(col, list)
880            sentence = sentence.replace(tok, 
881                                        "numpy.array(%s)" % str(xaxis))
882        for key, value in FUNC_DICT.iteritems():
883            sentence = sentence.replace(key.lower(), value)
884        x = eval(sentence)
885        #evaluate y
886        sentence = self.y_axis_label.GetValue()
887        if sentence.strip() == "":
888            msg = "select value for y axis"
889            raise ValueError, msg
890        dict = parse_string(sentence, column_names.keys())
891        for tok, (col_name, list) in dict.iteritems():
892            col = column_names[col_name]
893            yaxis = self.get_plot_axis(col, list)
894            sentence = sentence.replace(tok, 
895                                        "numpy.array(%s)" % str(yaxis))
896        for key, value in FUNC_DICT.iteritems():
897            sentence = sentence.replace(key, value)
898        y = eval(sentence)
899        if len(x) != len(y) and (len(x) == 0 or len(y) == 0):
900            msg = "Need same length for X and Y axis and both greater than 0"
901            msg += " to plot.\n"
902            msg += "Got X length = %s, Y length = %s" % (str(len(x)),
903                                                          str(len(y)))
904            wx.PostEvent(self.parent.parent, 
905                             StatusEvent(status=msg, info="error")) 
906            return
907           
908        #plotting
909        new_plot = Data1D(x=x, y=y)
910        new_plot.id =  wx.NewId()
911        new_plot.group_id = wx.NewId()
912        title = "%s vs %s" % (self.y_axis_title.GetValue(), 
913                              self.x_axis_title.GetValue())
914        new_plot.xaxis(self.x_axis_title.GetValue(), 
915                       self.x_axis_unit.GetValue())
916        new_plot.yaxis(self.y_axis_title.GetValue(), 
917                       self.y_axis_unit.GetValue())
918        try:
919            title = self.notebook.GetPageText(pos)
920            new_plot.name = title
921            new_plot.xtransform = "x"
922            new_plot.ytransform  = "y" 
923            wx.PostEvent(self.parent.parent, 
924                        NewPlotEvent(plot=new_plot, 
925                        group_id=str(new_plot.group_id), title =title)) 
926            msg = "Plotting completed!"
927            wx.PostEvent( self.parent.parent, 
928                                      StatusEvent(status=msg))   
929        except:
930             wx.PostEvent(self.parent.parent, 
931                             StatusEvent(status=msg, info="error")) 
932
933    def layout_grid(self):
934        """
935        Draw the area related to the grid
936        """
937        self.notebook = Notebook(parent=self)
938        self.notebook.set_data(self._data_inputs, self._data_outputs)
939        self.grid_sizer.Add(self.notebook, 1, wx.EXPAND, 0)
940       
941    def layout_plotting_area(self):
942        """
943        Draw area containing options to plot
944        """
945       
946        self.x_axis_title = wx.TextCtrl(self, -1)
947        self.y_axis_title = wx.TextCtrl(self, -1)
948        self.x_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
949        self.y_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
950        self.x_axis_add = wx.Button(self, -1, "Add")
951        self.x_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis, 
952                            id=self.x_axis_add.GetId())
953        self.y_axis_add = wx.Button(self, -1, "Add")
954        self.y_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis, 
955                            id=self.y_axis_add.GetId())
956        self.x_axis_unit = wx.TextCtrl(self, -1)
957        self.y_axis_unit = wx.TextCtrl(self, -1)
958        self.view_button = wx.Button(self, -1, "View Results")
959        view_tip = "Highlight the data set or the Chi2 column first."
960        self.view_button.SetToolTipString(view_tip)
961        wx.EVT_BUTTON(self, self.view_button.GetId(), self.on_view)
962        self.plot_button = wx.Button(self, -1, "Plot")
963        plot_tip = "Highlight a column for each axis and \n"
964        plot_tip += "click the Add buttons first."
965        self.plot_button.SetToolTipString(plot_tip)
966        self.button_sizer.AddMany( [ (500, 30),
967                                (self.view_button, 0, wx.RIGHT|wx.BOTTOM, 10),
968                                (self.plot_button, 0, wx.RIGHT|wx.BOTTOM, 10)])
969       
970        wx.EVT_BUTTON(self, self.plot_button.GetId(), self.on_plot)
971        self.plotting_sizer.AddMany([
972                    (wx.StaticText(self, -1, 
973                                   "X-axis Label\nSelection Range"), 1,
974                      wx.TOP|wx.BOTTOM|wx.LEFT, 10),
975                    (self.x_axis_label, 1, wx.TOP|wx.BOTTOM, 10),
976                    (self.x_axis_add, 1, wx.TOP|wx.BOTTOM|wx.RIGHT, 10),
977                    (wx.StaticText(self, -1, "X-axis Label"), 1, 
978                     wx.TOP|wx.BOTTOM|wx.LEFT, 10),
979                    (self.x_axis_title, 1, wx.TOP|wx.BOTTOM, 10),
980                    (wx.StaticText(self, -1 , "X-axis Unit"), 1, 
981                     wx.TOP|wx.BOTTOM, 10),
982                    (self.x_axis_unit, 1, wx.TOP|wx.BOTTOM, 10),
983                    (wx.StaticText(self, -1, 
984                                   "Y-axis Label\nSelection Range"), 1, 
985                     wx.BOTTOM|wx.LEFT, 10),
986                    (self.y_axis_label, wx.BOTTOM, 10),
987                    (self.y_axis_add, 1, wx.BOTTOM|wx.RIGHT, 10),
988                    (wx.StaticText(self, -1, "Y-axis Label"), 1, 
989                     wx.BOTTOM|wx.LEFT, 10),
990                    (self.y_axis_title,  wx.BOTTOM, 10),
991                    (wx.StaticText(self, -1 , "Y-axis Unit"), 1, wx.BOTTOM, 10),
992                    (self.y_axis_unit, 1, wx.BOTTOM, 10),
993                      (-1, -1),
994                      (-1, -1),
995                      (-1, -1),
996                      (-1, -1),
997                      (-1, -1),
998                      (-1, -1),
999                      (-1, 1)])
1000   
1001    def on_edit_axis(self, event):
1002        """
1003        Get the selected column on  the visible grid and set values for axis
1004        """
1005        try:
1006            cell_list = self.notebook.on_edit_axis()
1007        except:
1008            msg = str(sys.exc_value)
1009            wx.PostEvent(self.parent.parent, 
1010                             StatusEvent(status=msg, info="error")) 
1011            return 
1012        label, title = self.create_axis_label(cell_list)
1013        tcrtl = event.GetEventObject()
1014        if tcrtl == self.x_axis_add:
1015            self.edit_axis_helper(self.x_axis_label, self.x_axis_title,
1016                                   label, title)
1017        elif tcrtl == self.y_axis_add:
1018            self.edit_axis_helper(self.y_axis_label, self.y_axis_title,
1019                                   label, title)
1020           
1021    def create_axis_label(self, cell_list):
1022        """
1023        Receive a list of cells and  create a string presenting the selected
1024        cells.
1025        :param cell_list: list of tuple
1026       
1027        """
1028        if self.notebook is not None:
1029            return self.notebook.create_axis_label(cell_list)
1030   
1031    def edit_axis_helper(self, tcrtl_label, tcrtl_title, label, title):
1032        """
1033        get controls to modify
1034        """
1035        tcrtl_label.SetValue(str(label))
1036        tcrtl_title.SetValue(str(title))
1037       
1038    def add_column(self):
1039        """
1040        """
1041        if self.notebook is not None:
1042            self.notebook.add_column()
1043       
1044    def on_remove_column(self):
1045        """
1046        """
1047        if self.notebook is not None:
1048            self.notebook.on_remove_column()
1049       
1050       
1051class GridFrame(wx.Frame):
1052    def __init__(self, parent=None, data_inputs=None, data_outputs=None, id=-1, 
1053                 title="Batch Window", size=(800, 500)):
1054        wx.Frame.__init__(self, parent=parent, id=id, title=title, size=size)
1055        self.parent = parent
1056        self.panel = GridPanel(self, data_inputs, data_outputs)
1057        menubar = wx.MenuBar()
1058        self.SetMenuBar(menubar)
1059       
1060        self.curr_col = None
1061        self.curr_grid = None
1062        self.curr_col_name = ""
1063        file = wx.Menu()
1064        menubar.Append(file, "&File")
1065       
1066        hint = "Open file containing batch results"
1067        open_menu = file.Append(wx.NewId(), 'Open ', hint)
1068        wx.EVT_MENU(self, open_menu.GetId(), self.on_open)
1069       
1070        hint = "Open the the current grid into excel"
1071        open_excel_menu = file.Append(wx.NewId(), 'Open with Excel', hint)
1072        wx.EVT_MENU(self, open_excel_menu.GetId(), self.open_with_excel)
1073        file.AppendSeparator()
1074        save_menu = file.Append(wx.NewId(), 'Save As', 'Save into File')
1075        wx.EVT_MENU(self, save_menu.GetId(), self.on_save_page)
1076       
1077        self.edit = wx.Menu()
1078        hint = "Insert column before the selected column"
1079        self.insert_before_menu = wx.Menu()
1080        self.insert_sub_menu = self.edit.AppendSubMenu(self.insert_before_menu, 
1081                                                      'Insert Before', hint)
1082 
1083        hint = "Remove the selected column"
1084        self.remove_menu = self.edit.Append(-1, 'Remove Column', hint)
1085        wx.EVT_MENU(self, self.remove_menu.GetId(), self.on_remove_column)
1086       
1087        self.Bind(wx.EVT_MENU_OPEN, self.on_menu_open)
1088        menubar.Append(self.edit, "&Edit")
1089        self.Bind(wx.EVT_CLOSE, self.on_close)
1090       
1091    def GetLabelText(self, id):
1092        """
1093        """
1094        for item in self.insert_before_menu.GetMenuItems():
1095            m_id = item.GetId() 
1096            if m_id == id:
1097                return item.GetLabel() 
1098   
1099    def on_remove_column(self, event):
1100        """
1101        """
1102        pos = self.panel.notebook.GetSelection()
1103        grid = self.panel.notebook.GetPage(pos)
1104        grid.on_remove_column(event=None)
1105       
1106    def on_menu_open(self, event):
1107        """
1108       
1109        """
1110        if self.edit == event.GetMenu():
1111            #get the selected column
1112            pos = self.panel.notebook.GetSelection()
1113            grid = self.panel.notebook.GetPage(pos)
1114            col_list = grid.GetSelectedCols()
1115            if len(col_list) > 0:
1116                self.remove_menu.Enable(True)
1117            else:
1118                self.remove_menu.Enable(False)
1119            if len(col_list)== 0 or len(col_list) > 1:
1120                self.insert_sub_menu.Enable(False)
1121               
1122                label = "Insert Column Before"
1123                self.insert_sub_menu.SetText(label)
1124            else:
1125                self.insert_sub_menu.Enable(True)
1126               
1127                col = col_list[0]
1128                #GetColLabelValue(self, col)
1129                col_name = grid.GetCellValue(row=0, col=col)
1130                label = "Insert Column Before " + str(col_name)
1131                self.insert_sub_menu.SetText(label)
1132                for item in self.insert_before_menu.GetMenuItems():
1133                    self.insert_before_menu.DeleteItem(item)
1134                grid.insert_col_menu(menu=self.insert_before_menu, 
1135                                     label=col_name, window=self)
1136        event.Skip()
1137       
1138 
1139       
1140    def on_save_page(self, event):
1141        """
1142        """
1143        if self.parent is not None:
1144            pos = self.panel.notebook.GetSelection()
1145            grid = self.panel.notebook.GetPage(pos)
1146            if grid.file_name is None or grid.file_name.strip() == "" or \
1147                grid.data is None or len(grid.data) == 0:
1148                name = self.panel.notebook.GetPageText(pos)
1149                msg = " %s has not data to save" % str(name)
1150                wx.PostEvent(self.parent, 
1151                             StatusEvent(status=msg, info="error")) 
1152           
1153                return
1154            reader, ext = os.path.splitext(grid.file_name)
1155            path = None
1156            if self.parent is not None: 
1157                location = os.path.dirname(grid.file_name)
1158                dlg = wx.FileDialog(self, "Save Project file",
1159                            location, grid.file_name, ext, wx.SAVE)
1160                path = None
1161                if dlg.ShowModal() == wx.ID_OK:
1162                    path = dlg.GetPath()
1163                dlg.Destroy()
1164                if path != None:
1165                    if self.parent is not None:
1166                        data = grid.get_grid_view()
1167                        self.parent.write_batch_tofile(data=data, 
1168                                               file_name=path,
1169                                               details=grid.details)
1170   
1171    def on_open(self, event):
1172        """
1173        Open file containg batch result
1174        """
1175        if self.parent is not None:
1176            self.parent.on_read_batch_tofile(event)
1177           
1178    def open_with_excel(self, event):
1179        """
1180        open excel and display batch result in Excel
1181        """
1182        if self.parent is not None:
1183            pos = self.panel.notebook.GetSelection()
1184            grid = self.panel.notebook.GetPage(pos)
1185            data = grid.get_grid_view()
1186            if grid.file_name is None or grid.file_name.strip() == "" or \
1187                grid.data is None or len(grid.data) == 0:
1188                name = self.panel.notebook.GetPageText(pos)
1189                msg = " %s has not data to open on excel" % str(name)
1190                wx.PostEvent(self.parent, 
1191                             StatusEvent(status=msg, info="error")) 
1192           
1193                return
1194            self.parent.open_with_externalapp(data=data,
1195                                               file_name=grid.file_name, 
1196                                               details=grid.details)
1197           
1198    def on_close(self, event):
1199        """
1200        """
1201        self.Hide()
1202       
1203   
1204    def on_append_column(self, event):
1205        """
1206        Append a new column to the grid
1207        """
1208        self.panel.add_column()
1209       
1210    def set_data(self, data_inputs, data_outputs, details="", file_name=None):
1211        """
1212        """
1213       
1214        self.panel.notebook.set_data(data_inputs=data_inputs, 
1215                            file_name=file_name,
1216                            details=details,
1217                            data_outputs=data_outputs)
1218     
1219     
1220class BatchOutputFrame(wx.Frame):
1221    """
1222    Allow to select where the result of batch will be displayed or stored
1223    """
1224    def __init__(self, parent, data_inputs, data_outputs, file_name="",
1225                 details="", *args, **kwds):
1226        """
1227        :param parent: Window instantiating this dialog
1228        :param result: result to display in a grid or export to an external
1229                application.
1230        """
1231        #kwds['style'] = wx.CAPTION|wx.SYSTEM_MENU
1232        wx.Frame.__init__(self, parent, *args, **kwds)
1233        self.parent = parent
1234        self.panel = wx.Panel(self)
1235        self.file_name = file_name
1236        self.details = details
1237        self.data_inputs = data_inputs
1238        self.data_outputs = data_outputs
1239        self.data = {}
1240        for item in (self.data_outputs, self.data_inputs):
1241            self.data.update(item)
1242        self.flag = 1
1243        self.SetSize((300, 200))
1244        self.local_app_selected = None
1245        self.external_app_selected = None
1246        self.save_to_file = None
1247        self._do_layout()
1248   
1249    def _do_layout(self):
1250        """
1251        Draw the content of the current dialog window
1252        """
1253        vbox = wx.BoxSizer(wx.VERTICAL)
1254        box_description = wx.StaticBox(self.panel, -1, str("Batch Outputs"))
1255        hint_sizer = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1256        selection_sizer = wx.GridBagSizer(5, 5)
1257        button_sizer = wx.BoxSizer(wx.HORIZONTAL)
1258        text = "Open with %s" % self.parent.application_name
1259        self.local_app_selected = wx.RadioButton(self.panel, -1, text,
1260                                                style=wx.RB_GROUP)
1261        self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
1262                    id=self.local_app_selected.GetId())
1263        text = "Open with Excel"
1264        self.external_app_selected  = wx.RadioButton(self.panel, -1, text)
1265        self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
1266                    id=self.external_app_selected.GetId())
1267        text = "Save to File"
1268        self.save_to_file = wx.CheckBox(self.panel, -1, text)
1269        self.Bind(wx.EVT_CHECKBOX, self.onselect,
1270                    id=self.save_to_file.GetId())
1271        self.local_app_selected.SetValue(True)
1272        self.external_app_selected.SetValue(False)
1273        self.save_to_file.SetValue(False)
1274        button_close = wx.Button(self.panel, -1, "Close")
1275        button_close.Bind(wx.EVT_BUTTON, id=button_close.GetId(),
1276                           handler=self.on_close)
1277        button_apply = wx.Button(self.panel, -1, "Apply")
1278        button_apply.Bind(wx.EVT_BUTTON, id=button_apply.GetId(),
1279                        handler=self.on_apply)
1280        button_apply.SetFocus()
1281        hint = ""
1282        hint_sizer.Add(wx.StaticText(self.panel, -1, hint))
1283        hint_sizer.Add(selection_sizer)
1284        #draw area containing radio buttons
1285        ix = 0
1286        iy = 0
1287        selection_sizer.Add(self.local_app_selected, (iy, ix),
1288                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1289        iy += 1
1290        selection_sizer.Add(self.external_app_selected, (iy, ix),
1291                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1292        iy += 1
1293        selection_sizer.Add(self.save_to_file, (iy, ix),
1294                           (1, 1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1295        #contruction the sizer contaning button
1296        button_sizer.Add((20, 20), 1, wx.EXPAND|wx.ADJUST_MINSIZE, 0)
1297
1298        button_sizer.Add(button_close, 0,
1299                        wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
1300        button_sizer.Add(button_apply, 0,
1301                                wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10)
1302        vbox.Add(hint_sizer,  0, wx.EXPAND|wx.ALL, 10)
1303        vbox.Add(wx.StaticLine(self.panel, -1),  0, wx.EXPAND, 0)
1304        vbox.Add(button_sizer, 0 , wx.TOP|wx.BOTTOM, 10)
1305        self.SetSizer(vbox)
1306       
1307    def on_apply(self, event):
1308        """
1309        Get the user selection and display output to the selected application
1310        """
1311        if self.flag == 1:
1312            self.parent.open_with_localapp(data_inputs=self.data_inputs,
1313                                            data_outputs=self.data_outputs)
1314        elif self.flag == 2:
1315            self.parent.open_with_externalapp(data=self.data, 
1316                                           file_name=self.file_name,
1317                                           details=self.details)
1318    def on_close(self, event):
1319        """
1320        close the Window
1321        """
1322        self.Close()
1323       
1324    def onselect(self, event=None):
1325        """
1326        Receive event and display data into third party application
1327        or save data to file.
1328       
1329        """
1330        if self.save_to_file.GetValue():
1331            reader, ext = os.path.splitext(self.file_name)
1332            path = None
1333            location = os.getcwd()
1334            if self.parent is not None: 
1335                location = os.path.dirname(self.file_name)
1336                dlg = wx.FileDialog(self, "Save Project file",
1337                            location, self.file_name, ext, wx.SAVE)
1338                path = None
1339                if dlg.ShowModal() == wx.ID_OK:
1340                    path = dlg.GetPath()
1341                dlg.Destroy()
1342                if path != None:
1343                    if self.parent is not None and  self.data is not None:
1344                        self.parent.write_batch_tofile(data=self.data, 
1345                                               file_name=path,
1346                                               details=self.details)
1347        if self.local_app_selected.GetValue():
1348            self.flag = 1
1349        else:
1350            self.flag = 2
1351        return self.flag
1352   
1353 
1354       
1355if __name__ == "__main__":
1356    app = wx.App()
1357   
1358    try:
1359        data = {}
1360        j = 0
1361        for i in range(4):
1362            j += 1
1363            data["index"+str(i)] = [i/j, i*j, i, i+j]
1364       
1365        data_input =  copy.deepcopy(data)   
1366        data_input["index5"] = [10,20,40, 50]
1367        frame = GridFrame(data_outputs=data, data_inputs=data_input)
1368        frame.Show(True)
1369    except:
1370        print sys.exc_value
1371       
1372    app.MainLoop()
Note: See TracBrowser for help on using the repository browser.