def axi_density2(R, z=0, phi=0.): spherical_coords = coords.cyl_to_spher(R, z, phi) theta = spherical_coords[1] return rho_Zeeuw(R, z, phi) * (1 + numpy.cos(theta) + numpy.cos(theta)**2)
def density1(R, z=0, phi=0.): r, theta, phi = coords.cyl_to_spher(R, z, phi) h = potential.HernquistPotential(2) return h.dens(R, z, phi) * (1 + numpy.cos(theta) + numpy.cos(theta)** 2.) * (1 + numpy.cos(phi) + numpy.sin(phi))
def rho_Zeeuw(R, z, phi, a=1.): r, theta, phi = coords.cyl_to_spher(R, z, phi) return 3. / (4 * numpy.pi) * numpy.power((a + r), -4.) * a