1 | # -*- coding: utf-8 -*- |
---|
2 | #GSASII - image data display routines |
---|
3 | ########### SVN repository information ################### |
---|
4 | # $Date: 2014-11-21 22:12:53 +0000 (Fri, 21 Nov 2014) $ |
---|
5 | # $Author: toby $ |
---|
6 | # $Revision: 1580 $ |
---|
7 | # $URL: trunk/GSASIIimgGUI.py $ |
---|
8 | # $Id: GSASIIimgGUI.py 1580 2014-11-21 22:12:53Z toby $ |
---|
9 | ########### SVN repository information ################### |
---|
10 | ''' |
---|
11 | *GSASIIimgGUI: Image GUI* |
---|
12 | ------------------------- |
---|
13 | |
---|
14 | Control image display and processing |
---|
15 | |
---|
16 | ''' |
---|
17 | import os.path |
---|
18 | import wx |
---|
19 | import wx.lib.scrolledpanel as wxscroll |
---|
20 | import matplotlib as mpl |
---|
21 | import math |
---|
22 | import time |
---|
23 | import copy |
---|
24 | import GSASIIpath |
---|
25 | GSASIIpath.SetVersionNumber("$Revision: 1580 $") |
---|
26 | import GSASIIimage as G2img |
---|
27 | import GSASIImath as G2mth |
---|
28 | import GSASIIplot as G2plt |
---|
29 | import GSASIIIO as G2IO |
---|
30 | import GSASIIgrid as G2gd |
---|
31 | import numpy as np |
---|
32 | |
---|
33 | VERY_LIGHT_GREY = wx.Colour(235,235,235) |
---|
34 | WACV = wx.ALIGN_CENTER_VERTICAL |
---|
35 | |
---|
36 | # trig functions in degrees |
---|
37 | sind = lambda x: math.sin(x*math.pi/180.) |
---|
38 | tand = lambda x: math.tan(x*math.pi/180.) |
---|
39 | cosd = lambda x: math.cos(x*math.pi/180.) |
---|
40 | asind = lambda x: 180.*math.asin(x)/math.pi |
---|
41 | |
---|
42 | ################################################################################ |
---|
43 | ##### Image Data |
---|
44 | ################################################################################ |
---|
45 | |
---|
46 | def UpdateImageData(G2frame,data): |
---|
47 | |
---|
48 | def OnPixVal(event): |
---|
49 | Obj = event.GetEventObject() |
---|
50 | id = Indx[Obj.GetId()] |
---|
51 | try: |
---|
52 | data['pixelSize'][id] = min(500,max(10,float(Obj.GetValue()))) |
---|
53 | except ValueError: |
---|
54 | pass |
---|
55 | Obj.SetValue('%.3f'%(data['pixelSize'][id])) |
---|
56 | G2plt.PlotExposedImage(G2frame,newPlot=True,event=event) |
---|
57 | |
---|
58 | |
---|
59 | |
---|
60 | if G2frame.dataDisplay: |
---|
61 | G2frame.dataDisplay.Destroy() |
---|
62 | if not G2frame.dataFrame.GetStatusBar(): |
---|
63 | G2frame.dataFrame.CreateStatusBar() |
---|
64 | G2frame.dataDisplay = wx.Panel(G2frame.dataFrame) |
---|
65 | mainSizer = wx.BoxSizer(wx.VERTICAL) |
---|
66 | mainSizer.Add(wx.StaticText(G2frame.dataDisplay, |
---|
67 | label='Do not change anything here unless you are absolutely sure!'),0,WACV) |
---|
68 | mainSizer.Add(wx.StaticText(G2frame.dataDisplay,label=' Image size: %d by %d'%(data['size'][0],data['size'][1])),0,WACV) |
---|
69 | pixSize = wx.FlexGridSizer(0,4,5,5) |
---|
70 | pixLabels = [u' Pixel X-dimension (\xb5m)',u' Pixel Y-dimension (\xb5m)'] |
---|
71 | Indx = {} |
---|
72 | for i,[pixLabel,pix] in enumerate(zip(pixLabels,data['pixelSize'])): |
---|
73 | pixSize.Add(wx.StaticText(G2frame.dataDisplay,label=pixLabel),0,WACV) |
---|
74 | pixVal = wx.TextCtrl(G2frame.dataDisplay,value='%.3f'%(pix),style=wx.TE_PROCESS_ENTER) |
---|
75 | Indx[pixVal.GetId()] = i |
---|
76 | pixVal.Bind(wx.EVT_TEXT_ENTER,OnPixVal) |
---|
77 | pixVal.Bind(wx.EVT_KILL_FOCUS,OnPixVal) |
---|
78 | pixSize.Add(pixVal,0,WACV) |
---|
79 | mainSizer.Add(pixSize,0) |
---|
80 | |
---|
81 | mainSizer.Layout() |
---|
82 | G2frame.dataDisplay.SetSizer(mainSizer) |
---|
83 | fitSize = mainSizer.Fit(G2frame.dataFrame) |
---|
84 | G2frame.dataFrame.setSizePosLeft(fitSize) |
---|
85 | G2frame.dataDisplay.SetSize(fitSize) |
---|
86 | |
---|
87 | ################################################################################ |
---|
88 | ##### Image Controls |
---|
89 | ################################################################################ |
---|
90 | def UpdateImageControls(G2frame,data,masks): |
---|
91 | '''Shows and handles the controls on the "Image Controls" |
---|
92 | data tree entry |
---|
93 | ''' |
---|
94 | import ImageCalibrants as calFile |
---|
95 | #patch |
---|
96 | if 'GonioAngles' not in data: |
---|
97 | data['GonioAngles'] = [0.,0.,0.] |
---|
98 | if 'DetDepth' not in data: |
---|
99 | data['DetDepth'] = 0. |
---|
100 | data['DetDepthRef'] = False |
---|
101 | if 'SampleAbs' not in data: |
---|
102 | data['SampleShape'] = 'Cylinder' |
---|
103 | data['SampleAbs'] = [0.0,False] |
---|
104 | if 'binType' not in data: |
---|
105 | if 'PWDR' in data['type']: |
---|
106 | data['binType'] = '2-theta' |
---|
107 | elif 'SASD' in data['type']: |
---|
108 | data['binType'] = 'log(q)' |
---|
109 | #end patch |
---|
110 | |
---|
111 | # Menu items |
---|
112 | |
---|
113 | def OnCalibrate(event): |
---|
114 | G2frame.dataFrame.ImageEdit.Enable(id=G2gd.wxID_IMRECALIBRATE,enable=True) |
---|
115 | G2frame.dataFrame.GetStatusBar().SetStatusText('Select > 4 points on 1st used ring; LB to pick, RB on point to delete else RB to finish') |
---|
116 | G2frame.ifGetRing = True |
---|
117 | |
---|
118 | def OnRecalibrate(event): |
---|
119 | G2img.ImageRecalibrate(G2frame,data,masks) |
---|
120 | UpdateImageControls(G2frame,data,masks) |
---|
121 | |
---|
122 | def OnClearCalib(event): |
---|
123 | data['ring'] = [] |
---|
124 | data['rings'] = [] |
---|
125 | data['ellipses'] = [] |
---|
126 | # G2frame.dataFrame.ImageEdit.Enable(id=G2gd.wxID_IMRECALIBRATE,enable=False) |
---|
127 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
128 | |
---|
129 | def OnIntegrate(event): |
---|
130 | blkSize = 128 #this seems to be optimal; will break in polymask if >1024 |
---|
131 | Nx,Ny = data['size'] |
---|
132 | nXBlks = (Nx-1)/blkSize+1 |
---|
133 | nYBlks = (Ny-1)/blkSize+1 |
---|
134 | Nup = nXBlks*nYBlks*3+3 |
---|
135 | dlg = wx.ProgressDialog("Elapsed time","2D image integration",Nup, |
---|
136 | style = wx.PD_ELAPSED_TIME|wx.PD_AUTO_HIDE) |
---|
137 | try: |
---|
138 | sumImg = G2frame.ImageZ |
---|
139 | darkImg,darkScale = data['dark image'] |
---|
140 | if darkImg: |
---|
141 | Did = G2gd.GetPatternTreeItemId(G2frame, G2frame.root, darkImg) |
---|
142 | Npix,imagefile = G2frame.PatternTree.GetItemPyData(Did) |
---|
143 | darkImage = G2IO.GetImageData(G2frame,imagefile,True) |
---|
144 | sumImg += darkImage*darkScale |
---|
145 | backImg,backScale = data['background image'] |
---|
146 | if backImg: #ignores any transmission effect in the background image |
---|
147 | Bid = G2gd.GetPatternTreeItemId(G2frame, G2frame.root, backImg) |
---|
148 | Npix,imagefile = G2frame.PatternTree.GetItemPyData(Bid) |
---|
149 | backImage = G2IO.GetImageData(G2frame,imagefile,True) |
---|
150 | Bdata = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,Bid,'Image Controls')) |
---|
151 | BdarkImg,BdarkScale = Bdata['dark image'] |
---|
152 | if BdarkImg: |
---|
153 | BDid = G2gd.GetPatternTreeItemId(G2frame, G2frame.root,BdarkImg) |
---|
154 | Npix,imagefile = G2frame.PatternTree.GetItemPyData(BDid) |
---|
155 | BdarkImage = G2IO.GetImageData(G2frame,imagefile,True) |
---|
156 | backImage += BdarkImage*BdarkScale |
---|
157 | sumImg += backImage*backScale |
---|
158 | G2frame.Integrate = G2img.ImageIntegrate(sumImg,data,masks,blkSize,dlg) |
---|
159 | # G2plt.PlotIntegration(G2frame,newPlot=True) |
---|
160 | Id = G2IO.SaveIntegration(G2frame,G2frame.PickId,data) |
---|
161 | G2frame.PatternId = Id |
---|
162 | G2frame.PatternTree.SelectItem(Id) |
---|
163 | G2frame.PatternTree.Expand(Id) |
---|
164 | finally: |
---|
165 | dlg.Destroy() |
---|
166 | for item in G2frame.MakePDF: item.Enable(True) |
---|
167 | |
---|
168 | def OnIntegrateAll(event): |
---|
169 | print 'integrate all' |
---|
170 | TextList = [[False,'All IMG',0]] |
---|
171 | Names = [] |
---|
172 | if G2frame.PatternTree.GetCount(): |
---|
173 | id, cookie = G2frame.PatternTree.GetFirstChild(G2frame.root) |
---|
174 | while id: |
---|
175 | name = G2frame.PatternTree.GetItemText(id) |
---|
176 | Names.append(name) |
---|
177 | if 'IMG' in name: |
---|
178 | TextList.append([False,name,id]) |
---|
179 | id, cookie = G2frame.PatternTree.GetNextChild(G2frame.root, cookie) |
---|
180 | if len(TextList) == 1: |
---|
181 | G2frame.ErrorDialog('Nothing to integrate','There must some "IMG" patterns') |
---|
182 | return |
---|
183 | dlg = G2frame.CopyDialog(G2frame,'Image integration controls','Select images to integrate:',TextList) |
---|
184 | try: |
---|
185 | if dlg.ShowModal() == wx.ID_OK: |
---|
186 | result = dlg.GetData() |
---|
187 | if result[0][0]: #the 'All IMG' is True |
---|
188 | result = TextList[1:] |
---|
189 | for item in result: item[0] = True |
---|
190 | G2frame.EnablePlot = False |
---|
191 | for item in result: |
---|
192 | ifintegrate,name,id = item |
---|
193 | if ifintegrate: |
---|
194 | Id = G2gd.GetPatternTreeItemId(G2frame,id, 'Image Controls') |
---|
195 | Data = G2frame.PatternTree.GetItemPyData(Id) |
---|
196 | blkSize = 128 #this seems to be optimal; will break in polymask if >1024 |
---|
197 | Nx,Ny = Data['size'] |
---|
198 | nXBlks = (Nx-1)/blkSize+1 |
---|
199 | nYBlks = (Ny-1)/blkSize+1 |
---|
200 | Nup = nXBlks*nYBlks*3+3 |
---|
201 | dlgp = wx.ProgressDialog("Elapsed time","2D image integration",Nup, |
---|
202 | style = wx.PD_ELAPSED_TIME|wx.PD_AUTO_HIDE) |
---|
203 | try: |
---|
204 | id = G2gd.GetPatternTreeItemId(G2frame, G2frame.root, name) |
---|
205 | Npix,imagefile = G2frame.PatternTree.GetItemPyData(id) |
---|
206 | image = G2IO.GetImageData(G2frame,imagefile,True) |
---|
207 | backImage = [] |
---|
208 | if Data['background image'][0]: |
---|
209 | backImg = Data['background image'][0] |
---|
210 | backScale = Data['background image'][1] |
---|
211 | id = G2gd.GetPatternTreeItemId(G2frame, G2frame.root, backImg) |
---|
212 | Npix,imagefile = G2frame.PatternTree.GetItemPyData(id) |
---|
213 | backImage = G2IO.GetImageData(G2frame,imagefile,True)*backScale |
---|
214 | try: |
---|
215 | Masks = G2frame.PatternTree.GetItemPyData( |
---|
216 | G2gd.GetPatternTreeItemId(G2frame,G2frame.Image, 'Masks')) |
---|
217 | except TypeError: #missing Masks |
---|
218 | Imin,Imax = Data['Range'] |
---|
219 | Masks = {'Points':[],'Rings':[],'Arcs':[],'Polygons':[],'Frames':[],'Thresholds':[(Imin,Imax),[Imin,Imax]]} |
---|
220 | G2frame.PatternTree.SetItemPyData( |
---|
221 | G2gd.GetPatternTreeItemId(G2frame,G2frame.Image, 'Masks'),Masks) |
---|
222 | if len(backImage): |
---|
223 | G2frame.Integrate = G2img.ImageIntegrate(image+backImage,Data,Masks,blkSize,dlgp) |
---|
224 | else: |
---|
225 | G2frame.Integrate = G2img.ImageIntegrate(image,Data,Masks,blkSize,dlgp) |
---|
226 | pId = G2IO.SaveIntegration(G2frame,Id,Data) |
---|
227 | finally: |
---|
228 | dlgp.Destroy() |
---|
229 | else: |
---|
230 | G2frame.EnablePlot = True |
---|
231 | G2frame.PatternTree.SelectItem(pId) |
---|
232 | G2frame.PatternTree.Expand(pId) |
---|
233 | G2frame.PatternId = pId |
---|
234 | |
---|
235 | finally: |
---|
236 | dlg.Destroy() |
---|
237 | |
---|
238 | def OnCopyControls(event): |
---|
239 | TextList = [[False,'All IMG',0]] |
---|
240 | Names = [] |
---|
241 | if G2frame.PatternTree.GetCount(): |
---|
242 | id, cookie = G2frame.PatternTree.GetFirstChild(G2frame.root) |
---|
243 | while id: |
---|
244 | name = G2frame.PatternTree.GetItemText(id) |
---|
245 | Names.append(name) |
---|
246 | if 'IMG' in name: |
---|
247 | if id == G2frame.Image: |
---|
248 | Source = name |
---|
249 | Data = copy.deepcopy(data) |
---|
250 | # Data = copy.deepcopy(G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Image Controls'))) |
---|
251 | Data['showLines'] = True |
---|
252 | Data['ring'] = [] |
---|
253 | Data['rings'] = [] |
---|
254 | Data['ellipses'] = [] |
---|
255 | Data['setDefault'] = False |
---|
256 | else: |
---|
257 | TextList.append([False,name,id]) |
---|
258 | id, cookie = G2frame.PatternTree.GetNextChild(G2frame.root, cookie) |
---|
259 | if len(TextList) == 1: |
---|
260 | G2frame.ErrorDialog('Nothing to copy controls to','There must be more than one "IMG" pattern') |
---|
261 | return |
---|
262 | dlg = G2frame.CopyDialog(G2frame,'Copy image controls','Copy controls from '+Source+' to:',TextList) |
---|
263 | try: |
---|
264 | if dlg.ShowModal() == wx.ID_OK: |
---|
265 | result = dlg.GetData() |
---|
266 | if result[0][0]: |
---|
267 | result = TextList[1:] |
---|
268 | for item in result: item[0] = True |
---|
269 | for i,item in enumerate(result): |
---|
270 | ifcopy,name,id = item |
---|
271 | if ifcopy: |
---|
272 | oldData = copy.deepcopy(G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Image Controls'))) |
---|
273 | Data['range'] = oldData['range'] |
---|
274 | Data['size'] = oldData['size'] |
---|
275 | Data['GonioAngles'] = oldData.get('GonioAngles', [0.,0.,0.]) |
---|
276 | Data['ring'] = [] |
---|
277 | Data['rings'] = [] |
---|
278 | Data['ellipses'] = [] |
---|
279 | G2frame.PatternTree.SetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Image Controls'),copy.deepcopy(Data)) |
---|
280 | finally: |
---|
281 | dlg.Destroy() |
---|
282 | G2frame.PatternTree.SelectItem(G2frame.PickId) |
---|
283 | |
---|
284 | def OnSaveControls(event): |
---|
285 | dlg = wx.FileDialog(G2frame, 'Choose image controls file', '.', '', |
---|
286 | 'image control files (*.imctrl)|*.imctrl',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT|wx.CHANGE_DIR) |
---|
287 | try: |
---|
288 | if dlg.ShowModal() == wx.ID_OK: |
---|
289 | filename = dlg.GetPath() |
---|
290 | # make sure extension is .imctrl |
---|
291 | filename = os.path.splitext(filename)[0]+'.imctrl' |
---|
292 | File = open(filename,'w') |
---|
293 | save = {} |
---|
294 | keys = ['type','wavelength','calibrant','distance','center', |
---|
295 | 'tilt','rotation','azmthOff','fullIntegrate','LRazimuth', |
---|
296 | 'IOtth','outChannels','outAzimuths','invert_x','invert_y','DetDepth', |
---|
297 | 'calibskip','pixLimit','cutoff','calibdmin','chisq', |
---|
298 | 'binType','SampleShape','PolaVal','SampleAbs','dark image','background image'] |
---|
299 | for key in keys: |
---|
300 | if key not in data: #uncalibrated! |
---|
301 | continue |
---|
302 | File.write(key+':'+str(data[key])+'\n') |
---|
303 | File.close() |
---|
304 | finally: |
---|
305 | dlg.Destroy() |
---|
306 | |
---|
307 | def OnLoadControls(event): |
---|
308 | cntlList = ['wavelength','distance','tilt','invert_x','invert_y','type', |
---|
309 | 'fullIntegrate','outChannels','outAzimuths','LRazimuth','IOtth','azmthOff','DetDepth', |
---|
310 | 'calibskip','pixLimit','cutoff','calibdmin','chisq', |
---|
311 | 'PolaVal','SampleAbs','dark image','background image'] |
---|
312 | dlg = wx.FileDialog(G2frame, 'Choose image controls file', '.', '', |
---|
313 | 'image control files (*.imctrl)|*.imctrl',wx.OPEN|wx.CHANGE_DIR) |
---|
314 | try: |
---|
315 | if dlg.ShowModal() == wx.ID_OK: |
---|
316 | filename = dlg.GetPath() |
---|
317 | File = open(filename,'r') |
---|
318 | save = {} |
---|
319 | S = File.readline() |
---|
320 | while S: |
---|
321 | if S[0] == '#': |
---|
322 | S = File.readline() |
---|
323 | continue |
---|
324 | [key,val] = S[:-1].split(':') |
---|
325 | if key in ['type','calibrant','binType','SampleShape',]: #strings |
---|
326 | save[key] = val |
---|
327 | elif key in ['rotation']: |
---|
328 | save[key] = float(val) |
---|
329 | elif key in ['center',]: |
---|
330 | if ',' in val: |
---|
331 | save[key] = eval(val) |
---|
332 | else: |
---|
333 | vals = val.strip('[] ').split() |
---|
334 | save[key] = [float(vals[0]),float(vals[1])] |
---|
335 | elif key in cntlList: |
---|
336 | save[key] = eval(val) |
---|
337 | S = File.readline() |
---|
338 | data.update(save) |
---|
339 | G2frame.PatternTree.SetItemPyData(G2gd.GetPatternTreeItemId(G2frame,G2frame.Image, 'Image Controls'),copy.deepcopy(data)) |
---|
340 | UpdateImageControls(G2frame,data,masks) |
---|
341 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
342 | |
---|
343 | File.close() |
---|
344 | finally: |
---|
345 | dlg.Destroy() |
---|
346 | |
---|
347 | # Sizers |
---|
348 | |
---|
349 | def ComboSizer(): |
---|
350 | |
---|
351 | def OnDataType(event): |
---|
352 | data['type'] = typeSel.GetValue()[:4] |
---|
353 | if 'SASD' in data['type']: |
---|
354 | data['SampleAbs'][0] = np.exp(-data['SampleAbs'][0]) #switch from muT to trans! |
---|
355 | if data['binType'] == '2-theta': data['binType'] = 'log(q)' #switch default bin type |
---|
356 | elif 'PWDR' in data['type']: |
---|
357 | data['SampleAbs'][0] = -np.log(data['SampleAbs'][0]) #switch from trans to muT! |
---|
358 | if data['binType'] == 'log(q)': data['binType'] = '2-theta' #switch default bin type |
---|
359 | wx.CallAfter(UpdateImageControls,G2frame,data,masks) |
---|
360 | |
---|
361 | def OnNewColorBar(event): |
---|
362 | data['color'] = colSel.GetValue() |
---|
363 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
364 | |
---|
365 | def OnAzmthOff(event): |
---|
366 | try: |
---|
367 | azmthoff = float(azmthOff.GetValue()) |
---|
368 | data['azmthOff'] = azmthoff |
---|
369 | except ValueError: |
---|
370 | pass |
---|
371 | azmthOff.SetValue("%.2f"%(data['azmthOff'])) #reset in case of error |
---|
372 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
373 | |
---|
374 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
375 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Type of image data: '),0,WACV) |
---|
376 | typeSel = wx.ComboBox(parent=G2frame.dataDisplay,value=typeDict[data['type']],choices=typeList, |
---|
377 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
378 | typeSel.SetValue(data['type']) |
---|
379 | typeSel.Bind(wx.EVT_COMBOBOX, OnDataType) |
---|
380 | comboSizer.Add(typeSel,0,WACV) |
---|
381 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Color bar '),0,WACV) |
---|
382 | colSel = wx.ComboBox(parent=G2frame.dataDisplay,value=data['color'],choices=colorList, |
---|
383 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
384 | colSel.Bind(wx.EVT_COMBOBOX, OnNewColorBar) |
---|
385 | comboSizer.Add(colSel,0,WACV) |
---|
386 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Azimuth offset '),0,WACV) |
---|
387 | azmthOff = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.2f" % (data['azmthOff'])), |
---|
388 | style=wx.TE_PROCESS_ENTER) |
---|
389 | azmthOff.Bind(wx.EVT_TEXT_ENTER,OnAzmthOff) |
---|
390 | azmthOff.Bind(wx.EVT_KILL_FOCUS,OnAzmthOff) |
---|
391 | comboSizer.Add(azmthOff,0,WACV) |
---|
392 | return comboSizer |
---|
393 | |
---|
394 | def MaxSizer(): |
---|
395 | |
---|
396 | def OnMaxVal(event): |
---|
397 | try: |
---|
398 | value = min(data['range'][0][1],int(maxVal.GetValue())) |
---|
399 | if value < data['range'][1][0]+1: |
---|
400 | raise ValueError |
---|
401 | data['range'][1][1] = value |
---|
402 | except ValueError: |
---|
403 | pass |
---|
404 | maxVal.SetValue('%.0f'%(data['range'][1][1])) |
---|
405 | DeltOne = data['range'][1][1]-max(0.0,data['range'][0][0]) |
---|
406 | sqrtDeltOne = math.sqrt(DeltOne) |
---|
407 | maxSel.SetValue(int(100*sqrtDeltOne/sqrtDeltZero)) |
---|
408 | minSel.SetValue(int(100*(data['range'][1][0]/DeltOne))) |
---|
409 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
410 | |
---|
411 | def OnMinVal(event): |
---|
412 | try: |
---|
413 | value = int(minVal.GetValue()) |
---|
414 | if value > data['range'][1][1]-1: |
---|
415 | raise ValueError |
---|
416 | data['range'][1][0] = value |
---|
417 | except ValueError: |
---|
418 | pass |
---|
419 | minVal.SetValue('%.0f'%(data['range'][1][0])) |
---|
420 | minSel.SetValue(int(100*(data['range'][1][0]-max(0.0,data['range'][0][0]))/DeltOne)) |
---|
421 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
422 | |
---|
423 | def OnMaxSlider(event): |
---|
424 | sqrtDeltZero = math.sqrt(data['range'][0][1]) |
---|
425 | imax = int(maxSel.GetValue())*sqrtDeltZero/100. |
---|
426 | data['range'][1][1] = imax**2 |
---|
427 | data['range'][1][0] = max(0.0,min(data['range'][1][1]-1,data['range'][1][0])) |
---|
428 | DeltOne = max(1.0,data['range'][1][1]-data['range'][1][0]) |
---|
429 | minSel.SetValue(int(100*(data['range'][1][0]/DeltOne))) |
---|
430 | maxVal.SetValue('%.0f'%(data['range'][1][1])) |
---|
431 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
432 | |
---|
433 | def OnMinSlider(event): |
---|
434 | DeltOne = data['range'][1][1]-data['range'][1][0] |
---|
435 | imin = int(minSel.GetValue())*DeltOne/100. |
---|
436 | data['range'][1][0] = max(0.0,min(data['range'][1][1]-1,imin)) |
---|
437 | minVal.SetValue('%.0f'%(data['range'][1][0])) |
---|
438 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
439 | |
---|
440 | maxSizer = wx.FlexGridSizer(0,3,0,5) |
---|
441 | maxSizer.AddGrowableCol(1,1) |
---|
442 | maxSizer.SetFlexibleDirection(wx.HORIZONTAL) |
---|
443 | sqrtDeltZero = math.sqrt(data['range'][0][1]-max(0.0,data['range'][0][0])) |
---|
444 | DeltOne = data['range'][1][1]-max(0.0,data['range'][0][0]) |
---|
445 | sqrtDeltOne = math.sqrt(DeltOne) |
---|
446 | maxSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Max intensity'),0,WACV) |
---|
447 | maxSel = wx.Slider(parent=G2frame.dataDisplay,style=wx.SL_HORIZONTAL, |
---|
448 | value=int(100*sqrtDeltOne/sqrtDeltZero)) |
---|
449 | maxSizer.Add(maxSel,1,wx.EXPAND) |
---|
450 | maxSel.Bind(wx.EVT_SLIDER, OnMaxSlider) |
---|
451 | maxVal = wx.TextCtrl(parent=G2frame.dataDisplay,value='%.0f'%(data['range'][1][1])) |
---|
452 | maxVal.Bind(wx.EVT_TEXT_ENTER,OnMaxVal) |
---|
453 | maxVal.Bind(wx.EVT_KILL_FOCUS,OnMaxVal) |
---|
454 | maxSizer.Add(maxVal,0,WACV) |
---|
455 | maxSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Min intensity'),0,WACV) |
---|
456 | minSel = wx.Slider(parent=G2frame.dataDisplay,style=wx.SL_HORIZONTAL, |
---|
457 | value=int(100*(data['range'][1][0]-max(0.0,data['range'][0][0]))/DeltOne)) |
---|
458 | maxSizer.Add(minSel,1,wx.EXPAND) |
---|
459 | minSel.Bind(wx.EVT_SLIDER, OnMinSlider) |
---|
460 | minVal = wx.TextCtrl(parent=G2frame.dataDisplay,value='%.0f'%(data['range'][1][0])) |
---|
461 | minVal.Bind(wx.EVT_TEXT_ENTER,OnMinVal) |
---|
462 | minVal.Bind(wx.EVT_KILL_FOCUS,OnMinVal) |
---|
463 | maxSizer.Add(minVal,0,WACV) |
---|
464 | return maxSizer |
---|
465 | |
---|
466 | def CalibCoeffSizer(): |
---|
467 | |
---|
468 | def OnWavelength(event): |
---|
469 | try: |
---|
470 | wave = float(waveSel.GetValue()) |
---|
471 | if wave < .01: |
---|
472 | raise ValueError |
---|
473 | data['wavelength'] = wave |
---|
474 | except ValueError: |
---|
475 | pass |
---|
476 | waveSel.SetValue("%7.5f" % (data['wavelength'])) #reset in case of error |
---|
477 | |
---|
478 | def OnDetDepthRef(event): |
---|
479 | data['DetDepthRef'] = penSel.GetValue() |
---|
480 | |
---|
481 | def OnDetDepth(event): |
---|
482 | try: |
---|
483 | data['DetDepth'] = float(penVal.GetValue()) |
---|
484 | except ValueError: |
---|
485 | pass |
---|
486 | penVal.SetValue("%6.3f" % (data['DetDepth'])) #reset in case of error |
---|
487 | |
---|
488 | calibSizer = wx.FlexGridSizer(0,2,5,5) |
---|
489 | calibSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Calibration coefficients'),0,WACV) |
---|
490 | calibSizer.Add((5,0),0) |
---|
491 | cent = data['center'] |
---|
492 | for axis,cnt in zip(['X','Y'],cent): |
---|
493 | calibSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Beam center '+axis),0,WACV) |
---|
494 | centText = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%7.2f" % (cnt)),style=wx.TE_READONLY) |
---|
495 | centText.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
496 | calibSizer.Add(centText,0,WACV) |
---|
497 | calibSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Wavelength'),0,WACV) |
---|
498 | waveSel = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%7.5f" % (data['wavelength'])), |
---|
499 | style=wx.TE_PROCESS_ENTER) |
---|
500 | waveSel.Bind(wx.EVT_TEXT_ENTER,OnWavelength) |
---|
501 | waveSel.Bind(wx.EVT_KILL_FOCUS,OnWavelength) |
---|
502 | calibSizer.Add(waveSel,0,WACV) |
---|
503 | calibSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Distance'),0,WACV) |
---|
504 | # dist = wx.CheckBox(G2frame.dataDisplay,label=' Distance') #possible future mod. to allow fix of calib. parms.? |
---|
505 | distSel = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%8.2f"%(data['distance'])),style=wx.TE_READONLY) |
---|
506 | distSel.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
507 | calibSizer.Add(distSel,0,WACV) |
---|
508 | calibSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Tilt angle'),0,WACV) |
---|
509 | tiltSel = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%9.3f"%(data['tilt'])),style=wx.TE_READONLY) |
---|
510 | tiltSel.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
511 | calibSizer.Add(tiltSel,0,WACV) |
---|
512 | calibSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Tilt rotation'),0,WACV) |
---|
513 | rotSel = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%9.3f"%(data['rotation'])),style=wx.TE_READONLY) |
---|
514 | rotSel.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
515 | calibSizer.Add(rotSel,0,WACV) |
---|
516 | if 'PWDR' in data['type']: |
---|
517 | penSel = wx.CheckBox(parent=G2frame.dataDisplay,label='Penetration?') |
---|
518 | calibSizer.Add(penSel,0,WACV) |
---|
519 | penSel.Bind(wx.EVT_CHECKBOX, OnDetDepthRef) |
---|
520 | penSel.SetValue(data['DetDepthRef']) |
---|
521 | penVal = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%6.5f" % (data['DetDepth'])), |
---|
522 | style=wx.TE_PROCESS_ENTER) |
---|
523 | penVal.Bind(wx.EVT_TEXT_ENTER,OnDetDepth) |
---|
524 | penVal.Bind(wx.EVT_KILL_FOCUS,OnDetDepth) |
---|
525 | calibSizer.Add(penVal,0,WACV) |
---|
526 | |
---|
527 | return calibSizer |
---|
528 | |
---|
529 | def IntegrateSizer(): |
---|
530 | |
---|
531 | def OnNewBinType(event): |
---|
532 | data['binType'] = binSel.GetValue() |
---|
533 | |
---|
534 | def OnIOtth(event): |
---|
535 | Ltth = max(float(G2frame.InnerTth.GetValue()),0.001) |
---|
536 | Utth = float(G2frame.OuterTth.GetValue()) |
---|
537 | if Ltth > Utth: |
---|
538 | Ltth,Utth = Utth,Ltth |
---|
539 | data['IOtth'] = [Ltth,Utth] |
---|
540 | G2frame.InnerTth.SetValue("%8.3f" % (Ltth)) |
---|
541 | G2frame.OuterTth.SetValue("%8.2f" % (Utth)) |
---|
542 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
543 | |
---|
544 | def OnLRazim(event): |
---|
545 | Lazm = int(G2frame.Lazim.GetValue())%360 |
---|
546 | Razm = int(G2frame.Razim.GetValue())%360 |
---|
547 | if Lazm > Razm: |
---|
548 | Razm += 360 |
---|
549 | if data['fullIntegrate']: |
---|
550 | Razm = Lazm+360 |
---|
551 | G2frame.Lazim.SetValue("%6d" % (Lazm)) |
---|
552 | G2frame.Razim.SetValue("%6d" % (Razm)) |
---|
553 | data['LRazimuth'] = [Lazm,Razm] |
---|
554 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
555 | |
---|
556 | def OnNumOutChans(event): |
---|
557 | try: |
---|
558 | numChans = int(outChan.GetValue()) |
---|
559 | if numChans < 10: |
---|
560 | raise ValueError |
---|
561 | data['outChannels'] = numChans |
---|
562 | except ValueError: |
---|
563 | pass |
---|
564 | outChan.SetValue(str(data['outChannels'])) #reset in case of error |
---|
565 | |
---|
566 | def OnNumOutAzms(event): |
---|
567 | try: |
---|
568 | numAzms = int(outAzim.GetValue()) |
---|
569 | if numAzms < 1: |
---|
570 | raise ValueError |
---|
571 | data['outAzimuths'] = numAzms |
---|
572 | except ValueError: |
---|
573 | pass |
---|
574 | outAzim.SetValue(str(data['outAzimuths'])) #reset in case of error |
---|
575 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
576 | |
---|
577 | def OnOblique(event): |
---|
578 | if data['Oblique'][1]: |
---|
579 | data['Oblique'][1] = False |
---|
580 | else: |
---|
581 | data['Oblique'][1] = True |
---|
582 | |
---|
583 | def OnObliqVal(event): |
---|
584 | try: |
---|
585 | value = float(obliqVal.GetValue()) |
---|
586 | if 0.01 <= value <= 0.99: |
---|
587 | data['Oblique'][0] = value |
---|
588 | else: |
---|
589 | raise ValueError |
---|
590 | except ValueError: |
---|
591 | pass |
---|
592 | obliqVal.SetValue('%.3f'%(data['Oblique'][0])) |
---|
593 | |
---|
594 | def OnSamAbs(event): |
---|
595 | if data['SampleAbs'][1]: |
---|
596 | data['SampleAbs'][1] = False |
---|
597 | else: |
---|
598 | data['SampleAbs'][1] = True |
---|
599 | |
---|
600 | def OnSamAbsVal(event): |
---|
601 | try: |
---|
602 | value = float(samabsVal.GetValue()) |
---|
603 | minmax = [0.,2.] |
---|
604 | if 'SASD' in data['type']: |
---|
605 | minmax = [.05,1.0] |
---|
606 | if minmax[0] <= value <= minmax[1]: |
---|
607 | data['SampleAbs'][0] = value |
---|
608 | else: |
---|
609 | raise ValueError |
---|
610 | except ValueError: |
---|
611 | pass |
---|
612 | samabsVal.SetValue('%.3f'%(data['SampleAbs'][0])) |
---|
613 | |
---|
614 | def OnShowLines(event): |
---|
615 | if data['showLines']: |
---|
616 | data['showLines'] = False |
---|
617 | else: |
---|
618 | data['showLines'] = True |
---|
619 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
620 | |
---|
621 | def OnFullIntegrate(event): |
---|
622 | Lazm =int(G2frame.Lazim.GetValue()) |
---|
623 | if data['fullIntegrate']: |
---|
624 | data['fullIntegrate'] = False |
---|
625 | data['LRazimuth'] = [Lazm,Lazm+20] |
---|
626 | else: |
---|
627 | data['fullIntegrate'] = True |
---|
628 | data['LRazimuth'] = [Lazm,Lazm+360] |
---|
629 | UpdateImageControls(G2frame,data,masks) |
---|
630 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
631 | |
---|
632 | def OnSetDefault(event): |
---|
633 | if data['setDefault']: |
---|
634 | G2frame.imageDefault = {} |
---|
635 | data['setDefault'] = False |
---|
636 | else: |
---|
637 | G2frame.imageDefault = copy.copy(data) |
---|
638 | data['setDefault'] = True |
---|
639 | |
---|
640 | def OnCenterAzm(event): |
---|
641 | if data['centerAzm']: |
---|
642 | data['centerAzm'] = False |
---|
643 | else: |
---|
644 | data['centerAzm'] = True |
---|
645 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
646 | |
---|
647 | def OnApplyPola(event): |
---|
648 | if data['PolaVal'][1]: |
---|
649 | data['PolaVal'][1] = False |
---|
650 | else: |
---|
651 | data['PolaVal'][1] = True |
---|
652 | |
---|
653 | def OnPolaVal(event): |
---|
654 | try: |
---|
655 | value = float(polaVal.GetValue()) |
---|
656 | if 0.001 <= value <= 0.999: |
---|
657 | data['PolaVal'][0] = value |
---|
658 | else: |
---|
659 | raise ValueError |
---|
660 | except ValueError: |
---|
661 | pass |
---|
662 | polaVal.SetValue('%.3f'%(data['PolaVal'][0])) |
---|
663 | |
---|
664 | dataSizer = wx.FlexGridSizer(0,2,5,3) |
---|
665 | dataSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Integration coefficients'),0,WACV) |
---|
666 | dataSizer.Add((5,0),0) |
---|
667 | if 'PWDR' in data['type']: |
---|
668 | binChoice = ['2-theta','q'] |
---|
669 | elif 'SASD' in data['type']: |
---|
670 | binChoice = ['q','log(q)'] |
---|
671 | dataSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Bin style: Constant step bins in'),0,WACV) |
---|
672 | binSel = wx.ComboBox(parent=G2frame.dataDisplay,value=data['binType'],choices=binChoice, |
---|
673 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
674 | binSel.Bind(wx.EVT_COMBOBOX, OnNewBinType) |
---|
675 | dataSizer.Add(binSel,0,WACV) |
---|
676 | dataSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Inner/Outer 2-theta'),0,WACV) |
---|
677 | IOtth = data['IOtth'] |
---|
678 | littleSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
679 | G2frame.InnerTth = wx.TextCtrl(parent=G2frame.dataDisplay, |
---|
680 | value=("%8.3f" % (IOtth[0])),style=wx.TE_PROCESS_ENTER) |
---|
681 | G2frame.InnerTth.Bind(wx.EVT_TEXT_ENTER,OnIOtth) |
---|
682 | G2frame.InnerTth.Bind(wx.EVT_KILL_FOCUS,OnIOtth) |
---|
683 | littleSizer.Add(G2frame.InnerTth,0,WACV) |
---|
684 | G2frame.OuterTth = wx.TextCtrl(parent=G2frame.dataDisplay, |
---|
685 | value=("%8.2f" % (IOtth[1])),style=wx.TE_PROCESS_ENTER) |
---|
686 | G2frame.OuterTth.Bind(wx.EVT_TEXT_ENTER,OnIOtth) |
---|
687 | G2frame.OuterTth.Bind(wx.EVT_KILL_FOCUS,OnIOtth) |
---|
688 | littleSizer.Add(G2frame.OuterTth,0,WACV) |
---|
689 | dataSizer.Add(littleSizer,0,) |
---|
690 | dataSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Start/End azimuth'),0,WACV) |
---|
691 | LRazim = data['LRazimuth'] |
---|
692 | littleSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
693 | G2frame.Lazim = wx.TextCtrl(parent=G2frame.dataDisplay, |
---|
694 | value=("%6d" % (LRazim[0])),style=wx.TE_PROCESS_ENTER) |
---|
695 | G2frame.Lazim.Bind(wx.EVT_TEXT_ENTER,OnLRazim) |
---|
696 | G2frame.Lazim.Bind(wx.EVT_KILL_FOCUS,OnLRazim) |
---|
697 | littleSizer.Add(G2frame.Lazim,0,WACV) |
---|
698 | G2frame.Razim = wx.TextCtrl(parent=G2frame.dataDisplay, |
---|
699 | value=("%6d" % (LRazim[1])),style=wx.TE_PROCESS_ENTER) |
---|
700 | G2frame.Razim.Bind(wx.EVT_TEXT_ENTER,OnLRazim) |
---|
701 | G2frame.Razim.Bind(wx.EVT_KILL_FOCUS,OnLRazim) |
---|
702 | if data['fullIntegrate']: |
---|
703 | G2frame.Razim.Enable(False) |
---|
704 | G2frame.Razim.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
705 | G2frame.Razim.SetValue("%6d" % (LRazim[0]+360)) |
---|
706 | littleSizer.Add(G2frame.Razim,0,WACV) |
---|
707 | dataSizer.Add(littleSizer,0,) |
---|
708 | dataSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' No. 2-theta/azimuth bins'),0,WACV) |
---|
709 | littleSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
710 | outChan = wx.TextCtrl(parent=G2frame.dataDisplay,value=str(data['outChannels']),style=wx.TE_PROCESS_ENTER) |
---|
711 | outChan.Bind(wx.EVT_TEXT_ENTER,OnNumOutChans) |
---|
712 | outChan.Bind(wx.EVT_KILL_FOCUS,OnNumOutChans) |
---|
713 | littleSizer.Add(outChan,0,WACV) |
---|
714 | outAzim = wx.TextCtrl(parent=G2frame.dataDisplay,value=str(data['outAzimuths']),style=wx.TE_PROCESS_ENTER) |
---|
715 | outAzim.Bind(wx.EVT_TEXT_ENTER,OnNumOutAzms) |
---|
716 | outAzim.Bind(wx.EVT_KILL_FOCUS,OnNumOutAzms) |
---|
717 | littleSizer.Add(outAzim,0,WACV) |
---|
718 | dataSizer.Add(littleSizer,0,) |
---|
719 | littleSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
720 | samabs = wx.CheckBox(parent=G2frame.dataDisplay,label='Apply sample absorption?') |
---|
721 | dataSizer.Add(samabs,0,WACV) |
---|
722 | samabs.Bind(wx.EVT_CHECKBOX, OnSamAbs) |
---|
723 | samabs.SetValue(data['SampleAbs'][1]) |
---|
724 | if 'PWDR' in data['type']: |
---|
725 | littleSizer.Add(wx.StaticText(G2frame.dataDisplay,label='mu/R (0.00-2.0) '),0,WACV) |
---|
726 | elif 'SASD' in data['type']: |
---|
727 | littleSizer.Add(wx.StaticText(G2frame.dataDisplay,label='transmission '),0,WACV) |
---|
728 | samabsVal = wx.TextCtrl(parent=G2frame.dataDisplay,value='%.3f'%(data['SampleAbs'][0]),style=wx.TE_PROCESS_ENTER) |
---|
729 | samabsVal.Bind(wx.EVT_TEXT_ENTER,OnSamAbsVal) |
---|
730 | samabsVal.Bind(wx.EVT_KILL_FOCUS,OnSamAbsVal) |
---|
731 | littleSizer.Add(samabsVal,0,WACV) |
---|
732 | dataSizer.Add(littleSizer,0,) |
---|
733 | if 'PWDR' in data['type']: |
---|
734 | littleSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
735 | oblique = wx.CheckBox(parent=G2frame.dataDisplay,label='Apply detector absorption?') |
---|
736 | dataSizer.Add(oblique,0,WACV) |
---|
737 | oblique.Bind(wx.EVT_CHECKBOX, OnOblique) |
---|
738 | oblique.SetValue(data['Oblique'][1]) |
---|
739 | littleSizer.Add(wx.StaticText(G2frame.dataDisplay,label='Value (0.01-0.99) '),0,WACV) |
---|
740 | obliqVal = wx.TextCtrl(parent=G2frame.dataDisplay,value='%.3f'%(data['Oblique'][0]),style=wx.TE_PROCESS_ENTER) |
---|
741 | obliqVal.Bind(wx.EVT_TEXT_ENTER,OnObliqVal) |
---|
742 | obliqVal.Bind(wx.EVT_KILL_FOCUS,OnObliqVal) |
---|
743 | littleSizer.Add(obliqVal,0,WACV) |
---|
744 | dataSizer.Add(littleSizer,0,) |
---|
745 | if 'SASD' in data['type']: |
---|
746 | littleSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
747 | setPolariz = wx.CheckBox(parent=G2frame.dataDisplay,label='Apply polarization?') |
---|
748 | dataSizer.Add(setPolariz,0,WACV) |
---|
749 | setPolariz.Bind(wx.EVT_CHECKBOX, OnApplyPola) |
---|
750 | setPolariz.SetValue(data['PolaVal'][1]) |
---|
751 | littleSizer.Add(wx.StaticText(G2frame.dataDisplay,label='Value (0.001-0.999) '),0,WACV) |
---|
752 | polaVal = wx.TextCtrl(parent=G2frame.dataDisplay,value='%.3f'%(data['PolaVal'][0]), |
---|
753 | style=wx.TE_PROCESS_ENTER) |
---|
754 | polaVal.Bind(wx.EVT_TEXT_ENTER,OnPolaVal) |
---|
755 | polaVal.Bind(wx.EVT_KILL_FOCUS,OnPolaVal) |
---|
756 | littleSizer.Add(polaVal,0,WACV) |
---|
757 | dataSizer.Add(littleSizer,0,) |
---|
758 | |
---|
759 | showLines = wx.CheckBox(parent=G2frame.dataDisplay,label='Show integration limits?') |
---|
760 | dataSizer.Add(showLines,0,WACV) |
---|
761 | showLines.Bind(wx.EVT_CHECKBOX, OnShowLines) |
---|
762 | showLines.SetValue(data['showLines']) |
---|
763 | fullIntegrate = wx.CheckBox(parent=G2frame.dataDisplay,label='Do full integration?') |
---|
764 | dataSizer.Add(fullIntegrate,0,WACV) |
---|
765 | fullIntegrate.Bind(wx.EVT_CHECKBOX, OnFullIntegrate) |
---|
766 | fullIntegrate.SetValue(data['fullIntegrate']) |
---|
767 | setDefault = wx.CheckBox(parent=G2frame.dataDisplay,label='Use as default for all images?') |
---|
768 | dataSizer.Add(setDefault,0,WACV) |
---|
769 | setDefault.Bind(wx.EVT_CHECKBOX, OnSetDefault) |
---|
770 | setDefault.SetValue(data['setDefault']) |
---|
771 | centerAzm = wx.CheckBox(parent=G2frame.dataDisplay,label='Azimuth at bin center?') |
---|
772 | dataSizer.Add(centerAzm,0,WACV) |
---|
773 | centerAzm.Bind(wx.EVT_CHECKBOX, OnCenterAzm) |
---|
774 | centerAzm.SetValue(data['centerAzm']) |
---|
775 | return dataSizer |
---|
776 | |
---|
777 | def BackSizer(): |
---|
778 | |
---|
779 | def OnBackImage(event): |
---|
780 | data['background image'][0] = backImage.GetValue() |
---|
781 | |
---|
782 | def OnDarkImage(event): |
---|
783 | data['dark image'][0] = darkImage.GetValue() |
---|
784 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
785 | |
---|
786 | def OnBackMult(event): |
---|
787 | try: |
---|
788 | mult = float(backMult.GetValue()) |
---|
789 | data['background image'][1] = mult |
---|
790 | except ValueError: |
---|
791 | pass |
---|
792 | backMult.SetValue("%.3f" % (data['background image'][1])) #reset in case of error |
---|
793 | |
---|
794 | def OnDarkMult(event): |
---|
795 | try: |
---|
796 | mult = float(darkMult.GetValue()) |
---|
797 | data['dark image'][1] = mult |
---|
798 | except ValueError: |
---|
799 | pass |
---|
800 | darkMult.SetValue("%.3f" % (data['dark image'][1])) #reset in case of error |
---|
801 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
802 | |
---|
803 | backSizer = wx.FlexGridSizer(0,4,5,5) |
---|
804 | |
---|
805 | backSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,' Dark image'),0,WACV) |
---|
806 | Choices = ['',]+G2gd.GetPatternTreeDataNames(G2frame,['IMG ',]) |
---|
807 | darkImage = wx.ComboBox(parent=G2frame.dataDisplay,value=data['dark image'][0],choices=Choices, |
---|
808 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
809 | darkImage.Bind(wx.EVT_COMBOBOX,OnDarkImage) |
---|
810 | backSizer.Add(darkImage) |
---|
811 | backSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,' multiplier'),0,WACV) |
---|
812 | darkMult = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.3f" % (data['dark image'][1])), |
---|
813 | style=wx.TE_PROCESS_ENTER) |
---|
814 | darkMult.Bind(wx.EVT_TEXT_ENTER,OnDarkMult) |
---|
815 | darkMult.Bind(wx.EVT_KILL_FOCUS,OnDarkMult) |
---|
816 | backSizer.Add(darkMult,0,WACV) |
---|
817 | |
---|
818 | backSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,' Background image'),0,WACV) |
---|
819 | Choices = ['',]+G2gd.GetPatternTreeDataNames(G2frame,['IMG ',]) |
---|
820 | backImage = wx.ComboBox(parent=G2frame.dataDisplay,value=data['background image'][0],choices=Choices, |
---|
821 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
822 | backImage.Bind(wx.EVT_COMBOBOX,OnBackImage) |
---|
823 | backSizer.Add(backImage) |
---|
824 | backSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,' multiplier'),0,WACV) |
---|
825 | backMult = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.3f" % (data['background image'][1])), |
---|
826 | style=wx.TE_PROCESS_ENTER) |
---|
827 | backMult.Bind(wx.EVT_TEXT_ENTER,OnBackMult) |
---|
828 | backMult.Bind(wx.EVT_KILL_FOCUS,OnBackMult) |
---|
829 | backSizer.Add(backMult,0,WACV) |
---|
830 | return backSizer |
---|
831 | |
---|
832 | def CalibSizer(): |
---|
833 | |
---|
834 | def OnNewCalibrant(event): |
---|
835 | data['calibrant'] = calSel.GetValue() |
---|
836 | data['calibskip'] = calFile.Calibrants[data['calibrant']][3] |
---|
837 | limits = calFile.Calibrants[data['calibrant']][4] |
---|
838 | data['calibdmin'],data['pixLimit'],data['cutoff'] = limits |
---|
839 | pixLimit.SetValue(str(limits[1])) |
---|
840 | cutOff.SetValue('%.1f'%(limits[2])) |
---|
841 | calibSkip.SetValue(str(data['calibskip'])) |
---|
842 | calibDmin.SetValue('%.1f'%(limits[0])) |
---|
843 | |
---|
844 | def OnCalibSkip(event): |
---|
845 | data['calibskip'] = int(calibSkip.GetValue()) |
---|
846 | |
---|
847 | def OnCalibDmin(event): |
---|
848 | try: |
---|
849 | dmin = float(calibDmin.GetValue()) |
---|
850 | if dmin < 0.25: |
---|
851 | raise ValueError |
---|
852 | data['calibdmin'] = dmin |
---|
853 | except ValueError: |
---|
854 | pass |
---|
855 | calibDmin.SetValue("%.2f"%(data['calibdmin'])) #reset in case of error |
---|
856 | |
---|
857 | def OnCutOff(event): |
---|
858 | try: |
---|
859 | cutoff = float(cutOff.GetValue()) |
---|
860 | if cutoff < 0.1: |
---|
861 | raise ValueError |
---|
862 | data['cutoff'] = cutoff |
---|
863 | except ValueError: |
---|
864 | pass |
---|
865 | cutOff.SetValue("%.1f"%(data['cutoff'])) #reset in case of error |
---|
866 | |
---|
867 | def OnPixLimit(event): |
---|
868 | data['pixLimit'] = int(pixLimit.GetValue()) |
---|
869 | |
---|
870 | def OnSetRings(event): |
---|
871 | if data['setRings']: |
---|
872 | data['setRings'] = False |
---|
873 | else: |
---|
874 | data['setRings'] = True |
---|
875 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
876 | |
---|
877 | calibSizer = wx.FlexGridSizer(0,3,5,5) |
---|
878 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
879 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Calibrant '),0,WACV) |
---|
880 | calSel = wx.ComboBox(parent=G2frame.dataDisplay,value=data['calibrant'],choices=calList, |
---|
881 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
882 | calSel.Bind(wx.EVT_COMBOBOX, OnNewCalibrant) |
---|
883 | comboSizer.Add(calSel,0,WACV) |
---|
884 | calibSizer.Add(comboSizer,0) |
---|
885 | |
---|
886 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
887 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Calib lines to skip '),0,WACV) |
---|
888 | calibSkip = wx.ComboBox(parent=G2frame.dataDisplay,value=str(data['calibskip']),choices=[str(i) for i in range(25)], |
---|
889 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
890 | calibSkip.Bind(wx.EVT_COMBOBOX, OnCalibSkip) |
---|
891 | comboSizer.Add(calibSkip,0,WACV) |
---|
892 | calibSizer.Add(comboSizer,0) |
---|
893 | |
---|
894 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
895 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Min calib d-spacing '),0,WACV) |
---|
896 | calibDmin = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.2f" % (data['calibdmin'])), |
---|
897 | style=wx.TE_PROCESS_ENTER) |
---|
898 | calibDmin.Bind(wx.EVT_TEXT_ENTER,OnCalibDmin) |
---|
899 | calibDmin.Bind(wx.EVT_KILL_FOCUS,OnCalibDmin) |
---|
900 | comboSizer.Add(calibDmin,0,WACV) |
---|
901 | calibSizer.Add(comboSizer,0) |
---|
902 | |
---|
903 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
904 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Min ring I/Ib '),0,WACV) |
---|
905 | cutOff = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.1f" % (data['cutoff'])), |
---|
906 | style=wx.TE_PROCESS_ENTER) |
---|
907 | cutOff.Bind(wx.EVT_TEXT_ENTER,OnCutOff) |
---|
908 | cutOff.Bind(wx.EVT_KILL_FOCUS,OnCutOff) |
---|
909 | comboSizer.Add(cutOff,0,WACV) |
---|
910 | calibSizer.Add(comboSizer,0) |
---|
911 | |
---|
912 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
913 | comboSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Pixel search range '),0,WACV) |
---|
914 | pixLimit = wx.ComboBox(parent=G2frame.dataDisplay,value=str(data['pixLimit']),choices=['1','2','5','10','15','20'], |
---|
915 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
916 | pixLimit.Bind(wx.EVT_COMBOBOX, OnPixLimit) |
---|
917 | comboSizer.Add(pixLimit,0,WACV) |
---|
918 | calibSizer.Add(comboSizer,0) |
---|
919 | |
---|
920 | comboSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
921 | setRings = wx.CheckBox(parent=G2frame.dataDisplay,label='Show ring picks?') |
---|
922 | comboSizer.Add(setRings,0) |
---|
923 | setRings.Bind(wx.EVT_CHECKBOX, OnSetRings) |
---|
924 | setRings.SetValue(data['setRings']) |
---|
925 | calibSizer.Add(comboSizer,0) |
---|
926 | return calibSizer |
---|
927 | |
---|
928 | def GonioSizer(): |
---|
929 | |
---|
930 | ValObj = {} |
---|
931 | |
---|
932 | def OnGonioAngle(event): |
---|
933 | Obj = event.GetEventObject() |
---|
934 | item = ValObj[Obj.GetId()] |
---|
935 | try: |
---|
936 | value = float(Obj.GetValue()) |
---|
937 | except ValueError: |
---|
938 | value = data['GonioAngles'][item] |
---|
939 | data['GonioAngles'][item] = value |
---|
940 | Obj.SetValue('%8.2f'%(value)) |
---|
941 | |
---|
942 | gonioSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
943 | names = ['Omega','Chi','Phi'] |
---|
944 | gonioSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,'Sample goniometer angles: '),0,WACV) |
---|
945 | for i,name in enumerate(names): |
---|
946 | gonioSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,name),0,WACV) |
---|
947 | angle = wx.TextCtrl(G2frame.dataDisplay,-1,value='%8.2f'%(data['GonioAngles'][i]), |
---|
948 | style=wx.TE_PROCESS_ENTER) |
---|
949 | angle.Bind(wx.EVT_TEXT_ENTER,OnGonioAngle) |
---|
950 | angle.Bind(wx.EVT_KILL_FOCUS,OnGonioAngle) |
---|
951 | ValObj[angle.GetId()] = i |
---|
952 | gonioSizer.Add(angle,0,WACV) |
---|
953 | return gonioSizer |
---|
954 | |
---|
955 | # Image Controls main code |
---|
956 | |
---|
957 | #fix for old files: |
---|
958 | if 'azmthOff' not in data: |
---|
959 | data['azmthOff'] = 0.0 |
---|
960 | if 'background image' not in data: |
---|
961 | data['background image'] = ['',-1.0] |
---|
962 | if 'dark image' not in data: |
---|
963 | data['dark image'] = ['',-1.0] |
---|
964 | if 'centerAzm' not in data: |
---|
965 | data['centerAzm'] = False |
---|
966 | if 'Oblique' not in data: |
---|
967 | data['Oblique'] = [0.5,False] |
---|
968 | if 'PolaVal' not in data: |
---|
969 | data['PolaVal'] = [0.99,False] |
---|
970 | #end fix |
---|
971 | |
---|
972 | colorList = sorted([m for m in mpl.cm.datad.keys() if not m.endswith("_r")],key=lambda s: s.lower()) |
---|
973 | calList = sorted([m for m in calFile.Calibrants.keys()],key=lambda s: s.lower()) |
---|
974 | typeList = ['PWDR - powder diffraction data','SASD - small angle scattering data', |
---|
975 | 'REFL - reflectometry data'] |
---|
976 | if not data.get('type'): #patch for old project files |
---|
977 | data['type'] = 'PWDR' |
---|
978 | typeDict = {'PWDR':typeList[0],'SASD':typeList[1],'REFL':typeList[2]} |
---|
979 | if G2frame.dataDisplay: |
---|
980 | G2frame.dataDisplay.Destroy() |
---|
981 | G2gd.SetDataMenuBar(G2frame,G2frame.dataFrame.ImageMenu) |
---|
982 | if not G2frame.dataFrame.GetStatusBar(): |
---|
983 | G2frame.dataFrame.CreateStatusBar() |
---|
984 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnCalibrate, id=G2gd.wxID_IMCALIBRATE) |
---|
985 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnRecalibrate, id=G2gd.wxID_IMRECALIBRATE) |
---|
986 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnClearCalib, id=G2gd.wxID_IMCLEARCALIB) |
---|
987 | if 'chisq' not in data: |
---|
988 | G2frame.dataFrame.ImageEdit.Enable(id=G2gd.wxID_IMRECALIBRATE,enable=False) |
---|
989 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnIntegrate, id=G2gd.wxID_IMINTEGRATE) |
---|
990 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnIntegrateAll, id=G2gd.wxID_INTEGRATEALL) |
---|
991 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnCopyControls, id=G2gd.wxID_IMCOPYCONTROLS) |
---|
992 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnSaveControls, id=G2gd.wxID_IMSAVECONTROLS) |
---|
993 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnLoadControls, id=G2gd.wxID_IMLOADCONTROLS) |
---|
994 | G2frame.dataDisplay = wx.Panel(G2frame.dataFrame) |
---|
995 | |
---|
996 | mainSizer = wx.BoxSizer(wx.VERTICAL) |
---|
997 | mainSizer.Add((5,10),0) |
---|
998 | mainSizer.Add(ComboSizer(),0,wx.ALIGN_LEFT) |
---|
999 | mainSizer.Add((5,5),0) |
---|
1000 | mainSizer.Add(MaxSizer(),0,wx.ALIGN_LEFT|wx.EXPAND) |
---|
1001 | |
---|
1002 | mainSizer.Add((5,5),0) |
---|
1003 | DataSizer = wx.FlexGridSizer(0,2,5,5) |
---|
1004 | DataSizer.Add(CalibCoeffSizer(),0) |
---|
1005 | DataSizer.Add(IntegrateSizer(),0) |
---|
1006 | mainSizer.Add(DataSizer,0) |
---|
1007 | mainSizer.Add((5,5),0) |
---|
1008 | mainSizer.Add(BackSizer(),0) |
---|
1009 | mainSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Calibration controls:'),0,WACV) |
---|
1010 | mainSizer.Add((5,5),0) |
---|
1011 | mainSizer.Add(CalibSizer(),0,WACV) |
---|
1012 | mainSizer.Add((5,5),0) |
---|
1013 | mainSizer.Add(GonioSizer(),0,WACV) |
---|
1014 | |
---|
1015 | mainSizer.Layout() |
---|
1016 | G2frame.dataDisplay.SetSizer(mainSizer) |
---|
1017 | fitSize = mainSizer.Fit(G2frame.dataFrame) |
---|
1018 | G2frame.dataFrame.setSizePosLeft(fitSize) |
---|
1019 | G2frame.dataDisplay.SetSize(fitSize) |
---|
1020 | |
---|
1021 | ################################################################################ |
---|
1022 | ##### Masks |
---|
1023 | ################################################################################ |
---|
1024 | |
---|
1025 | def UpdateMasks(G2frame,data): |
---|
1026 | '''Shows and handles the controls on the "Masks" data tree entry |
---|
1027 | ''' |
---|
1028 | |
---|
1029 | def OnTextMsg(event): |
---|
1030 | Obj = event.GetEventObject() |
---|
1031 | Obj.SetToolTipString('Drag this mask on 2D Powder Image with mouse to change ') |
---|
1032 | |
---|
1033 | def Replot(*args,**kwargs): |
---|
1034 | G2plt.PlotExposedImage(G2frame) |
---|
1035 | |
---|
1036 | def onDeleteMask(event): |
---|
1037 | Obj = event.GetEventObject() |
---|
1038 | typ = Obj.locationcode.split('+')[1] |
---|
1039 | num = int(Obj.locationcode.split('+')[2]) |
---|
1040 | del(data[typ][num]) |
---|
1041 | wx.CallAfter(UpdateMasks,G2frame,data) |
---|
1042 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1043 | |
---|
1044 | def onDeleteFrame(event): |
---|
1045 | data['Frames'] = [] |
---|
1046 | wx.CallAfter(UpdateMasks,G2frame,data) |
---|
1047 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1048 | |
---|
1049 | def OnCopyMask(event): |
---|
1050 | TextList = [[False,'All IMG',0]] |
---|
1051 | Names = [] |
---|
1052 | if G2frame.PatternTree.GetCount(): |
---|
1053 | id, cookie = G2frame.PatternTree.GetFirstChild(G2frame.root) |
---|
1054 | while id: |
---|
1055 | name = G2frame.PatternTree.GetItemText(id) |
---|
1056 | Names.append(name) |
---|
1057 | if 'IMG' in name: |
---|
1058 | if id == G2frame.Image: |
---|
1059 | Source = name |
---|
1060 | Mask = copy.deepcopy(G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Masks'))) |
---|
1061 | Thresh = Mask.pop('Thresholds') #remove Thresholds from source mask & save it for later |
---|
1062 | else: |
---|
1063 | TextList.append([False,name,id]) |
---|
1064 | id, cookie = G2frame.PatternTree.GetNextChild(G2frame.root, cookie) |
---|
1065 | if len(TextList) == 1: |
---|
1066 | G2frame.ErrorDialog('Nothing to copy mask to','There must be more than one "IMG" pattern') |
---|
1067 | return |
---|
1068 | dlg = G2frame.CopyDialog(G2frame,'Copy mask information','Copy mask from '+Source+' to:',TextList) |
---|
1069 | try: |
---|
1070 | if dlg.ShowModal() == wx.ID_OK: |
---|
1071 | result = dlg.GetData() |
---|
1072 | if result[0][0]: |
---|
1073 | result = TextList[1:] |
---|
1074 | for item in result: item[0] = True |
---|
1075 | for i,item in enumerate(result): |
---|
1076 | ifcopy,name,id = item |
---|
1077 | if ifcopy: |
---|
1078 | mask = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Masks')) |
---|
1079 | # Mask['Thresholds'][0] = mask['Thresholds'][0] |
---|
1080 | # Mask['Thresholds'][1][1] = min(mask['Thresholds'][1][1],Mask['Thresholds'][1][1]) |
---|
1081 | mask.update(Mask) |
---|
1082 | mask['Thresholds'][1][0] = Thresh[1][0] #copy only lower threshold |
---|
1083 | G2frame.PatternTree.SetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Masks'),copy.deepcopy(mask)) |
---|
1084 | finally: |
---|
1085 | dlg.Destroy() |
---|
1086 | |
---|
1087 | def OnSaveMask(event): |
---|
1088 | dlg = wx.FileDialog(G2frame, 'Choose image mask file', '.', '', |
---|
1089 | 'image mask files (*.immask)|*.immask',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT|wx.CHANGE_DIR) |
---|
1090 | try: |
---|
1091 | if dlg.ShowModal() == wx.ID_OK: |
---|
1092 | filename = dlg.GetPath() |
---|
1093 | File = open(filename,'w') |
---|
1094 | save = {} |
---|
1095 | keys = ['Points','Rings','Arcs','Polygons','Frames','Thresholds'] |
---|
1096 | for key in keys: |
---|
1097 | File.write(key+':'+str(data[key])+'\n') |
---|
1098 | File.close() |
---|
1099 | finally: |
---|
1100 | dlg.Destroy() |
---|
1101 | |
---|
1102 | def OnLoadMask(event): |
---|
1103 | if event.Id == G2gd.wxID_MASKLOADNOT: |
---|
1104 | ignoreThreshold = True |
---|
1105 | else: |
---|
1106 | ignoreThreshold = False |
---|
1107 | dlg = wx.FileDialog(G2frame, 'Choose image mask file', '.', '', |
---|
1108 | 'image mask files (*.immask)|*.immask',wx.OPEN|wx.CHANGE_DIR) |
---|
1109 | try: |
---|
1110 | if dlg.ShowModal() == wx.ID_OK: |
---|
1111 | filename = dlg.GetPath() |
---|
1112 | File = open(filename,'r') |
---|
1113 | save = {} |
---|
1114 | oldThreshold = data['Thresholds'][0] |
---|
1115 | S = File.readline() |
---|
1116 | while S: |
---|
1117 | if S[0] == '#': |
---|
1118 | S = File.readline() |
---|
1119 | continue |
---|
1120 | [key,val] = S[:-1].split(':') |
---|
1121 | if key in ['Points','Rings','Arcs','Polygons','Frames','Thresholds']: |
---|
1122 | if ignoreThreshold and key == 'Thresholds': continue |
---|
1123 | save[key] = eval(val) |
---|
1124 | if key == 'Thresholds': |
---|
1125 | save[key][0] = oldThreshold |
---|
1126 | save[key][1][1] = min(oldThreshold[1],save[key][1][1]) |
---|
1127 | S = File.readline() |
---|
1128 | data.update(save) |
---|
1129 | wx.CallAfter(UpdateMasks,G2frame,data) |
---|
1130 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1131 | File.close() |
---|
1132 | finally: |
---|
1133 | dlg.Destroy() |
---|
1134 | |
---|
1135 | def OnNewSpotMask(event): |
---|
1136 | 'Start a new spot mask' |
---|
1137 | G2frame.MaskKey = 's' |
---|
1138 | G2plt.OnStartMask(G2frame) |
---|
1139 | |
---|
1140 | def OnNewArcMask(event): |
---|
1141 | 'Start a new arc mask' |
---|
1142 | G2frame.MaskKey = 'a' |
---|
1143 | G2plt.OnStartMask(G2frame) |
---|
1144 | |
---|
1145 | def OnNewRingMask(event): |
---|
1146 | 'Start a new ring mask' |
---|
1147 | G2frame.MaskKey = 'r' |
---|
1148 | G2plt.OnStartMask(G2frame) |
---|
1149 | |
---|
1150 | def OnNewPolyMask(event): |
---|
1151 | 'Start a new polygon mask' |
---|
1152 | G2frame.MaskKey = 'p' |
---|
1153 | G2plt.OnStartMask(G2frame) |
---|
1154 | |
---|
1155 | def OnNewFrameMask(event): |
---|
1156 | 'Start a new Frame mask' |
---|
1157 | G2frame.MaskKey = 'f' |
---|
1158 | G2plt.OnStartMask(G2frame) |
---|
1159 | |
---|
1160 | startScroll = None |
---|
1161 | if G2frame.dataDisplay: |
---|
1162 | startScroll = G2frame.dataDisplay.GetScrollPos(wx.VERTICAL) # save scroll position |
---|
1163 | G2frame.dataDisplay.Destroy() |
---|
1164 | G2gd.SetDataMenuBar(G2frame,G2frame.dataFrame.MaskMenu) |
---|
1165 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnCopyMask, id=G2gd.wxID_MASKCOPY) |
---|
1166 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnLoadMask, id=G2gd.wxID_MASKLOAD) |
---|
1167 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnLoadMask, id=G2gd.wxID_MASKLOADNOT) |
---|
1168 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnSaveMask, id=G2gd.wxID_MASKSAVE) |
---|
1169 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnNewSpotMask, id=G2gd.wxID_NEWMASKSPOT) |
---|
1170 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnNewArcMask, id=G2gd.wxID_NEWMASKARC) |
---|
1171 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnNewRingMask, id=G2gd.wxID_NEWMASKRING) |
---|
1172 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnNewPolyMask, id=G2gd.wxID_NEWMASKPOLY) |
---|
1173 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnNewFrameMask, id=G2gd.wxID_NEWMASKFRAME) |
---|
1174 | if not G2frame.dataFrame.GetStatusBar(): |
---|
1175 | Status = G2frame.dataFrame.CreateStatusBar() |
---|
1176 | if G2frame.MaskKey == 'f': |
---|
1177 | G2frame.dataFrame.GetStatusBar().SetStatusText('Frame mask active - LB pick next point, RB close polygon') |
---|
1178 | elif G2frame.MaskKey == 'p': |
---|
1179 | G2frame.dataFrame.GetStatusBar().SetStatusText('Polygon mask active - LB pick next point, RB close polygon') |
---|
1180 | elif G2frame.MaskKey == 's': |
---|
1181 | G2frame.dataFrame.GetStatusBar().SetStatusText('Spot mask active - LB pick spot location') |
---|
1182 | elif G2frame.MaskKey == 'a': |
---|
1183 | G2frame.dataFrame.GetStatusBar().SetStatusText('Arc mask active - LB pick arc location') |
---|
1184 | elif G2frame.MaskKey == 'r': |
---|
1185 | G2frame.dataFrame.GetStatusBar().SetStatusText('Ring mask active - LB pick ring location') |
---|
1186 | else: |
---|
1187 | G2frame.dataFrame.GetStatusBar().SetStatusText("To add mask: press a,r,s,p or f on 2D image for arc/ring/spot/polygon/frame") |
---|
1188 | G2frame.dataDisplay = wxscroll.ScrolledPanel(G2frame.dataFrame) |
---|
1189 | mainSizer = wx.BoxSizer(wx.VERTICAL) |
---|
1190 | mainSizer.Add((5,10),0) |
---|
1191 | |
---|
1192 | thresh = data['Thresholds'] #min/max intensity range |
---|
1193 | Spots = data['Points'] #x,y,radius in mm |
---|
1194 | Rings = data['Rings'] #radius, thickness |
---|
1195 | Polygons = data['Polygons'] #3+ x,y pairs |
---|
1196 | if 'Frames' not in data: |
---|
1197 | data['Frames'] = [] |
---|
1198 | frame = data['Frames'] #3+ x,y pairs |
---|
1199 | Arcs = data['Arcs'] #radius, start/end azimuth, thickness |
---|
1200 | |
---|
1201 | littleSizer = wx.FlexGridSizer(0,3,0,5) |
---|
1202 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Lower/Upper limits '),0,WACV) |
---|
1203 | Text = wx.TextCtrl(G2frame.dataDisplay,value=str(thresh[0][0]),style=wx.TE_READONLY) |
---|
1204 | littleSizer.Add(Text,0,WACV) |
---|
1205 | Text.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1206 | Text = wx.TextCtrl(G2frame.dataDisplay,value=str(thresh[0][1]),style=wx.TE_READONLY) |
---|
1207 | littleSizer.Add(Text,0,WACV) |
---|
1208 | Text.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1209 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' Lower/Upper thresholds '),0,WACV) |
---|
1210 | lowerThreshold = G2gd.ValidatedTxtCtrl(G2frame.dataDisplay,loc=thresh[1],key=0, |
---|
1211 | min=thresh[0][0],OnLeave=Replot,typeHint=int) |
---|
1212 | littleSizer.Add(lowerThreshold,0,WACV) |
---|
1213 | upperThreshold = G2gd.ValidatedTxtCtrl(G2frame.dataDisplay,loc=thresh[1],key=1, |
---|
1214 | max=thresh[0][1],OnLeave=Replot,typeHint=int) |
---|
1215 | littleSizer.Add(upperThreshold,0,WACV) |
---|
1216 | mainSizer.Add(littleSizer,0,) |
---|
1217 | if Spots: |
---|
1218 | lbl = wx.StaticText(parent=G2frame.dataDisplay,label=' Spot masks') |
---|
1219 | lbl.SetBackgroundColour(wx.Colour(200,200,210)) |
---|
1220 | mainSizer.Add(lbl,0,wx.EXPAND|wx.ALIGN_CENTER,0) |
---|
1221 | littleSizer = wx.FlexGridSizer(0,3,0,5) |
---|
1222 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' position, mm'),0,WACV) |
---|
1223 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' diameter, mm'),0,WACV) |
---|
1224 | littleSizer.Add((5,0),0) |
---|
1225 | for i in range(len(Spots)): |
---|
1226 | if Spots[i]: |
---|
1227 | x,y,d = Spots[i] |
---|
1228 | spotText = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.2f,%.2f" % (x,y)), |
---|
1229 | style=wx.TE_READONLY) |
---|
1230 | spotText.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1231 | littleSizer.Add(spotText,0,WACV) |
---|
1232 | spotText.Bind(wx.EVT_ENTER_WINDOW,OnTextMsg) |
---|
1233 | spotDiameter = G2gd.ValidatedTxtCtrl(G2frame.dataDisplay,loc=Spots[i],key=2, |
---|
1234 | max=100.,OnLeave=Replot,nDig=[8,2]) |
---|
1235 | littleSizer.Add(spotDiameter,0,WACV) |
---|
1236 | spotDelete = G2gd.G2LoggedButton(G2frame.dataDisplay,label='delete?', |
---|
1237 | locationcode='Delete+Points+'+str(i), |
---|
1238 | handler=onDeleteMask) |
---|
1239 | littleSizer.Add(spotDelete,0,WACV) |
---|
1240 | mainSizer.Add(littleSizer,0,) |
---|
1241 | if Rings: |
---|
1242 | lbl = wx.StaticText(parent=G2frame.dataDisplay,label=' Ring masks') |
---|
1243 | lbl.SetBackgroundColour(wx.Colour(200,200,210)) |
---|
1244 | mainSizer.Add(lbl,0,wx.EXPAND|wx.ALIGN_CENTER,0) |
---|
1245 | littleSizer = wx.FlexGridSizer(0,3,0,5) |
---|
1246 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' 2-theta,deg'),0,WACV) |
---|
1247 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' thickness, deg'),0,WACV) |
---|
1248 | littleSizer.Add((5,0),0) |
---|
1249 | for i in range(len(Rings)): |
---|
1250 | if Rings[i]: |
---|
1251 | ringText = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.3f" % (Rings[i][0])), |
---|
1252 | style=wx.TE_READONLY) |
---|
1253 | ringText.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1254 | ringText.Bind(wx.EVT_ENTER_WINDOW,OnTextMsg) |
---|
1255 | littleSizer.Add(ringText,0,WACV) |
---|
1256 | ringThick = G2gd.ValidatedTxtCtrl(G2frame.dataDisplay,loc=Rings[i],key=1, |
---|
1257 | min=0.001,max=1.,OnLeave=Replot,nDig=[8,3]) |
---|
1258 | littleSizer.Add(ringThick,0,WACV) |
---|
1259 | ringDelete = G2gd.G2LoggedButton(G2frame.dataDisplay,label='delete?', |
---|
1260 | locationcode='Delete+Rings+'+str(i), |
---|
1261 | handler=onDeleteMask) |
---|
1262 | littleSizer.Add(ringDelete,0,WACV) |
---|
1263 | mainSizer.Add(littleSizer,0,) |
---|
1264 | if Arcs: |
---|
1265 | lbl = wx.StaticText(parent=G2frame.dataDisplay,label=' Arc masks') |
---|
1266 | lbl.SetBackgroundColour(wx.Colour(200,200,210)) |
---|
1267 | mainSizer.Add(lbl,0,wx.EXPAND|wx.ALIGN_CENTER,0) |
---|
1268 | littleSizer = wx.FlexGridSizer(0,4,0,5) |
---|
1269 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' 2-theta,deg'),0,WACV) |
---|
1270 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' azimuth, deg'),0,WACV) |
---|
1271 | littleSizer.Add(wx.StaticText(parent=G2frame.dataDisplay,label=' thickness, deg'),0,WACV) |
---|
1272 | littleSizer.Add((5,0),0) |
---|
1273 | for i in range(len(Arcs)): |
---|
1274 | if Arcs[i]: |
---|
1275 | tth,azimuth,thick = Arcs[i] |
---|
1276 | arcText = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.3f" % (tth)), |
---|
1277 | style=wx.TE_READONLY) |
---|
1278 | arcText.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1279 | arcText.Bind(wx.EVT_ENTER_WINDOW,OnTextMsg) |
---|
1280 | littleSizer.Add(arcText,0,WACV) |
---|
1281 | azmText = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%d,%d" % (azimuth[0],azimuth[1])), |
---|
1282 | style=wx.TE_READONLY) |
---|
1283 | azmText.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1284 | azmText.Bind(wx.EVT_ENTER_WINDOW,OnTextMsg) |
---|
1285 | littleSizer.Add(azmText,0,WACV) |
---|
1286 | arcThick = G2gd.ValidatedTxtCtrl(G2frame.dataDisplay,loc=Arcs[i],key=2, |
---|
1287 | min=0.001,max=20.,OnLeave=Replot,nDig=[8,3]) |
---|
1288 | littleSizer.Add(arcThick,0,WACV) |
---|
1289 | arcDelete = G2gd.G2LoggedButton(G2frame.dataDisplay,label='delete?', |
---|
1290 | locationcode='Delete+Arcs+'+str(i), |
---|
1291 | handler=onDeleteMask) |
---|
1292 | littleSizer.Add(arcDelete,0,WACV) |
---|
1293 | mainSizer.Add(littleSizer,0,) |
---|
1294 | if Polygons: |
---|
1295 | lbl = wx.StaticText(parent=G2frame.dataDisplay, |
---|
1296 | label=' Polygon masks (on plot RB vertex drag to move,\nLB vertex drag to insert)') |
---|
1297 | lbl.SetBackgroundColour(wx.Colour(200,200,210)) |
---|
1298 | mainSizer.Add(lbl,0,wx.EXPAND|wx.ALIGN_CENTER,0) |
---|
1299 | littleSizer = wx.FlexGridSizer(0,2,0,5) |
---|
1300 | for i in range(len(Polygons)): |
---|
1301 | if Polygons[i]: |
---|
1302 | polyList = [] |
---|
1303 | for x,y in Polygons[i]: |
---|
1304 | polyList.append("%.2f, %.2f"%(x,y)) |
---|
1305 | polyText = wx.ComboBox(G2frame.dataDisplay,value=polyList[0],choices=polyList,style=wx.CB_READONLY) |
---|
1306 | littleSizer.Add(polyText,0,WACV) |
---|
1307 | polyDelete = G2gd.G2LoggedButton(G2frame.dataDisplay,label='delete?', |
---|
1308 | locationcode='Delete+Polygons+'+str(i), |
---|
1309 | handler=onDeleteMask) |
---|
1310 | littleSizer.Add(polyDelete,0,WACV) |
---|
1311 | mainSizer.Add(littleSizer,0,) |
---|
1312 | if frame: |
---|
1313 | lbl = wx.StaticText(parent=G2frame.dataDisplay, |
---|
1314 | label=' Frame mask (on plot RB vertex drag to move,\nLB vertex drag to insert)') |
---|
1315 | lbl.SetBackgroundColour(wx.Colour(200,200,210)) |
---|
1316 | mainSizer.Add(lbl,0,wx.EXPAND|wx.ALIGN_CENTER,0) |
---|
1317 | littleSizer = wx.FlexGridSizer(0,2,0,5) |
---|
1318 | frameList = [] |
---|
1319 | for x,y in frame: |
---|
1320 | frameList.append("%.2f, %.2f"%(x,y)) |
---|
1321 | frameText = wx.ComboBox(G2frame.dataDisplay,value=frameList[0],choices=frameList,style=wx.CB_READONLY) |
---|
1322 | littleSizer.Add(frameText,0,WACV) |
---|
1323 | frameDelete = G2gd.G2LoggedButton(G2frame.dataDisplay,label='delete?', |
---|
1324 | locationcode='Delete+Frame', |
---|
1325 | handler=onDeleteFrame) |
---|
1326 | littleSizer.Add(frameDelete,0,WACV) |
---|
1327 | mainSizer.Add(littleSizer,0,) |
---|
1328 | mainSizer.Layout() |
---|
1329 | G2frame.dataDisplay.SetSizer(mainSizer) |
---|
1330 | G2frame.dataDisplay.SetSize(mainSizer.Fit(G2frame.dataFrame)) |
---|
1331 | G2frame.dataDisplay.SetupScrolling() |
---|
1332 | Size = mainSizer.Fit(G2frame.dataFrame) |
---|
1333 | Size[0] += 50 # room for scrollbar & status msg |
---|
1334 | Size[1] = min(Size[1],500) |
---|
1335 | G2frame.dataDisplay.SetSize(Size) |
---|
1336 | G2frame.dataFrame.setSizePosLeft(Size) |
---|
1337 | wx.Yield() |
---|
1338 | if startScroll: # reset scroll to saved position |
---|
1339 | G2frame.dataDisplay.Scroll(0,startScroll) # set to saved scroll position |
---|
1340 | wx.Yield() |
---|
1341 | |
---|
1342 | ################################################################################ |
---|
1343 | ##### Stress/Strain |
---|
1344 | ################################################################################ |
---|
1345 | |
---|
1346 | def UpdateStressStrain(G2frame,data): |
---|
1347 | '''Shows and handles the controls on the "Stress/Strain" |
---|
1348 | data tree entry |
---|
1349 | ''' |
---|
1350 | |
---|
1351 | def OnAppendDzero(event): |
---|
1352 | data['d-zero'].append({'Dset':1.0,'Dcalc':0.0,'pixLimit':10,'cutoff':1.0, |
---|
1353 | 'ImxyObs':[[],[]],'ImtaObs':[[],[]],'ImtaCalc':[[],[]],'Emat':[1.0,1.0,1.0]}) |
---|
1354 | UpdateStressStrain(G2frame,data) |
---|
1355 | |
---|
1356 | def OnUpdateDzero(event): |
---|
1357 | for item in data['d-zero']: |
---|
1358 | if item['Dcalc']: #skip unrefined ones |
---|
1359 | item['Dset'] = item['Dcalc'] |
---|
1360 | UpdateStressStrain(G2frame,data) |
---|
1361 | |
---|
1362 | def OnCopyStrSta(event): |
---|
1363 | TextList = [[False,'All IMG',0,0]] |
---|
1364 | Names = [] |
---|
1365 | if G2frame.PatternTree.GetCount(): |
---|
1366 | id, cookie = G2frame.PatternTree.GetFirstChild(G2frame.root) |
---|
1367 | while id: |
---|
1368 | name = G2frame.PatternTree.GetItemText(id) |
---|
1369 | Names.append(name) |
---|
1370 | if 'IMG' in name: |
---|
1371 | Data = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Stress/Strain')) |
---|
1372 | if id == G2frame.Image: |
---|
1373 | Source = name |
---|
1374 | else: |
---|
1375 | TextList.append([False,name,id,Data.get('Sample load',0.0)]) |
---|
1376 | id, cookie = G2frame.PatternTree.GetNextChild(G2frame.root, cookie) |
---|
1377 | if len(TextList) == 1: |
---|
1378 | G2frame.ErrorDialog('Nothing to copy controls to','There must be more than one "IMG" pattern') |
---|
1379 | return |
---|
1380 | dlg = G2frame.CopyDialog(G2frame,'Copy stress/strain controls','Copy controls from '+Source+' to:',TextList) |
---|
1381 | try: |
---|
1382 | if dlg.ShowModal() == wx.ID_OK: |
---|
1383 | result = dlg.GetData() |
---|
1384 | if result[0][0]: |
---|
1385 | result = TextList[1:] |
---|
1386 | for item in result: item[0] = True |
---|
1387 | for i,item in enumerate(result): |
---|
1388 | ifcopy,name,id,load = item |
---|
1389 | if ifcopy: |
---|
1390 | Data = copy.deepcopy(data) |
---|
1391 | Data['Sample load'] = load |
---|
1392 | G2frame.PatternTree.SetItemPyData(G2gd.GetPatternTreeItemId(G2frame,id, 'Stress/Strain'),Data) |
---|
1393 | finally: |
---|
1394 | dlg.Destroy() |
---|
1395 | |
---|
1396 | def OnLoadStrSta(event): |
---|
1397 | dlg = wx.FileDialog(G2frame, 'Choose stress/strain file', '.', '', |
---|
1398 | 'image control files (*.strsta)|*.strsta',wx.OPEN|wx.CHANGE_DIR) |
---|
1399 | try: |
---|
1400 | if dlg.ShowModal() == wx.ID_OK: |
---|
1401 | filename = dlg.GetPath() |
---|
1402 | File = open(filename,'r') |
---|
1403 | S = File.read() |
---|
1404 | data = eval(S) |
---|
1405 | Controls = G2frame.PatternTree.GetItemPyData( |
---|
1406 | G2gd.GetPatternTreeItemId(G2frame,G2frame.Image, 'Image Controls')) |
---|
1407 | G2img.FitStrSta(G2frame.ImageZ,data,Controls) |
---|
1408 | UpdateStressStrain(G2frame,data) |
---|
1409 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1410 | G2plt.PlotStrain(G2frame,data,newPlot=True) |
---|
1411 | File.close() |
---|
1412 | finally: |
---|
1413 | dlg.Destroy() |
---|
1414 | |
---|
1415 | def OnSaveStrSta(event): |
---|
1416 | dlg = wx.FileDialog(G2frame, 'Choose stress/strain file', '.', '', |
---|
1417 | 'image control files (*.strsta)|*.strsta',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT|wx.CHANGE_DIR) |
---|
1418 | try: |
---|
1419 | if dlg.ShowModal() == wx.ID_OK: |
---|
1420 | filename = dlg.GetPath() |
---|
1421 | File = open(filename,'w') |
---|
1422 | save = {} |
---|
1423 | keys = ['Type','Sample phi','Sample z','Sample load'] |
---|
1424 | keys2 = ['Dset','Dcalc','pixLimit','cutoff','Emat'] |
---|
1425 | File.write('{\n\t') |
---|
1426 | for key in keys: |
---|
1427 | if key in 'Type': |
---|
1428 | File.write("'"+key+"':'"+data[key]+"',") |
---|
1429 | else: |
---|
1430 | File.write("'"+key+"':"+str(data[key])+',') |
---|
1431 | File.write('\n\t'+"'d-zero':[\n") |
---|
1432 | for data2 in data['d-zero']: |
---|
1433 | File.write('\t\t{') |
---|
1434 | for key in keys2: |
---|
1435 | File.write("'"+key+"':"+str(data2[key])+',') |
---|
1436 | File.write("'ImxyObs':[[],[]],'ImtaObs':[[],[]],'ImtaCalc':[[],[]]},\n") |
---|
1437 | File.write('\t]\n}') |
---|
1438 | File.close() |
---|
1439 | finally: |
---|
1440 | dlg.Destroy() |
---|
1441 | |
---|
1442 | def OnStrStaSample(event): |
---|
1443 | filename = '' |
---|
1444 | dlg = wx.FileDialog(G2frame, 'Choose multihistogram metadata text file', '.', '', |
---|
1445 | 'metadata file (*.*)|*.*',wx.OPEN|wx.CHANGE_DIR) |
---|
1446 | try: |
---|
1447 | if dlg.ShowModal() == wx.ID_OK: |
---|
1448 | filename = dlg.GetPath() |
---|
1449 | File = open(filename,'r') |
---|
1450 | S = File.readline() |
---|
1451 | newItems = [] |
---|
1452 | itemNames = [] |
---|
1453 | Comments = [] |
---|
1454 | while S: |
---|
1455 | if S[0] == '#': |
---|
1456 | Comments.append(S) |
---|
1457 | S = File.readline() |
---|
1458 | continue |
---|
1459 | S = S.replace(',',' ').replace('\t',' ') |
---|
1460 | Stuff = S[:-1].split() |
---|
1461 | itemNames.append(Stuff[0]) |
---|
1462 | newItems.append(Stuff[1:]) |
---|
1463 | S = File.readline() |
---|
1464 | File.close() |
---|
1465 | finally: |
---|
1466 | dlg.Destroy() |
---|
1467 | if not filename: |
---|
1468 | G2frame.ErrorDialog('Nothing to do','No file selected') |
---|
1469 | return |
---|
1470 | dataDict = dict(zip(itemNames,newItems)) |
---|
1471 | ifany = False |
---|
1472 | Names = [' ','Sample phi','Sample z','Sample load'] |
---|
1473 | dlg = G2gd.G2ColumnIDDialog( G2frame,' Choose multihistogram metadata columns:', |
---|
1474 | 'Select columns',Comments,Names,np.array(newItems).T) |
---|
1475 | try: |
---|
1476 | if dlg.ShowModal() == wx.ID_OK: |
---|
1477 | colNames,newData = dlg.GetSelection() |
---|
1478 | dataDict = dict(zip(itemNames,newData.T)) |
---|
1479 | for item in colNames: |
---|
1480 | if item != ' ': |
---|
1481 | ifany = True |
---|
1482 | finally: |
---|
1483 | dlg.Destroy() |
---|
1484 | if not ifany: |
---|
1485 | G2frame.ErrorDialog('Nothing to do','No columns identified') |
---|
1486 | return |
---|
1487 | histList = [] |
---|
1488 | item, cookie = G2frame.PatternTree.GetFirstChild(G2frame.root) |
---|
1489 | while item: |
---|
1490 | name = G2frame.PatternTree.GetItemText(item) |
---|
1491 | if name.startswith('IMG'): |
---|
1492 | histList.append(name) |
---|
1493 | item, cookie = G2frame.PatternTree.GetNextChild(G2frame.root, cookie) |
---|
1494 | colIds = {} |
---|
1495 | for i,name in enumerate(colNames): |
---|
1496 | if name != ' ': |
---|
1497 | colIds[name] = i |
---|
1498 | for hist in histList: |
---|
1499 | name = hist.split()[1] #this is file name |
---|
1500 | if name in dataDict: |
---|
1501 | newItems = {} |
---|
1502 | for item in colIds: |
---|
1503 | newItems[item] = float(dataDict[name][colIds[item]]) |
---|
1504 | Id = G2gd.GetPatternTreeItemId(G2frame,G2frame.root,hist) |
---|
1505 | stsrData = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,Id,'Stress/Strain')) |
---|
1506 | stsrData.update(newItems) |
---|
1507 | UpdateStressStrain(G2frame,data) |
---|
1508 | |
---|
1509 | def OnFitStrSta(event): |
---|
1510 | Controls = G2frame.PatternTree.GetItemPyData( |
---|
1511 | G2gd.GetPatternTreeItemId(G2frame,G2frame.Image, 'Image Controls')) |
---|
1512 | G2img.FitStrSta(G2frame.ImageZ,data,Controls) |
---|
1513 | print 'Strain fitting finished' |
---|
1514 | UpdateStressStrain(G2frame,data) |
---|
1515 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1516 | G2plt.PlotStrain(G2frame,data,newPlot=True) |
---|
1517 | |
---|
1518 | def OnFitAllStrSta(event): |
---|
1519 | TextList = [[False,'All IMG',0]] |
---|
1520 | Names = [] |
---|
1521 | if G2frame.PatternTree.GetCount(): |
---|
1522 | choices = G2gd.GetPatternTreeDataNames(G2frame,['IMG ',]) |
---|
1523 | if len(choices) == 1: |
---|
1524 | G2frame.ErrorDialog('Nothing to fit','There must some "IMG" patterns') |
---|
1525 | return |
---|
1526 | sel = [] |
---|
1527 | dlg = G2gd.G2MultiChoiceDialog(G2frame,'Stress/Strain fitting','Select images to fit:',choices) |
---|
1528 | dlg.SetSelections(sel) |
---|
1529 | names = [] |
---|
1530 | if dlg.ShowModal() == wx.ID_OK: |
---|
1531 | for sel in dlg.GetSelections(): |
---|
1532 | names.append(choices[sel]) |
---|
1533 | dlg.Destroy() |
---|
1534 | SeqResult = {} |
---|
1535 | dlg = wx.ProgressDialog('Sequential IMG Strain fit','Data set name = '+names[0],len(names), |
---|
1536 | style = wx.PD_ELAPSED_TIME|wx.PD_AUTO_HIDE|wx.PD_REMAINING_TIME|wx.PD_CAN_ABORT) |
---|
1537 | wx.BeginBusyCursor() |
---|
1538 | goodnames = [] |
---|
1539 | try: |
---|
1540 | for i,name in enumerate(names): |
---|
1541 | print ' Sequential strain fit for ',name |
---|
1542 | GoOn = dlg.Update(i,newmsg='Data set name = '+name)[0] |
---|
1543 | if not GoOn: |
---|
1544 | break |
---|
1545 | Id = G2gd.GetPatternTreeItemId(G2frame,G2frame.root,name) |
---|
1546 | Controls = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,Id, 'Image Controls')) |
---|
1547 | StaCtrls = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame,Id, 'Stress/Strain')) |
---|
1548 | if not len(StaCtrls['d-zero']): |
---|
1549 | continue |
---|
1550 | goodnames.append(name) |
---|
1551 | id = G2gd.GetPatternTreeItemId(G2frame, G2frame.root, name) |
---|
1552 | Npix,imagefile = G2frame.PatternTree.GetItemPyData(Id) |
---|
1553 | image = G2IO.GetImageData(G2frame,imagefile,True) |
---|
1554 | dark = Controls['dark image'] |
---|
1555 | if dark[0]: |
---|
1556 | darkfile = G2frame.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(G2frame, |
---|
1557 | G2frame.root,dark[0]))[1] |
---|
1558 | darkImg = G2IO.GetImageData(G2frame,darkfile,imageOnly=True) |
---|
1559 | image += dark[1]*darkImg |
---|
1560 | G2img.FitStrSta(image,StaCtrls,Controls) |
---|
1561 | G2plt.PlotStrain(G2frame,StaCtrls,newPlot=True) |
---|
1562 | parmDict = {'Sample load':StaCtrls['Sample load'],} |
---|
1563 | varyNames = ['e11','e12','e22'] |
---|
1564 | sig = [] |
---|
1565 | varyList = [] |
---|
1566 | variables = [] |
---|
1567 | for i,item in enumerate(StaCtrls['d-zero']): |
---|
1568 | variables += item['Emat'] |
---|
1569 | sig += item['Esig'] |
---|
1570 | varylist = ['%d%s%s'%(i,':',Name) for Name in varyNames] |
---|
1571 | varyList += varylist |
---|
1572 | parmDict.update(dict(zip(varylist,item['Emat']))) |
---|
1573 | parmDict['%d:Dcalc'%(i)] = item['Dcalc'] |
---|
1574 | SeqResult[name] = {'variables':variables,'varyList':varyList,'sig':sig,'Rvals':[], |
---|
1575 | 'covMatrix':np.eye(len(variables)),'title':name,'parmDict':parmDict} |
---|
1576 | else: |
---|
1577 | SeqResult['histNames'] = goodnames |
---|
1578 | dlg.Destroy() |
---|
1579 | print ' ***** Sequential strain refinement successful *****' |
---|
1580 | finally: |
---|
1581 | wx.EndBusyCursor() |
---|
1582 | Id = G2gd.GetPatternTreeItemId(G2frame,G2frame.root,'Sequential results') |
---|
1583 | if Id: |
---|
1584 | G2frame.PatternTree.SetItemPyData(Id,SeqResult) |
---|
1585 | else: |
---|
1586 | Id = G2frame.PatternTree.AppendItem(parent=G2frame.root,text='Sequential results') |
---|
1587 | G2frame.PatternTree.SetItemPyData(Id,SeqResult) |
---|
1588 | G2frame.PatternTree.SelectItem(Id) |
---|
1589 | print 'All images fitted' |
---|
1590 | |
---|
1591 | def SamSizer(): |
---|
1592 | |
---|
1593 | def OnStrainType(event): |
---|
1594 | data['Type'] = strType.GetValue() |
---|
1595 | |
---|
1596 | def OnSamPhi(event): |
---|
1597 | try: |
---|
1598 | value = float(samPhi.GetValue()) |
---|
1599 | except ValueError: |
---|
1600 | value = data['Sample phi'] |
---|
1601 | data['Sample phi'] = value |
---|
1602 | samPhi.SetValue("%.3f" % (data['Sample phi'])) |
---|
1603 | |
---|
1604 | def OnSamZ(event): |
---|
1605 | try: |
---|
1606 | value = float(samZ.GetValue()) |
---|
1607 | except ValueError: |
---|
1608 | value = data['Sample z'] |
---|
1609 | data['Sample z'] = value |
---|
1610 | samZ.SetValue("%.3f" % (data['Sample z'])) |
---|
1611 | |
---|
1612 | def OnSamLoad(event): |
---|
1613 | try: |
---|
1614 | value = float(samLoad.GetValue()) |
---|
1615 | except ValueError: |
---|
1616 | value = data['Sample load'] |
---|
1617 | data['Sample load'] = value |
---|
1618 | samLoad.SetValue("%.3f" % (data['Sample load'])) |
---|
1619 | |
---|
1620 | samSizer = wx.BoxSizer(wx.HORIZONTAL) |
---|
1621 | samSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=' Strain type: '),0,WACV) |
---|
1622 | strType = wx.ComboBox(G2frame.dataDisplay,value=data['Type'],choices=['True','Conventional'], |
---|
1623 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
1624 | strType.SetValue(data['Type']) |
---|
1625 | strType.Bind(wx.EVT_COMBOBOX, OnStrainType) |
---|
1626 | samSizer.Add(strType,0,WACV) |
---|
1627 | |
---|
1628 | samSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=' Sample phi: '),0,WACV) |
---|
1629 | samPhi = wx.TextCtrl(G2frame.dataDisplay,-1,value=("%.3f" % (data['Sample phi'])), |
---|
1630 | style=wx.TE_PROCESS_ENTER) |
---|
1631 | samSizer.Add(samPhi,0,WACV) |
---|
1632 | samPhi.Bind(wx.EVT_TEXT_ENTER,OnSamPhi) |
---|
1633 | samPhi.Bind(wx.EVT_KILL_FOCUS,OnSamPhi) |
---|
1634 | samSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=' Sample delta-z(mm): '),0,WACV) |
---|
1635 | samZ = wx.TextCtrl(G2frame.dataDisplay,-1,value=("%.3f" % (data['Sample z'])), |
---|
1636 | style=wx.TE_PROCESS_ENTER) |
---|
1637 | samSizer.Add(samZ,0,WACV) |
---|
1638 | samZ.Bind(wx.EVT_TEXT_ENTER,OnSamZ) |
---|
1639 | samZ.Bind(wx.EVT_KILL_FOCUS,OnSamZ) |
---|
1640 | samSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=' Sample load(MPa): '),0,WACV) |
---|
1641 | samLoad = G2gd.ValidatedTxtCtrl(G2frame.dataDisplay,data,'Sample load', |
---|
1642 | nDig=[8,3],typeHint=float,) |
---|
1643 | samSizer.Add(samLoad,0,WACV) |
---|
1644 | |
---|
1645 | return samSizer |
---|
1646 | |
---|
1647 | def DzeroSizer(): |
---|
1648 | |
---|
1649 | def OnDzero(event): |
---|
1650 | Obj = event.GetEventObject() |
---|
1651 | try: |
---|
1652 | value = min(20.0,max(0.25,float(Obj.GetValue()))) |
---|
1653 | except ValueError: |
---|
1654 | value = 1.0 |
---|
1655 | Obj.SetValue("%.5f"%(value)) |
---|
1656 | data['d-zero'][Indx[Obj.GetId()]]['Dset'] = value |
---|
1657 | data['d-zero'] = G2mth.sortArray(data['d-zero'],'Dset',reverse=True) |
---|
1658 | Ring,R = G2img.MakeStrStaRing(data['d-zero'][Indx[Obj.GetId()]],G2frame.ImageZ,Controls) |
---|
1659 | if len(Ring): |
---|
1660 | data['d-zero'][Indx[Obj.GetId()]].update(R) |
---|
1661 | else: |
---|
1662 | G2frame.ErrorDialog('Strain peak selection','WARNING - No points found for this ring selection') |
---|
1663 | |
---|
1664 | UpdateStressStrain(G2frame,data) |
---|
1665 | G2plt.PlotExposedImage(G2frame,event=event,newPlot=False) |
---|
1666 | G2plt.PlotStrain(G2frame,data,newPlot=True) |
---|
1667 | |
---|
1668 | def OnDeleteDzero(event): |
---|
1669 | Obj = event.GetEventObject() |
---|
1670 | del(data['d-zero'][delIndx.index(Obj)]) |
---|
1671 | UpdateStressStrain(G2frame,data) |
---|
1672 | G2plt.PlotExposedImage(G2frame,event=event,newPlot=True) |
---|
1673 | G2plt.PlotStrain(G2frame,data,newPlot=True) |
---|
1674 | |
---|
1675 | def OnCutOff(event): |
---|
1676 | Obj = event.GetEventObject() |
---|
1677 | try: |
---|
1678 | value = min(10.0,max(0.5,float(Obj.GetValue()))) |
---|
1679 | except ValueError: |
---|
1680 | value = 1.0 |
---|
1681 | Obj.SetValue("%.1f"%(value)) |
---|
1682 | data['d-zero'][Indx[Obj.GetId()]]['cutoff'] = value |
---|
1683 | Ring,R = G2img.MakeStrStaRing(data['d-zero'][Indx[Obj.GetId()]],G2frame.ImageZ,Controls) |
---|
1684 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1685 | G2plt.PlotStrain(G2frame,data,newPlot=True) |
---|
1686 | |
---|
1687 | def OnPixLimit(event): |
---|
1688 | Obj = event.GetEventObject() |
---|
1689 | data['d-zero'][Indx[Obj.GetId()]]['pixLimit'] = int(Obj.GetValue()) |
---|
1690 | Ring,R = G2img.MakeStrStaRing(data['d-zero'][Indx[Obj.GetId()]],G2frame.ImageZ,Controls) |
---|
1691 | G2plt.PlotExposedImage(G2frame,event=event) |
---|
1692 | G2plt.PlotStrain(G2frame,data,newPlot=True) |
---|
1693 | |
---|
1694 | Indx = {} |
---|
1695 | delIndx = [] |
---|
1696 | dzeroSizer = wx.FlexGridSizer(0,8,5,5) |
---|
1697 | for id,dzero in enumerate(data['d-zero']): |
---|
1698 | dzeroSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=(' d-zero #%d: '%(id))),0,WACV) |
---|
1699 | dZero = wx.TextCtrl(G2frame.dataDisplay,-1,value=('%.5f'%(dzero['Dset'])), |
---|
1700 | style=wx.TE_PROCESS_ENTER) |
---|
1701 | dzeroSizer.Add(dZero,0,WACV) |
---|
1702 | dZero.Bind(wx.EVT_TEXT_ENTER,OnDzero) |
---|
1703 | dZero.Bind(wx.EVT_KILL_FOCUS,OnDzero) |
---|
1704 | Indx[dZero.GetId()] = id |
---|
1705 | dzeroSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=(' d-zero ave: %.5f'%(dzero['Dcalc']))),0,WACV) |
---|
1706 | |
---|
1707 | dzeroSizer.Add(wx.StaticText(G2frame.dataDisplay,label=' Min ring I/Ib '),0,WACV) |
---|
1708 | cutOff = wx.TextCtrl(parent=G2frame.dataDisplay,value=("%.1f" % (dzero['cutoff'])), |
---|
1709 | style=wx.TE_PROCESS_ENTER) |
---|
1710 | cutOff.Bind(wx.EVT_TEXT_ENTER,OnCutOff) |
---|
1711 | cutOff.Bind(wx.EVT_KILL_FOCUS,OnCutOff) |
---|
1712 | Indx[cutOff.GetId()] = id |
---|
1713 | dzeroSizer.Add(cutOff,0,WACV) |
---|
1714 | |
---|
1715 | dzeroSizer.Add(wx.StaticText(G2frame.dataDisplay,label=' Pixel search range '),0,WACV) |
---|
1716 | pixLimit = wx.ComboBox(parent=G2frame.dataDisplay,value=str(dzero['pixLimit']),choices=['1','2','5','10','15','20'], |
---|
1717 | style=wx.CB_READONLY|wx.CB_DROPDOWN) |
---|
1718 | pixLimit.Bind(wx.EVT_COMBOBOX, OnPixLimit) |
---|
1719 | Indx[pixLimit.GetId()] = id |
---|
1720 | dzeroSizer.Add(pixLimit,0,WACV) |
---|
1721 | |
---|
1722 | dzeroDelete = wx.CheckBox(parent=G2frame.dataDisplay,label='delete?') |
---|
1723 | dzeroDelete.Bind(wx.EVT_CHECKBOX,OnDeleteDzero) |
---|
1724 | delIndx.append(dzeroDelete) |
---|
1725 | dzeroSizer.Add(dzeroDelete,0,WACV) |
---|
1726 | |
---|
1727 | dzeroSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=(' Strain tensor:')),WACV) |
---|
1728 | names = ['e11','e12','e22'] |
---|
1729 | for i in range(3): |
---|
1730 | dzeroSizer.Add(wx.StaticText(G2frame.dataDisplay,-1,label=names[i]),0,WACV) |
---|
1731 | tensorElem = wx.TextCtrl(G2frame.dataDisplay,-1,value='%.2f'%(dzero['Emat'][i]),style=wx.TE_READONLY) |
---|
1732 | tensorElem.SetBackgroundColour(VERY_LIGHT_GREY) |
---|
1733 | dzeroSizer.Add(tensorElem,0,WACV) |
---|
1734 | dzeroSizer.Add((5,5),0) |
---|
1735 | return dzeroSizer |
---|
1736 | |
---|
1737 | # patches |
---|
1738 | if 'Sample load' not in data: |
---|
1739 | data['Sample load'] = 0.0 |
---|
1740 | # end patches |
---|
1741 | |
---|
1742 | if G2frame.dataDisplay: |
---|
1743 | G2frame.dataDisplay.Destroy() |
---|
1744 | Controls = G2frame.PatternTree.GetItemPyData( |
---|
1745 | G2gd.GetPatternTreeItemId(G2frame,G2frame.Image, 'Image Controls')) |
---|
1746 | G2gd.SetDataMenuBar(G2frame,G2frame.dataFrame.StrStaMenu) |
---|
1747 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnAppendDzero, id=G2gd.wxID_APPENDDZERO) |
---|
1748 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnUpdateDzero, id=G2gd.wxID_UPDATEDZERO) |
---|
1749 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnFitStrSta, id=G2gd.wxID_STRSTAFIT) |
---|
1750 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnFitAllStrSta, id=G2gd.wxID_STRSTAALLFIT) |
---|
1751 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnCopyStrSta, id=G2gd.wxID_STRSTACOPY) |
---|
1752 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnLoadStrSta, id=G2gd.wxID_STRSTALOAD) |
---|
1753 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnSaveStrSta, id=G2gd.wxID_STRSTASAVE) |
---|
1754 | G2frame.dataFrame.Bind(wx.EVT_MENU, OnStrStaSample, id=G2gd.wxID_STRSTSAMPLE) |
---|
1755 | if not G2frame.dataFrame.GetStatusBar(): |
---|
1756 | Status = G2frame.dataFrame.CreateStatusBar() |
---|
1757 | if G2frame.StrainKey == 'a': #probably doesn't happen |
---|
1758 | G2frame.dataFrame.GetStatusBar().SetStatusText('Add strain ring active - LB pick d-zero value') |
---|
1759 | else: |
---|
1760 | G2frame.dataFrame.GetStatusBar().SetStatusText("To add strain data: On 2D Powder Image, key a:add ring") |
---|
1761 | |
---|
1762 | G2frame.dataDisplay = wxscroll.ScrolledPanel(G2frame.dataFrame) |
---|
1763 | mainSizer = wx.BoxSizer(wx.VERTICAL) |
---|
1764 | mainSizer.Add((5,10),0) |
---|
1765 | mainSizer.Add(SamSizer()) |
---|
1766 | mainSizer.Add((5,10),0) |
---|
1767 | mainSizer.Add(DzeroSizer()) |
---|
1768 | |
---|
1769 | mainSizer.Layout() |
---|
1770 | G2frame.dataDisplay.SetSizer(mainSizer) |
---|
1771 | G2frame.dataDisplay.SetAutoLayout(1) |
---|
1772 | G2frame.dataDisplay.SetupScrolling() |
---|
1773 | Size = mainSizer.Fit(G2frame.dataFrame) |
---|
1774 | Size[0] += 25 |
---|
1775 | G2frame.dataDisplay.SetSize(Size) |
---|
1776 | G2frame.dataFrame.setSizePosLeft(Size) |
---|