inversion_panel.py @ d65a00a

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 d65a00a was b35d3d1, checked in by Jae Cho <jhjcho@…>, 14 years ago
save state file: working, but more work to do
Property mode set to `100644`
File size: 39.0 KB

Rev	Line
[f3d51f6]	1	#!/usr/bin/env python
	2
	3	# version
	4	__id__ = "$Id: aboutdialog.py 1193 2007-05-03 17:29:59Z dmitriy $"
	5	__revision__ = "$Revision: 1193 $"
	6
	7	import wx
[5d98370]	8	import os
[ceaf16e]	9	import sys
	10	import logging
[9ff861b]	11	from wx.lib.scrolledpanel import ScrolledPanel
[f3d51f6]	12	from sans.guicomm.events import StatusEvent
[5d98370]	13	from inversion_state import InversionState
[9ff861b]	14	from pr_widgets import PrTextCtrl, DataFileTextCtrl, OutputTextCtrl
[f3d51f6]	15
	16
	17
[9ff861b]	18	class InversionControl(ScrolledPanel):
[7116b6e0]	19	"""
	20	"""
[f3d51f6]	21	window_name = 'pr_control'
	22	window_caption = "P(r) control panel"
	23	CENTER_PANE = True
	24
	25	# Figure of merit parameters [default]
[dfb58f8]	26
	27	## Oscillation parameters (sin function = 1.1)
[f3d51f6]	28	oscillation_max = 1.5
	29
[aa4b8379]	30	def __init__(self, parent, id = -1, plots = None, standalone=False, **kwargs):
[7116b6e0]	31	"""
	32	"""
[9ff861b]	33	ScrolledPanel.__init__(self, parent, id = id, **kwargs)
	34	self.SetupScrolling()
[f3d51f6]	35
	36	self.plots = plots
	37	self.radio_buttons = {}
[c405c69]	38	self.parent = parent
[f3d51f6]	39
	40	## Data file TextCtrl
[a4bd2ac]	41	self.data_file = None
	42	self.plot_data = None
	43	self.nfunc_ctl = None
	44	self.alpha_ctl = None
	45	self.dmax_ctl = None
	46	self.time_ctl = None
	47	self.chi2_ctl = None
	48	self.osc_ctl = None
[f3d51f6]	49	self.file_radio = None
	50	self.plot_radio = None
[634f1cf]	51	self.label_sugg = None
[a4bd2ac]	52	self.qmin_ctl = None
	53	self.qmax_ctl = None
[5d98370]	54	self.swidth_ctl = None
	55	self.sheight_ctl = None
[a4bd2ac]	56
	57	self.rg_ctl = None
	58	self.iq0_ctl = None
	59	self.bck_chk = None
	60	self.bck_ctl = None
[f3d51f6]	61
[dfb58f8]	62	# TextCtrl for fraction of positive P(r)
	63	self.pos_ctl = None
	64
	65	# TextCtrl for fraction of 1 sigma positive P(r)
	66	self.pos_err_ctl = None
	67
[f3d51f6]	68	## Estimates
	69	self.alpha_estimate_ctl = None
[35adaf6]	70	self.nterms_estimate_ctl = None
[f3d51f6]	71
[a00ee4c]	72	## D_max distance explorator
	73	self.distance_explorator_ctl = None
	74
[f3d51f6]	75	## Data manager
	76	self.manager = None
	77
[aa4b8379]	78	## Standalone flage
	79	self.standalone = standalone
	80
[5d98370]	81	## Default file location for save
	82	self._default_save_location = os.getcwd()
	83
[0ccd214]	84
	85	# Default width
	86	self._default_width = 350
[f3d51f6]	87	self._do_layout()
	88
	89	def __setattr__(self, name, value):
	90	"""
[7116b6e0]	91	Allow direct hooks to text boxes
[f3d51f6]	92	"""
	93	if name=='nfunc':
[5d98370]	94	self.nfunc_ctl.SetValue(str(int(value)))
[f3d51f6]	95	elif name=='d_max':
	96	self.dmax_ctl.SetValue(str(value))
	97	elif name=='alpha':
	98	self.alpha_ctl.SetValue(str(value))
	99	elif name=='chi2':
[a4bd2ac]	100	self.chi2_ctl.SetValue("%-5.2g" % value)
	101	elif name=='bck':
	102	self.bck_ctl.SetValue("%-5.2g" % value)
[634f1cf]	103	elif name=='q_min':
[a4bd2ac]	104	self.qmin_ctl.SetValue("%-5.2g" % value)
[634f1cf]	105	elif name=='q_max':
[a4bd2ac]	106	self.qmax_ctl.SetValue("%-5.2g" % value)
[f3d51f6]	107	elif name=='elapsed':
	108	self.time_ctl.SetValue("%-5.2g" % value)
[a4bd2ac]	109	elif name=='rg':
	110	self.rg_ctl.SetValue("%-5.2g" % value)
	111	elif name=='iq0':
	112	self.iq0_ctl.SetValue("%-5.2g" % value)
[f3d51f6]	113	elif name=='oscillation':
	114	self.osc_ctl.SetValue("%-5.2g" % value)
[5d98370]	115	elif name=='slit_width':
	116	self.swidth_ctl.SetValue("%-5.2g" % value)
	117	elif name=='slit_height':
	118	self.sheight_ctl.SetValue("%-5.2g" % value)
[dfb58f8]	119	elif name=='positive':
	120	self.pos_ctl.SetValue("%-5.2g" % value)
	121	elif name=='pos_err':
	122	self.pos_err_ctl.SetValue("%-5.2g" % value)
[f3d51f6]	123	elif name=='alpha_estimate':
[634f1cf]	124	self.alpha_estimate_ctl.SetToolTipString("Click to accept value.")
	125	self.alpha_estimate_ctl.Enable(True)
	126	self.alpha_estimate_ctl.SetLabel("%-3.1g" % value)
	127	#self.alpha_estimate_ctl.Show()
	128	#self.label_sugg.Show()
[35adaf6]	129	elif name=='nterms_estimate':
	130	self.nterms_estimate_ctl.SetToolTipString("Click to accept value.")
	131	self.nterms_estimate_ctl.Enable(True)
	132	self.nterms_estimate_ctl.SetLabel("%-g" % value)
[f3d51f6]	133	elif name=='plotname':
[0d88a09]	134	self.plot_data.SetValue(str(value))
	135	self._on_pars_changed(None)
[aa4b8379]	136	elif name=='datafile':
[0d88a09]	137	self.plot_data.SetValue(str(value))
	138	self._on_pars_changed(None)
[f3d51f6]	139	else:
	140	wx.Panel.__setattr__(self, name, value)
	141
	142	def __getattr__(self, name):
	143	"""
[7116b6e0]	144	Allow direct hooks to text boxes
[f3d51f6]	145	"""
	146	if name=='nfunc':
[3fd1ebc]	147	try:
	148	return int(self.nfunc_ctl.GetValue())
	149	except:
	150	return -1
[f3d51f6]	151	elif name=='d_max':
[3fd1ebc]	152	try:
	153	return self.dmax_ctl.GetValue()
	154	except:
	155	return -1.0
[f3d51f6]	156	elif name=='alpha':
[3fd1ebc]	157	try:
	158	return self.alpha_ctl.GetValue()
	159	except:
	160	return -1.0
[f3d51f6]	161	elif name=='chi2':
[3fd1ebc]	162	try:
	163	return float(self.chi2_ctl.GetValue())
	164	except:
[5d98370]	165	return None
[a4bd2ac]	166	elif name=='bck':
	167	try:
	168	return float(self.bck_ctl.GetValue())
	169	except:
[5d98370]	170	return None
[634f1cf]	171	elif name=='q_min':
[3fd1ebc]	172	try:
	173	return float(self.qmin_ctl.GetValue())
	174	except:
	175	return 0.0
[634f1cf]	176	elif name=='q_max':
[3fd1ebc]	177	try:
	178	return float(self.qmax_ctl.GetValue())
	179	except:
	180	return 0.0
[f3d51f6]	181	elif name=='elapsed':
[3fd1ebc]	182	try:
	183	return float(self.time_ctl.GetValue())
	184	except:
[5d98370]	185	return None
[a4bd2ac]	186	elif name=='rg':
	187	try:
	188	return float(self.rg_ctl.GetValue())
	189	except:
[5d98370]	190	return None
[a4bd2ac]	191	elif name=='iq0':
	192	try:
	193	return float(self.iq0_ctl.GetValue())
	194	except:
[5d98370]	195	return None
[f3d51f6]	196	elif name=='oscillation':
[3fd1ebc]	197	try:
	198	return float(self.osc_ctl.GetValue())
	199	except:
[5d98370]	200	return None
	201	elif name=='slit_width':
	202	try:
	203	return float(self.swidth_ctl.GetValue())
	204	except:
	205	return None
	206	elif name=='slit_height':
	207	try:
	208	return float(self.sheight_ctl.GetValue())
	209	except:
	210	return None
[dfb58f8]	211	elif name=='pos':
[3fd1ebc]	212	try:
	213	return float(self.pos_ctl.GetValue())
	214	except:
[5d98370]	215	return None
[dfb58f8]	216	elif name=='pos_err':
[5d98370]	217	try:
	218	return float(self.pos_err_ctl.GetValue())
	219	except:
	220	return None
[f3d51f6]	221	elif name=='alpha_estimate':
[5d98370]	222	try:
	223	return float(self.alpha_estimate_ctl.GetLabel())
	224	except:
	225	return None
[35adaf6]	226	elif name=='nterms_estimate':
[5d98370]	227	try:
	228	return int(self.nterms_estimate_ctl.GetLabel())
	229	except:
	230	return None
[f3d51f6]	231	elif name=='plotname':
[0d88a09]	232	return self.plot_data.GetValue()
[aa4b8379]	233	elif name=='datafile':
[0d88a09]	234	return self.plot_data.GetValue()
[f3d51f6]	235	else:
[a00ee4c]	236	return wx.Panel.__getattribute__(self, name)
[f3d51f6]	237
[5d98370]	238	def _save_state(self, evt=None):
	239	"""
[7116b6e0]	240	Method used to create a memento of the current state
[5d98370]	241
[7116b6e0]	242	:return: state object
[5d98370]	243	"""
	244	# Ask the user the location of the file to write to.
	245	path = None
	246	dlg = wx.FileDialog(self, "Choose a file", self._default_save_location, "", "*.prv", wx.SAVE)
	247	if dlg.ShowModal() == wx.ID_OK:
	248	path = dlg.GetPath()
	249	self._default_save_location = os.path.dirname(path)
[0d88a09]	250	else:
	251	return None
	252
[5d98370]	253	dlg.Destroy()
[b35d3d1]	254
	255	state = self.get_state()
	256
	257	self.manager.save_data(filepath=path, prstate=state)
	258
	259	return state
	260
	261	def get_state(self):
	262	"""
	263	Get the current state
	264
	265	: return: state object
	266	"""
[5d98370]	267	# Construct the state object
	268	state = InversionState()
	269
	270	# Read the panel's parameters
	271	flag, alpha, dmax, nfunc, qmin, \
	272	qmax, height, width = self._read_pars()
	273
	274	state.nfunc = nfunc
	275	state.d_max = dmax
	276	state.alpha = alpha
	277	state.qmin = qmin
	278	state.qmax = qmax
	279	state.width = width
	280	state.height = height
	281
	282	# Data file
[0d88a09]	283	state.file = self.plot_data.GetValue()
[5d98370]	284
	285	# Background evaluation checkbox
	286	state.estimate_bck = self.bck_chk.IsChecked()
	287
	288	# Estimates
	289	state.nterms_estimate = self.nterms_estimate
	290	state.alpha_estimate = self.alpha_estimate
	291
	292	# Read the output values
	293	state.chi2 = self.chi2
	294	state.elapsed = self.elapsed
	295	state.osc = self.oscillation
	296	state.pos = self.pos
	297	state.pos_err = self.pos_err
	298	state.rg = self.rg
	299	state.iq0 = self.iq0
	300	state.bck = self.bck
[fde249c]	301
[5d98370]	302	return state
	303
	304	def set_state(self, state):
	305	"""
[7116b6e0]	306	Set the state of the panel and inversion problem to
	307	the state passed as a parameter.
	308	Execute the inversion immediately after filling the
	309	controls.
	310
	311	:param state: InversionState object
[5d98370]	312	"""
[fde249c]	313	if state.nfunc is not None:
	314	self.nfunc = state.nfunc
	315	if state.d_max is not None:
	316	self.d_max = state.d_max
	317	if state.alpha is not None:
	318	self.alpha = state.alpha
	319	if state.qmin is not None:
	320	self.q_min = state.qmin
	321	if state.qmax is not None:
	322	self.q_max = state.qmax
	323	if state.width is not None:
	324	self.slit_width = state.width
	325	if state.height is not None:
	326	self.slit_height = state.height
[5d98370]	327
	328	# Data file
[0d88a09]	329	self.plot_data.SetValue(str(state.file))
[5d98370]	330
	331	# Background evaluation checkbox
	332	self.bck_chk.SetValue(state.estimate_bck)
	333
	334	# Estimates
[fde249c]	335	if state.nterms_estimate is not None:
	336	self.nterms_estimate = state.nterms_estimate
	337	if state.alpha_estimate is not None:
	338	self.alpha_estimate = state.alpha_estimate
[5d98370]	339
	340
	341	# Read the output values
[fde249c]	342	if state.chi2 is not None:
	343	self.chi2 = state.chi2
	344	if state.elapsed is not None:
	345	self.elapsed = state.elapsed
	346	if state.osc is not None:
	347	self.oscillation = state.osc
	348	if state.pos is not None:
	349	self.positive = state.pos
	350	if state.pos_err is not None:
	351	self.pos_err = state.pos_err
	352	if state.rg is not None:
	353	self.rg = state.rg
	354	if state.iq0 is not None:
	355	self.iq0 = state.iq0
	356	if state.bck is not None:
	357	self.bck = state.bck
[5d98370]	358
[0d88a09]	359	# Perform inversion
	360	if self.standalone == False:
[fde249c]	361	self._on_invert(None)
[5d98370]	362
[f3d51f6]	363	def set_manager(self, manager):
	364	self.manager = manager
	365
	366	def _do_layout(self):
[4318af7f]	367	vbox = wx.GridBagSizer(0,0)
	368	iy_vb = 0
[f3d51f6]	369
	370	# ----- I(q) data -----
[aa4b8379]	371	databox = wx.StaticBox(self, -1, "I(q) data source")
[f3d51f6]	372
	373	boxsizer1 = wx.StaticBoxSizer(databox, wx.VERTICAL)
[0ccd214]	374	boxsizer1.SetMinSize((self._default_width,50))
[f3d51f6]	375	pars_sizer = wx.GridBagSizer(5,5)
[0d21ac1]	376
[f3d51f6]	377	iy = 0
[d2ee6f6]	378	self.file_radio = wx.StaticText(self, -1, "Data:")
[f3d51f6]	379	pars_sizer.Add(self.file_radio, (iy,0), (1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	380
[9ff861b]	381	self.plot_data = DataFileTextCtrl(self, -1, size=(260,20))
	382
	383	pars_sizer.Add(self.plot_data, (iy,1), (1,1), wx.EXPAND\|wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 15)
[f3d51f6]	384
[a4bd2ac]	385	self.bck_chk = wx.CheckBox(self, -1, "Estimate background level")
	386	self.bck_chk.SetToolTipString("Check box to let the fit estimate the constant background level.")
	387	self.bck_chk.Bind(wx.EVT_CHECKBOX, self._on_pars_changed)
	388	iy += 1
	389	pars_sizer.Add(self.bck_chk, (iy,0), (1,2), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	390	boxsizer1.Add(pars_sizer, 0, wx.EXPAND)
[0ccd214]	391	vbox.Add(boxsizer1, (iy_vb,0), (1,1), wx.LEFT\|wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE\|wx.TOP, 5)
[a4bd2ac]	392
	393	# ----- Add slit parameters -----
[2a92852]	394	if True:
[a4bd2ac]	395	sbox = wx.StaticBox(self, -1, "Slit parameters")
	396	sboxsizer = wx.StaticBoxSizer(sbox, wx.VERTICAL)
[0ccd214]	397	sboxsizer.SetMinSize((self._default_width,20))
[a4bd2ac]	398
	399	sizer_slit = wx.GridBagSizer(5,5)
	400
[4318af7f]	401	label_sheight = wx.StaticText(self, -1, "Height", size=(40,20))
	402	label_swidth = wx.StaticText(self, -1, "Width", size=(40,20))
[a4bd2ac]	403	#label_sunits1 = wx.StaticText(self, -1, "[A^(-1)]")
[4318af7f]	404	label_sunits2 = wx.StaticText(self, -1, "[A^(-1)]", size=(55,20))
[9ff861b]	405	self.sheight_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
	406	self.swidth_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
[a4bd2ac]	407	self.sheight_ctl.SetToolTipString("Enter slit height in units of Q or leave blank.")
	408	self.swidth_ctl.SetToolTipString("Enter slit width in units of Q or leave blank.")
[2a92852]	409	#self.sheight_ctl.Bind(wx.EVT_TEXT, self._on_pars_changed)
	410	#self.swidth_ctl.Bind(wx.EVT_TEXT, self._on_pars_changed)
[a4bd2ac]	411
	412	iy = 0
[357b79b]	413	sizer_slit.Add(label_sheight, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 5)
	414	sizer_slit.Add(self.sheight_ctl, (iy,1), (1,1), wx.LEFT\|wx.EXPAND, 5)
[a4bd2ac]	415	#sizer_slit.Add(label_sunits1, (iy,2), (1,1), wx.LEFT\|wx.EXPAND, 10)
[357b79b]	416	sizer_slit.Add(label_swidth, (iy,2), (1,1), wx.LEFT\|wx.EXPAND, 5)
	417	sizer_slit.Add(self.swidth_ctl, (iy,3), (1,1), wx.LEFT\|wx.EXPAND, 5)
	418	sizer_slit.Add(label_sunits2, (iy,4), (1,1), wx.LEFT\|wx.EXPAND, 5)
[a4bd2ac]	419
	420	sboxsizer.Add(sizer_slit, wx.TOP, 15)
[4318af7f]	421	iy_vb += 1
[0ccd214]	422	vbox.Add(sboxsizer, (iy_vb,0), (1,1), wx.LEFT\|wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[a4bd2ac]	423
[357b79b]	424
	425	# ----- Q range -----
	426	qbox = wx.StaticBox(self, -1, "Q range")
	427	qboxsizer = wx.StaticBoxSizer(qbox, wx.VERTICAL)
[0ccd214]	428	qboxsizer.SetMinSize((self._default_width,20))
[357b79b]	429
	430	sizer_q = wx.GridBagSizer(5,5)
	431
[4318af7f]	432	label_qmin = wx.StaticText(self, -1, "Q min", size=(40,20))
	433	label_qmax = wx.StaticText(self, -1, "Q max", size=(40,20))
[357b79b]	434	#label_qunits1 = wx.StaticText(self, -1, "[A^(-1)]")
[4318af7f]	435	label_qunits2 = wx.StaticText(self, -1, "[A^(-1)]", size=(55,20))
[9ff861b]	436	self.qmin_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
	437	self.qmax_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
[357b79b]	438	self.qmin_ctl.SetToolTipString("Select a lower bound for Q or leave blank.")
	439	self.qmax_ctl.SetToolTipString("Select an upper bound for Q or leave blank.")
	440	self.qmin_ctl.Bind(wx.EVT_TEXT, self._on_pars_changed)
	441	self.qmax_ctl.Bind(wx.EVT_TEXT, self._on_pars_changed)
	442
	443	iy = 0
	444	sizer_q.Add(label_qmin, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 5)
	445	sizer_q.Add(self.qmin_ctl, (iy,1), (1,1), wx.LEFT\|wx.EXPAND, 5)
	446	#sizer_q.Add(label_qunits1, (iy,2), (1,1), wx.LEFT\|wx.EXPAND, 15)
	447	sizer_q.Add(label_qmax, (iy,2), (1,1), wx.LEFT\|wx.EXPAND, 5)
	448	sizer_q.Add(self.qmax_ctl, (iy,3), (1,1), wx.LEFT\|wx.EXPAND, 5)
	449	sizer_q.Add(label_qunits2, (iy,4), (1,1), wx.LEFT\|wx.EXPAND, 5)
	450	qboxsizer.Add(sizer_q, wx.TOP, 15)
[4318af7f]	451
	452	iy_vb += 1
[0ccd214]	453	vbox.Add(qboxsizer, (iy_vb,0), (1,1), wx.LEFT\|wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[357b79b]	454
	455
	456
[f3d51f6]	457
	458	# ----- Parameters -----
	459	parsbox = wx.StaticBox(self, -1, "Parameters")
	460	boxsizer2 = wx.StaticBoxSizer(parsbox, wx.VERTICAL)
[0ccd214]	461	boxsizer2.SetMinSize((self._default_width,50))
[f3d51f6]	462
[32dffae4]	463	explanation = "P(r) is found by fitting a set of base functions to I(Q). "
	464	explanation += "The minimization involves a regularization term to ensure "
[0bae207]	465	explanation += "a smooth P(r). The regularization constant gives the size of that "
[119a11d]	466	explanation += "term. The suggested value is the value above which the "
[32dffae4]	467	explanation += "output P(r) will have only one peak."
[0ccd214]	468	label_explain = wx.StaticText(self, -1, explanation, size=(280,120))
[32dffae4]	469	boxsizer2.Add(label_explain, wx.LEFT\|wx.BOTTOM, 5)
	470
	471
	472
[f3d51f6]	473	label_nfunc = wx.StaticText(self, -1, "Number of terms")
	474	label_nfunc.SetMinSize((120,20))
	475	label_alpha = wx.StaticText(self, -1, "Regularization constant")
	476	label_dmax = wx.StaticText(self, -1, "Max distance [A]")
[634f1cf]	477	self.label_sugg = wx.StaticText(self, -1, "Suggested value")
	478	#self.label_sugg.Hide()
[f3d51f6]	479
[9ff861b]	480	self.nfunc_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
[32dffae4]	481	self.nfunc_ctl.SetToolTipString("Number of terms in the expansion.")
[9ff861b]	482	self.alpha_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
[32dffae4]	483	self.alpha_ctl.SetToolTipString("Control parameter for the size of the regularization term.")
[9ff861b]	484	self.dmax_ctl = PrTextCtrl(self, -1, style=wx.TE_PROCESS_ENTER, size=(60,20))
[32dffae4]	485	self.dmax_ctl.SetToolTipString("Maximum distance between any two points in the system.")
[634f1cf]	486	id = wx.NewId()
	487	self.alpha_estimate_ctl = wx.Button(self, id, "")
	488	#self.alpha_estimate_ctl.Hide()
	489	self.Bind(wx.EVT_BUTTON, self._on_accept_alpha, id = id)
[f3d51f6]	490	self.alpha_estimate_ctl.Enable(False)
[634f1cf]	491	#self.alpha_estimate_ctl.SetBackgroundColour('#ffdf85')
	492	#self.alpha_estimate_ctl.SetBackgroundColour(self.GetBackgroundColour())
	493	self.alpha_estimate_ctl.SetToolTipString("Waiting for estimate...")
[f3d51f6]	494
[35adaf6]	495	id = wx.NewId()
	496	self.nterms_estimate_ctl = wx.Button(self, id, "")
	497	#self.nterms_estimate_ctl.Hide()
	498	self.Bind(wx.EVT_BUTTON, self._on_accept_nterms, id = id)
	499	self.nterms_estimate_ctl.Enable(False)
	500	#self.nterms_estimate_ctl.SetBackgroundColour('#ffdf85')
	501	#self.nterms_estimate_ctl.SetBackgroundColour(self.GetBackgroundColour())
	502	self.nterms_estimate_ctl.SetToolTipString("Waiting for estimate...")
	503
[d6113849]	504	self.nfunc_ctl.Bind(wx.EVT_TEXT, self._read_pars)
[a4bd2ac]	505	self.alpha_ctl.Bind(wx.EVT_TEXT, self._read_pars)
	506	self.dmax_ctl.Bind(wx.EVT_TEXT, self._on_pars_changed)
	507
[a00ee4c]	508	# Distance explorator
	509	id = wx.NewId()
	510	self.distance_explorator_ctl = wx.Button(self, id, "Explore")
	511	self.Bind(wx.EVT_BUTTON, self._on_explore, id = id)
[a4bd2ac]	512
	513
[f3d51f6]	514	sizer_params = wx.GridBagSizer(5,5)
	515
	516	iy = 0
[634f1cf]	517	sizer_params.Add(self.label_sugg, (iy,2), (1,1), wx.LEFT, 15)
[f3d51f6]	518	iy += 1
[32dffae4]	519	sizer_params.Add(label_nfunc, (iy,0), (1,1), wx.LEFT, 15)
	520	sizer_params.Add(self.nfunc_ctl, (iy,1), (1,1), wx.RIGHT, 0)
[35adaf6]	521	sizer_params.Add(self.nterms_estimate_ctl, (iy,2), (1,1), wx.LEFT, 15)
[f3d51f6]	522	iy += 1
[32dffae4]	523	sizer_params.Add(label_alpha, (iy,0), (1,1), wx.LEFT, 15)
	524	sizer_params.Add(self.alpha_ctl, (iy,1), (1,1), wx.RIGHT, 0)
	525	sizer_params.Add(self.alpha_estimate_ctl, (iy,2), (1,1), wx.LEFT, 15)
[f3d51f6]	526	iy += 1
[32dffae4]	527	sizer_params.Add(label_dmax, (iy,0), (1,1), wx.LEFT, 15)
	528	sizer_params.Add(self.dmax_ctl, (iy,1), (1,1), wx.RIGHT, 0)
[a00ee4c]	529	sizer_params.Add(self.distance_explorator_ctl, (iy,2), (1,1), wx.LEFT, 15)
[f3d51f6]	530
	531	boxsizer2.Add(sizer_params, 0)
[634f1cf]	532
[4318af7f]	533	iy_vb += 1
[0ccd214]	534	vbox.Add(boxsizer2, (iy_vb,0), (1,1), wx.LEFT\|wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[f3d51f6]	535
[634f1cf]	536
[f3d51f6]	537	# ----- Results -----
	538	resbox = wx.StaticBox(self, -1, "Outputs")
	539	ressizer = wx.StaticBoxSizer(resbox, wx.VERTICAL)
[0ccd214]	540	ressizer.SetMinSize((self._default_width,50))
[f3d51f6]	541
[a4bd2ac]	542	label_rg = wx.StaticText(self, -1, "Rg")
	543	label_rg_unit = wx.StaticText(self, -1, "[A]")
	544	label_iq0 = wx.StaticText(self, -1, "I(Q=0)")
	545	label_iq0_unit = wx.StaticText(self, -1, "[A^(-1)]")
	546	label_bck = wx.StaticText(self, -1, "Background")
	547	label_bck_unit = wx.StaticText(self, -1, "[A^(-1)]")
[9ff861b]	548	self.rg_ctl = OutputTextCtrl(self, -1, size=(60,20))
[a4bd2ac]	549	self.rg_ctl.SetToolTipString("Radius of gyration for the computed P(r).")
[9ff861b]	550	self.iq0_ctl = OutputTextCtrl(self, -1, size=(60,20))
[a4bd2ac]	551	self.iq0_ctl.SetToolTipString("Scattering intensity at Q=0 for the computed P(r).")
[9ff861b]	552	self.bck_ctl = OutputTextCtrl(self, -1, size=(60,20))
[a4bd2ac]	553	self.bck_ctl.SetToolTipString("Value of estimated constant background.")
	554
[f3d51f6]	555	label_time = wx.StaticText(self, -1, "Computation time")
	556	label_time_unit = wx.StaticText(self, -1, "secs")
	557	label_time.SetMinSize((120,20))
	558	label_chi2 = wx.StaticText(self, -1, "Chi2/dof")
	559	label_osc = wx.StaticText(self, -1, "Oscillations")
[dfb58f8]	560	label_pos = wx.StaticText(self, -1, "Positive fraction")
	561	label_pos_err = wx.StaticText(self, -1, "1-sigma positive fraction")
[f3d51f6]	562
[9ff861b]	563	self.time_ctl = OutputTextCtrl(self, -1, size=(60,20))
[32dffae4]	564	self.time_ctl.SetToolTipString("Computation time for the last inversion, in seconds.")
	565
[9ff861b]	566	self.chi2_ctl = OutputTextCtrl(self, -1, size=(60,20))
[32dffae4]	567	self.chi2_ctl.SetToolTipString("Chi^2 over degrees of freedom.")
	568
[dfb58f8]	569	# Oscillation parameter
[9ff861b]	570	self.osc_ctl = OutputTextCtrl(self, -1, size=(60,20))
[32dffae4]	571	self.osc_ctl.SetToolTipString("Oscillation parameter. P(r) for a sphere has an oscillation parameter of 1.1.")
[f3d51f6]	572
[dfb58f8]	573	# Positive fraction figure of merit
[9ff861b]	574	self.pos_ctl = OutputTextCtrl(self, -1, size=(60,20))
[dfb58f8]	575	self.pos_ctl.SetToolTipString("Fraction of P(r) that is positive. Theoretically, P(r) is defined positive.")
	576
	577	# 1-simga positive fraction figure of merit
[9ff861b]	578	self.pos_err_ctl = OutputTextCtrl(self, -1, size=(60,20))
[dfb58f8]	579	message = "Fraction of P(r) that is at least 1 standard deviation greater than zero.\n"
	580	message += "This figure of merit tells you about the size of the P(r) errors.\n"
	581	message += "If it is close to 1 and the other figures of merit are bad, consider changing "
	582	message += "the maximum distance."
	583	self.pos_err_ctl.SetToolTipString(message)
	584
[f3d51f6]	585	sizer_res = wx.GridBagSizer(5,5)
	586
	587	iy = 0
[a4bd2ac]	588	sizer_res.Add(label_rg, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	589	sizer_res.Add(self.rg_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	590	sizer_res.Add(label_rg_unit, (iy,2), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	591	iy += 1
	592	sizer_res.Add(label_iq0, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	593	sizer_res.Add(self.iq0_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	594	sizer_res.Add(label_iq0_unit, (iy,2), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	595	iy += 1
	596	sizer_res.Add(label_bck, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	597	sizer_res.Add(self.bck_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	598	sizer_res.Add(label_bck_unit, (iy,2), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	599	iy += 1
[f3d51f6]	600	sizer_res.Add(label_time, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	601	sizer_res.Add(self.time_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	602	sizer_res.Add(label_time_unit, (iy,2), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	603	iy += 1
	604	sizer_res.Add(label_chi2, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	605	sizer_res.Add(self.chi2_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	606	iy += 1
	607	sizer_res.Add(label_osc, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	608	sizer_res.Add(self.osc_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	609
[dfb58f8]	610	iy += 1
	611	sizer_res.Add(label_pos, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	612	sizer_res.Add(self.pos_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	613
	614	iy += 1
	615	sizer_res.Add(label_pos_err, (iy,0), (1,1), wx.LEFT\|wx.EXPAND, 15)
	616	sizer_res.Add(self.pos_err_ctl, (iy,1), (1,1), wx.RIGHT\|wx.EXPAND, 15)
	617
[f3d51f6]	618	ressizer.Add(sizer_res, 0)
[4318af7f]	619	iy_vb += 1
[0ccd214]	620	vbox.Add(ressizer, (iy_vb,0), (1,1), wx.LEFT\|wx.RIGHT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 5)
[f3d51f6]	621
	622	# ----- Buttons -----
	623	id = wx.NewId()
	624	button_OK = wx.Button(self, id, "Compute")
[32dffae4]	625	button_OK.SetToolTipString("Perform P(r) inversion.")
[f3d51f6]	626	self.Bind(wx.EVT_BUTTON, self._on_invert, id = id)
[b659551]	627
	628	id = wx.NewId()
	629	button_Reset = wx.Button(self, id, "Reset")
	630	button_Reset.SetToolTipString("Reset inversion parameters to default.")
	631	self.Bind(wx.EVT_BUTTON, self._on_reset, id = id)
[f3d51f6]	632	#button_Cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
	633
[5d98370]	634	id = wx.NewId()
	635	button_Save = wx.Button(self, id, "Save")
	636	button_Save.SetToolTipString("Save the current P(r) work to file.")
	637	self.Bind(wx.EVT_BUTTON, self._save_state, id = id)
	638
[f3d51f6]	639	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
	640	sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
[5d98370]	641	sizer_button.Add(button_Save, 0, wx.LEFT\|wx.ADJUST_MINSIZE, 10)
[b659551]	642	sizer_button.Add(button_Reset, 0, wx.LEFT\|wx.ADJUST_MINSIZE, 10)
[f3d51f6]	643	sizer_button.Add(button_OK, 0, wx.LEFT\|wx.ADJUST_MINSIZE, 10)
	644	#sizer_button.Add(button_Cancel, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
[4318af7f]	645	iy_vb += 1
[0ccd214]	646	vbox.Add(sizer_button, (iy_vb,0), (1,1), wx.EXPAND\|wx.BOTTOM\|wx.TOP\|wx.RIGHT, 10)
[f3d51f6]	647
[a00ee4c]	648	self.Bind(wx.EVT_TEXT_ENTER, self._on_invert)
[f3d51f6]	649
	650	self.SetSizer(vbox)
	651
[634f1cf]	652	def _on_accept_alpha(self, evt):
	653	"""
[7116b6e0]	654	User has accepted the estimated alpha,
	655	set it as part of the input parameters
[634f1cf]	656	"""
	657	try:
	658	alpha = self.alpha_estimate_ctl.GetLabel()
	659	tmp = float(alpha)
	660	self.alpha_ctl.SetValue(alpha)
	661	except:
	662	# No estimate or bad estimate, either do nothing
	663	import sys
	664	print "InversionControl._on_accept_alpha: %s" % sys.exc_value
	665	pass
[35adaf6]	666
	667	def _on_accept_nterms(self, evt):
	668	"""
[7116b6e0]	669	User has accepted the estimated number of terms,
	670	set it as part of the input parameters
[35adaf6]	671	"""
	672	try:
	673	nterms = self.nterms_estimate_ctl.GetLabel()
	674	tmp = float(nterms)
	675	self.nfunc_ctl.SetValue(nterms)
	676	except:
	677	# No estimate or bad estimate, either do nothing
	678	import sys
	679	print "InversionControl._on_accept_nterms: %s" % sys.exc_value
	680	pass
[634f1cf]	681
[b659551]	682	def _on_reset(self, evt):
	683	"""
[7116b6e0]	684	Resets inversion parameters
[b659551]	685	"""
	686	self.nfunc = self.manager.DEFAULT_NFUNC
	687	self.d_max = self.manager.DEFAULT_DMAX
	688	self.alpha = self.manager.DEFAULT_ALPHA
	689	self.qmin_ctl.SetValue("")
	690	self.qmax_ctl.SetValue("")
	691	self.time_ctl.SetValue("")
[a4bd2ac]	692	self.rg_ctl.SetValue("")
	693	self.iq0_ctl.SetValue("")
	694	self.bck_ctl.SetValue("")
[b659551]	695	self.chi2_ctl.SetValue("")
	696	self.osc_ctl.SetValue("")
	697	self.pos_ctl.SetValue("")
	698	self.pos_err_ctl.SetValue("")
	699	self.alpha_estimate_ctl.Enable(False)
	700	self.alpha_estimate_ctl.SetLabel("")
[35adaf6]	701	self.nterms_estimate_ctl.Enable(False)
	702	self.nterms_estimate_ctl.SetLabel("")
[b659551]	703	self._on_pars_changed()
[634f1cf]	704
[b659551]	705	def _on_pars_changed(self, evt=None):
[f3d51f6]	706	"""
[7116b6e0]	707	Called when an input parameter has changed
	708	We will estimate the alpha parameter behind the
	709	scenes.
[f3d51f6]	710	"""
[2a92852]	711	flag, alpha, dmax, nfunc, qmin, qmax, height, width = self._read_pars()
[a4bd2ac]	712	has_bck = self.bck_chk.IsChecked()
[f3d51f6]	713
	714	# If the pars are valid, estimate alpha
	715	if flag:
[2a92852]	716	self.nterms_estimate_ctl.Enable(False)
	717	self.alpha_estimate_ctl.Enable(False)
	718
[0d88a09]	719	dataset = self.plot_data.GetValue()
	720	self.manager.estimate_plot_inversion(alpha=alpha, nfunc=nfunc,
	721	d_max=dmax,
	722	q_min=qmin, q_max=qmax,
	723	bck=has_bck,
	724	height=height,
	725	width=width)
[f3d51f6]	726
[7116b6e0]	727	def _read_pars(self, evt=None):
	728	"""
	729	"""
[f3d51f6]	730	alpha = 0
	731	nfunc = 5
	732	dmax = 120
[3fd1ebc]	733	qmin = 0
	734	qmax = 0
[2a92852]	735	height = 0
	736	width = 0
[f3d51f6]	737
	738	flag = True
	739
[2a92852]	740
	741	# Read slit height
	742	try:
	743	height_str = self.sheight_ctl.GetValue()
	744	if len(height_str.lstrip().rstrip())==0:
	745	height = 0
	746	else:
	747	height = float(height_str)
	748	self.sheight_ctl.SetBackgroundColour(wx.WHITE)
	749	self.sheight_ctl.Refresh()
	750	except:
	751	flag = False
	752	self.sheight_ctl.SetBackgroundColour("pink")
	753	self.sheight_ctl.Refresh()
	754
	755	# Read slit width
	756	try:
	757	width_str = self.swidth_ctl.GetValue()
	758	if len(width_str.lstrip().rstrip())==0:
	759	width = 0
	760	else:
	761	width = float(width_str)
	762	self.swidth_ctl.SetBackgroundColour(wx.WHITE)
	763	self.swidth_ctl.Refresh()
	764	except:
	765	flag = False
	766	self.swidth_ctl.SetBackgroundColour("pink")
	767	self.swidth_ctl.Refresh()
	768
[f3d51f6]	769	# Read alpha
	770	try:
	771	alpha = float(self.alpha_ctl.GetValue())
	772	self.alpha_ctl.SetBackgroundColour(wx.WHITE)
	773	self.alpha_ctl.Refresh()
	774	except:
	775	flag = False
	776	self.alpha_ctl.SetBackgroundColour("pink")
	777	self.alpha_ctl.Refresh()
	778
	779	# Read d_max
	780	try:
	781	dmax = float(self.dmax_ctl.GetValue())
	782	self.dmax_ctl.SetBackgroundColour(wx.WHITE)
	783	self.dmax_ctl.Refresh()
	784	except:
	785	flag = False
	786	self.dmax_ctl.SetBackgroundColour("pink")
	787	self.dmax_ctl.Refresh()
	788
	789	# Read nfunc
	790	try:
	791	nfunc = int(self.nfunc_ctl.GetValue())
[b659551]	792	npts = self.manager.get_npts()
	793	if npts>0 and nfunc>npts:
	794	message = "Number of function terms should be smaller than the number of points"
[0d21ac1]	795	wx.PostEvent(self.manager.parent, StatusEvent(status=message))
[b659551]	796	raise ValueError, message
[f3d51f6]	797	self.nfunc_ctl.SetBackgroundColour(wx.WHITE)
	798	self.nfunc_ctl.Refresh()
	799	except:
	800	flag = False
	801	self.nfunc_ctl.SetBackgroundColour("pink")
	802	self.nfunc_ctl.Refresh()
	803
[634f1cf]	804	# Read qmin
	805	try:
	806	qmin_str = self.qmin_ctl.GetValue()
	807	if len(qmin_str.lstrip().rstrip())==0:
	808	qmin = None
	809	else:
	810	qmin = float(qmin_str)
	811	self.qmin_ctl.SetBackgroundColour(wx.WHITE)
	812	self.qmin_ctl.Refresh()
	813	except:
	814	flag = False
	815	self.qmin_ctl.SetBackgroundColour("pink")
	816	self.qmin_ctl.Refresh()
	817
	818	# Read qmax
	819	try:
	820	qmax_str = self.qmax_ctl.GetValue()
	821	if len(qmax_str.lstrip().rstrip())==0:
	822	qmax = None
	823	else:
	824	qmax = float(qmax_str)
	825	self.qmax_ctl.SetBackgroundColour(wx.WHITE)
	826	self.qmax_ctl.Refresh()
	827	except:
	828	flag = False
	829	self.qmax_ctl.SetBackgroundColour("pink")
	830	self.qmax_ctl.Refresh()
	831
[2a92852]	832	return flag, alpha, dmax, nfunc, qmin, qmax, height, width
[f3d51f6]	833
[a00ee4c]	834	def _on_explore(self, evt):
	835	"""
[7116b6e0]	836	Invoke the d_max exploration dialog
[a00ee4c]	837	"""
	838	from explore_dialog import ExploreDialog
	839	if self.manager._last_pr is not None:
	840	pr = self.manager._create_plot_pr()
	841	dialog = ExploreDialog(pr, 10, None, -1, "")
	842	dialog.ShowModal()
	843	else:
	844	message = "No data to analyze. Please load a data set to proceed."
	845	wx.PostEvent(self.manager.parent, StatusEvent(status=message))
	846
[f3d51f6]	847	def _on_invert(self, evt):
	848	"""
[7116b6e0]	849	Perform inversion
	850
	851	:param silent: when True, there will be no output for the user
	852
[f3d51f6]	853	"""
	854	# Get the data from the form
	855	# Push it to the manager
	856
[2a92852]	857	flag, alpha, dmax, nfunc, qmin, qmax, height, width = self._read_pars()
[a4bd2ac]	858	has_bck = self.bck_chk.IsChecked()
[f3d51f6]	859
	860	if flag:
[0d88a09]	861	dataset = self.plot_data.GetValue()
[9bc4ed23]	862	if dataset==None or len(dataset.strip())==0:
[0d88a09]	863	message = "No data to invert. Select a data set before proceeding with P(r) inversion."
	864	wx.PostEvent(self.manager.parent, StatusEvent(status=message))
[f3d51f6]	865	else:
[0d88a09]	866	self.manager.setup_plot_inversion(alpha=alpha, nfunc=nfunc,
	867	d_max=dmax,
	868	q_min=qmin, q_max=qmax,
	869	bck=has_bck,
	870	height=height,
	871	width=width)
[f3d51f6]	872	else:
	873	message = "The P(r) form contains invalid values: please submit it again."
	874	wx.PostEvent(self.parent, StatusEvent(status=message))
	875
[5d98370]	876	def _change_file(self, evt=None, filepath=None):
[f3d51f6]	877	"""
[7116b6e0]	878	Choose a new input file for I(q)
[f3d51f6]	879	"""
	880	import os
	881	if not self.manager==None:
[5d98370]	882	path = self.manager.choose_file(path=filepath)
[f3d51f6]	883
	884	if path and os.path.isfile(path):
[0d88a09]	885	self.plot_data.SetValue(str(path))
[ceaf16e]	886	try:
	887	self.manager.show_data(path, reset=True)
	888	self._on_pars_changed(None)
	889
	890	# Perform inversion
	891	if self.standalone == True:
	892	self._on_invert(None)
	893	except:
	894	# Invalid data
	895	logging.error("InversionControl._change_file: %s" % sys.exc_value)
[119a11d]	896
	897	class HelpDialog(wx.Dialog):
[7116b6e0]	898	"""
	899	"""
[119a11d]	900	def __init__(self, parent, id):
[7116b6e0]	901	"""
	902	"""
[119a11d]	903	from sans.pr.invertor import help
[fc4ab6e]	904	wx.Dialog.__init__(self, parent, id, size=(400, 420))
[119a11d]	905	self.SetTitle("P(r) help")
	906
	907
	908	vbox = wx.BoxSizer(wx.VERTICAL)
	909
	910	explanation = help()
	911
[fc4ab6e]	912	label_explain = wx.StaticText(self, -1, explanation, size=(350,320))
[119a11d]	913
	914	vbox.Add(label_explain, 0, wx.ALL\|wx.EXPAND, 15)
	915
	916
	917	static_line = wx.StaticLine(self, -1)
	918	vbox.Add(static_line, 0, wx.EXPAND, 0)
	919
	920	button_OK = wx.Button(self, wx.ID_OK, "OK")
	921	#button_Cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
	922
	923	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
	924	sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	925	sizer_button.Add(button_OK, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
	926	#sizer_button.Add(button_Cancel, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
	927	vbox.Add(sizer_button, 0, wx.EXPAND\|wx.BOTTOM\|wx.TOP, 10)
	928
	929	self.SetSizer(vbox)
	930	self.SetAutoLayout(True)
	931
	932	self.Layout()
	933	self.Centre()
	934
[b659551]	935	class PrDistDialog(wx.Dialog):
	936	"""
[7116b6e0]	937	Property dialog to let the user change the number
	938	of points on the P(r) plot.
[b659551]	939	"""
	940	def __init__(self, parent, id):
	941	from sans.pr.invertor import help
	942	wx.Dialog.__init__(self, parent, id, size=(250, 120))
	943	self.SetTitle("P(r) distribution")
	944
	945
	946	vbox = wx.BoxSizer(wx.VERTICAL)
	947
	948	label_npts = wx.StaticText(self, -1, "Number of points")
[9ff861b]	949	self.npts_ctl = PrTextCtrl(self, -1, size=(100,20))
[b659551]	950
	951	pars_sizer = wx.GridBagSizer(5,5)
	952	iy = 0
	953	pars_sizer.Add(label_npts, (iy,0), (1,1), wx.LEFT, 15)
	954	pars_sizer.Add(self.npts_ctl, (iy,1), (1,1), wx.RIGHT, 0)
	955
	956	vbox.Add(pars_sizer, 0, wx.ALL\|wx.EXPAND, 15)
	957
	958
	959	static_line = wx.StaticLine(self, -1)
	960	vbox.Add(static_line, 0, wx.EXPAND, 0)
	961
	962	button_OK = wx.Button(self, wx.ID_OK, "OK")
	963	self.Bind(wx.EVT_BUTTON, self._checkValues, button_OK)
	964	button_Cancel = wx.Button(self, wx.ID_CANCEL, "Cancel")
	965
	966	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
	967	sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	968	sizer_button.Add(button_OK, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
	969	sizer_button.Add(button_Cancel, 0, wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
	970	vbox.Add(sizer_button, 0, wx.EXPAND\|wx.BOTTOM\|wx.TOP, 10)
	971
	972	self.SetSizer(vbox)
	973	self.SetAutoLayout(True)
	974
	975	self.Layout()
	976	self.Centre()
	977
	978	def _checkValues(self, event):
	979	"""
[7116b6e0]	980	Check the dialog content.
[b659551]	981	"""
	982	flag = True
	983	try:
	984	int(self.npts_ctl.GetValue())
	985	self.npts_ctl.SetBackgroundColour(wx.WHITE)
	986	self.npts_ctl.Refresh()
	987	except:
	988	flag = False
	989	self.npts_ctl.SetBackgroundColour("pink")
	990	self.npts_ctl.Refresh()
	991	if flag:
	992	event.Skip(True)
	993
	994	def get_content(self):
	995	"""
[7116b6e0]	996	Return the content of the dialog.
	997	At this point the values have already been
	998	checked.
[b659551]	999	"""
	1000	value = int(self.npts_ctl.GetValue())
	1001	return value
	1002
	1003	def set_content(self, npts):
	1004	"""
[7116b6e0]	1005	Initialize the content of the dialog.
[b659551]	1006	"""
	1007	self.npts_ctl.SetValue("%i" % npts)
	1008
[f3d51f6]	1009	##### testing code ############################################################
[7116b6e0]	1010	"""
	1011	Example: ::
	1012
[f3d51f6]	1013	class TestPlot:
	1014	def __init__(self, text):
	1015	self.name = text
	1016
	1017	class MyApp(wx.App):
	1018	def OnInit(self):
	1019	wx.InitAllImageHandlers()
[b659551]	1020	dialog = PrDistDialog(None, -1)
[f3d51f6]	1021	if dialog.ShowModal() == wx.ID_OK:
[119a11d]	1022	pass
[f3d51f6]	1023	dialog.Destroy()
	1024
	1025	return 1
	1026
	1027	# end of class MyApp
	1028
	1029	if __name__ == "__main__":
	1030	app = MyApp(0)
	1031	app.MainLoop()
	1032
[7116b6e0]	1033	"""
[f3d51f6]	1034	##### end of testing code #####################################################

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

SasView

source: sasview/prview/perspectives/pr/inversion_panel.py @ d65a00a

Download in other formats: