Ruff format pylauncher notebook

Author: kks32

examples/pylauncher_sweep.ipynb

@@ -193,7 +193,87 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": "input_dir_opensees = MYDATA / \"opensees_sweep\"\ninput_dir_opensees.mkdir(parents=True, exist_ok=True)\n\ncantilever_script = '''\\\n# Ex1a.Canti2D.Push — OpenSeesPy cantilever pushover\n# Based on the OpenSees Examples Manual\n# Units: kip, inch, second\n#\n# Command-line arguments (set by PyLauncher per task):\n#   --NodalMass  mass at free node\n#   --LCol       column length\n#   --outDir     output directory for this run\n\nimport argparse\nimport os\n\nif os.path.exists(\"opensees.so\"):\n    import opensees as ops\nelse:\n    import openseespy.opensees as ops\n\nparser = argparse.ArgumentParser()\nparser.add_argument(\"--NodalMass\", type=float, required=True)\nparser.add_argument(\"--LCol\", type=float, required=True)\nparser.add_argument(\"--outDir\", type=str, required=True)\nargs = parser.parse_args()\n\nNodalMass = args.NodalMass\nLCol = args.LCol\noutDir = args.outDir\n\nos.makedirs(outDir, exist_ok=True)\nprint(f\"Running: NodalMass={NodalMass}, LCol={LCol}, outDir={outDir}\")\n\nops.wipe()\nops.model(\"basic\", \"-ndm\", 2, \"-ndf\", 3)\n\n# Geometry\nops.node(1, 0, 0)\nops.node(2, 0, LCol)\nops.fix(1, 1, 1, 1)\nops.mass(2, NodalMass, 0.0, 0.0)\n\n# Element\nops.geomTransf(\"Linear\", 1)\nops.element(\"elasticBeamColumn\", 1, 1, 2, 3600000000, 4227, 1080000, 1)\n\n# Recorders\nops.recorder(\"Node\", \"-file\", f\"{outDir}/DFree.out\", \"-time\", \"-node\", 2, \"-dof\", 1, 2, 3, \"disp\")\nops.recorder(\"Node\", \"-file\", f\"{outDir}/RBase.out\", \"-time\", \"-node\", 1, \"-dof\", 1, 2, 3, \"reaction\")\nops.recorder(\"Element\", \"-file\", f\"{outDir}/FCol.out\", \"-time\", \"-ele\", 1, \"globalForce\")\n\n# Gravity analysis\nops.timeSeries(\"Linear\", 1)\nops.pattern(\"Plain\", 1, 1)\nops.load(2, 0.0, -2000.0, 0.0)\nops.wipeAnalysis()\nops.constraints(\"Plain\")\nops.numberer(\"Plain\")\nops.system(\"BandGeneral\")\nops.test(\"NormDispIncr\", 1.0e-8, 6)\nops.algorithm(\"Newton\")\nops.integrator(\"LoadControl\", 0.1)\nops.analysis(\"Static\")\nops.analyze(10)\nops.loadConst(\"-time\", 0.0)\n\n# Pushover analysis\nops.timeSeries(\"Linear\", 2)\nops.pattern(\"Plain\", 2, 2)\nops.load(2, 2000.0, 0.0, 0.0)\nops.integrator(\"DisplacementControl\", 2, 1, 0.1)\nops.analyze(1000)\n\nprint(f\"Done: NodalMass={NodalMass}, LCol={LCol}\")\n'''\n\n(input_dir_opensees / \"cantilever.py\").write_text(cantilever_script)\nprint(f\"Wrote {input_dir_opensees}/cantilever.py\")"
+   "source": [
+    "input_dir_opensees = MYDATA / \"opensees_sweep\"\n",
+    "input_dir_opensees.mkdir(parents=True, exist_ok=True)\n",
+    "\n",
+    "cantilever_script = \"\"\"\\\n",
+    "# Ex1a.Canti2D.Push — OpenSeesPy cantilever pushover\n",
+    "# Based on the OpenSees Examples Manual\n",
+    "# Units: kip, inch, second\n",
+    "#\n",
+    "# Command-line arguments (set by PyLauncher per task):\n",
+    "#   --NodalMass  mass at free node\n",
+    "#   --LCol       column length\n",
+    "#   --outDir     output directory for this run\n",
+    "\n",
+    "import argparse\n",
+    "import os\n",
+    "\n",
+    "if os.path.exists(\"opensees.so\"):\n",
+    "    import opensees as ops\n",
+    "else:\n",
+    "    import openseespy.opensees as ops\n",
+    "\n",
+    "parser = argparse.ArgumentParser()\n",
+    "parser.add_argument(\"--NodalMass\", type=float, required=True)\n",
+    "parser.add_argument(\"--LCol\", type=float, required=True)\n",
+    "parser.add_argument(\"--outDir\", type=str, required=True)\n",
+    "args = parser.parse_args()\n",
+    "\n",
+    "NodalMass = args.NodalMass\n",
+    "LCol = args.LCol\n",
+    "outDir = args.outDir\n",
+    "\n",
+    "os.makedirs(outDir, exist_ok=True)\n",
+    "print(f\"Running: NodalMass={NodalMass}, LCol={LCol}, outDir={outDir}\")\n",
+    "\n",
+    "ops.wipe()\n",
+    "ops.model(\"basic\", \"-ndm\", 2, \"-ndf\", 3)\n",
+    "\n",
+    "# Geometry\n",
+    "ops.node(1, 0, 0)\n",
+    "ops.node(2, 0, LCol)\n",
+    "ops.fix(1, 1, 1, 1)\n",
+    "ops.mass(2, NodalMass, 0.0, 0.0)\n",
+    "\n",
+    "# Element\n",
+    "ops.geomTransf(\"Linear\", 1)\n",
+    "ops.element(\"elasticBeamColumn\", 1, 1, 2, 3600000000, 4227, 1080000, 1)\n",
+    "\n",
+    "# Recorders\n",
+    "ops.recorder(\"Node\", \"-file\", f\"{outDir}/DFree.out\", \"-time\", \"-node\", 2, \"-dof\", 1, 2, 3, \"disp\")\n",
+    "ops.recorder(\"Node\", \"-file\", f\"{outDir}/RBase.out\", \"-time\", \"-node\", 1, \"-dof\", 1, 2, 3, \"reaction\")\n",
+    "ops.recorder(\"Element\", \"-file\", f\"{outDir}/FCol.out\", \"-time\", \"-ele\", 1, \"globalForce\")\n",
+    "\n",
+    "# Gravity analysis\n",
+    "ops.timeSeries(\"Linear\", 1)\n",
+    "ops.pattern(\"Plain\", 1, 1)\n",
+    "ops.load(2, 0.0, -2000.0, 0.0)\n",
+    "ops.wipeAnalysis()\n",
+    "ops.constraints(\"Plain\")\n",
+    "ops.numberer(\"Plain\")\n",
+    "ops.system(\"BandGeneral\")\n",
+    "ops.test(\"NormDispIncr\", 1.0e-8, 6)\n",
+    "ops.algorithm(\"Newton\")\n",
+    "ops.integrator(\"LoadControl\", 0.1)\n",
+    "ops.analysis(\"Static\")\n",
+    "ops.analyze(10)\n",
+    "ops.loadConst(\"-time\", 0.0)\n",
+    "\n",
+    "# Pushover analysis\n",
+    "ops.timeSeries(\"Linear\", 2)\n",
+    "ops.pattern(\"Plain\", 2, 2)\n",
+    "ops.load(2, 2000.0, 0.0, 0.0)\n",
+    "ops.integrator(\"DisplacementControl\", 2, 1, 0.1)\n",
+    "ops.analyze(1000)\n",
+    "\n",
+    "print(f\"Done: NodalMass={NodalMass}, LCol={LCol}\")\n",
+    "\"\"\"\n",
+    "\n",
+    "(input_dir_opensees / \"cantilever.py\").write_text(cantilever_script)\n",
+    "print(f\"Wrote {input_dir_opensees}/cantilever.py\")"
+   ]
   },
   {
    "cell_type": "markdown",