def test_rectgal_to_sphergal():
#Test that this is the inverse of sphergal_to_rectgal
l,b,d= 90.,30.,1.
vr,pmll,pmbb= 10.,-20.,30.
X,Y,Z,vx,vy,vz= bovy_coords.sphergal_to_rectgal(l,b,d,vr,pmll,pmbb,
degree=True)
lt,bt,dt,vrt,pmllt,pmbbt= bovy_coords.rectgal_to_sphergal(X,Y,Z,
vx,vy,vz,
degree=True)
assert numpy.fabs(lt-l) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(bt-b) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(dt-d) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(vrt-vr) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(pmllt-pmll) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(pmbbt-pmbb) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
#Also test for degree=False
lt,bt,dt,vrt,pmllt,pmbbt= bovy_coords.rectgal_to_sphergal(X,Y,Z,
vx,vy,vz,
degree=False)
assert numpy.fabs(lt-l/180.*numpy.pi) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(bt-b/180.*numpy.pi) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(dt-d) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(vrt-vr) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(pmllt-pmll) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.fabs(pmbbt-pmbb) < 10.**-10., 'rectgal_to_sphergal conversion did not work as expected'
#Also test for arrays
os= numpy.ones(2)
lbdvrpmllpmbbt= bovy_coords.rectgal_to_sphergal(os*X,os*Y,os*Z,
os*vx,os*vy,
os*vz,
degree=True)
lt= lbdvrpmllpmbbt[:,0]
bt= lbdvrpmllpmbbt[:,1]
dt= lbdvrpmllpmbbt[:,2]
vrt= lbdvrpmllpmbbt[:,3]
pmllt= lbdvrpmllpmbbt[:,4]
pmbbt= lbdvrpmllpmbbt[:,5]
assert numpy.all(numpy.fabs(lt-l) < 10.**-10.), 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.all(numpy.fabs(bt-b) < 10.**-10.), 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.all(numpy.fabs(dt-d) < 10.**-10.), 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.all(numpy.fabs(vrt-vr) < 10.**-10.), 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.all(numpy.fabs(pmllt-pmll) < 10.**-10.), 'rectgal_to_sphergal conversion did not work as expected'
assert numpy.all(numpy.fabs(pmbbt-pmbb) < 10.**-10.), 'rectgal_to_sphergal conversion did not work as expected'
return None
def _fit_orbit_mlogl(new_vxvv, vxvv, vxvv_err, pot, radec, lb, customsky,
lb_to_customsky, pmllpmbb_to_customsky, tmockAA, ro, vo,
obs):
"""The log likelihood for fitting an orbit"""
#Use this _parse_args routine, which does forward and backward integration
iR, ivR, ivT, iz, ivz, iphi = tmockAA._parse_args(True, False, new_vxvv[0],
new_vxvv[1], new_vxvv[2],
new_vxvv[3], new_vxvv[4],
new_vxvv[5])
if radec or lb or customsky:
#Need to transform to (l,b), (ra,dec), or a custom set
#First transform to X,Y,Z,vX,vY,vZ (Galactic)
X, Y, Z = coords.galcencyl_to_XYZ(iR.flatten(),
iphi.flatten(),
iz.flatten(),
Xsun=obs[0] / ro,
Zsun=obs[2] / ro).T
vX,vY,vZ = coords.galcencyl_to_vxvyvz(ivR.flatten(),ivT.flatten(),
ivz.flatten(),iphi.flatten(),
vsun=nu.array(\
obs[3:6])/vo,Xsun=obs[0]/ro,Zsun=obs[2]/ro).T
bad_indx = (X == 0.) * (Y == 0.) * (Z == 0.)
if True in bad_indx: X[bad_indx] += ro / 10000.
lbdvrpmllpmbb = coords.rectgal_to_sphergal(X * ro,
Y * ro,
Z * ro,
vX * vo,
vY * vo,
vZ * vo,
degree=True)
if lb:
orb_vxvv = nu.array([
lbdvrpmllpmbb[:, 0], lbdvrpmllpmbb[:, 1], lbdvrpmllpmbb[:, 2],
lbdvrpmllpmbb[:, 4], lbdvrpmllpmbb[:, 5], lbdvrpmllpmbb[:, 3]
]).T
elif radec:
#Further transform to ra,dec,pmra,pmdec
radec = coords.lb_to_radec(lbdvrpmllpmbb[:, 0],
lbdvrpmllpmbb[:, 1],
degree=True,
epoch=None)
pmrapmdec = coords.pmllpmbb_to_pmrapmdec(lbdvrpmllpmbb[:, 4],
lbdvrpmllpmbb[:, 5],
lbdvrpmllpmbb[:, 0],
lbdvrpmllpmbb[:, 1],
degree=True,
epoch=None)
orb_vxvv = nu.array([
radec[:, 0], radec[:, 1], lbdvrpmllpmbb[:, 2], pmrapmdec[:, 0],
pmrapmdec[:, 1], lbdvrpmllpmbb[:, 3]
]).T
elif customsky:
#Further transform to ra,dec,pmra,pmdec
customradec = lb_to_customsky(lbdvrpmllpmbb[:, 0],
lbdvrpmllpmbb[:, 1],
degree=True)
custompmrapmdec = pmllpmbb_to_customsky(lbdvrpmllpmbb[:, 4],
lbdvrpmllpmbb[:, 5],
lbdvrpmllpmbb[:, 0],
lbdvrpmllpmbb[:, 1],
degree=True)
orb_vxvv = nu.array([
customradec[:, 0], customradec[:, 1], lbdvrpmllpmbb[:, 2],
custompmrapmdec[:, 0], custompmrapmdec[:, 1], lbdvrpmllpmbb[:,
3]
]).T
else:
#shape=(2tintJ-1,6)
orb_vxvv = nu.array([
iR.flatten(),
ivR.flatten(),
ivT.flatten(),
iz.flatten(),
ivz.flatten(),
iphi.flatten()
]).T
out = 0.
for ii in range(vxvv.shape[0]):
sub_vxvv = (orb_vxvv - vxvv[ii, :].flatten())**2.
#print(sub_vxvv[nu.argmin(nu.sum(sub_vxvv,axis=1))])
if not vxvv_err is None:
sub_vxvv /= vxvv_err[ii, :]**2.
else:
sub_vxvv /= 0.01**2.
out += logsumexp(-0.5 * nu.sum(sub_vxvv, axis=1))
return -out
def _fit_orbit_mlogl(new_vxvv,vxvv,vxvv_err,pot,radec,lb,tmockAA,
ro,vo,obs):
"""The log likelihood for fitting an orbit"""
#Use this _parse_args routine, which does forward and backward integration
iR,ivR,ivT,iz,ivz,iphi= tmockAA._parse_args(True,False,
new_vxvv[0],
new_vxvv[1],
new_vxvv[2],
new_vxvv[3],
new_vxvv[4],
new_vxvv[5])
if radec or lb:
#Need to transform to ra,dec
#First transform to X,Y,Z,vX,vY,vZ (Galactic)
X,Y,Z = coords.galcencyl_to_XYZ(iR.flatten(),iphi.flatten(),
iz.flatten(),
Xsun=obs[0]/ro,
Ysun=obs[1]/ro,
Zsun=obs[2]/ro)
vX,vY,vZ = coords.galcencyl_to_vxvyvz(ivR.flatten(),ivT.flatten(),
ivz.flatten(),iphi.flatten(),
vsun=nu.array(\
obs[3:6])/vo)
bad_indx= (X == 0.)*(Y == 0.)*(Z == 0.)
if True in bad_indx: X[bad_indx]+= ro/10000.
lbdvrpmllpmbb= coords.rectgal_to_sphergal(X*ro,Y*ro,Z*ro,
vX*vo,vY*vo,vZ*vo,
degree=True)
if lb:
orb_vxvv= nu.array([lbdvrpmllpmbb[:,0],
lbdvrpmllpmbb[:,1],
lbdvrpmllpmbb[:,2],
lbdvrpmllpmbb[:,4],
lbdvrpmllpmbb[:,5],
lbdvrpmllpmbb[:,3]]).T
else:
#Further transform to ra,dec,pmra,pmdec
radec= coords.lb_to_radec(lbdvrpmllpmbb[:,0],
lbdvrpmllpmbb[:,1],degree=True)
pmrapmdec= coords.pmllpmbb_to_pmrapmdec(lbdvrpmllpmbb[:,4],
lbdvrpmllpmbb[:,5],
lbdvrpmllpmbb[:,0],
lbdvrpmllpmbb[:,1],
degree=True)
orb_vxvv= nu.array([radec[:,0],radec[:,1],
lbdvrpmllpmbb[:,2],
pmrapmdec[:,0],pmrapmdec[:,1],
lbdvrpmllpmbb[:,3]]).T
else:
#shape=(2tintJ-1,6)
orb_vxvv= nu.array([iR.flatten(),ivR.flatten(),ivT.flatten(),
iz.flatten(),ivz.flatten(),iphi.flatten()]).T
out= 0.
for ii in range(vxvv.shape[0]):
sub_vxvv= (orb_vxvv-vxvv[ii,:].flatten())**2.
#print sub_vxvv[nu.argmin(nu.sum(sub_vxvv,axis=1))]
if not vxvv_err is None:
sub_vxvv/= vxvv_err[ii,:]**2.
else:
sub_vxvv/= 0.01**2.
out+= logsumexp(-0.5*nu.sum(sub_vxvv,axis=1))
return -out
