CSG/tests/CSGIntersectComparisonTest.py

0001 #!/usr/bin/env python
0002
0003
0004 import logging
0005 log = logging.getLogger(__name__)
0006 import numpy as np
0007 from opticks.ana.fold import Fold
0008
0009 import matplotlib.pyplot as mp
0010 from opticks.sysrap.sframe import sframe , X, Y, Z
0011 from opticks.ana.pvplt import mpplt_simtrace_selection_line, mpplt_hist
0012
0013 if __name__ == '__main__':
0014     logging.basicConfig(level=logging.INFO)
0015     s = Fold.Load(symbol="s")
0016     print(repr(s))
0017
0018     ab = s.a_simtrace - s.b_simtrace
0019     print(ab.max())
0020
0021     a_simtrace = s.a_simtrace.reshape(-1,4,4)
0022     b_simtrace = s.b_simtrace.reshape(-1,4,4)
0023
0024     fr = sframe.FakeXZ(e=300)
0025
0026     fig, ax = fr.mp_subplots(mp)
0027
0028     if not s is None:
0029         a_hit = a_simtrace[:,0,3]>0
0030         a_pos = a_simtrace[a_hit][:,1,:3]
0031         fr.mp_scatter(a_pos, label="a_pos", s=1 )
0032
0033         b_hit = b_simtrace[:,0,3]>0
0034         b_pos = b_simtrace[b_hit][:,1,:3]
0035         fr.mp_scatter(b_pos, label="b_pos", s=1 )
0036
0037         #w = np.logical_and( np.abs(s.b_simtrace[:,1,0]) < 10. , np.abs(s.b_simtrace[:,1,2]) < 10. )
0038     pass
0039
0040     if not "NOLEGEND" in os.environ:
0041         ax.legend()
0042     pass
0043     fig.show()
0044 pass
0045