{
  "cells": [
    {
      "cell_type": "markdown",
      "id": "markdown-000",
      "metadata": {},
      "source": [
        "# Two-body and N-body problems\n",
        "\n",
        "[Read the lesson](https://tirta.co/a-byte-of-physics/two-body-n-body-problems). This notebook includes the lesson text and its Python examples in reading order. Run the setup cell, then run the remaining cells from top to bottom. In Colab, use File → Save a copy in Drive to keep your edits. You can run and edit this view without changing the course source. Examples use illustrative data unless stated otherwise."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-001",
      "metadata": {},
      "source": [
        "import importlib.metadata\n",
        "import subprocess\n",
        "import sys\n",
        "\n",
        "requirements = {\"numpy\": (2, 0), \"sympy\": (1, 13), \"matplotlib\": (3, 8)}\n",
        "to_install = []\n",
        "for package, minimum in requirements.items():\n",
        "    try:\n",
        "        installed = importlib.metadata.version(package)\n",
        "        version = tuple(int(part) for part in installed.split(\".\")[:2])\n",
        "    except importlib.metadata.PackageNotFoundError:\n",
        "        version = (0, 0)\n",
        "    if version < minimum:\n",
        "        to_install.append(f\"{package}>={minimum[0]}.{minimum[1]}\")\n",
        "if to_install:\n",
        "    subprocess.check_call([sys.executable, \"-m\", \"pip\", \"install\", *to_install])\n",
        "\n",
        "import numpy as np\n",
        "import sympy as sp\n",
        "import matplotlib\n",
        "import matplotlib.pyplot as plt\n",
        "\n",
        "print(\"Python\", sys.version.split()[0])\n",
        "print(\"NumPy\", np.__version__)\n",
        "print(\"SymPy\", sp.__version__)\n",
        "print(\"Matplotlib\", matplotlib.__version__)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-002",
      "metadata": {},
      "source": [
        "In a two-body problem, each body accelerates because of the other. In an N-body problem,\n",
        "every body responds to the combined field of all the others. The rule is simple, but its\n",
        "coupling matters: the trajectories must be solved together from one simultaneous state.\n",
        "\n",
        "## Newton's law for a pair\n",
        "\n",
        "For bodies at positions\n",
        "$\\mathbf r_1$ and\n",
        "$\\mathbf r_2$, define\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf d=\\mathbf r_2-\\mathbf r_1\n",
        "$$\n",
        "\n",
        "\n",
        "The force on body 1 due to body 2 is\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf F_{1\\leftarrow2}=Gm_1m_2\\frac{\\mathbf d}{|\\mathbf d|^3}\n",
        "$$\n",
        "\n",
        "\n",
        "and Newton's third law gives\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf F_{2\\leftarrow1}=-\\mathbf F_{1\\leftarrow2}\n",
        "$$\n",
        "\n",
        "\n",
        "The forces are equal and opposite, but accelerations are generally not. Dividing by each\n",
        "body's mass gives acceleration inversely proportional to that body's inertial mass. A\n",
        "lighter body accelerates more, while the pair's total momentum remains constant in the\n",
        "absence of external force."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-003",
      "metadata": {},
      "source": [
        "import numpy as np\n",
        "\n",
        "masses = np.array([1.0, 2.0])\n",
        "positions = np.array([[-1.0, 0.0], [1.0, 0.0]])\n",
        "separation = positions[1] - positions[0]\n",
        "force = masses[0] * masses[1] * separation / np.linalg.norm(separation) ** 3\n",
        "print(\"Pair forces with G=1:\", force, -force)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-004",
      "metadata": {},
      "source": [
        "Computing both ordered pairs and adding each force twice is a common error. Visit only\n",
        "indices with $i<j$.\n",
        "\n",
        "## Reducing the two-body problem\n",
        "\n",
        "Center-of-mass position is\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf R=\\frac{m_1\\mathbf r_1+m_2\\mathbf r_2}{m_1+m_2}\n",
        "$$\n",
        "\n",
        "\n",
        "With no external force, it moves at constant velocity. Relative position\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf r=\\mathbf r_1-\\mathbf r_2\n",
        "$$\n",
        "\n",
        "\n",
        "obeys\n",
        "\n",
        "\n",
        "$$\n",
        "\\ddot{\\mathbf r}=-G(m_1+m_2)\\frac{\\mathbf r}{|\\mathbf r|^3}\n",
        "$$\n",
        "\n",
        "\n",
        "This turns the coupled relative motion into an equivalent one-body central-force problem.\n",
        "The reduced mass\n",
        "\n",
        "\n",
        "$$\n",
        "m_{\\mathrm r}=\\frac{m_1m_2}{m_1+m_2}\n",
        "$$\n",
        "\n",
        "\n",
        "is useful for writing the relative kinetic energy and angular momentum. After solving\n",
        "relative motion, recover each orbit about the center of mass. Both bodies trace similar\n",
        "conics scaled according to the mass ratio; the heavier body has the smaller orbit.\n",
        "\n",
        "When $m_1\\ll m_2$,\n",
        "the center of mass lies close to body 2 and treating body 2 as fixed can be an excellent\n",
        "approximation. It is still an approximation, not a different law of gravity.\n",
        "\n",
        "## Extending to N bodies\n",
        "\n",
        "For body $i$, acceleration is\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf a_i=G\\sum_{j\\ne i}m_j\\frac{\\mathbf r_j-\\mathbf r_i}{|\\mathbf r_j-\\mathbf r_i|^3}\n",
        "$$\n",
        "\n",
        "\n",
        "Direct evaluation requires roughly one pair calculation for every pair of bodies, so its\n",
        "cost grows as the square of body count. The exact pair count is\n",
        "$N(N-1)/2$."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-005",
      "metadata": {},
      "source": [
        "def all_forces(positions, masses):\n",
        "    forces = np.zeros_like(positions)\n",
        "    for i in range(len(masses) - 1):\n",
        "        for j in range(i + 1, len(masses)):\n",
        "            offset = positions[j] - positions[i]\n",
        "            radius = np.linalg.norm(offset)\n",
        "            if radius == 0:\n",
        "                raise ValueError(\"Point bodies overlap.\")\n",
        "            force = masses[i] * masses[j] * offset / radius**3\n",
        "            forces[i] += force\n",
        "            forces[j] -= force\n",
        "    return forces\n",
        "\n",
        "\n",
        "forces = all_forces(positions, masses)\n",
        "assert np.allclose(forces.sum(axis=0), 0.0)\n",
        "print(forces)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-006",
      "metadata": {},
      "source": [
        "The state used by every pair must correspond to the same time. Advancing body 1 before\n",
        "calculating body 2's interactions makes the answer depend on array order and no longer\n",
        "implements the intended simultaneous equations.\n",
        "\n",
        "## Conserved quantities of an isolated system\n",
        "\n",
        "Total linear momentum is\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf P=\\sum_i m_i\\mathbf v_i\n",
        "$$\n",
        "\n",
        "\n",
        "and pair-force cancellation gives a constant\n",
        "$\\mathbf P$. Therefore the center\n",
        "of mass\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf R_{\\mathrm{cm}}=\\frac{\\sum_i m_i\\mathbf r_i}{\\sum_i m_i}\n",
        "$$\n",
        "\n",
        "\n",
        "moves uniformly. A drifting center-of-mass velocity in an isolated calculation often\n",
        "reveals asymmetric force accumulation or numerical error.\n",
        "\n",
        "Total angular momentum about a fixed inertial origin is\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf L=\\sum_i m_i\\mathbf r_i\\times\\mathbf v_i\n",
        "$$\n",
        "\n",
        "\n",
        "and Newtonian pair forces conserve it because each force lies along its pair separation.\n",
        "Total mechanical energy is\n",
        "\n",
        "\n",
        "$$\n",
        "E=\\sum_i\\frac12m_i|\\mathbf v_i|^2-G\\sum_{i<j}\\frac{m_im_j}{r_{ij}}\n",
        "$$\n",
        "\n",
        "\n",
        "The potential sum also visits each unordered pair once. These quantities are exact\n",
        "invariants of the ideal isolated continuous equations. A numerical method only\n",
        "approximates them.\n",
        "\n",
        "## Choosing an integration method\n",
        "\n",
        "At each step, first assemble accelerations or forces. A symplectic method such as\n",
        "velocity Verlet often gives useful long-term energy behavior for conservative gravity.\n",
        "Adaptive high-order methods can resolve encounters and event times efficiently. The best\n",
        "choice depends on whether the goal is long statistical evolution, precise short arcs, or\n",
        "close-encounter timing.\n",
        "\n",
        "Whatever method is chosen, compare results at fixed physical times under step or tolerance\n",
        "refinement. Monitor each invariant separately. Small total energy error does not guarantee\n",
        "that individual trajectories or encounter times are accurate in a chaotic system.\n",
        "\n",
        "## Close encounters and collision policy\n",
        "\n",
        "Point-mass acceleration diverges when separation goes to zero. A real body may collide,\n",
        "deform, merge, fragment, or pass through only if the model deliberately permits that.\n",
        "Define radii and event handling when collisions matter.\n",
        "\n",
        "A Plummer-style softened acceleration is\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf a_i=G\\sum_{j\\ne i}m_j\\frac{\\mathbf r_j-\\mathbf r_i}{(r_{ij}^2+\\epsilon^2)^{3/2}}\n",
        "$$\n",
        "\n",
        "\n",
        "Softening length\n",
        "$\\epsilon$ removes the\n",
        "point singularity but changes small-scale force and potential. It can represent limited\n",
        "spatial resolution in a large collisionless system, but it is not an exact point-body\n",
        "solution. Report it in physical units and repeat important conclusions with other\n",
        "appropriate values.\n",
        "\n",
        "Adaptive time steps introduce another issue: if each body uses a different time without\n",
        "a consistent algorithm, pair symmetry and synchronization can be lost. Encounter\n",
        "methods must preserve a well-defined common evolution policy.\n",
        "\n",
        "## Merging bodies consistently\n",
        "\n",
        "If the model says colliding bodies merge perfectly inelastically, conserve total mass and\n",
        "linear momentum:\n",
        "\n",
        "\n",
        "$$\n",
        "m_{\\mathrm new}=m_1+m_2\\qquad\\mathbf v_{\\mathrm new}=\\frac{m_1\\mathbf v_1+m_2\\mathbf v_2}{m_1+m_2}\n",
        "$$\n",
        "\n",
        "\n",
        "Place the new body at the pair center of mass. Mechanical energy is not conserved by an\n",
        "inelastic merge; the missing kinetic energy represents deformation, heat, radiation, or\n",
        "other unresolved internal processes. Angular momentum associated with internal spin may\n",
        "also need storage if the merged object is not treated as a point."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-007",
      "metadata": {},
      "source": [
        "velocities = np.array([[0.0, 0.2], [0.0, -0.1]])\n",
        "merged_mass = masses.sum()\n",
        "merged_position = np.sum(masses[:, None] * positions, axis=0) / merged_mass\n",
        "merged_velocity = np.sum(masses[:, None] * velocities, axis=0) / merged_mass\n",
        "print(merged_mass, merged_position, merged_velocity)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-008",
      "metadata": {},
      "source": [
        "## Faster approximations\n",
        "\n",
        "Direct summation is clear and useful for validation, but large systems need other\n",
        "approaches. Tree methods group distant masses, particle-mesh methods solve a field on a\n",
        "grid, and hybrid methods combine near direct forces with distant approximations. Each\n",
        "introduces parameters such as opening angle, grid spacing, or assignment kernel.\n",
        "\n",
        "These methods should be compared against direct summation on smaller fixtures. Parallel\n",
        "reductions may change floating-point addition order, producing tiny differences that\n",
        "later grow in chaotic systems. Record deterministic ordering or acceptable statistical\n",
        "comparisons rather than expecting bit-identical trajectories automatically.\n",
        "\n",
        "## Verification sequence\n",
        "\n",
        "Start with two unequal masses in their center-of-mass frame. Confirm equal-and-opposite\n",
        "pair force, unequal acceleration, stationary center of mass, and agreement with a known\n",
        "circular or elliptic orbit. Then add a distant third body and predict the direction of its\n",
        "initial perturbation.\n",
        "\n",
        "Permute body storage order and confirm that physical output agrees within the declared\n",
        "floating-point policy. Reduce time step, examine minimum separations, and compare energy,\n",
        "momentum, angular momentum, and center-of-mass velocity. For softened or approximate\n",
        "forces, vary the approximation parameter independently of time step.\n",
        "\n",
        "An N-body trajectory is credible only with the initial state, unit system, force law,\n",
        "collision or softening policy, integration method, and error evidence. Determinism of the\n",
        "equations does not imply that a long chaotic trajectory is predictable to arbitrary\n",
        "precision.\n",
        "\n",
        "## A worked Python notebook\n",
        "\n",
        "Calculate the mutual force between point masses 1 and 2 in units with G equal to one. The question for this calculation is precise: how does **attractive force magnitude**\n",
        "change when we change **pair separation**, while keeping the other choices fixed?\n",
        "The earlier sections explain the physical relationship. Here we turn one part of that\n",
        "relationship into a calculation that you can run, inspect, and change in Google Colab.\n",
        "\n",
        "The same force magnitude acts on each mass in opposite directions. Keep that sentence beside the results. A formula can return a number outside\n",
        "its useful range, and a plotting library will usually draw that number without asking\n",
        "whether the physical assumptions still hold. The selected interval is from\n",
        "0.5 to 3 scaled length units. It gives us a specific domain to discuss\n",
        "instead of leaving the graph's range to a software default.\n",
        "\n",
        "Use the notebook link above the lesson, run the setup cell, and then run the cells in\n",
        "order. The code below is also included there. NumPy handles the arrays of numerical\n",
        "values, SymPy keeps the equation symbolic, and Matplotlib draws the results. If these\n",
        "libraries or notebook cells are unfamiliar, start with\n",
        "[Your first physics notebook](https://tirta.co/a-byte-of-physics/your-first-physics-notebook).\n",
        "\n",
        "### State the relationship before choosing a graph\n",
        "\n",
        "In this small example, `q` stands for pair separation, expressed in scaled length units. The output is\n",
        "attractive force magnitude, expressed in scaled force units. The equation below uses the numerical values in those\n",
        "units. Any fixed coefficients therefore carry the units needed to make the relation\n",
        "consistent. A value supplied in another unit must be converted before it enters this\n",
        "expression.\n",
        "\n",
        "\n",
        "$$\n",
        "f(q) = \\frac{2}{q^{2}}\n",
        "$$\n",
        "\n",
        "\n",
        "This is a function of one changing input. It is not a statement that the physical\n",
        "system has only one variable. We deliberately hold the other inputs fixed so that one\n",
        "cause of variation is visible. Changing several inputs at once would make it harder\n",
        "to tell which change produced the output difference. Write down the held-fixed choices\n",
        "before trying a different experiment.\n",
        "\n",
        "At 1 scaled length units, the expression gives 2 scaled force units. This is our\n",
        "reference calculation. It is a prediction of the stated example, not a new measurement.\n",
        "Its displayed digits help us compare implementations. They should not be read as a\n",
        "claim that physical inputs are known to six significant figures."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-009",
      "metadata": {},
      "source": [
        "import numpy as np\n",
        "import sympy as sp\n",
        "import matplotlib.pyplot as plt\n",
        "\n",
        "q = sp.Symbol(\"q\", real=True)\n",
        "expression = sp.sympify(\"2/q**2\", locals={\"q\": q})\n",
        "reference_input = 1.0\n",
        "reference_output = float(expression.subs(q, reference_input))\n",
        "print(\"Input:\", reference_input, \"scaled length units\")\n",
        "print(\"Predicted attractive force magnitude:\", reference_output, \"scaled force units\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-010",
      "metadata": {},
      "source": [
        "`sp.Symbol` declares a mathematical symbol. It does not assign a numerical measurement\n",
        "to that symbol. `subs` then replaces the symbol with our chosen input, and `float`\n",
        "converts the resulting symbolic number into a standard Python numerical value.\n",
        "Keeping these steps separate lets us inspect the equation before numerical substitution.\n",
        "It also prevents a later plot from becoming the only surviving record of the model.\n",
        "\n",
        "### Compare a few cases you can explain\n",
        "\n",
        "Before drawing hundreds of points, calculate three. The lower endpoint gives\n",
        "8 scaled force units. The midpoint, at 1.75 scaled length units, gives\n",
        "0.653061 scaled force units. The upper endpoint gives 0.222222 scaled force units.\n",
        "Read these together with the assumptions: they describe different choices of pair separation\n",
        "within the same example, with the other inputs held fixed.\n",
        "\n",
        "These three values are a useful first check, but they do not establish the shape between\n",
        "them. Two endpoints can be equal even when the curve bends or oscillates in between.\n",
        "Likewise, a large change between endpoints does not tell us whether the relationship is\n",
        "linear. We need the equation and intermediate samples to answer those questions."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-011",
      "metadata": {},
      "source": [
        "evaluate = sp.lambdify(q, expression, modules=\"numpy\")\n",
        "inputs = np.array([0.5, 1.75, 3.0])\n",
        "outputs = np.asarray(evaluate(inputs), dtype=float)\n",
        "outputs = np.broadcast_to(outputs, inputs.shape)\n",
        "for input_value, output_value in zip(inputs, outputs):\n",
        "    print(\n",
        "        f\"{input_value:.6g} scaled length units -> \"\n",
        "        f\"{output_value:.6g} scaled force units\"\n",
        "    )\n",
        "assert np.all(np.isfinite(outputs))\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-012",
      "metadata": {},
      "source": [
        "`lambdify` turns the symbolic expression into a numerical function. We explicitly select\n",
        "NumPy as its numerical backend so it can accept an array. The same expression now has\n",
        "two useful forms: one for algebra and one for calculating many cases. The assertion\n",
        "checks that the chosen inputs produce finite outputs. It would catch an infinity or an\n",
        "undefined numerical result, but a finite value can still be physically inappropriate.\n",
        "\n",
        "### Make a graph that answers the question\n",
        "\n",
        "The horizontal axis shows pair separation; the vertical axis shows attractive force magnitude. Every plotted\n",
        "point is a prediction of the same expression. We mark the reference case separately\n",
        "so that a reader can connect the earlier printed value to its place on the curve.\n",
        "The axis labels include units because a shape without a scale cannot tell us the size\n",
        "of a physical effect."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-013",
      "metadata": {},
      "source": [
        "input_grid = np.linspace(0.5, 3.0, 301)\n",
        "output_grid = np.asarray(evaluate(input_grid), dtype=float)\n",
        "output_grid = np.broadcast_to(output_grid, input_grid.shape)\n",
        "fig, ax = plt.subplots(figsize=(7, 4))\n",
        "ax.plot(input_grid, output_grid, label=\"Stated model\")\n",
        "ax.scatter([reference_input], [reference_output], label=\"Reference case\", zorder=3)\n",
        "ax.set_xlabel(\"Pair separation (scaled length units)\")\n",
        "ax.set_ylabel(\"Attractive force magnitude (scaled force units)\")\n",
        "ax.legend()\n",
        "ax.grid(alpha=0.25)\n",
        "fig.tight_layout()\n",
        "plt.show()\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-014",
      "metadata": {},
      "source": [
        "Look for a sign change, a turning point, or a region where the curve becomes steep.\n",
        "Connect each feature to the expression rather than to the colour of the line. If the\n",
        "curve has a minimum or maximum, changing the input in opposite directions may have\n",
        "similar effects on the output near that point. If it is nearly straight over a short\n",
        "interval, a local linear approximation may be useful even when the complete expression\n",
        "is nonlinear.\n",
        "\n",
        "Increasing the number of samples makes the line smoother and may reveal features that\n",
        "a coarse grid missed. It does not improve the physical inputs or enlarge the model's\n",
        "validity range. For quantities that can take only discrete values, use the line as a\n",
        "guide to the relationship and interpret only the allowed input values. A plotting\n",
        "function does not enforce such restrictions for you.\n",
        "\n",
        "### Measure the local sensitivity\n",
        "\n",
        "A derivative asks a more specific question than the whole graph: how quickly does\n",
        "attractive force magnitude change near the reference input? Its units are the output unit divided by\n",
        "the input unit. For this calculation, the derivative at 1 scaled length units is\n",
        "-4 in those ratio units. Locally, the output decreases as the\n",
        "input increases. This statement is local; the sign or size of the slope can be different\n",
        "elsewhere on the curve.\n",
        "\n",
        "\n",
        "$$\n",
        "f'(q) = - \\frac{4}{q^{3}}\n",
        "$$\n",
        "\n",
        "\n",
        "For a small input change, multiply the derivative by that change to estimate the output\n",
        "change. Then compare that estimate with a fresh evaluation of the full expression.\n",
        "The distinction between these two calculations matters: the derivative gives a tangent\n",
        "approximation, while substitution evaluates the chosen expression at the new point."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-015",
      "metadata": {},
      "source": [
        "slope_expression = sp.diff(expression, q)\n",
        "slope_at_reference = float(slope_expression.subs(q, reference_input))\n",
        "changed_input = 1.0125\n",
        "linear_prediction = reference_output + slope_at_reference * (\n",
        "    changed_input - reference_input\n",
        ")\n",
        "full_prediction = float(evaluate(changed_input))\n",
        "print(\"Symbolic derivative:\", slope_expression)\n",
        "print(\"Tangent prediction:\", linear_prediction, \"scaled force units\")\n",
        "print(\"Full expression:\", full_prediction, \"scaled force units\")\n",
        "print(\n",
        "    \"Approximation difference:\",\n",
        "    full_prediction - linear_prediction,\n",
        "    \"scaled force units\",\n",
        ")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-016",
      "metadata": {},
      "source": [
        "Here the changed input is 1.0125 scaled length units. The tangent estimate is\n",
        "1.95 scaled force units, while the full expression gives\n",
        "1.95092 scaled force units. If the two values are close, it means the chosen\n",
        "input change is small enough for this local approximation at this point. It does not\n",
        "mean the entire model is linear. If the slope is zero, a quadratic or higher-order term\n",
        "may supply the leading change instead.\n",
        "\n",
        "This is also the beginning of uncertainty propagation. If the input has a small stated\n",
        "standard uncertainty, the magnitude of the slope times that uncertainty estimates one\n",
        "contribution to output uncertainty. Do not automatically use the same rule for a wide\n",
        "interval, a discontinuity, a turning point with zero first derivative, or several\n",
        "correlated uncertain inputs. Those situations need more of the model than one slope.\n",
        "\n",
        "### Check the derivative with numerical differences\n",
        "\n",
        "We can estimate the same slope from values on either side of the reference point.\n",
        "Subtract the lower output from the upper output and divide by twice the input step.\n",
        "Parentheses are essential: the entire output difference must be divided by the entire\n",
        "input interval. Without them, Python can execute a different mathematical expression\n",
        "while still producing a plausible number.\n",
        "\n",
        "The calculation below uses three decreasing step sizes. Each estimate is compared with\n",
        "the symbolic derivative, so the reference is independent of the numerical difference\n",
        "formula. Near an endpoint, it is acceptable here to evaluate a small extension only\n",
        "because the displayed mathematical expression is smooth there; such a check does not\n",
        "assert that extrapolated values describe additional physical cases."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-017",
      "metadata": {},
      "source": [
        "input_span = 2.5\n",
        "for fraction in [1e-2, 1e-3, 1e-4]:\n",
        "    step = input_span * fraction\n",
        "    upper = float(evaluate(reference_input + step))\n",
        "    lower = float(evaluate(reference_input - step))\n",
        "    numerical_slope = (upper - lower) / (2 * step)\n",
        "    difference = numerical_slope - slope_at_reference\n",
        "    print(f\"step={step:.6g}: slope={numerical_slope:.6g}, difference={difference:.6g}\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-018",
      "metadata": {},
      "source": [
        "For a smooth function, reducing the step often reduces the approximation error at first.\n",
        "Very small steps can eventually make subtraction of nearly equal floating-point values\n",
        "less reliable. Therefore “smaller” is not an unlimited guarantee of “better.” Compare\n",
        "the observed behaviour over a useful range of steps instead of choosing the smallest\n",
        "number the computer can represent.\n",
        "\n",
        "An absolute slope difference is useful when the expected slope is near zero. A relative\n",
        "difference would divide by a very small reference and could look enormous even when the\n",
        "absolute discrepancy is tiny. Away from zero, a relative difference can help compare\n",
        "quantities with very different scales. The tolerance should reflect the question and\n",
        "the numerical method, rather than a universal decimal threshold copied into every test.\n",
        "\n",
        "### Change the example and preserve the reasoning\n",
        "\n",
        "Make one copy of the reference calculation before editing it. Choose a second value of\n",
        "pair separation inside the stated interval, predict whether attractive force magnitude will rise or fall,\n",
        "and then evaluate the expression. Place the prediction and the result in a text cell.\n",
        "If they disagree, identify whether the mistake was in the sign, the unit conversion,\n",
        "the algebra, or the assumption about how the curve behaves.\n",
        "\n",
        "Next choose a shorter plotting interval around the reference point. Compare the local\n",
        "curve with its tangent approximation. Then widen the interval again. The difference\n",
        "shows why a useful local estimate need not describe the whole range. Keep the axis\n",
        "labels and numerical scale visible when comparing the plots; automatic axis limits can\n",
        "make two different effects appear equally large.\n",
        "\n",
        "Finally, return to the physical limit stated at the start: The same force magnitude acts on each mass in opposite directions. Describe one change\n",
        "to the setup that would violate that limit. State what new input, equation, or observation\n",
        "would be needed before trusting an answer in the changed situation. This last step\n",
        "connects the Python experiment to the physics instead of treating successful execution\n",
        "as the only criterion for a useful result.\n",
        "\n",
        "Before saving, restart the runtime and run all cells in order. A result that depends on\n",
        "an earlier, now-deleted input cell is not reproducible from the saved notebook. Keep\n",
        "unrounded values for later calculations, round only the reported output, and label any\n",
        "new data as measured, assumed, or generated. The graph and printed values should remain\n",
        "traceable to the same expression and the same choices of inputs.\n",
        "\n",
        "### Recover the output change by integrating the slope\n",
        "\n",
        "There is another way to check that the expression and its derivative describe the same\n",
        "relationship. Add up the small output changes predicted by the slope across the chosen\n",
        "input interval. In the continuous limit, this integral equals the output at the upper\n",
        "endpoint minus the output at the lower endpoint. The result has units of scaled force units:\n",
        "the input unit in the integration cancels the input unit in the denominator of the slope.\n",
        "\n",
        "This statement is about the mathematics of this particular relationship. It does not\n",
        "assign a new physical meaning to an arbitrary area under a graph. For example, an area\n",
        "under force versus distance represents work, while an area under a different pair of\n",
        "quantities may represent something else or have no standard name. Always carry the\n",
        "axis units through the calculation before naming an integral."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-019",
      "metadata": {},
      "source": [
        "slope_function = sp.lambdify(q, slope_expression, modules=\"numpy\")\n",
        "endpoint_change = float(evaluate(3.0) - evaluate(0.5))\n",
        "for points in [101, 201, 401]:\n",
        "    grid = np.linspace(0.5, 3.0, points)\n",
        "    slopes = np.asarray(slope_function(grid), dtype=float)\n",
        "    slopes = np.broadcast_to(slopes, grid.shape)\n",
        "    integrated_change = np.trapezoid(slopes, grid)\n",
        "    print(\n",
        "        points,\n",
        "        \"integrated change:\",\n",
        "        integrated_change,\n",
        "        \"endpoint change:\",\n",
        "        endpoint_change,\n",
        "    )\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-020",
      "metadata": {},
      "source": [
        "The trapezoidal rule joins neighbouring slope samples with straight segments and adds\n",
        "their signed areas. A negative contribution remains negative; replacing every slope\n",
        "by its magnitude would calculate a different quantity. Increasing the sample count\n",
        "provides a practical refinement check. If the values do not settle toward the endpoint\n",
        "difference, first inspect the interval, singularities, and array ordering.\n",
        "\n",
        "Do not conclude that two agreeing calculations prove the model is physically complete.\n",
        "Both use the same selected relationship, so an omitted effect would be omitted from\n",
        "both. Their agreement is evidence that the numerical tools implement that relationship\n",
        "consistently. Comparing with an independent experiment or a more appropriate physical\n",
        "model is a separate test.\n",
        "\n",
        "Keep this comparison in the notebook when you change the expression. It is particularly\n",
        "useful after adding a coefficient, changing a sign convention, or converting an input\n",
        "unit. A code cell that produces a smooth graph can conceal these mistakes; a second\n",
        "calculation of the same output change gives you another place to notice them.\n",
        "\n",
        "## Sources\n",
        "\n",
        "- [OpenStax, *University Physics Volume 1*, Section 13.1, \"Newton's Law of Universal Gravitation\"](https://openstax.org/books/university-physics-volume-1/pages/13-1-newtons-law-of-universal-gravitation).\n",
        "- [OpenStax, *University Physics Volume 1*, Section 13.4, \"Satellite Orbits and Energy\"](https://openstax.org/books/university-physics-volume-1/pages/13-4-satellite-orbits-and-energy).\n",
        "- [MIT OpenCourseWare, *8.01SC Classical Mechanics*, Chapter 25, Sections 25.1-25.5](https://ocw.mit.edu/courses/8-01sc-classical-mechanics-fall-2016/resources/mit8_01scs22_chapter25new/).\n",
        "- [Benjamin Crowell, *Light and Matter*, center-of-mass and gravitation discussions](https://open.umn.edu/opentextbooks/textbooks/light-and-matter).\n",
        "- [The Physics Hypertextbook, \"Gravity\"](https://physics.info/gravity/).\n",
        "- [Physics LibreTexts, \"Newton's Universal Law of Gravitation\"](https://phys.libretexts.org/Bookshelves/University_Physics/University_Physics_%28OpenStax%29/Book%3A_University_Physics_I_-_Mechanics_Sound_Oscillations_and_Waves_%28OpenStax%29/13%3A_Gravitation/13.02%3A_Newtons_Law_of_Universal_Gravitation)."
      ]
    }
  ],
  "metadata": {
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "name": "python",
      "version": "3.12"
    },
    "colab": {
      "name": "two-body-n-body-problems.ipynb",
      "provenance": []
    },
    "tirta": {
      "lesson": "https://tirta.co/a-byte-of-physics/two-body-n-body-problems",
      "source_sha256": "922a24c9eaeb8e201fd8b2ad1bbec117afc40487fbb04f6263b49c90deca3ddb"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 5
}