{
  "cells": [
    {
      "cell_type": "markdown",
      "id": "markdown-000",
      "metadata": {},
      "source": [
        "# Nonlinear oscillators\n",
        "\n",
        "[Read the lesson](https://tirta.co/a-byte-of-physics/nonlinear-oscillators). 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": [
        "A nonlinear oscillator has a restoring force that is not exactly proportional to\n",
        "displacement. Its frequency can depend on amplitude, waveforms need not be sinusoidal,\n",
        "and responses cannot generally be added by superposition. The same position-velocity\n",
        "state architecture still works, but the force law and interpretation become richer.\n",
        "\n",
        "## Linear motion is a local approximation\n",
        "\n",
        "Near a stable equilibrium\n",
        "$x_0$, a smooth potential\n",
        "can be expanded as\n",
        "\n",
        "\n",
        "$$\n",
        "U(x_0+\\eta)=U(x_0)+\\frac12U''(x_0)\\eta^2+\\frac16U'''(x_0)\\eta^3+\\cdots\n",
        "$$\n",
        "\n",
        "\n",
        "The first derivative vanishes at equilibrium. Keeping only the positive quadratic term\n",
        "produces simple harmonic motion. Larger displacement makes higher powers important.\n",
        "Thus “nonlinear” often means that the small-oscillation approximation is no longer\n",
        "accurate for the amplitude of interest.\n",
        "\n",
        "The local small-oscillation angular frequency remains\n",
        "\n",
        "\n",
        "$$\n",
        "\\omega_0=\\sqrt{\\frac{U''(x_0)}{m}}\n",
        "$$\n",
        "\n",
        "\n",
        "but it need not describe finite-amplitude periods.\n",
        "\n",
        "## A cubic-stiffness model\n",
        "\n",
        "The Duffing-style unforced conservative equation is\n",
        "\n",
        "\n",
        "$$\n",
        "m\\ddot x+kx+\\beta x^3=0\n",
        "$$\n",
        "\n",
        "\n",
        "Its restoring force is\n",
        "\n",
        "\n",
        "$$\n",
        "F(x)=-kx-\\beta x^3\n",
        "$$\n",
        "\n",
        "\n",
        "and its potential is\n",
        "\n",
        "\n",
        "$$\n",
        "U(x)=\\frac12kx^2+\\frac14\\beta x^4\n",
        "$$\n",
        "\n",
        "\n",
        "because $F=-dU/dx$.\n",
        "The cubic coefficient has units force per length cubed."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-003",
      "metadata": {},
      "source": [
        "import numpy as np\n",
        "\n",
        "linear_stiffness, cubic_stiffness, mass, position = 1.0, 0.5, 1.0, 1.0\n",
        "force = -linear_stiffness * position - cubic_stiffness * position**3\n",
        "print(\"Acceleration (m/s^2):\", force / mass)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-004",
      "metadata": {},
      "source": [
        "Setting $\\beta=0$ must recover\n",
        "the linear oscillator exactly.\n",
        "\n",
        "## Hardening and softening behavior\n",
        "\n",
        "With positive linear stiffness and positive\n",
        "$\\beta$, the restoring force\n",
        "grows more rapidly than a linear spring at large displacement. This is a hardening\n",
        "oscillator, and its finite-amplitude frequency generally increases with amplitude.\n",
        "\n",
        "A negative cubic coefficient initially softens the spring. However, the simple quartic\n",
        "potential then falls without bound at sufficiently large displacement, so it is only a\n",
        "local model unless stabilizing higher-order terms or physical travel limits are added.\n",
        "Never extrapolate a fitted negative quartic model to arbitrary amplitude.\n",
        "\n",
        "For weak cubic nonlinearity and amplitude\n",
        "$A$, an approximate hardening\n",
        "frequency is\n",
        "\n",
        "\n",
        "$$\n",
        "\\omega(A)\\approx\\omega_0\\left(1+\\frac{3\\beta A^2}{8k}\\right)\n",
        "$$\n",
        "\n",
        "\n",
        "when the correction is small. This approximation shows the trend but should be checked\n",
        "against event-measured periods for the actual parameter range.\n",
        "\n",
        "## Energy and turning points\n",
        "\n",
        "Without damping or drive,\n",
        "\n",
        "\n",
        "$$\n",
        "E=\\frac12mv^2+\\frac12kx^2+\\frac14\\beta x^4\n",
        "$$\n",
        "\n",
        "\n",
        "At a turning point, velocity is zero, so energy determines amplitude. Period can be\n",
        "written as an energy integral:\n",
        "\n",
        "\n",
        "$$\n",
        "T=\\sqrt{2m}\\int_{x_-}^{x_+}\\frac{dx}{\\sqrt{E-U(x)}}\n",
        "$$\n",
        "\n",
        "\n",
        "with appropriate doubling or full-cycle limits depending on symmetry. Unlike the\n",
        "harmonic case, the integral generally depends on energy and hence amplitude.\n",
        "\n",
        "The phase-space trajectory remains on a constant-energy contour for an ideal conservative\n",
        "system, but it is not generally an ellipse. Comparing numerical energy with the analytic\n",
        "potential is a powerful check of the force sign and integrator.\n",
        "\n",
        "## The nonlinear pendulum\n",
        "\n",
        "A simple pendulum obeys\n",
        "\n",
        "\n",
        "$$\n",
        "\\ddot\\theta+\\frac{g}{L}\\sin\\theta=0\n",
        "$$\n",
        "\n",
        "\n",
        "The approximation\n",
        "$\\sin\\theta\\approx\\theta$\n",
        "produces harmonic motion with\n",
        "$\\omega_0=\\sqrt{g/L}$.\n",
        "At larger amplitude, the exact restoring torque is weaker than the linear extrapolation\n",
        "near the extremes and the period grows.\n",
        "\n",
        "For modest amplitude\n",
        "$\\theta_0$ in radians,\n",
        "\n",
        "\n",
        "$$\n",
        "T\\approx2\\pi\\sqrt{\\frac{L}{g}}\\left(1+\\frac{\\theta_0^2}{16}+\\cdots\\right)\n",
        "$$\n",
        "\n",
        "\n",
        "The exact finite-amplitude period involves an elliptic integral and diverges as the\n",
        "release approaches the upright separatrix from below.\n",
        "\n",
        "## Measuring period reproducibly\n",
        "\n",
        "Visual peak selection is imprecise. Define an event such as consecutive positive-\n",
        "direction equilibrium crossings. The time between such crossings is one period. Locate\n",
        "the crossing within a time-step bracket rather than assigning it to the nearest sample."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-005",
      "metadata": {},
      "source": [
        "x, v, dt, crossings = 1.0, 0.0, 0.001, []\n",
        "for step in range(20_000):\n",
        "    previous = x\n",
        "    v += (-x - 0.5 * x**3) * dt\n",
        "    x += v * dt\n",
        "    if previous < 0 <= x and v > 0:\n",
        "        fraction = -previous / (x - previous)\n",
        "        crossings.append((step + fraction) * dt)\n",
        "print(\"Periods from upward crossings (s):\", np.diff(crossings))\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-006",
      "metadata": {},
      "source": [
        "Measure several cycles after any transient and report mean and variation. Repeat at half\n",
        "the time step. Change amplitude without changing solver settings so physical amplitude\n",
        "dependence is not confused with numerical phase error.\n",
        "\n",
        "## Superposition no longer works\n",
        "\n",
        "For a linear equation, if two functions are solutions, their sum is also a solution under\n",
        "corresponding combined initial data or forcing. The cubic term breaks this property:\n",
        "\n",
        "\n",
        "$$\n",
        "(x_1+x_2)^3\\ne x_1^3+x_2^3\n",
        "$$\n",
        "\n",
        "\n",
        "A multi-frequency drive can therefore create harmonics and combination frequencies.\n",
        "Fourier analysis remains useful for describing output, but output frequencies need not\n",
        "match only the input frequencies.\n",
        "\n",
        "## Damping and drive\n",
        "\n",
        "A driven damped nonlinear oscillator can be written\n",
        "\n",
        "\n",
        "$$\n",
        "m\\ddot x+c\\dot x+kx+\\beta x^3=F_0\\cos(\\omega t)\n",
        "$$\n",
        "\n",
        "\n",
        "Its instantaneous mechanical-energy rate is\n",
        "\n",
        "\n",
        "$$\n",
        "\\frac{dE}{dt}=-cv^2+F(t)v\n",
        "$$\n",
        "\n",
        "\n",
        "Drive can replace energy removed by damping. The settled response may contain harmonics,\n",
        "multiple coexisting periodic states, jumps, or chaos. Initial condition and sweep\n",
        "direction can select different attractors at identical parameters."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-007",
      "metadata": {},
      "source": [
        "position, velocity, time = 0.2, 0.1, 0.5\n",
        "spring = -position - 0.5 * position**3\n",
        "damping = -0.2 * velocity\n",
        "drive = 0.3 * np.cos(1.2 * time)\n",
        "print(\"Net force (N):\", spring + damping + drive)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-008",
      "metadata": {},
      "source": [
        "Do not compare a nonlinear frequency sweep with the linear amplitude formula except as a\n",
        "small-amplitude limiting test.\n",
        "\n",
        "## Multiple equilibria\n",
        "\n",
        "Changing signs can create a double-well potential. For example,\n",
        "\n",
        "\n",
        "$$\n",
        "U(x)=-\\frac12ax^2+\\frac14\\beta x^4\\qquad a>0,\\ \\beta>0\n",
        "$$\n",
        "\n",
        "\n",
        "has an unstable equilibrium at the origin and stable minima at\n",
        "\n",
        "\n",
        "$$\n",
        "x=\\pm\\sqrt{\\frac{a}{\\beta}}\n",
        "$$\n",
        "\n",
        "\n",
        "Low-energy motion stays in one well; energy above the central barrier can cross between\n",
        "wells. Damping, drive, and noise can produce transitions whose interpretation depends on\n",
        "the full energy and transfer model.\n",
        "\n",
        "## Numerical care\n",
        "\n",
        "Nonlinear force can become extremely large, making a previously adequate time step\n",
        "unstable. Estimate the highest local curvature over the visited amplitude range and\n",
        "refine adaptively or choose a robust method. Reject nonfinite state rather than drawing\n",
        "it.\n",
        "\n",
        "Run the zero-nonlinearity regression, force odd-symmetry tests where appropriate,\n",
        "finite-difference checks of $F=-dU/dx$,\n",
        "and conservative energy tests. Compare period, turning points, phase portrait, harmonics,\n",
        "and energy transfer under time-step and duration refinement.\n",
        "\n",
        "A surprising nonlinear response becomes credible only after simpler limiting cases pass\n",
        "and the same qualitative result persists under numerical refinement and plausible\n",
        "parameter uncertainty.\n",
        "\n",
        "## Nondimensional parameters\n",
        "\n",
        "Scaling makes it easier to compare different oscillators and identify which terms are\n",
        "important. Let displacement be measured relative to a characteristic amplitude\n",
        "$X$ and time relative to\n",
        "$1/\\omega_0$. Write\n",
        "$x=Xq$ and\n",
        "$\\tau=\\omega_0t$. The\n",
        "conservative cubic equation becomes\n",
        "\n",
        "\n",
        "$$\n",
        "\\frac{d^2q}{d\\tau^2}+q+\\alpha q^3=0\n",
        "$$\n",
        "\n",
        "\n",
        "with\n",
        "\n",
        "\n",
        "$$\n",
        "\\alpha=\\frac{\\beta X^2}{k}\n",
        "$$\n",
        "\n",
        "\n",
        "If magnitude of $\\alpha$ is\n",
        "much less than one over the motion range, the linear approximation should be accurate.\n",
        "If it is order one, nonlinear effects are expected. This comparison is more informative\n",
        "than calling a dimensional cubic coefficient “large” without a displacement scale.\n",
        "\n",
        "Nondimensionalization also improves numerical conditioning when physical values span\n",
        "many orders of magnitude. Convert output back to physical units only after solving and\n",
        "retain the scale definitions with the result.\n",
        "\n",
        "## Connecting a model to measurements\n",
        "\n",
        "Nonlinear coefficients should be inferred from force-displacement data, period-versus-\n",
        "amplitude data, or another explicit measurement. Fit the simplest range that supports\n",
        "the intended prediction. A cubic approximation that matches moderate motion may fail\n",
        "near contact, yielding, or geometric limits.\n",
        "\n",
        "Validate with data not used for fitting. For a conservative candidate, integrate the\n",
        "measured force to obtain potential and check whether observed turning-point speeds match\n",
        "energy conversion. For a damped model, compare decay across several amplitudes. For a\n",
        "driven model, compare both upward and downward frequency sweeps when multiple responses\n",
        "may exist.\n",
        "\n",
        "Parameter uncertainty can move bifurcation or jump locations substantially. Report a\n",
        "range of responses over plausible coefficients instead of extra digits in one curve.\n",
        "If two different force laws fit the available data equally well but predict different\n",
        "large-amplitude behavior, the limitation is model identifiability, not a problem that a\n",
        "smaller time step can solve.\n",
        "\n",
        "## Symmetry as a diagnostic\n",
        "\n",
        "An odd restoring force satisfies\n",
        "$F(-x)=-F(x)$ and produces a\n",
        "symmetric potential. With symmetric initial conditions, positive and negative turning\n",
        "points should have equal magnitude. A persistent asymmetry can indicate an even force\n",
        "term, shifted equilibrium, gravity bias, contact, or an implementation error.\n",
        "\n",
        "Test symmetry directly at force-function level before integrating. Then check trajectory\n",
        "symmetry under refined time steps. This separates a wrong force evaluation from an\n",
        "otherwise correct solver and makes the nonlinear behavior easier to explain.\n",
        "\n",
        "## A worked Python notebook\n",
        "\n",
        "Add a cubic stiffness term to a linear spring and compare local slopes. The question for this calculation is precise: how does **restoring force**\n",
        "change when we change **displacement**, 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",
        "This hardening-spring model is an approximation to a specified force law, not every nonlinear oscillator. 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",
        "-2 to 2 m. 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 displacement, expressed in m. The output is\n",
        "restoring force, expressed in N. 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) = - 0.5 q^{3} - q\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 m, the expression gives -1.5 N. 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(\"-q-0.5*q**3\", locals={\"q\": q})\n",
        "reference_input = 1.0\n",
        "reference_output = float(expression.subs(q, reference_input))\n",
        "print(\"Input:\", reference_input, \"m\")\n",
        "print(\"Predicted restoring force:\", reference_output, \"N\")\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",
        "6 N. The midpoint, at 0 m, gives\n",
        "0 N. The upper endpoint gives -6 N.\n",
        "Read these together with the assumptions: they describe different choices of displacement\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([-2.0, 0.0, 2.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} m -> {output_value:.6g} 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 displacement; the vertical axis shows restoring force. 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(-2.0, 2.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(\"Displacement (m)\")\n",
        "ax.set_ylabel(\"Restoring force (N)\")\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",
        "restoring force change near the reference input? Its units are the output unit divided by\n",
        "the input unit. For this calculation, the derivative at 1 m is\n",
        "-2.5 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) = - 1.5 q^{2} - 1\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.02\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, \"N\")\n",
        "print(\"Full expression:\", full_prediction, \"N\")\n",
        "print(\"Approximation difference:\", full_prediction - linear_prediction, \"N\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-016",
      "metadata": {},
      "source": [
        "Here the changed input is 1.02 m. The tangent estimate is\n",
        "-1.55 N, while the full expression gives\n",
        "-1.5506 N. 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 = 4.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-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",
        "displacement inside the stated interval, predict whether restoring force 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: This hardening-spring model is an approximation to a specified force law, not every nonlinear oscillator. 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 N:\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(2.0) - evaluate(-2.0))\n",
        "for points in [101, 201, 401]:\n",
        "    grid = np.linspace(-2.0, 2.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 15.4, \"Pendulums\"](https://openstax.org/books/university-physics-volume-1/pages/15-4-pendulums).\n",
        "- [MIT OpenCourseWare, *8.01SC Classical Mechanics*, Chapter 23, Sections 23.3 and 23.7](https://ocw.mit.edu/courses/8-01sc-classical-mechanics-fall-2016/resources/mit8_01scs22_chapter23/).\n",
        "- [Benjamin Crowell, *Light and Matter*, nonlinear pendulum and oscillation discussions](https://open.umn.edu/opentextbooks/textbooks/light-and-matter).\n",
        "- [Physics LibreTexts, \"The Nonlinear Pendulum\"](https://phys.libretexts.org/Bookshelves/Classical_Mechanics/Graduate_Classical_Mechanics_%28Fowler%29/04%3A_Oscillations/4.04%3A_The_Nonlinear_Pendulum).\n",
        "- [The Physics Hypertextbook, \"Pendulum\"](https://physics.info/pendulum/)."
      ]
    }
  ],
  "metadata": {
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "name": "python",
      "version": "3.12"
    },
    "colab": {
      "name": "nonlinear-oscillators.ipynb",
      "provenance": []
    },
    "tirta": {
      "lesson": "https://tirta.co/a-byte-of-physics/nonlinear-oscillators",
      "source_sha256": "3b8c341b0fd098b571697caca8cf4255449cdfa13f61d0c4c41fb38608e3f43f"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 5
}
