ShapeParameters.py @ 686fd10

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 686fd10 was 79492222, checked in by krzywon, 10 years ago
Changed the file and folder names to remove all SANS references.
Property mode set to `100644`
File size: 20.9 KB

Rev	Line
[eaeb13e]	1	"""
	2	This software was developed by the University of Tennessee as part of the
	3	Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
	4	project funded by the US National Science Foundation.
	5
	6	See the license text in license.txt
	7
	8	copyright 2009, University of Tennessee
	9	"""
	10	import wx
	11	import sys
	12	import wx.lib.newevent
	13	from copy import deepcopy
	14	import SimCanvas
	15
	16	(CreateShapeEvent, EVT_ADD_SHAPE) = wx.lib.newevent.NewEvent()
	17	(EditShapeEvent, EVT_EDIT_SHAPE) = wx.lib.newevent.NewEvent()
[6ef7ac5a]	18	(DelShapeEvent, EVT_DEL_SHAPE) = wx.lib.newevent.NewEvent()
	19	(QRangeEvent, EVT_Q_RANGE) = wx.lib.newevent.NewEvent()
	20	(PtDensityEvent, EVT_PT_DENSITY) = wx.lib.newevent.NewEvent()
[eaeb13e]	21
	22	class ShapeParameterPanel(wx.Panel):
	23	#TODO: show units
	24	#TODO: order parameters properly
	25	CENTER_PANE = True
	26
[6ef7ac5a]	27	def __init__(self, parent, q_min=0.001, q_max=0.5, q_npts=10, pt_density=0.1, args, *kwargs):
[eaeb13e]	28	wx.Panel.__init__(self, parent, args, *kwargs)
	29
	30	self.window_name = "ShapeParams"
	31	self.window_caption = "Shape parameter"
	32
	33
	34	self.params = {}
	35	self.parent = parent
	36	self.type = None
	37	self.listeners = []
	38	self.parameters = []
	39	self.bck = wx.GridBagSizer(5,5)
	40	self.SetSizer(self.bck)
	41
	42	# position and orientation ctrl
	43	self.xctrl = None
	44	self.yctrl = None
	45	self.zctrl = None
	46	self.actrl = None
	47	self.bctrl = None
	48	self.cctrl = None
	49
	50	# Sizer to hold the shape parameters
	51	self.shape_sizer = wx.GridBagSizer(5,5)
	52
[cb78690]	53	ny = 0
[eaeb13e]	54	title = wx.StaticText(self, -1, "[Temporary form]", style=wx.ALIGN_LEFT)
[cb78690]	55	self.bck.Add(title, (ny,0), (1,2), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border=15)
	56
	57	# Shape list
	58	shape_text = wx.StaticText(self, -1, "Geometric shape", style=wx.ALIGN_LEFT)
	59	self.shape_list = SimCanvas.getShapes()
	60	value_list = []
	61	for item in self.shape_list:
	62	value_list.append(item['name'])
	63
	64	self.model_combo = wx.ComboBox(self, -1, value=value_list[0], choices=value_list, style=wx.CB_READONLY)
[6ef7ac5a]	65	self.model_combo.SetToolTip(wx.ToolTip("Select a geometric shape from the drop-down list"))
[eaeb13e]	66
[cb78690]	67	ny+=1
	68	self.bck.Add(shape_text, (ny,0), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border=15)
	69	self.bck.Add(self.model_combo, (ny,1), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border=15)
	70	wx.EVT_COMBOBOX(self.model_combo,-1, self._on_select_shape)
	71
	72	# Placeholder for parameter form
	73	ny+=1
	74	self.bck.Add(self.shape_sizer, (ny,0), (1,2), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border=0)
[eaeb13e]	75
	76	# Point density control
	77	point_density_text = wx.StaticText(self, -1, "Point density", style=wx.ALIGN_LEFT)
[cb78690]	78	ny+=1
	79	self.bck.Add(point_density_text, (ny,0), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
[eaeb13e]	80	self.point_density_ctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
[6ef7ac5a]	81	self.point_density_ctrl.SetValue(str(pt_density))
	82	self.point_density_ctrl.SetToolTip(wx.ToolTip("Enter the number of real-space points per Angstrom cube"))
	83	self.point_density_ctrl.Bind(wx.EVT_TEXT_ENTER, self._on_density_changed)
	84	self.point_density_ctrl.Bind(wx.EVT_KILL_FOCUS, self._on_density_changed)
[cb78690]	85	self.bck.Add(self.point_density_ctrl, (ny,1), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
[eaeb13e]	86
[cb78690]	87	# Q range
	88	q_min_text = wx.StaticText(self, -1, "Q min", style=wx.ALIGN_LEFT)
	89	ny+=1
	90	self.bck.Add(q_min_text, (ny,0), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	91	self.q_min_ctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
[6ef7ac5a]	92	self.q_min_ctrl.SetValue(str(q_min))
	93	self.q_min_ctrl.SetToolTip(wx.ToolTip("Enter the minimum Q value to be simulated"))
	94	self.q_min_ctrl.Bind(wx.EVT_TEXT_ENTER, self._on_q_range_changed)
	95	self.q_min_ctrl.Bind(wx.EVT_KILL_FOCUS, self._on_q_range_changed)
[cb78690]	96	self.bck.Add(self.q_min_ctrl, (ny,1), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
[eaeb13e]	97
[cb78690]	98	q_max_text = wx.StaticText(self, -1, "Q max", style=wx.ALIGN_LEFT)
	99	self.bck.Add(q_max_text, (ny,2), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	100	self.q_max_ctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
[6ef7ac5a]	101	self.q_max_ctrl.SetValue(str(q_max))
	102	self.q_min_ctrl.SetToolTip(wx.ToolTip("Enter the maximum Q value to be simulated"))
	103	self.q_max_ctrl.Bind(wx.EVT_TEXT_ENTER, self._on_q_range_changed)
	104	self.q_max_ctrl.Bind(wx.EVT_KILL_FOCUS, self._on_q_range_changed)
[cb78690]	105	self.bck.Add(self.q_max_ctrl, (ny,3), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	106
[6ef7ac5a]	107	q_npts_text = wx.StaticText(self, -1, "No. of Q pts", style=wx.ALIGN_LEFT)
	108	self.bck.Add(q_npts_text, (ny,4), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	109	self.q_npts_ctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	110	self.q_npts_ctrl.SetValue(str(q_npts))
	111	self.q_min_ctrl.SetToolTip(wx.ToolTip("Enter the number of Q points to be simulated"))
	112	self.q_npts_ctrl.Bind(wx.EVT_TEXT_ENTER, self._on_q_range_changed)
	113	self.q_npts_ctrl.Bind(wx.EVT_KILL_FOCUS, self._on_q_range_changed)
	114	self.bck.Add(self.q_npts_ctrl, (ny,5), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	115
[cb78690]	116	# Shape List
	117	# Internal counter of shapes in the listbox
	118	self.counter = 0
	119	# Buffer filled flag
	120	self.buffer_filled = False
	121	# Save current flag
	122	self.current_saved = False
	123
	124	id = wx.NewId()
	125	shape_listbox_text = wx.StaticText(self, -1, "List of shapes on 3D canvas", style=wx.ALIGN_LEFT)
	126	self.shape_listbox = wx.ListBox(self, id, wx.DefaultPosition, (295, 200),
	127	[], wx.LB_SINGLE \| wx.LB_HSCROLL)
	128
	129	# Listen to history events
	130	self.parent.Bind(EVT_ADD_SHAPE, self._on_add_shape_to_listbox)
	131	self.shape_listbox.Bind(wx.EVT_LISTBOX, self._on_select_from_listbox, id=id)
	132	self.shape_listbox.Bind(wx.EVT_CONTEXT_MENU, self._on_listbox_context_menu, id=id)
	133
	134	ny+=1
	135	self.bck.Add(shape_listbox_text, (ny,0), (1,2), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	136	ny+=1
	137	self.bck.Add(self.shape_listbox, (ny,0), (1,5), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
[eaeb13e]	138
	139	self.current_shape = None
	140
[cb78690]	141	# Default shape
	142	if type(self.shape_list)==list and len(self.shape_list)>0:
	143	shape = SimCanvas.getShapeClassByName(self.shape_list[0]['name'])()
	144	self.editShape(shape)
	145
	146	def _on_select_shape(self, event):
	147	shape = SimCanvas.getShapeClassByName(self.shape_list[event.GetEventObject().GetSelection()]['name'])()
	148	self.editShape(shape)
[eaeb13e]	149
[6ef7ac5a]	150	def _on_density_changed(self, event):
[eaeb13e]	151	"""
	152	Process event that might mean a change of the simulation point density
	153	"""
[6ef7ac5a]	154	npts = self._readCtrlFloat(self.point_density_ctrl)
	155	if npts is not None:
	156	event = PtDensityEvent(npts=npts)
	157	wx.PostEvent(self.parent, event)
[eaeb13e]	158
[6ef7ac5a]	159	def _on_q_range_changed(self, event):
[eaeb13e]	160	"""
[6ef7ac5a]	161	Process event that might mean a change in Q range
	162	@param event: EVT_Q_RANGE event
[eaeb13e]	163	"""
[6ef7ac5a]	164	q_min = self._readCtrlFloat(self.q_min_ctrl)
	165	q_max = self._readCtrlFloat(self.q_max_ctrl)
	166	npts = self._readCtrlInt(self.q_npts_ctrl)
	167	if q_min is not None or q_max is not None or npts is not None:
	168	event = QRangeEvent(q_min=q_min, q_max=q_max, npts=npts)
	169	wx.PostEvent(self.parent, event)
	170
[eaeb13e]	171	def _onEditShape(self, evt):
	172	"""
	173	Process an EVT_EDIT_SHAPE event
	174	@param evt: EVT_EDIT_SHAPE object
	175	"""
	176	evt.Skip()
	177	self.editShape(evt.shape, False)
	178
	179	def editShape(self, shape=None, new=True):
	180	"""
	181	Rebuild the panel
	182	"""
	183	self.current_shape = shape
	184	self.shape_sizer.Clear(True)
	185	#self.type = type
	186
	187	if shape==None:
	188	title = wx.StaticText(self, -1, "Use menu to add a shapes", style=wx.ALIGN_LEFT)
	189	self.shape_sizer.Add(title, (0,0), (1,2), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border=15)
	190
	191	else:
	192	if new==False:
	193	title = wx.StaticText(self, -1, "Edit shape parameters", style=wx.ALIGN_LEFT)
	194	else:
	195	title = wx.StaticText(self, -1, "Create shape from parameters", style=wx.ALIGN_LEFT)
	196
	197	self.shape_sizer.Add(title, (0,0), (1,2), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border=15)
	198
	199	n = 1
	200	self.parameters = []
	201	keys = shape.params.keys()
	202	keys.sort()
	203
	204	for item in keys:
	205	if item in ['contrast', 'order']:
	206	continue
	207	n += 1
	208	text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
	209	self.shape_sizer.Add(text, (n-1,0), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	210	ctl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	211
	212
	213	ctl.SetValue(str(shape.params[item]))
	214	self.Bind(wx.EVT_TEXT_ENTER, self.onTextEnter)
	215	ctl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	216	self.parameters.append([item, ctl])
	217	self.shape_sizer.Add(ctl, (n-1,1), flag=wx.TOP\|wx.BOTTOM, border = 0)
	218
	219	# Units
	220	units = wx.StaticText(self, -1, shape.details[item], style=wx.ALIGN_LEFT)
	221	self.shape_sizer.Add(units, (n-1,2), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	222
	223	for item in ['contrast', 'order']:
	224	n += 1
	225	text = wx.StaticText(self, -1, item, style=wx.ALIGN_LEFT)
	226	self.shape_sizer.Add(text, (n-1,0), flag = wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	227	ctl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	228
	229
	230	ctl.SetValue(str(shape.params[item]))
	231	self.Bind(wx.EVT_TEXT_ENTER, self.onTextEnter)
	232	ctl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	233	self.parameters.append([item, ctl])
	234	self.shape_sizer.Add(ctl, (n-1,1), flag=wx.TOP\|wx.BOTTOM, border = 0)
	235
	236	# Units
	237	units = wx.StaticText(self, -1, shape.details[item], style=wx.ALIGN_LEFT)
	238	self.shape_sizer.Add(units, (n-1,2), flag=wx.LEFT\|wx.ALIGN_CENTER_VERTICAL, border = 15)
	239
	240
	241	# Add position
	242	n += 1
	243
	244	pos_sizer = wx.GridBagSizer(0,0)
	245
	246	text = wx.StaticText(self, -1, 'x', style=wx.ALIGN_LEFT)
	247	pos_sizer.Add(text, (0,0), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	248	text = wx.StaticText(self, -1, 'y', style=wx.ALIGN_LEFT)
	249	pos_sizer.Add(text, (0,1), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	250	text = wx.StaticText(self, -1, 'z', style=wx.ALIGN_LEFT)
	251	pos_sizer.Add(text, (0,2), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	252
	253	self.xctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	254	self.xctrl.SetValue(str(shape.x))
	255	self.xctrl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	256	pos_sizer.Add(self.xctrl, (1,0), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	257	self.yctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	258	self.yctrl.SetValue(str(shape.y))
	259	self.yctrl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	260	pos_sizer.Add(self.yctrl, (1,1), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	261	self.zctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	262	self.zctrl.SetValue(str(shape.z))
	263	self.zctrl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	264	pos_sizer.Add(self.zctrl, (1,2), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	265
	266	self.shape_sizer.Add(pos_sizer, (n-1, 0), (1,3), flag=wx.LEFT\|wx.ALIGN_CENTER)
	267
	268	# Add orientation
	269	n += 1
	270
	271	pos_sizer = wx.GridBagSizer(0,0)
	272
	273	text = wx.StaticText(self, -1, 'theta_x', style=wx.ALIGN_LEFT)
	274	pos_sizer.Add(text, (0,0), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	275	text = wx.StaticText(self, -1, 'theta_y', style=wx.ALIGN_LEFT)
	276	pos_sizer.Add(text, (0,1), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	277	text = wx.StaticText(self, -1, 'theta_z', style=wx.ALIGN_LEFT)
	278	pos_sizer.Add(text, (0,2), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	279
	280	self.actrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	281	self.actrl.SetValue(str(shape.theta_x))
	282	self.actrl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	283	pos_sizer.Add(self.actrl, (1,0), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	284	self.bctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	285	self.bctrl.SetValue(str(shape.theta_y))
	286	self.bctrl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	287	pos_sizer.Add(self.bctrl, (1,1), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	288	self.cctrl = wx.TextCtrl(self, -1, size=(40,20), style=wx.TE_PROCESS_ENTER)
	289	self.cctrl.SetValue(str(shape.theta_z))
	290	self.cctrl.Bind(wx.EVT_KILL_FOCUS, self.onTextEnter)
	291	pos_sizer.Add(self.cctrl, (1,2), (1,1), flag = wx.LEFT\|wx.ALIGN_CENTER, border = 15)
	292
	293	self.shape_sizer.Add(pos_sizer, (n-1, 0), (1,3), flag=wx.LEFT\|wx.ALIGN_CENTER)
	294
	295
	296	# Add a button to create the new shape on the canvas
	297	n += 1
	298	if new==True:
	299	create = wx.Button(self, 1,"Create")
	300	self.shape_sizer.Add(create, (n-1,1), (1,2), flag=wx.ALIGN_RIGHT\|wx.ALL, border = 15)
	301	self.Bind(wx.EVT_BUTTON, self._onCreate, id = 1)
	302	else:
	303	create = wx.Button(self, 1,"Apply")
	304	self.shape_sizer.Add(create, (n-1,1), (1,2), flag=wx.ALIGN_RIGHT\|wx.ALL, border = 15)
	305	self.Bind(wx.EVT_BUTTON, self._onEdited, id = 1)
	306
	307
	308
	309	self.shape_sizer.Layout()
	310	self.shape_sizer.Fit(self)
[cb78690]	311	try:
	312	self.parent.GetSizer().Layout()
	313	except:
	314	print "TODO: move the Layout call of editShape up to the caller"
[eaeb13e]	315
	316	def _readCtrlFloat(self, ctrl):
[6ef7ac5a]	317	"""
	318	Parses a TextCtrl for a float value.
	319	Returns None is the value hasn't changed or the value is not a float.
	320	The control background turns pink if the value is not a float.
	321
	322	@param ctrl: TextCtrl object
	323	"""
[eaeb13e]	324	if ctrl.IsModified():
[6ef7ac5a]	325	ctrl.SetModified(False)
[eaeb13e]	326	str_value = ctrl.GetValue().lstrip().rstrip()
	327	try:
	328	flt_value = float(str_value)
	329	ctrl.SetBackgroundColour(
	330	wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
	331	ctrl.Refresh()
	332	return flt_value
	333	except:
	334	ctrl.SetBackgroundColour("pink")
	335	ctrl.Refresh()
	336	return None
	337
[6ef7ac5a]	338	def _readCtrlInt(self, ctrl):
	339	"""
	340	Parses a TextCtrl for a float value.
	341	Returns None is the value hasn't changed or the value is not a float.
	342	The control background turns pink if the value is not a float.
	343
	344	@param ctrl: TextCtrl object
	345	"""
	346	if ctrl.IsModified():
	347	ctrl.SetModified(False)
	348	str_value = ctrl.GetValue().lstrip().rstrip()
	349	try:
	350	int_value = int(str_value)
	351	ctrl.SetBackgroundColour(
	352	wx.SystemSettings_GetColour(wx.SYS_COLOUR_WINDOW))
	353	ctrl.Refresh()
	354	return int_value
	355	except:
	356	ctrl.SetBackgroundColour("pink")
	357	ctrl.Refresh()
	358	return None
	359
[eaeb13e]	360	def _parseCtrl(self):
	361	try:
	362	# Position
	363	tmp = self._readCtrlFloat(self.xctrl)
	364	if not tmp==None:
	365	self.current_shape.x = tmp
	366
	367	tmp = self._readCtrlFloat(self.yctrl)
	368	if not tmp==None:
	369	self.current_shape.y = tmp
	370
	371	tmp = self._readCtrlFloat(self.zctrl)
	372	if not tmp==None:
	373	self.current_shape.z = tmp
	374
	375	# Orientation
	376	tmp = self._readCtrlFloat(self.actrl)
	377	if not tmp==None:
	378	self.current_shape.theta_x = tmp
	379
	380	tmp = self._readCtrlFloat(self.bctrl)
	381	if not tmp==None:
	382	self.current_shape.theta_y = tmp
	383
	384	tmp = self._readCtrlFloat(self.cctrl)
	385	if not tmp==None:
	386	self.current_shape.theta_z = tmp
	387
	388	# Parameters
	389	for item in self.parameters:
	390	tmp = self._readCtrlFloat(item[1])
	391	if not tmp==None:
	392	self.current_shape.params[item[0]] = tmp
	393	except:
	394	print "Could not create"
	395	print sys.exc_value
	396
	397	def _onCreate(self, evt):
	398	"""
	399	Create a new shape with the parameters given by the user
	400	"""
	401	self._parseCtrl()
	402	event = CreateShapeEvent(shape=self.current_shape, new=True)
	403	wx.PostEvent(self.parent, event)
	404	self.editShape(self.current_shape, False)
	405
	406
	407	def _onEdited(self, evt):
	408	self._parseCtrl()
	409	event = CreateShapeEvent(shape=self.current_shape, new=False)
	410	wx.PostEvent(self.parent, event)
	411	self.editShape(self.current_shape, False)
	412
	413	def onTextEnter(self, evt):
	414	"""
	415	Read text field to check values
	416	"""
	417	self._readCtrlFloat(self.xctrl)
	418	self._readCtrlFloat(self.yctrl)
	419	self._readCtrlFloat(self.zctrl)
	420	self._readCtrlFloat(self.actrl)
	421	self._readCtrlFloat(self.bctrl)
	422	self._readCtrlFloat(self.cctrl)
	423	for item in self.parameters:
	424	self._readCtrlFloat(item[1])
	425
[cb78690]	426	#-- Methods to support list of shapes --
	427	def _on_add_shape_to_listbox(self, event):
[eaeb13e]	428	"""
[cb78690]	429	Process a new shape
	430	@param event: EVT_ADD_SHAPE event
[eaeb13e]	431	"""
[cb78690]	432	event.Skip()
	433	if event.new:
	434	self.counter += 1
	435	self.shape_listbox.Insert("%d: %s" % (self.counter, event.shape.name),
	436	self.shape_listbox.GetCount(), event.shape)
[eaeb13e]	437
[cb78690]	438	def _on_select_from_listbox(self, event):
[eaeb13e]	439	"""
[cb78690]	440	Process item selection events
[eaeb13e]	441	"""
[cb78690]	442	index = event.GetSelection()
	443	view_name = self.shape_listbox.GetString(index)
	444	self.editShape(shape=event.GetClientData(), new=False)
	445	# The following event is bound to the SimPanel to highlight the shape in blue
	446	#TODO: how come it doesn't work?
	447	wx.PostEvent(self.parent, EditShapeEvent(shape=event.GetClientData()))
	448	#TODO: select the shape from the drop down
[eaeb13e]	449
[cb78690]	450	def _on_listbox_context_menu(self, event):
[eaeb13e]	451	"""
	452	Popup context menu event
	453	"""
[cb78690]	454	# Create context menu
[eaeb13e]	455	popupmenu = wx.Menu()
[cb78690]	456
	457	# Create an item to delete the selected shape from the canvas
	458	id = wx.NewId()
	459	popupmenu.Append(id, "&Remove Selected")
	460	wx.EVT_MENU(self, id, self._on_remove_shape_from_listbox)
	461
	462	# Create an item to rename a shape to a more user friendly name
	463	#id = wx.NewId()
[eaeb13e]	464	#popupmenu.Append(102, "&Rename Selected")
	465	#wx.EVT_MENU(self, 102, self.onRename)
	466
	467	pos = event.GetPosition()
	468	pos = self.ScreenToClient(pos)
[cb78690]	469	self.PopupMenu(popupmenu, pos)
	470
	471	def _on_remove_shape_from_listbox(self, ev):
[eaeb13e]	472	"""
[cb78690]	473	Remove an item
[eaeb13e]	474	"""
[cb78690]	475	indices = self.shape_listbox.GetSelections()
	476	if len(indices)>0:
	477	name = self.shape_listbox.GetClientData(indices[0]).name
	478	self.shape_listbox.Delete(indices[0])
	479	wx.PostEvent(self.parent, DelShapeEvent(id = name))
	480
	481	def _on_rename_shape_from_listbox(self, ev):
[eaeb13e]	482	"""
[cb78690]	483	Rename an item
[eaeb13e]	484	"""
[cb78690]	485	indices = self.shape_listbox.GetSelections()
	486	if len(indices)>0:
	487	print "NOT YET IMPLMENTED"
	488	print "renaming", self.shape_listbox.GetString(indices[0])
[eaeb13e]	489

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

SasView

source: sasview/src/sas/perspectives/simulation/ShapeParameters.py @ 686fd10

Download in other formats: