Go to the source code of this file.
Namespaces | |
make_fwd_distortions | |
Functions | |
def | make_fwd_distortions.get_corr_points (hx, hy, hz) |
def | make_fwd_distortions.get_bin_points (h) |
def | make_fwd_distortions.set_distortions (hx, hy, hz, pts) |
def | make_fwd_distortions.get_distorted_pts (grid_pts, tri_in) |
Variables | |
make_fwd_distortions.parser = ap() | |
make_fwd_distortions.type | |
make_fwd_distortions.str | |
make_fwd_distortions.help | |
make_fwd_distortions.args = parser.parse_args() | |
make_fwd_distortions.fIn = RT.TFile(args.i, "OPEN") | |
make_fwd_distortions.bkwd_z_pos = fIn.Get("RecoBkwd_Displacement_Z_Pos") | |
make_fwd_distortions.bkwd_z_neg = fIn.Get("RecoBkwd_Displacement_Z_Neg") | |
make_fwd_distortions.bkwd_x_pos = fIn.Get("RecoBkwd_Displacement_X_Pos") | |
make_fwd_distortions.bkwd_x_neg = fIn.Get("RecoBkwd_Displacement_X_Neg") | |
make_fwd_distortions.bkwd_y_pos = fIn.Get("RecoBkwd_Displacement_Y_Pos") | |
make_fwd_distortions.bkwd_y_neg = fIn.Get("RecoBkwd_Displacement_Y_Neg") | |
make_fwd_distortions.fwd_z_pos = bkwd_z_pos.Clone("RecoFwd_Displacement_Z_Pos") | |
make_fwd_distortions.fwd_z_neg = bkwd_z_neg.Clone("RecoFwd_Displacement_Z_Neg") | |
make_fwd_distortions.fwd_x_pos = bkwd_x_pos.Clone("RecoFwd_Displacement_X_Pos") | |
make_fwd_distortions.fwd_x_neg = bkwd_x_neg.Clone("RecoFwd_Displacement_X_Neg") | |
make_fwd_distortions.fwd_y_pos = bkwd_y_pos.Clone("RecoFwd_Displacement_Y_Pos") | |
make_fwd_distortions.fwd_y_neg = bkwd_y_neg.Clone("RecoFwd_Displacement_Y_Neg") | |
make_fwd_distortions.bkwd_pos_points = get_bin_points(bkwd_z_pos) | |
make_fwd_distortions.bkwd_neg_points = get_bin_points(bkwd_z_neg) | |
make_fwd_distortions.bkwd_pos_corrected = get_corr_points(bkwd_x_pos, bkwd_y_pos, bkwd_z_pos) | |
make_fwd_distortions.bkwd_neg_corrected = get_corr_points(bkwd_x_neg, bkwd_y_neg, bkwd_z_neg) | |
make_fwd_distortions.bkwd_pos_corrected_array = np.array(bkwd_pos_corrected) | |
make_fwd_distortions.bkwd_neg_corrected_array = np.array(bkwd_neg_corrected) | |
make_fwd_distortions.tri_pos_corr = Delaunay(bkwd_pos_corrected_array) | |
make_fwd_distortions.tri_neg_corr = Delaunay(bkwd_neg_corrected_array) | |
make_fwd_distortions.distorted_pos = get_distorted_pts(bkwd_pos_points, tri_pos_corr) | |
make_fwd_distortions.distorted_neg = get_distorted_pts(bkwd_neg_points, tri_neg_corr) | |
make_fwd_distortions.fOut = RT.TFile(args.o, "RECREATE") | |
make_fwd_distortions.ex_pos = fIn.Get("Reco_ElecField_X_Pos") | |
make_fwd_distortions.ey_pos = fIn.Get("Reco_ElecField_Y_Pos") | |
make_fwd_distortions.ez_pos = fIn.Get("Reco_ElecField_Z_Pos") | |
make_fwd_distortions.ex_neg = fIn.Get("Reco_ElecField_X_Neg") | |
make_fwd_distortions.ey_neg = fIn.Get("Reco_ElecField_Y_Neg") | |
make_fwd_distortions.ez_neg = fIn.Get("Reco_ElecField_Z_Neg") | |