qtas.py 15.9 KB
Newer Older
Tobias WEBER's avatar
Tobias WEBER committed
1
2
3
4
5
6
7
#
# tascalc gui
# @author Tobias Weber <tweber@ill.fr>
# @date 24-oct-18
# @license see 'LICENSE' file
#

Tobias WEBER's avatar
Tobias WEBER committed
8
9
10
# -----------------------------------------------------------------------------
# dependencies
import sys
Tobias WEBER's avatar
Tobias WEBER committed
11
12
13
14
import tascalc as tas
import numpy as np
import numpy.linalg as la

Tobias WEBER's avatar
Tobias WEBER committed
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# try to import qt5...
try:
	import PyQt5 as qt
	import PyQt5.QtCore as qtc
	import PyQt5.QtWidgets as qtw
	qt_ver = 5
except ImportError:
	# ...and if not possible try to import qt4 instead
	try:
		import PyQt4 as qt
		import PyQt4.QtCore as qtc
		import PyQt4.QtGui as qtw
		qt_ver = 4
	except ImportError:
		print("Error: No suitable version of Qt was found!")
		exit(-1)
# -----------------------------------------------------------------------------
Tobias WEBER's avatar
Tobias WEBER committed
32
33
34



Tobias WEBER's avatar
Tobias WEBER committed
35
36
# -----------------------------------------------------------------------------
# main application
Tobias WEBER's avatar
Tobias WEBER committed
37
38
np.set_printoptions(suppress=True, precision=4)

Tobias WEBER's avatar
Tobias WEBER committed
39
app = qtw.QApplication(sys.argv)
Tobias WEBER's avatar
Tobias WEBER committed
40
app.setApplicationName("qtas")
Tobias WEBER's avatar
Tobias WEBER committed
41
42
43
44
app.setStyle("Fusion")

sett = qtc.QSettings("tobis_stuff", "qtas")
tabs = qtw.QTabWidget()
Tobias WEBER's avatar
Tobias WEBER committed
45
# -----------------------------------------------------------------------------
Tobias WEBER's avatar
Tobias WEBER committed
46
47
48
49
50
51
52



# -----------------------------------------------------------------------------
# variables
B = np.array([[1,0,0], [0,1,0], [0,0,1]])
orient_rlu = np.array([1,0,0])
Tobias WEBER's avatar
Tobias WEBER committed
53
#orient2_rlu = np.array([0,1,0])
Tobias WEBER's avatar
Tobias WEBER committed
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
orient_up_rlu = np.array([0,0,1])
# -----------------------------------------------------------------------------



# -----------------------------------------------------------------------------
# helpers
def getfloat(str):
	try:
		return float(str)
	except ValueError:
		return 0.
# -----------------------------------------------------------------------------



# -----------------------------------------------------------------------------
# crystal tab

xtalpanel = qtw.QWidget()
xtallayout = qtw.QGridLayout(xtalpanel)
Tobias WEBER's avatar
Tobias WEBER committed
75
76
scpanel = xtalpanel
sclayout = xtallayout
Tobias WEBER's avatar
Tobias WEBER committed
77
78
79
80
81
82
83

editA = qtw.QLineEdit(xtalpanel)
editB = qtw.QLineEdit(xtalpanel)
editC = qtw.QLineEdit(xtalpanel)
editAlpha = qtw.QLineEdit(xtalpanel)
editBeta = qtw.QLineEdit(xtalpanel)
editGamma = qtw.QLineEdit(xtalpanel)
Tobias WEBER's avatar
Tobias WEBER committed
84
85
separatorXtal = qtw.QFrame(xtalpanel)
separatorXtal.setFrameStyle(qtw.QFrame.HLine)
Tobias WEBER's avatar
Tobias WEBER committed
86
87
88
89
90
91
editAx = qtw.QLineEdit(scpanel)
editAy = qtw.QLineEdit(scpanel)
editAz = qtw.QLineEdit(scpanel)
editBx = qtw.QLineEdit(scpanel)
editBy = qtw.QLineEdit(scpanel)
editBz = qtw.QLineEdit(scpanel)
Tobias WEBER's avatar
Tobias WEBER committed
92
93
editBMat = qtw.QPlainTextEdit(xtalpanel)
editBMat.setReadOnly(True)
Tobias WEBER's avatar
Tobias WEBER committed
94
95
96


def xtalChanged():
Tobias WEBER's avatar
Tobias WEBER committed
97
	global B, orient_rlu, orient_up_rlu
Tobias WEBER's avatar
Tobias WEBER committed
98
99
	lattice = np.array([getfloat(editA.text()), getfloat(editB.text()), getfloat(editC.text())])
	angles = np.array([getfloat(editAlpha.text()), getfloat(editBeta.text()), getfloat(editGamma.text())])
Tobias WEBER's avatar
Tobias WEBER committed
100
101
102
	orient_rlu = np.array([getfloat(editAx.text()), getfloat(editAy.text()), getfloat(editAz.text())])
	orient2_rlu = np.array([getfloat(editBx.text()), getfloat(editBy.text()), getfloat(editBz.text())])

Tobias WEBER's avatar
Tobias WEBER committed
103
104
105
	try:
		B = tas.get_B(lattice, angles/180.*np.pi)
		invB = la.inv(B)
Tobias WEBER's avatar
Tobias WEBER committed
106
107
108
109
110
111
112
113
114
115
116
117
118

		metric = tas.get_metric(B)
		ang = tas.angle(orient_rlu, orient2_rlu, metric)

		orient_up_rlu = tas.cross(orient_rlu, orient2_rlu, B)
		orient_up_rlu_norm = orient_up_rlu / la.norm(orient_up_rlu)

		UB = tas.get_UB(B, orient_rlu, orient2_rlu, orient_up_rlu)
		invUB = la.inv(UB)

		editBMat.setPlainText("Scattering plane normal: %s rlu.\n" % str(orient_up_rlu_norm) \
			+"Angle between orientation vectors 1 and 2: %.4g deg.\n" % (ang/np.pi*180.) \
			+"\nB =\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n" \
Tobias WEBER's avatar
Tobias WEBER committed
119
			% (B[0,0],B[0,1],B[0,2], B[1,0],B[1,1],B[1,2], B[2,0],B[2,1],B[2,2]) \
Tobias WEBER's avatar
Tobias WEBER committed
120
			+"\nB^(-1) =\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n" \
Tobias WEBER's avatar
Tobias WEBER committed
121
122
123
124
125
126
			% (invB[0,0],invB[0,1],invB[0,2], invB[1,0],invB[1,1],invB[1,2], invB[2,0],invB[2,1],invB[2,2]) \
			+"\nUB =\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n" \
			% (UB[0,0],UB[0,1],UB[0,2], UB[1,0],UB[1,1],UB[1,2], UB[2,0],UB[2,1],UB[2,2]) \
			+"\n(UB)^(-1) =\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n%10.4f %10.4f %10.4f\n" \
			% (invUB[0,0],invUB[0,1],invUB[0,2], invUB[1,0],invUB[1,1],invUB[1,2], invUB[2,0],invUB[2,1],invUB[2,2]) \
		)
Tobias WEBER's avatar
Tobias WEBER committed
127
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
128
		editBMat.setPlainText("invalid")
Tobias WEBER's avatar
bugfix    
Tobias WEBER committed
129
	QChanged()
Tobias WEBER's avatar
Tobias WEBER committed
130

Tobias WEBER's avatar
Tobias WEBER committed
131
def planeChanged():
Tobias WEBER's avatar
Tobias WEBER committed
132
133
	xtalChanged()
	#QChanged()
Tobias WEBER's avatar
Tobias WEBER committed
134
135


Tobias WEBER's avatar
Tobias WEBER committed
136
137
138
139
140
141
editA.textEdited.connect(xtalChanged)
editB.textEdited.connect(xtalChanged)
editC.textEdited.connect(xtalChanged)
editAlpha.textEdited.connect(xtalChanged)
editBeta.textEdited.connect(xtalChanged)
editGamma.textEdited.connect(xtalChanged)
Tobias WEBER's avatar
Tobias WEBER committed
142
143
144
145
146
147
148
editAx.textEdited.connect(planeChanged)
editAy.textEdited.connect(planeChanged)
editAz.textEdited.connect(planeChanged)
editBx.textEdited.connect(planeChanged)
editBy.textEdited.connect(planeChanged)
editBz.textEdited.connect(planeChanged)

Tobias WEBER's avatar
Tobias WEBER committed
149
150
151
152
153
154
155

editA.setText("%.6g" % sett.value("a", 5., type=float))
editB.setText("%.6g" % sett.value("b", 5., type=float))
editC.setText("%.6g" % sett.value("c", 5., type=float))
editAlpha.setText("%.6g" % sett.value("alpha", 90., type=float))
editBeta.setText("%.6g" % sett.value("beta", 90., type=float))
editGamma.setText("%.6g" % sett.value("gamma", 90., type=float))
Tobias WEBER's avatar
Tobias WEBER committed
156
157
158
159
160
161
editAx.setText("%.6g" % sett.value("ax", 1., type=float))
editAy.setText("%.6g" % sett.value("ay", 0., type=float))
editAz.setText("%.6g" % sett.value("az", 0., type=float))
editBx.setText("%.6g" % sett.value("bx", 0., type=float))
editBy.setText("%.6g" % sett.value("by", 1., type=float))
editBz.setText("%.6g" % sett.value("bz", 0., type=float))
Tobias WEBER's avatar
Tobias WEBER committed
162

Tobias WEBER's avatar
Tobias WEBER committed
163
xtallayout.addWidget(qtw.QLabel(u"a (\u212b):", xtalpanel), 0,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
164
xtallayout.addWidget(editA, 0,1, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
165
xtallayout.addWidget(qtw.QLabel(u"b (\u212b):", xtalpanel), 1,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
166
xtallayout.addWidget(editB, 1,1, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
167
xtallayout.addWidget(qtw.QLabel(u"c (\u212b):", xtalpanel), 2,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
168
xtallayout.addWidget(editC, 2,1, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
169
xtallayout.addWidget(qtw.QLabel(u"\u03b1 (deg):", xtalpanel), 3,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
170
xtallayout.addWidget(editAlpha, 3,1, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
171
xtallayout.addWidget(qtw.QLabel(u"\u03b2 (deg):", xtalpanel), 4,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
172
xtallayout.addWidget(editBeta, 4,1, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
173
xtallayout.addWidget(qtw.QLabel(u"\u03b3 (deg):", xtalpanel), 5,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
174
xtallayout.addWidget(editGamma, 5,1, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
175
176
177
178
179
180
181
182
183
184
xtallayout.addWidget(separatorXtal, 6,0, 1,4)
sclayout.addWidget(qtw.QLabel("Orient. 1 (rlu):", scpanel), 7,0, 1,1)
sclayout.addWidget(editAx, 7,1, 1,1)
sclayout.addWidget(editAy, 7,2, 1,1)
sclayout.addWidget(editAz, 7,3, 1,1)
sclayout.addWidget(qtw.QLabel("Orient. 2 (rlu):", scpanel), 8,0, 1,1)
sclayout.addWidget(editBx, 8,1, 1,1)
sclayout.addWidget(editBy, 8,2, 1,1)
sclayout.addWidget(editBz, 8,3, 1,1)
xtallayout.addWidget(editBMat, 9,0, 2,4)
Tobias WEBER's avatar
Tobias WEBER committed
185
186
187
188
189
190
191
192
193
194
195

tabs.addTab(xtalpanel, "Crystal")
# -----------------------------------------------------------------------------



# -----------------------------------------------------------------------------
# tas tab

taspanel = qtw.QWidget()
taslayout = qtw.QGridLayout(taspanel)
Tobias WEBER's avatar
Tobias WEBER committed
196
197
Qpanel = taspanel
Qlayout = taslayout
Tobias WEBER's avatar
Tobias WEBER committed
198
199
200
201
202
203
204

editA1 = qtw.QLineEdit(taspanel)
editA2 = qtw.QLineEdit(taspanel)
editA3 = qtw.QLineEdit(taspanel)
editA4 = qtw.QLineEdit(taspanel)
editA5 = qtw.QLineEdit(taspanel)
editA6 = qtw.QLineEdit(taspanel)
Tobias WEBER's avatar
Tobias WEBER committed
205

Tobias WEBER's avatar
Tobias WEBER committed
206
207
208
editDm = qtw.QLineEdit(taspanel)
editDa = qtw.QLineEdit(taspanel)

Tobias WEBER's avatar
Tobias WEBER committed
209
210
211
212
213
214
edith = qtw.QLineEdit(Qpanel)
editk = qtw.QLineEdit(Qpanel)
editl = qtw.QLineEdit(Qpanel)
editE = qtw.QLineEdit(Qpanel)
editKi = qtw.QLineEdit(Qpanel)
editKf = qtw.QLineEdit(Qpanel)
Tobias WEBER's avatar
Tobias WEBER committed
215
216
editQAbs = qtw.QLineEdit(Qpanel)
editQAbs.setReadOnly(True)
Tobias WEBER's avatar
Tobias WEBER committed
217

Tobias WEBER's avatar
Tobias WEBER committed
218
219
tasstatus = qtw.QLabel(taspanel)

Tobias WEBER's avatar
Tobias WEBER committed
220
221
separatorTas = qtw.QFrame(Qpanel)
separatorTas.setFrameStyle(qtw.QFrame.HLine)
Tobias WEBER's avatar
Tobias WEBER committed
222
223
separatorTas2 = qtw.QFrame(Qpanel)
separatorTas2.setFrameStyle(qtw.QFrame.HLine)
Tobias WEBER's avatar
Tobias WEBER committed
224

Tobias WEBER's avatar
Tobias WEBER committed
225
226

def TASChanged():
Tobias WEBER's avatar
Tobias WEBER committed
227
228
	global orient_rlu, orient_up_rlu

Tobias WEBER's avatar
Tobias WEBER committed
229
230
231
232
233
234
235
236
237
238
239
240
	a1 = getfloat(editA1.text()) / 180. * np.pi
	a2 = a1 * 2.
	a3 = getfloat(editA3.text()) / 180. * np.pi
	a4 = getfloat(editA4.text()) / 180. * np.pi
	a5 = getfloat(editA5.text()) / 180. * np.pi
	a6 = a5 * 2.
	dmono = getfloat(editDm.text())
	dana = getfloat(editDa.text())

	editA2.setText("%.6g" % (a2 / np.pi * 180.))
	editA6.setText("%.6g" % (a6 / np.pi * 180.))

Tobias WEBER's avatar
Tobias WEBER committed
241
242
243
244
245
246
247
248
249
250
251
252
253
254
	try:
		ki = tas.get_monok(a1, dmono)
		kf = tas.get_monok(a5, dana)
		E = tas.get_E(ki, kf)
		Qlen = tas.get_Q(ki, kf, a4)
		Qvec = tas.get_hkl(ki, kf, a3, Qlen, orient_rlu, orient_up_rlu, B)

		edith.setText("%.6g" % Qvec[0])
		editk.setText("%.6g" % Qvec[1])
		editl.setText("%.6g" % Qvec[2])
		editQAbs.setText("%.6g" % Qlen)
		editKi.setText("%.6g" % ki)
		editKf.setText("%.6g" % kf)
		editE.setText("%.6g" % E)
Tobias WEBER's avatar
Tobias WEBER committed
255
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
256
257
258
259
260
261
		edith.setText("invalid")
		editk.setText("invalid")
		editl.setText("invalid")
		editKi.setText("invalid")
		editKf.setText("invalid")
		editE.setText("invalid")
Tobias WEBER's avatar
Tobias WEBER committed
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277


def A2Changed():
	a2 = getfloat(editA2.text()) / 180. * np.pi
	editA1.setText("%.6g" % (0.5*a2 / np.pi * 180.))
	TASChanged()

def A6Changed():
	a6 = getfloat(editA6.text()) / 180. * np.pi
	editA5.setText("%.6g" % (0.5*a6 / np.pi * 180.))
	TASChanged()

def DChanged():
	QChanged()

def QChanged():
Tobias WEBER's avatar
Tobias WEBER committed
278
	global orient_rlu, orient_up_rlu
Tobias WEBER's avatar
Tobias WEBER committed
279
280
281
282
283

	Q_rlu = np.array([getfloat(edith.text()), getfloat(editk.text()), getfloat(editl.text())])
	ki = getfloat(editKi.text())
	kf = getfloat(editKf.text())

Tobias WEBER's avatar
Tobias WEBER committed
284
285
	try:
		[a1, a2] = tas.get_a1a2(ki, getfloat(editDm.text()))
Tobias WEBER's avatar
Tobias WEBER committed
286

Tobias WEBER's avatar
Tobias WEBER committed
287
288
		editA1.setText("%.6g" % (a1 / np.pi * 180.))
		editA2.setText("%.6g" % (a2 / np.pi * 180.))
Tobias WEBER's avatar
Tobias WEBER committed
289
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
290
291
292
293
294
295
296
297
		editA1.setText("invalid")
		editA2.setText("invalid")

	try:
		[a5, a6] = tas.get_a1a2(kf, getfloat(editDa.text()))

		editA5.setText("%.6g" % (a5 / np.pi * 180.))
		editA6.setText("%.6g" % (a6 / np.pi * 180.))
Tobias WEBER's avatar
Tobias WEBER committed
298
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
299
300
301
302
		editA5.setText("invalid")
		editA6.setText("invalid")

	try:
Tobias WEBER's avatar
Tobias WEBER committed
303
		[a3, a4, dist_Q_plane] = tas.get_a3a4(ki, kf, Q_rlu, orient_rlu, orient_up_rlu, B)
Tobias WEBER's avatar
Tobias WEBER committed
304
		Qlen = tas.get_Q(ki, kf, a4)
Tobias WEBER's avatar
Tobias WEBER committed
305
		Q_in_plane = np.abs(dist_Q_plane) < 1e-4
Tobias WEBER's avatar
Tobias WEBER committed
306
307
308
309

		editA3.setText("%.6g" % (a3 / np.pi * 180.))
		editA4.setText("%.6g" % (a4 / np.pi * 180.))
		editQAbs.setText("%.6g" % Qlen)
Tobias WEBER's avatar
Tobias WEBER committed
310
311
312
313
		if Q_in_plane:
			tasstatus.setText("")
		else:
			tasstatus.setText(u"WARNING: Q is out of the plane by %.4g / \u212b!" % dist_Q_plane)
Tobias WEBER's avatar
Tobias WEBER committed
314
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
315
316
		editA3.setText("invalid")
		editA4.setText("invalid")
Tobias WEBER's avatar
Tobias WEBER committed
317

Tobias WEBER's avatar
Tobias WEBER committed
318
319
320
321
def KiKfChanged():
	ki = getfloat(editKi.text())
	kf = getfloat(editKf.text())

Tobias WEBER's avatar
Tobias WEBER committed
322
323
324
	try:
		E = tas.get_E(ki, kf)
		editE.setText("%.6g" % E)
Tobias WEBER's avatar
Tobias WEBER committed
325
326
327

		QChanged()
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
328
		editE.setText("invalid")
Tobias WEBER's avatar
Tobias WEBER committed
329
330
331
332

def EChanged():
	E = getfloat(editE.text())
	kf = getfloat(editKf.text())
Tobias WEBER's avatar
Tobias WEBER committed
333
334
335
336

	try:
		ki = tas.get_ki(kf, E)
		editKi.setText("%.6g" % ki)
Tobias WEBER's avatar
Tobias WEBER committed
337
338
339

		QChanged()
	except (ArithmeticError, la.LinAlgError) as err:
Tobias WEBER's avatar
Tobias WEBER committed
340
		editKi.setText("invalid")
Tobias WEBER's avatar
Tobias WEBER committed
341

Tobias WEBER's avatar
Tobias WEBER committed
342
343
344
345
346
347
348
349
350

editA1.textEdited.connect(TASChanged)
editA3.textEdited.connect(TASChanged)
editA4.textEdited.connect(TASChanged)
editA5.textEdited.connect(TASChanged)
editA2.textEdited.connect(A2Changed)
editA6.textEdited.connect(A6Changed)
editDm.textEdited.connect(DChanged)
editDa.textEdited.connect(DChanged)
Tobias WEBER's avatar
Tobias WEBER committed
351
352
353
edith.textEdited.connect(QChanged)
editk.textEdited.connect(QChanged)
editl.textEdited.connect(QChanged)
Tobias WEBER's avatar
Tobias WEBER committed
354
355
editKi.textEdited.connect(KiKfChanged)
editKf.textEdited.connect(KiKfChanged)
Tobias WEBER's avatar
Tobias WEBER committed
356
357
358
editE.textEdited.connect(EChanged)


Tobias WEBER's avatar
Tobias WEBER committed
359
360
editDm.setText("%.6g" % sett.value("dm", 3.355, type=float))
editDa.setText("%.6g" % sett.value("da", 3.355, type=float))
Tobias WEBER's avatar
Tobias WEBER committed
361
362
363
364
365
366
367
368
edith.setText("%.6g" % sett.value("h", 1., type=float))
editk.setText("%.6g" % sett.value("k", 0., type=float))
editl.setText("%.6g" % sett.value("l", 0., type=float))
#editE.setText("%.6g" % sett.value("E", 0., type=float))
editKi.setText("%.6g" % sett.value("ki", 2.662, type=float))
editKf.setText("%.6g" % sett.value("kf", 2.662, type=float))

Qlayout.addWidget(qtw.QLabel("h (rlu):", Qpanel), 0,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
369
Qlayout.addWidget(edith, 0,1, 1,2)
Tobias WEBER's avatar
Tobias WEBER committed
370
Qlayout.addWidget(qtw.QLabel("k (rlu):", Qpanel), 1,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
371
Qlayout.addWidget(editk, 1,1, 1,2)
Tobias WEBER's avatar
Tobias WEBER committed
372
Qlayout.addWidget(qtw.QLabel("l (rlu):", Qpanel), 2,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
373
Qlayout.addWidget(editl, 2,1, 1,2)
Tobias WEBER's avatar
Tobias WEBER committed
374
Qlayout.addWidget(qtw.QLabel("E (meV):", Qpanel), 3,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
375
Qlayout.addWidget(editE, 3,1, 1,2)
Tobias WEBER's avatar
Tobias WEBER committed
376
Qlayout.addWidget(qtw.QLabel(u"ki, kf (1/\u212b):", Qpanel), 4,0, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
377
Qlayout.addWidget(editKi, 4,1, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
378
Qlayout.addWidget(editKf, 4,2, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
379
380
381
382
383
384
385
386
387
388
389
390
Qlayout.addWidget(qtw.QLabel(u"|Q| (1/\u212b):", Qpanel), 5,0, 1,1)
Qlayout.addWidget(editQAbs, 5,1, 1,2)
Qlayout.addWidget(separatorTas, 6,0,1,3)
taslayout.addWidget(qtw.QLabel("a1, a2 (deg):", taspanel), 7,0, 1,1)
taslayout.addWidget(editA1, 7,1, 1,1)
taslayout.addWidget(editA2, 7,2, 1,1)
taslayout.addWidget(qtw.QLabel("a3, a4 (deg):", taspanel), 8,0, 1,1)
taslayout.addWidget(editA3, 8,1, 1,1)
taslayout.addWidget(editA4, 8,2, 1,1)
taslayout.addWidget(qtw.QLabel("a5, a6 (deg):", taspanel), 9,0, 1,1)
taslayout.addWidget(editA5, 9,1, 1,1)
taslayout.addWidget(editA6, 9,2, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
391
392
393
394
395
396
taslayout.addWidget(separatorTas2, 10,0, 1,3)
taslayout.addWidget(qtw.QLabel("Mono., Ana. d (A):", taspanel), 11,0, 1,1)
taslayout.addWidget(editDm, 11,1, 1,1)
taslayout.addWidget(editDa, 11,2, 1,1)
taslayout.addItem(qtw.QSpacerItem(16,16, qtw.QSizePolicy.Minimum, qtw.QSizePolicy.Expanding), 12,0, 1,3)
taslayout.addWidget(tasstatus, 13,0, 1,3)
Tobias WEBER's avatar
Tobias WEBER committed
397

Tobias WEBER's avatar
Tobias WEBER committed
398
tabs.addTab(taspanel, "TAS")
Tobias WEBER's avatar
Tobias WEBER committed
399
400
401
402
# -----------------------------------------------------------------------------



Tobias WEBER's avatar
Tobias WEBER committed
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
# -----------------------------------------------------------------------------
# info/settings tab
infopanel = qtw.QWidget()
infolayout = qtw.QGridLayout(infopanel)

comboA3 = qtw.QComboBox(infopanel)
comboA3.addItems(["Takin", "NOMAD", "SICS", "NICOS"])
a3_offs = [np.pi/2., np.pi, 0., 0.]
comboA3.setCurrentIndex(sett.value("a3_conv", 1, type=int))

def comboA3ConvChanged():
	idx = comboA3.currentIndex()
	tas.set_a3_offs(a3_offs[idx])
	QChanged()

comboA3.currentIndexChanged.connect(comboA3ConvChanged)


Tobias WEBER's avatar
Tobias WEBER committed
421
422
423
424
separatorInfo = qtw.QFrame(infopanel)
separatorInfo.setFrameStyle(qtw.QFrame.HLine)


Tobias WEBER's avatar
Tobias WEBER committed
425
426
427
infolayout.addWidget(qtw.QLabel("TAS Calculator.", infopanel), 0,0, 1,2)
infolayout.addWidget(qtw.QLabel("Written by Tobias Weber <tweber@ill.fr>.", infopanel), 1,0, 1,2)
infolayout.addWidget(qtw.QLabel("Date: October 24, 2018.", infopanel), 2,0, 1,2)
Tobias WEBER's avatar
Tobias WEBER committed
428
429
430
431
432
433
434
infolayout.addWidget(separatorInfo, 3,0, 1,2)
infolayout.addWidget(qtw.QLabel("Interpreter Version: " + sys.version + ".", infopanel), 4,0, 1,2)
infolayout.addWidget(qtw.QLabel("Numpy Version: " + np.__version__ + ".", infopanel), 5,0, 1,2)
infolayout.addWidget(qtw.QLabel("Qt Version: " + qtc.QT_VERSION_STR + ".", infopanel), 6,0, 1,2)
infolayout.addItem(qtw.QSpacerItem(16,16, qtw.QSizePolicy.Minimum, qtw.QSizePolicy.Expanding), 7,0, 1,2)
infolayout.addWidget(qtw.QLabel("A3 Convention:", infopanel), 8,0, 1,1)
infolayout.addWidget(comboA3, 8,1, 1,1)
Tobias WEBER's avatar
Tobias WEBER committed
435
436
437
438
439

tabs.addTab(infopanel, "Infos")
# -----------------------------------------------------------------------------


Tobias WEBER's avatar
Tobias WEBER committed
440
441
442
443
444
445
446
447
448

# -----------------------------------------------------------------------------
# main dialog window

dlg = qtw.QDialog()
dlg.setWindowTitle("TAS Calculator")
mainlayout = qtw.QGridLayout(dlg)
mainlayout.addWidget(tabs)

Tobias WEBER's avatar
Tobias WEBER committed
449
if sett.contains("geo"):
Tobias WEBER's avatar
Tobias WEBER committed
450
451
452
453
	geo = sett.value("geo")
	if qt_ver == 4:
		geo = geo.toByteArray()
	dlg.restoreGeometry(geo)
Tobias WEBER's avatar
Tobias WEBER committed
454

Tobias WEBER's avatar
Tobias WEBER committed
455
xtalChanged()
Tobias WEBER's avatar
Tobias WEBER committed
456
KiKfChanged()
Tobias WEBER's avatar
Tobias WEBER committed
457
458
comboA3ConvChanged()
#QChanged()
Tobias WEBER's avatar
Tobias WEBER committed
459

Tobias WEBER's avatar
Tobias WEBER committed
460
dlg.show()
Tobias WEBER's avatar
Tobias WEBER committed
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
app.exec_()


# save settings
sett.setValue("a", getfloat(editA.text()))
sett.setValue("b", getfloat(editB.text()))
sett.setValue("c", getfloat(editC.text()))
sett.setValue("alpha", getfloat(editAlpha.text()))
sett.setValue("beta", getfloat(editBeta.text()))
sett.setValue("gamma", getfloat(editGamma.text()))
sett.setValue("ax", getfloat(editAx.text()))
sett.setValue("ay", getfloat(editAy.text()))
sett.setValue("az", getfloat(editAz.text()))
sett.setValue("bx", getfloat(editBx.text()))
sett.setValue("by", getfloat(editBy.text()))
sett.setValue("bz", getfloat(editBz.text()))
sett.setValue("dm", getfloat(editDm.text()))
sett.setValue("da", getfloat(editDa.text()))
sett.setValue("h", getfloat(edith.text()))
sett.setValue("k", getfloat(editk.text()))
sett.setValue("l", getfloat(editl.text()))
#sett.setValue("E", getfloat(editE.text()))
sett.setValue("ki", getfloat(editKi.text()))
sett.setValue("kf", getfloat(editKf.text()))
Tobias WEBER's avatar
Tobias WEBER committed
485
sett.setValue("a3_conv", comboA3.currentIndex())
Tobias WEBER's avatar
Tobias WEBER committed
486
sett.setValue("geo", dlg.saveGeometry())
Tobias WEBER's avatar
Tobias WEBER committed
487
# -----------------------------------------------------------------------------