eic-opticks/CSGOptiX/cxs_min_AB.py

0001 #!/usr/bin/env python
0002 """
0003
0004 ~/o/cxs_min.sh AB
0005
0006 """
0007
0008 import os, logging, textwrap, numpy as np
0009 log = logging.getLogger(__name__)
0010
0011 print("[from opticks.sysrap.sevt import SEvt, SAB")
0012 from opticks.sysrap.sevt import SEvt, SAB
0013 print("]from opticks.sysrap.sevt import SEvt, SAB")
0014
0015
0016 TEST = os.environ.get("TEST","")
0017
0018
0019 if __name__ == '__main__':
0020     logging.basicConfig(level=logging.INFO)
0021
0022     a = SEvt.Load("$AFOLD", symbol="a")
0023     print(repr(a))
0024
0025     if "BFOLD" in os.environ:
0026         b = SEvt.Load("$BFOLD", symbol="b")
0027         print(repr(b))
0028         ab = SAB(a,b)
0029         print(repr(ab))
0030     pass
0031
0032
0033     EXPR = filter(None,textwrap.dedent(r"""
0034     np.all( a.f.genstep == b.f.genstep )
0035     np.all( a.f.hit == b.f.hit )
0036     a.f.hit.shape
0037     b.f.hit.shape
0038     """).split("\n"))
0039
0040     for expr in EXPR:
0041         val = str(eval(expr)) if not expr.startswith("#") else ""
0042         fmt = "%-80s \n%s\n" if len(val.split("\n")) > 1 else "%-80s : %s"
0043         print(fmt % (expr, val))
0044     pass
0045
0046
0047
0048     e = a
0049     qtab = e.minimal_qtab()
0050     print("qtab")
0051     print(qtab)
0052
0053