CSGOptiX/tests/CSGOptiXSimulateTest.py

0001 #!/usr/bin/env python
0002 import sys, numpy as np
0003 from opticks.ana.fold import Fold
0004 from opticks.ana.p import *
0005 PIDX = int(os.environ.get("PIDX","-1"))
0006
0007 if __name__ == '__main__':
0008     t = Fold.Load()
0009     r = t.record if hasattr(t,'record') else None
0010     p = t.photon if hasattr(t,'photon') else None
0011
0012     if p is None:
0013        print("ERROR p is None")
0014        sys.exit(0)
0015     pass
0016
0017     s = str(p[:,:3])
0018     a = np.array( s.split("\n") + [""] ).reshape(-1,4)
0019
0020
0021     for i in range(len(a)):
0022         if not (PIDX == -1 or PIDX == i): continue
0023         if PIDX > -1: print("PIDX %d " % PIDX)
0024
0025         if not r is None:
0026             print("r[i,:,:3]")
0027             print(r[i,:,:3])
0028             print("\n\nbflagdesc_(r[i,j])")
0029             for j in range(len(r[i])):
0030                 print(bflagdesc_(r[i,j])  )
0031             pass
0032         pass
0033
0034         print("\n")
0035         print("p")
0036         print("\n".join(a[i]))
0037         print(bflagdesc_(p[i]))
0038         print("\n")
0039
0040
0041
0042