[6fd89d4] | 1 | # -*- coding: utf-8 -*- |
---|
| 2 | """ |
---|
| 3 | Dialog panel to explore the P(r) inversion results for a range |
---|
| 4 | of D_max value. User picks a number of points and a range of |
---|
| 5 | distances, then can toggle between inversion outputs and see |
---|
| 6 | their distribution as a function of D_max. |
---|
| 7 | """ |
---|
| 8 | |
---|
| 9 | # global |
---|
| 10 | import sys |
---|
| 11 | import os |
---|
[6060a86] | 12 | import logging |
---|
| 13 | import numpy as np |
---|
[6fd89d4] | 14 | from PyQt4 import QtCore |
---|
| 15 | from PyQt4 import QtGui |
---|
| 16 | from PyQt4 import QtWebKit |
---|
| 17 | |
---|
| 18 | from twisted.internet import threads |
---|
| 19 | |
---|
| 20 | # sas-global |
---|
| 21 | from sas.qtgui.Plotting.PlotterData import Data1D |
---|
| 22 | from sas.qtgui.Plotting.Plotter import PlotterWidget |
---|
| 23 | import sas.qtgui.Utilities.GuiUtils as GuiUtils |
---|
| 24 | |
---|
| 25 | # local |
---|
| 26 | from UI.dmax import Ui_DmaxExplorer |
---|
| 27 | # from InvariantDetails import DetailsDialog |
---|
| 28 | # from InvariantUtils import WIDGETS |
---|
| 29 | |
---|
[6060a86] | 30 | logger = logging.getLogger(__name__) |
---|
| 31 | |
---|
[60bf0db] | 32 | def enum(*sequential, **named): |
---|
| 33 | enums = dict(zip(sequential, range(len(sequential))), **named) |
---|
| 34 | return type('Enum', (), enums) |
---|
| 35 | |
---|
| 36 | W = enum( 'NPTS', #0 |
---|
| 37 | 'DMIN', #1 |
---|
| 38 | 'DMAX', #2 |
---|
| 39 | 'VARIABLE', #3 |
---|
| 40 | ) |
---|
[6fd89d4] | 41 | |
---|
| 42 | class DmaxWindow(QtGui.QDialog, Ui_DmaxExplorer): |
---|
| 43 | # The controller which is responsible for managing signal slots connections |
---|
| 44 | # for the gui and providing an interface to the data model. |
---|
| 45 | name = "Dmax Explorer" # For displaying in the combo box |
---|
| 46 | |
---|
[60bf0db] | 47 | def __init__(self, pr_state, nfunc, parent=None): |
---|
[6fd89d4] | 48 | super(DmaxWindow, self).__init__() |
---|
| 49 | self.setupUi(self) |
---|
| 50 | |
---|
| 51 | self.setWindowTitle("Dâââ Explorer") |
---|
| 52 | |
---|
| 53 | self.pr_state = pr_state |
---|
[60bf0db] | 54 | self.nfunc = nfunc |
---|
[6fd89d4] | 55 | self.communicator = GuiUtils.Communicate() |
---|
| 56 | |
---|
| 57 | self.plot = PlotterWidget(self, self) |
---|
[5fe52594] | 58 | self.hasPlot = None |
---|
[6fd89d4] | 59 | self.verticalLayout.insertWidget(0, self.plot) |
---|
| 60 | |
---|
| 61 | # Let's choose the Standard Item Model. |
---|
| 62 | self.model = QtGui.QStandardItemModel(self) |
---|
[6060a86] | 63 | self.mapper = None |
---|
[6fd89d4] | 64 | |
---|
| 65 | # # Connect buttons to slots. |
---|
| 66 | # # Needs to be done early so default values propagate properly. |
---|
[60bf0db] | 67 | self.setupSlots() |
---|
[6fd89d4] | 68 | |
---|
[60bf0db] | 69 | # Set up the model. |
---|
| 70 | self.setupModel() |
---|
[6fd89d4] | 71 | |
---|
| 72 | # # Set up the mapper |
---|
[60bf0db] | 73 | self.setupMapper() |
---|
| 74 | |
---|
| 75 | def setupSlots(self): |
---|
| 76 | self.closeButton.clicked.connect(self.close) |
---|
| 77 | |
---|
| 78 | self.model.itemChanged.connect(self.modelChanged) |
---|
| 79 | |
---|
| 80 | def setupModel(self): |
---|
| 81 | self.model.setItem(W.NPTS, QtGui.QStandardItem(str(self.nfunc))) |
---|
| 82 | self.model.setItem(W.DMIN, |
---|
| 83 | QtGui.QStandardItem( |
---|
| 84 | str(0.9*self.pr_state.d_max))) |
---|
| 85 | self.model.setItem(W.DMAX, |
---|
| 86 | QtGui.QStandardItem( |
---|
| 87 | str(1.1*self.pr_state.d_max))) |
---|
[481c4f4] | 88 | self.model.setItem(W.VARIABLE, |
---|
| 89 | QtGui.QStandardItem( "ϲ/dof")) |
---|
[60bf0db] | 90 | |
---|
| 91 | def setupMapper(self): |
---|
| 92 | self.mapper = QtGui.QDataWidgetMapper(self) |
---|
| 93 | self.mapper.setOrientation(QtCore.Qt.Vertical) |
---|
| 94 | self.mapper.setModel(self.model) |
---|
| 95 | |
---|
| 96 | self.mapper.addMapping(self.Npts, W.NPTS) |
---|
| 97 | self.mapper.addMapping(self.minDist, W.DMIN) |
---|
| 98 | self.mapper.addMapping(self.maxDist, W.DMAX) |
---|
| 99 | self.mapper.addMapping(self.dependentVariable, W.VARIABLE) |
---|
| 100 | |
---|
| 101 | self.mapper.toFirst() |
---|
| 102 | |
---|
| 103 | def modelChanged(self, item): |
---|
[6060a86] | 104 | if not self.mapper: |
---|
| 105 | return |
---|
| 106 | |
---|
| 107 | iq0 = [] |
---|
| 108 | rg = [] |
---|
| 109 | pos = [] |
---|
| 110 | pos_err = [] |
---|
| 111 | osc = [] |
---|
[4aa62c8] | 112 | bck = [] |
---|
| 113 | chi2 = [] |
---|
| 114 | |
---|
[6060a86] | 115 | xs = np.linspace(float(self.model.item(W.DMIN).text()), |
---|
| 116 | float(self.model.item(W.DMAX).text()), |
---|
| 117 | float(self.model.item(W.NPTS).text())) |
---|
| 118 | |
---|
[3e20577] | 119 | original = self.pr_state.d_max |
---|
[6060a86] | 120 | for x in xs: |
---|
| 121 | self.pr_state.d_max = x |
---|
| 122 | try: |
---|
| 123 | out, cov = self.pr_state.invert(self.nfunc) |
---|
| 124 | |
---|
| 125 | iq0.append(self.pr_state.iq0(out)) |
---|
| 126 | rg.append(self.pr_state.rg(out)) |
---|
| 127 | pos.append(self.pr_state.get_positive(out)) |
---|
| 128 | pos_err.append(self.pr_state.get_pos_err(out, cov)) |
---|
| 129 | osc.append(self.pr_state.oscillations(out)) |
---|
[4aa62c8] | 130 | bck.append(self.pr_state.background) |
---|
| 131 | chi2.append(self.pr_state.chi2) |
---|
[6060a86] | 132 | except: |
---|
| 133 | # This inversion failed, skip this D_max value |
---|
| 134 | msg = "ExploreDialog: inversion failed " |
---|
| 135 | msg += "for D_max=%s\n%s" % (str(x), sys.exc_value) |
---|
| 136 | print(msg) |
---|
| 137 | # logger.error(msg) |
---|
[73a24fa] | 138 | |
---|
| 139 | #Return the invertor to its original state |
---|
[3e20577] | 140 | self.pr_state.d_max = original |
---|
[73a24fa] | 141 | self.pr_state.invert(self.nfunc) |
---|
[3e20577] | 142 | |
---|
[481c4f4] | 143 | plotter = unicode(self.model.item(W.VARIABLE).text()) |
---|
[e742b63c] | 144 | if plotter == u"ϲ/dof": |
---|
[481c4f4] | 145 | ys = chi2 |
---|
| 146 | elif plotter == u"I(Q=0)": |
---|
| 147 | ys = iq0 |
---|
| 148 | elif plotter == u"Rg": |
---|
| 149 | ys = rg |
---|
| 150 | elif plotter == u"Oscillation parameter": |
---|
| 151 | ys = osc |
---|
| 152 | elif plotter == u"Background": |
---|
| 153 | ys = bck |
---|
| 154 | elif plotter == u"Positive Fraction": |
---|
| 155 | ys = pos |
---|
[e742b63c] | 156 | else: |
---|
| 157 | ys = pos_err |
---|
[481c4f4] | 158 | |
---|
[5fe52594] | 159 | data = Data1D(xs, ys) |
---|
| 160 | if self.hasPlot: |
---|
| 161 | self.plot.removePlot(None) |
---|
| 162 | self.hasPlot = True |
---|
| 163 | self.plot.plot(data=data, marker="-") |
---|
[6fd89d4] | 164 | |
---|
| 165 | |
---|
| 166 | if __name__ == "__main__": |
---|
| 167 | APP = QtGui.QApplication([]) |
---|
| 168 | import qt4reactor |
---|
| 169 | qt4reactor.install() |
---|
| 170 | # DO NOT move the following import to the top! |
---|
| 171 | # (unless you know what you're doing) |
---|
| 172 | from twisted.internet import reactor |
---|
[60bf0db] | 173 | from sas.sascalc.pr.invertor import Invertor |
---|
[3e20577] | 174 | from sas.sascalc.dataloader.loader import Loader |
---|
| 175 | |
---|
| 176 | data = Loader().load("../../../../../test/pr_inversion/test/sphere_80.txt")[0] |
---|
[60bf0db] | 177 | pr_state = Invertor() |
---|
[73a24fa] | 178 | pr_state.d_max = 160.0 |
---|
| 179 | pr_state.alpha = 0.0007 |
---|
[3e20577] | 180 | pr_state.x = data.x |
---|
| 181 | pr_state.y = data.y |
---|
[73a24fa] | 182 | pr_state.err = 0.15*np.sqrt(data.y) |
---|
[3e20577] | 183 | |
---|
[73a24fa] | 184 | DLG = DmaxWindow(pr_state, 15, reactor) |
---|
[6fd89d4] | 185 | DLG.show() |
---|
| 186 | reactor.run() |
---|