{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "31dfbc3e",
   "metadata": {},
   "source": [
    "# Stochastic Robustness Explorer\n",
    "\n",
    "Analyse stochastic playback ensembles for the synthetic medium bundle using preset sampling configurations."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "8a93a5a4",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:56:54.209515Z",
     "iopub.status.busy": "2026-10-05T12:56:54.209379Z",
     "iopub.status.idle": "2026-10-05T12:56:57.142656Z",
     "shell.execute_reply": "2026-10-05T12:56:57.141884Z"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Samples: 12\n"
     ]
    },
    {
     "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>1</td>\n",
       "      <td>S1</td>\n",
       "      <td>M2</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>0</td>\n",
       "      <td>14.589</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.1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M2</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>0</td>\n",
       "      <td>6.484</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.6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>8</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>0</td>\n",
       "      <td>15.034</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>1</td>\n",
       "      <td>S1</td>\n",
       "      <td>M1</td>\n",
       "      <td>harvester</td>\n",
       "      <td>1</td>\n",
       "      <td>9.251</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>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2</td>\n",
       "      <td>S1</td>\n",
       "      <td>M2</td>\n",
       "      <td>forwarder</td>\n",
       "      <td>1</td>\n",
       "      <td>16.210</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    1       S1         M2    forwarder          0            14.589   \n",
       "1    2       S1         M2    forwarder          0             6.484   \n",
       "2    8       S1         M1    harvester          0            15.034   \n",
       "3    1       S1         M1    harvester          1             9.251   \n",
       "4    2       S1         M2    forwarder          1            16.210   \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              0.0                NaN             0.0   \n",
       "4                   0              8.0                1.0             0.0   \n",
       "\n",
       "   downtime_events  weather_severity_total  \n",
       "0                0                     0.1  \n",
       "1                0                     0.6  \n",
       "2                0                     0.0  \n",
       "3                0                     0.0  \n",
       "4                0                     0.0  "
      ]
     },
     "execution_count": 1,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "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",
    "\n",
    "SCENARIO = PROJECT_ROOT / \"examples/synthetic/medium/scenario.yaml\"\n",
    "ASSIGNMENTS = (\n",
    "    PROJECT_ROOT / \"docs/examples/analytics/data/scaling_medium_sa_assignments.csv\"\n",
    ")  # update with fresh assignments if desired\n",
    "\n",
    "(tables, sampling_config) = utils.run_stochastic_summary(SCENARIO, ASSIGNMENTS, tier=\"medium\")\n",
    "print(f\"Samples: {sampling_config.samples}\")\n",
    "tables.shift.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "bde45154",
   "metadata": {},
   "source": [
    "### Production Distribution\n",
    "\n",
    "Inspect the distribution of production across ensemble samples."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "137462ef",
   "metadata": {
    "execution": {
     "iopub.execute_input": "2026-10-05T12:56:57.144399Z",
     "iopub.status.busy": "2026-10-05T12:56:57.144209Z",
     "iopub.status.idle": "2026-10-05T12:56:57.161225Z",
     "shell.execute_reply": "2026-10-05T12:56:57.160415Z"
    }
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "<style>\n",
       "  #altair-viz-cf17c0d525c5419a8e8fbc7adbd6a950.vega-embed {\n",
       "    width: 100%;\n",
       "    display: flex;\n",
       "  }\n",
       "\n",
       "  #altair-viz-cf17c0d525c5419a8e8fbc7adbd6a950.vega-embed details,\n",
       "  #altair-viz-cf17c0d525c5419a8e8fbc7adbd6a950.vega-embed details summary {\n",
       "    position: relative;\n",
       "  }\n",
       "</style>\n",
       "<div id=\"altair-viz-cf17c0d525c5419a8e8fbc7adbd6a950\"></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-cf17c0d525c5419a8e8fbc7adbd6a950\") {\n",
       "      outputDiv = document.getElementById(\"altair-viz-cf17c0d525c5419a8e8fbc7adbd6a950\");\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-2e00d6ca209105fb9aad2bca08c3b2fc\"}, \"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\": \"Production Distribution\", \"width\": 400, \"$schema\": \"https://vega.github.io/schema/vega-lite/v6.4.1.json\", \"datasets\": {\"data-2e00d6ca209105fb9aad2bca08c3b2fc\": [{\"Production Units\": 36.107}, {\"Production Units\": 34.4814}, {\"Production Units\": 29.277}, {\"Production Units\": 20.260019338274923}, {\"Production Units\": 30.33595018644585}, {\"Production Units\": 40.495}, {\"Production Units\": 24.427}, {\"Production Units\": 19.835743414603776}, {\"Production Units\": 29.277}, {\"Production Units\": 24.427}, {\"Production Units\": 32.044}, {\"Production Units\": 32.044}]}}, {\"mode\": \"vega-lite\"});\n",
       "</script>"
      ],
      "text/plain": [
       "alt.Chart(...)"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "utils.plot_distribution(\n",
    "    tables.shift.groupby(\"sample_id\")[\"production_units\"].sum(),\n",
    "    title=\"Production Distribution\",\n",
    "    xlabel=\"Production Units\",\n",
    ")"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": ".venv",
   "language": "python",
   "name": "python3"
  },
  "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
}