qudarap/tests/cerenkov_generate.py

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