File EnergyDistribution.h¶

File List > EnergyDistributions > EnergyDistribution.h
#pragma once

#include "Beamline/EnergyDistribution.h"
#include "Core.h"
#include "Shader/Rand.h"
#include "Variant.h"

namespace rayx {

struct EnergyDistributionList {
    double* __restrict prefixWeights;
    double* __restrict energies;
    double weightSum;
    int size;
    bool continous;
};

struct EnergyDistributionDataBase {
    using HardEdge               = rayx::HardEdge;
    using SoftEdge               = rayx::SoftEdge;
    using SeparateEnergies       = rayx::SeparateEnergies;
    using EnergyDistributionList = rayx::EnergyDistributionList;
};

using EnergyDistributionDataVariant = Variant<EnergyDistributionDataBase, EnergyDistributionDataBase::HardEdge, EnergyDistributionDataBase::SoftEdge,
                                              SeparateEnergies, EnergyDistributionDataBase::EnergyDistributionList>;

RAYX_FN_ACC double selectEnergy(const HardEdge& __restrict hardEdge, Rand& __restrict rand);
RAYX_FN_ACC double selectEnergy(const SoftEdge& __restrict softEdge, Rand& __restrict rand);
RAYX_FN_ACC double selectEnergy(const SeparateEnergies& __restrict separateEnergies, Rand& __restrict rand);
RAYX_FN_ACC double selectEnergy(const EnergyDistributionList& __restrict energyDistributionList, Rand& __restrict rand);
RAYX_FN_ACC double selectEnergy(const EnergyDistributionDataVariant& __restrict energyDistribution, Rand& __restrict rand);

}  // namespace rayx