odisseo package

Submodules

• odisseo.dynamics module
  • direct_acc()
  • direct_acc_for_loop()
  • direct_acc_laxmap()
  • direct_acc_matrix()
  • direct_acc_sharding()
  • no_self_gravity()
  • single_body_acc()
• odisseo.initial_condition module
  • Plummer_sphere()
  • Plummer_sphere_reparam()
  • ic_two_body()
  • inclined_circular_velocity()
  • inclined_position()
  • sample_position_on_circle()
  • sample_position_on_sphere()
• odisseo.integrators module
  • RungeKutta4()
  • diffrax_solver()
  • leapfrog()
• odisseo.option_classes module
  • LogarithmicParams
    • LogarithmicParams.q
    • LogarithmicParams.v0
  • MNParams
    • MNParams.M
    • MNParams.a
    • MNParams.b
  • NFWParams
    • NFWParams.Mvir
    • NFWParams.r_s
  • PSPParams
    • PSPParams.M
    • PSPParams.alpha
    • PSPParams.r_c
  • PlummerParams
    • PlummerParams.Mtot
    • PlummerParams.a
  • PointMassParams
    • PointMassParams.M
  • SimulationConfig
    • SimulationConfig.MN3_positive_density
    • SimulationConfig.N_particles
    • SimulationConfig.acceleration_scheme
    • SimulationConfig.batch_size
    • SimulationConfig.differentation_mode
    • SimulationConfig.diffrax_adjoint_method
    • SimulationConfig.diffrax_solver
    • SimulationConfig.dimensions
    • SimulationConfig.double_map
    • SimulationConfig.external_accelerations
    • SimulationConfig.fixed_timestep
    • SimulationConfig.glorder
    • SimulationConfig.gradient_horizon
    • SimulationConfig.integrator
    • SimulationConfig.num_checkpoints
    • SimulationConfig.num_snapshots
    • SimulationConfig.num_timesteps
    • SimulationConfig.progress_bar
    • SimulationConfig.return_snapshots
    • SimulationConfig.sech2_z
    • SimulationConfig.softening
  • SimulationParams
    • SimulationParams.G
    • SimulationParams.Logarithmic_params
    • SimulationParams.MN_params
    • SimulationParams.NFW_params
    • SimulationParams.PSP_params
    • SimulationParams.Plummer_params
    • SimulationParams.PointMass_params
    • SimulationParams.ThickMN3Disk_params
    • SimulationParams.ThinMN3Disk_params
    • SimulationParams.TriaxialNFW_params
    • SimulationParams.TwoPowerTriaxial_params
    • SimulationParams.t_end
  • ThickMN3DiskParams
    • ThickMN3DiskParams.M
    • ThickMN3DiskParams.hr
    • ThickMN3DiskParams.hz
  • ThinMN3DiskParams
    • ThinMN3DiskParams.M
    • ThinMN3DiskParams.hr
    • ThinMN3DiskParams.hz
  • TriaxialNFWParams
    • TriaxialNFWParams.Mvir
    • TriaxialNFWParams.q1
    • TriaxialNFWParams.q2
    • TriaxialNFWParams.r_s
  • TwoPowerTriaxialParams
    • TwoPowerTriaxialParams.a
    • TwoPowerTriaxialParams.alpha
    • TwoPowerTriaxialParams.b
    • TwoPowerTriaxialParams.beta
    • TwoPowerTriaxialParams.c
    • TwoPowerTriaxialParams.rho
• odisseo.potentials module
  • MyamotoNagai()
  • NFW()
  • PowerSphericalPotentialwCutoff()
  • Thick_MN3DiskPotential()
  • Thin_MN3DiskPotential()
  • TriaxialNFW()
  • TwoPowerTriaxialPotential()
  • call_MyamotoNagai()
  • combined_external_acceleration_vmpa_switch()
  • logarithmic_potential()
  • point_mass()
• odisseo.time_integration module
  • SnapshotData
    • SnapshotData.angular_momentum
    • SnapshotData.current_checkpoint
    • SnapshotData.num_iterations
    • SnapshotData.runtime
    • SnapshotData.states
    • SnapshotData.times
    • SnapshotData.total_energy
  • time_integration()
• odisseo.units module
  • CodeUnits
• odisseo.utils module
  • Angular_momentum()
  • E_kin()
  • E_pot()
  • E_tot()
  • center_of_mass()
  • equat_to_gd1()
  • equat_to_gd1_velocity()
  • equat_to_gd1cart()
  • gal_to_equat()
  • gd1_projection_vmap()
  • gd1cart_to_gd1()
  • halo_to_gd1()
  • halo_to_gd1_all()
  • halo_to_gd1_velocity()
  • halo_to_gd1_velocity_vmap()
  • halo_to_gd1_vmap()
  • halo_to_sun()
  • jacobian_equat_to_gd1()
  • jacobian_halo_to_gd1()
  • projection_on_GD1()
  • sun_to_gal()
• odisseo.visualization module
  • build_core_stars()
  • build_spiral_arms()
  • build_spiral_stars()
  • create_3d_gif()
  • create_3d_gif_velocitycoding()
  • create_projection_gif()
  • energy_angular_momentum_plot()
  • haze()
  • plot_last_snapshot()
  • plot_orbit()
  • plot_sky_projection()
  • random()
  • spherical_coords()

Module contents

odisseo.construct_initial_state(position, velocity)

Constructs the initial state for the simulation.

Parameters:

• position (jnp.ndarray) – Initial positions of the bodies.

• velocity (jnp.ndarray) – Initial velocities of the bodies.

Returns:

jnp.ndarray – The initial state containing positions and velocities.