source: trunk/GSASIIimgGUI.py @ 1392

Last change on this file since 1392 was 1392, checked in by vondreele, 9 years ago

Speedup GetTthAzmDsp? - ~40% improvement
add size item to UpdateImageData?

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