Changeset 577 for trunk/GSASII.py

Timestamp:

Apr 27, 2012 9:45:26 AM (10 years ago)

Author:

toby

Message:

fix error in reporting constraint errors

File:

• trunk/GSASII.py (modified) (4 diffs)

trunk/GSASII.py

@@ -356 +356 @@
         return
         
+    def _init_Import_Sfact(self,parent):
+        '''import all the G2importsfact*.py files that are found in the
+        path and configure the Import Structure Factor menus accordingly
+        '''
+
+        path2GSAS2 = os.path.dirname(os.path.realpath(__file__)) # location of this file
+        pathlist = sys.path[:]
+        if path2GSAS2 not in pathlist: pathlist.append(path2GSAS2)
+        filelist = []
+        for path in pathlist:
+            for filename in glob.iglob(os.path.join(path, "G2importsfact*.py")):
+                filelist.append(filename)    
+                #print 'found',filename
+        filelist = sorted(list(set(filelist))) # remove duplicates
+        self.ImportSfactReaderlist = []
+        for filename in filelist:
+            path,rootname = os.path.split(filename)
+            pkg = os.path.splitext(rootname)[0]
+            try:
+                fp = None
+                fp, fppath,desc = imp.find_module(pkg,[path,])
+                pkg = imp.load_module(pkg,fp,fppath,desc)
+                for clss in inspect.getmembers(pkg): # find classes defined in package
+                    if clss[0].startswith('_'): continue
+                    if inspect.isclass(clss[1]):
+                        # check if we have the required methods
+                        for m in 'Reader','ExtensionValidator','ContentsValidator':
+                            if not hasattr(clss[1],m): break
+                            if not callable(getattr(clss[1],m)): break
+                        else:
+                            reader = clss[1]() # create a Structure Factor import instance
+                            self.ImportSfactReaderlist.append(reader)
+            except AttributeError:
+                print 'Import_Sfact: Attribute Error',filename
+                pass
+            except ImportError:
+                print 'Import_Sfact: Error importing file',filename
+                pass
+            finally:
+                if fp: fp.close()
+        submenu = wx.Menu()
+        item = parent.AppendMenu(wx.ID_ANY, 'Import Structure Factor menu',
+            submenu, help='Import Structure Factor data')
+        self.SfactImportMenuId = {}
+        item = submenu.Append(wx.ID_ANY,
+                              help='Import Structure Factor data based selected by extension',
+                              kind=wx.ITEM_NORMAL,text='Import Structure Factor by extension')
+        self.Bind(wx.EVT_MENU, self.OnImportSfactGeneric, id=item.GetId())
+        for reader in self.ImportSfactReaderlist:
+            item = submenu.Append(wx.ID_ANY,
+                help='Import specific format Structure Factor data',
+                kind=wx.ITEM_NORMAL,text='Import Structure Factor '+reader.formatName+'...')
+            self.SfactImportMenuId[item.GetId()] = reader
+            self.Bind(wx.EVT_MENU, self.OnImportSfactGeneric, id=item.GetId())
+
+    def OnImportSfactGeneric(self,event):
+        # find out which format was requested
+        reader = self.SfactImportMenuId.get(event.GetId())
+        if reader is None:
+            #print "use all formats"
+            readerlist = self.ImportSfactReaderlist
+            choices = "any file (*.*)|*.*"
+            extdict = {}
+            # compile a list of allowed extensions
+            for rd in readerlist:
+                fmt = rd.formatName
+                for extn in rd.extensionlist:
+                    if not extdict.get(extn): extdict[extn] = []
+                    extdict[extn] += [fmt,]
+            for extn in sorted(extdict.keys(),cmp=lambda x,y: cmp(x.lower(), y.lower())):
+                fmt = ''
+                for f in extdict[extn]:
+                    if fmt != "": fmt += ', '
+                    fmt += f
+                choices += "|" + fmt + " file (*" + extn + ")|*" + extn
+        else:
+            readerlist = [reader,]
+            # compile a list of allowed extensions
+            choices = reader.formatName + " file ("
+            w = ""
+            for extn in reader.extensionlist:
+                if w != "": w += ";"
+                w += "*" + extn
+            choices += w + ")|" + w
+            if not reader.strictExtension:
+                choices += "|any file (*.*)|*.*"
+        # get the file
+        dlg = wx.FileDialog(
+            self, message="Choose Structure Factor input file",
+            #defaultDir=os.getcwd(), 
+            defaultFile="",
+            wildcard=choices,
+            style=wx.OPEN | wx.CHANGE_DIR
+            )
+        try:
+            if dlg.ShowModal() == wx.ID_OK:
+                file = dlg.GetPath()
+            else: # cancel was pressed
+                return
+        finally:
+            dlg.Destroy()
+        # set what formats are compatible with this file
+        primaryReaders = []
+        secondaryReaders = []
+        for reader in readerlist:
+            flag = reader.ExtensionValidator(file)
+            if flag is None: 
+                secondaryReaders.append(reader)
+            elif flag:
+                primaryReaders.append(reader)
+        if len(secondaryReaders) + len(primaryReaders) == 0:
+            self.ErrorDialog('No Format','No matching format for file '+file)
+            return
+        
+        fp = None
+        try:
+            fp = open(file,'r')
+            # try the file first with Readers that specify the
+            # files extension and later with ones that allow it
+            for rd in primaryReaders+secondaryReaders:
+                if not rd.ContentsValidator(fp):
+                    continue # rejected on cursory check
+                #flag = rd.Reader(file,fp,self)
+                try:
+                    flag = rd.Reader(file,fp,self)
+                except:
+                    import traceback
+                    print traceback.format_exc()
+                    self.ErrorDialog('Read Error',
+                                     'Error reading file '+file
+                                     +' with format '+ rd.formatName)
+                    continue
+                if not flag: continue
+#                dlg = wx.TextEntryDialog( # allow editing of Structure Factor name
+#                    self, 'Enter the name for the new Structure Factor',
+#                    'Edit Structure Factor name', rd.Phase['General']['Name'],
+ #                   style=wx.OK)
+                dlg.CenterOnParent()
+#                if dlg.ShowModal() == wx.ID_OK:
+#                    rd.Phase['General']['Name'] = dlg.GetValue()
+#                dlg.Destroy()
+#                PhaseName = rd.Phase['General']['Name']
+#                if not G2gd.GetPatternTreeItemId(self,self.root,'Phases'):
+#                    sub = self.PatternTree.AppendItem(parent=self.root,text='Phases')
+#                else:
+#                    sub = G2gd.GetPatternTreeItemId(self,self.root,'Phases')
+#                psub = self.PatternTree.AppendItem(parent=sub,text=PhaseName)
+#                self.PatternTree.SetItemPyData(psub,rd.Phase)
+#                self.PatternTree.Expand(self.root) # make sure phases are seen
+#                self.PatternTree.Expand(sub) 
+#                self.PatternTree.Expand(psub) 
+                return # success
+        except:
+            import traceback
+            print traceback.format_exc()
+            self.ErrorDialog('Open Error',
+                             'Error on open of file '+file)
+        finally:
+            if fp: fp.close()
+
+        return
 
     def _init_coll_Import_Items(self,parent):
@@ -404 +565 @@
         self._init_coll_Calculate_Items(self.Calculate)
         self._init_Import_Phase(self.Import)
+        self._init_Import_Sfact(self.Import)
         self._init_coll_Import_Items(self.Import)
         self._init_coll_Export_Items(self.Export)
@@ -706 +868 @@
         finally:
             dlg.Destroy()
-        
+
+    # this will be removed eventually
     def OnSnglRead(self,event):
         self.CheckNotebook()
@@ -744 +907 @@
             
     def CheckNotebook(self):
+        '''Make sure the data tree has the minimally expected controls
+        (BHT) correct?
+        '''
         if not G2gd.GetPatternTreeItemId(self,self.root,'Notebook'):
             sub = self.PatternTree.AppendItem(parent=self.root,text='Notebook')