{
  "cells": [
    {
      "cell_type": "markdown",
      "id": "markdown-000",
      "metadata": {},
      "source": [
        "# Driven systems and resonance\n",
        "\n",
        "[Read the lesson](https://tirta.co/a-byte-of-physics/driven-systems-resonance). 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 driven oscillator receives energy from an external time-dependent force. Its response\n",
        "depends on drive frequency, natural frequency, damping, and what observable is measured.\n",
        "Resonance is a frequency-response phenomenon, not simply the largest displacement seen\n",
        "during startup.\n",
        "\n",
        "## The driven damped equation\n",
        "\n",
        "For a sinusoidal force,\n",
        "\n",
        "\n",
        "$$\n",
        "m\\ddot x+c\\dot x+kx=F_0\\cos(\\omega t)\n",
        "$$\n",
        "\n",
        "\n",
        "The mass is $m$, stiffness is\n",
        "$k$, damping coefficient is\n",
        "$c$, and drive amplitude is\n",
        "$F_0$. Drive angular\n",
        "frequency $\\omega$ is\n",
        "in radians per second. Ordinary frequency is\n",
        "\n",
        "\n",
        "$$\n",
        "f=\\frac{\\omega}{2\\pi}\n",
        "$$\n",
        "\n",
        "\n",
        "Confusing hertz with radians per second shifts a sweep by a factor of\n",
        "$2\\pi$.\n",
        "\n",
        "The state equations are\n",
        "\n",
        "\n",
        "$$\n",
        "\\dot x=v\\qquad\\dot v=\\frac{F_0\\cos(\\omega t)-cv-kx}{m}\n",
        "$$"
      ]
    },
    {
      "cell_type": "code",
      "id": "code-003",
      "metadata": {},
      "source": [
        "import numpy as np\n",
        "\n",
        "mass, stiffness, damping, amplitude, frequency = 1.0, 4.0, 0.4, 1.0, 2.0\n",
        "time, position, velocity = 0.5, 0.2, 0.1\n",
        "drive_force = amplitude * np.cos(frequency * time)\n",
        "acceleration = (drive_force - damping * velocity - stiffness * position) / mass\n",
        "print(\"Acceleration (m/s^2):\", acceleration)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-004",
      "metadata": {},
      "source": [
        "Drive phase matters during transients and for timestamped comparisons. In the settled\n",
        "linear response, changing only the arbitrary time origin does not change amplitude.\n",
        "\n",
        "## Transient plus steady response\n",
        "\n",
        "The complete solution is the sum of a homogeneous transient and a forced steady response:\n",
        "\n",
        "\n",
        "$$\n",
        "x(t)=x_{\\mathrm{transient}}(t)+x_{\\mathrm{steady}}(t)\n",
        "$$\n",
        "\n",
        "\n",
        "With positive damping, the transient decays. The steady displacement is sinusoidal at the\n",
        "drive frequency, not the natural frequency:\n",
        "\n",
        "\n",
        "$$\n",
        "x_{\\mathrm{steady}}(t)=A(\\omega)\\cos(\\omega t-\\delta)\n",
        "$$\n",
        "\n",
        "\n",
        "A startup maximum can include constructive addition of transient and steady components.\n",
        "It should not be used as the settled response without a documented reason.\n",
        "\n",
        "## Steady displacement amplitude\n",
        "\n",
        "Substitution into the equation gives\n",
        "\n",
        "\n",
        "$$\n",
        "A(\\omega)=\\frac{F_0}{\\sqrt{(k-m\\omega^2)^2+(c\\omega)^2}}\n",
        "$$\n",
        "\n",
        "\n",
        "At very low frequency, the response approaches the static displacement\n",
        "\n",
        "\n",
        "$$\n",
        "A(0)=\\frac{F_0}{k}\n",
        "$$\n",
        "\n",
        "\n",
        "At very high frequency, inertia dominates and amplitude falls approximately as\n",
        "$F_0/(m\\omega^2)$.\n",
        "Damping keeps the response finite near the natural frequency."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-005",
      "metadata": {},
      "source": [
        "def response(omega):\n",
        "    return amplitude / np.hypot(stiffness - mass * omega**2, damping * omega)\n",
        "\n",
        "\n",
        "print(f\"Steady amplitude: {response(frequency):.3f} m\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-006",
      "metadata": {},
      "source": [
        "This analytic formula is an excellent pointwise test of a linear numerical solver after\n",
        "transients are removed.\n",
        "\n",
        "## Phase lag\n",
        "\n",
        "The phase lag obeys a quadrant-aware relation\n",
        "\n",
        "\n",
        "$$\n",
        "\\delta=\\operatorname{atan2}(c\\omega,\\,k-m\\omega^2)\n",
        "$$\n",
        "\n",
        "\n",
        "At low frequency displacement is nearly in phase with force. Near the undamped natural\n",
        "frequency it lags by about one quarter cycle. At high frequency it approaches opposition\n",
        "to the force. Use a two-argument arctangent so the phase stays in the correct quadrant.\n",
        "\n",
        "Velocity leads displacement by one quarter cycle, so force-velocity phase controls\n",
        "average input power.\n",
        "\n",
        "## Which frequency is resonant?\n",
        "\n",
        "The undamped natural frequency is\n",
        "\n",
        "\n",
        "$$\n",
        "\\omega_0=\\sqrt{\\frac{k}{m}}\n",
        "$$\n",
        "\n",
        "\n",
        "For sufficiently light viscous damping, the displacement-amplitude maximum occurs at\n",
        "\n",
        "\n",
        "$$\n",
        "\\omega_{\\mathrm r}=\\sqrt{\\omega_0^2-\\frac{c^2}{2m^2}}=\\omega_0\\sqrt{1-2\\zeta^2}\n",
        "$$\n",
        "\n",
        "\n",
        "where damping ratio is\n",
        "$\\zeta=c/(2\\sqrt{mk})$. This\n",
        "real nonzero maximum exists only when damping is small enough. It lies below\n",
        "$\\omega_0$.\n",
        "\n",
        "Other observables have different peaks. Velocity amplitude and average absorbed power\n",
        "peak at $\\omega_0$ for this\n",
        "ideal force-driven linear model. Always say whether “resonance frequency” refers to\n",
        "displacement, velocity, acceleration, or power.\n",
        "\n",
        "## Average power and energy balance\n",
        "\n",
        "Instantaneous power supplied by the drive is\n",
        "\n",
        "\n",
        "$$\n",
        "P_{\\mathrm{in}}(t)=F(t)\\dot x(t)\n",
        "$$\n",
        "\n",
        "\n",
        "and damping removes mechanical energy at rate\n",
        "\n",
        "\n",
        "$$\n",
        "P_{\\mathrm d}(t)=-c\\dot x^2(t)\n",
        "$$\n",
        "\n",
        "\n",
        "In settled periodic motion, average mechanical energy no longer grows, so average input\n",
        "power equals average damping loss in magnitude:\n",
        "\n",
        "\n",
        "$$\n",
        "\\langle P_{\\mathrm{in}}\\rangle=\\frac12c\\omega^2A^2\n",
        "$$"
      ]
    },
    {
      "cell_type": "code",
      "id": "code-007",
      "metadata": {},
      "source": [
        "period = 2 * np.pi / frequency\n",
        "times = np.linspace(0, period, 2001)\n",
        "phase = np.arctan2(damping * frequency, stiffness - mass * frequency**2)\n",
        "velocity = -response(frequency) * frequency * np.sin(frequency * times - phase)\n",
        "drive_work = np.trapezoid(amplitude * np.cos(frequency * times) * velocity, times)\n",
        "damping_transfer = np.trapezoid(damping * velocity**2, times)\n",
        "assert np.isclose(drive_work, damping_transfer)\n",
        "print(\"Drive work and dissipated energy per cycle (J):\", drive_work, damping_transfer)\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-008",
      "metadata": {},
      "source": [
        "Over an integer number of settled cycles, drive work and damping transfer should agree\n",
        "within integration error.\n",
        "\n",
        "## Quality factor and bandwidth\n",
        "\n",
        "For light damping, quality factor is\n",
        "\n",
        "\n",
        "$$\n",
        "Q=\\frac{m\\omega_0}{c}=\\frac{1}{2\\zeta}\n",
        "$$\n",
        "\n",
        "\n",
        "A large $Q$ gives a tall, narrow\n",
        "resonance and a long transient decay. The approximate half-power bandwidth is\n",
        "\n",
        "\n",
        "$$\n",
        "\\Delta\\omega\\approx\\frac{\\omega_0}{Q}=\\frac{c}{m}\n",
        "$$\n",
        "\n",
        "\n",
        "These bandwidth relations are approximations for light damping. A narrow peak requires a\n",
        "fine frequency grid and long settling time, so high quality factor makes numerical\n",
        "measurement more demanding.\n",
        "\n",
        "## Build a reproducible frequency sweep\n",
        "\n",
        "At each drive frequency, define whether the oscillator restarts from the same initial\n",
        "state or continues from the previous frequency. Restarted sweeps isolate the linear\n",
        "frequency response when they run long enough. Continued sweeps can reveal hysteresis in\n",
        "nonlinear systems but make history part of the input."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-009",
      "metadata": {},
      "source": [
        "frequencies = np.linspace(0.1, 4.0, 100)\n",
        "amplitudes = response(frequencies)\n",
        "peak = np.argmax(amplitudes)\n",
        "print(\"Largest sampled response:\", frequencies[peak], amplitudes[peak])\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-010",
      "metadata": {},
      "source": [
        "Use the same estimator everywhere. Half peak-to-peak amplitude works for a clean settled\n",
        "sinusoid. A least-squares sine-cosine fit at the known drive frequency gives amplitude\n",
        "and phase and can tolerate sampling that misses exact extrema. RMS must be converted\n",
        "carefully if comparing with peak amplitude.\n",
        "\n",
        "## Decide when the transient is gone\n",
        "\n",
        "A fixed number of periods may be inadequate near a narrow resonance because the free\n",
        "transient decays on a damping timescale. Compare amplitude and phase in successive\n",
        "windows. Declare settled only when both change less than tolerances for enough cycles."
      ]
    },
    {
      "cell_type": "code",
      "id": "code-011",
      "metadata": {},
      "source": [
        "# Compare estimates over complete cycles of the analytic steady response.\n",
        "for cycles in [1, 2, 4]:\n",
        "    times = np.linspace(0, cycles * period, cycles * 1000 + 1)\n",
        "    signal = response(frequency) * np.cos(frequency * times - phase)\n",
        "    design = np.column_stack([np.cos(frequency * times), np.sin(frequency * times)])\n",
        "    coefficients = np.linalg.lstsq(design, signal, rcond=None)[0]\n",
        "    print(cycles, \"fitted amplitude:\", np.linalg.norm(coefficients))\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-012",
      "metadata": {},
      "source": [
        "If the duration limit is reached, report the point as unsettled instead of silently using\n",
        "a startup peak.\n",
        "\n",
        "## Numerical and physical checks\n",
        "\n",
        "Refine both time step and frequency spacing. A small time step does not prevent a coarse\n",
        "sweep from missing a narrow maximum, and a dense sweep does not repair an inaccurate\n",
        "integrator. Compare selected frequencies with the analytic amplitude and phase.\n",
        "\n",
        "Zero drive must recover free damped motion. Zero damping at exact resonance has no finite\n",
        "settled amplitude in the linear ideal model; amplitude grows with time, so the steady\n",
        "formula's denominator vanishes. Any finite plateau there comes from finite duration,\n",
        "nonlinearity, numerical damping, or another omitted effect.\n",
        "\n",
        "## Beyond the linear model\n",
        "\n",
        "Nonlinear stiffness can shift resonance with amplitude, produce harmonics, multiple\n",
        "steady responses, jumps, and sweep-direction hysteresis. Coupled systems have several\n",
        "normal modes. Real actuators may supply displacement rather than force or have their own\n",
        "frequency response.\n",
        "\n",
        "Large motion can change damping, geometry, and material behavior. A frequency response is\n",
        "therefore tied to drive amplitude, boundary conditions, measurement definition, and sweep\n",
        "history. A reliable resonance claim includes all of them plus convergence and energy\n",
        "evidence.\n",
        "\n",
        "## Reading a response curve\n",
        "\n",
        "A response curve should show more than a connected line through a few samples. Mark the\n",
        "actual drive frequencies, distinguish settled from unsettled points, and include the\n",
        "amplitude definition. If a peak lies between sampled frequencies, the largest sampled\n",
        "point is only a lower bound on the true maximum under the continuous model. Refine around\n",
        "the candidate peak or fit the justified analytic response.\n",
        "\n",
        "At frequencies well below resonance, stiffness dominates and displacement follows the\n",
        "force almost quasistatically. Through the resonance region, response grows and phase\n",
        "changes rapidly. Well above resonance, inertia dominates, displacement becomes small,\n",
        "and it is nearly opposite the applied force. Seeing all three regions is a useful\n",
        "qualitative check on sweep range and units.\n",
        "\n",
        "Plotting phase with amplitude helps diagnose mistakes. A supposed linear settled response\n",
        "that shows a discontinuous arbitrary phase jump may have used a one-argument inverse\n",
        "tangent or inconsistent wrapping. A response containing strong harmonics may indicate\n",
        "nonlinearity, clipping, or incomplete transient removal. Inspect the time trace and its\n",
        "frequency content instead of forcing every result into a single sinusoidal amplitude.\n",
        "\n",
        "## Sensitivity and uncertainty\n",
        "\n",
        "Uncertainty in mass and stiffness changes natural frequency, while uncertainty in damping\n",
        "strongly affects peak height and width. Near a sharp resonance, a small frequency error\n",
        "can create a large amplitude error. Report input uncertainty and sweep resolution rather\n",
        "than presenting the peak frequency with more digits than the experiment or grid supports.\n",
        "\n",
        "Measurement duration also controls frequency resolution. A short window cannot cleanly\n",
        "distinguish nearby frequencies and may capture a beat between transient and drive. Use an\n",
        "integer or sufficiently long number of drive cycles where possible, retain timestamps,\n",
        "and disclose windowing or fitting choices.\n",
        "\n",
        "If the driving device is coupled to the oscillator, its force may fall as response grows.\n",
        "Treating measured actuator command as a constant applied force can then exaggerate the\n",
        "ideal response. Measure force at the system boundary or include actuator dynamics. The\n",
        "same caution applies to sensors whose calibration or bandwidth changes near the peak.\n",
        "\n",
        "Finally, compare energy per cycle. A high amplitude without corresponding drive work is\n",
        "an accounting warning. In settled motion the oscillator returns to the same mechanical\n",
        "energy after each cycle, so all net drive work over that cycle must be transferred by\n",
        "damping or another explicitly modeled loss.\n",
        "\n",
        "## A worked Python notebook\n",
        "\n",
        "Drive a one kilogram oscillator with stiffness 4 and damping coefficient 0.4 using a one newton sinusoid. The question for this calculation is precise: how does **steady displacement amplitude**\n",
        "change when we change **drive angular frequency**, 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 expression describes the steady response after transients decay. 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.1 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 drive angular frequency, expressed in rad/s. The output is\n",
        "steady displacement amplitude, expressed in m. 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{1}{\\sqrt{0.16 q^{2} + \\left(4 - q^{2}\\right)^{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 2 rad/s, the expression gives 1.25 m. 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-013",
      "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(\"1/sqrt((4-q**2)**2+(0.4*q)**2)\", 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 steady displacement amplitude:\", reference_output, \"m\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-014",
      "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",
        "0.250614 m. The midpoint, at 2.05 rad/s, gives\n",
        "1.18395 m. The upper endpoint gives 0.0826023 m.\n",
        "Read these together with the assumptions: they describe different choices of drive angular frequency\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-015",
      "metadata": {},
      "source": [
        "evaluate = sp.lambdify(q, expression, modules=\"numpy\")\n",
        "inputs = np.array([0.1, 2.05, 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} m\")\n",
        "assert np.all(np.isfinite(outputs))\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-016",
      "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 drive angular frequency; the vertical axis shows steady displacement amplitude. 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-017",
      "metadata": {},
      "source": [
        "input_grid = np.linspace(0.1, 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(\"Drive angular frequency (rad/s)\")\n",
        "ax.set_ylabel(\"Steady displacement amplitude (m)\")\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-018",
      "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",
        "steady displacement amplitude 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",
        "-0.625 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{2 q \\left(4 - q^{2}\\right) - 0.16 q}{\\left(0.16 q^{2} + \\left(4 - q^{2}\\right)^{2}\\right)^{\\frac{3}{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-019",
      "metadata": {},
      "source": [
        "slope_expression = sp.diff(expression, q)\n",
        "slope_at_reference = float(slope_expression.subs(q, reference_input))\n",
        "changed_input = 2.0195\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, \"m\")\n",
        "print(\"Full expression:\", full_prediction, \"m\")\n",
        "print(\"Approximation difference:\", full_prediction - linear_prediction, \"m\")\n"
      ],
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "id": "markdown-020",
      "metadata": {},
      "source": [
        "Here the changed input is 2.0195 rad/s. The tangent estimate is\n",
        "1.23781 m, while the full expression gives\n",
        "1.23214 m. 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-021",
      "metadata": {},
      "source": [
        "input_span = 3.9\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-022",
      "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",
        "drive angular frequency inside the stated interval, predict whether steady displacement amplitude 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 expression describes the steady response after transients decay. 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 m:\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-023",
      "metadata": {},
      "source": [
        "slope_function = sp.lambdify(q, slope_expression, modules=\"numpy\")\n",
        "endpoint_change = float(evaluate(4.0) - evaluate(0.1))\n",
        "for points in [101, 201, 401]:\n",
        "    grid = np.linspace(0.1, 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-024",
      "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.6, \"Forced Oscillations\"](https://openstax.org/books/university-physics-volume-1/pages/15-6-forced-oscillations).\n",
        "- [MIT OpenCourseWare, *8.01SC Classical Mechanics*, Chapter 23, Section 23.6, \"Forced Damped Oscillator\"](https://ocw.mit.edu/courses/8-01sc-classical-mechanics-fall-2016/resources/mit8_01scs22_chapter23/).\n",
        "- [Benjamin Crowell, *Light and Matter*, resonance and driven-oscillation discussions](https://open.umn.edu/opentextbooks/textbooks/light-and-matter).\n",
        "- [The Physics Hypertextbook, \"Resonance\"](https://physics.info/resonance/).\n",
        "- [Physics LibreTexts, \"Forced Oscillations\"](https://phys.libretexts.org/Bookshelves/University_Physics/University_Physics_%28OpenStax%29/Book%3A_University_Physics_I_-_Mechanics_Sound_Oscillations_and_Waves_%28OpenStax%29/15%3A_Oscillations/15.07%3A_Forced_Oscillations)."
      ]
    }
  ],
  "metadata": {
    "kernelspec": {
      "display_name": "Python 3",
      "language": "python",
      "name": "python3"
    },
    "language_info": {
      "name": "python",
      "version": "3.12"
    },
    "colab": {
      "name": "driven-systems-resonance.ipynb",
      "provenance": []
    },
    "tirta": {
      "lesson": "https://tirta.co/a-byte-of-physics/driven-systems-resonance",
      "source_sha256": "c46ceebc27b92b62b1c3bf684f069d1fccbe4012d462dceefc6b069a65c69153"
    }
  },
  "nbformat": 4,
  "nbformat_minor": 5
}
