qudarap/tests/propagate_at_boundary.py

0001 #!/usr/bin/env python
0002
0003 import os, numpy as np
0004 from opticks.ana.fold import Fold
0005 from opticks.ana.eprint import eprint, epr
0006
0007 if __name__ == '__main__':
0008     FOLD = os.environ["FOLD"]
0009     t = Fold.Load(FOLD)
0010
0011     g = globals()
0012     l = locals()
0013
0014     epr("FOLD",          g, l )
0015     p0  = epr("t.p0",    g, l,  tail="\n\n", rprefix="\n"  )
0016     prd = epr("t.prd",   g, l,  tail="\n\n", rprefix="\n"  )
0017     s   = epr("t.s",     g, l,  tail="\n\n", rprefix="\n"  )
0018     p   = epr("t.p",     g, l,  tail="\n\n", rprefix="\n"  )
0019
0020     epr("FOLD", g, l )
0021
0022     assert not p is None
0023     flag       = epr("flag=p[:,3,3].view(np.uint32)",             g, l )
0024     uflag      = epr("uflag=np.unique(flag, return_counts=True)", g, l )
0025     flat       = epr("flat=p[:,0,3]",                             g, l, tail="HMM: sphoton.time not flat" )
0026     TransCoeff = epr("TransCoeff=p[:,1,3]",                       g, l, tail="HMM: sphoton.weight not TransCoeff" )
0027
0028
0029
0030