Emission Models¶
This notebook demonstrates how to construct emission models to be used in synthetic diagnostics.
Prerequisites: Pooch must be installed to download the example data.
[1]:
import numpy as np
import ultraplot as uplt
from raysect.core.math import Point3D, Vector3D
from raysect.core.workflow import MulticoreEngine
from raysect.optical import Ray, Spectrum, World
from rich import print as rprint
from rich.table import Table
from cherab.core.atomic import Isotope, Line, elements
from cherab.core.math import sample3d
from cherab.core.model import Bremsstrahlung, ExcitationLine, RecombinationLine
from cherab.imas.datasets import iter_jintrac
from cherab.imas.plasma import load_plasma
from cherab.openadas import OpenADAS
# Set dark background for plots
uplt.rc.style = "dark_background"
13:47:10 CRITICAL Could not import 'imas_core': No module named 'imas_core'. Some functionality is not available. @imas_interface.py:34
Retrieve atomic data¶
OpenADAS provides data for a variety of emission processes in plasmas. Here we demonstrate how to include some of these processes in a cherab simulation and generate synthetic spectra.
[2]:
atomic_data = OpenADAS(permit_extrapolation=True, missing_rates_return_null=False)
When using OpenADAS data first time, the relevant data files must be downloaded and installed.
Usually, they are downloaded by executing populate, but some files are not downloaded by default.
Currently, neon and tungsten atomic data files (related to adf11, adf15) are to be downloaded manually as a workaround.
[3]:
from cherab.openadas.install import install_files
from cherab.openadas.repository import populate
try:
atomic_data.ionisation_rate(elements.neon, 1)
atomic_data.recombination_pec(elements.neon, 1, ("2s1 2p6 2s0.5", "2s2 2p5 2p2.5"))
except RuntimeError:
print("Populating OpenADAS repository...")
# Download ADAS files from OpenADAS
populate()
# Install ADAS files that are not downloaded by default
# TODO: uncomment tungsten ADF15 once https://github.com/cherab/core/pull/497 is resolved
install_files(
{
# "adf11scd": ((elements.tungsten, "adf11/scd50/scd50_w.dat"),),
# "adf11acd": ((elements.tungsten, "adf11/acd50/acd50_w.dat"),),
"adf15": [
(elements.neon, i, f"adf15/pec96#ne/pec96#ne_pjr#ne{i}.dat")
for i in range(elements.neon.atomic_number)
],
# + [
# (elements.tungsten, i, f"adf15/pec40#w/pec40#w_ic#w{i}.dat")
# for i in range(elements.tungsten.atomic_number)
# ],
},
download=True,
)
Populating OpenADAS repository...
Installing adf11/scd12/scd12_h.dat...
- downloading ADF file 'adf11/scd12/scd12_h.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd12/scd12_h.dat'
Installing adf11/scd96/scd96_he.dat...
- downloading ADF file 'adf11/scd96/scd96_he.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd96/scd96_he.dat'
Installing adf11/scd96/scd96_li.dat...
- downloading ADF file 'adf11/scd96/scd96_li.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd96/scd96_li.dat'
Installing adf11/scd96/scd96_be.dat...
- downloading ADF file 'adf11/scd96/scd96_be.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd96/scd96_be.dat'
Installing adf11/scd89/scd89_b.dat...
- downloading ADF file 'adf11/scd89/scd89_b.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd89/scd89_b.dat'
Installing adf11/scd96/scd96_c.dat...
- downloading ADF file 'adf11/scd96/scd96_c.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd96/scd96_c.dat'
Installing adf11/scd96/scd96_n.dat...
- downloading ADF file 'adf11/scd96/scd96_n.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd96/scd96_n.dat'
Installing adf11/scd96/scd96_o.dat...
- downloading ADF file 'adf11/scd96/scd96_o.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd96/scd96_o.dat'
Installing adf11/scd96/scd96_ne.dat...
Installing adf11/scd89/scd89_ar.dat...
- downloading ADF file 'adf11/scd89/scd89_ar.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd89/scd89_ar.dat'
Installing adf11/scd89/scd89_kr.dat...
- downloading ADF file 'adf11/scd89/scd89_kr.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd89/scd89_kr.dat'
Installing adf11/scd89/scd89_xe.dat...
- downloading ADF file 'adf11/scd89/scd89_xe.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/scd89/scd89_xe.dat'
Installing adf11/acd12/acd12_h.dat...
- downloading ADF file 'adf11/acd12/acd12_h.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd12/acd12_h.dat'
Installing adf11/acd96/acd96_he.dat...
- downloading ADF file 'adf11/acd96/acd96_he.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd96/acd96_he.dat'
Installing adf11/acd96/acd96_li.dat...
- downloading ADF file 'adf11/acd96/acd96_li.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd96/acd96_li.dat'
Installing adf11/acd96/acd96_be.dat...
- downloading ADF file 'adf11/acd96/acd96_be.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd96/acd96_be.dat'
Installing adf11/acd89/acd89_b.dat...
- downloading ADF file 'adf11/acd89/acd89_b.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd89/acd89_b.dat'
Installing adf11/acd96/acd96_c.dat...
- downloading ADF file 'adf11/acd96/acd96_c.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd96/acd96_c.dat'
Installing adf11/acd96/acd96_n.dat...
- downloading ADF file 'adf11/acd96/acd96_n.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd96/acd96_n.dat'
Installing adf11/acd96/acd96_o.dat...
- downloading ADF file 'adf11/acd96/acd96_o.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd96/acd96_o.dat'
Installing adf11/acd96/acd96_ne.dat...
Installing adf11/acd89/acd89_ar.dat...
- downloading ADF file 'adf11/acd89/acd89_ar.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd89/acd89_ar.dat'
Installing adf11/acd89/acd89_kr.dat...
- downloading ADF file 'adf11/acd89/acd89_kr.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd89/acd89_kr.dat'
Installing adf11/acd89/acd89_xe.dat...
- downloading ADF file 'adf11/acd89/acd89_xe.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/acd89/acd89_xe.dat'
Installing adf11/ccd96/ccd96_h.dat...
- downloading ADF file 'adf11/ccd96/ccd96_h.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd96/ccd96_h.dat'
Installing adf11/ccd96/ccd96_he.dat...
- downloading ADF file 'adf11/ccd96/ccd96_he.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd96/ccd96_he.dat'
Installing adf11/ccd89/ccd89_li.dat...
- downloading ADF file 'adf11/ccd89/ccd89_li.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_li.dat'
Installing adf11/ccd89/ccd89_be.dat...
- downloading ADF file 'adf11/ccd89/ccd89_be.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_be.dat'
Installing adf11/ccd89/ccd89_b.dat...
- downloading ADF file 'adf11/ccd89/ccd89_b.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_b.dat'
Installing adf11/ccd96/ccd96_c.dat...
- downloading ADF file 'adf11/ccd96/ccd96_c.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd96/ccd96_c.dat'
Installing adf11/ccd89/ccd89_n.dat...
- downloading ADF file 'adf11/ccd89/ccd89_n.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_n.dat'
Installing adf11/ccd89/ccd89_o.dat...
- downloading ADF file 'adf11/ccd89/ccd89_o.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_o.dat'
Installing adf11/ccd89/ccd89_ne.dat...
- downloading ADF file 'adf11/ccd89/ccd89_ne.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_ne.dat'
Installing adf11/ccd89/ccd89_ar.dat...
- downloading ADF file 'adf11/ccd89/ccd89_ar.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_ar.dat'
Installing adf11/ccd89/ccd89_kr.dat...
- downloading ADF file 'adf11/ccd89/ccd89_kr.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_kr.dat'
Installing adf11/ccd89/ccd89_xe.dat...
- downloading ADF file 'adf11/ccd89/ccd89_xe.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/ccd89/ccd89_xe.dat'
Installing adf11/plt12/plt12_h.dat...
- downloading ADF file 'adf11/plt12/plt12_h.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt12/plt12_h.dat'
Installing adf11/plt96/plt96_he.dat...
- downloading ADF file 'adf11/plt96/plt96_he.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_he.dat'
Installing adf11/plt96/plt96_li.dat...
- downloading ADF file 'adf11/plt96/plt96_li.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_li.dat'
Installing adf11/plt96/plt96_be.dat...
- downloading ADF file 'adf11/plt96/plt96_be.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_be.dat'
Installing adf11/plt89/plt89_b.dat...
- downloading ADF file 'adf11/plt89/plt89_b.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt89/plt89_b.dat'
Installing adf11/plt96/plt96_c.dat...
- downloading ADF file 'adf11/plt96/plt96_c.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_c.dat'
Installing adf11/plt96/plt96_n.dat...
- downloading ADF file 'adf11/plt96/plt96_n.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_n.dat'
Installing adf11/plt96/plt96_o.dat...
- downloading ADF file 'adf11/plt96/plt96_o.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_o.dat'
Installing adf11/plt96/plt96_ne.dat...
- downloading ADF file 'adf11/plt96/plt96_ne.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt96/plt96_ne.dat'
Installing adf11/plt40/plt40_ar.dat...
- downloading ADF file 'adf11/plt40/plt40_ar.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt40/plt40_ar.dat'
Installing adf11/plt89/plt89_kr.dat...
- downloading ADF file 'adf11/plt89/plt89_kr.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt89/plt89_kr.dat'
Installing adf11/plt89/plt89_xe.dat...
- downloading ADF file 'adf11/plt89/plt89_xe.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/plt89/plt89_xe.dat'
Installing adf11/prb12/prb12_h.dat...
- downloading ADF file 'adf11/prb12/prb12_h.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb12/prb12_h.dat'
Installing adf11/prb96/prb96_he.dat...
- downloading ADF file 'adf11/prb96/prb96_he.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_he.dat'
Installing adf11/prb96/prb96_li.dat...
- downloading ADF file 'adf11/prb96/prb96_li.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_li.dat'
Installing adf11/prb96/prb96_be.dat...
- downloading ADF file 'adf11/prb96/prb96_be.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_be.dat'
Installing adf11/prb89/prb89_b.dat...
- downloading ADF file 'adf11/prb89/prb89_b.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb89/prb89_b.dat'
Installing adf11/prb96/prb96_c.dat...
- downloading ADF file 'adf11/prb96/prb96_c.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_c.dat'
Installing adf11/prb96/prb96_n.dat...
- downloading ADF file 'adf11/prb96/prb96_n.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_n.dat'
Installing adf11/prb96/prb96_o.dat...
- downloading ADF file 'adf11/prb96/prb96_o.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_o.dat'
Installing adf11/prb96/prb96_ne.dat...
- downloading ADF file 'adf11/prb96/prb96_ne.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb96/prb96_ne.dat'
Installing adf11/prb89/prb89_ar.dat...
- downloading ADF file 'adf11/prb89/prb89_ar.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb89/prb89_ar.dat'
Installing adf11/prb89/prb89_kr.dat...
- downloading ADF file 'adf11/prb89/prb89_kr.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb89/prb89_kr.dat'
Installing adf11/prb89/prb89_xe.dat...
- downloading ADF file 'adf11/prb89/prb89_xe.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prb89/prb89_xe.dat'
Installing adf11/prc96/prc96_h.dat...
- downloading ADF file 'adf11/prc96/prc96_h.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc96/prc96_h.dat'
Installing adf11/prc96/prc96_he.dat...
- downloading ADF file 'adf11/prc96/prc96_he.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc96/prc96_he.dat'
Installing adf11/prc89/prc89_li.dat...
- downloading ADF file 'adf11/prc89/prc89_li.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_li.dat'
Installing adf11/prc89/prc89_be.dat...
- downloading ADF file 'adf11/prc89/prc89_be.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_be.dat'
Installing adf11/prc89/prc89_b.dat...
- downloading ADF file 'adf11/prc89/prc89_b.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_b.dat'
Installing adf11/prc96/prc96_c.dat...
- downloading ADF file 'adf11/prc96/prc96_c.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc96/prc96_c.dat'
Installing adf11/prc89/prc89_n.dat...
- downloading ADF file 'adf11/prc89/prc89_n.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_n.dat'
Installing adf11/prc89/prc89_o.dat...
- downloading ADF file 'adf11/prc89/prc89_o.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_o.dat'
Installing adf11/prc89/prc89_ne.dat...
- downloading ADF file 'adf11/prc89/prc89_ne.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_ne.dat'
Installing adf11/prc89/prc89_ar.dat...
- downloading ADF file 'adf11/prc89/prc89_ar.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_ar.dat'
Installing adf11/prc89/prc89_kr.dat...
- downloading ADF file 'adf11/prc89/prc89_kr.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_kr.dat'
Installing adf11/prc89/prc89_xe.dat...
- downloading ADF file 'adf11/prc89/prc89_xe.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf11/prc89/prc89_xe.dat'
Installing adf12/qef93#h/qef93#h_h1.dat...
- downloading ADF file 'adf12/qef93#h/qef93#h_h1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef93#h/qef93#h_h1.dat'
Installing adf12/qef93#h/qef93#h_he2.dat...
- downloading ADF file 'adf12/qef93#h/qef93#h_he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef93#h/qef93#h_he2.dat'
Installing adf12/qef97#h/qef97#h_en2_kvi#he2.dat...
- downloading ADF file 'adf12/qef97#h/qef97#h_en2_kvi#he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef97#h/qef97#h_en2_kvi#he2.dat'
Installing adf12/qef93#h/qef93#h_be4.dat...
- downloading ADF file 'adf12/qef93#h/qef93#h_be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef93#h/qef93#h_be4.dat'
Installing adf12/qef97#h/qef97#h_en2_kvi#be4.dat...
- downloading ADF file 'adf12/qef97#h/qef97#h_en2_kvi#be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef97#h/qef97#h_en2_kvi#be4.dat'
Installing adf12/qef93#h/qef93#h_b5.dat...
- downloading ADF file 'adf12/qef93#h/qef93#h_b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef93#h/qef93#h_b5.dat'
Installing adf12/qef97#h/qef97#h_en2_kvi#b5.dat...
- downloading ADF file 'adf12/qef97#h/qef97#h_en2_kvi#b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef97#h/qef97#h_en2_kvi#b5.dat'
Installing adf12/qef93#h/qef93#h_c6.dat...
- downloading ADF file 'adf12/qef93#h/qef93#h_c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef93#h/qef93#h_c6.dat'
Installing adf12/qef97#h/qef97#h_en2_kvi#c6.dat...
- downloading ADF file 'adf12/qef97#h/qef97#h_en2_kvi#c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef97#h/qef97#h_en2_kvi#c6.dat'
Installing adf12/qef93#h/qef93#h_ne10.dat...
- downloading ADF file 'adf12/qef93#h/qef93#h_ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef93#h/qef93#h_ne10.dat'
Installing adf12/qef97#h/qef97#h_en2_kvi#ne10.dat...
- downloading ADF file 'adf12/qef97#h/qef97#h_en2_kvi#ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf12/qef97#h/qef97#h_en2_kvi#ne10.dat'
Installing adf15/pec12#h/pec12#h_pju#h0.dat...
- downloading ADF file 'adf15/pec12#h/pec12#h_pju#h0.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec12#h/pec12#h_pju#h0.dat'
Installing adf15/pec96#he/pec96#he_pju#he0.dat...
- downloading ADF file 'adf15/pec96#he/pec96#he_pju#he0.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#he/pec96#he_pju#he0.dat'
Installing adf15/pec96#he/pec96#he_pju#he1.dat...
- downloading ADF file 'adf15/pec96#he/pec96#he_pju#he1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#he/pec96#he_pju#he1.dat'
Installing adf15/pec96#be/pec96#be_pju#be0.dat...
- downloading ADF file 'adf15/pec96#be/pec96#be_pju#be0.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#be/pec96#be_pju#be0.dat'
Installing adf15/pec96#be/pec96#be_pju#be1.dat...
- downloading ADF file 'adf15/pec96#be/pec96#be_pju#be1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#be/pec96#be_pju#be1.dat'
Installing adf15/pec96#be/pec96#be_pju#be2.dat...
- downloading ADF file 'adf15/pec96#be/pec96#be_pju#be2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#be/pec96#be_pju#be2.dat'
Installing adf15/pec96#be/pec96#be_pju#be3.dat...
- downloading ADF file 'adf15/pec96#be/pec96#be_pju#be3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#be/pec96#be_pju#be3.dat'
Installing adf15/pec96#c/pec96#c_vsu#c0.dat...
- downloading ADF file 'adf15/pec96#c/pec96#c_vsu#c0.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#c/pec96#c_vsu#c0.dat'
Installing adf15/pec96#c/pec96#c_vsu#c1.dat...
- downloading ADF file 'adf15/pec96#c/pec96#c_vsu#c1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#c/pec96#c_vsu#c1.dat'
Installing adf15/pec96#c/pec96#c_vsu#c2.dat...
- downloading ADF file 'adf15/pec96#c/pec96#c_vsu#c2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#c/pec96#c_vsu#c2.dat'
Installing adf15/pec96#c/pec96#c_pju#c5.dat...
- downloading ADF file 'adf15/pec96#c/pec96#c_pju#c5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#c/pec96#c_pju#c5.dat'
Installing adf15/pec96#n/pec96#n_vsu#n0.dat...
- downloading ADF file 'adf15/pec96#n/pec96#n_vsu#n0.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#n/pec96#n_vsu#n0.dat'
Installing adf15/pec96#n/pec96#n_vsu#n1.dat...
- downloading ADF file 'adf15/pec96#n/pec96#n_vsu#n1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#n/pec96#n_vsu#n1.dat'
Installing adf21/bms97#h/bms97#h_h1.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_h1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_h1.dat'
Installing adf21/bms97#h/bms97#h_he2.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_he2.dat'
Installing adf21/bms97#h/bms97#h_li3.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_li3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_li3.dat'
Installing adf21/bms97#h/bms97#h_be4.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_be4.dat'
Installing adf21/bms97#h/bms97#h_b5.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_b5.dat'
Installing adf21/bms97#h/bms97#h_c6.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_c6.dat'
Installing adf21/bms97#h/bms97#h_n7.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_n7.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_n7.dat'
Installing adf21/bms97#h/bms97#h_o8.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_o8.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_o8.dat'
Installing adf21/bms97#h/bms97#h_f9.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_f9.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_f9.dat'
Installing adf21/bms97#h/bms97#h_ne10.dat...
- downloading ADF file 'adf21/bms97#h/bms97#h_ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf21/bms97#h/bms97#h_ne10.dat'
Installing adf22/bmp97#h/bmp97#h_2_h1.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_h1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_h1.dat'
Installing adf22/bmp97#h/bmp97#h_3_h1.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_h1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_h1.dat'
Installing adf22/bmp97#h/bmp97#h_4_h1.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_h1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_h1.dat'
Installing adf22/bmp97#h/bmp97#h_2_he2.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_he2.dat'
Installing adf22/bmp97#h/bmp97#h_3_he2.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_he2.dat'
Installing adf22/bmp97#h/bmp97#h_4_he2.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_he2.dat'
Installing adf22/bmp97#h/bmp97#h_2_li3.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_li3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_li3.dat'
Installing adf22/bmp97#h/bmp97#h_3_li3.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_li3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_li3.dat'
Installing adf22/bmp97#h/bmp97#h_4_li3.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_li3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_li3.dat'
Installing adf22/bmp97#h/bmp97#h_2_be4.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_be4.dat'
Installing adf22/bmp97#h/bmp97#h_3_be4.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_be4.dat'
Installing adf22/bmp97#h/bmp97#h_4_be4.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_be4.dat'
Installing adf22/bmp97#h/bmp97#h_2_b5.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_b5.dat'
Installing adf22/bmp97#h/bmp97#h_3_b5.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_b5.dat'
Installing adf22/bmp97#h/bmp97#h_4_b5.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_b5.dat'
Installing adf22/bmp97#h/bmp97#h_2_c6.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_c6.dat'
Installing adf22/bmp97#h/bmp97#h_3_c6.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_c6.dat'
Installing adf22/bmp97#h/bmp97#h_4_c6.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_c6.dat'
Installing adf22/bmp97#h/bmp97#h_2_n7.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_n7.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_n7.dat'
Installing adf22/bmp97#h/bmp97#h_3_n7.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_n7.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_n7.dat'
Installing adf22/bmp97#h/bmp97#h_4_n7.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_n7.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_n7.dat'
Installing adf22/bmp97#h/bmp97#h_2_o8.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_o8.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_o8.dat'
Installing adf22/bmp97#h/bmp97#h_3_o8.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_o8.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_o8.dat'
Installing adf22/bmp97#h/bmp97#h_4_o8.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_o8.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_o8.dat'
Installing adf22/bmp97#h/bmp97#h_2_f9.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_f9.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_f9.dat'
Installing adf22/bmp97#h/bmp97#h_3_f9.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_f9.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_f9.dat'
Installing adf22/bmp97#h/bmp97#h_4_f9.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_f9.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_f9.dat'
Installing adf22/bmp97#h/bmp97#h_2_ne10.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_2_ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_2_ne10.dat'
Installing adf22/bmp97#h/bmp97#h_3_ne10.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_3_ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_3_ne10.dat'
Installing adf22/bmp97#h/bmp97#h_4_ne10.dat...
- downloading ADF file 'adf22/bmp97#h/bmp97#h_4_ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bmp97#h/bmp97#h_4_ne10.dat'
Installing adf22/bme10#h/bme10#h_h1.dat...
- downloading ADF file 'adf22/bme10#h/bme10#h_h1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme10#h/bme10#h_h1.dat'
Installing adf22/bme97#h/bme97#h_he2.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_he2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_he2.dat'
Installing adf22/bme97#h/bme97#h_li3.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_li3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_li3.dat'
Installing adf22/bme97#h/bme97#h_be4.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_be4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_be4.dat'
Installing adf22/bme97#h/bme97#h_b5.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_b5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_b5.dat'
Installing adf22/bme97#h/bme97#h_c6.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_c6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_c6.dat'
Installing adf22/bme97#h/bme97#h_n7.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_n7.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_n7.dat'
Installing adf22/bme97#h/bme97#h_o8.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_o8.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_o8.dat'
Installing adf22/bme97#h/bme97#h_f9.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_f9.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_f9.dat'
Installing adf22/bme97#h/bme97#h_ne10.dat...
- downloading ADF file 'adf22/bme97#h/bme97#h_ne10.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme97#h/bme97#h_ne10.dat'
Installing adf22/bme99#h/bme99#h_ar18.dat...
- downloading ADF file 'adf22/bme99#h/bme99#h_ar18.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf22/bme99#h/bme99#h_ar18.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne0.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne0.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne0.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne1.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne1.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne1.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne2.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne2.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne2.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne3.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne3.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne3.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne4.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne4.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne4.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne5.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne5.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne5.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne6.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne6.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne6.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne7.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne7.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne7.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne8.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne8.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne8.dat'
Installing adf15/pec96#ne/pec96#ne_pjr#ne9.dat...
- downloading ADF file 'adf15/pec96#ne/pec96#ne_pjr#ne9.dat' to '/home/runner/.cherab/openadas/repository/_download_cache/adf15/pec96#ne/pec96#ne_pjr#ne9.dat'
Load full plasma from IMAS¶
The details of loading a full plasma are covered in the Full Plasma notebook. Here we simply load a plasma from an IMAS jintrac dataset.
[4]:
world = World()
path = iter_jintrac()
plasma = load_plasma(path, "r", parent=world, atomic_data=atomic_data)
13:50:15 INFO Parsing data dictionary version 4.1.1 @dd_zip.py:89
13:50:16 INFO Parsing data dictionary version 4.0.0 @dd_zip.py:89
/home/runner/work/imas/imas/src/cherab/imas/plasma/blend.py:174: RuntimeWarning: The 'get_slice' method is not implemented for the URI '/home/runner/.cache/cherab/imas/iter_scenario_53298_seq1_DD4_mod.nc'. Falling back to 'get' method to retrieve the entire IDS.
core_profiles_ids = get_ids_time_slice(
/home/runner/work/imas/imas/src/cherab/imas/plasma/blend.py:199: RuntimeWarning: The 'get_slice' method is not implemented for the URI '/home/runner/.cache/cherab/imas/iter_scenario_53298_seq1_DD4_mod.nc'. Falling back to 'get' method to retrieve the entire IDS.
edge_profiles_ids = get_ids_time_slice(
/home/runner/work/imas/imas/src/cherab/imas/plasma/equilibrium.py:97: RuntimeWarning: The 'get_slice' method is not implemented for the URI '/home/runner/.cache/cherab/imas/iter_scenario_53298_seq1_DD4_mod.nc'. Falling back to 'get' method to retrieve the entire IDS.
equilibrium_ids = get_ids_time_slice(
/home/runner/work/imas/imas/src/cherab/imas/plasma/equilibrium.py:179: RuntimeWarning: The 'get_slice' method is not implemented for the URI '/home/runner/.cache/cherab/imas/iter_scenario_53298_seq1_DD4_mod.nc'. Falling back to 'get' method to retrieve the entire IDS.
equilibrium_ids = get_ids_time_slice(
Skipping ion bundle W ion_bundle (z=2-6) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=2) is not available.
Skipping ion bundle W ion_bundle (z=7-12) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=7) is not available.
Skipping ion bundle W ion_bundle (z=13-22) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=13) is not available.
Skipping ion bundle W ion_bundle (z=23-73) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=23) is not available.
Warning! Unable to verify that the cell nodes are in the winding order.
Skipping ion bundle W ion_bundle (z=2-6) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=2) is not available.
Skipping ion bundle W ion_bundle (z=7-12) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=7) is not available.
Skipping ion bundle W ion_bundle (z=13-22) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=13) is not available.
Skipping ion bundle W ion_bundle (z=23-73) due to error in solving coronal equilibrium: Requested ionisation rate (element=W, charge=23) is not available.
Warning! Using average ion temperature for the W ion_bundle (z=2-6).
Warning! Using average ion temperature for the W ion_bundle (z=7-12).
Warning! Using average ion temperature for the W ion_bundle (z=13-22).
Warning! Using average ion temperature for the W ion_bundle (z=23-73).
Warning! Using average ion temperature for the D ion (z=+1).
Warning! Using average ion temperature for the He ion (z=+1).
Warning! Using average ion temperature for the He ion (z=+2).
Warning! Using average ion temperature for the Ne ion (z=+1).
Warning! Using average ion temperature for the Ne ion (z=+2).
Warning! Using average ion temperature for the Ne ion (z=+3).
Warning! Using average ion temperature for the Ne ion (z=+4).
Warning! Using average ion temperature for the Ne ion (z=+5).
Warning! Using average ion temperature for the Ne ion (z=+6).
Warning! Using average ion temperature for the Ne ion (z=+7).
Warning! Using average ion temperature for the Ne ion (z=+8).
Warning! Using average ion temperature for the Ne ion (z=+9).
Warning! Using average ion temperature for the Ne ion (z=+10).
Warning! Using average ion temperature for the W ion (z=+1).
Warning! Using average ion temperature for the W ion (z=+74).
Warning! Species of type 'ion_bundle' are currently not supported.
The following species will be skipped: tungsten; tungsten; tungsten; tungsten
Warning! Species of type 'ion_bundle' are currently not supported.
The following species will be skipped: tungsten; tungsten; tungsten; tungsten
Warning! Species of type 'molecule' are currently not supported.
The following species will be skipped: deuteriumdeuterium
Show which species the plasma is composed of.
[5]:
# Organize plasma species by symbol and charge
species = {}
for s in plasma.composition:
symbol = s.element.symbol
if symbol not in species:
species[symbol] = []
species[symbol].append(s)
# Sort species of each element by charge
for symbol in species:
species[symbol].sort(key=lambda x: (x.element.atomic_number, x.element.atomic_weight, x.charge))
# Print table of plasma species
table = Table(title="Plasma Species")
table.add_column("Element", style="cyan", justify="right")
table.add_column("Charge", justify="left")
for symbol in species:
charges = [s.charge for s in species[symbol]]
charges_split = [charges[i : i + 10] for i in range(0, len(charges), 10)]
charges_strs = [", ".join(f"{c:d}+" for c in split) for split in charges_split]
charges_str = "\n".join(charges_strs)
table.add_row(symbol, charges_str)
rprint(table)
Plasma Species ┏━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓ ┃ Element ┃ Charge ┃ ┡━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩ │ D │ 0+, 1+ │ │ T │ 0+, 1+ │ │ He │ 0+, 1+, 2+ │ │ Ne │ 0+, 1+, 2+, 3+, 4+, 5+, 6+, 7+, 8+, 9+ │ │ │ 10+ │ │ W │ 0+, 1+, 74+ │ └─────────┴────────────────────────────────────────┘
Construct emission models¶
Apply emission models to plasma species¶
Here we create emission models for Bremsstrahlung, excitation lines, and recombination lines for the main plasma species (D, T, He, Ne, W).
We consider all excitation/recombination lines stored in the OpenADAS adf15 files for these species, which are saved in the ~/.cherab/openadas/repository/pec/... directory after downloading the files from OpenADAS.
[6]:
import json
from pathlib import Path
from cherab.openadas.repository.utility import DEFAULT_REPOSITORY_PATH
lines_exc = []
lines_rec = []
for elm in [
elements.deuterium,
elements.tritium,
elements.helium,
elements.neon,
# elements.tungsten, # TODO: uncomment once https://github.com/cherab/core/pull/497 is resolved
]:
ion_elm = elm.element if isinstance(elm, Isotope) else elm
# === Excitation Lines ===
dir_exc = Path(DEFAULT_REPOSITORY_PATH) / f"pec/excitation/{ion_elm.symbol.lower()}"
paths_pec_exc = list(dir_exc.glob("*.json"))
if len(paths_pec_exc) == 0:
raise ValueError(
f"No excitation data found for element {ion_elm.symbol} in the repository {dir_exc}"
)
for path in paths_pec_exc:
with path.open("r") as f:
data_exc = json.load(f)
charge = int(path.stem)
for transition in data_exc.keys():
lines_exc.append(
ExcitationLine(
Line(elm, charge, tuple(transition.split(" -> "))),
),
)
# === Recombination Lines ===
if ion_elm == elements.tungsten:
# Skip recombination lines for tungsten since they are not available in the repository
continue
dir_rec = Path(DEFAULT_REPOSITORY_PATH) / f"pec/recombination/{ion_elm.symbol.lower()}"
paths_pec_rec = list(dir_rec.glob("*.json"))
if len(paths_pec_rec) == 0:
raise ValueError(
f"No recombination data found for element {ion_elm.symbol} in the repository {dir_rec}"
)
for path in paths_pec_rec:
with path.open("r") as f:
data_rec = json.load(f)
charge = int(path.stem)
for transition in data_rec.keys():
lines_rec.append(
RecombinationLine(
Line(elm, charge, tuple(transition.split(" -> "))),
),
)
Apply the Bremsstrahlung and line emission models to the plasma object.
[7]:
plasma.models = [
Bremsstrahlung(),
*lines_exc,
*lines_rec,
]
Visualize emission profiles¶
Let us visualize the total emission from all processes on a 2D grid in the R-Z plane. Due to the heavy computational load, we limit the grid resolution to 50 mm and wavelength range to 0.01–1000 nm with 50 bins per each decade.
[8]:
# Grid specification
R_MIN, R_MAX = 4.0, 8.5
Z_MIN, Z_MAX = -4.5, 4.4
RES = 50.0e-3 # resolution of grid in [m]
n_r = round((R_MAX - R_MIN) / RES) + 1
n_z = round((Z_MAX - Z_MIN) / RES) + 1
extent = (R_MIN, R_MAX, Z_MIN, Z_MAX)
# Define wavelength integration bands as (min_wavelength_nm, max_wavelength_nm).
# We split the full 0.01–1000 nm range into decade intervals.
#
# Why this structure:
# - Raysect/Cherab spectra use linearly spaced bins within each Spectrum object.
# - Using multiple adjacent decade bands approximates logarithmic coverage overall,
# while preserving linear binning locally in each band.
wavelength_bands = [
(0.01, 0.1),
(0.1, 1.0),
(1.0, 10.0),
(10.0, 100.0),
(100.0, 1000.0),
]
wavelength_bins = 50 # number of bins per decade band
Sample emission at each grid point integrating over the defined wavelength bands.
[9]:
class Emission2DSample:
def __init__(
self,
plasma,
r_pts,
z_pts,
wavelength_bands,
wavelength_bins,
) -> None:
self.plasma = plasma
self.r_pts = r_pts
self.z_pts = z_pts
self.indices = list(np.ndindex(len(r_pts), len(z_pts)))
self.wavelength_bands = wavelength_bands
self.wavelength_bins = wavelength_bins
self.engine = MulticoreEngine()
self.emission_2d = np.zeros((len(r_pts), len(z_pts)))
def run(self) -> None:
self.engine.run(
self.indices,
self._render,
self._update,
)
def _render(self, idx) -> float:
i_r, i_z = idx
x = self.r_pts[i_r]
z = self.z_pts[i_z]
n_e = self.plasma.electron_distribution.density(x, 0, z)
if n_e <= 0:
return idx, 0.0
total = 0.0
for min_wl, max_wl in self.wavelength_bands:
spectrum = Spectrum(
min_wavelength=min_wl,
max_wavelength=max_wl,
bins=self.wavelength_bins,
)
total += self._emission(x, 0, z, spectrum)
return idx, total
def _update(self, packed_result) -> None:
(i_r, i_z), value = packed_result
self.emission_2d[i_r, i_z] = value
def _emission(self, x, y, z, spectrum: Spectrum) -> float:
"""Calculate total emission at a point for a given spectrum."""
for model in self.plasma.models:
spectrum = model.emission(
Point3D(x, y, z),
Vector3D(0, 1, 0),
spectrum,
)
return spectrum.total()
# Get R, Z sampling points
r_pts, _, z_pts, _ = sample3d(
lambda x, y, z: 0,
(R_MIN, R_MAX, n_r),
(0, 0, 1),
(Z_MIN, Z_MAX, n_z),
)
# Create emission sampler and run
emission_sampler = Emission2DSample(
plasma,
r_pts,
z_pts,
wavelength_bands,
wavelength_bins,
)
emission_sampler.run()
Plot the 2-D emission map
[10]:
emission_2d = emission_sampler.emission_2d
emission_2d[emission_2d <= 0] = np.nan # for log plotting
fig, ax = uplt.subplots()
ax.imshow(
emission_2d.T,
discrete=False,
extent=extent,
origin="lower",
norm="log",
cmap="rocket",
colorbar="r",
colorbar_kw=dict(
tickdir="out",
formatter="log",
label="[W/m³/sr]",
),
)
ax.format(
aspect="equal",
title=f"Emissivity integrated over {wavelength_bands[0][0]:.3f} – {wavelength_bands[-1][1]:.0f} nm",
xlabel="$R$ [m]",
ylabel="$Z$ [m]",
xlocator=1,
ylocator=1,
)
[11]:
Total radiated power: 876 MW
Measure line-of-sight Spectra¶
Here we measure line-of-sight spectra through the above modeled emission.
[12]:
class MeasureLoS:
def __init__(self, world) -> None:
self.engine = MulticoreEngine()
self.world = world
self.results: list[Spectrum] = []
def run(self, rays) -> list[Spectrum]:
self.results.clear()
self.engine.run(rays, self.render, self.update)
result = sorted(
self.results,
key=lambda x: x.min_wavelength,
)
return result
def render(self, ray: Ray) -> Spectrum:
return ray.trace(self.world)
def update(self, result: Spectrum) -> None:
self.results.append(result)
measure = MeasureLoS(world)
[13]:
# Generate rays with defined wavelength bands
origin = Point3D(7.5, 0, 4.0)
endpoint = Point3D(4.8, 0, -4.2)
direction = origin.vector_to(endpoint).normalise()
rays = [
Ray(
origin=origin,
direction=direction,
bins=wavelength_bins,
max_wavelength=band[1],
min_wavelength=band[0],
)
for band in wavelength_bands
]
# Measure line of sight spectrum
spectra = measure.run(rays)
Visualize measured spectra¶
Let us visualize the ray trajectory projected onto the R-Z plane along with the measured spectra.
[14]:
fig, axs = uplt.subplots([1, 2, 2], share=False)
# Plot 2-D emissivity map
image = axs[0].imshow(
emission_2d.T,
extent=extent,
origin="lower",
norm="log",
cmap="rocket",
)
axs[0].colorbar(
image,
formatter="log",
tickdir="out",
loc="lr",
orientation="vertical",
ticklabelsize="small",
length=5,
frame=False,
)
# Plot line of sight
axs[0].plot(
[origin.x, endpoint.x],
[origin.z, endpoint.z],
color="C0",
label="Line of Sight",
legend="ur",
)
axs[0].format(
aspect="equal",
xlabel="$R$ [m]",
ylabel="$Z$ [m]",
xreverse=False,
ylocator=1,
xlocator=1,
title="Emissivity [W/m³/sr]",
gridalpha=0.1,
)
# Plot 1-D measured spectrum
for s in spectra:
axs[1].loglog(s.wavelengths, s.samples, color="C0")
axs[1].format(
title="Ray-traced emission spectrum",
xlabel="Wavelength [nm]",
ylabel="Spectral Radiance [W/m²/sr/nm]",
yformatter="log",
grid=True,
gridalpha=0.3,
)
[15]:
Total radiance along line of sight: 4.31e+05 W/m²/sr
Let us also plot the bar chart of the integrated spectrum for each wavelength band to see which wavelength bands dominate the total emission.
[16]:
fig, axs = uplt.subplots([[1, 2, 2], [1, 3, 3]], share=False)
# Plot 2-D emissivity map
image = axs[0].imshow(
emission_2d.T,
extent=extent,
origin="lower",
norm="log",
cmap="rocket",
)
axs[0].colorbar(
image,
formatter="log",
tickdir="out",
loc="lr",
orientation="vertical",
ticklabelsize="small",
length=5,
frame=False,
)
# Plot line of sight
axs[0].plot(
[origin.x, endpoint.x],
[origin.z, endpoint.z],
color="C0",
label="Line of Sight",
legend="ur",
)
axs[0].format(
aspect="equal",
xlabel="$R$ [m]",
ylabel="$Z$ [m]",
xreverse=False,
ylocator=1,
xlocator=1,
title="Emissivity [W/m³/sr]",
gridalpha=0.1,
)
# Plot 1-D measured spectrum
for s in spectra:
axs[1].loglog(s.wavelengths, s.samples, color="C0")
# Plot histogram of integrated spectrum
axs[2].bar(
np.arange(len(spectra)),
[s.total() for s in spectra]
)
axs[1:].format(
yscale="log",
yformatter="log",
grid=True,
gridalpha=0.3,
)
axs[1].format(
title="Ray-traced emission spectrum",
xlabel="Wavelength [nm]",
ylabel="Spectral Radiance [W/m²/sr/nm]",
)
axs[2].format(
xlabel="Wavelength [nm]",
title="Ray-traced emission spectrum integrated over wavelength bands",
ylabel="Radiance [W/m²/sr]",
xlocator=range(len(wavelength_bands)),
xticklabels=[f"{band[0]} – {band[1]}" for band in wavelength_bands],
)
axs.format(
abc="(a)",
abcloc="ul",
abcborder=False,
)
In this example, 1–10 nm wavelength band dominates the total emission. The radiance ratio between the soft X‑ray (0.01–10 nm) and UV/visible (10–1000 nm) bands is given by the following calculation:
[17]:
radiance_soft_xray = sum([s.total() for s in spectra[0:3]])
radiance_uv_visible = sum([s.total() for s in spectra[3:]])
print(f"soft X-ray radiance: {radiance_soft_xray:.3e} W/m²/sr")
print(f"UV/visible radiance: {radiance_uv_visible:.3e} W/m²/sr")
print(f"radiance ratio (soft X-ray / UV-visible): {radiance_soft_xray / radiance_uv_visible:.2f}")
soft X-ray radiance: 2.807e+05 W/m²/sr
UV/visible radiance: 1.501e+05 W/m²/sr
radiance ratio (soft X-ray / UV-visible): 1.87