PhysicsTools/TruthInfo/src/BranchSelector.cc - master - githubmirror/cmssw

PhysicsTools/TruthInfo/src/BranchSelector.cc

47 строк1 KB

Felice Pantaleo

TruthInfo: add author headers and prototype-status notice

13 июн 2026, 23:36

13 июн 2026, 23:362613821

// Original author: Felice Pantaleo (CERN) <felice.pantaleo@cern.ch>
// Part of the MC-truth-graph prototype - under heavy development, not yet open
// to external contributions (see PhysicsTools/TruthInfo/README.md).

#include "PhysicsTools/TruthInfo/interface/BranchSelector.h"

#include <algorithm>

#include "HepPDT/ParticleID.hh"

namespace truth {

  bool BranchSelector::operator()(Branch const& branch) const {
    if (!branch.valid())
      return false;

    if (config_.signalOnly && !branch.isSignal())
      return false;

    if (config_.intimeOnly && !branch.isInTime())
      return false;

    const int32_t pdgId = branch.rootPdgId();

    if (config_.chargedOnly && HepPDT::ParticleID(pdgId).threeCharge() == 0)
      return false;

    if (!config_.pdgIds.empty() &&
        std::find(config_.pdgIds.begin(), config_.pdgIds.end(), pdgId) == config_.pdgIds.end())
      return false;

    // Kinematics from the defining root particle. Copy by value: root() returns
    // a temporary Particle, so a reference to its momentum() would dangle.
    const auto p4 = branch.root().momentum();
    const double pt = p4.pt();
    if (pt < config_.ptMin || pt > config_.ptMax)
      return false;

    const double eta = p4.eta();
    const bool insideEta = eta >= config_.etaMin && eta <= config_.etaMax;
    if (config_.invertEta ? insideEta : !insideEta)
      return false;

    return true;
  }

}  // namespace truth