qudarap/tests/gentorch.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 from opticks.ana.p import *
0007
0008 if __name__ == '__main__':
0009
0010     t = Fold.Load()
0011     PIDX = int(os.environ.get("PIDX","-1"))
0012
0013     p= t.p
0014
0015     pos = p[:,0,:3]
0016     mom = p[:,1,:3]
0017     pol = p[:,2,:3]
0018
0019     pl = pvplt_plotter()
0020     pvplt_polarized(pl, pos, mom, pol, factor=20  )
0021     pl.show()
0022
0023
0024
0025