data_processor.py @ caf3a08f

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 caf3a08f was caf3a08f, checked in by Gervaise Alina <gervyh@…>, 13 years ago
add functon to output result to excel
Property mode set to `100644`
File size: 26.6 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	from sans.guiframe.panel_base import PanelBase
	16	import wx.lib.sheet as sheet
	17
	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
	25
	26
[904830e]	27	def parse_string(sentence, list):
	28	"""
	29	Return a dictionary of column label and index or row selected
	30	:param sentence: String to parse
	31	:param list: list of columns label
	32	"""
	33	toks = []
	34	p2 = re.compile(r'\d+')
	35	p = re.compile(r'[\+\-\*\%\/]')
	36	labels = p.split(sentence)
	37	col_dict = {}
	38	for elt in labels:
	39	rang = None
	40	temp_arr = []
	41	for label in list:
	42	label_pos = elt.find(label)
	43	if label_pos != -1:
	44	if elt.count(',') > 0:
	45	new_temp = []
	46	temp = elt.split(label)
	47	for item in temp:
	48	range_pos = item.find(":")
	49	if range_pos != -1:
	50	rang = p2.findall(item)
	51	for i in xrange(int(rang[0]), int(rang[1])+1 ):
	52	new_temp.append(i)
	53	temp_arr += new_temp
	54	else:
	55	temp = elt.split(label)
	56	for item in temp:
	57	range_pos = item.find(":")
	58	if range_pos != -1:
	59	rang = p2.findall(item)
	60	for i in xrange(int(rang[0]), int(rang[1])+1 ):
	61	temp_arr.append(i)
	62	col_dict[elt] = (label, temp_arr)
	63	return col_dict
[24adb89]	64
	65	class GridPage(sheet.CSheet):
[9c8f3ad]	66	"""
	67	"""
[24adb89]	68	def __init__(self, parent, panel=None):
	69	"""
	70	"""
	71	sheet.CSheet.__init__(self, parent)
	72	self.SetLabelBackgroundColour('#DBD4D4')
[9c8f3ad]	73	self.AutoSize()
[24adb89]	74	self.panel = panel
	75	self.col_names = []
[9c8f3ad]	76	self._cols = 50
	77	self._rows = 51
	78	col_with = 30
	79	row_height = 20
[49ad00b]	80	self.axis_value = []
	81	self.axis_label = ""
	82	self.selected_cells = []
	83	self.selected_cols = []
[9c8f3ad]	84	self.SetColMinimalAcceptableWidth(col_with)
	85	self.SetRowMinimalAcceptableHeight(row_height)
[24adb89]	86	self.SetNumberRows(self._cols)
	87	self.SetNumberCols(self._rows)
[49ad00b]	88	self.Bind(wx.grid.EVT_GRID_LABEL_LEFT_CLICK, self.on_left_click)
[24adb89]	89	self.Bind(wx.grid.EVT_GRID_LABEL_RIGHT_CLICK, self.on_right_click)
[49ad00b]	90	self.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.on_selected_cell)
	91
	92	def on_selected_cell(self, event):
	93	"""
	94	Handler catching cell selection
	95	"""
	96	flag = event.CmdDown() or event.ControlDown()
[904830e]	97	row, col = event.GetRow(), event.GetCol()
	98	cell = (row, col)
[49ad00b]	99	if not flag:
	100	self.selected_cells = []
[904830e]	101	self.axis_value = []
	102	self.axis_label = ""
[49ad00b]	103	if cell in self.selected_cells:
	104	self.selected_cells.remove(cell)
	105	else:
	106	self.selected_cells.append(cell)
[904830e]	107	if row > 1:
	108	if (cell) in self.selected_cells:
	109	self.axis_value.append(self.GetCellValue(row, col))
	110	self.axis_label = self.GetCellValue(row, col)
[49ad00b]	111	event.Skip()
	112
	113	def on_left_click(self, event):
	114	"""
	115	Catch the left click on label mouse event
	116	"""
	117	flag = event.CmdDown() or event.ControlDown()
	118	col = event.GetCol()
	119	if not flag:
[904830e]	120	self.selected_cols = []
	121	self.axis_value = []
	122	self.axis_label = ""
[49ad00b]	123	if col not in self.selected_cols:
	124	self.selected_cols.append(col)
[904830e]	125	if not flag :
	126	for row in range(2, self.GetNumberRows()+ 1):
	127	self.axis_value.append(self.GetCellValue(row, col))
	128	row = 1
	129	self.axis_label = self.GetCellValue(row, col)
[49ad00b]	130	event.Skip()
	131
[24adb89]	132	def on_right_click(self, event):
[9c8f3ad]	133	"""
	134	Catch the right click mouse
	135	"""
[24adb89]	136	col = event.GetCol()
	137	if col != -1 and len(self.col_names) > col:
	138	label = self.col_names[int(col)]
	139	self.axis_label = label
	140	if label in self.data.keys():
	141	col_val = self.data[label]
	142	self.axis_value = col_val
	143	# Slicer plot popup menu
	144	slicerpop = wx.Menu()
	145	id = wx.NewId()
	146	slicerpop.Append(id, '&Set X axis', 'Set X axis')
	147	wx.EVT_MENU(self, id, self.on_set_x_axis)
	148
	149	id = wx.NewId()
	150	slicerpop.Append(id, '&Set Y axis', 'Set Y axis')
	151	wx.EVT_MENU(self, id, self.on_set_y_axis)
	152	pos = event.GetPosition()
	153	pos = self.ScreenToClient(pos)
	154	self.PopupMenu(slicerpop, pos)
	155
	156	def on_set_x_axis(self, event):
[9c8f3ad]	157	"""
	158	"""
[24adb89]	159	self.panel.set_xaxis(x=self.axis_value, label=self.axis_label)
	160
	161	def on_set_y_axis(self, event):
[9c8f3ad]	162	"""
	163	"""
[24adb89]	164	self.panel.set_yaxis(y=self.axis_value, label=self.axis_label)
	165
	166	def set_data(self, data):
	167	"""
[9c8f3ad]	168	Add data to the grid
[24adb89]	169	"""
	170	if data is None:
	171	data = {}
	172	if len(data) > 0:
[9c8f3ad]	173	self._cols = self.GetNumberCols()
	174	self._rows = self.GetNumberRows()
[24adb89]	175	self.data = data
	176	self.col_names = data.keys()
	177	self.col_names.sort()
[9c8f3ad]	178	nbr_user_cols = len(self.col_names)
	179	#Add more columns to the grid if necessary
	180	if nbr_user_cols > self._cols:
	181	new_col_nbr = nbr_user_cols - self._cols
	182	self.AppendCols(new_col_nbr, True)
	183	#Add more rows to the grid if necessary
	184	nbr_user_row = len(self.data.values())
	185	if nbr_user_row > self._rows + 1:
	186	new_row_nbr = nbr_user_row - self._rows
	187	self.AppendRows(new_row_nbr, True)
	188	# add data to the grid
[10675c3]	189	row = 0
[9c8f3ad]	190	for index in range(nbr_user_cols):
[10675c3]	191	col = index
[9c8f3ad]	192	# use the first row of the grid to add user defined labels
[10675c3]	193	self.SetCellValue(row, col, str(self.col_names[index]))
[24adb89]	194	col = 0
	195	for value_list in self.data.values():
[10675c3]	196	row = 1
	197	for index in range(len(value_list)):
	198	self.SetCellValue(row, col, str(value_list[index]))
	199	row += 1
[24adb89]	200	col += 1
	201	self.AutoSize()
[9c8f3ad]	202
[24adb89]	203
	204	class Notebook(nb, PanelBase):
	205	"""
	206	## Internal name for the AUI manager
	207	window_name = "Fit panel"
	208	## Title to appear on top of the window
	209	"""
	210	window_caption = "Notebook "
	211
	212	def __init__(self, parent, manager=None, data=None, args, *kwargs):
	213	"""
	214	"""
	215	nb.__init__(self, parent, -1,
[7ad194fa]	216	style= wx.aui.AUI_BUTTON_DOWN\|
	217	wx.aui.AUI_NB_WINDOWLIST_BUTTON\|
[9c8f3ad]	218	wx.aui.AUI_NB_DEFAULT_STYLE\|
	219	wx.CLIP_CHILDREN)
[24adb89]	220	PanelBase.__init__(self, parent)
[9c8f3ad]	221	self.enable_close_button()
[24adb89]	222	self.parent = parent
	223	self.manager = manager
	224	self.data = data
[9c8f3ad]	225	self.Bind(wx.aui.EVT_AUINOTEBOOK_PAGE_CLOSE, self.on_close_page)
[7ad194fa]	226
[9c8f3ad]	227	def enable_close_button(self):
	228	"""
	229	display the close button on tab for more than 1 tabs else remove the
	230	close button
	231	"""
	232	if self.GetPageCount() <= 1:
	233	style = self.GetWindowStyleFlag()
	234	flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
	235	if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB == flag:
	236	style = style & ~wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
	237	self.SetWindowStyle(style)
	238	else:
	239	style = self.GetWindowStyleFlag()
	240	flag = wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
	241	if style & wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB != flag:
	242	style \|= wx.aui.AUI_NB_CLOSE_ON_ACTIVE_TAB
	243	self.SetWindowStyle(style)
[9680906d]	244
	245	def on_edit_axis(self):
	246	"""
	247	Return the select cell of a given selected column
	248	"""
	249	pos = self.GetSelection()
	250	grid = self.GetPage(pos)
[49ad00b]	251	if len(grid.selected_cols) > 1:
	252	msg = "Edit axis doesn't understand this selection.\n"
	253	msg += "Please select only one column"
	254	raise ValueError, msg
	255	list_of_cells = []
	256	if len(grid.selected_cols) == 1:
	257	col = grid.selected_cols[0]
[7ad194fa]	258	if len(grid.selected_cells) > 0:
	259	cell_row, cell_col = grid.selected_cells[0]
	260	if cell_col != col:
	261	msg = "Edit axis doesn't understand this selection.\n"
	262	msg += "Please select element of the same col"
	263	raise ValueError, msg
[49ad00b]	264	for row in range(grid.GetNumberRows()):
[7ad194fa]	265	list_of_cells.append((row + 1 , col))
	266	for item in grid.selected_cells:
	267	if item in list_of_cells:
	268	list_of_cells.remove(item)
	269	elif len(grid.selected_cols) == 0:
	270	list_of_cells = [(row + 1, col) for row, col in grid.selected_cells]
	271	return list_of_cells
	272
[904830e]	273	def get_column_labels(self):
	274	"""
	275	return dictionary of columns labels of the current page
	276	"""
	277	pos = self.GetSelection()
	278	grid = self.GetPage(pos)
	279	labels = {}
	280	row = 0
	281	for col in range(grid.GetNumberCols()):
	282	label = grid.GetCellValue(row, col)
	283	if label.strip() != "" :
	284	labels[label.strip()] = col
	285	return labels
	286
[7ad194fa]	287	def create_axis_label(self, cell_list):
	288	"""
	289	Receive a list of cells and create a string presenting the selected
	290	cells.
	291	:param cell_list: list of tuple
[9680906d]	292
[7ad194fa]	293	"""
	294	pos = self.GetSelection()
	295	grid = self.GetPage(pos)
	296	label = ""
	297	col_name = ""
	298	if len(cell_list) > 0:
	299	temp_list = copy.deepcopy(cell_list)
	300	temp_list.sort()
	301	temp = []
	302	for item in temp_list:
	303	if item[0] <= 1:
	304	temp.append(item)
	305	for element in temp:
	306	temp_list.remove(element)
[904830e]	307	row_min, col = temp_list[0]
[7ad194fa]	308	row_max = row_min
[904830e]	309	col_name = grid.GetCellValue(0, col)
[7ad194fa]	310	label += str(col_name) + "[" + str(row_min) + ":"
	311	for index in xrange(len(temp_list)):
	312	prev = index - 1
	313	row, _ = temp_list[index]
	314	if row > row_max + 1 :
	315	if prev > 0:
	316	row_max, _ = temp_list[prev]
	317	label += str(row_max) + "]" + ","
	318	row_min = row
	319	label += "[" + str(row_min) + ":"
	320	row_max = row
	321	if (index == len(temp_list)- 1):
[904830e]	322	label += str(row_max) + "]"
	323	return label, col_name
[7ad194fa]	324
[9c8f3ad]	325	def on_close_page(self, event):
	326	"""
	327	close the page
	328	"""
	329	if self.GetPageCount() == 1:
	330	event.Veto()
	331	self.enable_close_button()
[24adb89]	332
	333	def set_data(self, data):
	334	if data is None:
	335	return
	336
[36ce0b5]	337	grid = GridPage(self, panel=self.parent)
	338	grid.set_data(data)
[9c8f3ad]	339	self.AddPage(grid, "")
	340	pos = self.GetPageIndex(grid)
	341	title = "Batch " + str(self.GetPageCount())
	342	self.SetPageText(pos, title)
[24adb89]	343
[9c8f3ad]	344	def add_column(self):
	345	"""
	346	Append a new column to the grid
	347	"""
	348	pos = self.GetSelection()
	349	grid = self.GetPage(pos)
	350	grid.AppendCols(1, True)
	351
	352	def on_remove_column(self):
	353	"""
	354	Remove column to the current grid
	355	"""
	356	pos = self.GetSelection()
	357	grid = self.GetPage(pos)
	358	cols_pos = grid.GetSelectedCols()
	359	for cpos in cols_pos:
	360	grid.DeleteCols(cpos)
	361
	362
[24adb89]	363	class SPanel(ScrolledPanel):
	364	def __init__(self, parent, args, *kwds):
	365	ScrolledPanel.__init__(self, parent , args, *kwds)
	366	self.SetupScrolling()
	367
[9c8f3ad]	368
[24adb89]	369	class GridPanel(SPanel):
[904830e]	370	def __init__(self, parent, data=None, args, *kwds):
[24adb89]	371	SPanel.__init__(self, parent , args, *kwds)
	372	self.vbox = wx.BoxSizer(wx.VERTICAL)
	373
[9680906d]	374	self.plotting_sizer = wx.FlexGridSizer(3, 7, 10, 5)
[24adb89]	375	self.grid_sizer = wx.BoxSizer(wx.HORIZONTAL)
	376	self.vbox.AddMany([(self.grid_sizer, 1, wx.EXPAND, 0),
	377	(wx.StaticLine(self, -1), 0, wx.EXPAND, 0),
	378	(self.plotting_sizer)])
	379	self.parent = parent
	380	self._data = data
	381	self.x = []
	382	self.y = []
	383	self.x_axis_label = None
	384	self.y_axis_label = None
[904830e]	385	self.x_axis_title = None
	386	self.y_axis_title = None
[24adb89]	387	self.x_axis_unit = None
	388	self.y_axis_unit = None
	389	self.plot_button = None
[904830e]	390	self.notebook = None
[24adb89]	391	self.layout_grid()
	392	self.layout_plotting_area()
	393	self.SetSizer(self.vbox)
	394
	395	def set_data(self, data):
	396	"""
	397	"""
[904830e]	398	if self.notebook is not None:
	399	self.notebook.set_data(data)
[24adb89]	400
[904830e]	401	def set_xaxis(self, label="", x=None):
	402	"""
	403	"""
[24adb89]	404	if x is None:
	405	x = []
	406	self.x = x
[904830e]	407	self.x_axis_label.SetValue("%s[:]" % str(label))
	408	self.x_axis_title.SetValue(str(label))
[24adb89]	409
[904830e]	410	def set_yaxis(self, label="", y=None):
	411	"""
	412	"""
[24adb89]	413	if y is None:
	414	y = []
	415	self.y = y
[904830e]	416	self.y_axis_label.SetValue("%s[:]" % str(label))
	417	self.y_axis_title.SetValue(str(label))
[24adb89]	418
[904830e]	419	def get_plot_axis(self, col, list):
	420	"""
	421
	422	"""
	423	axis = []
	424	pos = self.notebook.GetSelection()
	425	grid = self.notebook.GetPage(pos)
	426	for row in list:
	427	label = grid.GetCellValue(row - 1, col)
	428	if label.strip() != "":
	429	axis.append(float(label.strip()))
	430	return axis
	431
[24adb89]	432	def on_plot(self, event):
	433	"""
[904830e]	434	Evaluate the contains of textcrtl and plot result
[24adb89]	435	"""
[904830e]	436	pos = self.notebook.GetSelection()
	437	grid = self.notebook.GetPage(pos)
	438	column_names = {}
	439	if grid is not None:
	440	column_names = self.notebook.get_column_labels()
	441	#evalue x
	442	sentence = self.x_axis_label.GetValue()
	443	if sentence.strip() == "":
	444	msg = "select value for x axis"
	445	raise ValueError, msg
	446	dict = parse_string(sentence, column_names.keys())
	447	for tok, (col_name, list) in dict.iteritems():
	448	col = column_names[col_name]
	449	xaxis = self.get_plot_axis(col, list)
	450	sentence = sentence.replace(tok,
	451	"numpy.array(%s)" % str(xaxis))
	452	for key, value in FUNC_DICT.iteritems():
	453	sentence = sentence.replace(key.lower(), value)
	454	x = eval(sentence)
	455	#evaluate y
	456	sentence = self.y_axis_label.GetValue()
	457	if sentence.strip() == "":
	458	msg = "select value for y axis"
	459	raise ValueError, msg
	460	dict = parse_string(sentence, column_names.keys())
	461	for tok, (col_name, list) in dict.iteritems():
	462	col = column_names[col_name]
	463	yaxis = self.get_plot_axis(col, list)
	464	sentence = sentence.replace(tok,
	465	"numpy.array(%s)" % str(yaxis))
	466	for key, value in FUNC_DICT.iteritems():
	467	sentence = sentence.replace(key, value)
	468	y = eval(sentence)
	469	#plotting
	470	new_plot = Data1D(x=x, y=y)
[24adb89]	471	new_plot.id = wx.NewId()
	472	new_plot.group_id = wx.NewId()
[904830e]	473	title = "%s vs %s" % (self.y_axis_title.GetValue(),
	474	self.x_axis_title.GetValue())
	475	new_plot.xaxis(self.x_axis_title.GetValue(), self.x_axis_unit.GetValue())
	476	new_plot.yaxis(self.y_axis_title.GetValue(), self.y_axis_unit.GetValue())
	477	try:
	478	title = self.notebook.GetPageText(pos)
	479	wx.PostEvent(self.parent.parent,
[49ad00b]	480	NewPlotEvent(plot=new_plot,
[904830e]	481	group_id=str(new_plot.group_id), title =title))
	482	except:
	483	pass
	484
[24adb89]	485	def layout_grid(self):
	486	"""
	487	Draw the area related to the grid
	488	"""
[904830e]	489	self.notebook = Notebook(parent=self)
	490	self.notebook.set_data(self._data)
	491	self.grid_sizer.Add(self.notebook, 1, wx.EXPAND, 0)
[24adb89]	492
	493	def layout_plotting_area(self):
	494	"""
	495	Draw area containing options to plot
	496	"""
[904830e]	497	self.x_axis_title = wx.TextCtrl(self, -1)
	498	self.y_axis_title = wx.TextCtrl(self, -1)
	499	self.x_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
	500	self.y_axis_label = wx.TextCtrl(self, -1, size=(200, -1))
[9680906d]	501	self.x_axis_add = wx.Button(self, -1, "Add")
	502	self.x_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis,
	503	id=self.x_axis_add.GetId())
	504	self.y_axis_add = wx.Button(self, -1, "Add")
	505	self.y_axis_add.Bind(event=wx.EVT_BUTTON, handler=self.on_edit_axis,
	506	id=self.y_axis_add.GetId())
[24adb89]	507	self.x_axis_unit = wx.TextCtrl(self, -1)
	508	self.y_axis_unit = wx.TextCtrl(self, -1)
	509	self.plot_button = wx.Button(self, -1, "Plot")
	510	wx.EVT_BUTTON(self, self.plot_button.GetId(), self.on_plot)
[9680906d]	511	self.plotting_sizer.AddMany([
[904830e]	512	(wx.StaticText(self, -1, "x-axis label"), 1,
	513	wx.TOP\|wx.BOTTOM\|wx.LEFT, 10),
	514	(self.x_axis_label, 1, wx.TOP\|wx.BOTTOM, 10),
	515	(self.x_axis_add, 1, wx.TOP\|wx.BOTTOM\|wx.RIGHT, 10),
	516	(wx.StaticText(self, -1, "x-axis title"), 1,
	517	wx.TOP\|wx.BOTTOM\|wx.LEFT, 10),
	518	(self.x_axis_title, 1, wx.TOP\|wx.BOTTOM, 10),
	519	(wx.StaticText(self, -1 , "unit"), 1,
	520	wx.TOP\|wx.BOTTOM, 10),
	521	(self.x_axis_unit, 1, wx.TOP\|wx.BOTTOM, 10),
	522	(wx.StaticText(self, -1, "y-axis label"), 1,
	523	wx.BOTTOM\|wx.LEFT, 10),
	524	(self.y_axis_label, wx.BOTTOM, 10),
	525	(self.y_axis_add, 1, wx.BOTTOM\|wx.RIGHT, 10),
	526	(wx.StaticText(self, -1, "y-axis title"), 1,
	527	wx.BOTTOM\|wx.LEFT, 10),
	528	(self.y_axis_title, wx.BOTTOM, 10),
	529	(wx.StaticText(self, -1 , "unit"), 1, wx.BOTTOM, 10),
	530	(self.y_axis_unit, 1, wx.BOTTOM, 10),
[9680906d]	531	(-1, -1),
	532	(-1, -1),
	533	(-1, -1),
	534	(-1, -1),
	535	(-1, -1),
	536	(-1, -1),
[904830e]	537	(self.plot_button, 1, wx.LEFT\|wx.BOTTOM, 10)])
[24adb89]	538
[9680906d]	539	def on_edit_axis(self, event):
	540	"""
	541	Get the selected column on the visible grid and set values for axis
	542	"""
[904830e]	543	cell_list = self.notebook.on_edit_axis()
	544	label, title = self.create_axis_label(cell_list)
	545	tcrtl = event.GetEventObject()
	546	if tcrtl == self.x_axis_add:
	547	self.edit_axis_helper(self.x_axis_label, self.x_axis_title,
	548	label, title)
	549	elif tcrtl == self.y_axis_add:
	550	self.edit_axis_helper(self.y_axis_label, self.y_axis_title,
	551	label, title)
	552
[7ad194fa]	553	def create_axis_label(self, cell_list):
	554	"""
	555	Receive a list of cells and create a string presenting the selected
	556	cells.
	557	:param cell_list: list of tuple
[24adb89]	558
[7ad194fa]	559	"""
[904830e]	560	if self.notebook is not None:
	561	return self.notebook.create_axis_label(cell_list)
[7ad194fa]	562
[904830e]	563	def edit_axis_helper(self, tcrtl_label, tcrtl_title, label, title):
[9680906d]	564	"""
[904830e]	565	get controls to modify
[9680906d]	566	"""
[904830e]	567	tcrtl_label.SetValue(str(label))
	568	tcrtl_title.SetValue(str(title))
	569
[24adb89]	570	def add_column(self):
[9c8f3ad]	571	"""
	572	"""
[904830e]	573	if self.notebook is not None:
	574	self.notebook.add_column()
[24adb89]	575
[9c8f3ad]	576	def on_remove_column(self):
	577	"""
	578	"""
[904830e]	579	if self.notebook is not None:
	580	self.notebook.on_remove_column()
[9c8f3ad]	581
[24adb89]	582
	583	class GridFrame(wx.Frame):
[7ad194fa]	584	def __init__(self, parent=None, data=None, id=-1,
	585	title="Batch Results", size=(700, 400)):
[24adb89]	586	wx.Frame.__init__(self, parent=parent, id=id, title=title, size=size)
	587	self.parent = parent
	588	self.panel = GridPanel(self, data)
	589	menubar = wx.MenuBar()
	590	self.SetMenuBar(menubar)
[9c8f3ad]	591	edit = wx.Menu()
	592	menubar.Append(edit, "&Edit")
[cb26857]	593	self.Bind(wx.EVT_CLOSE, self.on_close)
[24adb89]	594
[9c8f3ad]	595	add_col_menu = edit.Append(wx.NewId(), 'Add Column', 'Add column')
	596	wx.EVT_MENU(self, add_col_menu.GetId(), self.on_add_column)
	597	remove_col_menu = edit.Append(wx.NewId(), 'Remove Column',
	598	'Remove Column')
	599	wx.EVT_MENU(self, remove_col_menu.GetId(), self.on_remove_column)
	600
[cb26857]	601	def on_close(self, event):
	602	"""
	603	"""
	604	self.Hide()
	605
[9c8f3ad]	606	def on_remove_column(self, event):
	607	"""
	608	Remove the selected column to the grid
	609	"""
	610	self.panel.on_remove_column()
	611
[24adb89]	612	def on_add_column(self, event):
[cb26857]	613	"""
[9c8f3ad]	614	Append a new column to the grid
[cb26857]	615	"""
[24adb89]	616	self.panel.add_column()
[cb26857]	617
[24adb89]	618	def set_data(self, data):
[cb26857]	619	"""
	620	"""
[24adb89]	621	self.panel.set_data(data)
	622
	623
[73197d0]	624	class BatchOutputDialog(wx.Dialog):
	625	"""
	626	Allow to select where the result of batch will be displayed or stored
	627	"""
[850525c]	628	def __init__(self, parent=None, data=None, file_name="",
	629	details="", args, *kwds):
[73197d0]	630	"""
	631	:param parent: Window instantiating this dialog
	632	:param result: result to display in a grid or export to an external
	633	application.
	634	"""
[850525c]	635	#kwds['style'] = wx.CAPTION\|wx.SYSTEM_MENU
[73197d0]	636	wx.Dialog.__init__(self, parent, args, *kwds)
	637	self.parent = parent
[850525c]	638	self.file_name = file_name
	639	self.details = details
[73197d0]	640	self.data = data
	641	self.flag = 1
	642	self.SetSize((300, 200))
	643	self.local_app_selected = None
	644	self.external_app_selected = None
	645	self.save_to_file = None
	646	self._do_layout()
	647
	648	def _do_layout(self):
	649	"""
	650	Draw the content of the current dialog window
	651	"""
	652	vbox = wx.BoxSizer(wx.VERTICAL)
	653	box_description= wx.StaticBox(self, -1,str("Batch Outputs"))
	654	hint_sizer = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	655	selection_sizer = wx.GridBagSizer(5,5)
	656	button_sizer = wx.BoxSizer(wx.HORIZONTAL)
[caf3a08f]	657	text = "Open with %s" % self.parent.application_name
[73197d0]	658	self.local_app_selected = wx.RadioButton(self, -1, text,
	659	style=wx.RB_GROUP)
	660	self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
	661	id=self.local_app_selected.GetId())
	662	text = "Open with Excel"
	663	self.external_app_selected = wx.RadioButton(self, -1, text)
	664	self.Bind(wx.EVT_RADIOBUTTON, self.onselect,
	665	id=self.external_app_selected.GetId())
[caf3a08f]	666	text = "Save to File"
[73197d0]	667	self.save_to_file = wx.CheckBox(self, -1, text)
	668	self.Bind(wx.EVT_CHECKBOX, self.onselect,
	669	id=self.save_to_file.GetId())
	670	self.local_app_selected.SetValue(True)
	671	self.external_app_selected.SetValue(False)
	672	self.save_to_file.SetValue(False)
[caf3a08f]	673	button_close = wx.Button(self, wx.ID_OK, "Close")
	674	button_OK = wx.Button(self, wx.ID_CANCEL, "Apply")
[73197d0]	675	button_OK.SetFocus()
	676	hint = ""
	677	hint_sizer.Add(wx.StaticText(self, -1, hint))
	678	hint_sizer.Add(selection_sizer)
	679	#draw area containing radio buttons
	680	ix = 0
	681	iy = 0
	682	selection_sizer.Add(self.local_app_selected, (iy, ix),
	683	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	684	iy += 1
	685	selection_sizer.Add(self.external_app_selected, (iy, ix),
	686	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	687	iy += 1
	688	selection_sizer.Add(self.save_to_file, (iy, ix),
	689	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	690	#contruction the sizer contaning button
	691	button_sizer.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[caf3a08f]	692
	693	button_sizer.Add(button_close, 0,
	694	wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[73197d0]	695	button_sizer.Add(button_OK, 0,
	696	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
	697	vbox.Add(hint_sizer, 0, wx.EXPAND\|wx.ALL, 10)
	698	vbox.Add(wx.StaticLine(self, -1), 0, wx.EXPAND, 0)
	699	vbox.Add(button_sizer, 0 , wx.TOP\|wx.BOTTOM, 10)
	700	self.SetSizer(vbox)
	701
	702	def onselect(self, event=None):
	703	"""
	704	Receive event and display data into third party application
	705	or save data to file.
	706
	707	"""
	708	if self.save_to_file.GetValue():
[850525c]	709	reader, ext = os.path.splitext(self.file_name)
	710	path = None
	711	location = os.getcwd()
	712	if self.parent is not None:
	713	location = self.parent._default_save_location
	714	dlg = wx.FileDialog(self, "Save Project file",
	715	location, self.file_name, ext, wx.SAVE)
	716	path = None
	717	if dlg.ShowModal() == wx.ID_OK:
	718	path = dlg.GetPath()
	719	if self.parent is not None:
	720	self.parent._default_save_location = os.path.dirname(path)
	721	dlg.Destroy()
	722	if path != None:
	723	if self.parent is not None and self.data is not None:
	724	self.parent.write_batch_output(data=self.data,
	725	file_name=path,
	726	details=self.details)
	727
	728	if self.local_app_selected.GetValue():
[73197d0]	729	self.flag = 1
	730	else:
	731	self.flag = 2
	732	return self.flag
	733
[850525c]	734	def save_file(self):
	735	"""
	736	Save inot file
	737	"""
	738
	739
[73197d0]	740
	741
[24adb89]	742	if __name__ == "__main__":
	743	app = wx.App()
	744
	745	try:
	746	data = {}
	747	j = 0
	748	for i in range(4):
	749	j += 1
	750	data["index"+str(i)] = [i/j, i*j, i, i+j]
	751
[9c8f3ad]	752	frame = GridFrame(data=data)
[24adb89]	753	frame.Show(True)
	754	except:
	755	print sys.exc_value
	756
	757	app.MainLoop()

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

SasView

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

Download in other formats: