model_editor.py @ f89fa84

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

Rev	Line
[6f140f2]	1	################################################################################
	2	#This software was developed by the University of Tennessee as part of the
	3	#Distributed Data Analysis of Neutron Scattering Experiments (DANSE)
	4	#project funded by the US National Science Foundation.
	5	#
	6	#See the license text in license.txt
	7	#
	8	#copyright 2009, University of Tennessee
	9	################################################################################
	10	import wx
	11	import sys
	12	import os
[318b5bbb]	13	import math
[6f140f2]	14	from wx.py.editwindow import EditWindow
	15
	16	if sys.platform.count("win32") > 0:
	17	FONT_VARIANT = 0
[fdef956]	18	PNL_WIDTH = 450
[41a8cb3]	19	PNL_HITE = 320
[6f140f2]	20	else:
	21	FONT_VARIANT = 1
[fdef956]	22	PNL_WIDTH = 590
[9e00363]	23	PNL_HITE = 350
[fdef956]	24	M_NAME = 'Model'
[6f140f2]	25	EDITOR_WIDTH = 800
[318b5bbb]	26	EDITOR_HEIGTH = 735
[6f140f2]	27	PANEL_WIDTH = 500
[fdef956]	28	_BOX_WIDTH = 55
[6f140f2]	29
	30
	31	def _compileFile(path):
	32	"""
	33	Compile the file in the path
	34	"""
	35	try:
	36	import py_compile
	37	py_compile.compile(file=path, doraise=True)
	38	return ''
	39	except:
[d970df9]	40	_, value, _ = sys.exc_info()
[6f140f2]	41	return value
	42
	43	def _deleteFile(path):
	44	"""
	45	Delete file in the path
	46	"""
	47	try:
	48	os.remove(path)
	49	except:
	50	raise
	51
	52
	53	class TextDialog(wx.Dialog):
	54	"""
	55	Dialog for easy custom sum models
	56	"""
[d970df9]	57	def __init__(self, parent=None, base=None, id=None, title='',
	58	model_list=[], plugin_dir=None):
[6f140f2]	59	"""
[d970df9]	60	Dialog window popup when selecting 'Easy Custom Sum/Multiply'
	61	on the menu
[6f140f2]	62	"""
	63	wx.Dialog.__init__(self, parent=parent, id=id,
	64	title=title, size=(PNL_WIDTH, PNL_HITE))
[796c4d4]	65	self.parent = base
[6f140f2]	66	#Font
	67	self.SetWindowVariant(variant=FONT_VARIANT)
	68	# default
[796c4d4]	69	self.font = wx.SystemSettings_GetFont(wx.SYS_SYSTEM_FONT)
	70	self.font.SetPointSize(10)
	71	self.overwrite_name = False
	72	self.plugin_dir = plugin_dir
[6f140f2]	73	self.model_list = model_list
	74	self.model1_string = "SphereModel"
	75	self.model2_string = "CylinderModel"
[fdef956]	76	self.name = 'Sum' + M_NAME
[d970df9]	77	self.factor = 'scale_factor'
[41a8cb3]	78	self._notes = ''
[0008f54]	79	self.operator = '+'
[fdef956]	80	self.operator_cbox = None
	81	self.explanation = ''
	82	self.explanationctr = None
	83	self.sizer = None
[d970df9]	84	self.name_sizer = None
	85	self.name_hsizer = None
	86	self.desc_sizer = None
	87	self.desc_tcl = None
	88	self.model1 = None
	89	self.model2 = None
	90	self.static_line_1 = None
	91	self.okButton = None
	92	self.closeButton = None
[41a8cb3]	93	self._msg_box = None
	94	self.msg_sizer = None
[d970df9]	95	self.fname = None
	96	self.cm_list = None
	97	self.is_p1_custom = False
	98	self.is_p2_custom = False
[6f140f2]	99	self._build_sizer()
	100	self.model1_name = str(self.model1.GetValue())
	101	self.model2_name = str(self.model2.GetValue())
[796c4d4]	102	self.good_name = True
[fdef956]	103	self.fill_oprator_combox()
[796c4d4]	104
	105	def _layout_name(self):
	106	"""
	107	Do the layout for file/function name related widgets
	108	"""
	109	self.name_sizer = wx.BoxSizer(wx.VERTICAL)
	110	self.name_hsizer = wx.BoxSizer(wx.HORIZONTAL)
	111	#title name [string]
[0008f54]	112	name_txt = wx.StaticText(self, -1, 'Function Name : ')
[796c4d4]	113	self.name_tcl = wx.TextCtrl(self, -1, size=(PANEL_WIDTH*3/5, -1))
	114	self.name_tcl.Bind(wx.EVT_TEXT_ENTER, self.on_change_name)
	115	self.name_tcl.SetValue('')
	116	self.name_tcl.SetFont(self.font)
[0008f54]	117	hint_name = "Unique Sum/Multiply Model Function Name."
[796c4d4]	118	self.name_tcl.SetToolTipString(hint_name)
	119	self.name_hsizer.AddMany([(name_txt, 0, wx.LEFT\|wx.TOP, 10),
[fdef956]	120	(self.name_tcl, -1,
	121	wx.EXPAND\|wx.RIGHT\|wx.TOP\|wx.BOTTOM, 10)])
	122	self.name_sizer.AddMany([(self.name_hsizer, -1,
[796c4d4]	123	wx.LEFT\|wx.TOP, 10)])
	124
[6f140f2]	125
[796c4d4]	126	def _layout_description(self):
	127	"""
	128	Do the layout for description related widgets
	129	"""
	130	self.desc_sizer = wx.BoxSizer(wx.HORIZONTAL)
	131	#title name [string]
	132	desc_txt = wx.StaticText(self, -1, 'Description (optional) : ')
	133	self.desc_tcl = wx.TextCtrl(self, -1, size=(PANEL_WIDTH*3/5, -1))
	134	self.desc_tcl.SetValue('')
	135	#self.name_tcl.SetFont(self.font)
[fdef956]	136	hint_desc = "Write a short description of this model function."
[796c4d4]	137	self.desc_tcl.SetToolTipString(hint_desc)
	138	self.desc_sizer.AddMany([(desc_txt, 0, wx.LEFT\|wx.TOP, 10),
[d970df9]	139	(self.desc_tcl, -1,
	140	wx.EXPAND\|wx.RIGHT\|wx.TOP\|wx.BOTTOM, 10)])
[796c4d4]	141
[6f140f2]	142	def _build_sizer(self):
	143	"""
	144	Build gui
	145	"""
[fdef956]	146	box_width = 195 # combobox width
[6f140f2]	147	vbox = wx.BoxSizer(wx.VERTICAL)
[fdef956]	148	self.sizer = wx.GridBagSizer(1, 3)
[796c4d4]	149	self._layout_name()
	150	self._layout_description()
	151
	152
[d970df9]	153	sum_description = wx.StaticBox(self, -1, 'Select',
[6f140f2]	154	size=(PNL_WIDTH-30, 70))
	155	sum_box = wx.StaticBoxSizer(sum_description, wx.VERTICAL)
	156	model1_box = wx.BoxSizer(wx.HORIZONTAL)
	157	model2_box = wx.BoxSizer(wx.HORIZONTAL)
	158	model_vbox = wx.BoxSizer(wx.VERTICAL)
	159	self.model1 = wx.ComboBox(self, -1, style=wx.CB_READONLY)
	160	wx.EVT_COMBOBOX(self.model1, -1, self.on_model1)
[fdef956]	161	self.model1.SetMinSize((box_width*5/6, -1))
[6f140f2]	162	self.model1.SetToolTipString("model1")
[fdef956]	163
	164	self.operator_cbox = wx.ComboBox(self, -1, size=(50, -1),
	165	style=wx.CB_READONLY)
	166	wx.EVT_COMBOBOX(self.operator_cbox, -1, self.on_select_operator)
	167	operation_tip = "Add: +, Multiply: * "
	168	self.operator_cbox.SetToolTipString(operation_tip)
	169
[6f140f2]	170	self.model2 = wx.ComboBox(self, -1, style=wx.CB_READONLY)
	171	wx.EVT_COMBOBOX(self.model2, -1, self.on_model2)
[fdef956]	172	self.model2.SetMinSize((box_width*5/6, -1))
[6f140f2]	173	self.model2.SetToolTipString("model2")
	174	self._set_model_list()
	175
	176	# Buttons on the bottom
	177	self.static_line_1 = wx.StaticLine(self, -1)
[fdef956]	178	self.okButton = wx.Button(self,wx.ID_OK, 'Apply', size=(box_width/2, 25))
[796c4d4]	179	self.okButton.Bind(wx.EVT_BUTTON, self.check_name)
	180	self.closeButton = wx.Button(self,wx.ID_CANCEL, 'Close',
[fdef956]	181	size=(box_width/2, 25))
[6f140f2]	182	# Intro
[d970df9]	183	self.explanation = " custom model = %s %s "% (self.factor, '*')
	184	self.explanation += "(model1 %s model2)\n"% self.operator
[796c4d4]	185	#explanation += " Note: This will overwrite the previous sum model.\n"
[6f140f2]	186	model_string = " Model%s (p%s):"
[41a8cb3]	187	# msg
	188	self._msg_box = wx.StaticText(self, -1, self._notes)
	189	self.msg_sizer = wx.BoxSizer(wx.HORIZONTAL)
	190	self.msg_sizer.Add(self._msg_box, 0, wx.LEFT, 0)
[796c4d4]	191	vbox.Add(self.name_hsizer)
	192	vbox.Add(self.desc_sizer)
[fdef956]	193	vbox.Add(self.sizer)
[6f140f2]	194	ix = 0
	195	iy = 1
[fdef956]	196	self.explanationctr = wx.StaticText(self, -1, self.explanation)
	197	self.sizer.Add(self.explanationctr , (iy, ix),
[6f140f2]	198	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d970df9]	199	model1_box.Add(wx.StaticText(self, -1, model_string% (1, 1)), -1, 0)
	200	model1_box.Add((box_width-15, 10))
[6f140f2]	201	model1_box.Add(wx.StaticText(self, -1, model_string% (2, 2)), -1, 0)
	202	model2_box.Add(self.model1, -1, 0)
[d970df9]	203	model2_box.Add((15, 10))
[fdef956]	204	model2_box.Add(self.operator_cbox, 0, 0)
[d970df9]	205	model2_box.Add((15, 10))
[6f140f2]	206	model2_box.Add(self.model2, -1, 0)
	207	model_vbox.Add(model1_box, -1, 0)
	208	model_vbox.Add(model2_box, -1, 0)
	209	sum_box.Add(model_vbox, -1, 10)
	210	iy += 1
	211	ix = 0
[fdef956]	212	self.sizer.Add(sum_box, (iy, ix),
[6f140f2]	213	(1, 1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[d970df9]	214	vbox.Add((10, 10))
[6f140f2]	215	vbox.Add(self.static_line_1, 0, wx.EXPAND, 10)
[41a8cb3]	216	vbox.Add(self.msg_sizer, 0,
	217	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE\|wx.BOTTOM, 10)
[6f140f2]	218	sizer_button = wx.BoxSizer(wx.HORIZONTAL)
	219	sizer_button.Add((20, 20), 1, wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	220	sizer_button.Add(self.okButton, 0,
	221	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 0)
	222	sizer_button.Add(self.closeButton, 0,
[fdef956]	223	wx.LEFT\|wx.RIGHT\|wx.ADJUST_MINSIZE, 10)
[6f140f2]	224	vbox.Add(sizer_button, 0, wx.EXPAND\|wx.BOTTOM\|wx.TOP, 10)
[fdef956]	225
[6f140f2]	226	self.SetSizer(vbox)
	227	self.Centre()
[796c4d4]	228
	229	def on_change_name(self, event=None):
	230	"""
	231	Change name
	232	"""
	233	if event is not None:
	234	event.Skip()
	235	self.name_tcl.SetBackgroundColour('white')
	236	self.Refresh()
	237
	238	def check_name(self, event=None):
	239	"""
	240	Check name if exist already
	241	"""
[d970df9]	242	mname = M_NAME
[796c4d4]	243	self.on_change_name(None)
	244	list_fnames = os.listdir(self.plugin_dir)
[d970df9]	245	# fake existing regular model name list
	246	m_list = [model + ".py" for model in self.model_list]
	247	list_fnames.append(m_list)
[796c4d4]	248	# function/file name
	249	title = self.name_tcl.GetValue().lstrip().rstrip()
	250	if title == '':
[c8654a3]	251	text = self.operator
	252	if text.count('+') > 0:
[d970df9]	253	mname = 'Sum'
[19b9e43]	254	else:
[c8654a3]	255	mname = 'Multi'
[d970df9]	256	mname += M_NAME
	257	title = mname
[796c4d4]	258	self.name = title
	259	t_fname = title + '.py'
	260	if not self.overwrite_name:
[d970df9]	261	if t_fname in list_fnames and title != mname:
[796c4d4]	262	self.name_tcl.SetBackgroundColour('pink')
	263	self.good_name = False
	264	info = 'Error'
	265	msg = "Name exists already."
	266	wx.MessageBox(msg, info)
[cdc2ee2]	267	self._notes = msg
[41a8cb3]	268	color = 'red'
	269	self._msg_box.SetLabel(msg)
	270	self._msg_box.SetForegroundColour(color)
[796c4d4]	271	return self.good_name
	272	self.fname = os.path.join(self.plugin_dir, t_fname)
[41a8cb3]	273	s_title = title
	274	if len(title) > 20:
	275	s_title = title[0:19] + '...'
[0008f54]	276	self._notes = "Model function (%s) has been set! \n" % str(s_title)
[796c4d4]	277	self.good_name = True
	278	self.on_apply(self.fname)
	279	return self.good_name
	280
	281	def on_apply(self, path):
	282	"""
	283	On Apply
	284	"""
	285	try:
	286	label = self.getText()
	287	fname = path
	288	name1 = label[0]
	289	name2 = label[1]
	290	self.write_string(fname, name1, name2)
	291	self.compile_file(fname)
	292	self.parent.update_custom_combo()
[cdc2ee2]	293	msg = self._notes
[0008f54]	294	info = 'Info'
[41a8cb3]	295	color = 'blue'
[796c4d4]	296	except:
[fdef956]	297	msg= "Easy Custom Sum/Multipy: Error occurred..."
[cdc2ee2]	298	info = 'Error'
[41a8cb3]	299	color = 'red'
	300	self._msg_box.SetLabel(msg)
	301	self._msg_box.SetForegroundColour(color)
[cdc2ee2]	302	if self.parent.parent != None:
[79492222]	303	from sas.guiframe.events import StatusEvent
[cdc2ee2]	304	wx.PostEvent(self.parent.parent, StatusEvent(status = msg,
	305	info=info))
	306	else:
[796c4d4]	307	raise
	308
[6f140f2]	309	def _set_model_list(self):
	310	"""
	311	Set the list of models
	312	"""
	313	# list of model names
[d970df9]	314	cm_list = []
	315	# models
[6f140f2]	316	list = self.model_list
[d970df9]	317	# custom models
	318	al_list = os.listdir(self.plugin_dir)
	319	for c_name in al_list:
	320	if c_name.split('.')[-1] == 'py' and \
	321	c_name.split('.')[0] != '__init__':
	322	name = str(c_name.split('.')[0])
	323	cm_list.append(name)
	324	if name not in list:
	325	list.append(name)
	326	self.cm_list = cm_list
[6f140f2]	327	if len(list) > 1:
	328	list.sort()
	329	for idx in range(len(list)):
[d970df9]	330	self.model1.Append(str(list[idx]), idx)
	331	self.model2.Append(str(list[idx]), idx)
[6f140f2]	332	self.model1.SetStringSelection(self.model1_string)
	333	self.model2.SetStringSelection(self.model2_string)
[d970df9]	334
	335	def update_cm_list(self):
	336	"""
	337	Update custom model list
	338	"""
	339	cm_list = []
	340	al_list = os.listdir(self.plugin_dir)
	341	for c_name in al_list:
	342	if c_name.split('.')[-1] == 'py' and \
	343	c_name.split('.')[0] != '__init__':
	344	name = str(c_name.split('.')[0])
	345	cm_list.append(name)
	346	self.cm_list = cm_list
	347
[6f140f2]	348	def on_model1(self, event):
	349	"""
	350	Set model1
	351	"""
	352	event.Skip()
[d970df9]	353	self.update_cm_list()
[6f140f2]	354	self.model1_name = str(self.model1.GetValue())
	355	self.model1_string = self.model1_name
[d970df9]	356	if self.model1_name in self.cm_list:
	357	self.is_p1_custom = True
	358	else:
	359	self.is_p1_custom = False
[6f140f2]	360
	361	def on_model2(self, event):
	362	"""
	363	Set model2
	364	"""
	365	event.Skip()
[d970df9]	366	self.update_cm_list()
[6f140f2]	367	self.model2_name = str(self.model2.GetValue())
	368	self.model2_string = self.model2_name
[d970df9]	369	if self.model2_name in self.cm_list:
	370	self.is_p2_custom = True
	371	else:
	372	self.is_p2_custom = False
[fdef956]	373
	374	def on_select_operator(self, event=None):
	375	"""
	376	On Select an Operator
	377	"""
[9e00363]	378	# For Mac
	379	if event != None:
	380	event.Skip()
[c8654a3]	381	name = ''
[fdef956]	382	item = event.GetEventObject()
[c8654a3]	383	text = item.GetValue()
	384	if text.count('*') > 0:
[fdef956]	385	name = 'Multi'
[d970df9]	386	factor = 'BackGround'
	387	f_oper = '+'
[f5500b3]	388	else:
[848913c]	389	name = 'Sum'
	390	factor = 'scale_factor'
	391	f_oper = '*'
[c8654a3]	392
	393	self.factor = factor
[f5500b3]	394	self.operator = text
	395	self.explanation = " Custom Model = %s %s (model1 %s model2)\n"% \
	396	(self.factor, f_oper, self.operator)
[fdef956]	397	self.explanationctr.SetLabel(self.explanation)
	398	self.name = name + M_NAME
	399	self.sizer.Layout()
	400
	401	def fill_oprator_combox(self):
	402	"""
	403	fill the current combobox with the operator
	404	"""
	405	operator_list = [' +', ' *']
	406	for oper in operator_list:
	407	pos = self.operator_cbox.Append(str(oper))
	408	self.operator_cbox.SetClientData(pos, str(oper.strip()))
	409	self.operator_cbox.SetSelection(0)
	410
[6f140f2]	411	def getText(self):
	412	"""
	413	Returns model name string as list
	414	"""
	415	return [self.model1_name, self.model2_name]
	416
	417	def write_string(self, fname, name1, name2):
	418	"""
	419	Write and Save file
	420	"""
[796c4d4]	421	self.fname = fname
	422	description = self.desc_tcl.GetValue().lstrip().rstrip()
	423	if description == '':
[0008f54]	424	description = name1 + self.operator + name2
[796c4d4]	425	name = self.name_tcl.GetValue().lstrip().rstrip()
[c8654a3]	426	text = self.operator_cbox.GetValue()
	427	if text.count('+') > 0:
[d970df9]	428	factor = 'scale_factor'
	429	f_oper = '*'
	430	default_val = '1.0'
[c8654a3]	431	else:
[d970df9]	432	factor = 'BackGround'
	433	f_oper = '+'
	434	default_val = '0.0'
[796c4d4]	435	path = self.fname
[6f140f2]	436	try:
[796c4d4]	437	out_f = open(path,'w')
[6f140f2]	438	except :
	439	raise
	440	lines = SUM_TEMPLATE.split('\n')
	441	for line in lines:
[d970df9]	442	try:
	443	if line.count("scale_factor"):
	444	line = line.replace('scale_factor', factor)
	445	#print "scale_factor", line
	446	if line.count("= %s"):
	447	out_f.write(line % (default_val) + "\n")
	448	elif line.count("import Model as P1"):
	449	if self.is_p1_custom:
	450	line = line.replace('#', '')
	451	out_f.write(line % name1 + "\n")
	452	else:
	453	out_f.write(line + "\n")
	454	elif line.count("import %s as P1"):
	455	if not self.is_p1_custom:
	456	line = line.replace('#', '')
	457	out_f.write(line % (name1, name1) + "\n")
	458	else:
	459	out_f.write(line + "\n")
	460	elif line.count("import Model as P2"):
	461	if self.is_p2_custom:
	462	line = line.replace('#', '')
	463	out_f.write(line % name2 + "\n")
	464	else:
	465	out_f.write(line + "\n")
	466	elif line.count("import %s as P2"):
	467	if not self.is_p2_custom:
	468	line = line.replace('#', '')
	469	out_f.write(line % (name2, name2) + "\n")
	470	else:
	471	out_f.write(line + "\n")
	472	elif line.count("self.description = '%s'"):
	473	out_f.write(line % description + "\n")
	474	#elif line.count("run") and line.count("%s"):
	475	# out_f.write(line % self.operator + "\n")
	476	#elif line.count("evalDistribution") and line.count("%s"):
	477	# out_f.write(line % self.operator + "\n")
	478	elif line.count("return") and line.count("%s") == 2:
	479	#print "line return", line
	480	out_f.write(line % (f_oper, self.operator) + "\n")
	481	elif line.count("out2")and line.count("%s"):
	482	out_f.write(line % self.operator + "\n")
	483	else:
	484	out_f.write(line + "\n")
	485	except:
	486	raise
[796c4d4]	487	out_f.close()
	488	#else:
	489	# msg = "Name exists already."
[6f140f2]	490
	491	def compile_file(self, path):
	492	"""
	493	Compile the file in the path
	494	"""
[796c4d4]	495	path = self.fname
[6f140f2]	496	_compileFile(path)
	497
	498	def delete_file(self, path):
	499	"""
	500	Delete file in the path
	501	"""
	502	_deleteFile(path)
	503
	504
	505	class EditorPanel(wx.ScrolledWindow):
	506	"""
	507	Custom model function editor
	508	"""
	509	def __init__(self, parent, base, path, title, args, *kwds):
	510	kwds['name'] = title
	511	kwds["size"] = (EDITOR_WIDTH, EDITOR_HEIGTH)
[d970df9]	512	kwds["style"] = wx.FULL_REPAINT_ON_RESIZE
[6f140f2]	513	wx.ScrolledWindow.__init__(self, parent, args, *kwds)
	514	#self.SetupScrolling()
	515	self.parent = parent
	516	self.base = base
	517	self.path = path
	518	self.font = wx.SystemSettings_GetFont(wx.SYS_SYSTEM_FONT)
	519	self.font.SetPointSize(10)
	520	self.reader = None
	521	self.name = 'untitled'
[796c4d4]	522	self.overwrite_name = False
[d83549c]	523	self.is_2d = False
[6f140f2]	524	self.fname = None
	525	self.param_strings = ''
	526	self.function_strings = ''
	527	self._notes = ""
[41a8cb3]	528	self._msg_box = None
	529	self.msg_sizer = None
[6f140f2]	530	self.warning = ""
	531	self._description = "New Custom Model"
[318b5bbb]	532	self.function_tcl = None
[6f140f2]	533	#self._default_save_location = os.getcwd()
	534	self._do_layout()
	535	#self.bt_apply.Disable()
	536
	537
	538	def _define_structure(self):
	539	"""
	540	define initial sizer
	541	"""
	542	#w, h = self.parent.GetSize()
	543	self.main_sizer = wx.BoxSizer(wx.VERTICAL)
	544	self.name_sizer = wx.BoxSizer(wx.VERTICAL)
	545	self.name_hsizer = wx.BoxSizer(wx.HORIZONTAL)
	546	self.desc_sizer = wx.BoxSizer(wx.VERTICAL)
	547	self.param_sizer = wx.BoxSizer(wx.VERTICAL)
	548	self.function_sizer = wx.BoxSizer(wx.VERTICAL)
[318b5bbb]	549	self.func_horizon_sizer = wx.BoxSizer(wx.HORIZONTAL)
[6f140f2]	550	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
[41a8cb3]	551	self.msg_sizer = wx.BoxSizer(wx.HORIZONTAL)
[6f140f2]	552
	553	def _layout_name(self):
	554	"""
	555	Do the layout for file/function name related widgets
	556	"""
	557	#title name [string]
	558	name_txt = wx.StaticText(self, -1, 'Function Name : ')
	559	overwrite_cb = wx.CheckBox(self, -1, "Overwrite?", (10, 10))
[796c4d4]	560	overwrite_cb.SetValue(False)
[6f140f2]	561	overwrite_cb.SetToolTipString("Overwrite it if already exists?")
	562	wx.EVT_CHECKBOX(self, overwrite_cb.GetId(), self.on_over_cb)
[796c4d4]	563	#overwrite_cb.Show(False)
[6f140f2]	564	self.name_tcl = wx.TextCtrl(self, -1, size=(PANEL_WIDTH*3/5, -1))
	565	self.name_tcl.Bind(wx.EVT_TEXT_ENTER, self.on_change_name)
[7434020]	566	self.name_tcl.SetValue('MyFunction')
[6f140f2]	567	self.name_tcl.SetFont(self.font)
	568	hint_name = "Unique Model Function Name."
	569	self.name_tcl.SetToolTipString(hint_name)
	570	self.name_hsizer.AddMany([(self.name_tcl, 0, wx.LEFT\|wx.TOP, 0),
	571	(overwrite_cb, 0, wx.LEFT, 20)])
	572	self.name_sizer.AddMany([(name_txt, 0, wx.LEFT\|wx.TOP, 10),
	573	(self.name_hsizer, 0,
	574	wx.LEFT\|wx.TOP\|wx.BOTTOM, 10)])
	575
	576
	577	def _layout_description(self):
	578	"""
	579	Do the layout for description related widgets
	580	"""
	581	#title name [string]
	582	desc_txt = wx.StaticText(self, -1, 'Description (optional) : ')
	583	self.desc_tcl = wx.TextCtrl(self, -1, size=(PANEL_WIDTH*3/5, -1))
	584	self.desc_tcl.SetValue('')
	585	#self.name_tcl.SetFont(self.font)
	586	hint_desc = "Write a short description of the model function."
	587	self.desc_tcl.SetToolTipString(hint_desc)
	588	self.desc_sizer.AddMany([(desc_txt, 0, wx.LEFT\|wx.TOP, 10),
	589	(self.desc_tcl, 0,
	590	wx.LEFT\|wx.TOP\|wx.BOTTOM, 10)])
	591	def _layout_param(self):
	592	"""
	593	Do the layout for parameter related widgets
	594	"""
[e499ca0]	595	param_txt = wx.StaticText(self, -1, 'Fit Parameters (if any): ')
[318b5bbb]	596
[6f140f2]	597	param_tip = "#Set the parameters and initial values.\n"
	598	param_tip += "#Example:\n"
	599	param_tip += "A = 1\nB = 1"
	600	#param_txt.SetToolTipString(param_tip)
	601	id = wx.NewId()
	602	self.param_tcl = EditWindow(self, id, wx.DefaultPosition,
	603	wx.DefaultSize, wx.CLIP_CHILDREN\|wx.SUNKEN_BORDER)
[b25caad]	604	self.param_tcl.setDisplayLineNumbers(True)
[6f140f2]	605	self.param_tcl.SetToolTipString(param_tip)
[318b5bbb]	606
[6f140f2]	607	self.param_sizer.AddMany([(param_txt, 0, wx.LEFT, 10),
	608	(self.param_tcl, 1, wx.EXPAND\|wx.ALL, 10)])
[318b5bbb]	609
[6f140f2]	610	def _layout_function(self):
	611	"""
	612	Do the layout for function related widgets
	613	"""
	614	function_txt = wx.StaticText(self, -1, 'Function(x) : ')
	615	hint_function = "#Example:\n"
[e499ca0]	616	hint_function += "if x <= 0:\n"
[b25caad]	617	hint_function += " y = A + B\n"
	618	hint_function += "else:\n"
[e499ca0]	619	hint_function += " y = A + B * cos(2 * pi * x)\n"
[b25caad]	620	hint_function += "return y\n"
[318b5bbb]	621	math_txt = wx.StaticText(self, -1, '*Useful math functions: ')
	622	math_combo = self._fill_math_combo()
	623
[6f140f2]	624	id = wx.NewId()
	625	self.function_tcl = EditWindow(self, id, wx.DefaultPosition,
	626	wx.DefaultSize, wx.CLIP_CHILDREN\|wx.SUNKEN_BORDER)
[b25caad]	627	self.function_tcl.setDisplayLineNumbers(True)
[6f140f2]	628	self.function_tcl.SetToolTipString(hint_function)
[318b5bbb]	629
	630	self.func_horizon_sizer.Add(function_txt)
	631	self.func_horizon_sizer.Add(math_txt, 0, wx.LEFT, 250)
	632	self.func_horizon_sizer.Add(math_combo, 0, wx.LEFT, 10)
	633
	634	self.function_sizer.Add(self.func_horizon_sizer, 0, wx.LEFT, 10)
[6f140f2]	635	self.function_sizer.Add( self.function_tcl, 1, wx.EXPAND\|wx.ALL, 10)
[41a8cb3]	636
	637	def _layout_msg(self):
	638	"""
	639	Layout msg
	640	"""
[657e52c]	641	self._msg_box = wx.StaticText(self, -1, self._notes,
	642	size=(PANEL_WIDTH, -1))
[41a8cb3]	643	self.msg_sizer.Add(self._msg_box, 0, wx.LEFT, 10)
[6f140f2]	644
	645	def _layout_button(self):
	646	"""
	647	Do the layout for the button widgets
	648	"""
	649	self.bt_apply = wx.Button(self, -1, "Apply", size=(_BOX_WIDTH, -1))
	650	self.bt_apply.SetToolTipString("Save changes into the imported data.")
	651	self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
	652
	653	self.bt_close = wx.Button(self, -1, 'Close', size=(_BOX_WIDTH, -1))
	654	self.bt_close.Bind(wx.EVT_BUTTON, self.on_close)
	655	self.bt_close.SetToolTipString("Close this panel.")
	656
	657	self.button_sizer.AddMany([(self.bt_apply, 0,
	658	wx.LEFT, EDITOR_WIDTH * 0.8),
	659	(self.bt_close, 0,
	660	wx.LEFT\|wx.BOTTOM, 15)])
[41a8cb3]	661
[6f140f2]	662	def _do_layout(self):
	663	"""
	664	Draw the current panel
	665	"""
	666	self._define_structure()
	667	self._layout_name()
	668	self._layout_description()
	669	self._layout_param()
	670	self._layout_function()
[41a8cb3]	671	self._layout_msg()
[6f140f2]	672	self._layout_button()
	673	self.main_sizer.AddMany([(self.name_sizer, 0,
	674	wx.EXPAND\|wx.ALL, 5),
	675	(wx.StaticLine(self), 0,
	676	wx.ALL\|wx.EXPAND, 5),
	677	(self.desc_sizer, 0,
	678	wx.EXPAND\|wx.ALL, 5),
	679	(wx.StaticLine(self), 0,
	680	wx.ALL\|wx.EXPAND, 5),
[9ec0955]	681	(self.param_sizer, 1,
[6f140f2]	682	wx.EXPAND\|wx.ALL, 5),
	683	(wx.StaticLine(self), 0,
	684	wx.ALL\|wx.EXPAND, 5),
[9ec0955]	685	(self.function_sizer, 2,
[6f140f2]	686	wx.EXPAND\|wx.ALL, 5),
	687	(wx.StaticLine(self), 0,
	688	wx.ALL\|wx.EXPAND, 5),
[41a8cb3]	689	(self.msg_sizer, 0,
	690	wx.EXPAND\|wx.ALL, 5),
[6f140f2]	691	(self.button_sizer, 0,
	692	wx.EXPAND\|wx.ALL, 5)])
	693	self.SetSizer(self.main_sizer)
	694	self.SetAutoLayout(True)
[318b5bbb]	695
	696	def _fill_math_combo(self):
	697	"""
	698	Fill up the math combo box
	699	"""
	700	self.math_combo = wx.ComboBox(self, -1, size=(100, -1),
	701	style=wx.CB_READONLY)
	702	for item in dir(math):
	703	if item.count("_") < 1:
	704	try:
	705	exec "float(math.%s)"% item
	706	self.math_combo.Append(str(item))
	707	except:
	708	self.math_combo.Append(str(item) + "()" )
	709	self.math_combo.Bind(wx.EVT_COMBOBOX, self._on_math_select)
	710	self.math_combo.SetSelection(0)
	711	return self.math_combo
	712
	713	def _on_math_select(self, event):
	714	"""
	715	On math selection on ComboBox
	716	"""
	717	event.Skip()
	718	label = self.math_combo.GetLabel()
	719	self.function_tcl.SetFocus()
	720	# Put the text at the cursor position
	721	pos = self.function_tcl.GetCurrentPos()
	722	self.function_tcl.InsertText(pos, label)
	723	# Put the cursor at appropriate postion
	724	length = len(label)
	725	if label[-1] == ')':
	726	length -= 1
	727	f_pos = pos + length
	728	self.function_tcl.GotoPos(f_pos)
	729
[6f140f2]	730	def get_notes(self):
	731	"""
	732	return notes
	733	"""
	734	return self._notes
	735
	736	def on_change_name(self, event=None):
	737	"""
	738	Change name
	739	"""
	740	if event is not None:
	741	event.Skip()
	742	self.name_tcl.SetBackgroundColour('white')
	743	self.Refresh()
	744
	745	def check_name(self):
	746	"""
	747	Check name if exist already
	748	"""
[41a8cb3]	749	self._notes = ''
[6f140f2]	750	self.on_change_name(None)
	751	plugin_dir = self.path
	752	list_fnames = os.listdir(plugin_dir)
	753	# function/file name
	754	title = self.name_tcl.GetValue().lstrip().rstrip()
	755	self.name = title
	756	t_fname = title + '.py'
	757	if not self.overwrite_name:
	758	if t_fname in list_fnames:
	759	self.name_tcl.SetBackgroundColour('pink')
	760	return False
	761	self.fname = os.path.join(plugin_dir, t_fname)
[41a8cb3]	762	s_title = title
	763	if len(title) > 20:
	764	s_title = title[0:19] + '...'
[657e52c]	765	self._notes += "Model function name is set "
[41a8cb3]	766	self._notes += "to %s. \n" % str(s_title)
[6f140f2]	767	return True
	768
	769	def on_over_cb(self, event):
	770	"""
	771	Set overwrite name flag on cb event
	772	"""
	773	if event is not None:
	774	event.Skip()
	775	cb = event.GetEventObject()
	776	self.overwrite_name = cb.GetValue()
	777
	778	def on_click_apply(self, event):
	779	"""
	780	Changes are saved in data object imported to edit
	781	"""
	782	#must post event here
	783	event.Skip()
	784	info = 'Info'
[6057915]	785	msg = ''
[6f140f2]	786	# Sort out the errors if occur
	787	if self.check_name():
[e499ca0]	788	name = self.name_tcl.GetValue().lstrip().rstrip()
[6f140f2]	789	description = self.desc_tcl.GetValue()
	790	param_str = self.param_tcl.GetText()
	791	func_str = self.function_tcl.GetText()
	792	# No input for the model function
[6057915]	793	if func_str.lstrip().rstrip():
[0a90d92]	794	if func_str.count('return') > 0:
[6057915]	795	self.write_file(self.fname, description, param_str,
	796	func_str)
[0a90d92]	797	tr_msg = _compileFile(self.fname)
[0ed400d]	798	msg = str(tr_msg.__str__())
[0a90d92]	799	# Compile error
	800	if msg:
[0ed400d]	801	msg.replace(" ", "\n")
[6057915]	802	msg += "\nCompiling Failed"
[6f140f2]	803	else:
[0a90d92]	804	msg = "Error: The func(x) must 'return' a value at least.\n"
	805	msg += "For example: \n\nreturn 2*x"
[6f140f2]	806	else:
	807	msg = 'Error: Function is not defined.'
	808	else:
	809	msg = "Name exists already."
	810	# Prepare for the messagebox
[0ed400d]	811	if self.base != None and not msg:
	812	self.base.update_custom_combo()
	813	Model = None
	814	exec "from %s import Model"% name
	815	try:
	816	Model().run(0.01)
	817	except:
	818	msg = "Error "
	819	_, value, _ = sys.exc_info()
	820	msg += "in %s:\n%s\n" % (name, value)
	821	if msg:
[6f140f2]	822	info = 'Error'
[6057915]	823	color = 'red'
	824	try:
	825	# try to remove pyc file if exists
[8d0ec40]	826	_deleteFile(self.fname)
[6057915]	827	_deleteFile(self.fname + "c")
	828	except:
	829	pass
[0ed400d]	830	else:
[657e52c]	831	msg = "Successful! "
[0ed400d]	832	msg += " " + self._notes
[ea5fa58]	833	msg += " Please look for it in the Customized Models."
[0ed400d]	834	info = 'Info'
	835	color = 'blue'
	836	# Not to display long error msg
	837	if info == 'Error':
	838	mss = info
	839	else:
	840	mss = msg
	841	self._msg_box.SetLabel(mss)
[41a8cb3]	842	self._msg_box.SetForegroundColour(color)
[6f140f2]	843	# Send msg to the top window
	844	if self.base != None:
[79492222]	845	from sas.guiframe.events import StatusEvent
[6f140f2]	846	wx.PostEvent(self.base.parent, StatusEvent(status = msg,
	847	info=info))
	848	self.warning = msg
	849
	850
	851	def write_file(self, fname, desc_str, param_str, func_str):
	852	"""
	853	Write content in file
	854
	855	:param fname: full file path
	856	:param desc_str: content of the description strings
	857	:param param_str: content of params; Strings
	858	:param func_str: content of func; Strings
	859	"""
	860	try:
	861	out_f = open(fname,'w')
	862	except :
	863	raise
[e499ca0]	864	# Prepare the content of the function
[6f140f2]	865	lines = CUSTOM_TEMPLATE.split('\n')
[f9feff3]	866
[6f140f2]	867	has_scipy = func_str.count("scipy.")
[d83549c]	868	self.is_2d = func_str.count("#self.ndim = 2")
	869	line_2d = ''
	870	if self.is_2d:
	871	line_2d = CUSTOM_2D_TEMP.split('\n')
	872	line_test = TEST_TEMPLATE.split('\n')
[6f140f2]	873	local_params = ''
	874	spaces = ' '#8spaces
[d83549c]	875	# write function here
[6f140f2]	876	for line in lines:
	877	# The location where to put the strings is
	878	# hard-coded in the template as shown below.
	879	if line.count("#self.params here"):
	880	for param_line in param_str.split('\n'):
	881	p_line = param_line.lstrip().rstrip()
	882	if p_line:
	883	p0_line = self.set_param_helper(p_line)
	884	local_params += self.set_function_helper(p_line)
	885	out_f.write(p0_line)
	886	elif line.count("#local params here"):
	887	if local_params:
	888	out_f.write(local_params)
	889	elif line.count("self.description = "):
[e499ca0]	890	des0 = self.name + "\\n"
[6f140f2]	891	desc = str(desc_str.lstrip().rstrip().replace('\"', ''))
[e499ca0]	892	out_f.write(line% (des0 + desc) + "\n")
[d83549c]	893	elif line.count("def function(self, x=0.0%s):"):
	894	if self.is_2d:
	895	y_str = ', y=0.0'
	896	out_f.write(line% y_str + "\n")
	897	else:
	898	out_f.write(line% '' + "\n")
[6f140f2]	899	elif line.count("#function here"):
	900	for func_line in func_str.split('\n'):
[46472c61]	901	f_line = func_line.rstrip()
[6f140f2]	902	if f_line:
	903	out_f.write(spaces + f_line + "\n")
	904	if not func_str:
[d83549c]	905	dep_var = 'y'
	906	if self.is_2d:
	907	dep_var = 'z'
	908	out_f.write(spaces + 'return %s'% dep_var + "\n")
[6f140f2]	909	elif line.count("#import scipy?"):
	910	if has_scipy:
	911	out_f.write("import scipy" + "\n")
[316e231]	912	#elif line.count("name = "):
	913	# out_f.write(line % self.name + "\n")
[6f140f2]	914	elif line:
	915	out_f.write(line + "\n")
[d83549c]	916	# run string for 2d
	917	if line_2d:
	918	for line in line_2d:
	919	out_f.write(line + "\n")
	920	# Test strins
	921	for line in line_test:
	922	out_f.write(line + "\n")
	923
[6f140f2]	924	out_f.close()
	925
	926	def set_param_helper(self, line):
	927	"""
	928	Get string in line to define the params dictionary
	929
	930	:param line: one line of string got from the param_str
	931	"""
	932	flag = True
	933	params_str = ''
	934	spaces = ' '#8spaces
	935	items = line.split(";")
	936	for item in items:
	937	name = item.split("=")[0].lstrip().rstrip()
	938	try:
[e499ca0]	939	value = item.split("=")[1].lstrip().rstrip()
[6f140f2]	940	float(value)
	941	except:
[e499ca0]	942	value = 1.0 # default
[6f140f2]	943	params_str += spaces + "self.params['%s'] = %s\n"% (name, value)
	944
	945	return params_str
	946
	947	def set_function_helper(self, line):
	948	"""
	949	Get string in line to define the local params
	950
	951	:param line: one line of string got from the param_str
	952	"""
	953	flag = True
	954	params_str = ''
	955	spaces = ' '#8spaces
	956	items = line.split(";")
	957	for item in items:
	958	name = item.split("=")[0].lstrip().rstrip()
	959	params_str += spaces + "%s = self.params['%s']\n"% (name, name)
	960	return params_str
	961
	962	def get_warning(self):
	963	"""
	964	Get the warning msg
	965	"""
	966	return self.warning
	967
	968	def on_close(self, event):
	969	"""
	970	leave data as it is and close
	971	"""
[6057915]	972	self.parent.Show(False)#Close()
[6f140f2]	973	event.Skip()
	974
	975	class EditorWindow(wx.Frame):
[d970df9]	976	"""
	977	Editor Window
	978	"""
[6f140f2]	979	def __init__(self, parent, base, path, title,
	980	size=(EDITOR_WIDTH, EDITOR_HEIGTH), args, *kwds):
[d970df9]	981	"""
	982	Init
	983	"""
[6f140f2]	984	kwds["title"] = title
	985	kwds["size"] = size
	986	wx.Frame.__init__(self, parent=None, args, *kwds)
	987	self.parent = parent
	988	self.panel = EditorPanel(parent=self, base=parent,
	989	path=path, title=title)
	990	self.Show(True)
	991	wx.EVT_CLOSE(self, self.OnClose)
	992
	993	def OnClose(self, event):
	994	"""
	995	On close event
	996	"""
[6057915]	997	self.Show(False)
	998	#if self.parent != None:
	999	# self.parent.new_model_frame = None
	1000	#self.Destroy()
[6f140f2]	1001
	1002	## Templates for custom models
	1003	CUSTOM_TEMPLATE = """
[79492222]	1004	from sas.models.pluginmodel import Model1DPlugin
[e499ca0]	1005	from math import *
[316e231]	1006	import os
	1007	import sys
[6f140f2]	1008	import numpy
	1009	#import scipy?
	1010	class Model(Model1DPlugin):
[316e231]	1011	name = ""
[6f140f2]	1012	def __init__(self):
[316e231]	1013	Model1DPlugin.__init__(self, name=self.name)
	1014	#set name same as file name
	1015	self.name = self.get_fname()
[6f140f2]	1016	#self.params here
	1017	self.description = "%s"
	1018	self.set_details()
[d83549c]	1019	def function(self, x=0.0%s):
[6f140f2]	1020	#local params here
[d83549c]	1021	#function here
	1022	"""
	1023	CUSTOM_2D_TEMP = """
[f505470]	1024	def run(self, x=0.0, y=0.0):
[d83549c]	1025	if x.__class__.__name__ == 'list':
[f505470]	1026	x_val = x[0]
	1027	y_val = y[0]*0.0
[d83549c]	1028	return self.function(x_val, y_val)
	1029	elif x.__class__.__name__ == 'tuple':
	1030	msg = "Tuples are not allowed as input to BaseComponent models"
	1031	raise ValueError, msg
	1032	else:
[7485cfa]	1033	return self.function(x, 0.0)
[d83549c]	1034	def runXY(self, x=0.0, y=0.0):
	1035	if x.__class__.__name__ == 'list':
	1036	return self.function(x, y)
	1037	elif x.__class__.__name__ == 'tuple':
	1038	msg = "Tuples are not allowed as input to BaseComponent models"
	1039	raise ValueError, msg
	1040	else:
	1041	return self.function(x, y)
	1042	def evalDistribution(self, qdist):
	1043	if qdist.__class__.__name__ == 'list':
	1044	msg = "evalDistribution expects a list of 2 ndarrays"
	1045	if len(qdist)!=2:
	1046	raise RuntimeError, msg
	1047	if qdist[0].__class__.__name__ != 'ndarray':
	1048	raise RuntimeError, msg
	1049	if qdist[1].__class__.__name__ != 'ndarray':
	1050	raise RuntimeError, msg
	1051	v_model = numpy.vectorize(self.runXY, otypes=[float])
	1052	iq_array = v_model(qdist[0], qdist[1])
	1053	return iq_array
	1054	elif qdist.__class__.__name__ == 'ndarray':
	1055	v_model = numpy.vectorize(self.runXY, otypes=[float])
	1056	iq_array = v_model(qdist)
	1057	return iq_array
	1058	"""
[7485cfa]	1059	TEST_TEMPLATE = """
[316e231]	1060	def get_fname(self):
	1061	path = sys._getframe().f_code.co_filename
	1062	basename = os.path.basename(path)
	1063	name, _ = os.path.splitext(basename)
	1064	return name
[7485cfa]	1065	######################################################################
	1066	## THIS IS FOR TEST. DO NOT MODIFY THE FOLLOWING LINES!!!!!!!!!!!!!!!!
	1067	if __name__ == "__main__":
	1068	m= Model()
	1069	out1 = m.runXY(0.0)
	1070	out2 = m.runXY(0.01)
	1071	isfine1 = numpy.isfinite(out1)
	1072	isfine2 = numpy.isfinite(out2)
	1073	print "Testing the value at Q = 0.0:"
	1074	print out1, " : finite? ", isfine1
	1075	print "Testing the value at Q = 0.01:"
	1076	print out2, " : finite? ", isfine2
	1077	if isfine1 and isfine2:
	1078	print "===> Simple Test: Passed!"
	1079	else:
	1080	print "===> Simple Test: Failed!"
	1081	"""
[6f140f2]	1082	SUM_TEMPLATE = """
[0008f54]	1083	# A sample of an experimental model function for Sum/Multiply(Pmodel1,Pmodel2)
[6f140f2]	1084	import copy
[79492222]	1085	from sas.models.pluginmodel import Model1DPlugin
[6f140f2]	1086	# User can change the name of the model (only with single functional model)
	1087	#P1_model:
[79492222]	1088	#from sas.models.%s import %s as P1
[d970df9]	1089	#from %s import Model as P1
[6f140f2]	1090
	1091	#P2_model:
[79492222]	1092	#from sas.models.%s import %s as P2
[d970df9]	1093	#from %s import Model as P2
[316e231]	1094	import os
	1095	import sys
[6f140f2]	1096
	1097	class Model(Model1DPlugin):
	1098	name = ""
	1099	def __init__(self):
	1100	Model1DPlugin.__init__(self, name='')
	1101	p_model1 = P1()
	1102	p_model2 = P2()
	1103	## Setting model name model description
[796c4d4]	1104	self.description = '%s'
[316e231]	1105	self.name = self.get_fname()
[796c4d4]	1106	if self.name.rstrip().lstrip() == '':
	1107	self.name = self._get_name(p_model1.name, p_model2.name)
	1108	if self.description.rstrip().lstrip() == '':
	1109	self.description = p_model1.name
	1110	self.description += p_model2.name
	1111	self.fill_description(p_model1, p_model2)
[6f140f2]	1112
	1113	## Define parameters
	1114	self.params = {}
	1115
	1116	## Parameter details [units, min, max]
	1117	self.details = {}
[318b5bbb]	1118	## Magnetic Panrameters
	1119	self.magnetic_params = []
[6f140f2]	1120	# non-fittable parameters
	1121	self.non_fittable = p_model1.non_fittable
	1122	self.non_fittable += p_model2.non_fittable
	1123
	1124	##models
	1125	self.p_model1= p_model1
	1126	self.p_model2= p_model2
	1127
	1128
	1129	## dispersion
	1130	self._set_dispersion()
	1131	## Define parameters
	1132	self._set_params()
[d970df9]	1133	## New parameter:scaling_factor
	1134	self.params['scale_factor'] = %s
[6f140f2]	1135
	1136	## Parameter details [units, min, max]
	1137	self._set_details()
	1138	self.details['scale_factor'] = ['', None, None]
	1139
	1140
	1141	#list of parameter that can be fitted
	1142	self._set_fixed_params()
	1143	## parameters with orientation
	1144	for item in self.p_model1.orientation_params:
	1145	new_item = "p1_" + item
	1146	if not new_item in self.orientation_params:
	1147	self.orientation_params.append(new_item)
	1148
	1149	for item in self.p_model2.orientation_params:
	1150	new_item = "p2_" + item
	1151	if not new_item in self.orientation_params:
	1152	self.orientation_params.append(new_item)
[318b5bbb]	1153	## magnetic params
	1154	for item in self.p_model1.magnetic_params:
	1155	new_item = "p1_" + item
	1156	if not new_item in self.magnetic_params:
	1157	self.magnetic_params.append(new_item)
	1158
	1159	for item in self.p_model2.magnetic_params:
	1160	new_item = "p2_" + item
	1161	if not new_item in self.magnetic_params:
	1162	self.magnetic_params.append(new_item)
[6f140f2]	1163	# get multiplicity if model provide it, else 1.
	1164	try:
	1165	multiplicity1 = p_model1.multiplicity
	1166	try:
	1167	multiplicity2 = p_model2.multiplicity
	1168	except:
	1169	multiplicity2 = 1
	1170	except:
	1171	multiplicity1 = 1
	1172	multiplicity2 = 1
	1173	## functional multiplicity of the model
	1174	self.multiplicity1 = multiplicity1
	1175	self.multiplicity2 = multiplicity2
	1176	self.multiplicity_info = []
	1177
	1178	def _clone(self, obj):
	1179	obj.params = copy.deepcopy(self.params)
	1180	obj.description = copy.deepcopy(self.description)
	1181	obj.details = copy.deepcopy(self.details)
	1182	obj.dispersion = copy.deepcopy(self.dispersion)
	1183	obj.p_model1 = self.p_model1.clone()
	1184	obj.p_model2 = self.p_model2.clone()
	1185	#obj = copy.deepcopy(self)
	1186	return obj
	1187
	1188	def _get_name(self, name1, name2):
[2b4c8ca]	1189	p1_name = self._get_upper_name(name1)
	1190	if not p1_name:
	1191	p1_name = name1
	1192	name = p1_name
[30bc3045]	1193	name += "_and_"
[2b4c8ca]	1194	p2_name = self._get_upper_name(name2)
	1195	if not p2_name:
	1196	p2_name = name2
	1197	name += p2_name
[6f140f2]	1198	return name
	1199
	1200	def _get_upper_name(self, name=None):
	1201	if name == None:
	1202	return ""
	1203	upper_name = ""
	1204	str_name = str(name)
	1205	for index in range(len(str_name)):
	1206	if str_name[index].isupper():
	1207	upper_name += str_name[index]
	1208	return upper_name
	1209
	1210	def _set_dispersion(self):
	1211	##set dispersion only from p_model
	1212	for name , value in self.p_model1.dispersion.iteritems():
	1213	#if name.lower() not in self.p_model1.orientation_params:
	1214	new_name = "p1_" + name
	1215	self.dispersion[new_name]= value
	1216	for name , value in self.p_model2.dispersion.iteritems():
	1217	#if name.lower() not in self.p_model2.orientation_params:
	1218	new_name = "p2_" + name
	1219	self.dispersion[new_name]= value
	1220
	1221	def function(self, x=0.0):
	1222	return 0
	1223
	1224	def getProfile(self):
	1225	try:
	1226	x,y = self.p_model1.getProfile()
	1227	except:
	1228	x = None
	1229	y = None
	1230
	1231	return x, y
	1232
	1233	def _set_params(self):
	1234	for name , value in self.p_model1.params.iteritems():
	1235	# No 2D-supported
	1236	#if name not in self.p_model1.orientation_params:
	1237	new_name = "p1_" + name
	1238	self.params[new_name]= value
	1239
	1240	for name , value in self.p_model2.params.iteritems():
	1241	# No 2D-supported
	1242	#if name not in self.p_model2.orientation_params:
	1243	new_name = "p2_" + name
	1244	self.params[new_name]= value
	1245
	1246	# Set "scale" as initializing
	1247	self._set_scale_factor()
	1248
	1249
	1250	def _set_details(self):
	1251	for name ,detail in self.p_model1.details.iteritems():
	1252	new_name = "p1_" + name
	1253	#if new_name not in self.orientation_params:
	1254	self.details[new_name]= detail
	1255
	1256	for name ,detail in self.p_model2.details.iteritems():
	1257	new_name = "p2_" + name
	1258	#if new_name not in self.orientation_params:
	1259	self.details[new_name]= detail
	1260
	1261	def _set_scale_factor(self):
	1262	pass
	1263
	1264
	1265	def setParam(self, name, value):
[0008f54]	1266	# set param to this (p1, p2) model
[6f140f2]	1267	self._setParamHelper(name, value)
	1268
	1269	## setParam to p model
[be0fe41]	1270	model_pre = ''
	1271	new_name = ''
	1272	name_split = name.split('_', 1)
	1273	if len(name_split) == 2:
	1274	model_pre = name.split('_', 1)[0]
	1275	new_name = name.split('_', 1)[1]
[6f140f2]	1276	if model_pre == "p1":
	1277	if new_name in self.p_model1.getParamList():
	1278	self.p_model1.setParam(new_name, value)
	1279	elif model_pre == "p2":
	1280	if new_name in self.p_model2.getParamList():
	1281	self.p_model2.setParam(new_name, value)
[be0fe41]	1282	elif name == 'scale_factor':
[6f140f2]	1283	self.params['scale_factor'] = value
	1284	else:
	1285	raise ValueError, "Model does not contain parameter %s" % name
	1286
	1287	def getParam(self, name):
	1288	# Look for dispersion parameters
	1289	toks = name.split('.')
	1290	if len(toks)==2:
	1291	for item in self.dispersion.keys():
	1292	# 2D not supported
	1293	if item.lower()==toks[0].lower():
	1294	for par in self.dispersion[item]:
	1295	if par.lower() == toks[1].lower():
	1296	return self.dispersion[item][par]
	1297	else:
	1298	# Look for standard parameter
	1299	for item in self.params.keys():
	1300	if item.lower()==name.lower():
	1301	return self.params[item]
	1302	return
	1303	#raise ValueError, "Model does not contain parameter %s" % name
	1304
	1305	def _setParamHelper(self, name, value):
	1306	# Look for dispersion parameters
	1307	toks = name.split('.')
	1308	if len(toks)== 2:
	1309	for item in self.dispersion.keys():
	1310	if item.lower()== toks[0].lower():
	1311	for par in self.dispersion[item]:
	1312	if par.lower() == toks[1].lower():
	1313	self.dispersion[item][par] = value
	1314	return
	1315	else:
	1316	# Look for standard parameter
	1317	for item in self.params.keys():
	1318	if item.lower()== name.lower():
	1319	self.params[item] = value
	1320	return
	1321
	1322	raise ValueError, "Model does not contain parameter %s" % name
	1323
	1324
	1325	def _set_fixed_params(self):
	1326	for item in self.p_model1.fixed:
	1327	new_item = "p1" + item
	1328	self.fixed.append(new_item)
	1329	for item in self.p_model2.fixed:
	1330	new_item = "p2" + item
	1331	self.fixed.append(new_item)
	1332
	1333	self.fixed.sort()
	1334
	1335
	1336	def run(self, x = 0.0):
	1337	self._set_scale_factor()
[d970df9]	1338	return self.params['scale_factor'] %s \
[0008f54]	1339	(self.p_model1.run(x) %s self.p_model2.run(x))
[6f140f2]	1340
	1341	def runXY(self, x = 0.0):
	1342	self._set_scale_factor()
[d970df9]	1343	return self.params['scale_factor'] %s \
[0008f54]	1344	(self.p_model1.runXY(x) %s self.p_model2.runXY(x))
[6f140f2]	1345
	1346	## Now (May27,10) directly uses the model eval function
	1347	## instead of the for-loop in Base Component.
	1348	def evalDistribution(self, x = []):
	1349	self._set_scale_factor()
[d970df9]	1350	return self.params['scale_factor'] %s \
[0008f54]	1351	(self.p_model1.evalDistribution(x) %s \
[6f140f2]	1352	self.p_model2.evalDistribution(x))
	1353
	1354	def set_dispersion(self, parameter, dispersion):
	1355	value= None
	1356	new_pre = parameter.split("_", 1)[0]
	1357	new_parameter = parameter.split("_", 1)[1]
	1358	try:
	1359	if new_pre == 'p1' and \
	1360	new_parameter in self.p_model1.dispersion.keys():
	1361	value= self.p_model1.set_dispersion(new_parameter, dispersion)
	1362	if new_pre == 'p2' and \
	1363	new_parameter in self.p_model2.dispersion.keys():
	1364	value= self.p_model2.set_dispersion(new_parameter, dispersion)
	1365	self._set_dispersion()
	1366	return value
	1367	except:
	1368	raise
	1369
	1370	def fill_description(self, p_model1, p_model2):
	1371	description = ""
[0008f54]	1372	description += "This model gives the summation or multiplication of"
	1373	description += "%s and %s. "% ( p_model1.name, p_model2.name )
[6f140f2]	1374	self.description += description
[316e231]	1375
	1376	def get_fname(self):
	1377	path = sys._getframe().f_code.co_filename
	1378	basename = os.path.basename(path)
	1379	name, _ = os.path.splitext(basename)
	1380	return name
	1381
[6f140f2]	1382	if __name__ == "__main__":
	1383	m1= Model()
	1384	#m1.setParam("p1_scale", 25)
	1385	#m1.setParam("p1_length", 1000)
	1386	#m1.setParam("p2_scale", 100)
	1387	#m1.setParam("p2_rg", 100)
	1388	out1 = m1.runXY(0.01)
	1389
	1390	m2= Model()
	1391	#m2.p_model1.setParam("scale", 25)
	1392	#m2.p_model1.setParam("length", 1000)
	1393	#m2.p_model2.setParam("scale", 100)
	1394	#m2.p_model2.setParam("rg", 100)
[0008f54]	1395	out2 = m2.p_model1.runXY(0.01) %s m2.p_model2.runXY(0.01)\n
[fdef956]	1396	print "My name is %s."% m1.name
[6f140f2]	1397	print out1, " = ", out2
	1398	if out1 == out2:
	1399	print "===> Simple Test: Passed!"
	1400	else:
	1401	print "===> Simple Test: Failed!"
	1402	"""
	1403
	1404	#if __name__ == "__main__":
	1405	# app = wx.PySimpleApp()
	1406	# frame = TextDialog(id=1, model_list=["SphereModel", "CylinderModel"])
	1407	# frame.Show(True)
	1408	# app.MainLoop()
	1409
	1410	if __name__ == "__main__":
[79492222]	1411	from sas.perspectives.fitting import models
[6f140f2]	1412	dir_path = models.find_plugins_dir()
	1413	app = wx.App()
[7485cfa]	1414	window = EditorWindow(parent=None, base=None, path=dir_path, title="Editor")
[6f140f2]	1415	app.MainLoop()

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

SasView

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

Download in other formats: