source: trunk/testXNFF.py

Last change on this file was 4365, checked in by toby, 19 months ago

set keywords & eol on all G2 files

  • Property svn:eol-style set to native
  • Property svn:keywords set to Date Author Revision URL Id
File size: 4.5 KB
Line 
1# -*- coding: utf-8 -*-
2#testXNFF.py
3'''
4*testXNFF: Check x-ray & neutron form factor computation*
5=========================================
6
7Use this to check form factors used in neutron scattering
8
9'''
10
11import sys
12import wx
13import numpy as np
14import GSASIIpath
15GSASIIpath.SetBinaryPath()
16import GSASIItestplot as plot
17import GSASIIElem as G2el
18import GSASIIElemGUI as G2elG
19import atmdata
20WACV = wx.ALIGN_CENTER_VERTICAL
21
22try:
23    wx.NewIdRef
24    wx.NewId = wx.NewIdRef
25except AttributeError:
26    pass
27
28def create(parent):
29    return testXNFF(parent)
30   
31[wxID_FILEEXIT, wxID_PICKELEM,
32] = [wx.NewId() for _init_coll_File_Items in range(2)]
33
34class testXNFF(wx.Frame):
35    def _init_ctrls(self, parent):
36        wx.Frame.__init__(self, name='testXNFF', parent=parent,
37            size=wx.Size(460, 250),style=wx.DEFAULT_FRAME_STYLE, title='Test Xray & Neutron Form Factors')
38        self.testFFMenu = wx.MenuBar()
39        self.File = wx.Menu(title='')
40        self.File.Append(wxID_PICKELEM,'Pick element','Pick element')
41        self.File.Append(wxID_FILEEXIT,'Exit','Exit from testXNFF')
42        self.Bind(wx.EVT_MENU,self.OnPickElement,id=wxID_PICKELEM)
43        self.Bind(wx.EVT_MENU, self.OnFileExit, id=wxID_FILEEXIT)
44        self.testFFMenu.Append(menu=self.File, title='File')
45        self.SetMenuBar(self.testFFMenu)
46        self.testFFPanel = wx.ScrolledWindow(self)       
47        self.plotNB = plot.PlotNotebook()
48       
49    def __init__(self, parent):
50        self._init_ctrls(parent)
51        self.Bind(wx.EVT_CLOSE, self.ExitMain)
52
53    def ExitMain(self, event):
54        sys.exit()
55       
56    def OnFileExit(self,event):
57        self.Close()
58
59    def OnPickElement(self,Parms):
60
61        PE = G2elG.PickElement(self.testFFPanel,oneOnly=True)
62        if PE.ShowModal() == wx.ID_OK:
63            self.xrayFFs = G2el.GetFormFactorCoeff(PE.Elem)
64            self.Elem = PE.Elem
65            self.atmInfo = G2el.GetAtomInfo(PE.Elem)
66            self.magFFs = G2el.GetMagFormFacCoeff(PE.Elem)
67        PE.Destroy()
68        self.MakePlot()
69
70    def MakePlot(self):
71       
72        def test1():
73            fplot = self.plotNB.add('Neutron scattering lengths').gca()
74            lams = np.linspace(0.3,10.0,1000,True)
75           
76            BLtable = {}
77            El = self.Elem.capitalize()
78            for isotope in self.atmInfo['Isotopes']:
79                if 'Nat' in isotope:
80                    BLtable[isotope] = ['',atmdata.AtmBlens[El+'_']]
81                else:
82                    BLtable[isotope] = ['',atmdata.AtmBlens[El+'_'+isotope]]
83            for isotope in BLtable:
84                if 'BW-LS' in BLtable[isotope][1]:
85                    b0 = np.ones_like(lams)*BLtable[isotope][1]['BW-LS'][0]                   
86                else:
87                    b0 = np.ones_like(lams)*BLtable[isotope][1]['SL'][0]
88                bp,bpp = G2el.BlenResTOF([isotope,],BLtable,lams)
89                fplot.plot(lams,b0,label=isotope+El+' b0')
90                fplot.plot(lams,bp[0],label=isotope+El+" b'")
91                fplot.plot(lams,bpp[0],label=isotope+El+' b"')
92               
93            fplot.legend(loc='best')
94            fplot.set_xlabel('wavelength, A')
95            fplot.set_ylabel('b')
96
97        def test2():
98            fplot = self.plotNB.add('X-ray form factors').gca()
99            sq = np.linspace(0,2.,1000,True)
100            for El in self.xrayFFs:
101                F = G2el.ScatFac(El, sq**2)
102                fplot.plot(sq,F,label=El['Symbol'])
103            fplot.legend(loc='best')
104            fplot.set_xlabel('sin-theta/lambda')
105            fplot.set_ylabel('form factor')
106           
107        def test3():
108            fplot = self.plotNB.add('magnetic form factors').gca()
109            sq = np.linspace(0,1.,1000,True)
110            for El in self.magFFs:
111                El['gfac'] = 2.0
112                F = G2el.MagScatFac(El, sq**2)
113                fplot.plot(sq,F,label=El['Symbol'])
114            fplot.legend(loc='best')
115            fplot.set_xlabel('sin-theta/lambda')
116            fplot.set_ylabel('form factor')
117           
118        while self.plotNB.nb.GetPageCount():
119            self.plotNB.nb.DeletePage(0)
120        test1()
121        test2()
122        test3()
123       
124        self.plotNB.Show()
125       
126class testFFmain(wx.App):
127    def OnInit(self):
128        self.main = testXNFF(None)
129        self.main.Show()
130        self.SetTopWindow(self.main)
131        return True
132
133def main():
134    'Starts main application to compute and plot form factors'
135    application = testFFmain(0)
136    application.MainLoop()
137   
138if __name__ == '__main__':
139    GSASIIpath.InvokeDebugOpts()
140    main()
Note: See TracBrowser for help on using the repository browser.