RadiationModel.cpp

Go to the documentation of this file.

 
#include "RadiationModel.h"
#include "BufferIndexing.h"
#include <climits>
#include <cmath>
#include <ctime>
#include <fstream>
#include <iomanip>
#include <sstream>
#include <unordered_set>
 
using namespace helios;
 
RadiationModel::RadiationModel(helios::Context *context_a) {
 
    context = context_a;
 
    // Asset directory registration removed - now using HELIOS_BUILD resolution
 
    // All default values set here
 
    message_flag = true;
 
    directRayCount_default = 100;
    diffuseRayCount_default = 1000;
 
    diffuseFlux_default = -1.f;
 
    minScatterEnergy_default = 0.1;
    scatteringDepth_default = 0;
 
    rho_default = 0.f;
    tau_default = 0.f;
    eps_default = 1.f;
 
    kappa_default = 1.f;
    sigmas_default = 0.f;
 
    temperature_default = 300;
 
    periodic_flag = make_vec2(0, 0);
 
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/camera_spectral_library.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/light_spectral_library.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/soil_surface_spectral_library.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/leaf_surface_spectral_library.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/bark_surface_spectral_library.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/fruit_surface_spectral_library.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/solar_spectrum_ASTMG173.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/color_board/Calibrite_ColorChecker_Classic_colorboard.xml").string());
    spectral_library_files.push_back(helios::resolvePluginAsset("radiation", "spectral_data/color_board/DGK_DKK_colorboard.xml").string());
 
    // Initialize backend abstraction layer
    backend = helios::RayTracingBackend::create("optix6");
    backend->initialize();
}
 
RadiationModel::~RadiationModel() {
    // Backend's unique_ptr will automatically clean up OptiX context
}
 
void RadiationModel::disableMessages() {
    message_flag = false;
}
 
void RadiationModel::enableMessages() {
    message_flag = true;
}
 
void RadiationModel::optionalOutputPrimitiveData(const char *label) {
 
    if (strcmp(label, "reflectivity") == 0 || strcmp(label, "transmissivity") == 0) {
        output_prim_data.emplace_back(label);
    } else {
        std::cout << "WARNING (RadiationModel::optionalOutputPrimitiveData): unknown output primitive data " << label << std::endl;
    }
}
 
void RadiationModel::setDirectRayCount(const std::string &label, size_t N) {
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::setDirectRayCount): Cannot set ray count for band '" + label + "' because it is not a valid band.");
    }
    radiation_bands.at(label).directRayCount = N;
}
 
void RadiationModel::setDiffuseRayCount(const std::string &label, size_t N) {
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseRayCount): Cannot set ray count for band '" + label + "' because it is not a valid band.");
    }
    radiation_bands.at(label).diffuseRayCount = N;
}
 
void RadiationModel::setDiffuseRadiationFlux(const std::string &label, float flux) {
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseRadiationFlux): Cannot set flux value for band '" + label + "' because it is not a valid band.");
    }
    radiation_bands.at(label).diffuseFlux = flux;
}
 
void RadiationModel::setDiffuseRadiationExtinctionCoeff(const std::string &label, float K, const SphericalCoord &peak_dir) {
    setDiffuseRadiationExtinctionCoeff(label, K, sphere2cart(peak_dir));
}
 
void RadiationModel::setDiffuseRadiationExtinctionCoeff(const std::string &label, float K, const vec3 &peak_dir) {
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseRadiationExtinctionCoeff): Cannot set diffuse extinction value for band '" + label + "' because it is not a valid band.");
    }
 
    vec3 dir = peak_dir;
    dir.normalize();
 
    int N = 100;
    float norm = 0.f;
    for (int j = 0; j < N; j++) {
        for (int i = 0; i < N; i++) {
            float theta = 0.5f * M_PI / float(N) * (0.5f + float(i));
            float phi = 2.f * M_PI / float(N) * (0.5f + float(j));
            vec3 n = sphere2cart(make_SphericalCoord(0.5f * M_PI - theta, phi));
 
            float psi = acos_safe(n * dir);
            float fd;
            if (psi < M_PI / 180.f) {
                fd = powf(M_PI / 180.f, -K);
            } else {
                fd = powf(psi, -K);
            }
 
            norm += fd * cosf(theta) * sinf(theta) * M_PI / float(N * N);
            // note: the multipication factors are dtheta*dphi/pi = (0.5*pi/N)*(2*pi/N)/pi = pi/N^2
        }
    }
 
    radiation_bands.at(label).diffuseExtinction = K;
    radiation_bands.at(label).diffusePeakDir = dir;
    radiation_bands.at(label).diffuseDistNorm = 1.f / norm;
}
 
void RadiationModel::setDiffuseSpectrumIntegral(float spectrum_integral) {
 
    if (spectrum_integral < 0) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Spectrum integral must be non-negative.");
    } else if (global_diffuse_spectrum.empty()) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Global diffuse spectrum has not been set. Call setDiffuseSpectrum() first.");
    }
 
    // Scale the global spectrum
    float current_integral = integrateSpectrum(global_diffuse_spectrum);
    if (current_integral > 0) {
        float scale_factor = spectrum_integral / current_integral;
        for (vec2 &wavelength: global_diffuse_spectrum) {
            wavelength.y *= scale_factor;
        }
    }
 
    // Apply scaled spectrum to all existing bands
    for (auto &band: radiation_bands) {
        band.second.diffuse_spectrum = global_diffuse_spectrum;
    }
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::setDiffuseSpectrumIntegral(float spectrum_integral, float wavelength1, float wavelength2) {
 
    if (spectrum_integral < 0) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Spectrum integral must be non-negative.");
    } else if (global_diffuse_spectrum.empty()) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Global diffuse spectrum has not been set. Call setDiffuseSpectrum() first.");
    }
 
    // Scale the global spectrum based on the integral within the specified wavelength range
    float current_integral = integrateSpectrum(global_diffuse_spectrum, wavelength1, wavelength2);
    if (current_integral > 0) {
        float scale_factor = spectrum_integral / current_integral;
        for (vec2 &wavelength: global_diffuse_spectrum) {
            wavelength.y *= scale_factor;
        }
    }
 
    // Apply scaled spectrum to all existing bands
    for (auto &band: radiation_bands) {
        band.second.diffuse_spectrum = global_diffuse_spectrum;
    }
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::setDiffuseSpectrumIntegral(const std::string &band_label, float spectrum_integral) {
 
    if (spectrum_integral < 0) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Source integral must be non-negative.");
    } else if (!doesBandExist(band_label)) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Cannot set integral for band '" + band_label + "' because it is not a valid band.");
    } else if (radiation_bands.at(band_label).diffuse_spectrum.empty()) {
        std::cerr << "WARNING (RadiationModel::setDiffuseSpectrumIntegral): Diffuse spectral distribution has not been set for radiation band '" + band_label + "'. Cannot set its integral." << std::endl;
        return;
    }
 
    float current_integral = integrateSpectrum(radiation_bands.at(band_label).diffuse_spectrum);
 
    for (vec2 &wavelength: radiation_bands.at(band_label).diffuse_spectrum) {
        wavelength.y *= spectrum_integral / current_integral;
    }
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::setDiffuseSpectrumIntegral(const std::string &band_label, float spectrum_integral, float wavelength1, float wavelength2) {
 
    if (spectrum_integral < 0) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Source integral must be non-negative.");
    } else if (!doesBandExist(band_label)) {
        helios_runtime_error("ERROR (RadiationModel::setDiffuseSpectrumIntegral): Cannot set integral for band '" + band_label + "' because it is not a valid band.");
    }
 
    float current_integral = integrateSpectrum(radiation_bands.at(band_label).diffuse_spectrum, wavelength1, wavelength2);
 
    for (vec2 &wavelength: radiation_bands.at(band_label).diffuse_spectrum) {
        wavelength.y *= spectrum_integral / current_integral;
    }
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::addRadiationBand(const std::string &label) {
 
    if (radiation_bands.find(label) != radiation_bands.end()) {
        std::cerr << "WARNING (RadiationModel::addRadiationBand): Radiation band " << label << " has already been added. Skipping this call to addRadiationBand()." << std::endl;
        return;
    }
 
    RadiationBand band(label, directRayCount_default, diffuseRayCount_default, diffuseFlux_default, scatteringDepth_default, minScatterEnergy_default);
 
    // Apply global diffuse spectrum if one was set
    if (!global_diffuse_spectrum.empty()) {
        band.diffuse_spectrum = global_diffuse_spectrum;
    }
 
    radiation_bands.emplace(label, band);
 
    // Initialize all radiation source fluxes
    for (auto &source: radiation_sources) {
        source.source_fluxes[label] = -1.f;
    }
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::addRadiationBand(const std::string &label, float wavelength1, float wavelength2) {
 
    if (radiation_bands.find(label) != radiation_bands.end()) {
        std::cerr << "WARNING (RadiationModel::addRadiationBand): Radiation band " << label << " has already been added. Skipping this call to addRadiationBand()." << std::endl;
        return;
    } else if (wavelength1 > wavelength2) {
        helios_runtime_error("ERROR (RadiationModel::addRadiationBand): The upper wavelength bound for a band must be greater than the lower bound.");
    } else if (wavelength2 - wavelength1 < 1) {
        helios_runtime_error("ERROR (RadiationModel::addRadiationBand): The waveband range of a radiation band must be at least 1 nm.");
    }
 
    RadiationBand band(label, directRayCount_default, diffuseRayCount_default, diffuseFlux_default, scatteringDepth_default, minScatterEnergy_default);
 
    band.wavebandBounds = make_vec2(wavelength1, wavelength2);
 
    // Apply global diffuse spectrum if one was set
    if (!global_diffuse_spectrum.empty()) {
        band.diffuse_spectrum = global_diffuse_spectrum;
    }
 
    radiation_bands.emplace(label, band);
 
    // Initialize all radiation source fluxes
    for (auto &source: radiation_sources) {
        source.source_fluxes[label] = -1.f;
    }
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::copyRadiationBand(const std::string &old_label, const std::string &new_label) {
 
    if (!doesBandExist(old_label)) {
        helios_runtime_error("ERROR (RadiationModel::copyRadiationBand): Cannot copy band " + old_label + " because it does not exist.");
    }
 
    vec2 waveBounds = radiation_bands.at(old_label).wavebandBounds;
 
    copyRadiationBand(old_label, new_label, waveBounds.x, waveBounds.y);
}
 
void RadiationModel::copyRadiationBand(const std::string &old_label, const std::string &new_label, float wavelength_min, float wavelength_max) {
 
    if (!doesBandExist(old_label)) {
        helios_runtime_error("ERROR (RadiationModel::copyRadiationBand): Cannot copy band " + old_label + " because it does not exist.");
    }
 
    RadiationBand band = radiation_bands.at(old_label);
    band.label = new_label;
    band.wavebandBounds = make_vec2(wavelength_min, wavelength_max);
 
    radiation_bands.emplace(new_label, band);
 
    // copy source fluxes
    for (auto &source: radiation_sources) {
        source.source_fluxes[new_label] = source.source_fluxes.at(old_label);
    }
 
    radiativepropertiesneedupdate = true;
}
 
bool RadiationModel::doesBandExist(const std::string &label) const {
    if (radiation_bands.find(label) == radiation_bands.end()) {
        return false;
    } else {
        return true;
    }
}
 
void RadiationModel::disableEmission(const std::string &label) {
 
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::disableEmission): Cannot disable emission for band '" + label + "' because it is not a valid band.");
    }
 
    radiation_bands.at(label).emissionFlag = false;
}
 
void RadiationModel::enableEmission(const std::string &label) {
 
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::enableEmission): Cannot disable emission for band '" + label + "' because it is not a valid band.");
    }
 
    radiation_bands.at(label).emissionFlag = true;
}
 
uint RadiationModel::addCollimatedRadiationSource() {
 
    return addCollimatedRadiationSource(make_vec3(0, 0, 1));
}
 
uint RadiationModel::addCollimatedRadiationSource(const SphericalCoord &direction) {
    return addCollimatedRadiationSource(sphere2cart(direction));
}
 
uint RadiationModel::addCollimatedRadiationSource(const vec3 &direction) {
 
    if (direction.magnitude() == 0) {
        helios_runtime_error("ERROR (RadiationModel::addCollimatedRadiationSource): Invalid collimated source direction. Direction vector should not have length of zero.");
    }
 
    uint Nsources = radiation_sources.size() + 1;
    if (Nsources > 256) {
        helios_runtime_error("ERROR (RadiationModel::addCollimatedRadiationSource): A maximum of 256 radiation sources are allowed.");
    }
 
    bool warn_multiple_suns = false;
    for (auto &source: radiation_sources) {
        if (source.source_type == RADIATION_SOURCE_TYPE_COLLIMATED || source.source_type == RADIATION_SOURCE_TYPE_SUN_SPHERE) {
            warn_multiple_suns = true;
        }
    }
    if (warn_multiple_suns) {
        std::cerr << "WARNING (RadiationModel::addCollimatedRadiationSource): Multiple sun sources have been added to the radiation model. This may lead to unintended behavior." << std::endl;
    }
 
    RadiationSource collimated_source(direction);
 
    // initialize fluxes
    for (const auto &band: radiation_bands) {
        collimated_source.source_fluxes[band.first] = -1.f;
    }
 
    radiation_sources.emplace_back(collimated_source);
 
    radiativepropertiesneedupdate = true;
 
    return Nsources - 1;
}
 
uint RadiationModel::addSphereRadiationSource(const vec3 &position, float radius) {
 
    if (radius <= 0) {
        helios_runtime_error("ERROR (RadiationModel::addSphereRadiationSource): Spherical radiation source radius must be positive.");
    }
 
    uint Nsources = radiation_sources.size() + 1;
    if (Nsources > 256) {
        helios_runtime_error("ERROR (RadiationModel::addSphereRadiationSource): A maximum of 256 radiation sources are allowed.");
    }
 
    RadiationSource sphere_source(position, 2.f * fabsf(radius));
 
    // initialize fluxes
    for (const auto &band: radiation_bands) {
        sphere_source.source_fluxes[band.first] = -1.f;
    }
 
    radiation_sources.emplace_back(sphere_source);
 
    uint sourceID = Nsources - 1;
 
    if (islightvisualizationenabled) {
        buildLightModelGeometry(sourceID);
    }
 
    radiativepropertiesneedupdate = true;
 
    return sourceID;
}
 
uint RadiationModel::addSunSphereRadiationSource() {
    return addSunSphereRadiationSource(make_vec3(0, 0, 1));
}
 
uint RadiationModel::addSunSphereRadiationSource(const SphericalCoord &sun_direction) {
    return addSunSphereRadiationSource(sphere2cart(sun_direction));
}
 
uint RadiationModel::addSunSphereRadiationSource(const vec3 &sun_direction) {
 
    uint Nsources = radiation_sources.size() + 1;
    if (Nsources > 256) {
        helios_runtime_error("ERROR (RadiationModel::addSunSphereRadiationSource): A maximum of 256 radiation sources are allowed.");
    }
 
    bool warn_multiple_suns = false;
    for (auto &source: radiation_sources) {
        if (source.source_type == RADIATION_SOURCE_TYPE_COLLIMATED || source.source_type == RADIATION_SOURCE_TYPE_SUN_SPHERE) {
            warn_multiple_suns = true;
        }
    }
    if (warn_multiple_suns) {
        std::cerr << "WARNING (RadiationModel::addSunSphereRadiationSource): Multiple sun sources have been added to the radiation model. This may lead to unintended behavior." << std::endl;
    }
 
    RadiationSource sphere_source(150e9 * sun_direction / sun_direction.magnitude(), 150e9, 2.f * 695.5e6, sigma * powf(5700, 4) / 1288.437f);
 
    // initialize fluxes
    for (const auto &band: radiation_bands) {
        sphere_source.source_fluxes[band.first] = -1.f;
    }
 
    radiation_sources.emplace_back(sphere_source);
 
    radiativepropertiesneedupdate = true;
 
    return Nsources - 1;
}
 
uint RadiationModel::addRectangleRadiationSource(const vec3 &position, const vec2 &size, const vec3 &rotation_rad) {
 
    if (size.x <= 0 || size.y <= 0) {
        helios_runtime_error("ERROR (RadiationModel::addRectangleRadiationSource): Radiation source size must be positive.");
    }
 
    uint Nsources = radiation_sources.size() + 1;
    if (Nsources > 256) {
        helios_runtime_error("ERROR (RadiationModel::addRectangleRadiationSource): A maximum of 256 radiation sources are allowed.");
    }
 
    RadiationSource rectangle_source(position, size, rotation_rad);
 
    // initialize fluxes
    for (const auto &band: radiation_bands) {
        rectangle_source.source_fluxes[band.first] = -1.f;
    }
 
    radiation_sources.emplace_back(rectangle_source);
 
    uint sourceID = Nsources - 1;
 
    if (islightvisualizationenabled) {
        buildLightModelGeometry(sourceID);
    }
 
    radiativepropertiesneedupdate = true;
 
    return sourceID;
}
 
uint RadiationModel::addDiskRadiationSource(const vec3 &position, float radius, const vec3 &rotation_rad) {
 
    if (radius <= 0) {
        helios_runtime_error("ERROR (RadiationModel::addDiskRadiationSource): Disk radiation source radius must be positive.");
    }
 
    uint Nsources = radiation_sources.size() + 1;
    if (Nsources > 256) {
        helios_runtime_error("ERROR (RadiationModel::addDiskRadiationSource): A maximum of 256 radiation sources are allowed.");
    }
 
    RadiationSource disk_source(position, radius, rotation_rad);
 
    // initialize fluxes
    for (const auto &band: radiation_bands) {
        disk_source.source_fluxes[band.first] = -1.f;
    }
 
    radiation_sources.emplace_back(disk_source);
 
    uint sourceID = Nsources - 1;
 
    if (islightvisualizationenabled) {
        buildLightModelGeometry(sourceID);
    }
 
    radiativepropertiesneedupdate = true;
 
    return sourceID;
}
 
void RadiationModel::deleteRadiationSource(uint sourceID) {
 
    if (sourceID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::deleteRadiationSource): Source ID out of bounds. Only " + std::to_string(radiation_sources.size() - 1) + " radiation sources have been created.");
    }
 
    radiation_sources.erase(radiation_sources.begin() + sourceID);
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::setSourceSpectrumIntegral(uint source_ID, float source_integral) {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::setSourceSpectrumIntegral): Source ID out of bounds. Only " + std::to_string(radiation_sources.size() - 1) + " radiation sources have been created.");
    } else if (source_integral < 0) {
        helios_runtime_error("ERROR (RadiationModel::setSourceIntegral): Source integral must be non-negative.");
    }
 
    float current_integral = integrateSpectrum(radiation_sources.at(source_ID).source_spectrum);
 
    for (vec2 &wavelength: radiation_sources.at(source_ID).source_spectrum) {
        wavelength.y *= source_integral / current_integral;
    }
}
 
void RadiationModel::setSourceSpectrumIntegral(uint source_ID, float source_integral, float wavelength1, float wavelength2) {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::setSourceSpectrumIntegral): Source ID out of bounds. Only " + std::to_string(radiation_sources.size() - 1) + " radiation sources have been created.");
    } else if (source_integral < 0) {
        helios_runtime_error("ERROR (RadiationModel::setSourceSpectrumIntegral): Source integral must be non-negative.");
    } else if (radiation_sources.at(source_ID).source_spectrum.empty()) {
        std::cout << "WARNING (RadiationModel::setSourceSpectrumIntegral): Spectral distribution has not been set for radiation source. Cannot set its integral." << std::endl;
        return;
    }
 
    RadiationSource &source = radiation_sources.at(source_ID);
 
    float old_integral = integrateSpectrum(source.source_spectrum, wavelength1, wavelength2);
 
    for (vec2 &wavelength: source.source_spectrum) {
        wavelength.y *= source_integral / old_integral;
    }
}
 
void RadiationModel::setSourceFlux(uint source_ID, const std::string &label, float flux) {
 
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::setSourceFlux): Cannot add set source flux for band '" + label + "' because it is not a valid band.");
    } else if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::setSourceFlux): Source ID out of bounds. Only " + std::to_string(radiation_sources.size() - 1) + " radiation sources have been created.");
    } else if (flux < 0) {
        helios_runtime_error("ERROR (RadiationModel::setSourceFlux): Source flux must be non-negative.");
    }
 
    radiation_sources.at(source_ID).source_fluxes[label] = flux * radiation_sources.at(source_ID).source_flux_scaling_factor;
}
 
void RadiationModel::setSourceFlux(const std::vector<uint> &source_ID, const std::string &band_label, float flux) {
    for (auto ID: source_ID) {
        setSourceFlux(ID, band_label, flux);
    }
}
 
float RadiationModel::getSourceFlux(uint source_ID, const std::string &label) const {
 
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::getSourceFlux): Cannot get source flux for band '" + label + "' because it is not a valid band.");
    } else if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::getSourceFlux): Source ID out of bounds. Only " + std::to_string(radiation_sources.size() - 1) + " radiation sources have been created.");
    } else if (radiation_sources.at(source_ID).source_fluxes.find(label) == radiation_sources.at(source_ID).source_fluxes.end()) {
        helios_runtime_error("ERROR (RadiationModel::getSourceFlux): Cannot get flux for source #" + std::to_string(source_ID) + " because radiative band '" + label + "' does not exist.");
    }
 
    const RadiationSource &source = radiation_sources.at(source_ID);
 
    if (!source.source_spectrum.empty() && source.source_fluxes.at(label) < 0.f) { // source spectrum was specified (and not overridden by setting source flux manually)
        vec2 wavebounds = radiation_bands.at(label).wavebandBounds;
        if (wavebounds == make_vec2(0, 0)) {
            wavebounds = make_vec2(source.source_spectrum.front().x, source.source_spectrum.back().x);
        }
        return integrateSpectrum(source.source_spectrum, wavebounds.x, wavebounds.y) * source.source_flux_scaling_factor;
    } else if (source.source_fluxes.at(label) < 0.f) {
        return 0;
    }
 
    return source.source_fluxes.at(label);
}
 
void RadiationModel::setSourceSpectrum(uint source_ID, const std::vector<helios::vec2> &spectrum) {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::setSourceSpectrum): Cannot add radiation spectra for this source because it is not a valid radiation source ID.\n");
    }
 
    // validate spectrum
    for (auto s = 0; s < spectrum.size(); s++) {
        // check that wavelengths are monotonic
        if (s > 0 && spectrum.at(s).x <= spectrum.at(s - 1).x) {
            helios_runtime_error("ERROR (RadiationModel::setSourceSpectrum): Source spectral data validation failed. Wavelengths must increase monotonically.");
        }
        // check that wavelength is within a reasonable range
        if (spectrum.at(s).x < 0 || spectrum.at(s).x > 100000) {
            helios_runtime_error("ERROR (RadiationModel::setSourceSpectrum): Source spectral data validation failed. Wavelength value of " + std::to_string(spectrum.at(s).x) + " appears to be erroneous.");
        }
        // check that flux is non-negative
        if (spectrum.at(s).y < 0) {
            helios_runtime_error("ERROR (RadiationModel::setSourceSpectrum): Source spectral data validation failed. Flux value at wavelength of " + std::to_string(spectrum.at(s).x) + " appears is negative.");
        }
    }
 
    radiation_sources.at(source_ID).source_spectrum = spectrum;
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::setSourceSpectrum(const std::vector<uint> &source_ID, const std::vector<helios::vec2> &spectrum) {
    for (auto ID: source_ID) {
        setSourceSpectrum(ID, spectrum);
    }
}
 
void RadiationModel::setSourceSpectrum(uint source_ID, const std::string &spectrum_label) {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::setSourceSpectrum): Cannot add radiation spectra for this source because it is not a valid radiation source ID.\n");
    }
 
    std::vector<vec2> spectrum = loadSpectralData(spectrum_label);
 
    radiation_sources.at(source_ID).source_spectrum = spectrum;
    radiation_sources.at(source_ID).source_spectrum_label = spectrum_label;
    radiation_sources.at(source_ID).source_spectrum_version = context->getGlobalDataVersion(spectrum_label.c_str());
 
    radiativepropertiesneedupdate = true;
}
 
void RadiationModel::setSourceSpectrum(const std::vector<uint> &source_ID, const std::string &spectrum_label) {
    for (auto ID: source_ID) {
        setSourceSpectrum(ID, spectrum_label);
    }
}
 
void RadiationModel::setDiffuseSpectrum(const std::string &spectrum_label) {
 
    std::vector<vec2> spectrum;
 
    // standard solar spectrum
    if (spectrum_label == "ASTMG173") {
        spectrum = loadSpectralData("solar_spectrum_diffuse_ASTMG173");
        global_diffuse_spectrum_label = "solar_spectrum_diffuse_ASTMG173";
    } else {
        spectrum = loadSpectralData(spectrum_label);
        global_diffuse_spectrum_label = spectrum_label;
    }
 
    // Store globally so new bands will also get this spectrum
    global_diffuse_spectrum = spectrum;
    global_diffuse_spectrum_version = context->getGlobalDataVersion(global_diffuse_spectrum_label.c_str());
 
    // Apply to all existing bands
    for (auto &band_pair: radiation_bands) {
        band_pair.second.diffuse_spectrum = spectrum;
    }
 
    radiativepropertiesneedupdate = true;
}
 
float RadiationModel::getDiffuseFlux(const std::string &band_label) const {
 
    if (!doesBandExist(band_label)) {
        helios_runtime_error("ERROR (RadiationModel::getDiffuseFlux): Cannot get diffuse flux for band '" + band_label + "' because it is not a valid band.");
    }
 
    const RadiationBand &band = radiation_bands.at(band_label);
 
    // For emission-enabled bands: spectra are not relevant, only use manual flux
    if (band.emissionFlag) {
        if (band.diffuseFlux >= 0.f) {
            return band.diffuseFlux;
        }
        return 0.f;
    }
 
    // For non-emission bands: check manual flux first, then spectrum
    if (band.diffuseFlux >= 0.f) {
        return band.diffuseFlux;
    }
 
    const std::vector<vec2> &spectrum = band.diffuse_spectrum;
    if (!spectrum.empty()) {
        vec2 wavebounds = band.wavebandBounds;
        if (wavebounds == make_vec2(0, 0)) {
            wavebounds = make_vec2(spectrum.front().x, spectrum.back().x);
        }
        return integrateSpectrum(spectrum, wavebounds.x, wavebounds.y);
    }
 
    return 0.f;
}
 
void RadiationModel::enableLightModelVisualization() {
    islightvisualizationenabled = true;
 
    // build the geometry of any existing sources at this point
    for (int s = 0; s < radiation_sources.size(); s++) {
        buildLightModelGeometry(s);
    }
}
 
void RadiationModel::disableLightModelVisualization() {
    islightvisualizationenabled = false;
    for (auto &UUIDs: source_model_UUIDs) {
        context->deletePrimitive(UUIDs.second);
    }
}
 
void RadiationModel::enableCameraModelVisualization() {
    iscameravisualizationenabled = true;
 
    // build the geometry of any existing cameras at this point
    for (auto &cam: cameras) {
        buildCameraModelGeometry(cam.first);
    }
}
 
void RadiationModel::disableCameraModelVisualization() {
    iscameravisualizationenabled = false;
    for (auto &UUIDs: camera_model_UUIDs) {
        context->deletePrimitive(UUIDs.second);
    }
}
 
void RadiationModel::buildLightModelGeometry(uint sourceID) {
 
    assert(sourceID < radiation_sources.size());
 
    RadiationSource source = radiation_sources.at(sourceID);
    if (source.source_type == RADIATION_SOURCE_TYPE_SPHERE) {
        source_model_UUIDs[sourceID] = context->loadOBJ("SphereLightSource.obj", true);
    } else if (source.source_type == RADIATION_SOURCE_TYPE_SUN_SPHERE) {
        source_model_UUIDs[sourceID] = context->loadOBJ("SphereLightSource.obj", true);
    } else if (source.source_type == RADIATION_SOURCE_TYPE_DISK) {
        source_model_UUIDs[sourceID] = context->loadOBJ("DiskLightSource.obj", true);
        context->scalePrimitive(source_model_UUIDs.at(sourceID), make_vec3(source.source_width.x, source.source_width.y, 0.05f * source.source_width.x));
        std::vector<uint> UUIDs_arrow = context->loadOBJ("Arrow.obj", true);
        source_model_UUIDs.at(sourceID).insert(source_model_UUIDs.at(sourceID).begin(), UUIDs_arrow.begin(), UUIDs_arrow.end());
        context->scalePrimitive(UUIDs_arrow, make_vec3(1, 1, 1) * 0.25f * source.source_width.x);
    } else if (source.source_type == RADIATION_SOURCE_TYPE_RECTANGLE) {
        source_model_UUIDs[sourceID] = context->loadOBJ("RectangularLightSource.obj", true);
        context->scalePrimitive(source_model_UUIDs.at(sourceID), make_vec3(source.source_width.x, source.source_width.y, fmin(0.05f * (source.source_width.x + source.source_width.y), 0.5f * fmin(source.source_width.x, source.source_width.y))));
        std::vector<uint> UUIDs_arrow = context->loadOBJ("Arrow.obj", true);
        source_model_UUIDs.at(sourceID).insert(source_model_UUIDs.at(sourceID).begin(), UUIDs_arrow.begin(), UUIDs_arrow.end());
        context->scalePrimitive(UUIDs_arrow, make_vec3(1, 1, 1) * 0.15f * (source.source_width.x + source.source_width.y));
    } else {
        return;
    }
 
    if (source.source_type == RADIATION_SOURCE_TYPE_SPHERE) {
        context->scalePrimitive(source_model_UUIDs.at(sourceID), make_vec3(source.source_width.x, source.source_width.x, source.source_width.x));
        context->translatePrimitive(source_model_UUIDs.at(sourceID), source.source_position);
    } else if (source.source_type == RADIATION_SOURCE_TYPE_SUN_SPHERE) {
        vec3 center;
        float radius;
        context->getDomainBoundingSphere(center, radius);
        context->scalePrimitive(source_model_UUIDs.at(sourceID), make_vec3(1, 1, 1) * 0.1f * radius);
        vec3 sunvec = source.source_position;
        sunvec.normalize();
        context->translatePrimitive(source_model_UUIDs.at(sourceID), center + sunvec * radius);
    } else {
        context->rotatePrimitive(source_model_UUIDs.at(sourceID), source.source_rotation.x, "x");
        context->rotatePrimitive(source_model_UUIDs.at(sourceID), source.source_rotation.y, "y");
        context->rotatePrimitive(source_model_UUIDs.at(sourceID), source.source_rotation.z, "z");
        context->translatePrimitive(source_model_UUIDs.at(sourceID), source.source_position);
    }
 
    context->setPrimitiveData(source_model_UUIDs.at(sourceID), "twosided_flag", uint(3)); // source model does not interact with radiation field
}
 
void RadiationModel::buildCameraModelGeometry(const std::string &cameralabel) {
 
    assert(cameras.find(cameralabel) != cameras.end());
 
    RadiationCamera camera = cameras.at(cameralabel);
 
    vec3 viewvec = camera.lookat - camera.position;
    SphericalCoord viewsph = cart2sphere(viewvec);
 
    camera_model_UUIDs[cameralabel] = context->loadOBJ("Camera.obj", true);
 
    context->rotatePrimitive(camera_model_UUIDs.at(cameralabel), viewsph.elevation, "x");
    context->rotatePrimitive(camera_model_UUIDs.at(cameralabel), -viewsph.azimuth, "z");
 
    context->translatePrimitive(camera_model_UUIDs.at(cameralabel), camera.position);
 
    context->setPrimitiveData(camera_model_UUIDs.at(cameralabel), "twosided_flag", uint(3)); // camera model does not interact with radiation field
}
 
void RadiationModel::updateLightModelPosition(uint sourceID, const helios::vec3 &delta_position) {
 
    assert(sourceID < radiation_sources.size());
 
    RadiationSource source = radiation_sources.at(sourceID);
 
    if (source.source_type != RADIATION_SOURCE_TYPE_SPHERE && source.source_type != RADIATION_SOURCE_TYPE_DISK && source.source_type != RADIATION_SOURCE_TYPE_RECTANGLE) {
        return;
    }
 
    context->translatePrimitive(source_model_UUIDs.at(sourceID), delta_position);
}
 
void RadiationModel::updateCameraModelPosition(const std::string &cameralabel) {
 
    assert(cameras.find(cameralabel) != cameras.end());
 
    context->deletePrimitive(camera_model_UUIDs.at(cameralabel));
    buildCameraModelGeometry(cameralabel);
}
 
float RadiationModel::integrateSpectrum(uint source_ID, const std::vector<helios::vec2> &object_spectrum, float wavelength1, float wavelength2) const {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Radiation spectrum was not set for source ID. Make sure to set its spectrum using setSourceSpectrum() function.");
    } else if (object_spectrum.size() < 2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Radiation spectrum must have at least 2 wavelengths.");
    } else if (wavelength1 > wavelength2 || wavelength1 == wavelength2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Lower wavelength bound must be less than the upper wavelength bound.");
    }
 
    std::vector<helios::vec2> source_spectrum = radiation_sources.at(source_ID).source_spectrum;
 
    int istart = 0;
    int iend = (int) object_spectrum.size() - 1;
    for (auto i = 0; i < object_spectrum.size() - 1; i++) {
 
        if (object_spectrum.at(i).x <= wavelength1 && object_spectrum.at(i + 1).x > wavelength1) {
            istart = i;
        }
        if (object_spectrum.at(i).x <= wavelength2 && object_spectrum.at(i + 1).x > wavelength2) {
            iend = i + 1;
            break;
        }
    }
 
    float E = 0;
    float Etot = 0;
    for (auto i = istart; i < iend; i++) {
 
        float x0 = object_spectrum.at(i).x;
        float Esource0 = interp1(source_spectrum, object_spectrum.at(i).x);
        float Eobject0 = object_spectrum.at(i).y;
 
        float x1 = object_spectrum.at(i + 1).x;
        float Eobject1 = object_spectrum.at(i + 1).y;
        float Esource1 = interp1(source_spectrum, object_spectrum.at(i + 1).x);
 
        E += 0.5f * (Eobject0 * Esource0 + Eobject1 * Esource1) * (x1 - x0);
        Etot += 0.5f * (Esource1 + Esource0) * (x1 - x0);
    }
 
    return E / Etot;
}
 
float RadiationModel::integrateSpectrum(const std::vector<helios::vec2> &object_spectrum, float wavelength1, float wavelength2) const {
 
    if (object_spectrum.size() < 2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Radiation spectrum must have at least 2 wavelengths.");
    } else if (wavelength1 > wavelength2 || wavelength1 == wavelength2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Lower wavelength bound must be less than the upper wavelength bound.");
    }
 
    int istart = 1;
    int iend = (int) object_spectrum.size() - 1;
    for (auto i = 0; i < object_spectrum.size() - 1; i++) {
 
        if (object_spectrum.at(i).x <= wavelength1 && object_spectrum.at(i + 1).x > wavelength1) {
            istart = i;
        }
        if (object_spectrum.at(i).x <= wavelength2 && object_spectrum.at(i + 1).x > wavelength2) {
            iend = i + 1;
            break;
        }
    }
 
    float E = 0;
    for (auto i = istart; i < iend; i++) {
        float E0 = object_spectrum.at(i).y;
        float x0 = object_spectrum.at(i).x;
        float E1 = object_spectrum.at(i + 1).y;
        float x1 = object_spectrum.at(i + 1).x;
        E += (E0 + E1) * (x1 - x0) * 0.5f;
    }
 
    return E;
}
 
float RadiationModel::integrateSpectrum(const std::vector<helios::vec2> &object_spectrum) const {
    float wavelength1 = object_spectrum.at(0).x;
    float wavelength2 = object_spectrum.at(object_spectrum.size() - 1).x;
    float E = RadiationModel::integrateSpectrum(object_spectrum, wavelength1, wavelength2);
    return E;
}
 
float RadiationModel::integrateSpectrum(uint source_ID, const std::vector<helios::vec2> &object_spectrum, const std::vector<helios::vec2> &camera_spectrum) const {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Radiation spectrum was not set for source ID. Make sure to set its spectrum using setSourceSpectrum() function.");
    } else if (object_spectrum.size() < 2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Radiation spectrum must have at least 2 wavelengths.");
    }
 
    std::vector<helios::vec2> source_spectrum = radiation_sources.at(source_ID).source_spectrum;
 
    float E = 0;
    float Etot = 0;
    for (auto i = 1; i < object_spectrum.size(); i++) {
 
        if (object_spectrum.at(i).x <= source_spectrum.front().x || object_spectrum.at(i).x <= camera_spectrum.front().x) {
            continue;
        }
        if (object_spectrum.at(i).x > source_spectrum.back().x || object_spectrum.at(i).x > camera_spectrum.back().x) {
            break;
        }
        float x1 = object_spectrum.at(i).x;
        float Eobject1 = object_spectrum.at(i).y;
        float Esource1 = interp1(source_spectrum, x1);
        float Ecamera1 = interp1(camera_spectrum, x1);
 
 
        float x0 = object_spectrum.at(i - 1).x;
        float Eobject0 = object_spectrum.at(i - 1).y;
        float Esource0 = interp1(source_spectrum, x0);
        float Ecamera0 = interp1(camera_spectrum, x0);
 
        E += 0.5f * ((Eobject1 * Esource1 * Ecamera1) + (Eobject0 * Ecamera0 * Esource0)) * (x1 - x0);
        Etot += 0.5f * (Esource1 + Esource0) * (x1 - x0);
    }
 
 
    return E / Etot;
}
 
float RadiationModel::integrateSpectrum(const std::vector<helios::vec2> &object_spectrum, const std::vector<helios::vec2> &camera_spectrum) const {
 
    if (object_spectrum.size() < 2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSpectrum): Radiation spectrum must have at least 2 wavelengths.");
    }
 
    float E = 0;
    float Etot = 0;
    for (auto i = 1; i < object_spectrum.size(); i++) {
 
        if (object_spectrum.at(i).x <= camera_spectrum.front().x) {
            continue;
        }
        if (object_spectrum.at(i).x > camera_spectrum.back().x) {
            break;
        }
 
        float x1 = object_spectrum.at(i).x;
        float Eobject1 = object_spectrum.at(i).y;
        float Ecamera1 = interp1(camera_spectrum, x1);
 
 
        float x0 = object_spectrum.at(i - 1).x;
        float Eobject0 = object_spectrum.at(i - 1).y;
        float Ecamera0 = interp1(camera_spectrum, x0);
 
        E += 0.5f * ((Eobject1 * Ecamera1) + (Eobject0 * Ecamera0)) * (x1 - x0);
        Etot += 0.5f * (Ecamera1 + Ecamera0) * (x1 - x0);
    }
 
    return E / Etot;
}
 
float RadiationModel::integrateSourceSpectrum(uint source_ID, float wavelength1, float wavelength2) const {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::integrateSourceSpectrum): Radiation spectrum was not set for source ID. Make sure to set its spectrum using setSourceSpectrum() function.");
    } else if (wavelength1 > wavelength2 || wavelength1 == wavelength2) {
        helios_runtime_error("ERROR (RadiationModel::integrateSourceSpectrum): Lower wavelength bound must be less than the upper wavelength bound.");
    }
 
    return integrateSpectrum(radiation_sources.at(source_ID).source_spectrum, wavelength1, wavelength2);
}
 
void RadiationModel::scaleSpectrum(const std::string &existing_global_data_label, const std::string &new_global_data_label, float scale_factor) const {
 
    std::vector<helios::vec2> spectrum = loadSpectralData(existing_global_data_label);
 
    for (helios::vec2 &s: spectrum) {
        s.y *= scale_factor;
    }
 
    context->setGlobalData(new_global_data_label.c_str(), spectrum);
}
 
void RadiationModel::scaleSpectrum(const std::string &global_data_label, float scale_factor) const {
 
    std::vector<vec2> spectrum = loadSpectralData(global_data_label);
 
    for (vec2 &s: spectrum) {
        s.y *= scale_factor;
    }
 
    context->setGlobalData(global_data_label.c_str(), spectrum);
}
 
void RadiationModel::scaleSpectrumRandomly(const std::string &existing_global_data_label, const std::string &new_global_data_label, float minimum_scale_factor, float maximum_scale_factor) const {
 
    scaleSpectrum(existing_global_data_label, new_global_data_label, context->randu(minimum_scale_factor, maximum_scale_factor));
}
 
 
void RadiationModel::blendSpectra(const std::string &new_spectrum_label, const std::vector<std::string> &spectrum_labels, const std::vector<float> &weights) const {
 
    if (spectrum_labels.size() != weights.size()) {
        helios_runtime_error("ERROR (RadiationModel::blendSpectra): number of spectra and weights must be equal");
    } else if (sum(weights) != 1.f) {
        helios_runtime_error("ERROR (RadiationModel::blendSpectra): weights must sum to 1");
    }
 
    std::vector<vec2> new_spectrum;
    uint spectrum_size = 0;
 
    std::vector<std::vector<vec2>> spectrum(spectrum_labels.size());
 
    uint lambda_start = 0;
    uint lambda_end = 0;
    for (uint i = 0; i < spectrum_labels.size(); i++) {
 
        spectrum.at(i) = loadSpectralData(spectrum_labels.at(i));
 
        if (i == 0) {
            lambda_start = spectrum.at(i).front().x;
            lambda_end = spectrum.at(i).back().x;
        } else {
            if (spectrum.at(i).front().x > lambda_start) {
                lambda_start = spectrum.at(i).front().x;
            }
            if (spectrum.at(i).back().x < lambda_end) {
                lambda_end = spectrum.at(i).back().x;
            }
        }
    }
 
    spectrum_size = lambda_end - lambda_start + 1;
    new_spectrum.resize(spectrum_size);
    for (uint j = 0; j < spectrum_size; j++) {
        new_spectrum.at(j) = make_vec2(lambda_start + j, 0);
    }
 
    // trim front
    for (uint i = 0; i < spectrum_labels.size(); i++) {
        for (uint j = 0; j < spectrum.at(i).size(); j++) {
 
            if (spectrum.at(i).at(j).x >= lambda_start) {
                if (j > 0) {
                    spectrum.at(i).erase(spectrum.at(i).begin(), spectrum.at(i).begin() + j);
                }
                break;
            }
        }
    }
 
    // trim back
    for (uint i = 0; i < spectrum_labels.size(); i++) {
        for (int j = spectrum.at(i).size() - 1; j <= 0; j--) {
 
            if (spectrum.at(i).at(j).x <= lambda_end) {
                if (j < spectrum.at(i).size() - 1) {
                    spectrum.at(i).erase(spectrum.at(i).begin() + j + 1, spectrum.at(i).end());
                }
                break;
            }
        }
    }
 
    for (uint i = 0; i < spectrum_labels.size(); i++) {
        for (uint j = 0; j < spectrum_size; j++) {
            assert(new_spectrum.at(j).x == spectrum.at(i).at(j).x);
            new_spectrum.at(j).y += weights.at(i) * spectrum.at(i).at(j).y;
        }
    }
 
    context->setGlobalData(new_spectrum_label.c_str(), new_spectrum);
}
 
void RadiationModel::blendSpectraRandomly(const std::string &new_spectrum_label, const std::vector<std::string> &spectrum_labels) const {
 
    std::vector<float> weights;
    weights.resize(spectrum_labels.size());
    for (uint i = 0; i < spectrum_labels.size(); i++) {
        weights.at(i) = context->randu();
    }
    float sum_weights = sum(weights);
    for (uint i = 0; i < spectrum_labels.size(); i++) {
        weights.at(i) /= sum_weights;
    }
 
    blendSpectra(new_spectrum_label, spectrum_labels, weights);
}
 
void RadiationModel::interpolateSpectrumFromPrimitiveData(const std::vector<uint> &primitive_UUIDs, const std::vector<std::string> &spectra, const std::vector<float> &values, const std::string &primitive_data_query_label,
                                                          const std::string &primitive_data_radprop_label) {
 
    // Validate that spectra and values have the same length
    if (spectra.size() != values.size()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromPrimitiveData): The 'spectra' vector (size=" + std::to_string(spectra.size()) + ") and 'values' vector (size=" + std::to_string(values.size()) +
                             ") must have the same length.");
    }
 
    // Validate that vectors are not empty
    if (spectra.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromPrimitiveData): The 'spectra' and 'values' vectors cannot be empty.");
    }
 
    // Validate that primitive_UUIDs is not empty
    if (primitive_UUIDs.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromPrimitiveData): The 'primitive_UUIDs' vector cannot be empty.");
    }
 
    // Validate that query and target data labels are not empty
    if (primitive_data_query_label.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromPrimitiveData): The 'primitive_data_query_label' cannot be empty.");
    }
 
    if (primitive_data_radprop_label.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromPrimitiveData): The 'primitive_data_radprop_label' cannot be empty.");
    }
 
    // Search for existing config with matching query and target labels
    SpectrumInterpolationConfig *existing_config = nullptr;
    for (auto &config: spectrum_interpolation_configs) {
        if (config.query_data_label == primitive_data_query_label && config.target_data_label == primitive_data_radprop_label) {
            existing_config = &config;
            break;
        }
    }
 
    if (existing_config != nullptr) {
        // Check if spectra/values match the existing config
        bool spectra_match = (existing_config->spectra_labels == spectra && existing_config->mapping_values == values);
 
        if (spectra_match) {
            // Merge UUIDs into existing config (unordered_set handles duplicates automatically)
            existing_config->primitive_UUIDs.insert(primitive_UUIDs.begin(), primitive_UUIDs.end());
        } else {
            // Replace entire config with new spectra/values and UUIDs
            existing_config->spectra_labels = spectra;
            existing_config->mapping_values = values;
            existing_config->primitive_UUIDs.clear();
            existing_config->primitive_UUIDs.insert(primitive_UUIDs.begin(), primitive_UUIDs.end());
        }
    } else {
        // Create new config
        SpectrumInterpolationConfig config;
        config.primitive_UUIDs.insert(primitive_UUIDs.begin(), primitive_UUIDs.end());
        config.spectra_labels = spectra;
        config.mapping_values = values;
        config.query_data_label = primitive_data_query_label;
        config.target_data_label = primitive_data_radprop_label;
 
        spectrum_interpolation_configs.push_back(config);
    }
}
 
void RadiationModel::interpolateSpectrumFromObjectData(const std::vector<uint> &object_IDs, const std::vector<std::string> &spectra, const std::vector<float> &values, const std::string &object_data_query_label,
                                                       const std::string &primitive_data_radprop_label) {
 
    // Validate that spectra and values have the same length
    if (spectra.size() != values.size()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromObjectData): The 'spectra' vector (size=" + std::to_string(spectra.size()) + ") and 'values' vector (size=" + std::to_string(values.size()) + ") must have the same length.");
    }
 
    // Validate that vectors are not empty
    if (spectra.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromObjectData): The 'spectra' and 'values' vectors cannot be empty.");
    }
 
    // Validate that object_IDs is not empty
    if (object_IDs.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromObjectData): The 'object_IDs' vector cannot be empty.");
    }
 
    // Validate that query and target data labels are not empty
    if (object_data_query_label.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromObjectData): The 'object_data_query_label' cannot be empty.");
    }
 
    if (primitive_data_radprop_label.empty()) {
        helios_runtime_error("ERROR (RadiationModel::interpolateSpectrumFromObjectData): The 'primitive_data_radprop_label' cannot be empty.");
    }
 
    // Search for existing config with matching query and target labels
    SpectrumInterpolationConfig *existing_config = nullptr;
    for (auto &config: spectrum_interpolation_configs) {
        if (config.query_data_label == object_data_query_label && config.target_data_label == primitive_data_radprop_label) {
            existing_config = &config;
            break;
        }
    }
 
    if (existing_config != nullptr) {
        // Check if spectra/values match the existing config
        bool spectra_match = (existing_config->spectra_labels == spectra && existing_config->mapping_values == values);
 
        if (spectra_match) {
            // Merge object IDs into existing config (unordered_set handles duplicates automatically)
            existing_config->object_IDs.insert(object_IDs.begin(), object_IDs.end());
        } else {
            // Replace entire config with new spectra/values and object IDs
            existing_config->spectra_labels = spectra;
            existing_config->mapping_values = values;
            existing_config->object_IDs.clear();
            existing_config->object_IDs.insert(object_IDs.begin(), object_IDs.end());
        }
    } else {
        // Create new config
        SpectrumInterpolationConfig config;
        config.object_IDs.insert(object_IDs.begin(), object_IDs.end());
        config.spectra_labels = spectra;
        config.mapping_values = values;
        config.query_data_label = object_data_query_label;
        config.target_data_label = primitive_data_radprop_label;
 
        spectrum_interpolation_configs.push_back(config);
    }
}
 
void RadiationModel::setSourcePosition(uint source_ID, const vec3 &position) {
 
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::setSourcePosition): Source ID out of bounds. Only " + std::to_string(radiation_sources.size() - 1) + " radiation sources.");
    }
 
    vec3 old_position = radiation_sources.at(source_ID).source_position;
 
    if (radiation_sources.at(source_ID).source_type == RADIATION_SOURCE_TYPE_COLLIMATED) {
        radiation_sources.at(source_ID).source_position = position / position.magnitude();
    } else {
        radiation_sources.at(source_ID).source_position = position * radiation_sources.at(source_ID).source_position_scaling_factor;
    }
 
    if (islightvisualizationenabled) {
        updateLightModelPosition(source_ID, radiation_sources.at(source_ID).source_position - old_position);
    }
}
 
void RadiationModel::setSourcePosition(uint source_ID, const SphericalCoord &position) {
    setSourcePosition(source_ID, sphere2cart(position));
}
 
helios::vec3 RadiationModel::getSourcePosition(uint source_ID) const {
    if (source_ID >= radiation_sources.size()) {
        helios_runtime_error("ERROR (RadiationModel::getSourcePosition): Source ID does not exist.");
    }
    return radiation_sources.at(source_ID).source_position;
}
 
void RadiationModel::setScatteringDepth(const std::string &label, uint depth) {
 
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (RadiationModel::setScatteringDepth): Cannot set scattering depth for band '" + label + "' because it is not a valid band.");
    }
    radiation_bands.at(label).scatteringDepth = depth;
}
 
void RadiationModel::setMinScatterEnergy(const std::string &label, uint energy) {
 
    if (!doesBandExist(label)) {
        helios_runtime_error("ERROR (setMinScatterEnergy): Cannot set minimum scattering energy for band '" + label + "' because it is not a valid band.");
    }
    radiation_bands.at(label).minScatterEnergy = energy;
}
 
void RadiationModel::enforcePeriodicBoundary(const std::string &boundary) {
 
    if (boundary == "x") {
 
        periodic_flag.x = 1;
 
    } else if (boundary == "y") {
 
        periodic_flag.y = 1;
 
    } else if (boundary == "xy") {
 
        periodic_flag.x = 1;
        periodic_flag.y = 1;
 
    } else {
 
        std::cout << "WARNING (RadiationModel::enforcePeriodicBoundary()): unknown boundary of '" << boundary << "'. Possible choices are x, y, or xy." << std::endl;
    }
}
 
void RadiationModel::updateGeometry() {
    updateGeometry(context->getAllUUIDs());
}
 
 
void RadiationModel::updateGeometry(const std::vector<uint> &UUIDs) {
 
    if (message_flag) {
        std::cout << "Updating geometry in radiation transport model..." << std::flush;
    }
 
    // Upload geometry through backend abstraction layer
    buildGeometryData();
    buildUUIDMapping(); // Build UUID↔position mapping for efficient indexing
 
    // CRITICAL: context_UUIDs must match GPU buffer ordering (primitive_UUIDs_ordered)
    // Emission data is indexed by position, which corresponds to primitive_UUIDs order
    context_UUIDs = geometry_data.primitive_UUIDs;
 
    backend->updateGeometry(geometry_data);
    backend->buildAccelerationStructure();
 
    radiativepropertiesneedupdate = true;
    isgeometryinitialized = true;
 
    if (message_flag) {
        std::cout << "done." << std::endl;
    }
}
 
void RadiationModel::updateRadiativeProperties() {
 
    // Possible scenarios for specifying a primitive's radiative properties
    // 1. If primitive data of form reflectivity_band/transmissivity_band is given, this value is used and overrides any other option.
    // 2. If primitive data of form reflectivity_spectrum/transmissivity_spectrum is given that references global data containing spectral reflectivity/transmissivity:
    //    2a. If radiation source spectrum was not given, assume source spectral intensity is constant over band and calculate using primitive spectrum
    //    2b. If radiation source spectrum was given, calculate using both source and primitive spectrum.
 
    // Create warning aggregator
    helios::WarningAggregator warnings;
    warnings.setEnabled(message_flag);
 
    if (message_flag) {
        std::cout << "Updating radiative properties..." << std::flush;
    }
 
    uint Nbands = radiation_bands.size(); // number of radiative bands
    uint Nsources = radiation_sources.size();
    uint Ncameras = cameras.size();
    size_t Nobjects = primitiveID.size();
    size_t Nprimitives = context_UUIDs.size();
 
    scattering_iterations_needed.clear();
    for (auto &band: radiation_bands) {
        scattering_iterations_needed[band.first] = false;
    }
 
    std::vector<std::vector<std::vector<float>>> rho, tau; // first index is the source, second index is the primitive, third index is the band
    std::vector<std::vector<std::vector<std::vector<float>>>> rho_cam, tau_cam; // Fourth index is the camera
    float eps;
 
    std::string prop;
    std::vector<std::string> band_labels;
    for (auto &band: radiation_bands) {
        band_labels.push_back(band.first);
    }
 
    rho.resize(Nsources);
    tau.resize(Nsources);
    for (size_t s = 0; s < Nsources; s++) {
        rho.at(s).resize(Nprimitives);
        tau.at(s).resize(Nprimitives);
        for (size_t p = 0; p < Nprimitives; p++) {
            rho.at(s).at(p).resize(Nbands);
            tau.at(s).at(p).resize(Nbands);
        }
    }
    if (Ncameras) {
        rho_cam.resize(Nsources);
        tau_cam.resize(Nsources);
        for (size_t s = 0; s < Nsources; s++) {
            rho_cam.at(s).resize(Nprimitives);
            tau_cam.at(s).resize(Nprimitives);
            for (size_t p = 0; p < Nprimitives; p++) {
                rho_cam.at(s).at(p).resize(Nbands);
                tau_cam.at(s).at(p).resize(Nbands);
                for (size_t b = 0; b < Nbands; b++) {
                    rho_cam.at(s).at(p).at(b).resize(Ncameras);
                    tau_cam.at(s).at(p).at(b).resize(Ncameras);
                }
            }
        }
    }
 
    // Cache all unique camera spectral responses for all cameras and bands
    std::vector<std::vector<std::vector<helios::vec2>>> camera_response_unique;
    camera_response_unique.resize(Ncameras);
    if (Ncameras > 0) {
        uint cam = 0;
        for (const auto &camera: cameras) {
 
            camera_response_unique.at(cam).resize(Nbands);
 
            for (uint b = 0; b < Nbands; b++) {
 
                if (camera.second.band_spectral_response.find(band_labels.at(b)) == camera.second.band_spectral_response.end()) {
                    continue;
                }
 
                std::string camera_response = camera.second.band_spectral_response.at(band_labels.at(b));
 
                if (!camera_response.empty()) {
 
                    if (!context->doesGlobalDataExist(camera_response.c_str())) {
                        if (camera_response != "uniform") {
                            warnings.addWarning("missing_camera_response", "Camera spectral response \"" + camera_response + "\" does not exist. Assuming a uniform spectral response.");
                        }
                    } else if (context->getGlobalDataType(camera_response.c_str()) == helios::HELIOS_TYPE_VEC2) {
 
                        std::vector<helios::vec2> data = loadSpectralData(camera_response.c_str());
 
                        camera_response_unique.at(cam).at(b) = data;
 
                    } else if (context->getGlobalDataType(camera_response.c_str()) != helios::HELIOS_TYPE_VEC2 && context->getGlobalDataType(camera_response.c_str()) != helios::HELIOS_TYPE_STRING) {
                        camera_response.clear();
                        std::cout << "WARNING (RadiationModel::runBand): Camera spectral response \"" << camera_response << "\" is not of type HELIOS_TYPE_VEC2 or HELIOS_TYPE_STRING. Assuming a uniform spectral response..." << std::endl;
                    }
                }
            }
            cam++;
        }
    }
 
    // Spectral integration cache to avoid redundant computations
    std::unordered_map<std::string, float> spectral_integration_cache;
 
#ifdef USE_OPENMP
    // Temporary cache for this thread group (will be merged later)
    std::unordered_map<std::string, float> temp_spectral_cache;
#endif
 
    // Helper function to create cache keys for spectral integrations
    auto createCacheKey = [](const std::string &spectrum_label, uint source_id, uint band_id, uint camera_id, const std::string &type) -> std::string {
        return spectrum_label + "_" + std::to_string(source_id) + "_" + std::to_string(band_id) + "_" + std::to_string(camera_id) + "_" + type;
    };
 
    // Helper function to get from cache (thread-safe)
    auto getCachedValue = [&](const std::string &cache_key, bool &found) -> float {
        float result = 0.0f;
        found = false;
 
#ifdef USE_OPENMP
#pragma omp critical
        {
#endif
            // Check shared cache
            auto cache_it = spectral_integration_cache.find(cache_key);
            if (cache_it != spectral_integration_cache.end()) {
                found = true;
                result = cache_it->second;
            }
#ifdef USE_OPENMP
        }
#endif
        return result;
    };
 
    // Helper function to store in cache (thread-safe)
    auto setCachedValue = [&](const std::string &cache_key, float value) {
#ifdef USE_OPENMP
#pragma omp critical
        {
#endif
            spectral_integration_cache[cache_key] = value;
#ifdef USE_OPENMP
        }
#endif
    };
 
    // Helper function for cached interpolation (thread-safe)
    auto cachedInterp1 = [&](const std::vector<helios::vec2> &spectrum, float wavelength, const std::string &spectrum_id) -> float {
        // Create cache key for this specific interpolation
        std::string cache_key = "interp_" + spectrum_id + "_" + std::to_string(wavelength);
 
        bool found = false;
        float cached_result = getCachedValue(cache_key, found);
        if (found) {
            return cached_result;
        }
 
        // Perform interpolation and cache result
        float result = interp1(spectrum, wavelength);
        setCachedValue(cache_key, result);
        return result;
    };
 
    // Cached version of integrateSpectrum with source spectrum
    auto cachedIntegrateSpectrumWithSource = [&](uint source_ID, const std::vector<helios::vec2> &object_spectrum, float wavelength1, float wavelength2, const std::string &object_spectrum_id) -> float {
        if (source_ID >= radiation_sources.size() || object_spectrum.size() < 2 || wavelength1 >= wavelength2) {
            return 0.0f; // Handle edge cases gracefully
        }
 
        std::vector<helios::vec2> source_spectrum = radiation_sources.at(source_ID).source_spectrum;
        std::string source_id = "source_" + std::to_string(source_ID);
 
        int istart = 0;
        int iend = (int) object_spectrum.size() - 1;
        for (auto i = 0; i < object_spectrum.size() - 1; i++) {
            if (object_spectrum.at(i).x <= wavelength1 && object_spectrum.at(i + 1).x > wavelength1) {
                istart = i;
            }
            if (object_spectrum.at(i).x <= wavelength2 && object_spectrum.at(i + 1).x > wavelength2) {
                iend = i + 1;
                break;
            }
        }
 
        float E = 0;
        float Etot = 0;
        for (auto i = istart; i < iend; i++) {
            float x0 = object_spectrum.at(i).x;
            float Esource0 = cachedInterp1(source_spectrum, x0, source_id);
            float Eobject0 = object_spectrum.at(i).y;
 
            float x1 = object_spectrum.at(i + 1).x;
            float Eobject1 = object_spectrum.at(i + 1).y;
            float Esource1 = cachedInterp1(source_spectrum, x1, source_id);
 
            E += 0.5f * (Eobject0 * Esource0 + Eobject1 * Esource1) * (x1 - x0);
            Etot += 0.5f * (Esource1 + Esource0) * (x1 - x0);
        }
 
        return (Etot != 0.0f) ? E / Etot : 0.0f;
    };
 
    // Cached version of integrateSpectrum with source and camera spectra
    auto cachedIntegrateSpectrumWithSourceAndCamera = [&](uint source_ID, const std::vector<helios::vec2> &object_spectrum, const std::vector<helios::vec2> &camera_spectrum, uint camera_index, uint band_index,
                                                          const std::string &object_spectrum_id) -> float {
        if (source_ID >= radiation_sources.size() || object_spectrum.size() < 2) {
            return 0.0f;
        }
 
        std::vector<helios::vec2> source_spectrum = radiation_sources.at(source_ID).source_spectrum;
        std::string source_id = "source_" + std::to_string(source_ID);
        std::string camera_id = "camera_" + std::to_string(camera_index) + "_band_" + std::to_string(band_index); // Include band for unique cache key per band
 
        float E = 0;
        float Etot = 0;
        for (auto i = 1; i < object_spectrum.size(); i++) {
            if (object_spectrum.at(i).x <= source_spectrum.front().x || object_spectrum.at(i).x <= camera_spectrum.front().x) {
                continue;
            }
            if (object_spectrum.at(i).x > source_spectrum.back().x || object_spectrum.at(i).x > camera_spectrum.back().x) {
                break;
            }
 
            float x1 = object_spectrum.at(i).x;
            float Eobject1 = object_spectrum.at(i).y;
            float Esource1 = cachedInterp1(source_spectrum, x1, source_id);
            float Ecamera1 = cachedInterp1(camera_spectrum, x1, camera_id);
 
            float x0 = object_spectrum.at(i - 1).x;
            float Eobject0 = object_spectrum.at(i - 1).y;
            float Esource0 = cachedInterp1(source_spectrum, x0, source_id);
            float Ecamera0 = cachedInterp1(camera_spectrum, x0, camera_id);
 
            E += 0.5f * ((Eobject1 * Esource1 * Ecamera1) + (Eobject0 * Ecamera0 * Esource0)) * (x1 - x0);
            Etot += 0.5f * (Esource1 + Esource0) * (x1 - x0);
        }
 
        return (Etot != 0.0f) ? E / Etot : 0.0f;
    };
 
    // Apply spectral interpolation based on primitive data values
    for (const auto &config: spectrum_interpolation_configs) {
        // Validate that all spectra in this config exist in global data and have correct type
        for (const auto &spectrum_label: config.spectra_labels) {
            if (!context->doesGlobalDataExist(spectrum_label.c_str())) {
                helios_runtime_error("ERROR (RadiationModel::updateRadiativeProperties): Spectral interpolation config references global data '" + spectrum_label + "' which does not exist.");
            }
            if (context->getGlobalDataType(spectrum_label.c_str()) != helios::HELIOS_TYPE_VEC2) {
                helios_runtime_error("ERROR (RadiationModel::updateRadiativeProperties): Spectral interpolation config references global data '" + spectrum_label + "' which must be of type HELIOS_TYPE_VEC2 (std::vector<helios::vec2>).");
            }
        }
 
        for (uint uuid: config.primitive_UUIDs) {
            // Check if primitive still exists in context (it may have been deleted)
            if (!context->doesPrimitiveExist(uuid)) {
                continue;
            }
 
            // Check if the query data exists for this primitive and has correct type
            if (context->doesPrimitiveDataExist(uuid, config.query_data_label.c_str())) {
                // Check that query data is of type float
                if (context->getPrimitiveDataType(config.query_data_label.c_str()) != helios::HELIOS_TYPE_FLOAT) {
                    helios_runtime_error("ERROR (RadiationModel::updateRadiativeProperties): Primitive data '" + config.query_data_label + "' for UUID " + std::to_string(uuid) + " must be of type HELIOS_TYPE_FLOAT for spectral interpolation.");
                }
 
                // Get the query value
                float query_value;
                context->getPrimitiveData(uuid, config.query_data_label.c_str(), query_value);
 
                // Perform nearest-neighbor interpolation
                size_t nearest_idx = 0;
                float min_distance = std::abs(query_value - config.mapping_values[0]);
                for (size_t i = 1; i < config.mapping_values.size(); i++) {
                    float distance = std::abs(query_value - config.mapping_values[i]);
                    if (distance < min_distance) {
                        min_distance = distance;
                        nearest_idx = i;
                    }
                }
 
                // Set the target primitive data to the selected spectrum label
                context->setPrimitiveData(uuid, config.target_data_label.c_str(), config.spectra_labels[nearest_idx]);
            }
        }
 
        // Apply spectral interpolation based on object data values
        for (uint objID: config.object_IDs) {
            // Check if object still exists in context (it may have been deleted)
            if (!context->doesObjectExist(objID)) {
                continue;
            }
 
            // Check if the query data exists for this object and has correct type
            if (context->doesObjectDataExist(objID, config.query_data_label.c_str())) {
                // Check that query data is of type float
                if (context->getObjectDataType(config.query_data_label.c_str()) != helios::HELIOS_TYPE_FLOAT) {
                    helios_runtime_error("ERROR (RadiationModel::updateRadiativeProperties): Object data '" + config.query_data_label + "' for object ID " + std::to_string(objID) + " must be of type HELIOS_TYPE_FLOAT for spectral interpolation.");
                }
 
                // Get the query value
                float query_value;
                context->getObjectData(objID, config.query_data_label.c_str(), query_value);
 
                // Perform nearest-neighbor interpolation
                size_t nearest_idx = 0;
                float min_distance = std::abs(query_value - config.mapping_values.at(0));
                for (size_t i = 1; i < config.mapping_values.size(); i++) {
                    float distance = std::abs(query_value - config.mapping_values.at(i));
                    if (distance < min_distance) {
                        min_distance = distance;
                        nearest_idx = i;
                    }
                }
 
                // Get object's primitive UUIDs and set their primitive data using vector overload
                std::vector<uint> prim_uuids = context->getObjectPrimitiveUUIDs(objID);
                context->setPrimitiveData(prim_uuids, config.target_data_label.c_str(), config.spectra_labels.at(nearest_idx));
            }
        }
    }
 
    // Cache all unique primitive reflectivity and transmissivity spectra before assigning to primitives
 
    // first, figure out all of the spectra referenced by all primitives and store it in "surface_spectra" to avoid having to load it again
    std::map<std::string, std::vector<helios::vec2>> surface_spectra_rho;
    std::map<std::string, std::vector<helios::vec2>> surface_spectra_tau;
    for (size_t u = 0; u < Nprimitives; u++) {
 
        uint UUID = context_UUIDs.at(u);
 
        if (context->doesPrimitiveDataExist(UUID, "reflectivity_spectrum")) {
            if (context->getPrimitiveDataType("reflectivity_spectrum") == HELIOS_TYPE_STRING) {
                std::string spectrum_label;
                context->getPrimitiveData(UUID, "reflectivity_spectrum", spectrum_label);
 
                // get the spectral reflectivity data and store it in surface_spectra to avoid having to load it again
                if (surface_spectra_rho.find(spectrum_label) == surface_spectra_rho.end()) {
                    if (!context->doesGlobalDataExist(spectrum_label.c_str())) {
                        if (!spectrum_label.empty()) {
                            warnings.addWarning("missing_reflectivity_spectrum", "Primitive spectral reflectivity \"" + spectrum_label + "\" does not exist. Using default reflectivity of 0.");
                        }
                        std::vector<helios::vec2> data;
                        surface_spectra_rho.emplace(spectrum_label, data);
                    } else if (context->getGlobalDataType(spectrum_label.c_str()) == HELIOS_TYPE_VEC2) {
 
                        std::vector<helios::vec2> data = loadSpectralData(spectrum_label.c_str());
                        surface_spectra_rho.emplace(spectrum_label, data);
 
                    } else if (context->getGlobalDataType(spectrum_label.c_str()) != helios::HELIOS_TYPE_VEC2 && context->getGlobalDataType(spectrum_label.c_str()) != helios::HELIOS_TYPE_STRING) {
                        spectrum_label.clear();
                        std::cout << "WARNING (RadiationModel::runBand): Object spectral reflectivity \"" << spectrum_label << "\" is not of type HELIOS_TYPE_VEC2 or HELIOS_TYPE_STRING. Assuming a uniform spectral distribution..." << std::flush;
                    }
                }
            }
        }
 
        if (context->doesPrimitiveDataExist(UUID, "transmissivity_spectrum")) {
            if (context->getPrimitiveDataType("transmissivity_spectrum") == HELIOS_TYPE_STRING) {
                std::string spectrum_label;
                context->getPrimitiveData(UUID, "transmissivity_spectrum", spectrum_label);
 
                // get the spectral transmissivity data and store it in surface_spectra to avoid having to load it again
                if (surface_spectra_tau.find(spectrum_label) == surface_spectra_tau.end()) {
                    if (!context->doesGlobalDataExist(spectrum_label.c_str())) {
                        if (!spectrum_label.empty()) {
                            warnings.addWarning("missing_transmissivity_spectrum", "Primitive spectral transmissivity \"" + spectrum_label + "\" does not exist. Using default transmissivity of 0.");
                        }
                        std::vector<helios::vec2> data;
                        surface_spectra_tau.emplace(spectrum_label, data);
                    } else if (context->getGlobalDataType(spectrum_label.c_str()) == HELIOS_TYPE_VEC2) {
 
                        std::vector<helios::vec2> data = loadSpectralData(spectrum_label.c_str());
                        surface_spectra_tau.emplace(spectrum_label, data);
 
                    } else if (context->getGlobalDataType(spectrum_label.c_str()) != helios::HELIOS_TYPE_VEC2 && context->getGlobalDataType(spectrum_label.c_str()) != helios::HELIOS_TYPE_STRING) {
                        spectrum_label.clear();
                        std::cout << "WARNING (RadiationModel::runBand): Object spectral transmissivity \"" << spectrum_label << "\" is not of type HELIOS_TYPE_VEC2 or HELIOS_TYPE_STRING. Assuming a uniform spectral distribution..." << std::flush;
                    }
                }
            }
        }
    }
 
    // second, calculate unique values of rho and tau for all sources and bands
    std::map<std::string, std::vector<std::vector<float>>> rho_unique;
    std::map<std::string, std::vector<std::vector<float>>> tau_unique;
 
    std::map<std::string, std::vector<std::vector<std::vector<float>>>> rho_cam_unique;
    std::map<std::string, std::vector<std::vector<std::vector<float>>>> tau_cam_unique;
 
    std::vector<std::vector<float>> empty;
    empty.resize(Nbands);
    for (uint b = 0; b < Nbands; b++) {
        empty.at(b).resize(Nsources, 0);
    }
    std::vector<std::vector<std::vector<float>>> empty_cam;
    if (Ncameras > 0) {
        empty_cam.resize(Nbands);
        for (uint b = 0; b < Nbands; b++) {
            empty_cam.at(b).resize(Nsources);
            for (uint s = 0; s < Nsources; s++) {
                empty_cam.at(b).at(s).resize(Ncameras, 0);
            }
        }
    }
 
    // Convert maps to vectors for OpenMP indexing
    std::vector<std::pair<std::string, std::vector<helios::vec2>>> spectra_rho_vector(surface_spectra_rho.begin(), surface_spectra_rho.end());
 
    // Pre-initialize all map entries before parallel processing to avoid race conditions
    for (const auto &spectrum: spectra_rho_vector) {
        rho_unique[spectrum.first] = empty;
        if (Ncameras > 0) {
            rho_cam_unique[spectrum.first] = empty_cam;
        }
    }
 
    // Process reflectivity spectra with OpenMP parallelization
#ifdef USE_OPENMP
#pragma omp parallel for schedule(dynamic)
#endif
    for (int spectrum_idx = 0; spectrum_idx < (int) spectra_rho_vector.size(); spectrum_idx++) {
        const auto &spectrum = spectra_rho_vector[spectrum_idx];
 
        for (uint b = 0; b < Nbands; b++) {
            std::string band = band_labels.at(b);
 
            for (uint s = 0; s < Nsources; s++) {
 
                // integrate with caching
                auto band_it = radiation_bands.find(band);
                if (band_it != radiation_bands.end() && band_it->second.wavebandBounds.x != 0 && band_it->second.wavebandBounds.y != 0 && !spectrum.second.empty()) {
                    if (!radiation_sources.at(s).source_spectrum.empty()) {
                        std::string cache_key = createCacheKey(spectrum.first, s, b, 0, "rho_source");
                        bool found;
                        float cached_result = getCachedValue(cache_key, found);
                        if (found) {
                            rho_unique[spectrum.first][b][s] = cached_result;
                        } else {
                            float result = cachedIntegrateSpectrumWithSource(s, spectrum.second, band_it->second.wavebandBounds.x, band_it->second.wavebandBounds.y, spectrum.first);
                            setCachedValue(cache_key, result);
                            rho_unique[spectrum.first][b][s] = result;
                        }
                    } else {
                        // source spectrum not provided, assume source intensity is constant over the band
                        std::string cache_key = createCacheKey(spectrum.first, s, b, 0, "rho_no_source");
                        bool found;
                        float cached_result = getCachedValue(cache_key, found);
                        if (found) {
                            rho_unique[spectrum.first][b][s] = cached_result;
                        } else {
                            float result = integrateSpectrum(spectrum.second, band_it->second.wavebandBounds.x, band_it->second.wavebandBounds.y) / (band_it->second.wavebandBounds.y - band_it->second.wavebandBounds.x);
                            setCachedValue(cache_key, result);
                            rho_unique[spectrum.first][b][s] = result;
                        }
                    }
                } else {
                    // No wavelength bounds, can't integrate spectrum without camera response
                    // Set to default for now, will use camera average if available
                    rho_unique[spectrum.first][b][s] = rho_default;
                }
 
                // cameras
                if (Ncameras > 0) {
                    uint cam = 0;
                    float rho_cam_sum_for_averaging = 0.f;
                    for (const auto &camera: cameras) {
 
                        if (camera_response_unique.at(cam).at(b).empty()) {
                            rho_cam_unique[spectrum.first][b][s][cam] = rho_unique[spectrum.first][b][s];
                        } else {
 
                            // integrate with caching
                            if (!spectrum.second.empty()) {
                                if (!radiation_sources.at(s).source_spectrum.empty()) {
                                    std::string cache_key = createCacheKey(spectrum.first, s, b, cam, "rho_cam_source");
                                    bool found;
                                    float cached_result = getCachedValue(cache_key, found);
                                    if (found) {
                                        rho_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = cached_result;
                                        rho_cam_sum_for_averaging += cached_result;
                                    } else {
                                        float result = cachedIntegrateSpectrumWithSourceAndCamera(s, spectrum.second, camera_response_unique.at(cam).at(b), cam, b, spectrum.first);
                                        setCachedValue(cache_key, result);
                                        rho_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = result;
                                        rho_cam_sum_for_averaging += result;
                                    }
                                } else {
                                    std::string cache_key = createCacheKey(spectrum.first, s, b, cam, "rho_cam_no_source");
                                    bool found;
                                    float cached_result = getCachedValue(cache_key, found);
                                    if (found) {
                                        rho_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = cached_result;
                                        rho_cam_sum_for_averaging += cached_result;
                                    } else {
                                        float result = integrateSpectrum(spectrum.second, camera_response_unique.at(cam).at(b));
                                        setCachedValue(cache_key, result);
                                        rho_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = result;
                                        rho_cam_sum_for_averaging += result;
                                    }
                                }
                            } else {
                                rho_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = rho_default;
                            }
                        }
 
                        cam++;
                    }
 
                    // CRITICAL FIX: If wavelength bounds weren't set but camera integration produced values,
                    // use camera average as the base reflectivity. This allows regular scatter to work
                    // when only reflectivity_spectrum + camera response are provided.
                    if (rho_unique[spectrum.first][b][s] == rho_default && rho_cam_sum_for_averaging > 0 && cam > 0) {
                        rho_unique[spectrum.first][b][s] = rho_cam_sum_for_averaging / float(cam);
                    }
                }
            }
        }
    }
 
    // Convert tau spectra to vector for OpenMP indexing
    std::vector<std::pair<std::string, std::vector<helios::vec2>>> spectra_tau_vector(surface_spectra_tau.begin(), surface_spectra_tau.end());
 
    // Pre-initialize all map entries before parallel processing to avoid race conditions
    for (const auto &spectrum: spectra_tau_vector) {
        tau_unique[spectrum.first] = empty;
        if (Ncameras > 0) {
            tau_cam_unique[spectrum.first] = empty_cam;
        }
    }
 
    // Process transmissivity spectra with OpenMP parallelization
#ifdef USE_OPENMP
#pragma omp parallel for schedule(dynamic)
#endif
    for (int spectrum_idx = 0; spectrum_idx < (int) spectra_tau_vector.size(); spectrum_idx++) {
        const auto &spectrum = spectra_tau_vector[spectrum_idx];
 
        for (uint b = 0; b < Nbands; b++) {
            std::string band = band_labels.at(b);
 
            for (uint s = 0; s < Nsources; s++) {
 
                // integrate with caching
                auto band_it = radiation_bands.find(band);
                if (band_it != radiation_bands.end() && band_it->second.wavebandBounds.x != 0 && band_it->second.wavebandBounds.y != 0 && !spectrum.second.empty()) {
                    if (!radiation_sources.at(s).source_spectrum.empty()) {
                        std::string cache_key = createCacheKey(spectrum.first, s, b, 0, "tau_source");
                        bool found;
                        float cached_result = getCachedValue(cache_key, found);
                        if (found) {
                            tau_unique[spectrum.first][b][s] = cached_result;
                        } else {
                            float result = cachedIntegrateSpectrumWithSource(s, spectrum.second, band_it->second.wavebandBounds.x, band_it->second.wavebandBounds.y, spectrum.first);
                            setCachedValue(cache_key, result);
                            tau_unique[spectrum.first][b][s] = result;
                        }
                    } else {
                        std::string cache_key = createCacheKey(spectrum.first, s, b, 0, "tau_no_source");
                        bool found;
                        float cached_result = getCachedValue(cache_key, found);
                        if (found) {
                            tau_unique[spectrum.first][b][s] = cached_result;
                        } else {
                            float result = integrateSpectrum(spectrum.second, band_it->second.wavebandBounds.x, band_it->second.wavebandBounds.y) / (band_it->second.wavebandBounds.y - band_it->second.wavebandBounds.x);
                            setCachedValue(cache_key, result);
                            tau_unique[spectrum.first][b][s] = result;
                        }
                    }
                } else {
                    tau_unique[spectrum.first][b][s] = tau_default;
                }
 
                // cameras
                if (Ncameras > 0) {
                    uint cam = 0;
                    for (const auto &camera: cameras) {
 
                        if (camera_response_unique.at(cam).at(b).empty()) {
 
                            tau_cam_unique[spectrum.first][b][s][cam] = tau_unique[spectrum.first][b][s];
 
                        } else {
 
                            // integrate with caching
                            if (!spectrum.second.empty()) {
                                if (!radiation_sources.at(s).source_spectrum.empty()) {
                                    std::string cache_key = createCacheKey(spectrum.first, s, b, cam, "tau_cam_source");
                                    bool found;
                                    float cached_result = getCachedValue(cache_key, found);
                                    if (found) {
                                        tau_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = cached_result;
                                    } else {
                                        float result = cachedIntegrateSpectrumWithSourceAndCamera(s, spectrum.second, camera_response_unique.at(cam).at(b), cam, b, spectrum.first);
                                        setCachedValue(cache_key, result);
                                        tau_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = result;
                                    }
                                } else {
                                    std::string cache_key = createCacheKey(spectrum.first, s, b, cam, "tau_cam_no_source");
                                    bool found;
                                    float cached_result = getCachedValue(cache_key, found);
                                    if (found) {
                                        tau_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = cached_result;
                                    } else {
                                        float result = integrateSpectrum(spectrum.second, camera_response_unique.at(cam).at(b));
                                        setCachedValue(cache_key, result);
                                        tau_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = result;
                                    }
                                }
                            } else {
                                tau_cam_unique.at(spectrum.first).at(b).at(s).at(cam) = tau_default;
                            }
                        }
 
                        cam++;
                    }
                }
            }
        }
    }
 
    for (size_t u = 0; u < Nprimitives; u++) {
 
        uint UUID = context_UUIDs.at(u);
 
        helios::PrimitiveType type = context->getPrimitiveType(UUID);
 
        if (type == helios::PRIMITIVE_TYPE_VOXEL) {
 
        } else { // other than voxels
 
            // Reflectivity
 
            // check for primitive data of form "reflectivity_spectrum" that can be used to calculate reflectivity
            std::string spectrum_label;
            if (context->doesPrimitiveDataExist(UUID, "reflectivity_spectrum")) {
                if (context->getPrimitiveDataType("reflectivity_spectrum") == HELIOS_TYPE_STRING) {
                    context->getPrimitiveData(UUID, "reflectivity_spectrum", spectrum_label);
                }
            }
 
            uint b = 0;
            for (const auto &band: band_labels) {
 
                // check for primitive data of form "reflectivity_bandname"
                prop = "reflectivity_" + band;
 
                float rho_s = rho_default;
                if (context->doesPrimitiveDataExist(UUID, prop.c_str())) {
                    context->getPrimitiveData(UUID, prop.c_str(), rho_s);
                }
 
                for (uint s = 0; s < Nsources; s++) {
                    // if reflectivity was manually set, or a spectrum was given and the global data exists
                    if (rho_s != rho_default || spectrum_label.empty() || !context->doesGlobalDataExist(spectrum_label.c_str()) || rho_unique.find(spectrum_label) == rho_unique.end()) {
 
                        rho.at(s).at(u).at(b) = rho_s;
 
                        // cameras
                        for (uint cam = 0; cam < Ncameras; cam++) {
                            rho_cam.at(s).at(u).at(b).at(cam) = rho_s;
                        }
 
                        // use spectrum
                    } else {
 
                        rho.at(s).at(u).at(b) = rho_unique.at(spectrum_label).at(b).at(s);
 
                        // cameras
                        for (uint cam = 0; cam < Ncameras; cam++) {
                            rho_cam.at(s).at(u).at(b).at(cam) = rho_cam_unique.at(spectrum_label).at(b).at(s).at(cam);
                        }
                    }
 
                    // error checking
                    if (rho.at(s).at(u).at(b) < 0) {
                        rho.at(s).at(u).at(b) = 0.f;
                        warnings.addWarning("reflectivity_negative_clamped", "Reflectivity cannot be less than 0. Clamping to 0 for band " + band + ".");
                    } else if (rho.at(s).at(u).at(b) > 1.f) {
                        rho.at(s).at(u).at(b) = 1.f;
                        warnings.addWarning("reflectivity_exceeded_clamped", "Reflectivity cannot be greater than 1. Clamping to 1 for band " + band + ".");
                    }
                    if (rho.at(s).at(u).at(b) != 0) {
                        scattering_iterations_needed.at(band) = true;
                    }
                    for (auto &odata: output_prim_data) {
                        if (odata == "reflectivity") {
                            context->setPrimitiveData(UUID, ("reflectivity_" + std::to_string(s) + "_" + band).c_str(), rho.at(s).at(u).at(b));
                        }
                    }
                }
                b++;
            }
 
            // Transmissivity
 
            // check for primitive data of form "transmissivity_spectrum" that can be used to calculate transmissivity
            spectrum_label.resize(0);
            if (context->doesPrimitiveDataExist(UUID, "transmissivity_spectrum")) {
                if (context->getPrimitiveDataType("transmissivity_spectrum") == HELIOS_TYPE_STRING) {
                    context->getPrimitiveData(UUID, "transmissivity_spectrum", spectrum_label);
                }
            }
 
            b = 0;
            for (const auto &band: band_labels) {
 
                // check for primitive data of form "transmissivity_bandname"
                prop = "transmissivity_" + band;
 
                float tau_s = tau_default;
                if (context->doesPrimitiveDataExist(UUID, prop.c_str())) {
                    context->getPrimitiveData(UUID, prop.c_str(), tau_s);
                }
 
                for (uint s = 0; s < Nsources; s++) {
                    // if transmissivity was manually set, or a spectrum was given and the global data exists
                    if (tau_s != tau_default || spectrum_label.empty() || !context->doesGlobalDataExist(spectrum_label.c_str()) || tau_unique.find(spectrum_label) == tau_unique.end()) {
 
                        tau.at(s).at(u).at(b) = tau_s;
 
                        // cameras
                        for (uint cam = 0; cam < Ncameras; cam++) {
                            tau_cam.at(s).at(u).at(b).at(cam) = tau_s;
                        }
 
                    } else {
 
                        tau.at(s).at(u).at(b) = tau_unique.at(spectrum_label).at(b).at(s);
 
                        // cameras
                        for (uint cam = 0; cam < Ncameras; cam++) {
                            tau_cam.at(s).at(u).at(b).at(cam) = tau_cam_unique.at(spectrum_label).at(b).at(s).at(cam);
                        }
                    }
 
                    // error checking
                    if (tau.at(s).at(u).at(b) < 0) {
                        tau.at(s).at(u).at(b) = 0.f;
                        warnings.addWarning("transmissivity_negative_clamped", "Transmissivity cannot be less than 0. Clamping to 0 for band " + band + ".");
                    } else if (tau.at(s).at(u).at(b) > 1.f) {
                        tau.at(s).at(u).at(b) = 1.f;
                        warnings.addWarning("transmissivity_exceeded_clamped", "Transmissivity cannot be greater than 1. Clamping to 1 for band " + band + ".");
                    }
                    if (tau.at(s).at(u).at(b) != 0) {
                        scattering_iterations_needed.at(band) = true;
                    }
                    for (auto &odata: output_prim_data) {
                        if (odata == "transmissivity") {
                            context->setPrimitiveData(UUID, ("transmissivity_" + std::to_string(s) + "_" + band).c_str(), tau.at(s).at(u).at(b));
                        }
                    }
                }
                b++;
            }
 
            // Emissivity (only for error checking)
 
            b = 0;
            for (const auto &band: band_labels) {
 
                prop = "emissivity_" + band;
 
                if (context->doesPrimitiveDataExist(UUID, prop.c_str())) {
                    context->getPrimitiveData(UUID, prop.c_str(), eps);
                } else {
                    eps = eps_default;
                }
 
                if (eps < 0) {
                    eps = 0.f;
                    warnings.addWarning("emissivity_negative_clamped", "Emissivity cannot be less than 0. Clamping to 0 for band " + band + ".");
                } else if (eps > 1.f) {
                    eps = 1.f;
                    warnings.addWarning("emissivity_exceeded_clamped", "Emissivity cannot be greater than 1. Clamping to 1 for band " + band + ".");
                }
                if (eps != 1) {
                    scattering_iterations_needed.at(band) = true;
                }
 
                assert(doesBandExist(band));
 
                for (uint s = 0; s < Nsources; s++) {
                    if (radiation_bands.at(band).emissionFlag) { // emission enabled
                        if (eps != 1.f && rho.at(s).at(u).at(b) == 0 && tau.at(s).at(u).at(b) == 0) {
                            rho.at(s).at(u).at(b) = 1.f - eps;
                        } else if (eps + tau.at(s).at(u).at(b) + rho.at(s).at(u).at(b) != 1.f && eps > 0.f) {
                            helios_runtime_error("ERROR (RadiationModel): emissivity, transmissivity, and reflectivity must sum to 1 to ensure energy conservation. Band " + band + ", Primitive #" + std::to_string(UUID) + ": eps=" +
                                                 std::to_string(eps) + ", tau=" + std::to_string(tau.at(s).at(u).at(b)) + ", rho=" + std::to_string(rho.at(s).at(u).at(b)) + ". It is also possible that you forgot to disable emission for this band.");
                        } else if (radiation_bands.at(band).scatteringDepth == 0 && eps != 1.f) {
                            eps = 1.f;
                            rho.at(s).at(u).at(b) = 0.f;
                            tau.at(s).at(u).at(b) = 0.f;
                        }
                    } else if (tau.at(s).at(u).at(b) + rho.at(s).at(u).at(b) > 1.f) {
                        helios_runtime_error("ERROR (RadiationModel): transmissivity and reflectivity cannot sum to greater than 1 ensure energy conservation. Band " + band + ", Primitive #" + std::to_string(UUID) + ": eps=" + std::to_string(eps) +
                                             ", tau=" + std::to_string(tau.at(s).at(u).at(b)) + ", rho=" + std::to_string(rho.at(s).at(u).at(b)) + ". It is also possible that you forgot to disable emission for this band.");
                    }
                }
                b++;
            }
        }
    }
 
    std::vector<float> rho_flat = flatten(rho);
    std::vector<float> tau_flat = flatten(tau);
    std::vector<float> rho_cam_flat = flatten(rho_cam);
    std::vector<float> tau_cam_flat = flatten(tau_cam);
 
    // Upload material properties to backend
    material_data.num_primitives = Nprimitives;
    material_data.num_bands = radiation_bands.size();
    material_data.num_sources = radiation_sources.size();
    material_data.num_cameras = cameras.size();
    material_data.reflectivity = rho_flat;
    material_data.transmissivity = tau_flat;
    material_data.reflectivity_cam = rho_cam_flat;
    material_data.transmissivity_cam = tau_cam_flat;
 
    // Specular reflection properties
    material_data.specular_exponent.resize(Nprimitives, -1.f);
    material_data.specular_scale.resize(Nprimitives, 0.f);
 
    bool specular_exponent_specified = false;
    bool specular_scale_specified = false;
 
    for (size_t u = 0; u < Nprimitives; u++) {
        uint UUID = context_UUIDs.at(u);
 
        if (context->doesPrimitiveDataExist(UUID, "specular_exponent") && context->getPrimitiveDataType("specular_exponent") == HELIOS_TYPE_FLOAT) {
            context->getPrimitiveData(UUID, "specular_exponent", material_data.specular_exponent.at(u));
            if (material_data.specular_exponent.at(u) >= 0.f) {
                specular_exponent_specified = true;
            }
        }
 
        if (context->doesPrimitiveDataExist(UUID, "specular_scale") && context->getPrimitiveDataType("specular_scale") == HELIOS_TYPE_FLOAT) {
            context->getPrimitiveData(UUID, "specular_scale", material_data.specular_scale.at(u));
            if (material_data.specular_scale.at(u) > 0.f) {
                specular_scale_specified = true;
            }
        }
    }
 
    // Auto-enable specular reflection if specular properties are specified on any primitive
    if (specular_exponent_specified) {
        if (specular_scale_specified) {
            specular_reflection_mode = 2; // Mode 2: use primitive specular_scale
        } else {
            specular_reflection_mode = 1; // Mode 1: use default 0.25 scale
        }
    } else {
        specular_reflection_mode = 0; // Disabled
    }
 
    backend->updateMaterials(material_data);
 
    radiativepropertiesneedupdate = false;
 
    if (message_flag) {
        std::cout << "done\n";
    }
 
    // Report aggregated warnings
    warnings.report(std::cerr);
}
 
std::vector<float> RadiationModel::updateAtmosphericSkyModel(const std::vector<std::string> &band_labels, const RadiationCamera &camera) {
    // Prague Sky Model implementation for atmospheric sky radiance
    // Uses validated spectral radiance from brute-force atmospheric simulations
    // (Wilkie et al. 2021, Vévoda et al. 2022)
 
    size_t Nbands_launch = band_labels.size();
    std::vector<float> sky_base_radiances(Nbands_launch, 0.0f);
 
    // Only run atmospheric sky model if user has explicitly enabled it by setting atmospheric parameters
    // This prevents the model from running with default values in tests/scripts that don't want it
    bool has_atmospheric_data =
            context->doesGlobalDataExist("atmosphere_pressure_Pa") || context->doesGlobalDataExist("atmosphere_temperature_K") || context->doesGlobalDataExist("atmosphere_humidity_rel") || context->doesGlobalDataExist("atmosphere_turbidity");
 
    if (!has_atmospheric_data) {
        // No atmospheric parameters set - return zeros (camera will use user-set diffuse flux or 0)
        return sky_base_radiances;
    }
 
    // Read atmospheric parameters from Context global data (set by SolarPosition plugin)
    // Default values match SolarPosition::getAtmosphericConditions() defaults
    float pressure_Pa = 101325.f; // Standard atmosphere (1 atm)
    float temperature_K = 300.f; // 27°C
    float humidity_rel = 0.5f; // 50% relative humidity
    float turbidity = 0.02f; // Clear sky - Ångström's aerosol turbidity coefficient (AOD at 500nm)
 
    if (context->doesGlobalDataExist("atmosphere_pressure_Pa")) {
        context->getGlobalData("atmosphere_pressure_Pa", pressure_Pa);
    }
    if (context->doesGlobalDataExist("atmosphere_temperature_K")) {
        context->getGlobalData("atmosphere_temperature_K", temperature_K);
    }
    if (context->doesGlobalDataExist("atmosphere_humidity_rel")) {
        context->getGlobalData("atmosphere_humidity_rel", humidity_rel);
    }
    if (context->doesGlobalDataExist("atmosphere_turbidity")) {
        context->getGlobalData("atmosphere_turbidity", turbidity);
    }
 
    // --- Check Prague data availability from Context ---
    int prague_valid = 0;
    if (context->doesGlobalDataExist("prague_sky_valid")) {
        context->getGlobalData("prague_sky_valid", prague_valid);
    }
 
    // Get sun direction from first radiation source (assumed to be sun)
    helios::vec3 sun_dir(0, 0, 1); // Default zenith
    if (!radiation_sources.empty()) {
        sun_dir = radiation_sources[0].source_position;
        sun_dir.normalize();
    }
 
    // Compute per-band sky radiance parameters
    std::vector<helios::vec4> sky_params(Nbands_launch);
 
    // Check if Prague data is available
    bool use_prague_fallback = (prague_valid != 1);
    if (use_prague_fallback) {
        // Will use Rayleigh sky fallback - warn user once
        std::cerr << "WARNING (RadiationModel::updateAtmosphericSkyModel): "
                  << "Prague sky model data not available in Context. "
                  << "Using simple Rayleigh sky fallback. "
                  << "Call SolarPosition::updatePragueSkyModel() for accurate sky radiance." << std::endl;
    }
 
    // Prepare spectral data (either Prague or Rayleigh fallback)
    std::vector<float> wavelengths;
    std::vector<float> L_zenith_spectrum;
    std::vector<float> circ_str_spectrum;
    std::vector<float> circ_width_spectrum;
    std::vector<float> horiz_bright_spectrum;
    std::vector<float> norm_spectrum;
 
    if (use_prague_fallback) {
        // --- Create simple Rayleigh sky spectrum (λ^-4 dependence) ---
        // 360-750 nm at 10 nm spacing (visible range only for fallback)
        const int n_wavelengths = 40; // (750-360)/10 + 1
        wavelengths.resize(n_wavelengths);
        L_zenith_spectrum.resize(n_wavelengths);
        circ_str_spectrum.resize(n_wavelengths);
        circ_width_spectrum.resize(n_wavelengths);
        horiz_bright_spectrum.resize(n_wavelengths);
        norm_spectrum.resize(n_wavelengths);
 
        const float L_base = 0.4f; // W/m²/sr/nm at 550 nm (typical clear sky zenith)
        const float lambda_ref = 550.0f; // Reference wavelength
 
        for (int i = 0; i < n_wavelengths; ++i) {
            float lambda = 360.0f + i * 10.0f;
            wavelengths[i] = lambda;
 
            // Rayleigh scattering: L(λ) ∝ λ^-4 (blue sky)
            float rayleigh_factor = std::pow(lambda_ref / lambda, 4.0f);
            L_zenith_spectrum[i] = L_base * rayleigh_factor;
 
            // Simple angular parameters (no strong circumsolar for fallback)
            circ_str_spectrum[i] = 0.5f;
            circ_width_spectrum[i] = 20.0f;
            horiz_bright_spectrum[i] = 1.8f;
            norm_spectrum[i] = 0.7f;
        }
    } else {
        // --- Read spectral parameters from Context ---
        std::vector<float> spectral_params;
        context->getGlobalData("prague_sky_spectral_params", spectral_params);
 
        const int params_per_wavelength = 6;
        const int n_wavelengths = spectral_params.size() / params_per_wavelength;
 
        // Parse into structured format
        wavelengths.resize(n_wavelengths);
        L_zenith_spectrum.resize(n_wavelengths);
        circ_str_spectrum.resize(n_wavelengths);
        circ_width_spectrum.resize(n_wavelengths);
        horiz_bright_spectrum.resize(n_wavelengths);
        norm_spectrum.resize(n_wavelengths);
 
        for (int i = 0; i < n_wavelengths; ++i) {
            int base = i * params_per_wavelength;
            wavelengths[i] = spectral_params[base + 0];
            L_zenith_spectrum[i] = spectral_params[base + 1];
            circ_str_spectrum[i] = spectral_params[base + 2];
            circ_width_spectrum[i] = spectral_params[base + 3];
            horiz_bright_spectrum[i] = spectral_params[base + 4];
            norm_spectrum[i] = spectral_params[base + 5];
        }
    }
 
    // --- Process each band ---
    for (size_t b = 0; b < Nbands_launch; b++) {
        const std::string &band_label = band_labels[b];
        if (radiation_bands.find(band_label) == radiation_bands.end()) {
            continue;
        }
 
        const RadiationBand &band = radiation_bands.at(band_label);
 
        // Skip thermal/longwave bands - Prague Sky Model only handles shortwave radiation
        if (band.emissionFlag) {
            continue;
        }
 
        // Get camera spectral response for this band
        std::string spectral_response_label = "uniform";
        if (camera.band_spectral_response.find(band_label) != camera.band_spectral_response.end()) {
            spectral_response_label = camera.band_spectral_response.at(band_label);
            if (spectral_response_label.empty() || trim_whitespace(spectral_response_label).empty()) {
                spectral_response_label = "uniform";
            }
        }
 
        // Get camera spectral response data
        std::vector<helios::vec2> camera_response;
 
        if (spectral_response_label == "uniform") {
            helios::vec2 wavelength_range = band.wavebandBounds;
 
            if (wavelength_range.x <= 0.f || wavelength_range.y <= 0.f) {
                bool bounds_inferred = false;
 
                if (band_label == "red" || band_label == "R") {
                    wavelength_range = helios::make_vec2(620.f, 750.f);
                    bounds_inferred = true;
                } else if (band_label == "green" || band_label == "G") {
                    wavelength_range = helios::make_vec2(495.f, 570.f);
                    bounds_inferred = true;
                } else if (band_label == "blue" || band_label == "B") {
                    wavelength_range = helios::make_vec2(450.f, 495.f);
                    bounds_inferred = true;
                }
 
                if (!bounds_inferred) {
                    if (!band.diffuse_spectrum.empty()) {
                        wavelength_range.x = band.diffuse_spectrum.front().x;
                        wavelength_range.y = band.diffuse_spectrum.back().x;
                    } else {
                        helios_runtime_error("ERROR (RadiationModel::updateAtmosphericSkyModel): Camera '" + camera.label + "' band '" + band_label + "' has uniform spectral response but no wavelength bounds set.");
                    }
                }
            }
 
            camera_response.push_back(helios::make_vec2(wavelength_range.x, 1.0f));
            camera_response.push_back(helios::make_vec2(wavelength_range.y, 1.0f));
 
        } else {
            camera_response = loadSpectralData(spectral_response_label);
 
            if (camera_response.empty()) {
                helios_runtime_error("ERROR (RadiationModel::updateAtmosphericSkyModel): Camera spectral response '" + spectral_response_label + "' not found for camera '" + camera.label + "' band '" + band_label + "'.");
            }
        }
 
        // Integrate radiance and weight-average angular parameters over camera response
        // L_zenith: Integrate to get W/m²/sr (band-integrated radiance)
        float integrated_L_zenith = integrateOverResponse(wavelengths, L_zenith_spectrum, camera_response);
 
        // Angular parameters: Weighted average (unitless quantities)
        // Weight by L_zenith(λ) × R(λ) to get radiance-weighted average
        float integrated_circ_str = weightedAverageOverResponse(wavelengths, circ_str_spectrum, L_zenith_spectrum, camera_response);
        float integrated_circ_width = weightedAverageOverResponse(wavelengths, circ_width_spectrum, L_zenith_spectrum, camera_response);
        float integrated_horiz_bright = weightedAverageOverResponse(wavelengths, horiz_bright_spectrum, L_zenith_spectrum, camera_response);
 
        // Recompute normalization from averaged angular parameters
        float integrated_norm = computeAngularNormalization(integrated_circ_str, integrated_circ_width, integrated_horiz_bright);
 
        // CRITICAL: GPU multiplies by normalization (see rayHit.cu:evaluateSkyRadiance)
        // Since normalization < 1 (typically 0.6-0.7), this darkens the sky
        // Pre-divide by normalization so it cancels: GPU does (L/norm) × pattern × norm = L × pattern
        float base_radiance_for_gpu = integrated_L_zenith / std::max(integrated_norm, 0.1f);
 
        sky_base_radiances[b] = base_radiance_for_gpu;
        sky_params[b] = helios::make_vec4(integrated_circ_str, integrated_circ_width, integrated_horiz_bright, integrated_norm);
    }
 
    // Sky parameters will be uploaded to backend via updateSkyModel()
    return sky_base_radiances;
}
 
void RadiationModel::updatePragueParametersForGeneralDiffuse(const std::vector<std::string> &band_labels) {
    // Update Prague sky model angular parameters for general diffuse radiation
    // Reads spectral parameters from Context (set by SolarPosition::updatePragueSkyModel())
    // Integrates over band spectral response to get band-averaged parameters
 
    // Check Prague data availability
    int prague_valid = 0;
    if (!context->doesGlobalDataExist("prague_sky_valid") || (context->getGlobalData("prague_sky_valid", prague_valid), prague_valid != 1)) {
        // No Prague data - leave params at zero (will use power-law or isotropic)
        return;
    }
 
    // Read spectral parameters from Context
    std::vector<float> spectral_params;
    context->getGlobalData("prague_sky_spectral_params", spectral_params);
 
    // Parse into wavelength-resolved arrays
    const int params_per_wavelength = 6;
    const int n_wavelengths = spectral_params.size() / params_per_wavelength;
 
    std::vector<float> wavelengths(n_wavelengths);
    std::vector<float> L_zenith_spectrum(n_wavelengths);
    std::vector<float> circ_str_spectrum(n_wavelengths);
    std::vector<float> circ_width_spectrum(n_wavelengths);
    std::vector<float> horiz_bright_spectrum(n_wavelengths);
    std::vector<float> norm_spectrum(n_wavelengths);
 
    for (int i = 0; i < n_wavelengths; ++i) {
        int base = i * params_per_wavelength;
        wavelengths[i] = spectral_params[base + 0];
        L_zenith_spectrum[i] = spectral_params[base + 1];
        circ_str_spectrum[i] = spectral_params[base + 2];
        circ_width_spectrum[i] = spectral_params[base + 3];
        horiz_bright_spectrum[i] = spectral_params[base + 4];
        norm_spectrum[i] = spectral_params[base + 5];
    }
 
    // Get sun direction
    helios::vec3 sun_dir;
    context->getGlobalData("prague_sky_sun_direction", sun_dir);
 
    // Process each band
    for (const auto &label: band_labels) {
        RadiationBand &band = radiation_bands.at(label);
 
        // SKIP if user has explicitly set power-law (priority 1)
        if (band.diffuseExtinction > 0.0f) {
            continue;
        }
 
        // Integrate Prague parameters over band spectrum
        std::vector<helios::vec2> band_spectrum = band.diffuse_spectrum;
        if (band_spectrum.empty()) {
            // Use waveband bounds if no detailed spectrum
            float lambda_min = band.wavebandBounds.x;
            float lambda_max = band.wavebandBounds.y;
            if (lambda_min > 0 && lambda_max > lambda_min) {
                band_spectrum = {{lambda_min, 1.0f}, {lambda_max, 1.0f}};
            }
        }
 
        if (band_spectrum.empty()) {
            // No spectral info - skip Prague for this band
            continue;
        }
 
        // Weighted integration (weight by L_zenith for physical consistency)
        float int_circ_str = weightedAverageOverResponse(wavelengths, circ_str_spectrum, L_zenith_spectrum, band_spectrum);
        float int_circ_width = weightedAverageOverResponse(wavelengths, circ_width_spectrum, L_zenith_spectrum, band_spectrum);
        float int_horiz_bright = weightedAverageOverResponse(wavelengths, horiz_bright_spectrum, L_zenith_spectrum, band_spectrum);
 
        // Recompute normalization from integrated parameters
        float int_norm = computeAngularNormalization(int_circ_str, int_circ_width, int_horiz_bright);
 
        // Store in RadiationBand
        band.diffusePragueParams = helios::make_vec4(int_circ_str, int_circ_width, int_horiz_bright, int_norm);
        band.diffusePeakDir = sun_dir;
    }
}
 
float RadiationModel::integrateOverResponse(const std::vector<float> &wavelengths, const std::vector<float> &values, const std::vector<helios::vec2> &camera_response) const {
 
    if (wavelengths.empty() || camera_response.empty()) {
        return 0.0f;
    }
 
    // CRITICAL: This integrates spectral radiance L(λ) in W/m²/sr/nm over wavelength
    // to produce band-integrated radiance in W/m²/sr (same as Prague computeIntegratedSkyRadiance)
    double integrated_radiance = 0.0;
 
    // Trapezoidal integration over camera response wavelength range
    for (size_t i = 0; i < camera_response.size() - 1; ++i) {
        float lambda1 = camera_response[i].x;
        float lambda2 = camera_response[i + 1].x;
 
        // Skip if outside spectral data range
        if (lambda2 < wavelengths.front() || lambda1 > wavelengths.back()) {
            continue;
        }
 
        float r1 = camera_response[i].y;
        float r2 = camera_response[i + 1].y;
 
        // Interpolate spectral values at these wavelengths using linear interpolation
        float v1, v2;
 
        // Interpolate v1 at lambda1
        if (lambda1 <= wavelengths.front()) {
            v1 = values.front();
        } else if (lambda1 >= wavelengths.back()) {
            v1 = values.back();
        } else {
            auto it = std::lower_bound(wavelengths.begin(), wavelengths.end(), lambda1);
            size_t idx = std::distance(wavelengths.begin(), it);
            if (idx == 0)
                idx = 1;
            float t = (lambda1 - wavelengths[idx - 1]) / (wavelengths[idx] - wavelengths[idx - 1]);
            v1 = values[idx - 1] + t * (values[idx] - values[idx - 1]);
        }
 
        // Interpolate v2 at lambda2
        if (lambda2 <= wavelengths.front()) {
            v2 = values.front();
        } else if (lambda2 >= wavelengths.back()) {
            v2 = values.back();
        } else {
            auto it = std::lower_bound(wavelengths.begin(), wavelengths.end(), lambda2);
            size_t idx = std::distance(wavelengths.begin(), it);
            if (idx == 0)
                idx = 1;
            float t = (lambda2 - wavelengths[idx - 1]) / (wavelengths[idx] - wavelengths[idx - 1]);
            v2 = values[idx - 1] + t * (values[idx] - values[idx - 1]);
        }
 
        float dlambda = lambda2 - lambda1;
 
        // Integrate: ∫ L(λ) × R(λ) dλ
        // L(λ) in W/m²/sr/nm, R(λ) unitless, dλ in nm → result in W/m²/sr
        integrated_radiance += 0.5 * (v1 * r1 + v2 * r2) * dlambda;
    }
 
    // Return band-integrated radiance in W/m²/sr (matches Prague computeIntegratedSkyRadiance)
    return static_cast<float>(integrated_radiance);
}
 
float RadiationModel::weightedAverageOverResponse(const std::vector<float> &wavelengths, const std::vector<float> &param_values, const std::vector<float> &weight_values, const std::vector<helios::vec2> &camera_response) const {
 
    if (wavelengths.empty() || camera_response.empty()) {
        return 0.0f;
    }
 
    // CRITICAL: Angular parameters are unitless - compute radiance-weighted average
    // Formula: Σ param(λ) × L(λ) × R(λ) dλ / Σ L(λ) × R(λ) dλ
    double weighted_sum = 0.0;
    double total_weight = 0.0;
 
    for (size_t i = 0; i < camera_response.size() - 1; ++i) {
        float lambda1 = camera_response[i].x;
        float lambda2 = camera_response[i + 1].x;
 
        if (lambda2 < wavelengths.front() || lambda1 > wavelengths.back()) {
            continue;
        }
 
        float r1 = camera_response[i].y;
        float r2 = camera_response[i + 1].y;
 
        // Interpolate parameter values
        float p1, p2;
        if (lambda1 <= wavelengths.front()) {
            p1 = param_values.front();
        } else if (lambda1 >= wavelengths.back()) {
            p1 = param_values.back();
        } else {
            auto it = std::lower_bound(wavelengths.begin(), wavelengths.end(), lambda1);
            size_t idx = std::distance(wavelengths.begin(), it);
            if (idx == 0)
                idx = 1;
            float t = (lambda1 - wavelengths[idx - 1]) / (wavelengths[idx] - wavelengths[idx - 1]);
            p1 = param_values[idx - 1] + t * (param_values[idx] - param_values[idx - 1]);
        }
 
        if (lambda2 <= wavelengths.front()) {
            p2 = param_values.front();
        } else if (lambda2 >= wavelengths.back()) {
            p2 = param_values.back();
        } else {
            auto it = std::lower_bound(wavelengths.begin(), wavelengths.end(), lambda2);
            size_t idx = std::distance(wavelengths.begin(), it);
            if (idx == 0)
                idx = 1;
            float t = (lambda2 - wavelengths[idx - 1]) / (wavelengths[idx] - wavelengths[idx - 1]);
            p2 = param_values[idx - 1] + t * (param_values[idx] - param_values[idx - 1]);
        }
 
        // Interpolate weight (radiance) values
        float w1, w2;
        if (lambda1 <= wavelengths.front()) {
            w1 = weight_values.front();
        } else if (lambda1 >= wavelengths.back()) {
            w1 = weight_values.back();
        } else {
            auto it = std::lower_bound(wavelengths.begin(), wavelengths.end(), lambda1);
            size_t idx = std::distance(wavelengths.begin(), it);
            if (idx == 0)
                idx = 1;
            float t = (lambda1 - wavelengths[idx - 1]) / (wavelengths[idx] - wavelengths[idx - 1]);
            w1 = weight_values[idx - 1] + t * (weight_values[idx] - weight_values[idx - 1]);
        }
 
        if (lambda2 <= wavelengths.front()) {
            w2 = weight_values.front();
        } else if (lambda2 >= wavelengths.back()) {
            w2 = weight_values.back();
        } else {
            auto it = std::lower_bound(wavelengths.begin(), wavelengths.end(), lambda2);
            size_t idx = std::distance(wavelengths.begin(), it);
            if (idx == 0)
                idx = 1;
            float t = (lambda2 - wavelengths[idx - 1]) / (wavelengths[idx] - wavelengths[idx - 1]);
            w2 = weight_values[idx - 1] + t * (weight_values[idx] - weight_values[idx - 1]);
        }
 
        float dlambda = lambda2 - lambda1;
 
        // Weighted average: Σ param × weight × response × dλ
        weighted_sum += 0.5 * (p1 * w1 * r1 + p2 * w2 * r2) * dlambda;
        total_weight += 0.5 * (w1 * r1 + w2 * r2) * dlambda;
    }
 
    // Return weighted average (unitless)
    if (total_weight > 1e-10) {
        return static_cast<float>(weighted_sum / total_weight);
    }
    return 0.0f;
}
 
float RadiationModel::computeAngularNormalization(float circ_str, float circ_width, float horiz_bright) const {
    // Numerical integration of angular pattern over hemisphere
    const int N = 50;
    float integral = 0.0f;
 
    // Sun at zenith for normalization calculation
    helios::vec3 sun_dir = make_vec3(0, 0, 1);
 
    for (int j = 0; j < N; ++j) {
        for (int i = 0; i < N; ++i) {
            float theta = 0.5f * float(M_PI) * (i + 0.5f) / N; // 0 to π/2
            float phi = 2.0f * float(M_PI) * (j + 0.5f) / N; // 0 to 2π
 
            helios::vec3 dir = sphere2cart(make_SphericalCoord(0.5f * float(M_PI) - theta, phi));
 
            // Angular distance from sun (degrees) - matches GPU calculation
            float cos_gamma = std::max(-1.0f, std::min(1.0f, dir.x * sun_dir.x + dir.y * sun_dir.y + dir.z * sun_dir.z));
            float gamma = std::acos(cos_gamma) * 180.0f / float(M_PI);
 
            // Compute angular pattern (same as GPU: rayHit.cu evaluateDiffuseAngularDistribution)
            float cos_theta = std::max(0.0f, dir.z);
            float horizon_term = 1.0f + (horiz_bright - 1.0f) * (1.0f - cos_theta);
            float circ_term = 1.0f + circ_str * std::exp(-gamma / circ_width);
 
            float pattern = circ_term * horizon_term;
 
            // Solid angle element: sin(θ) dθ dφ
            integral += pattern * std::cos(theta) * std::sin(theta) * (float(M_PI) / (2.0f * N)) * (2.0f * float(M_PI) / N);
        }
    }
 
    return 1.0f / std::max(integral, 1e-10f);
}
 
std::vector<helios::vec2> RadiationModel::loadSpectralData(const std::string &global_data_label) const {
 
    std::vector<helios::vec2> spectrum;
 
    if (!context->doesGlobalDataExist(global_data_label.c_str())) {
 
        // check if spectral data exists in any of the library files
        bool data_found = false;
        for (const auto &file: spectral_library_files) {
            if (Context::scanXMLForTag(file, "globaldata_vec2", global_data_label)) {
                context->loadXML(file.c_str(), true);
                data_found = true;
                break;
            }
        }
 
        if (!data_found) {
            helios_runtime_error("ERROR (RadiationModel::loadSpectralData): Global data for spectrum '" + global_data_label + "' could not be found.");
        }
    }
 
    if (context->getGlobalDataType(global_data_label.c_str()) != HELIOS_TYPE_VEC2) {
        helios_runtime_error("ERROR (RadiationModel::loadSpectralData): Global data for spectrum '" + global_data_label + "' is not of type HELIOS_TYPE_VEC2.");
    }
 
    context->getGlobalData(global_data_label.c_str(), spectrum);
 
    // validate spectrum
    if (spectrum.empty()) {
        helios_runtime_error("ERROR (RadiationModel::loadSpectralData): Global data for spectrum '" + global_data_label + "' is empty.");
    }
    for (auto s = 0; s < spectrum.size(); s++) {
        // check that wavelengths are monotonic
        if (s > 0 && spectrum.at(s).x <= spectrum.at(s - 1).x) {
            helios_runtime_error("ERROR (RadiationModel::loadSpectralData): Source spectral data validation failed. Wavelengths must increase monotonically.");
        }
        // check that wavelength is within a reasonable range
        if (spectrum.at(s).x < 0 || spectrum.at(s).x > 100000) {
            helios_runtime_error("ERROR (RadiationModel::loadSpectralData): Source spectral data validation failed. Wavelength value of " + std::to_string(spectrum.at(s).x) + " appears to be erroneous.");
        }
        // check that flux is non-negative
        if (spectrum.at(s).y < 0) {
            helios_runtime_error("ERROR (RadiationModel::loadSpectralData): Source spectral data validation failed. Flux value at wavelength of " + std::to_string(spectrum.at(s).x) + " appears is negative.");
        }
    }
 
    return spectrum;
}
 
void RadiationModel::runBand(const std::string &label) {
    std::vector<std::string> labels{label};
    runBand(labels);
}
 
void RadiationModel::runBand(const std::vector<std::string> &label) {
 
    //----- VERIFICATIONS -----//
 
    // We need the band label strings to appear in the same order as in radiation_bands map.
    // this is because that was the order in which radiative properties were laid out when updateRadiativeProperties() was called
    std::vector<std::string> band_labels;
    for (auto &band: radiation_bands) {
        if (std::find(label.begin(), label.end(), band.first) != label.end()) {
            band_labels.push_back(band.first);
        }
    }
 
    // Check to make sure some geometry was added to the context
    if (context->getPrimitiveCount() == 0) {
        std::cerr << "WARNING (RadiationModel::runBand): No geometry was added to the context. There is nothing to simulate...exiting." << std::endl;
        return;
    }
 
    // Check to make sure geometry was built in OptiX
    if (!isgeometryinitialized) {
        updateGeometry();
    }
 
    // Check that all the bands passed to the runBand() method exist
    for (const std::string &band: label) {
        if (!doesBandExist(band)) {
            helios_runtime_error("ERROR (RadiationModel::runBand): Cannot run band " + band + " because it is not a valid band. Use addRadiationBand() function to add the band.");
        }
    }
 
    // if there are no radiation sources in the simulation, add at least one but with zero fluxes
    if (radiation_sources.empty()) {
        addCollimatedRadiationSource();
    }
 
    // Check if any source spectra have changed in global data and reload if necessary
    for (auto &source: radiation_sources) {
        if (!source.source_spectrum_label.empty() && source.source_spectrum_label != "none") {
            uint64_t current_version = context->getGlobalDataVersion(source.source_spectrum_label.c_str());
            if (current_version != source.source_spectrum_version) {
                // Reload spectrum from global data
                source.source_spectrum = loadSpectralData(source.source_spectrum_label);
                source.source_spectrum_version = current_version;
                radiativepropertiesneedupdate = true;
            }
        }
    }
 
    // Check if global diffuse spectrum has changed and reload if necessary
    if (!global_diffuse_spectrum_label.empty() && global_diffuse_spectrum_label != "none") {
        uint64_t current_version = context->getGlobalDataVersion(global_diffuse_spectrum_label.c_str());
        if (current_version != global_diffuse_spectrum_version) {
            // Reload diffuse spectrum from global data
            global_diffuse_spectrum = loadSpectralData(global_diffuse_spectrum_label);
            global_diffuse_spectrum_version = current_version;
            // Also update all band diffuse spectra
            for (auto &band_pair: radiation_bands) {
                band_pair.second.diffuse_spectrum = global_diffuse_spectrum;
            }
            radiativepropertiesneedupdate = true;
        }
    }
 
    if (radiativepropertiesneedupdate) {
        // Use old material path (handles spectrum interpolation)
        updateRadiativeProperties();
        // DON'T call backend->updateMaterials() - old code already uploaded via direct OptiX calls
    } else {
        // Use new backend path (per-band materials only)
        buildMaterialData();
        backend->updateMaterials(material_data);
    }
 
    // Upload sources to backend (always use new path)
    buildSourceData();
    backend->updateSources(source_data);
 
    // Prepare launch parameters (these will be passed to backend via RayTracingLaunchParams)
    size_t Nbands_launch = band_labels.size();
    size_t Nbands_global = radiation_bands.size();
 
    // Build band launch flags
    std::vector<char> band_launch_flag(Nbands_global);
    uint bb = 0;
    for (auto &band: radiation_bands) {
        if (std::find(band_labels.begin(), band_labels.end(), band.first) != band_labels.end()) {
            band_launch_flag.at(bb) = 1;
        }
        bb++;
    }
 
    // Get dimensions
    size_t Nobjects = primitiveID.size();
    size_t Nprimitives = context_UUIDs.size();
    uint Nsources = radiation_sources.size();
    uint Ncameras = cameras.size();
 
    // Note: Atmospheric sky radiance model is updated per-camera (see camera trace loop below)
    // This allows us to use camera-specific spectral responses for each band
 
    // Set scattering depth for each band
    std::vector<uint> scattering_depth(Nbands_launch);
    bool scatteringenabled = false;
    for (auto b = 0; b < Nbands_launch; b++) {
        scattering_depth.at(b) = radiation_bands.at(band_labels.at(b)).scatteringDepth;
        if (scattering_depth.at(b) > 0) {
            scatteringenabled = true;
        }
    }
 
    // Issue warning if rho>0, tau>0, or eps<1
    for (int b = 0; b < Nbands_launch; b++) {
        if (scattering_depth.at(b) == 0 && scattering_iterations_needed.at(band_labels.at(b))) {
            std::cout << "WARNING (RadiationModel::runBand): Surface radiative properties for band " << band_labels.at(b)
                      << " are set to non-default values, but scattering iterations are disabled. Surface radiative properties will be ignored unless scattering depth is non-zero." << std::endl;
        }
    }
 
    // Set diffuse flux for each band
    std::vector<float> diffuse_flux(Nbands_launch);
    bool diffuseenabled = false;
    for (auto b = 0; b < Nbands_launch; b++) {
        diffuse_flux.at(b) = getDiffuseFlux(band_labels.at(b));
        if (diffuse_flux.at(b) > 0.f) {
            diffuseenabled = true;
        }
    }
    // NOTE: diffuse_flux now passed to backend via launch params, not uploaded here
 
    // Initialize camera sky radiance buffer to zeros (will be set per-camera if atmospheric model is used)
    std::vector<float> camera_sky_radiance(Nbands_launch, 0.0f);
 
    // Update Prague parameters for general diffuse (if available in Context)
    // This must be done before uploading diffuse parameters to GPU
    if (diffuseenabled) {
        updatePragueParametersForGeneralDiffuse(band_labels);
    }
 
    // Set diffuse extinction coefficient for each band
    std::vector<float> diffuse_extinction(Nbands_launch, 0);
    if (diffuseenabled) {
        for (auto b = 0; b < Nbands_launch; b++) {
            diffuse_extinction.at(b) = radiation_bands.at(band_labels.at(b)).diffuseExtinction;
        }
    }
    // NOTE: diffuse_extinction now passed to backend via launch params, not uploaded here
 
    // Set diffuse distribution normalization factor for each band
    std::vector<float> diffuse_dist_norm(Nbands_launch, 0);
    if (diffuseenabled) {
        for (auto b = 0; b < Nbands_launch; b++) {
            diffuse_dist_norm.at(b) = radiation_bands.at(band_labels.at(b)).diffuseDistNorm;
        }
    }
    // NOTE: diffuse_dist_norm now passed to backend via launch params, not uploaded here
 
    // Set diffuse distribution peak direction for each band
    std::vector<helios::vec3> diffuse_peak_dir(Nbands_launch);
    if (diffuseenabled) {
        for (auto b = 0; b < Nbands_launch; b++) {
            helios::vec3 peak_dir = radiation_bands.at(band_labels.at(b)).diffusePeakDir;
            diffuse_peak_dir.at(b) = helios::make_vec3(peak_dir.x, peak_dir.y, peak_dir.z);
        }
    }
    // NOTE: diffuse_peak_dir now passed to backend via launch params, not uploaded here
 
    // Upload Prague parameters for general diffuse (reuses camera buffer)
    // This allows general diffuse to use Prague sky model if available
    std::vector<helios::vec4> prague_params(Nbands_launch);
    if (diffuseenabled) {
        for (auto b = 0; b < Nbands_launch; b++) {
            const auto &params = radiation_bands.at(band_labels.at(b)).diffusePragueParams;
            prague_params.at(b) = helios::make_vec4(params.x, params.y, params.z, params.w);
        }
        // Prague params will be uploaded to backend via updateSkyModel() during scattering
    }
 
    // Determine whether emission is enabled for any band
    bool emissionenabled = false;
    for (auto b = 0; b < Nbands_launch; b++) {
        if (radiation_bands.at(band_labels.at(b)).emissionFlag) {
            emissionenabled = true;
        }
    }
 
    // Figure out the maximum direct ray count for all bands in this run and use this as the launch size
    size_t directRayCount = 0;
    for (const auto &band: label) {
        if (radiation_bands.at(band).directRayCount > directRayCount) {
            directRayCount = radiation_bands.at(band).directRayCount;
        }
    }
 
    // Figure out the maximum diffuse ray count for all bands in this run and use this as the launch size
    size_t diffuseRayCount = 0;
    for (const auto &band: label) {
        if (radiation_bands.at(band).diffuseRayCount > diffuseRayCount) {
            diffuseRayCount = radiation_bands.at(band).diffuseRayCount;
        }
    }
 
    // Figure out the maximum diffuse ray count for all bands in this run and use this as the launch size
    size_t scatteringDepth = 0;
    for (const auto &band: label) {
        if (radiation_bands.at(band).scatteringDepth > scatteringDepth) {
            scatteringDepth = radiation_bands.at(band).scatteringDepth;
        }
    }
 
    // Zero radiation buffers via backend
    backend->zeroRadiationBuffers(Nbands_launch);
 
    std::vector<float> TBS_top, TBS_bottom;
    TBS_top.resize(Nbands_launch * Nprimitives, 0);
    TBS_bottom = TBS_top;
 
    std::map<std::string, std::vector<std::vector<float>>> radiation_in_camera;
 
    size_t maxRays = 1024 * 1024 * 1024; // maximum number of total rays in a launch
 
    // ***** DIRECT LAUNCH FROM ALL RADIATION SOURCES ***** //
 
    helios::int3 launch_dim_dir;
 
    bool rundirect = false;
    for (uint s = 0; s < Nsources; s++) {
        for (uint b = 0; b < Nbands_launch; b++) {
            if (getSourceFlux(s, band_labels.at(b)) > 0.f) {
                rundirect = true;
                break;
            }
        }
    }
 
    if (Nsources > 0 && rundirect) {
 
        // update radiation source buffers
 
        std::vector<std::vector<float>> fluxes; // first index is the source, second index is the band (only those passed to runBand() function)
        fluxes.resize(Nsources);
        std::vector<helios::vec3> positions(Nsources);
        std::vector<helios::vec2> widths(Nsources);
        std::vector<helios::vec3> rotations(Nsources);
        std::vector<uint> types(Nsources);
 
        size_t s = 0;
        for (const auto &source: radiation_sources) {
 
            fluxes.at(s).resize(Nbands_launch);
 
            for (auto b = 0; b < label.size(); b++) {
                fluxes.at(s).at(b) = getSourceFlux(s, band_labels.at(b));
            }
 
            positions.at(s) = helios::make_vec3(source.source_position.x, source.source_position.y, source.source_position.z);
            widths.at(s) = helios::make_vec2(source.source_width.x, source.source_width.y);
            rotations.at(s) = helios::make_vec3(source.source_rotation.x, source.source_rotation.y, source.source_rotation.z);
            types.at(s) = source.source_type;
 
            s++;
        }
 
        // Upload band-specific source fluxes to backend buffer (indexed by Nbands_launch, not Nbands_global)
        backend->uploadSourceFluxes(flatten(fluxes));
        // Note: positions, widths, rotations, types are uploaded once in buildSourceData()
        // Only fluxes need per-launch update because they depend on which bands are being run
 
        // Compute camera response weighting factors for specular reflection (if cameras exist)
        // Factor = ∫(source_spectrum × camera_response) / ∫(source_spectrum)
        // This must be done before ray tracing so the weights are available during miss_direct()
        if (Ncameras > 0) {
            std::vector<float> source_fluxes_cam;
            source_fluxes_cam.resize(Nsources * Nbands_launch * Ncameras, 1.0f);
 
            for (uint s = 0; s < Nsources; s++) {
                const RadiationSource &source = radiation_sources.at(s);
 
                uint cam = 0;
                for (const auto &camera: cameras) {
                    for (uint b = 0; b < Nbands_launch; b++) {
                        std::string band_label = band_labels.at(b);
 
                        // Default weighting factor (no camera response)
                        float weight = 1.0f;
 
                        // Check if camera has spectral response for this band
                        if (camera.second.band_spectral_response.find(band_label) != camera.second.band_spectral_response.end()) {
                            std::string response_label = camera.second.band_spectral_response.at(band_label);
 
                            if (!response_label.empty() && response_label != "uniform" && context->doesGlobalDataExist(response_label.c_str()) && context->getGlobalDataType(response_label.c_str()) == helios::HELIOS_TYPE_VEC2 &&
                                source.source_spectrum.size() > 0) {
 
                                // Load camera spectral response
                                std::vector<helios::vec2> camera_response;
                                context->getGlobalData(response_label.c_str(), camera_response);
 
                                // Get band wavelength range
                                helios::vec2 wavelength_range = radiation_bands.at(band_label).wavebandBounds;
 
                                // If no wavelength bounds, use overlapping range of source and camera
                                if (wavelength_range.x == 0 && wavelength_range.y == 0) {
                                    wavelength_range.x = fmax(source.source_spectrum.front().x, camera_response.front().x);
                                    wavelength_range.y = fmin(source.source_spectrum.back().x, camera_response.back().x);
                                }
 
                                // Integrate source_spectrum × camera_response over band
                                // Note: integrateSpectrum already returns ratio: ∫(source × camera) / ∫(source)
                                weight = integrateSpectrum(s, camera_response, wavelength_range.x, wavelength_range.y);
                            }
                        }
 
                        source_fluxes_cam[s * Nbands_launch * Ncameras + b * Ncameras + cam] = weight;
                    }
                    cam++;
                }
            }
 
            // Update source_data with camera-weighted fluxes and re-upload to backend
            for (uint s = 0; s < Nsources; s++) {
                source_data[s].fluxes_cam.clear();
                for (uint b = 0; b < Nbands_launch; b++) {
                    for (uint cam = 0; cam < Ncameras; cam++) {
                        source_data[s].fluxes_cam.push_back(source_fluxes_cam[s * Nbands_launch * Ncameras + b * Ncameras + cam]);
                    }
                }
            }
            backend->updateSources(source_data);
        }
 
        // -- Ray Trace (Using Backend) -- //
 
        if (message_flag) {
            std::cout << "Performing primary direct radiation ray trace for bands ";
            for (const auto &band: label) {
                std::cout << band << ", ";
            }
            std::cout << "..." << std::flush;
        }
 
        // Launch direct rays through backend
        helios::RayTracingLaunchParams params;
        params.launch_offset = 0;
        params.launch_count = Nprimitives; // Launch all primitives at once
        params.rays_per_primitive = directRayCount;
        params.random_seed = std::chrono::system_clock::now().time_since_epoch().count();
        params.num_bands_global = Nbands_global;
        params.num_bands_launch = Nbands_launch;
        params.specular_reflection_enabled = false;
 
        // Use the band_launch_flag already built above (lines 3375-3383)
        std::vector<bool> band_flags(band_launch_flag.begin(), band_launch_flag.end());
        params.band_launch_flag = band_flags;
 
        backend->launchDirectRays(params);
 
        if (message_flag) {
            std::cout << "done." << std::endl;
        }
 
    } // end direct source launch
 
    // --- Extract scattered energy from direct rays for diffuse/emission and scattering ---//
    // This needs to happen BEFORE diffuse/emission block so scattered direct energy
    // is available for both diffuse/emission (via flux_top/flux_bottom) and scattering (via radiation_out)
    std::vector<float> flux_top, flux_bottom;
    flux_top.resize(Nbands_launch * Nprimitives, 0);
    flux_bottom = flux_top;
 
    // Camera scatter accumulation vectors (declare early for use throughout ray tracing)
    std::vector<float> scatter_top_cam;
    std::vector<float> scatter_bottom_cam;
    if (Ncameras > 0) {
        scatter_top_cam.resize(Nprimitives * Nbands_launch, 0.0f);
        scatter_bottom_cam.resize(Nprimitives * Nbands_launch, 0.0f);
    }
 
    if (scatteringenabled && rundirect) {
        // Get scattered energy from direct rays for primary diffuse/emission
        helios::RayTracingResults scatter_results;
        backend->getRadiationResults(scatter_results);
        flux_top = scatter_results.scatter_buff_top;
        flux_bottom = scatter_results.scatter_buff_bottom;
 
        // Accumulate camera scatter from direct rays
        if (Ncameras > 0) {
            for (size_t i = 0; i < scatter_results.scatter_buff_top_cam.size(); i++) {
                scatter_top_cam[i] += scatter_results.scatter_buff_top_cam[i];
                scatter_bottom_cam[i] += scatter_results.scatter_buff_bottom_cam[i];
            }
            // Zero GPU camera scatter buffers to prevent double-counting on next iteration
            backend->zeroCameraScatterBuffers(Nbands_launch);
        }
 
        // For one-sided primitives, make scattered energy accessible from both faces
        // This is necessary because scattering rays can hit from either direction
        RadiationBufferIndexer rad_indexer(Nprimitives, Nbands_launch);
 
        for (size_t i = 0; i < Nprimitives; i++) {
            uint UUID = context_UUIDs.at(i);
            uint twosided = context->getPrimitiveTwosidedFlag(UUID, 1);
 
            if (twosided == 0) { // one-sided primitive - combine top+bottom scattered energy
                for (size_t b = 0; b < Nbands_launch; b++) {
                    size_t ind = rad_indexer(i, b);
                    float total = flux_top[ind] + flux_bottom[ind];
                    flux_top[ind] = total;
                    flux_bottom[ind] = total;
                }
            }
        }
 
        // Upload scattered energy to backend's radiation_out buffers for scattering iterations
        backend->uploadRadiationOut(flux_top, flux_bottom);
        backend->zeroScatterBuffers();
    }
 
    // --- Diffuse/Emission launch ---- //
 
    if (emissionenabled || diffuseenabled) {
 
        // add any emitted energy to the outgoing energy buffer
        if (emissionenabled) {
            // Update primitive outgoing emission
            float eps, temperature;
 
            // Create indexer for emission flux buffers
            RadiationBufferIndexer emission_indexer(Nprimitives, Nbands_launch);
 
            for (auto b = 0; b < Nbands_launch; b++) {
                //\todo For emissivity and twosided_flag, this should be done in updateRadiativeProperties() to avoid having to do it on every runBand() call
                if (radiation_bands.at(band_labels.at(b)).emissionFlag) {
                    std::string prop = "emissivity_" + band_labels.at(b);
                    for (size_t u = 0; u < Nprimitives; u++) {
                        // Use BufferIndexer: [primitive][band]
                        size_t ind = emission_indexer(u, b);
                        uint p = context_UUIDs.at(u);
                        if (context->doesPrimitiveDataExist(p, prop.c_str())) {
                            context->getPrimitiveData(p, prop.c_str(), eps);
                        } else {
                            eps = eps_default;
                        }
                        if (scattering_depth.at(b) == 0 && eps != 1.f) {
                            eps = 1.f;
                        }
                        if (context->doesPrimitiveDataExist(p, "temperature")) {
                            context->getPrimitiveData(p, "temperature", temperature);
                            if (temperature < 0) {
                                temperature = temperature_default;
                            }
                        } else {
                            temperature = temperature_default;
                        }
                        float out_top = sigma * eps * pow(temperature, 4);
                        flux_top.at(ind) += out_top;
                        if (Ncameras > 0) {
                            scatter_top_cam[ind] += out_top;
                        }
                        // Check twosided_flag - check material first, then primitive data
                        uint twosided_flag = context->getPrimitiveTwosidedFlag(p, 1);
                        if (twosided_flag != 0) { // If two-sided, emit from bottom face too
                            flux_bottom.at(ind) += flux_top.at(ind);
                            if (Ncameras > 0) {
                                scatter_bottom_cam[ind] += out_top;
                            }
                        }
                    }
                }
            }
        }
 
        // Upload camera scatter buffers accumulated from emission, direct rays, and primary diffuse
        // Camera scatter is accumulated on CPU from GPU after each ray launch
        if (Ncameras > 0) {
            backend->uploadCameraScatterBuffers(scatter_top_cam, scatter_bottom_cam);
        }
 
        // Note: radiation_specular_RTbuffer is populated on GPU via atomicFloatAdd during ray tracing, don't overwrite it here
 
        // Compute diffuse launch dimension
        size_t n = ceil(sqrt(double(diffuseRayCount)));
        uint rays_per_primitive = n * n;
 
        size_t maxPrims = floor(float(maxRays) / float(rays_per_primitive));
 
        int Nlaunches = ceil(rays_per_primitive * Nprimitives / float(maxRays));
 
        size_t prims_per_launch = fmin(Nprimitives, maxPrims);
 
        for (uint launch = 0; launch < Nlaunches; launch++) {
 
            size_t prims_this_launch;
            if ((launch + 1) * prims_per_launch > Nprimitives) {
                prims_this_launch = Nprimitives - launch * prims_per_launch;
            } else {
                prims_this_launch = prims_per_launch;
            }
 
            if (message_flag) {
                std::cout << "Performing primary diffuse radiation ray trace for bands ";
                for (const auto &band: label) {
                    std::cout << band << " ";
                }
                std::cout << " (batch " << launch + 1 << " of " << Nlaunches << ")..." << std::flush;
            }
 
            // Build launch parameters for diffuse rays
            helios::RayTracingLaunchParams params;
            params.launch_offset = launch * prims_per_launch;
            params.launch_count = prims_this_launch;
            params.rays_per_primitive = rays_per_primitive;
            params.random_seed = std::chrono::system_clock::now().time_since_epoch().count();
            params.current_band = 0;
            params.num_bands_global = Nbands_global;
            params.num_bands_launch = Nbands_launch;
            std::vector<bool> band_flags(band_launch_flag.begin(), band_launch_flag.end());
            params.band_launch_flag = band_flags;
            params.scattering_iteration = 0;
            params.max_scatters = scatteringDepth;
            params.radiation_out_top = flux_top;
            params.radiation_out_bottom = flux_bottom;
 
            // Pass diffuse radiation parameters to backend
            params.diffuse_flux = diffuse_flux;
            params.diffuse_extinction = diffuse_extinction;
            params.diffuse_dist_norm = diffuse_dist_norm;
            // Convert helios::vec3 to helios::vec3
            std::vector<helios::vec3> peak_dirs(diffuse_peak_dir.size());
            for (size_t i = 0; i < diffuse_peak_dir.size(); i++) {
                peak_dirs[i] = helios::make_vec3(diffuse_peak_dir[i].x, diffuse_peak_dir[i].y, diffuse_peak_dir[i].z);
            }
            params.diffuse_peak_dir = peak_dirs;
 
            // Top surface launch
            params.launch_face = 1;
            backend->launchDiffuseRays(params);
 
            // Bottom surface launch
            params.launch_face = 0;
            backend->launchDiffuseRays(params);
 
            // Retrieve and accumulate camera scatter from primary diffuse
            if (Ncameras > 0) {
                helios::RayTracingResults primary_results;
                backend->getRadiationResults(primary_results);
                for (size_t i = 0; i < primary_results.scatter_buff_top_cam.size(); i++) {
                    scatter_top_cam[i] += primary_results.scatter_buff_top_cam[i];
                    scatter_bottom_cam[i] += primary_results.scatter_buff_bottom_cam[i];
                }
                // Zero GPU camera scatter buffers to prevent double-counting on next iteration
                backend->zeroCameraScatterBuffers(Nbands_launch);
            }
 
            if (message_flag) {
                std::cout << "\r                                                                                                                               \r" << std::flush;
            }
        }
 
        if (message_flag) {
            std::cout << "Performing primary diffuse radiation ray trace for bands ";
            for (const auto &band: label) {
                std::cout << band << ", ";
            }
            std::cout << "...done." << std::endl;
        }
    }
 
    // After primary diffuse, prepare scatter_buff for scattering iterations
    // When direct rays ran, scatter_buff was already copied to radiation_out at line 3710
    // For emission/diffuse without direct rays, we need to do this now
    if (scatteringenabled && (emissionenabled || diffuseenabled) && !rundirect) {
        backend->copyScatterToRadiation();
        backend->zeroScatterBuffers();
    }
 
    if (scatteringenabled && (emissionenabled || diffuseenabled || rundirect)) {
 
        for (auto b = 0; b < Nbands_launch; b++) {
            diffuse_flux.at(b) = 0.f;
        }
        // NOTE: diffuse_flux zeroed for scattering, passed to backend via launch params
 
        size_t n = ceil(sqrt(double(diffuseRayCount)));
        uint rays_per_primitive = n * n;
 
        size_t maxPrims = floor(float(maxRays) / float(rays_per_primitive));
 
        int Nlaunches = ceil(n * n * Nobjects / float(maxRays));
 
        size_t prims_per_launch = fmin(Nobjects, maxPrims);
 
        uint s;
        // FIX: Use a copy of band_launch_flag for scattering so modifications don't affect primary launch indices
        std::vector<char> scatter_band_flags = band_launch_flag;
 
        for (s = 0; s < scatteringDepth; s++) {
            if (message_flag) {
                std::cout << "Performing scattering ray trace (iteration " << s + 1 << " of " << scatteringDepth << ")..." << std::flush;
            }
 
            int b = -1;
            int active_bands = 0;
            for (uint b_global = 0; b_global < Nbands_global; b_global++) {
 
                if (scatter_band_flags.at(b_global) == 0) {
                    continue;
                }
                b++;
 
                uint depth = radiation_bands.at(band_labels.at(b)).scatteringDepth;
                if (s + 1 > depth) {
                    if (message_flag) {
                        std::cout << "Skipping band " << band_labels.at(b) << " for scattering launch " << s + 1 << std::flush;
                    }
                    scatter_band_flags.at(b_global) = 0; // FIX: Modify copy, not original
                } else {
                    active_bands++;
                }
            }
 
            // Copy scatter buffers to radiation_out when needed
            // For s=0 with emission+direct: primary diffuse uploaded emission+scatter via params, but we need to copy scatter to avoid double-counting emission on next iteration
            // For s>0: scatter from previous iteration needs to be copied for next iteration
            if (s > 0 || (emissionenabled && rundirect)) {
                backend->copyScatterToRadiation();
            }
            backend->zeroScatterBuffers();
 
            // Extract radiation_out to ensure it's uploaded for scattering rays
            helios::RayTracingResults scatter_results;
            backend->getRadiationResults(scatter_results);
            std::vector<float> flux_top_scatter = scatter_results.radiation_out_top;
            std::vector<float> flux_bottom_scatter = scatter_results.radiation_out_bottom;
 
            for (uint launch = 0; launch < Nlaunches; launch++) {
 
                size_t prims_this_launch;
                if ((launch + 1) * prims_per_launch > Nprimitives) {
                    prims_this_launch = Nprimitives - launch * prims_per_launch;
                } else {
                    prims_this_launch = prims_per_launch;
                }
 
                // Build launch parameters for scattering diffuse rays
                helios::RayTracingLaunchParams params;
                params.launch_offset = launch * prims_per_launch;
                params.launch_count = prims_this_launch;
                params.rays_per_primitive = rays_per_primitive;
                params.random_seed = std::chrono::system_clock::now().time_since_epoch().count();
                params.current_band = 0;
                params.num_bands_global = Nbands_global;
                params.num_bands_launch = Nbands_launch;
                std::vector<bool> band_flags(scatter_band_flags.begin(), scatter_band_flags.end()); // FIX: Use scatter copy
                params.band_launch_flag = band_flags;
                params.scattering_iteration = s;
                params.max_scatters = scatteringDepth;
 
                // Pass diffuse radiation parameters to backend
                params.diffuse_flux = diffuse_flux;
                params.diffuse_extinction = diffuse_extinction;
                params.diffuse_dist_norm = diffuse_dist_norm;
                // Convert helios::vec3 to helios::vec3
                std::vector<helios::vec3> peak_dirs(diffuse_peak_dir.size());
                for (size_t i = 0; i < diffuse_peak_dir.size(); i++) {
                    peak_dirs[i] = helios::make_vec3(diffuse_peak_dir[i].x, diffuse_peak_dir[i].y, diffuse_peak_dir[i].z);
                }
                params.diffuse_peak_dir = peak_dirs;
 
                // Set radiation_out for scattering rays
                params.radiation_out_top = flux_top_scatter;
                params.radiation_out_bottom = flux_bottom_scatter;
 
                // Top surface launch
                params.launch_face = 1;
                backend->launchDiffuseRays(params);
 
 
                // Bottom surface launch
                params.launch_face = 0;
                backend->launchDiffuseRays(params);
            }
 
            // Accumulate camera scatter from this scattering iteration
            if (Ncameras > 0) {
                helios::RayTracingResults post_launch;
                backend->getRadiationResults(post_launch);
                for (size_t i = 0; i < post_launch.scatter_buff_top_cam.size(); i++) {
                    scatter_top_cam[i] += post_launch.scatter_buff_top_cam[i];
                    scatter_bottom_cam[i] += post_launch.scatter_buff_bottom_cam[i];
                }
                // Zero GPU camera scatter buffers to prevent double-counting on next iteration
                backend->zeroCameraScatterBuffers(Nbands_launch);
            }
 
            if (message_flag) {
                std::cout << "\r                                                                                                                           \r" << std::flush;
            }
        }
 
        if (message_flag) {
            std::cout << "Performing scattering ray trace...done." << std::endl;
        }
    }
 
    // **** CAMERA RAY TRACE **** //
    if (Ncameras > 0) {
 
        // Upload accumulated camera scatter to radiation_out for cameras to read
        // scatter_top_cam contains camera-weighted scattered energy from all ray types
        // Cameras read from radiation_out during hits, so we upload camera scatter there
        if (Ncameras > 0 && scatteringenabled) {
            backend->uploadRadiationOut(scatter_top_cam, scatter_bottom_cam);
        }
 
        // Setup solar disk rendering for cameras (enables lens flare effects)
        // Find sun-like sources (collimated or sun_sphere) and compute solar disk radiance
        vec3 sun_dir(0, 0, 1); // Default zenith
        std::vector<float> solar_radiances(Nbands_launch, 0.0f);
        bool has_sun_source = false;
 
        for (size_t s = 0; s < radiation_sources.size(); s++) {
            const RadiationSource &source = radiation_sources.at(s);
            if (source.source_type == RADIATION_SOURCE_TYPE_COLLIMATED || source.source_type == RADIATION_SOURCE_TYPE_SUN_SPHERE) {
                // Get sun direction from source position (normalized)
                sun_dir = source.source_position;
                sun_dir.normalize();
                has_sun_source = true;
 
                // Compute solar disk radiance for each band
                for (size_t b = 0; b < Nbands_launch; b++) {
                    float flux = getSourceFlux(s, band_labels.at(b));
 
                    if (source.source_type == RADIATION_SOURCE_TYPE_SUN_SPHERE) {
                        // For sun sphere: flux is surface exitance (σT⁴)
                        // Radiance as seen from Earth: L = F_surface / π
                        solar_radiances[b] = flux / M_PI;
                    } else {
                        // For collimated: flux is irradiance at Earth
                        // Solar solid angle: π × (4.63e-3)² ≈ 6.74×10⁻⁵ sr
                        const float solar_solid_angle = 6.74e-5f;
                        solar_radiances[b] = flux / solar_solid_angle;
                    }
                }
                break; // Use first sun-like source found
            }
        }
 
        if (scatteringenabled && (emissionenabled || diffuseenabled || rundirect)) {
            // re-set diffuse radiation fluxes (will be passed via launch params)
            if (diffuseenabled) {
                for (auto b = 0; b < Nbands_launch; b++) {
                    diffuse_flux.at(b) = getDiffuseFlux(band_labels.at(b));
                }
            }
 
            size_t n = ceil(sqrt(double(diffuseRayCount)));
 
            // Upload sky model parameters to backend (for camera rendering)
 
            if (!cameras.empty() && prague_params.size() == Nbands_launch) {
                // Get sky radiances for first camera (already computed above)
                std::vector<float> sky_for_backend = updateAtmosphericSkyModel(band_labels, cameras.begin()->second);
 
                // Upload to backend
                backend->updateSkyModel(prague_params, sky_for_backend, sun_dir, solar_radiances,
                                        has_sun_source ? 0.999989f : 0.0f // solar_disk_cos_angle
                );
            }
 
            uint cam = 0;
            for (auto &camera: cameras) {
 
 
                // Validate antialiasing samples don't exceed maximum
                if (camera.second.antialiasing_samples > maxRays) {
                    helios_runtime_error("ERROR (runBand): Camera '" + camera.second.label + "' antialiasing samples (" + std::to_string(camera.second.antialiasing_samples) + ") exceeds OptiX maximum launch size (" + std::to_string(maxRays) +
                                         "). Reduce antialiasing samples.");
                }
 
                // Compute tiling if needed
                std::vector<CameraTile> tiles = computeCameraTiles(camera.second, maxRays);
 
                if (message_flag && tiles.size() > 1) {
                    std::cout << "Camera '" << camera.second.label << "' requires " << tiles.size() << " tiles" << std::endl;
                }
 
                // Launch camera rays (tiled or full)
                for (size_t tile_idx = 0; tile_idx < tiles.size(); tile_idx++) {
                    const auto &tile = tiles[tile_idx];
 
                    // Build params for this tile
                    helios::RayTracingLaunchParams params = buildCameraLaunchParams(camera.second, cam, camera.second.antialiasing_samples, tile.resolution, tile.offset);
 
                    // Set band parameters (CRITICAL for materials!)
                    params.num_bands_launch = Nbands_launch;
                    params.num_bands_global = Nbands_global;
                    params.random_seed = std::chrono::system_clock::now().time_since_epoch().count();
                    std::vector<bool> band_flags(band_launch_flag.begin(), band_launch_flag.end());
                    params.band_launch_flag = band_flags;
 
                    // Progress message
                    if (message_flag) {
                        if (tiles.size() == 1) {
                            std::cout << "Performing scattering radiation camera ray trace for camera " << camera.second.label << "..." << std::flush;
                        } else {
                            std::cout << "Performing scattering radiation camera ray trace for camera " << camera.second.label << " (tile " << (tile_idx + 1) << " of " << tiles.size() << ")..." << std::flush;
                        }
                    }
 
                    // Launch through backend
                    backend->launchCameraRays(params);
 
                    if (message_flag) {
                        if (tiles.size() > 1) {
                            std::cout << "\r" << std::string(120, ' ') << "\r" << std::flush;
                        } else {
                            std::cout << "done." << std::endl;
                        }
                    }
                }
 
                if (message_flag && tiles.size() > 1) {
                    std::cout << "Performing scattering radiation camera ray trace for camera " << camera.second.label << "...done." << std::endl;
                }
 
                // Get results from backend
                std::vector<float> radiation_camera;
                std::vector<uint> dummy_labels;
                std::vector<float> dummy_depths;
                backend->getCameraResults(radiation_camera, dummy_labels, dummy_depths, cam, camera.second.resolution);
 
                // Process pixel data (KEEP EXISTING LOGIC)
                std::string camera_label = camera.second.label;
 
                for (auto b = 0; b < Nbands_launch; b++) {
 
                    camera.second.pixel_data[band_labels.at(b)].resize(camera.second.resolution.x * camera.second.resolution.y);
 
                    std::string data_label = "camera_" + camera_label + "_" + band_labels.at(b);
 
                    for (auto p = 0; p < camera.second.resolution.x * camera.second.resolution.y; p++) {
                        camera.second.pixel_data.at(band_labels.at(b)).at(p) = radiation_camera.at(p * Nbands_launch + b);
                    }
 
                    context->setGlobalData(data_label.c_str(), camera.second.pixel_data.at(band_labels.at(b)));
                }
 
                //--- Pixel Labeling Trace ---//
 
                // Compute tiling for pixel labeling (no antialiasing, 1 ray per pixel)
                RadiationCamera pixel_label_camera = camera.second;
                pixel_label_camera.antialiasing_samples = 1;
                std::vector<CameraTile> pixel_tiles = computeCameraTiles(pixel_label_camera, maxRays);
 
                // Zero camera pixel buffers once before tile loop
                backend->zeroCameraPixelBuffers(camera.second.resolution);
 
                // Launch pixel label rays (tiled or full)
                for (size_t tile_idx = 0; tile_idx < pixel_tiles.size(); tile_idx++) {
                    const auto &tile = pixel_tiles[tile_idx];
 
                    // Build params (reuse buildCameraLaunchParams, antialiasing=1)
                    helios::RayTracingLaunchParams params = buildCameraLaunchParams(pixel_label_camera, cam,
                                                                                    1, // No antialiasing for pixel labeling
                                                                                    tile.resolution, tile.offset);
 
                    // Progress message
                    if (message_flag) {
                        if (pixel_tiles.size() == 1) {
                            std::cout << "Performing camera pixel labeling ray trace for camera " << camera.second.label << "..." << std::flush;
                        } else {
                            std::cout << "Performing camera pixel labeling ray trace for camera " << camera.second.label << " (tile " << (tile_idx + 1) << " of " << pixel_tiles.size() << ")..." << std::flush;
                        }
                    }
 
                    // Launch through backend
                    backend->launchPixelLabelRays(params);
 
                    if (message_flag) {
                        if (pixel_tiles.size() > 1) {
                            std::cout << "\r" << std::string(120, ' ') << "\r" << std::flush;
                        } else {
                            std::cout << "done." << std::endl;
                        }
                    }
                }
 
                if (message_flag && pixel_tiles.size() > 1) {
                    std::cout << "Performing camera pixel labeling ray trace for camera " << camera.second.label << "...done." << std::endl;
                }
 
                // Get pixel label results
                std::vector<float> dummy_pixel_data;
                backend->getCameraResults(dummy_pixel_data, camera.second.pixel_label_UUID, camera.second.pixel_depth, cam, camera.second.resolution);
 
                // Convert IDs to actual UUIDs
                // Pixel labels contain position+1 (1-indexed), need to convert to UUIDs
                for (uint ID = 0; ID < camera.second.pixel_label_UUID.size(); ID++) {
                    if (camera.second.pixel_label_UUID.at(ID) > 0) {
                        uint position = camera.second.pixel_label_UUID.at(ID) - 1; // Convert to 0-indexed position
 
                        // Check if this is a bbox hit (position >= primitive_count)
                        if (position >= context_UUIDs.size()) {
                            // Bbox: calculate UUID directly (bbox_UUID = bbox_UUID_base + bbox_index)
                            uint bbox_index = position - context_UUIDs.size();
                            uint bbox_UUID = geometry_data.bbox_UUID_base + bbox_index;
                            camera.second.pixel_label_UUID.at(ID) = bbox_UUID + 1; // Store as 1-indexed
                        } else {
                            // Real primitive: look up UUID from context_UUIDs
                            camera.second.pixel_label_UUID.at(ID) = context_UUIDs.at(position) + 1;
                        }
                    }
                }
 
                // Store results in context (KEEP EXISTING LOGIC)
                std::string data_label = "camera_" + camera_label + "_pixel_UUID";
                context->setGlobalData(data_label.c_str(), camera.second.pixel_label_UUID);
 
                data_label = "camera_" + camera_label + "_pixel_depth";
                context->setGlobalData(data_label.c_str(), camera.second.pixel_depth);
 
                cam++;
            }
        } else {
            // if scattering is not enabled or all sources have zero flux, we still need to zero the camera buffers
            for (auto &camera: cameras) {
                for (auto b = 0; b < Nbands_launch; b++) {
                    camera.second.pixel_data[band_labels.at(b)].resize(camera.second.resolution.x * camera.second.resolution.y);
 
                    std::string data_label = "camera_" + camera.second.label + "_" + band_labels.at(b);
 
                    for (auto p = 0; p < camera.second.resolution.x * camera.second.resolution.y; p++) {
                        camera.second.pixel_data.at(band_labels.at(b)).at(p) = 0.f;
                    }
                    context->setGlobalData(data_label.c_str(), camera.second.pixel_data.at(band_labels.at(b)));
                }
            }
        }
    }
 
    // Apply camera exposure based on each camera's exposure setting
    for (auto &camera: cameras) {
        camera.second.applyCameraExposure(context);
    }
 
    // Apply camera white balance based on each camera's white_balance setting
    for (auto &camera: cameras) {
        camera.second.applyCameraWhiteBalance(context);
    }
 
    // deposit any energy that is left to make sure we satisfy conservation of energy
 
    // Extract ALL results from backend instead of old OptiX buffers
    helios::RayTracingResults results;
    backend->getRadiationResults(results);
 
    std::vector<float> radiation_flux_data = results.radiation_in;
 
    // Extract scatter buffer data from backend results
    TBS_top = results.scatter_buff_top;
    TBS_bottom = results.scatter_buff_bottom;
 
    std::vector<uint> UUIDs_context_all = context->getAllUUIDs();
 
    // Create indexer for result extraction
    RadiationBufferIndexer result_indexer(Nprimitives, Nbands_launch);
 
    for (auto b = 0; b < Nbands_launch; b++) {
 
        std::string prop = "radiation_flux_" + band_labels.at(b);
        std::vector<float> R(Nprimitives);
        for (size_t u = 0; u < Nprimitives; u++) {
            // Use BufferIndexer: [primitive][band]
            size_t ind = result_indexer(u, b);
            R.at(u) = radiation_flux_data.at(ind) + TBS_top.at(ind) + TBS_bottom.at(ind);
        }
        context->setPrimitiveData(context_UUIDs, prop.c_str(), R);
 
        if (UUIDs_context_all.size() != Nprimitives) {
            for (uint UUID: UUIDs_context_all) {
                if (context->doesPrimitiveExist(UUID) && !context->doesPrimitiveDataExist(UUID, prop.c_str())) {
                    context->setPrimitiveData(UUID, prop.c_str(), 0.f);
                }
            }
        }
    }
}
 
float RadiationModel::getSkyEnergy() {
 
    helios::RayTracingResults results;
    backend->getRadiationResults(results);
 
    float Rsky = 0.f;
    for (size_t i = 0; i < results.sky_energy.size(); i++) {
        Rsky += results.sky_energy.at(i);
    }
    return Rsky;
}
 
std::vector<float> RadiationModel::getTotalAbsorbedFlux() {
 
    std::vector<float> total_flux;
    total_flux.resize(context->getPrimitiveCount(), 0.f);
 
    for (const auto &band: radiation_bands) {
 
        std::string label = band.first;
 
        for (size_t u = 0; u < context_UUIDs.size(); u++) {
 
            uint p = context_UUIDs.at(u);
 
            std::string str = "radiation_flux_" + label;
 
            float R;
            context->getPrimitiveData(p, str.c_str(), R);
            total_flux.at(u) += R;
        }
    }
 
    return total_flux;
}
 
 
float RadiationModel::calculateGtheta(helios::Context *context, vec3 view_direction) {
 
    vec3 dir = view_direction;
    dir.normalize();
 
    float Gtheta = 0;
    float total_area = 0;
    for (std::size_t u = 0; u < primitiveID.size(); u++) {
 
        uint UUID = context_UUIDs.at(primitiveID.at(u));
 
        vec3 normal = context->getPrimitiveNormal(UUID);
        float area = context->getPrimitiveArea(UUID);
 
        Gtheta += fabsf(normal * dir) * area;
 
        total_area += area;
    }
 
    return Gtheta / total_area;
}
 
void RadiationModel::exportColorCorrectionMatrixXML(const std::string &file_path, const std::string &camera_label, const std::vector<std::vector<float>> &matrix, const std::string &source_image_path, const std::string &colorboard_type,
                                                    float average_delta_e) {
 
    std::ofstream file(file_path);
    if (!file.is_open()) {
        helios_runtime_error("ERROR (RadiationModel::exportColorCorrectionMatrixXML): Failed to open file for writing: " + file_path);
    }
 
    // Determine matrix type (3x3 or 4x3)
    std::string matrix_type = "3x3";
    if (matrix.size() == 4 || (matrix.size() >= 3 && matrix[0].size() == 4)) {
        matrix_type = "4x3";
    }
 
    // Write XML header with informative comments
    file << "<?xml version=\"1.0\" encoding=\"UTF-8\"?>" << std::endl;
    file << "<!-- Camera Color Correction Matrix -->" << std::endl;
    file << "<!-- Source Image: " << source_image_path << " -->" << std::endl;
    file << "<!-- Camera Label: " << camera_label << " -->" << std::endl;
    file << "<!-- Colorboard Type: " << colorboard_type << " -->" << std::endl;
    if (average_delta_e >= 0.0f) {
        file << "<!-- Average Delta E: " << std::fixed << std::setprecision(2) << average_delta_e << " -->" << std::endl;
    }
    file << "<!-- Matrix Type: " << matrix_type << " -->" << std::endl;
    file << "<!-- Generated: " << getCurrentDateTime() << " -->" << std::endl;
 
    // Write matrix data
    file << "<helios>" << std::endl;
    file << "  <ColorCorrectionMatrix camera_label=\"" << camera_label << "\" matrix_type=\"" << matrix_type << "\">" << std::endl;
 
    for (size_t i = 0; i < matrix.size(); i++) {
        file << "    <row>";
        for (size_t j = 0; j < matrix[i].size(); j++) {
            file << std::fixed << std::setprecision(6) << matrix[i][j];
            if (j < matrix[i].size() - 1) {
                file << " ";
            }
        }
        file << "</row>" << std::endl;
    }
 
    file << "  </ColorCorrectionMatrix>" << std::endl;
    file << "</helios>" << std::endl;
 
    file.close();
}
 
std::string RadiationModel::getCurrentDateTime() {
    auto now = std::time(nullptr);
    auto tm = *std::localtime(&now);
    std::stringstream ss;
    ss << std::put_time(&tm, "%Y-%m-%d %H:%M:%S");
    return ss.str();
}
 
std::vector<std::vector<float>> RadiationModel::loadColorCorrectionMatrixXML(const std::string &file_path, std::string &camera_label_out) {
 
    std::ifstream file(file_path);
    if (!file.is_open()) {
        helios_runtime_error("ERROR (RadiationModel::loadColorCorrectionMatrixXML): Failed to open file for reading: " + file_path);
    }
 
    std::vector<std::vector<float>> matrix;
    std::string line;
    bool in_matrix = false;
    std::string matrix_type = "";
 
    while (std::getline(file, line)) {
        // Remove leading/trailing whitespace
        line.erase(0, line.find_first_not_of(" \t"));
        line.erase(line.find_last_not_of(" \t") + 1);
 
        // Look for ColorCorrectionMatrix opening tag
        if (line.find("<ColorCorrectionMatrix") != std::string::npos) {
            in_matrix = true;
 
            // Extract camera_label attribute
            size_t camera_start = line.find("camera_label=\"");
            if (camera_start != std::string::npos) {
                camera_start += 14; // Length of "camera_label=\""
                size_t camera_end = line.find("\"", camera_start);
                if (camera_end != std::string::npos) {
                    camera_label_out = line.substr(camera_start, camera_end - camera_start);
                }
            }
 
            // Extract matrix_type attribute
            size_t type_start = line.find("matrix_type=\"");
            if (type_start != std::string::npos) {
                type_start += 13; // Length of "matrix_type=\""
                size_t type_end = line.find("\"", type_start);
                if (type_end != std::string::npos) {
                    matrix_type = line.substr(type_start, type_end - type_start);
                }
            }
            continue;
        }
 
        // Look for ColorCorrectionMatrix closing tag
        if (line.find("</ColorCorrectionMatrix>") != std::string::npos) {
            in_matrix = false;
            break;
        }
 
        // Parse row data
        if (in_matrix && line.find("<row>") != std::string::npos && line.find("</row>") != std::string::npos) {
            // Extract content between <row> and </row>
            size_t start = line.find("<row>") + 5;
            size_t end = line.find("</row>");
            std::string row_data = line.substr(start, end - start);
 
            // Parse float values from row
            std::vector<float> row;
            std::istringstream iss(row_data);
            float value;
            while (iss >> value) {
                row.push_back(value);
            }
 
            if (!row.empty()) {
                matrix.push_back(row);
            }
        }
    }
 
    file.close();
 
    // Validate loaded matrix
    if (matrix.empty()) {
        helios_runtime_error("ERROR (RadiationModel::loadColorCorrectionMatrixXML): No matrix data found in file: " + file_path);
    }
 
    if (matrix.size() != 3) {
        helios_runtime_error("ERROR (RadiationModel::loadColorCorrectionMatrixXML): Invalid matrix size. Expected 3 rows, found " + std::to_string(matrix.size()) + " rows in file: " + file_path);
    }
 
    // Validate matrix type consistency
    bool is_3x3 = (matrix[0].size() == 3 && matrix[1].size() == 3 && matrix[2].size() == 3);
    bool is_4x3 = (matrix[0].size() == 4 && matrix[1].size() == 4 && matrix[2].size() == 4);
 
    if (!is_3x3 && !is_4x3) {
        helios_runtime_error("ERROR (RadiationModel::loadColorCorrectionMatrixXML): Invalid matrix dimensions. All rows must have either 3 or 4 elements. File: " + file_path);
    }
 
    // Check matrix type attribute matches actual dimensions
    if (!matrix_type.empty()) {
        if ((matrix_type == "3x3" && !is_3x3) || (matrix_type == "4x3" && !is_4x3)) {
            helios_runtime_error("ERROR (RadiationModel::loadColorCorrectionMatrixXML): Matrix type attribute ('" + matrix_type + "') does not match actual matrix dimensions in file: " + file_path);
        }
    }
 
    return matrix;
}
 
std::string RadiationModel::autoCalibrateCameraImage(const std::string &camera_label, const std::string &red_band_label, const std::string &green_band_label, const std::string &blue_band_label, const std::string &output_file_path,
                                                     bool print_quality_report, ColorCorrectionAlgorithm algorithm, const std::string &ccm_export_file_path) {
 
    // Step 1: Validate camera exists and get pixel UUID data
    if (cameras.find(camera_label) == cameras.end()) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Camera '" + camera_label + "' does not exist. Make sure the camera was added to the radiation model.");
    }
 
    // Get camera pixel UUID data from global data (needed for segmentation)
    std::string pixel_UUID_label = "camera_" + camera_label + "_pixel_UUID";
    if (!context->doesGlobalDataExist(pixel_UUID_label.c_str())) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Camera pixel UUID data '" + pixel_UUID_label + "' does not exist for camera '" + camera_label + "'. Make sure the radiation model has been run.");
    }
 
    // Step 2: Detect all colorboard types using CameraCalibration helper
    CameraCalibration calibration(context);
    std::vector<std::string> colorboard_types;
    try {
        colorboard_types = calibration.detectColorBoardTypes();
    } catch (const std::exception &e) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Failed to detect colorboard types. " + std::string(e.what()));
    }
 
    // Step 3: Get reference Lab values for all detected colorboards
    std::vector<CameraCalibration::LabColor> reference_lab_values;
    std::vector<std::string> colorboard_type_per_patch; // Track which colorboard each patch belongs to
 
    for (const auto &colorboard_type: colorboard_types) {
        std::vector<CameraCalibration::LabColor> current_reference_values;
 
        if (colorboard_type == "DGK") {
            current_reference_values = calibration.getReferenceLab_DGK();
        } else if (colorboard_type == "Calibrite") {
            current_reference_values = calibration.getReferenceLab_Calibrite();
        } else if (colorboard_type == "SpyderCHECKR") {
            current_reference_values = calibration.getReferenceLab_SpyderCHECKR();
        } else {
            helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Unsupported colorboard type '" + colorboard_type + "'.");
        }
 
        // Add to combined list
        reference_lab_values.insert(reference_lab_values.end(), current_reference_values.begin(), current_reference_values.end());
 
        // Track which colorboard type each patch belongs to
        for (size_t i = 0; i < current_reference_values.size(); i++) {
            colorboard_type_per_patch.push_back(colorboard_type);
        }
    }
 
    // Step 4: Generate segmentation masks for all colorboard patches
    std::vector<uint> pixel_UUIDs;
    context->getGlobalData(pixel_UUID_label.c_str(), pixel_UUIDs);
    int2 camera_resolution = cameras.at(camera_label).resolution;
 
    // Create segmentation masks by finding pixels that belong to colorboard patches
    std::map<int, std::vector<std::vector<bool>>> patch_masks;
    int global_patch_idx = 0; // Global patch index across all colorboards
 
    for (const auto &colorboard_type: colorboard_types) {
        // Get the number of patches for this colorboard type
        int num_patches = 0;
        if (colorboard_type == "DGK") {
            num_patches = 18;
        } else if (colorboard_type == "Calibrite" || colorboard_type == "SpyderCHECKR") {
            num_patches = 24;
        }
 
        // Generate masks for each patch in this colorboard
        for (int local_patch_idx = 0; local_patch_idx < num_patches; local_patch_idx++) {
            std::vector<std::vector<bool>> mask(camera_resolution.y, std::vector<bool>(camera_resolution.x, false));
 
            // Find pixels that correspond to this colorboard patch
            for (int y = 0; y < camera_resolution.y; y++) {
                for (int x = 0; x < camera_resolution.x; x++) {
                    int pixel_index = y * camera_resolution.x + x;
                    uint pixel_UUID = pixel_UUIDs[pixel_index];
 
                    if (pixel_UUID > 0) { // Valid primitive
                        pixel_UUID--; // Convert from 1-based to 0-based indexing
 
                        // Check if this primitive belongs to this specific colorboard patch
                        std::string colorboard_data_label = "colorboard_" + colorboard_type;
                        if (context->doesPrimitiveDataExist(pixel_UUID, colorboard_data_label.c_str())) {
                            uint patch_id;
                            context->getPrimitiveData(pixel_UUID, colorboard_data_label.c_str(), patch_id);
                            // Patch indices are 0-based, compare directly
                            if ((int) patch_id == local_patch_idx) {
                                mask[y][x] = true;
                            }
                        }
                    }
                }
            }
 
            patch_masks[global_patch_idx] = mask;
            global_patch_idx++;
        }
    }
 
    // Step 5: Extract RGB colors from processed camera data (same source as writeCameraImage)
    // Use the same data source that writeCameraImage() uses: cameras.pixel_data
    std::vector<float> red_data, green_data, blue_data;
 
    // Check if bands exist in camera
    auto &camera_bands = cameras.at(camera_label).band_labels;
    if (std::find(camera_bands.begin(), camera_bands.end(), red_band_label) == camera_bands.end()) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Red band '" + red_band_label + "' not found in camera '" + camera_label + "'.");
    }
    if (std::find(camera_bands.begin(), camera_bands.end(), green_band_label) == camera_bands.end()) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Green band '" + green_band_label + "' not found in camera '" + camera_label + "'.");
    }
    if (std::find(camera_bands.begin(), camera_bands.end(), blue_band_label) == camera_bands.end()) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Blue band '" + blue_band_label + "' not found in camera '" + camera_label + "'.");
    }
 
    // Read processed camera data (same as writeCameraImage uses)
    red_data = cameras.at(camera_label).pixel_data.at(red_band_label);
    green_data = cameras.at(camera_label).pixel_data.at(green_band_label);
    blue_data = cameras.at(camera_label).pixel_data.at(blue_band_label);
 
    // Check data range and normalize if needed
    float max_r = *std::max_element(red_data.begin(), red_data.end());
    float max_g = *std::max_element(green_data.begin(), green_data.end());
    float max_b = *std::max_element(blue_data.begin(), blue_data.end());
 
    // Normalize camera data to [0,1] range if values are > 1
    float scale_factor = 1.0f;
    if (max_r > 1.0f || max_g > 1.0f || max_b > 1.0f) {
        scale_factor = 1.0f / std::max({max_r, max_g, max_b});
 
        for (size_t i = 0; i < red_data.size(); i++) {
            red_data[i] *= scale_factor;
            green_data[i] *= scale_factor;
            blue_data[i] *= scale_factor;
        }
    }
 
    std::vector<helios::vec3> measured_rgb_values;
    int visible_patches = 0;
 
    for (const auto &[patch_idx, mask]: patch_masks) {
        float sum_r = 0.0f, sum_g = 0.0f, sum_b = 0.0f;
        int pixel_count = 0;
 
        // Average RGB values over all pixels in this patch
        for (int y = 0; y < camera_resolution.y; y++) {
            for (int x = 0; x < camera_resolution.x; x++) {
                if (mask[y][x]) {
                    int pixel_index = y * camera_resolution.x + x;
                    sum_r += red_data[pixel_index];
                    sum_g += green_data[pixel_index];
                    sum_b += blue_data[pixel_index];
                    pixel_count++;
                }
            }
        }
 
        if (pixel_count > 10) { // Only consider patches with sufficient pixels
            helios::vec3 avg_rgb = make_vec3(sum_r / pixel_count, sum_g / pixel_count, sum_b / pixel_count);
            measured_rgb_values.push_back(avg_rgb);
 
            visible_patches++;
        } else {
            // Add placeholder for missing patch
            measured_rgb_values.push_back(make_vec3(0, 0, 0));
        }
    }
 
    // Convert measured RGB to Lab and calculate correction matrix
    std::vector<CameraCalibration::LabColor> measured_lab_values;
    for (const auto &rgb: measured_rgb_values) {
        if (rgb.magnitude() > 0) { // Only process non-zero values
            measured_lab_values.push_back(calibration.rgbToLab(rgb));
        }
    }
 
    // Calculate color correction matrix based on selected algorithm
    std::vector<std::vector<float>> correction_matrix = {{1.0f, 0.0f, 0.0f}, {0.0f, 1.0f, 0.0f}, {0.0f, 0.0f, 1.0f}};
 
    // Report which algorithm is being used
    std::string algorithm_name;
    switch (algorithm) {
        case ColorCorrectionAlgorithm::DIAGONAL_ONLY:
            algorithm_name = "Diagonal scaling (white balance only)";
            break;
        case ColorCorrectionAlgorithm::MATRIX_3X3_AUTO:
            algorithm_name = "3x3 matrix with auto-fallback to diagonal";
            break;
        case ColorCorrectionAlgorithm::MATRIX_3X3_FORCE:
            algorithm_name = "3x3 matrix (forced)";
            break;
    }
 
    if (measured_lab_values.size() >= 6 && reference_lab_values.size() >= 6) {
        // Convert reference Lab back to RGB for matrix fitting
        std::vector<helios::vec3> target_rgb;
 
        for (size_t i = 0; i < reference_lab_values.size(); i++) {
            CameraCalibration::LabColor ref_lab = reference_lab_values[i];
            helios::vec3 ref_rgb = calibration.labToRgb(ref_lab);
            target_rgb.push_back(ref_rgb);
        }
 
        // Build matrices for least squares: M * Measured = Target
        // We need to solve for M where M is 3x3 matrix
        // This becomes: M = Target * Measured^T * (Measured * Measured^T)^(-1)
 
        // Collect valid patches for matrix calculation
        std::vector<helios::vec3> valid_measured, valid_target;
        std::vector<float> patch_weights;
 
        for (size_t i = 0; i < std::min(measured_rgb_values.size(), target_rgb.size()); i++) {
            if (measured_rgb_values[i].magnitude() > 0.01f) {
                valid_measured.push_back(measured_rgb_values[i]);
                valid_target.push_back(target_rgb[i]);
 
                // Perceptually-weighted patch selection based on colour-science best practices
                float weight = 1.0f;
 
                // Neutral patches (white to black series) - highest priority for white balance
                if (i >= 18 && i <= 23) {
                    // White and light grays get highest weight (most visually important)
                    if (i == 18)
                        weight = 5.0f; // White patch - critical for white balance
                    else if (i == 19)
                        weight = 4.0f; // Light gray - very important for tone mapping
                    else
                        weight = 3.0f; // Darker grays - important for contrast
                }
                // Primary color patches - important for color accuracy
                else if (i == 14 || i == 13 || i == 12)
                    weight = 3.0f; // Red, Green, Blue primaries
 
                // Skin tone approximates - patches that represent common skin tones
                else if (i == 3 || i == 10)
                    weight = 2.5f; // Foliage and Yellow Green (skin-like hues)
 
                // Well-lit patches get higher weight based on luminance
                helios::vec3 measured_rgb = measured_rgb_values[i];
                float luminance = 0.299f * measured_rgb.x + 0.587f * measured_rgb.y + 0.114f * measured_rgb.z;
 
                // Boost weight for brighter patches (they're more visually prominent)
                if (luminance > 0.6f)
                    weight *= 1.5f;
                else if (luminance < 0.2f)
                    weight *= 0.7f; // Reduce weight for very dark patches
 
                // Additional quality checks for measured RGB values
                // Reduce weight for patches that seem poorly measured (extreme values)
                if (measured_rgb.magnitude() > 1.2f || measured_rgb.magnitude() < 0.1f) {
                    weight *= 0.5f; // Reduce weight for suspicious measurements
                }
 
                patch_weights.push_back(weight);
            }
        }
 
        if (valid_measured.size() >= 6 && algorithm != ColorCorrectionAlgorithm::DIAGONAL_ONLY) {
 
            // Compute robustly regularized weighted least squares solution
            // Use adaptive regularization to prevent extreme matrix coefficients
            bool matrix_valid = true;
 
            // Try moderate regularization values - avoid extreme regularization that creates bad matrices
            std::vector<float> lambda_values = {0.01f, 0.05f, 0.1f, 0.15f, 0.2f};
            int lambda_attempt = 0;
 
            while (lambda_attempt < lambda_values.size()) {
                float lambda = lambda_values[lambda_attempt];
                matrix_valid = true;
 
                for (int row = 0; row < 3; row++) {
                    // Build weighted normal equations with adaptive regularization
                    float ATA[3][3] = {{0}}; // A^T * W * A + λI
                    float ATb[3] = {0}; // A^T * W * b
 
                    for (size_t i = 0; i < valid_measured.size(); i++) {
                        float weight = patch_weights[i];
                        helios::vec3 m = valid_measured[i]; // measured RGB
                        float target_val = (row == 0) ? valid_target[i].x : (row == 1) ? valid_target[i].y : valid_target[i].z;
 
                        // Update normal equations
                        ATA[0][0] += weight * m.x * m.x;
                        ATA[0][1] += weight * m.x * m.y;
                        ATA[0][2] += weight * m.x * m.z;
                        ATA[1][0] += weight * m.y * m.x;
                        ATA[1][1] += weight * m.y * m.y;
                        ATA[1][2] += weight * m.y * m.z;
                        ATA[2][0] += weight * m.z * m.x;
                        ATA[2][1] += weight * m.z * m.y;
                        ATA[2][2] += weight * m.z * m.z;
 
                        ATb[0] += weight * m.x * target_val;
                        ATb[1] += weight * m.y * target_val;
                        ATb[2] += weight * m.z * target_val;
                    }
 
                    // Add color-preserving regularization
                    // Stronger regularization on diagonal (preserves primary colors)
                    // Weaker regularization on off-diagonal (allows some color mixing)
                    float diag_reg = lambda * 2.0f; // Stronger on diagonal
                    float offdiag_reg = lambda * 0.5f; // Weaker on off-diagonal
 
                    ATA[0][0] += diag_reg; // Red preservation
                    ATA[1][1] += diag_reg; // Green preservation
                    ATA[2][2] += diag_reg; // Blue preservation
 
                    // Light off-diagonal regularization to prevent extreme color mixing
                    ATA[0][1] += offdiag_reg;
                    ATA[1][0] += offdiag_reg;
                    ATA[0][2] += offdiag_reg;
                    ATA[2][0] += offdiag_reg;
                    ATA[1][2] += offdiag_reg;
                    ATA[2][1] += offdiag_reg;
 
                    // Solve regularized 3x3 system using Cramer's rule
                    float det = ATA[0][0] * (ATA[1][1] * ATA[2][2] - ATA[1][2] * ATA[2][1]) - ATA[0][1] * (ATA[1][0] * ATA[2][2] - ATA[1][2] * ATA[2][0]) + ATA[0][2] * (ATA[1][0] * ATA[2][1] - ATA[1][1] * ATA[2][0]);
 
                    if (fabs(det) < 1e-3f) {
                        if (algorithm != ColorCorrectionAlgorithm::MATRIX_3X3_FORCE) {
                            matrix_valid = false;
                            break;
                        }
                    }
 
                    float inv_det = 1.0f / det;
                    correction_matrix[row][0] = inv_det * (ATb[0] * (ATA[1][1] * ATA[2][2] - ATA[1][2] * ATA[2][1]) - ATb[1] * (ATA[0][1] * ATA[2][2] - ATA[0][2] * ATA[2][1]) + ATb[2] * (ATA[0][1] * ATA[1][2] - ATA[0][2] * ATA[1][1]));
 
                    correction_matrix[row][1] = inv_det * (ATb[1] * (ATA[0][0] * ATA[2][2] - ATA[0][2] * ATA[2][0]) - ATb[0] * (ATA[1][0] * ATA[2][2] - ATA[1][2] * ATA[2][0]) + ATb[2] * (ATA[1][0] * ATA[0][2] - ATA[1][2] * ATA[0][0]));
 
                    correction_matrix[row][2] = inv_det * (ATb[2] * (ATA[0][0] * ATA[1][1] - ATA[0][1] * ATA[1][0]) - ATb[0] * (ATA[1][0] * ATA[2][1] - ATA[1][1] * ATA[2][0]) + ATb[1] * (ATA[0][0] * ATA[2][1] - ATA[0][1] * ATA[2][0]));
                }
 
                // Validate computed matrix elements are reasonable
                if (matrix_valid) {
                    bool elements_reasonable = true;
                    for (int i = 0; i < 3; i++) {
                        for (int j = 0; j < 3; j++) {
                            if (fabs(correction_matrix[i][j]) > 5.0f) {
                                if (algorithm != ColorCorrectionAlgorithm::MATRIX_3X3_FORCE) {
                                    elements_reasonable = false;
                                    break;
                                }
                            }
                        }
                        if (!elements_reasonable)
                            break;
                    }
                    matrix_valid = elements_reasonable;
                }
 
                if (matrix_valid) {
                    break; // Success!
                } else {
                    lambda_attempt++;
                }
            }
 
            if (!matrix_valid) {
 
                // Enhanced perceptually-weighted diagonal correction
                // Calculate weighted averages for each color channel
                float total_weight = 0.0f;
                helios::vec3 weighted_correction = make_vec3(0, 0, 0);
 
                for (size_t i = 0; i < valid_measured.size(); i++) {
                    float weight = patch_weights[i];
                    helios::vec3 measured = valid_measured[i];
                    helios::vec3 target = valid_target[i];
 
                    // Calculate per-channel correction factors
                    if (measured.x > 0.01f && measured.y > 0.01f && measured.z > 0.01f) {
                        helios::vec3 channel_correction = make_vec3(target.x / measured.x, target.y / measured.y, target.z / measured.z);
 
                        weighted_correction.x += weight * channel_correction.x;
                        weighted_correction.y += weight * channel_correction.y;
                        weighted_correction.z += weight * channel_correction.z;
                        total_weight += weight;
                    }
                }
 
                if (total_weight > 0.1f) {
                    // Apply weighted average correction factors
                    correction_matrix[0][0] = weighted_correction.x / total_weight;
                    correction_matrix[1][1] = weighted_correction.y / total_weight;
                    correction_matrix[2][2] = weighted_correction.z / total_weight;
 
                    // Apply conservative limits
                    correction_matrix[0][0] = std::max(0.5f, std::min(2.0f, correction_matrix[0][0]));
                    correction_matrix[1][1] = std::max(0.5f, std::min(2.0f, correction_matrix[1][1]));
                    correction_matrix[2][2] = std::max(0.5f, std::min(2.0f, correction_matrix[2][2]));
                }
            }
 
            if (!matrix_valid || algorithm == ColorCorrectionAlgorithm::DIAGONAL_ONLY) {
                // Use diagonal correction using white patch
                correction_matrix = {{1.0f, 0.0f, 0.0f}, {0.0f, 1.0f, 0.0f}, {0.0f, 0.0f, 1.0f}};
 
                // Enhanced perceptually-weighted diagonal correction using all valid patches
                // This is more robust than using just the white patch
                if (valid_measured.size() > 0 && patch_weights.size() == valid_measured.size()) {
                    float total_weight = 0.0f;
                    helios::vec3 weighted_measured_avg = make_vec3(0, 0, 0);
                    helios::vec3 weighted_target_avg = make_vec3(0, 0, 0);
 
                    // Compute weighted averages using perceptual patch weights
                    for (size_t i = 0; i < valid_measured.size(); i++) {
                        float weight = patch_weights[i];
                        weighted_measured_avg = weighted_measured_avg + weight * valid_measured[i];
                        weighted_target_avg = weighted_target_avg + weight * valid_target[i];
                        total_weight += weight;
                    }
 
                    if (total_weight > 0) {
                        weighted_measured_avg = weighted_measured_avg / total_weight;
                        weighted_target_avg = weighted_target_avg / total_weight;
 
                        if (weighted_measured_avg.x > 0.05f && weighted_measured_avg.y > 0.05f && weighted_measured_avg.z > 0.05f) {
                            correction_matrix[0][0] = weighted_target_avg.x / weighted_measured_avg.x;
                            correction_matrix[1][1] = weighted_target_avg.y / weighted_measured_avg.y;
                            correction_matrix[2][2] = weighted_target_avg.z / weighted_measured_avg.z;
 
                            // Apply conservative limits for stability
                            correction_matrix[0][0] = std::max(0.5f, std::min(2.0f, correction_matrix[0][0]));
                            correction_matrix[1][1] = std::max(0.5f, std::min(2.0f, correction_matrix[1][1]));
                            correction_matrix[2][2] = std::max(0.5f, std::min(2.0f, correction_matrix[2][2]));
 
                        } else {
                            // Fallback to original white-patch method
                            size_t white_idx = 18;
                            if (white_idx < measured_rgb_values.size() && white_idx < target_rgb.size()) {
                                helios::vec3 measured_white = measured_rgb_values[white_idx];
                                helios::vec3 target_white = target_rgb[white_idx];
                                if (measured_white.x > 0.05f && measured_white.y > 0.05f && measured_white.z > 0.05f) {
                                    correction_matrix[0][0] = std::max(0.5f, std::min(2.0f, target_white.x / measured_white.x));
                                    correction_matrix[1][1] = std::max(0.5f, std::min(2.0f, target_white.y / measured_white.y));
                                    correction_matrix[2][2] = std::max(0.5f, std::min(2.0f, target_white.z / measured_white.z));
                                }
                            }
                        }
                    }
                } else {
                    // Original simple approach as ultimate fallback
                    size_t white_idx = 18;
                    if (white_idx < measured_rgb_values.size() && white_idx < target_rgb.size()) {
                        helios::vec3 measured_white = measured_rgb_values[white_idx];
                        helios::vec3 target_white = target_rgb[white_idx];
                        if (measured_white.x > 0.05f && measured_white.y > 0.05f && measured_white.z > 0.05f) {
                            correction_matrix[0][0] = std::max(0.5f, std::min(2.0f, target_white.x / measured_white.x));
                            correction_matrix[1][1] = std::max(0.5f, std::min(2.0f, target_white.y / measured_white.y));
                            correction_matrix[2][2] = std::max(0.5f, std::min(2.0f, target_white.z / measured_white.z));
                        }
                    }
                }
            }
        } else if (algorithm == ColorCorrectionAlgorithm::DIAGONAL_ONLY) {
            // Apply diagonal correction using white patch
            size_t white_idx = 18;
            if (white_idx < measured_rgb_values.size() && white_idx < target_rgb.size() && measured_rgb_values[white_idx].magnitude() > 0) {
 
                helios::vec3 measured_white = measured_rgb_values[white_idx];
                helios::vec3 target_white = target_rgb[white_idx];
 
                if (measured_white.x > 0.05f && measured_white.y > 0.05f && measured_white.z > 0.05f) {
                    correction_matrix[0][0] = target_white.x / measured_white.x;
                    correction_matrix[1][1] = target_white.y / measured_white.y;
                    correction_matrix[2][2] = target_white.z / measured_white.z;
 
                    // Apply limits
                    correction_matrix[0][0] = std::max(0.5f, std::min(2.0f, correction_matrix[0][0]));
                    correction_matrix[1][1] = std::max(0.5f, std::min(2.0f, correction_matrix[1][1]));
                    correction_matrix[2][2] = std::max(0.5f, std::min(2.0f, correction_matrix[2][2]));
                }
            }
        } else {
            std::cout << "Insufficient valid patches (" << valid_measured.size() << " available), using identity matrix" << std::endl;
        }
    } else if (algorithm == ColorCorrectionAlgorithm::DIAGONAL_ONLY && measured_lab_values.size() > 0) {
        // Apply diagonal correction using white patch
        size_t white_idx = 18;
        if (white_idx < measured_rgb_values.size() && white_idx < reference_lab_values.size() && measured_rgb_values[white_idx].magnitude() > 0) {
 
            CameraCalibration::LabColor ref_lab = reference_lab_values[white_idx];
            helios::vec3 target_white = calibration.labToRgb(ref_lab);
            helios::vec3 measured_white = measured_rgb_values[white_idx];
 
            if (measured_white.x > 0.05f && measured_white.y > 0.05f && measured_white.z > 0.05f) {
                correction_matrix[0][0] = target_white.x / measured_white.x;
                correction_matrix[1][1] = target_white.y / measured_white.y;
                correction_matrix[2][2] = target_white.z / measured_white.z;
 
                // Apply limits
                correction_matrix[0][0] = std::max(0.5f, std::min(2.0f, correction_matrix[0][0]));
                correction_matrix[1][1] = std::max(0.5f, std::min(2.0f, correction_matrix[1][1]));
                correction_matrix[2][2] = std::max(0.5f, std::min(2.0f, correction_matrix[2][2]));
            }
        }
    } else {
        std::cout << "Insufficient patches for correction (" << measured_lab_values.size() << " available), using identity matrix" << std::endl;
    }
 
    // Generate quality of fit report if requested
    if (print_quality_report) {
        std::cout << "\n========== COLOR CALIBRATION QUALITY REPORT ==========" << std::endl;
        std::cout << "Colorboard types: ";
        for (size_t i = 0; i < colorboard_types.size(); i++) {
            std::cout << colorboard_types[i];
            if (i < colorboard_types.size() - 1) {
                std::cout << ", ";
            }
        }
        std::cout << std::endl;
        std::cout << "Number of patches analyzed: " << visible_patches << std::endl;
        std::cout << "Algorithm used: " << algorithm_name << std::endl;
 
        // Display matrix conditioning information
        bool is_diagonal_only = true;
        for (int i = 0; i < 3; i++) {
            for (int j = 0; j < 3; j++) {
                if (i != j && fabs(correction_matrix[i][j]) > 1e-6f) {
                    is_diagonal_only = false;
                    break;
                }
            }
            if (!is_diagonal_only)
                break;
        }
 
        if (is_diagonal_only) {
            std::cout << "Color correction factors applied: R=" << correction_matrix[0][0] << ", G=" << correction_matrix[1][1] << ", B=" << correction_matrix[2][2] << std::endl;
            std::cout << "Matrix type: Diagonal (white balance only)" << std::endl;
        } else {
            std::cout << "Full 3x3 color correction matrix applied:" << std::endl;
            for (int i = 0; i < 3; i++) {
                std::cout << "[" << std::fixed << std::setprecision(4);
                for (int j = 0; j < 3; j++) {
                    std::cout << std::setw(8) << correction_matrix[i][j];
                    if (j < 2)
                        std::cout << " ";
                }
                std::cout << "]" << std::endl;
            }
            std::cout << "Matrix type: Full 3x3 (corrects color casts and chromatic errors)" << std::endl;
 
            // Calculate matrix determinant for conditioning info
            float det = correction_matrix[0][0] * (correction_matrix[1][1] * correction_matrix[2][2] - correction_matrix[1][2] * correction_matrix[2][1]) -
                        correction_matrix[0][1] * (correction_matrix[1][0] * correction_matrix[2][2] - correction_matrix[1][2] * correction_matrix[2][0]) +
                        correction_matrix[0][2] * (correction_matrix[1][0] * correction_matrix[2][1] - correction_matrix[1][1] * correction_matrix[2][0]);
 
            std::cout << "Matrix determinant: " << std::scientific << std::setprecision(3) << det << std::endl;
            if (fabs(det) > 0.1f) {
                std::cout << "Matrix conditioning: Good (well-conditioned)" << std::endl;
            } else if (fabs(det) > 0.01f) {
                std::cout << "Matrix conditioning: Fair (moderately conditioned)" << std::endl;
            } else {
                std::cout << "Matrix conditioning: Poor (ill-conditioned)" << std::endl;
            }
            std::cout << std::fixed; // Reset formatting
        }
 
        // Calculate and display quality metrics for each patch
        double total_delta_e = 0.0;
        int valid_patches = 0;
 
        std::cout << "\nPer-patch analysis (after correction):" << std::endl;
        std::cout << "Patch | Corrected RGB       | Reference RGB       | Delta E " << std::endl;
        std::cout << "------|--------------------|--------------------|---------" << std::endl;
 
        for (size_t i = 0; i < std::min(measured_rgb_values.size(), reference_lab_values.size()); i++) {
            if (measured_rgb_values[i].magnitude() > 0) {
                // Apply color correction to measured RGB values (with optional affine terms)
                helios::vec3 measured_rgb = measured_rgb_values[i];
                float corrected_r = correction_matrix[0][0] * measured_rgb.x + correction_matrix[0][1] * measured_rgb.y + correction_matrix[0][2] * measured_rgb.z;
                float corrected_g = correction_matrix[1][0] * measured_rgb.x + correction_matrix[1][1] * measured_rgb.y + correction_matrix[1][2] * measured_rgb.z;
                float corrected_b = correction_matrix[2][0] * measured_rgb.x + correction_matrix[2][1] * measured_rgb.y + correction_matrix[2][2] * measured_rgb.z;
 
 
                // Clamp corrected values to [0,1] - DISABLED FOR TESTING
                // corrected_r = std::max(0.0f, std::min(1.0f, corrected_r));
                // corrected_g = std::max(0.0f, std::min(1.0f, corrected_g));
                // corrected_b = std::max(0.0f, std::min(1.0f, corrected_b));
 
                helios::vec3 corrected_rgb = make_vec3(corrected_r, corrected_g, corrected_b);
 
                // Convert corrected RGB to Lab
                CameraCalibration::LabColor corrected_lab = calibration.rgbToLab(corrected_rgb);
                CameraCalibration::LabColor reference_lab = reference_lab_values[i];
 
                // Calculate Delta E between corrected and reference
                // Use ΔE2000 for better perceptual color difference assessment
                double delta_E = calibration.deltaE2000(corrected_lab, reference_lab);
 
                std::cout << std::setw(5) << i << " | " << std::fixed << std::setprecision(3) << "(" << std::setw(5) << corrected_rgb.x << "," << std::setw(5) << corrected_rgb.y << "," << std::setw(5) << corrected_rgb.z << ") | ";
 
                helios::vec3 ref_rgb = calibration.labToRgb(reference_lab);
                std::cout << "(" << std::setw(5) << ref_rgb.x << "," << std::setw(5) << ref_rgb.y << "," << std::setw(5) << ref_rgb.z << ") | " << std::setw(7) << delta_E << std::endl;
 
                total_delta_e += delta_E;
                valid_patches++;
            }
        }
 
        // Overall statistics
        double mean_delta_e = total_delta_e / valid_patches;
        std::cout << "\n========== OVERALL CALIBRATION QUALITY ==========" << std::endl;
        std::cout << "Mean Delta E: " << std::fixed << std::setprecision(2) << mean_delta_e << std::endl;
 
        std::cout << "======================================================\n" << std::endl;
    }
 
    // Step 7: Apply correction to entire image with same pixel ordering as writeCameraImage
    std::vector<helios::RGBcolor> corrected_pixels;
    corrected_pixels.resize(red_data.size());
 
    // Apply correction using the same pixel transformation as writeCameraImage uses
    for (int j = 0; j < camera_resolution.y; j++) {
        for (int i = 0; i < camera_resolution.x; i++) {
            // Get pixel from source data (no flip)
            int source_index = j * camera_resolution.x + i;
            float r = red_data[source_index];
            float g = green_data[source_index];
            float b = blue_data[source_index];
 
            // Apply correction matrix (with optional affine terms)
            float corrected_r = correction_matrix[0][0] * r + correction_matrix[0][1] * g + correction_matrix[0][2] * b;
            float corrected_g = correction_matrix[1][0] * r + correction_matrix[1][1] * g + correction_matrix[1][2] * b;
            float corrected_b = correction_matrix[2][0] * r + correction_matrix[2][1] * g + correction_matrix[2][2] * b;
 
 
            // Clamp values to [0,1] - DISABLED FOR TESTING
            // corrected_r = std::max(0.0f, std::min(1.0f, corrected_r));
            // corrected_g = std::max(0.0f, std::min(1.0f, corrected_g));
            // corrected_b = std::max(0.0f, std::min(1.0f, corrected_b));
 
            // Apply same coordinate transformation as writeCameraImage
            uint ii = camera_resolution.x - i - 1; // Horizontal flip
            uint jj = camera_resolution.y - j - 1; // Vertical flip
            uint dest_index = jj * camera_resolution.x + ii;
 
            corrected_pixels[dest_index] = make_RGBcolor(corrected_r, corrected_g, corrected_b);
        }
    }
 
    // Step 8: Write corrected image using writeJPEG
    std::string output_path = output_file_path;
    if (output_path.empty()) {
        output_path = "auto_calibrated_" + camera_label + ".jpg";
    }
 
    try {
        helios::writeJPEG(output_path, camera_resolution.x, camera_resolution.y, corrected_pixels);
        std::cout << "Wrote corrected image to: " << output_path << std::endl;
    } catch (const std::exception &e) {
        helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Failed to write corrected image. " + std::string(e.what()));
    }
 
    // Export CCM to XML file if requested
    if (!ccm_export_file_path.empty()) {
        try {
            // Calculate quality metrics for export (even if not printed)
            double total_delta_e = 0.0;
            int valid_patches = 0;
 
            for (size_t i = 0; i < std::min(measured_rgb_values.size(), reference_lab_values.size()); i++) {
                if (measured_rgb_values[i].magnitude() > 0) {
                    // Apply color correction to measured RGB values
                    helios::vec3 measured_rgb = measured_rgb_values[i];
                    float corrected_r = correction_matrix[0][0] * measured_rgb.x + correction_matrix[0][1] * measured_rgb.y + correction_matrix[0][2] * measured_rgb.z;
                    float corrected_g = correction_matrix[1][0] * measured_rgb.x + correction_matrix[1][1] * measured_rgb.y + correction_matrix[1][2] * measured_rgb.z;
                    float corrected_b = correction_matrix[2][0] * measured_rgb.x + correction_matrix[2][1] * measured_rgb.y + correction_matrix[2][2] * measured_rgb.z;
 
                    helios::vec3 corrected_rgb = make_vec3(corrected_r, corrected_g, corrected_b);
 
                    // Convert corrected RGB to Lab
                    CameraCalibration::LabColor corrected_lab = calibration.rgbToLab(corrected_rgb);
                    CameraCalibration::LabColor reference_lab = reference_lab_values[i];
 
                    // Calculate Delta E between corrected and reference
                    double delta_E = calibration.deltaE2000(corrected_lab, reference_lab);
                    total_delta_e += delta_E;
                    valid_patches++;
                }
            }
 
            double mean_delta_e = (valid_patches > 0) ? (total_delta_e / valid_patches) : -1.0;
 
            // Export CCM to XML file
            // Concatenate all colorboard types into a single string
            std::string colorboard_types_str;
            for (size_t i = 0; i < colorboard_types.size(); i++) {
                colorboard_types_str += colorboard_types[i];
                if (i < colorboard_types.size() - 1) {
                    colorboard_types_str += ", ";
                }
            }
            exportColorCorrectionMatrixXML(ccm_export_file_path, camera_label, correction_matrix, output_path, colorboard_types_str, (float) mean_delta_e);
 
            std::cout << "Exported color correction matrix to: " << ccm_export_file_path << std::endl;
        } catch (const std::exception &e) {
            helios_runtime_error("ERROR (RadiationModel::autoCalibrateCameraImage): Failed to export CCM to XML. " + std::string(e.what()));
        }
    }
 
    return output_path;
}
 
void RadiationModel::applyCameraColorCorrectionMatrix(const std::string &camera_label, const std::string &red_band_label, const std::string &green_band_label, const std::string &blue_band_label, const std::string &ccm_file_path) {
 
    // Step 1: Validate camera exists
    if (cameras.find(camera_label) == cameras.end()) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Camera '" + camera_label + "' does not exist. Make sure the camera was added to the radiation model.");
    }
 
    // Step 2: Validate band labels exist in camera
    auto &camera_bands = cameras.at(camera_label).band_labels;
    if (std::find(camera_bands.begin(), camera_bands.end(), red_band_label) == camera_bands.end()) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Red band '" + red_band_label + "' not found in camera '" + camera_label + "'.");
    }
    if (std::find(camera_bands.begin(), camera_bands.end(), green_band_label) == camera_bands.end()) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Green band '" + green_band_label + "' not found in camera '" + camera_label + "'.");
    }
    if (std::find(camera_bands.begin(), camera_bands.end(), blue_band_label) == camera_bands.end()) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Blue band '" + blue_band_label + "' not found in camera '" + camera_label + "'.");
    }
 
    // Step 3: Load color correction matrix from XML file
    std::string loaded_camera_label;
    std::vector<std::vector<float>> correction_matrix;
    try {
        correction_matrix = loadColorCorrectionMatrixXML(ccm_file_path, loaded_camera_label);
    } catch (const std::exception &e) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Failed to load CCM from XML file. " + std::string(e.what()));
    }
 
    // Step 4: Validate matrix dimensions (should be 3x3 or 4x3)
    if (correction_matrix.size() != 3) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Invalid matrix dimensions. Expected 3x3 or 4x3 matrix, got " + std::to_string(correction_matrix.size()) + " rows.");
    }
 
    bool is_3x3 = (correction_matrix[0].size() == 3);
    bool is_4x3 = (correction_matrix[0].size() == 4);
 
    if (!is_3x3 && !is_4x3) {
        helios_runtime_error("ERROR (RadiationModel::applyCameraColorCorrectionMatrix): Invalid matrix dimensions. Expected 3x3 or 4x3 matrix, got " + std::to_string(correction_matrix.size()) + "x" + std::to_string(correction_matrix[0].size()) +
                             " matrix.");
    }
 
    // Step 5: Get camera data (same approach as applyImageProcessingPipeline)
    std::vector<float> &red_data = cameras.at(camera_label).pixel_data.at(red_band_label);
    std::vector<float> &green_data = cameras.at(camera_label).pixel_data.at(green_band_label);
    std::vector<float> &blue_data = cameras.at(camera_label).pixel_data.at(blue_band_label);
 
    int2 camera_resolution = cameras.at(camera_label).resolution;
    size_t pixel_count = red_data.size();
 
    // Step 6: Apply color correction matrix to all pixels in-place
    for (size_t i = 0; i < pixel_count; i++) {
        float r = red_data[i];
        float g = green_data[i];
        float b = blue_data[i];
 
        // Apply color correction matrix (3x3 or 4x3)
        if (is_3x3) {
            // Standard 3x3 matrix transformation
            red_data[i] = correction_matrix[0][0] * r + correction_matrix[0][1] * g + correction_matrix[0][2] * b;
            green_data[i] = correction_matrix[1][0] * r + correction_matrix[1][1] * g + correction_matrix[1][2] * b;
            blue_data[i] = correction_matrix[2][0] * r + correction_matrix[2][1] * g + correction_matrix[2][2] * b;
        } else {
            // 4x3 matrix transformation with affine offset
            red_data[i] = correction_matrix[0][0] * r + correction_matrix[0][1] * g + correction_matrix[0][2] * b + correction_matrix[0][3];
            green_data[i] = correction_matrix[1][0] * r + correction_matrix[1][1] * g + correction_matrix[1][2] * b + correction_matrix[1][3];
            blue_data[i] = correction_matrix[2][0] * r + correction_matrix[2][1] * g + correction_matrix[2][2] * b + correction_matrix[2][3];
        }
    }
 
    if (message_flag) {
        std::cout << "Applied color correction matrix from '" << ccm_file_path << "' to camera '" << camera_label << "'" << std::endl;
        std::cout << "Matrix type: " << (is_3x3 ? "3x3" : "4x3") << ", processed " << pixel_count << " pixels" << std::endl;
    }
}
 
std::vector<float> RadiationModel::getCameraPixelData(const std::string &camera_label, const std::string &band_label) {
    if (cameras.find(camera_label) == cameras.end()) {
        helios_runtime_error("ERROR (RadiationModel::getCameraPixelData): Camera '" + camera_label + "' does not exist.");
    }
 
    auto &camera_pixel_data = cameras.at(camera_label).pixel_data;
    if (camera_pixel_data.find(band_label) == camera_pixel_data.end()) {
        helios_runtime_error("ERROR (RadiationModel::getCameraPixelData): Band '" + band_label + "' does not exist in camera '" + camera_label + "'.");
    }
 
    return camera_pixel_data.at(band_label);
}
 
void RadiationModel::setCameraPixelData(const std::string &camera_label, const std::string &band_label, const std::vector<float> &pixel_data) {
    if (cameras.find(camera_label) == cameras.end()) {
        helios_runtime_error("ERROR (RadiationModel::setCameraPixelData): Camera '" + camera_label + "' does not exist.");
    }
 
    cameras.at(camera_label).pixel_data[band_label] = pixel_data;
}
 
// ========== Phase 1: Backend Integration Methods ==========
 
void RadiationModel::queryBackendGPUMemory() const {
    if (backend) {
        backend->queryGPUMemory();
    } else {
        std::cout << "Backend not initialized - cannot query GPU memory." << std::endl;
    }
}
 
 
helios::RayTracingLaunchParams RadiationModel::buildCameraLaunchParams(const RadiationCamera &camera, uint camera_id, uint antialiasing_samples, const helios::int2 &tile_resolution, const helios::int2 &tile_offset) {
 
    helios::RayTracingLaunchParams params;
 
    // Camera position and orientation
    params.camera_position = camera.position;
    helios::SphericalCoord dir = cart2sphere(camera.lookat - camera.position);
    params.camera_direction = helios::make_vec2(dir.zenith, dir.azimuth);
 
    // Camera optical properties
    params.camera_focal_length = camera.focal_length;
    params.camera_lens_diameter = camera.lens_diameter;
    params.camera_fov_aspect = camera.FOV_aspect_ratio;
 
    // Resolution and tiling
    params.camera_resolution = tile_resolution;
    params.camera_resolution_full = camera.resolution;
    params.camera_pixel_offset = tile_offset;
    params.antialiasing_samples = antialiasing_samples;
    params.camera_id = camera_id;
 
    // Compute effective HFOV with zoom
    float effective_HFOV = camera.HFOV_degrees / camera.camera_zoom;
    params.camera_HFOV = effective_HFOV * M_PI / 180.0f;
    params.camera_viewplane_length = 0.5f / tanf(0.5f * effective_HFOV * M_PI / 180.f);
 
    // Compute pixel solid angle
    float HFOV_rad = effective_HFOV * M_PI / 180.f;
    float VFOV_rad = HFOV_rad / camera.FOV_aspect_ratio;
    float pixel_angle_h = HFOV_rad / float(camera.resolution.x);
    float pixel_angle_v = VFOV_rad / float(camera.resolution.y);
    params.camera_pixel_solid_angle = pixel_angle_h * pixel_angle_v;
 
    // Explicitly set scattering iteration for cameras (always iteration 0 for specular)
    params.scattering_iteration = 0;
 
    // Set specular reflection mode from auto-detection
    params.specular_reflection_enabled = specular_reflection_mode;
 
    return params;
}
 
std::vector<CameraTile> RadiationModel::computeCameraTiles(const RadiationCamera &camera, size_t maxRays) {
 
    std::vector<CameraTile> tiles;
 
    size_t total_rays = size_t(camera.antialiasing_samples) * size_t(camera.resolution.x) * size_t(camera.resolution.y);
 
    // No tiling needed
    if (total_rays <= maxRays) {
        tiles.push_back({camera.resolution, helios::make_int2(0, 0)});
        return tiles;
    }
 
    // Calculate tile dimensions
    size_t rays_per_row = size_t(camera.antialiasing_samples) * size_t(camera.resolution.x);
    size_t max_rows_per_tile = floor(float(maxRays) / float(rays_per_row));
 
    if (max_rows_per_tile == 0) {
        // 2D tiling - even one row is too large
        size_t max_pixels_per_tile = floor(float(maxRays) / float(camera.antialiasing_samples));
 
        float aspect = float(camera.resolution.x) / float(camera.resolution.y);
        size_t tile_width = round(sqrt(max_pixels_per_tile * aspect));
        size_t tile_height = floor(float(max_pixels_per_tile) / float(tile_width));
 
        tile_width = std::min(tile_width, size_t(camera.resolution.x));
        tile_height = std::min(tile_height, size_t(camera.resolution.y));
 
        int Ntiles_x = ceil(float(camera.resolution.x) / float(tile_width));
        int Ntiles_y = ceil(float(camera.resolution.y) / float(tile_height));
 
        for (int ty = 0; ty < Ntiles_y; ty++) {
            for (int tx = 0; tx < Ntiles_x; tx++) {
                size_t offset_x = tx * tile_width;
                size_t offset_y = ty * tile_height;
                size_t width_this = std::min(tile_width, camera.resolution.x - offset_x);
                size_t height_this = std::min(tile_height, camera.resolution.y - offset_y);
 
                tiles.push_back({helios::make_int2(width_this, height_this), helios::make_int2(offset_x, offset_y)});
            }
        }
    } else {
        // 1D tiling - tile along height only
        size_t rows_per_tile = std::min(max_rows_per_tile, size_t(camera.resolution.y));
        int Ntiles = ceil(float(camera.resolution.y) / float(rows_per_tile));
 
        for (int t = 0; t < Ntiles; t++) {
            size_t offset_y = t * rows_per_tile;
            size_t height_this = std::min(rows_per_tile, camera.resolution.y - offset_y);
 
            tiles.push_back({helios::make_int2(camera.resolution.x, height_this), helios::make_int2(0, offset_y)});
        }
    }
 
    return tiles;
}
 
void RadiationModel::buildGeometryData() {
    // Build backend-agnostic geometry data from Context primitives
    // This extracts all geometry information needed by the ray tracing backend
 
    std::vector<uint> UUIDs = context->getAllUUIDs();
 
    // Filter out invalid/zero-area primitives (same as old updateGeometry)
    std::vector<uint> valid_UUIDs;
    for (uint UUID: UUIDs) {
        if (!context->doesPrimitiveExist(UUID))
            continue;
 
        float area = context->getPrimitiveArea(UUID);
        uint parentID = context->getPrimitiveParentObjectID(UUID);
        if ((area == 0 || std::isnan(area)) && context->getObjectType(parentID) != helios::OBJECT_TYPE_TILE) {
            continue;
        }
        valid_UUIDs.push_back(UUID);
    }
 
    if (valid_UUIDs.empty()) {
        geometry_data = helios::RayTracingGeometry(); // Empty geometry
        return;
    }
 
    // Reorder primitives by parent object (same ordering as old code)
    std::vector<uint> objID_all = context->getUniquePrimitiveParentObjectIDs(valid_UUIDs, true);
    std::vector<uint> primitive_UUIDs_ordered;
    std::unordered_set<uint> valid_set(valid_UUIDs.begin(), valid_UUIDs.end());
 
    for (uint objID: objID_all) {
        const std::vector<uint> &prim_UUIDs = context->getObjectPrimitiveUUIDs(objID);
        for (uint UUID: prim_UUIDs) {
            if (context->doesPrimitiveExist(UUID) && valid_set.find(UUID) != valid_set.end()) {
                primitive_UUIDs_ordered.push_back(UUID);
            }
        }
    }
 
    size_t Nprimitives = primitive_UUIDs_ordered.size();
    geometry_data.primitive_count = Nprimitives;
 
    // Clear and allocate per-primitive arrays (important when updateGeometry is called multiple times)
    geometry_data.transform_matrices.clear();
    geometry_data.transform_matrices.resize(Nprimitives * 16);
    geometry_data.primitive_types.clear();
    // Initialize to UINT_MAX as sentinel - prevents uninitialized entries from matching type==0 (patch)
    geometry_data.primitive_types.resize(Nprimitives, UINT_MAX);
    geometry_data.primitive_UUIDs = primitive_UUIDs_ordered;
    geometry_data.primitive_IDs.clear();
    geometry_data.primitive_IDs.resize(Nprimitives); // Will be populated after primitiveID_indices is built
    geometry_data.object_IDs.clear();
    geometry_data.object_IDs.resize(Nprimitives);
    geometry_data.object_subdivisions.clear();
    geometry_data.object_subdivisions.resize(Nprimitives);
    geometry_data.twosided_flags.clear();
    geometry_data.twosided_flags.resize(Nprimitives);
    geometry_data.solid_fractions.clear();
    geometry_data.solid_fractions.resize(Nprimitives);
 
    // Clear type-specific arrays
    geometry_data.patches.vertices.clear();
    geometry_data.patches.UUIDs.clear();
    geometry_data.triangles.vertices.clear();
    geometry_data.triangles.UUIDs.clear();
    geometry_data.disk_centers.clear();
    geometry_data.disk_radii.clear();
    geometry_data.disk_normals.clear();
    geometry_data.disk_UUIDs.clear();
    geometry_data.tiles.vertices.clear();
    geometry_data.tiles.UUIDs.clear();
    geometry_data.voxels.vertices.clear();
    geometry_data.voxels.UUIDs.clear();
    geometry_data.bboxes.vertices.clear();
    geometry_data.bboxes.UUIDs.clear();
 
    // Track object IDs for compound objects
    uint current_objID = 0;
    uint last_parentID = 99999;
 
    std::vector<uint> primitiveID_indices; // Maps primitives to their "object" index
 
    for (size_t u = 0; u < Nprimitives; u++) {
        uint UUID = primitive_UUIDs_ordered[u];
        uint parentID = context->getPrimitiveParentObjectID(UUID);
 
        if (last_parentID != parentID || parentID == 0 || context->getObjectType(parentID) != helios::OBJECT_TYPE_TILE) {
            primitiveID_indices.push_back(u);
            last_parentID = parentID;
            current_objID++;
        } else {
            last_parentID = parentID;
        }
 
        geometry_data.object_IDs[u] = current_objID - 1;
    }
 
    size_t Nobjects = primitiveID_indices.size();
 
    // Populate primitiveID for runBand() compatibility
    primitiveID = primitiveID_indices;
 
    // For backend: primitiveID[position] must return the UUID for that primitive
    // Sized by Nprimitives (all primitives including subpatches), not Nobjects (object entries only)
    std::vector<uint> primitiveID_for_backend(Nprimitives);
    for (size_t i = 0; i < Nprimitives; i++) {
        primitiveID_for_backend[i] = primitive_UUIDs_ordered[i];
    }
 
    // Copy corrected primitiveID mapping to geometry_data for backend upload
    geometry_data.primitive_IDs = primitiveID_for_backend;
 
    // Populate geometry for each primitive
    size_t patch_idx = 0, tri_idx = 0, disk_idx = 0, tile_idx = 0, voxel_idx = 0, bbox_idx = 0;
 
    // Track which parent tile objects have been added to tile geometry
    // (tiles should have ONE geometry entry per parent object, not per subpatch)
    std::set<uint> processed_tile_parents;
 
    // Iterate over ALL primitives to set per-primitive data
    // (not just Nobjects, which only has one entry per object group)
    for (size_t prim_idx = 0; prim_idx < Nprimitives; prim_idx++) {
        uint UUID = primitive_UUIDs_ordered[prim_idx];
 
        // Transform matrix
        float m[16];
        uint parentID = context->getPrimitiveParentObjectID(UUID);
        helios::PrimitiveType type = context->getPrimitiveType(UUID);
 
        // Solid fraction
        geometry_data.solid_fractions[prim_idx] = context->getPrimitiveSolidFraction(UUID);
 
        // Two-sided flag
        geometry_data.twosided_flags[prim_idx] = context->getPrimitiveTwosidedFlag(UUID, 1) ? 1 : 0;
 
        if (parentID > 0 && context->getObjectType(parentID) == helios::OBJECT_TYPE_TILE) {
            // Tile object - set per-primitive data for ALL subpatches
            geometry_data.primitive_types[prim_idx] = 3; // tile
 
            context->getObjectTransformationMatrix(parentID, m);
            memcpy(&geometry_data.transform_matrices[prim_idx * 16], m, 16 * sizeof(float));
 
            // Individual tile subpatches should NOT be subdivided (they're already the result of subdivision)
            // Only the parent tile geometry entry uses the subdivision count
            geometry_data.object_subdivisions[prim_idx] = helios::make_int2(1, 1);
 
            // Only add ONE tile geometry entry per parent tile object (not per subpatch)
            // The tile intersection program handles subpatch selection internally
            if (processed_tile_parents.find(parentID) == processed_tile_parents.end()) {
                processed_tile_parents.insert(parentID);
 
                std::vector<vec3> verts = context->getTileObjectVertices(parentID);
                for (const auto &v: verts) {
                    geometry_data.tiles.vertices.push_back(v);
                }
 
                // Use the first subpatch's UUID as the tile geometry UUID
                geometry_data.tiles.UUIDs.push_back(UUID);
                tile_idx++;
            }
 
        } else if (type == helios::PRIMITIVE_TYPE_PATCH) {
            geometry_data.primitive_types[prim_idx] = 0; // patch
 
            context->getPrimitiveTransformationMatrix(UUID, m);
            memcpy(&geometry_data.transform_matrices[prim_idx * 16], m, 16 * sizeof(float));
 
            std::vector<vec3> verts = context->getPrimitiveVertices(UUID);
            for (const auto &v: verts) {
                geometry_data.patches.vertices.push_back(v);
            }
 
            geometry_data.object_subdivisions[prim_idx] = helios::make_int2(1, 1);
 
            // FIX: Add UUID inline to ensure consistent ordering with vertices
            geometry_data.patches.UUIDs.push_back(UUID);
 
            patch_idx++;
 
        } else if (type == helios::PRIMITIVE_TYPE_TRIANGLE) {
            geometry_data.primitive_types[prim_idx] = 1; // triangle
 
            context->getPrimitiveTransformationMatrix(UUID, m);
            memcpy(&geometry_data.transform_matrices[prim_idx * 16], m, 16 * sizeof(float));
 
            std::vector<vec3> verts = context->getPrimitiveVertices(UUID);
            for (const auto &v: verts) {
                geometry_data.triangles.vertices.push_back(v);
            }
 
            geometry_data.object_subdivisions[prim_idx] = helios::make_int2(1, 1);
            geometry_data.triangles.UUIDs.push_back(UUID); // Store actual UUID, not position
            tri_idx++;
 
        } else if (type == helios::PRIMITIVE_TYPE_VOXEL) {
            geometry_data.primitive_types[prim_idx] = 4; // voxel
 
            context->getPrimitiveTransformationMatrix(UUID, m);
            memcpy(&geometry_data.transform_matrices[prim_idx * 16], m, 16 * sizeof(float));
 
            std::vector<vec3> verts = context->getPrimitiveVertices(UUID);
            for (const auto &v: verts) {
                geometry_data.voxels.vertices.push_back(v);
            }
 
            geometry_data.object_subdivisions[prim_idx] = helios::make_int2(1, 1);
            geometry_data.voxels.UUIDs.push_back(UUID); // Store actual UUID, not position
            voxel_idx++;
        }
    }
 
    // Set counts
    geometry_data.patch_count = patch_idx;
    geometry_data.triangle_count = tri_idx;
    geometry_data.disk_count = disk_idx;
    geometry_data.tile_count = tile_idx;
    geometry_data.voxel_count = voxel_idx;
 
    // ========== Periodic Boundary Bboxes ==========
    // Create bbox geometry for periodic boundary conditions
    // Each bbox face is a rectangular boundary at domain edge
 
    // Get domain bounding box
    vec2 xbounds, ybounds, zbounds;
    context->getDomainBoundingBox(xbounds, ybounds, zbounds);
 
    // Validate camera positions if periodic boundaries enabled
    if (periodic_flag.x == 1 || periodic_flag.y == 1) {
        if (!cameras.empty()) {
            for (auto &camera: cameras) {
                vec3 camerapos = camera.second.position;
                if (camerapos.x < xbounds.x || camerapos.x > xbounds.y || camerapos.y < ybounds.x || camerapos.y > ybounds.y) {
                    std::cout << "WARNING (RadiationModel::buildGeometryData): camera position is outside of the domain bounding box. Disabling periodic boundary conditions." << std::endl;
                    periodic_flag.x = 0;
                    periodic_flag.y = 0;
                    break;
                }
                // Extend z-bounds to include camera
                if (camerapos.z < zbounds.x) {
                    zbounds.x = camerapos.z;
                }
                if (camerapos.z > zbounds.y) {
                    zbounds.y = camerapos.z;
                }
            }
        }
    }
 
    // Expand bounds slightly to ensure bbox faces are outside geometry
    xbounds.x -= 1e-5;
    xbounds.y += 1e-5;
    ybounds.x -= 1e-5;
    ybounds.y += 1e-5;
    zbounds.x -= 1e-5;
    zbounds.y += 1e-5;
 
    // Bbox UUIDs must not collide with real primitive UUIDs
    // Use max_UUID + 1 as base (not Nprimitives, which can cause collisions with sparse UUIDs)
    uint max_UUID = geometry_data.primitive_UUIDs.empty() ? 0 : *std::max_element(geometry_data.primitive_UUIDs.begin(), geometry_data.primitive_UUIDs.end());
    uint bbox_UUID_base = max_UUID + 1;
 
    // Create bbox faces based on periodic flags
    if (periodic_flag.x == 1) {
        // -x facing boundary (4 vertices: counter-clockwise from bottom-left)
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.x, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.y, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.y, zbounds.y));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.x, zbounds.y));
        geometry_data.bboxes.UUIDs.push_back(bbox_UUID_base + bbox_idx);
        bbox_idx++;
 
        // +x facing boundary
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.x, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.y, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.y, zbounds.y));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.x, zbounds.y));
        geometry_data.bboxes.UUIDs.push_back(bbox_UUID_base + bbox_idx);
        bbox_idx++;
    }
 
    if (periodic_flag.y == 1) {
        // -y facing boundary
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.x, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.x, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.x, zbounds.y));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.x, zbounds.y));
        geometry_data.bboxes.UUIDs.push_back(bbox_UUID_base + bbox_idx);
        bbox_idx++;
 
        // +y facing boundary
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.y, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.y, zbounds.x));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.y, ybounds.y, zbounds.y));
        geometry_data.bboxes.vertices.push_back(vec3(xbounds.x, ybounds.y, zbounds.y));
        geometry_data.bboxes.UUIDs.push_back(bbox_UUID_base + bbox_idx);
        bbox_idx++;
    }
 
    // Update bbox count and UUID base
    geometry_data.bbox_count = bbox_idx;
    if (bbox_idx > 0) {
        geometry_data.bbox_UUID_base = bbox_UUID_base;
    } else {
        // No bboxes: set sentinel value so GPU knows all UUIDs are real primitives
        geometry_data.bbox_UUID_base = UINT_MAX;
    }
 
    // Add bbox primitive data so hit/intersection shaders can access them
    // Bbox positions are after real primitives: primitive_count + bbox_idx
    if (bbox_idx > 0) {
        geometry_data.primitive_types.resize(geometry_data.primitive_count + bbox_idx, UINT_MAX);
        geometry_data.twosided_flags.resize(geometry_data.primitive_count + bbox_idx, 0);
        geometry_data.solid_fractions.resize(geometry_data.primitive_count + bbox_idx, 1.0f);
        geometry_data.object_IDs.resize(geometry_data.primitive_count + bbox_idx, UINT_MAX);
        geometry_data.object_subdivisions.resize(geometry_data.primitive_count + bbox_idx, make_int2(1, 1));
        geometry_data.transform_matrices.resize((geometry_data.primitive_count + bbox_idx) * 16, 0.0f);
        geometry_data.primitive_IDs.resize(geometry_data.primitive_count + bbox_idx, UINT_MAX);
 
        for (size_t i = 0; i < bbox_idx; i++) {
            size_t bbox_pos = geometry_data.primitive_count + i;
            geometry_data.primitive_types[bbox_pos] = 5; // type=5 for bbox
            geometry_data.twosided_flags[bbox_pos] = 1; // bboxes are two-sided
            geometry_data.solid_fractions[bbox_pos] = 1.0f; // bboxes are fully solid
            geometry_data.object_IDs[bbox_pos] = geometry_data.primitive_count + i; // Match master: Nobjects + i
            geometry_data.object_subdivisions[bbox_pos] = make_int2(1, 1); // no subdivisions
            geometry_data.primitive_IDs[bbox_pos] = bbox_UUID_base + i; // bbox UUID
 
            // Set identity transform matrix for bbox
            geometry_data.transform_matrices[bbox_pos * 16 + 0] = 1.0f; // m00
            geometry_data.transform_matrices[bbox_pos * 16 + 5] = 1.0f; // m11
            geometry_data.transform_matrices[bbox_pos * 16 + 10] = 1.0f; // m22
            geometry_data.transform_matrices[bbox_pos * 16 + 15] = 1.0f; // m33
        }
    }
 
    // Periodic boundary condition
    geometry_data.periodic_flag = periodic_flag;
 
    // Extract texture mask and UV data for primitives with transparency textures
    buildTextureData();
 
    // Build primitive_positions lookup table for GPU UUID→position conversion
    // Size by max UUID to create sparse lookup table (includes bbox UUIDs now that they don't collide)
    // Clear first to remove stale mappings from deleted primitives
    geometry_data.primitive_positions.clear();
    if (!geometry_data.primitive_UUIDs.empty()) {
        uint max_UUID = *std::max_element(geometry_data.primitive_UUIDs.begin(), geometry_data.primitive_UUIDs.end());
 
        // Expand to include bbox UUIDs if present (they now use max_UUID+1 base, so no collisions)
        uint bbox_max_UUID = max_UUID;
        if (geometry_data.bbox_count > 0) {
            bbox_max_UUID = geometry_data.bbox_UUID_base + geometry_data.bbox_count - 1;
        }
 
        geometry_data.primitive_positions.resize(bbox_max_UUID + 1, UINT_MAX); // UINT_MAX = invalid/unused
 
        // Map real primitive UUIDs
        for (size_t i = 0; i < geometry_data.primitive_count; i++) {
            uint UUID = geometry_data.primitive_UUIDs[i];
            geometry_data.primitive_positions[UUID] = i; // Map UUID → array position
        }
 
        // Map bbox UUIDs to their positions (after real primitives)
        // Now safe because bbox_UUID_base = max_UUID + 1 (no collisions)
        if (geometry_data.bbox_count > 0) {
            for (size_t i = 0; i < geometry_data.bbox_count; i++) {
                uint bbox_UUID = geometry_data.bbox_UUID_base + i;
                geometry_data.primitive_positions[bbox_UUID] = geometry_data.primitive_count + i;
            }
        }
    }
}
 
void RadiationModel::buildTextureData() {
    // Extract texture mask and UV data for all primitives with transparency textures
 
    size_t Nobjects = geometry_data.primitive_count;
 
    // Clear any previous texture data (important when updateGeometry is called multiple times)
    geometry_data.mask_data.clear();
    geometry_data.mask_sizes.clear();
    geometry_data.uv_data.clear();
 
    // Initialize with -1 (no texture)
    geometry_data.mask_IDs.clear();
    geometry_data.mask_IDs.resize(Nobjects, -1);
    geometry_data.uv_IDs.clear();
    geometry_data.uv_IDs.resize(Nobjects, -1);
 
    // Cache to avoid duplicate mask data for primitives using the same texture file
    std::map<std::string, int> texture_to_mask_idx;
 
    for (size_t prim_idx = 0; prim_idx < Nobjects; prim_idx++) {
        uint UUID = geometry_data.primitive_UUIDs[prim_idx];
 
        // Check if primitive has a texture file
        std::string texture_file = context->getPrimitiveTextureFile(UUID);
        if (texture_file.empty()) {
            continue; // No texture - mask_ID stays -1
        }
 
        // Check if texture has transparency channel (alpha)
        if (!context->primitiveTextureHasTransparencyChannel(UUID)) {
            continue; // No transparency - mask_ID stays -1 (e.g., JPEG files)
        }
 
        // Check cache for existing mask from same texture file
        int mask_idx;
        auto cache_it = texture_to_mask_idx.find(texture_file);
        if (cache_it != texture_to_mask_idx.end()) {
            // Reuse existing mask
            mask_idx = cache_it->second;
        } else {
            // New texture - extract mask data
            const std::vector<std::vector<bool>> *trans_data = context->getPrimitiveTextureTransparencyData(UUID);
            helios::int2 tex_size = context->getPrimitiveTextureSize(UUID);
 
            mask_idx = static_cast<int>(geometry_data.mask_sizes.size());
            texture_to_mask_idx[texture_file] = mask_idx;
 
            // Flatten 2D bool array to 1D (row-major: [y][x])
            // Backend expects: for each mask m, iterate [y][x] order
            for (int y = 0; y < tex_size.y; y++) {
                for (int x = 0; x < tex_size.x; x++) {
                    geometry_data.mask_data.push_back((*trans_data)[y][x]);
                }
            }
            geometry_data.mask_sizes.push_back(tex_size);
        }
 
        geometry_data.mask_IDs[prim_idx] = mask_idx;
 
        // Extract UV coordinates for this primitive
        // uv_IDs stores the position index (not offset), used to access uvdata[vertex][position] in CUDA
        std::vector<helios::vec2> uvs = context->getPrimitiveTextureUV(UUID);
        if (!uvs.empty()) {
            geometry_data.uv_IDs[prim_idx] = static_cast<int>(prim_idx); // Store position index for CUDA 2D buffer access
            for (const auto &uv: uvs) {
                geometry_data.uv_data.push_back(uv);
            }
            // Pad to 4 vertices if needed (CUDA expects max 4 vertices per primitive)
            size_t start_idx = geometry_data.uv_data.size() - uvs.size();
            while (geometry_data.uv_data.size() - start_idx < 4) {
                geometry_data.uv_data.push_back(uvs.back());
            }
        }
        // If uvs is empty, uv_ID stays -1 and CUDA will use default UV mapping
    }
}
 
size_t RadiationModel::testBuildGeometryData() {
    buildGeometryData();
    return geometry_data.primitive_count;
}
 
void RadiationModel::buildUUIDMapping() {
    // Build bidirectional UUID ↔ array position mapping
    // This enables efficient conversion between UUID values and array indices
 
    uuid_to_position.clear();
    position_to_uuid.clear();
 
    // geometry_data.primitive_UUIDs is already ordered by object
    // Build mapping from this ordered list
    for (size_t i = 0; i < geometry_data.primitive_count; i++) {
        uint UUID = geometry_data.primitive_UUIDs[i];
        uuid_to_position[UUID] = i;
        position_to_uuid.push_back(UUID);
    }
 
    // Build type-safe mapper (new indexing system)
    // Provides compile-time safety for UUID/position conversions
    geometry_data.mapper.build(geometry_data.primitive_UUIDs);
}
 
static void validateAndCorrectMaterialProperties(float &rho, float &tau, float eps, bool emission_enabled, uint scattering_depth, const std::string &band_label, uint UUID, helios::WarningAggregator *warnings = nullptr) {
    // Helper function to enforce energy conservation constraints on material properties
    // Mirrors the validation logic from updateRadiativeProperties() (lines 2672-2686)
 
    // 1. Clamp rho and tau to [0,1] with warnings for out-of-range values
    if (rho < 0.f || rho > 1.f) {
        if (warnings) {
            warnings->addWarning("material_property_clamping", "Reflectivity out of range [0,1] for band " + band_label + ", primitive #" + std::to_string(UUID) + ": rho=" + std::to_string(rho) + ". Clamping to valid range.");
        }
        rho = std::max(0.f, std::min(1.f, rho));
    }
 
    if (tau < 0.f || tau > 1.f) {
        if (warnings) {
            warnings->addWarning("material_property_clamping", "Transmissivity out of range [0,1] for band " + band_label + ", primitive #" + std::to_string(UUID) + ": tau=" + std::to_string(tau) + ". Clamping to valid range.");
        }
        tau = std::max(0.f, std::min(1.f, tau));
    }
 
    // 2. Apply emission-specific constraints
    if (emission_enabled) {
        // Special case: blackbody emission (scatteringDepth=0 requires eps=1, rho=0, tau=0)
        if (scattering_depth == 0 && eps != 1.f) {
            if (warnings && (rho != 0.f || tau != 0.f)) {
                warnings->addWarning("blackbody_override", "Band " + band_label + " has emission with scatteringDepth=0, " + "enforcing blackbody behavior (eps=1, rho=0, tau=0) for primitive #" + std::to_string(UUID));
            }
            rho = 0.f;
            tau = 0.f;
        }
        // General emission case: check energy conservation (eps + rho + tau = 1)
        else if (eps != 1.f && rho == 0 && tau == 0) {
            // Auto-correct: set rho = 1 - eps
            rho = 1.f - eps;
        } else if (std::abs(eps + rho + tau - 1.f) > 1e-5f && eps > 0.f) {
            // Cannot auto-correct, throw error
            helios_runtime_error(std::string("ERROR (RadiationModel): emissivity, transmissivity, and reflectivity ") + "must sum to 1 to ensure energy conservation. Band " + band_label + ", Primitive #" + std::to_string(UUID) +
                                 ": eps=" + std::to_string(eps) + ", tau=" + std::to_string(tau) + ", rho=" + std::to_string(rho) + ". It is also possible that you forgot to disable emission for this band.");
        }
    } else {
        // 3. Non-emission case: rho + tau must be ≤ 1
        if (rho + tau > 1.f) {
            helios_runtime_error(std::string("ERROR (RadiationModel): transmissivity and reflectivity cannot sum to ") + "greater than 1 to ensure energy conservation. Band " + band_label + ", Primitive #" + std::to_string(UUID) +
                                 ": eps=" + std::to_string(eps) + ", tau=" + std::to_string(tau) + ", rho=" + std::to_string(rho) + ". It is also possible that you forgot to disable emission for this band.");
        }
    }
}
 
void RadiationModel::buildMaterialData() {
    // Build backend-agnostic material data from Context primitive data
 
    // Warning aggregator for energy conservation issues
    helios::WarningAggregator warnings;
 
    size_t Nprims = geometry_data.primitive_count;
    size_t Nbands = radiation_bands.size();
    size_t Nsources = radiation_sources.size();
 
    material_data.num_primitives = Nprims;
    material_data.num_bands = Nbands;
    material_data.num_sources = Nsources;
    material_data.num_cameras = cameras.size();
 
    // Allocate arrays (indexed as [source][primitive][band] using MaterialPropertyIndexer)
    // NOTE: Bboxes don't need material properties (they only wrap rays for periodic boundaries)
    size_t total_size = Nsources * Nbands * Nprims;
    material_data.reflectivity.resize(total_size, 0.0f);
    material_data.transmissivity.resize(total_size, 0.0f);
    material_data.specular_exponent.resize(Nprims, -1.0f); // Default -1 means disabled
    material_data.specular_scale.resize(Nprims, 0.0f);
 
    // Create indexer for material properties: [source][primitive][band]
    MaterialPropertyIndexer mat_indexer(Nsources, Nprims, Nbands);
 
    // Cache unique spectral data to avoid redundant loads
    std::map<std::string, std::vector<helios::vec2>> unique_rho_spectra;
    std::map<std::string, std::vector<helios::vec2>> unique_tau_spectra;
 
    for (size_t p = 0; p < Nprims; p++) {
        uint UUID = geometry_data.primitive_UUIDs[p];
 
        // Cache reflectivity spectra
        if (context->doesPrimitiveDataExist(UUID, "reflectivity_spectrum")) {
            std::string spectrum_label;
            context->getPrimitiveData(UUID, "reflectivity_spectrum", spectrum_label);
            if (unique_rho_spectra.find(spectrum_label) == unique_rho_spectra.end()) {
                // Only load if spectrum exists in global data
                if (context->doesGlobalDataExist(spectrum_label.c_str())) {
                    unique_rho_spectra[spectrum_label] = loadSpectralData(spectrum_label);
                }
            }
        }
 
        // Cache transmissivity spectra
        if (context->doesPrimitiveDataExist(UUID, "transmissivity_spectrum")) {
            std::string spectrum_label;
            context->getPrimitiveData(UUID, "transmissivity_spectrum", spectrum_label);
            if (unique_tau_spectra.find(spectrum_label) == unique_tau_spectra.end()) {
                // Only load if spectrum exists in global data
                if (context->doesGlobalDataExist(spectrum_label.c_str())) {
                    unique_tau_spectra[spectrum_label] = loadSpectralData(spectrum_label);
                }
            }
        }
    }
 
    // Extract material properties from Context primitives
    size_t b_idx = 0;
    for (const auto &band_pair: radiation_bands) {
        std::string band_label = band_pair.second.label;
 
        for (size_t s = 0; s < Nsources; s++) {
            for (size_t p = 0; p < Nprims; p++) {
                uint UUID = geometry_data.primitive_UUIDs[p];
 
                // Use BufferIndexer for safe, verifiable indexing
                // Note: p is already the array position, so we use p directly (not UUID)
                size_t idx = mat_indexer(s, p, b_idx);
 
                // Get reflectivity - try spectrum first, then per-band label
                float rho = rho_default;
 
                if (context->doesPrimitiveDataExist(UUID, "reflectivity_spectrum")) {
                    // Spectrum-based reflectivity
                    std::string spectrum_label;
                    context->getPrimitiveData(UUID, "reflectivity_spectrum", spectrum_label);
 
                    // Get spectrum from cache
                    if (unique_rho_spectra.find(spectrum_label) != unique_rho_spectra.end()) {
                        const std::vector<helios::vec2> &spectrum = unique_rho_spectra.at(spectrum_label);
 
                        // Get band wavelength bounds
                        helios::vec2 wavebounds = band_pair.second.wavebandBounds;
 
                        // Only require wavelength bounds if band performs scattering/absorption
                        // Emission-only bands (scatteringDepth==0) use Stefan-Boltzmann and don't need spectral integration
                        // Ray launches for emission don't require wavelength bounds since emission properties are wavelength-independent
                        bool needs_spectral_integration = (band_pair.second.scatteringDepth > 0);
 
                        if (needs_spectral_integration && wavebounds.x == 0 && wavebounds.y == 0) {
                            helios_runtime_error("ERROR (RadiationModel::buildMaterialData): Band '" + band_label + "' has no wavelength bounds - required for spectral integration");
                        }
 
                        // Integrate spectrum over band wavelength range (only if bounds are defined)
                        if (wavebounds.x != 0 || wavebounds.y != 0) {
                            if (!radiation_sources[s].source_spectrum.empty()) {
                                // Weight by source spectrum
                                rho = integrateSpectrum(s, spectrum, wavebounds.x, wavebounds.y);
                            } else {
                                // Uniform integration (divide by wavelength range to normalize)
                                rho = integrateSpectrum(spectrum, wavebounds.x, wavebounds.y) / (wavebounds.y - wavebounds.x);
                            }
                        }
                        // else: emission-only band, rho remains at default value (should be 0 for blackbody)
                    }
                } else {
                    // Per-band reflectivity (backward compatibility)
                    std::string rho_label = "reflectivity_" + band_label;
                    if (context->doesPrimitiveDataExist(UUID, rho_label.c_str())) {
                        context->getPrimitiveData(UUID, rho_label.c_str(), rho);
                    }
                }
 
                // Get transmissivity - try spectrum first, then per-band label
                float tau = tau_default;
 
                if (context->doesPrimitiveDataExist(UUID, "transmissivity_spectrum")) {
                    // Spectrum-based transmissivity
                    std::string spectrum_label;
                    context->getPrimitiveData(UUID, "transmissivity_spectrum", spectrum_label);
 
                    // Get spectrum from cache
                    if (unique_tau_spectra.find(spectrum_label) != unique_tau_spectra.end()) {
                        const std::vector<helios::vec2> &spectrum = unique_tau_spectra.at(spectrum_label);
 
                        // Get band wavelength bounds
                        helios::vec2 wavebounds = band_pair.second.wavebandBounds;
 
                        // Only require wavelength bounds if band performs scattering/absorption
                        // Emission-only bands (scatteringDepth==0) use Stefan-Boltzmann and don't need spectral integration
                        // Ray launches for emission don't require wavelength bounds since emission properties are wavelength-independent
                        bool needs_spectral_integration = (band_pair.second.scatteringDepth > 0);
 
                        if (needs_spectral_integration && wavebounds.x == 0 && wavebounds.y == 0) {
                            helios_runtime_error("ERROR (RadiationModel::buildMaterialData): Band '" + band_label + "' has no wavelength bounds - required for spectral integration");
                        }
 
                        // Integrate spectrum over band wavelength range (only if bounds are defined)
                        if (wavebounds.x != 0 || wavebounds.y != 0) {
                            if (!radiation_sources[s].source_spectrum.empty()) {
                                // Weight by source spectrum
                                tau = integrateSpectrum(s, spectrum, wavebounds.x, wavebounds.y);
                            } else {
                                // Uniform integration
                                tau = integrateSpectrum(spectrum, wavebounds.x, wavebounds.y) / (wavebounds.y - wavebounds.x);
                            }
                        }
                        // else: emission-only band, tau remains at default value (should be 0 for blackbody)
                    }
                } else {
                    // Per-band transmissivity (backward compatibility)
                    std::string tau_label = "transmissivity_" + band_label;
                    if (context->doesPrimitiveDataExist(UUID, tau_label.c_str())) {
                        context->getPrimitiveData(UUID, tau_label.c_str(), tau);
                    }
                }
 
                // Get emissivity for validation
                float eps = eps_default;
                std::string eps_label = "emissivity_" + band_label;
                if (context->doesPrimitiveDataExist(UUID, eps_label.c_str())) {
                    context->getPrimitiveData(UUID, eps_label.c_str(), eps);
                }
 
                // Validate and correct material properties to ensure energy conservation
                const RadiationBand &band = band_pair.second;
                validateAndCorrectMaterialProperties(rho, tau, eps, band.emissionFlag, band.scatteringDepth, band_label, UUID, &warnings);
 
                // Store validated properties
                material_data.reflectivity[idx] = rho;
                material_data.transmissivity[idx] = tau;
            }
        }
        b_idx++;
    }
 
    // NOTE: Bboxes don't need material properties - they only wrap rays for periodic boundaries
    // Material buffers are sized for real primitives only (Nprims), not including bboxes
 
    // Load specular reflection properties from primitive data
    bool specular_exponent_specified = false;
    bool specular_scale_specified = false;
 
    for (size_t p = 0; p < Nprims; p++) {
        uint UUID = geometry_data.primitive_UUIDs[p];
 
        if (context->doesPrimitiveDataExist(UUID, "specular_exponent") && context->getPrimitiveDataType("specular_exponent") == helios::HELIOS_TYPE_FLOAT) {
            context->getPrimitiveData(UUID, "specular_exponent", material_data.specular_exponent.at(p));
            if (material_data.specular_exponent.at(p) >= 0.f) {
                specular_exponent_specified = true;
            }
        }
 
        if (context->doesPrimitiveDataExist(UUID, "specular_scale") && context->getPrimitiveDataType("specular_scale") == helios::HELIOS_TYPE_FLOAT) {
            context->getPrimitiveData(UUID, "specular_scale", material_data.specular_scale.at(p));
            if (material_data.specular_scale.at(p) > 0.f) {
                specular_scale_specified = true;
            }
        }
    }
 
    // Auto-enable specular reflection if specular properties are specified on any primitive
    if (specular_exponent_specified) {
        if (specular_scale_specified) {
            specular_reflection_mode = 2; // Mode 2: use primitive specular_scale
        } else {
            specular_reflection_mode = 1; // Mode 1: use default 0.25 scale
        }
    } else {
        specular_reflection_mode = 0; // Disabled
    }
 
    // Report any accumulated warnings
    warnings.report();
}
 
void RadiationModel::buildSourceData() {
    // Build backend-agnostic source data from radiation_sources
 
    source_data.clear();
    source_data.reserve(radiation_sources.size());
 
    for (size_t s = 0; s < radiation_sources.size(); s++) {
        const auto &src = radiation_sources[s];
        helios::RayTracingSource backend_src;
        backend_src.position = src.source_position;
        backend_src.rotation = src.source_rotation;
        backend_src.width = src.source_width;
        backend_src.type = src.source_type;
 
        // Flatten flux arrays - use getSourceFlux() to handle -1.f sentinel values
        backend_src.fluxes.clear();
        backend_src.fluxes_cam.clear();
        for (const auto &band_pair: radiation_bands) {
            std::string band_label = band_pair.second.label;
            // Use getSourceFlux() which properly handles -1.f sentinel (returns 0 or integrates spectrum)
            float flux = getSourceFlux(s, band_label);
            backend_src.fluxes.push_back(flux);
            backend_src.fluxes_cam.push_back(flux); // Same for now
        }
 
        source_data.push_back(backend_src);
    }
}
 
 
helios::RayTracingBackend *RadiationModel::getBackend() {
    return backend.get();
}
 
helios::RayTracingGeometry &RadiationModel::getGeometryData() {
    return geometry_data;
}
 
helios::RayTracingMaterial &RadiationModel::getMaterialData() {
    return material_data;
}
 
std::vector<helios::RayTracingSource> &RadiationModel::getSourceData() {
    return source_data;
}
 
 
void RadiationModel::testBuildAllBackendData() {
    buildGeometryData();
    buildMaterialData();
    buildSourceData();
}