data_processor.py @ afb93df

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.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since afb93df was 468c253, checked in by butler, 9 years ago
Fixes image not loading in dev docs. Problem is all images are slurped into user docs folder (not really a choice — no easy way to identify it is a developer doc image. anyway this is the only one I think. just adjusted path in doc to reflect way it is slurped — not beautiful but works.
Property mode set to `100644`
File size: 77.8 KB

Line
1	"""
2	Implement grid used to store results of a batch fit.
3
4	This is in Guiframe rather than fitting which is probably where it should be.
5	Actually could be a generic framework implemented in fit gui module. At this
6	point however there this grid behaves independently of the fitting panel and
7	only knows about information sent to it but not about the fits or fit panel and
8	thus cannot feed back to the fitting panel. This could change in the future.
9
10	The organization of the classes goes as:
11
12	.. note:: Path to this is: /sasview/src/sas/sasgui/guiframe/data_processor.py
13
14	.. note:: Path to image is: /sasview/src/sas/sasgui/guiframe/media/BatchGridClassLayout.png
15
16	.. image:: ../../user/sasgui/guiframe/BatchGridClassLayout.png
17	:align: center
18
19	"""
20	import wx
21	import numpy
22	import math
23	import re
24	import os
25	import sys
26	import copy
27	from wx.lib.scrolledpanel import ScrolledPanel
28	import wx.aui
29	from wx.aui import AuiNotebook as nb
30	import wx.lib.sheet as sheet
31	from sas.sasgui.guiframe.panel_base import PanelBase
32	from sas.sasgui.guiframe.events import NewPlotEvent
33	from sas.sasgui.guiframe.events import StatusEvent
34	from sas.sasgui.plottools import plottables
35	from sas.sasgui.guiframe.dataFitting import Data1D
36
37
38	FUNC_DICT = {"sqrt": "math.sqrt",
39	"pow": "math.sqrt"}
40
41	class BatchCell(object):
42	"""
43	Object describing a cell in the grid.
44	"""
45	def __init__(self):
46	"""
47	Initialize attributes of class (label, value, col, row, object)
48	"""
49	self.label = ""
50	self.value = None
51	self.col = -1
52	self.row = -1
53	self.object = []
54
55
56	def parse_string(sentence, list):
57	"""
58	Return a dictionary of column label and index or row selected
59
60	:param sentence: String to parse
61	:param list: list of columns label
62	:returns: col_dict
63	"""
64
65	p2 = re.compile(r'\d+')
66	p = re.compile(r'[\+\-\*\%\/]')
67	labels = p.split(sentence)
68	col_dict = {}
69	for elt in labels:
70	rang = None
71	temp_arr = []
72	for label in list:
73	label_pos = elt.find(label)
74	separator_pos = label_pos + len(label)
75	if label_pos != -1 and len(elt) >= separator_pos and\
76	elt[separator_pos] == "[":
77	# the label contain , meaning the range selected is not
78	# continuous
79	if elt.count(',') > 0:
80	new_temp = []
81	temp = elt.split(label)
82	for item in temp:
83	range_pos = item.find(":")
84	if range_pos != -1:
85	rang = p2.findall(item)
86	for i in xrange(int(rang[0]), int(rang[1]) + 1):
87	new_temp.append(i)
88	temp_arr += new_temp
89	else:
90	# continuous range
91	temp = elt.split(label)
92	for item in temp:
93	if item.strip() != "":
94	range_pos = item.find(":")
95	if range_pos != -1:
96	rang = p2.findall(item)
97	for i in xrange(int(rang[0]), int(rang[1]) + 1):
98	temp_arr.append(i)
99	col_dict[elt] = (label, temp_arr)
100	return col_dict
101
102
103	class SPanel(ScrolledPanel):
104	"""
105	ensure proper scrolling of GridPanel
106
107	Adds a SetupScrolling call to the normal ScrolledPanel init.
108	GridPanel then subclasses this class
109
110	"""
111	def __init__(self, parent, args, *kwds):
112	"""
113	initialize ScrolledPanel then force a call to SetupScrolling
114
115	"""
116	ScrolledPanel.__init__(self, parent, args, *kwds)
117	self.SetupScrolling()
118
119
120	class GridCellEditor(sheet.CCellEditor):
121	"""
122	Custom cell editor
123
124	This subclasses the sheet.CCellEditor (itself a subclass of
125	grid.GridCellEditor) in order to override two of its methods:
126	PaintBackrgound and EndEdit.
127
128	This is necessary as the sheet module is broken in wx 3.0.2 and
129	improperly subclasses grid.GridCellEditor
130	"""
131	def __init__(self, grid):
132	"""
133	Override of CCellEditor init. Runs the grid.GridCellEditor init code
134	"""
135	super(GridCellEditor, self).__init__(grid)
136
137	def PaintBackground(self, dc, rect, attr):
138	"""
139	Overrides wx.sheet.CCellEditor.PaintBackground which incorrectly calls
140	the base class method.
141
142	In wx3.0 all paint objects must explicitly
143	have a wxPaintDC (Device Context) object. Thus the paint event which
144	generates a call to this method provides such a DC object and the
145	base class in grid expects to receive that object. sheet was apparently
146	not updated to reflect this and hence fails. This could thus
147	become obsolete in a future bug fix of wxPython.
148
149	Apart from adding a dc variable in the list of arguments in the def
150	and in the call to the base class the rest of this method is copied
151	as is from sheet.CCellEditor.PaintBackground
152
153	From original GridCellEditor docs:
154
155	Draws the part of the cell not occupied by the edit control. The
156	base class version just fills it with background colour from the
157	attribute.
158
159	.. note:: There is no need to override this if you don't need
160	to do something out of the ordinary.
161
162	:param dc: the wxDC object for the paint
163	"""
164	# Call base class method.
165	DC = dc
166	super(sheet.CCellEditor,self).PaintBackground(DC, rect, attr)
167
168	def EndEdit(self, row, col, grid, previous):
169	"""
170	Commit editing the current cell. Returns True if the value has changed.
171
172	:param previous: previous value in the cell
173	"""
174	changed = False # Assume value not changed
175	val = self._tc.GetValue() # Get value in edit control
176	if val != self._startValue: # Compare
177	changed = True # If different then changed is True
178	grid.GetTable().SetValue(row, col, val) # Update the table
179	self._startValue = '' # Clear the class' start value
180	self._tc.SetValue('') # Clear contents of the edit control
181	return changed
182
183
184	class GridPage(sheet.CSheet):
185	"""
186	Class that receives the results of a batch fit.
187
188	GridPage displays the received results in a wx.grid using sheet. This is
189	then used by GridPanel and GridFrame to present the full GUI.
190	"""
191	def __init__(self, parent, panel=None):
192	"""
193	Initialize
194
195	Initialize all the attributes of GridPage, and the events. include
196	the init stuff from sheet.CSheet as well.
197	"""
198	#sheet.CSheet.__init__(self, parent)
199
200	# The following is the __init__ from CSheet. ##########################
201	# We re-write it here because the class is broken in wx 3.0,
202	# such that the cell editor is not able to receive the right
203	# number of parameters when it is called. The only way to
204	# pick a different cell editor is apparently to re-write the __init__.
205	wx.grid.Grid.__init__(self, parent, -1)
206
207	# Init variables
208	self._lastCol = -1 # Init last cell column clicked
209	self._lastRow = -1 # Init last cell row clicked
210	self._selected = None # Init range currently selected
211	# Map string datatype to default renderer/editor
212	self.RegisterDataType(wx.grid.GRID_VALUE_STRING,
213	wx.grid.GridCellStringRenderer(),
214	GridCellEditor(self))
215
216	self.CreateGrid(4, 3) # By default start with a 4 x 3 grid
217	self.SetColLabelSize(18) # Default sizes and alignment
218	self.SetRowLabelSize(50)
219	self.SetRowLabelAlignment(wx.ALIGN_RIGHT, wx.ALIGN_BOTTOM)
220	self.SetColSize(0, 75) # Default column sizes
221	self.SetColSize(1, 75)
222	self.SetColSize(2, 75)
223
224	# Sink events
225	self.Bind(wx.grid.EVT_GRID_RANGE_SELECT, self.OnRangeSelect)
226	self.Bind(wx.grid.EVT_GRID_ROW_SIZE, self.OnRowSize)
227	self.Bind(wx.grid.EVT_GRID_COL_SIZE, self.OnColSize)
228	self.Bind(wx.grid.EVT_GRID_SELECT_CELL, self.OnGridSelectCell)
229	# NOTE: the following bind to standard sheet methods that are
230	# overriden in this subclassn - actually we have currently
231	# disabled the on_context_menu that would override the OnRightClick
232	self.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnCellChange)
233	self.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnLeftClick)
234	self.Bind(wx.grid.EVT_GRID_CELL_RIGHT_CLICK, self.OnRightClick)
235	#self.Bind(wx.grid.EVT_GRID_CELL_LEFT_DCLICK, self.OnLeftDoubleClick)
236	# This ends the __init__ section for CSheet. ##########################
237
238
239
240	# The following events must be bound even if CSheet is working
241	# properly and does not need the above re-implementation of the
242	# CSheet init method. Basically these override any intrinsic binding
243	self.Bind(wx.grid.EVT_GRID_LABEL_RIGHT_CLICK, self.on_right_click)
244	self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_CLICK, self.on_left_click)
245
246	self.AdjustScrollbars()
247	#self.SetLabelBackgroundColour('#DBD4D4')
248	self.uid = wx.NewId()
249	self.parent = parent
250	self.panel = panel
251	self.col_names = []
252	self.data_inputs = {}
253	self.data_outputs = {}
254	self.data = None
255	self.details = ""
256	self.file_name = None
257	self._cols = 50
258	self._rows = 3001
259	self.last_selected_row = -1
260	self.last_selected_col = -1
261	self.col_width = 30
262	self.row_height = 20
263	self.max_row_touse = 0
264	self.axis_value = []
265	self.axis_label = ""
266	self.selected_cells = []
267	self.selected_cols = []
268	self.selected_rows = []
269	self.plottable_cells = []
270	self.plottable_flag = False
271	self.SetColMinimalAcceptableWidth(self.col_width)
272	self.SetRowMinimalAcceptableHeight(self.row_height)
273	self.SetNumberRows(self._rows)
274	self.SetNumberCols(self._cols)
275	color = self.parent.GetBackgroundColour()
276	for col in range(self._cols):
277	self.SetCellBackgroundColour(0, col, color)
278	self.AutoSize()
279	self.list_plot_panels = {}
280	self.default_col_width = 75
281	self.EnableEditing(True)
282	if self.GetNumberCols() > 0:
283	self.default_col_width = self.GetColSize(0)
284	# We have moved these to the top of the init section with the
285	# rest of the grid event bindings from the sheet init when
286	# appropriate
287	#self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_CLICK, self.on_left_click)
288	#self.Bind(wx.grid.EVT_GRID_LABEL_RIGHT_CLICK, self.on_right_click)
289	#self.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.on_selected_cell)
290	#self.Bind(wx.grid.EVT_GRID_CMD_CELL_CHANGE, self.on_edit_cell)
291	#self.Bind(wx.grid.EVT_GRID_CELL_RIGHT_CLICK, self.onContextMenu)
292
293	def OnLeftClick(self, event):
294	"""
295	Overrides sheet.CSheet.OnLefClick.
296
297	Processes when a cell is selected by left clicking on that cell. First
298	process the base Sheet method then the current class specific method
299	"""
300	sheet.CSheet.OnLeftClick(self, event)
301	self.on_selected_cell(event)
302
303
304	def OnCellChange(self, event):
305	"""
306	Overrides sheet.CSheet.OnCellChange.
307
308	Processes when a cell has been edited by a cell editor. Checks for the
309	edited row being outside the max row to use attribute and if so updates
310	the last row. Then calls the base handler using skip.
311	"""
312	row, _ = event.GetRow(), event.GetCol()
313	if row > self.max_row_touse:
314	self.max_row_touse = row
315	if self.data == None:
316	self.data = {}
317	event.Skip()
318
319	def on_selected_cell(self, event):
320	"""
321	Handler catching cell selection.
322
323	Called after calling base 'on left click' method.
324	"""
325
326	flag = event.CmdDown() or event.ControlDown()
327	flag_shift = event.ShiftDown()
328	row, col = event.GetRow(), event.GetCol()
329	cell = (row, col)
330	if not flag and not flag_shift:
331	self.selected_cols = []
332	self.selected_rows = []
333	self.selected_cells = []
334	self.axis_label = ""
335	self.axis_value = []
336	self.plottable_list = []
337	self.plottable_cells = []
338	self.plottable_flag = False
339	self.last_selected_col = col
340	self.last_selected_row = row
341	if col >= 0:
342	if flag:
343	label_row = row
344	else:
345	label_row = 0
346	self.axis_label = self.GetCellValue(label_row, col)
347	self.selected_cols.append(col)
348	if flag_shift:
349	if not self.selected_rows:
350	min_r = 1
351	else:
352	min_r = min(self.selected_rows)
353	for row_s in range(min_r, row + 1):
354	cel = (row_s, col)
355	if cel not in self.selected_cells:
356	if row > 0:
357	self.selected_cells.append(cel)
358	self.selected_rows.append(row)
359	for row_s in self.selected_rows:
360	cel = (row_s, col)
361	if row_s > row:
362	try:
363	self.selected_cells.remove(cel)
364	except:
365	pass
366	try:
367	self.selected_rows.remove(row_s)
368	except:
369	pass
370	elif flag:
371	if cell not in self.selected_cells:
372	if row > 0:
373	self.selected_cells.append(cell)
374	self.selected_rows.append(row)
375	else:
376	try:
377	self.selected_cells.remove(cell)
378	except:
379	pass
380	try:
381	self.selected_rows.remove(row)
382	except:
383	pass
384	else:
385	self.selected_cells.append(cell)
386	self.selected_rows.append(row)
387	self.axis_value = []
388	for cell_row, cell_col in self.selected_cells:
389	if cell_row > 0 and cell_row < self.max_row_touse:
390	self.axis_value.append(self.GetCellValue(cell_row, cell_col))
391	event.Skip()
392
393	def on_left_click(self, event):
394	"""
395	Is triggered when the left mouse button is clicked while the mouse
396	is hovering over the column 'label.'
397
398	This processes the information on the selected column: the column name
399	(in row 0 of column) and the range of cells with a valid value to be
400	used by the GridPanel set_axis methods.
401	"""
402
403	flag = event.CmdDown() or event.ControlDown()
404
405	col = event.GetCol()
406	row = event.GetRow()
407
408	if not flag:
409	self.selected_cols = []
410	self.selected_rows = []
411	self.selected_cells = []
412	self.axis_label = ""
413	self.axis_value = []
414	self.plottable_list = []
415	self.plottable_cells = []
416	self.plottable_flag = False
417
418	self.last_selected_col = col
419	self.last_selected_row = row
420	if row != -1 and row not in self.selected_rows:
421	self.selected_rows.append(row)
422
423	if col != -1:
424	for row in range(1, self.GetNumberRows() + 1):
425	cell = (row, col)
426	if row > 0 and row < self.max_row_touse:
427	if cell not in self.selected_cells:
428	self.selected_cells.append(cell)
429	else:
430	if flag:
431	self.selected_cells.remove(cell)
432	self.selected_cols.append(col)
433	self.axis_value = []
434	for cell_row, cell_col in self.selected_cells:
435	val = self.GetCellValue(cell_row, cell_col)
436	if not val:
437	self.axis_value.append(self.GetCellValue(cell_row, cell_col))
438	self.axis_label = self.GetCellValue(0, col)
439	if not self.axis_label:
440	self.axis_label = " "
441	event.Skip()
442
443	def on_right_click(self, event):
444	"""
445	Is triggered when the right mouse button is clicked while the mouse
446	is hovering over the column 'label.'
447
448	This brings up a context menu that allows the deletion of the column,
449	or the insertion of a new column either to the right or left of the
450	current column. If inserting a new column can insert a blank column or
451	choose a number of hidden columns. By default all the error parameters
452	are in hidden columns so as to save space on the grid. Also any other
453	intrinsic variables stored with the data such as Temperature, pressure,
454	time etc can be used to populate this menu.
455	"""
456
457	col = event.GetCol()
458	row = event.GetRow()
459	# Ignore the index column
460	if col < 0 or row != -1:
461	return
462	self.selected_cols = []
463	self.selected_cols.append(col)
464	# Slicer plot popup menu
465	slicerpop = wx.Menu()
466	col_label_menu = wx.Menu()
467	c_name = self.GetCellValue(0, col)
468	label = "Insert column before %s " % str(c_name)
469	slicerpop.AppendSubMenu(col_label_menu, '&%s' % str(label), str(label))
470	row = 0
471	label = self.GetCellValue(row, col)
472	self.insert_col_menu(col_label_menu, label, self)
473
474	col_after_menu = wx.Menu()
475	label = "Insert column after %s " % str(c_name)
476	slicerpop.AppendSubMenu(col_after_menu, '&%s' % str(label), str(label))
477	self.insert_after_col_menu(col_after_menu, label, self)
478
479	wx_id = wx.NewId()
480	hint = 'Remove selected column %s'
481	slicerpop.Append(wx_id, '&Remove Column', hint)
482	wx.EVT_MENU(self, wx_id, self.on_remove_column)
483
484	pos = wx.GetMousePosition()
485	pos = self.ScreenToClient(pos)
486	self.PopupMenu(slicerpop, pos)
487	event.Skip()
488
489	def insert_col_menu(self, menu, label, window):
490	"""
491	method called to populate the 'insert column before current column'
492	submenu.
493	"""
494
495	if self.data is None:
496	return
497	id = wx.NewId()
498	title = "Empty"
499	hint = 'Insert empty column before %s' % str(label)
500	menu.Append(id, title, hint)
501	wx.EVT_MENU(window, id, self.on_insert_column)
502	row = 0
503	col_name = [self.GetCellValue(row, col) for col in range(self.GetNumberCols())]
504	for c_name in self.data.keys():
505	if c_name not in col_name and self.data[c_name]:
506	wx_id = wx.NewId()
507	hint = "Insert %s column before the " % str(c_name)
508	hint += " %s column" % str(label)
509	menu.Append(wx_id, '&%s' % str(c_name), hint)
510	wx.EVT_MENU(window, wx_id, self.on_insert_column)
511
512	def insert_after_col_menu(self, menu, label, window):
513	"""
514	Method called to populate the 'insert column after current column'
515	submenu
516	"""
517
518	if self.data is None:
519	return
520	wx_id = wx.NewId()
521	title = "Empty"
522	hint = 'Insert empty column after %s' % str(label)
523	menu.Append(wx_id, title, hint)
524	wx.EVT_MENU(window, wx_id, self.on_insert_after_column)
525	row = 0
526	col_name = [self.GetCellValue(row, col)
527	for col in range(self.GetNumberCols())]
528	for c_name in self.data.keys():
529	if c_name not in col_name and self.data[c_name]:
530	wx_id = wx.NewId()
531	hint = "Insert %s column after the " % str(c_name)
532	hint += " %s column" % str(label)
533	menu.Append(wx_id, '&%s' % str(c_name), hint)
534	wx.EVT_MENU(window, wx_id, self.on_insert_after_column)
535
536	def on_remove_column(self, event=None):
537	"""
538	Called when user chooses remove from the column right click menu
539	Checks the columnn exists then calls the remove_column method
540	"""
541
542	if self.selected_cols is not None or len(self.selected_cols) > 0:
543	col = self.selected_cols[0]
544	self.remove_column(col=col, numCols=1)
545
546	def remove_column(self, col, numCols=1):
547	"""
548	Remove the col column from the current grid
549	"""
550
551	# add data to the grid
552	row = 0
553	col_name = self.GetCellValue(row, col)
554	self.data[col_name] = []
555	for row in range(1, self.GetNumberRows() + 1):
556	if row < self.max_row_touse:
557	value = self.GetCellValue(row, col)
558	self.data[col_name].append(value)
559	for k, value_list in self.data.iteritems():
560	if k != col_name:
561	length = len(value_list)
562	if length < self.max_row_touse:
563	diff = self.max_row_touse - length
564	for i in range(diff):
565	self.data[k].append("")
566	self.DeleteCols(pos=col, numCols=numCols, updateLabels=True)
567
568	def on_insert_column(self, event):
569	"""
570	Called when user chooses insert 'column before' submenu
571	of the column context menu obtained when right clicking on a given
572	column header.
573
574	Sets up to insert column into the current grid before the current
575	highlighted column location and sets up what to populate that column
576	with. Then calls insert_column method to actually do the insertion.
577	"""
578
579	if self.selected_cols is not None or len(self.selected_cols) > 0:
580	col = self.selected_cols[0]
581	# add data to the grid
582	wx_id = event.GetId()
583	col_name = event.GetEventObject().GetLabelText(wx_id)
584	self.insert_column(col=col, col_name=col_name)
585	if not issubclass(event.GetEventObject().__class__, wx.Menu):
586	col += 1
587	self.selected_cols[0] += 1
588
589	def on_insert_after_column(self, event):
590	"""
591	Called when user chooses insert 'column after' submenu
592	of the column context menu obtained when right clicking on a given
593	column header.
594
595	Sets up to insert column into the current grid after the current
596	highlighted column location and sets up what to populate that column
597	with. Then calls insert_column method to actually do the insertion.
598	"""
599
600	if self.selected_cols is not None or len(self.selected_cols) > 0:
601	col = self.selected_cols[0] + 1
602	# add data to the grid
603	wx_id = event.GetId()
604	col_name = event.GetEventObject().GetLabelText(wx_id)
605	self.insert_column(col=col, col_name=col_name)
606	if not issubclass(event.GetEventObject().__class__, wx.Menu):
607	self.selected_cols[0] += 1
608
609	def insert_column(self, col, col_name):
610	"""
611	Insert column at position col with data[col_name] into the current
612	grid.
613	"""
614
615	row = 0
616	self.InsertCols(pos=col, numCols=1, updateLabels=True)
617	if col_name.strip() != "Empty":
618	self.SetCellValue(row, col, str(col_name.strip()))
619	if col_name in self.data.keys():
620	value_list = self.data[col_name]
621	cell_row = 1
622	for value in value_list:
623	label = value#format_number(value, high=True)
624	self.SetCellValue(cell_row, col, str(label))
625	cell_row += 1
626	self.AutoSizeColumn(col, True)
627	width = self.GetColSize(col)
628	if width < self.default_col_width:
629	self.SetColSize(col, self.default_col_width)
630	color = self.parent.GetBackgroundColour()
631	self.SetCellBackgroundColour(0, col, color)
632	self.ForceRefresh()
633
634	def on_set_x_axis(self, event):
635	"""
636	Just calls the panel version of the method
637	"""
638
639	self.panel.set_xaxis(x=self.axis_value, label=self.axis_label)
640
641	def on_set_y_axis(self, event):
642	"""
643	Just calls the panel version of the method
644	"""
645
646	self.panel.set_yaxis(y=self.axis_value, label=self.axis_label)
647
648	def set_data(self, data_inputs, data_outputs, details, file_name):
649	"""
650	Add data to the grid
651
652	:param data_inputs: data to use from the context menu of the grid
653	:param data_ouputs: default columns displayed
654	"""
655
656	self.file_name = file_name
657	self.details = details
658
659	if data_outputs is None:
660	data_outputs = {}
661	self.data_outputs = data_outputs
662	if data_inputs is None:
663	data_inputs = {}
664	self.data_inputs = data_inputs
665	self.data = {}
666	for item in (self.data_outputs, self.data_inputs):
667	self.data.update(item)
668
669	if len(self.data_outputs) > 0:
670	self._cols = self.GetNumberCols()
671	self._rows = self.GetNumberRows()
672	self.col_names = self.data_outputs.keys()
673	self.col_names.sort()
674	nbr_user_cols = len(self.col_names)
675	#Add more columns to the grid if necessary
676	if nbr_user_cols > self._cols:
677	new_col_nbr = nbr_user_cols - self._cols + 1
678	self.AppendCols(new_col_nbr, True)
679	#Add more rows to the grid if necessary
680	nbr_user_row = len(self.data_outputs.values()[0])
681	if nbr_user_row > self._rows + 1:
682	new_row_nbr = nbr_user_row - self._rows + 1
683	self.AppendRows(new_row_nbr, True)
684	# add data to the grid
685	wx.CallAfter(self.set_grid_values)
686	self.ForceRefresh()
687
688	def set_grid_values(self):
689	"""
690	Set the values in grids
691	"""
692
693	# add data to the grid
694	row = 0
695	col = 0
696	cell_col = 0
697	for col_name in self.col_names:
698	# use the first row of the grid to add user defined labels
699	self.SetCellValue(row, col, str(col_name))
700	col += 1
701	cell_row = 1
702	value_list = self.data_outputs[col_name]
703
704	for value in value_list:
705	label = value
706	if issubclass(value.__class__, BatchCell):
707	label = value.label
708	try:
709	float(label)
710	label = str(label)#format_number(label, high=True)
711	except:
712	label = str(label)
713	self.SetCellValue(cell_row, cell_col, label)
714	self.AutoSizeColumn(cell_col, True)
715	width = self.GetColSize(cell_col)
716	if width < self.default_col_width:
717	self.SetColSize(cell_col, self.default_col_width)
718
719	cell_row += 1
720	cell_col += 1
721	if cell_row > self.max_row_touse:
722	self.max_row_touse = cell_row
723
724	def get_grid_view(self):
725	"""
726	Return value contained in the grid
727	"""
728
729	grid_view = {}
730	for col in xrange(self.GetNumberCols()):
731	label = self.GetCellValue(row=0, col=col)
732	label = label.strip()
733	if label != "":
734	grid_view[label] = []
735	for row in range(1, self.max_row_touse):
736	value = self.GetCellValue(row=row, col=col)
737	if value != "":
738	grid_view[label].append(value)
739	else:
740	grid_view[label].append(None)
741	return grid_view
742
743	def get_nofrows(self):
744	"""
745	Return number of total rows
746	"""
747	return self._rows
748
749	def onContextMenu(self, event):
750	"""
751	Method to handle cell right click context menu.
752
753	THIS METHOD IS NOT CURRENTLY USED. It is designed to provide a
754	cell pop up context by right clicking on a cell and gives the
755	option to cut, paste, and clear. This will probably be removed in
756	future versions and is being superceded by more traditional cut and
757	paste options.
758	"""
759
760	wx_id = wx.NewId()
761	c_menu = wx.Menu()
762	copy_menu = c_menu.Append(wx_id, '&Copy', 'Copy the selected cells')
763	wx.EVT_MENU(self, wx_id, self.on_copy)
764
765	wx_id = wx.NewId()
766	c_menu.Append(wx_id, '&Paste', 'Paste the selected cells')
767	wx.EVT_MENU(self, wx_id, self.on_paste)
768
769	wx_id = wx.NewId()
770	clear_menu = c_menu.Append(wx_id, '&Clear', 'Clear the selected cells')
771	wx.EVT_MENU(self, wx_id, self.on_clear)
772
773	# enable from flag
774	has_selection = False
775	selected_cel = self.selected_cells
776	if len(selected_cel) > 0:
777	_row, _col = selected_cel[0]
778	has_selection = self.IsInSelection(_row, _col)
779	if len(self.selected_cols) > 0:
780	has_selection = True
781	if len(self.selected_rows) > 0:
782	has_selection = True
783	copy_menu.Enable(has_selection)
784	clear_menu.Enable(has_selection)
785	try:
786	# mouse event pos
787	pos_evt = event.GetPosition()
788	self.PopupMenu(c_menu, pos_evt)
789	except:
790	return
791
792	def on_copy(self, event):
793	"""
794	Called when copy is chosen from cell right click context menu
795
796	THIS METHOD IS NOT CURRENTLY USED. it is part of right click cell
797	context menu which is being removed. This will probably be removed in
798	future versions and is being superceded by more traditional cut and
799	paste options
800	"""
801
802	self.Copy()
803
804	def on_paste(self, event):
805	"""
806	Called when paste is chosen from cell right click context menu
807
808	THIS METHOD IS NOT CURRENTLY USED. it is part of right click cell
809	context menu which is being removed. This will probably be removed in
810	future versions and is being superceded by more traditional cut and
811	paste options
812	"""
813
814	if self.data == None:
815	self.data = {}
816	if self.file_name == None:
817	self.file_name = 'copied_data'
818	self.Paste()
819
820	def on_clear(self, event):
821	"""
822	Called when clear cell is chosen from cell right click context menu
823
824	THIS METHOD IS NOT CURRENTLY USED. it is part of right click cell
825	context menu which is being removed. This will probably be removed in
826	future versions and is being superceded by more traditional cut and
827	paste options
828	"""
829
830	self.Clear()
831
832	class Notebook(nb, PanelBase):
833	"""
834	## Internal name for the AUI manager
835	window_name = "Fit panel"
836	## Title to appear on top of the window
837	"""
838
839	window_caption = "Notebook "
840
841	def __init__(self, parent, manager=None, data=None, args, *kwargs):
842	"""
843	"""
844	nb.__init__(self, parent, -1,
845	style=wx.aui.AUI_NB_WINDOWLIST_BUTTON \|
846	wx.aui.AUI_BUTTON_DOWN \|
847	wx.aui.AUI_NB_DEFAULT_STYLE \|
848	wx.CLIP_CHILDREN)
849	PanelBase.__init__(self, parent)
850	self.gpage_num = 1
851	self.enable_close_button()
852	self.parent = parent
853	self.manager = manager
854	self.data = data
855	#add empty page
856	self.add_empty_page()
857	self.pageClosedEvent = wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE
858	self.Bind(self.pageClosedEvent, self.on_close_page)
859
860	def add_empty_page(self):
861	"""
862	"""
863	grid = GridPage(self, panel=self.parent)
864	self.AddPage(grid, "", True)
865	pos = self.GetPageIndex(grid)
866	title = "Table" + str(self.gpage_num)
867	self.SetPageText(pos, title)
868	self.SetSelection(pos)
869	self.enable_close_button()
870	self.gpage_num += 1
871	return grid, pos
872
873	def enable_close_button(self):
874	"""
875	display the close button on the tab if more than 1 tab exits.
876	Otherwise remove the close button
877	"""
878
879	if self.GetPageCount() <= 1:
880	style = self.GetWindowStyleFlag()
881	flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
882	if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB == flag:
883	style = style & ~wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
884	self.SetWindowStyle(style)
885	else:
886	style = self.GetWindowStyleFlag()
887	flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
888	if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB != flag:
889	style \|= wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
890	self.SetWindowStyle(style)
891
892	def on_edit_axis(self):
893	"""
894	Return the select cell range from a given selected column. Checks that
895	all cells are from the same column
896	"""
897
898	pos = self.GetSelection()
899	grid = self.GetPage(pos)
900	#grid.selected_cols = [grid.GetSelectedRows()]#
901	if len(grid.selected_cols) >= 1:
902	col = grid.selected_cols[0]
903	for c in grid.selected_cols:
904	if c != col:
905	msg = "Edit axis doesn't understand this selection.\n"
906	msg += "Please select only one column"
907	raise ValueError, msg
908	for (_, cell_col) in grid.selected_cells:
909	if cell_col != col:
910	msg = "Cannot use cells from different columns for "
911	msg += "this operation.\n"
912	msg += "Please select elements of the same col.\n"
913	raise ValueError, msg
914
915	# Finally check the highlighted cell if any cells missing
916	self.get_highlighted_row(True)
917	else:
918	msg = "No item selected.\n"
919	msg += "Please select only one column or one cell"
920	raise ValueError, msg
921	return grid.selected_cells
922
923	def get_highlighted_row(self, is_number=True):
924	"""
925	Add highlight rows
926	"""
927
928	pos = self.GetSelection()
929	grid = self.GetPage(pos)
930	col = grid.selected_cols[0]
931	# Finally check the highlighted cell if any cells missing
932	for row in range(grid.get_nofrows()):
933	if grid.IsInSelection(row, col):
934	cel = (row, col)
935	if row < 1 and not is_number:
936	continue
937	# empty cell
938	if not grid.GetCellValue(row, col).lstrip().rstrip():
939	if cel in grid.selected_cells:
940	grid.selected_cells.remove(cel)
941	continue
942	if is_number:
943	try:
944	float(grid.GetCellValue(row, col))
945	except:
946	# non numeric cell
947	if cel in grid.selected_cells:
948	grid.selected_cells.remove(cel)
949	continue
950	if cel not in grid.selected_cells:
951	grid.selected_cells.append(cel)
952
953	def get_column_labels(self):
954	"""
955	return dictionary of columns labels on the current page
956	"""
957
958	pos = self.GetSelection()
959	grid = self.GetPage(pos)
960	labels = {}
961	for col in range(grid.GetNumberCols()):
962	label = grid.GetColLabelValue(int(col))
963	if label.strip() != "":
964	labels[label.strip()] = col
965	return labels
966
967	def create_axis_label(self, cell_list):
968	"""
969	Receive a list of cells and create a string presenting the selected
970	cells that can be used as data for one axis of a plot.
971
972	:param cell_list: list of tuple
973	"""
974	pos = self.GetSelection()
975	grid = self.GetPage(pos)
976	label = ""
977	col_name = ""
978	def create_label(col_name, row_min=None, row_max=None):
979	"""
980	"""
981	result = " "
982	if row_min is not None or row_max is not None:
983	if row_min is None:
984	result = str(row_max) + "]"
985	elif row_max is None:
986	result = str(col_name) + "[" + str(row_min) + ":"
987	else:
988	result = str(col_name) + "[" + str(row_min) + ":"
989	result += str(row_max) + "]"
990	return str(result)
991
992	if len(cell_list) > 0:
993	if len(cell_list) == 1:
994	row_min, col = cell_list[0]
995	col_name = grid.GetColLabelValue(int(col))
996
997	col_title = grid.GetCellValue(0, col)
998	label = create_label(col_name, row_min + 1, row_min + 1)
999	return label, col_title
1000	else:
1001	temp_list = copy.deepcopy(cell_list)
1002	temp_list.sort()
1003	length = len(temp_list)
1004	row_min, col = temp_list[0]
1005	row_max, _ = temp_list[length - 1]
1006	col_name = grid.GetColLabelValue(int(col))
1007	col_title = grid.GetCellValue(0, col)
1008
1009	index = 0
1010	for row in xrange(row_min, row_max + 1):
1011	if index > 0 and index < len(temp_list):
1012	new_row, _ = temp_list[index]
1013	if row != new_row:
1014	temp_list.insert(index, (None, None))
1015	if index - 1 >= 0:
1016	new_row, _ = temp_list[index - 1]
1017	if not new_row == None and new_row != ' ':
1018	label += create_label(col_name, None,
1019	int(new_row) + 1)
1020	else:
1021	label += "]"
1022	label += ","
1023	if index + 1 < len(temp_list):
1024	new_row, _ = temp_list[index + 1]
1025	if not new_row == None:
1026	label += create_label(col_name,
1027	int(new_row) + 1, None)
1028	if row_min != None and row_max != None:
1029	if index == 0:
1030	label += create_label(col_name,
1031	int(row_min) + 1, None)
1032	elif index == len(temp_list) - 1:
1033	label += create_label(col_name, None,
1034	int(row_max) + 1)
1035	index += 1
1036	# clean up the list
1037	label_out = ''
1038	for item in label.split(','):
1039	if item.split(":")[1] == "]":
1040	continue
1041	else:
1042	label_out += item + ","
1043
1044	return label_out, col_title
1045
1046	def on_close_page(self, event):
1047	"""
1048	close the page
1049	"""
1050
1051	if self.GetPageCount() == 1:
1052	event.Veto()
1053	wx.CallAfter(self.enable_close_button)
1054
1055	def set_data(self, data_inputs, data_outputs, details="", file_name=None):
1056	"""
1057	"""
1058	if data_outputs is None or data_outputs == {}:
1059	return
1060	inputs, outputs = self.get_odered_results(data_inputs, data_outputs)
1061	for pos in range(self.GetPageCount()):
1062	grid = self.GetPage(pos)
1063	if grid.data is None:
1064	#Found empty page
1065	grid.set_data(data_inputs=inputs,
1066	data_outputs=outputs,
1067	details=details,
1068	file_name=file_name)
1069	self.SetSelection(pos)
1070	return
1071
1072	grid, pos = self.add_empty_page()
1073	grid.set_data(data_inputs=inputs,
1074	data_outputs=outputs,
1075	file_name=file_name,
1076	details=details)
1077
1078	def get_odered_results(self, inputs, outputs=None):
1079	"""
1080	Order a list of 'inputs.' Used to sort rows and columns to present
1081	in batch results grid.
1082	"""
1083
1084	# Let's re-order the data from the keys in 'Data' name.
1085	if outputs == None:
1086	return
1087	try:
1088	# For outputs from batch
1089	to_be_sort = [str(item.label) for item in outputs['Data']]
1090	except:
1091	# When inputs are from an external file
1092	return inputs, outputs
1093	inds = numpy.lexsort((to_be_sort, to_be_sort))
1094	for key in outputs.keys():
1095	key_list = outputs[key]
1096	temp_key = [item for item in key_list]
1097	for ind in inds:
1098	temp_key[ind] = key_list[inds[ind]]
1099	outputs[key] = temp_key
1100	for key in inputs.keys():
1101	key_list = inputs[key]
1102	if len(key_list) == len(inds):
1103	temp_key = [item for item in key_list]
1104	for ind in inds:
1105	temp_key[ind] = key_list[inds[ind]]
1106	inputs[key] = temp_key
1107	else:
1108	inputs[key] = []
1109
1110	return inputs, outputs
1111
1112	def add_column(self):
1113	"""
1114	Append a new column to the grid
1115	"""
1116
1117	# I Believe this is no longer used now that we have removed the
1118	# edit menu from the menubar - PDB July 12, 2015
1119	pos = self.GetSelection()
1120	grid = self.GetPage(pos)
1121	grid.AppendCols(1, True)
1122
1123	def on_remove_column(self):
1124	"""
1125	Remove the selected column from the grid
1126	"""
1127	# I Believe this is no longer used now that we have removed the
1128	# edit menu from the menubar - PDB July 12, 2015
1129	pos = self.GetSelection()
1130	grid = self.GetPage(pos)
1131	grid.on_remove_column(event=None)
1132
1133	class GridPanel(SPanel):
1134	"""
1135	A ScrolledPanel class that contains the grid sheet as well as a number of
1136	widgets to create interesting plots and buttons for help etc.
1137	"""
1138
1139	def __init__(self, parent, data_inputs=None,
1140	data_outputs=None, args, *kwds):
1141	"""
1142	Initialize the GridPanel
1143	"""
1144
1145	SPanel.__init__(self, parent, args, *kwds)
1146
1147	self.vbox = wx.BoxSizer(wx.VERTICAL)
1148
1149	self.plotting_sizer = wx.FlexGridSizer(3, 7, 10, 5)
1150	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
1151	self.grid_sizer = wx.BoxSizer(wx.HORIZONTAL)
1152	self.vbox.AddMany([(self.grid_sizer, 1, wx.EXPAND, 0),
1153	(wx.StaticLine(self, -1), 0, wx.EXPAND, 0),
1154	(self.plotting_sizer),
1155	(self.button_sizer, 0, wx.BOTTOM, 10)])
1156	self.parent = parent
1157	self._data_inputs = data_inputs
1158	self._data_outputs = data_outputs
1159	self.x = []
1160	self.y = []
1161	self.dy = []
1162	self.x_axis_label = None
1163	self.y_axis_label = None
1164	self.dy_axis_label = None
1165	self.x_axis_title = None
1166	self.y_axis_title = None
1167	self.x_axis_unit = None
1168	self.y_axis_unit = None
1169	self.view_button = None
1170	self.plot_button = None
1171	self.notebook = None
1172	self.plot_num = 1
1173
1174	self.layout_grid()
1175	self.layout_plotting_area()
1176	self.SetSizer(self.vbox)
1177
1178	def set_xaxis(self, label="", x=None):
1179	"""
1180	"""
1181	if x is None:
1182	x = []
1183	self.x = x
1184	self.x_axis_label.SetValue("%s[:]" % str(label))
1185	self.x_axis_title.SetValue(str(label))
1186
1187	def set_yaxis(self, label="", y=None):
1188	"""
1189	"""
1190	if y is None:
1191	y = []
1192	self.y = y
1193	self.y_axis_label.SetValue("%s[:]" % str(label))
1194	self.y_axis_title.SetValue(str(label))
1195
1196	def set_dyaxis(self, label="", dy=None):
1197	"""
1198	"""
1199	if dy is None:
1200	dy = []
1201	self.dy = dy
1202	self.dy_axis_label.SetValue("%s[:]" % str(label))
1203
1204	def get_plot_axis(self, col, list):
1205	"""
1206	"""
1207	axis = []
1208	pos = self.notebook.GetSelection()
1209	grid = self.notebook.GetPage(pos)
1210	for row in list:
1211	label = grid.GetCellValue(0, col)
1212	value = grid.GetCellValue(row - 1, col).strip()
1213	if value != "":
1214	if label.lower().strip() == "data":
1215	axis.append(float(row - 1))
1216	else:
1217	try:
1218	axis.append(float(value))
1219	except:
1220	msg = "Invalid data in row %s column %s" % (str(row), str(col))
1221	wx.PostEvent(self.parent.parent,
1222	StatusEvent(status=msg, info="error"))
1223	return None
1224	else:
1225	axis.append(None)
1226	return axis
1227
1228	def on_view(self, event):
1229	"""
1230	Get object represented by the given cells and plot them. Basically
1231	plot the colum in y vs the column in x.
1232	"""
1233
1234	pos = self.notebook.GetSelection()
1235	grid = self.notebook.GetPage(pos)
1236	title = self.notebook.GetPageText(pos)
1237	self.notebook.get_highlighted_row(False)
1238	if len(grid.selected_cells) == 0:
1239	msg = "Highlight a Data or Chi2 column first..."
1240	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1241	return
1242	elif len(grid.selected_cells) > 20:
1243	msg = "Too many data (> 20) to plot..."
1244	msg += "\n Please select no more than 20 data."
1245	wx.MessageDialog(self, msg, 'Plotting', wx.OK)
1246	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1247	return
1248
1249	for cell in grid.selected_cells:
1250	row, col = cell
1251	label_row = 0
1252	label = grid.GetCellValue(label_row, col)
1253	if label in grid.data:
1254	values = grid.data[label]
1255	if row > len(values) or row < 1:
1256	msg = "Invalid cell was chosen."
1257	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1258	continue
1259	else:
1260	value = values[row - 1]
1261	if issubclass(value.__class__, BatchCell):
1262	if value.object is None or len(value.object) == 0:
1263	msg = "Row %s , " % str(row)
1264	msg += "Column %s is NOT " % str(label)
1265	msg += "the results of fits to view..."
1266	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1267	return
1268	for new_plot in value.object:
1269	if new_plot is None or \
1270	not issubclass(new_plot.__class__,
1271	plottables.Plottable):
1272	msg = "Row %s , " % str(row)
1273	msg += "Column %s is NOT " % str(label)
1274	msg += "the results of fits to view..."
1275	wx.PostEvent(self.parent.parent,
1276	StatusEvent(status=msg, info="error"))
1277	return
1278	if issubclass(new_plot.__class__, Data1D):
1279	if label in grid.list_plot_panels.keys():
1280	group_id = grid.list_plot_panels[label]
1281	else:
1282	group_id = str(new_plot.group_id) + str(grid.uid)
1283	grid.list_plot_panels[label] = group_id
1284	if group_id not in new_plot.list_group_id:
1285	new_plot.group_id = group_id
1286	new_plot.list_group_id.append(group_id)
1287	else:
1288	if label.lower() in ["data", "chi2"]:
1289	if len(grid.selected_cells) != 1:
1290	msg = "2D View: Please select one data set"
1291	msg += " at a time for View Fit Results."
1292	wx.PostEvent(self.parent.parent,
1293	StatusEvent(status=msg, info="error"))
1294	return
1295
1296	wx.PostEvent(self.parent.parent,
1297	NewPlotEvent(plot=new_plot,
1298	group_id=str(new_plot.group_id),
1299	title=title))
1300	msg = "Plotting the View Fit Results completed!"
1301	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
1302	else:
1303	msg = "Row %s , " % str(row)
1304	msg += "Column %s is NOT " % str(label)
1305	msg += "the results of fits to view..."
1306	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1307	return
1308
1309	def on_plot(self, event):
1310	"""
1311	Evaluate the contains of textcrtl and plot result
1312	"""
1313
1314	pos = self.notebook.GetSelection()
1315	grid = self.notebook.GetPage(pos)
1316	column_names = {}
1317	if grid is not None:
1318	column_names = self.notebook.get_column_labels()
1319	#evaluate x
1320	sentence = self.x_axis_label.GetValue()
1321	try:
1322	if sentence.strip() == "":
1323	msg = "Select column values for x axis"
1324	raise ValueError, msg
1325	except:
1326	msg = "X axis value error."
1327	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1328	return
1329	dict = parse_string(sentence, column_names.keys())
1330
1331	try:
1332	sentence = self.get_sentence(dict, sentence, column_names)
1333	x = eval(sentence)
1334	except:
1335	msg = "Need a proper x-range."
1336	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1337	return
1338	#evaluate y
1339	sentence = self.y_axis_label.GetValue()
1340	try:
1341	if sentence.strip() == "":
1342	msg = "select value for y axis"
1343	raise ValueError, msg
1344	except:
1345	msg = "Y axis value error."
1346	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1347	return
1348	dict = parse_string(sentence, column_names.keys())
1349	try:
1350	sentence = self.get_sentence(dict, sentence, column_names)
1351	y = eval(sentence)
1352	except:
1353	msg = "Need a proper y-range."
1354	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1355	return
1356	#evaluate y
1357	sentence = self.dy_axis_label.GetValue()
1358	dy = None
1359	if sentence.strip() != "":
1360	dict = parse_string(sentence, column_names.keys())
1361	sentence = self.get_sentence(dict, sentence, column_names)
1362	try:
1363	dy = eval(sentence)
1364	except:
1365	msg = "Need a proper dy-range."
1366	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1367	return
1368	if len(x) != len(y) or (len(x) == 0 or len(y) == 0):
1369	msg = "Need same length for X and Y axis and both greater than 0"
1370	msg += " to plot.\n"
1371	msg += "Got X length = %s, Y length = %s" % (str(len(x)), str(len(y)))
1372	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1373	return
1374	if dy != None and (len(y) != len(dy)):
1375	msg = "Need same length for Y and dY axis and both greater than 0"
1376	msg += " to plot.\n"
1377	msg += "Got Y length = %s, dY length = %s" % (str(len(y)), str(len(dy)))
1378	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1379	return
1380	if dy == None:
1381	dy = numpy.zeros(len(y))
1382	#plotting
1383	new_plot = Data1D(x=x, y=y, dy=dy)
1384	new_plot.id = wx.NewId()
1385	new_plot.is_data = False
1386	new_plot.group_id = wx.NewId()
1387	y_title = self.y_axis_title.GetValue()
1388	x_title = self.x_axis_title.GetValue()
1389	title = "%s_vs_%s" % (y_title, x_title)
1390	new_plot.xaxis(x_title, self.x_axis_unit.GetValue())
1391	new_plot.yaxis(y_title, self.y_axis_unit.GetValue())
1392	try:
1393	title = y_title.strip()
1394	title += "_" + self.notebook.GetPageText(pos)
1395	title += "_" + str(self.plot_num)
1396	self.plot_num += 1
1397	new_plot.name = title
1398	new_plot.xtransform = "x"
1399	new_plot.ytransform = "y"
1400	wx.PostEvent(self.parent.parent,
1401	NewPlotEvent(plot=new_plot,
1402	group_id=str(new_plot.group_id), title=title))
1403	msg = "Plotting completed!"
1404	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
1405	self.parent.parent.update_theory(data_id=new_plot.id, theory=new_plot)
1406	except:
1407	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1408
1409	def on_help(self, event):
1410	"""
1411	Bring up the Batch Grid Panel Usage Documentation whenever
1412	the HELP button is clicked.
1413
1414	Calls DocumentationWindow with the path of the location within the
1415	documentation tree (after /doc/ ....". Note that when using old
1416	versions of Wx (before 2.9) and thus not the release version of
1417	installers, the help comes up at the top level of the file as
1418	webbrowser does not pass anything past the # to the browser when it is
1419	running "file:///...."
1420
1421	:param evt: Triggers on clicking the help button
1422	"""
1423
1424	#import documentation window here to avoid circular imports
1425	#if put at top of file with rest of imports.
1426	from documentation_window import DocumentationWindow
1427
1428	_TreeLocation = "user/sasgui/perspectives/fitting/fitting_help.html"
1429	_PageAnchor = "#batch-fit-mode"
1430	_doc_viewer = DocumentationWindow(self, -1, _TreeLocation, _PageAnchor,
1431	"Batch Mode Help")
1432
1433	def get_sentence(self, dict, sentence, column_names):
1434	"""
1435	Get sentence from dict
1436	"""
1437
1438	for tok, (col_name, list) in dict.iteritems():
1439	col = column_names[col_name]
1440	axis = self.get_plot_axis(col, list)
1441	if axis == None:
1442	return None
1443	sentence = sentence.replace(tok, "numpy.array(%s)" % str(axis))
1444	for key, value in FUNC_DICT.iteritems():
1445	sentence = sentence.replace(key.lower(), value)
1446	return sentence
1447
1448	def layout_grid(self):
1449	"""
1450	Draw the area related to the grid by adding it as the first element
1451	in the panel's grid_sizer
1452	"""
1453
1454	self.notebook = Notebook(parent=self)
1455	self.notebook.set_data(self._data_inputs, self._data_outputs)
1456	self.grid_sizer.Add(self.notebook, 1, wx.EXPAND, 0)
1457
1458	def layout_plotting_area(self):
1459	"""
1460	Add the area containing all the plot options, buttons etc to a plotting
1461	area sizer to later be added to the top level grid_sizer
1462	"""
1463
1464	view_description = wx.StaticBox(self, -1, 'Plot Fits/Residuals')
1465	note = "To plot the fits (or residuals), click the 'View Fits' button"
1466	note += "\n after highlighting the Data names (or Chi2 values)."
1467	note_text = wx.StaticText(self, -1, note)
1468	boxsizer1 = wx.StaticBoxSizer(view_description, wx.HORIZONTAL)
1469	self.x_axis_title = wx.TextCtrl(self, -1)
1470	self.y_axis_title = wx.TextCtrl(self, -1)
1471	self.x_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
1472	self.y_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
1473	self.dy_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
1474	self.x_axis_add = wx.Button(self, -1, "Add")
1475	self.x_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis,
1476	id=self.x_axis_add.GetId())
1477	self.y_axis_add = wx.Button(self, -1, "Add")
1478	self.y_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis,
1479	id=self.y_axis_add.GetId())
1480	self.dy_axis_add = wx.Button(self, -1, "Add")
1481	self.dy_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis,
1482	id=self.dy_axis_add.GetId())
1483	self.x_axis_unit = wx.TextCtrl(self, -1)
1484	self.y_axis_unit = wx.TextCtrl(self, -1)
1485	self.view_button = wx.Button(self, -1, "View Fits")
1486	view_tip = "Highlight the data set or the Chi2 column first."
1487	self.view_button.SetToolTipString(view_tip)
1488	wx.EVT_BUTTON(self, self.view_button.GetId(), self.on_view)
1489	self.plot_button = wx.Button(self, -1, "Plot")
1490	plot_tip = "Highlight a column for each axis and \n"
1491	plot_tip += "click the Add buttons first."
1492
1493	self.plot_button.SetToolTipString(plot_tip)
1494
1495	self.help_button = wx.Button(self, -1, "HELP")
1496	self.help_button.SetToolTipString("Get Help for Batch Mode")
1497	self.help_button.Bind(wx.EVT_BUTTON, self.on_help)
1498
1499	boxsizer1.AddMany([(note_text, 0, wx.LEFT, 10),
1500	(self.view_button, 0, wx.LEFT \| wx.RIGHT, 10)])
1501	self.button_sizer.AddMany([(boxsizer1, 0,
1502	wx.LEFT \| wx.RIGHT \| wx.BOTTOM, 10),
1503	(self.plot_button, 0,
1504	wx.LEFT \| wx.TOP \| wx.BOTTOM, 12),
1505	(self.help_button,0,
1506	wx.LEFT \| wx.TOP \| wx.BOTTOM, 12)])
1507
1508	wx.EVT_BUTTON(self, self.plot_button.GetId(), self.on_plot)
1509	self.plotting_sizer.AddMany(\
1510	[(wx.StaticText(self, -1, "X-axis Label\nSelection Range"), 1,
1511	wx.TOP \| wx.BOTTOM \| wx.LEFT, 10),
1512	(self.x_axis_label, 1, wx.TOP \| wx.BOTTOM, 10),
1513	(self.x_axis_add, 1, wx.TOP \| wx.BOTTOM \| wx.RIGHT, 10),
1514	(wx.StaticText(self, -1, "X-axis Label"), 1, wx.TOP \| wx.BOTTOM \| wx.LEFT, 10),
1515	(self.x_axis_title, 1, wx.TOP \| wx.BOTTOM, 10),
1516	(wx.StaticText(self, -1, "X-axis Unit"), 1, wx.TOP \| wx.BOTTOM, 10),
1517	(self.x_axis_unit, 1, wx.TOP \| wx.BOTTOM, 10),
1518	(wx.StaticText(self, -1, "Y-axis Label\nSelection Range"), 1,
1519	wx.BOTTOM \| wx.LEFT, 10),
1520	(self.y_axis_label, wx.BOTTOM, 10),
1521	(self.y_axis_add, 1, wx.BOTTOM \| wx.RIGHT, 10),
1522	(wx.StaticText(self, -1, "Y-axis Label"), 1,
1523	wx.BOTTOM \| wx.LEFT, 10),
1524	(self.y_axis_title, wx.BOTTOM, 10),
1525	(wx.StaticText(self, -1, "Y-axis Unit"), 1, wx.BOTTOM, 10),
1526	(self.y_axis_unit, 1, wx.BOTTOM, 10),
1527	(wx.StaticText(self, -1, "dY-Bar (Optional)\nSelection Range"),
1528	1, wx.BOTTOM \| wx.LEFT, 10),
1529	(self.dy_axis_label, wx.BOTTOM, 10),
1530	(self.dy_axis_add, 1, wx.BOTTOM \| wx.RIGHT, 10),
1531	(-1, -1),
1532	(-1, -1),
1533	(-1, -1),
1534	(-1, 1)])
1535
1536	def on_edit_axis(self, event):
1537	"""
1538	Get the selected column on the visible grid and set values for axis
1539	"""
1540
1541	try:
1542	cell_list = self.notebook.on_edit_axis()
1543	label, title = self.create_axis_label(cell_list)
1544	except:
1545	msg = str(sys.exc_value)
1546	wx.PostEvent(self.parent.parent, StatusEvent(status=msg, info="error"))
1547	return
1548	tcrtl = event.GetEventObject()
1549	if tcrtl == self.x_axis_add:
1550	self.edit_axis_helper(self.x_axis_label, self.x_axis_title, label, title)
1551	elif tcrtl == self.y_axis_add:
1552	self.edit_axis_helper(self.y_axis_label, self.y_axis_title, label, title)
1553	elif tcrtl == self.dy_axis_add:
1554	self.edit_axis_helper(self.dy_axis_label, None, label, None)
1555
1556	def create_axis_label(self, cell_list):
1557	"""
1558	Receive a list of cells and create a string presenting the selected
1559	cells.
1560
1561	:param cell_list: list of tuple
1562	"""
1563
1564	if self.notebook is not None:
1565	return self.notebook.create_axis_label(cell_list)
1566
1567	def edit_axis_helper(self, tcrtl_label, tcrtl_title, label, title):
1568	"""
1569	get controls to modify
1570	"""
1571
1572	if label != None:
1573	tcrtl_label.SetValue(str(label))
1574	if title != None:
1575	tcrtl_title.SetValue(str(title))
1576
1577	def add_column(self):
1578	"""
1579	"""
1580	# I Believe this is no longer used now that we have removed the
1581	# edit menu from the menubar - PDB July 12, 2015
1582	if self.notebook is not None:
1583	self.notebook.add_column()
1584
1585	def on_remove_column(self):
1586	"""
1587	"""
1588	# I Believe this is no longer used now that we have removed the
1589	# edit menu from the menubar - PDB July 12, 2015
1590	if self.notebook is not None:
1591	self.notebook.on_remove_column()
1592
1593
1594	class GridFrame(wx.Frame):
1595	"""
1596	The main wx.Frame for the batch results grid
1597	"""
1598
1599	def __init__(self, parent=None, data_inputs=None, data_outputs=None, id=-1,
1600	title="Batch Fitting Results Panel", size=(800, 500)):
1601	"""
1602	Initialize the Frame
1603	"""
1604
1605	wx.Frame.__init__(self, parent=parent, id=id, title=title, size=size)
1606	self.parent = parent
1607	self.panel = GridPanel(self, data_inputs, data_outputs)
1608	menubar = wx.MenuBar()
1609	self.SetMenuBar(menubar)
1610
1611	self.curr_col = None
1612	self.curr_grid = None
1613	self.curr_col_name = ""
1614	self.file = wx.Menu()
1615	menubar.Append(self.file, "&File")
1616
1617	hint = "Open file containing batch results"
1618	open_menu = self.file.Append(wx.NewId(), 'Open ', hint)
1619	wx.EVT_MENU(self, open_menu.GetId(), self.on_open)
1620
1621	hint = "Open the the current grid into excel"
1622	self.open_excel_menu = self.file.Append(wx.NewId(), 'Open with Excel', hint)
1623	wx.EVT_MENU(self, self.open_excel_menu.GetId(), self.open_with_excel)
1624	self.file.AppendSeparator()
1625	self.save_menu = self.file.Append(wx.NewId(), 'Save As', 'Save into File')
1626	wx.EVT_MENU(self, self.save_menu.GetId(), self.on_save_page)
1627
1628	# We need to grab a WxMenu handle here, otherwise the next one to grab
1629	# the handle will be treated as the Edit Menu handle when checking in
1630	# on_menu_open event handler and thus raise an exception when it hits an
1631	# unitialized object. Alternative is to comment out that whole section
1632	# in on_menu_open, but that would make it more difficult to undo the
1633	# hidding of the menu. PDB July 12, 2015.
1634	#
1635	# To enable the Edit menubar comment out next line and uncomment the
1636	# following line.
1637	self.edit = wx.Menu()
1638	#self.add_edit_menu()
1639
1640	self.Bind(wx.EVT_MENU_OPEN, self.on_menu_open)
1641	self.Bind(wx.EVT_CLOSE, self.on_close)
1642
1643	def add_edit_menu(self, menubar):
1644	"""
1645	populates the edit menu on the menubar. Not activated as of SasView
1646	3.1.0
1647	"""
1648	self.edit = wx.Menu()
1649
1650	add_table_menu = self.edit.Append(-1, 'New Table',
1651	'Add a New Table')
1652	self.edit.AppendSeparator()
1653	wx.EVT_MENU(self, add_table_menu.GetId(), self.add_table)
1654
1655	self.copy_menu = self.edit.Append(-1, 'Copy',
1656	'Copy the selected cells')
1657	wx.EVT_MENU(self, self.copy_menu.GetId(), self.on_copy)
1658	self.paste_menu = self.edit.Append(-1, 'Paste',
1659	'Paste the selected Cells')
1660	wx.EVT_MENU(self, self.paste_menu.GetId(), self.on_paste)
1661	self.clear_menu = self.edit.Append(-1, 'Clear',
1662	'Clear the selected Cells')
1663	wx.EVT_MENU(self, self.clear_menu.GetId(), self.on_clear)
1664
1665	self.edit.AppendSeparator()
1666	hint = "Insert column before the selected column"
1667	self.insert_before_menu = wx.Menu()
1668	self.insertb_sub_menu = self.edit.AppendSubMenu(self.insert_before_menu,
1669	'Insert Before', hint)
1670	hint = "Insert column after the selected column"
1671	self.insert_after_menu = wx.Menu()
1672	self.inserta_sub_menu = self.edit.AppendSubMenu(self.insert_after_menu,
1673	'Insert After', hint)
1674	hint = "Remove the selected column"
1675	self.remove_menu = self.edit.Append(-1, 'Remove Column', hint)
1676	wx.EVT_MENU(self, self.remove_menu.GetId(), self.on_remove_column)
1677	menubar.Append(self.edit, "&Edit")
1678
1679	def on_copy(self, event):
1680	"""
1681	On Copy from the Edit menu item on the menubar
1682	"""
1683	# I Believe this is no longer used now that we have removed the
1684	# edit menu from the menubar - PDB July 12, 2015
1685	if event != None:
1686	event.Skip()
1687	pos = self.panel.notebook.GetSelection()
1688	grid = self.panel.notebook.GetPage(pos)
1689	grid.Copy()
1690
1691	def on_paste(self, event):
1692	"""
1693	On Paste from the Edit menu item on the menubar
1694	"""
1695	# I Believe this is no longer used now that we have removed the
1696	# edit menu from the menubar - PDB July 12, 2015
1697	if event != None:
1698	event.Skip()
1699	pos = self.panel.notebook.GetSelection()
1700	grid = self.panel.notebook.GetPage(pos)
1701	grid.on_paste(None)
1702
1703	def on_clear(self, event):
1704	"""
1705	On Clear from the Edit menu item on the menubar
1706	"""
1707	# I Believe this is no longer used now that we have removed the
1708	# edit menu from the menubar - PDB July 12, 2015
1709	pos = self.panel.notebook.GetSelection()
1710	grid = self.panel.notebook.GetPage(pos)
1711	grid.Clear()
1712
1713	def GetLabelText(self, id):
1714	"""
1715	Get Label Text
1716	"""
1717	for item in self.insert_before_menu.GetMenuItems():
1718	m_id = item.GetId()
1719	if m_id == id:
1720	return item.GetLabel()
1721
1722	def on_remove_column(self, event):
1723	"""
1724	On remove column from the Edit menu Item on the menubar
1725	"""
1726	# I Believe this is no longer used now that we have removed the
1727	# edit menu from the menubar - PDB July 12, 2015
1728	pos = self.panel.notebook.GetSelection()
1729	grid = self.panel.notebook.GetPage(pos)
1730	grid.on_remove_column(event=None)
1731
1732	def on_menu_open(self, event):
1733	"""
1734	On menu open
1735	"""
1736	if self.file == event.GetMenu():
1737	pos = self.panel.notebook.GetSelection()
1738	grid = self.panel.notebook.GetPage(pos)
1739	has_data = (grid.data != None and grid.data != {})
1740	self.open_excel_menu.Enable(has_data)
1741	self.save_menu.Enable(has_data)
1742
1743	if self.edit == event.GetMenu():
1744	#get the selected column
1745	pos = self.panel.notebook.GetSelection()
1746	grid = self.panel.notebook.GetPage(pos)
1747	col_list = grid.GetSelectedCols()
1748	has_selection = False
1749	selected_cel = grid.selected_cells
1750	if len(selected_cel) > 0:
1751	_row, _col = selected_cel[0]
1752	has_selection = grid.IsInSelection(_row, _col)
1753	if len(grid.selected_cols) > 0:
1754	has_selection = True
1755	if len(grid.selected_rows) > 0:
1756	has_selection = True
1757	self.copy_menu.Enable(has_selection)
1758	self.clear_menu.Enable(has_selection)
1759
1760	if len(col_list) > 0:
1761	self.remove_menu.Enable(True)
1762	else:
1763	self.remove_menu.Enable(False)
1764	if len(col_list) == 0 or len(col_list) > 1:
1765	self.insertb_sub_menu.Enable(False)
1766	self.inserta_sub_menu.Enable(False)
1767	label = "Insert Column Before"
1768	self.insertb_sub_menu.SetText(label)
1769	label = "Insert Column After"
1770	self.inserta_sub_menu.SetText(label)
1771	else:
1772	self.insertb_sub_menu.Enable(True)
1773	self.inserta_sub_menu.Enable(True)
1774
1775	col = col_list[0]
1776	col_name = grid.GetCellValue(row=0, col=col)
1777	label = "Insert Column Before " + str(col_name)
1778	self.insertb_sub_menu.SetText(label)
1779	for item in self.insert_before_menu.GetMenuItems():
1780	self.insert_before_menu.DeleteItem(item)
1781	grid.insert_col_menu(menu=self.insert_before_menu,
1782	label=col_name, window=self)
1783	label = "Insert Column After " + str(col_name)
1784	self.inserta_sub_menu.SetText(label)
1785	for item in self.insert_after_menu.GetMenuItems():
1786	self.insert_after_menu.DeleteItem(item)
1787	grid.insert_after_col_menu(menu=self.insert_after_menu,
1788	label=col_name, window=self)
1789	event.Skip()
1790
1791
1792
1793	def on_save_page(self, event):
1794	"""
1795	Saves data in grid to a csv file.
1796
1797	At this time only the columns displayed get saved. Thus any error
1798	bars not inserted before saving will not be saved in the file
1799	"""
1800
1801	if self.parent is not None:
1802	pos = self.panel.notebook.GetSelection()
1803	grid = self.panel.notebook.GetPage(pos)
1804	if grid.file_name is None or grid.file_name.strip() == "" or \
1805	grid.data is None or len(grid.data) == 0:
1806	name = self.panel.notebook.GetPageText(pos)
1807	msg = " %s has not data to save" % str(name)
1808	wx.PostEvent(self.parent,
1809	StatusEvent(status=msg, info="error"))
1810
1811	return
1812	reader, ext = os.path.splitext(grid.file_name)
1813	path = None
1814	if self.parent is not None:
1815	location = os.path.dirname(grid.file_name)
1816	dlg = wx.FileDialog(self, "Save Project file",
1817	location, grid.file_name, ext, wx.SAVE)
1818	path = None
1819	if dlg.ShowModal() == wx.ID_OK:
1820	path = dlg.GetPath()
1821	dlg.Destroy()
1822	if path != None:
1823	if self.parent is not None:
1824	data = grid.get_grid_view()
1825	self.parent.write_batch_tofile(data=data,
1826	file_name=path,
1827	details=grid.details)
1828
1829	def on_open(self, event):
1830	"""
1831	Open file containing batch result
1832	"""
1833
1834	if self.parent is not None:
1835	self.parent.on_read_batch_tofile(self)
1836
1837	def open_with_excel(self, event):
1838	"""
1839	open excel and display batch result in Excel
1840	"""
1841
1842	if self.parent is not None:
1843	pos = self.panel.notebook.GetSelection()
1844	grid = self.panel.notebook.GetPage(pos)
1845	data = grid.get_grid_view()
1846	if grid.file_name is None or grid.file_name.strip() == "" or \
1847	grid.data is None or len(grid.data) == 0:
1848	name = self.panel.notebook.GetPageText(pos)
1849	msg = " %s has not data to open on excel" % str(name)
1850	wx.PostEvent(self.parent,
1851	StatusEvent(status=msg, info="error"))
1852
1853	return
1854	self.parent.open_with_externalapp(data=data,
1855	file_name=grid.file_name,
1856	details=grid.details)
1857
1858	def on_close(self, event):
1859	"""
1860	"""
1861	self.Hide()
1862
1863	def on_append_column(self, event):
1864	"""
1865	Append a new column to the grid
1866	"""
1867	self.panel.add_column()
1868
1869	def set_data(self, data_inputs, data_outputs, details="", file_name=None):
1870	"""
1871	Set data
1872	"""
1873	self.panel.notebook.set_data(data_inputs=data_inputs,
1874	file_name=file_name,
1875	details=details,
1876	data_outputs=data_outputs)
1877
1878	def add_table(self, event):
1879	"""
1880	Add a new table
1881	"""
1882	# DO not event.Skip(): it will make 2 pages
1883	self.panel.notebook.add_empty_page()
1884
1885	class BatchOutputFrame(wx.Frame):
1886	"""
1887	Allow to select where the result of batch will be displayed or stored
1888	"""
1889	def __init__(self, parent, data_inputs, data_outputs, file_name="",
1890	details="", args, *kwds):
1891	"""
1892	Initialize dialog
1893
1894	:param parent: Window instantiating this dialog
1895	:param result: result to display in a grid or export to an external\
1896	application.
1897	"""
1898
1899	#kwds['style'] = wx.CAPTION\|wx.SYSTEM_MENU
1900	wx.Frame.__init__(self, parent, args, *kwds)
1901	self.parent = parent
1902	self.panel = wx.Panel(self)
1903	self.file_name = file_name
1904	self.details = details
1905	self.data_inputs = data_inputs
1906	self.data_outputs = data_outputs
1907	self.data = {}
1908	for item in (self.data_outputs, self.data_inputs):
1909	self.data.update(item)
1910	self.flag = 1
1911	self.SetSize((300, 200))
1912	self.local_app_selected = None
1913	self.external_app_selected = None
1914	self.save_to_file = None
1915	self._do_layout()
1916
1917	def _do_layout(self):
1918	"""
1919	Draw the content of the current dialog window
1920	"""
1921
1922	vbox = wx.BoxSizer(wx.VERTICAL)
1923	box_description = wx.StaticBox(self.panel, -1, str("Batch Outputs"))
1924	hint_sizer = wx.StaticBoxSizer(box_description, wx.VERTICAL)
1925	selection_sizer = wx.GridBagSizer(5, 5)
1926	button_sizer = wx.BoxSizer(wx.HORIZONTAL)
1927	text = "Open with %s" % self.parent.application_name
1928	self.local_app_selected = wx.RadioButton(self.panel, -1, text, style=wx.RB_GROUP)
1929	self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
1930	id=self.local_app_selected.GetId())
1931	text = "Open with Excel"
1932	self.external_app_selected = wx.RadioButton(self.panel, -1, text)
1933	self.Bind(wx.EVT_RADIOBUTTON, self.onselect, id=self.external_app_selected.GetId())
1934	text = "Save to File"
1935	self.save_to_file = wx.CheckBox(self.panel, -1, text)
1936	self.Bind(wx.EVT_CHECKBOX, self.onselect, id=self.save_to_file.GetId())
1937	self.local_app_selected.SetValue(True)
1938	self.external_app_selected.SetValue(False)
1939	self.save_to_file.SetValue(False)
1940	button_close = wx.Button(self.panel, -1, "Close")
1941	button_close.Bind(wx.EVT_BUTTON, id=button_close.GetId(), handler=self.on_close)
1942	button_apply = wx.Button(self.panel, -1, "Apply")
1943	button_apply.Bind(wx.EVT_BUTTON, id=button_apply.GetId(), handler=self.on_apply)
1944	button_apply.SetFocus()
1945	hint = ""
1946	hint_sizer.Add(wx.StaticText(self.panel, -1, hint))
1947	hint_sizer.Add(selection_sizer)
1948	#draw area containing radio buttons
1949	ix = 0
1950	iy = 0
1951	selection_sizer.Add(self.local_app_selected, (iy, ix),
1952	(1, 1), wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
1953	iy += 1
1954	selection_sizer.Add(self.external_app_selected, (iy, ix),
1955	(1, 1), wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
1956	iy += 1
1957	selection_sizer.Add(self.save_to_file, (iy, ix),
1958	(1, 1), wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
1959	#contruction the sizer contaning button
1960	button_sizer.Add((20, 20), 1, wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
1961
1962	button_sizer.Add(button_close, 0,
1963	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
1964	button_sizer.Add(button_apply, 0,
1965	wx.LEFT \| wx.RIGHT \| wx.ADJUST_MINSIZE, 10)
1966	vbox.Add(hint_sizer, 0, wx.EXPAND \| wx.ALL, 10)
1967	vbox.Add(wx.StaticLine(self.panel, -1), 0, wx.EXPAND, 0)
1968	vbox.Add(button_sizer, 0, wx.TOP \| wx.BOTTOM, 10)
1969	self.SetSizer(vbox)
1970
1971	def on_apply(self, event):
1972	"""
1973	Get the user selection and display output to the selected application
1974	"""
1975
1976	if self.flag == 1:
1977	self.parent.open_with_localapp(data_inputs=self.data_inputs,
1978	data_outputs=self.data_outputs)
1979	elif self.flag == 2:
1980	self.parent.open_with_externalapp(data=self.data,
1981	file_name=self.file_name,
1982	details=self.details)
1983	def on_close(self, event):
1984	"""
1985	close the Window
1986	"""
1987
1988	self.Close()
1989
1990	def onselect(self, event=None):
1991	"""
1992	Receive event and display data into third party application
1993	or save data to file.
1994	"""
1995	if self.save_to_file.GetValue():
1996	_, ext = os.path.splitext(self.file_name)
1997	path = None
1998	location = os.getcwd()
1999	if self.parent is not None:
2000	location = os.path.dirname(self.file_name)
2001	dlg = wx.FileDialog(self, "Save Project file",
2002	location, self.file_name, ext, wx.SAVE)
2003	path = None
2004	if dlg.ShowModal() == wx.ID_OK:
2005	path = dlg.GetPath()
2006	dlg.Destroy()
2007	if path != None:
2008	if self.parent is not None and self.data is not None:
2009	self.parent.write_batch_tofile(data=self.data,
2010	file_name=path,
2011	details=self.details)
2012	if self.local_app_selected.GetValue():
2013	self.flag = 1
2014	else:
2015	self.flag = 2
2016	return self.flag
2017
2018
2019
2020	if __name__ == "__main__":
2021	app = wx.App()
2022
2023	try:
2024	data = {}
2025	j = 0
2026	for i in range(4):
2027	j += 1
2028	data["index" + str(i)] = [i / j, i * j, i, i + j]
2029
2030	data_input = copy.deepcopy(data)
2031	data_input["index5"] = [10, 20, 40, 50]
2032	frame = GridFrame(data_outputs=data, data_inputs=data_input)
2033	frame.Show(True)
2034	except:
2035	print sys.exc_value
2036
2037	app.MainLoop()

Note: See TracBrowser for help on using the repository browser.

SasView

source: sasview/src/sas/sasgui/guiframe/data_processor.py @ afb93df

Download in other formats: