{
  "cells": [
    {
      "cell_type": "markdown",
      "id": "markdown-000",
      "metadata": {},
      "source": [
        "# Numerical drift in rotation\n",
        "\n",
        "[Read the lesson](https://tirta.co/a-byte-of-physics/numerical-drift-in-rotation). 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": [
        "Numerical drift is a systematic departure from a model's expected behavior as repeated\n",
        "approximations accumulate. A rotating object may look smooth while its calculated\n",
        "energy, angular momentum, phase, or orientation constraints gradually become wrong.\n",
        "Drift must be measured against the correct physical balance, not judged from appearance.\n",
        "\n",
        "## What can drift\n",
        "\n",
        "For a fixed-axis rigid body with constant moment of inertia\n",
        "$I$, angular momentum and rotational\n",
        "kinetic energy are\n",
        "\n",
        "\n",
        "$$\n",
        "L=I\\omega,\\qquad K_{\\mathrm{rot}}=\\frac12I\\omega^2\n",
        "$$\n",
        "\n",
        "\n",
        "With no external torque,\n",
        "$L$ should be constant. Energy is\n",
        "constant only when the modeled system is conservative and includes all relevant energy\n",
        "stores. Friction, damping, a motor, or a changing constraint can produce physical energy\n",
        "change.\n",
        "\n",
        "Even when invariants remain close, phase can drift. An oscillator may preserve nearly\n",
        "the right energy but reach its turning point too early on every cycle. Orientation can\n",
        "also violate its mathematical representation: a quaternion can lose unit length, or a\n",
        "rotation matrix can cease to be orthogonal.\n",
        "\n",
        "## Residuals rather than raw change\n",
        "\n",
        "The angular impulse balance is\n",
        "\n",
        "\n",
        "$$\n",
        "L(t_b)-L(t_a)=\\int_{t_a}^{t_b}\\tau_{\\mathrm{ext}}(t)\\,dt\n",
        "$$\n",
        "\n",
        "\n",
        "Define a numerical residual by subtracting the expected transfer:\n",
        "\n",
        "\n",
        "$$\n",
        "r_L=L_b-L_a-J_{\\tau,\\mathrm{ext}}\n",
        "$$\n",
        "\n",
        "\n",
        "For energy, use\n",
        "\n",
        "\n",
        "$$\n",
        "r_E=E_b-E_a-W_{\\mathrm{external}}-Q_{\\mathrm{modeled}}\n",
        "$$\n",
        "\n",
        "\n",
        "The sign of work and heat terms follows the declared system convention. The residual is\n",
        "the unexplained remainder. Demanding raw energy constancy while a drive performs work\n",
        "mislabels correct physics as numerical drift."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-003",
      "metadata": {},
      "source": [
        "initial_l, final_l, angular_impulse = 1.0, 1.6, 0.6\n",
        "initial_energy, final_energy, work = 1.0, 2.56, 1.56\n",
        "print(\"Angular momentum residual:\", final_l - initial_l - angular_impulse)\n",
        "print(\"Energy residual:\", final_energy - initial_energy - work)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-004",
      "metadata": {},
      "source": [
        "Report vector angular-momentum residuals in three dimensions. A small magnitude alone can\n",
        "hide which component or sign is wrong.\n",
        "\n",
        "## Forward Euler and artificial energy growth\n",
        "\n",
        "For an undamped torsional oscillator,\n",
        "\n",
        "\n",
        "$$\n",
        "I\\ddot\\theta=-\\kappa\\theta\n",
        "$$\n",
        "\n",
        "\n",
        "the exact mechanical energy is\n",
        "\n",
        "\n",
        "$$\n",
        "E=\\frac12I\\omega^2+\\frac12\\kappa\\theta^2\n",
        "$$\n",
        "\n",
        "\n",
        "Forward Euler uses\n",
        "\n",
        "\n",
        "$$\n",
        "\\theta_{n+1}=\\theta_n+\\omega_n\\Delta t,\\qquad \\omega_{n+1}=\\omega_n-\\frac{\\kappa}{I}\\theta_n\\Delta t\n",
        "$$\n",
        "\n",
        "\n",
        "Substitution shows that its energy grows by a positive term proportional to the square\n",
        "of the step for any nonzero state. Each individual increase can be small, yet repeated\n",
        "steps spiral outward in angular phase space. A smooth trace does not remove this\n",
        "algorithmic bias.\n",
        "\n",
        "Semi-implicit Euler updates angular velocity first and uses the new value for angle. It\n",
        "often keeps oscillatory energy error bounded, although the exact energy still fluctuates.\n",
        "Higher-order methods can reduce short-time truncation error but do not automatically\n",
        "preserve long-time geometric structure.\n",
        "\n",
        "## Compare at the same physical time\n",
        "\n",
        "If one calculation uses step\n",
        "$\\Delta t$ and another uses half that\n",
        "step, the second takes twice as many updates to reach the same physical duration.\n",
        "Comparing both after the same step count compares different experiments.\n",
        "\n",
        "Choose checkpoint times first. Advance each method to those times, using dense output or\n",
        "careful final substeps if needed. Compare angle, angular velocity, invariant residuals,\n",
        "and event timing at those shared times."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-005",
      "metadata": {},
      "source": [
        "for steps in [100, 200, 400]:\n",
        "    theta, omega, dt = 1.0, 0.0, 10.0 / steps\n",
        "    errors = []\n",
        "    for _ in range(steps):\n",
        "        omega -= theta * dt\n",
        "        theta += omega * dt\n",
        "        errors.append(abs(0.5 * (theta**2 + omega**2) - 0.5))\n",
        "    print(steps, \"maximum torsional oscillator energy error:\", max(errors))\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-006",
      "metadata": {},
      "source": [
        "Convergence should be assessed before selecting a tolerance. If halving the step no\n",
        "longer changes the result because roundoff, event error, or another approximation\n",
        "dominates, further refinement of that one parameter does not improve accuracy.\n",
        "\n",
        "## Absolute and relative measures\n",
        "\n",
        "An absolute residual retains physical units and matters near zero. A relative residual\n",
        "helps compare across scales:\n",
        "\n",
        "\n",
        "$$\n",
        "\\varepsilon_E=\\frac{|r_E|}{E_{\\mathrm{scale}}}\n",
        "$$\n",
        "\n",
        "\n",
        "The scale should not be a nearly cancelling total if large positive and negative terms\n",
        "exist. Use a characteristic sum of magnitudes or another physically justified reference.\n",
        "For angular momentum that is theoretically zero, only a relative error divided by the\n",
        "initial value is undefined; an absolute scale from typical individual angular momenta is\n",
        "needed.\n",
        "\n",
        "Track maximum, final, and root-mean-square residual. A final value near zero can hide\n",
        "large intermediate error that later cancels. Plotting only rounded values can hide small\n",
        "but systematic trends.\n",
        "\n",
        "## Angle wrapping can imitate drift\n",
        "\n",
        "Angles used for display may be wrapped into one turn. A transition from just below\n",
        "$2\\pi$ to just above zero is continuous\n",
        "physical rotation but appears as a large negative jump. Never calculate phase error,\n",
        "angular velocity, or accumulated rotation by naive subtraction of wrapped angles.\n",
        "\n",
        "Keep an unwrapped angle or turn count. When comparing orientations where completed turns\n",
        "do not matter, use a periodic difference such as\n",
        "\n",
        "\n",
        "$$\n",
        "\\delta\\theta=\\operatorname{atan2}\\left[\\sin(\\theta_1-\\theta_2),\\cos(\\theta_1-\\theta_2)\\right]\n",
        "$$\n",
        "\n",
        "\n",
        "This returns a principal signed orientation difference, not accumulated displacement.\n",
        "Choose the comparison that matches the question.\n",
        "\n",
        "## Roundoff and long duration\n",
        "\n",
        "Floating-point arithmetic rounds every operation. Random-looking roundoff may grow like a\n",
        "walk, while biased summation or asymmetric force evaluation can produce steady drift.\n",
        "Using a smaller time step increases the number of operations, so truncation error falls\n",
        "while accumulated roundoff may eventually become more important.\n",
        "\n",
        "Keep time with an integer step count when using a fixed step, then derive physical time\n",
        "from count and step. Repeatedly adding a floating-point step can accumulate clock error.\n",
        "Use compensated summation for long energy or impulse ledgers when cancellation is severe.\n",
        "\n",
        "## Constraint drift in rolling\n",
        "\n",
        "For ideal rolling on a stationary surface, a signed constraint has the form\n",
        "\n",
        "\n",
        "$$\n",
        "C=v_{\\mathrm{cm}}+R\\omega=0\n",
        "$$\n",
        "\n",
        "\n",
        "Independent translation and rotation updates can make\n",
        "$C$ grow even if each\n",
        "subsystem looks reasonable. Projection can force the residual back to zero, but an\n",
        "arbitrary correction may change momentum or energy. A constraint impulse should account\n",
        "for mass, inertia, contact geometry, and friction limits.\n",
        "\n",
        "Measure constraint residual separately from conservation residuals. Perfect constraint\n",
        "satisfaction can coexist with incorrect energy transfer.\n",
        "\n",
        "## Three-dimensional orientation drift\n",
        "\n",
        "A unit quaternion\n",
        "$q$ must satisfy\n",
        "\n",
        "\n",
        "$$\n",
        "q\\cdot q=1\n",
        "$$\n",
        "\n",
        "\n",
        "A rotation matrix must satisfy\n",
        "\n",
        "\n",
        "$$\n",
        "\\mathbf R^{\\mathsf T}\\mathbf R=\\mathbf I,\\qquad \\det\\mathbf R=1\n",
        "$$\n",
        "\n",
        "\n",
        "Naive additive updates can violate these conditions. Renormalizing a quaternion or\n",
        "re-orthogonalizing a matrix controls representation drift but does not prove that the\n",
        "physical trajectory is accurate. Compare orientation against a refined or analytic\n",
        "solution and monitor angular momentum and energy too.\n",
        "\n",
        "The frame of angular velocity matters. Body-frame and world-frame rates multiply the\n",
        "orientation update on different sides. Mixing them can yield a normalized but physically\n",
        "wrong rotation.\n",
        "\n",
        "## Discontinuous events\n",
        "\n",
        "Impacts, hard stops, clutches, and friction transitions create events. Error may be\n",
        "dominated by locating event time rather than by the smooth integrator between events.\n",
        "Record pre-event state, event impulse, post-event state, and balance residual. Reduce the\n",
        "event tolerance separately from the regular step.\n",
        "\n",
        "Applying the same contact impulse over several overlapping steps can create angular-\n",
        "momentum drift. Conversely, silently clamping angular velocity at a stop removes energy\n",
        "and momentum without a ledger. Event changes must be part of the physical accounting.\n",
        "\n",
        "## A layered validation sequence\n",
        "\n",
        "Begin with zero torque and fixed inertia: angular velocity must remain constant and angle\n",
        "must be linear in time. Continue with constant torque, which has an exact quadratic angle.\n",
        "Then use a conservative torsional oscillator to test phase and bounded energy error. Add\n",
        "damping only after conservative behavior is understood, and verify its predicted energy\n",
        "loss.\n",
        "\n",
        "For each layer, change one numerical parameter at a time. A reference result calculated\n",
        "with the same flawed method at only a smaller step is evidence of convergence, not an\n",
        "independent proof. Where possible, retain analytic solutions and balance identities.\n",
        "\n",
        "## Reliability checks\n",
        "\n",
        "- Define the physical invariant or transfer balance before measuring drift.\n",
        "- Compare calculations at identical physical times and initial conditions.\n",
        "- Track maximum and component-wise residuals, not only the final norm.\n",
        "- Use both absolute and physically scaled relative tolerances.\n",
        "- Keep unwrapped angle separate from display wrapping.\n",
        "- Refine step, event tolerance, and constraint tolerance independently.\n",
        "- Check phase error as well as energy and angular momentum.\n",
        "- Monitor quaternion norm or rotation-matrix orthogonality in three dimensions.\n",
        "- Record physical impulses and work instead of hiding them as corrections.\n",
        "- Distinguish convergence to the chosen model from accuracy of the model itself.\n",
        "\n",
        "Numerical drift becomes diagnosable when every expected transfer and representation\n",
        "constraint has its own residual. The goal is not a perfectly flat graph at any cost, but\n",
        "a calculation whose remaining error decreases predictably and whose physical changes are\n",
        "fully accounted for.\n",
        "\n",
        "## A worked Python notebook\n",
        "\n",
        "Evaluate the energy diagnostic for a rotor with inertia 0.5 kilogram metres squared. The question for this calculation is precise: how does **rotational energy**\n",
        "change when we change **angular velocity**, 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",
        "An energy change can be physical when external torque does work; drift is the unexplained residual. 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",
        "-4 to 4 rad/s. 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 angular velocity, expressed in rad/s. The output is\n",
        "rotational energy, expressed in J. 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{q^{2}}{4}\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 2 rad/s, the expression gives 1 J. 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-007",
      "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(\"q**2/4\", locals={\"q\": q})\n",
        "reference_input = 2.0\n",
        "reference_output = float(expression.subs(q, reference_input))\n",
        "print(\"Input:\", reference_input, \"rad/s\")\n",
        "print(\"Predicted rotational energy:\", reference_output, \"J\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-008",
      "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",
        "4 J. The midpoint, at 0 rad/s, gives\n",
        "0 J. The upper endpoint gives 4 J.\n",
        "Read these together with the assumptions: they describe different choices of angular velocity\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-009",
      "metadata": {},
      "source": [
        "evaluate = sp.lambdify(q, expression, modules=\"numpy\")\n",
        "inputs = np.array([-4.0, 0.0, 4.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(f\"{input_value:.6g} rad/s -> {output_value:.6g} J\")\n",
        "assert np.all(np.isfinite(outputs))\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-010",
      "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 angular velocity; the vertical axis shows rotational energy. 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-011",
      "metadata": {},
      "source": [
        "input_grid = np.linspace(-4.0, 4.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(\"Angular velocity (rad/s)\")\n",
        "ax.set_ylabel(\"Rotational energy (J)\")\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-012",
      "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",
        "rotational energy change near the reference input? Its units are the output unit divided by\n",
        "the input unit. For this calculation, the derivative at 2 rad/s is\n",
        "1 in those ratio units. Locally, the output increases 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{q}{2}\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-013",
      "metadata": {},
      "source": [
        "slope_expression = sp.diff(expression, q)\n",
        "slope_at_reference = float(slope_expression.subs(q, reference_input))\n",
        "changed_input = 2.04\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, \"J\")\n",
        "print(\"Full expression:\", full_prediction, \"J\")\n",
        "print(\"Approximation difference:\", full_prediction - linear_prediction, \"J\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-014",
      "metadata": {},
      "source": [
        "Here the changed input is 2.04 rad/s. The tangent estimate is\n",
        "1.04 J, while the full expression gives\n",
        "1.0404 J. 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-015",
      "metadata": {},
      "source": [
        "input_span = 8.0\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-016",
      "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",
        "angular velocity inside the stated interval, predict whether rotational energy 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: An energy change can be physical when external torque does work; drift is the unexplained residual. 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 J:\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-017",
      "metadata": {},
      "source": [
        "slope_function = sp.lambdify(q, slope_expression, modules=\"numpy\")\n",
        "endpoint_change = float(evaluate(4.0) - evaluate(-4.0))\n",
        "for points in [101, 201, 401]:\n",
        "    grid = np.linspace(-4.0, 4.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-018",
      "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 10.7, \"Newton's Second Law for Rotation\"](https://openstax.org/books/university-physics-volume-1/pages/10-7-newtons-second-law-for-rotation).\n",
        "- [OpenStax, *University Physics Volume 1*, Section 10.8, \"Work and Power for Rotational Motion\"](https://openstax.org/books/university-physics-volume-1/pages/10-8-work-and-power-for-rotational-motion).\n",
        "- [OpenStax, *University Physics Volume 1*, Section 11.2, \"Angular Momentum\"](https://openstax.org/books/university-physics-volume-1/pages/11-2-angular-momentum).\n",
        "- [MIT OpenCourseWare, *8.01SC Classical Mechanics*, Chapter 17, \"Two Dimensional Rotational Dynamics\"](https://ocw.mit.edu/courses/8-01sc-classical-mechanics-fall-2016/resources/mit8_01scs22_chapter17/).\n",
        "- [MIT OpenCourseWare, *8.01SC Classical Mechanics*, Chapter 19, \"Angular Momentum\"](https://ocw.mit.edu/courses/8-01sc-classical-mechanics-fall-2016/resources/mit8_01scs22_chapter19/).\n",
        "- [Benjamin Crowell, *Light and Matter*, Chapter 8, \"Rotation\"](https://open.umn.edu/opentextbooks/textbooks/light-and-matter)."
      ]
    }
  ],
  "metadata": {
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "name": "python",
      "version": "3.12"
    },
    "colab": {
      "name": "numerical-drift-in-rotation.ipynb",
      "provenance": []
    },
    "tirta": {
      "lesson": "https://tirta.co/a-byte-of-physics/numerical-drift-in-rotation",
      "source_sha256": "b05d186901476be66ceba18a2de019927cb1e100ae07e5c3be826e41289a9479"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 5
}
