Source code for sas.qtgui.Perspectives.Fitting.SmearingWidget
"""
Widget/logic for smearing data.
"""
import copy
import numpy as np
from PyQt5 import QtCore
from PyQt5 import QtGui
from PyQt5 import QtWidgets
from sas.sascalc.fit.qsmearing import smear_selection, PySmear, PySmear2D
from sas.qtgui.Plotting.PlotterData import Data1D
from sas.qtgui.Plotting.PlotterData import Data2D
import sas.qtgui.Utilities.GuiUtils as GuiUtils
from sasmodels.resolution import Slit1D, Pinhole1D
from sasmodels.sesans import SesansTransform
# Local UI
from sas.qtgui.Perspectives.Fitting.UI.SmearingWidgetUI import Ui_SmearingWidgetUI
[docs]class DataWidgetMapper(QtWidgets.QDataWidgetMapper):
"""
Custom version of the standard QDataWidgetMapper allowing for proper
response to index change in comboboxes
"""
[docs] def addMapping(self, widget, section, propertyName=None):
if propertyName is None:
super(DataWidgetMapper, self).addMapping(widget, section)
else:
super(DataWidgetMapper, self).addMapping(widget, section, propertyName)
if isinstance(widget, QtWidgets.QComboBox):
delegate = self.itemDelegate()
widget.currentIndexChanged.connect(lambda: delegate.commitData.emit(widget))
SMEARING_1D = ["Custom Pinhole Smear", "Custom Slit Smear"]
SMEARING_2D = ["Custom Pinhole Smear"]
SMEARING_QD = "Use dQ Data"
MODEL = [
'SMEARING',
'PINHOLE_MIN',
'PINHOLE_MAX',
'ACCURACY']
ACCURACY_DICT={'Low': 'low',
'Medium': 'med',
'High': 'high',
'Extra high': 'xhigh'}
DEFAULT_PINHOLE_UP=0.0
DEFAULT_PINHOLE_DOWN=0.0
[docs]class SmearingWidget(QtWidgets.QWidget, Ui_SmearingWidgetUI):
smearingChangedSignal = QtCore.pyqtSignal()
[docs] def __init__(self, parent=None):
super(SmearingWidget, self).__init__()
self.setupUi(self)
# Local model for holding data
self.model = None
# Mapper for model update
self.mapper = None
# Data from the widget
self.data = None
self.current_smearer = None
self.kernel_model = None
# dQ data variables
self.smear_type = None
self.dq_l = None
self.dq_r = None
# current pinhole/slot values
self.pinhole = 0.0
self.slit_height = 0.0
self.slit_width = 0.0
# current accuracy option
self.accuracy = ""
# Let only floats in the line edits
self.txtSmearDown.setValidator(GuiUtils.DoubleValidator())
self.txtSmearUp.setValidator(GuiUtils.DoubleValidator())
# Attach slots
self.cbSmearing.currentIndexChanged.connect(self.onIndexChange)
self.cbSmearing.setCurrentIndex(0)
self.txtSmearUp.setText(str(DEFAULT_PINHOLE_UP))
self.txtSmearDown.setText(str(DEFAULT_PINHOLE_DOWN))
self.initModel()
self.initMapper()
[docs] def initModel(self):
"""
Initialize the state
"""
self.model = QtGui.QStandardItemModel()
for model_item in range(len(MODEL)):
self.model.setItem(model_item, QtGui.QStandardItem())
# Attach slot
self.model.dataChanged.connect(self.onModelChange)
[docs] def initMapper(self):
"""
Initialize model item <-> UI element mapping
"""
self.mapper = DataWidgetMapper(self)
self.mapper.setModel(self.model)
self.mapper.setOrientation(QtCore.Qt.Vertical)
self.mapper.addMapping(self.txtSmearUp, MODEL.index('PINHOLE_MIN'))
self.mapper.addMapping(self.txtSmearDown, MODEL.index('PINHOLE_MAX'))
self.mapper.addMapping(self.cbAccuracy, MODEL.index('ACCURACY'))
self.mapper.toFirst()
[docs] def updateData(self, data=None):
"""
Update control elements based on data and model passed
"""
# retain the combobox index
old_index = self.cbSmearing.currentIndex()
self.cbSmearing.clear()
self.cbSmearing.addItem("None")
self.gAccuracy.setVisible(False)
self.data = data
if data is None:
self.setElementsVisibility(False)
model = self.kernel_model
self.updateKernelModel(model, keep_order = True, old_index=old_index)
[docs] def updateKernelModel(self, kernel_model=None, keep_order=False, old_index=None):
"""
Update the model
"""
self.kernel_model = kernel_model
# keep the original cbSmearing value, if already set
index_to_show = self.cbSmearing.currentIndex()
if old_index is not None:
index_to_show = old_index
self.cbSmearing.blockSignals(True)
self.cbSmearing.clear()
self.cbSmearing.addItem("None")
if self.data is None:
self.setElementsVisibility(False)
return
# Find out if data has dQ
self.current_smearer = smear_selection(self.data, self.kernel_model)
self.setSmearInfo()
if self.smear_type is not None:
self.cbSmearing.addItem(SMEARING_QD)
index_to_show = 1 if keep_order else index_to_show
if self.kernel_model is None:
# No model definend yet - just use data file smearing, if any
self.cbSmearing.blockSignals(False)
self.cbSmearing.setCurrentIndex(index_to_show)
return
if isinstance(self.data, Data1D):
self.cbSmearing.addItems(SMEARING_1D)
else:
self.cbSmearing.addItems(SMEARING_2D)
self.cbSmearing.blockSignals(False)
self.cbSmearing.setCurrentIndex(index_to_show)
[docs] def onIndexChange(self, index):
"""
Callback for smearing combobox index change
"""
text = self.cbSmearing.currentText()
if text == 'None':
self.setElementsVisibility(False)
self.current_smearer = None
elif text == "Use dQ Data":
self.setElementsVisibility(True)
self.setDQLabels()
self.onDQSmear()
elif text == "Custom Pinhole Smear":
self.setElementsVisibility(True)
self.setPinholeLabels()
self.onPinholeSmear()
elif text == "Custom Slit Smear":
self.setElementsVisibility(True)
self.setSlitLabels()
self.onSlitSmear()
self.smearingChangedSignal.emit()
[docs] def onModelChange(self):
"""
Respond to model change by notifying any listeners
"""
# Recalculate the smearing
index = self.cbSmearing.currentIndex()
# update the backup values based on model choice
smearing, accuracy, d_down, d_up = self.state()
# don't save the state if dQ Data
if smearing == "Custom Pinhole Smear":
self.pinhole = d_up
elif smearing == 'Custom Slit Smear':
self.slit_height = d_up
self.slit_width = d_down
# check changes in accuracy
if self.accuracy != accuracy:
self.accuracy = accuracy
if accuracy == 'High' or accuracy == 'Extra high':
QtWidgets.QMessageBox.information(self, "Accuracy Warning",
"Higher accuracy is very expensive, \nso fitting can be very slow!")
self.onIndexChange(index)
[docs] def setElementsVisibility(self, visible):
"""
Labels and linedits visibility control
"""
self.lblSmearDown.setVisible(visible)
self.lblSmearUp.setVisible(visible)
self.txtSmearDown.setVisible(visible)
self.txtSmearUp.setVisible(visible)
self.lblUnitUp.setVisible(visible)
self.lblUnitDown.setVisible(visible)
self.setAccuracyVisibility()
[docs] def setAccuracyVisibility(self):
"""
Accuracy combobox visibility
"""
if isinstance(self.data, Data2D) and self.cbSmearing.currentIndex() >= 1:
self.gAccuracy.setVisible(True)
else:
self.gAccuracy.setVisible(False)
[docs] def setPinholeLabels(self):
"""
Use pinhole labels
"""
self.txtSmearDown.setVisible(False)
self.lblSmearDown.setText('')
self.lblUnitDown.setText('')
self.lblSmearUp.setText('<html><head/><body><p>dQ/Q</p></body></html>')
self.lblUnitUp.setText('%')
self.txtSmearUp.setText(str(self.pinhole))
self.txtSmearDown.setEnabled(True)
self.txtSmearUp.setEnabled(True)
[docs] def setSlitLabels(self):
"""
Use pinhole labels
"""
self.lblSmearUp.setText('Slit height')
self.lblSmearDown.setText('Slit width')
self.lblUnitUp.setText('<html><head/><body><p>Å<span style=" vertical-align:super;">-1</span></p></body></html>')
self.lblUnitDown.setText('<html><head/><body><p>Å<span style=" vertical-align:super;">-1</span></p></body></html>')
self.txtSmearUp.setText(str(self.slit_height))
self.txtSmearDown.setText(str(self.slit_width))
self.txtSmearDown.setEnabled(True)
self.txtSmearUp.setEnabled(True)
[docs] def setDQLabels(self):
"""
Use appropriate labels
"""
if self.smear_type == "Pinhole":
text_down = '<html><head/><body><p>[dQ/Q]<span style=" vertical-align:sub;">max</span></p></body></html>'
text_up = '<html><head/><body><p>[dQ/Q]<span style=" vertical-align:sub;">min</span></p></body></html>'
text_unit = '%'
elif self.smear_type == "Slit":
text_down = '<html><head/><body><p>Slit width</p></body></html>'
text_up = '<html><head/><body><p>Slit height</p></body></html>'
text_unit = '<html><head/><body><p>Å<span style=" vertical-align:super;">-1</span></p></body></html>'
else:
text_unit = '%'
text_up = '<html><head/><body><p>‹dQ/Q›<span style=" vertical-align:sub;">r</span></p></body></html>'
text_down = '<html><head/><body><p>‹dQ/Q›<span style=" vertical-align:sub;">φ</span></p></body></html>'
self.lblSmearDown.setText(text_down)
self.lblSmearUp.setText(text_up)
self.lblUnitUp.setText(text_unit)
self.lblUnitDown.setText(text_unit)
self.txtSmearDown.setText(str(self.dq_r))
self.txtSmearUp.setText(str(self.dq_l))
self.txtSmearDown.setEnabled(False)
self.txtSmearUp.setEnabled(False)
[docs] def state(self):
"""
Returns current state of controls
"""
smearing = self.cbSmearing.currentText()
accuracy = ""
d_down = None
d_up = None
if smearing != "None":
accuracy = str(self.model.item(MODEL.index('ACCURACY')).text())
try:
d_down = float(self.txtSmearDown.text())
except ValueError:
d_down = 0.0
try:
d_up = float(self.txtSmearUp.text())
except ValueError:
d_up = 0.0
return (smearing, accuracy, d_down, d_up)
[docs] def setState(self, smearing, accuracy, d_down, d_up):
"""
Sets new values for the controls
"""
# Update the model -> controls update automatically
if accuracy is not None:
self.model.item(MODEL.index('ACCURACY')).setText(accuracy)
if d_down is not None:
self.model.item(MODEL.index('PINHOLE_MIN')).setText(str(d_down))
if d_up is not None:
self.model.item(MODEL.index('PINHOLE_MAX')).setText(str(d_up))
[docs] def onDQSmear(self):
"""
Create a custom dQ smear object that will change the way residuals
are compute when fitting
"""
# resolution information already in data.dx (if 1D) or
# data.dqx_data & data.dqy_data (if 2D),
# so only need to set accuracy for 2D
_, accuracy, _, _ = self.state()
self.current_smearer = smear_selection(self.data, self.kernel_model)
if isinstance(self.data, Data2D):
backend_accuracy = ACCURACY_DICT.get(accuracy)
if backend_accuracy:
self.current_smearer.set_accuracy(accuracy=backend_accuracy)
else:
self.current_smearer.set_accuracy(accuracy='low')
[docs] def onPinholeSmear(self):
"""
Create a custom pinhole smear object that will change the way residuals
are compute when fitting
"""
_, accuracy, _, d_percent = self.state()
self.pinhole = d_percent
if d_percent is None or d_percent == 0.0:
self.current_smearer = None
return
percent = d_percent/100.0
# copy data
data = copy.deepcopy(self.data)
if isinstance(self.data, Data2D):
len_data = len(data.data)
data.dqx_data = np.zeros(len_data)
data.dqy_data = np.zeros(len_data)
q = np.sqrt(data.qx_data**2 + data.qy_data**2)
data.dqx_data = data.dqy_data = percent*q
else:
len_data = len(data.x)
data.dx = np.zeros(len_data)
data.dx = percent * data.x
data.dxl = None
data.dxw = None
self.current_smearer = smear_selection(data, self.kernel_model)
# need to set accuracy for 2D
if isinstance(self.data, Data2D):
backend_accuracy = ACCURACY_DICT.get(accuracy)
if backend_accuracy:
self.current_smearer.set_accuracy(accuracy=backend_accuracy)
else:
self.current_smearer.set_accuracy(accuracy='low')
[docs] def onSlitSmear(self):
"""
Create a custom slit smear object that will change the way residuals
are compute when fitting
"""
_, accuracy, d_width, d_height = self.state()
# Check changes in slit width
if d_width is None:
d_width = 0.0
if d_height is None:
d_height = 0.0
self.slit_width = d_width
self.slit_height = d_height
if isinstance(self.data, Data2D):
self.current_smearer = smear_selection(self.data, self.kernel_model)
return
# make sure once more if it is smearer
data = copy.deepcopy(self.data)
data_len = len(data.x)
data.dx = None
data.dxl = None
data.dxw = None
try:
self.dxl = d_height
data.dxl = self.dxl * np.ones(data_len)
except:
self.dxl = None
data.dxl = np.zeros(data_len)
try:
self.dxw = d_width
data.dxw = self.dxw * np.ones(data_len)
except:
self.dxw = None
data.dxw = np.zeros(data_len)
self.current_smearer = smear_selection(data, self.kernel_model)
[docs] def setSmearInfo(self):
"""
Set default smear_type, dq_l, and dq_r based on the q-resolution information found in the data.
"""
# default
self.smear_type = None
self.dq_l = None
self.dq_r = None
data = self.data
if self.data is None:
return
# First check if data is 2D - If so check that data set has smearing info.
elif isinstance(data, Data2D):
if isinstance(self.smearer(), PySmear2D):
self.smear_type = "Pinhole2d"
self.dq_l = GuiUtils.formatNumber(np.average(data.dqx_data/np.abs(data.qx_data))*100., high=True)
self.dq_r = GuiUtils.formatNumber(np.average(data.dqy_data/np.abs(data.qy_data))*100., high=True)
# Check for pinhole smearing and get min max if it is.
elif (isinstance(self.smearer(), PySmear)
and isinstance(self.smearer().resolution, (Pinhole1D, SesansTransform))):
self.smear_type = "Pinhole"
self.dq_r = GuiUtils.formatNumber(data.dx[0]/data.x[0] *100., high=True)
self.dq_l = GuiUtils.formatNumber(data.dx[-1]/data.x[-1] *100., high=True)
# Check for slit smearing and get min max if it is.
elif isinstance(self.smearer(), PySmear) and isinstance(self.smearer().resolution, Slit1D):
self.smear_type = "Slit"
if data.dxl is not None and np.all(data.dxl, 0):
self.dq_l = GuiUtils.formatNumber(data.dxl[0])
if data.dxw is not None and np.all(data.dxw, 0):
self.dq_r = GuiUtils.formatNumber(data.dxw[0])
[docs] def resetSmearer(self):
self.current_smearer = None
self.cbSmearing.blockSignals(True)
self.cbSmearing.clear()
self.cbSmearing.blockSignals(False)