source: trunk/GSASII.py @ 2619

Last change on this file since 2619 was 2619, checked in by toby, 5 years ago

note when Phase window is resized and use that size

  • Property svn:eol-style set to native
  • Property svn:keywords set to Date Author Revision URL Id
File size: 193.9 KB
Line 
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3#GSASII
4########### SVN repository information ###################
5# $Date: 2017-01-11 04:25:29 +0000 (Wed, 11 Jan 2017) $
6# $Author: toby $
7# $Revision: 2619 $
8# $URL: trunk/GSASII.py $
9# $Id: GSASII.py 2619 2017-01-11 04:25:29Z toby $
10########### SVN repository information ###################
11'''
12*GSAS-II Main Module*
13=====================
14
15Main routines for the GSAS-II program
16'''
17
18import os
19import sys
20import math
21import copy
22import random as ran
23import glob
24import imp
25import inspect
26import numpy as np
27import scipy as sp
28import wx
29import wx.lib.scrolledpanel as wxscroll
30try:  # patch for LANG environment var problem on occasional OSX machines
31    import locale
32    locale.getdefaultlocale()
33except ValueError:
34    print('Fixing location (see https://github.com/matplotlib/matplotlib/issues/5420.)')
35    os.environ['LC_ALL'] = 'en_US.UTF-8'
36    locale.getdefaultlocale()
37import matplotlib as mpl
38try:
39    import OpenGL as ogl
40except ImportError:
41    print('*******************************************************')
42    print('PyOpenGL is missing from your python installation')
43    print('     - we will try to install it')
44    print('*******************************************************')
45    def install_with_easyinstall(package):
46        try: 
47            print "trying a system-wide PyOpenGl install"
48            easy_install.main(['-f',os.path.split(__file__)[0],package])
49            return
50        except:
51            pass
52        try: 
53            print "trying a user level PyOpenGl install"
54            easy_install.main(['-f',os.path.split(__file__)[0],'--user',package])
55            return
56        except:
57            print "Install of '+package+' failed. Please report this information:"
58            import traceback
59            print traceback.format_exc()
60            sys.exit()
61    from setuptools.command import easy_install
62    install_with_easyinstall('PyOpenGl')
63    print('*******************************************************')         
64    print('OpenGL has been installed. Restarting GSAS-II')
65    print('*******************************************************')         
66    loc = os.path.dirname(__file__)
67    import subprocess
68    subprocess.Popen([sys.executable,os.path.join(loc,'GSASII.py')])
69    sys.exit()
70   
71# load the GSAS routines
72import GSASIIpath
73GSASIIpath.SetVersionNumber("$Revision: 2619 $")
74import GSASIIIO as G2IO
75import GSASIIElem as G2elem
76import GSASIIgrid as G2gd
77import GSASIIctrls as G2G
78import GSASIIplot as G2plt
79import GSASIIpwd as G2pwd
80import GSASIIpwdGUI as G2pdG
81import GSASIIspc as G2spc
82import GSASIIstrMain as G2stMn
83import GSASIIstrIO as G2stIO
84import GSASIImath as G2mth
85import GSASIImapvars as G2mv
86import GSASIIobj as G2obj
87import GSASIIlattice as G2lat
88import GSASIIlog as log
89WACV = wx.ALIGN_CENTER_VERTICAL
90#                GSASIIpath.IPyBreak()
91
92__version__ = '0.2.0'
93
94# PATCH: for Mavericks (OS X 10.9.x), wx produces an annoying warning about LucidaGrandeUI.
95# In case stderr has been suppressed there, redirect python error output to stdout. Nobody
96# else should care much about this.
97sys.stderr = sys.stdout
98
99def create(parent):
100    return GSASII(parent)
101
102def SetDefaultDData(dType,histoName,NShkl=0,NDij=0):
103    if dType in ['SXC','SNC']:
104        return {'Histogram':histoName,'Show':False,'Scale':[1.0,True],
105            'Babinet':{'BabA':[0.0,False],'BabU':[0.0,False]},
106            'Extinction':['Lorentzian','None', {'Tbar':0.1,'Cos2TM':0.955,
107            'Eg':[1.e-10,False],'Es':[1.e-10,False],'Ep':[1.e-10,False]}],
108            'Flack':[0.0,False]}
109    elif dType == 'SNT':
110        return {'Histogram':histoName,'Show':False,'Scale':[1.0,True],
111            'Babinet':{'BabA':[0.0,False],'BabU':[0.0,False]},
112            'Extinction':['Lorentzian','None', {
113            'Eg':[1.e-10,False],'Es':[1.e-10,False],'Ep':[1.e-10,False]}]}
114    elif 'P' in dType:
115        return {'Histogram':histoName,'Show':False,'Scale':[1.0,False],
116            'Pref.Ori.':['MD',1.0,False,[0,0,1],0,{},[],0.1],
117            'Size':['isotropic',[1.,1.,1.],[False,False,False],[0,0,1],
118                [1.,1.,1.,0.,0.,0.],6*[False,]],
119            'Mustrain':['isotropic',[1000.0,1000.0,1.0],[False,False,False],[0,0,1],
120                NShkl*[0.01,],NShkl*[False,]],
121            'HStrain':[NDij*[0.0,],NDij*[False,]],                         
122            'Extinction':[0.0,False],'Babinet':{'BabA':[0.0,False],'BabU':[0.0,False]}}
123
124class GSASII(wx.Frame):
125    '''Define the main GSAS-II frame and its associated menu items
126    '''
127    def MenuBinding(self,event):
128        '''Called when a menu is clicked upon; looks up the binding in table
129        '''
130        log.InvokeMenuCommand(event.GetId(),self,event)
131           
132    def Bind(self,eventtype,handler,*args,**kwargs):
133        '''Override the Bind function so that we can wrap calls that will be logged.
134       
135        N.B. This is a bit kludgy. Menu bindings with an id are wrapped and
136        menu bindings with an object and no id are not.
137        '''
138        if eventtype == wx.EVT_MENU and 'id' in kwargs:
139            menulabels = log.SaveMenuCommand(kwargs['id'],self,handler)
140            if menulabels:
141                wx.Frame.Bind(self,eventtype,self.MenuBinding,*args,**kwargs)
142                return
143        wx.Frame.Bind(self,eventtype,handler,*args,**kwargs)     
144   
145    def _Add_FileMenuItems(self, parent):
146        item = parent.Append(
147            help='Open a GSAS-II project file (*.gpx)', id=wx.ID_ANY,
148            kind=wx.ITEM_NORMAL,text='&Open project...')
149        self.Bind(wx.EVT_MENU, self.OnFileOpen, id=item.GetId())
150        item = parent.Append(
151            help='Save project under current name', id=wx.ID_ANY,
152            kind=wx.ITEM_NORMAL,text='&Save project')
153        self.Bind(wx.EVT_MENU, self.OnFileSave, id=item.GetId())
154        item = parent.Append(
155            help='Save current project to new file', id=wx.ID_ANY,
156            kind=wx.ITEM_NORMAL,text='Save project as...')
157        self.Bind(wx.EVT_MENU, self.OnFileSaveas, id=item.GetId())
158        item = parent.Append(
159            help='Create empty new project, saving current is optional', id=wx.ID_ANY,
160            kind=wx.ITEM_NORMAL,text='&New project')
161        self.Bind(wx.EVT_MENU, self.OnFileClose, id=item.GetId())
162        item = parent.Append(wx.ID_PREFERENCES, text = "&Preferences")
163        self.Bind(wx.EVT_MENU, self.OnPreferences, item)
164        if GSASIIpath.GetConfigValue('debug'):
165            def OnIPython(event):
166                GSASIIpath.IPyBreak()
167            item = parent.Append(wx.ID_ANY, text = "IPython Console")
168            self.Bind(wx.EVT_MENU, OnIPython, item)
169        item = parent.Append(
170            help='Exit from GSAS-II', id=wx.ID_ANY,
171            kind=wx.ITEM_NORMAL,text='&Exit')
172        self.Bind(wx.EVT_MENU, self.OnFileExit, id=item.GetId())
173       
174    def _Add_DataMenuItems(self,parent):
175        item = parent.Append(
176            help='',id=wx.ID_ANY,
177            kind=wx.ITEM_NORMAL,
178            text='Read image data...')
179        self.Bind(wx.EVT_MENU, self.OnImageRead, id=item.GetId())
180        item = parent.Append(
181            help='',id=wx.ID_ANY,
182            kind=wx.ITEM_NORMAL,
183            text='Read Powder Pattern Peaks...')
184        self.Bind(wx.EVT_MENU, self.OnReadPowderPeaks, id=item.GetId())
185        item = parent.Append(
186            help='',id=wx.ID_ANY,
187            kind=wx.ITEM_NORMAL,
188            text='Sum powder data')
189        self.Bind(wx.EVT_MENU, self.OnPwdrSum, id=item.GetId())
190        item = parent.Append(
191            help='',id=wx.ID_ANY,
192            kind=wx.ITEM_NORMAL,
193            text='Sum image data')
194        self.Bind(wx.EVT_MENU, self.OnImageSum, id=item.GetId())
195        item = parent.Append(
196            help='',id=wx.ID_ANY,
197            kind=wx.ITEM_NORMAL,
198            text='Add new phase')
199        self.Bind(wx.EVT_MENU, self.OnAddPhase, id=item.GetId())
200        item = parent.Append(
201            help='',id=wx.ID_ANY,
202            kind=wx.ITEM_NORMAL,
203            text='Delete phase')
204        self.Bind(wx.EVT_MENU, self.OnDeletePhase, id=item.GetId())
205        item = parent.Append(
206            help='',id=wx.ID_ANY,
207            kind=wx.ITEM_NORMAL,
208            text='Rename data') 
209        self.Bind(wx.EVT_MENU, self.OnRenameData, id=item.GetId())
210        item = parent.Append(
211            help='',id=wx.ID_ANY,
212            kind=wx.ITEM_NORMAL,
213            text='Delete data')
214        self.Bind(wx.EVT_MENU, self.OnDataDelete, id=item.GetId())
215               
216    def _Add_CalculateMenuItems(self,parent):
217        item = parent.Append(help='Make new PDFs from selected powder patterns', 
218            id=wx.ID_ANY, kind=wx.ITEM_NORMAL,text='Make new PDFs')
219        self.MakePDF.append(item)
220#        item.Enable(False)
221        self.Bind(wx.EVT_MENU, self.OnMakePDFs, id=item.GetId())
222       
223        item = parent.Append(help='View least squares parameters', 
224            id=wx.ID_ANY, kind=wx.ITEM_NORMAL,text='&View LS parms')
225        self.Bind(wx.EVT_MENU, self.ShowLSParms, id=item.GetId())
226       
227        item = parent.Append(help='', id=wx.ID_ANY, kind=wx.ITEM_NORMAL,
228            text='&Refine')
229        if len(self.Refine): # extend state for new menus to match main (on mac)
230            state = self.Refine[0].IsEnabled()
231        else:
232            state = False
233        item.Enable(state)
234        self.Refine.append(item)
235        self.Bind(wx.EVT_MENU, self.OnRefine, id=item.GetId())
236       
237        item = parent.Append(help='', id=wx.ID_ANY, kind=wx.ITEM_NORMAL,
238            text='Sequential refine')
239        self.Bind(wx.EVT_MENU, self.OnSeqRefine, id=item.GetId())
240        if len(self.SeqRefine): # extend state for new menus to match main (on mac)
241            state = self.SeqRefine[0].IsEnabled()
242        else:
243            state = False
244        item.Enable(state)
245        self.SeqRefine.append(item) # save menu obj for use in self.EnableSeqRefineMenu
246#        if GSASIIpath.GetConfigValue('debug'): # allow exceptions for debugging
247#            item = parent.Append(help='', id=wx.ID_ANY, kind=wx.ITEM_NORMAL,
248#                text='tree test')
249#            self.Bind(wx.EVT_MENU, self.TreeTest, id=item.GetId())
250
251    def _init_Imports(self):
252        '''import all the G2phase*.py & G2sfact*.py & G2pwd*.py files that
253        are found in the path
254        '''
255
256        self.ImportPhaseReaderlist = []
257        self._init_Import_routines('phase',self.ImportPhaseReaderlist,'Phase')
258        self.ImportSfactReaderlist = []
259        self._init_Import_routines('sfact',self.ImportSfactReaderlist,'Struct_Factor')
260        self.ImportPowderReaderlist = []
261        self._init_Import_routines('pwd',self.ImportPowderReaderlist,'Powder_Data')
262        self.ImportSmallAngleReaderlist = []
263        self._init_Import_routines('sad',self.ImportSmallAngleReaderlist,'SmallAngle_Data')
264        self.ImportImageReaderlist = []
265        self._init_Import_routines('img',self.ImportImageReaderlist,'Images')
266        self.ImportMenuId = {}
267
268    def _init_Import_routines(self,prefix,readerlist,errprefix):
269        '''import all the import readers matching the prefix
270        '''
271        #path2GSAS2 = os.path.dirname(os.path.realpath(__file__)) # location of this file
272        #pathlist = sys.path[:]
273        #if path2GSAS2 not in pathlist: pathlist.append(path2GSAS2)
274        #path2GSAS2 = os.path.join(
275        #    os.path.dirname(os.path.realpath(__file__)), # location of this file
276        #    'imports')
277        pathlist = sys.path[:]
278        #if path2GSAS2 not in pathlist: pathlist.append(path2GSAS2)
279        if '.' not in pathlist: pathlist.append('.') # insert the directory where G2 is started
280
281        filelist = []
282        for path in pathlist:
283            for filename in glob.iglob(os.path.join(
284                path,
285                "G2"+prefix+"*.py")):
286                filelist.append(filename)   
287                #print 'debug: found',filename
288        filelist = sorted(list(set(filelist))) # remove duplicates
289        for filename in filelist:
290            path,rootname = os.path.split(filename)
291            pkg = os.path.splitext(rootname)[0]
292            try:
293                fp = None
294                fp, fppath,desc = imp.find_module(pkg,[path,])
295                pkg = imp.load_module(pkg,fp,fppath,desc)
296                for clss in inspect.getmembers(pkg): # find classes defined in package
297                    if clss[0].startswith('_'): continue
298                    if inspect.isclass(clss[1]):
299                        # check if we have the required methods
300                        for m in 'Reader','ExtensionValidator','ContentsValidator':
301                            if not hasattr(clss[1],m): break
302                            if not callable(getattr(clss[1],m)): break
303                        else:
304                            reader = clss[1]() # create an import instance
305                            if reader.UseReader:
306                                readerlist.append(reader)
307            except AttributeError:
308                print 'Import_'+errprefix+': Attribute Error '+ filename
309            #except ImportError:
310            #    print 'Import_'+errprefix+': Error importing file '+ filename
311            except Exception,errmsg:
312                print('\nImport_'+errprefix+': Error importing file '+ filename)
313                print('Error message: '+str(errmsg)+'\n')
314            if fp: fp.close()
315
316    def EnableSeqRefineMenu(self):
317        '''Enable or disable the sequential refinement menu items based on the
318        contents of the Controls 'Seq Data' item (if present)
319        '''
320        controls = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,self.root, 'Controls'))
321        if controls.get('Seq Data'):
322            for i in self.SeqRefine: i.Enable(True)
323        else:
324            for i in self.SeqRefine: i.Enable(False)
325
326    def PreviewFile(self,filename,fp):
327        'confirm we have the right file'
328        rdmsg = 'File '+ filename +' begins:\n\n'
329        try:
330            rdmsg += fp.read(80)
331            rdmsg += '\n\nDo you want to read this file?'
332        except UnicodeDecodeError:
333            rdmsg = None
334        if rdmsg is None or not all([ord(c) < 128 and ord(c) != 0 for c in rdmsg]): # show only if ASCII
335            rdmsg = 'File '+ filename +' is a binary file. Do you want to read this file?'
336        # it would be better to use something that
337        # would resize better, but this will do for now
338        dlg = wx.MessageDialog(
339            self, rdmsg,
340            'Is this the file you want?', 
341            wx.YES_NO | wx.ICON_QUESTION,
342            )
343        dlg.SetSize((700,300)) # does not resize on Mac
344        result = wx.ID_NO
345        try:
346            result = dlg.ShowModal()
347        finally:
348            dlg.Destroy()
349        if result == wx.ID_NO: return True
350        return False
351   
352    def OnImportGeneric(self,reader,readerlist,label,multiple=False,
353                        usedRanIdList=[],Preview=True,
354                        load2Tree=False):
355        '''Used for all imports, including Phases, datasets, images...
356
357        Called from :meth:`GSASII.OnImportPhase`, :meth:`GSASII.OnImportImage`,
358        :meth:`GSASII.OnImportSfact`, :meth:`GSASII.OnImportPowder` and
359        :meth:`GSASII.OnImportSmallAngle`
360
361        Uses reader_objects subclassed from :class:`GSASIIIO.ImportPhase`,
362        :class:`GSASIIIO.ImportStructFactor`,
363        :class:`GSASIIIO.ImportPowderData`,
364        :class:`GSASIIIO.ImportSmallAngleData` or
365        :class:`GSASIIIO.ImportImage`.
366        If a specific reader is specified, only that method will be called,
367        but if no reader is specified, every one that is potentially
368        compatible (by file extension) will be tried on the file(s)
369        selected in the Open File dialog.
370
371        :param reader_object reader: This will be a reference to
372          a particular object to be used to read a file or None,
373          if every appropriate reader should be used.
374
375        :param list readerlist: a list of reader objects appropriate for
376          the current read attempt. At present, this will be either
377          self.ImportPhaseReaderlist, self.ImportSfactReaderlist
378          self.ImportPowderReaderlist or self.ImportImageReaderlist
379          (defined in _init_Imports from the files found in the path),
380          but in theory this list could be tailored.
381          Used only when reader is None.
382
383        :param str label: string to place on the open file dialog:
384          Open `label` input file
385
386        :param bool multiple: True if multiple files can be selected
387          in the file dialog. False is default. At present True is used
388          only for reading of powder data.
389
390        :param list usedRanIdList: an optional list of random Ids that
391          have been used and should not be reused
392
393        :param bool Preview: indicates if a preview of the file should
394          be shown. Default is True, but set to False for image files
395          which are all binary.
396
397        :param bool load2Tree: indicates if the file should be loaded
398          into the data tree immediately (used for images only). True
399          only when called from :meth:`OnImportImage`; causes return
400          value to change to a list of True values rather than
401          reader objects.
402
403        :returns: a list of reader objects (rd_list) that were able
404          to read the specified file(s). This list may be empty.
405        '''
406        self.lastimport = ''
407        self.zipfile = None
408        singlereader = True
409        if reader is None:
410            singlereader = False
411            multiple = False
412            #print "use all formats"
413            choices = "any file (*.*)|*.*"
414            choices += "|zip archive (.zip)|*.zip"
415            extdict = {}
416            # compile a list of allowed extensions
417            for rd in readerlist:
418                fmt = rd.formatName
419                for extn in rd.extensionlist:
420                    if not extdict.get(extn): extdict[extn] = []
421                    extdict[extn] += [fmt,]
422            for extn in sorted(extdict.keys(),cmp=lambda x,y: cmp(x.lower(), y.lower())):
423                fmt = ''
424                for f in extdict[extn]:
425                    if fmt != "": fmt += ', '
426                    fmt += f
427                choices += "|" + fmt + " file (*" + extn + ")|*" + extn
428        else:
429            readerlist = [reader,]
430            # compile a list of allowed extensions
431            choices = reader.formatName + " file ("
432            w = ""
433            for extn in reader.extensionlist:
434                if w != "": w += ";"
435                w += "*" + extn
436            choices += w + ")|" + w
437            choices += "|zip archive (.zip)|*.zip"
438            if not reader.strictExtension:
439                choices += "|any file (*.*)|*.*"
440        # get the file(s)
441        if multiple:
442            mode = wx.OPEN|wx.MULTIPLE
443        else:
444            mode = wx.OPEN
445        filelist = G2G.GetImportFile(self,message="Choose "+label+" input file",
446                    defaultFile="",wildcard=choices,style=mode)
447        rd_list = []
448        filelist1 = []
449        for filename in filelist:
450            # is this a zip file?
451            if os.path.splitext(filename)[1].lower() == '.zip':
452                extractedfiles = G2IO.ExtractFileFromZip(
453                    filename,parent=self,
454                    multipleselect=True)
455                if extractedfiles is None: continue # error or Cancel
456                if extractedfiles != filename:
457                    self.zipfile = filename # save zip name
458                    filelist1 += extractedfiles
459                    continue
460            filelist1.append(filename)
461        filelist = filelist1
462        Start = True    #1st time read - clear selections below
463        for filename in filelist:
464            # is this a zip file?
465            if os.path.splitext(filename)[1].lower() == '.zip':
466                extractedfile = G2IO.ExtractFileFromZip(filename,parent=self)
467                if extractedfile is None: continue # error or Cancel
468                if extractedfile != filename:
469                    filename,self.zipfile = extractedfile,filename # now use the file that was created
470            # determine which formats are compatible with this file
471            primaryReaders = []
472            secondaryReaders = []
473            for rd in readerlist:
474                flag = rd.ExtensionValidator(filename)
475                if flag is None: 
476                    secondaryReaders.append(rd)
477                elif flag:
478                    primaryReaders.append(rd)
479            if len(secondaryReaders) + len(primaryReaders) == 0 and reader:
480                self.ErrorDialog('Not supported','The selected reader cannot read file '+filename)
481                return []
482            elif len(secondaryReaders) + len(primaryReaders) == 0:
483                self.ErrorDialog('No Format','No matching format for file '+filename)
484                return []
485
486            fp = None
487            msg = ''
488            fp = open(filename,'Ur')
489            if len(filelist) == 1 and Preview:
490                if self.PreviewFile(filename,fp): return []
491            self.lastimport = filename # this is probably not what I want to do -- it saves only the
492            # last name in a series. See rd.readfilename for a better name.
493
494            # try the file first with Readers that specify the
495            # file's extension and later with ones that merely allow it
496            errorReport = ''
497            for rd in primaryReaders+secondaryReaders:
498                if Start:   #clear old bank selections to allow new ones to be selected by user
499                    rd.selections = []
500                    Start = False
501                rd.ReInitialize() # purge anything from a previous read
502                fp.seek(0)  # rewind
503                rd.errors = "" # clear out any old errors
504                if not rd.ContentsValidator(fp): # rejected on cursory check
505                    errorReport += "\n  "+rd.formatName + ' validator error'
506                    if rd.errors: 
507                        errorReport += ': '+rd.errors
508                    continue 
509                repeat = True
510                rdbuffer = {} # create temporary storage for file reader
511                block = 0
512                while repeat: # loop if the reader asks for another pass on the file
513                    block += 1
514                    repeat = False
515                    fp.seek(0)  # rewind
516                    rd.objname = os.path.basename(filename)
517                    flag = False
518                    if GSASIIpath.GetConfigValue('debug'): # allow exceptions for debugging
519                        flag = rd.Reader(filename,fp,self,
520                                         buffer=rdbuffer,
521                                         blocknum=block,
522                                         usedRanIdList=usedRanIdList,
523                                         )
524                    else:
525                        try:
526                            flag = rd.Reader(filename,fp,self,
527                                             buffer=rdbuffer,
528                                             blocknum=block,
529                                             usedRanIdList=usedRanIdList,
530                                             )
531                        except rd.ImportException as detail:
532                            rd.errors += "\n  Read exception: "+str(detail)
533                        except Exception as detail:
534                            import traceback
535                            rd.errors += "\n  Unhandled read exception: "+str(detail)
536                            rd.errors += "\n  Traceback info:\n"+str(traceback.format_exc())
537                    if flag: # this read succeeded
538                        rd.readfilename = filename
539                        if load2Tree:   #images only
540                            if rd.repeatcount == 1 and not rd.repeat: # skip image number if only one in set
541                                rd.Data['ImageTag'] = None
542                            else:
543                                rd.Data['ImageTag'] = rd.repeatcount
544                            rd.Data['formatName'] = rd.formatName
545#                            print rd.readfilename
546#                            print rd.sumfile
547#                            print rd.Image.shape,rd.Image.dtype
548                            if rd.sumfile:
549                                rd.readfilename = rd.sumfile
550                            G2IO.LoadImage2Tree(rd.readfilename,self,rd.Comments,rd.Data,rd.Npix,rd.Image)
551                            rd_list.append(True) # save a stub the result before it is written over
552                            del rd.Image
553                        else:                                                   
554                            rd_list.append(copy.deepcopy(rd)) # save the result before it is written over
555                        if rd.repeat:
556                            repeat = True
557                        continue
558                    errorReport += '\n'+rd.formatName + ' read error'
559                    if rd.errors:
560                        errorReport += ': '+rd.errors
561                if rd_list: # read succeeded, was there a warning or any errors?
562                    if rd.warnings:
563                        self.ErrorDialog('Read Warning','The '+ rd.formatName+
564                                         ' reader reported a warning message:\n\n'+
565                                         rd.warnings)
566                    break # success in reading, try no further
567            else:
568                if singlereader:
569                    print('The '+ rd.formatName+
570                            ' reader was not able to read file '+filename+msg)
571                    try:
572                        print('\n\nError message(s):\n\t'+errorReport)
573                    except:
574                        pass
575                    self.ErrorDialog('Read Error','The '+ rd.formatName+
576                                     ' reader was not able to read file '+filename+msg)
577                else:
578                    print('No reader was able to read file '+filename+msg)
579                    try:
580                        print('\n\nError message(s):\n\t'+errorReport)
581                    except:
582                        pass
583                    self.ErrorDialog('Read Error','No reader was able to read file '+filename+msg)
584            if fp: fp.close()
585        return rd_list
586
587    def _Add_ImportMenu_Phase(self,parent):
588        '''configure the Import Phase menus accord to the readers found in _init_Imports
589        '''
590        submenu = wx.Menu()
591        item = parent.AppendMenu(wx.ID_ANY, 'Phase',
592            submenu, help='Import phase data')
593        for reader in self.ImportPhaseReaderlist:
594            item = submenu.Append(wx.ID_ANY,help=reader.longFormatName,
595                kind=wx.ITEM_NORMAL,text='from '+reader.formatName+' file')
596            self.ImportMenuId[item.GetId()] = reader
597            self.Bind(wx.EVT_MENU, self.OnImportPhase, id=item.GetId())
598        item = submenu.Append(wx.ID_ANY,
599                              help='Import phase data, use file to try to determine format',
600                              kind=wx.ITEM_NORMAL,
601                              text='guess format from file')
602        self.Bind(wx.EVT_MENU, self.OnImportPhase, id=item.GetId())
603       
604    def OnImportPhase(self,event):
605        '''Called in response to an Import/Phase/... menu item
606        to read phase information.
607        dict self.ImportMenuId is used to look up the specific
608        reader item associated with the menu item, which will be
609        None for the last menu item, which is the "guess" option
610        where all appropriate formats will be tried.
611        '''
612        # look up which format was requested
613        reqrdr = self.ImportMenuId.get(event.GetId())
614       
615        # make a list of phase names, ranId's and the histograms used in those phases
616        phaseRIdList,usedHistograms = self.GetPhaseInfofromTree()
617        phaseNameList = usedHistograms.keys() # phase names in use
618        usedHKLFhists = [] # used single-crystal histograms
619        for p in usedHistograms:
620            for h in usedHistograms[p]:
621                if h.startswith('HKLF ') and h not in usedHKLFhists:
622                    usedHKLFhists.append(h)
623        rdlist = self.OnImportGeneric(reqrdr,
624                                  self.ImportPhaseReaderlist,
625                                  'phase',usedRanIdList=phaseRIdList)
626        if len(rdlist) == 0: return
627        # for now rdlist is only expected to have one element
628        # but below will allow multiple phases to be imported
629        # if ever the import routines ever implement multiple phase reads.
630        self.CheckNotebook()
631        newPhaseList = []
632        for rd in rdlist:
633            PhaseName = ''
634            dlg = wx.TextEntryDialog( # allow editing of phase name
635                                    self, 'Enter the name for the new phase',
636                                    'Edit phase name', rd.Phase['General']['Name'],
637                                    style=wx.OK)
638            while PhaseName == '':
639                dlg.CenterOnParent()
640                if dlg.ShowModal() == wx.ID_OK:
641                    PhaseName = dlg.GetValue().strip()
642                else:
643                    dlg.Destroy()
644                    return
645            dlg.Destroy()
646            # make new phase names unique
647            rd.Phase['General']['Name'] = G2obj.MakeUniqueLabel(PhaseName,phaseNameList)
648            PhaseName = rd.Phase['General']['Name'][:]
649            newPhaseList.append(PhaseName)
650            print 'Read phase '+str(PhaseName)+' from file '+str(self.lastimport)
651            if not G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
652                sub = self.PatternTree.AppendItem(parent=self.root,text='Phases')
653            else:
654                sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
655            psub = self.PatternTree.AppendItem(parent=sub,text=PhaseName)
656            self.PatternTree.SetItemPyData(psub,rd.Phase)
657            self.PatternTree.Expand(self.root) # make sure phases are seen
658            self.PatternTree.Expand(sub) 
659            self.PatternTree.Expand(psub)
660            self.PatternTree.SelectItem(psub) # show the page to complete the initialization (yuk!)
661            wx.Yield() # make sure call of GSASII.OnDataTreeSelChanged happens before we go on
662
663            if rd.Constraints:
664                sub = G2gd.GetPatternTreeItemId(self,self.root,'Constraints') # was created in CheckNotebook if needed
665                Constraints = self.PatternTree.GetItemPyData(sub)               
666                # TODO: make sure that NEWVAR names are unique here?
667                for i in rd.Constraints:
668                    if type(i) is dict:
669                        #for j in i: print j,' --> ',i[j]
670                        if '_Explain' not in Constraints: Constraints['_Explain'] = {}
671                        Constraints['_Explain'].update(i)
672                        continue
673                    Constraints['Phase'].append(i)
674        if not newPhaseList: return # somehow, no new phases
675        # get a list of existing histograms
676        PWDRlist = []
677        HKLFlist = []
678        if self.PatternTree.GetCount():
679            item, cookie = self.PatternTree.GetFirstChild(self.root)
680            while item:
681                name = self.PatternTree.GetItemText(item)
682                if name.startswith('PWDR ') and name not in PWDRlist:
683                    PWDRlist.append(name)
684                if name.startswith('HKLF ') and name not in HKLFlist:
685                    HKLFlist.append(name)
686                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
687        TextList = PWDRlist + HKLFlist
688        if not len(TextList): return # no data loaded yet
689        header = 'Select histogram(s) to add to new phase(s):'
690        for phaseName in newPhaseList:
691            header += '\n  '+str(phaseName)
692
693        notOK = True
694        while notOK:
695            result = G2G.ItemSelector(TextList,self,header,header='Add histogram(s)',multiple=True)
696            if not result: return
697            # check that selected single crystal histograms are not already in use!
698            used = [TextList[i] for i in result if TextList[i] in usedHKLFhists]
699            #for i in result:
700            #    if TextList[i] in usedHKLFhists: used.append(TextList[i])
701            if used:
702                msg = 'The following single crystal histogram(s) are already in use'
703                for i in used:
704                    msg += '\n  '+str(i)
705                msg += '\nAre you sure you want to add them to this phase? '
706                msg += 'Associating a single crystal dataset to >1 histogram is usually an error, '
707                msg += 'so No is suggested here.'
708                if self.ErrorDialog('Likely error',msg,self,wtype=wx.YES_NO) == wx.ID_YES: notOK = False
709            else:
710                notOK = False
711        # connect new phases to histograms
712        sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
713        if not sub:
714            raise Exception('ERROR -- why are there no phases here?')
715        wx.BeginBusyCursor()
716        item, cookie = self.PatternTree.GetFirstChild(sub)
717        while item: # loop over (new) phases
718            phaseName = self.PatternTree.GetItemText(item)
719            data = self.PatternTree.GetItemPyData(item)
720            item, cookie = self.PatternTree.GetNextChild(sub, cookie)
721            if phaseName not in newPhaseList: continue
722            generalData = data['General']
723            SGData = generalData['SGData']
724            Super = generalData.get('Super',0)
725            SuperVec = []
726            if Super:
727                SuperVec = np.array(generalData['SuperVec'][0])
728            UseList = data['Histograms']
729            NShkl = len(G2spc.MustrainNames(SGData))
730            NDij = len(G2spc.HStrainNames(SGData))
731            for i in result:
732                histoName = TextList[i]
733                if histoName in HKLFlist:
734                    #redo UpdateHKLFdata(histoName) here:
735                    Id = G2gd.GetPatternTreeItemId(self,self.root,histoName)
736                    refDict,reflData = self.PatternTree.GetItemPyData(Id)
737                    G,g = G2lat.cell2Gmat(generalData['Cell'][1:7])
738                    Super = reflData.get('Super',0)
739                    for iref,ref in enumerate(reflData['RefList']):
740                        hkl = ref[:3]
741                        if Super:
742                            H = list(hkl+SuperVec*ref[3])
743                        else:
744                            H = hkl
745                        ref[4+Super] = np.sqrt(1./G2lat.calc_rDsq2(H,G))
746                        iabsnt = G2spc.GenHKLf(H,SGData)[0]
747                        if iabsnt:  #flag space gp. absences
748                            if Super: 
749                                if not ref[2+Super]:
750                                    ref[3+Super] = 0
751                                else:
752                                    ref[3+Super] = 1    #twin id
753                            else:
754                                ref[3] = 0
755                    UseList[histoName] = SetDefaultDData(reflData['Type'],histoName)
756                elif histoName in PWDRlist:
757                    Id = G2gd.GetPatternTreeItemId(self,self.root,histoName)
758                    refList = self.PatternTree.GetItemPyData(
759                        G2gd.GetPatternTreeItemId(self,Id,'Reflection Lists'))
760                    refList[generalData['Name']] = {}
761                    UseList[histoName] = SetDefaultDData('PWDR',histoName,NShkl=NShkl,NDij=NDij)
762                else:
763                    raise Exception('Unexpected histogram '+str(histoName))
764        wx.EndBusyCursor()
765        self.EnableRefineCommand()
766        return # success
767       
768    def _Add_ImportMenu_Image(self,parent):
769        '''configure the Import Image menus accord to the readers found in _init_Imports
770        '''
771        submenu = wx.Menu()
772        item = parent.AppendMenu(wx.ID_ANY, 'Image',
773            submenu, help='Import image file')
774        for reader in self.ImportImageReaderlist:
775            item = submenu.Append(wx.ID_ANY,help=reader.longFormatName,
776                kind=wx.ITEM_NORMAL,text='from '+reader.formatName+' file')
777            self.ImportMenuId[item.GetId()] = reader
778            self.Bind(wx.EVT_MENU, self.OnImportImage, id=item.GetId())
779        item = submenu.Append(wx.ID_ANY,
780                              help='Import image data, use file to try to determine format',
781                              kind=wx.ITEM_NORMAL,
782                              text='guess format from file')
783        self.Bind(wx.EVT_MENU, self.OnImportImage, id=item.GetId())
784       
785    def OnImportImage(self,event):
786        '''Called in response to an Import/Image/... menu item
787        to read an image from a file. Like all the other imports,
788        dict self.ImportMenuId is used to look up the specific
789        reader item associated with the menu item, which will be
790        None for the last menu item, which is the "guess" option
791        where all appropriate formats will be tried.
792
793        A reader object is filled each time an image is read.
794        '''
795        self.CheckNotebook()
796        # look up which format was requested
797        reqrdr = self.ImportMenuId.get(event.GetId())
798        rdlist = self.OnImportGeneric(reqrdr,
799                    self.ImportImageReaderlist,
800                    'image',multiple=True,Preview=False,
801                    load2Tree=True)
802        if rdlist: 
803            self.PatternTree.SelectItem(G2gd.GetPatternTreeItemId(self,self.Image,'Image Controls'))             #show last image to have beeen read
804                   
805    def _Add_ImportMenu_Sfact(self,parent):
806        '''configure the Import Structure Factor menus accord to the readers found in _init_Imports
807        '''
808        submenu = wx.Menu()
809        item = parent.AppendMenu(wx.ID_ANY, 'Structure Factor',
810            submenu, help='Import Structure Factor data')
811        for reader in self.ImportSfactReaderlist:
812            item = submenu.Append(wx.ID_ANY,help=reader.longFormatName,               
813                kind=wx.ITEM_NORMAL,text='from '+reader.formatName+' file')
814            self.ImportMenuId[item.GetId()] = reader
815            self.Bind(wx.EVT_MENU, self.OnImportSfact, id=item.GetId())
816        item = submenu.Append(wx.ID_ANY,
817            help='Import Structure Factor, use file to try to determine format',
818            kind=wx.ITEM_NORMAL,
819            text='guess format from file')
820        self.Bind(wx.EVT_MENU, self.OnImportSfact, id=item.GetId())
821
822    def OnImportSfact(self,event):
823        '''Called in response to an Import/Structure Factor/... menu item
824        to read single crystal datasets.
825        dict self.ImportMenuId is used to look up the specific
826        reader item associated with the menu item, which will be
827        None for the last menu item, which is the "guess" option
828        where all appropriate formats will be tried.
829        '''
830        # get a list of existing histograms
831        HKLFlist = []
832        if self.PatternTree.GetCount():
833            item, cookie = self.PatternTree.GetFirstChild(self.root)
834            while item:
835                name = self.PatternTree.GetItemText(item)
836                if name.startswith('HKLF ') and name not in HKLFlist:
837                    HKLFlist.append(name)
838                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
839        # look up which format was requested
840        reqrdr = self.ImportMenuId.get(event.GetId())
841        rdlist = self.OnImportGeneric(reqrdr,self.ImportSfactReaderlist,
842            'Structure Factor',multiple=True)
843        if len(rdlist) == 0: return
844        self.CheckNotebook()
845        newHistList = []
846        for rd in rdlist:
847            HistName = rd.objname
848            if len(rdlist) <= 2: 
849                dlg = wx.TextEntryDialog( # allow editing of Structure Factor name
850                    self, 'Enter the name for the new Structure Factor',
851                    'Edit Structure Factor name', HistName,
852                    style=wx.OK)
853                dlg.CenterOnParent()
854                if dlg.ShowModal() == wx.ID_OK:
855                    HistName = dlg.GetValue()
856                dlg.Destroy()
857            HistName = 'HKLF '+HistName
858            # make new histogram names unique
859            if len(rd.Banks):
860                for Bank in rd.Banks:
861                    valuesdict = {'wtFactor':1.0,'Dummy':False,'ranId':ran.randint(0,sys.maxint),}
862                    HistName = G2obj.MakeUniqueLabel(HistName,HKLFlist)
863                    print 'Read structure factor table '+str(HistName)+' from file '+str(self.lastimport)
864                    Id = self.PatternTree.AppendItem(parent=self.root,text=HistName)
865                    if not Bank['RefDict'].get('FF'):
866                        Bank['RefDict']['FF'] = {}
867                    self.PatternTree.SetItemPyData(Id,[valuesdict,Bank['RefDict']])
868                    Sub = self.PatternTree.AppendItem(Id,text='Instrument Parameters')
869                    self.PatternTree.SetItemPyData(Sub,copy.copy(rd.Parameters))
870                    self.PatternTree.SetItemPyData(
871                        self.PatternTree.AppendItem(Id,text='Reflection List'),{})  #dummy entry for GUI use
872                    newHistList.append(HistName)
873            else:
874                valuesdict = {'wtFactor':1.0,'Dummy':False,'ranId':ran.randint(0,sys.maxint),}
875                HistName = G2obj.MakeUniqueLabel(HistName,HKLFlist)
876                print 'Read structure factor table '+str(HistName)+' from file '+str(self.lastimport)
877                if not rd.RefDict.get('FF'):
878                    rd.RefDict['FF'] = {}
879                Id = self.PatternTree.AppendItem(parent=self.root,text=HistName)
880                self.PatternTree.SetItemPyData(Id,[valuesdict,rd.RefDict])
881                Sub = self.PatternTree.AppendItem(Id,text='Instrument Parameters')
882                self.PatternTree.SetItemPyData(Sub,rd.Parameters)
883                self.PatternTree.SetItemPyData(
884                    self.PatternTree.AppendItem(Id,text='Reflection List'),{})  #dummy entry for GUI use
885                newHistList.append(HistName)
886               
887            self.PatternTree.SelectItem(Id)
888            self.PatternTree.Expand(Id)
889            self.Sngl = True
890
891        if not newHistList: return # somehow, no new histograms
892        # make a list of phase names
893        phaseRIdList,usedHistograms = self.GetPhaseInfofromTree()
894        phaseNameList = usedHistograms.keys() # phase names in use
895        if not phaseNameList: return # no phases yet, nothing to do
896        header = 'Select phase(s) to add the new\nsingle crystal dataset(s) to:'
897        for Name in newHistList:
898            header += '\n  '+str(Name)
899        result = G2G.ItemSelector(phaseNameList,self,header,header='Add to phase(s)',multiple=True)
900        if not result: return
901        # connect new phases to histograms
902        sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
903        if not sub:
904            raise Exception('ERROR -- why are there no phases here?')
905        wx.BeginBusyCursor()
906        item, cookie = self.PatternTree.GetFirstChild(sub)
907        iph = -1
908        while item: # loop over (new) phases
909            iph += 1
910            data = self.PatternTree.GetItemPyData(item)
911            item, cookie = self.PatternTree.GetNextChild(sub, cookie)
912            if iph not in result: continue
913            generalData = data['General']
914            SGData = generalData['SGData']
915            Super = generalData.get('Super',0)
916            SuperVec = []
917            if Super:
918                SuperVec = np.array(generalData['SuperVec'][0])
919            UseList = data['Histograms']
920            for histoName in newHistList:
921                #redo UpdateHKLFdata(histoName) here:
922                Id = G2gd.GetPatternTreeItemId(self,self.root,histoName)
923                refDict,reflData = self.PatternTree.GetItemPyData(Id)
924                UseList[histoName] = SetDefaultDData(reflData['Type'],histoName)
925                G,g = G2lat.cell2Gmat(generalData['Cell'][1:7])
926                if 'TwMax' in reflData:     #nonmerohedral twins present
927                    UseList[histoName]['Twins'] = []
928                    for iT in range(reflData['TwMax'][0]+1):
929                        if iT in reflData['TwMax'][1]:
930                            UseList[histoName]['Twins'].append([False,0.0])
931                        else:
932                            UseList[histoName]['Twins'].append([np.array([[1,0,0],[0,1,0],[0,0,1]]),[1.0,False,reflData['TwMax'][0]]])
933                else:   #no nonmerohedral twins
934                    UseList[histoName]['Twins'] = [[np.array([[1,0,0],[0,1,0],[0,0,1]]),[1.0,False,0]],]
935                for iref,ref in enumerate(reflData['RefList']):
936                    hkl = ref[:3]
937                    if Super:
938                        H = list(hkl+SuperVec*ref[3])
939                    else:
940                        H = hkl
941                    ref[4+Super] = np.sqrt(1./G2lat.calc_rDsq2(H,G))
942                    iabsnt,mul,Uniq,phi = G2spc.GenHKLf(H,SGData)
943                    if iabsnt:  #flag space gp. absences
944                        if Super: 
945                            if not ref[2+Super]:
946                                ref[3+Super] = 0
947                            else:
948                                ref[3+Super] = 1    #twin id?
949                        else:
950                            ref[3] = 0
951        wx.EndBusyCursor()
952        self.EnableRefineCommand()       
953        return # success
954
955    def _Add_ImportMenu_powder(self,parent):
956        '''configure the Powder Data menus accord to the readers found in _init_Imports
957        '''
958        submenu = wx.Menu()
959        item = parent.AppendMenu(wx.ID_ANY, 'Powder Data',
960            submenu, help='Import Powder data')
961        for reader in self.ImportPowderReaderlist:
962            item = submenu.Append(wx.ID_ANY,help=reader.longFormatName,
963                kind=wx.ITEM_NORMAL,text='from '+reader.formatName+' file')
964            self.ImportMenuId[item.GetId()] = reader
965            self.Bind(wx.EVT_MENU, self.OnImportPowder, id=item.GetId())
966        item = submenu.Append(wx.ID_ANY,
967            help='Import powder data, use file to try to determine format',
968            kind=wx.ITEM_NORMAL,text='guess format from file')
969        self.Bind(wx.EVT_MENU, self.OnImportPowder, id=item.GetId())
970        submenu.AppendSeparator()
971        item = submenu.Append(wx.ID_ANY,
972            help='Create a powder data set entry that will be simulated',
973            kind=wx.ITEM_NORMAL,text='Simulate a dataset')
974        self.Bind(wx.EVT_MENU, self.OnDummyPowder, id=item.GetId())
975       
976    def OpenPowderInstprm(self,instfile):
977        '''Read a GSAS-II (new) instrument parameter file
978
979        :param str instfile: name of instrument parameter file
980
981        '''
982        File = open(instfile,'r')
983        lines = File.readlines()
984        File.close()
985        return lines       
986           
987    def ReadPowderInstprm(self,instLines,bank,databanks,rd):
988        '''Read lines from a GSAS-II (new) instrument parameter file
989        similar to G2pwdGUI.OnLoad
990        If instprm file has multiple banks each with header #Bank n: ..., this
991        finds matching bank no. to load - problem with nonmatches?
992
993        :param list instLines: strings from GSAS-II parameter file; can be concatenated with ';'
994        :param int  bank: bank number to check when instprm file has '#BANK n:...' strings
995            when bank = n then use parameters; otherwise skip that set. Ignored if BANK n:
996            not present. NB: this kind of instprm file made by a Save all profile command in Instrument Parameters
997        :return dict: Inst  instrument parameter dict if OK, or
998                str: Error message if failed   
999        '''
1000        if 'GSAS-II' not in instLines[0]: # not a valid file
1001            return 'Not a valid GSAS-II instprm file'
1002        newItems = []
1003        newVals = []
1004        Found = False
1005        il = 0
1006        if bank is None: # no bank was specified in the input file, is more than one present in file?
1007            banklist = set([])
1008            for S in instLines:
1009                if S[0] == '#' and 'Bank' in S:
1010                    banklist.add(int(S.split(':')[0].split()[1]))
1011            if len(banklist) > 1: # yes, the user must make a selection
1012                choices = [str(i) for i in banklist]
1013                bank = int(G2G.ItemSelector(choices,self.G2frame,multiple=False))
1014            else:
1015                bank = 1
1016            rd.powderentry[2] = bank
1017        while il < len(instLines):
1018            S = instLines[il]
1019            if S[0] == '#':
1020                if Found:
1021                    break
1022                if 'Bank' in S:
1023                    if bank == int(S.split(':')[0].split()[1]):
1024                        il += 1
1025                        S = instLines[il]
1026                    else:
1027                        il += 1
1028                        S = instLines[il]
1029                        while il < len(instLines) and '#Bank' not in S:
1030                            il += 1
1031                            if il == len(instLines):
1032                                return 'Bank %d not found in instprm file'%(bank)
1033                            S = instLines[il]
1034                        continue
1035                else:   #a non #Bank file
1036                    il += 1
1037                    S = instLines[il]
1038            Found = True
1039            if '"""' in S:
1040                delim = '"""' 
1041            elif "'''" in S:
1042                delim = "'''"
1043            else:
1044                S = S.replace(' ','')
1045                SS = S.strip().split(';')
1046                for s in SS:
1047                    [item,val] = s.split(':',1)
1048                    newItems.append(item)
1049                    try:
1050                        newVals.append(float(val))
1051                    except ValueError:
1052                        newVals.append(val)
1053                il += 1
1054                continue
1055            # read multiline values, delimited by ''' or """
1056            item,val = S.strip().split(':',1)
1057            val = val.replace(delim,'').rstrip()
1058            val += '\n'
1059            while True:
1060                il += 1
1061                if il >= len(instLines): break
1062                S = instLines[il]
1063                if delim in S:
1064                    val += S.replace(delim,'').rstrip()
1065                    val += '\n'
1066                    break
1067                else:
1068                    val += S.rstrip()
1069                    val += '\n'
1070            newItems.append(item)
1071            newVals.append(val)
1072            il += 1
1073        return [G2IO.makeInstDict(newItems,newVals,len(newVals)*[False,]),{}]
1074       
1075    def ReadPowderIparm(self,instfile,bank,databanks,rd):
1076        '''Read a GSAS (old) instrument parameter file
1077
1078        :param str instfile: name of instrument parameter file
1079        :param int bank: the bank number read in the raw data file
1080        :param int databanks: the number of banks in the raw data file.
1081          If the number of banks in the data and instrument parameter files
1082          agree, then the sets of banks are assumed to match up and bank
1083          is used to select the instrument parameter file. If not, the user
1084          is asked to make a selection.
1085        :param obj rd: the raw data (histogram) data object. This
1086          sets rd.instbank.
1087
1088        '''
1089        if not os.path.exists(instfile): # no such file
1090            return {}
1091        fp = 0
1092        try:
1093            fp = open(instfile,'Ur')
1094            Iparm = {}
1095            for S in fp:
1096                if '#' in S[0]:
1097                    continue
1098                Iparm[S[:12]] = S[12:-1]
1099        except IOError:
1100            print('Error reading file:'+str(instfile))
1101        if fp:       
1102            fp.close()
1103
1104        ibanks = int(Iparm.get('INS   BANK  ','1').strip())
1105        if ibanks == 1: # there is only one bank here, return it
1106            rd.instbank = 1
1107            rd.powderentry[2] = 1
1108            return Iparm
1109#        if 'PNT' in hType:  # not sure what this is about
1110#            rd.instbank = bank
1111#        elif ibanks != databanks or bank is None:
1112        if (ibanks != databanks and databanks != 1) or bank is None:
1113            # number of banks in data and prm file not not agree, need a
1114            # choice from a human here
1115            choices = []
1116            for i in range(1,1+ibanks):
1117                choices.append('Bank '+str(i))
1118            bank = 1 + rd.BlockSelector(
1119                choices, self,
1120                title='Select an instrument parameter bank for '+
1121                os.path.split(rd.powderentry[0])[1]+' BANK '+str(bank)+
1122                '\nOr use Cancel to select from the default parameter sets',
1123                header='Block Selector')
1124        if bank is None: return {}
1125        # pull out requested bank # bank from the data, and change the bank to 1
1126        IparmS = {}
1127        for key in Iparm:
1128            if 'INS' in key[:3]:    #skip around rubbish lines in some old iparm files
1129                if key[4:6] == "  ":
1130                    IparmS[key] = Iparm[key]
1131                elif int(key[4:6].strip()) == bank:
1132                    IparmS[key[:4]+' 1'+key[6:]] = Iparm[key]
1133        rd.instbank = bank
1134        return IparmS
1135                       
1136    def GetPowderIparm(self,rd, prevIparm, lastIparmfile, lastdatafile):
1137        '''Open and read an instrument parameter file for a data file
1138        Returns the list of parameters used in the data tree
1139
1140        :param obj rd: the raw data (histogram) data object.
1141
1142        :param str prevIparm: not used
1143
1144        :param str lastIparmfile: Name of last instrument parameter
1145          file that was read, or a empty string.
1146
1147        :param str lastdatafile: Name of last data file that was read.
1148
1149        :returns: a list of two dicts, the first containing instrument parameters
1150          and the second used for TOF lookup tables for profile coeff.
1151
1152        '''
1153        def SetPowderInstParms(Iparm, rd):
1154            '''extracts values from instrument parameters in rd.instdict
1155            or in array Iparm.
1156            Create and return the contents of the instrument parameter tree entry.
1157            '''
1158            Irads = {0:' ',1:'CrKa',2:'FeKa',3:'CuKa',4:'MoKa',5:'AgKa',6:'TiKa',7:'CoKa'}
1159            DataType = Iparm['INS   HTYPE '].strip()[:3]  # take 1st 3 chars
1160            # override inst values with values read from data file
1161            Bank = rd.powderentry[2]    #should be used in multibank iparm files
1162            if rd.instdict.get('type'):
1163                DataType = rd.instdict.get('type')
1164            data = [DataType,]
1165            instname = Iparm.get('INS  1INAME ')
1166            irad = int(Iparm.get('INS  1 IRAD ','0'))
1167            if instname:
1168                rd.Sample['InstrName'] = instname.strip()
1169            if 'C' in DataType:
1170                wave1 = None
1171                wave2 = 0.0
1172                if rd.instdict.get('wave'):
1173                    wl = rd.instdict.get('wave')
1174                    wave1 = wl[0]
1175                    if len(wl) > 1: wave2 = wl[1]
1176                s = Iparm['INS  1 ICONS']
1177                if not wave1:
1178                    wave1 = G2IO.sfloat(s[:10])
1179                    wave2 = G2IO.sfloat(s[10:20])
1180                v = (wave1,wave2,
1181                     G2IO.sfloat(s[20:30]),G2IO.sfloat(s[55:65]),G2IO.sfloat(s[40:50])) #get lam1, lam2, zero, pola & ratio
1182                if not v[1]:
1183                    names = ['Type','Lam','Zero','Polariz.','U','V','W','X','Y','SH/L','Azimuth'] 
1184                    v = (v[0],v[2],v[4])
1185                    codes = [0,0,0,0]
1186                else:
1187                    names = ['Type','Lam1','Lam2','Zero','I(L2)/I(L1)','Polariz.','U','V','W','X','Y','SH/L','Azimuth']
1188                    codes = [0,0,0,0,0,0]
1189                data.extend(v)
1190                if 'INS  1PRCF  ' in Iparm:
1191                    v1 = Iparm['INS  1PRCF  '].split()                                                 
1192                    v = Iparm['INS  1PRCF 1'].split()
1193                    data.extend([float(v[0]),float(v[1]),float(v[2])])                  #get GU, GV & GW - always here
1194                    azm = float(Iparm.get('INS  1DETAZM','0.0'))
1195                    v = Iparm['INS  1PRCF 2'].split()
1196                    if v1[0] == 3:
1197                        data.extend([float(v[0]),float(v[1]),float(v[2])+float(v[3],azm)])  #get LX, LY, S+H/L & azimuth
1198                    else:
1199                        data.extend([0.0,0.0,0.002,azm])                                      #OK defaults if fxn #3 not 1st in iprm file                   
1200                else:
1201                    v1 = Iparm['INS  1PRCF1 '].split()                                                 
1202                    v = Iparm['INS  1PRCF11'].split()
1203                    data.extend([float(v[0]),float(v[1]),float(v[2])])                  #get GU, GV & GW - always here
1204                    azm = float(Iparm.get('INS  1DETAZM','0.0'))
1205                    v = Iparm['INS  1PRCF12'].split()
1206                    if v1[0] == 3:
1207                        data.extend([float(v[0]),float(v[1]),float(v[2])+float(v[3],azm)])  #get LX, LY, S+H/L & azimuth
1208                    else:
1209                        data.extend([0.0,0.0,0.002,azm])                                      #OK defaults if fxn #3 not 1st in iprm file
1210                codes.extend([0,0,0,0,0,0,0])
1211                Iparm1 = G2IO.makeInstDict(names,data,codes)
1212                Iparm1['Source'] = [Irads[irad],Irads[irad]]
1213                Iparm1['Bank'] = [Bank,Bank,0]
1214                return [Iparm1,{}]
1215            elif 'T' in DataType:
1216                names = ['Type','fltPath','2-theta','difC','difA', 'difB','Zero','alpha','beta-0','beta-1',
1217                    'beta-q','sig-0','sig-1','sig-2','sig-q', 'X','Y','Azimuth',]
1218                codes = [0,0,0,0,0, 0,0,0,0,0, 0,0,0,0,0, 0,0,0,]
1219                azm = 0.
1220                if 'INS  1DETAZM' in Iparm:
1221                    azm = float(Iparm['INS  1DETAZM'])
1222                s = Iparm['INS   FPATH1'].split()
1223                fltPath0 = G2IO.sfloat(s[0])
1224                if 'INS  1BNKPAR' not in Iparm:     #bank missing from Iparm file
1225                    return []
1226                s = Iparm['INS  1BNKPAR'].split()
1227                fltPath1 = G2IO.sfloat(s[0])
1228                data.extend([fltPath0+fltPath1,])               #Flight path source-sample-detector
1229                data.extend([G2IO.sfloat(s[1]),])               #2-theta for bank
1230                s = Iparm['INS  1 ICONS'].split()
1231                data.extend([G2IO.sfloat(s[0]),G2IO.sfloat(s[1]),0.0,G2IO.sfloat(s[2])])    #difC,difA,difB,Zero
1232                if 'INS  1PRCF  ' in Iparm:
1233                    s = Iparm['INS  1PRCF  '].split()
1234                    pfType = int(s[0])
1235                    s = Iparm['INS  1PRCF 1'].split()
1236                    if abs(pfType) == 1:
1237                        data.extend([G2IO.sfloat(s[1]),G2IO.sfloat(s[2]),G2IO.sfloat(s[3])]) #alpha, beta-0, beta-1
1238                        s = Iparm['INS  1PRCF 2'].split()
1239                        data.extend([0.0,0.0,G2IO.sfloat(s[1]),G2IO.sfloat(s[2]),0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y
1240                    elif abs(pfType) in [3,4,5]:
1241                        data.extend([G2IO.sfloat(s[0]),G2IO.sfloat(s[1]),G2IO.sfloat(s[2])]) #alpha, beta-0, beta-1
1242                        if abs(pfType) == 4:
1243                            data.extend([0.0,0.0,G2IO.sfloat(s[3]),0.0,0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y
1244                        else:
1245                            s = Iparm['INS  1PRCF 2'].split()
1246                            data.extend([0.0,0.0,G2IO.sfloat(s[0]),G2IO.sfloat(s[1]),0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y                       
1247                    elif abs(pfType) == 2:
1248                        data.extend([G2IO.sfloat(s[1]),0.0,1./G2IO.sfloat(s[3])]) #alpha, beta-0, beta-1
1249                        data.extend([0.0,0.0,G2IO.sfloat(s[1]),0.0,0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y                           
1250                else:
1251                    s = Iparm['INS  1PRCF1 '].split()
1252                    pfType = int(s[0])
1253                    s = Iparm['INS  1PRCF11'].split()
1254                    if abs(pfType) == 1:
1255                        data.extend([G2IO.sfloat(s[1]),G2IO.sfloat(s[2]),G2IO.sfloat(s[3])]) #alpha, beta-0, beta-1
1256                        s = Iparm['INS  1PRCF12'].split()
1257                        data.extend([0.0,0.0,G2IO.sfloat(s[1]),G2IO.sfloat(s[2]),0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y
1258                    elif abs(pfType) in [3,4,5]:
1259                        data.extend([G2IO.sfloat(s[0]),G2IO.sfloat(s[1]),G2IO.sfloat(s[2])]) #alpha, beta-0, beta-1
1260                        if abs(pfType) == 4:
1261                            data.extend([0.0,0.0,G2IO.sfloat(s[3]),0.0,0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y
1262                        else:
1263                            s = Iparm['INS  1PRCF12'].split()
1264                            data.extend([0.0,0.0,G2IO.sfloat(s[0]),G2IO.sfloat(s[1]),0.0,0.0,0.0,azm])    #beta-q, sig-0, sig-1, sig-2, sig-q, X, Y                       
1265                Inst1 = G2IO.makeInstDict(names,data,codes)
1266                Inst1['Bank'] = [Bank,Bank,0]
1267                Inst2 = {}
1268                if pfType < 0:
1269                    Ipab = 'INS  1PAB'+str(-pfType)
1270                    Npab = int(Iparm[Ipab+'  '].strip())
1271                    Inst2['Pdabc'] = []
1272                    for i in range(Npab):
1273                        k = Ipab+str(i+1).rjust(2)
1274                        s = Iparm[k].split()
1275                        Inst2['Pdabc'].append([float(t) for t in s])
1276                    Inst2['Pdabc'] = np.array(Inst2['Pdabc'])
1277                    Inst2['Pdabc'].T[3] += Inst2['Pdabc'].T[0]*Inst1['difC'][0] #turn 3rd col into TOF
1278                if 'INS  1I ITYP' in Iparm:
1279                    s = Iparm['INS  1I ITYP'].split()
1280                    Ityp = int(s[0])
1281                    Tminmax = [float(s[1])*1000.,float(s[2])*1000.]
1282                    Itypes = ['Exponential','Maxwell/Exponential','','Maxwell/Chebyschev','']
1283                    if Ityp in [1,2,4]:
1284                        Inst2['Itype'] = Itypes[Ityp-1]
1285                        Inst2['Tminmax'] = Tminmax
1286                        Icoeff = []
1287                        Iesd = []
1288                        Icovar = []                   
1289                        for i in range(3):
1290                            s = Iparm['INS  1ICOFF'+str(i+1)].split()
1291                            Icoeff += [float(S) for S in s]
1292                            s = Iparm['INS  1IECOF'+str(i+1)].split()
1293                            Iesd += [float(S) for S in s]
1294                        NT = 10
1295                        for i in range(8):
1296                            s = Iparm['INS  1IECOR'+str(i+1)]
1297                            if i == 7:
1298                                NT = 8
1299                            Icovar += [float(s[6*j:6*j+6]) for j in range(NT)]
1300                        Inst2['Icoeff'] = Icoeff
1301                        Inst2['Iesd'] = Iesd
1302                        Inst2['Icovar'] = Icovar
1303                return [Inst1,Inst2]
1304               
1305        def GetDefaultParms(self,rd):
1306            '''Solicits from user a default set of parameters & returns Inst parm dict
1307            param: self: refers to the GSASII main class
1308            param: rd: importer data structure
1309            returns: dict: Instrument parameter dictionary
1310            '''       
1311            sind = lambda x: math.sin(x*math.pi/180.)
1312            tand = lambda x: math.tan(x*math.pi/180.)
1313            import defaultIparms as dI
1314            while True: # loop until we get a choice
1315                choices = []
1316                head = 'Select from default instrument parameters for '+rd.idstring
1317   
1318                for l in dI.defaultIparm_lbl:
1319                    choices.append('Defaults for '+l)
1320                res = rd.BlockSelector(choices,ParentFrame=self,title=head,
1321                    header='Select default inst parms',useCancel=True)
1322                if res is None: return None
1323                rd.instfile = ''
1324                if 'Generic' in choices[res]:
1325                    dlg = G2G.MultiFloatDialog(self,title='Generic TOF detector bank',
1326                        prompts=['Total FP','2-theta',],values=[25.0,150.,],
1327                            limits=[[6.,200.],[5.,175.],],formats=['%6.2f','%6.1f',])
1328                    if dlg.ShowModal() == wx.ID_OK: #strictly empirical approx.
1329                        FP,tth = dlg.GetValues()
1330                        difC = 505.632*FP*sind(tth/2.)
1331                        sig1 = 50.+2.5e-6*(difC/tand(tth/2.))**2
1332                        bet1 = .00226+7.76e+11/difC**4
1333                        rd.instmsg = 'default: '+dI.defaultIparm_lbl[res]
1334                        Inst = self.ReadPowderInstprm(dI.defaultIparms[res],bank,numbanks,rd)
1335                        Inst[0]['difC'] = [difC,difC,0]
1336                        Inst[0]['sig-1'] = [sig1,sig1,0]
1337                        Inst[0]['beta-1'] = [bet1,bet1,0]
1338                        return Inst    #this is [Inst1,Inst2] a pair of dicts
1339                    dlg.Destroy()
1340                else:
1341                    rd.instmsg = 'default: '+dI.defaultIparm_lbl[res]
1342                    return self.ReadPowderInstprm(dI.defaultIparms[res],bank,numbanks,rd)    #this is [Inst1,Inst2] a pair of dicts
1343
1344        # stuff we might need from the reader
1345        filename = rd.powderentry[0]
1346        bank = rd.powderentry[2]
1347        numbanks = rd.numbanks
1348        #1st priority: is there an instrument parameter file matching the current file
1349        # with extension .instprm, .prm, .inst, or .ins? If so read it
1350        basename = os.path.splitext(filename)[0]
1351        for ext in '.prm','.inst','.ins','.instprm':
1352            if self.zipfile:
1353                instfile = G2IO.ExtractFileFromZip(self.zipfile,
1354                    selection=os.path.split(basename + ext)[1],parent=self)
1355                if instfile == None:
1356                    continue
1357            else:
1358                instfile = basename + ext
1359            if not os.path.exists(instfile):
1360                continue
1361            if 'instprm' in instfile:
1362                Lines = self.OpenPowderInstprm(instfile)
1363                instParmList = self.ReadPowderInstprm(Lines,bank,numbanks,rd)    #this is [Inst1,Inst2] a pair of dicts
1364                if 'list' in str(type(instParmList)):
1365                    rd.instfile = instfile
1366                    rd.instmsg = 'GSAS-II file '+instfile
1367                    return instParmList
1368                else:
1369                    #print 'debug: open/read failed',instfile
1370                    pass # fail silently
1371            else:
1372                Iparm = self.ReadPowderIparm(instfile,bank,numbanks,rd)
1373                if Iparm:
1374                    #print 'debug: success'
1375                    rd.instfile = instfile
1376                    rd.instmsg = instfile + ' bank ' + str(rd.instbank)
1377                    return SetPowderInstParms(Iparm,rd)
1378                else:
1379                    #print 'debug: open/read failed',instfile
1380                    pass # fail silently
1381
1382        #2nd priority: is there an instrument parameter file defined for the current data set?
1383        # or if this is a read on a set of set of files, use the last one again
1384        #if rd.instparm as found in data file header or (lastdatafile == filename and lastIparmfile):
1385        if rd.instparm or lastIparmfile:
1386            if rd.instparm:
1387                instfile = os.path.join(os.path.split(filename)[0],rd.instparm)
1388            else:
1389                # for multiple reads of one data file, reuse the inst parm file
1390                instfile = lastIparmfile
1391#            if self.zipfile:
1392#                instfile = G2IO.ExtractFileFromZip(self.zipfile,
1393#                    selection=os.path.split(instfile)[1],parent=self)
1394            if instfile != None and os.path.exists(instfile):
1395                #print 'debug: try read',instfile
1396                if 'instprm' in instfile:   #GSAS-II file must have .instprm as extension
1397                    Lines = self.OpenPowderInstprm(instfile)
1398                    if Lines is not None:
1399                        instParmList = self.ReadPowderInstprm(Lines,bank,numbanks,rd)   #this is [Inst1,Inst2] a pair of dicts
1400                else:   #old GSAS style iparm file - could be named anything!
1401                    Iparm = self.ReadPowderIparm(instfile,bank,numbanks,rd)
1402                    if Iparm:
1403                        #print 'debug: success'
1404                        rd.instfile = instfile
1405                        rd.instmsg = instfile + ' bank ' + str(rd.instbank)
1406                        instParmList = SetPowderInstParms(Iparm,rd)     #this is [Inst1,Inst2] a pair of dicts
1407                if 'list' in str(type(instParmList)):   #record stuff & return stuff
1408                    rd.instfile = instfile
1409                    rd.instmsg = 'GSAS-II file '+instfile
1410                    return instParmList
1411                else:   #bad iparms - try default
1412                    rd.instmsg = instParmList   #an error message
1413                    return GetDefaultParms(self,rd)
1414            else:
1415                self.ErrorDialog('Open Error','Error opening instrument parameter file '
1416                    +str(instfile)+' requested by file '+ filename)
1417        #Finally - ask user for Instrument parametrs file - seems it can't be in a zip file
1418        while True: # loop until we get a file that works or we get a cancel
1419            instfile = ''
1420            pth = G2G.GetImportPath(self)
1421            if not pth: pth = '.'
1422            dlg = wx.FileDialog(self,
1423                'Choose inst. param file for "'+rd.idstring+'" (or Cancel for default)',
1424                pth, '',
1425                'GSAS iparm file (*.prm,*.inst,*.ins)|*.prm;*.inst;*.ins|'
1426                'GSAS-II iparm file (*.instprm)|*.instprm|'
1427                'All files (*.*)|*.*', wx.OPEN)
1428            if os.path.exists(lastIparmfile):
1429                dlg.SetFilename(lastIparmfile)
1430            if dlg.ShowModal() == wx.ID_OK:
1431                instfile = dlg.GetPath()
1432            dlg.Destroy()
1433            if not instfile: 
1434                return GetDefaultParms(self,rd) #on Cancel/break
1435            if 'instprm' in instfile:
1436                Lines = self.OpenPowderInstprm(instfile)
1437                if Lines is not None:
1438                    instParmList = self.ReadPowderInstprm(Lines,bank,numbanks,rd)    #this is [Inst1,Inst2] a pair of dicts
1439                if 'list' in str(type(instParmList)):
1440                    rd.instfile = instfile
1441                    rd.instmsg = 'GSAS-II file '+instfile
1442                    return instParmList
1443                else:
1444                    rd.instmsg = instParmList   #an error message
1445                    return GetDefaultParms(self,rd)
1446            else:
1447                Iparm = self.ReadPowderIparm(instfile,bank,numbanks,rd)
1448                if Iparm:
1449                    #print 'debug: success with',instfile
1450                    rd.instfile = instfile
1451                    rd.instmsg = instfile + ' bank ' + str(rd.instbank)
1452                    return SetPowderInstParms(Iparm,rd)
1453                else:
1454                    self.ErrorDialog('Read Error',
1455                                     'Error opening/reading file '+str(instfile))
1456    def EnableRefineCommand(self):
1457        haveData = False
1458        # check for phases connected to histograms
1459        sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
1460        if not sub: return
1461        item, cookie = self.PatternTree.GetFirstChild(sub)
1462        while item: # loop over phases
1463            data = self.PatternTree.GetItemPyData(item)
1464            item, cookie = self.PatternTree.GetNextChild(sub, cookie)
1465            UseList = data['Histograms']
1466            if UseList: haveData = True
1467        if haveData:
1468            self.dataFrame.DataMenu.Enable(G2gd.wxID_DATADELETE,True)
1469            for item in self.Refine: item.Enable(True)
1470        else:
1471            self.dataFrame.DataMenu.Enable(G2gd.wxID_DATADELETE,False)
1472            for item in self.Refine: item.Enable(False)
1473
1474       
1475    def OnImportPowder(self,event):
1476        '''Called in response to an Import/Powder Data/... menu item
1477        to read a powder diffraction data set.
1478        dict self.ImportMenuId is used to look up the specific
1479        reader item associated with the menu item, which will be
1480        None for the last menu item, which is the "guess" option
1481        where all appropriate formats will be tried.
1482
1483        Also reads an instrument parameter file for each dataset.
1484        '''
1485        # get a list of existing histograms
1486        PWDRlist = []
1487        if self.PatternTree.GetCount():
1488            item, cookie = self.PatternTree.GetFirstChild(self.root)
1489            while item:
1490                name = self.PatternTree.GetItemText(item)
1491                if name.startswith('PWDR ') and name not in PWDRlist:
1492                    PWDRlist.append(name)
1493                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
1494        # look up which format was requested
1495        reqrdr = self.ImportMenuId.get(event.GetId()) 
1496        rdlist = self.OnImportGeneric(
1497            reqrdr,self.ImportPowderReaderlist,'Powder Data',multiple=True)
1498        if len(rdlist) == 0: return
1499        self.CheckNotebook()
1500        Iparm = None
1501        lastIparmfile = ''
1502        lastdatafile = ''
1503        newHistList = []
1504        self.EnablePlot = False
1505        for rd in rdlist:
1506            if 'Instrument Parameters' not in rd.pwdparms:
1507                # get instrument parameters for each dataset, unless already set
1508                Iparms = self.GetPowderIparm(rd, Iparm, lastIparmfile, lastdatafile)
1509                if not Iparms:  #may have bailed out
1510                    Id = 0
1511                    continue
1512                Iparm1,Iparm2 = Iparms
1513                if rd.repeat_instparm: 
1514                    lastIparmfile = rd.instfile
1515                # override any keys in read instrument parameters with ones set in import
1516                for key in Iparm1: 
1517                    if key in rd.instdict:
1518                        Iparm1[key] = rd.instdict[key]
1519            else:
1520                Iparm1,Iparm2 = rd.pwdparms['Instrument Parameters']
1521            lastdatafile = rd.powderentry[0]
1522            HistName = rd.idstring
1523            HistName = 'PWDR '+HistName
1524            # make new histogram names unique
1525            if HistName in PWDRlist:
1526                dlg = wx.MessageDialog(self,'Skip %s?'%(HistName),'Duplicate data name',wx.YES_NO)
1527                try:
1528                    if dlg.ShowModal() == wx.ID_YES:
1529                        Id = 0
1530                        continue
1531                finally:
1532                    dlg.Destroy()
1533            HistName = G2obj.MakeUniqueLabel(HistName,PWDRlist)
1534            print('Read powder data '+HistName+ 
1535                ' from file '+rd.readfilename +
1536                ' (format: '+ rd.formatName + 
1537                '). Inst parameters from '+rd.instmsg)
1538            # data are read, now store them in the tree
1539            Id = self.PatternTree.AppendItem(parent=self.root,text=HistName)
1540            if 'T' in Iparm1['Type'][0]:
1541                if not rd.clockWd and rd.GSAS:
1542                    rd.powderdata[0] *= 100.        #put back the CW centideg correction
1543                cw = np.diff(rd.powderdata[0])
1544                rd.powderdata[0] = rd.powderdata[0][:-1]+cw/2.
1545                if rd.GSAS:     #NB: old GSAS wanted intensities*CW even if normalized!
1546                    npts = min(len(rd.powderdata[0]),len(rd.powderdata[1]),len(cw))
1547                    rd.powderdata[1] = rd.powderdata[1][:npts]/cw[:npts]
1548                    rd.powderdata[2] = rd.powderdata[2][:npts]*cw[:npts]**2  #1/var=w at this point
1549                else:       #NB: from topas/fullprof type files
1550                    rd.powderdata[1] = rd.powderdata[1][:-1]
1551                    rd.powderdata[2] = rd.powderdata[2][:-1]
1552                if 'Itype' in Iparm2:
1553                    Ibeg = np.searchsorted(rd.powderdata[0],Iparm2['Tminmax'][0])
1554                    Ifin = np.searchsorted(rd.powderdata[0],Iparm2['Tminmax'][1])
1555                    rd.powderdata[0] = rd.powderdata[0][Ibeg:Ifin]
1556                    YI,WYI = G2pwd.calcIncident(Iparm2,rd.powderdata[0])
1557                    rd.powderdata[1] = rd.powderdata[1][Ibeg:Ifin]/YI
1558                    var = 1./rd.powderdata[2][Ibeg:Ifin]
1559                    var += WYI*rd.powderdata[1]**2
1560                    var /= YI**2
1561                    rd.powderdata[2] = 1./var
1562                rd.powderdata[3] = np.zeros_like(rd.powderdata[0])
1563                rd.powderdata[4] = np.zeros_like(rd.powderdata[0])
1564                rd.powderdata[5] = np.zeros_like(rd.powderdata[0])
1565            Ymax = np.max(rd.powderdata[1])                 
1566            valuesdict = {
1567                'wtFactor':1.0,
1568                'Dummy':False,
1569                'ranId':ran.randint(0,sys.maxint),
1570                'Offset':[0.0,0.0],'delOffset':0.02*Ymax,'refOffset':-.1*Ymax,'refDelt':0.1*Ymax,
1571                'qPlot':False,'dPlot':False,'sqrtPlot':False
1572                }
1573            # apply user-supplied corrections to powder data
1574            if 'CorrectionCode' in Iparm1:
1575                print('Applying corrections from instprm file')
1576                corr = Iparm1['CorrectionCode'][0]
1577                try:
1578                    exec(corr)
1579                    print('done')
1580                except Exception as err:
1581                    print('error: '+str(err))
1582                    print('with commands -------------------')
1583                    print(str(corr))
1584                    print('---------------------------------')
1585                finally:
1586                    del Iparm1['CorrectionCode']
1587            rd.Sample['ranId'] = valuesdict['ranId'] # this should be removed someday
1588            self.PatternTree.SetItemPyData(Id,[valuesdict,rd.powderdata])
1589            self.PatternTree.SetItemPyData(
1590                self.PatternTree.AppendItem(Id,text='Comments'),
1591                rd.comments)
1592            Tmin = min(rd.powderdata[0])
1593            Tmax = max(rd.powderdata[0])
1594            self.PatternTree.SetItemPyData(
1595                self.PatternTree.AppendItem(Id,text='Limits'),
1596                rd.pwdparms.get('Limits',[(Tmin,Tmax),[Tmin,Tmax]])
1597                )
1598            self.PatternId = G2gd.GetPatternTreeItemId(self,Id,'Limits')
1599            self.PatternTree.SetItemPyData(
1600                self.PatternTree.AppendItem(Id,text='Background'),
1601                rd.pwdparms.get('Background',
1602                    [['chebyschev',True,3,1.0,0.0,0.0],{'nDebye':0,'debyeTerms':[],'nPeaks':0,'peaksList':[]}])
1603                    )
1604            self.PatternTree.SetItemPyData(
1605                self.PatternTree.AppendItem(Id,text='Instrument Parameters'),
1606                [Iparm1,Iparm2])
1607            self.PatternTree.SetItemPyData(
1608                self.PatternTree.AppendItem(Id,text='Sample Parameters'),
1609                rd.Sample)
1610            self.PatternTree.SetItemPyData(
1611                self.PatternTree.AppendItem(Id,text='Peak List')
1612                ,{'peaks':[],'sigDict':{}})
1613            self.PatternTree.SetItemPyData(
1614                self.PatternTree.AppendItem(Id,text='Index Peak List'),
1615                [[],[]])
1616            self.PatternTree.SetItemPyData(
1617                self.PatternTree.AppendItem(Id,text='Unit Cells List'),
1618                [])
1619            self.PatternTree.SetItemPyData(
1620                self.PatternTree.AppendItem(Id,text='Reflection Lists'),
1621                {})
1622            # if any Control values have been set, move them into tree
1623            Controls = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,self.root, 'Controls'))
1624            Controls.update(rd.Controls)
1625            newHistList.append(HistName)
1626            rd.repeat_instparm = False  #clear the iparm reuse flag
1627        else:
1628            self.EnablePlot = True
1629            if Id:
1630                self.PatternTree.Expand(Id)
1631                self.PatternTree.SelectItem(Id)
1632
1633        if not newHistList: return # somehow, no new histograms
1634        # make a list of phase names
1635        phaseRIdList,usedHistograms = self.GetPhaseInfofromTree()
1636        phaseNameList = usedHistograms.keys() # phase names in use
1637        if not phaseNameList: return # no phases yet, nothing to do
1638        header = 'Select phase(s) to link\nto the newly-read data:'
1639        for Name in newHistList:
1640            header += '\n  '+str(Name)
1641
1642        result = G2G.ItemSelector(phaseNameList,self,header,header='Add to phase(s)',multiple=True)
1643        if not result: return
1644        # connect new phases to histograms
1645        sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
1646        if not sub:
1647            raise Exception('ERROR -- why are there no phases here?')
1648        item, cookie = self.PatternTree.GetFirstChild(sub)
1649        iph = -1
1650        while item: # loop over (new) phases
1651            iph += 1
1652            data = self.PatternTree.GetItemPyData(item)
1653            item, cookie = self.PatternTree.GetNextChild(sub, cookie)
1654            if iph not in result: continue
1655            generalData = data['General']
1656            SGData = generalData['SGData']
1657            UseList = data['Histograms']
1658            NShkl = len(G2spc.MustrainNames(SGData))
1659            NDij = len(G2spc.HStrainNames(SGData))
1660            for histoName in newHistList:
1661                UseList[histoName] = SetDefaultDData('PWDR',histoName,NShkl=NShkl,NDij=NDij)
1662                Id = G2gd.GetPatternTreeItemId(self,self.root,histoName)
1663                refList = self.PatternTree.GetItemPyData(
1664                    G2gd.GetPatternTreeItemId(self,Id,'Reflection Lists'))
1665                refList[generalData['Name']] = []
1666        self.EnableRefineCommand()
1667        return # success
1668
1669    def OnDummyPowder(self,event):
1670        '''Called in response to Import/Powder Data/Simulate menu item
1671        to create a Dummy powder diffraction data set.
1672
1673        Reads an instrument parameter file and then gets input from the user
1674        '''
1675        # get a list of existing histograms
1676        PWDRlist = []
1677        if self.PatternTree.GetCount():
1678            item, cookie = self.PatternTree.GetFirstChild(self.root)
1679            while item:
1680                name = self.PatternTree.GetItemText(item)
1681                if name.startswith('PWDR ') and name not in PWDRlist:
1682                    PWDRlist.append(name)
1683                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
1684        # Initialize a base class reader
1685        rd = G2IO.ImportPowderData(
1686            extensionlist=tuple(),
1687            strictExtension=False,
1688            formatName = 'Simulate dataset',
1689            longFormatName = 'Compute a simulated pattern')
1690        rd.powderentry[0] = '' # no filename
1691        # #self.powderentry[1] = pos # bank offset (N/A here)
1692        rd.powderentry[2] = 1 # only one bank
1693        rd.comments.append('This is a dummy dataset for powder pattern simulation')
1694        self.CheckNotebook()
1695        Iparm = None
1696        lastIparmfile = ''
1697        lastdatafile = ''
1698        self.zipfile = None
1699        # get instrument parameters for it
1700        Iparm1,Iparm2 = self.GetPowderIparm(rd, Iparm, lastIparmfile, lastdatafile)
1701        if 'T' in Iparm1['Type'][0]:
1702            print('TOF simulation not supported yet')
1703            return False
1704        else:
1705            # need to get name, 2theta start, end, step
1706            rd.idstring = ' CW'
1707            if 'X' in Iparm1['Type'][0]:
1708                rd.idstring = 'CW x-ray simulation'
1709            else:
1710                rd.idstring = 'CW neutron simulation'
1711            # base initial range on wavelength
1712            wave = Iparm1.get('Lam')
1713            if wave:
1714                wave = wave[0]
1715            else:
1716                wave = Iparm1.get('Lam1')
1717                if wave:
1718                    wave = wave[0]
1719        N = 0
1720        while (N < 3): # insist on a dataset with a few points
1721            names = ('dataset name', 'start angle', 'end angle', 'step size')
1722            if not wave or wave < 1.0:
1723                inp = [rd.idstring, 10.,40.,0.005] # see names for what's what
1724            else:
1725                inp = [rd.idstring, 10.,80.,0.01] # see names for what's what
1726            dlg = G2G.ScrolledMultiEditor(
1727                self,[inp] * len(inp),range(len(inp)),names,
1728                header='Enter simulation name and range',
1729                minvals=(None,0.001,0.001,0.0001),
1730                maxvals=(None,180.,180.,.1),
1731                sizevals=((225,-1),)
1732                )
1733            dlg.CenterOnParent()
1734            if dlg.ShowModal() == wx.ID_OK:
1735                if inp[1] > inp[2]:
1736                    end,start,step = inp[1:]
1737                else:               
1738                    start,end,step = inp[1:]
1739                step = abs(step)
1740            else:
1741                return False
1742            N = int((end-start)/step)+1
1743            x = np.linspace(start,end,N,True)
1744            N = len(x)
1745        rd.powderdata = [
1746            np.array(x), # x-axis values
1747            np.zeros_like(x), # powder pattern intensities
1748            np.ones_like(x), # 1/sig(intensity)^2 values (weights)
1749            np.zeros_like(x), # calc. intensities (zero)
1750            np.zeros_like(x), # calc. background (zero)
1751            np.zeros_like(x), # obs-calc profiles
1752            ]
1753        Tmin = rd.powderdata[0][0]
1754        Tmax = rd.powderdata[0][-1]
1755        # data are read, now store them in the tree
1756        HistName = inp[0]
1757        HistName = 'PWDR '+HistName
1758        HistName = G2obj.MakeUniqueLabel(HistName,PWDRlist)  # make new histogram names unique
1759        Id = self.PatternTree.AppendItem(parent=self.root,text=HistName)
1760        Ymax = np.max(rd.powderdata[1])
1761        valuesdict = {
1762            'wtFactor':1.0,
1763            'Dummy':True,
1764            'ranId':ran.randint(0,sys.maxint),
1765            'Offset':[0.0,0.0],'delOffset':0.02*Ymax,'refOffset':-.1*Ymax,'refDelt':0.1*Ymax,
1766            'qPlot':False,'dPlot':False,'sqrtPlot':False
1767            }
1768        self.PatternTree.SetItemPyData(Id,[valuesdict,rd.powderdata])
1769        self.PatternTree.SetItemPyData(
1770            self.PatternTree.AppendItem(Id,text='Comments'),
1771            rd.comments)
1772        self.PatternTree.SetItemPyData(
1773            self.PatternTree.AppendItem(Id,text='Limits'),
1774            [(Tmin,Tmax),[Tmin,Tmax]])
1775        self.PatternId = G2gd.GetPatternTreeItemId(self,Id,'Limits')
1776        self.PatternTree.SetItemPyData(
1777            self.PatternTree.AppendItem(Id,text='Background'),
1778            [['chebyschev',True,3,1.0,0.0,0.0],
1779             {'nDebye':0,'debyeTerms':[],'nPeaks':0,'peaksList':[]}])
1780        self.PatternTree.SetItemPyData(
1781            self.PatternTree.AppendItem(Id,text='Instrument Parameters'),
1782            [Iparm1,Iparm2])
1783        self.PatternTree.SetItemPyData(
1784            self.PatternTree.AppendItem(Id,text='Sample Parameters'),
1785            rd.Sample)
1786        self.PatternTree.SetItemPyData(
1787            self.PatternTree.AppendItem(Id,text='Peak List')
1788            ,{'peaks':[],'sigDict':{}})
1789        self.PatternTree.SetItemPyData(
1790            self.PatternTree.AppendItem(Id,text='Index Peak List'),
1791            [[],[]])
1792        self.PatternTree.SetItemPyData(
1793            self.PatternTree.AppendItem(Id,text='Unit Cells List'),
1794            [])
1795        self.PatternTree.SetItemPyData(
1796            self.PatternTree.AppendItem(Id,text='Reflection Lists'),
1797            {})
1798        self.PatternTree.Expand(Id)
1799        self.PatternTree.SelectItem(Id)
1800        print('Added simulation powder data '+str(HistName)+ 
1801              ' with parameters from '+str(rd.instmsg))
1802
1803        # make a list of phase names
1804        phaseRIdList,usedHistograms = self.GetPhaseInfofromTree()
1805        phaseNameList = usedHistograms.keys() # phase names in use
1806        if not phaseNameList: return # no phases yet, nothing to do
1807        header = 'Select phase(s) to add the new\npowder simulation (dummy) dataset to:'
1808        result = G2G.ItemSelector(phaseNameList,self,header,header='Add to phase(s)',multiple=True)
1809        if not result: return
1810        # connect new phases to histograms
1811        sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
1812        if not sub:
1813            raise Exception('ERROR -- why are there no phases here?')
1814        item, cookie = self.PatternTree.GetFirstChild(sub)
1815        iph = -1
1816        while item: # loop over (new) phases
1817            iph += 1
1818            data = self.PatternTree.GetItemPyData(item)
1819            item, cookie = self.PatternTree.GetNextChild(sub, cookie)
1820            if iph not in result: continue
1821            generalData = data['General']
1822            SGData = generalData['SGData']
1823            UseList = data['Histograms']
1824            NShkl = len(G2spc.MustrainNames(SGData))
1825            NDij = len(G2spc.HStrainNames(SGData))
1826            UseList[HistName] = SetDefaultDData('PWDR',HistName,NShkl=NShkl,NDij=NDij)
1827            Id = G2gd.GetPatternTreeItemId(self,self.root,HistName)
1828            refList = self.PatternTree.GetItemPyData(
1829                G2gd.GetPatternTreeItemId(self,Id,'Reflection Lists'))
1830            refList[generalData['Name']] = []
1831        self.EnableRefineCommand()
1832        return # success
1833       
1834    def OnPreferences(self,event):
1835        'Edit the GSAS-II configuration variables'
1836        dlg = G2G.SelectConfigSetting(self)
1837        dlg.ShowModal() == wx.ID_OK
1838        dlg.Destroy()
1839
1840    def _Add_ImportMenu_smallangle(self,parent):
1841        '''configure the Small Angle Data menus accord to the readers found in _init_Imports
1842        '''
1843        submenu = wx.Menu()
1844        item = parent.AppendMenu(wx.ID_ANY, 'Small Angle Data',
1845            submenu, help='Import small angle data')
1846        for reader in self.ImportSmallAngleReaderlist:
1847            item = submenu.Append(wx.ID_ANY,help=reader.longFormatName,
1848                kind=wx.ITEM_NORMAL,text='from '+reader.formatName+' file')
1849            self.ImportMenuId[item.GetId()] = reader
1850            self.Bind(wx.EVT_MENU, self.OnImportSmallAngle, id=item.GetId())
1851        # item = submenu.Append(wx.ID_ANY,
1852        #     help='Import small angle data, use file to try to determine format',
1853        #     kind=wx.ITEM_NORMAL,text='guess format from file')
1854        # self.Bind(wx.EVT_MENU, self.OnImportSmallAngle, id=item.GetId())
1855
1856    def OnImportSmallAngle(self,event):
1857        '''Called in response to an Import/Small Angle Data/... menu item
1858        to read a small angle diffraction data set.
1859        dict self.ImportMenuId is used to look up the specific
1860        reader item associated with the menu item, which will be
1861        None for the last menu item, which is the "guess" option
1862        where all appropriate formats will be tried.
1863
1864        '''
1865       
1866        def GetSASDIparm(reader):
1867            parm = reader.instdict
1868            Iparm = {'Type':[parm['type'],parm['type'],0],'Lam':[parm['wave'],
1869                parm['wave'],0],'Azimuth':[0.,0.,0]}           
1870            return Iparm,{}
1871           
1872        # get a list of existing histograms
1873        SASDlist = []
1874        if self.PatternTree.GetCount():
1875            item, cookie = self.PatternTree.GetFirstChild(self.root)
1876            while item:
1877                name = self.PatternTree.GetItemText(item)
1878                if name.startswith('SASD ') and name not in SASDlist:
1879                    SASDlist.append(name)
1880                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
1881        # look up which format was requested
1882        reqrdr = self.ImportMenuId.get(event.GetId()) 
1883        rdlist = self.OnImportGeneric(
1884            reqrdr,self.ImportSmallAngleReaderlist,'Small Angle Data',multiple=True)
1885        if len(rdlist) == 0: return
1886        self.CheckNotebook()
1887        newHistList = []
1888        self.EnablePlot = False
1889        for rd in rdlist:
1890            HistName = rd.idstring
1891            HistName = 'SASD '+HistName
1892            # make new histogram names unique
1893            HistName = G2obj.MakeUniqueLabel(HistName,SASDlist)
1894            print 'Read small angle data '+str(HistName)+ \
1895                ' from file '+str(self.lastimport)
1896            # data are read, now store them in the tree
1897            Id = self.PatternTree.AppendItem(parent=self.root,text=HistName)
1898            Iparm1,Iparm2 = GetSASDIparm(rd)
1899#            if 'T' in Iparm1['Type'][0]:
1900#                if not rd.clockWd and rd.GSAS:
1901#                    rd.powderdata[0] *= 100.        #put back the CW centideg correction
1902#                cw = np.diff(rd.powderdata[0])
1903#                rd.powderdata[0] = rd.powderdata[0][:-1]+cw/2.
1904#                rd.powderdata[1] = rd.powderdata[1][:-1]/cw
1905#                rd.powderdata[2] = rd.powderdata[2][:-1]*cw**2  #1/var=w at this point
1906#                if 'Itype' in Iparm2:
1907#                    Ibeg = np.searchsorted(rd.powderdata[0],Iparm2['Tminmax'][0])
1908#                    Ifin = np.searchsorted(rd.powderdata[0],Iparm2['Tminmax'][1])
1909#                    rd.powderdata[0] = rd.powderdata[0][Ibeg:Ifin]
1910#                    YI,WYI = G2pwd.calcIncident(Iparm2,rd.powderdata[0])
1911#                    rd.powderdata[1] = rd.powderdata[1][Ibeg:Ifin]/YI
1912#                    var = 1./rd.powderdata[2][Ibeg:Ifin]
1913#                    var += WYI*rd.powderdata[1]**2
1914#                    var /= YI**2
1915#                    rd.powderdata[2] = 1./var
1916#                rd.powderdata[3] = np.zeros_like(rd.powderdata[0])                                       
1917#                rd.powderdata[4] = np.zeros_like(rd.powderdata[0])                                       
1918#                rd.powderdata[5] = np.zeros_like(rd.powderdata[0])                                       
1919            Tmin = min(rd.smallangledata[0])
1920            Tmax = max(rd.smallangledata[0])
1921            valuesdict = {
1922                'wtFactor':1.0,
1923                'Dummy':False,
1924                'ranId':ran.randint(0,sys.maxint),
1925                }
1926            rd.Sample['ranId'] = valuesdict['ranId'] # this should be removed someday
1927            self.PatternTree.SetItemPyData(Id,[valuesdict,rd.smallangledata])
1928            self.PatternTree.SetItemPyData(
1929                self.PatternTree.AppendItem(Id,text='Comments'),
1930                rd.comments)
1931            self.PatternTree.SetItemPyData(
1932                self.PatternTree.AppendItem(Id,text='Limits'),
1933                [(Tmin,Tmax),[Tmin,Tmax]])
1934            self.PatternId = G2gd.GetPatternTreeItemId(self,Id,'Limits')
1935            self.PatternTree.SetItemPyData(
1936                self.PatternTree.AppendItem(Id,text='Instrument Parameters'),
1937                [Iparm1,Iparm2])
1938            self.PatternTree.SetItemPyData(
1939                self.PatternTree.AppendItem(Id,text='Substances'),G2pdG.SetDefaultSubstances())
1940            self.PatternTree.SetItemPyData(
1941                self.PatternTree.AppendItem(Id,text='Sample Parameters'),
1942                rd.Sample)
1943            self.PatternTree.SetItemPyData(
1944                self.PatternTree.AppendItem(Id,text='Models'),G2pdG.SetDefaultSASDModel())
1945            newHistList.append(HistName)
1946        else:
1947            self.EnablePlot = True
1948            self.PatternTree.Expand(Id)
1949            self.PatternTree.SelectItem(Id)
1950           
1951        if not newHistList: return # somehow, no new histograms
1952        return # success
1953
1954    def OnMacroRecordStatus(self,event,setvalue=None):
1955        '''Called when the record macro menu item is used which toggles the
1956        value. Alternately a value to be set can be provided. Note that this
1957        routine is made more complex because on the Mac there are lots of menu
1958        items (listed in self.MacroStatusList) and this loops over all of them.
1959        '''
1960        nextvalue = log.ShowLogStatus() != True
1961        if setvalue is not None:
1962            nextvalue = setvalue
1963        if nextvalue:
1964            log.LogOn()
1965            set2 = True
1966        else:
1967            log.LogOff()
1968            set2 = False
1969        for menuitem in self.MacroStatusList:
1970            menuitem.Check(set2)
1971
1972    def _init_Macro(self):
1973        '''Define the items in the macro menu.
1974        '''
1975        menu = self.MacroMenu
1976        item = menu.Append(
1977                help='Start or stop recording of menu actions, etc.', id=wx.ID_ANY,
1978                kind=wx.ITEM_CHECK,text='Record actions')
1979        self.MacroStatusList.append(item)
1980        item.Check(log.ShowLogStatus())
1981        self.Bind(wx.EVT_MENU, self.OnMacroRecordStatus, item)
1982
1983        # this may only be of value for development work
1984        item = menu.Append(
1985            help='Show logged commands', id=wx.ID_ANY,
1986            kind=wx.ITEM_NORMAL,text='Show log')
1987        def OnShowLog(event):
1988            print 70*'='
1989            print 'List of logged actions'
1990            for i,line in enumerate(log.G2logList):
1991                if line: print i,line
1992            print 70*'='
1993        self.Bind(wx.EVT_MENU, OnShowLog, item)
1994
1995        item = menu.Append(
1996            help='Clear logged commands', id=wx.ID_ANY,
1997            kind=wx.ITEM_NORMAL,text='Clear log')
1998        def OnClearLog(event): log.G2logList=[None]
1999        self.Bind(wx.EVT_MENU, OnClearLog, item)
2000       
2001        item = menu.Append(
2002            help='Save logged commands to file', id=wx.ID_ANY,
2003            kind=wx.ITEM_NORMAL,text='Save log')
2004        def OnSaveLog(event):
2005            import cPickle
2006            defnam = os.path.splitext(
2007                os.path.split(self.GSASprojectfile)[1]
2008                )[0]+'.gcmd'
2009            dlg = wx.FileDialog(self,
2010                'Choose an file to save past actions', '.', defnam, 
2011                'GSAS-II cmd output (*.gcmd)|*.gcmd',
2012                wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
2013            dlg.CenterOnParent()
2014            try:
2015                if dlg.ShowModal() == wx.ID_OK:
2016                    filename = dlg.GetPath()
2017                    # make sure extension is correct
2018                    filename = os.path.splitext(filename)[0]+'.gcmd'
2019                else:
2020                    filename = None
2021            finally:
2022                dlg.Destroy()
2023            if filename:
2024                fp = open(filename,'wb')
2025                fp.write(str(len(log.G2logList)-1)+'\n')
2026                for item in log.G2logList:
2027                    if item: cPickle.dump(item,fp)
2028                fp.close()
2029        self.Bind(wx.EVT_MENU, OnSaveLog, item)
2030
2031        item = menu.Append(
2032            help='Load logged commands from file', id=wx.ID_ANY,
2033            kind=wx.ITEM_NORMAL,text='Load log')
2034        def OnLoadLog(event):
2035            # this appends. Perhaps we should ask to clear?
2036            import cPickle
2037            defnam = os.path.splitext(
2038                os.path.split(self.GSASprojectfile)[1]
2039                )[0]+'.gcmd'
2040            dlg = wx.FileDialog(self,
2041                'Choose an file to read saved actions', '.', defnam, 
2042                'GSAS-II cmd output (*.gcmd)|*.gcmd',
2043                wx.OPEN)
2044            dlg.CenterOnParent()
2045            try:
2046                if dlg.ShowModal() == wx.ID_OK:
2047                    filename = dlg.GetPath()
2048                    # make sure extension is correct
2049                    filename = os.path.splitext(filename)[0]+'.gcmd'
2050                else:
2051                    filename = None
2052            finally:
2053                dlg.Destroy()
2054            if filename and os.path.exists(filename):
2055                fp = open(filename,'rb')
2056                lines = fp.readline()
2057                for i in range(int(lines)):
2058                    log.G2logList.append(cPickle.load(fp))
2059                fp.close()
2060        self.Bind(wx.EVT_MENU, OnLoadLog, item)
2061
2062        item = menu.Append(
2063            help='Replay saved commands', id=wx.ID_ANY,
2064            kind=wx.ITEM_NORMAL,text='Replay log')
2065        self.Bind(wx.EVT_MENU, log.ReplayLog, item)
2066
2067    def _init_Exports(self,menu):
2068        '''Find exporter routines and add them into menus
2069        '''
2070        # set up the top-level menus
2071        projectmenu = wx.Menu()
2072        item = menu.AppendMenu(
2073            wx.ID_ANY, 'Entire project as',
2074            projectmenu, help='Export entire project')
2075
2076        phasemenu = wx.Menu()
2077        item = menu.AppendMenu(
2078            wx.ID_ANY, 'Phase as',
2079            phasemenu, help='Export phase or sometimes phases')
2080
2081        powdermenu = wx.Menu()
2082        item = menu.AppendMenu(
2083            wx.ID_ANY, 'Powder data as',
2084            powdermenu, help='Export powder diffraction histogram(s)')
2085
2086        singlemenu = wx.Menu()
2087        item = menu.AppendMenu(
2088            wx.ID_ANY, 'Single crystal data as',
2089            singlemenu, help='Export single crystal histogram(s)')
2090
2091        imagemenu = wx.Menu()
2092        item = menu.AppendMenu(
2093            wx.ID_ANY, 'Image data as',
2094            imagemenu, help='Export powder image(s) data')
2095
2096        mapmenu = wx.Menu()
2097        item = menu.AppendMenu(
2098            wx.ID_ANY, 'Maps as',
2099            mapmenu, help='Export density map(s)')
2100
2101        # pdfmenu = wx.Menu()
2102        # item = menu.AppendMenu(
2103        #     wx.ID_ANY, 'PDFs as',
2104        #     pdfmenu, help='Export pair distribution function(s)')
2105
2106        # find all the exporter files
2107        if not self.exporterlist: # this only needs to be done once
2108            pathlist = sys.path
2109            filelist = []
2110            for path in pathlist:
2111                for filename in glob.iglob(os.path.join(path,"G2export*.py")):
2112                    filelist.append(filename)   
2113            filelist = sorted(list(set(filelist))) # remove duplicates
2114            # go through the routines and import them, saving objects that
2115            # have export routines (method Exporter)
2116            for filename in filelist:
2117                path,rootname = os.path.split(filename)
2118                pkg = os.path.splitext(rootname)[0]
2119#                try:
2120                fp = None
2121                fp, fppath,desc = imp.find_module(pkg,[path,])
2122                pkg = imp.load_module(pkg,fp,fppath,desc)
2123                for clss in inspect.getmembers(pkg): # find classes defined in package
2124                    if clss[0].startswith('_'): continue
2125                    if inspect.isclass(clss[1]):
2126                        # check if we have the required methods
2127                        for m in 'Exporter','loadParmDict':
2128                            if not hasattr(clss[1],m): break
2129                            if not callable(getattr(clss[1],m)): break
2130                        else:
2131                            exporter = clss[1](self) # create an export instance
2132                            self.exporterlist.append(exporter)
2133#                except AttributeError:
2134#                    print 'Import_'+errprefix+': Attribute Error'+str(filename)
2135#                    pass
2136#                except ImportError:
2137#                    print 'Import_'+errprefix+': Error importing file'+str(filename)
2138#                    pass
2139                if fp: fp.close()
2140        # Add submenu item(s) for each Exporter by its self-declared type (can be more than one)
2141        for obj in self.exporterlist:
2142            #print 'exporter',obj
2143            for typ in obj.exporttype:
2144                if typ == "project":
2145                    submenu = projectmenu
2146                elif typ == "phase":
2147                    submenu = phasemenu
2148                elif typ == "powder":
2149                    submenu = powdermenu
2150                elif typ == "single":
2151                    submenu = singlemenu
2152                elif typ == "image":
2153                    submenu = imagemenu
2154                elif typ == "map":
2155                    submenu = mapmenu
2156                # elif typ == "pdf":
2157                #     submenu = pdfmenu
2158                else:
2159                    print("Error, unknown type in "+str(obj))
2160                    break
2161                item = submenu.Append(
2162                    wx.ID_ANY,
2163                    help=obj.longFormatName,
2164                    kind=wx.ITEM_NORMAL,
2165                    text=obj.formatName)
2166                self.Bind(wx.EVT_MENU, obj.Exporter, id=item.GetId())
2167                self.ExportLookup[item.GetId()] = typ # lookup table for submenu item
2168        item = imagemenu.Append(wx.ID_ANY,
2169                        help='Export image controls and masks for multiple images',
2170                        kind=wx.ITEM_NORMAL,
2171                        text='Multiple image controls and masks')
2172        self.Bind(wx.EVT_MENU, self.OnSaveMultipleImg, id=item.GetId())
2173        #code to debug an Exporter. hard-code the routine below, to allow a reload before use
2174        # def DebugExport(event):
2175        #      print 'start reload'
2176        #      reload(G2IO)
2177        #      import G2export_pwdr as dev
2178        #      reload(dev)
2179        #      dev.ExportPowderFXYE(self).Exporter(event)
2180        # item = menu.Append(
2181        #     wx.ID_ANY,kind=wx.ITEM_NORMAL,
2182        #     help="debug exporter",text="test Export FXYE")
2183        # self.Bind(wx.EVT_MENU, DebugExport, id=item.GetId())
2184        # # #self.ExportLookup[item.GetId()] = 'image'
2185        # self.ExportLookup[item.GetId()] = 'powder'
2186
2187    def _Add_ExportMenuItems(self,parent):
2188        # item = parent.Append(
2189        #     help='Select PWDR item to enable',id=wx.ID_ANY,
2190        #     kind=wx.ITEM_NORMAL,
2191        #     text='Export Powder Patterns...')
2192        # self.ExportPattern.append(item)
2193        # item.Enable(False)
2194        # self.Bind(wx.EVT_MENU, self.OnExportPatterns, id=item.GetId())
2195
2196        item = parent.Append(
2197            help='',id=wx.ID_ANY,
2198            kind=wx.ITEM_NORMAL,
2199            text='Export All Peak Lists...')
2200        self.ExportPeakList.append(item)
2201        item.Enable(True)
2202        self.Bind(wx.EVT_MENU, self.OnExportPeakList, id=item.GetId())
2203
2204        item = parent.Append(
2205            help='',id=wx.ID_ANY,
2206            kind=wx.ITEM_NORMAL,
2207            text='Export HKLs...')
2208        self.ExportHKL.append(item)
2209        self.Bind(wx.EVT_MENU, self.OnExportHKL, id=item.GetId())
2210
2211        item = parent.Append(
2212            help='Select PDF item to enable',
2213            id=wx.ID_ANY,
2214            kind=wx.ITEM_NORMAL,
2215            text='Export PDF...')
2216        self.ExportPDF.append(item)
2217        item.Enable(False)
2218        self.Bind(wx.EVT_MENU, self.OnExportPDF, id=item.GetId())
2219
2220    def FillMainMenu(self,menubar,addhelp=True):
2221        '''Define contents of the main GSAS-II menu for the (main) data tree window.
2222        For the mac, this is also called for the data item windows as well so that
2223        the main menu items are data menu as well.
2224        '''
2225        File = wx.Menu(title='')
2226        menubar.Append(menu=File, title='&File')
2227        self._Add_FileMenuItems(File)
2228        Data = wx.Menu(title='')
2229        menubar.Append(menu=Data, title='Data')
2230        self._Add_DataMenuItems(Data)
2231        Calculate = wx.Menu(title='')       
2232        menubar.Append(menu=Calculate, title='&Calculate')
2233        self._Add_CalculateMenuItems(Calculate)
2234        Import = wx.Menu(title='')       
2235        menubar.Append(menu=Import, title='Import')
2236        self._Add_ImportMenu_Image(Import)
2237        self._Add_ImportMenu_Phase(Import)
2238        self._Add_ImportMenu_powder(Import)
2239        self._Add_ImportMenu_Sfact(Import)
2240        self._Add_ImportMenu_smallangle(Import)
2241
2242        #======================================================================
2243        # Code to help develop/debug an importer, much is hard-coded below
2244        # but module is reloaded before each use, allowing faster testing
2245        # def DebugImport(event):
2246        #     print 'start reload'
2247        #     import G2phase_ISO as dev
2248        #     reload(dev)
2249        #     rd = dev.ISODISTORTPhaseReader()
2250        #     self.ImportMenuId[event.GetId()] = rd
2251        #     self.OnImportPhase(event)
2252            # or ----------------------------------------------------------------------
2253            #self.OnImportGeneric(rd,[],'test of ISODISTORTPhaseReader')
2254            # special debug code
2255            # or ----------------------------------------------------------------------
2256            # filename = '/Users/toby/projects/branton/subgroup_cif.txt'
2257            # fp = open(filename,'Ur')
2258            # if not rd.ContentsValidator(fp):
2259            #     print 'not validated'
2260            #     # make a list of used phase ranId's
2261            # phaseRIdList = []
2262            # sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
2263            # if sub:
2264            #     item, cookie = self.PatternTree.GetFirstChild(sub)
2265            #     while item:
2266            #         phaseName = self.PatternTree.GetItemText(item)
2267            #         ranId = self.PatternTree.GetItemPyData(item).get('ranId')
2268            #         if ranId: phaseRIdList.append(ranId)
2269            #         item, cookie = self.PatternTree.GetNextChild(sub, cookie)
2270            # if rd.Reader(filename,fp,usedRanIdList=phaseRIdList):
2271            #     print 'read OK'
2272        # item = Import.Append(
2273        #     wx.ID_ANY,kind=wx.ITEM_NORMAL,
2274        #     help="debug importer",text="test importer")
2275        # self.Bind(wx.EVT_MENU, DebugImport, id=item.GetId())
2276        #======================================================================
2277        self.ExportMenu = wx.Menu(title='')
2278        menubar.Append(menu=self.ExportMenu, title='Export')
2279        self._init_Exports(self.ExportMenu)
2280        self._Add_ExportMenuItems(self.ExportMenu)
2281        if GSASIIpath.GetConfigValue('Enable_logging'):
2282            self.MacroMenu = wx.Menu(title='')
2283            menubar.Append(menu=self.MacroMenu, title='Macro')
2284            self._init_Macro()
2285        if addhelp:
2286            HelpMenu=G2G.MyHelp(self,includeTree=True,
2287                morehelpitems=[('&Tutorials','Tutorials'),])
2288            menubar.Append(menu=HelpMenu,title='&Help')
2289           
2290    def _init_ctrls(self, parent):
2291        wx.Frame.__init__(self, name='GSASII', parent=parent,
2292            size=wx.Size(400, 250),style=wx.DEFAULT_FRAME_STYLE, title='GSAS-II data tree')
2293        clientSize = wx.ClientDisplayRect()
2294        Size = self.GetSize()
2295        xPos = clientSize[2]-Size[0]
2296        self.SetPosition(wx.Point(xPos,clientSize[1]))
2297        self._init_Imports()
2298        #initialize Menu item objects (these contain lists of menu items that are enabled or disabled)
2299        self.MakePDF = []
2300        self.Refine = []
2301        self.SeqRefine = [] # pointer(s) to Sequential Refinement menu objects
2302        #self.ExportPattern = []
2303        self.ExportPeakList = []
2304        self.ExportHKL = []
2305        self.ExportPDF = []
2306        self.ExportPhase = []
2307        self.ExportCIF = []
2308        #
2309        self.GSASIIMenu = wx.MenuBar()
2310        # create a list of all dataframe menus (appended in PrefillDataMenu)
2311        self.dataMenuBars = [self.GSASIIMenu]
2312        self.MacroStatusList = []
2313        self.FillMainMenu(self.GSASIIMenu)
2314        self.SetMenuBar(self.GSASIIMenu)
2315        self.Bind(wx.EVT_SIZE, self.OnSize)
2316        self.Status = self.CreateStatusBar()
2317        self.mainPanel = wx.Panel(self,-1)
2318       
2319        wxID_PATTERNTREE = wx.NewId()
2320        #self.PatternTree = wx.TreeCtrl(id=wxID_PATTERNTREE, # replaced for logging
2321        self.PatternTree = G2G.G2TreeCtrl(id=wxID_PATTERNTREE,
2322            parent=self.mainPanel, pos=wx.Point(0, 0),style=wx.TR_DEFAULT_STYLE )
2323        self.PatternTree.Bind(wx.EVT_TREE_SEL_CHANGED,self.OnDataTreeSelChanged)
2324        self.PatternTree.Bind(wx.EVT_TREE_ITEM_RIGHT_CLICK,self.OnDataTreeSelChanged)
2325        self.PatternTree.Bind(wx.EVT_TREE_ITEM_COLLAPSED,
2326            self.OnPatternTreeItemCollapsed, id=wxID_PATTERNTREE)
2327        self.PatternTree.Bind(wx.EVT_TREE_ITEM_EXPANDED,
2328            self.OnPatternTreeItemExpanded, id=wxID_PATTERNTREE)
2329        self.PatternTree.Bind(wx.EVT_TREE_DELETE_ITEM,
2330            self.OnPatternTreeItemDelete, id=wxID_PATTERNTREE)
2331        self.PatternTree.Bind(wx.EVT_TREE_KEY_DOWN,
2332            self.OnPatternTreeKeyDown, id=wxID_PATTERNTREE)
2333        self.PatternTree.Bind(wx.EVT_TREE_BEGIN_RDRAG,
2334            self.OnPatternTreeBeginRDrag, id=wxID_PATTERNTREE)       
2335        self.PatternTree.Bind(wx.EVT_TREE_END_DRAG,
2336            self.OnPatternTreeEndDrag, id=wxID_PATTERNTREE)       
2337        #self.root = self.PatternTree.AddRoot('Loaded Data: ')
2338        self.root = self.PatternTree.root
2339        plotFrame = wx.Frame(None,-1,'GSASII Plots',size=wx.Size(700,600), \
2340            style=wx.DEFAULT_FRAME_STYLE ^ wx.CLOSE_BOX)
2341        self.G2plotNB = G2plt.G2PlotNoteBook(plotFrame,G2frame=self)
2342        plotFrame.Show()
2343       
2344        self.dataDisplay = None
2345       
2346    def __init__(self, parent):
2347        self.ExportLookup = {}
2348        self.exporterlist = []
2349        self._init_ctrls(parent)
2350        self.Image = wx.Image(
2351            os.path.join(GSASIIpath.path2GSAS2,'gsas2.ico'),
2352            wx.BITMAP_TYPE_ICO)
2353        if "wxMSW" in wx.PlatformInfo:
2354            img = self.Image.Scale(16, 16).ConvertToBitmap()
2355        elif "wxGTK" in wx.PlatformInfo:
2356            img = self.Image.Scale(22, 22).ConvertToBitmap()
2357        else:
2358            img = self.Image.ConvertToBitmap()
2359        self.SetIcon(wx.IconFromBitmap(img))
2360        self.Bind(wx.EVT_CLOSE, self.ExitMain)
2361        # various defaults
2362        self.oldFocus = None
2363        self.GSASprojectfile = ''
2364        self.undofile = ''
2365        self.TreeItemDelete = False
2366        self.plotStyle = {'qPlot':False,'dPlot':False,'sqrtPlot':False}
2367        self.Weight = False
2368        self.IfPlot = False
2369        self.DDShowAll = False
2370        self.atmSel = ''
2371        self.PatternId = 0
2372        self.PickId = 0
2373        self.PickIdText = None
2374        self.PeakTable = []
2375        self.LimitsTable = []
2376        self.ifX20 = True   #use M20 /= (1+X20) in powder indexing, etc.
2377        self.HKL = []
2378        self.Lines = []
2379        self.itemPicked = None
2380        self.dataFrame = None
2381        self.Interpolate = 'nearest'
2382        self.ContourColor = 'Paired'
2383        self.VcovColor = 'RdYlGn'
2384        self.RamaColor = 'Blues'
2385        self.Projection = 'equal area'
2386        self.logPlot = False
2387        self.plusPlot = True
2388        self.sqPlot = False
2389        self.ErrorBars = False
2390        self.Contour = False
2391        self.Legend = False
2392        self.SinglePlot = True
2393        self.selections= None
2394        self.PDFselections = None
2395        self.SubBack = False
2396        self.seqReverse = False
2397        self.seqLines = True #draw lines between points
2398        self.plotView = 0
2399        self.Image = 0
2400        self.oldImagefile = '' # the name of the last image file read
2401        self.oldImageTag = None # the name of the tag for multi-image files
2402        self.PauseIntegration = False
2403        self.ImageZ = []
2404        self.Integrate = 0
2405        self.imageDefault = {}
2406        self.IntgOutList = [] # list of integration tree item Ids created in G2IO.SaveIntegration
2407        self.AutointPWDRnames = [] # list of autoint created PWDR tree item names (to be deleted on a reset)
2408        self.autoIntFrame = None
2409        self.IntegratedList = [] # list of already integrated IMG tree items
2410        self.Sngl = False
2411        self.ifGetRing = False
2412        self.MaskKey = ''           #trigger for making image masks
2413        self.MskDelete = False      #trigger for mask delete
2414        self.StrainKey = ''         #ditto for new strain d-zeros
2415        self.EnablePlot = True
2416        self.hist = ''              # selected histogram in Phase/Data tab
2417        self.dirname = os.path.expanduser('~')       #start in the users home directory by default; may be meaningless
2418        self.TutorialImportDir = None  # location to read tutorial files, set when a tutorial is viewed
2419        self.LastImportDir = None # last-used directory where an import was done
2420        self.LastGPXdir = None    # directory where a GPX file was last read
2421        self.LastExportDir = None  # the last directory used for exports, if any.
2422        self.dataDisplayPhaseText = ''
2423        self.lastTreeSetting = []
2424       
2425        arg = sys.argv
2426        if len(arg) > 1 and arg[1]:
2427            self.GSASprojectfile = os.path.splitext(arg[1])[0]+'.gpx'
2428            self.dirname = os.path.dirname(arg[1])
2429            if self.dirname: os.chdir(self.dirname)
2430            try:
2431                self.StartProject()         #open the file if possible
2432                return
2433            except Exception:
2434                print 'Error opening or reading file',arg[1]
2435                import traceback
2436                print traceback.format_exc()
2437               
2438        if GSASIIpath.GetConfigValue('Starting_directory'):
2439            try:
2440                pth = GSASIIpath.GetConfigValue('Starting_directory')
2441                pth = os.path.expanduser(pth) 
2442                os.chdir(pth)
2443                self.LastGPXdir = pth
2444            except:
2445                print('Ignoring Config Starting_directory value: '+
2446                      GSASIIpath.GetConfigValue('Starting_directory'))
2447
2448    def GetTreeItemsList(self,item):
2449        return self.PatternTree._getTreeItemsList(item)
2450
2451    def OnSize(self,event):
2452        'Called to make PatternTree fill mainPanel'
2453        w,h = self.GetClientSizeTuple()
2454        self.mainPanel.SetSize(wx.Size(w,h))
2455        self.PatternTree.SetSize(wx.Size(w,h))
2456                       
2457    def OnDataTreeSelChanged(self, event):
2458        '''Called when a data tree item is selected'''
2459        if self.TreeItemDelete:
2460            self.TreeItemDelete = False
2461        else:
2462            pltNum = self.G2plotNB.nb.GetSelection()
2463            if pltNum >= 0:                         #to avoid the startup with no plot!
2464                self.G2plotNB.nb.GetPage(pltNum)
2465            item = event.GetItem()
2466            G2gd.SelectDataTreeItem(self,item)
2467            if self.oldFocus:
2468                self.oldFocus.SetFocus()
2469       
2470    def OnPatternTreeItemCollapsed(self, event):
2471        'Called when a tree item is collapsed - all children will be collapsed'
2472        self.PatternTree.CollapseAllChildren(event.GetItem())
2473
2474    def OnPatternTreeItemExpanded(self, event):
2475        'Called when a tree item is expanded'
2476        self.OnDataTreeSelChanged(event)
2477        event.Skip()
2478       
2479    def OnPatternTreeItemDelete(self, event):
2480        'Called when a tree item is deleted -- not sure what this does'
2481        self.TreeItemDelete = True
2482
2483    def OnPatternTreeItemActivated(self, event):
2484        'Called when a tree item is activated'
2485        event.Skip()
2486       
2487    def OnPatternTreeBeginRDrag(self,event):
2488        event.Allow()
2489        self.BeginDragId = event.GetItem()
2490        self.ParentId = self.PatternTree.GetItemParent(self.BeginDragId)
2491        DragText = self.PatternTree.GetItemText(self.BeginDragId)
2492        self.DragData = [[DragText,self.PatternTree.GetItemPyData(self.BeginDragId)],]
2493        item, cookie = self.PatternTree.GetFirstChild(self.BeginDragId)
2494        while item:     #G2 data tree has no sub children under a child of a tree item
2495            name = self.PatternTree.GetItemText(item)
2496            self.DragData.append([name,self.PatternTree.GetItemPyData(item)])
2497            item, cookie = self.PatternTree.GetNextChild(self.BeginDragId, cookie)                           
2498       
2499    def OnPatternTreeEndDrag(self,event):
2500        event.Allow()
2501        self.EndDragId = event.GetItem()
2502        try:
2503            NewParent = self.PatternTree.GetItemParent(self.EndDragId)
2504        except:
2505            self.EndDragId = self.PatternTree.GetLastChild(self.root)
2506            NewParent = self.root
2507        if self.ParentId != NewParent:
2508            self.ErrorDialog('Drag not allowed','Wrong parent for item dragged')
2509        else:
2510            Name,Item = self.DragData[0]
2511            NewId = self.PatternTree.InsertItem(self.ParentId,self.EndDragId,Name,data=None)
2512            self.PatternTree.SetItemPyData(NewId,Item)
2513            for name,item in self.DragData[1:]:     #loop over children
2514                Id = self.PatternTree.AppendItem(parent=NewId,text=name)
2515                self.PatternTree.SetItemPyData(Id,item)
2516            self.PatternTree.Delete(self.BeginDragId)
2517            G2gd.SelectDataTreeItem(self,NewId)
2518       
2519    def OnPatternTreeKeyDown(self,event): #doesn't exactly work right with Shift key down
2520        'Allows stepping through the tree with the up/down arrow keys'
2521        self.oldFocus = wx.Window.FindFocus()
2522        keyevt = event.GetKeyEvent()
2523        key = event.GetKeyCode()
2524        item = self.PatternTree.GetSelection()
2525        if type(item) is int: return # is this the toplevel in tree?
2526        name = self.PatternTree.GetItemText(item)
2527        parent = self.PatternTree.GetItemParent(item)
2528        if key == wx.WXK_UP:
2529            if keyevt.GetModifiers() == wx.MOD_SHIFT and parent != self.root:
2530                if type(parent) is int: return # is this the toplevel in tree?
2531                prev = self.PatternTree.GetPrevSibling(parent)
2532                NewId = G2gd.GetPatternTreeItemId(self,prev,name)
2533                if NewId:
2534                    self.PatternTree.Collapse(parent)
2535                    self.PatternTree.Expand(prev)
2536                    self.oldFocus = wx.Window.FindFocus()
2537                    wx.CallAfter(self.PatternTree.SelectItem,NewId)
2538                else:
2539                    wx.CallAfter(self.PatternTree.SelectItem,item)
2540            else:   
2541                self.PatternTree.GetPrevSibling(item)
2542                self.PatternTree.SelectItem(item)
2543        elif key == wx.WXK_DOWN:
2544            if keyevt.GetModifiers() == wx.MOD_SHIFT and parent != self.root:
2545                next = self.PatternTree.GetNextSibling(parent)
2546                NewId = G2gd.GetPatternTreeItemId(self,next,name)
2547                if NewId:
2548                    self.PatternTree.Collapse(parent)
2549                    self.PatternTree.Expand(next)
2550                    self.oldFocus = wx.Window.FindFocus()
2551                    wx.CallAfter(self.PatternTree.SelectItem,NewId)
2552                else:
2553                    wx.CallAfter(self.PatternTree.SelectItem,item)
2554            else:   
2555                self.PatternTree.GetNextSibling(item)
2556                self.PatternTree.SelectItem(item)
2557               
2558    def OnReadPowderPeaks(self,event):
2559        'Bound to menu Data/Read Powder Peaks'
2560        self.CheckNotebook()
2561        pth = G2G.GetImportPath(self)
2562        if not pth: pth = '.'
2563        dlg = wx.FileDialog(self, 'Choose file with peak list', pth, '', 
2564            'peak files (*.txt)|*.txt|All files (*.*)|*.*',wx.OPEN)
2565        try:
2566            if dlg.ShowModal() == wx.ID_OK:
2567                self.HKL = []
2568                self.powderfile = dlg.GetPath()
2569                comments,peaks,limits,wave = G2IO.GetPowderPeaks(self.powderfile)
2570                Id = self.PatternTree.AppendItem(parent=self.root,text='PKS '+os.path.basename(self.powderfile))
2571                data = ['PKS',wave,0.0]
2572                names = ['Type','Lam','Zero'] 
2573                codes = [0,0,0]
2574                inst = [G2IO.makeInstDict(names,data,codes),{}]
2575                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Instrument Parameters'),inst)
2576                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),comments)
2577                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Limits'),[tuple(limits),limits])
2578                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Index Peak List'),[peaks,[]])
2579                self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Unit Cells List'),[])             
2580                self.PatternTree.Expand(Id)
2581                self.PatternTree.SelectItem(Id)
2582                os.chdir(dlg.GetDirectory())           # to get Mac/Linux to change directory!
2583        finally:
2584            dlg.Destroy()
2585                       
2586    def OnImageRead(self,event):
2587        '''Called to read in an image in any known format. *** Depreciated. ***
2588        '''
2589        G2G.G2MessageBox(self,'Please use the Import/Image/... menu item rather than this','depreciating menu item')
2590
2591    def CheckNotebook(self):
2592        '''Make sure the data tree has the minimally expected controls.
2593        '''
2594        if not G2gd.GetPatternTreeItemId(self,self.root,'Notebook'):
2595            sub = self.PatternTree.AppendItem(parent=self.root,text='Notebook')
2596            self.PatternTree.SetItemPyData(sub,[''])
2597        if not G2gd.GetPatternTreeItemId(self,self.root,'Controls'):
2598            sub = self.PatternTree.AppendItem(parent=self.root,text='Controls')
2599            self.PatternTree.SetItemPyData(sub,copy.copy(G2obj.DefaultControls))
2600        if not G2gd.GetPatternTreeItemId(self,self.root,'Covariance'):
2601            sub = self.PatternTree.AppendItem(parent=self.root,text='Covariance')
2602            self.PatternTree.SetItemPyData(sub,{})
2603        if not G2gd.GetPatternTreeItemId(self,self.root,'Constraints'):
2604            sub = self.PatternTree.AppendItem(parent=self.root,text='Constraints')
2605            self.PatternTree.SetItemPyData(sub,{'Hist':[],'HAP':[],'Phase':[]})
2606        if not G2gd.GetPatternTreeItemId(self,self.root,'Restraints'):
2607            sub = self.PatternTree.AppendItem(parent=self.root,text='Restraints')
2608            self.PatternTree.SetItemPyData(sub,{})
2609        if not G2gd.GetPatternTreeItemId(self,self.root,'Rigid bodies'):
2610            sub = self.PatternTree.AppendItem(parent=self.root,text='Rigid bodies')
2611            self.PatternTree.SetItemPyData(sub,{'Vector':{'AtInfo':{}},
2612                'Residue':{'AtInfo':{}},'RBIds':{'Vector':[],'Residue':[]}})
2613               
2614    class CopyDialog(wx.Dialog):
2615        '''Creates a dialog for copying control settings between
2616        data tree items'''
2617        def __init__(self,parent,title,text,data):
2618            wx.Dialog.__init__(self,parent,-1,title, 
2619                pos=wx.DefaultPosition,style=wx.DEFAULT_DIALOG_STYLE)
2620            self.data = data
2621            panel = wx.Panel(self)
2622            mainSizer = wx.BoxSizer(wx.VERTICAL)
2623            topLabl = wx.StaticText(panel,-1,text)
2624            mainSizer.Add((10,10),1)
2625            mainSizer.Add(topLabl,0,wx.ALIGN_CENTER_VERTICAL|wx.LEFT,10)
2626            mainSizer.Add((10,10),1)
2627            ncols = len(data)/40+1
2628            dataGridSizer = wx.FlexGridSizer(cols=ncols,hgap=2,vgap=2)
2629            for id,item in enumerate(self.data):
2630                ckbox = wx.CheckBox(panel,id,item[1])
2631                ckbox.Bind(wx.EVT_CHECKBOX,self.OnCopyChange)                   
2632                dataGridSizer.Add(ckbox,0,wx.LEFT,10)
2633            mainSizer.Add(dataGridSizer,0,wx.EXPAND)
2634            OkBtn = wx.Button(panel,-1,"Ok")
2635            OkBtn.Bind(wx.EVT_BUTTON, self.OnOk)
2636            cancelBtn = wx.Button(panel,-1,"Cancel")
2637            cancelBtn.Bind(wx.EVT_BUTTON, self.OnCancel)
2638            btnSizer = wx.BoxSizer(wx.HORIZONTAL)
2639            btnSizer.Add((20,20),1)
2640            btnSizer.Add(OkBtn)
2641            btnSizer.Add((20,20),1)
2642            btnSizer.Add(cancelBtn)
2643            btnSizer.Add((20,20),1)
2644           
2645            mainSizer.Add(btnSizer,0,wx.EXPAND|wx.BOTTOM|wx.TOP, 10)
2646            panel.SetSizer(mainSizer)
2647            panel.Fit()
2648            self.Fit()
2649       
2650        def OnCopyChange(self,event):
2651            id = event.GetId()
2652            self.data[id][0] = self.FindWindowById(id).GetValue()       
2653           
2654        def OnOk(self,event):
2655            parent = self.GetParent()
2656            parent.Raise()
2657            self.EndModal(wx.ID_OK)             
2658           
2659        def OnCancel(self,event):
2660            parent = self.GetParent()
2661            parent.Raise()
2662            self.EndModal(wx.ID_CANCEL)             
2663           
2664        def GetData(self):
2665            return self.data
2666       
2667    class SumDialog(wx.Dialog):
2668        '''Allows user to supply scale factor(s) when summing data -
2669        TODO: CAN WE PREVIEW RESULT HERE?'''
2670        def __init__(self,parent,title,text,dataType,data,dataList):
2671            wx.Dialog.__init__(self,parent,-1,title,size=(400,250),
2672                pos=wx.DefaultPosition,style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER)
2673            self.plotFrame = wx.Frame(self,-1,'Sum Plots',size=wx.Size(700,600), \
2674                style=wx.DEFAULT_FRAME_STYLE ^ wx.CLOSE_BOX)
2675            self.G2plotNB = G2plt.G2PlotNoteBook(self.plotFrame,G2frame=self)
2676            self.data = data
2677            self.dataList = dataList
2678            self.dataType = dataType
2679            size = (400,250)
2680            panel = wxscroll.ScrolledPanel(self, wx.ID_ANY,size=size,
2681                style = wx.TAB_TRAVERSAL|wx.SUNKEN_BORDER)
2682            mainSizer = wx.BoxSizer(wx.VERTICAL)
2683            topLabl = wx.StaticText(panel,-1,text)
2684            mainSizer.Add((10,10),1)
2685            mainSizer.Add(topLabl,0,wx.ALIGN_CENTER_VERTICAL|wx.LEFT,10)
2686            mainSizer.Add((10,10),1)
2687            self.dataGridSizer = wx.FlexGridSizer(cols=2,hgap=2,vgap=2)
2688            for id,item in enumerate(self.data[:-1]):
2689                name = wx.TextCtrl(panel,-1,item[1],size=wx.Size(300,20))
2690                name.SetEditable(False)
2691#        azmthOff = G2G.ValidatedTxtCtrl(G2frame.dataDisplay,data,'azmthOff',nDig=(10,2),typeHint=float,OnLeave=OnAzmthOff)
2692                scale = wx.TextCtrl(panel,id,'%.3f'%(item[0]),style=wx.TE_PROCESS_ENTER)
2693                scale.Bind(wx.EVT_TEXT_ENTER,self.OnScaleChange)
2694                scale.Bind(wx.EVT_KILL_FOCUS,self.OnScaleChange)
2695                self.dataGridSizer.Add(scale,0,wx.LEFT,10)
2696                self.dataGridSizer.Add(name,0,wx.RIGHT,10)
2697            if self.dataType:
2698                self.dataGridSizer.Add(wx.StaticText(panel,-1,'Sum result name: '+self.dataType),0, \
2699                    wx.LEFT|wx.TOP|wx.ALIGN_CENTER_VERTICAL,10)
2700                self.name = wx.TextCtrl(panel,-1,self.data[-1],size=wx.Size(300,20),style=wx.TE_PROCESS_ENTER)
2701                self.name.Bind(wx.EVT_TEXT_ENTER,self.OnNameChange)
2702                self.name.Bind(wx.EVT_KILL_FOCUS,self.OnNameChange)
2703                self.dataGridSizer.Add(self.name,0,wx.RIGHT|wx.TOP,10)
2704                self.dataGridSizer.Add(wx.StaticText(panel,label='All scales value: '),0,  \
2705                    wx.LEFT|wx.TOP|wx.ALIGN_CENTER_VERTICAL,10)
2706#        azmthOff = G2G.ValidatedTxtCtrl(G2frame.dataDisplay,data,'azmthOff',nDig=(10,2),typeHint=float,OnLeave=OnAzmthOff)
2707                allScale = wx.TextCtrl(panel,value='',style=wx.TE_PROCESS_ENTER)
2708                allScale.Bind(wx.EVT_TEXT_ENTER,self.OnAllScale)
2709                allScale.Bind(wx.EVT_KILL_FOCUS,self.OnAllScale)
2710                self.dataGridSizer.Add(allScale,0,WACV)
2711            mainSizer.Add(self.dataGridSizer,0,wx.EXPAND)
2712            TestBtn = wx.Button(panel,-1,"Test")
2713            TestBtn.Bind(wx.EVT_BUTTON, self.OnTest)
2714            OkBtn = wx.Button(panel,-1,"Ok")
2715            OkBtn.Bind(wx.EVT_BUTTON, self.OnOk)
2716            cancelBtn = wx.Button(panel,-1,"Cancel")
2717            cancelBtn.Bind(wx.EVT_BUTTON, self.OnCancel)
2718            btnSizer = wx.FlexGridSizer(0,3,10,20)
2719            if self.dataType =='PWDR':  btnSizer.Add(TestBtn)
2720            btnSizer.Add(OkBtn)
2721            btnSizer.Add(cancelBtn)
2722           
2723            panel.SetSizer(mainSizer)
2724            panel.SetAutoLayout(1)
2725            panel.SetupScrolling()
2726            mainSizer.Add((10,10),1)
2727            mainSizer.Add(btnSizer,0,wx.CENTER)
2728            panel.SetSizer(mainSizer)
2729            panel.Fit()
2730            self.Fit()
2731
2732        def OnScaleChange(self,event):
2733            event.Skip()
2734            id = event.GetId()
2735            value = self.FindWindowById(id).GetValue()
2736            try:
2737                self.data[id][0] = float(value)
2738                self.FindWindowById(id).SetValue('%.3f'%(self.data[id][0]))
2739            except ValueError:
2740                if value and '-' not in value[0]:
2741                    print 'bad input - numbers only'
2742                    self.FindWindowById(id).SetValue('0.000')
2743                   
2744        def OnAllScale(self,event):
2745            event.Skip()
2746            id = event.GetId()
2747            try:
2748                scale = float(self.FindWindowById(id).GetValue())
2749                self.FindWindowById(id).SetValue('%.3f'%(scale))
2750                entries = self.dataGridSizer.GetChildren()
2751                for i,item in enumerate(self.data[:-1]):
2752                    item[0] = scale
2753                    entries[2*i].GetWindow().SetValue('%.3f'%(scale))
2754                 
2755            except ValueError:
2756                print 'bad input - numbers only'
2757                self.FindWindowById(id).SetValue('')
2758                   
2759           
2760        def OnNameChange(self,event):
2761            event.Skip()
2762            self.data[-1] = self.name.GetValue()
2763           
2764        def OnTest(self,event):
2765            lenX = 0
2766            Xminmax = [0,0]
2767            XY = []
2768            Xsum = []
2769            Ysum = []
2770            Vsum = []
2771            result = self.data
2772            for i,item in enumerate(result[:-1]):
2773                scale,name = item
2774                data = self.dataList[i]
2775                if scale:
2776                    x,y,w,yc,yb,yd = data   #numpy arrays!
2777                    XY.append([x,scale*y])
2778                    v = 1./w
2779                    if lenX:
2780                        if lenX != len(x):
2781                            self.ErrorDialog('Data length error','Data to be summed must have same number of points'+ \
2782                                '\nExpected:'+str(lenX)+ \
2783                                '\nFound:   '+str(len(x))+'\nfor '+name)
2784                            self.OnCancel(event)
2785                    else:
2786                        lenX = len(x)
2787                    if Xminmax[1]:
2788                        if Xminmax != [x[0],x[-1]]:
2789                            self.ErrorDialog('Data range error','Data to be summed must span same range'+ \
2790                                '\nExpected:'+str(Xminmax[0])+' '+str(Xminmax[1])+ \
2791                                '\nFound:   '+str(x[0])+' '+str(x[-1])+'\nfor '+name)
2792                            self.OnCancel(event)
2793                        else:
2794                            for j,yi in enumerate(y):
2795                                 Ysum[j] += scale*yi
2796                                 Vsum[j] += abs(scale)*v[j]
2797                    else:
2798                        Xminmax = [x[0],x[-1]]
2799                        Xsum = x
2800                        Ysum = scale*y
2801                        Vsum = abs(scale*v)
2802            Wsum = 1./np.array(Vsum)
2803            YCsum = np.zeros(lenX)
2804            YBsum = np.zeros(lenX)
2805            YDsum = np.zeros(lenX)
2806            XY.append([Xsum,Ysum])
2807            self.result = [Xsum,Ysum,Wsum,YCsum,YBsum,YDsum]
2808            # N.B. PlotXY expects the first arg to point to G2frame. In this case, we
2809            # create a duplicate (temporary) Plot notebook window that is a child of the
2810            # modal SumDialog dialog (self). This nicely gets deleted when the dialog is destroyed,
2811            # but the plot window is not fully functional, at least on the Mac.
2812            G2plt.PlotXY(self,XY,lines=True,Title='Sum:'+self.data[-1],labelY='Intensity',)
2813            self.plotFrame.Show()
2814                       
2815        def OnOk(self,event):
2816            if self.dataType == 'PWDR': self.OnTest(event)
2817            parent = self.GetParent()
2818            parent.Raise()
2819            self.EndModal(wx.ID_OK)             
2820           
2821        def OnCancel(self,event):
2822            parent = self.GetParent()
2823            parent.Raise()
2824            self.EndModal(wx.ID_CANCEL)             
2825           
2826        def GetData(self):
2827            if self.dataType == 'PWDR':
2828                return self.data,self.result
2829            else:
2830                return self.data
2831                       
2832    def OnPwdrSum(self,event):
2833        'Sum together powder data(?)'
2834        TextList = []
2835        DataList = []
2836        Names = []
2837        Inst = None
2838        Comments = ['Sum equals: \n']
2839        if self.PatternTree.GetCount():
2840            item, cookie = self.PatternTree.GetFirstChild(self.root)
2841            while item:
2842                name = self.PatternTree.GetItemText(item)
2843                Names.append(name)
2844                if 'PWDR' in name:
2845                    TextList.append([0.0,name])
2846                    DataList.append(self.PatternTree.GetItemPyData(item)[1])    # (x,y,w,yc,yb,yd)
2847                    if not Inst:
2848                        Inst = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,item, 'Instrument Parameters'))
2849                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
2850            if len(TextList) < 2:
2851                self.ErrorDialog('Not enough data to sum','There must be more than one "PWDR" pattern')
2852                return
2853            TextList.append('default_sum_name')               
2854            dlg = self.SumDialog(self,'Sum data','Enter scale for each pattern in summation','PWDR',TextList,DataList)
2855            try:
2856                if dlg.ShowModal() == wx.ID_OK:
2857                    result,sumData = dlg.GetData()
2858                    Xsum,Ysum,Wsum,YCsum,YBsum,YDsum = sumData
2859                    Xminmax = [Xsum[0],Xsum[-1]]
2860                    outname = 'PWDR '+result[-1]
2861                    Id = 0
2862                    if outname in Names:
2863                        dlg2 = wx.MessageDialog(self,'Overwrite data?','Duplicate data name',wx.OK|wx.CANCEL)
2864                        try:
2865                            if dlg2.ShowModal() == wx.ID_OK:
2866                                Id = G2gd.GetPatternTreeItemId(self,self.root,name)
2867                                self.PatternTree.Delete(Id)
2868                        finally:
2869                            dlg2.Destroy()
2870                    Id = self.PatternTree.AppendItem(parent=self.root,text=outname)
2871                    if Id:
2872                        Sample = G2pdG.SetDefaultSample()
2873                        Ymax = np.max(Ysum)
2874                        valuesdict = {
2875                            'wtFactor':1.0,
2876                            'Dummy':False,
2877                            'ranId':ran.randint(0,sys.maxint),
2878                            'Offset':[0.0,0.0],'delOffset':0.02*Ymax,'refOffset':-.1*Ymax,'refDelt':0.1*Ymax,
2879                            'qPlot':False,'dPlot':False,'sqrtPlot':False
2880                            }
2881                        self.PatternTree.SetItemPyData(Id,[valuesdict,[np.array(Xsum),np.array(Ysum),np.array(Wsum),
2882                            np.array(YCsum),np.array(YBsum),np.array(YDsum)]])
2883                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),Comments)                   
2884                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Limits'),[tuple(Xminmax),Xminmax])
2885                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Background'),[['chebyschev',True,3,1.0,0.0,0.0],
2886                            {'nDebye':0,'debyeTerms':[],'nPeaks':0,'peaksList':[]}])
2887                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Instrument Parameters'),Inst)
2888                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Sample Parameters'),Sample)
2889                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Peak List'),{'peaks':[],'sigDict':{}})
2890                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Index Peak List'),[[],[]])
2891                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Unit Cells List'),[])             
2892                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Reflection Lists'),{})             
2893                        self.PatternTree.SelectItem(Id)
2894                        self.PatternTree.Expand(Id)
2895            finally:
2896                dlg.Destroy()
2897
2898    def OnImageSum(self,event):
2899        'Sum together image data'
2900        TextList = []
2901        DataList = []
2902        Names = []
2903        Comments = ['Sum equals: \n']
2904        if self.PatternTree.GetCount():
2905            item, cookie = self.PatternTree.GetFirstChild(self.root)
2906            while item:
2907                name = self.PatternTree.GetItemText(item)
2908                Names.append(name)
2909                if 'IMG' in name:
2910                    TextList.append([0.0,name])
2911                    DataList.append(self.PatternTree.GetImageLoc(item))        #Size,Image,Tag
2912                    Data = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,item,'Image Controls'))
2913                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
2914            if len(TextList) < 2:
2915                self.ErrorDialog('Not enough data to sum','There must be more than one "IMG" pattern')
2916                return
2917            TextList.append('default_sum_name')               
2918            dlg = self.SumDialog(self,'Sum data','Enter scale for each image in summation','IMG',TextList,DataList)
2919            try:
2920                if dlg.ShowModal() == wx.ID_OK:
2921                    imSize = 0
2922                    result = dlg.GetData()
2923                    First = True
2924                    Found = False
2925                    for i,item in enumerate(result[:-1]):
2926                        scale,name = item
2927                        if scale:
2928                            Found = True                               
2929                            Comments.append("%10.3f %s" % (scale,' * '+name))
2930                            Npix,imagefile,imagetag = DataList[i]
2931                            imagefile = G2IO.CheckImageFile(self,imagefile)
2932                            image = G2IO.GetImageData(self,imagefile,imageOnly=True,ImageTag=imagetag)
2933                            if First:
2934                                newImage = np.zeros_like(image)
2935                                First = False
2936                            if imSize:
2937                                if imSize != Npix:
2938                                    self.ErrorDialog('Image size error','Images to be summed must be same size'+ \
2939                                        '\nExpected:'+str(imSize)+ \
2940                                        '\nFound:   '+str(Npix)+'\nfor '+name)
2941                                    return
2942                                newImage = newImage+scale*image
2943                            else:
2944                                imSize = Npix
2945                                newImage = newImage+scale*image
2946                            del(image)
2947                    if not Found:
2948                        self.ErrorDialog('Image sum error','No nonzero image multipliers found')
2949                        return
2950                       
2951                       
2952                    newImage = np.array(newImage,dtype=np.int32)                       
2953                    outname = 'IMG '+result[-1]
2954                    Id = 0
2955                    if outname in Names:
2956                        dlg2 = wx.MessageDialog(self,'Overwrite data?','Duplicate data name',wx.OK|wx.CANCEL)
2957                        try:
2958                            if dlg2.ShowModal() == wx.ID_OK:
2959                                Id = G2gd.GetPatternTreeItemId(self,self.root,name)
2960                        finally:
2961                            dlg2.Destroy()
2962                    else:
2963                        Id = self.PatternTree.AppendItem(parent=self.root,text=outname)
2964                    if Id:
2965                        pth = G2G.GetExportPath(self)
2966                        dlg = wx.FileDialog(self, 'Choose sum image filename', pth,outname.split('IMG ')[1], 
2967                            'G2img files (*.G2img)|*.G2img', 
2968                            wx.SAVE|wx.FD_OVERWRITE_PROMPT)
2969                        if dlg.ShowModal() == wx.ID_OK:
2970                            newimagefile = dlg.GetPath()
2971                            newimagefile = G2IO.FileDlgFixExt(dlg,newimagefile)
2972                            G2IO.PutG2Image(newimagefile,Comments,Data,Npix,newImage)
2973                            Imax = np.amax(newImage)
2974                            Imin = np.amin(newImage)
2975                            newImage = []
2976                            self.PatternTree.SetItemPyData(Id,[imSize,newimagefile])
2977                            self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Comments'),Comments)
2978                        del(newImage)
2979                        if self.imageDefault:
2980                            Data = copy.copy(self.imageDefault)
2981                        Data['formatName'] = 'GSAS-II image'
2982                        Data['showLines'] = True
2983                        Data['ring'] = []
2984                        Data['rings'] = []
2985                        Data['cutoff'] = 10
2986                        Data['pixLimit'] = 20
2987                        Data['ellipses'] = []
2988                        Data['calibrant'] = ''
2989                        Data['range'] = [(Imin,Imax),[Imin,Imax]]
2990                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Image Controls'),Data)                                           
2991                        Masks = {'Points':[],'Rings':[],'Arcs':[],'Polygons':[],'Frames':[],'Thresholds':[(Imin,Imax),[Imin,Imax]]}
2992                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Masks'),Masks)
2993                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='Stress/Strain'),
2994                            {'Type':'True','d-zero':[],'Sample phi':0.0,'Sample z':0.0,'Sample load':0.0})
2995                        self.PatternTree.SelectItem(Id)
2996                        self.PatternTree.Expand(Id)
2997                        self.PickId = G2gd.GetPatternTreeItemId(self,self.root,outname)
2998                        self.Image = self.PickId
2999            finally:
3000                dlg.Destroy()
3001                     
3002    def OnAddPhase(self,event):
3003        'Add a new, empty phase to the tree. Called by Data/Add Phase menu'
3004        if not G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
3005            sub = self.PatternTree.AppendItem(parent=self.root,text='Phases')
3006        else:
3007            sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
3008        PhaseName = ''
3009        dlg = wx.TextEntryDialog(None,'Enter a name for this phase','Phase Name Entry','New phase',
3010            style=wx.OK)
3011        if dlg.ShowModal() == wx.ID_OK:
3012            PhaseName = dlg.GetValue()
3013        dlg.Destroy()
3014        sub = self.PatternTree.AppendItem(parent=sub,text=PhaseName)
3015        E,SGData = G2spc.SpcGroup('P 1')
3016        self.PatternTree.SetItemPyData(sub,G2IO.SetNewPhase(Name=PhaseName,SGData=SGData))
3017        G2gd.SelectDataTreeItem(self,sub) #bring up new phase General tab
3018       
3019    def OnDeletePhase(self,event):
3020        'Delete a phase from the tree. Called by Data/Delete Phase menu'
3021        #Hmm, also need to delete this phase from Reflection Lists for each PWDR histogram
3022        if self.dataFrame:
3023            self.dataFrame.Clear() 
3024        TextList = []
3025        DelList = []
3026        DelItemList = []
3027        if G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
3028            sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
3029        else:
3030            return
3031        if sub:
3032            item, cookie = self.PatternTree.GetFirstChild(sub)
3033            while item:
3034                TextList.append(self.PatternTree.GetItemText(item))
3035                item, cookie = self.PatternTree.GetNextChild(sub, cookie)               
3036            dlg = wx.MultiChoiceDialog(self, 'Which phase to delete?', 'Delete phase', TextList, wx.CHOICEDLG_STYLE)
3037            try:
3038                if dlg.ShowModal() == wx.ID_OK:
3039                    result = dlg.GetSelections()
3040                    for i in result: DelList.append([i,TextList[i]])
3041                    item, cookie = self.PatternTree.GetFirstChild(sub)
3042                    i = 0
3043                    while item:
3044                        if [i,self.PatternTree.GetItemText(item)] in DelList: DelItemList.append(item)
3045                        item, cookie = self.PatternTree.GetNextChild(sub, cookie)
3046                        i += 1
3047                    for item in DelItemList:
3048                        name = self.PatternTree.GetItemText(item)
3049                        self.PatternTree.Delete(item)
3050                        self.G2plotNB.Delete(name)
3051                    item, cookie = self.PatternTree.GetFirstChild(self.root)
3052                    while item:
3053                        name = self.PatternTree.GetItemText(item)
3054                        if 'PWDR' in name:
3055                            Id = G2gd.GetPatternTreeItemId(self,item, 'Reflection Lists')
3056                            refList = self.PatternTree.GetItemPyData(Id)
3057                            if len(refList):
3058                                for i,item in DelList:
3059                                    if item in refList:
3060                                        del(refList[item])
3061#                            self.PatternTree.SetItemPyData(Id,refList)
3062                        elif 'HKLF' in name:
3063                            data = self.PatternTree.GetItemPyData(item)
3064                            data[0] = {}
3065#                            self.PatternTree.SetItemPyData(item,data)
3066                           
3067                        item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3068            finally:
3069                dlg.Destroy()
3070               
3071    def OnRenameData(self,event):
3072        'Renames an existing phase. Called by Data/Rename Phase menu'
3073        name = self.PatternTree.GetItemText(self.PickId)     
3074        if 'PWDR' in name or 'HKLF' in name or 'IMG' in name:
3075            if 'Bank' in name:
3076                names = name.split('Bank')
3077                names[1] = ' Bank'+names[1]
3078            elif 'Azm' in name:
3079                names = name.split('Azm')
3080                names[1] = ' Azm'+names[1]
3081            else:
3082                names = [name,'']
3083            dataType = names[0][:names[0].index(' ')+1]                 #includes the ' '
3084            dlg = wx.TextEntryDialog(self,'Data name: '+name,'Change data name',
3085                defaultValue=names[0][names[0].index(' ')+1:])
3086            try:
3087                if dlg.ShowModal() == wx.ID_OK:
3088                    name = dataType+dlg.GetValue()+names[1]
3089                    self.PatternTree.SetItemText(self.PickId,name)
3090            finally:
3091                dlg.Destroy()
3092       
3093    def GetFileList(self,fileType,skip=None):        #potentially useful?
3094        'Appears unused. Note routine of same name in GSASIIpwdGUI'
3095        fileList = []
3096        Source = ''
3097        id, cookie = self.PatternTree.GetFirstChild(self.root)
3098        while id:
3099            name = self.PatternTree.GetItemText(id)
3100            if fileType in name:
3101                if id == skip:
3102                    Source = name
3103                else:
3104                    fileList.append([False,name,id])
3105            id, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3106        if skip:
3107            return fileList,Source
3108        else:
3109            return fileList
3110           
3111    def OnDataDelete(self, event):
3112        '''Delete one or more histograms from data tree. Called by the
3113        Data/DeleteData menu
3114        '''
3115        TextList = []
3116        DelList = []
3117        DelItemList = []
3118        nItems = {'PWDR':0,'SASD':0,'IMG':0,'HKLF':0,'PDF':0}
3119        if self.PatternTree.GetCount():
3120            item, cookie = self.PatternTree.GetFirstChild(self.root)
3121            while item:
3122                name = self.PatternTree.GetItemText(item)
3123                if name not in ['Notebook','Controls','Covariance','Constraints',
3124                    'Restraints','Phases','Rigid bodies'] or 'Sequential' not in name:
3125                    if 'PWDR' in name: nItems['PWDR'] += 1
3126                    if 'SASD' in name: nItems['SASD'] += 1
3127                    if 'IMG' in name:  nItems['IMG'] += 1
3128                    if 'HKLF' in name: nItems['HKLF'] += 1
3129                    if 'PDF' in name:  nItems['PDF'] += 1
3130                    TextList.append(name)
3131                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3132            dlg = G2G.G2MultiChoiceDialog(self, 'Which data to delete?', 'Delete data', TextList, wx.CHOICEDLG_STYLE)
3133            try:
3134                if dlg.ShowModal() == wx.ID_OK:
3135                    result = dlg.GetSelections()
3136                    for i in result: DelList.append(TextList[i])
3137                    item, cookie = self.PatternTree.GetFirstChild(self.root)
3138                    while item:
3139                        itemName = self.PatternTree.GetItemText(item)
3140                        if itemName in DelList:
3141                            if 'PWDR' in itemName: nItems['PWDR'] -= 1
3142                            elif 'SASD' in itemName: nItems['SASD'] -= 1
3143                            elif 'IMG' in itemName: nItems['IMG'] -= 1
3144                            elif 'HKLF' in itemName: nItems['HKLF'] -= 1
3145                            elif 'PDF' in itemName: nItems['PDF'] -= 1
3146                            DelItemList.append(item)
3147                        item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3148                    for item in DelItemList:
3149                        self.PatternTree.Delete(item)
3150                    self.PickId = 0
3151                    self.PickIdText = None
3152                    self.PatternId = 0
3153                    if nItems['PWDR']:
3154                        wx.CallAfter(G2plt.PlotPatterns,self,True)
3155                    else:
3156                        self.G2plotNB.Delete('Powder Patterns')
3157                    if not nItems['IMG']:
3158                        self.G2plotNB.Delete('2D Powder Image')
3159                    if not nItems['HKLF']:
3160                        self.G2plotNB.Delete('Structure Factors')
3161                        if '3D Structure Factors' in self.G2plotNB.plotList:
3162                            self.G2plotNB.Delete('3D Structure Factors')
3163            finally:
3164                dlg.Destroy()
3165
3166    def OnFileOpen(self, event, filename=None):
3167        '''Gets a GSAS-II .gpx project file in response to the
3168        File/Open Project menu button
3169        '''
3170        result = wx.ID_OK
3171        self.EnablePlot = False
3172        if self.PatternTree.GetChildrenCount(self.root,False):
3173            if self.dataFrame:
3174                self.dataFrame.Clear() 
3175            dlg = wx.MessageDialog(
3176                self,
3177                'Do you want to overwrite the current project? '+
3178                'Any unsaved changes in current project will be lost. Press OK to continue.',
3179                'Overwrite?',  wx.OK | wx.CANCEL)
3180            try:
3181                result = dlg.ShowModal()
3182                if result == wx.ID_OK:
3183                    self.PatternTree.DeleteChildren(self.root)
3184                    self.GSASprojectfile = ''
3185                    self.HKL = []
3186                    if self.G2plotNB.plotList:
3187                        self.G2plotNB.clear()
3188            finally:
3189                dlg.Destroy()
3190        if result != wx.ID_OK: return
3191
3192        if not filename:
3193            if self.dataDisplay: self.dataDisplay.Destroy()
3194            if self.LastGPXdir:
3195                pth = self.LastGPXdir
3196            else:
3197                pth = '.'
3198            dlg = wx.FileDialog(self, 'Choose GSAS-II project file', pth, 
3199                wildcard='GSAS-II project file (*.gpx)|*.gpx',style=wx.OPEN)
3200            try:
3201                if dlg.ShowModal() != wx.ID_OK: return
3202                self.GSASprojectfile = dlg.GetPath()
3203                self.GSASprojectfile = G2IO.FileDlgFixExt(dlg,self.GSASprojectfile)
3204                self.dirname = dlg.GetDirectory()
3205            finally:
3206                dlg.Destroy()
3207        else:
3208            self.GSASprojectfile = os.path.splitext(filename)[0]+'.gpx'
3209            self.dirname = os.path.split(filename)[0]
3210
3211        try:
3212            self.StartProject()         #open the file if possible
3213        except:
3214            print '\nError opening file ',filename
3215            import traceback
3216            print traceback.format_exc()
3217       
3218    def StartProject(self):
3219        '''Opens a GSAS-II project file & selects the 1st available data set to
3220        display (PWDR, HKLF or SASD)
3221        '''
3222       
3223        Id = 0
3224        phaseId = None
3225        G2IO.ProjFileOpen(self)
3226        self.PatternTree.SetItemText(self.root,'Loaded Data: '+self.GSASprojectfile)
3227        self.PatternTree.Expand(self.root)
3228        self.HKL = []
3229        item, cookie = self.PatternTree.GetFirstChild(self.root)
3230        while item and not Id:
3231            name = self.PatternTree.GetItemText(item)
3232            if name[:4] in ['PWDR','HKLF','IMG ','PDF ','SASD',]:
3233                Id = item
3234            elif name == "Phases":
3235                phaseId = item
3236            elif name == 'Controls':
3237                data = self.PatternTree.GetItemPyData(item)
3238                if data:
3239                    for item in self.Refine: item.Enable(True)
3240                    self.EnableSeqRefineMenu()
3241            item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3242        if phaseId: # show all phases
3243            self.PatternTree.Expand(phaseId)
3244        if Id:
3245            self.EnablePlot = True
3246            self.PatternTree.SelectItem(Id)
3247            self.PatternTree.Expand(Id)
3248        elif phaseId:
3249            self.PatternTree.SelectItem(phaseId)
3250        self.CheckNotebook()
3251        if self.dirname: os.chdir(self.dirname)           # to get Mac/Linux to change directory!
3252        pth = os.path.split(os.path.abspath(self.GSASprojectfile))[0]
3253        if GSASIIpath.GetConfigValue('Save_paths'): G2G.SaveGPXdirectory(pth)
3254        self.LastGPXdir = pth
3255
3256    def OnFileClose(self, event):
3257        '''Clears the data tree in response to the
3258        File/New Project menu button. User is given option to save
3259        the project.
3260        '''
3261        if self.dataFrame:
3262            self.dataFrame.Clear()
3263            self.dataFrame.SetLabel('GSAS-II data display') 
3264        dlg = wx.MessageDialog(self, 'Save current project?', ' ', wx.YES | wx.NO | wx.CANCEL)
3265        try:
3266            result = dlg.ShowModal()
3267            if result == wx.ID_OK:
3268                self.OnFileSaveMenu(event)
3269            if result != wx.ID_CANCEL:
3270                self.GSASprojectfile = ''
3271                self.PatternTree.SetItemText(self.root,'Loaded Data: ')
3272                self.PatternTree.DeleteChildren(self.root)
3273                if self.HKL: self.HKL = []
3274                if self.G2plotNB.plotList:
3275                    self.G2plotNB.clear()
3276        finally:
3277            dlg.Destroy()
3278
3279    def OnFileSave(self, event):
3280        '''Save the current project in response to the
3281        File/Save Project menu button
3282        '''
3283       
3284        if self.GSASprojectfile: 
3285            self.PatternTree.SetItemText(self.root,'Loaded Data: '+self.GSASprojectfile)
3286            self.CheckNotebook()
3287            G2IO.ProjFileSave(self)
3288        else:
3289            self.OnFileSaveas(event)
3290
3291    def OnFileSaveas(self, event):
3292        '''Save the current project in response to the
3293        File/Save as menu button
3294        '''
3295        if GSASIIpath.GetConfigValue('Starting_directory'):
3296            pth = GSASIIpath.GetConfigValue('Starting_directory')
3297            pth = os.path.expanduser(pth) 
3298        elif self.LastGPXdir:
3299            pth = self.LastGPXdir
3300        else:
3301            pth = '.'
3302        dlg = wx.FileDialog(self, 'Choose GSAS-II project file name', pth, '', 
3303            'GSAS-II project file (*.gpx)|*.gpx',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
3304        try:
3305            if dlg.ShowModal() == wx.ID_OK:
3306                self.GSASprojectfile = dlg.GetPath()
3307                self.GSASprojectfile = G2IO.FileDlgFixExt(dlg,self.GSASprojectfile)
3308                self.PatternTree.SetItemText(self.root,'Saving project as'+self.GSASprojectfile)
3309                self.SetTitle("GSAS-II data tree: "+os.path.split(self.GSASprojectfile)[1])
3310                self.CheckNotebook()
3311                G2IO.ProjFileSave(self)
3312                os.chdir(dlg.GetDirectory())           # to get Mac/Linux to change directory!
3313        finally:
3314            dlg.Destroy()
3315
3316    def ExitMain(self, event):
3317        '''Called if the main window is closed'''
3318        if self.G2plotNB:
3319            self.G2plotNB.Destroy()
3320        if self.dataFrame:
3321            self.dataFrame.Clear() 
3322            self.dataFrame.Destroy()
3323        if self.undofile:
3324            os.remove(self.undofile)
3325        sys.exit()
3326       
3327    def OnFileExit(self, event):
3328        '''Called in response to the File/Quit menu button'''
3329        if self.G2plotNB:
3330            self.G2plotNB.Destroy()
3331        if self.dataFrame:
3332            self.dataFrame.Clear() 
3333            self.dataFrame.Destroy()
3334        self.Close()
3335       
3336    def OnExportPeakList(self,event):
3337        nptand = lambda x: np.tan(x*math.pi/180.)
3338        pth = G2G.GetExportPath(self)
3339        dlg = wx.FileDialog(self, 'Choose output peak list file name', pth, '', 
3340            '(*.*)|*.*',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
3341        try:
3342            if dlg.ShowModal() == wx.ID_OK:
3343                self.peaklistfile = dlg.GetPath()
3344                self.peaklistfile = G2IO.FileDlgFixExt(dlg,self.peaklistfile)
3345                file = open(self.peaklistfile,'w')               
3346                item, cookie = self.PatternTree.GetFirstChild(self.root)
3347                while item:
3348                    name = self.PatternTree.GetItemText(item)
3349                    if 'PWDR' in name:
3350                        item2, cookie2 = self.PatternTree.GetFirstChild(item)
3351                        wave = 0.0
3352                        while item2:
3353                            name2 = self.PatternTree.GetItemText(item2)
3354                            if name2 == 'Instrument Parameters':
3355                                Inst = self.PatternTree.GetItemPyData(item2)[0]
3356                                Type = Inst['Type'][0]
3357                                if 'T' not in Type:
3358                                    wave = G2mth.getWave(Inst)
3359                            elif name2 == 'Peak List':
3360                                pkdata = self.PatternTree.GetItemPyData(item2)
3361                                peaks = pkdata['peaks']
3362                                sigDict = pkdata['sigDict']
3363                            item2, cookie2 = self.PatternTree.GetNextChild(item, cookie2)                           
3364                        file.write("#%s \n" % (name+' Peak List'))
3365                        if wave:
3366                            file.write('#wavelength = %10.6f\n'%(wave))
3367                        if 'T' in Type:
3368                            file.write('#%9s %10s %10s %12s %10s %10s %10s %10s %10s\n'%('pos','dsp','esd','int','alp','bet','sig','gam','FWHM'))                                   
3369                        else:
3370                            file.write('#%9s %10s %10s %12s %10s %10s %10s\n'%('pos','dsp','esd','int','sig','gam','FWHM'))
3371                        for ip,peak in enumerate(peaks):
3372                            dsp = G2lat.Pos2dsp(Inst,peak[0])
3373                            if 'T' in Type:  #TOF - more cols
3374                                esds = {'pos':0.,'int':0.,'alp':0.,'bet':0.,'sig':0.,'gam':0.}
3375                                for name in esds.keys():
3376                                    esds[name] = sigDict.get('%s%d'%(name,ip),0.)
3377                                sig = np.sqrt(peak[8])
3378                                gam = peak[10]
3379                                esddsp = G2lat.Pos2dsp(Inst,esds['pos'])
3380                                FWHM = G2pwd.getgamFW(gam,sig)      #to get delta-TOF from Gam(peak)
3381                                file.write("%10.2f %10.5f %10.5f %12.2f %10.3f %10.3f %10.3f %10.3f %10.3f\n" % \
3382                                    (peak[0],dsp,esddsp,peak[2],np.sqrt(max(0.0001,peak[4])),peak[6],peak[8],peak[10],FWHM))
3383                            else:               #CW
3384                                #get esds from sigDict for each peak & put in output - esds for sig & gam from UVWXY?
3385                                esds = {'pos':0.,'int':0.,'sig':0.,'gam':0.}
3386                                for name in esds.keys():
3387                                    esds[name] = sigDict.get('%s%d'%(name,ip),0.)
3388                                sig = np.sqrt(peak[4]) #var -> sig
3389                                gam = peak[6]
3390                                esddsp = 0.5*esds['pos']*dsp/nptand(peak[0]/2.)
3391                                FWHM = G2pwd.getgamFW(gam,sig)      #to get delta-2-theta in deg. from Gam(peak)
3392                                file.write("%10.4f %10.5f %10.5f %12.2f %10.5f %10.5f %10.5f \n" % \
3393                                    (peak[0],dsp,esddsp,peak[2],np.sqrt(max(0.0001,peak[4]))/100.,peak[6]/100.,FWHM/100.)) #convert to deg
3394                    item, cookie = self.PatternTree.GetNextChild(self.root, cookie)                           
3395                file.close()
3396        finally:
3397            dlg.Destroy()
3398       
3399    def OnExportHKL(self,event):
3400        pth = G2G.GetExportPath(self)
3401        dlg = wx.FileDialog(self, 'Choose output reflection list file name', pth, '', 
3402            '(*.*)|*.*',wx.FD_SAVE|wx.FD_OVERWRITE_PROMPT)
3403        try:
3404            if dlg.ShowModal() == wx.ID_OK:
3405                self.peaklistfile = dlg.GetPath()
3406                self.peaklistfile = G2IO.FileDlgFixExt(dlg,self.peaklistfile)
3407                file = open(self.peaklistfile,'w')               
3408                item, cookie = self.PatternTree.GetFirstChild(self.root)
3409                while item:
3410                    name = self.PatternTree.GetItemText(item)
3411                    if 'PWDR' in name:
3412                        item2, cookie2 = self.PatternTree.GetFirstChild(item)
3413                        while item2:
3414                            name2 = self.PatternTree.GetItemText(item2)
3415                            if name2 == 'Reflection Lists':
3416                                data = self.PatternTree.GetItemPyData(item2)
3417                                phases = data.keys()
3418                                for phase in phases:
3419                                    peaks = data[phase]
3420                                    file.write("%s %s %s \n" % (name,phase,' Reflection List'))
3421                                    if 'T' in peaks.get('Type','PXC'):
3422                                        file.write('%s \n'%('   h   k   l   m    d-space     TOF         wid        F**2'))
3423                                    else:               
3424                                        file.write('%s \n'%('   h   k   l   m    d-space   2-theta       wid        F**2'))
3425                                    for peak in peaks['RefList']:
3426                                        if 'T' in peaks.get('Type','PXC'):
3427                                            sig = np.sqrt(peak[6])
3428                                            gam = peak[7]
3429                                            FWHM = G2pwd.getgamFW(gam,sig)
3430                                            file.write(" %3d %3d %3d %3d %10.5f %10.2f %10.5f %10.3f \n" % \
3431                                                (int(peak[0]),int(peak[1]),int(peak[2]),int(peak[3]),peak[4],peak[5],FWHM,peak[8]))
3432                                        else:
3433                                            sig = np.sqrt(peak[6])
3434                                            gam = peak[7]
3435                                            FWHM = G2pwd.getgamFW(gam,sig)
3436                                            file.write(" %3d %3d %3d %3d %10.5f %10.5f %10.5f %10.3f \n" % \
3437                                                (int(peak[0]),int(peak[1]),int(peak[2]),int(peak[3]),peak[4],peak[5],FWHM/100.,peak[8]))
3438                            item2, cookie2 = self.PatternTree.GetNextChild(item, cookie2)                           
3439                    item, cookie = self.PatternTree.GetNextChild(self.root, cookie)                           
3440                file.close()
3441        finally:
3442            dlg.Destroy()
3443       
3444    def OnExportPDF(self,event):
3445        #need S(Q) and G(R) to be saved here - probably best from selection?
3446        names = ['All']
3447        exports = []
3448        item, cookie = self.PatternTree.GetFirstChild(self.root)
3449        while item:
3450            name = self.PatternTree.GetItemText(item)
3451            if 'PDF' in name:
3452                names.append(name)
3453            item, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3454        if names:
3455            dlg = wx.MultiChoiceDialog(self,'Select','PDF patterns to export',names)
3456            if dlg.ShowModal() == wx.ID_OK:
3457                sel = dlg.GetSelections()
3458                if sel[0] == 0:
3459                    exports = names[1:]
3460                else:
3461                    for x in sel:
3462                        exports.append(names[x])
3463            dlg.Destroy()
3464        if exports:
3465            G2IO.PDFSave(self,exports)
3466       
3467    def OnMakePDFs(self,event):
3468        '''Sets up PDF data structure filled with defaults; if found chemical formula is inserted
3469        so a default PDF can be made.
3470        '''
3471        sind = lambda x: math.sin(x*math.pi/180.)
3472        tth2q = lambda t,w:4.0*math.pi*sind(t/2.0)/w
3473        tof2q = lambda t,C:2.0*math.pi*C/t
3474        TextList = []
3475        ElLists = []
3476        Qlimits = []
3477        Names = []
3478        if self.PatternTree.GetCount():
3479            id, cookie = self.PatternTree.GetFirstChild(self.root)
3480            while id:
3481                name = self.PatternTree.GetItemText(id)
3482                Names.append(name)
3483                if 'PWDR' in name:
3484                    TextList.append(name)
3485                    Comments = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,id,'Comments'))
3486                    Parms = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,id,'Instrument Parameters'))[0]
3487                    fullLimits = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,id,'Limits'))[0]
3488                    if 'C' in Parms['Type'][0]:
3489                        wave = G2mth.getWave(Parms)
3490                        qMax = tth2q(fullLimits[1],wave)
3491                    else:   #'T'of
3492                        qMax = tof2q(fullLimits[0],Parms['difC'][1])
3493                    Qlimits.append([0.9*qMax,qMax])
3494                    ElList = {}
3495                    for item in Comments:           #grab chemical formula from Comments
3496                        if 'formula' in item[:15].lower():
3497                            formula = item.split('=')[1].split()
3498                            elems = formula[::2]
3499                            nums = formula[1::2]
3500                            formula = zip(elems,nums)
3501                            for [elem,num] in formula:
3502                                ElData = G2elem.GetElInfo(elem,Parms)
3503                                ElData['FormulaNo'] = float(num)
3504                                ElList[elem] = ElData
3505                    ElLists.append(ElList)
3506                id, cookie = self.PatternTree.GetNextChild(self.root, cookie)
3507            if len(TextList) < 1:
3508                self.ErrorDialog('Nothing to make PDFs for','There must be at least one "PWDR" pattern')
3509                return
3510            dlg = G2G.G2MultiChoiceDialog(self,'Make PDF controls','Make PDF controls for:',TextList, wx.CHOICEDLG_STYLE)
3511            try:
3512                if dlg.ShowModal() == wx.ID_OK:
3513                    for i in dlg.GetSelections():
3514                        item = TextList[i]
3515                        ElList = ElLists[i]
3516                        PWDRname = item[4:]
3517                        Id = self.PatternTree.AppendItem(parent=self.root,text='PDF '+PWDRname)
3518                        Data = {
3519                            'Sample':{'Name':item,'Mult':1.0,'Add':0.0},
3520                            'Sample Bkg.':{'Name':'','Mult':-1.0,'Add':0.0},
3521                            'Container':{'Name':'','Mult':-1.0,'Add':0.0},
3522                            'Container Bkg.':{'Name':'','Mult':-1.0,'Add':0.0},'ElList':ElList,
3523                            'Geometry':'Cylinder','Diam':1.0,'Pack':0.50,'Form Vol':10.0,
3524                            'DetType':'Image plate','ObliqCoeff':0.2,'Ruland':0.025,'QScaleLim':Qlimits[i],
3525                            'Lorch':False,'BackRatio':0.0,'Rmax':100.,'noRing':False,'IofQmin':1.0}
3526                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='PDF Controls'),Data)
3527                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='I(Q)'+PWDRname),[])       
3528                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='S(Q)'+PWDRname),[])       
3529                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='F(Q)'+PWDRname),[])       
3530                        self.PatternTree.SetItemPyData(self.PatternTree.AppendItem(Id,text='G(R)'+PWDRname),[])       
3531                for item in self.ExportPDF: item.Enable(True)
3532            finally:
3533                dlg.Destroy()
3534               
3535    def GetPWDRdatafromTree(self,PWDRname):
3536        ''' Returns powder data from GSASII tree
3537
3538        :param str PWDRname: a powder histogram name as obtained from
3539          :meth:`GSASIIstruct.GetHistogramNames`
3540
3541        :returns: PWDRdata = powder data dictionary with
3542          Powder data arrays, Limits, Instrument Parameters,
3543          Sample Parameters           
3544        '''
3545        PWDRdata = {}
3546        try:
3547            PWDRdata.update(self.PatternTree.GetItemPyData(PWDRname)[0])            #wtFactor + ?
3548        except ValueError:
3549            PWDRdata['wtFactor'] = 1.0
3550        PWDRdata['Data'] = self.PatternTree.GetItemPyData(PWDRname)[1]          #powder data arrays
3551        PWDRdata['Limits'] = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PWDRname,'Limits'))
3552        PWDRdata['Background'] = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PWDRname,'Background'))
3553        PWDRdata['Instrument Parameters'] = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PWDRname,'Instrument Parameters'))
3554        PWDRdata['Sample Parameters'] = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PWDRname,'Sample Parameters'))
3555        PWDRdata['Reflection Lists'] = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,PWDRname,'Reflection Lists'))
3556        if 'ranId' not in PWDRdata:  # patch, add a random Id
3557            PWDRdata['ranId'] = ran.randint(0,sys.maxint)
3558        if 'ranId' not in PWDRdata['Sample Parameters']:  # I hope this becomes obsolete at some point
3559            PWDRdata['Sample Parameters']['ranId'] = PWDRdata['ranId']
3560        return PWDRdata
3561
3562    def GetHKLFdatafromTree(self,HKLFname):
3563        ''' Returns single crystal data from GSASII tree
3564
3565        :param str HKLFname: a single crystal histogram name as obtained
3566          from
3567          :meth:`GSASIIstruct.GetHistogramNames`
3568
3569        :returns: HKLFdata = single crystal data list of reflections
3570
3571        '''
3572        HKLFdata = {}
3573        HKLFdata.update(self.PatternTree.GetItemPyData(HKLFname)[0])            #wtFactor + ?
3574#        try:
3575#            HKLFdata.update(self.PatternTree.GetItemPyData(HKLFname)[0])            #wtFactor + ?
3576#        except ValueError:
3577#            HKLFdata['wtFactor'] = 1.0
3578        HKLFdata['Data'] = self.PatternTree.GetItemPyData(HKLFname)[1]
3579        HKLFdata['Instrument Parameters'] = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,HKLFname,'Instrument Parameters'))
3580        return HKLFdata
3581       
3582    def GetPhaseData(self):
3583        '''Returns a dict with defined phases.
3584        Note routine :func:`GSASIIstrIO.GetPhaseData` also exists to
3585        get same info from GPX file.
3586        '''
3587        phaseData = {}
3588        if G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
3589            sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
3590        else:
3591            print 'no phases found in GetPhaseData'
3592            sub = None
3593        if sub:
3594            item, cookie = self.PatternTree.GetFirstChild(sub)
3595            while item:
3596                phaseName = self.PatternTree.GetItemText(item)
3597                phaseData[phaseName] =  self.PatternTree.GetItemPyData(item)
3598                if 'ranId' not in phaseData[phaseName]:
3599                    phaseData[phaseName]['ranId'] = ran.randint(0,sys.maxint)         
3600                item, cookie = self.PatternTree.GetNextChild(sub, cookie)
3601        return phaseData
3602
3603    def GetPhaseInfofromTree(self):
3604        '''Get the phase names and their rId values,
3605        also the histograms used in each phase.
3606
3607        :returns: (phaseRIdList, usedHistograms) where
3608
3609          * phaseRIdList is a list of random Id values for each phase
3610          * usedHistograms is a dict where the keys are the phase names
3611            and the values for each key are a list of the histogram names
3612            used in each phase.
3613        '''
3614        phaseRIdList = []
3615        usedHistograms = {}
3616        sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
3617        if sub:
3618            item, cookie = self.PatternTree.GetFirstChild(sub)
3619            while item:
3620                phaseName = self.PatternTree.GetItemText(item)
3621                ranId = self.PatternTree.GetItemPyData(item).get('ranId')
3622                if ranId: phaseRIdList.append(ranId)
3623                data = self.PatternTree.GetItemPyData(item)
3624                UseList = data['Histograms']
3625                usedHistograms[phaseName] = UseList.keys()
3626                item, cookie = self.PatternTree.GetNextChild(sub, cookie)
3627        return phaseRIdList,usedHistograms
3628
3629    def GetPhaseNames(self):
3630        '''Returns a list of defined phases.
3631        Note routine :func:`GSASIIstrIO.GetPhaseNames` also exists to
3632        get same info from GPX file.
3633        '''
3634        phaseNames = []
3635        if G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
3636            sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
3637        else:
3638            print 'no phases found in GetPhaseNames'
3639            sub = None
3640        if sub:
3641            item, cookie = self.PatternTree.GetFirstChild(sub)
3642            while item:
3643                phase = self.PatternTree.GetItemText(item)
3644                phaseNames.append(phase)
3645                item, cookie = self.PatternTree.GetNextChild(sub, cookie)
3646        return phaseNames
3647   
3648    def GetHistogramNames(self,hType):
3649        """ Returns a list of histogram names found in the GSASII data tree
3650        Note routine :func:`GSASIIstrIO.GetHistogramNames` also exists to
3651        get same info from GPX file.
3652       
3653        :param str hType: list of histogram types
3654        :return: list of histogram names
3655       
3656        """
3657        HistogramNames = []
3658        if self.PatternTree.GetCount():
3659            item, cookie = self.PatternTree.GetFirstChild(self.root)
3660            while item:
3661                name = self.PatternTree.GetItemText(item)
3662                if name[:4] in hType:
3663                    HistogramNames.append(name)       
3664                item, cookie = self.PatternTree.GetNextChild(self.root, cookie)               
3665
3666        return HistogramNames
3667                   
3668    def GetUsedHistogramsAndPhasesfromTree(self):
3669        ''' Returns all histograms that are found in any phase
3670        and any phase that uses a histogram.
3671        This also assigns numbers to used phases and histograms by the
3672        order they appear in the file.
3673        Note routine :func:`GSASIIstrIO.GetUsedHistogramsAndPhasesfromTree` also exists to
3674        get same info from GPX file.
3675
3676        :returns: (Histograms,Phases)
3677
3678            * Histograms = dictionary of histograms as {name:data,...}
3679            * Phases = dictionary of phases that use histograms
3680        '''
3681        Histograms = {}
3682        Phases = {}
3683        phaseNames = self.GetPhaseNames()
3684        phaseData = self.GetPhaseData()
3685        histoList = self.GetHistogramNames(['PWDR','HKLF'])
3686
3687        for phase in phaseData:
3688            Phase = phaseData[phase]
3689            pId = phaseNames.index(phase)
3690            Phase['pId'] = pId
3691            if Phase['Histograms']:
3692                if phase not in Phases:
3693                    Phases[phase] = Phase
3694                for hist in Phase['Histograms']:
3695                    if 'Use' not in Phase['Histograms'][hist]:      #patch: add Use flag as True
3696                        Phase['Histograms'][hist]['Use'] = True         
3697                    if hist not in Histograms and Phase['Histograms'][hist]['Use']:
3698                        item = G2gd.GetPatternTreeItemId(self,self.root,hist)
3699                        if item:
3700                            if 'PWDR' in hist[:4]: 
3701                                Histograms[hist] = self.GetPWDRdatafromTree(item)
3702                            elif 'HKLF' in hist[:4]:
3703                                Histograms[hist] = self.GetHKLFdatafromTree(item)
3704                            hId = histoList.index(hist)
3705                            Histograms[hist]['hId'] = hId
3706                        else: # would happen if a referenced histogram were renamed or deleted
3707                            print('For phase "'+str(phase)+
3708                                  '" unresolved reference to histogram "'+str(hist)+'"')
3709        #G2obj.IndexAllIds(Histograms=Histograms,Phases=Phases)
3710        G2obj.IndexAllIds(Histograms=Histograms,Phases=phaseData)
3711        return Histograms,Phases
3712       
3713    def MakeLSParmDict(self):
3714        '''Load all parameters used for computation from the tree into a
3715        dict of paired values [value, refine flag]. Note that this is
3716        different than the parmDict used in the refinement, which only has
3717        values.
3718
3719        Note that similar things are done in
3720        :meth:`GSASIIIO.ExportBaseclass.loadParmDict` (from the tree) and
3721        :func:`GSASIIstrMain.Refine` and :func:`GSASIIstrMain.SeqRefine` (from
3722        a GPX file).
3723
3724        :returns: (parmDict,varyList) where:
3725
3726         * parmDict is a dict with values and refinement flags
3727           for each parameter and
3728         * varyList is a list of variables (refined parameters).
3729        '''
3730        parmDict = {}
3731        Histograms,Phases = self.GetUsedHistogramsAndPhasesfromTree()
3732        for phase in Phases:
3733            if 'pId' not in Phases[phase]:
3734                self.ErrorDialog('View parameter error','You must run least squares at least once')
3735                raise Exception,'No pId for phase '+str(phase)
3736        rigidbodyDict = self.PatternTree.GetItemPyData(   
3737            G2gd.GetPatternTreeItemId(self,self.root,'Rigid bodies'))
3738        rbVary,rbDict = G2stIO.GetRigidBodyModels(rigidbodyDict,Print=False)
3739        rbIds = rigidbodyDict.get('RBIds',{'Vector':[],'Residue':[]})
3740        Natoms,atomIndx,phaseVary,phaseDict,pawleyLookup,FFtable,BLtable,MFtable,maxSSwave = G2stIO.GetPhaseData(Phases,RestraintDict=None,rbIds=rbIds,Print=False)       
3741        hapVary,hapDict,controlDict = G2stIO.GetHistogramPhaseData(Phases,Histograms,Print=False)
3742        histVary,histDict,controlDict = G2stIO.GetHistogramData(Histograms,Print=False)
3743        varyList = rbVary+phaseVary+hapVary+histVary
3744        parmDict.update(rbDict)
3745        parmDict.update(phaseDict)
3746        parmDict.update(hapDict)
3747        parmDict.update(histDict)
3748        for parm in parmDict:
3749            if parm.split(':')[-1] in ['Azimuth','Gonio. radius','Lam1','Lam2',
3750                'Omega','Chi','Phi','nDebye','nPeaks']:
3751                parmDict[parm] = [parmDict[parm],'-']
3752            elif parm.split(':')[-2] in ['Ax','Ay','Az','SHmodel','SHord']:
3753                parmDict[parm] = [parmDict[parm],'-']
3754            elif parm in varyList:
3755                parmDict[parm] = [parmDict[parm],'T']
3756            else:
3757                parmDict[parm] = [parmDict[parm],'F']
3758        # for i in parmDict: print i,'\t',parmDict[i]
3759        # fl = open('parmDict.dat','wb')
3760        # import cPickle
3761        # cPickle.dump(parmDict,fl,1)
3762        # fl.close()
3763        return parmDict,varyList
3764
3765    def ShowLSParms(self,event):
3766        '''Displays a window showing all parameters in the refinement.
3767        Called from the Calculate/View LS Parms menu.
3768        '''
3769        parmDict,varyList = self.MakeLSParmDict()
3770        parmValDict = {}
3771        for i in parmDict:
3772            parmValDict[i] = parmDict[i][0]
3773           
3774        reqVaryList = tuple(varyList) # save requested variables
3775        try:
3776            # process constraints
3777            sub = G2gd.GetPatternTreeItemId(self,self.root,'Constraints') 
3778            Constraints = self.PatternTree.GetItemPyData(sub)
3779            constList = []
3780            for item in Constraints:
3781                if item.startswith('_'): continue
3782                constList += Constraints[item]
3783            G2mv.InitVars()
3784            constrDict,fixedList,ignored = G2stIO.ProcessConstraints(constList)
3785            groups,parmlist = G2mv.GroupConstraints(constrDict)
3786            G2mv.GenerateConstraints(groups,parmlist,varyList,constrDict,fixedList,parmValDict)
3787            G2mv.Map2Dict(parmValDict,varyList)
3788        except:
3789            pass
3790        dlg = G2gd.ShowLSParms(self,'Least Squares Parameters',parmValDict,varyList,reqVaryList)
3791        dlg.ShowModal()
3792        dlg.Destroy()
3793
3794    def OnRefine(self,event):
3795        '''Perform a refinement.
3796        Called from the Calculate/Refine menu.
3797        '''
3798        Id = G2gd.GetPatternTreeItemId(self,self.root,'Sequential results')
3799        if Id:
3800            dlg = wx.MessageDialog(
3801                self,
3802                'Your last refinement was sequential. Continue with "Refine", removing previous sequential results?',
3803                'Remove sequential results?',wx.OK|wx.CANCEL)
3804            if dlg.ShowModal() == wx.ID_OK:
3805                self.PatternTree.Delete(Id)
3806                dlg.Destroy()
3807            else:
3808                dlg.Destroy()
3809                return
3810        self.OnFileSave(event)
3811        # check that constraints are OK here
3812        errmsg, warnmsg = G2stIO.ReadCheckConstraints(self.GSASprojectfile)
3813        if errmsg:
3814            self.ErrorDialog('Refinement error',errmsg)
3815            return
3816        if warnmsg:
3817            print('Conflict between refinment flag settings and constraints:\n'+
3818                warnmsg+'\nRefinement not possible')
3819            self.ErrorDialog('Refinement Flag Error',
3820                'Conflict between refinement flag settings and constraints:\n'+
3821                warnmsg+'\nRefinement not possible')
3822            return
3823        dlg = wx.ProgressDialog('Residual','All data Rw =',101.0, 
3824            style = wx.PD_ELAPSED_TIME|wx.PD_AUTO_HIDE|wx.PD_CAN_ABORT,
3825            parent=self)
3826        Size = dlg.GetSize()
3827        if 50 < Size[0] < 500: # sanity check on size, since this fails w/Win & wx3.0
3828            dlg.SetSize((int(Size[0]*1.2),Size[1])) # increase size a bit along x
3829        dlg.CenterOnParent()
3830        Rw = 100.00
3831        self.SaveTreeSetting()
3832        self.PatternTree.SaveExposedItems()       
3833        try:
3834            OK,Msg = G2stMn.Refine(self.GSASprojectfile,dlg)    #Msg is Rvals dict if Ok=True
3835        finally:
3836            dlg.Update(101.) # forces the Auto_Hide; needed after move w/Win & wx3.0
3837            dlg.Destroy()
3838            wx.Yield()
3839        if OK:
3840            Rw = Msg['Rwp']
3841            lamMax = Msg.get('lamMax',0.001)
3842            text = 'Load new result?'
3843            if lamMax >= 10.:
3844                text += '\nWARNING: Steepest descents dominates;'+   \
3845                ' minimum may not have been reached\nor result may be false minimum.'+  \
3846                ' You should reconsider your parameter suite'
3847            dlg2 = wx.MessageDialog(self,text,'Refinement results, Rw =%.3f'%(Rw),wx.OK|wx.CANCEL)
3848            try:
3849                if dlg2.ShowModal() == wx.ID_OK:
3850                    self.PatternTree.DeleteChildren(self.root)
3851                    self.HKL = []
3852                    G2IO.ProjFileOpen(self,False)
3853                    self.PatternTree.RestoreExposedItems()       
3854                    self.ResetPlots()
3855            finally:
3856                dlg2.Destroy()
3857        else:
3858            self.ErrorDialog('Refinement error',Msg)
3859       
3860    def SaveTreeSetting(self):
3861        'Save the last tree setting'
3862        oldId =  self.PatternTree.GetSelection()        #retain current selection
3863        oldPath = self.GetTreeItemsList(oldId)
3864        self.lastTreeSetting = oldPath
3865        # note that for reasons unclear, it does not seem necessary to reload the Atoms tab
3866        #parentName = ''
3867        #tabId = None
3868        # parentId = self.PatternTree.GetItemParent(oldId)
3869        # if parentId:
3870        #     parentName = self.PatternTree.GetItemText(parentId)     #find the current data tree name
3871        #     if 'Phases' in parentName:
3872        #         tabId = self.dataDisplay.GetSelection()
3873        #self.lastTreeSetting = oldPath,tabId
3874        #GSASIIpath.IPyBreak()
3875       
3876    def TestResetPlot(self,event):
3877        '''Debug code to test cleaning up plots after a refinement'''
3878        #for i in range(self.G2plotNB.nb.GetPageCount()):
3879        #    [self.G2plotNB.nb.GetPageText(i)
3880        # save current tree item and (if needed) atoms tab
3881        self.SaveTreeSetting()
3882        self.ResetPlots()
3883       
3884    def ResetPlots(self):
3885        '''This reloads the current tree item, often drawing a plot. It refreshes any plots
3886        that have registered a refresh routine (see G2plotNB.RegisterRedrawRoutine)
3887        and deletes all plots that have not been refreshed and
3888        require one (see G2plotNB.SetNoDelete).
3889        '''
3890        lastRaisedPlotTab = self.G2plotNB.lastRaisedPlotTab # save the last page saved
3891        #print 'lastRaisedPlotTab=',lastRaisedPlotTab
3892        self.G2plotNB.lastRaisedPlotTab = None
3893        # mark displayed plots as invalid
3894        for lbl,frame in zip(self.G2plotNB.plotList,self.G2plotNB.panelList):
3895            frame.plotInvalid = True
3896        # reload current tree item, triggering the routine to redraw the data window and possibly a plot
3897        #oldPath,tabId = self.lastTreeSetting
3898        oldPath = self.lastTreeSetting
3899        Id = self.root
3900        for txt in oldPath:
3901            Id = G2gd.GetPatternTreeItemId(self, Id, txt)
3902        self.PickIdText = None  #force reload of page
3903        if Id:
3904            self.PickId = Id
3905            self.PatternTree.SelectItem(Id)
3906        # update other self-updating plots
3907        for lbl,frame in zip(self.G2plotNB.plotList,self.G2plotNB.panelList):
3908            if frame.plotInvalid and frame.replotFunction:
3909                frame.replotFunction(*frame.replotArgs,**frame.replotKWargs)
3910        # delete any remaining plots that are still invalid and need a refresh
3911        for lbl,frame in zip(self.G2plotNB.plotList,self.G2plotNB.panelList):
3912            if frame.plotInvalid and frame.plotRequiresRedraw:
3913                self.G2plotNB.Delete(lbl)
3914        # put the previously last-raised tab on top, if present. If not, use the one corresponding to
3915        # the last tree item to be selected
3916        wx.CallAfter(self.G2plotNB.RaiseLastPage,lastRaisedPlotTab,self.G2plotNB.lastRaisedPlotTab)
3917       
3918    def OnSeqRefine(self,event):
3919        '''Perform a sequential refinement.
3920        Called from the Calculate/Sequential refine menu.
3921        '''
3922        Id = G2gd.GetPatternTreeItemId(self,self.root,'Sequential results')
3923        if not Id:
3924            Id = self.PatternTree.AppendItem(self.root,text='Sequential results')
3925            self.PatternTree.SetItemPyData(Id,{})           
3926        Controls = self.PatternTree.GetItemPyData(G2gd.GetPatternTreeItemId(self,self.root, 'Controls'))
3927        if not Controls.get('Seq Data'):
3928            print('Error: a sequential refinement has not been set up')
3929            return
3930        Controls['ShowCell'] = True
3931        self.OnFileSave(event)
3932        # check that constraints are OK here
3933        errmsg, warnmsg = G2stIO.ReadCheckConstraints(self.GSASprojectfile)
3934        if errmsg:
3935            self.ErrorDialog('Refinement error',errmsg)
3936            return
3937        if warnmsg:
3938            print('Conflict between refinment flag settings and constraints:\n'+
3939                  warnmsg+'\nRefinement not possible')
3940            self.ErrorDialog('Refinement Flag Error',
3941                             'Conflict between refinment flag settings and constraints:\n'+
3942                             warnmsg+'\nRefinement not possible')
3943            return
3944        self.PatternTree.SaveExposedItems()       
3945        dlg = wx.ProgressDialog('Residual for histogram 0','Powder profile Rwp =',101.0, 
3946            style = wx.PD_ELAPSED_TIME|wx.PD_AUTO_HIDE|wx.PD_CAN_ABORT,
3947            parent=self)           
3948        Size = dlg.GetSize()
3949        if 50 < Size[0] < 500: # sanity check on size, since this fails w/Win & wx3.0
3950            dlg.SetSize((int(Size[0]*1.2),Size[1])) # increase size a bit along x
3951        dlg.CenterOnParent()
3952        try:
3953            OK,Msg = G2stMn.SeqRefine(self.GSASprojectfile,dlg)     #Msg is Rvals dict if Ok=True
3954        finally:
3955            dlg.Update(101.) # forces the Auto_Hide; needed after move w/Win & wx3.0
3956            dlg.Destroy()
3957            wx.Yield()
3958        if OK:
3959            dlg = wx.MessageDialog(self,'Load new result?','Refinement results',wx.OK|wx.CANCEL)
3960            try:
3961                if dlg.ShowModal() == wx.ID_OK:
3962                    self.PickIdText = None  #force reload of PickId contents
3963                    self.PatternTree.DeleteChildren(self.root)
3964                    if len(self.HKL): self.HKL = []
3965                    if self.G2plotNB.plotList:
3966                        self.G2plotNB.clear()
3967                    G2IO.ProjFileOpen(self,False)
3968                    self.PatternTree.RestoreExposedItems()
3969                    self.ResetPlots()
3970                    Id = G2gd.GetPatternTreeItemId(self,self.root,'Sequential results')
3971                    self.PatternTree.SelectItem(Id)
3972            finally:
3973                dlg.Destroy()
3974        else:
3975            self.ErrorDialog('Sequential refinement error',Msg)
3976       
3977    def ErrorDialog(self,title,message,parent=None, wtype=wx.OK):
3978        'Display an error message'
3979        result = None
3980        if parent is None:
3981            dlg = wx.MessageDialog(self, message, title,  wtype)
3982        else:
3983            dlg = wx.MessageDialog(parent, message, title,  wtype)
3984            dlg.CenterOnParent() # not working on Mac
3985        try:
3986            result = dlg.ShowModal()
3987        finally:
3988            dlg.Destroy()
3989        return result
3990   
3991    def OnSaveMultipleImg(self,event):
3992        '''Select and save multiple image parameter and mask files
3993        '''
3994        G2IO.SaveMultipleImg(self)
3995       
3996class GSASIImain(wx.App):
3997    '''Defines a wxApp for GSAS-II
3998
3999    Creates a wx frame (self.main) which contains the display of the
4000    data tree.
4001    '''
4002    def OnInit(self):
4003        '''Called automatically when the app is created.'''
4004        import platform
4005        if '2.7' not in sys.version[:5]:
4006            dlg = wx.MessageDialog(None, 
4007                'GSAS-II requires Python 2.7.x\n Yours is '+sys.version.split()[0],
4008                'Python version error',  wx.OK)
4009            try:
4010                dlg.ShowModal()
4011            finally:
4012                dlg.Destroy()
4013            sys.exit()
4014        self.main = GSASII(None)
4015        self.main.Show()
4016        self.SetTopWindow(self.main)
4017        # save the current package versions
4018        self.main.PackageVersions = []
4019        self.main.PackageVersions.append(['Python',sys.version.split()[0]])
4020        for p in (wx,mpl,np,sp,ogl):
4021            self.main.PackageVersions.append([p.__name__,p.__version__])
4022        try:
4023            self.main.PackageVersions.append([Image.__name__,Image.VERSION])
4024        except:
4025            try:
4026                from PIL import PILLOW_VERSION
4027                self.main.PackageVersions.append([Image.__name__,PILLOW_VERSION])
4028            except:
4029                pass
4030        self.main.PackageVersions.append(['Platform',sys.platform+' '+platform.architecture()[0]+' '+platform.machine()])
4031       
4032        return True
4033    # def MacOpenFile(self, filename):
4034    #     '''Called on Mac every time a file is dropped on the app when it is running,
4035    #     treat this like a File/Open project menu action.
4036    #     Should be ignored on other platforms
4037    #     '''
4038    #     # PATCH: Canopy 1.4 script main seems dropped on app; ignore .py files
4039    #     print 'MacOpen',filename
4040    #     if os.path.splitext(filename)[1] == '.py': return
4041    #     # end PATCH
4042    #     self.main.OnFileOpen(None,filename)
4043    # removed because this gets triggered when a file is on the command line in canopy 1.4 -- not likely used anyway
4044       
4045def main():
4046    '''Start up the GSAS-II application'''
4047    #application = GSASIImain() # don't redirect output, someday we
4048    # may want to do this if we can
4049    application = GSASIImain(0)
4050    if GSASIIpath.GetConfigValue('wxInspector'):
4051        import wx.lib.inspection as wxeye
4052        wxeye.InspectionTool().Show()
4053
4054    #application.main.OnRefine(None)
4055    application.MainLoop()
4056   
4057if __name__ == '__main__':
4058    # print versions
4059    print "Python module versions loaded:"
4060    print "  Python:     ",sys.version.split()[0]
4061    print "  wx:         ",wx.__version__
4062    print "  matplotlib: ",mpl.__version__
4063    print "  numpy:      ",np.__version__
4064    print "  scipy:      ",sp.__version__
4065    print "  OpenGL:     ",ogl.__version__
4066    try:
4067        from PIL import Image
4068        try:
4069            from PIL import PILLOW_VERSION
4070            version = PILLOW_VERSION
4071        except:
4072            version = Image.VERSION
4073        print "  PIL.Image:  ",version
4074    except ImportError:
4075        try:
4076            import Image
4077            print "Image (PIL):",Image.VERSION
4078        except ImportError:
4079            print "Image module not present; Note that PIL (Python Imaging Library) or pillow is needed for some image operations"
4080    import platform
4081    print "  Platform:   ",sys.platform,platform.architecture()[0],platform.machine()
4082    try:
4083        import mkl
4084        print "  Max threads:",mkl.get_max_threads()
4085    except:
4086        pass
4087    #print "wxPython description",wx.PlatformInfo
4088    print "This is GSAS-II revision "+str(GSASIIpath.GetVersionNumber())+'\n'
4089    GSASIIpath.InvokeDebugOpts()
4090    main() # start the GUI
Note: See TracBrowser for help on using the repository browser.