diff --git a/PyART/analytic/coordschange.py b/PyART/analytic/coordschange.py index deafbbb..6ea9b3a 100644 --- a/PyART/analytic/coordschange.py +++ b/PyART/analytic/coordschange.py @@ -271,6 +271,10 @@ def eob_ID_to_ADM(eob_Wave, verbose=False, PN_order=2, rotate_on_x_axis=True): 'x2' : x coordinate of puncture 2 'D' : coordinate separation between punctures 'x_offset' : offset to be added to x coordinates + 'qe' : EOB coordinates (x,y) + 'pe' : EOB momenta (px, py), mu-normalized + 'qe_chk' : EOB coordinates (x,y), EOB->ADM->EOB + 'pe_chk' : EOB momenta (px, py), mu-normalized, EOB->ADM->EOB """ # Get info from EOB dynamics q = eob_Wave.pars["q"] @@ -308,6 +312,8 @@ def eob_ID_to_ADM(eob_Wave, verbose=False, PN_order=2, rotate_on_x_axis=True): x2 = -d_ADM * q / (q + 1) x_offset = -b_par + d_ADM / (q + 1) + qe_check, pe_check = Adm2Eob(qa, pa, nu, PN_order=PN_order) + # wrap output out = { "q_cart": qa, @@ -318,12 +324,13 @@ def eob_ID_to_ADM(eob_Wave, verbose=False, PN_order=2, rotate_on_x_axis=True): "x2": x2, "D": d_ADM, "x_offset": x_offset, + "qe": qe, + "pe": pe, + "qe_chk": qe_check, # EOB->ADM->EOB + "pe_chk": pe_check, # EOB->ADM->EOB } if verbose: - # for testing - qe_check, pe_check = Adm2Eob(qa, pa, nu, PN_order=PN_order) - dashes = "-" * 50 print("{}\nPunctures\n{}".format(dashes, dashes)) print("b_par : {:.15f}".format(b_par))