Add plotting to stiffened plate example

smdogroup · Jul 12, 2024 · 2617a02 · 2617a02
1 parent 73e5d48
commit 2617a02
Show file tree

Hide file tree

Showing 2 changed files with 192 additions and 52 deletions.
diff --git a/examples/stiffened_plate/generate_plate.py b/examples/stiffened_plate/generate_plate.py
@@ -6,11 +6,15 @@
 from scipy.spatial.transform import Rotation
 
 parser = argparse.ArgumentParser()
-parser.add_argument("--nex", type=int, default=20, help="# elements from top to bottom")
+parser.add_argument(
+    "--nex", type=int, default=20, help="# elements from top to bottom"
+)
 parser.add_argument(
     "--ney", type=int, default=40, help="# elements around circumference"
 )
-parser.add_argument("--pinEdges", action="store_true", help="Pin edges of plate")
+parser.add_argument(
+    "--pinEdges", action="store_true", help="Pin edges of plate"
+)
 args = parser.parse_args()
 
 # Overall dimensions
@@ -124,7 +128,9 @@ def write_bulk_line(key, items, format="small"):
 
 # Write nodes
 for i in range(n1 * n2):
-    write_bulk_line("GRID", [i + 1, 0, nmat[i, 0], nmat[i, 1], nmat[i, 2], 0, 0, 0])
+    write_bulk_line(
+        "GRID", [i + 1, 0, nmat[i, 0], nmat[i, 1], nmat[i, 2], 0, 0, 0]
+    )
 
 # Write elements
 compID = 1