Calculate and Compare Spectra for Multiple Molecules

This example demonstrates how to:

• Compute a spectrum with multiple molecules simultaneously

• Combine precomputed spectra of individual molecules

• Compare both approaches and highlight performance trade-offs

CO2-hitran-300K-#7776 new quantities added: ['emisscoeff']
H2O-hitran-300K-#9216 new quantities added: ['emisscoeff']
/home/docs/checkouts/readthedocs.org/user_builds/radis/checkouts/master/radis/spectrum/rescale.py:2488: UserWarning: Large rescaling from 0.1 to 0.8. There may be significant changes in pressure broadening coefficients.You should calculate a new spectrum instead.
  warn(
Single Iteration Performance Results:
Direct Calculation time: 0.2338 seconds
Precomputed then Combined time: 0.1564 seconds
/home/docs/checkouts/readthedocs.org/user_builds/radis/checkouts/master/radis/misc/warning.py:427: UnevenWaverangeWarning: When resampling the spectrum, the new waverange had unequal spacing.
  warnings.warn(WarningType(message))

(<Figure size 640x480 with 2 Axes>, [<Axes: >, <Axes: xlabel='Wavelength (nm)'>])

import time

from radis import calc_spectrum, config, plot_diff

# Configure RADIS to handle missing broadening coefficients
config["MISSING_BROAD_COEF"] = "air"

# Direct Calculation
start_time_direct = time.perf_counter()
direct_h2o_co2 = calc_spectrum(
    wavelength_min=4800,
    wavelength_max=5000,
    Tgas=300,
    path_length=0.1,
    mole_fraction={"H2O": 0.2, "CO2": 0.8},
    isotope="1,2",
    verbose=False,
)
time_direct = time.perf_counter() - start_time_direct

# Precompute and Combine Individual Spectra
start_time_precomputed = time.perf_counter()
s_h2o = calc_spectrum(
    wavelength_min=4800,
    wavelength_max=5000,
    Tgas=300,
    path_length=0.1,
    mole_fraction={"H2O": 0.1},
    isotope="1,2",
    verbose=False,
)
s_co2 = calc_spectrum(
    wavelength_min=4800,
    wavelength_max=5000,
    Tgas=300,
    path_length=0.1,
    mole_fraction={"CO2": 0.1},
    isotope="1,2",
    verbose=False,
)
s_h2o_selected = s_h2o.take("abscoeff")
s_co2_selected = s_co2.take("abscoeff")
combined_spectrum = s_h2o_selected.rescale_mole_fraction(
    0.2
) + s_co2_selected.rescale_mole_fraction(0.8)
time_precomputed = time.perf_counter() - start_time_precomputed

print("Single Iteration Performance Results:")
print(f"Direct Calculation time: {time_direct:.4f} seconds")
print(f"Precomputed then Combined time: {time_precomputed:.4f} seconds")

# Convert combined_spectrum from wavenumber to wavelength in nm
combined_spectrum_nm = combined_spectrum.resample(
    combined_spectrum.get_wavelength(), "nm", inplace=False
)
# Plot the difference between the combined spectrum and the directly calculated spectrum.
plot_diff(combined_spectrum_nm, direct_h2o_co2)