sysrap/tests/SGenerate_test.py

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