make_fwd_distortions.py File Reference

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")