def vesc(Pot,R):
"""
NAME:
vesc
PURPOSE:
calculate the escape velocity at R for potential Pot
INPUT:
Pot - Potential instances or list thereof
R - Galactocentric radius
OUTPUT:
escape velocity
HISTORY:
2011-10-09 - Written - Bovy (IAS)
"""
from planarPotential import evaluateplanarPotentials
from Potential import PotentialError
try:
return nu.sqrt(2.*(evaluateplanarPotentials(_INF,Pot)-evaluateplanarPotentials(R,Pot)))
except PotentialError:
from planarPotential import RZToplanarPotential
Pot= RZToplanarPotential(Pot)
return nu.sqrt(2.*(evaluateplanarPotentials(_INF,Pot)-evaluateplanarPotentials(R,Pot)))
def calcEscapecurve(Pot,Rs):
"""
NAME:
calcEscapecurve
PURPOSE:
calculate the escape velocity curve for this potential (in the
z=0 plane for non-spherical potentials)
INPUT:
Pot - Potential or list of Potential instances
Rs - (array of) radius(i)
OUTPUT:
array of v_esc
HISTORY:
2011-04-16 - Written - Bovy (NYU)
"""
isList= isinstance(Pot,list)
isNonAxi= ((isList and Pot[0].isNonAxi) or (not isList and Pot.isNonAxi))
if isNonAxi:
raise AttributeError("Escape velocity curve plotting for non-axisymmetric potentials is not currently supported")
try:
grid= len(Rs)
except TypeError:
grid=1
Rs= nu.array([Rs])
esccurve= nu.zeros(grid)
from planarPotential import evaluateplanarPotentials
for ii in range(grid):
try:
esccurve[ii]= nu.sqrt(2.*(evaluateplanarPotentials(_INF,Pot)-evaluateplanarPotentials(Rs[ii],Pot)))
except TypeError:
from planarPotential import RZToplanarPotential
Pot= RZToplanarPotential(Pot)
esccurve[ii]= nu.sqrt(2.*(evaluateplanarPotentials(_INF,Pot)-evaluateplanarPotentials(Rs[ii],Pot)))
return esccurve