Poincare Map Example

This example demonstrates how to compute Poincare maps in the Circular Restricted Three-Body Problem (CR3BP).

Computing Poincare maps in the CR3BP

"""Example script: generating and displaying a Poincare map for the Earth-Moon hiten.system.

python examples/poincare_map.py
"""

import os
import sys

sys.path.append(os.path.join(os.path.dirname(__file__), "..", "src"))

from hiten.system import System
from hiten.utils.log_config import logger


def main() -> None:
    """Generate and interactively display a Poincare map."""

    system = System.from_bodies("earth", "moon")

    l_point = system.get_libration_point(1)
    logger.info("Generating Poincare map for L%s of the %s-%s system...", 1, "Earth", "Moon")
    cm = l_point.get_center_manifold(degree=6)
    cm.compute()

    pm = cm.poincare_map(energy=0.7)
    overrides = {
        "n_seeds": 20,
        "n_iter": 20,
        "seed_strategy": "axis_aligned",
        "n_workers": 8
    }
    pm.compute(section_coord="p3", overrides=overrides)

    pm.plot(axes=("p2", "q3"))


if __name__ == "__main__":
    main()