source: trunk/GSASII.py @ 62

Last change on this file since 62 was 62, checked in by vondreel, 12 years ago

more cleanup of image integration; force 32 bit arrays to reduce memory problems. Some cost in CPU time.
First use of masking to define 2-theta/azimuth limits
Allow multiple image selection for readin.

File size: 51.7 KB
Line 
1#GSASII
2
3import os.path as ospath
4import sys
5import math
6import cPickle
7import time
8import copy
9import numpy as np
10import wx
11import matplotlib as mpl
12
13# load the GSAS routines
14import GSASIIIO as G2IO
15import GSASIIcomp as G2cmp
16import GSASIIgrid as G2gd
17import GSASIIplot as G2plt
18
19# print versions
20print "Available python module versions for pyGSASII:"
21print "python:     ",sys.version[:5]
22print "wxpython:   ",wx.__version__
23print "matplotlib: ",mpl.__version__
24print "numpy:      ",np.__version__
25
26__version__ = '0.1.3'
27
28# useful degree trig functions
29sind = lambda x: math.sin(x*math.pi/180.)
30cosd = lambda x: math.cos(x*math.pi/180.)
31tand = lambda x: math.tan(x*math.pi/180.)
32asind = lambda x: 180.*math.asin(x)/math.pi
33acosd = lambda x: 180.*math.acos(x)/math.pi
34atan2d = lambda x,y: 180.*math.atan2(y,x)/math.pi
35
36def create(parent):
37    return GSASII(parent)
38
39[wxID_GSASII, wxID_GSASIIPATTERNTREE, wxID_GSASIIDATA, wxID_GSASIIPICKGRID,
40] = [wx.NewId() for _init_ctrls in range(4)]
41
42[wxID_GSASIIFILECLOSE, wxID_GSASIIFILEEXIT, wxID_GSASIIFILEOPEN, 
43 wxID_GSASIIFILESAVE, wxID_GSASIIFILESAVEAS, wxID_GSASIIUNDO, 
44] = [wx.NewId() for _init_coll_File_Items in range(6)]
45
46[wxID_GSASIIPWDRREAD,wxID_GSASIISNGLREAD,wxID_GSASIIADDPHASE,wxID_GSASIIDELETEPHASE,
47 wxID_GSASIIDATADELETE,wxID_GSASIIREADPEAKS,wxID_GSASIIPWDSUM,wxID_GSASIIIMGREAD,
48 wxID_GSASIIIMSUM,
49] = [wx.NewId() for _init_coll_Data_Items in range(9)]
50
51[wxID_GSASIIIMPORT, wxID_GSASIIIMPORTPATTERN, wxID_GSASIIIMPORTHKL, wxID_GSASIIIMPORTPHASE,
52wxID_GSASIIIMPORTCIF, wxID_GSASIIIMPORTPDB, 
53] = [wx.NewId() for _init_coll_Import_Items in range(6)]
54
55[wxID_GSAIIEXPORT, wxID_GSASIIEXPORTPATTERN, wxID_GSASIIEXPORTHKL, wxID_GSASIIEXPORTPHASE,
56wxID_GSASIIEXPORTCIF, wxID_GSASIIEXPORTPEAKLIST
57] = [wx.NewId() for _init_coll_Export_Items in range(6)]
58
59[wxID_GSASIIHELPABOUT, wxID_GSASIIHELPHELP, 
60] = [wx.NewId() for _init_coll_Help_Items in range(2)]
61
62class GSASII(wx.Frame):
63   
64    def _init_coll_GSASIIMenu_Menus(self, parent):
65        parent.Append(menu=self.File, title='File')
66        parent.Append(menu=self.Data, title='Data')
67        parent.Append(menu=self.Calculate, title='Calculate')
68        parent.Append(menu=self.Import, title='Import')
69        parent.Append(menu=self.Export, title='Export')
70        parent.Append(menu=self.Help, title='Help')
71
72    def _init_coll_File_Items(self, parent):
73        parent.Append(help='Open a gsasii project file (*.gpx)', id=wxID_GSASIIFILEOPEN,
74             kind=wx.ITEM_NORMAL,text='Open project')
75        parent.Append(help='SAve project to old file', id=wxID_GSASIIFILESAVE, 
76            kind=wx.ITEM_NORMAL,text='Save project')
77        parent.Append(help='Save project to new file', id=wxID_GSASIIFILESAVEAS, 
78            kind=wx.ITEM_NORMAL,text='SaveAs')
79        parent.Append(help='Close project, saving is optional', id=wxID_GSASIIFILECLOSE, 
80            kind=wx.ITEM_NORMAL,text='Close project')
81        parent.Append(help='Exit from gsasii', id=wxID_GSASIIFILEEXIT, kind=wx.ITEM_NORMAL,
82            text='Exit')
83        self.Bind(wx.EVT_MENU, self.OnFileOpenMenu, id=wxID_GSASIIFILEOPEN)
84        self.Bind(wx.EVT_MENU, self.OnFileSaveMenu, id=wxID_GSASIIFILESAVE)
85        self.Bind(wx.EVT_MENU, self.OnFileSaveasMenu, id=wxID_GSASIIFILESAVEAS)
86        self.Bind(wx.EVT_MENU, self.OnFileCloseMenu, id=wxID_GSASIIFILECLOSE)
87        self.Bind(wx.EVT_MENU, self.OnFileExitMenu, id=wxID_GSASIIFILEEXIT)
88       
89    def _init_coll_Data_Items(self,parent):
90        parent.Append(help='', id=wxID_GSASIIPWDRREAD, kind=wx.ITEM_NORMAL,
91            text='Read powder data')
92        parent.Append(help='',id=wxID_GSASIIIMGREAD, kind=wx.ITEM_NORMAL,
93            text='Read image data')
94        parent.Append(help='',id=wxID_GSASIIREADPEAKS, kind=wx.ITEM_NORMAL,
95            text='Read Powder Pattern Peaks')
96        parent.Append(help='', id=wxID_GSASIISNGLREAD, kind=wx.ITEM_NORMAL,
97            text='Read single crystal data')
98        parent.Append(help='', id=wxID_GSASIIPWDSUM, kind=wx.ITEM_NORMAL,
99            text='Sum powder data')
100        parent.Append(help='',id=wxID_GSASIIIMSUM, kind=wx.ITEM_NORMAL,
101            text='Sum image data')
102        parent.Append(help='', id=wxID_GSASIIADDPHASE, kind=wx.ITEM_NORMAL,
103            text='Add phase')
104        parent.Append(help='', id=wxID_GSASIIDELETEPHASE, kind=wx.ITEM_NORMAL,
105            text='Delete phase')
106        parent.Append(help='', id=wxID_GSASIIDATADELETE, kind=wx.ITEM_NORMAL,
107            text='Delete data')
108        self.Bind(wx.EVT_MENU, self.OnPwdrReadMenu, id=wxID_GSASIIPWDRREAD)
109        self.Bind(wx.EVT_MENU, self.OnPwdrSumMenu, id=wxID_GSASIIPWDSUM)
110        self.Bind(wx.EVT_MENU, self.OnReadPowderPeaks, id=wxID_GSASIIREADPEAKS)
111        self.Bind(wx.EVT_MENU, self.OnImageRead, id=wxID_GSASIIIMGREAD)
112        self.Bind(wx.EVT_MENU, self.OnImageSum, id=wxID_GSASIIIMSUM)
113        self.Bind(wx.EVT_MENU, self.OnSnglReadMenu, id=wxID_GSASIISNGLREAD)
114        self.Bind(wx.EVT_MENU, self.OnAddPhase, id=wxID_GSASIIADDPHASE)
115        self.Bind(wx.EVT_MENU, self.OnDeletePhase, id=wxID_GSASIIDELETEPHASE)
116        self.Bind(wx.EVT_MENU, self.OnDataDeleteMenu, id=wxID_GSASIIDATADELETE)
117               
118    def _init_coll_Calculate_Items(self,parent):
119        self.UnDo = parent.Append(help='', id=wxID_GSASIIUNDO, kind=wx.ITEM_NORMAL,
120            text='UnDo')
121        self.UnDo.Enable(False)
122        self.Bind(wx.EVT_MENU, self.OnUnDo, id=wxID_GSASIIUNDO)
123       
124    def _init_coll_Import_Items(self,parent):
125        self.ImportPhase = parent.Append(help='Import phase data from GSAS EXP file',
126            id=wxID_GSASIIIMPORTPHASE, kind=wx.ITEM_NORMAL,text='Import GSAS EXP Phase')
127        self.ImportPDB = parent.Append(help='Import phase data from PDB file',
128            id=wxID_GSASIIIMPORTPDB, kind=wx.ITEM_NORMAL,text='Import PDB Phase')
129        self.ImportPattern = parent.Append(help='',id=wxID_GSASIIIMPORTPATTERN, kind=wx.ITEM_NORMAL,
130            text='Import Powder Pattern')
131        self.ImportHKL = parent.Append(help='',id=wxID_GSASIIIMPORTHKL, kind=wx.ITEM_NORMAL,
132            text='Import HKLs')
133        self.ImportCIF = parent.Append(help='',id=wxID_GSASIIIMPORTCIF, kind=wx.ITEM_NORMAL,
134            text='Import CIF')
135        self.Bind(wx.EVT_MENU, self.OnImportPhase, id=wxID_GSASIIIMPORTPHASE)
136        self.Bind(wx.EVT_MENU, self.OnImportPDB, id=wxID_GSASIIIMPORTPDB)
137        self.Bind(wx.EVT_MENU, self.OnImportPattern, id=wxID_GSASIIIMPORTPATTERN)
138        self.Bind(wx.EVT_MENU, self.OnImportHKL, id=wxID_GSASIIIMPORTHKL)
139        self.Bind(wx.EVT_MENU, self.OnImportCIF, id=wxID_GSASIIIMPORTCIF)
140
141    def _init_coll_Export_Items(self,parent):
142        self.ExportPattern = parent.Append(help='Select PWDR item to enable',id=wxID_GSASIIEXPORTPATTERN, kind=wx.ITEM_NORMAL,
143            text='Export Powder Pattern')
144        self.ExportPeakList = parent.Append(help='',id=wxID_GSASIIEXPORTPEAKLIST, kind=wx.ITEM_NORMAL,
145            text='Export All Peak Lists')
146        self.ExportHKL = parent.Append(help='',id=wxID_GSASIIEXPORTHKL, kind=wx.ITEM_NORMAL,
147            text='Export HKLs')
148        self.ExportPhase = parent.Append(help='',id=wxID_GSASIIEXPORTPHASE, kind=wx.ITEM_NORMAL,
149            text='Export Phase')
150        self.ExportCIF = parent.Append(help='',id=wxID_GSASIIEXPORTCIF, kind=wx.ITEM_NORMAL,
151            text='Export CIF')
152        self.ExportPattern.Enable(False)
153        self.ExportPeakList.Enable(True)
154        self.ExportHKL.Enable(False)
155        self.ExportPhase.Enable(False)
156        self.ExportCIF.Enable(False)
157        self.Bind(wx.EVT_MENU, self.OnExportPattern, id=wxID_GSASIIEXPORTPATTERN)
158        self.Bind(wx.EVT_MENU, self.OnExportPeakList, id=wxID_GSASIIEXPORTPEAKLIST)
159        self.Bind(wx.EVT_MENU, self.OnExportHKL, id=wxID_GSASIIEXPORTHKL)
160        self.Bind(wx.EVT_MENU, self.OnExportPhase, id=wxID_GSASIIEXPORTPHASE)
161        self.Bind(wx.EVT_MENU, self.OnExportCIF, id=wxID_GSASIIEXPORTCIF)
162               
163    def _init_coll_Help_Items(self, parent):
164        parent.Append(help='', id=wxID_GSASIIHELPHELP, kind=wx.ITEM_NORMAL,
165            text='Help')
166        parent.Append(help='', id=wxID_GSASIIHELPABOUT, kind=wx.ITEM_NORMAL,
167            text='About')
168        self.Bind(wx.EVT_MENU, self.OnHelpHelpMenu, id=wxID_GSASIIHELPHELP)
169        self.Bind(wx.EVT_MENU, self.OnHelpAboutMenu, id=wxID_GSASIIHELPABOUT)
170
171    def _init_utils(self):
172        self.GSASIIMenu = wx.MenuBar()
173        self.File = wx.Menu(title='')
174        self.Data = wx.Menu(title='')       
175        self.Calculate = wx.Menu(title='')       
176        self.Import = wx.Menu(title='')       
177        self.Export = wx.Menu(title='')       
178        self.Help = wx.Menu(title='')
179
180        self._init_coll_GSASIIMenu_Menus(self.GSASIIMenu)
181        self._init_coll_File_Items(self.File)
182        self._init_coll_Data_Items(self.Data)
183        self._init_coll_Calculate_Items(self.Calculate)
184        self._init_coll_Import_Items(self.Import)
185        self._init_coll_Export_Items(self.Export)
186        self._init_coll_Help_Items(self.Help)
187       
188    def _init_ctrls(self, parent):
189        wx.Frame.__init__(self, id=wxID_GSASII, name='GSASII', parent=parent,
190            size=wx.Size(300, 250),style=wx.DEFAULT_FRAME_STYLE, title='GSAS-II')
191        screenSize = wx.DisplaySize()
192        Size = self.GetSize()
193        xPos = screenSize[0]-Size[0]
194        self.SetPosition(wx.Point(xPos,0))
195        self._init_utils()
196        self.SetMenuBar(self.GSASIIMenu)
197        self.Bind(wx.EVT_SIZE, self.OnSize)
198        self.CreateStatusBar()
199        self.mainPanel = wx.Panel(self,-1)
200       
201        self.PatternTree = wx.TreeCtrl(id=wxID_GSASIIPATTERNTREE,
202            parent=self.mainPanel, pos=wx.Point(0, 0),style=wx.TR_DEFAULT_STYLE )
203        self.PatternTree.Bind(wx.EVT_TREE_SEL_CHANGED,
204            self.OnPatternTreeSelChanged, id=wxID_GSASIIPATTERNTREE)
205        self.PatternTree.Bind(wx.EVT_TREE_ITEM_COLLAPSED,
206            self.OnPatternTreeItemCollapsed, id=wxID_GSASIIPATTERNTREE)
207        self.PatternTree.Bind(wx.EVT_TREE_ITEM_EXPANDED,
208            self.OnPatternTreeItemExpanded, id=wxID_GSASIIPATTERNTREE)
209        self.root = self.PatternTree.AddRoot("Loaded Data")
210       
211        plotFrame = wx.Frame(None,-1,'GSASII Plots',size=wx.Size(700,600), \
212            style=wx.DEFAULT_FRAME_STYLE ^ wx.CLOSE_BOX)
213        self.G2plotNB = G2plt.G2PlotNoteBook(plotFrame)
214        plotFrame.Show()
215       
216        self.dataDisplay = None
217       
218    def __init__(self, parent):
219        self._init_ctrls(parent)
220        self.Bind(wx.EVT_CLOSE, self.ExitMain)
221        self.dirname = ''
222        self.GSASprojectfile = ''
223        self.Offset = 0.0
224        self.Weight = False
225        self.IparmName = ''
226        self.IfPlot = False
227        self.PatternId = 0
228        self.PickId = 0
229        self.PeakTable = []
230        self.LimitsTable = []
231        self.HKL = []
232        self.Lines = []
233        self.itemPicked = None
234        self.dataFrame = None
235        self.Contour = False
236        self.plotView = 0
237        self.Image = 0
238        self.oldImagefile = ''
239        self.Img = 0
240        self.Integrate = 0
241        self.Pwdr = False
242        self.imageDefault = {}
243        self.Sngl = 0
244        self.ifGetRing = False
245
246    def OnSize(self,event):
247        w,h = self.GetClientSizeTuple()
248        self.mainPanel.SetSize(wx.Size(w,h))
249        self.PatternTree.SetSize(wx.Size(w,h))
250                       
251    def OnPatternTreeSelChanged(self, event):
252        pltNum = self.G2plotNB.nb.GetSelection()
253        if pltNum >= 0:                         #to avoid the startup with no plot!
254            pltPage = self.G2plotNB.nb.GetPage(pltNum)
255            pltPlot = pltPage.figure.gca()
256        item = event.GetItem()
257        G2gd.MovePatternTreeToGrid(self,item)
258       
259    def OnPatternTreeItemCollapsed(self, event):
260        event.Skip()
261
262    def OnPatternTreeItemExpanded(self, event):
263        self.PatternTree.ScrollTo(self.PatternTree.GetLastChild(event.GetItem()))
264       
265    def OnPatternTreeDeleteItem(self, event):
266        event.Skip()
267
268    def OnPatternTreeItemActivated(self, event):
269        event.Skip()
270       
271    def OnPwdrReadMenu(self, event):
272        self.CheckNotebook()
273        dlg = wx.FileDialog(self, 'Choose files', '.', '', 
274            'GSAS fxye files (*.fxye)|*.fxye|GSAS fxy files (*.fxy)|*.fxy|All files (*.*)|*.*', 
275            wx.OPEN | wx.MULTIPLE)
276        if self.dirname: dlg.SetDirectory(self.dirname)
277        try:
278            if dlg.ShowModal() == wx.ID_OK:
279                filenames = dlg.GetPaths()
280                filenames.sort()
281                self.dirname = dlg.GetDirectory()
282                for filename in filenames:
283                    Data,Iparm,Comments = G2IO.SelectPowderData(self, filename)              #Data: list of tuples (filename,Pos,Bank)
284                    if not Data:                                                    #if Data rejected by user - go to next one
285                        continue
286                    DataType = Iparm['INS   HTYPE ']                                #expect only 4 char string
287                    DataType = DataType.strip()[0:3]                                #just 1st 3 chars
288                    wx.BeginBusyCursor()
289                    try:
290                        for Item in Data:
291                            vals = Item[2].split()          #split up the BANK record
292                            Id = self.PatternTree.AppendItem(parent=self.root,text='PWDR '+ospath.basename(Item[0])+': '+vals[0]+vals[1])
293                            data = G2IO.GetPowderData(filename,Item[1],Item[2],DataType)
294                            self.PatternTree.SetItemPyData(Id,[Item,data])
295                            '''
296                            Each tree item data is a list with:
297                            Item: the (filename,Pos,Bank) tuple
298                            data: (x,y,w,yc,yb,yd) list  of np.arrays from GetPowderData
299                            '''
300                           
301                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),Comments)                           
302                            Tmin = min(data[0])
303                            Tmax = max(data[0])
304                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Limits'),[(Tmin,Tmax),[Tmin,Tmax]])
305                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Background'),[['chebyschev',1,3,1.0,0.0,0.0]])
306       
307                            data = [DataType,]
308                            if 'C' in DataType:
309                                s = Iparm['INS  1 ICONS']
310                                v = (G2IO.sfloat(s[:10]),G2IO.sfloat(s[10:20]),G2IO.sfloat(s[20:30]),G2IO.sfloat(s[55:65]),G2IO.sfloat(s[40:50])) #get lam1, lam2, zero, pola & ratio
311                                if not v[1]:
312                                    names = ['Type','Lam','Zero','Polariz.','U','V','W','X','Y','SH/L','Azimuth'] 
313                                    v = (v[0],v[2],v[4])
314                                    codes = [0,0,0,0]
315                                else:
316                                    names = ['Type','Lam1','Lam2','Zero','I(L2)/I(L1)','Polariz.','U','V','W','X','Y','SH/L','Azimuth']
317                                    codes = [0,0,0,0,0,0]
318                                data.extend(v)
319                                v1 = Iparm['INS  1PRCF1 '].split()                                                 
320                                v = Iparm['INS  1PRCF11'].split()
321                                data.extend([float(v[0]),float(v[1]),float(v[2])])                  #get GU, GV & GW - always here
322                                v = Iparm['INS  1PRCF12'].split()
323                                if v1[0] == 3:
324                                    data.extend([float(v[0]),float(v[1]),float(v[2])+float(v[3],0.0)])  #get LX, LY, S+H/L & azimuth
325                                else:
326                                    data.extend([0.0,0.0,0.002,0.0])                                      #OK defaults if fxn #3 not 1st in iprm file
327                                codes.extend([0,0,0,0,0,0,0])
328                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Instrument Parameters'),[tuple(data),data,codes,names])
329                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Peak List'),[])
330                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Index Peak List'),[])
331                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Unit Cells List'),[])             
332                            self.PatternId = G2gd.GetPatternTreeItemId(self,Id,'Limits')
333                    finally:
334                        wx.EndBusyCursor()
335                self.PatternTree.Expand(Id)
336                self.PatternTree.SelectItem(Id)
337   
338        finally:
339            dlg.Destroy()
340       
341    def OnReadPowderPeaks(self,event):
342        Cuka = 1.54052
343        self.CheckNotebook()
344        dlg = wx.FileDialog(self, 'Choose file with peak list', '.', '', 
345            'peak files (*.txt)|*.txt|All files (*.*)|*.*',wx.OPEN)
346        if self.dirname:
347            dlg.SetDirectory(self.dirname)
348        try:
349            if dlg.ShowModal() == wx.ID_OK:
350                self.HKL = []
351                self.powderfile = dlg.GetPath()
352                self.dirname = dlg.GetDirectory()
353                comments,peaks = G2IO.GetPowderPeaks(self.powderfile)
354                Id = self.PatternTree.AppendItem(parent=self.root,text='PKS '+ospath.basename(self.powderfile))
355                data = ['PKS',Cuka,0.0]
356                names = ['Type','Lam','Zero'] 
357                codes = [0,0]
358                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Instrument Parameters'),[tuple(data),data,codes,names])
359                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),comments)
360                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Index Peak List'),peaks)
361                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Unit Cells List'),[])             
362                self.PatternTree.Expand(Id)
363                self.PatternTree.SelectItem(Id)
364        finally:
365            dlg.Destroy()
366           
367    def OnImageRead(self,event):
368        self.CheckNotebook()
369        dlg = wx.FileDialog(self, 'Choose image files', '.', '',\
370        'MAR345 (*.mar3450;*.mar2300)|*.mar3450;*.mar2300|ADSC Image (*.img)\
371        |*.img|Perkin-Elmer TIF (*.tif)|*.tif|GE Image sum (*.sum)\
372        |*.sum|GE Image avg (*.avg)|*.avg|All files (*.*)|*.*',wx.OPEN | wx.MULTIPLE)
373        if self.dirname:
374            dlg.SetDirectory(self.dirname)
375        try:
376            if dlg.ShowModal() == wx.ID_OK:
377                self.dirname = dlg.GetDirectory()
378                imagefiles = dlg.GetPaths()
379                imagefiles.sort()
380                for imagefile in imagefiles:
381                    Comments,Data,Size,Image = G2IO.GetImageData(imagefile)
382                    if Comments:
383                        Id = self.PatternTree.AppendItem(parent=self.root,text='IMG '+ospath.basename(imagefile))
384                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),Comments)
385                        Imax = np.amax(Image)
386                        Imin = np.amin(Image)
387                        if self.imageDefault:
388                            Data = copy.copy(self.imageDefault)
389                            Data['showLines'] = True
390                            Data['ring'] = []
391                            Data['rings'] = []
392                            Data['cutoff'] = 10
393                            Data['pixLimit'] = 20
394                            Data['ellipses'] = []
395                            Data['masks'] = []
396                            Data['calibrant'] = ''
397                        else:
398                            Data['color'] = 'binary'
399                            Data['tilt'] = 0.0
400                            Data['rotation'] = 0.0
401                            Data['showLines'] = False
402                            Data['ring'] = []
403                            Data['rings'] = []
404                            Data['cutoff'] = 10
405                            Data['pixLimit'] = 20
406                            Data['ellipses'] = []
407                            Data['masks'] = []
408                            Data['calibrant'] = ''
409                            Data['IOtth'] = [2.0,5.0]
410                            Data['LRazimuth'] = [-135,-45]
411                            Data['outChannels'] = 2500
412                            Data['outAzimuths'] = 1
413                            Data['fullIntegrate'] = False
414                            Data['setRings'] = False
415                        Data['setDefault'] = False
416                        Data['range'] = [(Imin,Imax),[Imin,Imax]]
417                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Image Controls'),Data)
418                        self.PatternTree.SetItemPyData(Id,[Size,imagefile])
419                        self.PickId = Id
420                        self.Image = Id
421                self.PatternTree.SelectItem(Id)             #show last one
422                self.PatternTree.Expand(Id)
423        finally:
424            dlg.Destroy()
425       
426    def OnSnglReadMenu(self,event):
427        self.CheckNotebook()
428        dlg = wx.FileDialog(self, 'Choose file', '.', '', 
429            'hkl files (*.hkl)|*.hkl|All files (*.*)|*.*', 
430            wx.OPEN)
431        if self.dirname: dlg.SetDirectory(self.dirname)
432        try:
433            if dlg.ShowModal() == wx.ID_OK:
434                filename = dlg.GetPath()
435                self.dirname = dlg.GetDirectory()
436                wx.BeginBusyCursor()
437                try:
438                    Data = {}
439                    names = ['Type','Lam']
440                    HKLref,HKLmin,HKLmax,FoMax,ifFc = G2IO.GetHKLData(filename)
441                    Id = self.PatternTree.AppendItem(parent=self.root,text='SXTL '+ospath.basename(filename))
442                    self.PatternTree.SetItemPyData(Id,HKLref)
443                    Sub = self.PatternTree.AppendItem(Id,text='Instrument Parameters')
444                    data = ['SXC',1.5428,]
445                    self.PatternTree.SetItemPyData(Sub,[tuple(data),data,names])
446                    Data['Type'] = 'Fosq'
447                    Data['ifFc'] = ifFc
448                    Data['HKLmax'] = HKLmax
449                    Data['HKLmin'] = HKLmin
450                    Data['FoMax'] = FoMax
451                    Data['Zone'] = '001'
452                    Data['Layer'] = 0
453                    Data['Scale'] = 1.0
454                    Data['log-lin'] = 'lin'                   
455                    self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='HKL Plot Controls'),Data)
456                    self.PatternTree.SelectItem(Id)
457                    self.PatternTree.Expand(Id)
458                    self.Sngl = Id
459                finally:
460                    wx.EndBusyCursor()   
461        finally:
462            dlg.Destroy()
463           
464    def CheckNotebook(self):
465        if not G2gd.GetPatternTreeItemId(self,self.root,'Notebook'):
466            sub = self.PatternTree.AppendItem(parent=self.root,text='Notebook')
467            self.PatternTree.SetItemPyData(sub,[''])
468            sub = self.PatternTree.AppendItem(parent=self.root,text='Controls')
469            self.PatternTree.SetItemPyData(sub,[0])
470               
471    class SumDialog(wx.Dialog):
472        def __init__(self,parent,title,text,type,data):
473            wx.Dialog.__init__(self,parent,-1,title, 
474                pos=wx.DefaultPosition,style=wx.DEFAULT_DIALOG_STYLE)
475            self.data = data
476            panel = wx.Panel(self)
477            mainSizer = wx.BoxSizer(wx.VERTICAL)
478            topLabl = wx.StaticText(panel,-1,text)
479            mainSizer.Add((10,10),1)
480            mainSizer.Add(topLabl,0,wx.ALIGN_CENTER_VERTICAL|wx.LEFT,10)
481            mainSizer.Add((10,10),1)
482            dataGridSizer = wx.FlexGridSizer(rows=len(data),cols=2,hgap=2,vgap=2)
483            for id,item in enumerate(self.data[:-1]):
484                name = wx.TextCtrl(panel,-1,item[1],size=wx.Size(200,20))
485                name.SetEditable(False)
486                scale = wx.TextCtrl(panel,id,str(item[0]),style=wx.TE_PROCESS_ENTER)
487                scale.Bind(wx.EVT_TEXT,self.OnScaleChange)                   
488                dataGridSizer.Add(scale,0,wx.LEFT,10)
489                dataGridSizer.Add(name,0,wx.RIGHT,10)
490            dataGridSizer.Add(wx.StaticText(panel,-1,'Sum result name: '+type),0, \
491                wx.LEFT|wx.TOP|wx.ALIGN_CENTER_VERTICAL,10)
492            self.name = wx.TextCtrl(panel,-1,self.data[-1],size=wx.Size(200,20),style=wx.TE_PROCESS_ENTER)
493            self.name.Bind(wx.EVT_TEXT,self.OnNameChange)
494            dataGridSizer.Add(self.name,0,wx.RIGHT|wx.TOP,10)
495            mainSizer.Add(dataGridSizer,0,wx.EXPAND)
496            OkBtn = wx.Button(panel,-1,"Ok")
497            OkBtn.Bind(wx.EVT_BUTTON, self.OnOk)
498            cancelBtn = wx.Button(panel,-1,"Cancel")
499            cancelBtn.Bind(wx.EVT_BUTTON, self.OnCancel)
500            btnSizer = wx.BoxSizer(wx.HORIZONTAL)
501            btnSizer.Add((20,20),1)
502            btnSizer.Add(OkBtn)
503            btnSizer.Add((20,20),1)
504            btnSizer.Add(cancelBtn)
505            btnSizer.Add((20,20),1)
506           
507            mainSizer.Add(btnSizer,0,wx.EXPAND|wx.BOTTOM|wx.TOP, 10)
508            panel.SetSizer(mainSizer)
509            panel.Fit()
510            self.Fit()
511           
512        def OnNameChange(self,event):
513            self.data[-1] = self.name.GetValue() 
514           
515        def OnScaleChange(self,event):
516            id = event.GetId()
517            value = self.FindWindowById(id).GetValue()
518            try:
519                self.data[id][0] = float(value)
520                self.FindWindowById(id).SetValue('%.3f'%(self.data[id][0]))
521            except ValueError:
522                if value and '-' not in value[0]:
523                    print 'bad input - numbers only'
524                    self.FindWindowById(id).SetValue('0.0')
525           
526        def OnOk(self,event):
527            parent = self.GetParent()
528            parent.Raise()
529            self.SetReturnCode(wx.ID_OK)
530            self.MakeModal(False)             
531            self.Destroy()
532           
533        def OnCancel(self,event):
534            parent = self.GetParent()
535            parent.Raise()
536            self.SetReturnCode(wx.ID_CANCEL)
537            self.MakeModal(False)             
538            self.Destroy()
539           
540        def GetData(self):
541                return self.data
542           
543    def OnPwdrSumMenu(self,event):
544        TextList = []
545        DataList = []
546        SumList = []
547        Names = []
548        Inst = []
549        SumItemList = []
550        Comments = ['Sum equals: \n']
551        if self.PatternTree.GetCount():
552            item, cookie = self.PatternTree.GetFirstChild(self.root)
553            while item:
554                name = self.PatternTree.GetItemText(item)
555                Names.append(name)
556                if 'PWDR' in name:
557                    TextList.append([0.0,name])
558                    DataList.append(self.PatternTree.GetItemPyData(item)[1])    # (x,y,w,yc,yb,yd)
559                    if not Inst:
560                        Inst = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,item, 'Instrument Parameters'))
561                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
562            if len(TextList) < 2:
563                self.ErrorDialog('Not enough data to sum','There must be more than one "PWDR" pattern')
564                return
565            TextList.append('default sum name')               
566            dlg = self.SumDialog(self,'Sum data','Enter scale for each pattern in summation','PWDR',TextList)
567            try:
568                if dlg.ShowModal() == wx.ID_OK:
569                    lenX = 0
570                    Xminmax = [0,0]
571                    Xsum = []
572                    Ysum = []
573                    Wsum = []
574                    result = dlg.GetData()
575                    for i,item in enumerate(result[:-1]):
576                        scale,name = item
577                        data = DataList[i]
578                        if scale:
579                            Comments.append("%10.3f %s" % (scale,' * '+name))
580                            x,y,w,yc,yb,yd = data
581                            if lenX:
582                                if lenX != len(x):
583                                    self.ErrorDialog('Data length error','Data to be summed must have same number of points'+ \
584                                        '\nExpected:'+str(lenX)+ \
585                                        '\nFound:   '+str(len(x))+'\nfor '+name)
586                                    return
587                            else:
588                                lenX = len(x)
589                            if Xminmax[1]:
590                                if Xminmax != [x[0],x[-1]]:
591                                    self.ErrorDialog('Data range error','Data to be summed must span same range'+ \
592                                        '\nExpected:'+str(Xminmax[0])+' '+str(Xminmax[1])+ \
593                                        '\nFound:   '+str(x[0])+' '+str(x[-1])+'\nfor '+name)
594                                    return
595                                else:
596                                    for j,yi in enumerate(y):
597                                         Ysum[j] += scale*yi
598                            else:
599                                Xminmax = [x[0],x[-1]]
600                                YCsum = YBsum = YDsum = [0.0 for i in range(lenX)]
601                                for j,yi in enumerate(y):
602                                    Xsum.append(x[j])
603                                    Ysum.append(scale*yi)
604                                    Wsum.append(w[j])
605                    outname = 'PWDR '+result[-1]
606                    Id = 0
607                    if outname in Names:
608                        dlg2 = wx.MessageDialog(self,'Overwrite data?','Duplicate data name',wx.OK|wx.CANCEL)
609                        try:
610                            if dlg2.ShowModal() == wx.ID_OK:
611                                Id = G2gd.GetPatternTreeItemId(self,self.root,name)
612                        finally:
613                            dlg2.Destroy()
614                    else:
615                        Id = self.PatternTree.AppendItem(parent=self.root,text=outname)
616                    if Id:
617                        self.PatternTree.SetItemPyData(Id,[[''],[Xsum,Ysum,Wsum,YCsum,YBsum,YDsum]])
618                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),Comments)                   
619                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Limits'),[tuple(Xminmax),Xminmax])
620                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Background'),[['chebyschev',1,3,1.0,0.0,0.0]])
621                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Instrument Parameters'),Inst)
622                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Peak List'),[])
623                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Index Peak List'),[])
624                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Unit Cells List'),[])             
625                        self.PatternTree.SelectItem(Id)
626                        self.PatternTree.Expand(Id)
627                   
628            finally:
629                dlg.Destroy()
630
631    def OnImageSum(self,event):
632        TextList = []
633        DataList = []
634        SumList = []
635        Names = []
636        Inst = []
637        SumItemList = []
638        Comments = ['Sum equals: \n']
639        if self.PatternTree.GetCount():
640            item, cookie = self.PatternTree.GetFirstChild(self.root)
641            while item:
642                name = self.PatternTree.GetItemText(item)
643                Names.append(name)
644                if 'IMG' in name:
645                    TextList.append([0.0,name])
646                    DataList.append(self.PatternTree.GetItemPyData(item))        #Size,Image
647                    Data = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,item,'Image Controls'))
648                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
649            if len(TextList) < 2:
650                self.ErrorDialog('Not enough data to sum','There must be more than one "IMG" pattern')
651                return
652            TextList.append('default sum name')               
653            dlg = self.SumDialog(self,'Sum data','Enter scale for each image in summation','IMG',TextList)
654            try:
655                if dlg.ShowModal() == wx.ID_OK:
656                    imSize = 0
657                    result = dlg.GetData()
658                    First = True
659                    for i,item in enumerate(result[:-1]):
660                        scale,name = item
661                        data = DataList[i]
662                        if scale:                               
663                            Comments.append("%10.3f %s" % (scale,' * '+name))
664                            size,imagefile = data
665                            image = G2IO.GetImageData(imagefile,imageOnly=True)
666                            if First:
667                                newImage = np.zeros_like(image)
668                                First = False
669                            if imSize:
670                                if imSize != size:
671                                    self.ErrorDialog('Image size error','Images to be summed must be same size'+ \
672                                        '\nExpected:'+str(imSize)+ \
673                                        '\nFound:   '+str(size)+'\nfor '+name)
674                                    return
675                                newImage = newImage+scale*image
676                            else:
677                                imSize = size
678                                newImage = newImage+scale*image
679                    newImage = np.asfarray(newImage,dtype=np.float32)                       
680                    print 'result dtype',newImage.dtype                               
681                    outname = 'IMG '+result[-1]
682                    Id = 0
683                    if outname in Names:
684                        dlg2 = wx.MessageDialog(self,'Overwrite data?','Duplicate data name',wx.OK|wx.CANCEL)
685                        try:
686                            if dlg2.ShowModal() == wx.ID_OK:
687                                Id = G2gd.GetPatternTreeItemId(self,self.root,name)
688                        finally:
689                            dlg2.Destroy()
690                    else:
691                        Id = self.PatternTree.AppendItem(parent=self.root,text=outname)
692                    if Id:
693                        dlg = wx.FileDialog(self, 'Choose sum image filename', '.', '', 
694                            'G2img files (*.G2img)|*.G2img', 
695                            wx.SAVE|wx.FD_OVERWRITE_PROMPT)
696                        if self.dirname: dlg.SetDirectory(self.dirname)
697                        if dlg.ShowModal() == wx.ID_OK:
698                            self.dirname = dlg.GetDirectory()
699                            newimagefile = dlg.GetPath()
700                            G2IO.PutG2Image(newimagefile,newImage)
701                            newImage = []
702                            self.PatternTree.SetItemPyData(Id,[imSize,newimagefile])
703                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),Comments)
704                        if self.imageDefault:
705                            Data = copy.copy(self.imageDefault)
706                            Data['showLines'] = True
707                            Data['ring'] = []
708                            Data['rings'] = []
709                            Data['cutoff'] = 10
710                            Data['pixLimit'] = 20
711                            Data['ellipses'] = []
712                            Data['masks'] = []
713                            Data['calibrant'] = ''
714                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Image Controls'),Data)                                           
715                            self.PatternTree.SelectItem(Id)
716                            self.PatternTree.Expand(Id)
717                        self.PickId = Id
718                        self.Image = Id
719            finally:
720                dlg.Destroy()
721                     
722    def OnAddPhase(self,event):
723        if not G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
724            sub = self.PatternTree.AppendItem(parent=self.root,text='Phases')
725        else:
726            sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
727        PhaseName = ''
728        dlg = wx.TextEntryDialog(None,'Enter a name for this phase','Phase Name Entry','New phase',
729            style=wx.OK)
730        if dlg.ShowModal() == wx.ID_OK:
731            PhaseName = dlg.GetValue()
732        dlg.Destroy()
733        sub = self.PatternTree.AppendItem(parent=sub,text=PhaseName)
734        SGData = {'SpGrp':'P 1'}
735        self.PatternTree.SetItemPyData(sub, \
736            {'General':[PhaseName,'nuclear',SGData,[False,10.,10.,10.,90.,90.,90,1000.],
737            [False,1.0],[],{},[],[],[]],'Atoms':[]})
738       
739    def OnDeletePhase(self,event):
740        if self.dataFrame:
741            self.dataFrame.Clear() 
742        TextList = []
743        DelList = []
744        DelItemList = []
745        if G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
746            sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
747        else:
748            return
749        if sub:
750            item, cookie = self.PatternTree.GetFirstChild(sub)
751            while item:
752                TextList.append(self.PatternTree.GetItemText(item))
753                item, cookie = self.PatternTree.GetNextChild(sub, cookie)               
754            dlg = wx.MultiChoiceDialog(self, 'Which phase to delete?', 'Delete phase', TextList, wx.CHOICEDLG_STYLE)
755            try:
756                if dlg.ShowModal() == wx.ID_OK:
757                    result = dlg.GetSelections()
758                    for i in result: DelList.append([i,TextList[i]])
759                    item, cookie = self.PatternTree.GetFirstChild(sub)
760                    i = 0
761                    while item:
762                        if [i,self.PatternTree.GetItemText(item)] in DelList: DelItemList.append(item)
763                        item, cookie = self.PatternTree.GetNextChild(sub, cookie)
764                        i += 1
765                    for item in DelItemList:
766                        self.PatternTree.Delete(item)
767            finally:
768                dlg.Destroy()       
769       
770    def OnDataDeleteMenu(self, event):
771        TextList = []
772        DelList = []
773        DelItemList = []
774        if self.PatternTree.GetCount():
775            item, cookie = self.PatternTree.GetFirstChild(self.root)
776            while item:
777                name = self.PatternTree.GetItemText(item)
778                if 'PWDR' in name or 'SXTL' in name or 'IMG' in name:
779                    TextList.append(name)
780                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)               
781            dlg = wx.MultiChoiceDialog(self, 'Which data to delete?', 'Delete data', TextList, wx.CHOICEDLG_STYLE)
782            try:
783                if dlg.ShowModal() == wx.ID_OK:
784                    result = dlg.GetSelections()
785                    for i in result: DelList.append(TextList[i])
786                    item, cookie = self.PatternTree.GetFirstChild(self.root)
787                    while item:
788                        if self.PatternTree.GetItemText(item) in DelList: DelItemList.append(item)
789                        item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
790                    for item in DelItemList:
791                        self.PatternTree.Delete(item)
792                    G2plt.PlotPatterns(self,True)                        #so plot gets updated
793            finally:
794                dlg.Destroy()
795
796    def OnFileOpenMenu(self, event):
797        result = ''
798        Id = 0
799        if self.PatternTree.GetChildrenCount(self.root,False):
800            if self.dataFrame:
801                self.dataFrame.Clear() 
802            dlg = wx.MessageDialog(self, 'Overwrite?','Project exists!',  wx.OK | wx.CANCEL)
803            try:
804                result = dlg.ShowModal()
805                if result == wx.ID_OK:
806                    self.PatternTree.DeleteChildren(self.root)
807                    print 'children deleted'
808            finally:
809                dlg.Destroy()
810        if result != wx.ID_CANCEL:   
811            if self.dataDisplay: self.dataDisplay.Destroy()
812            dlg = wx.FileDialog(self, 'Choose GSAS-II project file', '.', '', 
813                'GSAS-II project file (*.gpx)|*.gpx',wx.OPEN)
814            if self.dirname: dlg.SetDirectory(self.dirname)
815            try:
816                if dlg.ShowModal() == wx.ID_OK:
817                    self.GSASprojectfile = dlg.GetPath()
818                    self.dirname = dlg.GetDirectory()
819                    G2IO.ProjFileOpen(self)
820                    self.PatternTree.Expand(self.root)
821                    self.HKL = []
822                    item, cookie = self.PatternTree.GetFirstChild(self.root)
823                    while item and not Id:
824                        name = self.PatternTree.GetItemText(item)
825                        if 'PWDR' in name or 'SXTL' in name or 'IMG' in name:
826                            Id = item
827                        item, cookie = self.PatternTree.GetNextChild(self.root, cookie)               
828                    if Id:
829                        self.PatternTree.SelectItem(Id)
830            finally:
831                dlg.Destroy()
832
833    def OnFileCloseMenu(self, event):
834        if self.dataFrame:
835            self.dataFrame.Clear()
836            self.dataFrame.SetLabel('GSAS-II data display') 
837        dlg = wx.MessageDialog(self, 'Save current project?', ' ', wx.YES | wx.NO | wx.CANCEL)
838        try:
839            result = dlg.ShowModal()
840            if result == wx.ID_OK:
841                self.OnFileSaveMenu(event)
842            if result != wx.ID_CANCEL:
843                self.GSASprojectfile = ''
844                self.PatternTree.DeleteChildren(self.root)
845                if self.HKL: self.HKL = []
846        finally:
847            dlg.Destroy()
848
849    def OnFileSaveMenu(self, event):
850        if self.GSASprojectfile: 
851            G2IO.ProjFileSave(self)
852        else:
853            self.OnFileSaveasMenu(event)
854
855    def OnFileSaveasMenu(self, event):
856        dlg = wx.FileDialog(self, 'Choose GSAS-II project file name', '.', '', 
857            'GSAS-II project file (*.gpx)|*.gpx',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
858        if self.dirname:
859            dlg.SetDirectory(self.dirname)
860        try:
861            if dlg.ShowModal() == wx.ID_OK:
862                self.GSASprojectfile = dlg.GetPath()
863                G2IO.ProjFileSave(self)
864                self.dirname = dlg.GetDirectory()
865        finally:
866            dlg.Destroy()
867
868    def ExitMain(self, event):
869        sys.exit()
870       
871    def OnFileExitMenu(self, event):
872        if self.dataFrame:
873            self.dataFrame.Clear() 
874            self.dataFrame.Destroy()
875        self.Close()
876       
877    def OnImportPattern(self,event):
878            dlg = wx.FileDialog(self, 'Choose nonGSAS powder file', '.', '', 
879                '(*.*)|*.*',wx.OPEN)
880            if self.dirname:
881                dlg.SetDirectory(self.dirname)
882            try:
883                if dlg.ShowModal() == wx.ID_OK:
884                    self.powderfile = dlg.GetPath()
885                    self.dirname = dlg.GetDirectory()
886            finally:
887                dlg.Destroy()
888               
889    def OnImportHKL(self,event):
890            dlg = wx.FileDialog(self, 'Choose structure factor file', '.', '', 
891                '(*.*)|*.*',wx.OPEN)
892            if self.dirname:
893                dlg.SetDirectory(self.dirname)
894            try:
895                if dlg.ShowModal() == wx.ID_OK:
896                    self.HKLfile = dlg.GetPath()
897                    self.dirname = dlg.GetDirectory()
898            finally:
899                dlg.Destroy()
900       
901    def OnImportPhase(self,event):
902            dlg = wx.FileDialog(self, 'Choose GSAS EXP file', '.', '', 
903                'EXP file (*.EXP)|*.EXP',wx.OPEN)
904            if self.dirname:
905                dlg.SetDirectory(self.dirname)
906            try:
907                Phase = {}
908                if dlg.ShowModal() == wx.ID_OK:
909                    EXPfile = dlg.GetPath()
910                    self.dirname = dlg.GetDirectory()
911                    Phase = G2IO.ReadEXPPhase(EXPfile)
912            finally:
913                dlg.Destroy()
914            if Phase:
915                PhaseName = Phase['General'][0]
916                if not G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
917                    sub = self.PatternTree.AppendItem(parent=self.root,text='Phases')
918                else:
919                    sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
920                sub = self.PatternTree.AppendItem(parent=sub,text=PhaseName)
921                self.PatternTree.SetItemPyData(sub,Phase)
922               
923    def OnImportPDB(self,event):
924            dlg = wx.FileDialog(self, 'Choose PDB file', '.', '', 
925                'PDB file (*.pdb,*.ent)|*.pdb;*.ent|All files (*.*)|*.*',wx.OPEN)
926            if self.dirname:
927                dlg.SetDirectory(self.dirname)
928            try:
929                if dlg.ShowModal() == wx.ID_OK:
930                    PDBfile = dlg.GetPath()
931                    self.dirname = dlg.GetDirectory()
932                    Phase = G2IO.ReadPDBPhase(PDBfile)
933            finally:
934                dlg.Destroy()
935            if Phase:
936                PhaseName = Phase['General'][0]
937                if not G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
938                    sub = self.PatternTree.AppendItem(parent=self.root,text='Phases')
939                else:
940                    sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
941                sub = self.PatternTree.AppendItem(parent=sub,text=PhaseName)
942                self.PatternTree.SetItemPyData(sub,Phase)       
943       
944    def OnImportCIF(self,event):
945            dlg = wx.FileDialog(self, 'Choose CIF file', '.', '', 
946                'CIF file (*.cif)|*.cif',wx.OPEN)
947            if self.dirname:
948                dlg.SetDirectory(self.dirname)
949            try:
950                if dlg.ShowModal() == wx.ID_OK:
951                    self.CIFfile = dlg.GetPath()
952                    self.dirname = dlg.GetDirectory()
953            finally:
954                dlg.Destroy()
955       
956    def OnExportPattern(self,event):
957        dlg = wx.FileDialog(self, 'Choose output powder file name', '.', '', 
958            'GSAS fxye file (*.fxye)|*.fxye|xye file (*.xye)|*.xye',
959            wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
960        if self.dirname:
961            dlg.SetDirectory(self.dirname)
962        try:
963            if dlg.ShowModal() == wx.ID_OK:
964                powderfile = dlg.GetPath()
965                if 'fxye' in powderfile:
966                    G2IO.powderFxyeSave(self,powderfile)
967                else:       #just xye
968                    G2IO.powderXyeSave(self,powderfile)
969                self.dirname = dlg.GetDirectory()
970        finally:
971            dlg.Destroy()
972       
973    def OnExportPeakList(self,event):
974        dlg = wx.FileDialog(self, 'Choose output peak list file name', '.', '', 
975            '(*.*)|*.*',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
976        if self.dirname:
977            dlg.SetDirectory(self.dirname)
978        try:
979            if dlg.ShowModal() == wx.ID_OK:
980                self.peaklistfile = dlg.GetPath()
981                file = open(self.peaklistfile,'wa')               
982                item, cookie = self.PatternTree.GetFirstChild(self.root)
983                while item:
984                    name = self.PatternTree.GetItemText(item)
985                    if 'PWDR' in name:
986                        item2, cookie2 = self.PatternTree.GetFirstChild(item)
987                        while item2:
988                            name2 = self.PatternTree.GetItemText(item2)
989                            if name2 == 'Peak List':
990                                peaks = self.PatternTree.GetItemPyData(item2)
991                                file.write("%s \n" % (name+' Peak List'))               
992                                for peak in peaks:
993                                    file.write("%10.4f %12.2f %10.3f %10.3f \n" % \
994                                        (peak[0],peak[2],peak[4],peak[6]))
995                            item2, cookie2 = self.PatternTree.GetNextChild(item, cookie2)                           
996                    item, cookie = self.PatternTree.GetNextChild(self.root, cookie)                           
997                file.close()
998                self.dirname = dlg.GetDirectory()
999        finally:
1000            dlg.Destroy()
1001       
1002    def OnExportHKL(self,event):
1003        event.Skip()
1004       
1005    def OnExportPhase(self,event):
1006        event.Skip()
1007       
1008    def OnExportCIF(self,event):
1009        event.Skip()
1010       
1011    def OnUnDo(self,event):
1012        self.DoUnDo()
1013        self.UnDo.Enable(False)
1014       
1015    def DoUnDo(self):
1016        print 'Undo last refinement'
1017        file = open('GSASII.save','rb')
1018        PatternId = self.PatternId
1019        for item in ['Background','Instrument Parameters','Peak List']:
1020            self.PatternTree.SetItemPyData(G2gd.GetPatternTreeItemId(self,PatternId, item),cPickle.load(file))
1021            if self.dataDisplay.GetName() == item:
1022                if item == 'Background':
1023                    G2gd.UpdateBackgroundGrid(self,self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PatternId, item)))
1024                elif item == 'Instrument Parameters':
1025                    G2gd.UpdateInstrumentGrid(self,self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PatternId, item)))
1026                elif item == 'Peak List':
1027                    G2gd.UpdatePeakGrid(self,self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PatternId, item)))
1028            print item,' recovered'
1029        file.close()
1030       
1031    def SaveState(self):
1032        file = open('GSASII.save','wb')
1033        PatternId = self.PatternId
1034        for item in ['Background','Instrument Parameters','Peak List']:
1035            cPickle.dump(self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PatternId,item)),file,1)
1036        file.close()
1037        self.UnDo.Enable(True)
1038               
1039    def ErrorDialog(self,title,message):
1040        dlg = wx.MessageDialog(self, message, title,  wx.OK)
1041        try:
1042            result = dlg.ShowModal()
1043        finally:
1044            dlg.Destroy()
1045
1046    def OnHelpHelpMenu(self, event):
1047        event.Skip()
1048       
1049    def OnHelpAboutMenu(self, event):
1050        info = wx.AboutDialogInfo()
1051        info.Name = 'GSAS-II'
1052        info.Version = '0.0.1'
1053        info.Copyright = '''
1054Robert B. Von Dreele
1055Argonne National Laboratory(C)
1056This product includes software developed
1057by the UChicago Argonne, LLC, as
1058Operator of Argonne National Laboratory.         '''
1059        info.Description = '''
1060General Structure Analysis System - II
1061        '''
1062        wx.AboutBox(info)
1063       
1064class GSASIImain(wx.App):
1065    def OnInit(self):
1066        self.main = GSASII(None)
1067        self.main.Show()
1068        self.SetTopWindow(self.main)
1069        return True
1070
1071def main():
1072    application = GSASIImain(0)
1073    application.MainLoop()
1074
1075if __name__ == '__main__':
1076    main()
Note: See TracBrowser for help on using the repository browser.