simfitpage.py @ e28f34d

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.2ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since e28f34d was 998ca90, checked in by krzywon, 8 years ago
#189 #12 Saving and loading project now fully saves and loads a simultaneous fits. Loading a project will reset Sasview state to its base state (with a warning). One error on load to fix before calling this finished.
Property mode set to `100644`
File size: 43.0 KB

Line
1	"""
2	Simultaneous fit page
3	"""
4	import sys
5	from collections import namedtuple
6
7	import wx
8	import wx.lib.newevent
9	from wx.lib.scrolledpanel import ScrolledPanel
10
11	from sas.sasgui.guiframe.events import StatusEvent, PanelOnFocusEvent
12	from sas.sasgui.guiframe.panel_base import PanelBase
13	from sas.sasgui.guiframe.utils import IdList
14	from sas.sasgui.guiframe.documentation_window import DocumentationWindow
15
16	# Control panel width
17	if sys.platform.count("darwin") == 0:
18	PANEL_WID = 420
19	FONT_VARIANT = 0
20	else:
21	PANEL_WID = 490
22	FONT_VARIANT = 1
23
24
25	# Each constraint requires five widgets and sizer. Package them in
26	# a named tuple for easy access.
27	ConstraintLine = namedtuple('ConstraintLine',
28	'model_cbox param_cbox egal_txt constraint btRemove sizer')
29
30
31	def get_fittableParam(model):
32	"""
33	return list of fittable parameters from a model
34
35	:param model: the model used
36
37	"""
38	fittable_param = []
39	for item in model.getParamList():
40	if not item in model.getDispParamList():
41	if not item in model.non_fittable:
42	fittable_param.append(item)
43
44	for item in model.fixed:
45	fittable_param.append(item)
46
47	return fittable_param
48
49
50	class SimultaneousFitPage(ScrolledPanel, PanelBase):
51	"""
52	Simultaneous fitting panel
53	All that needs to be defined are the
54	two data members window_name and window_caption
55	"""
56	# Internal name for the AUI manager
57	window_name = "Simultaneous Fit Page"
58	# Title to appear on top of the window
59	window_caption = "Simultaneous Fit Page"
60	ID_DOC = wx.NewId()
61	ID_SET_ALL = wx.NewId()
62	ID_FIT = wx.NewId()
63	ID_ADD = wx.NewId()
64	_id_pool = IdList()
65
66	def __init__(self, parent, page_finder={}, id=wx.ID_ANY, batch_on=False,
67	args, *kwargs):
68	ScrolledPanel.__init__(self, parent, id=id,
69	style=wx.FULL_REPAINT_ON_RESIZE,
70	args, *kwargs)
71	PanelBase.__init__(self, parent)
72	"""
73	Simultaneous page display
74	"""
75	self._ids = iter(self._id_pool)
76	self.SetupScrolling()
77	# Font size
78	self.SetWindowVariant(variant=FONT_VARIANT)
79	self.uid = wx.NewId()
80	self.parent = parent
81	self.batch_on = batch_on
82	# store page_finder
83	self.page_finder = page_finder
84	# list containing info to set constraint
85	# look like self.constraint_dict[page_id]= page
86	self.constraint_dict = {}
87	# item list
88	# self.constraints_list=[combobox1, combobox2,=,textcrtl, button ]
89	self.constraints_list = []
90	# list of current model
91	self.model_list = []
92	# selected model to fit
93	self.model_to_fit = []
94	# Control the fit state
95	self.fit_started = False
96	# number of constraint
97	self.nb_constraint = 0
98	self.state = SimFitPageState()
99	self.model_cbox_left = None
100	self.model_cbox_right = None
101	# draw page
102	self.define_page_structure()
103	self.draw_page()
104	self._set_save_flag(False)
105
106	def define_page_structure(self):
107	"""
108	Create empty sizers, their hierarchy and set the sizer for the panel
109	"""
110	self.vbox = wx.BoxSizer(wx.VERTICAL)
111	self.data_selection_sizer = wx.BoxSizer(wx.VERTICAL)
112	self.constraints_sizer = wx.BoxSizer(wx.VERTICAL)
113	self.run_fit_sizer = wx.BoxSizer(wx.VERTICAL)
114
115	self.data_selection_sizer.SetMinSize((PANEL_WID, -1))
116	self.constraints_sizer.SetMinSize((PANEL_WID, -1))
117	self.run_fit_sizer.SetMinSize((PANEL_WID, -1))
118	self.vbox.Add(self.data_selection_sizer)
119	self.vbox.Add(self.constraints_sizer)
120	self.vbox.Add(self.run_fit_sizer)
121	self.SetSizer(self.vbox)
122	self.Centre()
123
124	def set_state(self):
125	"""
126	Define a set of state parameters for saving simultaneous fits.
127	"""
128	self._set_constraint()
129	self.state.fit_page_no = self.uid
130	self.state.select_all = self.cb1.GetValue()
131	self.state.model_list = self.model_list
132	self.state.model_to_fit = self.model_to_fit
133	self.state.no_constraint = self.nb_constraint
134	self.state.constraint_dict = self.constraint_dict
135	self.state.constraints_list = self.constraints_list
136	return self.get_state()
137
138	def get_state(self):
139	"""
140	Return the state of the current page
141	:return: self.state
142	"""
143	return self.state
144
145	def draw_page(self):
146	"""
147	Construct the Simultaneous/Constrained fit page. fills the first
148	region (sizer1) with the list of available fit page pairs of data
149	and models. Then fills sizer2 with the checkbox for adding
150	constraints, and finally fills sizer3 with the fit button and
151	instructions.
152	"""
153
154	# create blank list of constraints
155	self.model_list = []
156	self.model_to_fit = []
157	self.constraints_list = []
158	self.constraint_dict = {}
159	self.nb_constraint = 0
160	self.model_cbox_left = None
161	self.model_cbox_right = None
162
163	if len(self.model_list) > 0:
164	for item in self.model_list:
165	item[0].SetValue(False)
166	self.manager.schedule_for_fit(value=0, uid=item[2])
167
168	#-------------------------------------------------------
169	# setup sizer1 (which fitpages to include)
170	self.data_selection_sizer.Clear(True)
171	box_description = wx.StaticBox(self, wx.ID_ANY, "Fit Combinations")
172	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
173	sizer_title = wx.BoxSizer(wx.HORIZONTAL)
174	sizer_couples = wx.GridBagSizer(5, 5)
175
176	# The wx GUI has a flag to enable a menu item, but can still be
177	# reached via scripting. There is no guearantee future GUI
178	# implementations force this check, either.
179	# IMHO, this if statement should stay -- JRK 2016-OCT-05
180	if len(self.page_finder) == 0:
181	msg = " No fit combinations are found! \n\n"
182	msg += " Please load data and set up "
183	msg += "at least one fit panels first..."
184	sizer_title.Add(wx.StaticText(self, wx.ID_ANY, msg))
185	else:
186	# store model
187	self._store_model()
188
189	self.cb1 = wx.CheckBox(self, wx.ID_ANY, 'Select all')
190	self.cb1.SetValue(False)
191	wx.EVT_CHECKBOX(self, self.cb1.GetId(), self.check_all_model_name)
192
193	sizer_title.Add((10, 10), 0,
194	wx.TOP \| wx.BOTTOM \| wx.EXPAND \| wx.ADJUST_MINSIZE, border=5)
195	sizer_title.Add(self.cb1, 0,
196	wx.TOP \| wx.BOTTOM \| wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE,
197	border=5)
198
199	# draw list of model and data names
200	self._fill_sizer_model_list(sizer_couples)
201
202	boxsizer1.Add(sizer_title, flag=wx.TOP \| wx.BOTTOM, border=5)
203	boxsizer1.Add(sizer_couples, 1, flag=wx.TOP \| wx.BOTTOM, border=5)
204	self.data_selection_sizer.Add(boxsizer1, 1, wx.EXPAND \| wx.ALL, 10)
205	# self.sizer1.Layout()
206
207	#--------------------------------------------------------
208	# set up the other 2 sizers: the constraints list and the
209	# buttons (fit, help etc) sizer at the bottom of the page.
210	# Note: the if statement should be removed along with the above
211	# if statement as soon as it can be properly tested.
212	# Nov. 22 2015 --PDB
213	# As above, this page can be accessed through other means than the
214	# base SasView GUI.
215	# Oct. 5, 2016 --JRK
216	if len(self.page_finder) > 0:
217	# draw the sizer containing constraint info
218	if not self.batch_on:
219	self._fill_sizer_constraint()
220	# draw fit button sizer
221	self._fill_sizer_fit()
222
223	def _fill_sizer_model_list(self, sizer):
224	"""
225	Receive a dictionary containing information to display model name
226	"""
227	ix = 0
228	iy = 0
229	sizer.Clear(True)
230
231	new_name = wx.StaticText(self, wx.ID_ANY, ' Model Title ',
232	style=wx.ALIGN_CENTER)
233	new_name.SetBackgroundColour('orange')
234	new_name.SetForegroundColour(wx.WHITE)
235	sizer.Add(new_name, (iy, ix), (1, 1),
236	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
237	ix += 2
238	model_type = wx.StaticText(self, wx.ID_ANY, ' Model ')
239	model_type.SetBackgroundColour('grey')
240	model_type.SetForegroundColour(wx.WHITE)
241	sizer.Add(model_type, (iy, ix), (1, 1),
242	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
243	ix += 1
244	data_used = wx.StaticText(self, wx.ID_ANY, ' Data ')
245	data_used.SetBackgroundColour('grey')
246	data_used.SetForegroundColour(wx.WHITE)
247	sizer.Add(data_used, (iy, ix), (1, 1),
248	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
249	ix += 1
250	tab_used = wx.StaticText(self, wx.ID_ANY, ' FitPage ')
251	tab_used.SetBackgroundColour('grey')
252	tab_used.SetForegroundColour(wx.WHITE)
253	sizer.Add(tab_used, (iy, ix), (1, 1),
254	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
255	for id, value in self.page_finder.iteritems():
256	if id not in self.parent.opened_pages:
257	continue
258
259	if self.batch_on != self.parent.get_page_by_id(id).batch_on:
260	continue
261
262	data_list = []
263	model_list = []
264	# get data name and model objetta
265	for fitproblem in value.get_fit_problem():
266
267	data = fitproblem.get_fit_data()
268	if not data.is_data:
269	continue
270	name = '-'
271	if data is not None and data.is_data:
272	name = str(data.name)
273	data_list.append(name)
274
275	model = fitproblem.get_model()
276	if model is None:
277	continue
278	model_list.append(model)
279
280	if len(model_list) == 0:
281	continue
282	# Draw sizer
283	ix = 0
284	iy += 1
285	model = model_list[0]
286	name = '_'
287	if model is not None:
288	name = str(model.name)
289	cb = wx.CheckBox(self, wx.ID_ANY, name)
290	cb.SetValue(False)
291	cb.Enable(model is not None and data.is_data)
292	sizer.Add(cb, (iy, ix), (1, 1),
293	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
294	wx.EVT_CHECKBOX(self, cb.GetId(), self.check_model_name)
295	ix += 2
296	model_type = wx.StaticText(self, wx.ID_ANY,
297	model.__class__.__name__)
298	sizer.Add(model_type, (iy, ix), (1, 1),
299	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
300	if self.batch_on:
301	data_used = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
302	data_used.AppendItems(data_list)
303	data_used.SetSelection(0)
304	else:
305	data_used = wx.StaticText(self, wx.ID_ANY, data_list[0])
306
307	ix += 1
308	sizer.Add(data_used, (iy, ix), (1, 1),
309	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
310	ix += 1
311	caption = value.get_fit_tab_caption()
312	tab_caption_used = wx.StaticText(self, wx.ID_ANY, str(caption))
313	sizer.Add(tab_caption_used, (iy, ix), (1, 1),
314	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
315
316	self.model_list.append([cb, value, id, model])
317
318	iy += 1
319	sizer.Add((20, 20), (iy, ix), (1, 1),
320	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
321
322	def _fill_sizer_constraint(self):
323	"""
324	Fill sizer containing constraint info
325	"""
326	msg = "Select at least 1 model to add constraint "
327	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
328
329	self.constraints_sizer.Clear(True)
330	if self.batch_on:
331	if self.constraints_sizer.IsShown():
332	self.constraints_sizer.Show(False)
333	return
334	box_description = wx.StaticBox(self, wx.ID_ANY, "Fit Constraints")
335	box_sizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
336	sizer_title = wx.BoxSizer(wx.HORIZONTAL)
337	self.sizer_all_constraints = wx.BoxSizer(wx.HORIZONTAL)
338	self.sizer_constraints = wx.BoxSizer(wx.VERTICAL)
339	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
340
341	self.hide_constraint = wx.RadioButton(self, wx.ID_ANY, 'No', (10, 10),
342	style=wx.RB_GROUP)
343	self.show_constraint = wx.RadioButton(self, wx.ID_ANY, 'Yes', (10, 30))
344	self.Bind(wx.EVT_RADIOBUTTON, self._display_constraint,
345	id=self.hide_constraint.GetId())
346	self.Bind(wx.EVT_RADIOBUTTON, self._display_constraint,
347	id=self.show_constraint.GetId())
348	if self.batch_on:
349	self.hide_constraint.Enable(False)
350	self.show_constraint.Enable(False)
351	self.hide_constraint.SetValue(True)
352	self.show_constraint.SetValue(False)
353
354	sizer_title.Add(wx.StaticText(self, wx.ID_ANY, " Model"))
355	sizer_title.Add((10, 10))
356	sizer_title.Add(wx.StaticText(self, wx.ID_ANY, " Parameter"))
357	sizer_title.Add((10, 10))
358	sizer_title.Add(wx.StaticText(self, wx.ID_ANY, " Add Constraint?"))
359	sizer_title.Add((10, 10))
360	sizer_title.Add(self.show_constraint)
361	sizer_title.Add(self.hide_constraint)
362	sizer_title.Add((10, 10))
363
364	self.btAdd = wx.Button(self, self.ID_ADD, 'Add')
365	self.btAdd.Bind(wx.EVT_BUTTON, self._on_add_constraint,
366	id=self.btAdd.GetId())
367	self.btAdd.SetToolTipString("Add another constraint?")
368	self.btAdd.Hide()
369
370	text_hint = wx.StaticText(self, wx.ID_ANY,
371	"Example: [M0][parameter] = M1.parameter")
372	sizer_button.Add(text_hint, 0,
373	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 10)
374	sizer_button.Add(self.btAdd, 0,
375	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 10)
376
377	box_sizer1.Add(sizer_title, flag=wx.TOP \| wx.BOTTOM, border=10)
378	box_sizer1.Add(self.sizer_all_constraints, flag=wx.TOP \| wx.BOTTOM,
379	border=10)
380	box_sizer1.Add(self.sizer_constraints, flag=wx.TOP \| wx.BOTTOM,
381	border=10)
382	box_sizer1.Add(sizer_button, flag=wx.TOP \| wx.BOTTOM, border=10)
383
384	self.constraints_sizer.Add(box_sizer1, 0, wx.EXPAND \| wx.ALL, 10)
385
386	def _fill_sizer_fit(self):
387	"""
388	Draw fit button
389	"""
390	self.run_fit_sizer.Clear(True)
391	box_description = wx.StaticBox(self, wx.ID_ANY, "Fit ")
392	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
393	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
394
395	# Fit button
396	self.btFit = wx.Button(self, self.ID_FIT, 'Fit', size=wx.DefaultSize)
397	self.btFit.Bind(wx.EVT_BUTTON, self.on_fit, id=self.btFit.GetId())
398	self.btFit.SetToolTipString("Perform fit.")
399
400	# General Help button
401	self.btHelp = wx.Button(self, wx.ID_HELP, 'HELP')
402	self.btHelp.SetToolTipString("Simultaneous/Constrained Fitting help.")
403	self.btHelp.Bind(wx.EVT_BUTTON, self._on_help)
404
405	# hint text on button line
406	if self.batch_on:
407	text = " Fit in Parallel all Data sets\n"
408	text += "and model selected."
409	else:
410	text = " At least one set of model and data\n"
411	text += " must be selected for fitting."
412	text_hint = wx.StaticText(self, wx.ID_ANY, text)
413
414	sizer_button.Add(text_hint)
415	sizer_button.Add(self.btFit, 0, wx.LEFT \| wx.ADJUST_MINSIZE, 10)
416	sizer_button.Add(self.btHelp, 0, wx.LEFT \| wx.ADJUST_MINSIZE, 10)
417
418	boxsizer1.Add(sizer_button, flag=wx.TOP \| wx.BOTTOM, border=10)
419	self.run_fit_sizer.Add(boxsizer1, 0, wx.EXPAND \| wx.ALL, 10)
420
421	def on_remove(self, event):
422	"""
423	Remove constraint fields
424	"""
425	if len(self.constraints_list) == 1:
426	self.hide_constraint.SetValue(True)
427	self._hide_constraint()
428	return
429	if len(self.constraints_list) == 0:
430	return
431	wx.CallAfter(self._remove_after, event.GetId())
432	# self._onAdd_constraint(None)
433
434	def _remove_after(self, id):
435	for item in self.constraints_list:
436	if id == item.btRemove.GetId():
437	self.sizer_constraints.Hide(item.sizer)
438	item.sizer.Clear(True)
439	self.sizer_constraints.Remove(item.sizer)
440	self.constraints_list.remove(item)
441	self.nb_constraint -= 1
442	self.constraints_sizer.Layout()
443	self.FitInside()
444	break
445
446	def on_fit(self, event):
447	"""
448	signal for fitting
449
450	"""
451	if self.fit_started:
452	self._stop_fit()
453	self.fit_started = False
454	return
455
456	flag = False
457	# check if the current page a simultaneous fit page or a batch page
458	if self == self._manager.sim_page:
459	flag = (self._manager.sim_page.uid == self.uid)
460
461	# making sure all parameters content a constraint
462	if not self.batch_on and self.show_constraint.GetValue():
463	if not self._set_constraint():
464	return
465	# model was actually selected from this page to be fit
466	if len(self.model_to_fit) >= 1:
467	self.manager._reset_schedule_problem(value=0)
468	for item in self.model_list:
469	if item[0].GetValue():
470	self.manager.schedule_for_fit(value=1, uid=item[2])
471	try:
472	self.fit_started = True
473	wx.CallAfter(self.set_fitbutton)
474	if not self.manager.onFit(uid=self.uid):
475	return
476	except:
477	msg = "Select at least one parameter to fit in the FitPages."
478	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
479	else:
480	msg = "Select at least one model check box to fit "
481	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
482	self.set_state()
483
484	def _on_fit_complete(self):
485	"""
486	Set the completion flag and display the updated fit button label.
487	"""
488	self.fit_started = False
489	self.set_fitbutton()
490
491	def _stop_fit(self, event=None):
492	"""
493	Attempt to stop the fitting thread
494
495	:param event: Event handler when stop fit is clicked
496	"""
497	if event is not None:
498	event.Skip()
499	self.manager.stop_fit(self.uid)
500	self.manager._reset_schedule_problem(value=0)
501	self._on_fit_complete()
502
503	def set_fitbutton(self):
504	"""
505	Set fit button label depending on the fit_started
506	"""
507	label = "Stop" if self.fit_started else "Fit"
508	color = "red" if self.fit_started else "black"
509
510	self.btFit.SetLabel(label)
511	self.btFit.SetForegroundColour(color)
512	self.btFit.Enable(True)
513
514	def _on_help(self, event):
515	"""
516	Bring up the simultaneous Fitting Documentation whenever the HELP
517	button is clicked.
518
519	Calls DocumentationWindow with the path of the location within the
520	documentation tree (after /doc/ ....". Note that when using old
521	versions of Wx (before 2.9) and thus not the release version of
522	installers, the help comes up at the top level of the file as
523	web browser does not pass anything past the # to the browser when it is
524	running "file:///...."
525
526	:param event: Triggers on clicking the help button
527	"""
528	_TreeLocation = "user/sasgui/perspectives/fitting/fitting_help.html"
529	_PageAnchor = "#simultaneous-fit-mode"
530	_doc_viewer = DocumentationWindow(self, self.ID_DOC, _TreeLocation,
531	_PageAnchor,
532	"Simultaneous/Constrained Fitting Help")
533
534	def set_manager(self, manager):
535	"""
536	set panel manager
537
538	:param manager: instance of plugin fitting
539	"""
540	self.manager = manager
541
542	def check_all_model_name(self, event=None):
543	"""
544	check all models names
545	"""
546	self.model_to_fit = []
547	if self.cb1.GetValue():
548	for item in self.model_list:
549	if item[0].IsEnabled():
550	item[0].SetValue(True)
551	self.model_to_fit.append(item)
552
553	# constraint info
554	self._store_model()
555	if not self.batch_on:
556	# display constraint fields
557	if (self.show_constraint.GetValue() and
558	len(self.constraints_list) == 0):
559	self._show_all_constraint()
560	self._show_constraint()
561	else:
562	for item in self.model_list:
563	item[0].SetValue(False)
564
565	if not self.batch_on:
566	# constraint info
567	self._hide_constraint()
568
569	self._update_easy_setup_cb()
570	self.FitInside()
571
572	def check_model_name(self, event):
573	"""
574	Save information related to checkbox and their states
575	"""
576	self.model_to_fit = []
577	for item in self.model_list:
578	if item[0].GetValue():
579	self.model_to_fit.append(item)
580	else:
581	if item in self.model_to_fit:
582	self.model_to_fit.remove(item)
583	self.cb1.SetValue(False)
584
585	# display constraint fields
586	if len(self.model_to_fit) >= 1:
587	self._store_model()
588	if not self.batch_on and self.show_constraint.GetValue() and\
589	len(self.constraints_list) == 0:
590	self._show_all_constraint()
591	self._show_constraint()
592
593	elif len(self.model_to_fit) < 1:
594	# constraint info
595	self._hide_constraint()
596
597	self._update_easy_setup_cb()
598	# set the value of the main check button
599	if len(self.model_list) == len(self.model_to_fit):
600	self.cb1.SetValue(True)
601	self.FitInside()
602	return
603	else:
604	self.cb1.SetValue(False)
605	self.FitInside()
606
607	def _update_easy_setup_cb(self):
608	"""
609	Update easy setup combobox on selecting a model
610	"""
611	if self.model_cbox_left is None or self.model_cbox_right is None:
612	return
613
614	models = [(item[3].name, item[3]) for item in self.model_to_fit]
615	setComboBoxItems(self.model_cbox_left, models)
616	setComboBoxItems(self.model_cbox_right, models)
617	for item in self.constraints_list:
618	setComboBoxItems(item[0], models)
619	if self.model_cbox_left.GetSelection() == wx.NOT_FOUND:
620	self.model_cbox_left.SetSelection(0)
621	self.constraints_sizer.Layout()
622
623	def _store_model(self):
624	"""
625	Store selected model
626	"""
627	if len(self.model_to_fit) < 1:
628	return
629	for item in self.model_to_fit:
630	model = item[3]
631	page_id = item[2]
632	self.constraint_dict[page_id] = model
633
634	def _display_constraint(self, event):
635	"""
636	Show fields to add constraint
637	"""
638	if len(self.model_to_fit) < 1:
639	msg = "Select at least 1 model to add constraint "
640	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
641	# hide button
642	self._hide_constraint()
643	return
644	if self.show_constraint.GetValue():
645	self._show_all_constraint()
646	self._show_constraint()
647	self.FitInside()
648	return
649	else:
650	self._hide_constraint()
651	return
652
653	def _show_all_constraint(self):
654	"""
655	Show constraint fields
656	"""
657	box_description = wx.StaticBox(self, wx.ID_ANY, "Easy Setup ")
658	box_sizer = wx.StaticBoxSizer(box_description, wx.HORIZONTAL)
659	sizer_constraint = wx.BoxSizer(wx.HORIZONTAL)
660	self.model_cbox_left = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
661	self.model_cbox_left.Clear()
662	self.model_cbox_right = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
663	self.model_cbox_right.Clear()
664	wx.EVT_COMBOBOX(self.model_cbox_left, wx.ID_ANY, self._on_select_modelcb)
665	wx.EVT_COMBOBOX(self.model_cbox_right, wx.ID_ANY, self._on_select_modelcb)
666	egal_txt = wx.StaticText(self, wx.ID_ANY, " = ")
667	self.set_button = wx.Button(self, self.ID_SET_ALL, 'Set All')
668	self.set_button.Bind(wx.EVT_BUTTON, self._on_set_all_equal,
669	id=self.set_button.GetId())
670	set_tip = "Add constraints for all the adjustable parameters "
671	set_tip += "(checked in FitPages) if exist."
672	self.set_button.SetToolTipString(set_tip)
673	self.set_button.Disable()
674
675	for id, model in self.constraint_dict.iteritems():
676	# check if all parameters have been selected for constraint
677	# then do not allow add constraint on parameters
678	self.model_cbox_left.Append(str(model.name), model)
679	self.model_cbox_left.Select(0)
680	for id, model in self.constraint_dict.iteritems():
681	# check if all parameters have been selected for constraint
682	# then do not allow add constraint on parameters
683	self.model_cbox_right.Append(str(model.name), model)
684	box_sizer.Add(self.model_cbox_left,
685	flag=wx.RIGHT \| wx.EXPAND, border=10)
686	# box_sizer.Add(wx.StaticText(self, wx.ID_ANY, ".parameters"),
687	# flag=wx.RIGHT \| wx.EXPAND, border=5)
688	box_sizer.Add(egal_txt, flag=wx.RIGHT \| wx.EXPAND, border=5)
689	box_sizer.Add(self.model_cbox_right,
690	flag=wx.RIGHT \| wx.EXPAND, border=10)
691	# box_sizer.Add(wx.StaticText(self, wx.ID_ANY, ".parameters"),
692	# flag=wx.RIGHT \| wx.EXPAND, border=5)
693	box_sizer.Add((20, -1))
694	box_sizer.Add(self.set_button, flag=wx.RIGHT \| wx.EXPAND, border=5)
695	sizer_constraint.Add(box_sizer, flag=wx.RIGHT \| wx.EXPAND, border=5)
696	self.sizer_all_constraints.Insert(before=0,
697	item=sizer_constraint,
698	flag=wx.TOP \| wx.BOTTOM \| wx.EXPAND, border=5)
699	self.FitInside()
700
701	def _on_select_modelcb(self, event):
702	"""
703	On select model left or right combobox
704	"""
705	event.Skip()
706	flag = True
707	if self.model_cbox_left.GetValue().strip() == '':
708	flag = False
709	if self.model_cbox_right.GetValue().strip() == '':
710	flag = False
711	if (self.model_cbox_left.GetValue() ==
712	self.model_cbox_right.GetValue()):
713	flag = False
714	self.set_button.Enable(flag)
715
716	def _on_set_all_equal(self, event):
717	"""
718	On set button
719	"""
720	event.Skip()
721	length = len(self.constraints_list)
722	if length < 1:
723	return
724	param_list = []
725	param_list_b = []
726	selection = self.model_cbox_left.GetCurrentSelection()
727	model_left = self.model_cbox_left.GetValue()
728	model = self.model_cbox_left.GetClientData(selection)
729	selection_b = self.model_cbox_right.GetCurrentSelection()
730	model_right = self.model_cbox_right.GetValue()
731	model_b = self.model_cbox_right.GetClientData(selection_b)
732	for id, dic_model in self.constraint_dict.iteritems():
733	if model == dic_model:
734	param_list = self.page_finder[id].get_param2fit()
735	if model_b == dic_model:
736	param_list_b = self.page_finder[id].get_param2fit()
737	if len(param_list) > 0 and len(param_list_b) > 0:
738	break
739	num_cbox = 0
740	has_param = False
741	for param in param_list:
742	num_cbox += 1
743	if param in param_list_b:
744	item = self.constraints_list[-1]
745	item.model_cbox.SetStringSelection(model_left)
746	self._on_select_model(None)
747	item.param_cbox.Clear()
748	item.param_cbox.Append(str(param), model)
749	item.param_cbox.SetStringSelection(str(param))
750	item.constraint.SetValue(str(model_right + "." + str(param)))
751	has_param = True
752	if num_cbox == (len(param_list) + 1):
753	break
754	self._show_constraint()
755
756	self.FitInside()
757	if not has_param:
758	msg = " There is no adjustable parameter (checked to fit)"
759	msg += " either one of the models."
760	wx.PostEvent(self.parent.parent, StatusEvent(info="warning",
761	status=msg))
762	else:
763	msg = " The constraints are added."
764	wx.PostEvent(self.parent.parent, StatusEvent(info="info",
765	status=msg))
766
767	def _show_constraint(self):
768	"""
769	Show constraint fields
770	:param dict: dictionary mapping constraint values
771	"""
772	self.btAdd.Show(True)
773	if len(self.constraints_list) != 0:
774	nb_fit_param = 0
775	for id, model in self.constraint_dict.iteritems():
776	nb_fit_param += len(self.page_finder[id].get_param2fit())
777	# Don't add anymore
778	if len(self.constraints_list) == nb_fit_param:
779	msg = "Cannot add another constraint. Maximum of number "
780	msg += "Parameters name reached %s" % str(nb_fit_param)
781	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
782	self.sizer_constraints.Layout()
783	self.constraints_sizer.Layout()
784	return
785	if len(self.model_to_fit) < 1:
786	msg = "Select at least 1 model to add constraint "
787	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
788	self.sizer_constraints.Layout()
789	self.constraints_sizer.Layout()
790	return
791
792	sizer_constraint = wx.BoxSizer(wx.HORIZONTAL)
793
794	# Model list
795	model_cbox = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
796	model_cbox.Clear()
797	for id, model in self.constraint_dict.iteritems():
798	# check if all parameters have been selected for constraint
799	# then do not allow add constraint on parameters
800	model_cbox.Append(str(model.name), model)
801	wx.EVT_COMBOBOX(model_cbox, wx.ID_ANY, self._on_select_model)
802
803	# Parameters in model
804	param_cbox = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY,
805	size=(100, -1))
806	param_cbox.Hide()
807	wx.EVT_COMBOBOX(param_cbox, wx.ID_ANY, self._on_select_param)
808
809	egal_txt = wx.StaticText(self, wx.ID_ANY, " = ")
810
811	# Parameter constraint
812	constraint = wx.TextCtrl(self, wx.ID_ANY)
813
814	# Remove button
815	#btRemove = wx.Button(self, self.ID_REMOVE, 'Remove')
816	bt_remove = wx.Button(self, self._ids.next(), 'Remove')
817	bt_remove.Bind(wx.EVT_BUTTON, self.on_remove,
818	id=bt_remove.GetId())
819	bt_remove.SetToolTipString("Remove constraint.")
820	bt_remove.Hide()
821
822	# Hid the add button, if it exists
823	if hasattr(self, "btAdd"):
824	self.btAdd.Hide()
825
826	sizer_constraint.Add((5, -1))
827	sizer_constraint.Add(model_cbox, flag=wx.RIGHT \| wx.EXPAND, border=10)
828	sizer_constraint.Add(param_cbox, flag=wx.RIGHT \| wx.EXPAND, border=5)
829	sizer_constraint.Add(egal_txt, flag=wx.RIGHT \| wx.EXPAND, border=5)
830	sizer_constraint.Add(constraint, flag=wx.RIGHT \| wx.EXPAND, border=10)
831	sizer_constraint.Add(bt_remove, flag=wx.RIGHT \| wx.EXPAND, border=10)
832
833	self.sizer_constraints.Insert(before=self.nb_constraint,
834	item=sizer_constraint, flag=wx.TOP \| wx.BOTTOM \| wx.EXPAND,
835	border=5)
836	c = ConstraintLine(model_cbox, param_cbox, egal_txt,
837	constraint, bt_remove, sizer_constraint)
838	self.constraints_list.append(c)
839
840	self.nb_constraint += 1
841	self.sizer_constraints.Layout()
842	self.constraints_sizer.Layout()
843	self.Layout()
844
845	def _hide_constraint(self):
846	"""
847	hide buttons related constraint
848	"""
849	for id in self.page_finder.iterkeys():
850	self.page_finder[id].clear_model_param()
851
852	self.nb_constraint = 0
853	self.constraint_dict = {}
854	if hasattr(self, "btAdd"):
855	self.btAdd.Hide()
856	self._store_model()
857	if self.model_cbox_left is not None:
858	self.model_cbox_left.Clear()
859	self.model_cbox_left = None
860	if self.model_cbox_right is not None:
861	self.model_cbox_right.Clear()
862	self.model_cbox_right = None
863	self.constraints_list = []
864	self.sizer_all_constraints.Clear(True)
865	self.sizer_all_constraints.Layout()
866	self.sizer_constraints.Clear(True)
867	self.sizer_constraints.Layout()
868	self.constraints_sizer.Layout()
869	self.Layout()
870	self.FitInside()
871
872	def _on_select_model(self, event):
873	"""
874	fill combo box with list of parameters
875	"""
876	if not self.constraints_list:
877	return
878
879	# This way PC/MAC both work, instead of using event.GetClientData().
880	model_cbox = self.constraints_list[-1].model_cbox
881	n = model_cbox.GetCurrentSelection()
882	if n == wx.NOT_FOUND:
883	return
884
885	model = model_cbox.GetClientData(n)
886	param_list = []
887	for id, dic_model in self.constraint_dict.iteritems():
888	if model == dic_model:
889	param_list = self.page_finder[id].get_param2fit()
890	break
891
892	param_cbox = self.constraints_list[-1].param_cbox
893	param_cbox.Clear()
894	# insert only fittable paramaters
895	for param in param_list:
896	param_cbox.Append(str(param), model)
897	param_cbox.Show(True)
898
899	bt_remove = self.constraints_list[-1].btRemove
900	bt_remove.Show(True)
901	self.btAdd.Show(True)
902	# self.Layout()
903	self.FitInside()
904
905	def _on_select_param(self, event):
906	"""
907	Store the appropriate constraint in the page_finder
908	"""
909	# This way PC/MAC both work, instead of using event.GetClientData().
910	# n = self.param_cbox.GetCurrentSelection()
911	# model = self.param_cbox.GetClientData(n)
912	# param = event.GetString()
913
914	if self.constraints_list:
915	self.constraints_list[-1].egal_txt.Show(True)
916	self.constraints_list[-1].constraint.Show(True)
917
918	def _on_add_constraint(self, event):
919	"""
920	Add another line for constraint
921	"""
922	if not self.show_constraint.GetValue():
923	msg = " Select Yes to add Constraint "
924	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
925	return
926	# check that a constraint is added
927	# before allow to add another constraint
928	for item in self.constraints_list:
929	if item.model_cbox.GetString(0) == "":
930	msg = " Select a model Name! "
931	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
932	return
933	if item.param_cbox.GetString(0) == "":
934	msg = " Select a parameter Name! "
935	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
936	return
937	if item.constraint.GetValue().lstrip().rstrip() == "":
938	model = item.param_cbox.GetClientData(
939	item.param_cbox.GetCurrentSelection())
940	if model is not None:
941	msg = " Enter a constraint for %s.%s! " % (model.name,
942	item.param_cbox.GetString(0))
943	else:
944	msg = " Enter a constraint"
945	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
946	return
947	# some model or parameters can be constrained
948	self._show_constraint()
949	self.FitInside()
950
951	def _set_constraint(self):
952	"""
953	get values from the constraint textcrtl ,parses them into model name
954	parameter name and parameters values.
955	store them in a list self.params .when when params is not empty
956	set_model uses it to reset the appropriate model
957	and its appropriates parameters
958	"""
959	for item in self.constraints_list:
960	select0 = item.model_cbox.GetSelection()
961	if select0 == wx.NOT_FOUND:
962	continue
963	model = item.model_cbox.GetClientData(select0)
964	select1 = item.param_cbox.GetSelection()
965	if select1 == wx.NOT_FOUND:
966	continue
967	param = item.param_cbox.GetString(select1)
968	constraint = item.constraint.GetValue().lstrip().rstrip()
969	if param.lstrip().rstrip() == "":
970	param = None
971	msg = " Constraint will be ignored!. missing parameters"
972	msg += " in combobox to set constraint! "
973	wx.PostEvent(self.parent.parent, StatusEvent(status=msg))
974	for id, value in self.constraint_dict.iteritems():
975	if model == value:
976	if constraint == "":
977	msg = " Constraint will be ignored!. missing value"
978	msg += " in textcrtl to set constraint! "
979	wx.PostEvent(self.parent.parent,
980	StatusEvent(status=msg))
981	constraint = None
982	if str(param) in self.page_finder[id].get_param2fit():
983	msg = " Checking constraint for parameter: %s ", param
984	wx.PostEvent(self.parent.parent,
985	StatusEvent(info="info", status=msg))
986	else:
987	model_name = item[0].GetLabel()
988	fitpage = self.page_finder[id].get_fit_tab_caption()
989	msg = "All constrainted parameters must be set "
990	msg += " adjustable: '%s.%s' " % (model_name, param)
991	msg += "is NOT checked in '%s'. " % fitpage
992	msg += " Please check it to fit or"
993	msg += " remove the line of the constraint."
994	wx.PostEvent(self.parent.parent,
995	StatusEvent(info="error", status=msg))
996	return False
997
998	for fid in self.page_finder[id].iterkeys():
999	# wrap in param/constraint in str() to remove unicode
1000	self.page_finder[id].set_model_param(str(param),
1001	str(constraint), fid=fid)
1002	break
1003	return True
1004
1005	def on_set_focus(self, event=None):
1006	"""
1007	The derivative class is on focus if implemented
1008	"""
1009	if self.parent is not None:
1010	if self.parent.parent is not None:
1011	wx.PostEvent(self.parent.parent, PanelOnFocusEvent(panel=self))
1012	self.page_finder = self.parent._manager.get_page_finder()
1013
1014
1015	def setComboBoxItems(cbox, items):
1016	assert isinstance(cbox, wx.ComboBox)
1017	selected = cbox.GetStringSelection()
1018	cbox.Clear()
1019	for k, (name, value) in enumerate(items):
1020	cbox.Append(name, value)
1021	cbox.SetStringSelection(selected)
1022
1023
1024	class SimFitPageState:
1025	"""
1026	State of the simultaneous fit page for saving purposes
1027	"""
1028
1029	def __init__(self):
1030	# Sim Fit Page Number
1031	self.fit_page_no = None
1032	# Select all data
1033	self.select_all = False
1034	# Data sets sent to fit page
1035	self.model_list = []
1036	# Data sets to be fit
1037	self.model_to_fit = []
1038	# Number of constraints
1039	self.no_constraint = 0
1040	# Dictionary of constraints
1041	self.constraint_dict = {}
1042	# List of constraints
1043	self.constraints_list = []
1044
1045	def load_from_save_state(self, fit):
1046	"""
1047	Load in a simultaneous/constrained fit from a save state
1048	:param fit: Fitpanel object
1049	:return: None
1050	"""
1051
1052	model_map = {}
1053	if fit.fit_panel.sim_page is None:
1054	fit.fit_panel.add_sim_page()
1055	sim_page = fit.fit_panel.sim_page
1056
1057	# Process each model and associate old M# with new M#
1058	i = 0
1059	for model in sim_page.model_list:
1060	model_id = self._format_id(model[1].keys()[0])
1061	for saved_model in self.model_list:
1062	save_id = saved_model.pop('name')
1063	saved_model['name'] = save_id
1064	save_id = self._format_id(save_id)
1065	if save_id == model_id:
1066	model_map[saved_model.pop('fit_page_source')] = \
1067	model[3].name
1068	check = bool(saved_model.pop('checked'))
1069	sim_page.model_list[i][0].SetValue(check)
1070	break
1071	i += 1
1072	sim_page.check_model_name(None)
1073
1074	if len(self.constraints_list) > 0:
1075	sim_page.hide_constraint.SetValue(False)
1076	sim_page.show_constraint.SetValue(True)
1077	sim_page._display_constraint(None)
1078
1079	for index, item in enumerate(self.constraints_list):
1080	model_cbox = item.pop('model_cbox')
1081	if model_cbox != "":
1082	constraint_value = item.pop('constraint')
1083	param = item.pop('param_cbox')
1084	equality = item.pop('egal_txt')
1085	for key, value in model_map.iteritems():
1086	model_cbox.replace(key, value)
1087	constraint_value.replace(key, value)
1088
1089	sim_page.constraints_list[index][0].SetValue(model_cbox)
1090	sim_page._on_select_model(None)
1091	sim_page.constraints_list[index][1].SetValue(param)
1092	sim_page.constraints_list[index][2].SetLabel(equality)
1093	sim_page.constraints_list[index][3].SetValue(constraint_value)
1094	sim_page._on_add_constraint(None)
1095
1096	def _format_id(self, original_id):
1097	original_id = original_id.rstrip('1234567890.')
1098	new_id_list = original_id.split()
1099	new_id = ' '.join(new_id_list)
1100	return new_id

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

SasView

source: sasview/src/sas/sasgui/perspectives/fitting/simfitpage.py @ e28f34d

Download in other formats: