Skip to content

Compute heat sources only up to a given rho_norm in the PowerScalingFormationModel #2105

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
copybara-service wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Compute heat sources only up to a given rho_norm in the PowerScalingFormationModel #2105

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 12 additions & 5 deletions torax/_src/pedestal_model/formation/power_scaling_formation_model.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,22 +38,26 @@ class PowerScalingFormationRuntimeParams(
"""Runtime params for power scaling pedestal formation models."""

P_LH_prefactor: array_typing.FloatScalar = 1.0
rho_norm_for_P_SOL: array_typing.FloatScalar = 0.9


def _calculate_P_SOL_total(
def _calculate_P_SOL_total_up_to_rho_norm(
rho_norm_for_P_SOL: array_typing.FloatScalar,
internal_plasma_energy: state.PlasmaInternalEnergy,
core_sources: source_profiles_lib.SourceProfiles,
geo: geometry.Geometry,
) -> jax.Array:
"""Calculates the total power out of the separatrix."""
mask = geo.rho_norm <= rho_norm_for_P_SOL
P_heat_e = sum(
math_utils.volume_integration(source, geo)
math_utils.volume_integration(source * mask, geo)
for source in core_sources.T_e.values()
)
P_heat_i = sum(
math_utils.volume_integration(source, geo)
math_utils.volume_integration(source * mask, geo)
for source in core_sources.T_i.values()
)
# TODO(b/323504363): Also recompute dW_dt with the mask.
return P_heat_e + P_heat_i - internal_plasma_energy.dW_thermal_dt_smoothed


Expand Down Expand Up @@ -84,8 +88,11 @@ def __call__(
runtime_params.pedestal.formation, PowerScalingFormationRuntimeParams
)

P_SOL_total = _calculate_P_SOL_total(
core_profiles.internal_plasma_energy, core_sources, geo
P_SOL_total = _calculate_P_SOL_total_up_to_rho_norm(
runtime_params.pedestal.formation.rho_norm_for_P_SOL,
core_profiles.internal_plasma_energy,
core_sources,
geo,
)

P_LH, _ = scaling_laws.calculate_P_LH(
Expand Down
Loading