data_processor.py @ 51f14603

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

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

SasView

source: sasview/src/sans/guiframe/data_processor.py @ 51f14603

Download in other formats: