invariant_details.py @ dca6188

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 dca6188 was 78a205a, checked in by Mathieu Doucet <doucetm@…>, 10 years ago
pylint fixes
Property mode set to `100644`
File size: 22.3 KB

Rev	Line
[a3efdeb]	1	"""
	2	Invariant panel
	3	"""
[9ce7641c]	4	import wx
	5	import sys
[f6518f8]	6
[79492222]	7	from sas.guiframe.utils import format_number
[518d35d]	8	from invariant_widgets import OutputTextCtrl
[9ce7641c]	9	# Dimensions related to chart
[78a205a]	10	RECTANGLE_WIDTH = 400.0
[9ce7641c]	11	RECTANGLE_HEIGHT = 20
[78a205a]	12	# Invariant panel size
[9ce7641c]	13	_BOX_WIDTH = 76
[d0cc0bbc]	14
[78a205a]	15	# scale to use for a bar of value zero
	16	RECTANGLE_SCALE = 0.0001
	17	DEFAULT_QSTAR = 1.0
	18
[4a2b054]	19	if sys.platform.count("win32") > 0:
[9ce7641c]	20	_STATICBOX_WIDTH = 450
	21	PANEL_WIDTH = 500
[d0cc0bbc]	22	PANEL_HEIGHT = 430
[9ce7641c]	23	FONT_VARIANT = 0
	24	else:
	25	_STATICBOX_WIDTH = 480
	26	PANEL_WIDTH = 530
[d0cc0bbc]	27	PANEL_HEIGHT = 430
[9ce7641c]	28	FONT_VARIANT = 1
[78a205a]	29
	30	ERROR_COLOR = wx.Colour(255, 0, 0, 128)
	31	EXTRAPOLATION_COLOR = wx.Colour(169, 169, 168, 128)
	32	INVARIANT_COLOR = wx.Colour(67, 208, 128, 128)
	33
	34
[d0cc0bbc]	35	class InvariantContainer(wx.Object):
[45802d4]	36	"""
[d7a39e5]	37	This class stores some values resulting resulting from invariant
[78a205a]	38	calculations. Given the value of total invariant, this class can also
[d7a39e5]	39	determine the percentage of invariants resulting from extrapolation.
[45802d4]	40	"""
[d0cc0bbc]	41	def __init__(self):
[78a205a]	42	# invariant at low range
[eed601e]	43	self.qstar_low = None
[78a205a]	44	# invariant at low range error
[eed601e]	45	self.qstar_low_err = None
[78a205a]	46	# invariant
[eed601e]	47	self.qstar = None
[78a205a]	48	# invariant error
[eed601e]	49	self.qstar_err = None
[78a205a]	50	# invariant at high range
[eed601e]	51	self.qstar_high = None
[78a205a]	52	# invariant at high range error
[eed601e]	53	self.qstar_high_err = None
[78a205a]	54	# invariant total
[d0cc0bbc]	55	self.qstar_total = None
[78a205a]	56	# invariant error
[d0cc0bbc]	57	self.qstar_total_err = None
[78a205a]	58	# scale
[eed601e]	59	self.qstar_low_percent = None
	60	self.qstar_high_percent = None
	61	self.qstar_percent = None
[d0cc0bbc]	62	# warning message
	63	self.existing_warning = False
	64	self.warning_msg = "No Details on calculations available...\n"
[78a205a]	65
[d0cc0bbc]	66	def compute_percentage(self):
	67	"""
[d7a39e5]	68	Compute percentage of each invariant
[d0cc0bbc]	69	"""
[eed601e]	70	if self.qstar_total is None:
	71	self.qstar_percent = None
	72	self.qstar_low = None
	73	self.qstar_high = None
	74	self.check_values()
[78a205a]	75	return
	76
	77	# compute invariant percentage
[eed601e]	78	if self.qstar is None:
	79	self.qstar_percent = None
	80	else:
	81	try:
[78a205a]	82	self.qstar_percent = float(self.qstar) / float(self.qstar_total)
[eed601e]	83	except:
	84	self.qstar_percent = 'error'
[78a205a]	85	# compute low q invariant percentage
[eed601e]	86	if self.qstar_low is None:
	87	self.qstar_low_percent = None
	88	else:
	89	try:
[4a2b054]	90	self.qstar_low_percent = float(self.qstar_low)\
[78a205a]	91	/ float(self.qstar_total)
[eed601e]	92	except:
	93	self.qstar_low_percent = 'error'
[78a205a]	94	# compute high q invariant percentage
[eed601e]	95	if self.qstar_high is None:
	96	self.qstar_high_percent = None
	97	else:
	98	try:
[4a2b054]	99	self.qstar_high_percent = float(self.qstar_high)\
[78a205a]	100	/ float(self.qstar_total)
[eed601e]	101	except:
	102	self.qstar_high_percent = 'error'
[ee54fcf0]	103	wx.CallAfter(self.check_values)
[78a205a]	104
[d0cc0bbc]	105	def check_values(self):
	106	"""
[d7a39e5]	107	check the validity if invariant
[d0cc0bbc]	108	"""
[eed601e]	109	if self.qstar_total is None and self.qstar is None:
	110	self.warning_msg = "Invariant not calculated.\n"
[78a205a]	111	return
[eed601e]	112	if self.qstar_total == 0:
[4a2b054]	113	self.existing_warning = True
	114	self.warning_msg = "Invariant is zero. \n"
	115	self.warning_msg += "The calculations are likely "
	116	self.warning_msg += "to be unreliable!\n"
[78a205a]	117	return
	118	# warning to the user when the extrapolated invariant is greater than %5
[3fab6ef]	119	msg = ''
[eed601e]	120	if self.qstar_percent == 'error':
[ee54fcf0]	121	try:
	122	float(self.qstar)
	123	except:
	124	self.existing_warning = True
	125	msg += 'Error occurred when computing invariant from data.\n '
[f6518f8]	126	if self.qstar_percent > 1:
	127	self.existing_warning = True
	128	msg += "Invariant Q contribution is greater "
	129	msg += "than 100% .\n"
[eed601e]	130	if self.qstar_low_percent == 'error':
[ee54fcf0]	131	try:
	132	float(self.qstar_low)
	133	except:
	134	self.existing_warning = True
	135	msg += "Error occurred when computing extrapolated invariant"
	136	msg += " at low-Q region.\n"
[78a205a]	137	elif self.qstar_low_percent is not None:
[f6518f8]	138	if self.qstar_low_percent >= 0.05:
	139	self.existing_warning = True
	140	msg += "Extrapolated contribution at Low Q is higher "
	141	msg += "than 5% of the invariant.\n"
	142	elif self.qstar_low_percent < 0:
	143	self.existing_warning = True
	144	msg += "Extrapolated contribution at Low Q < 0.\n"
	145	elif self.qstar_low_percent > 1:
	146	self.existing_warning = True
	147	msg += "Extrapolated contribution at Low Q is greater "
	148	msg += "than 100% .\n"
[eed601e]	149	if self.qstar_high_percent == 'error':
[ee54fcf0]	150	try:
	151	float(self.qstar_high)
	152	except:
	153	self.existing_warning = True
	154	msg += 'Error occurred when computing extrapolated'
	155	msg += ' invariant at high-Q region.\n'
[f6518f8]	156	elif self.qstar_high_percent is not None:
	157	if self.qstar_high_percent >= 0.05:
	158	self.existing_warning = True
	159	msg += "Extrapolated contribution at High Q is higher "
	160	msg += "than 5% of the invariant.\n"
	161	elif self.qstar_high_percent < 0:
	162	self.existing_warning = True
	163	msg += "Extrapolated contribution at High Q < 0.\n"
	164	elif self.qstar_high_percent > 1:
	165	self.existing_warning = True
	166	msg += "Extrapolated contribution at High Q is greater "
	167	msg += "than 100% .\n"
[eed601e]	168	if (self.qstar_low_percent not in [None, "error"]) and \
	169	(self.qstar_high_percent not in [None, "error"])\
	170	and self.qstar_low_percent + self.qstar_high_percent >= 0.05:
[d0cc0bbc]	171	self.existing_warning = True
[3fab6ef]	172	msg += "The sum of all extrapolated contributions is higher "
	173	msg += "than 5% of the invariant.\n"
[78a205a]	174
[d0cc0bbc]	175	if self.existing_warning:
[3fab6ef]	176	self.warning_msg = ''
[78a205a]	177	self.warning_msg += msg
[4a2b054]	178	self.warning_msg += "The calculations are likely to be"
	179	self.warning_msg += " unreliable!\n"
[d0cc0bbc]	180	else:
	181	self.warning_msg = "No Details on calculations available...\n"
[78a205a]	182
[d3fac18]	183	class InvariantDetailsPanel(wx.Dialog):
[9ce7641c]	184	"""
[78a205a]	185	This panel describes proportion of invariants
[9ce7641c]	186	"""
[78a205a]	187	def __init__(self, parent=None, id=-1, qstar_container=None,
	188	title="Invariant Details",
	189	size=(PANEL_WIDTH, PANEL_HEIGHT)):
[4a2b054]	190	wx.Dialog.__init__(self, parent=parent, id=id, title=title, size=size)
[78a205a]	191
	192	# Font size
[f75ea4a]	193	self.SetWindowVariant(variant=FONT_VARIANT)
	194	self.parent = parent
[78a205a]	195	# self.qstar_container
[f75ea4a]	196	self.qstar_container = qstar_container
[78a205a]	197	# warning message
[f75ea4a]	198	self.warning_msg = self.qstar_container.warning_msg
[78a205a]	199
	200	# Define scale of each bar
[f75ea4a]	201	self.low_inv_percent = self.qstar_container.qstar_low_percent
	202	self.low_scale = self.get_scale(percentage=self.low_inv_percent,
[78a205a]	203	scale_name="Extrapolated at Low Q")
[f75ea4a]	204	self.inv_percent = self.qstar_container.qstar_percent
[78a205a]	205	self.inv_scale = self.get_scale(percentage=self.inv_percent,
	206	scale_name="Inv in Q range")
[f75ea4a]	207	self.high_inv_percent = self.qstar_container.qstar_high_percent
	208	self.high_scale = self.get_scale(percentage=self.high_inv_percent,
	209	scale_name="Extrapolated at High Q")
[78a205a]	210
	211	# Default color the extrapolation bar is grey
[9cec2dd]	212	self.extrapolation_color_low = EXTRAPOLATION_COLOR
	213	self.extrapolation_color_high = EXTRAPOLATION_COLOR
	214	self.invariant_color = INVARIANT_COLOR
[78a205a]	215	# change color of high and low bar when necessary
[f75ea4a]	216	self.set_color_bar()
[78a205a]	217	# draw the panel itself
[f75ea4a]	218	self._do_layout()
	219	self.set_values()
[78a205a]	220
[9ce7641c]	221	def _define_structure(self):
	222	"""
[d7a39e5]	223	Define main sizers needed for this panel
[9ce7641c]	224	"""
[78a205a]	225	# Box sizers must be defined first before defining buttons/textctrls
[4a2b054]	226	# (MAC).
[9ce7641c]	227	self.main_sizer = wx.BoxSizer(wx.VERTICAL)
[78a205a]	228	# Sizer related to chart
[9ce7641c]	229	chart_box = wx.StaticBox(self, -1, "Invariant Chart")
	230	self.chart_sizer = wx.StaticBoxSizer(chart_box, wx.VERTICAL)
[4a2b054]	231	self.chart_sizer.SetMinSize((PANEL_WIDTH - 50, 110))
[78a205a]	232	# Sizer related to invariant values
	233	self.invariant_sizer = wx.GridBagSizer(4, 4)
[d0cc0bbc]	234	invariant_box = wx.StaticBox(self, -1, "Numerical Values")
[78a205a]	235	self.invariant_box_sizer = wx.StaticBoxSizer(invariant_box, wx.HORIZONTAL)
[d0cc0bbc]	236
[4a2b054]	237	self.invariant_box_sizer.SetMinSize((PANEL_WIDTH - 50, -1))
[78a205a]	238	# Sizer related to warning message
[9ce7641c]	239	warning_box = wx.StaticBox(self, -1, "Warning")
	240	self.warning_sizer = wx.StaticBoxSizer(warning_box, wx.VERTICAL)
[78a205a]	241	self.warning_sizer.SetMinSize((PANEL_WIDTH - 50, -1))
	242	# Sizer related to button
[9ce7641c]	243	self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
[78a205a]	244
[9ce7641c]	245	def _layout_shart(self):
	246	"""
[d7a39e5]	247	Draw widgets related to chart
[9ce7641c]	248	"""
	249	self.panel_chart = wx.Panel(self)
	250	self.panel_chart.Bind(wx.EVT_PAINT, self.on_paint)
[78a205a]	251	self.chart_sizer.Add(self.panel_chart, 1, wx.EXPAND \| wx.ALL, 0)
	252
[9ce7641c]	253	def _layout_invariant(self):
	254	"""
[d7a39e5]	255	Draw widgets related to invariant
[9ce7641c]	256	"""
[78a205a]	257	uncertainty = "+/-"
[3df5ecf]	258	unit_invariant = '[1/(cm * A^3)]'
[78a205a]	259
[eed601e]	260	invariant_txt = wx.StaticText(self, -1, 'Q* from Data ')
	261	invariant_txt.SetToolTipString("Invariant in the data set's Q range.")
[4a2b054]	262	self.invariant_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1))
	263	hint_msg = "Invariant in the data set's Q range."
	264	self.invariant_tcl.SetToolTipString(hint_msg)
	265	self.invariant_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1))
	266	hint_msg = "Uncertainty on the invariant from data's range."
	267	self.invariant_err_tcl.SetToolTipString(hint_msg)
[9ce7641c]	268	invariant_units_txt = wx.StaticText(self, -1, unit_invariant)
[4a2b054]	269	hint_msg = "Unit of the invariant from data's Q range"
	270	invariant_units_txt.SetToolTipString(hint_msg)
[78a205a]	271
[eed601e]	272	invariant_low_txt = wx.StaticText(self, -1, 'Q* from Low-Q')
[4a2b054]	273	hint_msg = "Extrapolated invariant from low-Q range."
	274	invariant_low_txt.SetToolTipString(hint_msg)
	275	self.invariant_low_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1))
	276	hint_msg = "Extrapolated invariant from low-Q range."
	277	self.invariant_low_tcl.SetToolTipString(hint_msg)
	278	self.invariant_low_err_tcl = OutputTextCtrl(self, -1,
	279	size=(_BOX_WIDTH, -1))
	280	hint_msg = "Uncertainty on the invariant from low-Q range."
	281	self.invariant_low_err_tcl.SetToolTipString(hint_msg)
	282	invariant_low_units_txt = wx.StaticText(self, -1, unit_invariant)
	283	hint_msg = "Unit of the extrapolated invariant from low-Q range"
	284	invariant_low_units_txt.SetToolTipString(hint_msg)
[78a205a]	285
[eed601e]	286	invariant_high_txt = wx.StaticText(self, -1, 'Q* from High-Q')
[4a2b054]	287	hint_msg = "Extrapolated invariant from high-Q range"
	288	invariant_high_txt.SetToolTipString(hint_msg)
	289	self.invariant_high_tcl = OutputTextCtrl(self, -1,
	290	size=(_BOX_WIDTH, -1))
	291	hint_msg = "Extrapolated invariant from high-Q range"
	292	self.invariant_high_tcl.SetToolTipString(hint_msg)
	293	self.invariant_high_err_tcl = OutputTextCtrl(self, -1,
	294	size=(_BOX_WIDTH, -1))
	295	hint_msg = "Uncertainty on the invariant from high-Q range."
[98eefdb]	296	self.invariant_high_err_tcl.SetToolTipString(hint_msg)
[4a2b054]	297	invariant_high_units_txt = wx.StaticText(self, -1, unit_invariant)
	298	hint_msg = "Unit of the extrapolated invariant from high-Q range"
	299	invariant_high_units_txt.SetToolTipString(hint_msg)
[78a205a]	300
	301	# Invariant low
[d0cc0bbc]	302	iy = 0
[78a205a]	303	ix = 0
[4a2b054]	304	self.invariant_sizer.Add(invariant_low_txt, (iy, ix), (1, 1),
[78a205a]	305	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
[9ce7641c]	306	ix += 1
[4a2b054]	307	self.invariant_sizer.Add(self.invariant_low_tcl, (iy, ix), (1, 1),
[78a205a]	308	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	309	ix += 1
[78a205a]	310	self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty),
	311	(iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	312	ix += 1
[4a2b054]	313	self.invariant_sizer.Add(self.invariant_low_err_tcl, (iy, ix), (1, 1),
[78a205a]	314	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	315	ix += 1
[78a205a]	316	self.invariant_sizer.Add(invariant_low_units_txt,
	317	(iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
	318	# Invariant
[9ce7641c]	319	iy += 1
[78a205a]	320	ix = 0
[4a2b054]	321	self.invariant_sizer.Add(invariant_txt, (iy, ix), (1, 1),
[78a205a]	322	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
[9ce7641c]	323	ix += 1
[4a2b054]	324	self.invariant_sizer.Add(self.invariant_tcl, (iy, ix), (1, 1),
[78a205a]	325	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	326	ix += 1
	327	self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty),
[78a205a]	328	(iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	329	ix += 1
[4a2b054]	330	self.invariant_sizer.Add(self.invariant_err_tcl, (iy, ix), (1, 1),
[78a205a]	331	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
	332	ix += 1
	333	self.invariant_sizer.Add(invariant_units_txt,
	334	(iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
	335	# Invariant high
[9ce7641c]	336	iy += 1
[78a205a]	337	ix = 0
[4a2b054]	338	self.invariant_sizer.Add(invariant_high_txt, (iy, ix), (1, 1),
[78a205a]	339	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 15)
[9ce7641c]	340	ix += 1
[4a2b054]	341	self.invariant_sizer.Add(self.invariant_high_tcl, (iy, ix), (1, 1),
[78a205a]	342	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	343	ix += 1
	344	self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty),
[78a205a]	345	(iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	346	ix += 1
[4a2b054]	347	self.invariant_sizer.Add(self.invariant_high_err_tcl, (iy, ix), (1, 1),
[78a205a]	348	wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
[9ce7641c]	349	ix += 1
[78a205a]	350	self.invariant_sizer.Add(invariant_high_units_txt,
	351	(iy, ix), (1, 1), wx.EXPAND \| wx.ADJUST_MINSIZE, 0)
	352	self.invariant_box_sizer.Add(self.invariant_sizer, 0, wx.TOP \| wx.BOTTOM, 10)
	353
[9ce7641c]	354	def _layout_warning(self):
	355	"""
[d7a39e5]	356	Draw widgets related to warning
[9ce7641c]	357	"""
[78a205a]	358	# Warning [string]
	359	self.warning_msg_txt = wx.StaticText(self, -1, self.warning_msg)
[d0cc0bbc]	360	if self.qstar_container.existing_warning:
[78a205a]	361	self.warning_msg_txt.SetForegroundColour('red')
[277fad8]	362	self.warning_sizer.AddMany([(self.warning_msg_txt, 0,
[78a205a]	363	wx.LEFT \| wx.EXPAND \| wx.ADJUST_MINSIZE, 10)])
	364
[9ce7641c]	365	def _layout_button(self):
	366	"""
[d7a39e5]	367	Draw widgets related to button
[9ce7641c]	368	"""
[78a205a]	369	# Close button
[9ce7641c]	370	id = wx.NewId()
	371	button_ok = wx.Button(self, id, "Ok")
	372	button_ok.SetToolTipString("Give Details on Computation")
	373	self.Bind(wx.EVT_BUTTON, self.on_close, id=id)
[78a205a]	374	self.button_sizer.AddMany([((20, 20), 0, wx.LEFT, 350),
	375	(button_ok, 0, wx.RIGHT, 10)])
[9ce7641c]	376	def _do_layout(self):
	377	"""
[d7a39e5]	378	Draw window content
[9ce7641c]	379	"""
	380	self._define_structure()
	381	self._layout_shart()
	382	self._layout_invariant()
	383	self._layout_warning()
	384	self._layout_button()
[d0cc0bbc]	385	self.main_sizer.AddMany([(self.chart_sizer, 0, wx.ALL, 10),
[78a205a]	386	(self.invariant_box_sizer, 0, wx.ALL, 10),
	387	(self.warning_sizer, 0, wx.ALL, 10),
	388	(self.button_sizer, 0, wx.ALL, 10)])
[9ce7641c]	389	self.SetSizer(self.main_sizer)
[78a205a]	390
	391
[9ce7641c]	392	def set_values(self):
	393	"""
[d7a39e5]	394	Set value of txtcrtl
[9ce7641c]	395	"""
[4a2b054]	396	value = format_number(self.qstar_container.qstar)
	397	self.invariant_tcl.SetValue(value)
	398	value = format_number(self.qstar_container.qstar_err)
[78a205a]	399	self.invariant_err_tcl.SetValue(value)
[4a2b054]	400	value = format_number(self.qstar_container.qstar_low)
	401	self.invariant_low_tcl.SetValue(value)
	402	value = format_number(self.qstar_container.qstar_low_err)
[78a205a]	403	self.invariant_low_err_tcl.SetValue(value)
[4a2b054]	404	value = format_number(self.qstar_container.qstar_high)
	405	self.invariant_high_tcl.SetValue(value)
	406	value = format_number(self.qstar_container.qstar_high_err)
[78a205a]	407	self.invariant_high_err_tcl.SetValue(value)
[d0cc0bbc]	408
	409	def get_scale(self, percentage, scale_name='scale'):
[9ce7641c]	410	"""
[78a205a]	411	Check scale receive in this panel.
[9ce7641c]	412	"""
[eed601e]	413	scale = RECTANGLE_SCALE
[78a205a]	414	try:
[eed601e]	415	if percentage in [None, 0.0, "error"]:
[78a205a]	416	scale = RECTANGLE_SCALE
	417	return scale
[f6518f8]	418	elif percentage < 0:
	419	scale = RECTANGLE_SCALE
	420	return scale
	421	scale = float(percentage)
[9ce7641c]	422	except:
[eed601e]	423	scale = RECTANGLE_SCALE
[a3efdeb]	424	self.warning_msg += "Recieve an invalid scale for %s\n"
	425	self.warning_msg += "check this value : %s\n" % str(percentage)
[eed601e]	426	return scale
[78a205a]	427
[9ce7641c]	428	def set_color_bar(self):
	429	"""
[d7a39e5]	430	Change the color for low and high bar when necessary
[9ce7641c]	431	"""
[9cec2dd]	432	self.extrapolation_color_low = EXTRAPOLATION_COLOR
	433	self.extrapolation_color_high = EXTRAPOLATION_COLOR
	434	self.invariant_color = INVARIANT_COLOR
[78a205a]	435	# warning to the user when the extrapolated invariant is greater than %5
[f6518f8]	436	if self.low_scale >= 0.05 or self.low_scale > 1 or self.low_scale < 0:
[78a205a]	437	self.extrapolation_color_low = ERROR_COLOR
[f6518f8]	438	if self.high_scale >= 0.05 or self.high_scale > 1 or self.high_scale < 0:
[78a205a]	439	self.extrapolation_color_high = ERROR_COLOR
[9cec2dd]	440	if self.inv_scale > 1 or self.inv_scale < 0:
[78a205a]	441	self.invariant_color = ERROR_COLOR
	442
[9ce7641c]	443	def on_close(self, event):
	444	"""
[d7a39e5]	445	Close the current window
[9ce7641c]	446	"""
	447	self.Close()
[78a205a]	448
[9ce7641c]	449	def on_paint(self, event):
	450	"""
[d7a39e5]	451	Draw the chart
[9ce7641c]	452	"""
	453	dc = wx.PaintDC(self.panel_chart)
	454	try:
	455	gc = wx.GraphicsContext.Create(dc)
	456	except NotImplementedError:
[4a2b054]	457	msg = "This build of wxPython does not support "
	458	msg += "the wx.GraphicsContext family of classes."
	459	dc.DrawText(msg, 25, 25)
[9ce7641c]	460	return
[78a205a]	461	# Start the drawing
[9ce7641c]	462	font = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT)
	463	font.SetWeight(wx.BOLD)
	464	gc.SetFont(font)
	465	# Draw a rectangle
	466	path = gc.CreatePath()
[78a205a]	467	path.AddRectangle(-RECTANGLE_WIDTH / 2, -RECTANGLE_HEIGHT / 2,
	468	RECTANGLE_WIDTH / 2, RECTANGLE_HEIGHT / 2)
[d0cc0bbc]	469	x_origine = 20
[9ce7641c]	470	y_origine = 15
[78a205a]	471	# Draw low rectangle
	472	gc.PushState()
[f6518f8]	473	label = "Q* from Low-Q "
[9ce7641c]	474	PathFunc = gc.DrawPath
	475	w, h = gc.GetTextExtent(label)
	476	gc.DrawText(label, x_origine, y_origine)
[78a205a]	477	# Translate the rectangle
	478	x_center = x_origine + RECTANGLE_WIDTH * self.low_scale / 2 + w + 10
[9ce7641c]	479	y_center = y_origine + h
[78a205a]	480	gc.Translate(x_center, y_center)
[9ce7641c]	481	gc.SetPen(wx.Pen("black", 1))
	482	gc.SetBrush(wx.Brush(self.extrapolation_color_low))
[eed601e]	483	if self.low_inv_percent is None:
	484	low_percent = 'Not Computed'
	485	elif self.low_inv_percent == 'error':
	486	low_percent = 'Error'
	487	else:
[78a205a]	488	low_percent = format_number(self.low_inv_percent * 100) + '%'
[d0cc0bbc]	489	x_center = 20
	490	y_center = -h
	491	gc.DrawText(low_percent, x_center, y_center)
[9ce7641c]	492	# Increase width by self.low_scale
[78a205a]	493	gc.Scale(self.low_scale, 1.0)
[9ce7641c]	494	PathFunc(path)
[78a205a]	495	gc.PopState()
	496	# Draw rectangle for invariant
	497	gc.PushState() # save it again
	498	y_origine += 20
[f6518f8]	499	gc.DrawText("Q* from Data ", x_origine, y_origine)
[9ce7641c]	500	# offset to the lower part of the window
[78a205a]	501	x_center = x_origine + RECTANGLE_WIDTH * self.inv_scale / 2 + w + 10
[9ce7641c]	502	y_center = y_origine + h
	503	gc.Translate(x_center, y_center)
	504	# 128 == half transparent
[78a205a]	505	gc.SetBrush(wx.Brush(self.invariant_color))
[9ce7641c]	506	# Increase width by self.inv_scale
[eed601e]	507	if self.inv_percent is None:
	508	inv_percent = 'Not Computed'
	509	elif self.inv_percent == 'error':
	510	inv_percent = 'Error'
	511	else:
[78a205a]	512	inv_percent = format_number(self.inv_percent * 100) + '%'
	513	x_center = 20
[d0cc0bbc]	514	y_center = -h
	515	gc.DrawText(inv_percent, x_center, y_center)
[78a205a]	516	gc.Scale(self.inv_scale, 1.0)
[9ce7641c]	517	gc.DrawPath(path)
	518	gc.PopState()
	519	# restore saved state
[78a205a]	520	# Draw rectangle for high invariant
	521	gc.PushState()
	522	y_origine += 20
	523	gc.DrawText("Q* from High-Q ", x_origine, y_origine)
	524	# define the position of the new rectangle
	525	x_center = x_origine + RECTANGLE_WIDTH * self.high_scale / 2 + w + 10
[9ce7641c]	526	y_center = y_origine + h
	527	gc.Translate(x_center, y_center)
[78a205a]	528	gc.SetBrush(wx.Brush(self.extrapolation_color_high))
[9ce7641c]	529	# increase scale by self.high_scale
[eed601e]	530	if self.high_inv_percent is None:
	531	high_percent = 'Not Computed'
	532	elif self.high_inv_percent == 'error':
	533	high_percent = 'Error'
	534	else:
[78a205a]	535	high_percent = format_number(self.high_inv_percent * 100) + '%'
[d0cc0bbc]	536	x_center = 20
	537	y_center = -h
	538	gc.DrawText(high_percent, x_center, y_center)
[78a205a]	539
	540	gc.Scale(self.high_scale, 1.0)
[9ce7641c]	541	gc.DrawPath(path)
	542	gc.PopState()

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

SasView

source: sasview/src/sas/perspectives/invariant/invariant_details.py @ dca6188

Download in other formats: