ResolutionCalculatorPanel.py @ a2e8ea5

ESS_GUIESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc

Last change on this file since a2e8ea5 was 144fe21, checked in by Piotr Rozyczko <rozyczko@…>, 6 years ago
new unit test runner script + test fixes. SASVIEW-970
Property mode set to `100644`
File size: 30.6 KB

Rev	Line
[01cda57]	1	"""
	2	This object is a small tool to allow user to quickly
	3	determine the variance in q from the
	4	instrumental parameters.
	5	"""
[4992ff2]	6	from PyQt5 import QtCore
	7	from PyQt5 import QtGui
	8	from PyQt5 import QtWidgets
[01cda57]	9
	10	from twisted.internet import threads
	11	import sas.qtgui.Utilities.GuiUtils as GuiUtils
	12	from sas.qtgui.Plotting.PlotterData import Data2D
	13	from sas.qtgui.Plotting.Plotter2D import Plotter2DWidget
	14	from sas.sascalc.calculator.resolution_calculator import ResolutionCalculator
	15	import matplotlib.patches as patches
	16
	17	import numpy
	18	import sys
	19	import logging
	20	import os
	21	import re
	22
[b3e8629]	23	from .UI.ResolutionCalculatorPanelUI import Ui_ResolutionCalculatorPanel
[01cda57]	24
	25	_SOURCE_MASS = {'Alpha': 6.64465620E-24,
	26	'Deuteron': 3.34358320E-24,
	27	'Neutron': 1.67492729E-24,
	28	'Photon': 0.0,
	29	'Proton': 1.67262137E-24,
	30	'Triton': 5.00826667E-24}
	31
	32	BG_WHITE = "background-color: rgb(255, 255, 255);"
	33	BG_RED = "background-color: rgb(244, 170, 164);"
	34
	35
[4992ff2]	36	class ResolutionCalculatorPanel(QtWidgets.QDialog, Ui_ResolutionCalculatorPanel):
[01cda57]	37	"""
	38	compute resolution in 2D
	39	"""
	40	def __init__(self, parent=None):
	41	super(ResolutionCalculatorPanel, self).__init__()
	42	self.setupUi(self)
	43	self.manager = parent
	44
	45	# New font to display angstrom symbol
	46	new_font = 'font-family: -apple-system, "Helvetica Neue", "Ubuntu";'
	47	self.lblUnitWavelength.setStyleSheet(new_font)
	48	self.lblUnitQx.setStyleSheet(new_font)
	49	self.lblUnitQy.setStyleSheet(new_font)
	50	self.lblUnitSigmax.setStyleSheet(new_font)
	51	self.lblUnitSigmay.setStyleSheet(new_font)
	52	self.lblUnitSigmalamd.setStyleSheet(new_font)
	53	self.lblUnit1DSigma.setStyleSheet(new_font)
	54
	55	# by default Spectrum label and cbCustomSpectrum are not visible
	56	self.cbCustomSpectrum.setVisible(False)
	57	self.lblSpectrum.setVisible(False)
	58	# self.onReset()
	59
	60	# change index of comboboxes
	61	self.cbWaveColor.currentIndexChanged.connect(self.onSelectWaveColor)
	62	self.cbCustomSpectrum.currentIndexChanged.connect(self.onSelectCustomSpectrum)
	63
	64	# push buttons
	65	self.cmdClose.clicked.connect(self.accept)
	66	self.cmdHelp.clicked.connect(self.onHelp)
	67	self.cmdCompute.clicked.connect(self.onCompute)
	68	self.cmdReset.clicked.connect(self.onReset)
	69
	70	# input defaults
	71	self.qx = []
	72	self.qy = []
	73	# dQ defaults
	74	self.sigma_r = None
	75	self.sigma_phi = None
	76	self.sigma_1d = None
	77
	78	# number of bins for wavelength and wavelength spread
	79	self.num_wave = 10
	80	self.spectrum_dic = {}
	81
	82	# dQ 2d image
	83	self.image = None
	84	# Source selection dic
	85	self.source_mass = _SOURCE_MASS
	86	# detector coordinate of estimation of sigmas
	87	self.det_coordinate = 'cartesian'
	88
	89	self.resolution = ResolutionCalculator()
	90	self.spectrum_dic['Add new'] = ''
	91	self.spectrum_dic['Flat'] = self.resolution.get_default_spectrum()
	92	self.resolution.set_spectrum(self.spectrum_dic['Flat'])
	93
	94	# validators
	95	self.txtWavelength.editingFinished.connect(self.checkWavelength)
	96	self.txtWavelengthSpread.editingFinished.connect(self.checkWavelengthSpread)
	97
	98	self.txtDetectorPixSize.editingFinished.connect(self.checkPixels)
	99	self.txtDetectorSize.editingFinished.connect(self.checkPixels)
	100
	101	self.txtSourceApertureSize.editingFinished.connect(self.checkAperture)
	102	self.txtSampleApertureSize.editingFinished.connect(self.checkAperture)
	103
	104	self.txtQx.editingFinished.connect(self.checkQx_y)
	105	self.txtQy.editingFinished.connect(self.checkQx_y)
	106
	107	# double validator
[d6b8a1d]	108	self.txtSource2SampleDistance.setValidator(GuiUtils.DoubleValidator())
	109	self.txtSample2DetectorDistance.setValidator(GuiUtils.DoubleValidator())
	110	self.txtSampleOffset.setValidator(GuiUtils.DoubleValidator())
[01cda57]	111
	112	# call compute to calculate with default values
	113	self.createTemplate2DPlot()
[7fb471d]	114	#self.onCompute()
[01cda57]	115
	116	# #################################
	117	# Validators: red background in line edits when wrong input
	118	# and display of info logging message
	119	# #################################
	120
	121	def checkWavelength(self):
	122	""" Validator for Wavelength
	123	if TOF, wavelength = min - max else only one number """
	124	text_edit = self.txtWavelength # self.sender()
	125	if text_edit.isModified():
[7fb471d]	126	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	127	input_string = str(text_edit.text())
	128	if self.cbWaveColor.currentText() != 'TOF':
	129	input_wavelength = re.match('\d+\.?\d*', input_string)
	130	if input_wavelength is None:
[7fb471d]	131	text_edit.setStyleSheet(BG_RED)
[01cda57]	132	self.cmdCompute.setEnabled(False)
	133	logging.info('Wavelength has to be a number.')
	134	else:
[7fb471d]	135	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	136	self.cmdCompute.setEnabled(True)
	137	else:
	138	interval_wavelength = re.match('^\d+\.?\d\s-\s\d+\.?\d$',
	139	input_string)
	140
	141	if interval_wavelength is None:
[7fb471d]	142	text_edit.setStyleSheet(BG_RED)
[01cda57]	143	self.cmdCompute.setEnabled(False)
	144	logging.info("Wavelength's input has to be an interval: "
	145	"min - max.")
	146	else:
	147	# check on min < max
	148	[wavelength_min, wavelength_max] = \
	149	re.findall('\d+\.?\d*', interval_wavelength.group())
	150
	151	if float(wavelength_min) >= float(wavelength_max):
[7fb471d]	152	text_edit.setStyleSheet(BG_RED)
[01cda57]	153	self.cmdCompute.setEnabled(False)
	154	logging.info("Wavelength: min must be smaller than max.")
	155
	156	else:
[7fb471d]	157	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	158	self.cmdCompute.setEnabled(True)
	159
	160	def checkWavelengthSpread(self):
	161	""" Validator for WavelengthSpread
	162	Input can be a 'number or min - max (; Number of bins)' """
	163	text_edit = self.sender()
	164
	165	if text_edit.isModified():
[7fb471d]	166	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	167	if self.cbWaveColor.currentText() != 'TOF':
	168	pattern = '^\d+\.?\d(\|;\s\d+)$'
	169	input_string = str(text_edit.text())
	170	wavelength_spread_input = re.match(pattern, input_string)
	171
	172	if wavelength_spread_input is None:
[7fb471d]	173	text_edit.setStyleSheet(BG_RED)
[01cda57]	174	self.cmdCompute.setEnabled(False)
	175	logging.info('Wavelength spread has to be specified: '
	176	'single value or value; integer number of bins.')
	177
	178	else:
	179	split_input = wavelength_spread_input.group().split(';')
	180	self.num_wave = split_input[1] if len(split_input) > 1 else 10
[7fb471d]	181	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	182	self.cmdCompute.setEnabled(True)
	183	else:
	184	pattern = '^\d+\.?\d\s-\s\d+\.?\d(\|;\s*\d+)$'
	185	input_string = str(text_edit.text())
	186	wavelength_spread_input = re.match(pattern, input_string)
	187
	188	if wavelength_spread_input is None:
[7fb471d]	189	text_edit.setStyleSheet(BG_RED)
[01cda57]	190	self.cmdCompute.setEnabled(False)
	191	logging.info("Wavelength spread has to be specified: "
	192	"doublet separated by '-' with optional "
	193	"number of bins (given after ';'). "
	194	"For example, 0.1 - 0.1 (; 20).")
	195
	196	else:
	197	split_input = wavelength_spread_input.group().split(';')
	198	self.num_wave = split_input[1] if len(
	199	split_input) > 1 else 10
[7fb471d]	200	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	201	self.cmdCompute.setEnabled(True)
	202
	203	def checkPixels(self):
	204	""" Validator for detector pixel size and number """
	205	text_edit = self.sender()
	206
	207	if text_edit.isModified():
[7fb471d]	208	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	209	pattern = '^\d+\.?\d,\s\d+\.?\d*$'
	210	input_string = str(text_edit.text())
	211	pixels_input = re.match(pattern, input_string)
	212
	213	if pixels_input is None:
[7fb471d]	214	text_edit.setStyleSheet(BG_RED)
[01cda57]	215	self.cmdCompute.setEnabled(False)
	216	logging.info('The input for the detector should contain 2 '
	217	'values separated by a comma.')
	218
	219	else:
[7fb471d]	220	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	221	self.cmdCompute.setEnabled(True)
	222
	223	def checkQx_y(self):
	224	""" Validator for qx and qy inputs """
	225	Q_modified = [self.txtQx.isModified(), self.txtQy.isModified()]
	226	if any(Q_modified):
	227	pattern = '^-?\d+\.?\d(,\s-?\d+\.?\d)$'
	228	text_edit = self.txtQx if Q_modified[0] else self.txtQy
	229	input_string = str(text_edit.text())
	230	q_input = re.match(pattern, input_string)
	231	if q_input is None:
[7fb471d]	232	text_edit.setStyleSheet(BG_RED)
[01cda57]	233	self.cmdCompute.setEnabled(False)
[fc4fec8]	234	logging.info('Qx and Qy should contain one or more comma-separated numbers.')
[01cda57]	235	else:
[7fb471d]	236	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	237	self.cmdCompute.setEnabled(True)
	238	qx = str(self.txtQx.text()).split(',')
	239	qy = str(self.txtQy.text()).split(',')
	240
	241	if len(qx) == 1 and len(qy) > 1:
	242	fill_qx = ', '.join([qx[0]] * len(qy))
	243	self.txtQx.setText(fill_qx)
	244
	245	elif len(qy) == 1 and len(qx) > 1:
	246	fill_qy = ', '.join([qy[0]] * len(qx))
	247	self.txtQy.setText(fill_qy)
	248
	249	elif len(qx) != len(qy):
[7fb471d]	250	text_edit.setStyleSheet(BG_RED)
[01cda57]	251	self.cmdCompute.setEnabled(False)
	252	logging.info(
[fc4fec8]	253	'Qx and Qy should have the same number of elements.')
[01cda57]	254
	255	else:
[7fb471d]	256	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	257	self.cmdCompute.setEnabled(True)
	258
	259	def checkAperture(self):
	260	""" Validator for Sample and Source apertures"""
	261	text_edit = self.sender()
	262
	263	if text_edit.isModified():
[7fb471d]	264	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	265	input_string = str(text_edit.text())
	266	pattern = '^\d+\.?\d(\|,\s\d+)$'
	267	aperture_input = re.match(pattern, input_string)
	268
	269	if aperture_input is None:
[7fb471d]	270	text_edit.setStyleSheet(BG_RED)
[01cda57]	271	self.cmdCompute.setEnabled(False)
	272	logging.info('A circular aperture is defined by a single '
	273	'value (diameter). A rectangular aperture is '
	274	'defined by 2 values separated by a comma.')
	275
	276	else:
[7fb471d]	277	text_edit.setStyleSheet(BG_WHITE)
[01cda57]	278	self.cmdCompute.setEnabled(True)
	279
	280	# #################################
	281	# Slots associated with signals from comboboxes
	282	# #################################
	283
	284	def onSelectWaveColor(self):
	285	""" Modify layout of GUI when TOF selected: add elements
	286	and modify default entry of Wavelength """
	287	list_wdata = self.resolution.get_wave_list()
	288	min_lambda = min(list_wdata[0])
	289
	290	min_wspread = min(list_wdata[1])
	291	max_wspread = max(list_wdata[1])
	292
	293	if self.cbWaveColor.currentText() == 'TOF':
	294	self.cbCustomSpectrum.setVisible(True)
	295	self.lblSpectrum.setVisible(True)
	296	# Get information about wavelength and spread
	297
	298	if len(list_wdata[0]) < 2:
	299	max_lambda = 2 * min_lambda
	300	else:
	301	max_lambda = max(list_wdata[0])
	302	self.txtWavelength.setText('{} - {}'.format(min_lambda, max_lambda))
	303	self.txtWavelengthSpread.setText('{} - {}'.format(min_wspread,
	304	max_wspread))
	305
	306	else:
	307	self.cbCustomSpectrum.setVisible(False)
	308	self.lblSpectrum.setVisible(False)
	309	# modify Wavelength line edit only if set for TOF (2 elements)
	310
	311	if len(self.txtWavelength.text().split('-')) >= 2:
	312	self.txtWavelength.setText(str(min_lambda))
	313	self.txtWavelengthSpread.setText(str(min_wspread))
	314
	315	def onSelectCustomSpectrum(self):
	316	""" On Spectrum Combobox event"""
	317	if self.cbCustomSpectrum.currentText() == 'Add New':
[4992ff2]	318	datafile = QtWidgets.QFileDialog.getOpenFileName(
[fbfc488]	319	self, "Choose a spectral distribution file","",
	320	"All files (.)", None,
	321	QtWidgets.QFileDialog.DontUseNativeDialog)[0]
[01cda57]	322
	323	if datafile is None or str(datafile) == '':
	324	logging.info("No spectral distribution data chosen.")
	325	self.cbCustomSpectrum.setCurrentIndex(0)
	326	self.resolution.set_spectrum(self.spectrum_dic['Flat'])
	327	return
	328
[fa05c6c1]	329	basename = os.path.basename(datafile)
	330
	331	input_f = open(datafile, 'r')
	332	buff = input_f.read()
	333	lines = buff.split('\n')
	334
	335	wavelength = []
	336	intensity = []
	337
	338	for line in lines:
	339	toks = line.split()
	340	try:
	341	wave = float(toks[0])
	342	intens = float(toks[1])
	343	wavelength.append(wave)
	344	intensity.append(intens)
	345	except:
	346	logging.info('Could not extract values from file')
	347	if wavelength and intensity:
[b3e8629]	348	if basename not in list(self.spectrum_dic.keys()):
[01cda57]	349	self.cbCustomSpectrum.addItem(basename)
[fa05c6c1]	350	self.spectrum_dic[basename] = [wavelength, intensity]
	351	self.resolution.set_spectrum(self.spectrum_dic[basename])
[01cda57]	352	return
	353
	354	# #################################
	355	# Slots associated with signals from push buttons
	356	# #################################
	357
	358	def onHelp(self):
	359	"""
	360	Bring up the Resolution Calculator Documentation whenever
	361	the HELP button is clicked.
	362	Calls Documentation Window with the path of the location within the
	363	documentation tree (after /doc/ ....".
	364	"""
[aed0532]	365	location = "/user/qtgui/Calculators/resolution_calculator_help.html"
[e90988c]	366	self.manager.showHelp(location)
[01cda57]	367
	368	def onReset(self):
	369	# by default Spectrum label and cbCustomSpectrum are not visible
	370	self.cbCustomSpectrum.setVisible(False)
	371	self.lblSpectrum.setVisible(False)
	372	# Comboboxes
	373	self.cbCustomSpectrum.setCurrentIndex([self.cbCustomSpectrum.itemText(i)
	374	for i in range(self.cbCustomSpectrum.count())].index('Flat'))
	375	self.cbSource.setCurrentIndex([self.cbSource.itemText(i) for i in
	376	range(self.cbSource.count())].index('Neutron'))
	377	self.cbWaveColor.setCurrentIndex([self.cbWaveColor.itemText(i) for i
	378	in range(self.cbWaveColor.count())].index('Monochromatic'))
	379	# LineEdits
	380	self.txtDetectorPixSize.setText('0.5, 0.5')
	381	self.txtDetectorSize.setText('128, 128')
	382	self.txtSample2DetectorDistance.setText('1000')
	383	self.txtSampleApertureSize.setText('1.27')
	384	self.txtSampleOffset.setText('0')
	385	self.txtSource2SampleDistance.setText('1627')
	386	self.txtSourceApertureSize.setText('3.81')
	387	self.txtWavelength.setText('6.0')
	388	self.txtWavelengthSpread.setText('0.125')
	389	self.txtQx.setText('0.0')
	390	self.txtQy.setText('0.0')
	391	self.txt1DSigma.setText('0.0008289')
	392	self.txtSigma_x.setText('0.0008288')
	393	self.txtSigma_y.setText('0.0008288')
	394	self.txtSigma_lamd.setText('3.168e-05')
	395
	396	self.image = None
	397	self.source_mass = _SOURCE_MASS
	398	self.det_coordinate = 'cartesian'
	399	self.num_wave = 10
	400	self.spectrum_dic = {}
	401	self.spectrum_dic['Add new'] = ''
	402	self.spectrum_dic['Flat'] = self.resolution.get_default_spectrum()
	403	self.resolution.set_spectrum(self.spectrum_dic['Flat'])
	404	# Reset plot
	405	self.onCompute()
	406
	407	# TODO Keep legacy validators??
	408	def onCompute(self):
	409	"""
	410	Execute the computation of resolution
	411	"""
	412	# Q min max list default
	413	qx_min = []
	414	qx_max = []
	415	qy_min = []
	416	qy_max = []
	417	# possible max qrange
	418	self.resolution.qxmin_limit = 0
	419	self.resolution.qxmax_limit = 0
	420	self.resolution.qymin_limit = 0
	421	self.resolution.qymax_limit = 0
	422
	423	try:
	424	# Get all the values to compute
	425	wavelength = self._str2longlist(self.txtWavelength.text())
	426
	427	source = self.cbSource.currentText()
	428	mass = self.source_mass[str(source)]
	429	self.resolution.set_neutron_mass(float(mass))
	430
	431	wavelength_spread = self._str2longlist(\
	432	self.txtWavelengthSpread.text().split(';')[0])
	433	# Validate the wave inputs
	434	wave_input = self._validate_q_input(wavelength, wavelength_spread)
	435	if wave_input is not None:
	436	wavelength, wavelength_spread = wave_input
	437
	438	self.resolution.set_wave(wavelength)
	439	self.resolution.set_wave_spread(wavelength_spread)
	440
	441	# use legacy validator for correct input assignment
	442
	443	source_aperture_size = self.txtSourceApertureSize.text()
	444	source_aperture_size = self._str2longlist(source_aperture_size)
	445	self.resolution.set_source_aperture_size(source_aperture_size)
	446
	447	sample_aperture_size = self.txtSampleApertureSize.text()
	448	sample_aperture_size = self._string2list(sample_aperture_size)
	449	self.resolution.set_sample_aperture_size(sample_aperture_size)
	450
	451	source2sample_distance = self.txtSource2SampleDistance.text()
	452	source2sample_distance = self._string2list(source2sample_distance)
	453	self.resolution.set_source2sample_distance(source2sample_distance)
	454
	455	sample2sample_distance = self.txtSampleOffset.text()
	456	sample2sample_distance = self._string2list(sample2sample_distance)
	457	self.resolution.set_sample2sample_distance(sample2sample_distance)
	458
	459	sample2detector_distance = self.txtSample2DetectorDistance.text()
	460	sample2detector_distance = self._string2list(
	461	sample2detector_distance)
	462	self.resolution.set_sample2detector_distance(
	463	sample2detector_distance)
	464
	465	detector_size = self.txtDetectorSize.text()
	466	detector_size = self._string2list(detector_size)
	467	self.resolution.set_detector_size(detector_size)
	468
	469	detector_pix_size = self.txtDetectorPixSize.text()
	470	detector_pix_size = self._string2list(detector_pix_size)
	471	self.resolution.set_detector_pix_size(detector_pix_size)
	472
	473	self.qx = self._string2inputlist(self.txtQx.text())
	474	self.qy = self._string2inputlist(self.txtQy.text())
	475
	476	# Find min max of qs
	477	xmin = min(self.qx)
	478	xmax = max(self.qx)
	479	ymin = min(self.qy)
	480	ymax = max(self.qy)
	481	if not self._validate_q_input(self.qx, self.qy):
[fc4fec8]	482	raise ValueError("Invalid Q input")
[01cda57]	483	except:
	484	msg = "An error occurred during the resolution computation."
	485	msg += "Please check your inputs..."
	486	logging.warning(msg)
	487	return
	488
	489	# Validate the q inputs
	490	q_input = self._validate_q_input(self.qx, self.qy)
	491	if q_input is not None:
	492	self.qx, self.qy = q_input
	493
	494	# Make list of q min max for mapping
	495	for i in range(len(self.qx)):
	496	qx_min.append(xmin)
	497	qx_max.append(xmax)
	498	for i in range(len(self.qy)):
	499	qy_min.append(ymin)
	500	qy_max.append(ymax)
	501
	502	# Compute the resolution
	503	if self.image is not None:
	504	self.resolution.reset_image()
	505
	506	# Compute and get the image plot
	507	try:
[fc4fec8]	508	cal_res = threads.deferToThread(self.map_wrapper,
	509	self.calc_func,
	510	self.qx,
	511	self.qy,
	512	qx_min,
	513	qx_max,
	514	qy_min, qy_max)
	515
	516	cal_res.addCallback(self.complete)
[7fb471d]	517	cal_res.addErrback(self.calculateFailed)
[fc4fec8]	518
	519	# logging.info("Computation is in progress...")
[01cda57]	520	self.cmdCompute.setText('Wait...')
	521	self.cmdCompute.setEnabled(False)
	522	except:
	523	raise
	524
[7fb471d]	525	def calculateFailed(self, reason):
	526	print("calculateFailed Failed with:\n", reason)
	527	pass
	528
[fc4fec8]	529	def complete(self, image):
[01cda57]	530	"""
	531	Complete computation
	532	"""
	533	self.image = image
	534
	535	# Get and format the sigmas
	536	sigma_r = self.formatNumber(self.resolution.sigma_1)
	537	sigma_phi = self.formatNumber(self.resolution.sigma_2)
	538	sigma_lamd = self.formatNumber(self.resolution.sigma_lamd)
	539	sigma_1d = self.formatNumber(self.resolution.sigma_1d)
	540
	541	# Set output values
	542	self.txtSigma_x.setText(str(sigma_r))
	543	self.txtSigma_y.setText(str(sigma_phi))
	544	self.txtSigma_lamd.setText(str(sigma_lamd))
	545	self.txt1DSigma.setText(str(sigma_1d))
	546
	547	self.cmdCompute.setText('Compute')
	548	self.cmdCompute.setEnabled(True)
[fc4fec8]	549
	550	self.new2DPlot()
	551
[01cda57]	552	return
	553
[fc4fec8]	554	def map_wrapper(self, func, qx, qy, qx_min, qx_max, qy_min, qy_max):
[01cda57]	555	"""
	556	Prepare the Mapping for the computation
	557	: params qx, qy, qx_min, qx_max, qy_min, qy_max:
	558	: return: image (numpy array)
	559	"""
[fbfc488]	560	image = list(map(func, qx, qy,
[fc4fec8]	561	qx_min, qx_max,
[fbfc488]	562	qy_min, qy_max))[0]
	563
[fc4fec8]	564	return image
	565
	566	def calc_func(self, qx, qy, qx_min, qx_max, qy_min, qy_max):
	567	"""
	568	Perform the calculation for a given set of Q values.
	569	: return: image (numpy array)
	570	"""
[01cda57]	571	try:
	572	qx_value = float(qx)
	573	qy_value = float(qy)
[fc4fec8]	574	except :
	575	raise ValueError
	576
[01cda57]	577	# calculate 2D resolution distribution image
	578	image = self.resolution.compute_and_plot(qx_value, qy_value,
	579	qx_min, qx_max, qy_min,
	580	qy_max,
	581	self.det_coordinate)
	582	return image
	583
	584	# #################################
	585	# Legacy validators
	586	# #################################
[fc4fec8]	587	def _string2list(self, input_string):
[01cda57]	588	"""
	589	Change NNN, NNN to list,ie. [NNN, NNN] where NNN is a number
	590	"""
[fc4fec8]	591	new_numbers_list = []
[01cda57]	592	# check the number of floats
	593	try:
[fc4fec8]	594	strg = float(input_string)
	595	new_numbers_list.append(strg)
[01cda57]	596	except:
[fc4fec8]	597	string_split = input_string.split(',')
[170e95d]	598	if len(string_split) == 1 or len(string_split) == 2:
	599	new_numbers_list = [float(item) for item in string_split]
[01cda57]	600	else:
[fc4fec8]	601	msg = "The numbers must be one or two (separated by ',')"
[01cda57]	602	logging.info(msg)
[b3e8629]	603	raise RuntimeError(msg)
[01cda57]	604
[fc4fec8]	605	return new_numbers_list
[01cda57]	606
[fc4fec8]	607	def _string2inputlist(self, input_string):
[01cda57]	608	"""
	609	Change NNN, NNN,... to list,ie. [NNN, NNN,...] where NNN is a number
[fc4fec8]	610	: return new_list: string like list
[01cda57]	611	"""
[fc4fec8]	612	new_list = []
	613	string_split = input_string.split(',')
	614	try:
	615	new_list = [float(t) for t in string_split]
	616	except:
[b3e8629]	617	logging.error(sys.exc_info()[1])
[fc4fec8]	618	return new_list
[01cda57]	619
[fc4fec8]	620	def _str2longlist(self, input_string):
[01cda57]	621	"""
[fc4fec8]	622	Change NNN, NNN,... to list, NNN - NNN ; NNN to list, or float to list
	623	: return new_string: string like list
	624	"""
[01cda57]	625	try:
	626	# is float
[fc4fec8]	627	out = [float(input_string)]
[01cda57]	628	return out
	629	except:
	630	if self.cbWaveColor.currentText() == 'Monochromatic':
	631	logging.warning("Wrong format of inputs.")
	632	else:
	633	try:
	634	# has a '-'
[fc4fec8]	635	if input_string.count('-') > 0:
	636	value = input_string.split('-')
[01cda57]	637	if value[1].count(';') > 0:
	638	# has a ';'
	639	last_list = value[1].split(';')
[fc4fec8]	640	num = numpy.ceil(float(last_list[1]))
[01cda57]	641	max_value = float(last_list[0])
	642	self.num_wave = num
	643	else:
	644	# default num
	645	num = self.num_wave
	646	max_value = float(value[1])
	647	min_value = float(value[0])
	648	# make a list
[fc4fec8]	649	bin_size = numpy.fabs(max_value - min_value) / (num - 1)
[01cda57]	650	out = [min_value + bin_size * bnum for bnum in
	651	range(num)]
	652	return out
[fc4fec8]	653	if input_string.count(',') > 0:
	654	out = self._string2inputlist(input_string)
[01cda57]	655	return out
	656	except:
[b3e8629]	657	logging.error(sys.exc_info()[1])
[01cda57]	658
	659	def _validate_q_input(self, qx, qy):
	660	"""
	661	Check if q inputs are valid
	662	: params qx: qx as a list
	663	: params qy: qy as a list
	664	: return: True/False
	665	"""
	666	# check qualifications
	667	if qx.__class__.__name__ != 'list':
	668	return None
	669	if qy.__class__.__name__ != 'list':
	670	return None
	671	if len(qx) < 1:
	672	return None
	673	if len(qy) < 1:
	674	return None
	675	# allow one input
	676	if len(qx) == 1 and len(qy) > 1:
	677	qx = [qx[0] for ind in range(len(qy))]
	678
	679	if len(qy) == 1 and len(qx) > 1:
	680	qy = [qy[0] for ind in range(len(qx))]
	681	# check length
	682	if len(qx) != len(qy):
	683	return None
	684	if qx is None or qy is None:
	685	return None
	686	return qx, qy
	687
	688	def formatNumber(self, value=None):
	689	"""
	690	Return a float in a standardized, human-readable formatted string
	691	"""
	692	try:
	693	value = float(value)
	694	except:
	695	output = None
	696	return output
	697
	698	output = "%-7.4g" % value
	699	return output.lstrip().rstrip()
	700
	701	# #################################
	702	# Plot
	703	# #################################
	704
	705	def createTemplate2DPlot(self):
	706	"""
	707	Create a template for 2D data
	708	"""
[144fe21]	709	self.plotter = Plotter2DWidget(self, manager=self.manager, quickplot=True)
[01cda57]	710	self.plotter.scale = 'linear'
[fc4fec8]	711	self.plotter.cmap = None
[4992ff2]	712	layout = QtWidgets.QHBoxLayout()
[01cda57]	713	layout.setContentsMargins(0, 0, 0, 0)
	714	self.graphicsView.setLayout(layout)
	715	layout.addWidget(self.plotter)
	716
[fc4fec8]	717	def new2DPlot(self):
[01cda57]	718	"""
	719	Create a new 2D data instance based on computing results
	720	"""
	721	qx_min, qx_max, qy_min, qy_max = self.resolution.get_detector_qrange()
	722
	723	dx_size = (qx_max - qx_min) / (1000 - 1)
	724	dy_size = (qy_max - qy_min) / (1000 - 1)
	725	x_val = numpy.arange(qx_min, qx_max, dx_size)
	726	y_val = numpy.arange(qy_max, qy_min, -dy_size)
	727
[170e95d]	728	if len(self.plotter.ax.patches):
	729	self.plotter.ax.patches[0].remove()
	730
[01cda57]	731	self.drawLines()
	732
	733	self.plotter.data = Data2D(image=self.image,
	734	qx_data=x_val,
	735	qy_data=y_val,
	736	xmin=qx_min, xmax=qx_max,
	737	ymin=qy_min, ymax=qy_max)
	738
	739	self.plotter.plot()
	740	self.plotter.show()
[fbfc488]	741	self.plotter.update()
[01cda57]	742
	743	def drawLines(self):
	744	"""
	745	Draw lines in image if applicable
	746	"""
	747	wave_list, _ = self.resolution.get_wave_list()
	748	if len(wave_list) > 1 and wave_list[-1] == max(wave_list):
	749	color = 'g'
	750	# draw a green rectangle(limit for the longest wavelength
	751	# to be involved) for tof inputs
	752	# Get the params from resolution
	753	# plotting range for largest wavelength
	754	qx_min = self.resolution.qx_min
	755	qx_max = self.resolution.qx_max
	756	qy_min = self.resolution.qy_min
	757	qy_max = self.resolution.qy_max
	758	# detector range
	759	detector_qx_min = self.resolution.detector_qx_min
	760	detector_qx_max = self.resolution.detector_qx_max
	761	detector_qy_min = self.resolution.detector_qy_min
	762	detector_qy_max = self.resolution.detector_qy_max
	763
	764	rect = patches.Rectangle((detector_qx_min + 0.0002,
	765	detector_qy_min + 0.0002),
	766	detector_qx_max - detector_qx_min,
	767	detector_qy_max - detector_qy_min,
	768	linewidth=2,
	769	edgecolor=color, facecolor='none')
	770	self.plotter.ax.add_patch(rect)
	771	else:
	772	qx_min, qx_max, qy_min, qy_max = self.resolution.get_detector_qrange()
	773	# detector range
	774	detector_qx_min = self.resolution.qxmin_limit
	775	detector_qx_max = self.resolution.qxmax_limit
	776	detector_qy_min = self.resolution.qymin_limit
	777	detector_qy_max = self.resolution.qymax_limit
	778
	779	xmin = min(self.qx)
	780	xmax = max(self.qx)
	781	ymin = min(self.qy)
	782	ymax = max(self.qy)
	783
	784	if xmin < detector_qx_min or xmax > detector_qx_max or \
	785	ymin < detector_qy_min or ymax > detector_qy_max:
	786	# message
	787	msg = 'At least one q value located out side of\n'
	788	msg += " the detector range (%s < qx < %s, %s < qy < %s),\n" % \
	789	(self.formatNumber(detector_qx_min),
	790	self.formatNumber(detector_qx_max),
	791	self.formatNumber(detector_qy_min),
	792	self.formatNumber(detector_qy_max))
	793	msg += " is ignored in computation.\n"
	794
	795	logging.warning(msg)
	796
	797	# Draw zero axis lines.
	798	if qy_min < 0 <= qy_max:
	799	self.plotter.ax.axhline(linewidth=1)
	800
	801	if qx_min < 0 <= qx_max:
	802	self.plotter.ax.axvline(linewidth=1)

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

SasView

source: sasview/src/sas/qtgui/Calculators/ResolutionCalculatorPanel.py @ a2e8ea5

Download in other formats: