qudarap/tests/scint_generate.py

0001 #!/usr/bin/env python
0002
0003 import os, numpy as np
0004 MODE = int(os.environ.get("MODE","0"))
0005
0006 PIDX = int(os.environ.get("PIDX","-1"))
0007
0008 from opticks.ana.fold import Fold
0009 from opticks.ana.p import *
0010
0011 if MODE in [2,3]:
0012     from opticks.ana.pvplt import *
0013 pass
0014
0015 if __name__ == '__main__':
0016     t = Fold.Load(symbol="t",globals=True)
0017     print(repr(t))
0018
0019     lim = slice(0, 1000)
0020
0021     pos = t.p[lim,0,:3]
0022     mom = t.p[lim,1,:3]
0023     pol = t.p[lim,2,:3]
0024
0025     print("t.p.shape\n", t.p.shape)
0026     print("pos.shape\n", pos.shape)
0027     print("mom.shape\n", mom.shape)
0028     print("pol.shape\n", pol.shape)
0029
0030     if MODE == 3:
0031         pl = pvplt_plotter()
0032         pvplt_polarized(pl, pos, mom, pol, factor=20  )
0033         pl.show()
0034     pass
0035
0036
0037
0038