InteractionLengthModifier.hpp

/*
 * (c) Copyright 2021 CORSIKA Project, corsika-project@lists.kit.edu
 *
 * This software is distributed under the terms of the GNU General Public
 * Licence version 3 (GPL Version 3). See file LICENSE for a full version of
 * the license.
 */

#pragma once

#include <functional>

#include <corsika/framework/core/ParticleProperties.hpp>
#include <corsika/framework/core/PhysicalUnits.hpp>
#include <corsika/framework/process/InteractionProcess.hpp>

namespace corsika {
  template <class TUnderlyingProcess>
  class InteractionLengthModifier
      : public InteractionProcess<InteractionLengthModifier<TUnderlyingProcess>> {

    static auto constexpr non_modifying_functor = [](GrammageType original, corsika::Code,
                                                     HEPEnergyType) { return original; };

  public:
    using functor_signature = GrammageType(GrammageType, corsika::Code, HEPEnergyType);

    InteractionLengthModifier(TUnderlyingProcess&& process,
                              std::function<functor_signature> modifier);

    template <typename TSecondaryView>
    void doInteraction(TSecondaryView& view);

    template <typename TParticle>
    GrammageType getInteractionLength(TParticle const& particle);

    TUnderlyingProcess const& getProcess() const;
    TUnderlyingProcess& getProcess();

  private:
    TUnderlyingProcess process_;
    std::function<functor_signature> const modifier_{non_modifying_functor};
  };


} // namespace corsika

#include <corsika/detail/framework/process/InteractionLengthModifier.inl>