basepage.py @ b0eee0f0

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 b0eee0f0 was 0aeabc6, checked in by Gervaise Alina <gervyh@…>, 15 years ago
refactor draw model .working on save option
Property mode set to `100644`
File size: 58.8 KB

Rev	Line
[c77d859]	1
[60132ef]	2	import sys, os
[c77d859]	3	import wx
	4	import numpy
[a074145]	5	import time
[3370922]	6	import copy
[69bee6d]	7	from sans.guiframe.utils import format_number,check_float
[c77d859]	8	from sans.guicomm.events import StatusEvent
[cfc0913]	9	import pagestate
	10	from pagestate import PageState
	11	(PageInfoEvent, EVT_PAGE_INFO) = wx.lib.newevent.NewEvent()
[c77d859]	12	_BOX_WIDTH = 80
	13
	14	class BasicPage(wx.ScrolledWindow):
	15	"""
	16	This class provide general structure of fitpanel page
	17	"""
[b787e68c]	18	## Internal name for the AUI manager
	19	window_name = "Basic Page"
	20	## Title to appear on top of the window
	21	window_caption = "Basic page "
[59a7f2d]	22
[cfc0913]	23	def __init__(self,parent, page_info):
[e814734]	24	wx.ScrolledWindow.__init__(self, parent,
	25	style= wx.FULL_REPAINT_ON_RESIZE )
[cfc0913]	26	##window_name
	27	self.window_name = page_info.window_name
	28	##window_caption
	29	self.window_caption = page_info.window_caption
[c77d859]	30	## parent of the page
	31	self.parent = parent
[77e23a2]	32	## manager is the fitting plugin
[cfc0913]	33	self.manager= page_info.manager
[c77d859]	34	## owner of the page (fitting plugin)
[cfc0913]	35	self.event_owner= page_info.event_owner
	36	## current model
	37	self.model = page_info.model
	38	## data
	39	self.data = page_info.data
[c77d859]	40	## dictionary containing list of models
[cfc0913]	41	self.model_list_box = page_info.model_list_box
[c77d859]	42	## Data member to store the dispersion object created
	43	self._disp_obj_dict = {}
[cfc0913]	44	## selected parameters to apply dispersion
[c77d859]	45	self.disp_cb_dict ={}
[997131a]	46	## smearer object
	47	self.smearer = None
[c77d859]	48	##list of model parameters. each item must have same length
	49	## each item related to a given parameters
	50	##[cb state, name, value, "+/-", error of fit, min, max , units]
	51	self.parameters=[]
	52	## list of parameters to fit , must be like self.parameters
	53	self.param_toFit=[]
	54	## list of looking like parameters but with non fittable parameters info
	55	self.fixed_param=[]
	56	## list of looking like parameters but with fittable parameters info
	57	self.fittable_param=[]
	58	##list of dispersion parameters
	59	self.disp_list=[]
[780d095]	60	## list of orientation parameters
	61	self.orientation_params=[]
[60132ef]	62	self.orientation_params_disp=[]
[cfc0913]	63	if self.model !=None:
	64	self.disp_list= self.model.getDispParamList()
[70c57ebf]	65
[c77d859]	66	##enable model 2D draw
	67	self.enable2D= False
	68	## check that the fit range is correct to plot the model again
	69	self.fitrange= True
	70	## Q range
[df61301]	71	self.qmin_x= 0.0001
[45ed4dad]	72	self.qmax_x= 0.13
[0aeabc6]	73	self.num_points= 50
[cfc0913]	74	## Create memento to save the current state
	75
	76	self.state= PageState(parent= self.parent,model=self.model, data=self.data)
[71f0373]	77	## save customized array
	78	self.values=[]
	79	self.weights=[]
[a074145]	80	## retrieve saved state
	81	self.number_saved_state= 0
	82	## dictionary of saved state
	83	self.saved_states={}
	84	self.slicerpop = wx.Menu()
[60132ef]	85	## Default locations
	86	self._default_save_location = os.getcwd()
[a074145]	87	## save initial state on context menu
	88	self.onSave(event=None)
	89	self.Bind(wx.EVT_CONTEXT_MENU, self.onContextMenu)
	90
[cfc0913]	91	## create the basic structure of the panel with empty sizer
	92	self.define_page_structure()
[c77d859]	93	## drawing Initial dispersion parameters sizer
	94	self.set_dispers_sizer()
[00c3aac]	95	self._fill_save_sizer()
[c77d859]	96	## layout
	97	self.set_layout()
	98
[a074145]	99
	100	def onContextMenu(self, event):
	101	"""
	102	Retrieve the state selected state
	103	"""
[6d1235b]	104	# Skipping the save state functionality for release 0.9.0
[dad49a0]	105	#return
[6d1235b]	106
[a074145]	107	pos = event.GetPosition()
	108	pos = self.ScreenToClient(pos)
[6d1235b]	109	#TODO: why is the state menu called "slicerpop"?
[a074145]	110	self.PopupMenu(self.slicerpop, pos)
	111
	112
[c77d859]	113
	114	def define_page_structure(self):
	115	"""
	116	Create empty sizer for a panel
	117	"""
	118	self.vbox = wx.BoxSizer(wx.VERTICAL)
	119	self.sizer0 = wx.BoxSizer(wx.VERTICAL)
	120	self.sizer1 = wx.BoxSizer(wx.VERTICAL)
	121	self.sizer2 = wx.BoxSizer(wx.VERTICAL)
	122	self.sizer3 = wx.BoxSizer(wx.VERTICAL)
	123	self.sizer4 = wx.BoxSizer(wx.VERTICAL)
	124	self.sizer5 = wx.BoxSizer(wx.VERTICAL)
[0a518e4c]	125	self.sizer6 = wx.BoxSizer(wx.VERTICAL)
[c77d859]	126
[8bd4dc0]	127	self.sizer0.SetMinSize((375,-1))
	128	self.sizer1.SetMinSize((375,-1))
	129	self.sizer2.SetMinSize((375,-1))
	130	self.sizer3.SetMinSize((375,-1))
	131	self.sizer4.SetMinSize((375,-1))
	132	self.sizer5.SetMinSize((375,-1))
[0a518e4c]	133	self.sizer6.SetMinSize((375,-1))
[8bd4dc0]	134
[c77d859]	135	self.vbox.Add(self.sizer0)
	136	self.vbox.Add(self.sizer1)
	137	self.vbox.Add(self.sizer2)
	138	self.vbox.Add(self.sizer3)
	139	self.vbox.Add(self.sizer4)
	140	self.vbox.Add(self.sizer5)
[0a518e4c]	141	self.vbox.Add(self.sizer6)
[c77d859]	142
[a074145]	143
[c77d859]	144	def set_layout(self):
	145	"""
	146	layout
	147	"""
	148	self.vbox.Layout()
	149	self.vbox.Fit(self)
	150	self.SetSizer(self.vbox)
	151
	152	self.set_scroll()
	153	self.Centre()
	154
[a074145]	155
[c77d859]	156	def set_scroll(self):
	157	self.SetScrollbars(20,20,200,100)
	158	self.Layout()
[0a518e4c]	159	self.SetAutoLayout(True)
[c77d859]	160
[a074145]	161
[c77d859]	162	def set_owner(self,owner):
	163	"""
	164	set owner of fitpage
	165	@param owner: the class responsible of plotting
	166	"""
	167	self.event_owner = owner
[cfc0913]	168	self.state.event_owner = owner
[c77d859]	169
	170	def set_manager(self, manager):
	171	"""
	172	set panel manager
	173	@param manager: instance of plugin fitting
	174	"""
	175	self.manager = manager
[cfc0913]	176	self.state.manager = manager
[c77d859]	177
	178	def populate_box(self, dict):
	179	"""
	180	Store list of model
	181	@param dict: dictionary containing list of models
	182	"""
	183	self.model_list_box = dict
[cfc0913]	184	self.state.model_list_box = self.model_list_box
[c77d859]	185
	186
	187
	188	def set_dispers_sizer(self):
	189	"""
	190	fill sizer containing dispersity info
	191	"""
	192	self.sizer4.Clear(True)
[87fbc60]	193	name="Polydispersity and Orientational Distribution"
	194	box_description= wx.StaticBox(self, -1,name)
[c77d859]	195	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	196	#----------------------------------------------------
[fa58441]	197	self.disable_disp = wx.RadioButton(self, -1, 'Off', (10, 10), style=wx.RB_GROUP)
	198	self.enable_disp = wx.RadioButton(self, -1, 'On', (10, 30))
[71f0373]	199
[cfc0913]	200
[c77d859]	201	self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param, id=self.disable_disp.GetId())
	202	self.Bind(wx.EVT_RADIOBUTTON, self._set_dipers_Param, id=self.enable_disp.GetId())
	203
	204	sizer_dispersion = wx.BoxSizer(wx.HORIZONTAL)
	205	sizer_dispersion.Add((20,20))
[fa58441]	206	name=""#Polydispersity and \nOrientational Distribution "
[a074145]	207	sizer_dispersion.Add(wx.StaticText(self,-1,name))
[c77d859]	208	sizer_dispersion.Add(self.enable_disp )
	209	sizer_dispersion.Add((20,20))
	210	sizer_dispersion.Add(self.disable_disp )
	211	sizer_dispersion.Add((10,10))
	212
	213	## fill a sizer with the combobox to select dispersion type
	214	sizer_select_dispers = wx.BoxSizer(wx.HORIZONTAL)
[fa58441]	215	self.model_disp = wx.StaticText(self, -1, 'Distribution Function ')
[c77d859]	216
	217	import sans.models.dispersion_models
	218	self.polydisp= sans.models.dispersion_models.models
[0a518e4c]	219	self.disp_box = wx.ComboBox(self, -1)
[b787e68c]	220
[c77d859]	221	for key in self.polydisp.iterkeys():
	222	name = str(key.__name__)
	223	if name=="ArrayDispersion":
	224	# Remove the option until the rest of the code is ready for it
	225	self.disp_box.Append("Select customized Model",key)
[b787e68c]	226
[c77d859]	227	else:
[b787e68c]	228	self.disp_box.Append(name,key)
	229	self.disp_box.SetSelection(0)
	230
[c77d859]	231	wx.EVT_COMBOBOX(self.disp_box,-1, self._on_select_Disp)
	232
	233	sizer_select_dispers.Add((10,10))
	234	sizer_select_dispers.Add(self.model_disp)
[997131a]	235	sizer_select_dispers.Add(self.disp_box,0,
	236	wx.TOP\|wx.BOTTOM\|wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,border=5)
	237	#sizer_select_dispers.Add((10,10))
[c77d859]	238	self.model_disp.Hide()
	239	self.disp_box.Hide()
	240
[997131a]	241	boxsizer1.Add( sizer_dispersion,0,
	242	wx.TOP\|wx.BOTTOM\|wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE,border=5)
	243	#boxsizer1.Add( (10,10) )
[c77d859]	244	boxsizer1.Add( sizer_select_dispers )
	245	self.sizer4_4 = wx.GridBagSizer(5,5)
	246	boxsizer1.Add( self.sizer4_4 )
	247	#-----------------------------------------------------
	248	self.sizer4.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
[3b605bb]	249	self.sizer4_4.Layout()
[c77d859]	250	self.sizer4.Layout()
[3b605bb]	251	self.Layout()
[c77d859]	252	self.SetScrollbars(20,20,200,100)
[3b605bb]	253	self.Refresh()
[71f0373]	254	## saving the state of enable dispersity button
	255	self.state.enable_disp= self.enable_disp.GetValue()
	256	self.state.disable_disp= self.disable_disp.GetValue()
	257
[c77d859]	258
	259	def select_disp_angle(self, event):
	260	"""
	261	Event for when a user select a parameter to average over.
	262	@param event: check box event
	263	"""
[71f0373]	264	self.values=[]
	265	self.weights=[]
[c77d859]	266	# Go through the list of dispersion check boxes to identify which one has changed
	267	for p in self.disp_cb_dict:
	268	# Catch which one of the box was just checked or unchecked.
	269	if event.GetEventObject() == self.disp_cb_dict[p]:
	270
[71f0373]	271
[c77d859]	272	if self.disp_cb_dict[p].GetValue() == True:
	273	# The user wants this parameter to be averaged.
	274	# Pop up the file selection dialog.
	275	path = self._selectDlg()
	276
	277	# If nothing was selected, just return
	278	if path is None:
	279	self.disp_cb_dict[p].SetValue(False)
	280	return
[60132ef]	281	try:
	282	self._default_save_location = os.path.dirname(path)
	283	except:
	284	pass
[c77d859]	285	try:
[71f0373]	286	self.values,self.weights = self.read_file(path)
[c77d859]	287	except:
	288	msg="Could not read input file"
	289	wx.PostEvent(self.parent.parent, StatusEvent(status= msg))
	290	return
	291
	292	# If any of the two arrays is empty, notify the user that we won't
	293	# proceed
[71f0373]	294	if self.values is None or self.weights is None or \
	295	self.values ==[] or self.weights ==[]:
[c77d859]	296	wx.PostEvent(self.parent.parent, StatusEvent(status=\
	297	"The loaded %s distrubtion is corrupted or empty" % p))
	298	return
	299
	300	# Tell the user that we are about to apply the distribution
	301	wx.PostEvent(self.parent.parent, StatusEvent(status=\
	302	"Applying loaded %s distribution: %s" % (p, path)))
	303
	304	# Create the dispersion objects
	305	from sans.models.dispersion_models import ArrayDispersion
	306	disp_model = ArrayDispersion()
[71f0373]	307	disp_model.set_weights(self.values, self.weights)
	308
[c77d859]	309	# Store the object to make it persist outside the scope of this method
	310	#TODO: refactor model to clean this up?
	311	self._disp_obj_dict[p] = disp_model
[dad49a0]	312	self.state._disp_obj_dict [p]= disp_model
	313	self.state.values=[]
	314	self.state.weights=[]
[71f0373]	315	self.state.values = self.values
	316	self.state.weights = self.weights
[c77d859]	317	# Set the new model as the dispersion object for the selected parameter
	318	self.model.set_dispersion(p, disp_model)
[9e6c27f]	319	# Store a reference to the weights in the model object so that
	320	# it's not lost when we use the model within another thread.
	321	#TODO: total hack - fix this
[71f0373]	322	self.state.model= self.model.clone()
[0aeabc6]	323	#if not hasattr(self.model, "_persistency_dict"):
	324	self.model._persistency_dict = {}
[71f0373]	325	self.model._persistency_dict[p] = [self.values, self.weights]
	326	self.state.model._persistency_dict[p] = [self.values, self.weights]
[c77d859]	327	else:
	328	# The parameter was un-selected. Go back to Gaussian model (with 0 pts)
	329	self._reset_dispersity()
[dad49a0]	330
[c77d859]	331	## Redraw the model
[0aeabc6]	332
[c77d859]	333	self._draw_model()
[dad49a0]	334	## post state to fit panel
[71f0373]	335
	336	self.state.disp_cb_dict[p]= self.disp_cb_dict[p].GetValue()
[dad49a0]	337	event = PageInfoEvent(page = self)
	338	wx.PostEvent(self.parent, event)
[c77d859]	339	return
	340
[a074145]	341
	342	def onResetModel(self, event):
	343	"""
	344	Reset model state
	345	"""
	346	## post help message for the selected model
	347	msg = self.slicerpop.GetHelpString(event.GetId())
	348	msg +=" reloaded"
	349	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	350
	351	name= self.slicerpop.GetLabel(event.GetId())
	352	if name in self.saved_states.keys():
	353	previous_state = self.saved_states[name]
	354	self.reset_page(previous_state)
	355
	356
[00c3aac]	357	def onSave(self, event):
	358	"""
	359	save history of the data and model
	360	"""
[71f0373]	361
[a074145]	362	if self.model==None:
	363	return
	364	if hasattr(self,"enable_disp"):
[fc6ea43]	365	self.state.enable_disp = copy.deepcopy(self.enable_disp.GetValue())
[a074145]	366	if hasattr(self, "disp_box"):
[fc6ea43]	367	self.state.disp_box = copy.deepcopy(self.disp_box.GetSelection())
[a074145]	368
	369	self.state.model = self.model.clone()
[0aeabc6]	370	#if not hasattr(self.model, "_persistency_dict"):
	371	self.model._persistency_dict = {}
[71f0373]	372	self.state.model._persistency_dict= copy.deepcopy(self.model._persistency_dict)
	373
[a074145]	374	new_state = self.state.clone()
[c477b31]	375	new_state.enable2D = copy.deepcopy(self.enable2D)
[a074145]	376	##Add model state on context menu
	377	self.number_saved_state += 1
	378	name= self.model.name+"[%g]"%self.number_saved_state
	379	self.saved_states[name]= new_state
	380
	381	## Add item in the context menu
[fc6ea43]	382
	383	year, month, day,hour,minute,second,tda,ty,tm_isdst= time.localtime()
	384	my_time= str(hour)+" : "+str(minute)+" : "+str(second)+" "
[a074145]	385	date= str( month)+"\|"+str(day)+"\|"+str(year)
	386	msg= "Model saved at %s on %s"%(my_time, date)
	387	## post help message for the selected model
	388	msg +=" Saved! right click on this page to retrieve this model"
	389	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	390
	391	id = wx.NewId()
	392	self.slicerpop.Append(id,name,str(msg))
	393	wx.EVT_MENU(self, id, self.onResetModel)
	394
[71f0373]	395
[1328e03]	396	def onSetFocus(self, evt):
[77e23a2]	397	"""
	398	highlight the current textcrtl and hide the error text control shown
	399	after fitting
	400	"""
[cfc0913]	401
[77e23a2]	402	if hasattr(self,"text2_3"):
	403	self.text2_3.Hide()
	404	if len(self.parameters)>0:
	405	for item in self.parameters:
	406	## hide statictext +/-
	407	if item[3]!=None:
	408	item[3].Hide()
	409	## hide textcrtl for error after fit
	410	if item[4]!=None:
	411	item[4].Clear()
	412	item[4].Hide()
	413	if len(self.fittable_param)>0:
	414	for item in self.fittable_param:
	415	## hide statictext +/-
	416	if item[3]!=None:
	417	item[3].Hide()
	418	## hide textcrtl for error after fit
	419	if item[4]!=None:
	420	item[4].Clear()
	421	item[4].Hide()
	422	self.Layout()
[1328e03]	423	# Get a handle to the TextCtrl
	424	widget = evt.GetEventObject()
	425	# Select the whole control, after this event resolves
[77e23a2]	426	wx.CallAfter(widget.SetSelection, -1,-1)
[1328e03]	427	return
[c77d859]	428
[60d6c23]	429
[c77d859]	430	def read_file(self, path):
	431	"""
	432	Read two columns file
	433	@param path: the path to the file to read
	434	"""
	435	try:
	436	if path==None:
	437	wx.PostEvent(self.parent.parent, StatusEvent(status=\
	438	" Selected Distribution was not loaded: %s"%path))
	439	return None, None
	440	input_f = open(path, 'r')
	441	buff = input_f.read()
	442	lines = buff.split('\n')
	443
	444	angles = []
	445	weights=[]
	446	for line in lines:
	447	toks = line.split()
[502de0a]	448	try:
	449	angle = float(toks[0])
	450	weight = float(toks[1])
	451	except:
	452	# Skip non-data lines
	453	pass
	454	angles.append(angle)
	455	weights.append(weight)
[c77d859]	456	return numpy.array(angles), numpy.array(weights)
	457	except:
	458	raise
[00c3aac]	459
[cfc0913]	460
	461	def createMemento(self):
	462	"""
	463	return the current state of the page
	464	"""
[71f0373]	465
[cfc0913]	466	return self.state.clone()
	467
	468
	469	def save_current_state(self):
	470	"""
	471	Store current state
	472	"""
[0aeabc6]	473	## save model option
[87fbc60]	474	if self.model!= None:
[71f0373]	475	self.disp_list= self.model.getDispParamList()
	476	self.state.disp_list= copy.deepcopy(self.disp_list)
[87fbc60]	477	self.state.model = self.model.clone()
[0aeabc6]	478
[c477b31]	479	self.state.enable2D = copy.deepcopy(self.enable2D)
[71f0373]	480	self.state.values= copy.deepcopy(self.values)
	481	self.state.weights = copy.deepcopy( self.weights)
[0aeabc6]	482	## save data
	483	self.state.save_data(self.data)
	484
[70c57ebf]	485	try:
	486	n = self.disp_box.GetCurrentSelection()
	487	dispersity= self.disp_box.GetClientData(n)
	488	name= dispersity.__name__
	489	if name == "GaussianDispersion":
	490	if hasattr(self,"cb1"):
	491	self.state.cb1= self.cb1.GetValue()
	492	except:
	493	pass
	494
[cfc0913]	495	if hasattr(self,"enable_disp"):
	496	self.state.enable_disp= self.enable_disp.GetValue()
[fc6ea43]	497	self.state.disable_disp = self.disable_disp.GetValue()
	498
[3370922]	499	self.state.smearer = copy.deepcopy(self.smearer)
[cfc0913]	500	if hasattr(self,"enable_smearer"):
[3370922]	501	self.state.enable_smearer = copy.deepcopy(self.enable_smearer.GetValue())
[fc6ea43]	502	self.state.disable_smearer = copy.deepcopy(self.disable_smearer.GetValue())
	503
[b787e68c]	504	if hasattr(self,"disp_box"):
	505	self.state.disp_box = self.disp_box.GetCurrentSelection()
[c477b31]	506	if len(self.disp_cb_dict)>0:
	507	for k , v in self.disp_cb_dict.iteritems():
	508	if v ==None :
	509	self.state.disp_cb_dict[k]= v
	510	else:
	511	try:
	512	self.state.disp_cb_dict[k]=v.GetValue()
	513	except:
	514	self.state.disp_cb_dict[k]= None
	515
	516	if len(self._disp_obj_dict)>0:
	517	for k , v in self._disp_obj_dict.iteritems():
	518	self.state._disp_obj_dict[k]= v
[71f0373]	519	self.state.values = copy.deepcopy(self.values)
	520	self.state.weights = copy.deepcopy(self.weights)
[0aeabc6]	521	## save plotting range
[b787e68c]	522	self._save_plotting_range()
[d2d0cfdf]	523
	524	self.state.orientation_params =[]
	525	self.state.orientation_params_disp =[]
	526	self.state.parameters =[]
	527	self.state.fittable_param =[]
	528	self.state.fixed_param =[]
	529
[cfc0913]	530	## save checkbutton state and txtcrtl values
[fc6ea43]	531	self._copy_parameters_state(self.orientation_params,
	532	self.state.orientation_params)
	533	self._copy_parameters_state(self.orientation_params_disp,
	534	self.state.orientation_params_disp)
[d2d0cfdf]	535	self._copy_parameters_state(self.parameters, self.state.parameters)
	536	self._copy_parameters_state(self.fittable_param, self.state.fittable_param)
	537	self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
[1c1436d]	538
[cfc0913]	539	## post state to fit panel
	540	event = PageInfoEvent(page = self)
	541	wx.PostEvent(self.parent, event)
	542
[a074145]	543
[7c845cb]	544	def reset_page_helper(self, state):
[cfc0913]	545	"""
[b787e68c]	546	Use page_state and change the state of existing page
[71f0373]	547	@precondition: the page is already drawn or created
	548	@postcondition: the state of the underlying data change as well as the
	549	state of the graphic interface
[cfc0913]	550	"""
[0aeabc6]	551	self.model= state.model
	552	self.data = state.data
	553	self.smearer= state.smearer
[c477b31]	554	self.enable2D= state.enable2D
[dad49a0]	555
[71f0373]	556	##draw sizer containing model parameters value for the current model
	557	self._set_model_sizer_selection( self.model )
	558	self.set_model_param_sizer(self.model)
	559	## enable the view 2d button if this is a modelpage type
[c477b31]	560	if hasattr(self,"model_view"):
	561	if not self.enable2D:
	562	self.model_view.Enable()
	563	else:
	564	self.model_view.Disable()
[71f0373]	565	## set the select all check box to the a given state
[998b6b8]	566	if hasattr(self, "cb1"):
[71f0373]	567	self.cb1.SetValue(state.cb1)
	568	## reset state of checkbox,textcrtl and regular parameters value
	569	self._reset_parameters_state(self.orientation_params_disp,
	570	state.orientation_params_disp)
	571	self._reset_parameters_state(self.orientation_params,
	572	state.orientation_params)
	573	self._reset_parameters_state(self.parameters,state.parameters)
	574
[b787e68c]	575	## display dispersion info layer
[0aeabc6]	576	self.enable_disp.SetValue(state.enable_disp)
	577	self.disable_disp.SetValue(state.disable_disp)
[71f0373]	578
[b787e68c]	579	if hasattr(self, "disp_box"):
[0aeabc6]	580	self.disp_box.SetSelection(state.disp_box)
[dad49a0]	581	n = self.disp_box.GetCurrentSelection()
	582	dispersity= self.disp_box.GetClientData(n)
[71f0373]	583
	584	self._set_dipers_Param(event=None)
[dad49a0]	585	name= dispersity.__name__
	586	if name == "GaussianDispersion":
	587	self.disp_cb_dict = {}
[c477b31]	588	self.disp_cb_dict = copy.deepcopy(state.disp_cb_dict)
[71f0373]	589
[dad49a0]	590	self._disp_obj_dict={}
[0aeabc6]	591	for k , v in state._disp_obj_dict.iteritems():
[dad49a0]	592	self._disp_obj_dict[k]=v
[71f0373]	593
[dad49a0]	594	if name=="ArrayDispersion":
[71f0373]	595	self.model_disp.Show(True)
	596	self.disp_box.Show(True)
	597	self._set_sizer_arraydispersion()
	598
[dad49a0]	599	for k,v in state.disp_cb_dict.iteritems():
	600	self.disp_cb_dict[k].SetValue(v)
[0aeabc6]	601
[71f0373]	602	if self.disp_cb_dict[k].GetValue():
	603	# Create the dispersion objects
	604	from sans.models.dispersion_models import ArrayDispersion
	605	disp_model = ArrayDispersion()
	606
	607	disp_model.set_weights(state.values, state.weights)
	608
	609	# Store the object to make it persist outside the scope of this method
	610	#TODO: refactor model to clean this up?
	611	self._disp_obj_dict[k] = disp_model
[0aeabc6]	612
[71f0373]	613	self.values=[]
	614	self.weights = []
	615	self.values= state.values
	616	self.weights = state.weights
[dad49a0]	617
[71f0373]	618	# Set the new model as the dispersion object for the selected parameter
[0aeabc6]	619	#if not hasattr(self.model, "_persistency_dict"):
	620	self.model._persistency_dict = {}
[71f0373]	621	self.model._persistency_dict[k] = [state.values, state.weights]
	622	self.model.set_dispersion(k, disp_model)
	623
[3370922]	624	##plotting range restore
[0aeabc6]	625	self._reset_plotting_range(state)
[b787e68c]	626	## smearing info restore
[cfc0913]	627	if hasattr(self,"enable_smearer"):
[3370922]	628	## set smearing value whether or not the data contain the smearing info
[0aeabc6]	629	self.enable_smearer.SetValue(state.enable_smearer)
	630	self.disable_smearer.SetValue(state.disable_smearer)
	631	self.tcChi.SetLabel(str( state.tcChi))
	632	self.sizer5.Layout()
	633	self.Layout()
	634	self.Refresh()
	635	#self.compute_chisqr(smearer= self.smearer)
[3370922]	636
[71f0373]	637	## reset state of checkbox,textcrtl and dispersity parameters value
[d2d0cfdf]	638	self._reset_parameters_state(self.fittable_param,state.fittable_param)
	639	self._reset_parameters_state(self.fixed_param,state.fixed_param)
[71f0373]	640
	641
[dad49a0]	642	## draw the model with previous parameters value
[c477b31]	643	self._onparamEnter_helper()
[0aeabc6]	644
[d2d0cfdf]	645	## reset context menu items
	646	self._reset_context_menu()
[0aeabc6]	647	## set the value of the current state to the state given as parameter
	648	self.state = state.clone()
	649	## post state to fit panel
	650	event = PageInfoEvent(page = self)
	651	wx.PostEvent(self.parent, event)
[cfc0913]	652
[c77d859]	653	def _selectDlg(self):
	654	"""
	655	open a dialog file to selected the customized dispersity
	656	"""
	657	import os
[60132ef]	658	dlg = wx.FileDialog(self, "Choose a weight file",
	659	self._default_save_location , "", ".", wx.OPEN)
[c77d859]	660	path = None
	661	if dlg.ShowModal() == wx.ID_OK:
	662	path = dlg.GetPath()
	663	dlg.Destroy()
	664	return path
	665
[cfc0913]	666
[a074145]	667	def _reset_context_menu(self):
	668	"""
	669	reset the context menu
	670	"""
	671	for name, state in self.state.saved_states.iteritems():
	672	self.number_saved_state += 1
	673	## Add item in the context menu
	674	id = wx.NewId()
	675	self.slicerpop.Append(id,name, 'Save model and state %g'%self.number_saved_state)
	676	wx.EVT_MENU(self, id, self.onResetModel)
	677
[71f0373]	678
[0aeabc6]	679	def _reset_plotting_range(self, state):
[cfc0913]	680	"""
	681	Reset the plotting range to a given state
	682	"""
[0aeabc6]	683	self.qmin.SetValue(str(state.qmin))
	684	self.qmax.SetValue(str(state.qmax))
[cfc0913]	685	if self.state.npts!=None:
[0aeabc6]	686	self.npts.SetValue(str(state.npts))
[cfc0913]	687
	688
	689	def _save_plotting_range(self ):
	690	"""
	691	save the state of plotting range
	692	"""
[b787e68c]	693	self.state.qmin = self.qmin_x
	694	self.state.qmax = self.qmax_x
[cfc0913]	695	if self.npts!=None:
[b787e68c]	696	self.state.npts= self.num_points
[cfc0913]	697
	698
[c77d859]	699	def _onparamEnter_helper(self):
	700	"""
	701	check if values entered by the user are changed and valid to replot
	702	model
	703	use : _check_value_enter
	704	"""
	705	if self.model !=None:
[71f0373]	706
[a074145]	707	## save current state
	708	self.save_current_state()
[71f0373]	709
[c77d859]	710	# Flag to register when a parameter has changed.
	711	is_modified = False
	712	is_modified =self._check_value_enter( self.fittable_param ,is_modified)
	713	is_modified =self._check_value_enter( self.fixed_param ,is_modified)
	714	is_modified =self._check_value_enter( self.parameters ,is_modified)
	715
[0a518e4c]	716	self.Layout()
[c77d859]	717	# Here we should check whether the boundaries have been modified.
	718	# If qmin and qmax have been modified, update qmin and qmax and
	719	# set the is_modified flag to True
[69bee6d]	720	from sans.guiframe.utils import check_value
[c77d859]	721	if check_value( self.qmin, self.qmax):
	722	if float(self.qmin.GetValue()) != self.qmin_x:
	723	self.qmin_x = float(self.qmin.GetValue())
	724	is_modified = True
	725	if float(self.qmax.GetValue()) != self.qmax_x:
	726	self.qmax_x = float(self.qmax.GetValue())
	727	is_modified = True
	728	self.fitrange = True
	729	else:
	730	self.fitrange = False
	731	if self.npts != None:
[69bee6d]	732	if check_float(self.npts):
	733	if float(self.npts.GetValue()) != self.num_points:
	734	self.num_points = float(self.npts.GetValue())
	735	is_modified = True
	736	else:
	737	msg= "Cannot Plot :Must enter a number!!! "
	738	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	739
[b787e68c]	740
[c77d859]	741	## if any value is modify draw model with new value
	742	if is_modified:
[0aeabc6]	743
[c77d859]	744	self._draw_model()
[c477b31]	745	self.save_current_state()
[c77d859]	746
	747
[cfc0913]	748	def _reset_parameters_state(self, listtorestore,statelist):
	749	"""
	750	Reset the parameters at the given state
	751	"""
[6999659]	752	if len(statelist)==0 or len(listtorestore)==0 :
	753	return
	754	if len(statelist)!= len(listtorestore) :
[cfc0913]	755	return
	756	for j in range(len(listtorestore)):
	757	item_page = listtorestore[j]
	758	item_page_info = statelist[j]
	759	##change the state of the check box for simple parameters
	760	if item_page[0]!=None:
	761	item_page[0].SetValue(item_page_info[0])
	762
	763	if item_page[2]!=None:
	764	item_page[2].SetValue(item_page_info[2])
	765
	766	if item_page[3]!=None:
	767	## show or hide text +/-
	768	if item_page_info[2]:
	769	item_page[3].Show(True)
	770	else:
	771	item_page[3].Hide()
	772	if item_page[4]!=None:
	773	## show of hide the text crtl for fitting error
	774	if item_page_info[4][0]:
	775	item_page[4].Show(True)
	776	item_page[4].SetValue(item_page_info[4][1])
	777	else:
	778	item_page[3].Hide()
	779	if item_page[5]!=None:
	780	## show of hide the text crtl for fitting error
	781	if item_page_info[5][0]:
	782	item_page[5].Show(True)
	783	item_page[5].SetValue(item_page_info[4][1])
	784	else:
	785	item_page[5].Hide()
	786
	787	if item_page[6]!=None:
	788	## show of hide the text crtl for fitting error
	789	if item_page_info[6][0]:
	790	item_page[6].Show(True)
	791	item_page[6].SetValue(item_page_info[6][1])
	792	else:
	793	item_page[6].Hide()
	794
[b787e68c]	795
[cfc0913]	796	def _copy_parameters_state(self, listtocopy, statelist):
	797	"""
	798	copy the state of button
	799	@param listtocopy: the list of check button to copy
	800	@param statelist: list of state object to store the current state
	801	"""
[d2d0cfdf]	802
[cfc0913]	803	if len(listtocopy)==0:
	804	return
[1c1436d]	805
[cfc0913]	806	for item in listtocopy:
	807	checkbox_state = None
	808	if item[0]!= None:
	809	checkbox_state= item[0].GetValue()
	810	parameter_name = item[1]
	811	parameter_value = None
	812	if item[2]!=None:
	813	parameter_value = item[2].GetValue()
	814	static_text = None
	815	if item[3]!=None:
	816	static_text = item[3].IsShown()
	817	error_value = None
	818	error_state = None
	819	if item[4]!= None:
	820	error_value = item[4].GetValue()
	821	error_state = item[4].IsShown()
	822
	823	min_value = None
	824	min_state = None
	825	if item[5]!= None:
	826	min_value = item[5].GetValue()
	827	min_state = item[5].IsShown()
	828
	829	max_value = None
	830	max_state = None
	831	if item[6]!= None:
	832	max_value = item[6].GetValue()
	833	max_state = item[6].IsShown()
	834
	835	statelist.append([checkbox_state, parameter_name, parameter_value,
	836	static_text ,[error_state,error_value],
	837	[min_state,min_value],[max_state , max_value],None])
	838
	839
	840
[c77d859]	841	def _set_model_sizer_selection(self, model):
	842	"""
	843	Display the sizer according to the type of the current model
	844	"""
[70c57ebf]	845	if model ==None:
	846	return
[c77d859]	847	if hasattr(model ,"model2"):
	848
	849	class_name= model.model2.__class__
	850	name= model.model2.name
	851	flag= name != "NoStructure"
	852	if flag and (class_name in self.model_list_box["Structure Factors"]):
[3b605bb]	853	self.structurebox.Enable()
[c097f02]	854	self.text2.Enable()
[c77d859]	855	items = self.structurebox.GetItems()
	856	self.sizer1.Layout()
	857	self.SetScrollbars(20,20,200,100)
	858	for i in range(len(items)):
	859	if items[i]== str(name):
	860	self.structurebox.SetSelection(i)
	861	break
	862
	863	if hasattr(model ,"model1"):
	864	class_name = model.model1.__class__
	865	name = model.model1.name
	866	self.formfactorbox.Clear()
	867
	868	for k, list in self.model_list_box.iteritems():
[376916c]	869	if k in["P(Q)*S(Q)","Shapes" ] and class_name in self.model_list_box["Shapes"]:
[c77d859]	870	self.shape_rbutton.SetValue(True)
[376916c]	871	## fill the form factor list with new model
	872	self._populate_box(self.formfactorbox,self.model_list_box["Shapes"])
	873	items = self.formfactorbox.GetItems()
	874	## set comboxbox to the selected item
	875	for i in range(len(items)):
	876	if items[i]== str(name):
	877	self.formfactorbox.SetSelection(i)
	878	break
	879	return
	880	elif k == "Shape-Independent":
[c77d859]	881	self.shape_indep_rbutton.SetValue(True)
	882	elif k == "Structure Factors":
	883	self.struct_rbutton.SetValue(True)
	884	else:
	885	self.plugin_rbutton.SetValue(True)
[376916c]	886
[c77d859]	887	if class_name in list:
	888	## fill the form factor list with new model
	889	self._populate_box(self.formfactorbox, list)
	890	items = self.formfactorbox.GetItems()
	891	## set comboxbox to the selected item
	892	for i in range(len(items)):
	893	if items[i]== str(name):
	894	self.formfactorbox.SetSelection(i)
	895	break
	896	break
	897	else:
[cdbe88e]	898
	899	## Select the model from the menu
[c77d859]	900	class_name = model.__class__
	901	name = model.name
	902	self.formfactorbox.Clear()
	903	items = self.formfactorbox.GetItems()
[376916c]	904
[cdbe88e]	905	for k, list in self.model_list_box.iteritems():
[376916c]	906	if k in["P(Q)*S(Q)","Shapes" ] and class_name in self.model_list_box["Shapes"]:
[3b605bb]	907	if class_name in self.model_list_box["P(Q)*S(Q)"]:
	908	self.structurebox.Enable()
[cdbe88e]	909	self.structurebox.SetSelection(0)
[c097f02]	910	self.text2.Enable()
[3b605bb]	911	else:
	912	self.structurebox.Disable()
	913	self.structurebox.SetSelection(0)
[c097f02]	914	self.text2.Disable()
[3b605bb]	915
[c77d859]	916	self.shape_rbutton.SetValue(True)
[376916c]	917	## fill the form factor list with new model
	918	self._populate_box(self.formfactorbox,self.model_list_box["Shapes"])
	919	items = self.formfactorbox.GetItems()
	920	## set comboxbox to the selected item
	921	for i in range(len(items)):
	922	if items[i]== str(name):
	923	self.formfactorbox.SetSelection(i)
	924	break
	925	return
	926	elif k == "Shape-Independent":
[c77d859]	927	self.shape_indep_rbutton.SetValue(True)
	928	elif k == "Structure Factors":
[cdbe88e]	929	self.struct_rbutton.SetValue(True)
[c77d859]	930	else:
	931	self.plugin_rbutton.SetValue(True)
	932	if class_name in list:
[3b605bb]	933	self.structurebox.SetSelection(0)
	934	self.structurebox.Disable()
[c097f02]	935	self.text2.Disable()
[376916c]	936	## fill the form factor list with new model
[c77d859]	937	self._populate_box(self.formfactorbox, list)
	938	items = self.formfactorbox.GetItems()
	939	## set comboxbox to the selected item
	940	for i in range(len(items)):
	941	if items[i]== str(name):
	942	self.formfactorbox.SetSelection(i)
	943	break
	944	break
	945
	946
	947	def _draw_model(self):
	948	"""
	949	Method to draw or refresh a plotted model.
	950	The method will use the data member from the model page
	951	to build a call to the fitting perspective manager.
	952
	953	[Note to coder: This way future changes will be done in only one place.]
	954	"""
	955	if self.model !=None:
[fca9cbd9]	956	temp_smear=None
	957	if hasattr(self, "enable_smearer"):
	958	if self.enable_smearer.GetValue():
	959	temp_smear= self.smearer
[0aeabc6]	960
[c77d859]	961	self.manager.draw_model(self.model, data=self.data,
[c16557c]	962	smearer= temp_smear,
[4e6b5939]	963	qmin=float(self.qmin_x), qmax=float(self.qmax_x),
	964	qstep= float(self.num_points),
[c77d859]	965	enable2D=self.enable2D)
	966
[71f0373]	967
[c77d859]	968	def _set_model_sizer(self, sizer, title="", object=None):
	969	"""
	970	Use lists to fill a sizer for model info
	971	"""
	972
	973	sizer.Clear(True)
	974	box_description= wx.StaticBox(self, -1,str(title))
	975	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	976	#--------------------------------------------------------
[376916c]	977	self.shape_rbutton = wx.RadioButton(self, -1, 'Shapes', style=wx.RB_GROUP)
	978	self.shape_indep_rbutton = wx.RadioButton(self, -1, "Shape-Independent")
	979	self.struct_rbutton = wx.RadioButton(self, -1, "Structure Factor ")
	980	self.plugin_rbutton = wx.RadioButton(self, -1, "Customized Models")
[c77d859]	981
	982	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
	983	id= self.shape_rbutton.GetId() )
	984	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
	985	id= self.shape_indep_rbutton.GetId() )
	986	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
	987	id= self.struct_rbutton.GetId() )
	988	self.Bind( wx.EVT_RADIOBUTTON, self._show_combox,
	989	id= self.plugin_rbutton.GetId() )
[3b605bb]	990
[cfc0913]	991
[3b605bb]	992	sizer_radiobutton = wx.GridSizer(2, 2,5, 5)
[c77d859]	993	sizer_radiobutton.Add(self.shape_rbutton)
	994	sizer_radiobutton.Add(self.shape_indep_rbutton)
	995	sizer_radiobutton.Add(self.plugin_rbutton)
	996	sizer_radiobutton.Add(self.struct_rbutton)
	997
	998	sizer_selection = wx.BoxSizer(wx.HORIZONTAL)
	999
[fa58441]	1000	self.text1 = wx.StaticText( self,-1,"" )
	1001	self.text2 = wx.StaticText( self,-1,"P(Q)*S(Q)" )
[c77d859]	1002
	1003
[8bd4dc0]	1004	self.formfactorbox = wx.ComboBox(self, -1,style=wx.CB_READONLY)
[c77d859]	1005	if self.model!=None:
	1006	self.formfactorbox.SetValue(self.model.name)
[c097f02]	1007
[59a7f2d]	1008
[8bd4dc0]	1009	self.structurebox = wx.ComboBox(self, -1,style=wx.CB_READONLY)
[c77d859]	1010	wx.EVT_COMBOBOX(self.formfactorbox,-1, self._on_select_model)
	1011	wx.EVT_COMBOBOX(self.structurebox,-1, self._on_select_model)
[3b605bb]	1012
[71f0373]	1013
[c77d859]	1014	## fill combox box
	1015	if len(self.model_list_box)>0:
[376916c]	1016	self._populate_box( self.formfactorbox,self.model_list_box["Shapes"])
[c77d859]	1017
	1018	if len(self.model_list_box)>0:
	1019	self._populate_box( self.structurebox,
	1020	self.model_list_box["Structure Factors"])
[3b605bb]	1021	self.structurebox.Insert("None", 0,None)
	1022	self.structurebox.SetSelection(0)
	1023	self.structurebox.Disable()
[c097f02]	1024	self.text2.Disable()
	1025
[3b605bb]	1026	if self.model.__class__ in self.model_list_box["P(Q)*S(Q)"]:
	1027	self.structurebox.Enable()
[c097f02]	1028	self.text2.Enable()
[c77d859]	1029
[59a7f2d]	1030	## check model type to show sizer
[c77d859]	1031	if self.model !=None:
	1032	self._set_model_sizer_selection( self.model )
	1033
	1034	sizer_selection.Add(self.text1)
	1035	sizer_selection.Add((5,5))
	1036	sizer_selection.Add(self.formfactorbox)
	1037	sizer_selection.Add((5,5))
	1038	sizer_selection.Add(self.text2)
	1039	sizer_selection.Add((5,5))
	1040	sizer_selection.Add(self.structurebox)
	1041	sizer_selection.Add((5,5))
	1042
	1043	boxsizer1.Add( sizer_radiobutton )
	1044	boxsizer1.Add( (20,20))
	1045	boxsizer1.Add( sizer_selection )
	1046	if object !=None:
[f6b65b8]	1047	boxsizer1.Add( (-72,-72))
	1048	boxsizer1.Add( object, 0, wx.ALIGN_RIGHT\| wx.RIGHT, 10)
	1049	boxsizer1.Add( (60,60))
[c77d859]	1050	#--------------------------------------------------------
	1051	sizer.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	1052	sizer.Layout()
	1053	self.SetScrollbars(20,20,200,100)
	1054
	1055
	1056	def _show_combox(self, event):
	1057	"""
	1058	Show combox box associate with type of model selected
	1059	"""
[376916c]	1060	## Don't want to populate combo box again if the event comes from check box
	1061	if self.shape_rbutton.GetValue()and\
	1062	event.GetEventObject()==self.shape_rbutton:
	1063	##fill the combobox with form factor list
[3b605bb]	1064	self.structurebox.SetSelection(0)
	1065	self.structurebox.Disable()
[c77d859]	1066	self.formfactorbox.Clear()
[376916c]	1067	self._populate_box( self.formfactorbox,self.model_list_box["Shapes"])
[c77d859]	1068
[376916c]	1069	if self.shape_indep_rbutton.GetValue()and\
	1070	event.GetEventObject()==self.shape_indep_rbutton:
	1071	##fill the combobox with shape independent factor list
[3b605bb]	1072	self.structurebox.SetSelection(0)
	1073	self.structurebox.Disable()
[c77d859]	1074	self.formfactorbox.Clear()
	1075	self._populate_box( self.formfactorbox,
[376916c]	1076	self.model_list_box["Shape-Independent"])
[c77d859]	1077
[376916c]	1078	if self.struct_rbutton.GetValue() and\
	1079	event.GetEventObject()==self.struct_rbutton:
	1080	##fill the combobox with structure factor list
[3b605bb]	1081	self.structurebox.SetSelection(0)
	1082	self.structurebox.Disable()
[c77d859]	1083	self.formfactorbox.Clear()
	1084	self._populate_box( self.formfactorbox,
	1085	self.model_list_box["Structure Factors"])
	1086
[376916c]	1087	if self.plugin_rbutton.GetValue()and\
	1088	event.GetEventObject()==self.plugin_rbutton:
	1089
	1090	##fill the combobox with form factor list
[3b605bb]	1091	self.structurebox.Disable()
[c77d859]	1092	self.formfactorbox.Clear()
	1093	self._populate_box( self.formfactorbox,
[376916c]	1094	self.model_list_box["Customized Models"])
	1095
[77e23a2]	1096	self._on_select_model(event=None)
[3b605bb]	1097	self.sizer4_4.Layout()
[9853ad0]	1098	self.sizer4.Layout()
[3b605bb]	1099	self.Layout()
	1100	self.Refresh()
	1101	self.SetScrollbars(20,20,200,100)
[c77d859]	1102
	1103	def _populate_box(self, combobox, list):
	1104	"""
	1105	fill combox box with dict item
	1106	@param list: contains item to fill the combox
	1107	item must model class
	1108	"""
	1109	for models in list:
	1110	model= models()
	1111	name = model.__class__.__name__
[0a518e4c]	1112	if models.__name__!="NoStructure":
	1113	if hasattr(model, "name"):
	1114	name = model.name
	1115	combobox.Append(name,models)
[71f0373]	1116
[c77d859]	1117	return 0
	1118
	1119
[59a7f2d]	1120	def _on_select_model_helper(self):
[c77d859]	1121	"""
	1122	call back for model selection
	1123	"""
[376916c]	1124	## reset dictionary containing reference to dispersion
	1125	self._disp_obj_dict = {}
	1126	self.disp_cb_dict ={}
[c77d859]	1127	f_id = self.formfactorbox.GetCurrentSelection()
	1128	form_factor = self.formfactorbox.GetClientData( f_id )
[376916c]	1129	if not form_factor in self.model_list_box["multiplication"]:
[3b605bb]	1130	self.structurebox.Disable()
	1131	self.structurebox.SetSelection(0)
[c097f02]	1132	self.text2.Disable()
[87fbc60]	1133	else:
	1134	self.structurebox.Enable()
[c097f02]	1135	self.text2.Enable()
[a074145]	1136
[3b605bb]	1137	s_id = self.structurebox.GetCurrentSelection()
	1138	struct_factor = self.structurebox.GetClientData( s_id )
[a074145]	1139
[3b605bb]	1140	if struct_factor !=None:
[9853ad0]	1141	from sans.models.MultiplicationModel import MultiplicationModel
	1142	self.model= MultiplicationModel(form_factor(),struct_factor())
[0aeabc6]	1143
[c77d859]	1144	else:
[fbf4bf8]	1145	if form_factor != None:
	1146	self.model= form_factor()
	1147	else:
	1148	self.model = None
	1149	return self.model
[a074145]	1150
[cfc0913]	1151	## post state to fit panel
[fc6ea43]	1152	self.state.model =self.model
[71f0373]	1153	self.disp_list =self.model.getDispParamList()
	1154	self.state.disp_list = self.disp_list
	1155
[fc6ea43]	1156	## post state to fit panel
	1157	event = PageInfoEvent(page = self)
	1158	wx.PostEvent(self.parent, event)
	1159
[3b605bb]	1160	self.sizer4_4.Layout()
	1161	self.sizer4.Layout()
	1162	self.Layout()
	1163	self.SetScrollbars(20,20,200,100)
	1164	self.Refresh()
	1165
[c77d859]	1166
	1167
	1168	def _onparamEnter(self,event):
	1169	"""
	1170	when enter value on panel redraw model according to changed
	1171	"""
[69bee6d]	1172	tcrtl= event.GetEventObject()
	1173	if check_float(tcrtl):
	1174	self._onparamEnter_helper()
	1175	else:
	1176	msg= "Cannot Plot :Must enter a number!!! "
	1177	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	1178	return
[c77d859]	1179
	1180
	1181	def _check_value_enter(self, list, modified):
	1182	"""
	1183	@param list: model parameter and panel info
	1184	each item of the list should be as follow:
	1185	item=[cb state, name, value, "+/-", error of fit, min, max , units]
	1186	"""
	1187	is_modified = modified
	1188	if len(list)==0:
	1189	return is_modified
	1190	for item in list:
	1191	try:
[77e23a2]	1192	name = str(item[1])
[0a518e4c]	1193	if hasattr(self,"text2_3"):
	1194	self.text2_3.Hide()
[77e23a2]	1195	## check model parameters range
	1196	## check minimun value
	1197	param_min= None
	1198	param_max= None
	1199	if item[5]!= None:
[298b762]	1200	if item[5].GetValue().lstrip().rstrip()!="":
[77e23a2]	1201	param_min = float(item[5].GetValue())
[cfc0913]	1202
[77e23a2]	1203	## check maximum value
	1204	if item[6]!= None:
[298b762]	1205	if item[6].GetValue().lstrip().rstrip()!="":
[77e23a2]	1206	param_max = float(item[6].GetValue())
	1207
	1208	from sans.guiframe.utils import check_value
	1209	if param_min != None and param_max !=None:
[87a43fe]	1210	if not check_value(item[5], item[6]):
[77e23a2]	1211	msg= "Wrong Fit range entered for parameter "
	1212	msg+= "name %s of model %s "%(name, self.model.name)
	1213	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
[cfc0913]	1214	if name in self.model.details.keys():
	1215	self.model.details[name][1:]= param_min,param_max
[87a43fe]	1216
[77e23a2]	1217	## hide statictext +/-
[0a518e4c]	1218	if item[3]!=None:
	1219	item[3].Hide()
[77e23a2]	1220	## hide textcrtl for error after fit
[0a518e4c]	1221	if item[4]!=None:
	1222	item[4].Clear()
	1223	item[4].Hide()
[69bee6d]	1224
[0a518e4c]	1225	value= float(item[2].GetValue())
[69bee6d]	1226
[0a518e4c]	1227	# If the value of the parameter has changed,
[1328e03]	1228	# +update the model and set the is_modified flag
[0a518e4c]	1229	if value != self.model.getParam(name):
	1230	self.model.setParam(name,value)
[69bee6d]	1231	is_modified = True
	1232
[c77d859]	1233	except:
[1328e03]	1234	msg= "Model Drawing Error:wrong value entered : %s"% sys.exc_value
	1235	wx.PostEvent(self.parent.parent, StatusEvent(status = msg ))
	1236	return
[0a518e4c]	1237
[c77d859]	1238	return is_modified
	1239
	1240
	1241	def _set_dipers_Param(self, event):
	1242	"""
	1243	Add more item to select user dispersity
	1244	"""
[1c1436d]	1245	self._reset_dispersity()
[70c57ebf]	1246	if self.model ==None:
	1247	self.model_disp.Hide()
	1248	self.disp_box.Hide()
	1249	self.sizer4_4.Clear(True)
	1250	## post state to fit panel
	1251	self.save_current_state()
	1252	return
	1253
[1c1436d]	1254	if self.enable_disp.GetValue():
	1255	self.model_disp.Show(True)
	1256	self.disp_box.Show(True)
	1257	## layout for model containing no dispersity parameters
[71f0373]	1258	self.disp_list= self.model.getDispParamList()
[1c1436d]	1259	if len(self.disp_list)==0:
	1260	self._layout_sizer_noDipers()
[c77d859]	1261	else:
[1c1436d]	1262	## set gaussian sizer
	1263	self._on_select_Disp(event=None)
	1264	else:
	1265	self.model_disp.Hide()
	1266	self.disp_box.Hide()
[71f0373]	1267	self.disp_box.SetSelection(0)
[1c1436d]	1268	self.sizer4_4.Clear(True)
[71f0373]	1269	#self._reset_dispersity()
[1c1436d]	1270
	1271	## post state to fit panel
	1272	self.save_current_state()
[71f0373]	1273	self.sizer4_4.Layout()
	1274	self.sizer4.Layout()
	1275	self.Layout()
	1276	self.Refresh()
	1277	self.SetScrollbars(20,20,200,100)
[3b605bb]	1278
[0a518e4c]	1279
[c77d859]	1280
	1281	def _layout_sizer_noDipers(self):
	1282	"""
	1283	Draw a sizer with no dispersity info
	1284	"""
	1285	ix=0
	1286	iy=1
	1287	self.fittable_param=[]
	1288	self.fixed_param=[]
[1c1436d]	1289	self.orientation_params_disp=[]
	1290
[c77d859]	1291	self.model_disp.Hide()
	1292	self.disp_box.Hide()
	1293	self.sizer4_4.Clear(True)
	1294	model_disp = wx.StaticText(self, -1, 'No PolyDispersity for this model')
	1295	self.sizer4_4.Add(model_disp,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
[dcf29d7]	1296	self.sizer4_4.Layout()
[c77d859]	1297	self.sizer4.Layout()
	1298	self.SetScrollbars(20,20,200,100)
[3b605bb]	1299
[c77d859]	1300
	1301	def _reset_dispersity(self):
	1302	"""
	1303	put gaussian dispersity into current model
	1304	"""
[1c1436d]	1305	if len(self.param_toFit)>0:
	1306	for item in self.fittable_param:
[513115c]	1307	if item in self.param_toFit:
[1c1436d]	1308	self.param_toFit.remove(item)
	1309	for item in self.orientation_params_disp:
[513115c]	1310	if item in self.param_toFit:
[1c1436d]	1311	self.param_toFit.remove(item)
[513115c]	1312
[c77d859]	1313	self.fittable_param=[]
	1314	self.fixed_param=[]
[5812a55]	1315	self.orientation_params_disp=[]
[71f0373]	1316	self.values=[]
	1317	self.weights=[]
[1c1436d]	1318
[71f0373]	1319
[c77d859]	1320	from sans.models.dispersion_models import GaussianDispersion
	1321	if len(self.disp_cb_dict)==0:
[3b605bb]	1322	self.sizer4_4.Clear(True)
	1323	self.sizer4_4.Layout()
	1324	self.sizer4.Layout()
	1325	self.Layout()
	1326	self.Refresh()
	1327	self.SetScrollbars(20,20,200,100)
[c77d859]	1328	return
	1329
[71f0373]	1330
[c77d859]	1331	for p in self.disp_cb_dict:
	1332	# The parameter was un-selected. Go back to Gaussian model (with 0 pts)
	1333	disp_model = GaussianDispersion()
	1334	# Store the object to make it persist outside the scope of this method
	1335	#TODO: refactor model to clean this up?
	1336	self._disp_obj_dict[p] = disp_model
	1337	# Set the new model as the dispersion object for the selected parameter
	1338	self.model.set_dispersion(p, disp_model)
	1339	# Redraw the model
[0aeabc6]	1340
[376916c]	1341	self._draw_model()
[c77d859]	1342
[3b605bb]	1343	self.sizer4_4.Layout()
	1344	self.sizer4.Layout()
	1345	self.Layout()
	1346	self.SetScrollbars(20,20,200,100)
	1347	self.Refresh()
	1348
	1349
[c77d859]	1350
	1351	def _on_select_Disp(self,event):
	1352	"""
	1353	allow selecting different dispersion
	1354	self.disp_list should change type later .now only gaussian
	1355	"""
[b787e68c]	1356
	1357	n = self.disp_box.GetCurrentSelection()
	1358	dispersity= self.disp_box.GetClientData(n)
[c77d859]	1359	name= dispersity.__name__
[376916c]	1360	if name == "GaussianDispersion":
[c77d859]	1361	self._set_sizer_gaussian()
	1362
[71f0373]	1363	if name == "ArrayDispersion":
[c77d859]	1364	self._set_sizer_arraydispersion()
[3b605bb]	1365
[b787e68c]	1366	self.state.disp_box= n
	1367	## post state to fit panel
	1368	event = PageInfoEvent(page = self)
	1369	wx.PostEvent(self.parent, event)
	1370
[3b605bb]	1371	self.sizer4_4.Layout()
	1372	self.sizer4.Layout()
	1373	self.Layout()
	1374	self.SetScrollbars(20,20,200,100)
	1375	self.Refresh()
	1376
	1377
	1378
[b787e68c]	1379
[c77d859]	1380	def _set_sizer_arraydispersion(self):
	1381	"""
	1382	draw sizer with array dispersity parameters
	1383	"""
[513115c]	1384	if len(self.param_toFit)>0:
	1385	for item in self.fittable_param:
	1386	if item in self.param_toFit:
	1387	self.param_toFit.remove(item)
	1388	for item in self.orientation_params_disp:
	1389	if item in self.param_toFit:
	1390	self.param_toFit.remove(item)
[71f0373]	1391
[1c1436d]	1392	self.fittable_param=[]
	1393	self.fixed_param=[]
[60132ef]	1394	self.orientation_params_disp=[]
[c77d859]	1395	self.sizer4_4.Clear(True)
	1396	ix=0
	1397	iy=1
	1398	disp1 = wx.StaticText(self, -1, 'Array Dispersion')
	1399	self.sizer4_4.Add(disp1,( iy, ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	1400
	1401	# Look for model parameters to which we can apply an ArrayDispersion model
	1402	# Add a check box for each parameter.
	1403	self.disp_cb_dict = {}
	1404	for p in self.model.dispersion.keys():
	1405	ix+=1
	1406	self.disp_cb_dict[p] = wx.CheckBox(self, -1, p, (10, 10))
	1407
	1408	wx.EVT_CHECKBOX(self, self.disp_cb_dict[p].GetId(), self.select_disp_angle)
	1409	self.sizer4_4.Add(self.disp_cb_dict[p], (iy, ix), (1,1), wx.EXPAND\|wx.ADJUST_MINSIZE, 0)
	1410
	1411	ix =0
	1412	iy +=1
	1413	self.sizer4_4.Add((20,20),(iy,ix),(1,1), wx.LEFT\|wx.EXPAND\|wx.ADJUST_MINSIZE, 15)
	1414	self.sizer4_4.Layout()
	1415	self.sizer4.Layout()
	1416	self.SetScrollbars(20,20,200,100)
[71f0373]	1417
[70c57ebf]	1418
[c77d859]	1419	def _set_range_sizer(self, title, object1=None,object=None):
	1420	"""
	1421	Fill the
	1422	"""
	1423	self.sizer5.Clear(True)
	1424	box_description= wx.StaticBox(self, -1,str(title))
	1425	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	1426	#--------------------------------------------------------------
	1427	self.qmin = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[b5e98540]	1428	self.qmin.SetValue(str(self.qmin_x))
[c77d859]	1429	self.qmin.SetToolTipString("Minimun value of Q in linear scale.")
[1328e03]	1430	self.qmin.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[dcf29d7]	1431	self.qmin.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
[c77d859]	1432	self.qmin.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
	1433
	1434	self.qmax = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20))
[b5e98540]	1435	self.qmax.SetValue(str(self.qmax_x))
[c77d859]	1436	self.qmax.SetToolTipString("Maximum value of Q in linear scale.")
[1328e03]	1437	self.qmax.Bind(wx.EVT_SET_FOCUS, self.onSetFocus)
[dcf29d7]	1438	self.qmax.Bind(wx.EVT_KILL_FOCUS, self._onparamEnter)
[c77d859]	1439	self.qmax.Bind(wx.EVT_TEXT_ENTER, self._onparamEnter)
	1440
	1441	sizer_horizontal=wx.BoxSizer(wx.HORIZONTAL)
	1442	sizer= wx.GridSizer(3, 3,5, 5)
	1443
	1444	sizer.Add((5,5))
	1445	sizer.Add(wx.StaticText(self, -1, 'Min'))
	1446	sizer.Add(wx.StaticText(self, -1, 'Max'))
	1447	sizer.Add(wx.StaticText(self, -1, 'Q range'))
	1448
	1449	sizer.Add(self.qmin)
	1450	sizer.Add(self.qmax)
	1451	sizer_horizontal.Add(sizer)
	1452	if object!=None:
	1453	sizer_horizontal.Add(object)
	1454
	1455	if object1!=None:
	1456	boxsizer1.Add(object1)
	1457	boxsizer1.Add((10,10))
	1458	boxsizer1.Add(sizer_horizontal)
[cfc0913]	1459	## save state
	1460	self.save_current_state()
[c77d859]	1461	#----------------------------------------------------------------
	1462	self.sizer5.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	1463	self.sizer5.Layout()
[dcf29d7]	1464	self.Layout()
[c77d859]	1465	self.SetScrollbars(20,20,200,100)
	1466
[cfc0913]	1467
[00c3aac]	1468	def _fill_save_sizer(self):
	1469	"""
	1470	Draw the layout for saving option
	1471	"""
[6d1235b]	1472	# Skipping save state functionality for release 0.9.0
[dad49a0]	1473	#return
[6d1235b]	1474
[00c3aac]	1475	self.sizer6.Clear(True)
[45ed4dad]	1476	box_description= wx.StaticBox(self, -1,"Save Status")
[00c3aac]	1477	boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
	1478	sizer_save = wx.BoxSizer(wx.HORIZONTAL)
	1479
[45ed4dad]	1480	self.btSave_title = wx.StaticText(self, -1, 'Save the current panel status')
[00c3aac]	1481	self.btSave = wx.Button(self,wx.NewId(),'Save')
	1482	self.btSave.Bind(wx.EVT_BUTTON, self.onSave,id= self.btSave.GetId())
[45ed4dad]	1483	self.btSave.SetToolTipString("Save current panel status")
	1484
	1485	sizer_save.Add(self.btSave_title)
	1486	sizer_save.Add((20,20),0, wx.LEFT\|wx.RIGHT\|wx.EXPAND,45)
	1487
[00c3aac]	1488	sizer_save.Add(self.btSave)
	1489
	1490	boxsizer1.Add(sizer_save)
	1491	self.sizer6.Add(boxsizer1,0, wx.EXPAND \| wx.ALL, 10)
	1492	self.sizer6.Layout()
	1493	self.SetScrollbars(20,20,200,100)
	1494
[c77d859]	1495
	1496

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

SasView

source: sasview/sansview/perspectives/fitting/basepage.py @ b0eee0f0

Download in other formats: