Changeset 2216

Timestamp:

Apr 15, 2016 2:57:57 PM (7 years ago)

Author:

vondreele

Message:

implement import layer from another gpx file

Location:

trunk

Files:

• GSASIIphsGUI.py (modified) (2 diffs)
• GSASIIplot.py (modified) (2 diffs)

trunk/GSASIIphsGUI.py

@@ -44 +44 @@
 import GSASIIIO as G2IO
 import GSASIIstrMain as G2stMn
+import GSASIIstrIO as G2strIO
 import GSASIImath as G2mth
 import GSASIIpwd as G2pwd
@@ -2587 +2588 @@
             
         def OnImportLayer(event):
-            print 'Import atoms for a layer - TBD'
-            #from where? DIFFaX files? other phases? NB: transformation issues
-            event.Skip()
+            dlg = wx.FileDialog(G2frame, 'Choose GSAS-II project file', 
+                wildcard='GSAS-II project file (*.gpx)|*.gpx',style=wx.OPEN| wx.CHANGE_DIR)
+            try:
+                if dlg.ShowModal() == wx.ID_OK:
+                    GPXFile = dlg.GetPath()
+                    phaseNames = G2strIO.GetPhaseNames(GPXFile)
+            finally:
+                dlg.Destroy()
+            dlg = wx.SingleChoiceDialog(G2frame,'Phase to use for layer','Select',phaseNames)
+            if dlg.ShowModal() == wx.ID_OK:
+                sel = dlg.GetSelection()
+                PhaseName = phaseNames[sel]
+            else:
+                return
+            Phase = G2strIO.GetAllPhaseData(GPXFile,PhaseName)
+            #need cell compatibility check here
+            Layer = {'Name':Phase['General']['Name'],'SameAs':'','Symm':'None'}
+            cx,ct,cs,cia = Phase['General']['AtomPtrs']
+            atoms = Phase['Atoms']
+            Atoms = []
+            for atom in atoms:
+                x,y,z,f = atom[cx:cx+4]
+                u = atom[cia+1]
+                if not u: u = 0.01
+                Atoms.append([atom[ct-1],atom[ct],x,y,z,f,u])
+                if atom[ct] not in data['Layers']['AtInfo']:
+                    data['Layers']['AtInfo'][atom[ct]] = G2elem.GetAtomInfo(atom[ct])
+            Layer['Atoms'] = Atoms
+            data['Layers']['Layers'].append(Layer)
+            Trans = data['Layers']['Transitions']
+            if len(Trans):
+                Trans.append([[0.,0.,0.,0.,'',False] for trans in Trans])
+                for trans in Trans:
+                    trans.append([0.,0.,0.,0.,'',False])
+            else:
+                Trans = [[[1.,0.,0.,0.,'',False],],]
+            data['Layers']['Transitions'] = Trans
+            UpdateLayerData()
             
         def LayerSizer(il,Layer):

trunk/GSASIIplot.py

@@ -67 +67 @@
 Gkrho = unichr(0x03C1)
 nxs = np.newaxis
+#    GSASIIpath.IPyBreak()
     
 class _tabPlotWin(wx.Panel):    
@@ -6046 +6047 @@
             newXYZ = np.concatenate((newXYZ,XYZ))
         XYZ = newXYZ
-        na = int(8./cell[0])
-        nb = int(8./cell[1])
-        if any([na,nb]):
-            nunit = [na,nb,0]
-            indA = range(-na,na)
-            indB = range(-nb,nb)
-            Units = np.array([[h,k,0] for h in indA for k in indB])
-            newXYZ = np.zeros((0,3))
-            for unit in Units:
-                newXYZ = np.concatenate((newXYZ,unit+XYZ))
-            if len(Units):
-                AtNames *= len(Units)
-                AtTypes *= len(Units)
-            XYZ = newXYZ
+        na = max(int(8./cell[0]),1)
+        nb = max(int(8./cell[1]),1)
+        nunit = [na,nb,0]
+        indA = range(-na,na)
+        indB = range(-nb,nb)
+        Units = np.array([[h,k,0] for h in indA for k in indB])
+        newXYZ = np.zeros((0,3))
+        for unit in Units:
+            newXYZ = np.concatenate((newXYZ,unit+XYZ))
+        if len(Units):
+            AtNames *= len(Units)
+            AtTypes *= len(Units)
+        XYZ = newXYZ
+#        GSASIIpath.IPyBreak()
         
         Bonds = FindBonds(AtTypes,XYZ)