source: sasview/src/sas/qtgui/Perspectives/PrInversion/dmax.py @ 73a24fa

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalc
Last change on this file since 73a24fa was 73a24fa, checked in by Adam Washington <adam.washington@…>, 6 years ago

Start stealing from the unit tests

I still can't figure out how to get a proper Invertor object, but the
unit tests seem to be a good guide.

  • Property mode set to 100644
File size: 5.7 KB
Line 
1# -*- coding: utf-8 -*-
2"""
3Dialog panel to explore the P(r) inversion results for a range
4of D_max value. User picks a number of points and a range of
5distances, then can toggle between inversion outputs and see
6their distribution as a function of D_max.
7"""
8
9# global
10import sys
11import os
12import logging
13import numpy as np
14from PyQt4 import QtCore
15from PyQt4 import QtGui
16from PyQt4 import QtWebKit
17
18from twisted.internet import threads
19
20# sas-global
21from sas.qtgui.Plotting.PlotterData import Data1D
22from sas.qtgui.Plotting.Plotter import PlotterWidget
23import sas.qtgui.Utilities.GuiUtils as GuiUtils
24
25# local
26from UI.dmax import Ui_DmaxExplorer
27# from InvariantDetails import DetailsDialog
28# from InvariantUtils import WIDGETS
29
30logger = logging.getLogger(__name__)
31
32def enum(*sequential, **named):
33    enums = dict(zip(sequential, range(len(sequential))), **named)
34    return type('Enum', (), enums)
35
36W = enum( 'NPTS',           #0
37          'DMIN',               #1
38          'DMAX',               #2
39          'VARIABLE',         #3
40)
41
42class 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
47    def __init__(self, pr_state, nfunc, parent=None):
48        super(DmaxWindow, self).__init__()
49        self.setupUi(self)
50
51        self.setWindowTitle("Dₘₐₓ Explorer")
52
53        self.pr_state = pr_state
54        self.nfunc = nfunc
55        self.communicator = GuiUtils.Communicate()
56
57        self.plot = PlotterWidget(self, self)
58        self.hasPlot = None
59        self.verticalLayout.insertWidget(0, self.plot)
60
61        # Let's choose the Standard Item Model.
62        self.model = QtGui.QStandardItemModel(self)
63        self.mapper = None
64
65        # # Connect buttons to slots.
66        # # Needs to be done early so default values propagate properly.
67        self.setupSlots()
68
69        # Set up the model.
70        self.setupModel()
71
72        # # Set up the mapper
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)))
88        self.model.setItem(W.VARIABLE,
89                           QtGui.QStandardItem( "χ²/dof"))
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):
104        if not self.mapper:
105            return
106
107        iq0 = []
108        rg = []
109        pos = []
110        pos_err = []
111        osc = []
112        bck = []
113        chi2 = []
114
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
119        original = self.pr_state.d_max
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))
130                bck.append(self.pr_state.background)
131                chi2.append(self.pr_state.chi2)
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)
138
139        #Return the invertor to its original state
140        self.pr_state.d_max = original
141        self.pr_state.invert(self.nfunc)
142
143        plotter = unicode(self.model.item(W.VARIABLE).text())
144        if plotter == u"χ²/dof":
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
156        else:
157            ys = pos_err
158
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="-")
164
165
166if __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
173    from sas.sascalc.pr.invertor import Invertor
174    from sas.sascalc.dataloader.loader import Loader
175
176    data = Loader().load("../../../../../test/pr_inversion/test/sphere_80.txt")[0]
177    pr_state = Invertor()
178    pr_state.d_max = 160.0
179    pr_state.alpha = 0.0007
180    pr_state.x = data.x
181    pr_state.y = data.y
182    pr_state.err = 0.15*np.sqrt(data.y)
183
184    DLG = DmaxWindow(pr_state, 15, reactor)
185    DLG.show()
186    reactor.run()
Note: See TracBrowser for help on using the repository browser.