merge and update lag case files to generate their own input file

Author: wilfonba

examples/2D_lagrange_in_crossflow/case.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import json
 import math
+import os
 
 # Bubble screen
 # Description: A planar acoustic wave interacts with a bubble cloud
@@ -54,6 +55,27 @@
 Nx = 240  # number of elements into x direction
 Ny = 50  # number of elements into y direction
 
+# Lagrangian bubble initial conditions
+# Columns: x, y, z, u, v, w, R, pad (all nondimensional; lengths in units of x0)
+lag_bubbles = [
+    (-10.0, -2.0, 0.0, 0.0, 0.0, 0.0, 0.1000, 0.0),
+    (-10.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.1500, 0.0),
+    (-10.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.1500, 0.0),
+    (-10.0, 1.0, 0.0, 0.0, 0.0, 0.0, 0.1500, 0.0),
+    (-10.0, 2.0, 0.0, 0.0, 0.0, 0.0, 0.1009, 0.0),
+]
+
+
+def write_lag_bubbles_file():
+    input_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "input")
+    os.makedirs(input_dir, exist_ok=True)
+    with open(os.path.join(input_dir, "lag_bubbles.dat"), "w") as f:
+        for row in lag_bubbles:
+            f.write("".join(f"{v:15.6E}" for v in row) + "\n")
+
+
+write_lag_bubbles_file()
+
 # Configuring case dictionary
 print(
     json.dumps(
@@ -117,7 +139,7 @@
             "thermal": 3,
             "polytropic": "F",
             "fd_order": 2,
-            "lag_params%nBubs_glb": 5,
+            "lag_params%nBubs_glb": len(lag_bubbles),
             "lag_params%vel_model": 2,
             "lag_params%drag_model": 3,
             "lag_params%solver_approach": 1,

examples/2D_lagrange_in_crossflow/input/lag_bubbles.dat

@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 import json
 import math
+import os
 
 # Bubble screen
 # Description: A planar acoustic wave interacts with a bubble cloud
@@ -53,6 +54,22 @@
 Nx = 50  # number of elements into x direction
 Ny = 50  # number of elements into y direction
 
+# Lagrangian bubble seeding: 5 bubbles along y=-2 with varying radii
+lag_bubbles = [
+    # x,   y,    z,   u, v, w,   R,     interface velocity
+    (-2.0, -2.0, 0.0, 0.0, 0.0, 0.0, 0.100, 0.0),
+    (-1.0, -2.0, 0.0, 0.0, 0.0, 0.0, 0.125, 0.0),
+    (0.0, -2.0, 0.0, 0.0, 0.0, 0.0, 0.150, 0.0),
+    (1.0, -2.0, 0.0, 0.0, 0.0, 0.0, 0.125, 0.0),
+    (2.0, -2.0, 0.0, 0.0, 0.0, 0.0, 0.075, 0.0),
+]
+
+input_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "input")
+os.makedirs(input_dir, exist_ok=True)
+with open(os.path.join(input_dir, "lag_bubbles.dat"), "w") as f:
+    for row in lag_bubbles:
+        f.write("\t".join(f"{v: .6E}" for v in row) + "\n")
+
 # Configuring case dictionary
 print(
     json.dumps(

examples/2D_lagrange_rising_bubble/case.py

@@ -2,6 +2,7 @@
 import argparse
 import json
 import math
+import os
 
 import numpy as np
 
@@ -199,4 +200,48 @@
 
 mods = {}
 
+
+def write_lag_bubbles_file():
+    nBubs = 2000
+    r_inner = 0.25 / L0
+    r_outer = 0.50 / L0
+    R_min, R_max = 0.5, 1.0
+    z_half = 5.0  # virtual z depth (must satisfy |z| < lag_params%charwidth/2)
+    max_attempts = 10000
+
+    rng = np.random.default_rng(seed=42)
+    x_pos = np.zeros(nBubs)
+    y_pos = np.zeros(nBubs)
+    z_pos = np.zeros(nBubs)
+    R_bub = np.zeros(nBubs)
+
+    for n in range(nBubs):
+        for _ in range(max_attempts):
+            u = rng.uniform(0.0, 1.0)
+            r = math.sqrt(r_inner**2 + u * (r_outer**2 - r_inner**2))
+            theta = rng.uniform(0.0, 2.0 * math.pi)
+            xi = r * math.cos(theta)
+            yi = r * math.sin(theta)
+            zi = rng.uniform(-z_half, z_half)
+            Ri = rng.uniform(R_min, R_max)
+            if n == 0:
+                break
+            dx = x_pos[:n] - xi
+            dy = y_pos[:n] - yi
+            dist = np.sqrt(dx * dx + dy * dy)
+            if np.all(dist >= 2 * (R_bub[:n] + Ri)):
+                break
+        else:
+            raise RuntimeError(f"Could not place bubble {n} without overlap after {max_attempts} attempts")
+        x_pos[n], y_pos[n], z_pos[n], R_bub[n] = xi, yi, zi, Ri
+
+    input_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "input")
+    os.makedirs(input_dir, exist_ok=True)
+    with open(os.path.join(input_dir, "lag_bubbles.dat"), "w") as f:
+        for i in range(nBubs):
+            f.write(f"{x_pos[i]:12.6f}\t{y_pos[i]:12.6f}\t{z_pos[i]:12.6f}\t{0.0:12.6f}\t{0.0:12.6f}\t{0.0:12.6f}\t{R_bub[i]:12.6f}\t{0.0:12.6f}\n")
+
+
+write_lag_bubbles_file()
+
 print(json.dumps({**data, **mods}, indent=4))