ProcessTests.cpp

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

// 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/.

#include <boost/test/unit_test.hpp>

#include "Acts/Material/MaterialSlab.hpp"
#include "Acts/Tests/CommonHelpers/PredefinedMaterials.hpp"
#include "Acts/Utilities/Units.hpp"
#include "ActsFatras/EventData/Particle.hpp"
#include "ActsFatras/Kernel/Process.hpp"

#include <array>
#include <random>

using namespace Acts::UnitLiterals;
using namespace ActsFatras;

namespace {
struct MakeChildren {
  template <typename generator_t>
  std::array<ActsFatras::Particle, 4> operator()(generator_t &,
                                                 const Acts::MaterialSlab &,
                                                 ActsFatras::Particle &) const {
    // create daughter particles
    return {
        Particle().setAbsMomentum(1_GeV),
        Particle().setAbsMomentum(2_GeV),
        Particle().setAbsMomentum(3_GeV),
        Particle().setAbsMomentum(4_GeV),
    };
  }
};

struct HighP {
  double minP = 10_GeV;

  bool operator()(const ActsFatras::Particle &particle) const {
    return (minP <= particle.absMomentum());
  }
};

struct Fixture {
  std::default_random_engine generator;
  Acts::MaterialSlab slab{Acts::Test::makeBeryllium(), 1_mm};
  Particle parent = Particle().setAbsMomentum(10_GeV);
  std::vector<Particle> children;
};
}  // namespace

BOOST_AUTO_TEST_SUITE(FatrasProcess)

BOOST_AUTO_TEST_CASE(NoSelectors) {
  Fixture f;
  Process<MakeChildren> process;

  // process should not abort
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 4u);
}

BOOST_AUTO_TEST_CASE(WithInputSelector) {
  Fixture f;
  Process<MakeChildren, AsInputSelector<HighP>> process;
  process.selectInput.minP = 10_GeV;

  // above threshold should not abort
  f.parent.setAbsMomentum(20_GeV);
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 4u);
  // on threshold should still not abort
  f.parent.setAbsMomentum(10_GeV);
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 8u);
  // below threshold should abort and not run the process at all
  f.parent.setAbsMomentum(2_GeV);
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  // process did not run -> no new children
  BOOST_CHECK_EQUAL(f.children.size(), 8u);
}

BOOST_AUTO_TEST_CASE(WithOutputSelector) {
  Fixture f;
  Process<MakeChildren, EveryInput, HighP, EveryParticle> process;
  process.selectOutputParticle.minP = 10_GeV;

  // above threshold should not abort
  f.parent.setAbsMomentum(20_GeV);
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 4u);
  // on threshold should still not abort
  f.parent.setAbsMomentum(10_GeV);
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 8u);
  // below threshold should abort but only after running the process
  f.parent.setAbsMomentum(2_GeV);
  BOOST_CHECK(process(f.generator, f.slab, f.parent, f.children));
  // process did still run -> new children
  BOOST_CHECK_EQUAL(f.children.size(), 12u);
}

BOOST_AUTO_TEST_CASE(WithChildSelector) {
  Fixture f;
  Process<MakeChildren, EveryInput, EveryParticle, HighP> process;
  process.selectChildParticle.minP = 10_GeV;

  // all process should not abort regardless of child selection
  // select no daughters
  process.selectChildParticle.minP = 5_GeV;
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 0u);
  // select highest daughter
  process.selectChildParticle.minP = 3.5_GeV;
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 1u);
  // select all but the lowest daughter
  process.selectChildParticle.minP = 1.5_GeV;
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 4u);
  // select all daughters
  process.selectChildParticle.minP = 0.5_GeV;
  BOOST_CHECK(not process(f.generator, f.slab, f.parent, f.children));
  BOOST_CHECK_EQUAL(f.children.size(), 8u);
}

BOOST_AUTO_TEST_SUITE_END()