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: