qudarap/tests/propagate_at_surface.py

0001 #!/usr/bin/env python
0002
0003 import os, numpy as np
0004 from opticks.ana.fold import Fold
0005
0006 def eprint( expr, lprefix="", rprefix="", tail="" ):
0007     ret = eval(expr)
0008     lhs = "%s%s" % (lprefix, expr)
0009     rhs = "%s%s" % (rprefix, ret )
0010     print("%-50s : %s%s" % ( lhs, rhs, tail )   )
0011     return ret
0012
0013 def epr(arg, **kwa):
0014     p = arg.find("=")
0015     if p > -1:
0016         var_eq = arg[:p+1]
0017         expr = arg[p+1:]
0018         label = var_eq
0019     else:
0020         label, expr = "", arg
0021     pass
0022     return eprint(expr, lprefix=label,  **kwa)
0023
0024
0025 if __name__ == '__main__':
0026     FOLD = os.environ["FOLD"]
0027     t = Fold.Load(FOLD)
0028
0029     if hasattr(t,'p0'):
0030         p0 = t.p0
0031         epr("p0", tail="\n\n", rprefix="\n"  )
0032     else:
0033         p0 = None
0034     pass
0035
0036     if hasattr(t,'prd'):
0037         prd = t.prd
0038         epr("prd", tail="\n\n", rprefix="\n" )
0039     else:
0040         prd = None
0041     pass
0042
0043     if hasattr(t,'s'):
0044         s = t.s
0045         epr("s", tail="\n\n", rprefix="\n" )
0046     else:
0047         s = None
0048     pass
0049
0050
0051     if hasattr(t,'p'):
0052         p = t.p
0053         epr("p", tail="\n\n", rprefix="\n" )
0054     else:
0055         p = None
0056     pass
0057
0058
0059     epr("FOLD")
0060     assert not p is None
0061     flag = epr("flag=p[:,3,3].view(np.uint32)")
0062     uflag = epr("uflag=np.unique(flag, return_counts=True)")
0063     flat       = epr("flat=p[:,0,3]")
0064     TransCoeff = epr("TransCoeff=p[:,1,3]")
0065
0066