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 smearer(self): """ Returns the current smearer """ return self.current_smearer
[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>&lsaquo;dQ/Q&rsaquo;<span style=" vertical-align:sub;">r</span></p></body></html>' text_down = '<html><head/><body><p>&lsaquo;dQ/Q&rsaquo;<span style=" vertical-align:sub;">&phi;</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)