data_processor.py @ fb0de166

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 fb0de166 was fb0de166, checked in by Gervaise Alina <gervyh@…>, 13 years ago
minor refactor
Property mode set to `100644`
File size: 44.4 KB

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

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

SasView

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

Download in other formats: