model_editor.py @ cb44d66

ticket-1094-headless

Last change on this file since cb44d66 was 1752f38, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago
Added newlines to import statements in the plugin model template.
Property mode set to `100644`
File size: 41.2 KB

Line
1	'''
2	This module provides three model editor classes: the composite model editor,
3	the easy editor which provides a simple interface with tooltip help to enter
4	the parameters of the model and their default value and a panel to input a
5	function of y (usually the intensity). It also provides a drop down of
6	standard available math functions. Finally a full python editor panel for
7	complete customization is provided.
8
9	:TODO the writing of the file and name checking (and maybe some other
10	functions?) should be moved to a computational module which could be called
11	from a python script. Basically one just needs to pass the name,
12	description text and function text (or in the case of the composite editor
13	the names of the first and second model and the operator to be used).
14	'''
15
16	################################################################################
17	#This software was developed by the University of Tennessee as part of the
18	#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
19	#project funded by the US National Science Foundation.
20	#
21	#See the license text in license.txt
22	#
23	#copyright 2009, University of Tennessee
24	################################################################################
25	from __future__ import print_function
26
27	import wx
28	import sys
29	import os
30	import math
31	import re
32	import logging
33	import datetime
34
35	from wx.py.editwindow import EditWindow
36
37	from sas.sasgui.guiframe.documentation_window import DocumentationWindow
38
39	from .pyconsole import show_model_output, check_model
40
41	logger = logging.getLogger(__name__)
42
43	if sys.platform.count("win32") > 0:
44	FONT_VARIANT = 0
45	PNL_WIDTH = 500
46	PNL_HEIGHT = 320
47	else:
48	FONT_VARIANT = 1
49	PNL_WIDTH = 500
50	PNL_HEIGHT = 350
51	M_NAME = 'Model'
52	EDITOR_WIDTH = 800
53	EDITOR_HEIGTH = 735
54	PANEL_WIDTH = 500
55	_BOX_WIDTH = 55
56
57	def _delete_file(path):
58	"""
59	Delete file in the path
60	"""
61	try:
62	os.remove(path)
63	except:
64	raise
65
66
67	class TextDialog(wx.Dialog):
68	"""
69	Dialog for easy custom composite models. Provides a wx.Dialog panel
70	to choose two existing models (including pre-existing Plugin Models which
71	may themselves be composite models) as well as an operation on those models
72	(add or multiply) the resulting model will add a scale parameter for summed
73	models and a background parameter for a multiplied model.
74
75	The user also gives a brief help for the model in a description box and
76	must provide a unique name which is verified as unique before the new
77	model is saved.
78
79	This Dialog pops up for the user when they press 'Sum\|Multi(p1,p2)' under
80	'Plugin Model Operations' under 'Fitting' menu. This is currently called as
81	a Modal Dialog.
82
83	:TODO the built in compiler currently balks at when it tries to import
84	a model whose name contains spaces or symbols (such as + ... underscore
85	should be fine). Have fixed so the editor cannot save such a file name
86	but if a file is dropped in the plugin directory from outside this class
87	will create a file that cannot be compiled. Should add the check to
88	the write method or to the on_modelx method.
89
90	- PDB:April 5, 2015
91	"""
92	def __init__(self, parent=None, base=None, id=None, title='',
93	model_list=[], plugin_dir=None):
94	"""
95	This class is run when instatiated. The __init__ initializes and
96	calls the internal methods necessary. On exiting the wx.Dialog
97	window should be destroyed.
98	"""
99	wx.Dialog.__init__(self, parent=parent, id=id,
100	title=title, size=(PNL_WIDTH, PNL_HEIGHT))
101	self.parent = base
102	#Font
103	self.SetWindowVariant(variant=FONT_VARIANT)
104	# default
105	self.overwrite_name = False
106	self.plugin_dir = plugin_dir
107	self.model_list = model_list
108	self.model1_string = "sphere"
109	self.model2_string = "cylinder"
110	self.name = 'Sum' + M_NAME
111	self._notes = ''
112	self._operator = '+'
113	self._operator_choice = None
114	self.explanation = ''
115	self.explanationctr = None
116	self.type = None
117	self.name_sizer = None
118	self.name_tcl = None
119	self.desc_sizer = None
120	self.desc_tcl = None
121	self._selection_box = None
122	self.model1 = None
123	self.model2 = None
124	self.static_line_1 = None
125	self.ok_button = None
126	self.close_button = None
127	self._msg_box = None
128	self.msg_sizer = None
129	self.fname = None
130	self.cm_list = None
131	self.is_p1_custom = False
132	self.is_p2_custom = False
133	self._build_sizer()
134	self.model1_name = str(self.model1.GetValue())
135	self.model2_name = str(self.model2.GetValue())
136	self.good_name = True
137	self.fill_operator_combox()
138
139	def _layout_name(self):
140	"""
141	Do the layout for file/function name related widgets
142	"""
143	#container for new model name input
144	self.name_sizer = wx.BoxSizer(wx.HORIZONTAL)
145
146	#set up label and input box with tool tip and event handling
147	name_txt = wx.StaticText(self, -1, 'Function Name : ')
148	self.name_tcl = wx.TextCtrl(self, -1, value='MySumFunction')
149	self.name_tcl.Bind(wx.EVT_TEXT_ENTER, self.on_change_name)
150	hint_name = "Unique Sum/Multiply Model Function Name."
151	self.name_tcl.SetToolTipString(hint_name)
152
153	self.name_sizer.AddMany([(name_txt, 0, wx.LEFT \| wx.TOP, 10),
154	(self.name_tcl, -1,
155	wx.EXPAND \| wx.RIGHT \| wx.TOP \| wx.BOTTOM,
156	10)])
157
158
159	def _layout_description(self):
160	"""
161	Do the layout for description related widgets
162	"""
163	#container for new model description input
164	self.desc_sizer = wx.BoxSizer(wx.HORIZONTAL)
165
166	#set up description label and input box with tool tip and event handling
167	desc_txt = wx.StaticText(self, -1, 'Description (optional) : ')
168	self.desc_tcl = wx.TextCtrl(self, -1)
169	hint_desc = "Write a short description of this model function."
170	self.desc_tcl.SetToolTipString(hint_desc)
171
172	self.desc_sizer.AddMany([(desc_txt, 0, wx.LEFT \| wx.TOP, 10),
173	(self.desc_tcl, -1,
174	wx.EXPAND \| wx.RIGHT \| wx.TOP \| wx.BOTTOM,
175	10)])
176
177
178	def _layout_model_selection(self):
179	"""
180	Do the layout for model selection related widgets
181	"""
182	box_width = 195 # combobox width
183
184	#First set up main sizer for the selection
185	selection_box_title = wx.StaticBox(self, -1, 'Select',
186	size=(PNL_WIDTH - 30, 70))
187	self._selection_box = wx.StaticBoxSizer(selection_box_title,
188	wx.VERTICAL)
189
190	#Next create the help labels for the model selection
191	select_help_box = wx.BoxSizer(wx.HORIZONTAL)
192	model_string = " Model%s (p%s):"
193	select_help_box.Add(wx.StaticText(self, -1, model_string % (1, 1)),
194	0, 0)
195	select_help_box.Add((box_width - 25, 10), 0, 0)
196	select_help_box.Add(wx.StaticText(self, -1, model_string % (2, 2)),
197	0, 0)
198	self._selection_box.Add(select_help_box, 0, 0)
199
200	#Next create the actual selection box with 3 combo boxes
201	selection_box_choose = wx.BoxSizer(wx.HORIZONTAL)
202
203	self.model1 = wx.ComboBox(self, -1, style=wx.CB_READONLY)
204	wx.EVT_COMBOBOX(self.model1, -1, self.on_model1)
205	self.model1.SetMinSize((box_width * 5 / 6, -1))
206	self.model1.SetToolTipString("model1")
207
208	self._operator_choice = wx.ComboBox(self, -1, size=(50, -1),
209	style=wx.CB_READONLY)
210	wx.EVT_COMBOBOX(self._operator_choice, -1, self.on_select_operator)
211	operation_tip = "Add: +, Multiply: * "
212	self._operator_choice.SetToolTipString(operation_tip)
213
214	self.model2 = wx.ComboBox(self, -1, style=wx.CB_READONLY)
215	wx.EVT_COMBOBOX(self.model2, -1, self.on_model2)
216	self.model2.SetMinSize((box_width * 5 / 6, -1))
217	self.model2.SetToolTipString("model2")
218	self._set_model_list()
219
220	selection_box_choose.Add(self.model1, 0, 0)
221	selection_box_choose.Add((15, 10))
222	selection_box_choose.Add(self._operator_choice, 0, 0)
223	selection_box_choose.Add((15, 10))
224	selection_box_choose.Add(self.model2, 0, 0)
225	# add some space between labels and selection
226	self._selection_box.Add((20, 5), 0, 0)
227	self._selection_box.Add(selection_box_choose, 0, 0)
228
229	def _build_sizer(self):
230	"""
231	Build GUI with calls to _layout_name, _layout Description
232	and _layout_model_selection which each build a their portion of the
233	GUI.
234	"""
235	mainsizer = wx.BoxSizer(wx.VERTICAL) # create main sizer for dialog
236
237	# build fromm top by calling _layout_name and _layout_description
238	# and adding to main sizer
239	self._layout_name()
240	mainsizer.Add(self.name_sizer, 0, wx.EXPAND)
241	self._layout_description()
242	mainsizer.Add(self.desc_sizer, 0, wx.EXPAND)
243
244	# Add an explanation of dialog (short help)
245	self.explanationctr = wx.StaticText(self, -1, self.explanation)
246	self.fill_explanation_helpstring(self._operator)
247	mainsizer.Add(self.explanationctr, 0, wx.LEFT \| wx.EXPAND, 15)
248
249	# Add the selection box stuff with border and labels built
250	# by _layout_model_selection
251	self._layout_model_selection()
252	mainsizer.Add(self._selection_box, 0, wx.LEFT, 15)
253
254	# Add a space and horizontal line before the notification
255	#messages and the buttons at the bottom
256	mainsizer.Add((10, 10))
257	self.static_line_1 = wx.StaticLine(self, -1)
258	mainsizer.Add(self.static_line_1, 0, wx.EXPAND, 10)
259
260	# Add action status notification line (null at startup)
261	self._msg_box = wx.StaticText(self, -1, self._notes)
262	self.msg_sizer = wx.BoxSizer(wx.HORIZONTAL)
263	self.msg_sizer.Add(self._msg_box, 0, wx.LEFT, 0)
264	mainsizer.Add(self.msg_sizer, 0,
265	wx.LEFT \| wx.RIGHT \| wx.ADJUST_MINSIZE \| wx.BOTTOM, 10)
266
267	# Finally add the buttons (apply and close) on the bottom
268	# Eventually need to add help here
269	self.ok_button = wx.Button(self, wx.ID_OK, 'Apply')
270	_app_tip = "Save the new Model."
271	self.ok_button.SetToolTipString(_app_tip)
272	self.ok_button.Bind(wx.EVT_BUTTON, self.check_name)
273	self.help_button = wx.Button(self, -1, 'HELP')
274	_app_tip = "Help on composite model creation."
275	self.help_button.SetToolTipString(_app_tip)
276	self.help_button.Bind(wx.EVT_BUTTON, self.on_help)
277	self.close_button = wx.Button(self, wx.ID_CANCEL, 'Close')
278	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
279	sizer_button.AddMany([((20, 20), 1, 0),
280	(self.ok_button, 0, 0),
281	(self.help_button, 0, 0),
282	(self.close_button, 0, wx.LEFT \| wx.RIGHT, 10)])
283	mainsizer.Add(sizer_button, 0, wx.EXPAND \| wx.BOTTOM \| wx.TOP, 10)
284
285	self.SetSizer(mainsizer)
286	self.Centre()
287
288	def on_change_name(self, event=None):
289	"""
290	Change name
291	"""
292	if event is not None:
293	event.Skip()
294	self.name_tcl.SetBackgroundColour('white')
295	self.Refresh()
296
297	def check_name(self, event=None):
298	"""
299	Check that proposed new model name is a valid Python module name
300	and that it does not already exist. If not show error message and
301	pink background in text box else call on_apply
302
303	:TODO this should be separated out from the GUI code. For that we
304	need to pass it the name (or if we want to keep the default name
305	option also need to pass the self._operator attribute) We just need
306	the function to return an error code that the name is good or if
307	not why (not a valid name, name exists already). The rest of the
308	error handling should be done in this module. so on_apply would then
309	start by checking the name and then either raise errors or do the
310	deed.
311	"""
312	#Get the function/file name
313	mname = M_NAME
314	self.on_change_name(None)
315	title = self.name_tcl.GetValue().lstrip().rstrip()
316	if title == '':
317	text = self._operator
318	if text.count('+') > 0:
319	mname = 'Sum'
320	else:
321	mname = 'Multi'
322	mname += M_NAME
323	title = mname
324	self.name = title
325	t_fname = title + '.py'
326
327	#First check if the name is a valid Python name
328	if re.match('^[A-Za-z0-9_]*$', title):
329	self.good_name = True
330	else:
331	self.good_name = False
332	msg = ("%s is not a valid Python name. Only alphanumeric \n" \
333	"and underscore allowed" % self.name)
334
335	#Now check if the name already exists
336	if not self.overwrite_name and self.good_name:
337	#Create list of existing model names for comparison
338	list_fnames = os.listdir(self.plugin_dir)
339	# fake existing regular model name list
340	m_list = [model + ".py" for model in self.model_list]
341	list_fnames.append(m_list)
342	if t_fname in list_fnames and title != mname:
343	self.good_name = False
344	msg = "Name exists already."
345
346	if not self.good_name:
347	self.name_tcl.SetBackgroundColour('pink')
348	info = 'Error'
349	wx.MessageBox(msg, info)
350	self._notes = msg
351	color = 'red'
352	self._msg_box.SetLabel(msg)
353	self._msg_box.SetForegroundColour(color)
354	return self.good_name
355	self.fname = os.path.join(self.plugin_dir, t_fname)
356	s_title = title
357	if len(title) > 20:
358	s_title = title[0:19] + '...'
359	self._notes = "Model function (%s) has been set! \n" % str(s_title)
360	self.good_name = True
361	self.on_apply(self.fname)
362	return self.good_name
363
364	def on_apply(self, path):
365	"""
366	This method is a misnomer - it is not bound to the apply button
367	event. Instead the apply button event goes to check_name which
368	then calls this method if the name of the new file is acceptable.
369
370	:TODO this should be bound to the apply button. The first line
371	should call the check_name method which itself should be in another
372	module separated from the the GUI modules.
373	"""
374	self.name_tcl.SetBackgroundColour('white')
375	try:
376	label = self.get_textnames()
377	fname = path
378	name1 = label[0]
379	name2 = label[1]
380	self.write_string(fname, name1, name2)
381	success = show_model_output(self, fname)
382	if success:
383	self.parent.update_custom_combo()
384	msg = self._notes
385	info = 'Info'
386	color = 'blue'
387	except:
388	msg = "Easy Sum/Multipy Plugin: Error occurred..."
389	info = 'Error'
390	color = 'red'
391	self._msg_box.SetLabel(msg)
392	self._msg_box.SetForegroundColour(color)
393	self._set_model_list()
394	if self.parent.parent is not None:
395	from sas.sasgui.guiframe.events import StatusEvent
396	wx.PostEvent(self.parent.parent, StatusEvent(status=msg,
397	info=info))
398
399	def on_help(self, event):
400	"""
401	Bring up the Composite Model Editor Documentation whenever
402	the HELP button is clicked.
403
404	Calls DocumentationWindow with the path of the location within the
405	documentation tree (after /doc/ ....". Note that when using old
406	versions of Wx (before 2.9) and thus not the release version of
407	installers, the help comes up at the top level of the file as
408	webbrowser does not pass anything past the # to the browser when it is
409	running "file:///...."
410
411	:param evt: Triggers on clicking the help button
412	"""
413
414	_TreeLocation = "user/sasgui/perspectives/fitting/fitting_help.html"
415	_PageAnchor = "#sum-multi-p1-p2"
416	_doc_viewer = DocumentationWindow(self, -1, _TreeLocation, _PageAnchor,
417	"Composite Model Editor Help")
418
419	def _set_model_list(self):
420	"""
421	Set the list of models
422	"""
423	# list of model names
424	# get regular models
425	main_list = self.model_list
426	# get custom models
427	self.update_cm_list()
428	# add custom models to model list
429	for name in self.cm_list:
430	if name not in main_list:
431	main_list.append(name)
432
433	if len(main_list) > 1:
434	main_list.sort()
435	self.model1.Clear()
436	self.model2.Clear()
437	for idx in range(len(main_list)):
438	self.model1.Append(str(main_list[idx]), idx)
439	self.model2.Append(str(main_list[idx]), idx)
440	self.model1.SetStringSelection(self.model1_string)
441	self.model2.SetStringSelection(self.model2_string)
442
443	def update_cm_list(self):
444	"""
445	Update custom model list
446	"""
447	cm_list = []
448	al_list = os.listdir(self.plugin_dir)
449	for c_name in al_list:
450	if c_name.split('.')[-1] == 'py' and \
451	c_name.split('.')[0] != '__init__':
452	name = str(c_name.split('.')[0])
453	cm_list.append(name)
454	self.cm_list = cm_list
455
456	def on_model1(self, event):
457	"""
458	Set model1
459	"""
460	event.Skip()
461	self.update_cm_list()
462	self.model1_name = str(self.model1.GetValue())
463	self.model1_string = self.model1_name
464	if self.model1_name in self.cm_list:
465	self.is_p1_custom = True
466	else:
467	self.is_p1_custom = False
468
469	def on_model2(self, event):
470	"""
471	Set model2
472	"""
473	event.Skip()
474	self.update_cm_list()
475	self.model2_name = str(self.model2.GetValue())
476	self.model2_string = self.model2_name
477	if self.model2_name in self.cm_list:
478	self.is_p2_custom = True
479	else:
480	self.is_p2_custom = False
481
482	def on_select_operator(self, event=None):
483	"""
484	On Select an Operator
485	"""
486	# For Mac
487	if event is not None:
488	event.Skip()
489	item = event.GetEventObject()
490	text = item.GetValue()
491	self.fill_explanation_helpstring(text)
492
493	def fill_explanation_helpstring(self, operator):
494	"""
495	Choose the equation to use depending on whether we now have
496	a sum or multiply model then create the appropriate string
497
498	for the sum model the result will be:
499	scale_factor * (scale1 * model_1 + scale2 * model_2) + background
500	while for the multiply model it will just be:
501	scale_factor * (model_1* model_2) + background
502	"""
503	name = ''
504	if operator == '*':
505	name = 'Multi'
506	factor_1 = ''
507	factor_2 = ''
508	else:
509	name = 'Sum'
510	factor_1 = 'scale_1 * '
511	factor_2 = 'scale_2 * '
512
513	self._operator = operator
514	self.explanation = (" Plugin_model = scale_factor * ({}model_1 {} "
515	"{}model_2) + background").format(factor_1,operator,factor_2)
516	self.explanationctr.SetLabel(self.explanation)
517	self.name = name + M_NAME
518
519
520	def fill_operator_combox(self):
521	"""
522	fill the current combobox with the operator
523	"""
524	operator_list = ['+', '*']
525	for oper in operator_list:
526	pos = self._operator_choice.Append(str(oper))
527	self._operator_choice.SetClientData(pos, str(oper))
528	self._operator_choice.SetSelection(0)
529
530	def get_textnames(self):
531	"""
532	Returns model name string as list
533	"""
534	return [self.model1_name, self.model2_name]
535
536	def write_string(self, fname, model1_name, model2_name):
537	"""
538	Write and Save file
539	"""
540	self.fname = fname
541	description = self.desc_tcl.GetValue().lstrip().rstrip()
542	desc_line = ''
543	if description.strip() != '':
544	# Sasmodels generates a description for us. If the user provides
545	# their own description, add a line to overwrite the sasmodels one
546	desc_line = "\nmodel_info.description = '{}'".format(description)
547	name = os.path.splitext(os.path.basename(self.fname))[0]
548	output = SUM_TEMPLATE.format(name=name, model1=model1_name,
549	model2=model2_name, operator=self._operator, desc_line=desc_line)
550	with open(self.fname, 'w') as out_f:
551	out_f.write(output)
552
553	def compile_file(self, path):
554	"""
555	Compile the file in the path
556	"""
557	path = self.fname
558	show_model_output(self, path)
559
560	def delete_file(self, path):
561	"""
562	Delete file in the path
563	"""
564	_delete_file(path)
565
566
567	class EditorPanel(wx.ScrolledWindow):
568	"""
569	Simple Plugin Model function editor
570	"""
571	def __init__(self, parent, base, path, title, args, *kwds):
572	kwds['name'] = title
573	# kwds["size"] = (EDITOR_WIDTH, EDITOR_HEIGTH)
574	kwds["style"] = wx.FULL_REPAINT_ON_RESIZE
575	wx.ScrolledWindow.__init__(self, parent, args, *kwds)
576	self.SetScrollbars(1,1,1,1)
577	self.parent = parent
578	self.base = base
579	self.path = path
580	self.font = wx.SystemSettings_GetFont(wx.SYS_SYSTEM_FONT)
581	self.font.SetPointSize(10)
582	self.reader = None
583	self.name = 'untitled'
584	self.overwrite_name = False
585	self.is_2d = False
586	self.fname = None
587	self.main_sizer = None
588	self.name_sizer = None
589	self.name_hsizer = None
590	self.name_tcl = None
591	self.overwrite_cb = None
592	self.desc_sizer = None
593	self.desc_tcl = None
594	self.param_sizer = None
595	self.param_tcl = None
596	self.function_sizer = None
597	self.func_horizon_sizer = None
598	self.button_sizer = None
599	self.param_strings = ''
600	self.function_strings = ''
601	self._notes = ""
602	self._msg_box = None
603	self.msg_sizer = None
604	self.warning = ""
605	#This does not seem to be used anywhere so commenting out for now
606	# -- PDB 2/26/17
607	#self._description = "New Plugin Model"
608	self.function_tcl = None
609	self.math_combo = None
610	self.bt_apply = None
611	self.bt_close = None
612	#self._default_save_location = os.getcwd()
613	self._do_layout()
614
615
616
617	def _define_structure(self):
618	"""
619	define initial sizer
620	"""
621	#w, h = self.parent.GetSize()
622	self.main_sizer = wx.BoxSizer(wx.VERTICAL)
623	self.name_sizer = wx.BoxSizer(wx.VERTICAL)
624	self.name_hsizer = wx.BoxSizer(wx.HORIZONTAL)
625	self.desc_sizer = wx.BoxSizer(wx.VERTICAL)
626	self.param_sizer = wx.BoxSizer(wx.VERTICAL)
627	self.function_sizer = wx.BoxSizer(wx.VERTICAL)
628	self.func_horizon_sizer = wx.BoxSizer(wx.HORIZONTAL)
629	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
630	self.msg_sizer = wx.BoxSizer(wx.HORIZONTAL)
631
632	def _layout_name(self):
633	"""
634	Do the layout for file/function name related widgets
635	"""
636	#title name [string]
637	name_txt = wx.StaticText(self, -1, 'Function Name : ')
638	self.overwrite_cb = wx.CheckBox(self, -1, "Overwrite existing plugin model of this name?", (10, 10))
639	self.overwrite_cb.SetValue(False)
640	self.overwrite_cb.SetToolTipString("Overwrite it if already exists?")
641	wx.EVT_CHECKBOX(self, self.overwrite_cb.GetId(), self.on_over_cb)
642	self.name_tcl = wx.TextCtrl(self, -1, size=(PANEL_WIDTH * 3 / 5, -1))
643	self.name_tcl.Bind(wx.EVT_TEXT_ENTER, self.on_change_name)
644	self.name_tcl.SetValue('')
645	self.name_tcl.SetFont(self.font)
646	hint_name = "Unique Model Function Name."
647	self.name_tcl.SetToolTipString(hint_name)
648	self.name_hsizer.AddMany([(self.name_tcl, 0, wx.LEFT \| wx.TOP, 0),
649	(self.overwrite_cb, 0, wx.LEFT, 20)])
650	self.name_sizer.AddMany([(name_txt, 0, wx.LEFT \| wx.TOP, 10),
651	(self.name_hsizer, 0,
652	wx.LEFT \| wx.TOP \| wx.BOTTOM, 10)])
653
654
655	def _layout_description(self):
656	"""
657	Do the layout for description related widgets
658	"""
659	#title name [string]
660	desc_txt = wx.StaticText(self, -1, 'Description (optional) : ')
661	self.desc_tcl = wx.TextCtrl(self, -1, size=(PANEL_WIDTH * 3 / 5, -1))
662	self.desc_tcl.SetValue('')
663	hint_desc = "Write a short description of the model function."
664	self.desc_tcl.SetToolTipString(hint_desc)
665	self.desc_sizer.AddMany([(desc_txt, 0, wx.LEFT \| wx.TOP, 10),
666	(self.desc_tcl, 0,
667	wx.LEFT \| wx.TOP \| wx.BOTTOM, 10)])
668	def _layout_param(self):
669	"""
670	Do the layout for parameter related widgets
671	"""
672	param_txt = wx.StaticText(self, -1, 'Fit Parameters: ')
673
674	param_tip = "#Set the parameters and their initial values.\n"
675	param_tip += "#Example:\n"
676	param_tip += "A = 1\nB = 1"
677	#param_txt.SetToolTipString(param_tip)
678	newid = wx.NewId()
679	self.param_tcl = EditWindow(self, newid, wx.DefaultPosition,
680	wx.DefaultSize,
681	wx.CLIP_CHILDREN \| wx.SUNKEN_BORDER)
682	self.param_tcl.setDisplayLineNumbers(True)
683	self.param_tcl.SetToolTipString(param_tip)
684
685	self.param_sizer.AddMany([(param_txt, 0, wx.LEFT, 10),
686	(self.param_tcl, 1, wx.EXPAND \| wx.ALL, 10)])
687
688
689	def _layout_function(self):
690	"""
691	Do the layout for function related widgets
692	"""
693	function_txt = wx.StaticText(self, -1, 'Function(x) : ')
694	hint_function = "#Example:\n"
695	hint_function += "if x <= 0:\n"
696	hint_function += " y = A + B\n"
697	hint_function += "else:\n"
698	hint_function += " y = A + B * cos(2 * pi * x)\n"
699	hint_function += "return y\n"
700	math_txt = wx.StaticText(self, -1, '*Useful math functions: ')
701	math_combo = self._fill_math_combo()
702
703	newid = wx.NewId()
704	self.function_tcl = EditWindow(self, newid, wx.DefaultPosition,
705	wx.DefaultSize,
706	wx.CLIP_CHILDREN \| wx.SUNKEN_BORDER)
707	self.function_tcl.setDisplayLineNumbers(True)
708	self.function_tcl.SetToolTipString(hint_function)
709
710	self.func_horizon_sizer.Add(function_txt)
711	self.func_horizon_sizer.Add(math_txt, 0, wx.LEFT, 250)
712	self.func_horizon_sizer.Add(math_combo, 0, wx.LEFT, 10)
713
714	self.function_sizer.Add(self.func_horizon_sizer, 0, wx.LEFT, 10)
715	self.function_sizer.Add(self.function_tcl, 1, wx.EXPAND \| wx.ALL, 10)
716
717	def _layout_msg(self):
718	"""
719	Layout msg
720	"""
721	self._msg_box = wx.StaticText(self, -1, self._notes,
722	size=(PANEL_WIDTH, -1))
723	self.msg_sizer.Add(self._msg_box, 0, wx.LEFT, 10)
724
725	def _layout_button(self):
726	"""
727	Do the layout for the button widgets
728	"""
729	self.bt_apply = wx.Button(self, -1, "Apply", size=(_BOX_WIDTH, -1))
730	self.bt_apply.SetToolTipString("Save changes into the imported data.")
731	self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
732
733	self.bt_help = wx.Button(self, -1, "HELP", size=(_BOX_WIDTH, -1))
734	self.bt_help.SetToolTipString("Get Help For Model Editor")
735	self.bt_help.Bind(wx.EVT_BUTTON, self.on_help)
736
737	self.bt_close = wx.Button(self, -1, 'Close', size=(_BOX_WIDTH, -1))
738	self.bt_close.Bind(wx.EVT_BUTTON, self.on_close)
739	self.bt_close.SetToolTipString("Close this panel.")
740
741	self.button_sizer.AddMany([(self.bt_apply, 0,0),
742	(self.bt_help, 0, wx.LEFT \| wx.BOTTOM,15),
743	(self.bt_close, 0, wx.LEFT \| wx.RIGHT, 15)])
744
745	def _do_layout(self):
746	"""
747	Draw the current panel
748	"""
749	self._define_structure()
750	self._layout_name()
751	self._layout_description()
752	self._layout_param()
753	self._layout_function()
754	self._layout_msg()
755	self._layout_button()
756	self.main_sizer.AddMany([(self.name_sizer, 0, wx.EXPAND \| wx.ALL, 5),
757	(wx.StaticLine(self), 0,
758	wx.ALL \| wx.EXPAND, 5),
759	(self.desc_sizer, 0, wx.EXPAND \| wx.ALL, 5),
760	(wx.StaticLine(self), 0,
761	wx.ALL \| wx.EXPAND, 5),
762	(self.param_sizer, 1, wx.EXPAND \| wx.ALL, 5),
763	(wx.StaticLine(self), 0,
764	wx.ALL \| wx.EXPAND, 5),
765	(self.function_sizer, 2,
766	wx.EXPAND \| wx.ALL, 5),
767	(wx.StaticLine(self), 0,
768	wx.ALL \| wx.EXPAND, 5),
769	(self.msg_sizer, 0, wx.EXPAND \| wx.ALL, 5),
770	(self.button_sizer, 0, wx.ALIGN_RIGHT)])
771	self.SetSizer(self.main_sizer)
772	self.SetAutoLayout(True)
773
774	def _fill_math_combo(self):
775	"""
776	Fill up the math combo box
777	"""
778	self.math_combo = wx.ComboBox(self, -1, size=(100, -1),
779	style=wx.CB_READONLY)
780	for item in dir(math):
781	if item.count("_") < 1:
782	try:
783	exec "float(math.%s)" % item
784	self.math_combo.Append(str(item))
785	except Exception:
786	self.math_combo.Append(str(item) + "()")
787	self.math_combo.Bind(wx.EVT_COMBOBOX, self._on_math_select)
788	self.math_combo.SetSelection(0)
789	return self.math_combo
790
791	def _on_math_select(self, event):
792	"""
793	On math selection on ComboBox
794	"""
795	event.Skip()
796	label = self.math_combo.GetValue()
797	self.function_tcl.SetFocus()
798	# Put the text at the cursor position
799	pos = self.function_tcl.GetCurrentPos()
800	self.function_tcl.InsertText(pos, label)
801	# Put the cursor at appropriate position
802	length = len(label)
803	print(length)
804	if label[length-1] == ')':
805	length -= 1
806	f_pos = pos + length
807	self.function_tcl.GotoPos(f_pos)
808
809	def get_notes(self):
810	"""
811	return notes
812	"""
813	return self._notes
814
815	def on_change_name(self, event=None):
816	"""
817	Change name
818	"""
819	if event is not None:
820	event.Skip()
821	self.name_tcl.SetBackgroundColour('white')
822	self.Refresh()
823
824	def check_name(self):
825	"""
826	Check name if exist already
827	"""
828	self._notes = ''
829	self.on_change_name(None)
830	plugin_dir = self.path
831	list_fnames = os.listdir(plugin_dir)
832	# function/file name
833	title = self.name_tcl.GetValue().lstrip().rstrip()
834	self.name = title
835	t_fname = title + '.py'
836	if not self.overwrite_name:
837	if t_fname in list_fnames:
838	self.name_tcl.SetBackgroundColour('pink')
839	return False
840	self.fname = os.path.join(plugin_dir, t_fname)
841	s_title = title
842	if len(title) > 20:
843	s_title = title[0:19] + '...'
844	self._notes += "Model function name is set "
845	self._notes += "to %s. \n" % str(s_title)
846	return True
847
848	def on_over_cb(self, event):
849	"""
850	Set overwrite name flag on cb event
851	"""
852	if event is not None:
853	event.Skip()
854	cb_value = event.GetEventObject()
855	self.overwrite_name = cb_value.GetValue()
856
857	def on_click_apply(self, event):
858	"""
859	Changes are saved in data object imported to edit.
860
861	checks firs for valid name, then if it already exists then checks
862	that a function was entered and finally that if entered it contains at
863	least a return statement. If all passes writes file then tries to
864	compile. If compile fails or import module fails or run method fails
865	tries to remove any .py and pyc files that may have been created and
866	sets error message.
867
868	:todo this code still could do with a careful going over to clean
869	up and simplify. the non GUI methods such as this one should be removed
870	to computational code of SasView. Most of those computational methods
871	would be the same for both the simple editors.
872	"""
873	#must post event here
874	event.Skip()
875	name = self.name_tcl.GetValue().lstrip().rstrip()
876	info = 'Info'
877	msg = ''
878	result, check_err = '', ''
879	# Sort out the errors if occur
880	# First check for valid python name then if the name already exists
881	if not name or not bool(re.match('^[A-Za-z0-9_]*$', name)):
882	msg = '"%s" '%name
883	msg += "is not a valid model name. Name must not be empty and \n"
884	msg += "may include only alpha numeric or underline characters \n"
885	msg += "and no spaces"
886	elif self.check_name():
887	description = self.desc_tcl.GetValue()
888	param_str = self.param_tcl.GetText()
889	func_str = self.function_tcl.GetText()
890	# No input for the model function
891	if func_str.lstrip().rstrip():
892	if func_str.count('return') > 0:
893	self.write_file(self.fname, name, description, param_str,
894	func_str)
895	try:
896	result, msg = check_model(self.fname), None
897	except Exception:
898	import traceback
899	result, msg = None, "error building model"
900	check_err = "\n"+traceback.format_exc(limit=2)
901	else:
902	msg = "Error: The func(x) must 'return' a value at least.\n"
903	msg += "For example: \n\nreturn 2*x"
904	else:
905	msg = 'Error: Function is not defined.'
906	else:
907	msg = "Name exists already."
908
909	#
910	if self.base is not None and not msg:
911	self.base.update_custom_combo()
912
913	# Prepare the messagebox
914	if msg:
915	info = 'Error'
916	color = 'red'
917	self.overwrite_cb.SetValue(True)
918	self.overwrite_name = True
919	else:
920	self._notes = result
921	msg = "Successful! Please look for %s in Plugin Models."%name
922	msg += " " + self._notes
923	info = 'Info'
924	color = 'blue'
925	self._msg_box.SetLabel(msg)
926	self._msg_box.SetForegroundColour(color)
927	# Send msg to the top window
928	if self.base is not None:
929	from sas.sasgui.guiframe.events import StatusEvent
930	wx.PostEvent(self.base.parent,
931	StatusEvent(status=msg+check_err, info=info))
932	self.warning = msg
933
934	def write_file(self, fname, name, desc_str, param_str, func_str):
935	"""
936	Write content in file
937
938	:param fname: full file path
939	:param desc_str: content of the description strings
940	:param param_str: content of params; Strings
941	:param func_str: content of func; Strings
942	"""
943	out_f = open(fname, 'w')
944
945	out_f.write(CUSTOM_TEMPLATE % {
946	'name': name,
947	'title': 'User model for ' + name,
948	'description': desc_str,
949	'date': datetime.datetime.now().strftime('%YYYY-%mm-%dd'),
950	})
951
952	# Write out parameters
953	param_names = [] # to store parameter names
954	out_f.write('parameters = [ \n')
955	out_f.write('# ["name", "units", default, [lower, upper], "type", "description"],\n')
956	for pname, pvalue, desc in self.get_param_helper(param_str):
957	param_names.append(pname)
958	out_f.write(" ['%s', '', %s, [-inf, inf], '', '%s'],\n"
959	% (pname, pvalue, desc))
960	out_f.write(' ]\n')
961
962	# Write out function definition
963	out_f.write('\n')
964	out_f.write('def Iq(%s):\n' % ', '.join(['x'] + param_names))
965	out_f.write(' """Absolute scattering"""\n')
966	if "scipy." in func_str:
967	out_f.write(' import scipy\n')
968	if "numpy." in func_str:
969	out_f.write(' import numpy\n')
970	if "np." in func_str:
971	out_f.write(' import numpy as np\n')
972	for func_line in func_str.split('\n'):
973	out_f.write('%s%s\n' % (' ', func_line))
974	out_f.write('## uncomment the following if Iq works for vector x\n')
975	out_f.write('#Iq.vectorized = True\n')
976
977	# Create place holder for Iqxy
978	out_f.write('\n')
979	out_f.write('#def Iqxy(%s):\n' % ', '.join(["x", "y"] + param_names))
980	out_f.write('# """Absolute scattering of oriented particles."""\n')
981	out_f.write('# ...\n')
982	out_f.write('# return oriented_form(x, y, args)\n')
983	out_f.write('## uncomment the following if Iqxy works for vector x, y\n')
984	out_f.write('#Iqxy.vectorized = True\n')
985
986	out_f.close()
987
988	def get_param_helper(self, param_str):
989	"""
990	yield a sequence of name, value pairs for the parameters in param_str
991
992	Parameters can be defined by one per line by name=value, or multiple
993	on the same line by separating the pairs by semicolon or comma. The
994	value is optional and defaults to "1.0".
995	"""
996	for line in param_str.replace(';', ',').split('\n'):
997	for item in line.split(','):
998	defn, desc = item.split('#', 1) if '#' in item else (item, '')
999	name, value = defn.split('=', 1) if '=' in defn else (defn, '1.0')
1000	if name:
1001	yield [v.strip() for v in (name, value, desc)]
1002
1003	def set_function_helper(self, line):
1004	"""
1005	Get string in line to define the local params
1006
1007	:param line: one line of string got from the param_str
1008	"""
1009	params_str = ''
1010	spaces = ' '#8spaces
1011	items = line.split(";")
1012	for item in items:
1013	name = item.split("=")[0].lstrip().rstrip()
1014	params_str += spaces + "%s = self.params['%s']\n" % (name, name)
1015	return params_str
1016
1017	def get_warning(self):
1018	"""
1019	Get the warning msg
1020	"""
1021	return self.warning
1022
1023	def on_help(self, event):
1024	"""
1025	Bring up the New Plugin Model Editor Documentation whenever
1026	the HELP button is clicked.
1027
1028	Calls DocumentationWindow with the path of the location within the
1029	documentation tree (after /doc/ ....". Note that when using old
1030	versions of Wx (before 2.9) and thus not the release version of
1031	installers, the help comes up at the top level of the file as
1032	webbrowser does not pass anything past the # to the browser when it is
1033	running "file:///...."
1034
1035	:param evt: Triggers on clicking the help button
1036	"""
1037
1038	_TreeLocation = "user/sasgui/perspectives/fitting/fitting_help.html"
1039	_PageAnchor = "#new-plugin-model"
1040	_doc_viewer = DocumentationWindow(self, -1, _TreeLocation, _PageAnchor,
1041	"Plugin Model Editor Help")
1042
1043	def on_close(self, event):
1044	"""
1045	leave data as it is and close
1046	"""
1047	self.parent.Show(False)#Close()
1048	event.Skip()
1049
1050	class EditorWindow(wx.Frame):
1051	"""
1052	Editor Window
1053	"""
1054	def __init__(self, parent, base, path, title,
1055	size=(EDITOR_WIDTH, EDITOR_HEIGTH), args, *kwds):
1056	"""
1057	Init
1058	"""
1059	kwds["title"] = title
1060	kwds["size"] = size
1061	wx.Frame.__init__(self, parent=None, args, *kwds)
1062	self.parent = parent
1063	self.panel = EditorPanel(parent=self, base=parent,
1064	path=path, title=title)
1065	self.Show(True)
1066	wx.EVT_CLOSE(self, self.on_close)
1067
1068	def on_close(self, event):
1069	"""
1070	On close event
1071	"""
1072	self.Show(False)
1073	#if self.parent is not None:
1074	# self.parent.new_model_frame = None
1075	#self.Destroy()
1076
1077	## Templates for plugin models
1078
1079	CUSTOM_TEMPLATE = '''\
1080	r"""
1081	Definition
1082	----------
1083
1084	Calculates %(name)s.
1085
1086	%(description)s
1087
1088	References
1089	----------
1090
1091	Authorship and Verification
1092	---------------------------
1093
1094	* Author: --- Date: %(date)s
1095	* Last Modified by: --- Date: %(date)s
1096	* Last Reviewed by: --- Date: %(date)s
1097	"""
1098
1099	from math import *
1100	from numpy import inf
1101
1102	name = "%(name)s"
1103	title = "%(title)s"
1104	description = """%(description)s"""
1105
1106	'''
1107
1108	SUM_TEMPLATE = """
1109	from sasmodels.core import load_model_info
1110	from sasmodels.sasview_model import make_model_from_info
1111
1112	model_info = load_model_info('{model1}{operator}{model2}')
1113	model_info.name = '{name}'{desc_line}
1114	Model = make_model_from_info(model_info)
1115	"""
1116	if __name__ == "__main__":
1117	main_app = wx.App()
1118	main_frame = TextDialog(id=1, model_list=["SphereModel", "CylinderModel"],
1119	plugin_dir='../fitting/plugin_models')
1120	main_frame.ShowModal()
1121	main_app.MainLoop()

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

SasView

source: sasview/src/sas/sasgui/perspectives/calculator/model_editor.py @ cb44d66

Download in other formats: