sysrap/tests/scarrier_test.py

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