Process.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Process.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2018-2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Material/MaterialProperties.hpp"
#include "ActsFatras/EventData/Particle.hpp"

namespace ActsFatras {

struct EveryParticle {
  constexpr bool operator()(const Particle &) const { return true; }
};

// No-op input selector that selects all configurations.
struct EveryInput {
  constexpr bool operator()(const Particle &,
                            const Acts::MaterialProperties &) const {
    return true;
  }
};

template <typename ParticleSelector>
struct AsInputSelector : public ParticleSelector {
  bool operator()(const Particle &particle,
                  const Acts::MaterialProperties &) const {
    return ParticleSelector::operator()(particle);
  }
};

template <typename physics_t, typename input_selector_t = EveryInput,
          typename output_particle_selector_t = EveryParticle,
          typename child_particle_selector_t = output_particle_selector_t>
struct Process {
  physics_t physics;
  input_selector_t selectInput;
  output_particle_selector_t selectOutputParticle;
  child_particle_selector_t selectChildParticle;

  template <typename generator_t>
  bool operator()(generator_t &generator, const Acts::MaterialProperties &slab,
                  Particle &particle, std::vector<Particle> &generated) const {
    // not selecting this process is not a break condition
    if (not selectInput(particle, slab)) {
      return false;
    }
    // modify particle according to the physics process
    auto children = physics(generator, slab, particle);
    // move selected child particles to the output container
    std::copy_if(children.begin(), children.end(),
                 std::back_inserter(generated), selectChildParticle);
    // break condition is defined by whether the output particle is still valid
    // or not e.g. because it has fallen below a momentum threshold.
    return not selectOutputParticle(particle);
  }
};

}  // namespace ActsFatras