def pot(p) :
"""Calculates the potential in the midplane - can be expensive"""
#from . import gravity
import pynbody.gravity.calc as gravity
if pynbody.config['verbose'] :
print 'Profile: pot() -- warning, disk must be in the x-y plane'
grav_sim = p.sim
# Go up to the halo level
while hasattr(grav_sim,'base') and grav_sim.base.properties.has_key("halo_id") :
grav_sim = grav_sim.base
if pynbody.config['tracktime']:
import time
start = time.clock()
pot = gravity.midplane_potential(grav_sim, p['rbins']).in_units(p.sim['vel'].units**2)
end = time.clock()
if pynbody.config['verbose']: print 'Potential calculated in %5.3g s'%(end-start)
return pot
else:
return gravity.midplane_potential(grav_sim, p['rbins']).in_units(p.sim['vel'].units**2)
def pot(p):
"""Calculates the potential in the midplane - can be expensive"""
#from . import gravity
import pynbody.gravity.calc as gravity
logger.warn(
"Profile pot -- this routine assumes the disk is in the x-y plane")
grav_sim = p.sim
# Go up to the halo level
while hasattr(grav_sim, 'base') and grav_sim.base.properties.has_key("halo_id"):
grav_sim = grav_sim.base
start = time.clock()
pot = gravity.midplane_potential(
grav_sim, p['rbins']).in_units(p.sim['vel'].units ** 2)
end = time.clock()
logger.info("Potential calculated in %5.3g s" % (end - start))
return pot
def pot(p):
"""Calculates the potential in the midplane - can be expensive"""
#from . import gravity
import pynbody.gravity.calc as gravity
logger.warn(
"Profile pot -- this routine assumes the disk is in the x-y plane")
grav_sim = p.sim
# Go up to the halo level
while hasattr(grav_sim, 'base') and "halo_id" in grav_sim.base.properties:
grav_sim = grav_sim.base
start = backcompat.clock()
pot = gravity.midplane_potential(grav_sim, p['rbins']).in_units(
p.sim['vel'].units**2)
end = backcompat.clock()
logger.info("Potential calculated in %5.3g s" % (end - start))
return pot
