source: sasview/src/sas/qtgui/Perspectives/Corfunc/CorfuncPerspective.py @ ff8cb73

Last change on this file since ff8cb73 was ff8cb73, checked in by Adam Washington <adam.washington@…>, 7 years ago

Update to latest version of corfunc
  • Property mode set to 100644
File size: 10.5 KB
Line 
1# global
2import sys
3from PyQt4 import QtCore
4from PyQt4 import QtGui
5from PyQt4 import QtWebKit
6
7from twisted.internet import threads
8from twisted.internet import reactor
9
10# sas-global
11from sas.qtgui.Plotting.PlotterData import Data1D
12import sas.qtgui.Utilities.GuiUtils as GuiUtils
13from sas.sascalc.corfunc.corfunc_calculator import CorfuncCalculator
14
15# local
16from UI.CorfuncPanel import Ui_CorfuncDialog
17# from InvariantDetails import DetailsDialog
18from CorfuncUtils import WIDGETS as W
19
20from matplotlib.backends import qt_compat
21from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
22from matplotlib.figure import Figure
23
24
25class MyMplCanvas(FigureCanvas):
26    """Ultimately, this is a QWidget (as well as a FigureCanvasAgg, etc.)."""
27    def __init__(self, model, parent=None, width=5, height=4, dpi=100):
28        self.model = model
29        self.fig = Figure(figsize=(width, height), dpi=dpi)
30        self.axes = self.fig.add_subplot(111)
31
32        FigureCanvas.__init__(self, self.fig)
33        # self.reparent(parent, QPoint(0, 0))
34
35        # FigureCanvas.setSizePolicy(self,
36        #                            QSizePolicy.Expanding,
37        #                            QSizePolicy.Expanding)
38        # FigureCanvas.updateGeometry(self)
39
40        self.data = None
41        self.extrap = None
42
43    def drawQSpace(self):
44        self.fig.clf()
45
46        self.qmin = None
47        self.qmax1 = None
48        self.qmax2 = None
49
50        self.axes = self.fig.add_subplot(111)
51        self.axes.set_xscale("log")
52        self.axes.set_yscale("log")
53
54        qmin = float(self.model.item(W.W_QMIN).text())
55        qmax1 = float(self.model.item(W.W_QMAX).text())
56        qmax2 = float(self.model.item(W.W_QCUTOFF).text())
57
58        if self.data:
59            self.axes.plot(self.data.x, self.data.y)
60            self.axes.axvline(qmin)
61            self.axes.axvline(qmax1)
62            self.axes.axvline(qmax2)
63            self.axes.set_xlim(min(self.data.x), max(self.data.x)*1.5-0.5*min(self.data.x))
64        if self.extrap:
65            self.axes.plot(self.extrap.x, self.extrap.y)
66
67        self.draw()
68
69    def draw_real_space(self):
70        self.fig.clf()
71
72        self.axes = self.fig.add_subplot(111)
73        self.axes.set_xscale("linear")
74        self.axes.set_yscale("linear")
75
76        if self.data:
77            self.axes.plot(self.data.x, self.data.y, label="1D Correlation")
78            self.axes.plot(self.data3.x, self.data3.y, label="3D Correlation")
79            self.axes.plot(self.data_idf.x, self.data_idf.y, label="Interface Distribution Function")
80            self.axes.set_xlim(min(self.data.x), max(self.data.x) / 4)
81            self.axes.legend()
82
83        self.draw()
84
85
86    # def sizeHint(self):
87    #     w, h = self.get_width_height()
88    #     return QSize(w, h)
89
90    # def minimumSizeHint(self):
91    #     return QSize(10, 10)
92
93
94class CorfuncWindow(QtGui.QDialog, Ui_CorfuncDialog):
95    # The controller which is responsible for managing signal slots connections
96    # for the gui and providing an interface to the data model.
97    name = "Corfunc" # For displaying in the combo box
98    #def __init__(self, manager=None, parent=None):
99    def __init__(self, parent=None):
100        super(CorfuncWindow, self).__init__()
101        self.setupUi(self)
102
103        self.setWindowTitle("Corfunc Perspective")
104
105        self.model = QtGui.QStandardItemModel(self)
106        self.communicate = GuiUtils.Communicate()
107        self._calculator = CorfuncCalculator()
108
109        self._canvas = MyMplCanvas(self.model, self)
110        self._realplot = MyMplCanvas(self.model, self)
111        self.verticalLayout_7.insertWidget(0, self._canvas)
112        self.verticalLayout_7.insertWidget(1, self._realplot)
113
114        # Connect buttons to slots.
115        # Needs to be done early so default values propagate properly.
116        self.setupSlots()
117
118        # Set up the model.
119        self.setupModel()
120
121        # Set up the mapper
122        self.setupMapper()
123
124    def setupSlots(self):
125        self.extrapolateBtn.clicked.connect(self.extrapolate)
126        self.transformBtn.clicked.connect(self.transform)
127
128        self.calculateBgBtn.clicked.connect(self.calculateBackground)
129
130        self.model.itemChanged.connect(self.modelChanged)
131
132    def setupModel(self):
133        self.model.setItem(W.W_QMIN,
134                           QtGui.QStandardItem("0.01"))
135        self.model.setItem(W.W_QMAX,
136                           QtGui.QStandardItem("0.20"))
137        self.model.setItem(W.W_QCUTOFF,
138                           QtGui.QStandardItem("0.22"))
139        self.model.setItem(W.W_BACKGROUND,
140                           QtGui.QStandardItem("0"))
141        self.model.setItem(W.W_TRANSFORM,
142                           QtGui.QStandardItem("Fourier"))
143        self.model.setItem(W.W_GUINIERA,
144                           QtGui.QStandardItem("0.0"))
145        self.model.setItem(W.W_GUINIERB,
146                           QtGui.QStandardItem("0.0"))
147        self.model.setItem(W.W_PORODK,
148                           QtGui.QStandardItem("0.0"))
149        self.model.setItem(W.W_PORODSIGMA,
150                           QtGui.QStandardItem("0.0"))
151        self.model.setItem(W.W_CORETHICK, QtGui.QStandardItem(str(0)))
152        self.model.setItem(W.W_INTTHICK, QtGui.QStandardItem(str(0)))
153        self.model.setItem(W.W_HARDBLOCK, QtGui.QStandardItem(str(0)))
154        self.model.setItem(W.W_CRYSTAL, QtGui.QStandardItem(str(0)))
155        self.model.setItem(W.W_POLY, QtGui.QStandardItem(str(0)))
156        self.model.setItem(W.W_PERIOD, QtGui.QStandardItem(str(0)))
157
158    def modelChanged(self, item):
159        self.mapper.toFirst()
160        self._canvas.drawQSpace()
161
162    def _update_calculator(self):
163        self._calculator.lowerq = float(self.model.item(W.W_QMIN).text())
164        qmax1 = float(self.model.item(W.W_QMAX).text())
165        qmax2 = float(self.model.item(W.W_QCUTOFF).text())
166        self._calculator.upperq = (qmax1, qmax2)
167        self._calculator.background = float(self.model.item(W.W_BACKGROUND).text())
168
169    def extrapolate(self):
170        self._update_calculator()
171        params, extrapolation = self._calculator.compute_extrapolation()
172
173        self.model.setItem(W.W_GUINIERA, QtGui.QStandardItem(str(params['A'])))
174        self.model.setItem(W.W_GUINIERB, QtGui.QStandardItem(str(params['B'])))
175        self.model.setItem(W.W_PORODK, QtGui.QStandardItem(str(params['K'])))
176        self.model.setItem(W.W_PORODSIGMA, QtGui.QStandardItem(str(params['sigma'])))
177
178        self._canvas.extrap = extrapolation
179        self._canvas.drawQSpace()
180
181
182    def transform(self):
183        if self.fourierBtn.isChecked():
184            method = "fourier"
185        elif self.hilbertBtn.isChecked():
186            method = "hilbert"
187
188        extrap = self._canvas.extrap
189        bg = float(self.model.item(W.W_BACKGROUND).text())
190        def updatefn(*args, **kwargs):
191            pass
192
193        def completefn(transforms):
194            (trans1, trans3, idf) = transforms
195            self._realplot.data = trans1
196            self._realplot.data3 = trans3
197            self._realplot.data_idf = idf
198            self._realplot.draw_real_space()
199            params = self._calculator.extract_parameters(trans1)
200            self.model.setItem(W.W_CORETHICK,
201                               QtGui.QStandardItem(str(params['d0'])))
202            self.model.setItem(W.W_INTTHICK,
203                               QtGui.QStandardItem(str(params['dtr'])))
204            self.model.setItem(W.W_HARDBLOCK,
205                               QtGui.QStandardItem(str(params['Lc'])))
206            self.model.setItem(W.W_CRYSTAL,
207                               QtGui.QStandardItem(str(params['fill'])))
208            self.model.setItem(W.W_POLY,
209                               QtGui.QStandardItem(str(params['A'])))
210            self.model.setItem(W.W_PERIOD,
211                               QtGui.QStandardItem(str(params['max'])))
212
213        self._update_calculator()
214        self._calculator.compute_transform(extrap, method, bg, completefn, updatefn)
215
216
217    def setupMapper(self):
218        self.mapper = QtGui.QDataWidgetMapper(self)
219        self.mapper.setOrientation(QtCore.Qt.Vertical)
220        self.mapper.setModel(self.model)
221
222        self.mapper.addMapping(self.qMin, W.W_QMIN)
223        self.mapper.addMapping(self.qMax1, W.W_QMAX)
224        self.mapper.addMapping(self.qMax2, W.W_QCUTOFF)
225        self.mapper.addMapping(self.bg, W.W_BACKGROUND)
226
227        self.mapper.addMapping(self.guinierA, W.W_GUINIERA)
228        self.mapper.addMapping(self.guinierB, W.W_GUINIERB)
229        self.mapper.addMapping(self.porodK, W.W_PORODK)
230        self.mapper.addMapping(self.porodSigma, W.W_PORODSIGMA)
231
232        self.mapper.addMapping(self.avgCoreThick, W.W_CORETHICK)
233        self.mapper.addMapping(self.avgIntThick, W.W_INTTHICK)
234        self.mapper.addMapping(self.avgHardBlock, W.W_HARDBLOCK)
235        self.mapper.addMapping(self.polydisp, W.W_POLY)
236        self.mapper.addMapping(self.longPeriod, W.W_PERIOD)
237        self.mapper.addMapping(self.localCrystal, W.W_CRYSTAL)
238
239        self.mapper.toFirst()
240
241    def calculateBackground(self):
242        self._update_calculator()
243        bg = self._calculator.compute_background()
244        temp = QtGui.QStandardItem(str(bg))
245        self.model.setItem(W.W_BACKGROUND, temp)
246
247    def allowBatch(self):
248        """
249        We cannot perform corfunc analysis in batch at this time.
250        """
251        return False
252
253    def setData(self, data_item, is_batch=False):
254        """
255        Obtain a QStandardItem object and dissect it to get Data1D/2D
256        Pass it over to the calculator
257        """
258        if not isinstance(data_item, list):
259            msg = "Incorrect type passed to the Corfunc Perpsective"
260            raise AttributeError(msg)
261
262        if not isinstance(data_item[0], QtGui.QStandardItem):
263            msg = "Incorrect type passed to the Corfunc Perspective"
264            raise AttributeError(msg)
265
266        self._model_item = data_item[0]
267        data = GuiUtils.dataFromItem(self._model_item)
268        self._calculator.set_data(data)
269
270        self._canvas.data = data
271        self._canvas.drawQSpace()
272
273        # self.model.item(WIDGETS.W_FILENAME).setData(QtCoreQVariant(self._model_item.text()))
274
275    def setClosable(self, value=True):
276        """
277        Allow outsiders close this widget
278        """
279        assert isinstance(value, bool)
280
281        self._allow_close = value
282
283
284if __name__ == "__main__":
285    app = QtGui.QApplication([])
286    import qt4reactor
287    # qt4reactor.install()
288    # DO NOT move the following import to the top!
289    # (unless you know what you're doing)
290    from twisted.internet import reactor
291    dlg = CorfuncWindow(reactor)
292    print(dlg)
293    dlg.show()
294    # print(reactor)
295    # reactor.run()
296    sys.exit(app.exec_())
Note: See TracBrowser for help on using the repository browser.