def rho_gen(self):
"""
A wrapper for calc_rho_zr.
Upon executing, generates rho and rho cdf inverse
"""
# Check that sigma has been generated
if not hasattr(self._parent, 'sigma'):
raise RuntimeError, 'Must load/generate sigma before calculating rho'
# Numerically calculate rho(z,r) for a given sigma. rho(z,r)
# obeys vertical hydrostatic equilibrium (approximately)
rho_array, z, r = calc_rho_zr.rho_zr(self._parent)
# Create a complete rho object. Includes rho spline and CDF inverse
rho = calc_rho_zr.rho_from_array(self._parent, rho_array, z, r)
# Save to ICobj
self._parent.rho = rho
print 'rho stored in <IC instance>.rho'
def rho_gen(self):
"""
A wrapper for calc_rho_zr.
Upon executing, generates rho and rho cdf inverse
"""
# Check that sigma has been generated
if not hasattr(self._parent, 'sigma'):
raise RuntimeError,'Must load/generate sigma before calculating rho'
# Numerically calculate rho(z,r) for a given sigma. rho(z,r)
# obeys vertical hydrostatic equilibrium (approximately)
rho_array, z, r = calc_rho_zr.rho_zr(self._parent)
# Create a complete rho object. Includes rho spline and CDF inverse
rho = calc_rho_zr.rho_from_array(self._parent, rho_array, z, r)
# Save to ICobj
self._parent.rho = rho
print 'rho stored in <IC instance>.rho'
import calc_sigma
ICgenDir = os.path.dirname(os.path.realpath(__file__))
# File containing all the settings definitions
settings_file = 'ICgen_settings.py'
execfile(settings_file)
""" **************************************************
EXECUTION
************************************************** """
if initial_step < 2:
print '***********************************************'
print 'STEP 1 OF 3: Generating PDF(r,z) [rho(z,r)]'
print '***********************************************'
rho = calc_rho_zr.rho_zr(sigmaFileName, nr=nr, nz=nz, zmax=zmax, \
m=m, M=M, rho_tol=rho_tol, output=rhoFileName, rmin=rmin, rmax=rmax, \
T0=T0, r0=r0, Tpower=Tpower)
else:
rho = pickle.load(open(rhoFileName, 'rb'))
if initial_step < 3:
print '***********************************************'
print 'STEP 2 OF 3: Generating random positions'
print '***********************************************'
pos = pos_gen.make(rhoFileName, sigmaFileName, nParticles, zlim=zlim, \
rlim=rlim, savename=posFileName)
else:
pos = pickle.load(open(posFileName, 'rb'))
if initial_step < 4:
print '***********************************************'
print 'STEP 3 OF 3: Generating tipsy snapshot'
print '***********************************************'
""" **************************************************
EXECUTION
************************************************** """
if initial_step < 2:
print "***********************************************"
print "STEP 1 OF 3: Generating PDF(r,z) [rho(z,r)]"
print "***********************************************"
rho = calc_rho_zr.rho_zr(
sigmaFileName,
nr=nr,
nz=nz,
zmax=zmax,
m=m,
M=M,
rho_tol=rho_tol,
output=rhoFileName,
rmin=rmin,
rmax=rmax,
T0=T0,
r0=r0,
Tpower=Tpower,
)
else:
rho = pickle.load(open(rhoFileName, "rb"))
if initial_step < 3:
print "***********************************************"
print "STEP 2 OF 3: Generating random positions"
print "***********************************************"
pos = pos_gen.make(rhoFileName, sigmaFileName, nParticles, zlim=zlim, rlim=rlim, savename=posFileName)
else:
