d0fus 2.2.2__tar.gz → 2.3.0__tar.gz

{d0fus-2.2.2 → d0fus-2.3.0}/D0FUS.py

@@ -170,12 +170,39 @@ def detect_mode_from_input(input_file):
 #%% Main functions
 
 def print_banner():
-    """Display D0FUS banner"""
-    banner = """
+    """Display the D0FUS startup banner.
+
+    Simple double-line framed banner with title, tagline, version, author,
+    license and repository URL. Can be silenced with the environment variable
+    ``D0FUS_NO_BANNER`` (useful for batch scans, HPC jobs, CI).
+    """
+    if os.environ.get("D0FUS_NO_BANNER", "").strip() not in ("", "0", "false", "False"):
+        return
+
+    # Resolve installed package version, fall back to a static string
+    try:
+        from importlib.metadata import version, PackageNotFoundError
+        try:
+            v = version("d0fus")
+        except PackageNotFoundError:
+            v = "dev"
+    except ImportError:
+        v = "dev"
+
+    # The version is the only dynamic line; the others are hardcoded for clarity.
+    # Inside width is 51 characters; the version is centered programmatically.
+    version_line = f"Version {v}".center(51)
+
+    banner = f"""
     ╔═══════════════════════════════════════════════════╗
     ║                                                   ║
     ║                       D0FUS                       ║
-    ║     Design 0-dimensional for Fusion Systems       ║
+    ║      Design 0-dimensional for Fusion Systems      ║
+    ║                                                   ║
+    ╠═══════════════════════════════════════════════════╣
+    ║{version_line}║
+    ║         T. Auclair, CEA-IRFM  |  CeCILL-C         ║
+    ║           https://github.com/IRFM/D0FUS           ║
     ║                                                   ║
     ╚═══════════════════════════════════════════════════╝
     """

{d0fus-2.2.2 → d0fus-2.3.0}/D0FUS_BIB/D0FUS_figures.py

@@ -49,6 +49,7 @@ if __name__ != "__main__":
         f_He_fraction,
         f_q_profile,
         f_sigmav,
+        _two_point_core, M_F_DT,
     )
     from .D0FUS_radial_build_functions import (
         J_non_Cu_NbTi, J_non_Cu_Nb3Sn, J_non_Cu_REBCO,
@@ -78,6 +79,7 @@ else:
         f_He_fraction,
         f_q_profile,
         f_sigmav,
+        _two_point_core, M_F_DT,
     )
     from D0FUS_BIB.D0FUS_radial_build_functions import (
         J_non_Cu_NbTi, J_non_Cu_Nb3Sn, J_non_Cu_REBCO,
@@ -2214,6 +2216,85 @@ def plot_radiation_profile(
     _save_or_show(fig, save_dir, "run_radiation_profile")
 
 
+def plot_divertor_two_point(
+    run: dict,
+    n_pts: int = 300,
+    save_dir: str | None = None,
+) -> None:
+    """
+    Simple two-point-model view: target temperature versus SOL dissipation.
+
+    A single panel shows the target electron temperature T_et as a function of
+    the SOL power-loss fraction f_cooling, at the converged upstream conditions
+    (q_par_u, p_u read from the run). The dashed line is the detachment
+    threshold T_et = 10 eV (Stangeby 2018). The flat plateau at high T_et is the
+    attached sheath-limited branch where the two-point model no longer applies.
+    The minimum dissipation required for target survival (Eq. 14) is given in
+    the title and marked by the dotted vertical line.
+
+    Parameters
+    ----------
+    run      : dict   D0FUS run output; reads the 'divertor' sub-dict produced
+                      by f_heat_two_point.
+    n_pts    : int    Number of f_cooling samples.
+    save_dir : str or None
+
+    References
+    ----------
+    P.C. Stangeby, Plasma Phys. Control. Fusion 60 (2018) 044022.
+    """
+    div = run.get("divertor", {}) or {}
+    if not {"q_par_u", "T_eu", "n_sep"}.issubset(div):
+        print("  [skip] no two-point-model divertor solution in run dict")
+        return
+
+    q_par_u   = div["q_par_u"]                 # [MW/m^2]
+    T_eu      = div["T_eu"]                     # [eV]
+    n_sep     = div["n_sep"]                    # [m^-3]
+    f_c_op    = div.get("f_cooling", 0.0)
+    f_m_op    = div.get("f_mom", 0.0)
+    T_et_op   = div.get("T_et", np.nan)
+    f_pwr_req = div.get("f_pwr_loss_req", np.nan)
+    R0        = run.get("R0", "?")
+
+    q_par_u_SI = q_par_u * 1e6
+    p_u        = 2.0 * n_sep * E_ELEM * T_eu
+
+    f_c  = np.linspace(0.0, 0.995, n_pts)
+    T_et = np.array([min(_two_point_core(q_par_u_SI, p_u, fc, f_m_op, 7.0, M_F_DT)[0],
+                         T_eu) for fc in f_c])
+
+    fig, ax = plt.subplots(figsize=(7, 4.5))
+    ax.plot(f_c, T_et, color="#4477AA", lw=2.4)
+    ax.axhline(10.0, color="#EE6677", lw=1.3, ls="--")
+    ax.text(0.015, 11.0, "detachment (10 eV)", color="#EE6677",
+            fontsize=9, va="bottom")
+    if np.isfinite(f_pwr_req):
+        ax.axvline(f_pwr_req, color="0.55", lw=1.0, ls=":")
+    if np.isfinite(T_et_op):
+        ax.plot([f_c_op], [T_et_op], "o", ms=9, color="k", zorder=5)
+        near_left = f_c_op < 0.5
+        near_top  = T_et_op > 0.3 * T_eu
+        ax.annotate(f"operating point\n$T_{{e,t}}$ = {T_et_op:.1f} eV",
+                    (f_c_op, T_et_op), textcoords="offset points",
+                    xytext=(14 if near_left else -12, -16 if near_top else 14),
+                    ha="left" if near_left else "right",
+                    va="top" if near_top else "bottom", fontsize=9)
+
+    ax.set_yscale("log")
+    ax.set_xlim(0, 1)
+    ax.set_xlabel(r"SOL power-loss fraction $f_{\rm cooling}$", fontsize=12)
+    ax.set_ylabel(r"Target electron temperature $T_{e,t}$ [eV]", fontsize=12)
+    ttl = (rf"Divertor two-point model:  $R_0$={R0} m,  "
+           rf"$q_{{\parallel u}}$={q_par_u/1e3:.2f} GW/m$^2$")
+    if np.isfinite(f_pwr_req):
+        ttl += f",  required dissipation = {f_pwr_req:.2f}"
+    ax.set_title(ttl, fontsize=11)
+    ax.grid(True, which="both", alpha=0.3)
+    plt.tight_layout()
+    _save_or_show(fig, save_dir, "run_divertor_two_point")
+
+
 # ---------------------------------------------------------------------------
 # A — Convenience wrapper
 # ---------------------------------------------------------------------------
@@ -2998,23 +3079,24 @@ def plot_run(
     save_dir: str | None = None,
 ) -> None:
     """
-    Render the run-specific figure set (10 figures).
+    Render the run-specific figure set (11 figures).
 
     This is the subset called after each D0FUS run.  It contains only the
     figures that depend on the current run configuration and results —
     no validation curves, no benchmarks, no scaling-law surveys.
 
     Figures produced:
-      [ 1/10]  Tokamak LCFS comparison (with D0FUS overlay)
-      [ 2/10]  Miller flux surfaces (run geometry)
-      [ 3/10]  Shaping profiles κ(ρ), δ(ρ)
-      [ 4/10]  Kinetic profiles n(ρ), T(ρ), p(ρ)
-      [ 5/10]  Safety factor q(ρ) and current decomposition
-      [ 6/10]  Radiation profiles
-      [ 7/10]  TF coil side view
-      [ 8/10]  CICC TF conductor
-      [ 9/10]  CS cross-section
-      [10/10]  CICC CS conductor
+      [ 1/11]  Tokamak LCFS comparison (with D0FUS overlay)
+      [ 2/11]  Miller flux surfaces (run geometry)
+      [ 3/11]  Shaping profiles κ(ρ), δ(ρ)
+      [ 4/11]  Kinetic profiles n(ρ), T(ρ), p(ρ)
+      [ 5/11]  Safety factor q(ρ) and current decomposition
+      [ 6/11]  Radiation profiles
+      [ 7/11]  Divertor two-point model (detachment vs SOL dissipation)
+      [ 8/11]  TF coil side view
+      [ 9/11]  CICC TF conductor
+      [10/11]  CS cross-section
+      [11/11]  CICC CS conductor
 
     Parameters
     ----------
@@ -3023,7 +3105,7 @@ def plot_run(
         If provided, figures are saved as PNG files.
         Pass ``None`` to display interactively.
     """
-    N = 10
+    N = 11
 
     def _p(i, label):
         print(f"  [{i:2d}/{N}] {label}")
@@ -3051,17 +3133,20 @@ def plot_run(
     _p(6, "Radiation profiles")
     plot_radiation_profile(run, save_dir=save_dir)
 
+    _p(7, "Divertor two-point model")
+    plot_divertor_two_point(run, save_dir=save_dir)
+
     # ── Coils & conductors ────────────────────────────────────────────
-    _p(7, "TF coil side view")
+    _p(8, "TF coil side view")
     plot_TF_side_view(run, save_dir=save_dir)
 
-    _p(8, "CICC TF conductor")
+    _p(9, "CICC TF conductor")
     plot_CICC_cross_section(build_conductor_from_run(run, coil="TF"), save_dir=save_dir)
 
-    _p(9, "CS cross-section")
+    _p(10, "CS cross-section")
     plot_CS_cross_section(run, save_dir=save_dir)
 
-    _p(10, "CICC CS conductor")
+    _p(11, "CICC CS conductor")
     plot_CICC_cross_section(build_conductor_from_run(run, coil="CS"), save_dir=save_dir)
 
     print("Done.")
@@ -3308,11 +3393,11 @@ def plot_TF_benchmark_table(cfg=None, save_dir=None) -> None:
         "ARC":     {"a": 1.10, "b": 0.89, "R0": 3.30, "σ": 1000e6, "T_op": 20.0,
                     "B_max": 23.0,  "n_TF": 1,   "sc": "REBCO", "config": "Plug",
                     "κ": 1.84, "I_cond": 50e3, "V_max": 10e3, "N_sub": 6,
-                    "tau_h": 20,   "J_wost": 120e6},
+                    "tau_h": 20,   "J_wost": 200e6},
         "SPARC":   {"a": 0.57, "b": 0.18, "R0": 1.85, "σ": 1000e6, "T_op": 20.0,
                     "B_max": 20.0,  "n_TF": 1,   "sc": "REBCO", "config": "Bucking",
-                    "κ": 1.75, "I_cond": 40.5e3,"V_max": 10e3, "N_sub": 4,
-                    "tau_h": 20,   "J_wost": 120e6},
+                    "κ": 1.75, "I_cond": 40.5e3,"V_max": 10e3, "N_sub": 6,
+                    "tau_h": 20,   "J_wost": 200e6},
     }
 
     def _clean(val):

{d0fus-2.2.2 → d0fus-2.3.0}/D0FUS_BIB/D0FUS_import.py

@@ -18,11 +18,13 @@ os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
 
 import json
 import math
+import multiprocessing
 import random
 import re
 import shutil
 import sys
 import time
+import traceback
 import warnings
 import importlib
 from datetime import datetime
@@ -34,7 +36,7 @@ import numpy as np
 import pandas as pd
 import sympy as sp
 from typing import List, Tuple
-from dataclasses import dataclass
+from dataclasses import dataclass, replace, asdict
 
 #%% Scipy - Optimization and Numerical Methods
 
@@ -66,11 +68,13 @@ import matplotlib.colors as mcolors
 import matplotlib.lines as mlines
 import matplotlib.patches as mpatches
 import matplotlib.cm as cm
+from matplotlib.animation import FuncAnimation, PillowWriter
 from matplotlib.colors import Normalize
 from matplotlib.gridspec import GridSpec
 from matplotlib.patches import Circle, Rectangle, Patch
 from matplotlib.ticker import MultipleLocator
 from mpl_toolkits.axes_grid1 import make_axes_locatable
+from mpl_toolkits.mplot3d import Axes3D  # noqa: F401  (registers the '3d' projection)
 from pandas.plotting import table
 from tqdm import tqdm
 from dataclasses import dataclass

{d0fus-2.2.2 → d0fus-2.3.0}/D0FUS_BIB/D0FUS_parameterization.py

@@ -97,8 +97,10 @@ class GlobalConfig:
     Bmax_TF  : float = 12.0        # Peak magnetic field on TF conductor [T]
     Bmax_CS_adm  : float = 25.0    # Peak magnetic field allowed on the CS [T]
     P_fus : float = 2000.0         # Total fusion power [MW]
-    Tbar  : float = 14.0           # Volume-averaged ion temperature [keV]
-    H     : float = 1.0            # Confinement enhancement factor (H-factor) [-]
+    Tbar  : float = 14.0           # Volume-averaged electron temperature T_e [keV]
+    tau_i_e : float = 1.0          # Ion-to-electron temperature ratio T_i/T_e [-]
+                                   # 1.0 -> single-temperature plasma (T_i = T_e).
+                                   # Prescribed: T_i(rho) = tau_i_e * T_e(rho),
 
     # ── 2. Physics and operation ─────────────────────────────────────────────
     Operation_mode     : str   = 'Pulsed'       # 'Steady-State' or 'Pulsed'
@@ -122,7 +124,7 @@ class GlobalConfig:
     #   'Sauter'    — uses LCFS values (κ_edge, δ_edge). Sauter, FED 112 (2016) Eq. 30.
     #   'ITER_1989' — uses ψ_N = 0.95 values (κ₉₅, δ₉₅). Uckan (1989), also Johner (2011).
     Option_q95         : str = 'Sauter'         # q₉₅ formula: 'Sauter' (default) or 'ITER_1989'
-    Option_Kappa       : str = 'Wenninger'    # Elongation model: 'Wenninger', 'Stambaugh', 'Freidberg', 'Manual'
+    Option_Kappa       : str = 'Wenninger'      # Elongation model: 'Wenninger', 'Stambaugh', 'Freidberg', 'Manual'
     κ_manual           : float = 1.9            # Elongation (Manual mode only) [-]
 
     # ── 2a. Safety factor and current density profiles ────────────────────────
@@ -173,7 +175,15 @@ class GlobalConfig:
 
     # ── 4. Plasma composition ────────────────────────────────────────────────
     Atomic_mass : float = 2.5   # Volume-averaged ionic mass [AMU]  (D-T: 2.5)
-    Zeff        : float = 2.0   # Effective plasma charge [-]
+    Zeff        : float = None   # Effective plasma charge [-].
+                                 # None  -> computed self-consistently from the impurity
+                                 #          inventory (impurity_species + f_imp_core) and the
+                                 #          helium ash fraction (see _compute_Zeff_effective):
+                                 #              Z_eff = 1 + 2 f_He - sum_j <Z_j> c_j
+                                 #                                  + sum_j <Z_j>^2 c_j
+                                 #          An empty impurity inventory therefore yields a clean
+                                 #          D-T+He plasma with Z_eff ~ 1 + 2 f_He (~1.1).
+                                 # float -> manual override (legacy behaviour, e.g. Zeff = 2.0).
     r_synch     : float = 0.5   # Synchrotron radiation wall reflectivity [-]
     C_Alpha     : float = 7.0   # Helium ash dilution tuning parameter [-] (default taken from PROCESS)
     # Impurity line radiation (0D bulk-plasma estimate).
@@ -215,6 +225,26 @@ class GlobalConfig:
     Young_modul_Steel    : float = 200e9    # Steel Young's modulus [Pa] (only used in CIRCE model)
     Young_modul_GF       : float = 90e9     # S-glass fiber Young's modulus [Pa] (only used in CIRCE model)
     fatigue_CS           : float = 2.0      # CS fatigue knockdown factor (pulsed & wedging only) [-]
+    # ── Steel allowable safety factors ───────────────────────────────────────
+    # Dimensionless knockdown applied to the steel mechanical allowable inside
+    # the TF and CS thickness solvers: σ_eff = σ_allowable / SF.
+    # Captures effects not represented by the idealised CICC area model:
+    #   – Realistic winding-pack filling factor: round/square cables inside a
+    #     rectangular envelope leave 15–30% of the WP cross-section occupied
+    #     by ground insulation, inter-pancake plates, helium manifolds,
+    #     terminations and assembly clearances. The structural jacket carries
+    #     the electromagnetic load through a section reduced by this factor.
+    #   – Stress concentrations at jacket corners and transitions (K_t ≈ 1.2–1.5).
+    #   – Weld efficiency on jacket seams (η_w ≈ 0.85–0.90).
+    #   – Manufacturing thickness tolerances (≈ ±5%).
+    # Default 1.0 preserves backward compatibility (raw material allowable used
+    # directly). Realistic engineering value for both TF and CS large-magnet
+    # CICC designs is ≈ 1.5 (composition of the four contributions above).
+    # The CS fatigue knockdown (fatigue_CS) multiplies on top of SF_CS, since
+    # the two factors address disjoint phenomena: primary static stress (SF)
+    # versus cyclic damage accumulation (fatigue_CS).
+    SF_TF                : float = 1.0      # Safety factor on TF steel allowable [-] (suggested ≈ 1.5 for realistic CICC packing)
+    SF_CS                : float = 1.0      # Safety factor on CS steel allowable [-] (suggested ≈ 1.5 for realistic CICC packing)
 
     # ── 7. TF coil engineering ───────────────────────────────────────────────
     Radial_build_model   : str   = 'refined' # Stress model: 'academic', 'refined', 'CIRCE'
@@ -257,7 +287,11 @@ class GlobalConfig:
     f_void    : float = None    # Interstitial void fraction in strand bundle [-]
                                 # None = auto: 0.33 (LTS) / 0.00 (REBCO).
                                 # Set explicitly (e.g. 0.30) to override.
-    f_In      : float = 0.15    # Insulation area fraction [-]
+    f_In      : float = 0.05    # Insulation area fraction [-]
+    # Winding-pack overhead fraction in wost, treated exactly like the
+    # helium fraction: ground and inter-pancake insulation plus assembly
+    # clearances that occupy volume but carry neither current nor load.
+    f_gap     : float = 0.15    # WP insulation + clearance fraction in wost [-]
 
     # Temperature margins above T_helium defining T_operating [K]
     # Conservative baseline: Corato et al., "Common operating values for DEMO…" (2016)
@@ -297,7 +331,7 @@ class GlobalConfig:
     # P_CD_total = P_LH + P_ECRH + P_NBI + P_ICRH  (P_aux_input ignored).
     P_LH   : float = 0.0      # LHCD plasma power [MW]
     P_ECRH : float = 0.0      # ECRH plasma power [MW]
-    P_NBI  : float = 0.0      # NBI injected power [MW]  (5 % losses applied internally)
+    P_NBI  : float = 0.0      # NBI injected power[MW]
     P_ICRH : float = 0.0      # ICRH plasma power [MW]   (heating only, no current drive)
     # --- Steady-State mode: power fractions per source [-] ---
     # Used when Operation_mode = 'Steady-State' and CD_source = 'Multi'
@@ -325,6 +359,16 @@ class GlobalConfig:
     # ── 12. Plasma-facing components ─────────────────────────────────────────
     theta_deg : float = 2.7      # Divertor strike-point grazing angle [deg]
 
+    # Refined divertor exhaust (two-point model, Stangeby 2018). All used only
+    # in the post-convergence f_heat_two_point evaluation; they do not feed the
+    # core power balance.
+    f_n_sep       : float = 0.20  # n_sep / volume-average density n̄ [-].
+                                  # Groth value taken from https://arxiv.org/pdf/2406.15693
+    f_cooling_div : float = 0.0   # SOL volumetric power-loss fraction [-] (2PM input)
+    f_mom_div     : float = 0.0   # SOL volumetric momentum-loss fraction [-] (2PM input)
+    q_dep_limit   : float = 5.0   # Tolerable deposited target heat flux [MW/m²]
+    flux_expansion: float = 1.0   # Target/upstream flux expansion R_t/R_u [-]
+
     # ── 13. Maintenance constraints ──────────────────────────────────────────
     ripple_adm : float = 0.01    # Admissible toroidal field ripple [-]  (default 1%)
     L_min      : float = 3.00    # Minimum toroidal maintenance access width [m]