embedded3 S2-7b: the curved instrument's per-wall operator probe (curved_operator_probe) and RTX_E3_CURVED_SCHEME; host prototype RTX_E3_WALL_FOOT=centroid (the shear closure's wall velocity at the foot from the open part's CENTROID, not the face centre — on a rotating wall the two feet are ½h(1−α) apart: rigid-mode probe residual 4.5× smaller, the rotating wall's offset −0.022 → ±0.002 h, its cut pressure zeroth → first order); default off, device refuses it

Co-Authored-By: Claude Fable 5.1 <[email protected]>
2026-09-20 08:48:41 -05:00
parent 599023d786
commit 7fffbb1301
6 changed files with 197 additions and 6 deletions
@@ -10,7 +10,7 @@
 //! (`u = Ω r e_θ`, `p = ρ Ω² r²/2`).
 use rtx_cfd::solvers::incompressible::ConvectionScheme;
 use rtx_cfd::solvers::incompressible::embedded3::{
-    Body, Boundaries, Field, Fluid, Grid, Parameters, Side, Solver, WallScheme,
+    Body, Boundaries, FaceKind, Field, Fluid, Grid, Parameters, Side, Solver, WallScheme,
 };

 const MU: f64 = 0.1;
@@ -77,7 +77,11 @@ fn parameters() -> Parameters {
    Parameters {
        corrector_steps: 2,
        tolerance: 1e-10,
-        convection_scheme: ConvectionScheme::TvdVanAlbada,
+        convection_scheme: if std::env::var("RTX_E3_CURVED_SCHEME").is_ok_and(|v| v == "upwind") {
+            ConvectionScheme::Upwind
+        } else {
+            ConvectionScheme::TvdVanAlbada
+        },
        wall_scheme: WallScheme::CutCell,
        boundaries: Boundaries {
            z0: Side::Periodic,
@@ -326,3 +330,168 @@ fn curved_wall_effective_position_and_pressure() {
        reading(n, rigid);
    }
 }
+
+/// S2-7b B1: the operator probe on this instrument — one predictor and one
+/// corrector from the exact field at the faces' open-part centroids (exact
+/// p at the cells); the predictor's residual per face, recovered from the
+/// one correction, in units of the exact field's largest acceleration
+/// (Ω² R2), by aperture band and by WALL (inner / outer); the correction's
+/// pressure at each wall's cut cells (of ρ(ΩR1)²).
+fn probe(n: usize, rigid: bool) {
+    let h = 1.0 / n as f64;
+    let (nx, ny, nz) = ((LX * n as f64) as usize, (LX * n as f64) as usize, 2);
+    let ex = Exact::new(rigid);
+    let mut params = parameters();
+    params.corrector_steps = 1;
+    let mut solver = Solver::new(
+        Fluid {
+            density: RHO,
+            viscosity: MU,
+            reference_velocity: OMEGA * R1,
+            reference_length: R2 - R1,
+        },
+        params,
+    );
+    let exb = Exact::new(rigid);
+    solver.set_boundary_velocity(move |x, y, _z, _t| {
+        let (u, v) = exb.velocity(x, y, rigid);
+        (u, v, 0.0)
+    });
+    solver.set_body(
+        Body::from_sdf(move |x, y, _z, _t| {
+            let r = ((x - CENTRE.0).powi(2) + (y - CENTRE.1).powi(2)).sqrt();
+            (r - R1).min(R2 - r)
+        })
+        .with_surface_velocity(move |x, y, _z, _t| {
+            let (dx, dy) = (x - CENTRE.0, y - CENTRE.1);
+            let r = (dx * dx + dy * dy).sqrt().max(1e-12);
+            let inner_side = r < 0.5 * (R1 + R2);
+            let moving = rigid || (inner_side != outer_drives());
+            let ut = if moving { OMEGA * r } else { 0.0 };
+            (-ut * dy / r, ut * dx / r, 0.0)
+        }),
+    );
+    let g = Grid::cubic(nx, ny, nz, h);
+    let mut field = Field::new(g);
+    solver.initialize(&mut field);
+    let tables = solver.mask().expect("mask").face_shift_tables().expect("shift tables").clone();
+    for k in 0..nz {
+        for j in 0..ny {
+            for i in 0..=nx {
+                let f = g.uface(k, j, i);
+                let t = &tables[0][3 * f..3 * f + 3];
+                field.u[f] = ex.velocity(i as f64 * h + t[0], (j as f64 + 0.5) * h + t[1], rigid).0;
+            }
+        }
+        for j in 0..=ny {
+            for i in 0..nx {
+                let f = g.vface(k, j, i);
+                let t = &tables[1][3 * f..3 * f + 3];
+                field.v[f] = ex.velocity((i as f64 + 0.5) * h + t[0], j as f64 * h + t[1], rigid).1;
+            }
+        }
+    }
+    let r_of = |x: f64, y: f64| ((x - CENTRE.0).powi(2) + (y - CENTRE.1).powi(2)).sqrt();
+    for idx in 0..g.cells() {
+        let (_, j, i) = g.kji(idx);
+        let r = r_of((i as f64 + 0.5) * h, (j as f64 + 0.5) * h).clamp(R1, R2);
+        field.p[idx] = ex.p(r);
+    }
+    {
+        let (body, mask) = (solver.body().expect("body"), solver.mask().expect("mask"));
+        mask.impose(body, &mut field.u, &mut field.v, &mut field.w, solver.time());
+    }
+    let dt = 0.5 * h * h / (6.0 * MU);
+    solver.advance(&mut field, dt);
+    let mask = solver.mask().expect("mask");
+    let pp = &field.p_prime;
+    let a_max = OMEGA * OMEGA * R2;
+    let is_cut = |c: usize| mask.vol(c) < 1.0 - 1e-9;
+    // [wall][band]: bands α<¼, ¼–½, ½–¾, ¾–1, full next to cut.
+    let mut acc: [[Vec<f64>; 5]; 2] = Default::default();
+    let bin_of = |a: f64, near: bool| -> Option<usize> {
+        if a < 1.0 {
+            Some(((a * 4.0).floor() as usize).min(3))
+        } else if near {
+            Some(4)
+        } else {
+            None
+        }
+    };
+    let wall_of = |x: f64, y: f64| usize::from(r_of(x, y) >= 0.5 * (R1 + R2));
+    for j in 0..ny {
+        for i in 1..nx {
+            let f = g.uface(0, j, i);
+            if mask.u_kind(f) != FaceKind::Fluid {
+                continue;
+            }
+            let (cm, cp) = (g.cell(0, j, i - 1), g.cell(0, j, i));
+            let Some(b) = bin_of(mask.a_u(f), is_cut(cm) || is_cut(cp)) else { continue };
+            let star = field.u[f] + (dt / RHO) * mask.grad_weight(0, f) * (pp[cp] - pp[cm]) / h;
+            acc[wall_of(i as f64 * h, (j as f64 + 0.5) * h)][b].push((star - field.u_old[f]) / dt / a_max);
+        }
+    }
+    for j in 1..ny {
+        for i in 0..nx {
+            let f = g.vface(0, j, i);
+            if mask.v_kind(f) != FaceKind::Fluid {
+                continue;
+            }
+            let (cm, cp) = (g.cell(0, j - 1, i), g.cell(0, j, i));
+            let Some(b) = bin_of(mask.a_v(f), is_cut(cm) || is_cut(cp)) else { continue };
+            let star = field.v[f] + (dt / RHO) * mask.grad_weight(1, f) * (pp[cp] - pp[cm]) / h;
+            acc[wall_of((i as f64 + 0.5) * h, j as f64 * h)][b].push((star - field.v_old[f]) / dt / a_max);
+        }
+    }
+    let scale = RHO * (OMEGA * R1).powi(2);
+    let (mut sum, mut cnt) = (0.0, 0usize);
+    for j in 0..ny {
+        for i in 0..nx {
+            let c = g.cell(0, j, i);
+            if mask.cell_active(c) && !is_cut(c) && mask.master(c).is_none() {
+                sum += pp[c];
+                cnt += 1;
+            }
+        }
+    }
+    let lvl = sum / cnt.max(1) as f64;
+    let mut psq = [0.0f64; 3];
+    let mut pn = [0usize; 3];
+    for j in 0..ny {
+        for i in 0..nx {
+            let c = g.cell(0, j, i);
+            if !mask.cell_active(c) || mask.master(c).is_some() {
+                continue;
+            }
+            let e = (pp[c] - lvl) / scale;
+            let w = if is_cut(c) { wall_of((i as f64 + 0.5) * h, (j as f64 + 0.5) * h) } else { 2 };
+            psq[w] += e * e;
+            pn[w] += 1;
+        }
+    }
+    let rms = |v: &[f64]| (v.iter().map(|x| x * x).sum::<f64>() / v.len().max(1) as f64).sqrt();
+    let names = ["α<¼", "¼–½", "½–¾", "¾–1", "full next to cut"];
+    let mode = if rigid { "rigid" } else if outer_drives() { "outer-driven" } else { "couette" };
+    for (w, wname) in ["inner wall", "outer wall"].iter().enumerate() {
+        let mut line = format!("  probe {mode} n {n} {wname}:");
+        for (b, name) in names.iter().enumerate() {
+            line += &format!(" {name} {} rms {:.3e};", acc[w][b].len(), rms(&acc[w][b]));
+        }
+        line += &format!(" p' at its cut cells {:.3e} ({})", (psq[w] / pn[w].max(1) as f64).sqrt(), pn[w]);
+        println!("{line}");
+    }
+    println!("  probe {mode} n {n} interior p' {:.3e} ({})", (psq[2] / pn[2].max(1) as f64).sqrt(), pn[2]);
+}
+
+#[test]
+#[ignore = "S2-7b B1 probe: the discrete operator on the exact Taylor–Couette field (seconds per rung)"]
+fn curved_operator_probe() {
+    let ns: Vec<usize> = std::env::var("RTX_E3_CURVED_NS")
+        .ok()
+        .map(|v| v.split(',').filter_map(|t| t.trim().parse().ok()).collect())
+        .unwrap_or_else(|| vec![16, 32, 64]);
+    let rigid = std::env::var("RTX_E3_CURVED_MODE").is_ok_and(|v| v == "rigid");
+    for n in ns {
+        probe(n, rigid);
+    }
+}