R8-a harness: RTX_E3FSI_LOAD_PLANES (the load route on the n mid planes; cost knob, whole span by default)

Co-Authored-By: Claude Opus 5.5 (1M context) <[email protected]>
This commit is contained in:
Omar Sobh
2026-09-25 18:14:55 -05:00
co-authored by Claude Opus 5.5
parent 9867a8b831
commit 0f386df871
2 changed files with 26 additions and 11 deletions
@@ -301,7 +301,7 @@ fn fsi2_on_embedded3() {
if (step + 1) % 50 == 0 || step + 1 == rigid_steps { if (step + 1) % 50 == 0 || step + 1 == rigid_steps {
last = fl.loads(); last = fl.loads();
let (tot, contrib) = &last; let (tot, contrib) = &last;
let (_, on_flag, on_cyl, _) = distribute(&flag_geo, &rest, contrib, fl.width); let (_, on_flag, on_cyl, _) = distribute(&flag_geo, &rest, contrib, fl.load_width);
let t = fl.time(); let t = fl.time();
if let Some(f) = csv.as_mut() { if let Some(f) = csv.as_mut() {
writeln!( writeln!(
@@ -405,7 +405,7 @@ fn fsi2_on_embedded3() {
(state, nodal, s.line.clone(), s.c_fluid.clone()) (state, nodal, s.line.clone(), s.c_fluid.clone())
} }
None => { None => {
let (nodal0, _, _, _) = distribute(&flag_geo, &rest, &contrib0, fl.width); let (nodal0, _, _, _) = distribute(&flag_geo, &rest, &contrib0, fl.load_width);
flag.borrow_mut().set_nodal_forces(&nodal0); flag.borrow_mut().set_nodal_forces(&nodal0);
let state = flag.borrow_mut().rest_state().unwrap(); let state = flag.borrow_mut().rest_state().unwrap();
// The fluid's own previous line and centreline (its geometry's history). // The fluid's own previous line and centreline (its geometry's history).
@@ -467,7 +467,8 @@ fn fsi2_on_embedded3() {
} }
let tl = std::time::Instant::now(); let tl = std::time::Instant::now();
let (tot, contrib) = f.loads(); let (tot, contrib) = f.loads();
let (nodal, on_flag, on_cyl, parts_y) = distribute(&flag_geo, &line, &contrib, f.width); let (nodal, on_flag, on_cyl, parts_y) =
distribute(&flag_geo, &line, &contrib, f.load_width);
t_loads.set(t_loads.get() + tl.elapsed().as_secs_f64()); t_loads.set(t_loads.get() + tl.elapsed().as_secs_f64());
let ts = std::time::Instant::now(); let ts = std::time::Instant::now();
let mut st = flag.borrow_mut(); let mut st = flag.borrow_mut();
@@ -144,8 +144,10 @@ pub struct E3Fluid {
pub grid: Grid, pub grid: Grid,
pub h: f64, pub h: f64,
pub dt: f64, pub dt: f64,
/// The z extent the loads are divided by (per unit span). /// The duct's z extent.
pub width: f64, pub width: f64,
/// The z extent the last `loads` integrated over (per unit span).
pub load_width: f64,
pub lines: Arc<RwLock<Lines>>, pub lines: Arc<RwLock<Lines>>,
sink: Arc<Mutex<Vec<Contribution>>>, sink: Arc<Mutex<Vec<Contribution>>>,
} }
@@ -284,6 +286,7 @@ impl E3Fluid {
h, h,
dt, dt,
width, width,
load_width: width,
lines, lines,
sink: Arc::new(Mutex::new(Vec::new())), sink: Arc::new(Mutex::new(Vec::new())),
} }
@@ -324,14 +327,25 @@ impl E3Fluid {
assert!(previous.is_none(), "a load sink was already installed"); assert!(previous.is_none(), "a load sink was already installed");
let mask = self.device.solver.mask().expect("mask"); let mask = self.device.solver.mask().expect("mask");
let body = self.device.solver.body().expect("body"); let body = self.device.solver.body().expect("body");
// `RTX_E3FSI_LOAD_PLANES=n`: the route on the n mid planes only (per
// unit span of those planes) — a cost knob for a spanwise-uniform
// flow; the whole span by default.
let nz = self.grid.nz;
let n = (super::env_f("RTX_E3FSI_LOAD_PLANES", 0.0) as usize).min(nz);
let f = if n > 0 && n < nz {
let k0 = (nz - n) / 2;
self.load_width = n as f64 * self.h;
mask.cut_wall_force_per_span(body, &self.field, RHO * NU, t, (k0, k0 + n))
.expect("cut wall force per span")
} else {
self.load_width = self.width;
let f = mask let f = mask
.cut_wall_force(body, &self.field, RHO * NU, t) .cut_wall_force(body, &self.field, RHO * NU, t)
.expect("cut wall force"); .expect("cut wall force");
[f[0] / self.width, f[1] / self.width, f[2] / self.width]
};
set_load_sink(None); set_load_sink(None);
let contributions = std::mem::take(&mut *self.sink.lock().expect("sink")); let contributions = std::mem::take(&mut *self.sink.lock().expect("sink"));
( (f, contributions)
[f[0] / self.width, f[1] / self.width, f[2] / self.width],
contributions,
)
} }
} }