01_simple_projectile.py

#!/usr/bin/env python3
"""Example 1: Simple Projectile Motion Calculation

This example shows how to use the physics MCP server to calculate
the trajectory of a projectile (like a cannonball or basketball).
"""

import asyncio
from chuk_mcp_physics.providers.analytic import AnalyticProvider
from chuk_mcp_physics.models import ProjectileMotionRequest


async def main():
    """Calculate projectile motion for a cannonball."""

    # Create the analytic provider
    provider = AnalyticProvider()

    print("=" * 60)
    print("CANNONBALL TRAJECTORY CALCULATOR")
    print("=" * 60)
    print()

    # Example 1: 45-degree shot (maximum range)
    print("1. Cannonball fired at 45° (maximum range angle)")
    print("-" * 60)

    request = ProjectileMotionRequest(
        initial_velocity=50.0,  # 50 m/s
        angle_degrees=45.0,  # 45 degrees
        initial_height=0.0,  # Ground level
        gravity=9.81,  # Earth gravity
    )

    result = await provider.calculate_projectile_motion(request)

    print(f"Initial velocity: {request.initial_velocity} m/s")
    print(f"Launch angle: {request.angle_degrees}°")
    print(f"Initial height: {request.initial_height} m")
    print()
    print("RESULTS:")
    print(f"  Maximum height: {result.max_height:.2f} m")
    print(f"  Range: {result.range:.2f} m")
    print(f"  Time of flight: {result.time_of_flight:.2f} s")
    print(f"  Trajectory points: {len(result.trajectory_points)} samples")
    print()

    # Show some trajectory points
    print("Sample trajectory points (x, y):")
    for i in range(0, len(result.trajectory_points), 10):
        x, y = result.trajectory_points[i]
        print(
            f"  t={i * result.time_of_flight / len(result.trajectory_points):.2f}s: "
            f"({x:.1f}m, {y:.1f}m)"
        )
    print()

    # Example 2: Basketball free throw
    print("2. Basketball free throw simulation")
    print("-" * 60)

    # Free throw is 4.6m from basket, basket is 3.05m high, player releases at 2.0m
    # We need to find the right angle for a given velocity

    request = ProjectileMotionRequest(
        initial_velocity=7.0,  # 7 m/s (typical free throw speed)
        angle_degrees=52.0,  # Optimal angle for free throw
        initial_height=2.0,  # Player's release height
        gravity=9.81,
    )

    result = await provider.calculate_projectile_motion(request)

    print(f"Initial velocity: {request.initial_velocity} m/s")
    print(f"Launch angle: {request.angle_degrees}°")
    print(f"Release height: {request.initial_height} m")
    print()
    print("RESULTS:")
    print(f"  Maximum height: {result.max_height:.2f} m")
    print(f"  Range: {result.range:.2f} m")
    print(f"  Time of flight: {result.time_of_flight:.2f} s")
    print()

    # Check if it goes through the basket (4.6m away, 3.05m high)
    basket_distance = 4.6
    basket_height = 3.05

    # Find the height at basket distance
    for i, (x, y) in enumerate(result.trajectory_points):
        if x >= basket_distance:
            print(f"At basket position ({basket_distance}m):")
            print(f"  Ball height: {y:.2f} m")
            print(f"  Basket height: {basket_height:.2f} m")
            if abs(y - basket_height) < 0.5:
                print("  ✓ SWISH! Ball passes through basket!")
            elif y > basket_height:
                print(f"  ↑ Ball is {y - basket_height:.2f}m above basket")
            else:
                print(f"  ↓ Ball is {basket_height - y:.2f}m below basket")
            break
    print()

    # Example 3: Compare different angles
    print("3. Comparing different launch angles (same velocity)")
    print("-" * 60)
    print(f"{'Angle':<8} {'Max Height':<12} {'Range':<12} {'Flight Time':<12}")
    print("-" * 60)

    for angle in [15, 30, 45, 60, 75]:
        request = ProjectileMotionRequest(
            initial_velocity=30.0, angle_degrees=float(angle), initial_height=0.0, gravity=9.81
        )
        result = await provider.calculate_projectile_motion(request)

        print(
            f"{angle}°{'':<6} {result.max_height:<12.2f} {result.range:<12.2f} {result.time_of_flight:<12.2f}"
        )

    print()
    print("=" * 60)
    print("Note: 45° gives maximum range for projectiles!")
    print("=" * 60)


if __name__ == "__main__":
    asyncio.run(main())