{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "ddd4c056",
   "metadata": {},
   "source": [
    "# Landing Congestion Analysis\n",
    "\n",
    "Explore the impact of landing shock events on stochastic playback outputs for the synthetic medium scenario."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "16d5fc8e",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:57:04.800215Z",
     "iopub.status.busy": "2026-10-05T12:57:04.800096Z",
     "iopub.status.idle": "2026-10-05T12:57:07.907173Z",
     "shell.execute_reply": "2026-10-05T12:57:07.906310Z"
    }
   },
   "outputs": [],
   "source": [
    "import sys\n",
    "from pathlib import Path\n",
    "\n",
    "PROJECT_ROOT = Path.cwd().resolve()\n",
    "while PROJECT_ROOT != PROJECT_ROOT.parent and not (PROJECT_ROOT / \"pyproject.toml\").exists():\n",
    "    PROJECT_ROOT = PROJECT_ROOT.parent\n",
    "if not (PROJECT_ROOT / \"pyproject.toml\").exists():\n",
    "    raise RuntimeError(\n",
    "        \"Notebook must be executed within a FHOPS checkout (pyproject.toml not found).\"\n",
    "    )\n",
    "if str(PROJECT_ROOT) not in sys.path:\n",
    "    sys.path.insert(0, str(PROJECT_ROOT))\n",
    "\n",
    "from docs.examples.analytics import utils\n",
    "from fhops.scenario.synthetic import SyntheticDatasetConfig, sampling_config_for\n",
    "\n",
    "SCENARIO = PROJECT_ROOT / \"examples/synthetic/medium/scenario.yaml\"\n",
    "ASSIGNMENTS = PROJECT_ROOT / \"docs/examples/analytics/data/synthetic_medium_sa_assignments.csv\"\n",
    "\n",
    "config = SyntheticDatasetConfig(\n",
    "    name=\"synthetic-medium\",\n",
    "    tier=\"medium\",\n",
    "    num_blocks=8,\n",
    "    num_days=12,\n",
    "    num_machines=4,\n",
    ")\n",
    "\n",
    "baseline_sampling = sampling_config_for(config).model_copy()\n",
    "baseline_sampling.landing.enabled = False\n",
    "baseline_tables, baseline_sampling = utils.run_stochastic_summary(\n",
    "    SCENARIO, ASSIGNMENTS, sampling_config=baseline_sampling\n",
    ")\n",
    "\n",
    "shock_sampling = baseline_sampling.model_copy()\n",
    "shock_sampling.landing.enabled = True\n",
    "shock_sampling.landing.probability = 0.3\n",
    "shock_sampling.landing.capacity_multiplier_range = (0.35, 0.7)\n",
    "shock_sampling.landing.duration_days = 3\n",
    "shock_tables, shock_sampling = utils.run_stochastic_summary(\n",
    "    SCENARIO, ASSIGNMENTS, sampling_config=shock_sampling\n",
    ")"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "45420d70",
   "metadata": {},
   "source": [
    "## Baseline Ensemble Snapshot"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "9dcd3bd4",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:57:07.909164Z",
     "iopub.status.busy": "2026-10-05T12:57:07.908997Z",
     "iopub.status.idle": "2026-10-05T12:57:07.918025Z",
     "shell.execute_reply": "2026-10-05T12:57:07.917449Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>day</th>\n",
       "      <th>shift_id</th>\n",
       "      <th>machine_id</th>\n",
       "      <th>machine_role</th>\n",
       "      <th>sample_id</th>\n",
       "      <th>production_units</th>\n",
       "      <th>total_hours</th>\n",
       "      <th>idle_hours</th>\n",
       "      <th>mobilisation_cost</th>\n",
       "      <th>sequencing_violations</th>\n",
       "      <th>blackout_conflicts</th>\n",
       "      <th>available_hours</th>\n",
       "      <th>utilisation_ratio</th>\n",
       "      <th>downtime_hours</th>\n",
       "      <th>downtime_events</th>\n",
       "      <th>weather_severity_total</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M4</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>0</td>\n",
       "      <td>5.1558</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>8</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>0</td>\n",
       "      <td>15.0340</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>1</td>\n",
       "      <td>12.6600</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>8</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>1</td>\n",
       "      <td>9.0204</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M4</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>2</td>\n",
       "      <td>8.5930</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   day shift_id machine_id machine_role  sample_id  production_units  \\\n",
       "0    2       S1         M4    forwarder          0            5.1558   \n",
       "1    8       S1         M1    harvester          0           15.0340   \n",
       "2    2       S1         M1    harvester          1           12.6600   \n",
       "3    8       S1         M1    harvester          1            9.0204   \n",
       "4    2       S1         M4    forwarder          2            8.5930   \n",
       "\n",
       "   total_hours  idle_hours  mobilisation_cost  sequencing_violations  \\\n",
       "0          8.0         0.0                0.0                      0   \n",
       "1          8.0         0.0                0.0                      0   \n",
       "2          8.0         0.0                0.0                      0   \n",
       "3          8.0         0.0                0.0                      0   \n",
       "4          8.0         0.0                0.0                      0   \n",
       "\n",
       "   blackout_conflicts  available_hours  utilisation_ratio  downtime_hours  \\\n",
       "0                   0              8.0                1.0             0.0   \n",
       "1                   0              8.0                1.0             0.0   \n",
       "2                   0              8.0                1.0             0.0   \n",
       "3                   0              8.0                1.0             0.0   \n",
       "4                   0              8.0                1.0             0.0   \n",
       "\n",
       "   downtime_events  weather_severity_total  \n",
       "0                0                     0.4  \n",
       "1                0                     0.0  \n",
       "2                0                     0.0  \n",
       "3                0                     0.4  \n",
       "4                0                     0.0  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "baseline_tables.shift.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "12a88345",
   "metadata": {},
   "source": [
    "## Landing Shock Ensemble Snapshot"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "97c7830c",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:57:07.919473Z",
     "iopub.status.busy": "2026-10-05T12:57:07.919359Z",
     "iopub.status.idle": "2026-10-05T12:57:07.925913Z",
     "shell.execute_reply": "2026-10-05T12:57:07.925369Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>day</th>\n",
       "      <th>shift_id</th>\n",
       "      <th>machine_id</th>\n",
       "      <th>machine_role</th>\n",
       "      <th>sample_id</th>\n",
       "      <th>production_units</th>\n",
       "      <th>total_hours</th>\n",
       "      <th>idle_hours</th>\n",
       "      <th>mobilisation_cost</th>\n",
       "      <th>sequencing_violations</th>\n",
       "      <th>blackout_conflicts</th>\n",
       "      <th>available_hours</th>\n",
       "      <th>utilisation_ratio</th>\n",
       "      <th>downtime_hours</th>\n",
       "      <th>downtime_events</th>\n",
       "      <th>weather_severity_total</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M4</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>0</td>\n",
       "      <td>5.15580</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>8</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>0</td>\n",
       "      <td>15.03400</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>1</td>\n",
       "      <td>12.66000</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>8</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>1</td>\n",
       "      <td>9.02040</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M4</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>2</td>\n",
       "      <td>4.84338</td>\n",
       "      <td>8.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   day shift_id machine_id machine_role  sample_id  production_units  \\\n",
       "0    2       S1         M4    forwarder          0           5.15580   \n",
       "1    8       S1         M1    harvester          0          15.03400   \n",
       "2    2       S1         M1    harvester          1          12.66000   \n",
       "3    8       S1         M1    harvester          1           9.02040   \n",
       "4    2       S1         M4    forwarder          2           4.84338   \n",
       "\n",
       "   total_hours  idle_hours  mobilisation_cost  sequencing_violations  \\\n",
       "0          8.0         0.0                0.0                      0   \n",
       "1          8.0         0.0                0.0                      0   \n",
       "2          8.0         0.0                0.0                      0   \n",
       "3          8.0         0.0                0.0                      0   \n",
       "4          8.0         0.0                0.0                      0   \n",
       "\n",
       "   blackout_conflicts  available_hours  utilisation_ratio  downtime_hours  \\\n",
       "0                   0              8.0                1.0             0.0   \n",
       "1                   0              8.0                1.0             0.0   \n",
       "2                   0              8.0                1.0             0.0   \n",
       "3                   0              8.0                1.0             0.0   \n",
       "4                   0              8.0                1.0             0.0   \n",
       "\n",
       "   downtime_events  weather_severity_total  \n",
       "0                0                     0.4  \n",
       "1                0                     0.0  \n",
       "2                0                     0.0  \n",
       "3                0                     0.4  \n",
       "4                0                     0.0  "
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "shock_tables.shift.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "6854bc62",
   "metadata": {},
   "source": [
    "## KPI Comparison"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "fb6f580b",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:57:07.927189Z",
     "iopub.status.busy": "2026-10-05T12:57:07.927067Z",
     "iopub.status.idle": "2026-10-05T12:57:07.933625Z",
     "shell.execute_reply": "2026-10-05T12:57:07.933024Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>scenario</th>\n",
       "      <th>mean_production</th>\n",
       "      <th>mean_downtime_hours</th>\n",
       "      <th>mean_utilisation</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>baseline</td>\n",
       "      <td>20.266083</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>landing_shock</td>\n",
       "      <td>19.929846</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        scenario  mean_production  mean_downtime_hours  mean_utilisation\n",
       "0       baseline        20.266083                  0.0               1.0\n",
       "1  landing_shock        19.929846                  0.0               1.0"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import pandas as pd\n",
    "\n",
    "comparison = pd.DataFrame(\n",
    "    {\n",
    "        \"scenario\": [\"baseline\", \"landing_shock\"],\n",
    "        \"mean_production\": [\n",
    "            baseline_tables.shift.groupby(\"sample_id\")[\"production_units\"].sum().mean(),\n",
    "            shock_tables.shift.groupby(\"sample_id\")[\"production_units\"].sum().mean(),\n",
    "        ],\n",
    "        \"mean_downtime_hours\": [\n",
    "            baseline_tables.shift[\"downtime_hours\"].mean(),\n",
    "            shock_tables.shift[\"downtime_hours\"].mean(),\n",
    "        ],\n",
    "        \"mean_utilisation\": [\n",
    "            baseline_tables.shift[\"utilisation_ratio\"].mean(),\n",
    "            shock_tables.shift[\"utilisation_ratio\"].mean(),\n",
    "        ],\n",
    "    }\n",
    ")\n",
    "comparison"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "4ab289eb",
   "metadata": {},
   "source": [
    "## Production Distributions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "e87408e5",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:57:07.934944Z",
     "iopub.status.busy": "2026-10-05T12:57:07.934832Z",
     "iopub.status.idle": "2026-10-05T12:57:07.949748Z",
     "shell.execute_reply": "2026-10-05T12:57:07.949282Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "<style>\n",
       "  #altair-viz-cea9d92688e44a209270f9e7c17567b3.vega-embed {\n",
       "    width: 100%;\n",
       "    display: flex;\n",
       "  }\n",
       "\n",
       "  #altair-viz-cea9d92688e44a209270f9e7c17567b3.vega-embed details,\n",
       "  #altair-viz-cea9d92688e44a209270f9e7c17567b3.vega-embed details summary {\n",
       "    position: relative;\n",
       "  }\n",
       "</style>\n",
       "<div id=\"altair-viz-cea9d92688e44a209270f9e7c17567b3\"></div>\n",
       "<script type=\"text/javascript\">\n",
       "  var VEGA_DEBUG = (typeof VEGA_DEBUG == \"undefined\") ? {} : VEGA_DEBUG;\n",
       "  (function(spec, embedOpt){\n",
       "    let outputDiv = document.currentScript.previousElementSibling;\n",
       "    if (outputDiv.id !== \"altair-viz-cea9d92688e44a209270f9e7c17567b3\") {\n",
       "      outputDiv = document.getElementById(\"altair-viz-cea9d92688e44a209270f9e7c17567b3\");\n",
       "    }\n",
       "\n",
       "    const paths = {\n",
       "      \"vega\": \"https://cdn.jsdelivr.net/npm/vega@6?noext\",\n",
       "      \"vega-lib\": \"https://cdn.jsdelivr.net/npm/vega-lib?noext\",\n",
       "      \"vega-lite\": \"https://cdn.jsdelivr.net/npm/vega-lite@6.4.1?noext\",\n",
       "      \"vega-embed\": \"https://cdn.jsdelivr.net/npm/vega-embed@7?noext\",\n",
       "    };\n",
       "\n",
       "    function maybeLoadScript(lib, version) {\n",
       "      var key = `${lib.replace(\"-\", \"\")}_version`;\n",
       "      return (VEGA_DEBUG[key] == version) ?\n",
       "        Promise.resolve(paths[lib]) :\n",
       "        new Promise(function(resolve, reject) {\n",
       "          var s = document.createElement('script');\n",
       "          document.getElementsByTagName(\"head\")[0].appendChild(s);\n",
       "          s.async = true;\n",
       "          s.onload = () => {\n",
       "            VEGA_DEBUG[key] = version;\n",
       "            return resolve(paths[lib]);\n",
       "          };\n",
       "          s.onerror = () => reject(`Error loading script: ${paths[lib]}`);\n",
       "          s.src = paths[lib];\n",
       "        });\n",
       "    }\n",
       "\n",
       "    function showError(err) {\n",
       "      outputDiv.innerHTML = `<div class=\"error\" style=\"color:red;\">${err}</div>`;\n",
       "      throw err;\n",
       "    }\n",
       "\n",
       "    function displayChart(vegaEmbed) {\n",
       "      vegaEmbed(outputDiv, spec, embedOpt)\n",
       "        .catch(err => showError(`Javascript Error: ${err.message}<br>This usually means there's a typo in your chart specification. See the javascript console for the full traceback.`));\n",
       "    }\n",
       "\n",
       "    if(typeof define === \"function\" && define.amd) {\n",
       "      requirejs.config({paths});\n",
       "      let deps = [\"vega-embed\"];\n",
       "      require(deps, displayChart, err => showError(`Error loading script: ${err.message}`));\n",
       "    } else {\n",
       "      maybeLoadScript(\"vega\", \"6\")\n",
       "        .then(() => maybeLoadScript(\"vega-lite\", \"6.4.1\"))\n",
       "        .then(() => maybeLoadScript(\"vega-embed\", \"7\"))\n",
       "        .catch(showError)\n",
       "        .then(() => displayChart(vegaEmbed));\n",
       "    }\n",
       "  })({\"config\": {\"view\": {\"continuousWidth\": 300, \"continuousHeight\": 300}}, \"data\": {\"name\": \"data-e334ffe63eeda7f73d7c347c47cfa83c\"}, \"mark\": {\"type\": \"bar\", \"opacity\": 0.75}, \"encoding\": {\"x\": {\"bin\": {\"maxbins\": 20}, \"field\": \"Production Units\", \"title\": \"Production Units\", \"type\": \"quantitative\"}, \"y\": {\"aggregate\": \"count\", \"title\": \"Frequency\", \"type\": \"quantitative\"}}, \"height\": 240, \"title\": \"Baseline Production Distribution\", \"width\": 400, \"$schema\": \"https://vega.github.io/schema/vega-lite/v6.4.1.json\", \"datasets\": {\"data-e334ffe63eeda7f73d7c347c47cfa83c\": [{\"Production Units\": 20.1898}, {\"Production Units\": 21.6804}, {\"Production Units\": 23.627000000000002}, {\"Production Units\": 11.738800000000001}, {\"Production Units\": 11.647}, {\"Production Units\": 27.694000000000003}, {\"Production Units\": 14.179}, {\"Production Units\": 14.179}, {\"Production Units\": 23.627000000000002}, {\"Production Units\": 19.243000000000002}, {\"Production Units\": 27.694000000000003}, {\"Production Units\": 27.694000000000003}]}}, {\"mode\": \"vega-lite\"});\n",
       "</script>"
      ],
      "text/plain": [
       "alt.Chart(...)"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "baseline_prod = baseline_tables.shift.groupby(\"sample_id\")[\"production_units\"].sum()\n",
    "baseline_chart = utils.plot_distribution(\n",
    "    baseline_prod, title=\"Baseline Production Distribution\", xlabel=\"Production Units\"\n",
    ")\n",
    "baseline_chart"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "bafa8bd1",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:57:07.951323Z",
     "iopub.status.busy": "2026-10-05T12:57:07.951181Z",
     "iopub.status.idle": "2026-10-05T12:57:07.957959Z",
     "shell.execute_reply": "2026-10-05T12:57:07.957278Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "<style>\n",
       "  #altair-viz-7fc5502345bd4afabbb92f0081ebd279.vega-embed {\n",
       "    width: 100%;\n",
       "    display: flex;\n",
       "  }\n",
       "\n",
       "  #altair-viz-7fc5502345bd4afabbb92f0081ebd279.vega-embed details,\n",
       "  #altair-viz-7fc5502345bd4afabbb92f0081ebd279.vega-embed details summary {\n",
       "    position: relative;\n",
       "  }\n",
       "</style>\n",
       "<div id=\"altair-viz-7fc5502345bd4afabbb92f0081ebd279\"></div>\n",
       "<script type=\"text/javascript\">\n",
       "  var VEGA_DEBUG = (typeof VEGA_DEBUG == \"undefined\") ? {} : VEGA_DEBUG;\n",
       "  (function(spec, embedOpt){\n",
       "    let outputDiv = document.currentScript.previousElementSibling;\n",
       "    if (outputDiv.id !== \"altair-viz-7fc5502345bd4afabbb92f0081ebd279\") {\n",
       "      outputDiv = document.getElementById(\"altair-viz-7fc5502345bd4afabbb92f0081ebd279\");\n",
       "    }\n",
       "\n",
       "    const paths = {\n",
       "      \"vega\": \"https://cdn.jsdelivr.net/npm/vega@6?noext\",\n",
       "      \"vega-lib\": \"https://cdn.jsdelivr.net/npm/vega-lib?noext\",\n",
       "      \"vega-lite\": \"https://cdn.jsdelivr.net/npm/vega-lite@6.4.1?noext\",\n",
       "      \"vega-embed\": \"https://cdn.jsdelivr.net/npm/vega-embed@7?noext\",\n",
       "    };\n",
       "\n",
       "    function maybeLoadScript(lib, version) {\n",
       "      var key = `${lib.replace(\"-\", \"\")}_version`;\n",
       "      return (VEGA_DEBUG[key] == version) ?\n",
       "        Promise.resolve(paths[lib]) :\n",
       "        new Promise(function(resolve, reject) {\n",
       "          var s = document.createElement('script');\n",
       "          document.getElementsByTagName(\"head\")[0].appendChild(s);\n",
       "          s.async = true;\n",
       "          s.onload = () => {\n",
       "            VEGA_DEBUG[key] = version;\n",
       "            return resolve(paths[lib]);\n",
       "          };\n",
       "          s.onerror = () => reject(`Error loading script: ${paths[lib]}`);\n",
       "          s.src = paths[lib];\n",
       "        });\n",
       "    }\n",
       "\n",
       "    function showError(err) {\n",
       "      outputDiv.innerHTML = `<div class=\"error\" style=\"color:red;\">${err}</div>`;\n",
       "      throw err;\n",
       "    }\n",
       "\n",
       "    function displayChart(vegaEmbed) {\n",
       "      vegaEmbed(outputDiv, spec, embedOpt)\n",
       "        .catch(err => showError(`Javascript Error: ${err.message}<br>This usually means there's a typo in your chart specification. See the javascript console for the full traceback.`));\n",
       "    }\n",
       "\n",
       "    if(typeof define === \"function\" && define.amd) {\n",
       "      requirejs.config({paths});\n",
       "      let deps = [\"vega-embed\"];\n",
       "      require(deps, displayChart, err => showError(`Error loading script: ${err.message}`));\n",
       "    } else {\n",
       "      maybeLoadScript(\"vega\", \"6\")\n",
       "        .then(() => maybeLoadScript(\"vega-lite\", \"6.4.1\"))\n",
       "        .then(() => maybeLoadScript(\"vega-embed\", \"7\"))\n",
       "        .catch(showError)\n",
       "        .then(() => displayChart(vegaEmbed));\n",
       "    }\n",
       "  })({\"config\": {\"view\": {\"continuousWidth\": 300, \"continuousHeight\": 300}}, \"data\": {\"name\": \"data-93de22a7456338a283aaa3a7fde9f212\"}, \"mark\": {\"type\": \"bar\", \"opacity\": 0.75}, \"encoding\": {\"x\": {\"bin\": {\"maxbins\": 20}, \"field\": \"Production Units\", \"title\": \"Production Units\", \"type\": \"quantitative\"}, \"y\": {\"aggregate\": \"count\", \"title\": \"Frequency\", \"type\": \"quantitative\"}}, \"height\": 240, \"title\": \"Landing Shock Production Distribution\", \"width\": 400, \"$schema\": \"https://vega.github.io/schema/vega-lite/v6.4.1.json\", \"datasets\": {\"data-93de22a7456338a283aaa3a7fde9f212\": [{\"Production Units\": 20.1898}, {\"Production Units\": 21.6804}, {\"Production Units\": 19.877379983200207}, {\"Production Units\": 11.738800000000001}, {\"Production Units\": 15.218273612992983}, {\"Production Units\": 27.694000000000003}, {\"Production Units\": 14.179}, {\"Production Units\": 14.179}, {\"Production Units\": 23.627000000000002}, {\"Production Units\": 15.386498313559676}, {\"Production Units\": 27.694000000000003}, {\"Production Units\": 27.694000000000003}]}}, {\"mode\": \"vega-lite\"});\n",
       "</script>"
      ],
      "text/plain": [
       "alt.Chart(...)"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "shock_prod = shock_tables.shift.groupby(\"sample_id\")[\"production_units\"].sum()\n",
    "shock_chart = utils.plot_distribution(\n",
    "    shock_prod, title=\"Landing Shock Production Distribution\", xlabel=\"Production Units\"\n",
    ")\n",
    "shock_chart"
   ]
  }
 ],
 "metadata": {
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}